DeclObjC.h revision 91b51a92f2e9fc8025b6a9df88442840eb62823a
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;
26a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCIvarDecl;
27a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCMethodDecl;
28a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCProtocolDecl;
29a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCCategoryDecl;
30a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCPropertyDecl;
31f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanianclass ObjCPropertyImplDecl;
323db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
333db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
343db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// ObjCList - This is a simple template class used to hold various lists of
353db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// decls etc, which is heavily used by the ObjC front-end.  This only use case
363db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// this supports is setting the list all at once and then reading elements out
373db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner/// of it.
383db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattnertemplate <typename T>
393db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattnerclass ObjCList {
403db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  /// List is a new[]'d array of pointers to objects that are not owned by this
413db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  /// list.
423db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  T **List;
433db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  unsigned NumElts;
443db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
453db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  void operator=(const ObjCList &); // DO NOT IMPLEMENT
463db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ObjCList(const ObjCList&);        // DO NOT IMPLEMENT
473db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattnerpublic:
483db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ObjCList() : List(0), NumElts(0) {}
493db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ~ObjCList() {
503db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    delete[] List;
513db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
523db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
533db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  void set(T* const* InList, unsigned Elts) {
543db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    assert(List == 0 && "Elements already set!");
553db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    List = new T*[Elts];
563db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    NumElts = Elts;
573db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    memcpy(List, InList, sizeof(T*)*Elts);
583db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
593db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
603db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  typedef T* const * iterator;
613db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  iterator begin() const { return List; }
623db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  iterator end() const { return List+NumElts; }
633db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
643db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  unsigned size() const { return NumElts; }
653db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  bool empty() const { return NumElts == 0; }
663db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
67780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  T* operator[](unsigned idx) const {
683db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    assert(idx < NumElts && "Invalid access");
693db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner    return List[idx];
703db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
713db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner};
723db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
733db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
7458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
75a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCMethodDecl - Represents an instance or class method declaration.
7658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// ObjC methods can be declared within 4 contexts: class interfaces,
7758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// categories, protocols, and class implementations. While C++ member
7858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// functions leverage C syntax, Objective-C method syntax is modeled after
7958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Smalltalk (using colons to specify argument types/expressions).
8058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Here are some brief examples:
8158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
8258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Setter/getter instance methods:
8358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// - (void)setMenu:(NSMenu *)menu;
8458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// - (NSMenu *)menu;
8558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
8658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Instance method that takes 2 NSView arguments:
8758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// - (void)replaceSubview:(NSView *)oldView with:(NSView *)newView;
8858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
8958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// Getter class method:
9058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// + (NSMenu *)defaultMenu;
9158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
9258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// A selector represents a unique name for a method. The selector names for
9358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff/// the above methods are setMenu:, menu, replaceSubview:with:, and defaultMenu.
9458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff///
952e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregorclass ObjCMethodDecl : public NamedDecl, public DeclContext {
9658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroffpublic:
9758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  enum ImplementationControl { None, Required, Optional };
9858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroffprivate:
9958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// Bitfields must be first fields in this class so they pack with those
10058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// declared in class Decl.
10158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// instance (true) or class (false) method.
10258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool IsInstance : 1;
10358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool IsVariadic : 1;
10458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
1054607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian  // Synthesized declaration method for a property setter/getter
1064607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian  bool IsSynthesized : 1;
1074607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian
108ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  // NOTE: VC++ treats enums as signed, avoid using ImplementationControl enum
10958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// @required/@optional
110ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned DeclImplementation : 2;
11158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
112ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  // NOTE: VC++ treats enums as signed, avoid using the ObjCDeclQualifier enum
11358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// in, inout, etc.
114ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned objcDeclQualifier : 6;
11558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
11658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Context this method is declared in.
11758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  NamedDecl *MethodContext;
11858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
11958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Type of this method.
12058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  QualType MethodDeclType;
12158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// ParamInfo - new[]'d array of pointers to VarDecls for the formal
12258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// parameters of this Method.  This is null if there are no formals.
12358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  ParmVarDecl **ParamInfo;
12458cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  unsigned NumMethodParams;
12558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
12658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  /// List of attributes for this method declaration.
12758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  SourceLocation EndLoc; // the location of the ';' or '{'.
12858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
12958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // The following are only used for method definitions, null otherwise.
13058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // FIXME: space savings opportunity, consider a sub-class.
13158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  Stmt *Body;
132451318c08a6342c10b8986060386fd9274418437Daniel Dunbar
133451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// SelfDecl - Decl for the implicit self parameter. This is lazily
134451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// constructed by createImplicitParams.
1354111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl *SelfDecl;
136451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// CmdDecl - Decl for the implicit _cmd parameter. This is lazily
137451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// constructed by createImplicitParams.
1384111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl *CmdDecl;
1396c4ae5de0c356777446f823b573821fb95560d91Chris Lattner
140a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl(SourceLocation beginLoc, SourceLocation endLoc,
14158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff                 Selector SelInfo, QualType T,
14258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff                 Decl *contextDecl,
143f6414927e67e27d9324d8d179c5f7ea620443924Daniel Dunbar                 bool isInstance = true,
14458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff                 bool isVariadic = false,
1454607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian                 bool isSynthesized = false,
14658cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner                 ImplementationControl impControl = None)
1472e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  : NamedDecl(ObjCMethod, beginLoc, SelInfo),
148b048c9835969c4f7fe06264748be18ed4b442116Chris Lattner    DeclContext(ObjCMethod),
14958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    IsInstance(isInstance), IsVariadic(isVariadic),
1504607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian    IsSynthesized(isSynthesized),
15158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    DeclImplementation(impControl), objcDeclQualifier(OBJC_TQ_None),
15258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    MethodContext(static_cast<NamedDecl*>(contextDecl)),
1532e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor    MethodDeclType(T),
154f6414927e67e27d9324d8d179c5f7ea620443924Daniel Dunbar    ParamInfo(0), NumMethodParams(0),
1554111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner    EndLoc(endLoc), Body(0), SelfDecl(0), CmdDecl(0) {}
1568a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
1571c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual ~ObjCMethodDecl();
1581c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
1596c4ae5de0c356777446f823b573821fb95560d91Chris Lattnerpublic:
1608a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
1618a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  /// Destroy - Call destructors and release memory.
1628a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  virtual void Destroy(ASTContext& C);
1636c4ae5de0c356777446f823b573821fb95560d91Chris Lattner
1640ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCMethodDecl *Create(ASTContext &C,
1650ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                SourceLocation beginLoc,
1666c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                SourceLocation endLoc, Selector SelInfo,
1676c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                QualType T, Decl *contextDecl,
168f6414927e67e27d9324d8d179c5f7ea620443924Daniel Dunbar                                bool isInstance = true,
1696c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                bool isVariadic = false,
1704607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian                                bool isSynthesized = false,
171b06fa3b86951b9f179c99c3768331536c32e902dChris Lattner                                ImplementationControl impControl = None);
17258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
173ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  ObjCDeclQualifier getObjCDeclQualifier() const {
174ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek    return ObjCDeclQualifier(objcDeclQualifier);
175ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  }
176a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setObjCDeclQualifier(ObjCDeclQualifier QV) { objcDeclQualifier = QV; }
17758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
17858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Location information, modeled after the Stmt API.
17958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  SourceLocation getLocStart() const { return getLocation(); }
18058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
1819776ba0d844cf9f6888e871e3fd246ae782f76f4Daniel Dunbar  SourceRange getSourceRange() const {
1829776ba0d844cf9f6888e871e3fd246ae782f76f4Daniel Dunbar    return SourceRange(getLocation(), EndLoc);
1839776ba0d844cf9f6888e871e3fd246ae782f76f4Daniel Dunbar  }
18458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
18558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  NamedDecl *getMethodContext() const { return MethodContext; }
18658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
1875619688510185081cbb4621d703daf7ee24cf39aChris Lattner  ObjCInterfaceDecl *getClassInterface();
1885619688510185081cbb4621d703daf7ee24cf39aChris Lattner  const ObjCInterfaceDecl *getClassInterface() const {
1895619688510185081cbb4621d703daf7ee24cf39aChris Lattner    return const_cast<ObjCMethodDecl*>(this)->getClassInterface();
190e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  }
19158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
1922e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  Selector getSelector() const { return getDeclName().getObjCSelector(); }
193faf5e779d16bb4590f2a97e1c7ded255eddd90f3Fariborz Jahanian  unsigned getSynthesizedMethodSize() const;
19458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  QualType getResultType() const { return MethodDeclType; }
19558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
196d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  // Iterator access to formal parameters.
19758cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  unsigned param_size() const { return NumMethodParams; }
198d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  typedef ParmVarDecl **param_iterator;
199d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  typedef ParmVarDecl * const *param_const_iterator;
200d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_iterator param_begin() { return ParamInfo; }
201d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_iterator param_end() { return ParamInfo+param_size(); }
202d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_const_iterator param_begin() const { return ParamInfo; }
203d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner  param_const_iterator param_end() const { return ParamInfo+param_size(); }
204d57f635d520e8cb5b93d3d770ff58db06c62de54Chris Lattner
20558cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  unsigned getNumParams() const { return NumMethodParams; }
20658cce3b0dcbdcc95b7e713795834b4cb2c8a008aChris Lattner  ParmVarDecl *getParamDecl(unsigned i) const {
20758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    assert(i < getNumParams() && "Illegal param #");
20858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    return ParamInfo[i];
20958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
2102338d58a905191c4205d4f73affd0bdaa13493b4Fariborz Jahanian  void setParamDecl(int i, ParmVarDecl *pDecl) {
2112338d58a905191c4205d4f73affd0bdaa13493b4Fariborz Jahanian    ParamInfo[i] = pDecl;
2122338d58a905191c4205d4f73affd0bdaa13493b4Fariborz Jahanian  }
21358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  void setMethodParams(ParmVarDecl **NewParamInfo, unsigned NumParams);
2144111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner
215451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// createImplicitParams - Used to lazily create the self and cmd
216451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// implict parameters. This must be called prior to using getSelfDecl()
217451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// or getCmdDecl(). The call is ignored if the implicit paramters
218451318c08a6342c10b8986060386fd9274418437Daniel Dunbar  /// have already been created.
219fef30b55230064d334a669a065a1c9acdb87cdfeFariborz Jahanian  void createImplicitParams(ASTContext &Context, const ObjCInterfaceDecl *ID);
220451318c08a6342c10b8986060386fd9274418437Daniel Dunbar
2214111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl * getSelfDecl() const { return SelfDecl; }
2224111024be81e7c0525e42dadcc126d27e5bf2425Chris Lattner  ImplicitParamDecl * getCmdDecl() const { return CmdDecl; }
22358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
22458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool isInstance() const { return IsInstance; }
22558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  bool isVariadic() const { return IsVariadic; }
22658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
2274607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian  bool isSynthesized() const { return IsSynthesized; }
22891b51a92f2e9fc8025b6a9df88442840eb62823aFariborz Jahanian  void setIsSynthesized() { IsSynthesized = true; }
2294607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian
23058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Related to protocols declared in  @protocol
23158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  void setDeclImplementation(ImplementationControl ic) {
23258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    DeclImplementation = ic;
23358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
23458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  ImplementationControl getImplementationControl() const {
235ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek    return ImplementationControl(DeclImplementation);
23658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
237792481eec23d8c1aa92173be589e2ae9d02514a5Ted Kremenek
23869c8f0aab655257e9e532d9d53756acf4f7a2d78Ted Kremenek  virtual Stmt *getBody() const { return Body; }
23958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  void setBody(Stmt *B) { Body = B; }
24058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
24158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Implement isa/cast/dyncast/etc.
242a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCMethod; }
243a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCMethodDecl *D) { return true; }
24442220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  static DeclContext *castToDeclContext(const ObjCMethodDecl *D) {
24542220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis    return static_cast<DeclContext *>(const_cast<ObjCMethodDecl*>(D));
24642220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  }
24742220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  static ObjCMethodDecl *castFromDeclContext(const DeclContext *DC) {
24842220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis    return static_cast<ObjCMethodDecl *>(const_cast<DeclContext*>(DC));
24942220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  }
25058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff};
251b245a33e6f562377946dc9cb5694a2d3cb8a98a3Fariborz Jahanian
252a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCInterfaceDecl - Represents an ObjC class declaration. For example:
2530c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
2540c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   // MostPrimitive declares no super class (not particularly useful).
2550c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @interface MostPrimitive
2560c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     // no instance variables or methods.
2570c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @end
2580c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
259fd5de471478a507dd2495c4ea9bcab801ea5fa65Chris Lattner///   // NSResponder inherits from NSObject & implements NSCoding (a protocol).
2600c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @interface NSResponder : NSObject <NSCoding>
261a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek///   { // instance variables are represented by ObjCIvarDecl.
2620c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id nextResponder; // nextResponder instance variable.
2630c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   }
2640c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   - (NSResponder *)nextResponder; // return a pointer to NSResponder.
2650c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   - (void)mouseMoved:(NSEvent *)theEvent; // return void, takes a pointer
2660c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @end                                    // to an NSEvent.
2670c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
2680c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   Unlike C/C++, forward class declarations are accomplished with @class.
2690c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   Unlike C/C++, @class allows for a list of classes to be forward declared.
2700c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   Unlike C++, ObjC is a single-rooted class model. In Cocoa, classes
2710c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   typically inherit from NSObject (an exception is NSProxy).
2720c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
273b048c9835969c4f7fe06264748be18ed4b442116Chris Lattnerclass ObjCInterfaceDecl : public NamedDecl, public DeclContext {
2743110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  /// TypeForDecl - This indicates the Type object that represents this
2753110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  /// TypeDecl.  It is a cache maintained by ASTContext::getObjCInterfaceType
2763110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  Type *TypeForDecl;
2773110251f13981689f384eb3c0aba2afffea18d9dSteve Naroff  friend class ASTContext;
278980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
279980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Class's super class.
280a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *SuperClass;
281980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
282980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Protocols referenced in interface header declaration
2833db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  ObjCList<ObjCProtocolDecl> ReferencedProtocols;
284980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
285980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Ivars/NumIvars - This is a new[]'d array of pointers to Decls.
286a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCIvarDecl **Ivars;   // Null if not defined.
287f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned NumIvars;      // 0 if none.
288980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
289980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// instance methods
290a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **InstanceMethods;  // Null if not defined
291b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner  unsigned NumInstanceMethods;  // 0 if none.
292980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
293980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// class methods
294a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **ClassMethods;  // Null if not defined
29562db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned NumClassMethods;  // 0 if none
296980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
297980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// List of categories defined for this class.
298a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl *CategoryList;
29982a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
30082a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  /// class properties
301a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCPropertyDecl **PropertyDecl;  // Null if no property
30255d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner  unsigned NumPropertyDecl;  // 0 if none.
303980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
3043a3ca1b35a7121aea0bf465a192dce748465e10fFariborz Jahanian  bool ForwardDecl:1; // declared with @class.
3053a3ca1b35a7121aea0bf465a192dce748465e10fFariborz Jahanian  bool InternalInterface:1; // true - no @interface for @implementation
30660fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff
307d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation ClassLoc; // location of the class identifier.
308d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation SuperClassLoc; // location of the super class identifier.
309f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation EndLoc; // marks the '>', '}', or identifier.
310f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation AtEndLoc; // marks the end of the entire interface.
3110e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner
312b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner  ObjCInterfaceDecl(SourceLocation atLoc, IdentifierInfo *Id,
313b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner                    SourceLocation CLoc, bool FD, bool isInternal)
314b048c9835969c4f7fe06264748be18ed4b442116Chris Lattner    : NamedDecl(ObjCInterface, atLoc, Id), DeclContext(ObjCInterface),
3150ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner      TypeForDecl(0), SuperClass(0),
3163db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner      Ivars(0), NumIvars(0),
317b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner      InstanceMethods(0), NumInstanceMethods(0),
31862db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner      ClassMethods(0), NumClassMethods(0),
319f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner      CategoryList(0), PropertyDecl(0), NumPropertyDecl(0),
320d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff      ForwardDecl(FD), InternalInterface(isInternal),
321d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff      ClassLoc(CLoc) {
322980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff      }
3238a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
3241c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual ~ObjCInterfaceDecl();
3258a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
3260e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattnerpublic:
3270e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner
3288a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  /// Destroy - Call destructors and release memory.
3298a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek  virtual void Destroy(ASTContext& C);
3308a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
3310ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCInterfaceDecl *Create(ASTContext &C,
3320ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                   SourceLocation atLoc,
333d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff                                   IdentifierInfo *Id,
334d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff                                   SourceLocation ClassLoc = SourceLocation(),
3350e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner                                   bool ForwardDecl = false,
3360e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner                                   bool isInternal = false);
3373db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  const ObjCList<ObjCProtocolDecl> &getReferencedProtocols() const {
3387ed9e0f97f4645edc5d4670385b985ea4c617ce7Fariborz Jahanian    return ReferencedProtocols;
3397ed9e0f97f4645edc5d4670385b985ea4c617ce7Fariborz Jahanian  }
340980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
341559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCPropertyDecl *FindPropertyDeclaration(IdentifierInfo *PropertyId) const;
342559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCCategoryDecl *FindCategoryDeclaration(IdentifierInfo *CategoryId) const;
343559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCIvarDecl *FindIvarDeclaration(IdentifierInfo *IvarId) const;
34431afbf02a381ae9c77d225aa54f972d152838b3aFariborz Jahanian  bool isPropertyReadonly(ObjCPropertyDecl *PropertyDecl) const;
3453db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner
3463db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  typedef ObjCList<ObjCProtocolDecl>::iterator protocol_iterator;
3473db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  protocol_iterator protocol_begin() const {return ReferencedProtocols.begin();}
3483db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  protocol_iterator protocol_end() const { return ReferencedProtocols.end(); }
349aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian
350a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCIvarDecl * const *ivar_iterator;
351be6df088d69bca0e99c7845a6cd8c1ca85034f31Chris Lattner  ivar_iterator ivar_begin() const { return Ivars; }
3525564e07af5e62875f3b83fc2e5a8823588b5adeaChris Lattner  ivar_iterator ivar_end() const { return Ivars + ivar_size();}
353f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned ivar_size() const { return NumIvars; }
354a0fb5861dec7aa1da0d21d5759678d76b00464f4Ted Kremenek  bool ivar_empty() const { return NumIvars == 0; }
355be6df088d69bca0e99c7845a6cd8c1ca85034f31Chris Lattner
356b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner  unsigned getNumInstanceMethods() const { return NumInstanceMethods; }
35762db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned getNumClassMethods() const { return NumClassMethods; }
358980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
359a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * instmeth_iterator;
3604c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  instmeth_iterator instmeth_begin() const { return InstanceMethods; }
3614c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  instmeth_iterator instmeth_end() const {
362b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner    return InstanceMethods+NumInstanceMethods;
3634c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  }
3644c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner
365a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * classmeth_iterator;
3664c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  classmeth_iterator classmeth_begin() const { return ClassMethods; }
3674c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  classmeth_iterator classmeth_end() const {
36862db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return ClassMethods+NumClassMethods;
3694c52509ffb9fdd6c93aa7b50812e316f1d920a26Chris Lattner  }
370b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner
371b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner  /// addReferencedProtocols - Set the list of protocols that this interface
372b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner  /// implements.
373780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  void addReferencedProtocols(ObjCProtocolDecl *const*List, unsigned NumRPs) {
374780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    ReferencedProtocols.set(List, NumRPs);
3753db6cae19c236fe2cef343c90105ce7cca7de965Chris Lattner  }
376b752f289026ad8e5f44851b20e009a27ed61eefcChris Lattner
377a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addInstanceVariablesToClass(ObjCIvarDecl **ivars, unsigned numIvars,
37860fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                                   SourceLocation RBracLoc);
379980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
380a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addMethods(ObjCMethodDecl **insMethods, unsigned numInsMembers,
381a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                  ObjCMethodDecl **clsMethods, unsigned numClsMembers,
38260fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                  SourceLocation AtEnd);
383980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
38455d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner  void addProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
38555d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner
386aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian  void mergeProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
387aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian
38833de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  void addPropertyMethods(ASTContext &Context,
38933de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian                          ObjCPropertyDecl* Property,
390b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::SmallVector<ObjCMethodDecl*, 32> &insMethods,
391b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::DenseMap<Selector, const ObjCMethodDecl*> &InsMap);
39233de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
3933dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  typedef ObjCPropertyDecl * const * classprop_iterator;
3943dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_begin() const { return PropertyDecl; }
3953dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_end() const {
3963dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian    return PropertyDecl+NumPropertyDecl;
3973dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  }
39855d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner
399768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  bool isForwardDecl() const { return ForwardDecl; }
400768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  void setForwardDecl(bool val) { ForwardDecl = val; }
401980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
402a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *getSuperClass() const { return SuperClass; }
403a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setSuperClass(ObjCInterfaceDecl * superCls) { SuperClass = superCls; }
404980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
405a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl* getCategoryList() const { return CategoryList; }
406a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setCategoryList(ObjCCategoryDecl *category) {
40753efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    CategoryList = category;
408980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
40953efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner
41053efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  /// isSuperClassOf - Return true if this class is the specified class or is a
41153efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  /// super class of the specified interface class.
41253efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  bool isSuperClassOf(const ObjCInterfaceDecl *I) const {
41353efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    // If RHS is derived from LHS it is OK; else it is not OK.
41453efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    while (I != NULL) {
41553efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner      if (this == I)
41653efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner        return true;
41753efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner      I = I->getSuperClass();
41853efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    }
41953efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner    return false;
42053efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner  }
42153efc251792bf2c9c5f295bd3507facc51a1fe7eChris Lattner
42268a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner  ObjCIvarDecl *lookupInstanceVariable(IdentifierInfo *IVarName,
42368a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner                                       ObjCInterfaceDecl *&ClassDeclared);
42468a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner  ObjCIvarDecl *lookupInstanceVariable(IdentifierInfo *IVarName) {
42568a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner    ObjCInterfaceDecl *ClassDeclared;
42668a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner    return lookupInstanceVariable(IVarName, ClassDeclared);
42768a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner  }
42868a057b4292f5ff814ec8da53f6cda8cdcfbd2aeChris Lattner
429c81c8144a661a49d7b9dae8d2080dee2e43186ecChris Lattner
43058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local instance method declared in this interface.
4313216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const {
43258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (instmeth_iterator I = instmeth_begin(), E = instmeth_end();
4330e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner         I != E; ++I) {
43458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
43558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
43658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
4370e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner    return 0;
43858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
43958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local class method declared in this interface.
4403216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const {
44158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
4420e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner         I != E; ++I) {
44358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
44458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
44558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
4460e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner    return 0;
44758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
44894a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Lookup a method. First, we search locally. If a method isn't
44994a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // found, we search referenced protocols and class categories.
450a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupInstanceMethod(Selector Sel);
451a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupClassMethod(Selector Sel);
45260fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff
453f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  // Location information, modeled after the Stmt API.
45460fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff  SourceLocation getLocStart() const { return getLocation(); } // '@'interface
455f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
456f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
45760fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff
458d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation getClassLoc() const { return ClassLoc; }
459d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  void setSuperClassLoc(SourceLocation Loc) { SuperClassLoc = Loc; }
460d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff  SourceLocation getSuperClassLoc() const { return SuperClassLoc; }
461d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff
46260fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff  // We also need to record the @end location.
463f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff  SourceLocation getAtEndLoc() const { return AtEndLoc; }
46482a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
465f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner  unsigned getNumPropertyDecl() const { return NumPropertyDecl; }
46682a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
46755d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner  ObjCPropertyDecl * const * getPropertyDecl() const { return PropertyDecl; }
468a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCPropertyDecl **getPropertyDecl() { return PropertyDecl; }
469f908a87299d278164540f90b5b6e6cab7b14fb41Steve Naroff
4704b6df3fa953267c5d755628c8afd818bb571e579Fariborz Jahanian  /// ImplicitInterfaceDecl - check that this is an implicitely declared
471a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  /// ObjCInterfaceDecl node. This is for legacy objective-c @implementation
4724b6df3fa953267c5d755628c8afd818bb571e579Fariborz Jahanian  /// declaration without an @interface declaration.
4733a3ca1b35a7121aea0bf465a192dce748465e10fFariborz Jahanian  bool ImplicitInterfaceDecl() const { return InternalInterface; }
4744b6df3fa953267c5d755628c8afd818bb571e579Fariborz Jahanian
475a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCInterface; }
476a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCInterfaceDecl *D) { return true; }
47742220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  static DeclContext *castToDeclContext(const ObjCInterfaceDecl *D) {
47842220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis    return static_cast<DeclContext *>(const_cast<ObjCInterfaceDecl*>(D));
47942220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  }
48042220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  static ObjCInterfaceDecl *castFromDeclContext(const DeclContext *DC) {
48142220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis    return static_cast<ObjCInterfaceDecl *>(const_cast<DeclContext*>(DC));
48242220c5432c141d47cc8ce786e472b49dc907378Argyrios Kyrtzidis  }
483980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
484980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
485a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCIvarDecl - Represents an ObjC instance variable. In general, ObjC
4860c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// instance variables are identical to C. The only exception is Objective-C
4870c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// supports C++ style access control. For example:
4880c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
4890c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @interface IvarExample : NSObject
4900c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   {
491f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek///     id defaultToProtected;
4920c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @public:
4930c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id canBePublic; // same as C++.
4940c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @protected:
4950c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id canBeProtected; // same as C++.
4960c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   @package:
4970c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///     id canBePackage; // framework visibility (not available in C++).
4980c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///   }
4990c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
500a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCIvarDecl : public FieldDecl {
5010e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattnerpublic:
502980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  enum AccessControl {
503980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff    None, Private, Protected, Public, Package
504980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  };
505f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek
506b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenekprivate:
507b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek  ObjCIvarDecl(SourceLocation L, IdentifierInfo *Id, QualType T,
508b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek               AccessControl ac, Expr *BW)
5098f50b9c02abe7cd4945a9b77a8550a5207ea21cfTed Kremenek    : FieldDecl(ObjCIvar, L, Id, T, BW), DeclAccess(ac) {}
510b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek
511b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenekpublic:
512b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek  static ObjCIvarDecl *Create(ASTContext &C, SourceLocation L,
513b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek                              IdentifierInfo *Id, QualType T,
514b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek                              AccessControl ac, Expr *BW = NULL);
515b8db21d4df5fbb6ce1ace6411b82d3d623d789deTed Kremenek
516980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  void setAccessControl(AccessControl ac) { DeclAccess = ac; }
517f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek
518ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  AccessControl getAccessControl() const { return AccessControl(DeclAccess); }
519f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek
520f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek  AccessControl getCanonicalAccessControl() const {
521f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek    return DeclAccess == None ? Protected : AccessControl(DeclAccess);
522f079570fcad0d0053e75ebae29c883ec4276e020Ted Kremenek  }
523980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
524980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  // Implement isa/cast/dyncast/etc.
525a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCIvar; }
526a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCIvarDecl *D) { return true; }
527980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroffprivate:
528ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  // NOTE: VC++ treats enums as signed, avoid using the AccessControl enum
529ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned DeclAccess : 3;
530980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
531980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
53201e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
53301e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek/// ObjCAtDefsFieldDecl - Represents a field declaration created by an
53401e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek///  @defs(...).
53501e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenekclass ObjCAtDefsFieldDecl : public FieldDecl {
53601e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenekprivate:
53701e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  ObjCAtDefsFieldDecl(SourceLocation L, IdentifierInfo *Id,
53801e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek                      QualType T, Expr *BW)
53901e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  : FieldDecl(ObjCAtDefsField, L, Id, T, BW) {}
54001e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
54101e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenekpublic:
54201e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  static ObjCAtDefsFieldDecl *Create(ASTContext &C, SourceLocation L,
54301e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek                                     IdentifierInfo *Id, QualType T,
54401e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek                                     Expr *BW);
54501e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
54601e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  virtual void Destroy(ASTContext& C);
54701e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
54801e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  // Implement isa/cast/dyncast/etc.
54901e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCAtDefsField; }
55001e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek  static bool classof(const ObjCAtDefsFieldDecl *D) { return true; }
55101e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek};
552980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
553a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCProtocolDecl - Represents a protocol declaration. ObjC protocols
5540c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// declare a pure abstract type (i.e no instance variables are permitted).
5550c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Protocols orginally drew inspiration from C++ pure virtual functions (a C++
5560c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// feature with nice semantics and lousy syntax:-). Here is an example:
5570c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
558eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// @protocol NSDraggingInfo <refproto1, refproto2>
5590c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - (NSWindow *)draggingDestinationWindow;
5600c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - (NSImage *)draggedImage;
5610c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
5620c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
563eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// This says that NSDraggingInfo requires two methods and requires everything
564eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// that the two "referenced protocols" 'refproto1' and 'refproto2' require as
565eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner/// well.
566eca7be6b7ebd93682eeaab2c71d59f2995dacdccChris Lattner///
5670c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @interface ImplementsNSDraggingInfo : NSObject <NSDraggingInfo>
5680c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
5690c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
570a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjC protocols inspired Java interfaces. Unlike Java, ObjC classes and
5710c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// protocols are in distinct namespaces. For example, Cocoa defines both
5720c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// an NSObject protocol and class (which isn't allowed in Java). As a result,
5730c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// protocols are referenced using angle brackets as follows:
5740c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
5750c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// id <NSDraggingInfo> anyObjectThatImplementsNSDraggingInfo;
5760c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
577a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCProtocolDecl : public NamedDecl {
578780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  /// Referenced protocols
579780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  ObjCList<ObjCProtocolDecl> ReferencedProtocols;
580980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
581980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// protocol instance methods
582a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **InstanceMethods;  // Null if not defined
58362db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned NumInstanceMethods;  // 0 if none
584980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
585980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// protocol class methods
586a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **ClassMethods;  // Null if not defined
58762db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned NumClassMethods;  // 0 if none
5883dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
5893dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  /// protocol properties
5903dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  ObjCPropertyDecl **PropertyDecl;  // Null if no property
5913dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  unsigned NumPropertyDecl;  // 0 if none
5923dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
593980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  bool isForwardProtoDecl; // declared with @protocol.
594423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
595423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation EndLoc; // marks the '>' or identifier.
596423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation AtEndLoc; // marks the end of the entire interface.
597cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner
598780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  ObjCProtocolDecl(SourceLocation L, IdentifierInfo *Id)
599a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCProtocol, L, Id),
60062db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner      InstanceMethods(0), NumInstanceMethods(0),
60162db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner      ClassMethods(0), NumClassMethods(0),
6023dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian      PropertyDecl(0), NumPropertyDecl(0),
603c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner      isForwardProtoDecl(true) {
604cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner  }
6051c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
6061c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual ~ObjCProtocolDecl();
6071c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
608cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattnerpublic:
6091c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
6101c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  /// Destroy - Call destructors and release memory.
6111c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek  virtual void Destroy(ASTContext& C);
6121c8a413c1e00636c77666ddf1e3b0311f3fa8c81Ted Kremenek
613cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner  static ObjCProtocolDecl *Create(ASTContext &C, SourceLocation L,
614780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner                                  IdentifierInfo *Id);
615cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner
616a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addMethods(ObjCMethodDecl **insMethods, unsigned numInsMembers,
617a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                  ObjCMethodDecl **clsMethods, unsigned numClsMembers,
61860fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                  SourceLocation AtEndLoc);
619980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
620780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  const ObjCList<ObjCProtocolDecl> &getReferencedProtocols() const {
621780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    return ReferencedProtocols;
622980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
623dbc933701d20918add13b6a3c9d47ff8c75419cfDaniel Dunbar  typedef ObjCList<ObjCProtocolDecl>::iterator protocol_iterator;
624780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_begin() const {return ReferencedProtocols.begin();}
625780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_end() const { return ReferencedProtocols.end(); }
626780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
627780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  /// addReferencedProtocols - Set the list of protocols that this interface
628780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  /// implements.
629780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  void addReferencedProtocols(ObjCProtocolDecl *const*List, unsigned NumRPs) {
630780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    ReferencedProtocols.set(List, NumRPs);
631aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian  }
632aebf0cba02c014ac8b19d615c654248e0e93779fFariborz Jahanian
633c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner  unsigned getNumInstanceMethods() const { return NumInstanceMethods; }
63462db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned getNumClassMethods() const { return NumClassMethods; }
6353d2c22b1d334fa74d26a5f21841cb55df5dfdd1aSteve Naroff
6363d2c22b1d334fa74d26a5f21841cb55df5dfdd1aSteve Naroff  ObjCPropertyDecl *FindPropertyDeclaration(IdentifierInfo *PropertyId) const;
6373dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6383dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  unsigned getNumPropertyDecl() const { return NumPropertyDecl; }
6393dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6403dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  ObjCPropertyDecl * const * getPropertyDecl() const { return PropertyDecl; }
6413dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  ObjCPropertyDecl **getPropertyDecl() { return PropertyDecl; }
6423dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6433dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  void addProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
6447b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar
6457b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar  void addPropertyMethods(ASTContext &Context,
6467b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar                          ObjCPropertyDecl* Property,
647b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::SmallVector<ObjCMethodDecl*, 32> &insMethods,
648b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::DenseMap<Selector, const ObjCMethodDecl*> &InsMap);
6493dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
6503dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  typedef ObjCPropertyDecl * const * classprop_iterator;
6513dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_begin() const { return PropertyDecl; }
6523dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_end() const {
6533dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian    return PropertyDecl+NumPropertyDecl;
6543dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  }
65558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
656a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * instmeth_iterator;
65758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_begin() const { return InstanceMethods; }
65858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_end() const {
65962db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return InstanceMethods+NumInstanceMethods;
66058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
66158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
662a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * classmeth_iterator;
66358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_begin() const { return ClassMethods; }
66458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_end() const {
66562db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return ClassMethods+NumClassMethods;
66658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
66758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
66858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local instance method declared in this interface.
6693216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const {
67062db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    for (instmeth_iterator I = instmeth_begin(), E = instmeth_end();
67162db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner         I != E; ++I) {
67258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
67358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
67458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
67562db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return 0;
67658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
67758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local class method declared in this interface.
6783216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const {
67958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
68062db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner         I != E; ++I) {
68158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
68258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
68358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
68462db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner    return 0;
68558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
686980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
68794a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Lookup a method. First, we search locally. If a method isn't
68894a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // found, we search referenced protocols and class categories.
689a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupInstanceMethod(Selector Sel);
690a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl *lookupClassMethod(Selector Sel);
6917dd82836dca87cf828ce994f161b53a34f6cdb7eFariborz Jahanian
692768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  bool isForwardDecl() const { return isForwardProtoDecl; }
693768f26ee5892cd63ff0335a15d71a2385ba7c5eaSteve Naroff  void setForwardDecl(bool val) { isForwardProtoDecl = val; }
694980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
695423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // Location information, modeled after the Stmt API.
696423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocStart() const { return getLocation(); } // '@'protocol
697423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
698423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
699423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
700423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // We also need to record the @end location.
701423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getAtEndLoc() const { return AtEndLoc; }
702423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
703a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCProtocol; }
704a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCProtocolDecl *D) { return true; }
705980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
706980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
707a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCClassDecl - Specifies a list of forward class declarations. For example:
70806ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff///
70906ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff/// @class NSCursor, NSImage, NSPasteboard, NSWindow;
7100c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
711a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCClassDecl : public Decl {
712a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl **ForwardDecls;
7137e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner  unsigned NumForwardDecls;
71461f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
715a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCClassDecl(SourceLocation L, ObjCInterfaceDecl **Elts, unsigned nElts)
716a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : Decl(ObjCClass, L) {
71706ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    if (nElts) {
718a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      ForwardDecls = new ObjCInterfaceDecl*[nElts];
719a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      memcpy(ForwardDecls, Elts, nElts*sizeof(ObjCInterfaceDecl*));
7207e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner    } else {
7217e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner      ForwardDecls = 0;
72206ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    }
72306ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    NumForwardDecls = nElts;
72406ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
725400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
726400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  virtual ~ObjCClassDecl();
727400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
72861f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattnerpublic:
729400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
730400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  /// Destroy - Call destructors and release memory.
731400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  virtual void Destroy(ASTContext& C);
732400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
73361f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner  static ObjCClassDecl *Create(ASTContext &C, SourceLocation L,
73461f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner                               ObjCInterfaceDecl **Elts, unsigned nElts);
73561f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
736a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setInterfaceDecl(unsigned idx, ObjCInterfaceDecl *OID) {
7377e620729fcb5a0042107f999dcf524f7357ce819Chris Lattner    assert(idx < NumForwardDecls && "index out of range");
73806ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    ForwardDecls[idx] = OID;
73906ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
740a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl** getForwardDecls() const { return ForwardDecls; }
7417e15891fc89256fc013bd1003676ad3197b85c25Steve Naroff  int getNumForwardDecls() const { return NumForwardDecls; }
7427e15891fc89256fc013bd1003676ad3197b85c25Steve Naroff
743400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  typedef ObjCInterfaceDecl * const * iterator;
744400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  iterator begin() const { return ForwardDecls; }
745400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek  iterator end() const { return ForwardDecls+NumForwardDecls; }
746400d95fb7bb9fac609f8613862b84f3a2a7d510fTed Kremenek
747a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCClass; }
748a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCClassDecl *D) { return true; }
74906ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff};
75006ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff
751a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCForwardProtocolDecl - Specifies a list of forward protocol declarations.
75206ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff/// For example:
75306ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff///
7540c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @protocol NSTextInput, NSChangeSpelling, NSDraggingInfo;
7550c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
756a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCForwardProtocolDecl : public Decl {
757a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCProtocolDecl **ReferencedProtocols;
7589fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  unsigned NumReferencedProtocols;
75961f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
7600ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  ObjCForwardProtocolDecl(SourceLocation L,
761a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                          ObjCProtocolDecl **Elts, unsigned nElts)
762a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  : Decl(ObjCForwardProtocol, L) {
763b97de3eddf0a0f7ff97204e7def6b7e0aa8953f2Chris Lattner    NumReferencedProtocols = nElts;
76406ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff    if (nElts) {
765a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      ReferencedProtocols = new ObjCProtocolDecl*[nElts];
766a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek      memcpy(ReferencedProtocols, Elts, nElts*sizeof(ObjCProtocolDecl*));
7679fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    } else {
7689fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner      ReferencedProtocols = 0;
769980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff    }
77006ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
77105ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek
77205ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  virtual ~ObjCForwardProtocolDecl();
77305ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek
77461f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattnerpublic:
77561f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner  static ObjCForwardProtocolDecl *Create(ASTContext &C, SourceLocation L,
77661f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner                                         ObjCProtocolDecl **Elts, unsigned Num);
77761f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
77861f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
779a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setForwardProtocolDecl(unsigned idx, ObjCProtocolDecl *OID) {
7809fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    assert(idx < NumReferencedProtocols && "index out of range");
7817ed9e0f97f4645edc5d4670385b985ea4c617ce7Fariborz Jahanian    ReferencedProtocols[idx] = OID;
78206ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
7839fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner
7849fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  unsigned getNumForwardDecls() const { return NumReferencedProtocols; }
7859fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner
786a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCProtocolDecl *getForwardProtocolDecl(unsigned idx) {
7879fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    assert(idx < NumReferencedProtocols && "index out of range");
7889fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    return ReferencedProtocols[idx];
7899fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  }
790a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  const ObjCProtocolDecl *getForwardProtocolDecl(unsigned idx) const {
7919fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    assert(idx < NumReferencedProtocols && "index out of range");
7929fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner    return ReferencedProtocols[idx];
7939fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner  }
7949fa5e65d08aee1875c5f2a841c8b0b4069bd00e5Chris Lattner
79505ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  typedef ObjCProtocolDecl * const * iterator;
79605ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  iterator begin() const { return ReferencedProtocols; }
79705ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek  iterator end() const { return ReferencedProtocols+NumReferencedProtocols; }
79805ac3ef08f9d06e0a4439073c9edabf7f912f946Ted Kremenek
79906ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  static bool classof(const Decl *D) {
800a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    return D->getKind() == ObjCForwardProtocol;
80106ae8d68ef258ccd40c9cd1ce762eaae6f3d4432Steve Naroff  }
802a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCForwardProtocolDecl *D) { return true; }
803980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
804980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
805a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCCategoryDecl - Represents a category declaration. A category allows
8060c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// you to add methods to an existing class (without subclassing or modifying
8070c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// the original class interface or implementation:-). Categories don't allow
8080c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// you to add instance data. The following example adds "myMethod" to all
8090c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// NSView's within a process:
8100c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
8110c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @interface NSView (MyViewMethods)
8120c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - myMethod;
8130c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
8140c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
8150c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Cateogries also allow you to split the implementation of a class across
8160c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// several files (a feature more naturally supported in C++).
8170c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
8180c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Categories were originally inspired by dynamic languages such as Common
81968c82cf61228102aba1194efef222fa1478af2a8Chris Lattner/// Lisp and Smalltalk.  More traditional class-based languages (C++, Java)
8200c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// don't support this level of dynamism, which is both powerful and dangerous.
8210c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
822a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCCategoryDecl : public NamedDecl {
823980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Interface belonging to this category
824a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *ClassInterface;
825980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
82668c82cf61228102aba1194efef222fa1478af2a8Chris Lattner  /// referenced protocols in this category.
827780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  ObjCList<ObjCProtocolDecl> ReferencedProtocols;
828980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
829980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// category instance methods
830a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **InstanceMethods;  // Null if not defined
831c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner  unsigned NumInstanceMethods;  // 0 if none
832980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
833980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// category class methods
834a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCMethodDecl **ClassMethods;  // Null if not defined
835a906135721c350435319347d2672bbb3bf494f91Chris Lattner  unsigned NumClassMethods;  // 0 if not defined
836980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
837980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Next category belonging to this class
838a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl *NextClassCategory;
839ca3adf7e8cac8c9fbaf592b1e5c2be6f082de7baFariborz Jahanian
8407e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  /// category properties
8417e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  ObjCPropertyDecl **PropertyDecl;  // Null if no property
8427e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  unsigned NumPropertyDecl;  // 0 if none
8437e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian
844423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation EndLoc; // marks the '>' or identifier.
845423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation AtEndLoc; // marks the end of the entire interface.
84661f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
84768c82cf61228102aba1194efef222fa1478af2a8Chris Lattner  ObjCCategoryDecl(SourceLocation L, IdentifierInfo *Id)
848a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCCategory, L, Id),
849780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner      ClassInterface(0),
850c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner      InstanceMethods(0), NumInstanceMethods(0),
851a906135721c350435319347d2672bbb3bf494f91Chris Lattner      ClassMethods(0), NumClassMethods(0),
8527e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian      NextClassCategory(0), PropertyDecl(0),  NumPropertyDecl(0) {
853a906135721c350435319347d2672bbb3bf494f91Chris Lattner  }
85461f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattnerpublic:
85561f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
8560ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCCategoryDecl *Create(ASTContext &C,
8570ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                  SourceLocation L, IdentifierInfo *Id);
85861f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
859e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
860e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
861a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void setClassInterface(ObjCInterfaceDecl *IDecl) { ClassInterface = IDecl; }
862980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
863f7b2c98c16dfb2261ea57d40a1d5bc4738e73175Chris Lattner  /// addReferencedProtocols - Set the list of protocols that this interface
864f7b2c98c16dfb2261ea57d40a1d5bc4738e73175Chris Lattner  /// implements.
865780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  void addReferencedProtocols(ObjCProtocolDecl *const*List, unsigned NumRPs) {
866780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    ReferencedProtocols.set(List, NumRPs);
867780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  }
868980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
869780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  const ObjCList<ObjCProtocolDecl> &getReferencedProtocols() const {
870780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner    return ReferencedProtocols;
8718f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian  }
872780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
873780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  typedef ObjCProtocolDecl * const * protocol_iterator;
874780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_begin() const {return ReferencedProtocols.begin();}
875780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner  protocol_iterator protocol_end() const { return ReferencedProtocols.end(); }
876780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
877780f329cb011bff0da5763e2e9744eec093d0509Chris Lattner
878c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner  unsigned getNumInstanceMethods() const { return NumInstanceMethods; }
879a906135721c350435319347d2672bbb3bf494f91Chris Lattner  unsigned getNumClassMethods() const { return NumClassMethods; }
88058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
8817e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  unsigned getNumPropertyDecl() const { return NumPropertyDecl; }
8827e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian
8837e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian  ObjCPropertyDecl * const * getPropertyDecl() const { return PropertyDecl; }
8847e7e3872b584bc5e7de7a34c8b9c092032303b72Fariborz Jahanian
8853dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  void addProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
8867b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar
8871ac2bc44781ec8451f880dcf586768a71824d3a6Fariborz Jahanian  void mergeProperties(ObjCPropertyDecl **Properties, unsigned NumProperties);
8881ac2bc44781ec8451f880dcf586768a71824d3a6Fariborz Jahanian
8897b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar  void addPropertyMethods(ASTContext &Context,
8907b4732296b5c7e1a53021a8dbb97f39dedbd961fDaniel Dunbar                          ObjCPropertyDecl* Property,
891b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::SmallVector<ObjCMethodDecl*, 32> &insMethods,
892b85cce6498c8c1c20f701571d85a3b2fe53338ebFariborz Jahanian                          llvm::DenseMap<Selector, const ObjCMethodDecl*> &InsMap);
8933dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
894559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  ObjCPropertyDecl *FindPropertyDeclaration(IdentifierInfo *PropertyId) const;
895559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian
8963dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  typedef ObjCPropertyDecl * const * classprop_iterator;
8973dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_begin() const { return PropertyDecl; }
8983dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  classprop_iterator classprop_end() const {
8993dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian    return PropertyDecl+NumPropertyDecl;
9003dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian  }
9013dd4ba4068e953125b95ce85c723322cdd0a3cb5Fariborz Jahanian
902a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * instmeth_iterator;
90358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_begin() const { return InstanceMethods; }
90458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  instmeth_iterator instmeth_end() const {
905c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner    return InstanceMethods+NumInstanceMethods;
90658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
90758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
908a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCMethodDecl * const * classmeth_iterator;
90958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_begin() const { return ClassMethods; }
91058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  classmeth_iterator classmeth_end() const {
911a906135721c350435319347d2672bbb3bf494f91Chris Lattner    return ClassMethods+NumClassMethods;
91258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
91358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff
91458dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local instance method declared in this interface.
9153216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const {
91658dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (instmeth_iterator I = instmeth_begin(), E = instmeth_end();
91768c82cf61228102aba1194efef222fa1478af2a8Chris Lattner         I != E; ++I) {
91858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
91958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
92058dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
92168c82cf61228102aba1194efef222fa1478af2a8Chris Lattner    return 0;
92258dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
92358dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  // Get the local class method declared in this interface.
9243216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const {
92558dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
92668c82cf61228102aba1194efef222fa1478af2a8Chris Lattner         I != E; ++I) {
92758dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff      if ((*I)->getSelector() == Sel)
92858dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff        return *I;
92958dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff    }
93068c82cf61228102aba1194efef222fa1478af2a8Chris Lattner    return 0;
93158dbdeb69c063f82d644504fc638120198f7fad2Steve Naroff  }
9328f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
933a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addMethods(ObjCMethodDecl **insMethods, unsigned numInsMembers,
934a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                  ObjCMethodDecl **clsMethods, unsigned numClsMembers,
93560fcceeedbfc8b4a99cb942e2bc5aeb9e2f92a1fSteve Naroff                  SourceLocation AtEndLoc);
936980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
937a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryDecl *getNextClassCategory() const { return NextClassCategory; }
938980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  void insertNextClassCategory() {
9393d58138992b9bc7b34aaa680f3ddf3971292eb7dSteve Naroff    NextClassCategory = ClassInterface->getCategoryList();
9403d58138992b9bc7b34aaa680f3ddf3971292eb7dSteve Naroff    ClassInterface->setCategoryList(this);
941980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
942423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // Location information, modeled after the Stmt API.
943423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocStart() const { return getLocation(); } // '@'interface
944423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
945423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
946423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff
947423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  // We also need to record the @end location.
948423cb565abc681b770fb4b9b4bc24d398c98157bSteve Naroff  SourceLocation getAtEndLoc() const { return AtEndLoc; }
949ca3adf7e8cac8c9fbaf592b1e5c2be6f082de7baFariborz Jahanian
950a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCCategory; }
951a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCCategoryDecl *D) { return true; }
952980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
9530c6b6243d3efd958c17943130e2a773653511edcSteve Naroff
954a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCCategoryImplDecl - An object of this class encapsulates a category
955559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @implementation declaration. If a category class has declaration of a
956559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// property, its implementation must be specified in the category's
957559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @implementation declaration. Example:
958559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @interface I @end
959559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @interface I(CATEGORY)
960559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian///    @property int p1, d1;
961559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @end
962559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @implementation I(CATEGORY)
963559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian///  @dynamic p1,d1;
964559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian/// @end
965559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian///
966a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCCategoryImplDecl : public NamedDecl {
9678f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian  /// Class interface for this category implementation
968a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *ClassInterface;
9698f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
970e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  /// implemented instance methods
971a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> InstanceMethods;
9728f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
973e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  /// implemented class methods
974a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> ClassMethods;
975f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
976559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  /// Property Implementations in this category
977f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  llvm::SmallVector<ObjCPropertyImplDecl*, 8> PropertyImplementations;
978e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
979e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  SourceLocation EndLoc;
98075c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner
981a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCategoryImplDecl(SourceLocation L, IdentifierInfo *Id,
982a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                       ObjCInterfaceDecl *classInterface)
983a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCCategoryImpl, L, Id), ClassInterface(classInterface) {}
98475c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattnerpublic:
9850ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCCategoryImplDecl *Create(ASTContext &C,
9860ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                      SourceLocation L, IdentifierInfo *Id,
98775c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner                                      ObjCInterfaceDecl *classInterface);
9888f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
989e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
990e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
9918f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
992ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  unsigned getNumInstanceMethods() const { return InstanceMethods.size(); }
993ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  unsigned getNumClassMethods() const { return ClassMethods.size(); }
994e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
995a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addInstanceMethod(ObjCMethodDecl *method) {
996e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff    InstanceMethods.push_back(method);
997e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  }
998a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addClassMethod(ObjCMethodDecl *method) {
999e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff    ClassMethods.push_back(method);
1000f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  }
100194a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the instance method definition for this implementation.
10023216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const;
100394a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff
100494a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the class method definition for this implementation.
10053216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const;
1006f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
1007f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  void addPropertyImplementation(ObjCPropertyImplDecl *property) {
1008f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian    PropertyImplementations.push_back(property);
1009f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  }
1010e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
1011ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian  ObjCPropertyImplDecl *FindPropertyImplDecl(IdentifierInfo *propertyId) const;
1012ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian  ObjCPropertyImplDecl *FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const;
1013ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian
1014f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  unsigned getNumPropertyImplementations() const
1015f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  { return PropertyImplementations.size(); }
1016f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
1017559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian
1018559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  typedef llvm::SmallVector<ObjCPropertyImplDecl*, 8>::const_iterator
1019559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    propimpl_iterator;
1020559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_begin() const {
1021559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.begin();
1022559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1023559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_end() const {
1024559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.end();
1025559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1026559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian
1027a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
1028ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner    instmeth_iterator;
1029ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  instmeth_iterator instmeth_begin() const { return InstanceMethods.begin(); }
1030ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  instmeth_iterator instmeth_end() const { return InstanceMethods.end(); }
1031ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner
1032a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
1033ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner    classmeth_iterator;
1034ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  classmeth_iterator classmeth_begin() const { return ClassMethods.begin(); }
1035ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner  classmeth_iterator classmeth_end() const { return ClassMethods.end(); }
1036ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner
1037ab4c4d5e5ececa77aae7e291fafcba3049319cdcChris Lattner
1038e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  // Location information, modeled after the Stmt API.
1039e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  SourceLocation getLocStart() const { return getLocation(); }
1040e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
1041e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
1042e1e6c0d5c79c0ee7ed62fef47a19aa7ecef40db4Steve Naroff
1043a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const Decl *D) { return D->getKind() == ObjCCategoryImpl;}
1044a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCCategoryImplDecl *D) { return true; }
10458f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian};
10468f3fde00ad4d4f943321e338b914ae4740711c84Fariborz Jahanian
1047a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCImplementationDecl - Represents a class definition - this is where
10480c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// method definitions are specified. For example:
10490c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
10500c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @implementation MyClass
10510c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// - (void)myMethod { /* do something */ }
10520c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// @end
10530c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
10540c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// Typically, instance variables are specified in the class interface,
10550c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// *not* in the implemenentation. Nevertheless (for legacy reasons), we
10560c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// allow instance variables to be specified in the implementation. When
10570c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// specified, they need to be *identical* to the interface. Now that we
10580c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// have support for non-fragile ivars in ObjC 2.0, we can consider removing
10590c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// the legacy semantics and allow developers to move private ivar declarations
10600c6b6243d3efd958c17943130e2a773653511edcSteve Naroff/// from the class interface to the class implementation (but I digress:-)
10610c6b6243d3efd958c17943130e2a773653511edcSteve Naroff///
1062a526c5c67e5a0473c340903ee542ce570119665fTed Kremenekclass ObjCImplementationDecl : public NamedDecl {
1063dae1a1a2aa4f245b1958dc8db6089f24c575ef13Fariborz Jahanian  /// Class interface for this implementation
1064a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *ClassInterface;
1065f4d331dd922f92478ebf30e808c0ca97ce49418bFariborz Jahanian
1066980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Implementation Class's super class.
1067a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *SuperClass;
1068980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1069980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// Optional Ivars/NumIvars - This is a new[]'d array of pointers to Decls.
1070a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCIvarDecl **Ivars;   // Null if not specified
1071f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned NumIvars;      // 0 if none.
10720416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1073980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// implemented instance methods
1074a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> InstanceMethods;
10750416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1076980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  /// implemented class methods
1077a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  llvm::SmallVector<ObjCMethodDecl*, 32> ClassMethods;
10780416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1079f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  /// Propertys' being implemented
1080f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  llvm::SmallVector<ObjCPropertyImplDecl*, 8> PropertyImplementations;
1081f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
10820416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  SourceLocation EndLoc;
108375c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner
1084a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCImplementationDecl(SourceLocation L, IdentifierInfo *Id,
1085a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                         ObjCInterfaceDecl *classInterface,
1086a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek                         ObjCInterfaceDecl *superDecl)
1087a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    : NamedDecl(ObjCImplementation, L, Id),
10880416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff      ClassInterface(classInterface), SuperClass(superDecl),
1089f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner      Ivars(0), NumIvars(0) {}
109075c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattnerpublic:
10910ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCImplementationDecl *Create(ASTContext &C,
10920ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                        SourceLocation L, IdentifierInfo *Id,
109375c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner                                        ObjCInterfaceDecl *classInterface,
109475c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner                                        ObjCInterfaceDecl *superDecl);
109575c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner
1096980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1097a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void ObjCAddInstanceVariablesToClassImpl(ObjCIvarDecl **ivars,
1098980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff                                           unsigned numIvars);
1099980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1100a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addInstanceMethod(ObjCMethodDecl *method) {
11010416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff    InstanceMethods.push_back(method);
11020416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  }
1103a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  void addClassMethod(ObjCMethodDecl *method) {
11040416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff    ClassMethods.push_back(method);
11050416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  }
1106f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
1107f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  void addPropertyImplementation(ObjCPropertyImplDecl *property) {
1108f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian    PropertyImplementations.push_back(property);
1109f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  }
1110ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian
1111ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian  ObjCPropertyImplDecl *FindPropertyImplDecl(IdentifierInfo *propertyId) const;
1112ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian  ObjCPropertyImplDecl *FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const;
1113ae6f6fd1527a1da84679a6f0439dec3bbbd6ca7bFariborz Jahanian
1114559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  typedef llvm::SmallVector<ObjCPropertyImplDecl*, 8>::const_iterator
1115559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator;
1116559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_begin() const {
1117559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.begin();
1118559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1119559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  propimpl_iterator propimpl_end() const {
1120559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian    return PropertyImplementations.end();
1121559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  }
1122f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
11230416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  // Location information, modeled after the Stmt API.
11240416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  SourceLocation getLocStart() const { return getLocation(); }
11250416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  SourceLocation getLocEnd() const { return EndLoc; }
11260416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff  void setLocEnd(SourceLocation LE) { EndLoc = LE; };
11270416fb9f379b49abb3eb0c1cb2ca75107e5a71d1Steve Naroff
1128e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
1129e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
1130e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  const ObjCInterfaceDecl *getSuperClass() const { return SuperClass; }
1131e0def7589a8afa8a6acef13476bb3f882c104b91Chris Lattner  ObjCInterfaceDecl *getSuperClass() { return SuperClass; }
1132980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1133f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  void setSuperClass(ObjCInterfaceDecl * superCls) { SuperClass = superCls; }
1134980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1135b5feb35360526da42ad7e3fa7ef3d44a5992b83cChris Lattner  unsigned getNumInstanceMethods() const { return InstanceMethods.size(); }
113662db2f4214c1589082960f12c9cb8924fe0cf8c5Chris Lattner  unsigned getNumClassMethods() const { return ClassMethods.size(); }
1137f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian
1138f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian  unsigned getNumPropertyImplementations() const
1139f624f8186d8fe474350051c6d3f00b2c204fbeaeFariborz Jahanian    { return PropertyImplementations.size(); }
1140c43d868355374d48296ad3be2c9c536698a5e9a8Steve Naroff
1141a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
11420157c5144513438bb74aebf50d18f95df4104acbChris Lattner       instmeth_iterator;
11430157c5144513438bb74aebf50d18f95df4104acbChris Lattner  instmeth_iterator instmeth_begin() const { return InstanceMethods.begin(); }
11440157c5144513438bb74aebf50d18f95df4104acbChris Lattner  instmeth_iterator instmeth_end() const { return InstanceMethods.end(); }
11450157c5144513438bb74aebf50d18f95df4104acbChris Lattner
1146a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef llvm::SmallVector<ObjCMethodDecl*, 32>::const_iterator
11470157c5144513438bb74aebf50d18f95df4104acbChris Lattner    classmeth_iterator;
11480157c5144513438bb74aebf50d18f95df4104acbChris Lattner  classmeth_iterator classmeth_begin() const { return ClassMethods.begin(); }
11490157c5144513438bb74aebf50d18f95df4104acbChris Lattner  classmeth_iterator classmeth_end() const { return ClassMethods.end(); }
11500157c5144513438bb74aebf50d18f95df4104acbChris Lattner
115194a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the instance method definition for this implementation.
11523216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getInstanceMethod(Selector Sel) const;
11530157c5144513438bb74aebf50d18f95df4104acbChris Lattner
115494a5c3334bba3cc8cd1da85ba1118bc2c080add9Steve Naroff  // Get the class method definition for this implementation.
11553216dcdebb8ae0f2993ac5f5249caa217444bacfDaniel Dunbar  ObjCMethodDecl *getClassMethod(Selector Sel) const;
11560157c5144513438bb74aebf50d18f95df4104acbChris Lattner
1157a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  typedef ObjCIvarDecl * const *ivar_iterator;
11580157c5144513438bb74aebf50d18f95df4104acbChris Lattner  ivar_iterator ivar_begin() const { return Ivars; }
1159f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  ivar_iterator ivar_end() const { return Ivars+NumIvars; }
1160f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  unsigned ivar_size() const { return NumIvars; }
1161f3a7af901231535191d14cb524d58f22907ac4d2Chris Lattner  bool ivar_empty() const { return NumIvars == 0; }
11620157c5144513438bb74aebf50d18f95df4104acbChris Lattner
1163980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  static bool classof(const Decl *D) {
1164a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek    return D->getKind() == ObjCImplementation;
1165980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff  }
1166a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCImplementationDecl *D) { return true; }
1167980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff};
1168243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian
1169a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek/// ObjCCompatibleAliasDecl - Represents alias of a class. This alias is
1170243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian/// declared as @compatibility_alias alias class.
1171e8043c39176e7f253fbd92982b077eca6bf2fd59Steve Naroffclass ObjCCompatibleAliasDecl : public NamedDecl {
1172243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian  /// Class that this is an alias of.
1173a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCInterfaceDecl *AliasedClass;
1174243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian
1175a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  ObjCCompatibleAliasDecl(SourceLocation L, IdentifierInfo *Id,
1176e8043c39176e7f253fbd92982b077eca6bf2fd59Steve Naroff                          ObjCInterfaceDecl* aliasedClass)
1177e8043c39176e7f253fbd92982b077eca6bf2fd59Steve Naroff    : NamedDecl(ObjCCompatibleAlias, L, Id), AliasedClass(aliasedClass) {}
1178f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattnerpublic:
11790ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner  static ObjCCompatibleAliasDecl *Create(ASTContext &C,
11800ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                         SourceLocation L, IdentifierInfo *Id,
1181f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner                                         ObjCInterfaceDecl* aliasedClass);
1182f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner
1183f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner  const ObjCInterfaceDecl *getClassInterface() const { return AliasedClass; }
1184f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner  ObjCInterfaceDecl *getClassInterface() { return AliasedClass; }
1185980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
1186243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian  static bool classof(const Decl *D) {
11878a934233d1582b5bde9d270bc0705aa81e471a79Chris Lattner    return D->getKind() == ObjCCompatibleAlias;
1188243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian  }
1189a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCCompatibleAliasDecl *D) { return true; }
1190243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian
1191243b64b0001172405ff803c61bdcaa8e98ec1552Fariborz Jahanian};
11921de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian
11931de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian/// ObjCPropertyDecl - Represents one property declaration in an interface.
11941de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian/// For example:
11951de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian/// @property (assign, readwrite) int MyProperty;
11961de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian///
11971de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanianclass ObjCPropertyDecl : public NamedDecl {
119882a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanianpublic:
1199a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner  enum PropertyAttributeKind {
1200a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_noattr    = 0x00,
1201a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_readonly  = 0x01,
1202a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_getter    = 0x02,
1203a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_assign    = 0x04,
1204a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_readwrite = 0x08,
1205a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_retain    = 0x10,
1206a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_copy      = 0x20,
1207a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_nonatomic = 0x40,
1208a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    OBJC_PR_setter    = 0x80
1209a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner  };
1210af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
1211af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  enum SetterKind { Assign, Retain, Copy };
121246b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  enum PropertyControl { None, Required, Optional };
121382a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanianprivate:
1214dae1a1a2aa4f245b1958dc8db6089f24c575ef13Fariborz Jahanian  QualType DeclType;
1215ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  unsigned PropertyAttributes : 8;
121682a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
121746b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  // @required/@optional
121846b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  unsigned PropertyImplementation : 2;
121946b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian
12205251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector GetterName;    // getter name of NULL if no getter
12215251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector SetterName;    // setter name of NULL if no setter
122282a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
122333de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *GetterMethodDecl; // Declaration of getter instance method
122433de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *SetterMethodDecl; // Declaration of setter instance method
122533de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
12261de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian  ObjCPropertyDecl(SourceLocation L, IdentifierInfo *Id, QualType T)
12271de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian    : NamedDecl(ObjCProperty, L, Id), DeclType(T),
122833de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      PropertyAttributes(OBJC_PR_noattr), PropertyImplementation(None),
122933de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      GetterName(Selector()),
123033de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      SetterName(Selector()),
123133de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian      GetterMethodDecl(0), SetterMethodDecl(0) {}
1232f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattnerpublic:
12331de1e74541c25c1a7b721f1c3991ea34c8403420Fariborz Jahanian  static ObjCPropertyDecl *Create(ASTContext &C, SourceLocation L,
123446b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian                                  IdentifierInfo *Id, QualType T,
123546b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian                                  PropertyControl propControl = None);
1236dae1a1a2aa4f245b1958dc8db6089f24c575ef13Fariborz Jahanian  QualType getType() const { return DeclType; }
1237c35b9e4e2efad727538c848cf30d4b0eb1031dc9Fariborz Jahanian
1238a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner  PropertyAttributeKind getPropertyAttributes() const {
1239f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner    return PropertyAttributeKind(PropertyAttributes);
1240f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner  }
1241564576b225386cbff375351597dd5e2a92872d38Fariborz Jahanian  void setPropertyAttributes(PropertyAttributeKind PRVal) {
1242a5674258f5e6f74f1c0ed3ece4d64fbb1b9afb11Chris Lattner    PropertyAttributes |= PRVal;
124382a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  }
1244394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar
12458cf0bb3c2a798ce3acacaac2d3178648cd4c65c6Fariborz Jahanian void makeitReadWriteAttribute(void) {
12468cf0bb3c2a798ce3acacaac2d3178648cd4c65c6Fariborz Jahanian    PropertyAttributes &= ~OBJC_PR_readonly;
12478cf0bb3c2a798ce3acacaac2d3178648cd4c65c6Fariborz Jahanian    PropertyAttributes |= OBJC_PR_readwrite;
12488cf0bb3c2a798ce3acacaac2d3178648cd4c65c6Fariborz Jahanian }
12498cf0bb3c2a798ce3acacaac2d3178648cd4c65c6Fariborz Jahanian
1250af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  // Helper methods for accessing attributes.
1251af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
1252af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// isReadOnly - Return true iff the property has a setter.
1253394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar  bool isReadOnly() const {
1254394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar    return (PropertyAttributes & OBJC_PR_readonly);
1255394d33f1f602f7681032a659dff5bb09061ee510Daniel Dunbar  }
1256af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
1257af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// getSetterKind - Return the method used for doing assignment in
1258af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// the property setter. This is only valid if the property has been
1259af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  /// defined to have a setter.
1260af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  SetterKind getSetterKind() const {
1261af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar    if (PropertyAttributes & OBJC_PR_retain)
1262af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar      return Retain;
1263af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar    if (PropertyAttributes & OBJC_PR_copy)
1264af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar      return Copy;
1265af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar    return Assign;
1266af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  }
1267af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar
12685251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector getGetterName() const { return GetterName; }
12695251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  void setGetterName(Selector Sel) { GetterName = Sel; }
127082a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
12715251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  Selector getSetterName() const { return SetterName; }
12725251e130a23d997f7c0dfdc250cdc41f179e5bedFariborz Jahanian  void setSetterName(Selector Sel) { SetterName = Sel; }
127382a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian
127433de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *getGetterMethodDecl() const { return GetterMethodDecl; }
127533de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  void setGetterMethodDecl(ObjCMethodDecl *gDecl) { GetterMethodDecl = gDecl; }
127633de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
127733de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  ObjCMethodDecl *getSetterMethodDecl() const { return SetterMethodDecl; }
127833de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian  void setSetterMethodDecl(ObjCMethodDecl *gDecl) { SetterMethodDecl = gDecl; }
127933de3f0333ca0b5274291b8d76c86758c0484691Fariborz Jahanian
128046b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  // Related to @optional/@required declared in @protocol
128146b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  void setPropertyImplementation(PropertyControl pc) {
128246b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian    PropertyImplementation = pc;
128346b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  }
128446b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  PropertyControl getPropertyImplementation() const {
128546b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian    return PropertyControl(PropertyImplementation);
128646b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian  }
128746b55e56d029aec699fc2701e43d70264da9ecd8Fariborz Jahanian
128882a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  static bool classof(const Decl *D) {
1289670aa9d7639278f507930e95dc89c12032ab7c7eSam Bishop    return D->getKind() == ObjCProperty;
129082a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian  }
1291a526c5c67e5a0473c340903ee542ce570119665fTed Kremenek  static bool classof(const ObjCPropertyDecl *D) { return true; }
129282a5fe3d1cf204b672cdab24d72275b6ad2c3527Fariborz Jahanian};
1293980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff
129461d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian/// ObjCPropertyImplDecl - Represents implementation declaration of a property
129561d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian/// in a class or category implementation block. For example:
129661d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian/// @synthesize prop1 = ivar1;
129761d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian///
129861d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanianclass ObjCPropertyImplDecl : public Decl {
129961d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanianpublic:
13009f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar  enum Kind {
13019f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    Synthesize,
13029f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    Dynamic
130361d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  };
130461d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanianprivate:
1305559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  SourceLocation AtLoc;   // location of @synthesize or @dynamic
130661d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  /// Property declaration being implemented
130761d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  ObjCPropertyDecl *PropertyDecl;
1308be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek
130961d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  /// Null for @dynamic. Required for @synthesize.
131061d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  ObjCIvarDecl *PropertyIvarDecl;
1311be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek
1312628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian  ObjCPropertyImplDecl(SourceLocation atLoc, SourceLocation L,
1313628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                       ObjCPropertyDecl *property,
13149f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar                       Kind PK,
1315628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                       ObjCIvarDecl *ivarDecl)
13169f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    : Decl(ObjCPropertyImpl, L), AtLoc(atLoc), PropertyDecl(property),
13179f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar      PropertyIvarDecl(ivarDecl) {
13189f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    assert (PK == Dynamic || PropertyIvarDecl);
13199f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar  }
1320628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian
13219f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbarpublic:
1322628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian  static ObjCPropertyImplDecl *Create(ASTContext &C, SourceLocation atLoc,
1323628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                      SourceLocation L,
1324628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                      ObjCPropertyDecl *property,
13259f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar                                      Kind PK,
1326628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                      ObjCIvarDecl *ivarDecl);
132761d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
1328d40910b581b09c937a8c1fdcde9b8ec724398fb9Steve Naroff  SourceLocation getLocStart() const { return AtLoc; }
1329d40910b581b09c937a8c1fdcde9b8ec724398fb9Steve Naroff
1330be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  ObjCPropertyDecl *getPropertyDecl() const {
1331be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek    return PropertyDecl;
1332be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  }
133361d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
13349f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar  Kind getPropertyImplementation() const {
13359f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar    return PropertyIvarDecl ? Synthesize : Dynamic;
1336be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  }
133761d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
1338af05bb9073319d8381b71c4325188853fd4b8ed6Daniel Dunbar  ObjCIvarDecl *getPropertyIvarDecl() const {
1339be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek    return PropertyIvarDecl;
1340be57c3a3fef0776fca57ad88b2db263f37b074c4Ted Kremenek  }
134161d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
134261d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  static bool classof(const Decl *D) {
134361d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian    return D->getKind() == ObjCPropertyImpl;
134461d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  }
134561d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian  static bool classof(const ObjCPropertyImplDecl *D) { return true; }
134661d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian};
134761d46159af2a740207de8dc024211d531ae290d9Fariborz Jahanian
1348980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff}  // end namespace clang
1349980e508ca70d6de75d2abfd96b4681fc98bb2698Steve Naroff#endif
1350