DeclObjC.h revision a769c004a2874504c17ea8afccbc4ad35fc33c9f
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"
19b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian#include "llvm/ADT/DenseMap.h"
20980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
21980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffnamespace clang {
22980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffclass Expr;
23980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffclass Stmt;
24980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffclass FunctionDecl;
25980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffclass AttributeList;
2660f8c868ffb346b78451a3eccaecd0461d2ae498Fariborz Jahanianclass RecordDecl;
27a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCIvarDecl;
28a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCMethodDecl;
29a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCProtocolDecl;
30a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCCategoryDecl;
31a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCPropertyDecl;
32f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanianclass ObjCPropertyImplDecl;
333db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
343db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
353db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// ObjCList - This is a simple template class used to hold various lists of
363db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// decls etc, which is heavily used by the ObjC front-end.  This only use case
373db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// this supports is setting the list all at once and then reading elements out
383db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// of it.
393db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattnertemplate <typename T>
403db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattnerclass ObjCList {
413db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  /// List is a new[]'d array of pointers to objects that are not owned by this
423db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  /// list.
433db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  T **List;
443db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  unsigned NumElts;
453db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
463db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  void operator=(const ObjCList &); // DO NOT IMPLEMENT
473db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ObjCList(const ObjCList&);        // DO NOT IMPLEMENT
483db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattnerpublic:
493db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ObjCList() : List(0), NumElts(0) {}
503db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ~ObjCList() {
513db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    delete[] List;
523db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
533db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
543db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  void set(T* const* InList, unsigned Elts) {
553db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    assert(List == 0 && "Elements already set!");
563db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    List = new T*[Elts];
573db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    NumElts = Elts;
583db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    memcpy(List, InList, sizeof(T*)*Elts);
593db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
603db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
613db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  typedef T* const * iterator;
623db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  iterator begin() const { return List; }
633db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  iterator end() const { return List+NumElts; }
643db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
653db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  unsigned size() const { return NumElts; }
663db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  bool empty() const { return NumElts == 0; }
673db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
68780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  T* operator[](unsigned idx) const {
693db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    assert(idx < NumElts && "Invalid access");
703db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    return List[idx];
713db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
723db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner};
733db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
743db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
7558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
76a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCMethodDecl - Represents an instance or class method declaration.
7758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// ObjC methods can be declared within 4 contexts: class interfaces,
7858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// categories, protocols, and class implementations. While C++ member
7958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// functions leverage C syntax, Objective-C method syntax is modeled after
8058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Smalltalk (using colons to specify argument types/expressions).
8158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Here are some brief examples:
8258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
8358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Setter/getter instance methods:
8458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// - (void)setMenu:(NSMenu *)menu;
8558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// - (NSMenu *)menu;
8658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
8758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Instance method that takes 2 NSView arguments:
8858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// - (void)replaceSubview:(NSView *)oldView with:(NSView *)newView;
8958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
9058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Getter class method:
9158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// + (NSMenu *)defaultMenu;
9258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
9358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// A selector represents a unique name for a method. The selector names for
9458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// the above methods are setMenu:, menu, replaceSubview:with:, and defaultMenu.
9558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
962e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregorclass ObjCMethodDecl : public NamedDecl, public DeclContext {
9758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroffpublic:
9858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  enum ImplementationControl { None, Required, Optional };
9958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroffprivate:
10058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// Bitfields must be first fields in this class so they pack with those
10158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// declared in class Decl.
10258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// instance (true) or class (false) method.
10358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool IsInstance : 1;
10458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool IsVariadic : 1;
10558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
1064607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian  // Synthesized declaration method for a property setter/getter
1074607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian  bool IsSynthesized : 1;
1084607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian
109ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  // NOTE: VC++ treats enums as signed, avoid using ImplementationControl enum
11058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// @required/@optional
111ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned DeclImplementation : 2;
11258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
113ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  // NOTE: VC++ treats enums as signed, avoid using the ObjCDeclQualifier enum
11458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// in, inout, etc.
115ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned objcDeclQualifier : 6;
11658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
11758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Context this method is declared in.
11858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  NamedDecl *MethodContext;
11958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
12058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Type of this method.
12158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  QualType MethodDeclType;
12258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// ParamInfo - new[]'d array of pointers to VarDecls for the formal
12358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// parameters of this Method.  This is null if there are no formals.
12458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  ParmVarDecl **ParamInfo;
12558cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  unsigned NumMethodParams;
12658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
12758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// List of attributes for this method declaration.
12858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  SourceLocation EndLoc; // the location of the ';' or '{'.
12958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
13058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // The following are only used for method definitions, null otherwise.
13158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // FIXME: space savings opportunity, consider a sub-class.
13258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  Stmt *Body;
133451318c08a6342c10b8986060386fd9274418437Daniel Dunbar
134451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// SelfDecl - Decl for the implicit self parameter. This is lazily
135451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// constructed by createImplicitParams.
1364111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl *SelfDecl;
137451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// CmdDecl - Decl for the implicit _cmd parameter. This is lazily
138451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// constructed by createImplicitParams.
1394111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl *CmdDecl;
1406c4ae5de0c356777446f823b573821fb95560d91Chris Lattner
141a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl(SourceLocation beginLoc, SourceLocation endLoc,
14258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff                 Selector SelInfo, QualType T,
14358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff                 Decl *contextDecl,
144f6414927e67e27d9324d8d179c5f7ea620443924Daniel Dunbar                 bool isInstance = true,
14558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff                 bool isVariadic = false,
1464607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian                 bool isSynthesized = false,
14758cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner                 ImplementationControl impControl = None)
1482e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  : NamedDecl(ObjCMethod, beginLoc, SelInfo),
149b048c9835969c4f7fe06264748be18ed4b442116Chris Lattner    DeclContext(ObjCMethod),
15058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    IsInstance(isInstance), IsVariadic(isVariadic),
1514607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian    IsSynthesized(isSynthesized),
15258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    DeclImplementation(impControl), objcDeclQualifier(OBJC_TQ_None),
15358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    MethodContext(static_cast<NamedDecl*>(contextDecl)),
1542e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor    MethodDeclType(T),
155f6414927e67e27d9324d8d179c5f7ea620443924Daniel Dunbar    ParamInfo(0), NumMethodParams(0),
1564111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner    EndLoc(endLoc), Body(0), SelfDecl(0), CmdDecl(0) {}
1578a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
1581c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual ~ObjCMethodDecl();
1591c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
1606c4ae5de0c356777446f823b573821fb95560d91Chris Lattnerpublic:
1618a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
1628a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  /// Destroy - Call destructors and release memory.
1638a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  virtual void Destroy(ASTContext& C);
1646c4ae5de0c356777446f823b573821fb95560d91Chris Lattner
1650ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCMethodDecl *Create(ASTContext &C,
1660ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                SourceLocation beginLoc,
1676c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                SourceLocation endLoc, Selector SelInfo,
1686c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                QualType T, Decl *contextDecl,
169f6414927e67e27d9324d8d179c5f7ea620443924Daniel Dunbar                                bool isInstance = true,
1706c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                bool isVariadic = false,
1714607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian                                bool isSynthesized = false,
172b06fa3b86951b9f179c99c3768331536c32e902dChris Lattner                                ImplementationControl impControl = None);
17358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
174ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  ObjCDeclQualifier getObjCDeclQualifier() const {
175ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek    return ObjCDeclQualifier(objcDeclQualifier);
176ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  }
177a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setObjCDeclQualifier(ObjCDeclQualifier QV) { objcDeclQualifier = QV; }
17858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
17958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Location information, modeled after the Stmt API.
18058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  SourceLocation getLocStart() const { return getLocation(); }
18158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
1829776ba0d844cf9f6888e871e3fd246ae782f76f4Daniel Dunbar  SourceRange getSourceRange() const {
1839776ba0d844cf9f6888e871e3fd246ae782f76f4Daniel Dunbar    return SourceRange(getLocation(), EndLoc);
1849776ba0d844cf9f6888e871e3fd246ae782f76f4Daniel Dunbar  }
18558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
18658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  NamedDecl *getMethodContext() const { return MethodContext; }
18758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
1885619688510185081cbb4621d703daf7ee24cf39aChris Lattner  ObjCInterfaceDecl *getClassInterface();
1895619688510185081cbb4621d703daf7ee24cf39aChris Lattner  const ObjCInterfaceDecl *getClassInterface() const {
1905619688510185081cbb4621d703daf7ee24cf39aChris Lattner    return const_cast<ObjCMethodDecl*>(this)->getClassInterface();
191e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  }
19258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
1932e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  Selector getSelector() const { return getDeclName().getObjCSelector(); }
194faf5e779d16bb4590f2a97e1c7ded255eddd90f3Fariborz Jahanian  unsigned getSynthesizedMethodSize() const;
19558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  QualType getResultType() const { return MethodDeclType; }
19658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
197d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  // Iterator access to formal parameters.
19858cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  unsigned param_size() const { return NumMethodParams; }
199d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  typedef ParmVarDecl **param_iterator;
200d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  typedef ParmVarDecl * const *param_const_iterator;
201d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_iterator param_begin() { return ParamInfo; }
202d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_iterator param_end() { return ParamInfo+param_size(); }
203d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_const_iterator param_begin() const { return ParamInfo; }
204d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_const_iterator param_end() const { return ParamInfo+param_size(); }
205d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner
20658cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  unsigned getNumParams() const { return NumMethodParams; }
20758cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  ParmVarDecl *getParamDecl(unsigned i) const {
20858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    assert(i < getNumParams() && "Illegal param #");
20958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    return ParamInfo[i];
21058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
2112338d58a905191c4205d4f73affd0bdaa13493b4Fariborz Jahanian  void setParamDecl(int i, ParmVarDecl *pDecl) {
2122338d58a905191c4205d4f73affd0bdaa13493b4Fariborz Jahanian    ParamInfo[i] = pDecl;
2132338d58a905191c4205d4f73affd0bdaa13493b4Fariborz Jahanian  }
21458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  void setMethodParams(ParmVarDecl **NewParamInfo, unsigned NumParams);
2154111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner
216451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// createImplicitParams - Used to lazily create the self and cmd
217451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// implict parameters. This must be called prior to using getSelfDecl()
218451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// or getCmdDecl(). The call is ignored if the implicit paramters
219451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// have already been created.
220fef30b55230064d334a669a065a1c9acdb87cdfeFariborz Jahanian  void createImplicitParams(ASTContext &Context, const ObjCInterfaceDecl *ID);
221451318c08a6342c10b8986060386fd9274418437Daniel Dunbar
2224111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl * getSelfDecl() const { return SelfDecl; }
2234111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl * getCmdDecl() const { return CmdDecl; }
22458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
22558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool isInstance() const { return IsInstance; }
22658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool isVariadic() const { return IsVariadic; }
22758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
2284607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian  bool isSynthesized() const { return IsSynthesized; }
22991b51a92f2e9fc8025b6a9df88442840eb62823aFariborz Jahanian  void setIsSynthesized() { IsSynthesized = true; }
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; }
24542220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  static DeclContext *castToDeclContext(const ObjCMethodDecl *D) {
24642220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis    return static_cast<DeclContext *>(const_cast<ObjCMethodDecl*>(D));
24742220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  }
24842220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  static ObjCMethodDecl *castFromDeclContext(const DeclContext *DC) {
24942220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis    return static_cast<ObjCMethodDecl *>(const_cast<DeclContext*>(DC));
25042220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  }
25158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff};
252b245a33e6f562377946dc9cb5694a2d3cb8a98a3Fariborz Jahanian
253a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCInterfaceDecl - Represents an ObjC class declaration. For example:
2540c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
2550c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   // MostPrimitive declares no super class (not particularly useful).
2560c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @interface MostPrimitive
2570c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     // no instance variables or methods.
2580c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @end
2590c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
260fd5de471478a507dd2495c4ea9bcab801ea5fa65Chris Lattner///   // NSResponder inherits from NSObject & implements NSCoding (a protocol).
2610c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @interface NSResponder : NSObject <NSCoding>
262a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek///   { // instance variables are represented by ObjCIvarDecl.
2630c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id nextResponder; // nextResponder instance variable.
2640c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   }
2650c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   - (NSResponder *)nextResponder; // return a pointer to NSResponder.
2660c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   - (void)mouseMoved:(NSEvent *)theEvent; // return void, takes a pointer
2670c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @end                                    // to an NSEvent.
2680c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
2690c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   Unlike C/C++, forward class declarations are accomplished with @class.
2700c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   Unlike C/C++, @class allows for a list of classes to be forward declared.
2710c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   Unlike C++, ObjC is a single-rooted class model. In Cocoa, classes
2720c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   typically inherit from NSObject (an exception is NSProxy).
2730c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
274b048c9835969c4f7fe06264748be18ed4b442116Chris Lattnerclass ObjCInterfaceDecl : public NamedDecl, public DeclContext {
2753110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  /// TypeForDecl - This indicates the Type object that represents this
2763110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  /// TypeDecl.  It is a cache maintained by ASTContext::getObjCInterfaceType
2773110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  Type *TypeForDecl;
2783110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  friend class ASTContext;
279980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
280980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Class's super class.
281a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *SuperClass;
282980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
283980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Protocols referenced in interface header declaration
2843db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ObjCList<ObjCProtocolDecl> ReferencedProtocols;
285980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
286980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Ivars/NumIvars - This is a new[]'d array of pointers to Decls.
287a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCIvarDecl **Ivars;   // Null if not defined.
288f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned NumIvars;      // 0 if none.
289980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
290980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// instance methods
291a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **InstanceMethods;  // Null if not defined
292b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner  unsigned NumInstanceMethods;  // 0 if none.
293980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
294980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// class methods
295a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **ClassMethods;  // Null if not defined
29662db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned NumClassMethods;  // 0 if none
297980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
298980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// List of categories defined for this class.
299a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl *CategoryList;
30082a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
30182a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  /// class properties
302a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCPropertyDecl **PropertyDecl;  // Null if no property
30355d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner  unsigned NumPropertyDecl;  // 0 if none.
304980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
3053a3ca1b35a7121aea0bf465a192dce748465e10fFariborz Jahanian  bool ForwardDecl:1; // declared with @class.
3063a3ca1b35a7121aea0bf465a192dce748465e10fFariborz Jahanian  bool InternalInterface:1; // true - no @interface for @implementation
30760fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff
308d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation ClassLoc; // location of the class identifier.
309d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation SuperClassLoc; // location of the super class identifier.
310f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation EndLoc; // marks the '>', '}', or identifier.
311f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation AtEndLoc; // marks the end of the entire interface.
3120e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner
313b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner  ObjCInterfaceDecl(SourceLocation atLoc, IdentifierInfo *Id,
314b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner                    SourceLocation CLoc, bool FD, bool isInternal)
315b048c9835969c4f7fe06264748be18ed4b442116Chris Lattner    : NamedDecl(ObjCInterface, atLoc, Id), DeclContext(ObjCInterface),
316a769c004a2874504c17ea8afccbc4ad35fc33c9fFariborz Jahanian      TypeForDecl(0), SuperClass(0),
3173db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner      Ivars(0), NumIvars(0),
318b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner      InstanceMethods(0), NumInstanceMethods(0),
31962db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner      ClassMethods(0), NumClassMethods(0),
320f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner      CategoryList(0), PropertyDecl(0), NumPropertyDecl(0),
321d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff      ForwardDecl(FD), InternalInterface(isInternal),
322d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff      ClassLoc(CLoc) {
323980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff      }
3248a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
3251c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual ~ObjCInterfaceDecl();
3268a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
3270e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattnerpublic:
3280e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner
3298a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  /// Destroy - Call destructors and release memory.
3308a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  virtual void Destroy(ASTContext& C);
3318a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
3320ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCInterfaceDecl *Create(ASTContext &C,
3330ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                   SourceLocation atLoc,
334d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff                                   IdentifierInfo *Id,
335d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff                                   SourceLocation ClassLoc = SourceLocation(),
3360e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner                                   bool ForwardDecl = false,
3370e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner                                   bool isInternal = false);
3383db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  const ObjCList<ObjCProtocolDecl> &getReferencedProtocols() const {
3397ed9e0f97f4645edc5d4670385b985ea4c617ce7Fariborz Jahanian    return ReferencedProtocols;
3407ed9e0f97f4645edc5d4670385b985ea4c617ce7Fariborz Jahanian  }
341980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
342559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCPropertyDecl *FindPropertyDeclaration(IdentifierInfo *PropertyId) const;
343559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCCategoryDecl *FindCategoryDeclaration(IdentifierInfo *CategoryId) const;
344559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCIvarDecl *FindIvarDeclaration(IdentifierInfo *IvarId) const;
34531afbf02a381ae9c77d225aa54f972d152838b3aFariborz Jahanian  bool isPropertyReadonly(ObjCPropertyDecl *PropertyDecl) const;
3463db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
3473db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  typedef ObjCList<ObjCProtocolDecl>::iterator protocol_iterator;
3483db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  protocol_iterator protocol_begin() const {return ReferencedProtocols.begin();}
3493db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  protocol_iterator protocol_end() const { return ReferencedProtocols.end(); }
350aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian
351a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCIvarDecl * const *ivar_iterator;
352be6df088d69bca0e99c7845a6cd8c1ca85034f31Chris Lattner  ivar_iterator ivar_begin() const { return Ivars; }
3535564e07af5e62875f3b83fc2e5a8823588b5adeaChris Lattner  ivar_iterator ivar_end() const { return Ivars + ivar_size();}
354f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned ivar_size() const { return NumIvars; }
355a0fb5861dec7aa1da0d21d5759678d76b00464f4Ted Kremenek  bool ivar_empty() const { return NumIvars == 0; }
356be6df088d69bca0e99c7845a6cd8c1ca85034f31Chris Lattner
357b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner  unsigned getNumInstanceMethods() const { return NumInstanceMethods; }
35862db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned getNumClassMethods() const { return NumClassMethods; }
359980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
360a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * instmeth_iterator;
3614c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  instmeth_iterator instmeth_begin() const { return InstanceMethods; }
3624c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  instmeth_iterator instmeth_end() const {
363b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner    return InstanceMethods+NumInstanceMethods;
3644c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  }
3654c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner
366a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * classmeth_iterator;
3674c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  classmeth_iterator classmeth_begin() const { return ClassMethods; }
3684c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  classmeth_iterator classmeth_end() const {
36962db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return ClassMethods+NumClassMethods;
3704c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  }
371b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner
372b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner  /// addReferencedProtocols - Set the list of protocols that this interface
373b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner  /// implements.
374780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  void addReferencedProtocols(ObjCProtocolDecl *const*List, unsigned NumRPs) {
375780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    ReferencedProtocols.set(List, NumRPs);
3763db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
377b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner
378a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addInstanceVariablesToClass(ObjCIvarDecl **ivars, unsigned numIvars,
37960fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                                   SourceLocation RBracLoc);
380fd64bb635dc221baa19f81d5d2a084f7eb269f7fFariborz Jahanian  FieldDecl *lookupFieldDeclForIvar(ASTContext &Context,
381fd64bb635dc221baa19f81d5d2a084f7eb269f7fFariborz Jahanian                                    const ObjCIvarDecl *ivar);
382980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
383a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addMethods(ObjCMethodDecl **insMethods, unsigned numInsMembers,
384a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                  ObjCMethodDecl **clsMethods, unsigned numClsMembers,
38560fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                  SourceLocation AtEnd);
386980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
38755d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner  void addProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
38855d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner
389aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian  void mergeProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
390aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian
39133de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  void addPropertyMethods(ASTContext &Context,
39233de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian                          ObjCPropertyDecl* Property,
393b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::SmallVector<ObjCMethodDecl*, 32> &insMethods,
394b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::DenseMap<Selector, const ObjCMethodDecl*> &InsMap);
39533de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
3963dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  typedef ObjCPropertyDecl * const * classprop_iterator;
3973dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_begin() const { return PropertyDecl; }
3983dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_end() const {
3993dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian    return PropertyDecl+NumPropertyDecl;
4003dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  }
40155d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner
402768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  bool isForwardDecl() const { return ForwardDecl; }
403768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  void setForwardDecl(bool val) { ForwardDecl = val; }
404980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
405a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *getSuperClass() const { return SuperClass; }
406a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setSuperClass(ObjCInterfaceDecl * superCls) { SuperClass = superCls; }
407980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
408a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl* getCategoryList() const { return CategoryList; }
409a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setCategoryList(ObjCCategoryDecl *category) {
41053efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    CategoryList = category;
411980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
41253efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner
41353efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  /// isSuperClassOf - Return true if this class is the specified class or is a
41453efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  /// super class of the specified interface class.
41553efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  bool isSuperClassOf(const ObjCInterfaceDecl *I) const {
41653efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    // If RHS is derived from LHS it is OK; else it is not OK.
41753efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    while (I != NULL) {
41853efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner      if (this == I)
41953efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner        return true;
42053efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner      I = I->getSuperClass();
42153efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    }
42253efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    return false;
42353efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  }
42453efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner
42568a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner  ObjCIvarDecl *lookupInstanceVariable(IdentifierInfo *IVarName,
42668a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner                                       ObjCInterfaceDecl *&ClassDeclared);
42768a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner  ObjCIvarDecl *lookupInstanceVariable(IdentifierInfo *IVarName) {
42868a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner    ObjCInterfaceDecl *ClassDeclared;
42968a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner    return lookupInstanceVariable(IVarName, ClassDeclared);
43068a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner  }
43168a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner
432c81c8144a661a49d7b9dae8d2080dee2e43186ecChris Lattner
43358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local instance method declared in this interface.
4343216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const {
43558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (instmeth_iterator I = instmeth_begin(), E = instmeth_end();
4360e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner         I != E; ++I) {
43758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
43858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
43958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
4400e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner    return 0;
44158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
44258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local class method declared in this interface.
4433216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const {
44458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
4450e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner         I != E; ++I) {
44658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
44758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
44858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
4490e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner    return 0;
45058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
45194a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Lookup a method. First, we search locally. If a method isn't
45294a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // found, we search referenced protocols and class categories.
453a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupInstanceMethod(Selector Sel);
454a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupClassMethod(Selector Sel);
45560fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff
456f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  // Location information, modeled after the Stmt API.
45760fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff  SourceLocation getLocStart() const { return getLocation(); } // '@'interface
458f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
459f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
46060fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff
461d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation getClassLoc() const { return ClassLoc; }
462d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  void setSuperClassLoc(SourceLocation Loc) { SuperClassLoc = Loc; }
463d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation getSuperClassLoc() const { return SuperClassLoc; }
464d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff
46560fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff  // We also need to record the @end location.
466f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation getAtEndLoc() const { return AtEndLoc; }
46782a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
468f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner  unsigned getNumPropertyDecl() const { return NumPropertyDecl; }
46982a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
47055d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner  ObjCPropertyDecl * const * getPropertyDecl() const { return PropertyDecl; }
471a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCPropertyDecl **getPropertyDecl() { return PropertyDecl; }
472f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff
4734b6df3fa953267c5d755628c8afd818bb571e579Fariborz Jahanian  /// ImplicitInterfaceDecl - check that this is an implicitely declared
474a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  /// ObjCInterfaceDecl node. This is for legacy objective-c @implementation
4754b6df3fa953267c5d755628c8afd818bb571e579Fariborz Jahanian  /// declaration without an @interface declaration.
4763a3ca1b35a7121aea0bf465a192dce748465e10fFariborz Jahanian  bool ImplicitInterfaceDecl() const { return InternalInterface; }
4774b6df3fa953267c5d755628c8afd818bb571e579Fariborz Jahanian
478a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCInterface; }
479a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCInterfaceDecl *D) { return true; }
48042220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  static DeclContext *castToDeclContext(const ObjCInterfaceDecl *D) {
48142220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis    return static_cast<DeclContext *>(const_cast<ObjCInterfaceDecl*>(D));
48242220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  }
48342220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  static ObjCInterfaceDecl *castFromDeclContext(const DeclContext *DC) {
48442220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis    return static_cast<ObjCInterfaceDecl *>(const_cast<DeclContext*>(DC));
48542220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  }
486980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
487980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
488a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCIvarDecl - Represents an ObjC instance variable. In general, ObjC
4890c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// instance variables are identical to C. The only exception is Objective-C
4900c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// supports C++ style access control. For example:
4910c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
4920c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @interface IvarExample : NSObject
4930c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   {
494f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek///     id defaultToProtected;
4950c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @public:
4960c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id canBePublic; // same as C++.
4970c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @protected:
4980c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id canBeProtected; // same as C++.
4990c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @package:
5000c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id canBePackage; // framework visibility (not available in C++).
5010c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   }
5020c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
503a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCIvarDecl : public FieldDecl {
5040e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattnerpublic:
505980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  enum AccessControl {
506980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff    None, Private, Protected, Public, Package
507980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  };
508f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek
509b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenekprivate:
510b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek  ObjCIvarDecl(SourceLocation L, IdentifierInfo *Id, QualType T,
511b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek               AccessControl ac, Expr *BW)
51244b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor    : FieldDecl(ObjCIvar, 0, L, Id, T, BW, /*Mutable=*/false, 0),
51344b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor      DeclAccess(ac) {}
514b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek
515b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenekpublic:
516b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek  static ObjCIvarDecl *Create(ASTContext &C, SourceLocation L,
517b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek                              IdentifierInfo *Id, QualType T,
518b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek                              AccessControl ac, Expr *BW = NULL);
519b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek
520980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  void setAccessControl(AccessControl ac) { DeclAccess = ac; }
521f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek
522ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  AccessControl getAccessControl() const { return AccessControl(DeclAccess); }
523f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek
524f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek  AccessControl getCanonicalAccessControl() const {
525f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek    return DeclAccess == None ? Protected : AccessControl(DeclAccess);
526f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek  }
527980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
528980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  // Implement isa/cast/dyncast/etc.
529a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCIvar; }
530a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCIvarDecl *D) { return true; }
531980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffprivate:
532ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  // NOTE: VC++ treats enums as signed, avoid using the AccessControl enum
533ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned DeclAccess : 3;
534980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
535980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
53601e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
53701e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek/// ObjCAtDefsFieldDecl - Represents a field declaration created by an
53801e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek///  @defs(...).
53901e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenekclass ObjCAtDefsFieldDecl : public FieldDecl {
54001e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenekprivate:
54144b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor  ObjCAtDefsFieldDecl(DeclContext *DC, SourceLocation L, IdentifierInfo *Id,
54201e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek                      QualType T, Expr *BW)
54344b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor    : FieldDecl(ObjCAtDefsField, DC, L, Id, T, BW, /*Mutable=*/false, 0) {}
54401e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
54501e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenekpublic:
54644b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor  static ObjCAtDefsFieldDecl *Create(ASTContext &C, DeclContext *DC,
54744b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor                                     SourceLocation L,
54801e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek                                     IdentifierInfo *Id, QualType T,
54901e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek                                     Expr *BW);
55001e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
55101e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  virtual void Destroy(ASTContext& C);
55201e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
55301e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  // Implement isa/cast/dyncast/etc.
55401e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCAtDefsField; }
55501e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  static bool classof(const ObjCAtDefsFieldDecl *D) { return true; }
55601e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek};
557980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
558a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCProtocolDecl - Represents a protocol declaration. ObjC protocols
5590c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// declare a pure abstract type (i.e no instance variables are permitted).
5600c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Protocols orginally drew inspiration from C++ pure virtual functions (a C++
5610c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// feature with nice semantics and lousy syntax:-). Here is an example:
5620c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
563eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// @protocol NSDraggingInfo <refproto1, refproto2>
5640c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - (NSWindow *)draggingDestinationWindow;
5650c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - (NSImage *)draggedImage;
5660c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
5670c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
568eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// This says that NSDraggingInfo requires two methods and requires everything
569eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// that the two "referenced protocols" 'refproto1' and 'refproto2' require as
570eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// well.
571eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner///
5720c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @interface ImplementsNSDraggingInfo : NSObject <NSDraggingInfo>
5730c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
5740c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
575a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjC protocols inspired Java interfaces. Unlike Java, ObjC classes and
5760c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// protocols are in distinct namespaces. For example, Cocoa defines both
5770c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// an NSObject protocol and class (which isn't allowed in Java). As a result,
5780c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// protocols are referenced using angle brackets as follows:
5790c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
5800c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// id <NSDraggingInfo> anyObjectThatImplementsNSDraggingInfo;
5810c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
582a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCProtocolDecl : public NamedDecl {
583780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  /// Referenced protocols
584780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  ObjCList<ObjCProtocolDecl> ReferencedProtocols;
585980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
586980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// protocol instance methods
587a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **InstanceMethods;  // Null if not defined
58862db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned NumInstanceMethods;  // 0 if none
589980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
590980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// protocol class methods
591a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **ClassMethods;  // Null if not defined
59262db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned NumClassMethods;  // 0 if none
5933dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
5943dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  /// protocol properties
5953dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  ObjCPropertyDecl **PropertyDecl;  // Null if no property
5963dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  unsigned NumPropertyDecl;  // 0 if none
5973dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
598980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  bool isForwardProtoDecl; // declared with @protocol.
599423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
600423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation EndLoc; // marks the '>' or identifier.
601423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation AtEndLoc; // marks the end of the entire interface.
602cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner
603780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  ObjCProtocolDecl(SourceLocation L, IdentifierInfo *Id)
604a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCProtocol, L, Id),
60562db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner      InstanceMethods(0), NumInstanceMethods(0),
60662db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner      ClassMethods(0), NumClassMethods(0),
6073dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian      PropertyDecl(0), NumPropertyDecl(0),
608c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner      isForwardProtoDecl(true) {
609cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner  }
6101c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
6111c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual ~ObjCProtocolDecl();
6121c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
613cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattnerpublic:
6141c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
6151c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  /// Destroy - Call destructors and release memory.
6161c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual void Destroy(ASTContext& C);
6171c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
618cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner  static ObjCProtocolDecl *Create(ASTContext &C, SourceLocation L,
619780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner                                  IdentifierInfo *Id);
620cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner
621a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addMethods(ObjCMethodDecl **insMethods, unsigned numInsMembers,
622a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                  ObjCMethodDecl **clsMethods, unsigned numClsMembers,
62360fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                  SourceLocation AtEndLoc);
624980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
625780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  const ObjCList<ObjCProtocolDecl> &getReferencedProtocols() const {
626780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    return ReferencedProtocols;
627980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
628dbc933701d20918add13b6a3c9d47ff8c75419cfDaniel Dunbar  typedef ObjCList<ObjCProtocolDecl>::iterator protocol_iterator;
629780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_begin() const {return ReferencedProtocols.begin();}
630780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_end() const { return ReferencedProtocols.end(); }
631780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
632780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  /// addReferencedProtocols - Set the list of protocols that this interface
633780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  /// implements.
634780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  void addReferencedProtocols(ObjCProtocolDecl *const*List, unsigned NumRPs) {
635780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    ReferencedProtocols.set(List, NumRPs);
636aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian  }
637aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian
638c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner  unsigned getNumInstanceMethods() const { return NumInstanceMethods; }
63962db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned getNumClassMethods() const { return NumClassMethods; }
6403d2c22b1d334fa74d26a5f21841cb55df5dfdd1aSteve Naroff
6413d2c22b1d334fa74d26a5f21841cb55df5dfdd1aSteve Naroff  ObjCPropertyDecl *FindPropertyDeclaration(IdentifierInfo *PropertyId) const;
6423dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6433dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  unsigned getNumPropertyDecl() const { return NumPropertyDecl; }
6443dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6453dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  ObjCPropertyDecl * const * getPropertyDecl() const { return PropertyDecl; }
6463dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  ObjCPropertyDecl **getPropertyDecl() { return PropertyDecl; }
6473dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6483dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  void addProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
6497b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar
6507b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar  void addPropertyMethods(ASTContext &Context,
6517b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar                          ObjCPropertyDecl* Property,
652b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::SmallVector<ObjCMethodDecl*, 32> &insMethods,
653b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::DenseMap<Selector, const ObjCMethodDecl*> &InsMap);
6543dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6553dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  typedef ObjCPropertyDecl * const * classprop_iterator;
6563dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_begin() const { return PropertyDecl; }
6573dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_end() const {
6583dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian    return PropertyDecl+NumPropertyDecl;
6593dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  }
66058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
661a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * instmeth_iterator;
66258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_begin() const { return InstanceMethods; }
66358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_end() const {
66462db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return InstanceMethods+NumInstanceMethods;
66558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
66658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
667a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * classmeth_iterator;
66858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_begin() const { return ClassMethods; }
66958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_end() const {
67062db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return ClassMethods+NumClassMethods;
67158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
67258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
67358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local instance method declared in this interface.
6743216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const {
67562db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    for (instmeth_iterator I = instmeth_begin(), E = instmeth_end();
67662db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner         I != E; ++I) {
67758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
67858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
67958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
68062db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return 0;
68158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
68258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local class method declared in this interface.
6833216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const {
68458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
68562db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner         I != E; ++I) {
68658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
68758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
68858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
68962db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return 0;
69058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
691980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
69294a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Lookup a method. First, we search locally. If a method isn't
69394a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // found, we search referenced protocols and class categories.
694a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupInstanceMethod(Selector Sel);
695a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupClassMethod(Selector Sel);
6967dd82836dca87cf828ce994f161b53a34f6cdb7eFariborz Jahanian
697768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  bool isForwardDecl() const { return isForwardProtoDecl; }
698768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  void setForwardDecl(bool val) { isForwardProtoDecl = val; }
699980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
700423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // Location information, modeled after the Stmt API.
701423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocStart() const { return getLocation(); } // '@'protocol
702423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
703423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
704423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
705423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // We also need to record the @end location.
706423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getAtEndLoc() const { return AtEndLoc; }
707423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
708a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCProtocol; }
709a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCProtocolDecl *D) { return true; }
710980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
711980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
712a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCClassDecl - Specifies a list of forward class declarations. For example:
71306ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff///
71406ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff/// @class NSCursor, NSImage, NSPasteboard, NSWindow;
7150c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
716a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCClassDecl : public Decl {
717a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl **ForwardDecls;
7187e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner  unsigned NumForwardDecls;
71961f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
720a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCClassDecl(SourceLocation L, ObjCInterfaceDecl **Elts, unsigned nElts)
721a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : Decl(ObjCClass, L) {
72206ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    if (nElts) {
723a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      ForwardDecls = new ObjCInterfaceDecl*[nElts];
724a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      memcpy(ForwardDecls, Elts, nElts*sizeof(ObjCInterfaceDecl*));
7257e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner    } else {
7267e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner      ForwardDecls = 0;
72706ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    }
72806ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    NumForwardDecls = nElts;
72906ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
730400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
731400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  virtual ~ObjCClassDecl();
732400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
73361f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattnerpublic:
734400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
735400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  /// Destroy - Call destructors and release memory.
736400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  virtual void Destroy(ASTContext& C);
737400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
73861f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner  static ObjCClassDecl *Create(ASTContext &C, SourceLocation L,
73961f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner                               ObjCInterfaceDecl **Elts, unsigned nElts);
74061f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
741a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setInterfaceDecl(unsigned idx, ObjCInterfaceDecl *OID) {
7427e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner    assert(idx < NumForwardDecls && "index out of range");
74306ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    ForwardDecls[idx] = OID;
74406ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
745a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl** getForwardDecls() const { return ForwardDecls; }
7467e15891fc89256fc013bd1003676ad3197b85c25Steve Naroff  int getNumForwardDecls() const { return NumForwardDecls; }
7477e15891fc89256fc013bd1003676ad3197b85c25Steve Naroff
748400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  typedef ObjCInterfaceDecl * const * iterator;
749400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  iterator begin() const { return ForwardDecls; }
750400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  iterator end() const { return ForwardDecls+NumForwardDecls; }
751400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
752a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCClass; }
753a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCClassDecl *D) { return true; }
75406ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff};
75506ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff
756a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCForwardProtocolDecl - Specifies a list of forward protocol declarations.
75706ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff/// For example:
75806ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff///
7590c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @protocol NSTextInput, NSChangeSpelling, NSDraggingInfo;
7600c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
761a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCForwardProtocolDecl : public Decl {
762a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCProtocolDecl **ReferencedProtocols;
7639fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  unsigned NumReferencedProtocols;
76461f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
7650ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  ObjCForwardProtocolDecl(SourceLocation L,
766a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                          ObjCProtocolDecl **Elts, unsigned nElts)
767a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  : Decl(ObjCForwardProtocol, L) {
768b97de3eddf0a0f7ff97204e7def6b7e0aa8953f2Chris Lattner    NumReferencedProtocols = nElts;
76906ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    if (nElts) {
770a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      ReferencedProtocols = new ObjCProtocolDecl*[nElts];
771a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      memcpy(ReferencedProtocols, Elts, nElts*sizeof(ObjCProtocolDecl*));
7729fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    } else {
7739fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner      ReferencedProtocols = 0;
774980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff    }
77506ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
77605ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek
77705ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  virtual ~ObjCForwardProtocolDecl();
77805ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek
77961f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattnerpublic:
78061f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner  static ObjCForwardProtocolDecl *Create(ASTContext &C, SourceLocation L,
78161f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner                                         ObjCProtocolDecl **Elts, unsigned Num);
78261f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
78361f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
784a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setForwardProtocolDecl(unsigned idx, ObjCProtocolDecl *OID) {
7859fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    assert(idx < NumReferencedProtocols && "index out of range");
7867ed9e0f97f4645edc5d4670385b985ea4c617ce7Fariborz Jahanian    ReferencedProtocols[idx] = OID;
78706ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
7889fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner
7899fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  unsigned getNumForwardDecls() const { return NumReferencedProtocols; }
7909fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner
791a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCProtocolDecl *getForwardProtocolDecl(unsigned idx) {
7929fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    assert(idx < NumReferencedProtocols && "index out of range");
7939fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    return ReferencedProtocols[idx];
7949fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  }
795a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  const ObjCProtocolDecl *getForwardProtocolDecl(unsigned idx) const {
7969fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    assert(idx < NumReferencedProtocols && "index out of range");
7979fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    return ReferencedProtocols[idx];
7989fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  }
7999fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner
80005ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  typedef ObjCProtocolDecl * const * iterator;
80105ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  iterator begin() const { return ReferencedProtocols; }
80205ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  iterator end() const { return ReferencedProtocols+NumReferencedProtocols; }
80305ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek
80406ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  static bool classof(const Decl *D) {
805a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    return D->getKind() == ObjCForwardProtocol;
80606ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
807a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCForwardProtocolDecl *D) { return true; }
808980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
809980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
810a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCCategoryDecl - Represents a category declaration. A category allows
8110c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// you to add methods to an existing class (without subclassing or modifying
8120c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// the original class interface or implementation:-). Categories don't allow
8130c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// you to add instance data. The following example adds "myMethod" to all
8140c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// NSView's within a process:
8150c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
8160c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @interface NSView (MyViewMethods)
8170c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - myMethod;
8180c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
8190c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
8200c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Cateogries also allow you to split the implementation of a class across
8210c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// several files (a feature more naturally supported in C++).
8220c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
8230c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Categories were originally inspired by dynamic languages such as Common
82468c82cf61228102aba1194efef222fa1478af2a8Chris Lattner/// Lisp and Smalltalk.  More traditional class-based languages (C++, Java)
8250c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// don't support this level of dynamism, which is both powerful and dangerous.
8260c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
827a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCCategoryDecl : public NamedDecl {
828980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Interface belonging to this category
829a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *ClassInterface;
830980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
83168c82cf61228102aba1194efef222fa1478af2a8Chris Lattner  /// referenced protocols in this category.
832780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  ObjCList<ObjCProtocolDecl> ReferencedProtocols;
833980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
834980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// category instance methods
835a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **InstanceMethods;  // Null if not defined
836c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner  unsigned NumInstanceMethods;  // 0 if none
837980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
838980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// category class methods
839a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **ClassMethods;  // Null if not defined
840a906135721c350435319347d2672bbb3bf494f91Chris Lattner  unsigned NumClassMethods;  // 0 if not defined
841980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
842980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Next category belonging to this class
843a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl *NextClassCategory;
844ca3adf7e8cac8c9fbaf592b1e5c2be6f082de7baFariborz Jahanian
8457e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  /// category properties
8467e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  ObjCPropertyDecl **PropertyDecl;  // Null if no property
8477e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  unsigned NumPropertyDecl;  // 0 if none
8487e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian
849423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation EndLoc; // marks the '>' or identifier.
850423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation AtEndLoc; // marks the end of the entire interface.
85161f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
85268c82cf61228102aba1194efef222fa1478af2a8Chris Lattner  ObjCCategoryDecl(SourceLocation L, IdentifierInfo *Id)
853a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCCategory, L, Id),
854780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner      ClassInterface(0),
855c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner      InstanceMethods(0), NumInstanceMethods(0),
856a906135721c350435319347d2672bbb3bf494f91Chris Lattner      ClassMethods(0), NumClassMethods(0),
8577e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian      NextClassCategory(0), PropertyDecl(0),  NumPropertyDecl(0) {
858a906135721c350435319347d2672bbb3bf494f91Chris Lattner  }
85961f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattnerpublic:
86061f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
8610ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCCategoryDecl *Create(ASTContext &C,
8620ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                  SourceLocation L, IdentifierInfo *Id);
86361f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
864e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
865e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
866a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setClassInterface(ObjCInterfaceDecl *IDecl) { ClassInterface = IDecl; }
867980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
868f7b2c98c16dfb2261ea57d40a1d5bc4738e73175Chris Lattner  /// addReferencedProtocols - Set the list of protocols that this interface
869f7b2c98c16dfb2261ea57d40a1d5bc4738e73175Chris Lattner  /// implements.
870780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  void addReferencedProtocols(ObjCProtocolDecl *const*List, unsigned NumRPs) {
871780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    ReferencedProtocols.set(List, NumRPs);
872780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  }
873980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
874780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  const ObjCList<ObjCProtocolDecl> &getReferencedProtocols() const {
875780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    return ReferencedProtocols;
8768f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian  }
877780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
878780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  typedef ObjCProtocolDecl * const * protocol_iterator;
879780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_begin() const {return ReferencedProtocols.begin();}
880780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_end() const { return ReferencedProtocols.end(); }
881780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
882780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
883c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner  unsigned getNumInstanceMethods() const { return NumInstanceMethods; }
884a906135721c350435319347d2672bbb3bf494f91Chris Lattner  unsigned getNumClassMethods() const { return NumClassMethods; }
88558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
8867e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  unsigned getNumPropertyDecl() const { return NumPropertyDecl; }
8877e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian
8887e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  ObjCPropertyDecl * const * getPropertyDecl() const { return PropertyDecl; }
8897e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian
8903dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  void addProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
8917b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar
8921ac2bc44781ec8451f880dcf586768a71824d3a6Fariborz Jahanian  void mergeProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
8931ac2bc44781ec8451f880dcf586768a71824d3a6Fariborz Jahanian
8947b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar  void addPropertyMethods(ASTContext &Context,
8957b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar                          ObjCPropertyDecl* Property,
896b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::SmallVector<ObjCMethodDecl*, 32> &insMethods,
897b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::DenseMap<Selector, const ObjCMethodDecl*> &InsMap);
8983dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
899559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCPropertyDecl *FindPropertyDeclaration(IdentifierInfo *PropertyId) const;
900559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian
9013dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  typedef ObjCPropertyDecl * const * classprop_iterator;
9023dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_begin() const { return PropertyDecl; }
9033dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_end() const {
9043dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian    return PropertyDecl+NumPropertyDecl;
9053dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  }
9063dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
907a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * instmeth_iterator;
90858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_begin() const { return InstanceMethods; }
90958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_end() const {
910c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner    return InstanceMethods+NumInstanceMethods;
91158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
91258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
913a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * classmeth_iterator;
91458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_begin() const { return ClassMethods; }
91558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_end() const {
916a906135721c350435319347d2672bbb3bf494f91Chris Lattner    return ClassMethods+NumClassMethods;
91758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
91858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
91958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local instance method declared in this interface.
9203216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const {
92158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (instmeth_iterator I = instmeth_begin(), E = instmeth_end();
92268c82cf61228102aba1194efef222fa1478af2a8Chris Lattner         I != E; ++I) {
92358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
92458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
92558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
92668c82cf61228102aba1194efef222fa1478af2a8Chris Lattner    return 0;
92758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
92858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local class method declared in this interface.
9293216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const {
93058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
93168c82cf61228102aba1194efef222fa1478af2a8Chris Lattner         I != E; ++I) {
93258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
93358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
93458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
93568c82cf61228102aba1194efef222fa1478af2a8Chris Lattner    return 0;
93658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
9378f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
938a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addMethods(ObjCMethodDecl **insMethods, unsigned numInsMembers,
939a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                  ObjCMethodDecl **clsMethods, unsigned numClsMembers,
94060fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                  SourceLocation AtEndLoc);
941980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
942a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl *getNextClassCategory() const { return NextClassCategory; }
943980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  void insertNextClassCategory() {
9443d58138992b9bc7b34aaa680f3ddf3971292eb7dSteve Naroff    NextClassCategory = ClassInterface->getCategoryList();
9453d58138992b9bc7b34aaa680f3ddf3971292eb7dSteve Naroff    ClassInterface->setCategoryList(this);
946980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
947423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // Location information, modeled after the Stmt API.
948423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocStart() const { return getLocation(); } // '@'interface
949423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
950423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
951423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
952423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // We also need to record the @end location.
953423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getAtEndLoc() const { return AtEndLoc; }
954ca3adf7e8cac8c9fbaf592b1e5c2be6f082de7baFariborz Jahanian
955a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCCategory; }
956a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCCategoryDecl *D) { return true; }
957980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
9580c6b6243d3efd958c17943130e2a773653511edcSteve Naroff
959a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCCategoryImplDecl - An object of this class encapsulates a category
960559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @implementation declaration. If a category class has declaration of a
961559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// property, its implementation must be specified in the category's
962559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @implementation declaration. Example:
963559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @interface I @end
964559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @interface I(CATEGORY)
965559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian///    @property int p1, d1;
966559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @end
967559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @implementation I(CATEGORY)
968559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian///  @dynamic p1,d1;
969559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @end
970559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian///
971a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCCategoryImplDecl : public NamedDecl {
9728f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian  /// Class interface for this category implementation
973a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *ClassInterface;
9748f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
975e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  /// implemented instance methods
976a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> InstanceMethods;
9778f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
978e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  /// implemented class methods
979a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> ClassMethods;
980f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
981559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  /// Property Implementations in this category
982f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  llvm::SmallVector<ObjCPropertyImplDecl*, 8> PropertyImplementations;
983e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
984e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  SourceLocation EndLoc;
98575c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner
986a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryImplDecl(SourceLocation L, IdentifierInfo *Id,
987a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                       ObjCInterfaceDecl *classInterface)
988a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCCategoryImpl, L, Id), ClassInterface(classInterface) {}
98975c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattnerpublic:
9900ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCCategoryImplDecl *Create(ASTContext &C,
9910ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                      SourceLocation L, IdentifierInfo *Id,
99275c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner                                      ObjCInterfaceDecl *classInterface);
9938f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
994e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
995e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
9968f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
997ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  unsigned getNumInstanceMethods() const { return InstanceMethods.size(); }
998ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  unsigned getNumClassMethods() const { return ClassMethods.size(); }
999e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
1000a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addInstanceMethod(ObjCMethodDecl *method) {
1001e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff    InstanceMethods.push_back(method);
1002e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  }
1003a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addClassMethod(ObjCMethodDecl *method) {
1004e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff    ClassMethods.push_back(method);
1005f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  }
100694a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the instance method definition for this implementation.
10073216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const;
100894a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff
100994a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the class method definition for this implementation.
10103216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const;
1011f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
1012f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  void addPropertyImplementation(ObjCPropertyImplDecl *property) {
1013f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian    PropertyImplementations.push_back(property);
1014f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  }
1015e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
1016ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian  ObjCPropertyImplDecl *FindPropertyImplDecl(IdentifierInfo *propertyId) const;
1017ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian  ObjCPropertyImplDecl *FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const;
1018ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian
1019f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  unsigned getNumPropertyImplementations() const
1020f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  { return PropertyImplementations.size(); }
1021f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
1022559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian
1023559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  typedef llvm::SmallVector<ObjCPropertyImplDecl*, 8>::const_iterator
1024559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    propimpl_iterator;
1025559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_begin() const {
1026559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.begin();
1027559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1028559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_end() const {
1029559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.end();
1030559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1031559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian
1032a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
1033ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner    instmeth_iterator;
1034ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  instmeth_iterator instmeth_begin() const { return InstanceMethods.begin(); }
1035ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  instmeth_iterator instmeth_end() const { return InstanceMethods.end(); }
1036ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner
1037a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
1038ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner    classmeth_iterator;
1039ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  classmeth_iterator classmeth_begin() const { return ClassMethods.begin(); }
1040ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  classmeth_iterator classmeth_end() const { return ClassMethods.end(); }
1041ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner
1042ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner
1043e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  // Location information, modeled after the Stmt API.
1044e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  SourceLocation getLocStart() const { return getLocation(); }
1045e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
1046e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
1047e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
1048a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCCategoryImpl;}
1049a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCCategoryImplDecl *D) { return true; }
10508f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian};
10518f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
1052a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCImplementationDecl - Represents a class definition - this is where
10530c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// method definitions are specified. For example:
10540c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
10550c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @implementation MyClass
10560c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - (void)myMethod { /* do something */ }
10570c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
10580c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
10590c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Typically, instance variables are specified in the class interface,
10600c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// *not* in the implemenentation. Nevertheless (for legacy reasons), we
10610c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// allow instance variables to be specified in the implementation. When
10620c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// specified, they need to be *identical* to the interface. Now that we
10630c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// have support for non-fragile ivars in ObjC 2.0, we can consider removing
10640c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// the legacy semantics and allow developers to move private ivar declarations
10650c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// from the class interface to the class implementation (but I digress:-)
10660c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
1067a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCImplementationDecl : public NamedDecl {
1068dae1a1a2aa4f245b1958dc8db6089f24c575ef13Fariborz Jahanian  /// Class interface for this implementation
1069a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *ClassInterface;
1070f4d331dd922f92478ebf30e808c0ca97ce49418bFariborz Jahanian
1071980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Implementation Class's super class.
1072a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *SuperClass;
1073980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1074980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Optional Ivars/NumIvars - This is a new[]'d array of pointers to Decls.
1075a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCIvarDecl **Ivars;   // Null if not specified
1076f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned NumIvars;      // 0 if none.
10770416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1078980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// implemented instance methods
1079a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> InstanceMethods;
10800416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1081980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// implemented class methods
1082a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> ClassMethods;
10830416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1084f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  /// Propertys' being implemented
1085f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  llvm::SmallVector<ObjCPropertyImplDecl*, 8> PropertyImplementations;
1086f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
10870416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  SourceLocation EndLoc;
108875c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner
1089a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCImplementationDecl(SourceLocation L, IdentifierInfo *Id,
1090a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                         ObjCInterfaceDecl *classInterface,
1091a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                         ObjCInterfaceDecl *superDecl)
1092a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCImplementation, L, Id),
10930416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff      ClassInterface(classInterface), SuperClass(superDecl),
1094f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner      Ivars(0), NumIvars(0) {}
109575c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattnerpublic:
10960ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCImplementationDecl *Create(ASTContext &C,
10970ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                        SourceLocation L, IdentifierInfo *Id,
109875c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner                                        ObjCInterfaceDecl *classInterface,
109975c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner                                        ObjCInterfaceDecl *superDecl);
110075c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner
1101980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1102a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void ObjCAddInstanceVariablesToClassImpl(ObjCIvarDecl **ivars,
1103980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff                                           unsigned numIvars);
1104980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1105a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addInstanceMethod(ObjCMethodDecl *method) {
11060416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff    InstanceMethods.push_back(method);
11070416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  }
1108a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addClassMethod(ObjCMethodDecl *method) {
11090416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff    ClassMethods.push_back(method);
11100416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  }
1111f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
1112f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  void addPropertyImplementation(ObjCPropertyImplDecl *property) {
1113f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian    PropertyImplementations.push_back(property);
1114f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  }
1115ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian
1116ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian  ObjCPropertyImplDecl *FindPropertyImplDecl(IdentifierInfo *propertyId) const;
1117ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian  ObjCPropertyImplDecl *FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const;
1118ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian
1119559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  typedef llvm::SmallVector<ObjCPropertyImplDecl*, 8>::const_iterator
1120559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator;
1121559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_begin() const {
1122559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.begin();
1123559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1124559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_end() const {
1125559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.end();
1126559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1127f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
11280416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  // Location information, modeled after the Stmt API.
11290416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  SourceLocation getLocStart() const { return getLocation(); }
11300416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
11310416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
11320416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1133e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
1134e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
1135e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getSuperClass() const { return SuperClass; }
1136e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getSuperClass() { return SuperClass; }
1137980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1138f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  void setSuperClass(ObjCInterfaceDecl * superCls) { SuperClass = superCls; }
1139980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1140b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner  unsigned getNumInstanceMethods() const { return InstanceMethods.size(); }
114162db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned getNumClassMethods() const { return ClassMethods.size(); }
1142f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
1143f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  unsigned getNumPropertyImplementations() const
1144f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian    { return PropertyImplementations.size(); }
1145c43d868355374d48296ad3be2c9c536698a5e9a8Steve Naroff
1146a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
11470157c5144513438bb74aebf50d18f95df4104acbChris Lattner       instmeth_iterator;
11480157c5144513438bb74aebf50d18f95df4104acbChris Lattner  instmeth_iterator instmeth_begin() const { return InstanceMethods.begin(); }
11490157c5144513438bb74aebf50d18f95df4104acbChris Lattner  instmeth_iterator instmeth_end() const { return InstanceMethods.end(); }
11500157c5144513438bb74aebf50d18f95df4104acbChris Lattner
1151a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
11520157c5144513438bb74aebf50d18f95df4104acbChris Lattner    classmeth_iterator;
11530157c5144513438bb74aebf50d18f95df4104acbChris Lattner  classmeth_iterator classmeth_begin() const { return ClassMethods.begin(); }
11540157c5144513438bb74aebf50d18f95df4104acbChris Lattner  classmeth_iterator classmeth_end() const { return ClassMethods.end(); }
11550157c5144513438bb74aebf50d18f95df4104acbChris Lattner
115694a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the instance method definition for this implementation.
11573216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const;
11580157c5144513438bb74aebf50d18f95df4104acbChris Lattner
115994a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the class method definition for this implementation.
11603216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const;
11610157c5144513438bb74aebf50d18f95df4104acbChris Lattner
1162a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCIvarDecl * const *ivar_iterator;
11630157c5144513438bb74aebf50d18f95df4104acbChris Lattner  ivar_iterator ivar_begin() const { return Ivars; }
1164f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  ivar_iterator ivar_end() const { return Ivars+NumIvars; }
1165f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned ivar_size() const { return NumIvars; }
1166f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  bool ivar_empty() const { return NumIvars == 0; }
11670157c5144513438bb74aebf50d18f95df4104acbChris Lattner
1168980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  static bool classof(const Decl *D) {
1169a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    return D->getKind() == ObjCImplementation;
1170980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
1171a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCImplementationDecl *D) { return true; }
1172980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
1173243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian
1174a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCCompatibleAliasDecl - Represents alias of a class. This alias is
1175243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian/// declared as @compatibility_alias alias class.
1176e8043c39176e7f253fbd92982b077eca6bf2fd59Steve Naroffclass ObjCCompatibleAliasDecl : public NamedDecl {
1177243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian  /// Class that this is an alias of.
1178a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *AliasedClass;
1179243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian
1180a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCompatibleAliasDecl(SourceLocation L, IdentifierInfo *Id,
1181e8043c39176e7f253fbd92982b077eca6bf2fd59Steve Naroff                          ObjCInterfaceDecl* aliasedClass)
1182e8043c39176e7f253fbd92982b077eca6bf2fd59Steve Naroff    : NamedDecl(ObjCCompatibleAlias, L, Id), AliasedClass(aliasedClass) {}
1183f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattnerpublic:
11840ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCCompatibleAliasDecl *Create(ASTContext &C,
11850ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                         SourceLocation L, IdentifierInfo *Id,
1186f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner                                         ObjCInterfaceDecl* aliasedClass);
1187f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner
1188f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return AliasedClass; }
1189f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner  ObjCInterfaceDecl *getClassInterface() { return AliasedClass; }
1190980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1191243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian  static bool classof(const Decl *D) {
11928a934233d1582b5bde9d270bc0705aa81e471a79Chris Lattner    return D->getKind() == ObjCCompatibleAlias;
1193243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian  }
1194a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCCompatibleAliasDecl *D) { return true; }
1195243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian
1196243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian};
11971de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian
11981de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian/// ObjCPropertyDecl - Represents one property declaration in an interface.
11991de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian/// For example:
12001de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian/// @property (assign, readwrite) int MyProperty;
12011de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian///
12021de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanianclass ObjCPropertyDecl : public NamedDecl {
120382a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanianpublic:
1204a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner  enum PropertyAttributeKind {
1205a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_noattr    = 0x00,
1206a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_readonly  = 0x01,
1207a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_getter    = 0x02,
1208a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_assign    = 0x04,
1209a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_readwrite = 0x08,
1210a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_retain    = 0x10,
1211a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_copy      = 0x20,
1212a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_nonatomic = 0x40,
1213a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_setter    = 0x80
1214a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner  };
1215af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
1216af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  enum SetterKind { Assign, Retain, Copy };
121746b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  enum PropertyControl { None, Required, Optional };
121882a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanianprivate:
1219dae1a1a2aa4f245b1958dc8db6089f24c575ef13Fariborz Jahanian  QualType DeclType;
1220ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned PropertyAttributes : 8;
122182a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
122246b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  // @required/@optional
122346b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  unsigned PropertyImplementation : 2;
122446b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian
12255251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector GetterName;    // getter name of NULL if no getter
12265251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector SetterName;    // setter name of NULL if no setter
122782a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
122833de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *GetterMethodDecl; // Declaration of getter instance method
122933de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *SetterMethodDecl; // Declaration of setter instance method
123033de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
12311de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian  ObjCPropertyDecl(SourceLocation L, IdentifierInfo *Id, QualType T)
12321de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian    : NamedDecl(ObjCProperty, L, Id), DeclType(T),
123333de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      PropertyAttributes(OBJC_PR_noattr), PropertyImplementation(None),
123433de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      GetterName(Selector()),
123533de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      SetterName(Selector()),
123633de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      GetterMethodDecl(0), SetterMethodDecl(0) {}
1237f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattnerpublic:
12381de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian  static ObjCPropertyDecl *Create(ASTContext &C, SourceLocation L,
123946b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian                                  IdentifierInfo *Id, QualType T,
124046b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian                                  PropertyControl propControl = None);
1241dae1a1a2aa4f245b1958dc8db6089f24c575ef13Fariborz Jahanian  QualType getType() const { return DeclType; }
1242c35b9e4e2efad727538c848cf30d4b0eb1031dc9Fariborz Jahanian
1243a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner  PropertyAttributeKind getPropertyAttributes() const {
1244f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner    return PropertyAttributeKind(PropertyAttributes);
1245f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner  }
1246564576b225386cbff375351597dd5e2a92872d38Fariborz Jahanian  void setPropertyAttributes(PropertyAttributeKind PRVal) {
1247a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    PropertyAttributes |= PRVal;
124882a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  }
1249394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar
12508cf0bb3c2a798ce3acacaac2d3178648cd4c65c6Fariborz Jahanian void makeitReadWriteAttribute(void) {
12518cf0bb3c2a798ce3acacaac2d3178648cd4c65c6Fariborz Jahanian    PropertyAttributes &= ~OBJC_PR_readonly;
12528cf0bb3c2a798ce3acacaac2d3178648cd4c65c6Fariborz Jahanian    PropertyAttributes |= OBJC_PR_readwrite;
12538cf0bb3c2a798ce3acacaac2d3178648cd4c65c6Fariborz Jahanian }
12548cf0bb3c2a798ce3acacaac2d3178648cd4c65c6Fariborz Jahanian
1255af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  // Helper methods for accessing attributes.
1256af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
1257af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// isReadOnly - Return true iff the property has a setter.
1258394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar  bool isReadOnly() const {
1259394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar    return (PropertyAttributes & OBJC_PR_readonly);
1260394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar  }
1261af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
1262af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// getSetterKind - Return the method used for doing assignment in
1263af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// the property setter. This is only valid if the property has been
1264af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// defined to have a setter.
1265af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  SetterKind getSetterKind() const {
1266af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar    if (PropertyAttributes & OBJC_PR_retain)
1267af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar      return Retain;
1268af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar    if (PropertyAttributes & OBJC_PR_copy)
1269af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar      return Copy;
1270af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar    return Assign;
1271af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  }
1272af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
12735251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector getGetterName() const { return GetterName; }
12745251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  void setGetterName(Selector Sel) { GetterName = Sel; }
127582a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
12765251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector getSetterName() const { return SetterName; }
12775251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  void setSetterName(Selector Sel) { SetterName = Sel; }
127882a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
127933de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *getGetterMethodDecl() const { return GetterMethodDecl; }
128033de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  void setGetterMethodDecl(ObjCMethodDecl *gDecl) { GetterMethodDecl = gDecl; }
128133de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
128233de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *getSetterMethodDecl() const { return SetterMethodDecl; }
128333de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  void setSetterMethodDecl(ObjCMethodDecl *gDecl) { SetterMethodDecl = gDecl; }
128433de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
128546b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  // Related to @optional/@required declared in @protocol
128646b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  void setPropertyImplementation(PropertyControl pc) {
128746b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian    PropertyImplementation = pc;
128846b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  }
128946b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  PropertyControl getPropertyImplementation() const {
129046b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian    return PropertyControl(PropertyImplementation);
129146b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  }
129246b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian
129382a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  static bool classof(const Decl *D) {
1294670aa9d7639278f507930e95dc89c12032ab7c7eSam Bishop    return D->getKind() == ObjCProperty;
129582a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  }
1296a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCPropertyDecl *D) { return true; }
129782a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian};
1298980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
129961d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian/// ObjCPropertyImplDecl - Represents implementation declaration of a property
130061d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian/// in a class or category implementation block. For example:
130161d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian/// @synthesize prop1 = ivar1;
130261d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian///
130361d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanianclass ObjCPropertyImplDecl : public Decl {
130461d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanianpublic:
13059f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar  enum Kind {
13069f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    Synthesize,
13079f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    Dynamic
130861d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  };
130961d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanianprivate:
1310559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  SourceLocation AtLoc;   // location of @synthesize or @dynamic
131161d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  /// Property declaration being implemented
131261d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  ObjCPropertyDecl *PropertyDecl;
1313be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek
131461d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  /// Null for @dynamic. Required for @synthesize.
131561d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  ObjCIvarDecl *PropertyIvarDecl;
1316be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek
1317628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian  ObjCPropertyImplDecl(SourceLocation atLoc, SourceLocation L,
1318628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                       ObjCPropertyDecl *property,
13199f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar                       Kind PK,
1320628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                       ObjCIvarDecl *ivarDecl)
13219f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    : Decl(ObjCPropertyImpl, L), AtLoc(atLoc), PropertyDecl(property),
13229f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar      PropertyIvarDecl(ivarDecl) {
13239f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    assert (PK == Dynamic || PropertyIvarDecl);
13249f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar  }
1325628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian
13269f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbarpublic:
1327628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian  static ObjCPropertyImplDecl *Create(ASTContext &C, SourceLocation atLoc,
1328628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                      SourceLocation L,
1329628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                      ObjCPropertyDecl *property,
13309f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar                                      Kind PK,
1331628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                      ObjCIvarDecl *ivarDecl);
133261d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
1333d40910b581b09c937a8c1fdcde9b8ec724398fb9Steve Naroff  SourceLocation getLocStart() const { return AtLoc; }
1334d40910b581b09c937a8c1fdcde9b8ec724398fb9Steve Naroff
1335be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  ObjCPropertyDecl *getPropertyDecl() const {
1336be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek    return PropertyDecl;
1337be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  }
133861d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
13399f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar  Kind getPropertyImplementation() const {
13409f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    return PropertyIvarDecl ? Synthesize : Dynamic;
1341be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  }
134261d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
1343af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  ObjCIvarDecl *getPropertyIvarDecl() const {
1344be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek    return PropertyIvarDecl;
1345be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  }
134661d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
134761d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  static bool classof(const Decl *D) {
134861d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian    return D->getKind() == ObjCPropertyImpl;
134961d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  }
135061d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  static bool classof(const ObjCPropertyImplDecl *D) { return true; }
135161d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian};
135261d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
1353980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff}  // end namespace clang
1354980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff#endif
1355