DeclObjC.h revision 9776ba0d844cf9f6888e871e3fd246ae782f76f4
1980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff//===--- DeclObjC.h - Classes for representing declarations -----*- C++ -*-===//
2980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff//
3980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff//                     The LLVM Compiler Infrastructure
4980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff//
50bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// This file is distributed under the University of Illinois Open Source
60bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// License. See LICENSE.TXT for details.
7980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff//
8980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff//===----------------------------------------------------------------------===//
9980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff//
10980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff//  This file defines the DeclObjC interface and subclasses.
11980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff//
12980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff//===----------------------------------------------------------------------===//
13980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
14980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff#ifndef LLVM_CLANG_AST_DECLOBJC_H
15980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff#define LLVM_CLANG_AST_DECLOBJC_H
16980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
17980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff#include "clang/AST/Decl.h"
18c7229c338c21ef26b01ef3ecf9eec4fd373fa9ecChris Lattner#include "clang/Basic/IdentifierTable.h"
19980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
20980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffnamespace clang {
21980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffclass Expr;
22980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffclass Stmt;
23980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffclass FunctionDecl;
24980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffclass AttributeList;
25a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCIvarDecl;
26a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCMethodDecl;
27a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCProtocolDecl;
28a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCCategoryDecl;
29a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCPropertyDecl;
30f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanianclass ObjCPropertyImplDecl;
313db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
323db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
333db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// ObjCList - This is a simple template class used to hold various lists of
343db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// decls etc, which is heavily used by the ObjC front-end.  This only use case
353db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// this supports is setting the list all at once and then reading elements out
363db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// of it.
373db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattnertemplate <typename T>
383db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattnerclass ObjCList {
393db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  /// List is a new[]'d array of pointers to objects that are not owned by this
403db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  /// list.
413db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  T **List;
423db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  unsigned NumElts;
433db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
443db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  void operator=(const ObjCList &); // DO NOT IMPLEMENT
453db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ObjCList(const ObjCList&);        // DO NOT IMPLEMENT
463db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattnerpublic:
473db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ObjCList() : List(0), NumElts(0) {}
483db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ~ObjCList() {
493db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    delete[] List;
503db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
513db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
523db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  void set(T* const* InList, unsigned Elts) {
533db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    assert(List == 0 && "Elements already set!");
543db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    List = new T*[Elts];
553db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    NumElts = Elts;
563db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    memcpy(List, InList, sizeof(T*)*Elts);
573db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
583db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
593db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  typedef T* const * iterator;
603db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  iterator begin() const { return List; }
613db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  iterator end() const { return List+NumElts; }
623db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
633db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  unsigned size() const { return NumElts; }
643db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  bool empty() const { return NumElts == 0; }
653db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
66780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  T* operator[](unsigned idx) const {
673db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    assert(idx < NumElts && "Invalid access");
683db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    return List[idx];
693db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
703db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner};
713db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
723db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
7358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
74a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCMethodDecl - Represents an instance or class method declaration.
7558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// ObjC methods can be declared within 4 contexts: class interfaces,
7658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// categories, protocols, and class implementations. While C++ member
7758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// functions leverage C syntax, Objective-C method syntax is modeled after
7858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Smalltalk (using colons to specify argument types/expressions).
7958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Here are some brief examples:
8058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
8158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Setter/getter instance methods:
8258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// - (void)setMenu:(NSMenu *)menu;
8358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// - (NSMenu *)menu;
8458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
8558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Instance method that takes 2 NSView arguments:
8658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// - (void)replaceSubview:(NSView *)oldView with:(NSView *)newView;
8758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
8858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Getter class method:
8958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// + (NSMenu *)defaultMenu;
9058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
9158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// A selector represents a unique name for a method. The selector names for
9258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// the above methods are setMenu:, menu, replaceSubview:with:, and defaultMenu.
9358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
94b048c9835969c4f7fe06264748be18ed4b442116Chris Lattnerclass ObjCMethodDecl : public Decl, public DeclContext {
9558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroffpublic:
9658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  enum ImplementationControl { None, Required, Optional };
9758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroffprivate:
9858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// Bitfields must be first fields in this class so they pack with those
9958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// declared in class Decl.
10058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// instance (true) or class (false) method.
10158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool IsInstance : 1;
10258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool IsVariadic : 1;
10358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
1044607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian  // Synthesized declaration method for a property setter/getter
1054607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian  bool IsSynthesized : 1;
1064607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian
107ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  // NOTE: VC++ treats enums as signed, avoid using ImplementationControl enum
10858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// @required/@optional
109ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned DeclImplementation : 2;
11058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
111ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  // NOTE: VC++ treats enums as signed, avoid using the ObjCDeclQualifier enum
11258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// in, inout, etc.
113ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned objcDeclQualifier : 6;
11458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
11558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Context this method is declared in.
11658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  NamedDecl *MethodContext;
11758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
11858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // A unigue name for this method.
11958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  Selector SelName;
12058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
12158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Type of this method.
12258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  QualType MethodDeclType;
12358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// ParamInfo - new[]'d array of pointers to VarDecls for the formal
12458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// parameters of this Method.  This is null if there are no formals.
12558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  ParmVarDecl **ParamInfo;
12658cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  unsigned NumMethodParams;
12758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
12858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// List of attributes for this method declaration.
12958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  SourceLocation EndLoc; // the location of the ';' or '{'.
13058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
13158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // The following are only used for method definitions, null otherwise.
13258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // FIXME: space savings opportunity, consider a sub-class.
13358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  Stmt *Body;
134451318c08a6342c10b8986060386fd9274418437Daniel Dunbar
135451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// SelfDecl - Decl for the implicit self parameter. This is lazily
136451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// constructed by createImplicitParams.
1374111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl *SelfDecl;
138451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// CmdDecl - Decl for the implicit _cmd parameter. This is lazily
139451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// constructed by createImplicitParams.
1404111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl *CmdDecl;
1416c4ae5de0c356777446f823b573821fb95560d91Chris Lattner
142a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl(SourceLocation beginLoc, SourceLocation endLoc,
14358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff                 Selector SelInfo, QualType T,
14458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff                 Decl *contextDecl,
145f6414927e67e27d9324d8d179c5f7ea620443924Daniel Dunbar                 bool isInstance = true,
14658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff                 bool isVariadic = false,
1474607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian                 bool isSynthesized = false,
14858cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner                 ImplementationControl impControl = None)
149a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  : Decl(ObjCMethod, beginLoc),
150b048c9835969c4f7fe06264748be18ed4b442116Chris Lattner    DeclContext(ObjCMethod),
15158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    IsInstance(isInstance), IsVariadic(isVariadic),
1524607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian    IsSynthesized(isSynthesized),
15358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    DeclImplementation(impControl), objcDeclQualifier(OBJC_TQ_None),
15458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    MethodContext(static_cast<NamedDecl*>(contextDecl)),
15558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    SelName(SelInfo), MethodDeclType(T),
156f6414927e67e27d9324d8d179c5f7ea620443924Daniel Dunbar    ParamInfo(0), NumMethodParams(0),
1574111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner    EndLoc(endLoc), Body(0), SelfDecl(0), CmdDecl(0) {}
1588a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
1591c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual ~ObjCMethodDecl();
1601c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
1616c4ae5de0c356777446f823b573821fb95560d91Chris Lattnerpublic:
1628a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
1638a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  /// Destroy - Call destructors and release memory.
1648a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  virtual void Destroy(ASTContext& C);
1656c4ae5de0c356777446f823b573821fb95560d91Chris Lattner
1660ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCMethodDecl *Create(ASTContext &C,
1670ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                SourceLocation beginLoc,
1686c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                SourceLocation endLoc, Selector SelInfo,
1696c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                QualType T, Decl *contextDecl,
170f6414927e67e27d9324d8d179c5f7ea620443924Daniel Dunbar                                bool isInstance = true,
1716c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                bool isVariadic = false,
1724607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian                                bool isSynthesized = false,
173b06fa3b86951b9f179c99c3768331536c32e902dChris Lattner                                ImplementationControl impControl = None);
17458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
175ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  ObjCDeclQualifier getObjCDeclQualifier() const {
176ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek    return ObjCDeclQualifier(objcDeclQualifier);
177ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  }
178a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setObjCDeclQualifier(ObjCDeclQualifier QV) { objcDeclQualifier = QV; }
17958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
18058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Location information, modeled after the Stmt API.
18158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  SourceLocation getLocStart() const { return getLocation(); }
18258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
1839776ba0d844cf9f6888e871e3fd246ae782f76f4Daniel Dunbar  SourceRange getSourceRange() const {
1849776ba0d844cf9f6888e871e3fd246ae782f76f4Daniel Dunbar    return SourceRange(getLocation(), EndLoc);
1859776ba0d844cf9f6888e871e3fd246ae782f76f4Daniel Dunbar  }
18658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
18758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  NamedDecl *getMethodContext() const { return MethodContext; }
18858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
1895619688510185081cbb4621d703daf7ee24cf39aChris Lattner  ObjCInterfaceDecl *getClassInterface();
1905619688510185081cbb4621d703daf7ee24cf39aChris Lattner  const ObjCInterfaceDecl *getClassInterface() const {
1915619688510185081cbb4621d703daf7ee24cf39aChris Lattner    return const_cast<ObjCMethodDecl*>(this)->getClassInterface();
192e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  }
19358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
19458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  Selector getSelector() const { return SelName; }
195faf5e779d16bb4590f2a97e1c7ded255eddd90f3Fariborz Jahanian  unsigned getSynthesizedMethodSize() const;
19658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  QualType getResultType() const { return MethodDeclType; }
19758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
198d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  // Iterator access to formal parameters.
19958cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  unsigned param_size() const { return NumMethodParams; }
200d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  typedef ParmVarDecl **param_iterator;
201d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  typedef ParmVarDecl * const *param_const_iterator;
202d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_iterator param_begin() { return ParamInfo; }
203d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_iterator param_end() { return ParamInfo+param_size(); }
204d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_const_iterator param_begin() const { return ParamInfo; }
205d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_const_iterator param_end() const { return ParamInfo+param_size(); }
206d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner
20758cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  unsigned getNumParams() const { return NumMethodParams; }
20858cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  ParmVarDecl *getParamDecl(unsigned i) const {
20958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    assert(i < getNumParams() && "Illegal param #");
21058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    return ParamInfo[i];
21158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
2122338d58a905191c4205d4f73affd0bdaa13493b4Fariborz Jahanian  void setParamDecl(int i, ParmVarDecl *pDecl) {
2132338d58a905191c4205d4f73affd0bdaa13493b4Fariborz Jahanian    ParamInfo[i] = pDecl;
2142338d58a905191c4205d4f73affd0bdaa13493b4Fariborz Jahanian  }
21558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  void setMethodParams(ParmVarDecl **NewParamInfo, unsigned NumParams);
2164111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner
217451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// createImplicitParams - Used to lazily create the self and cmd
218451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// implict parameters. This must be called prior to using getSelfDecl()
219451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// or getCmdDecl(). The call is ignored if the implicit paramters
220451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// have already been created.
221451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  void createImplicitParams(ASTContext &Context);
222451318c08a6342c10b8986060386fd9274418437Daniel Dunbar
2234111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl * getSelfDecl() const { return SelfDecl; }
2244111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl * getCmdDecl() const { return CmdDecl; }
22558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
22658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool isInstance() const { return IsInstance; }
22758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool isVariadic() const { return IsVariadic; }
22858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
2294607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian  bool isSynthesized() const { return IsSynthesized; }
2304607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian
23158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Related to protocols declared in  @protocol
23258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  void setDeclImplementation(ImplementationControl ic) {
23358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    DeclImplementation = ic;
23458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
23558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  ImplementationControl getImplementationControl() const {
236ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek    return ImplementationControl(DeclImplementation);
23758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
238792481eec23d8c1aa92173be589e2ae9d02514a5Ted Kremenek
23969c8f0aab655257e9e532d9d53756acf4f7a2d78Ted Kremenek  virtual Stmt *getBody() const { return Body; }
24058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  void setBody(Stmt *B) { Body = B; }
24158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
24258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Implement isa/cast/dyncast/etc.
243a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCMethod; }
244a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCMethodDecl *D) { return true; }
24558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff};
246b245a33e6f562377946dc9cb5694a2d3cb8a98a3Fariborz Jahanian
247a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCInterfaceDecl - Represents an ObjC class declaration. For example:
2480c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
2490c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   // MostPrimitive declares no super class (not particularly useful).
2500c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @interface MostPrimitive
2510c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     // no instance variables or methods.
2520c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @end
2530c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
254fd5de471478a507dd2495c4ea9bcab801ea5fa65Chris Lattner///   // NSResponder inherits from NSObject & implements NSCoding (a protocol).
2550c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @interface NSResponder : NSObject <NSCoding>
256a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek///   { // instance variables are represented by ObjCIvarDecl.
2570c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id nextResponder; // nextResponder instance variable.
2580c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   }
2590c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   - (NSResponder *)nextResponder; // return a pointer to NSResponder.
2600c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   - (void)mouseMoved:(NSEvent *)theEvent; // return void, takes a pointer
2610c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @end                                    // to an NSEvent.
2620c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
2630c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   Unlike C/C++, forward class declarations are accomplished with @class.
2640c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   Unlike C/C++, @class allows for a list of classes to be forward declared.
2650c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   Unlike C++, ObjC is a single-rooted class model. In Cocoa, classes
2660c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   typically inherit from NSObject (an exception is NSProxy).
2670c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
268b048c9835969c4f7fe06264748be18ed4b442116Chris Lattnerclass ObjCInterfaceDecl : public NamedDecl, public DeclContext {
2693110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  /// TypeForDecl - This indicates the Type object that represents this
2703110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  /// TypeDecl.  It is a cache maintained by ASTContext::getObjCInterfaceType
2713110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  Type *TypeForDecl;
2723110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  friend class ASTContext;
273980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
274980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Class's super class.
275a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *SuperClass;
276980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
277980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Protocols referenced in interface header declaration
2783db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ObjCList<ObjCProtocolDecl> ReferencedProtocols;
279980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
280980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Ivars/NumIvars - This is a new[]'d array of pointers to Decls.
281a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCIvarDecl **Ivars;   // Null if not defined.
282f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned NumIvars;      // 0 if none.
283980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
284980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// instance methods
285a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **InstanceMethods;  // Null if not defined
286b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner  unsigned NumInstanceMethods;  // 0 if none.
287980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
288980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// class methods
289a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **ClassMethods;  // Null if not defined
29062db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned NumClassMethods;  // 0 if none
291980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
292980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// List of categories defined for this class.
293a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl *CategoryList;
29482a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
29582a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  /// class properties
296a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCPropertyDecl **PropertyDecl;  // Null if no property
29755d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner  unsigned NumPropertyDecl;  // 0 if none.
298980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
2993a3ca1b35a7121aea0bf465a192dce748465e10fFariborz Jahanian  bool ForwardDecl:1; // declared with @class.
3003a3ca1b35a7121aea0bf465a192dce748465e10fFariborz Jahanian  bool InternalInterface:1; // true - no @interface for @implementation
30160fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff
302d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation ClassLoc; // location of the class identifier.
303d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation SuperClassLoc; // location of the super class identifier.
304f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation EndLoc; // marks the '>', '}', or identifier.
305f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation AtEndLoc; // marks the end of the entire interface.
3060e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner
307b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner  ObjCInterfaceDecl(SourceLocation atLoc, IdentifierInfo *Id,
308b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner                    SourceLocation CLoc, bool FD, bool isInternal)
309b048c9835969c4f7fe06264748be18ed4b442116Chris Lattner    : NamedDecl(ObjCInterface, atLoc, Id), DeclContext(ObjCInterface),
3100ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner      TypeForDecl(0), SuperClass(0),
3113db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner      Ivars(0), NumIvars(0),
312b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner      InstanceMethods(0), NumInstanceMethods(0),
31362db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner      ClassMethods(0), NumClassMethods(0),
314f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner      CategoryList(0), PropertyDecl(0), NumPropertyDecl(0),
315d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff      ForwardDecl(FD), InternalInterface(isInternal),
316d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff      ClassLoc(CLoc) {
317980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff      }
3188a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
3191c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual ~ObjCInterfaceDecl();
3208a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
3210e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattnerpublic:
3220e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner
3238a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  /// Destroy - Call destructors and release memory.
3248a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  virtual void Destroy(ASTContext& C);
3258a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
3260ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCInterfaceDecl *Create(ASTContext &C,
3270ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                   SourceLocation atLoc,
328d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff                                   IdentifierInfo *Id,
329d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff                                   SourceLocation ClassLoc = SourceLocation(),
3300e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner                                   bool ForwardDecl = false,
3310e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner                                   bool isInternal = false);
3323db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  const ObjCList<ObjCProtocolDecl> &getReferencedProtocols() const {
3337ed9e0f97f4645edc5d4670385b985ea4c617ce7Fariborz Jahanian    return ReferencedProtocols;
3347ed9e0f97f4645edc5d4670385b985ea4c617ce7Fariborz Jahanian  }
335980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
336559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCPropertyDecl *FindPropertyDeclaration(IdentifierInfo *PropertyId) const;
337559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCCategoryDecl *FindCategoryDeclaration(IdentifierInfo *CategoryId) const;
338559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCIvarDecl *FindIvarDeclaration(IdentifierInfo *IvarId) const;
3393db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
3403db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  typedef ObjCList<ObjCProtocolDecl>::iterator protocol_iterator;
3413db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  protocol_iterator protocol_begin() const {return ReferencedProtocols.begin();}
3423db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  protocol_iterator protocol_end() const { return ReferencedProtocols.end(); }
343aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian
344a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCIvarDecl * const *ivar_iterator;
345be6df088d69bca0e99c7845a6cd8c1ca85034f31Chris Lattner  ivar_iterator ivar_begin() const { return Ivars; }
3465564e07af5e62875f3b83fc2e5a8823588b5adeaChris Lattner  ivar_iterator ivar_end() const { return Ivars + ivar_size();}
347f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned ivar_size() const { return NumIvars; }
348a0fb5861dec7aa1da0d21d5759678d76b00464f4Ted Kremenek  bool ivar_empty() const { return NumIvars == 0; }
349be6df088d69bca0e99c7845a6cd8c1ca85034f31Chris Lattner
350b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner  unsigned getNumInstanceMethods() const { return NumInstanceMethods; }
35162db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned getNumClassMethods() const { return NumClassMethods; }
352980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
353a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * instmeth_iterator;
3544c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  instmeth_iterator instmeth_begin() const { return InstanceMethods; }
3554c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  instmeth_iterator instmeth_end() const {
356b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner    return InstanceMethods+NumInstanceMethods;
3574c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  }
3584c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner
359a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * classmeth_iterator;
3604c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  classmeth_iterator classmeth_begin() const { return ClassMethods; }
3614c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  classmeth_iterator classmeth_end() const {
36262db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return ClassMethods+NumClassMethods;
3634c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  }
364b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner
365b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner  /// addReferencedProtocols - Set the list of protocols that this interface
366b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner  /// implements.
367780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  void addReferencedProtocols(ObjCProtocolDecl *const*List, unsigned NumRPs) {
368780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    ReferencedProtocols.set(List, NumRPs);
3693db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
370b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner
371a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addInstanceVariablesToClass(ObjCIvarDecl **ivars, unsigned numIvars,
37260fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                                   SourceLocation RBracLoc);
373980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
374a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addMethods(ObjCMethodDecl **insMethods, unsigned numInsMembers,
375a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                  ObjCMethodDecl **clsMethods, unsigned numClsMembers,
37660fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                  SourceLocation AtEnd);
377980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
37855d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner  void addProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
37955d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner
380aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian  void mergeProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
381aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian
38233de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  void addPropertyMethods(ASTContext &Context,
38333de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian                          ObjCPropertyDecl* Property,
38433de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian                          llvm::SmallVector<ObjCMethodDecl*, 32> &insMethods);
38533de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
3863dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  typedef ObjCPropertyDecl * const * classprop_iterator;
3873dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_begin() const { return PropertyDecl; }
3883dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_end() const {
3893dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian    return PropertyDecl+NumPropertyDecl;
3903dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  }
39155d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner
392768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  bool isForwardDecl() const { return ForwardDecl; }
393768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  void setForwardDecl(bool val) { ForwardDecl = val; }
394980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
395a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *getSuperClass() const { return SuperClass; }
396a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setSuperClass(ObjCInterfaceDecl * superCls) { SuperClass = superCls; }
397980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
398a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl* getCategoryList() const { return CategoryList; }
399a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setCategoryList(ObjCCategoryDecl *category) {
40053efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    CategoryList = category;
401980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
40253efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner
40353efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  /// isSuperClassOf - Return true if this class is the specified class or is a
40453efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  /// super class of the specified interface class.
40553efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  bool isSuperClassOf(const ObjCInterfaceDecl *I) const {
40653efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    // If RHS is derived from LHS it is OK; else it is not OK.
40753efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    while (I != NULL) {
40853efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner      if (this == I)
40953efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner        return true;
41053efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner      I = I->getSuperClass();
41153efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    }
41253efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    return false;
41353efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  }
41453efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner
41568a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner  ObjCIvarDecl *lookupInstanceVariable(IdentifierInfo *IVarName,
41668a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner                                       ObjCInterfaceDecl *&ClassDeclared);
41768a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner  ObjCIvarDecl *lookupInstanceVariable(IdentifierInfo *IVarName) {
41868a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner    ObjCInterfaceDecl *ClassDeclared;
41968a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner    return lookupInstanceVariable(IVarName, ClassDeclared);
42068a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner  }
42168a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner
422c81c8144a661a49d7b9dae8d2080dee2e43186ecChris Lattner
42358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local instance method declared in this interface.
4243216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const {
42558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (instmeth_iterator I = instmeth_begin(), E = instmeth_end();
4260e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner         I != E; ++I) {
42758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
42858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
42958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
4300e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner    return 0;
43158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
43258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local class method declared in this interface.
4333216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const {
43458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
4350e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner         I != E; ++I) {
43658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
43758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
43858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
4390e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner    return 0;
44058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
44194a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Lookup a method. First, we search locally. If a method isn't
44294a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // found, we search referenced protocols and class categories.
443a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupInstanceMethod(Selector Sel);
444a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupClassMethod(Selector Sel);
44560fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff
446f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  // Location information, modeled after the Stmt API.
44760fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff  SourceLocation getLocStart() const { return getLocation(); } // '@'interface
448f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
449f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
45060fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff
451d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation getClassLoc() const { return ClassLoc; }
452d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  void setSuperClassLoc(SourceLocation Loc) { SuperClassLoc = Loc; }
453d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation getSuperClassLoc() const { return SuperClassLoc; }
454d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff
45560fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff  // We also need to record the @end location.
456f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation getAtEndLoc() const { return AtEndLoc; }
45782a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
458f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner  unsigned getNumPropertyDecl() const { return NumPropertyDecl; }
45982a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
46055d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner  ObjCPropertyDecl * const * getPropertyDecl() const { return PropertyDecl; }
461a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCPropertyDecl **getPropertyDecl() { return PropertyDecl; }
462f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff
4634b6df3fa953267c5d755628c8afd818bb571e579Fariborz Jahanian  /// ImplicitInterfaceDecl - check that this is an implicitely declared
464a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  /// ObjCInterfaceDecl node. This is for legacy objective-c @implementation
4654b6df3fa953267c5d755628c8afd818bb571e579Fariborz Jahanian  /// declaration without an @interface declaration.
4663a3ca1b35a7121aea0bf465a192dce748465e10fFariborz Jahanian  bool ImplicitInterfaceDecl() const { return InternalInterface; }
4674b6df3fa953267c5d755628c8afd818bb571e579Fariborz Jahanian
468a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCInterface; }
469a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCInterfaceDecl *D) { return true; }
470980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
471980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
472a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCIvarDecl - Represents an ObjC instance variable. In general, ObjC
4730c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// instance variables are identical to C. The only exception is Objective-C
4740c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// supports C++ style access control. For example:
4750c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
4760c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @interface IvarExample : NSObject
4770c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   {
478f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek///     id defaultToProtected;
4790c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @public:
4800c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id canBePublic; // same as C++.
4810c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @protected:
4820c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id canBeProtected; // same as C++.
4830c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @package:
4840c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id canBePackage; // framework visibility (not available in C++).
4850c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   }
4860c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
487a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCIvarDecl : public FieldDecl {
4880e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattnerpublic:
489980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  enum AccessControl {
490980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff    None, Private, Protected, Public, Package
491980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  };
492f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek
493b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenekprivate:
494b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek  ObjCIvarDecl(SourceLocation L, IdentifierInfo *Id, QualType T,
495b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek               AccessControl ac, Expr *BW)
4968f50b9c02abe7cd4945a9b77a8550a5207ea21cfTed Kremenek    : FieldDecl(ObjCIvar, L, Id, T, BW), DeclAccess(ac) {}
497b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek
498b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenekpublic:
499b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek  static ObjCIvarDecl *Create(ASTContext &C, SourceLocation L,
500b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek                              IdentifierInfo *Id, QualType T,
501b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek                              AccessControl ac, Expr *BW = NULL);
502b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek
503980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  void setAccessControl(AccessControl ac) { DeclAccess = ac; }
504f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek
505ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  AccessControl getAccessControl() const { return AccessControl(DeclAccess); }
506f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek
507f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek  AccessControl getCanonicalAccessControl() const {
508f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek    return DeclAccess == None ? Protected : AccessControl(DeclAccess);
509f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek  }
510980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
511980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  // Implement isa/cast/dyncast/etc.
512a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCIvar; }
513a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCIvarDecl *D) { return true; }
514980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffprivate:
515ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  // NOTE: VC++ treats enums as signed, avoid using the AccessControl enum
516ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned DeclAccess : 3;
517980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
518980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
51901e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
52001e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek/// ObjCAtDefsFieldDecl - Represents a field declaration created by an
52101e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek///  @defs(...).
52201e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenekclass ObjCAtDefsFieldDecl : public FieldDecl {
52301e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenekprivate:
52401e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  ObjCAtDefsFieldDecl(SourceLocation L, IdentifierInfo *Id,
52501e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek                      QualType T, Expr *BW)
52601e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  : FieldDecl(ObjCAtDefsField, L, Id, T, BW) {}
52701e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
52801e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenekpublic:
52901e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  static ObjCAtDefsFieldDecl *Create(ASTContext &C, SourceLocation L,
53001e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek                                     IdentifierInfo *Id, QualType T,
53101e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek                                     Expr *BW);
53201e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
53301e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  virtual void Destroy(ASTContext& C);
53401e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
53501e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  // Implement isa/cast/dyncast/etc.
53601e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCAtDefsField; }
53701e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  static bool classof(const ObjCAtDefsFieldDecl *D) { return true; }
53801e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek};
539980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
540a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCProtocolDecl - Represents a protocol declaration. ObjC protocols
5410c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// declare a pure abstract type (i.e no instance variables are permitted).
5420c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Protocols orginally drew inspiration from C++ pure virtual functions (a C++
5430c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// feature with nice semantics and lousy syntax:-). Here is an example:
5440c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
545eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// @protocol NSDraggingInfo <refproto1, refproto2>
5460c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - (NSWindow *)draggingDestinationWindow;
5470c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - (NSImage *)draggedImage;
5480c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
5490c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
550eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// This says that NSDraggingInfo requires two methods and requires everything
551eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// that the two "referenced protocols" 'refproto1' and 'refproto2' require as
552eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// well.
553eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner///
5540c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @interface ImplementsNSDraggingInfo : NSObject <NSDraggingInfo>
5550c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
5560c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
557a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjC protocols inspired Java interfaces. Unlike Java, ObjC classes and
5580c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// protocols are in distinct namespaces. For example, Cocoa defines both
5590c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// an NSObject protocol and class (which isn't allowed in Java). As a result,
5600c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// protocols are referenced using angle brackets as follows:
5610c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
5620c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// id <NSDraggingInfo> anyObjectThatImplementsNSDraggingInfo;
5630c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
564a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCProtocolDecl : public NamedDecl {
565780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  /// Referenced protocols
566780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  ObjCList<ObjCProtocolDecl> ReferencedProtocols;
567980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
568980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// protocol instance methods
569a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **InstanceMethods;  // Null if not defined
57062db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned NumInstanceMethods;  // 0 if none
571980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
572980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// protocol class methods
573a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **ClassMethods;  // Null if not defined
57462db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned NumClassMethods;  // 0 if none
5753dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
5763dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  /// protocol properties
5773dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  ObjCPropertyDecl **PropertyDecl;  // Null if no property
5783dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  unsigned NumPropertyDecl;  // 0 if none
5793dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
580980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  bool isForwardProtoDecl; // declared with @protocol.
581423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
582423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation EndLoc; // marks the '>' or identifier.
583423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation AtEndLoc; // marks the end of the entire interface.
584cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner
585780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  ObjCProtocolDecl(SourceLocation L, IdentifierInfo *Id)
586a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCProtocol, L, Id),
58762db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner      InstanceMethods(0), NumInstanceMethods(0),
58862db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner      ClassMethods(0), NumClassMethods(0),
5893dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian      PropertyDecl(0), NumPropertyDecl(0),
590c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner      isForwardProtoDecl(true) {
591cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner  }
5921c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
5931c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual ~ObjCProtocolDecl();
5941c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
595cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattnerpublic:
5961c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
5971c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  /// Destroy - Call destructors and release memory.
5981c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual void Destroy(ASTContext& C);
5991c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
600cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner  static ObjCProtocolDecl *Create(ASTContext &C, SourceLocation L,
601780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner                                  IdentifierInfo *Id);
602cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner
603a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addMethods(ObjCMethodDecl **insMethods, unsigned numInsMembers,
604a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                  ObjCMethodDecl **clsMethods, unsigned numClsMembers,
60560fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                  SourceLocation AtEndLoc);
606980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
607780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  const ObjCList<ObjCProtocolDecl> &getReferencedProtocols() const {
608780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    return ReferencedProtocols;
609980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
610dbc933701d20918add13b6a3c9d47ff8c75419cfDaniel Dunbar  typedef ObjCList<ObjCProtocolDecl>::iterator protocol_iterator;
611780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_begin() const {return ReferencedProtocols.begin();}
612780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_end() const { return ReferencedProtocols.end(); }
613780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
614780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  /// addReferencedProtocols - Set the list of protocols that this interface
615780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  /// implements.
616780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  void addReferencedProtocols(ObjCProtocolDecl *const*List, unsigned NumRPs) {
617780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    ReferencedProtocols.set(List, NumRPs);
618aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian  }
619aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian
620c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner  unsigned getNumInstanceMethods() const { return NumInstanceMethods; }
62162db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned getNumClassMethods() const { return NumClassMethods; }
6223d2c22b1d334fa74d26a5f21841cb55df5dfdd1aSteve Naroff
6233d2c22b1d334fa74d26a5f21841cb55df5dfdd1aSteve Naroff  ObjCPropertyDecl *FindPropertyDeclaration(IdentifierInfo *PropertyId) const;
6243dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6253dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  unsigned getNumPropertyDecl() const { return NumPropertyDecl; }
6263dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6273dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  ObjCPropertyDecl * const * getPropertyDecl() const { return PropertyDecl; }
6283dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  ObjCPropertyDecl **getPropertyDecl() { return PropertyDecl; }
6293dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6303dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  void addProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
6317b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar
6327b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar  void addPropertyMethods(ASTContext &Context,
6337b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar                          ObjCPropertyDecl* Property,
6347b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar                          llvm::SmallVector<ObjCMethodDecl*, 32> &insMethods);
6353dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6363dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  typedef ObjCPropertyDecl * const * classprop_iterator;
6373dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_begin() const { return PropertyDecl; }
6383dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_end() const {
6393dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian    return PropertyDecl+NumPropertyDecl;
6403dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  }
64158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
642a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * instmeth_iterator;
64358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_begin() const { return InstanceMethods; }
64458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_end() const {
64562db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return InstanceMethods+NumInstanceMethods;
64658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
64758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
648a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * classmeth_iterator;
64958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_begin() const { return ClassMethods; }
65058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_end() const {
65162db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return ClassMethods+NumClassMethods;
65258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
65358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
65458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local instance method declared in this interface.
6553216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const {
65662db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    for (instmeth_iterator I = instmeth_begin(), E = instmeth_end();
65762db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner         I != E; ++I) {
65858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
65958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
66058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
66162db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return 0;
66258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
66358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local class method declared in this interface.
6643216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const {
66558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
66662db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner         I != E; ++I) {
66758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
66858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
66958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
67062db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return 0;
67158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
672980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
67394a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Lookup a method. First, we search locally. If a method isn't
67494a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // found, we search referenced protocols and class categories.
675a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupInstanceMethod(Selector Sel);
676a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupClassMethod(Selector Sel);
6777dd82836dca87cf828ce994f161b53a34f6cdb7eFariborz Jahanian
678768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  bool isForwardDecl() const { return isForwardProtoDecl; }
679768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  void setForwardDecl(bool val) { isForwardProtoDecl = val; }
680980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
681423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // Location information, modeled after the Stmt API.
682423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocStart() const { return getLocation(); } // '@'protocol
683423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
684423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
685423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
686423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // We also need to record the @end location.
687423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getAtEndLoc() const { return AtEndLoc; }
688423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
689a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCProtocol; }
690a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCProtocolDecl *D) { return true; }
691980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
692980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
693a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCClassDecl - Specifies a list of forward class declarations. For example:
69406ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff///
69506ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff/// @class NSCursor, NSImage, NSPasteboard, NSWindow;
6960c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
697a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCClassDecl : public Decl {
698a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl **ForwardDecls;
6997e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner  unsigned NumForwardDecls;
70061f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
701a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCClassDecl(SourceLocation L, ObjCInterfaceDecl **Elts, unsigned nElts)
702a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : Decl(ObjCClass, L) {
70306ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    if (nElts) {
704a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      ForwardDecls = new ObjCInterfaceDecl*[nElts];
705a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      memcpy(ForwardDecls, Elts, nElts*sizeof(ObjCInterfaceDecl*));
7067e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner    } else {
7077e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner      ForwardDecls = 0;
70806ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    }
70906ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    NumForwardDecls = nElts;
71006ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
711400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
712400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  virtual ~ObjCClassDecl();
713400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
71461f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattnerpublic:
715400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
716400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  /// Destroy - Call destructors and release memory.
717400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  virtual void Destroy(ASTContext& C);
718400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
71961f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner  static ObjCClassDecl *Create(ASTContext &C, SourceLocation L,
72061f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner                               ObjCInterfaceDecl **Elts, unsigned nElts);
72161f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
722a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setInterfaceDecl(unsigned idx, ObjCInterfaceDecl *OID) {
7237e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner    assert(idx < NumForwardDecls && "index out of range");
72406ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    ForwardDecls[idx] = OID;
72506ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
726a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl** getForwardDecls() const { return ForwardDecls; }
7277e15891fc89256fc013bd1003676ad3197b85c25Steve Naroff  int getNumForwardDecls() const { return NumForwardDecls; }
7287e15891fc89256fc013bd1003676ad3197b85c25Steve Naroff
729400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  typedef ObjCInterfaceDecl * const * iterator;
730400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  iterator begin() const { return ForwardDecls; }
731400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  iterator end() const { return ForwardDecls+NumForwardDecls; }
732400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
733a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCClass; }
734a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCClassDecl *D) { return true; }
73506ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff};
73606ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff
737a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCForwardProtocolDecl - Specifies a list of forward protocol declarations.
73806ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff/// For example:
73906ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff///
7400c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @protocol NSTextInput, NSChangeSpelling, NSDraggingInfo;
7410c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
742a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCForwardProtocolDecl : public Decl {
743a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCProtocolDecl **ReferencedProtocols;
7449fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  unsigned NumReferencedProtocols;
74561f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
7460ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  ObjCForwardProtocolDecl(SourceLocation L,
747a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                          ObjCProtocolDecl **Elts, unsigned nElts)
748a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  : Decl(ObjCForwardProtocol, L) {
749b97de3eddf0a0f7ff97204e7def6b7e0aa8953f2Chris Lattner    NumReferencedProtocols = nElts;
75006ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    if (nElts) {
751a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      ReferencedProtocols = new ObjCProtocolDecl*[nElts];
752a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      memcpy(ReferencedProtocols, Elts, nElts*sizeof(ObjCProtocolDecl*));
7539fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    } else {
7549fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner      ReferencedProtocols = 0;
755980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff    }
75606ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
75705ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek
75805ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  virtual ~ObjCForwardProtocolDecl();
75905ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek
76061f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattnerpublic:
76161f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner  static ObjCForwardProtocolDecl *Create(ASTContext &C, SourceLocation L,
76261f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner                                         ObjCProtocolDecl **Elts, unsigned Num);
76361f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
76461f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
765a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setForwardProtocolDecl(unsigned idx, ObjCProtocolDecl *OID) {
7669fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    assert(idx < NumReferencedProtocols && "index out of range");
7677ed9e0f97f4645edc5d4670385b985ea4c617ce7Fariborz Jahanian    ReferencedProtocols[idx] = OID;
76806ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
7699fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner
7709fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  unsigned getNumForwardDecls() const { return NumReferencedProtocols; }
7719fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner
772a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCProtocolDecl *getForwardProtocolDecl(unsigned idx) {
7739fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    assert(idx < NumReferencedProtocols && "index out of range");
7749fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    return ReferencedProtocols[idx];
7759fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  }
776a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  const ObjCProtocolDecl *getForwardProtocolDecl(unsigned idx) const {
7779fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    assert(idx < NumReferencedProtocols && "index out of range");
7789fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    return ReferencedProtocols[idx];
7799fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  }
7809fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner
78105ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  typedef ObjCProtocolDecl * const * iterator;
78205ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  iterator begin() const { return ReferencedProtocols; }
78305ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  iterator end() const { return ReferencedProtocols+NumReferencedProtocols; }
78405ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek
78506ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  static bool classof(const Decl *D) {
786a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    return D->getKind() == ObjCForwardProtocol;
78706ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
788a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCForwardProtocolDecl *D) { return true; }
789980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
790980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
791a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCCategoryDecl - Represents a category declaration. A category allows
7920c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// you to add methods to an existing class (without subclassing or modifying
7930c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// the original class interface or implementation:-). Categories don't allow
7940c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// you to add instance data. The following example adds "myMethod" to all
7950c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// NSView's within a process:
7960c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
7970c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @interface NSView (MyViewMethods)
7980c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - myMethod;
7990c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
8000c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
8010c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Cateogries also allow you to split the implementation of a class across
8020c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// several files (a feature more naturally supported in C++).
8030c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
8040c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Categories were originally inspired by dynamic languages such as Common
80568c82cf61228102aba1194efef222fa1478af2a8Chris Lattner/// Lisp and Smalltalk.  More traditional class-based languages (C++, Java)
8060c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// don't support this level of dynamism, which is both powerful and dangerous.
8070c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
808a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCCategoryDecl : public NamedDecl {
809980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Interface belonging to this category
810a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *ClassInterface;
811980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
81268c82cf61228102aba1194efef222fa1478af2a8Chris Lattner  /// referenced protocols in this category.
813780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  ObjCList<ObjCProtocolDecl> ReferencedProtocols;
814980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
815980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// category instance methods
816a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **InstanceMethods;  // Null if not defined
817c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner  unsigned NumInstanceMethods;  // 0 if none
818980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
819980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// category class methods
820a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **ClassMethods;  // Null if not defined
821a906135721c350435319347d2672bbb3bf494f91Chris Lattner  unsigned NumClassMethods;  // 0 if not defined
822980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
823980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Next category belonging to this class
824a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl *NextClassCategory;
825ca3adf7e8cac8c9fbaf592b1e5c2be6f082de7baFariborz Jahanian
8267e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  /// category properties
8277e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  ObjCPropertyDecl **PropertyDecl;  // Null if no property
8287e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  unsigned NumPropertyDecl;  // 0 if none
8297e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian
830423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation EndLoc; // marks the '>' or identifier.
831423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation AtEndLoc; // marks the end of the entire interface.
83261f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
83368c82cf61228102aba1194efef222fa1478af2a8Chris Lattner  ObjCCategoryDecl(SourceLocation L, IdentifierInfo *Id)
834a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCCategory, L, Id),
835780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner      ClassInterface(0),
836c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner      InstanceMethods(0), NumInstanceMethods(0),
837a906135721c350435319347d2672bbb3bf494f91Chris Lattner      ClassMethods(0), NumClassMethods(0),
8387e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian      NextClassCategory(0), PropertyDecl(0),  NumPropertyDecl(0) {
839a906135721c350435319347d2672bbb3bf494f91Chris Lattner  }
84061f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattnerpublic:
84161f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
8420ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCCategoryDecl *Create(ASTContext &C,
8430ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                  SourceLocation L, IdentifierInfo *Id);
84461f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
845e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
846e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
847a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setClassInterface(ObjCInterfaceDecl *IDecl) { ClassInterface = IDecl; }
848980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
849f7b2c98c16dfb2261ea57d40a1d5bc4738e73175Chris Lattner  /// addReferencedProtocols - Set the list of protocols that this interface
850f7b2c98c16dfb2261ea57d40a1d5bc4738e73175Chris Lattner  /// implements.
851780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  void addReferencedProtocols(ObjCProtocolDecl *const*List, unsigned NumRPs) {
852780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    ReferencedProtocols.set(List, NumRPs);
853780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  }
854980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
855780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  const ObjCList<ObjCProtocolDecl> &getReferencedProtocols() const {
856780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    return ReferencedProtocols;
8578f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian  }
858780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
859780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  typedef ObjCProtocolDecl * const * protocol_iterator;
860780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_begin() const {return ReferencedProtocols.begin();}
861780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_end() const { return ReferencedProtocols.end(); }
862780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
863780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
864c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner  unsigned getNumInstanceMethods() const { return NumInstanceMethods; }
865a906135721c350435319347d2672bbb3bf494f91Chris Lattner  unsigned getNumClassMethods() const { return NumClassMethods; }
86658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
8677e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  unsigned getNumPropertyDecl() const { return NumPropertyDecl; }
8687e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian
8697e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  ObjCPropertyDecl * const * getPropertyDecl() const { return PropertyDecl; }
8707e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian
8713dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  void addProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
8727b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar
8737b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar  void addPropertyMethods(ASTContext &Context,
8747b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar                          ObjCPropertyDecl* Property,
8757b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar                          llvm::SmallVector<ObjCMethodDecl*, 32> &insMethods);
8763dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
877559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCPropertyDecl *FindPropertyDeclaration(IdentifierInfo *PropertyId) const;
878559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian
8793dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  typedef ObjCPropertyDecl * const * classprop_iterator;
8803dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_begin() const { return PropertyDecl; }
8813dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_end() const {
8823dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian    return PropertyDecl+NumPropertyDecl;
8833dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  }
8843dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
885a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * instmeth_iterator;
88658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_begin() const { return InstanceMethods; }
88758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_end() const {
888c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner    return InstanceMethods+NumInstanceMethods;
88958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
89058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
891a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * classmeth_iterator;
89258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_begin() const { return ClassMethods; }
89358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_end() const {
894a906135721c350435319347d2672bbb3bf494f91Chris Lattner    return ClassMethods+NumClassMethods;
89558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
89658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
89758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local instance method declared in this interface.
8983216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const {
89958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (instmeth_iterator I = instmeth_begin(), E = instmeth_end();
90068c82cf61228102aba1194efef222fa1478af2a8Chris Lattner         I != E; ++I) {
90158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
90258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
90358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
90468c82cf61228102aba1194efef222fa1478af2a8Chris Lattner    return 0;
90558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
90658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local class method declared in this interface.
9073216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const {
90858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
90968c82cf61228102aba1194efef222fa1478af2a8Chris Lattner         I != E; ++I) {
91058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
91158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
91258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
91368c82cf61228102aba1194efef222fa1478af2a8Chris Lattner    return 0;
91458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
9158f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
916a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addMethods(ObjCMethodDecl **insMethods, unsigned numInsMembers,
917a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                  ObjCMethodDecl **clsMethods, unsigned numClsMembers,
91860fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                  SourceLocation AtEndLoc);
919980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
920a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl *getNextClassCategory() const { return NextClassCategory; }
921980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  void insertNextClassCategory() {
9223d58138992b9bc7b34aaa680f3ddf3971292eb7dSteve Naroff    NextClassCategory = ClassInterface->getCategoryList();
9233d58138992b9bc7b34aaa680f3ddf3971292eb7dSteve Naroff    ClassInterface->setCategoryList(this);
924980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
925423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // Location information, modeled after the Stmt API.
926423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocStart() const { return getLocation(); } // '@'interface
927423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
928423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
929423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
930423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // We also need to record the @end location.
931423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getAtEndLoc() const { return AtEndLoc; }
932ca3adf7e8cac8c9fbaf592b1e5c2be6f082de7baFariborz Jahanian
933a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCCategory; }
934a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCCategoryDecl *D) { return true; }
935980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
9360c6b6243d3efd958c17943130e2a773653511edcSteve Naroff
937a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCCategoryImplDecl - An object of this class encapsulates a category
938559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @implementation declaration. If a category class has declaration of a
939559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// property, its implementation must be specified in the category's
940559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @implementation declaration. Example:
941559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @interface I @end
942559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @interface I(CATEGORY)
943559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian///    @property int p1, d1;
944559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @end
945559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @implementation I(CATEGORY)
946559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian///  @dynamic p1,d1;
947559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @end
948559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian///
949a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCCategoryImplDecl : public NamedDecl {
9508f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian  /// Class interface for this category implementation
951a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *ClassInterface;
9528f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
953e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  /// implemented instance methods
954a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> InstanceMethods;
9558f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
956e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  /// implemented class methods
957a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> ClassMethods;
958f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
959559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  /// Property Implementations in this category
960f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  llvm::SmallVector<ObjCPropertyImplDecl*, 8> PropertyImplementations;
961e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
962e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  SourceLocation EndLoc;
96375c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner
964a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryImplDecl(SourceLocation L, IdentifierInfo *Id,
965a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                       ObjCInterfaceDecl *classInterface)
966a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCCategoryImpl, L, Id), ClassInterface(classInterface) {}
96775c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattnerpublic:
9680ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCCategoryImplDecl *Create(ASTContext &C,
9690ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                      SourceLocation L, IdentifierInfo *Id,
97075c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner                                      ObjCInterfaceDecl *classInterface);
9718f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
972e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
973e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
9748f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
975ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  unsigned getNumInstanceMethods() const { return InstanceMethods.size(); }
976ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  unsigned getNumClassMethods() const { return ClassMethods.size(); }
977e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
978a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addInstanceMethod(ObjCMethodDecl *method) {
979e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff    InstanceMethods.push_back(method);
980e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  }
981a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addClassMethod(ObjCMethodDecl *method) {
982e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff    ClassMethods.push_back(method);
983f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  }
98494a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the instance method definition for this implementation.
9853216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const;
98694a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff
98794a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the class method definition for this implementation.
9883216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const;
989f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
990f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  void addPropertyImplementation(ObjCPropertyImplDecl *property) {
991f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian    PropertyImplementations.push_back(property);
992f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  }
993e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
994f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  unsigned getNumPropertyImplementations() const
995f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  { return PropertyImplementations.size(); }
996f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
997559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian
998559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  typedef llvm::SmallVector<ObjCPropertyImplDecl*, 8>::const_iterator
999559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    propimpl_iterator;
1000559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_begin() const {
1001559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.begin();
1002559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1003559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_end() const {
1004559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.end();
1005559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1006559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian
1007a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
1008ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner    instmeth_iterator;
1009ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  instmeth_iterator instmeth_begin() const { return InstanceMethods.begin(); }
1010ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  instmeth_iterator instmeth_end() const { return InstanceMethods.end(); }
1011ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner
1012a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
1013ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner    classmeth_iterator;
1014ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  classmeth_iterator classmeth_begin() const { return ClassMethods.begin(); }
1015ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  classmeth_iterator classmeth_end() const { return ClassMethods.end(); }
1016ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner
1017ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner
1018e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  // Location information, modeled after the Stmt API.
1019e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  SourceLocation getLocStart() const { return getLocation(); }
1020e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
1021e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
1022e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
1023a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCCategoryImpl;}
1024a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCCategoryImplDecl *D) { return true; }
10258f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian};
10268f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
1027a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCImplementationDecl - Represents a class definition - this is where
10280c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// method definitions are specified. For example:
10290c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
10300c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @implementation MyClass
10310c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - (void)myMethod { /* do something */ }
10320c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
10330c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
10340c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Typically, instance variables are specified in the class interface,
10350c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// *not* in the implemenentation. Nevertheless (for legacy reasons), we
10360c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// allow instance variables to be specified in the implementation. When
10370c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// specified, they need to be *identical* to the interface. Now that we
10380c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// have support for non-fragile ivars in ObjC 2.0, we can consider removing
10390c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// the legacy semantics and allow developers to move private ivar declarations
10400c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// from the class interface to the class implementation (but I digress:-)
10410c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
1042a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCImplementationDecl : public NamedDecl {
1043dae1a1a2aa4f245b1958dc8db6089f24c575ef13Fariborz Jahanian  /// Class interface for this implementation
1044a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *ClassInterface;
1045f4d331dd922f92478ebf30e808c0ca97ce49418bFariborz Jahanian
1046980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Implementation Class's super class.
1047a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *SuperClass;
1048980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1049980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Optional Ivars/NumIvars - This is a new[]'d array of pointers to Decls.
1050a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCIvarDecl **Ivars;   // Null if not specified
1051f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned NumIvars;      // 0 if none.
10520416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1053980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// implemented instance methods
1054a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> InstanceMethods;
10550416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1056980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// implemented class methods
1057a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> ClassMethods;
10580416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1059f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  /// Propertys' being implemented
1060f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  llvm::SmallVector<ObjCPropertyImplDecl*, 8> PropertyImplementations;
1061f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
10620416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  SourceLocation EndLoc;
106375c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner
1064a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCImplementationDecl(SourceLocation L, IdentifierInfo *Id,
1065a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                         ObjCInterfaceDecl *classInterface,
1066a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                         ObjCInterfaceDecl *superDecl)
1067a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCImplementation, L, Id),
10680416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff      ClassInterface(classInterface), SuperClass(superDecl),
1069f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner      Ivars(0), NumIvars(0) {}
107075c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattnerpublic:
10710ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCImplementationDecl *Create(ASTContext &C,
10720ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                        SourceLocation L, IdentifierInfo *Id,
107375c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner                                        ObjCInterfaceDecl *classInterface,
107475c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner                                        ObjCInterfaceDecl *superDecl);
107575c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner
1076980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1077a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void ObjCAddInstanceVariablesToClassImpl(ObjCIvarDecl **ivars,
1078980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff                                           unsigned numIvars);
1079980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1080a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addInstanceMethod(ObjCMethodDecl *method) {
10810416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff    InstanceMethods.push_back(method);
10820416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  }
1083a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addClassMethod(ObjCMethodDecl *method) {
10840416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff    ClassMethods.push_back(method);
10850416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  }
1086f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
1087f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  void addPropertyImplementation(ObjCPropertyImplDecl *property) {
1088f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian    PropertyImplementations.push_back(property);
1089f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  }
1090559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  typedef llvm::SmallVector<ObjCPropertyImplDecl*, 8>::const_iterator
1091559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator;
1092559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_begin() const {
1093559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.begin();
1094559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1095559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_end() const {
1096559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.end();
1097559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1098f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
10990416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  // Location information, modeled after the Stmt API.
11000416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  SourceLocation getLocStart() const { return getLocation(); }
11010416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
11020416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
11030416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1104e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
1105e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
1106e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getSuperClass() const { return SuperClass; }
1107e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getSuperClass() { return SuperClass; }
1108980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1109f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  void setSuperClass(ObjCInterfaceDecl * superCls) { SuperClass = superCls; }
1110980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1111b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner  unsigned getNumInstanceMethods() const { return InstanceMethods.size(); }
111262db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned getNumClassMethods() const { return ClassMethods.size(); }
1113f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
1114f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  unsigned getNumPropertyImplementations() const
1115f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian    { return PropertyImplementations.size(); }
1116c43d868355374d48296ad3be2c9c536698a5e9a8Steve Naroff
1117a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
11180157c5144513438bb74aebf50d18f95df4104acbChris Lattner       instmeth_iterator;
11190157c5144513438bb74aebf50d18f95df4104acbChris Lattner  instmeth_iterator instmeth_begin() const { return InstanceMethods.begin(); }
11200157c5144513438bb74aebf50d18f95df4104acbChris Lattner  instmeth_iterator instmeth_end() const { return InstanceMethods.end(); }
11210157c5144513438bb74aebf50d18f95df4104acbChris Lattner
1122a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
11230157c5144513438bb74aebf50d18f95df4104acbChris Lattner    classmeth_iterator;
11240157c5144513438bb74aebf50d18f95df4104acbChris Lattner  classmeth_iterator classmeth_begin() const { return ClassMethods.begin(); }
11250157c5144513438bb74aebf50d18f95df4104acbChris Lattner  classmeth_iterator classmeth_end() const { return ClassMethods.end(); }
11260157c5144513438bb74aebf50d18f95df4104acbChris Lattner
112794a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the instance method definition for this implementation.
11283216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const;
11290157c5144513438bb74aebf50d18f95df4104acbChris Lattner
113094a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the class method definition for this implementation.
11313216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const;
11320157c5144513438bb74aebf50d18f95df4104acbChris Lattner
1133a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCIvarDecl * const *ivar_iterator;
11340157c5144513438bb74aebf50d18f95df4104acbChris Lattner  ivar_iterator ivar_begin() const { return Ivars; }
1135f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  ivar_iterator ivar_end() const { return Ivars+NumIvars; }
1136f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned ivar_size() const { return NumIvars; }
1137f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  bool ivar_empty() const { return NumIvars == 0; }
11380157c5144513438bb74aebf50d18f95df4104acbChris Lattner
1139980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  static bool classof(const Decl *D) {
1140a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    return D->getKind() == ObjCImplementation;
1141980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
1142a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCImplementationDecl *D) { return true; }
1143980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
1144243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian
1145a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCCompatibleAliasDecl - Represents alias of a class. This alias is
1146243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian/// declared as @compatibility_alias alias class.
1147e8043c39176e7f253fbd92982b077eca6bf2fd59Steve Naroffclass ObjCCompatibleAliasDecl : public NamedDecl {
1148243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian  /// Class that this is an alias of.
1149a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *AliasedClass;
1150243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian
1151a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCompatibleAliasDecl(SourceLocation L, IdentifierInfo *Id,
1152e8043c39176e7f253fbd92982b077eca6bf2fd59Steve Naroff                          ObjCInterfaceDecl* aliasedClass)
1153e8043c39176e7f253fbd92982b077eca6bf2fd59Steve Naroff    : NamedDecl(ObjCCompatibleAlias, L, Id), AliasedClass(aliasedClass) {}
1154f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattnerpublic:
11550ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCCompatibleAliasDecl *Create(ASTContext &C,
11560ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                         SourceLocation L, IdentifierInfo *Id,
1157f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner                                         ObjCInterfaceDecl* aliasedClass);
1158f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner
1159f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return AliasedClass; }
1160f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner  ObjCInterfaceDecl *getClassInterface() { return AliasedClass; }
1161980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1162243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian  static bool classof(const Decl *D) {
11638a934233d1582b5bde9d270bc0705aa81e471a79Chris Lattner    return D->getKind() == ObjCCompatibleAlias;
1164243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian  }
1165a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCCompatibleAliasDecl *D) { return true; }
1166243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian
1167243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian};
11681de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian
11691de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian/// ObjCPropertyDecl - Represents one property declaration in an interface.
11701de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian/// For example:
11711de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian/// @property (assign, readwrite) int MyProperty;
11721de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian///
11731de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanianclass ObjCPropertyDecl : public NamedDecl {
117482a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanianpublic:
1175a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner  enum PropertyAttributeKind {
1176a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_noattr    = 0x00,
1177a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_readonly  = 0x01,
1178a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_getter    = 0x02,
1179a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_assign    = 0x04,
1180a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_readwrite = 0x08,
1181a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_retain    = 0x10,
1182a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_copy      = 0x20,
1183a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_nonatomic = 0x40,
1184a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_setter    = 0x80
1185a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner  };
1186af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
1187af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  enum SetterKind { Assign, Retain, Copy };
118846b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  enum PropertyControl { None, Required, Optional };
118982a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanianprivate:
1190dae1a1a2aa4f245b1958dc8db6089f24c575ef13Fariborz Jahanian  QualType DeclType;
1191ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned PropertyAttributes : 8;
119282a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
119346b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  // @required/@optional
119446b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  unsigned PropertyImplementation : 2;
119546b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian
11965251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector GetterName;    // getter name of NULL if no getter
11975251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector SetterName;    // setter name of NULL if no setter
119882a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
119933de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *GetterMethodDecl; // Declaration of getter instance method
120033de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *SetterMethodDecl; // Declaration of setter instance method
120133de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
12021de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian  ObjCPropertyDecl(SourceLocation L, IdentifierInfo *Id, QualType T)
12031de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian    : NamedDecl(ObjCProperty, L, Id), DeclType(T),
120433de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      PropertyAttributes(OBJC_PR_noattr), PropertyImplementation(None),
120533de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      GetterName(Selector()),
120633de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      SetterName(Selector()),
120733de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      GetterMethodDecl(0), SetterMethodDecl(0) {}
1208f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattnerpublic:
12091de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian  static ObjCPropertyDecl *Create(ASTContext &C, SourceLocation L,
121046b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian                                  IdentifierInfo *Id, QualType T,
121146b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian                                  PropertyControl propControl = None);
1212dae1a1a2aa4f245b1958dc8db6089f24c575ef13Fariborz Jahanian  QualType getType() const { return DeclType; }
1213c35b9e4e2efad727538c848cf30d4b0eb1031dc9Fariborz Jahanian
1214a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner  PropertyAttributeKind getPropertyAttributes() const {
1215f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner    return PropertyAttributeKind(PropertyAttributes);
1216f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner  }
1217564576b225386cbff375351597dd5e2a92872d38Fariborz Jahanian  void setPropertyAttributes(PropertyAttributeKind PRVal) {
1218a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    PropertyAttributes |= PRVal;
121982a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  }
1220394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar
1221af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  // Helper methods for accessing attributes.
1222af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
1223af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// isReadOnly - Return true iff the property has a setter.
1224394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar  bool isReadOnly() const {
1225394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar    return (PropertyAttributes & OBJC_PR_readonly);
1226394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar  }
1227af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
1228af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// getSetterKind - Return the method used for doing assignment in
1229af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// the property setter. This is only valid if the property has been
1230af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// defined to have a setter.
1231af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  SetterKind getSetterKind() const {
1232af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar    if (PropertyAttributes & OBJC_PR_retain)
1233af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar      return Retain;
1234af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar    if (PropertyAttributes & OBJC_PR_copy)
1235af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar      return Copy;
1236af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar    return Assign;
1237af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  }
1238af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
12395251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector getGetterName() const { return GetterName; }
12405251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  void setGetterName(Selector Sel) { GetterName = Sel; }
124182a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
12425251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector getSetterName() const { return SetterName; }
12435251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  void setSetterName(Selector Sel) { SetterName = Sel; }
124482a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
124533de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *getGetterMethodDecl() const { return GetterMethodDecl; }
124633de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  void setGetterMethodDecl(ObjCMethodDecl *gDecl) { GetterMethodDecl = gDecl; }
124733de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
124833de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *getSetterMethodDecl() const { return SetterMethodDecl; }
124933de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  void setSetterMethodDecl(ObjCMethodDecl *gDecl) { SetterMethodDecl = gDecl; }
125033de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
125146b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  // Related to @optional/@required declared in @protocol
125246b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  void setPropertyImplementation(PropertyControl pc) {
125346b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian    PropertyImplementation = pc;
125446b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  }
125546b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  PropertyControl getPropertyImplementation() const {
125646b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian    return PropertyControl(PropertyImplementation);
125746b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  }
125846b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian
125982a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  static bool classof(const Decl *D) {
1260670aa9d7639278f507930e95dc89c12032ab7c7eSam Bishop    return D->getKind() == ObjCProperty;
126182a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  }
1262a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCPropertyDecl *D) { return true; }
126382a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian};
1264980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
126561d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian/// ObjCPropertyImplDecl - Represents implementation declaration of a property
126661d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian/// in a class or category implementation block. For example:
126761d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian/// @synthesize prop1 = ivar1;
126861d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian///
126961d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanianclass ObjCPropertyImplDecl : public Decl {
127061d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanianpublic:
12719f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar  enum Kind {
12729f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    Synthesize,
12739f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    Dynamic
127461d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  };
127561d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanianprivate:
1276559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  SourceLocation AtLoc;   // location of @synthesize or @dynamic
127761d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  /// Property declaration being implemented
127861d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  ObjCPropertyDecl *PropertyDecl;
1279be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek
128061d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  /// Null for @dynamic. Required for @synthesize.
128161d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  ObjCIvarDecl *PropertyIvarDecl;
1282be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek
1283628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian  ObjCPropertyImplDecl(SourceLocation atLoc, SourceLocation L,
1284628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                       ObjCPropertyDecl *property,
12859f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar                       Kind PK,
1286628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                       ObjCIvarDecl *ivarDecl)
12879f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    : Decl(ObjCPropertyImpl, L), AtLoc(atLoc), PropertyDecl(property),
12889f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar      PropertyIvarDecl(ivarDecl) {
12899f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    assert (PK == Dynamic || PropertyIvarDecl);
12909f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar  }
1291628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian
12929f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbarpublic:
1293628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian  static ObjCPropertyImplDecl *Create(ASTContext &C, SourceLocation atLoc,
1294628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                      SourceLocation L,
1295628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                      ObjCPropertyDecl *property,
12969f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar                                      Kind PK,
1297628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                      ObjCIvarDecl *ivarDecl);
129861d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
1299be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  ObjCPropertyDecl *getPropertyDecl() const {
1300be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek    return PropertyDecl;
1301be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  }
130261d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
13039f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar  Kind getPropertyImplementation() const {
13049f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    return PropertyIvarDecl ? Synthesize : Dynamic;
1305be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  }
130661d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
1307af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  ObjCIvarDecl *getPropertyIvarDecl() const {
1308be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek    return PropertyIvarDecl;
1309be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  }
131061d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
131161d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  static bool classof(const Decl *D) {
131261d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian    return D->getKind() == ObjCPropertyImpl;
131361d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  }
131461d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  static bool classof(const ObjCPropertyImplDecl *D) { return true; }
131561d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian};
131661d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
1317980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff}  // end namespace clang
1318980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff#endif
1319