11e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//===--- DeclObjC.cpp - ObjC Declaration AST Node Implementation ----------===//
21e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//
31e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//                     The LLVM Compiler Infrastructure
41e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//
51e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// This file is distributed under the University of Illinois Open Source
61e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// License. See LICENSE.TXT for details.
71e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//
81e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//===----------------------------------------------------------------------===//
91e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//
101e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// This file implements the Objective-C related Decl classes.
111e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//
121e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//===----------------------------------------------------------------------===//
131e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner
141e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner#include "clang/AST/DeclObjC.h"
151e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner#include "clang/AST/ASTContext.h"
16e91593ef084479340582b2ba177b44be50a717b7Daniel Dunbar#include "clang/AST/Stmt.h"
17e6b8d68a927368b06ac06cc9ac9e7f60aa966d5fArgyrios Kyrtzidis#include "clang/AST/ASTMutationListener.h"
180de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff#include "llvm/ADT/STLExtras.h"
191e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattnerusing namespace clang;
201e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner
216c4ae5de0c356777446f823b573821fb95560d91Chris Lattner//===----------------------------------------------------------------------===//
2211e1e1af2641affb378080a4f3d1a30da1cad082Chris Lattner// ObjCListBase
2311e1e1af2641affb378080a4f3d1a30da1cad082Chris Lattner//===----------------------------------------------------------------------===//
2411e1e1af2641affb378080a4f3d1a30da1cad082Chris Lattner
2538af2deb27cdfa1a95bde96e30dab15dce53fcefChris Lattnervoid ObjCListBase::set(void *const* InList, unsigned Elts, ASTContext &Ctx) {
26ff331c15729f7d4439d253c97f4d60f2a7ffd0c6Douglas Gregor  List = 0;
2711e1e1af2641affb378080a4f3d1a30da1cad082Chris Lattner  if (Elts == 0) return;  // Setting to an empty list is a noop.
281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
304ee413ba81c8030c195a9166847928054ed01ca4Chris Lattner  List = new (Ctx) void*[Elts];
3111e1e1af2641affb378080a4f3d1a30da1cad082Chris Lattner  NumElts = Elts;
3211e1e1af2641affb378080a4f3d1a30da1cad082Chris Lattner  memcpy(List, InList, sizeof(void*)*Elts);
3311e1e1af2641affb378080a4f3d1a30da1cad082Chris Lattner}
3411e1e1af2641affb378080a4f3d1a30da1cad082Chris Lattner
3518df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregorvoid ObjCProtocolList::set(ObjCProtocolDecl* const* InList, unsigned Elts,
3618df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregor                           const SourceLocation *Locs, ASTContext &Ctx) {
3718df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregor  if (Elts == 0)
3818df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregor    return;
3918df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregor
4018df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregor  Locations = new (Ctx) SourceLocation[Elts];
4118df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregor  memcpy(Locations, Locs, sizeof(SourceLocation) * Elts);
4218df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregor  set(InList, Elts, Ctx);
4318df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregor}
4418df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregor
4511e1e1af2641affb378080a4f3d1a30da1cad082Chris Lattner//===----------------------------------------------------------------------===//
46ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCInterfaceDecl
47ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
48ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
4999ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCContainerDecl::anchor() { }
5099ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
51496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian/// getIvarDecl - This method looks up an ivar in this ContextDecl.
52496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian///
53496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz JahanianObjCIvarDecl *
5417945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios KyrtzidisObjCContainerDecl::getIvarDecl(IdentifierInfo *Id) const {
55496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian  lookup_const_iterator Ivar, IvarEnd;
5617945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis  for (llvm::tie(Ivar, IvarEnd) = lookup(Id); Ivar != IvarEnd; ++Ivar) {
57496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian    if (ObjCIvarDecl *ivar = dyn_cast<ObjCIvarDecl>(*Ivar))
58496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian      return ivar;
59496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian  }
60496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian  return 0;
61496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian}
62496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian
63467c0b165072689ef87fe8d9cd47a5b63485bcdcArgyrios Kyrtzidis// Get the local instance/class method declared in this interface.
646ab3524f72a6e64aa04973fa9433b5559abb3525Douglas GregorObjCMethodDecl *
65467c0b165072689ef87fe8d9cd47a5b63485bcdcArgyrios KyrtzidisObjCContainerDecl::getMethod(Selector Sel, bool isInstance) const {
660de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff  // Since instance & class methods can have the same name, the loop below
670de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff  // ensures we get the correct method.
680de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff  //
690de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff  // @interface Whatever
700de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff  // - (int) class_method;
710de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff  // + (float) class_method;
720de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff  // @end
730de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff  //
740de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff  lookup_const_iterator Meth, MethEnd;
7517945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis  for (llvm::tie(Meth, MethEnd) = lookup(Sel); Meth != MethEnd; ++Meth) {
760de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff    ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(*Meth);
77467c0b165072689ef87fe8d9cd47a5b63485bcdcArgyrios Kyrtzidis    if (MD && MD->isInstanceMethod() == isInstance)
780de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff      return MD;
790de21fd85d79bccd32f04256f5b3328ab5ed7c95Steve Naroff  }
80ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  return 0;
81ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner}
82ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
839f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted KremenekObjCPropertyDecl *
84de09d0c9694f01a99870a8825266d44a29ebb325Ted KremenekObjCPropertyDecl::findPropertyDecl(const DeclContext *DC,
859f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted Kremenek                                   IdentifierInfo *propertyID) {
869f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted Kremenek
87de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek  DeclContext::lookup_const_iterator I, E;
889f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted Kremenek  llvm::tie(I, E) = DC->lookup(propertyID);
899f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted Kremenek  for ( ; I != E; ++I)
909f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted Kremenek    if (ObjCPropertyDecl *PD = dyn_cast<ObjCPropertyDecl>(*I))
919f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted Kremenek      return PD;
929f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted Kremenek
939f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted Kremenek  return 0;
949f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted Kremenek}
959f550ff05d496e6b9480e5619a21d9da0c9e27c1Ted Kremenek
96ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// FindPropertyDeclaration - Finds declaration of the property given its name
97ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// in 'PropertyId' and returns it. It returns 0, if not found.
98ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris LattnerObjCPropertyDecl *
9917945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios KyrtzidisObjCContainerDecl::FindPropertyDeclaration(IdentifierInfo *PropertyId) const {
1001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
101de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek  if (ObjCPropertyDecl *PD =
102de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek        ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(this), PropertyId))
103de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek    return PD;
104de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek
105de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek  switch (getKind()) {
106de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek    default:
107de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      break;
108de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek    case Decl::ObjCProtocol: {
109de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      const ObjCProtocolDecl *PID = cast<ObjCProtocolDecl>(this);
110de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
111de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek           E = PID->protocol_end(); I != E; ++I)
112de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek        if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
11325760611365be23556b32332f8a66ae21ea93ecfFariborz Jahanian          return P;
114de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      break;
115ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    }
116de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek    case Decl::ObjCInterface: {
117de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      const ObjCInterfaceDecl *OID = cast<ObjCInterfaceDecl>(this);
118de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      // Look through categories.
119de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      for (ObjCCategoryDecl *Cat = OID->getCategoryList();
120de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek           Cat; Cat = Cat->getNextClassCategory())
121de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek        if (!Cat->IsClassExtension())
122de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek          if (ObjCPropertyDecl *P = Cat->FindPropertyDeclaration(PropertyId))
123de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek            return P;
124de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek
125de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      // Look through protocols.
12653b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek      for (ObjCInterfaceDecl::all_protocol_iterator
12753b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek            I = OID->all_referenced_protocol_begin(),
12853b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek            E = OID->all_referenced_protocol_end(); I != E; ++I)
129de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek        if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
130de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek          return P;
131de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek
132de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      // Finally, check the super class.
133de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      if (const ObjCInterfaceDecl *superClass = OID->getSuperClass())
134de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek        return superClass->FindPropertyDeclaration(PropertyId);
135de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      break;
136ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    }
137de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek    case Decl::ObjCCategory: {
138de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      const ObjCCategoryDecl *OCD = cast<ObjCCategoryDecl>(this);
139de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      // Look through protocols.
140de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      if (!OCD->IsClassExtension())
141de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek        for (ObjCCategoryDecl::protocol_iterator
142de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek              I = OCD->protocol_begin(), E = OCD->protocol_end(); I != E; ++I)
14325760611365be23556b32332f8a66ae21ea93ecfFariborz Jahanian        if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
14425760611365be23556b32332f8a66ae21ea93ecfFariborz Jahanian          return P;
145de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek
146de09d0c9694f01a99870a8825266d44a29ebb325Ted Kremenek      break;
147ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    }
148ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  }
149ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  return 0;
150ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner}
151ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
15299ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCInterfaceDecl::anchor() { }
15399ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
154a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz Jahanian/// FindPropertyVisibleInPrimaryClass - Finds declaration of the property
155a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz Jahanian/// with name 'PropertyId' in the primary class; including those in protocols
15637cafb077ad5b170acae77e566638603011ef4c0Ted Kremenek/// (direct or indirect) used by the primary class.
157a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz Jahanian///
158a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz JahanianObjCPropertyDecl *
15937cafb077ad5b170acae77e566638603011ef4c0Ted KremenekObjCInterfaceDecl::FindPropertyVisibleInPrimaryClass(
160a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz Jahanian                                            IdentifierInfo *PropertyId) const {
1612e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  // FIXME: Should make sure no callers ever do this.
1622e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (!hasDefinition())
1632e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    return 0;
1642e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor
1652e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (data().ExternallyCompleted)
16626ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor    LoadExternalDefinition();
16726ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor
16837cafb077ad5b170acae77e566638603011ef4c0Ted Kremenek  if (ObjCPropertyDecl *PD =
16937cafb077ad5b170acae77e566638603011ef4c0Ted Kremenek      ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(this), PropertyId))
17037cafb077ad5b170acae77e566638603011ef4c0Ted Kremenek    return PD;
17137cafb077ad5b170acae77e566638603011ef4c0Ted Kremenek
172a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz Jahanian  // Look through protocols.
17353b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek  for (ObjCInterfaceDecl::all_protocol_iterator
17453b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek        I = all_referenced_protocol_begin(),
17553b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek        E = all_referenced_protocol_end(); I != E; ++I)
176a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz Jahanian    if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
177a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz Jahanian      return P;
17837cafb077ad5b170acae77e566638603011ef4c0Ted Kremenek
179a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz Jahanian  return 0;
180a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz Jahanian}
181a6f14e1a7ee3a9343a838297c73ca87fddb9ed4aFariborz Jahanian
182339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanianvoid ObjCInterfaceDecl::mergeClassExtensionProtocolList(
183339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian                              ObjCProtocolDecl *const* ExtList, unsigned ExtNum,
184339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian                              ASTContext &C)
185339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian{
1862e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (data().ExternallyCompleted)
18726ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor    LoadExternalDefinition();
18826ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor
1892e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (data().AllReferencedProtocols.empty() &&
1902e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor      data().ReferencedProtocols.empty()) {
1912e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    data().AllReferencedProtocols.set(ExtList, ExtNum, C);
192339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian    return;
193339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian  }
19453b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek
195339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian  // Check for duplicate protocol in class's protocol list.
19653b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek  // This is O(n*m). But it is extremely rare and number of protocols in
197339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian  // class or its extension are very few.
1985f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<ObjCProtocolDecl*, 8> ProtocolRefs;
199339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian  for (unsigned i = 0; i < ExtNum; i++) {
200339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian    bool protocolExists = false;
201339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian    ObjCProtocolDecl *ProtoInExtension = ExtList[i];
20253b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek    for (all_protocol_iterator
20353b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek          p = all_referenced_protocol_begin(),
20453b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek          e = all_referenced_protocol_end(); p != e; ++p) {
205339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian      ObjCProtocolDecl *Proto = (*p);
206339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian      if (C.ProtocolCompatibleWithProtocol(ProtoInExtension, Proto)) {
207339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian        protocolExists = true;
208339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian        break;
209339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian      }
210339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian    }
211339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian    // Do we want to warn on a protocol in extension class which
212339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian    // already exist in the class? Probably not.
21353b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek    if (!protocolExists)
214339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian      ProtocolRefs.push_back(ProtoInExtension);
215339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian  }
21653b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek
217339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian  if (ProtocolRefs.empty())
218339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian    return;
21953b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek
220b106fc635b1523952332131785b700453a936e49Fariborz Jahanian  // Merge ProtocolRefs into class's protocol list;
22153b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek  for (all_protocol_iterator p = all_referenced_protocol_begin(),
22253b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek        e = all_referenced_protocol_end(); p != e; ++p) {
223339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian    ProtocolRefs.push_back(*p);
22418df52bbb5d28ca082064d31ae7558dbdae52377Douglas Gregor  }
22553b9441b5a81a24fa1f66f3f6416f1e36baa9c2fTed Kremenek
2262e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  data().AllReferencedProtocols.set(ProtocolRefs.data(), ProtocolRefs.size(),C);
227339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian}
228339798eae1eb61c50ca68766ed028c0a16d0a284Fariborz Jahanian
2292e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregorvoid ObjCInterfaceDecl::allocateDefinitionData() {
2302e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  assert(!hasDefinition() && "ObjC class already has a definition");
23126fec63b14565e9e2d8c9935b276b99be950444aDouglas Gregor  Data = new (getASTContext()) DefinitionData();
2328d2dbbf9ddfd9d762a341d83f83d840ff68ce03dDouglas Gregor  Data->Definition = this;
2338d2dbbf9ddfd9d762a341d83f83d840ff68ce03dDouglas Gregor
2348d2dbbf9ddfd9d762a341d83f83d840ff68ce03dDouglas Gregor  // Make the type point at the definition, now that we have one.
2358d2dbbf9ddfd9d762a341d83f83d840ff68ce03dDouglas Gregor  if (TypeForDecl)
2368d2dbbf9ddfd9d762a341d83f83d840ff68ce03dDouglas Gregor    cast<ObjCInterfaceType>(TypeForDecl)->Decl = this;
2370af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregor}
2380af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregor
2390af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregorvoid ObjCInterfaceDecl::startDefinition() {
2400af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregor  allocateDefinitionData();
2410af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregor
24253df7a1d34f21d8f2309311d1067d463e9064c60Douglas Gregor  // Update all of the declarations with a pointer to the definition.
24353df7a1d34f21d8f2309311d1067d463e9064c60Douglas Gregor  for (redecl_iterator RD = redecls_begin(), RDEnd = redecls_end();
24453df7a1d34f21d8f2309311d1067d463e9064c60Douglas Gregor       RD != RDEnd; ++RD) {
24553df7a1d34f21d8f2309311d1067d463e9064c60Douglas Gregor    if (*RD != this)
24626fec63b14565e9e2d8c9935b276b99be950444aDouglas Gregor      RD->Data = Data;
24753df7a1d34f21d8f2309311d1067d463e9064c60Douglas Gregor  }
248ad834d534e9a5db3d3baa09593775f83ceaff1f2Argyrios Kyrtzidis}
249ad834d534e9a5db3d3baa09593775f83ceaff1f2Argyrios Kyrtzidis
25080aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian/// getFirstClassExtension - Find first class extension of the given class.
25180aa1cd7973561889e51c1c152c8990a8de9c953Fariborz JahanianObjCCategoryDecl* ObjCInterfaceDecl::getFirstClassExtension() const {
25280aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian  for (ObjCCategoryDecl *CDecl = getCategoryList(); CDecl;
2530e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian       CDecl = CDecl->getNextClassCategory())
2540e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian    if (CDecl->IsClassExtension())
2550e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian      return CDecl;
2560e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian  return 0;
2570e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian}
2580e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian
25980aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian/// getNextClassCategory - Find next class extension in list of categories.
26080aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanianconst ObjCCategoryDecl* ObjCCategoryDecl::getNextClassExtension() const {
26180aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian  for (const ObjCCategoryDecl *CDecl = getNextClassCategory(); CDecl;
26280aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian        CDecl = CDecl->getNextClassCategory())
26380aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian    if (CDecl->IsClassExtension())
26480aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian      return CDecl;
26580aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian  return 0;
26680aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian}
26780aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian
26817945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios KyrtzidisObjCIvarDecl *ObjCInterfaceDecl::lookupInstanceVariable(IdentifierInfo *ID,
26917945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis                                              ObjCInterfaceDecl *&clsDeclared) {
2702e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  // FIXME: Should make sure no callers ever do this.
2712e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (!hasDefinition())
2722e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    return 0;
2732e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor
2742e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (data().ExternallyCompleted)
2757c81c2a5915878e4aa6908a097290fd47fb3a154Argyrios Kyrtzidis    LoadExternalDefinition();
2767c81c2a5915878e4aa6908a097290fd47fb3a154Argyrios Kyrtzidis
277ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  ObjCInterfaceDecl* ClassDecl = this;
278ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  while (ClassDecl != NULL) {
27917945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis    if (ObjCIvarDecl *I = ClassDecl->getIvarDecl(ID)) {
280496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian      clsDeclared = ClassDecl;
281496b5a894c5ec5425de53909f5aac3fb4771a2ecFariborz Jahanian      return I;
282ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    }
28380aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian    for (const ObjCCategoryDecl *CDecl = ClassDecl->getFirstClassExtension();
28480aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian         CDecl; CDecl = CDecl->getNextClassExtension()) {
2850e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian      if (ObjCIvarDecl *I = CDecl->getIvarDecl(ID)) {
2860e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian        clsDeclared = ClassDecl;
2870e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian        return I;
2880e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian      }
28980aa1cd7973561889e51c1c152c8990a8de9c953Fariborz Jahanian    }
2900e5ad255729ee86b8ed57e659029008984517cdeFariborz Jahanian
291ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    ClassDecl = ClassDecl->getSuperClass();
292ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  }
293ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  return NULL;
294ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner}
295ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
296cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian/// lookupInheritedClass - This method returns ObjCInterfaceDecl * of the super
297cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian/// class whose name is passed as argument. If it is not one of the super classes
298cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian/// the it returns NULL.
299cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz JahanianObjCInterfaceDecl *ObjCInterfaceDecl::lookupInheritedClass(
300cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian                                        const IdentifierInfo*ICName) {
3012e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  // FIXME: Should make sure no callers ever do this.
3022e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (!hasDefinition())
3032e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    return 0;
3042e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor
3052e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (data().ExternallyCompleted)
3067c81c2a5915878e4aa6908a097290fd47fb3a154Argyrios Kyrtzidis    LoadExternalDefinition();
3077c81c2a5915878e4aa6908a097290fd47fb3a154Argyrios Kyrtzidis
308cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian  ObjCInterfaceDecl* ClassDecl = this;
309cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian  while (ClassDecl != NULL) {
310cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian    if (ClassDecl->getIdentifier() == ICName)
311cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian      return ClassDecl;
312cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian    ClassDecl = ClassDecl->getSuperClass();
313cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian  }
314cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian  return NULL;
315cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian}
316cd1876207f5564beba74e4b2524b664bdba0ba9fFariborz Jahanian
317aa5420c1e36ab8e0e4bb87239d8b73a3a8ce75dbArgyrios Kyrtzidis/// lookupMethod - This method returns an instance/class method by looking in
318ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// the class, its categories, and its super classes (using a linear search).
319bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz JahanianObjCMethodDecl *ObjCInterfaceDecl::lookupMethod(Selector Sel,
320bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian                                     bool isInstance,
321bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian                                     bool shallowCategoryLookup) const {
3222e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  // FIXME: Should make sure no callers ever do this.
3232e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (!hasDefinition())
3242e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    return 0;
3252e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor
326aa5420c1e36ab8e0e4bb87239d8b73a3a8ce75dbArgyrios Kyrtzidis  const ObjCInterfaceDecl* ClassDecl = this;
327ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  ObjCMethodDecl *MethodDecl = 0;
3281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3292e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (data().ExternallyCompleted)
33026ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor    LoadExternalDefinition();
33126ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor
332ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  while (ClassDecl != NULL) {
333aa5420c1e36ab8e0e4bb87239d8b73a3a8ce75dbArgyrios Kyrtzidis    if ((MethodDecl = ClassDecl->getMethod(Sel, isInstance)))
334ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner      return MethodDecl;
3351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
336ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    // Didn't find one yet - look through protocols.
337a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis    for (ObjCInterfaceDecl::protocol_iterator I = ClassDecl->protocol_begin(),
338a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis                                              E = ClassDecl->protocol_end();
339a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis           I != E; ++I)
340aa5420c1e36ab8e0e4bb87239d8b73a3a8ce75dbArgyrios Kyrtzidis      if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
341ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner        return MethodDecl;
342bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian
343bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian    // Didn't find one yet - now look through categories.
344bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian    ObjCCategoryDecl *CatDecl = ClassDecl->getCategoryList();
345bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian    while (CatDecl) {
346bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian      if ((MethodDecl = CatDecl->getMethod(Sel, isInstance)))
347bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian        return MethodDecl;
348bb3d14e55d267bf5228699c7bf0c8ccdb71c8f46Fariborz Jahanian
349bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian      if (!shallowCategoryLookup) {
350bb3d14e55d267bf5228699c7bf0c8ccdb71c8f46Fariborz Jahanian        // Didn't find one yet - look through protocols.
351bb3d14e55d267bf5228699c7bf0c8ccdb71c8f46Fariborz Jahanian        const ObjCList<ObjCProtocolDecl> &Protocols =
352bb3d14e55d267bf5228699c7bf0c8ccdb71c8f46Fariborz Jahanian          CatDecl->getReferencedProtocols();
353bb3d14e55d267bf5228699c7bf0c8ccdb71c8f46Fariborz Jahanian        for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
354bb3d14e55d267bf5228699c7bf0c8ccdb71c8f46Fariborz Jahanian             E = Protocols.end(); I != E; ++I)
355bb3d14e55d267bf5228699c7bf0c8ccdb71c8f46Fariborz Jahanian          if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
356bb3d14e55d267bf5228699c7bf0c8ccdb71c8f46Fariborz Jahanian            return MethodDecl;
357bb3d14e55d267bf5228699c7bf0c8ccdb71c8f46Fariborz Jahanian      }
358bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian      CatDecl = CatDecl->getNextClassCategory();
359ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    }
360bf393be8a0b8b573ceb23ed19ac953832a2a69e4Fariborz Jahanian
361ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    ClassDecl = ClassDecl->getSuperClass();
362ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  }
363ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  return NULL;
364ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner}
365ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
36674b2756bc1f1f5f7c189996fe7e4cd3efef70263Fariborz JahanianObjCMethodDecl *ObjCInterfaceDecl::lookupPrivateMethod(
36774b2756bc1f1f5f7c189996fe7e4cd3efef70263Fariborz Jahanian                                   const Selector &Sel,
36874b2756bc1f1f5f7c189996fe7e4cd3efef70263Fariborz Jahanian                                   bool Instance) {
3692e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  // FIXME: Should make sure no callers ever do this.
3702e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (!hasDefinition())
3712e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    return 0;
3722e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor
3732e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (data().ExternallyCompleted)
3747c81c2a5915878e4aa6908a097290fd47fb3a154Argyrios Kyrtzidis    LoadExternalDefinition();
3757c81c2a5915878e4aa6908a097290fd47fb3a154Argyrios Kyrtzidis
376d789d3d985e28c9404e62157af46dcb7331920e0Steve Naroff  ObjCMethodDecl *Method = 0;
377d789d3d985e28c9404e62157af46dcb7331920e0Steve Naroff  if (ObjCImplementationDecl *ImpDecl = getImplementation())
37874b2756bc1f1f5f7c189996fe7e4cd3efef70263Fariborz Jahanian    Method = Instance ? ImpDecl->getInstanceMethod(Sel)
37974b2756bc1f1f5f7c189996fe7e4cd3efef70263Fariborz Jahanian                      : ImpDecl->getClassMethod(Sel);
380d789d3d985e28c9404e62157af46dcb7331920e0Steve Naroff
381d789d3d985e28c9404e62157af46dcb7331920e0Steve Naroff  if (!Method && getSuperClass())
38274b2756bc1f1f5f7c189996fe7e4cd3efef70263Fariborz Jahanian    return getSuperClass()->lookupPrivateMethod(Sel, Instance);
383d789d3d985e28c9404e62157af46dcb7331920e0Steve Naroff  return Method;
384d789d3d985e28c9404e62157af46dcb7331920e0Steve Naroff}
385ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
386ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
387ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCMethodDecl
3886c4ae5de0c356777446f823b573821fb95560d91Chris Lattner//===----------------------------------------------------------------------===//
3896c4ae5de0c356777446f823b573821fb95560d91Chris Lattner
3900ed844b04ea4387caa4e1cf3dc375d269657536bChris LattnerObjCMethodDecl *ObjCMethodDecl::Create(ASTContext &C,
3911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                       SourceLocation beginLoc,
3926c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                       SourceLocation endLoc,
3936c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                       Selector SelInfo, QualType T,
3944bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor                                       TypeSourceInfo *ResultTInfo,
3950701bbb228dfd87e1fe82a0a4b7b9facfecb43daSteve Naroff                                       DeclContext *contextDecl,
396f6414927e67e27d9324d8d179c5f7ea620443924Daniel Dunbar                                       bool isInstance,
3976c4ae5de0c356777446f823b573821fb95560d91Chris Lattner                                       bool isVariadic,
3984607034e621aa77378ec75249d1e9eaf5de49b6aFariborz Jahanian                                       bool isSynthesized,
39975cf3e86d33ce810c12084126385371b335c30baArgyrios Kyrtzidis                                       bool isImplicitlyDeclared,
4003fe104154dd2e8ffb351142d74f308938b5c99bfFariborz Jahanian                                       bool isDefined,
4017732cc9c0fdc97a2f8cce4e5933d8103213d1aefFariborz Jahanian                                       ImplementationControl impControl,
402da92a7f91cf88f49e02050919676f7fb8e3bdff8Argyrios Kyrtzidis                                       bool HasRelatedResultType) {
4033e9704981d7691fdd44913bf1786e8d760d8a627Steve Naroff  return new (C) ObjCMethodDecl(beginLoc, endLoc,
4044bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor                                SelInfo, T, ResultTInfo, contextDecl,
4054bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor                                isInstance,
40675cf3e86d33ce810c12084126385371b335c30baArgyrios Kyrtzidis                                isVariadic, isSynthesized, isImplicitlyDeclared,
40775cf3e86d33ce810c12084126385371b335c30baArgyrios Kyrtzidis                                isDefined,
4083fe104154dd2e8ffb351142d74f308938b5c99bfFariborz Jahanian                                impControl,
409da92a7f91cf88f49e02050919676f7fb8e3bdff8Argyrios Kyrtzidis                                HasRelatedResultType);
4100e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner}
4110e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner
4121e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCMethodDecl *ObjCMethodDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
4131e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCMethodDecl));
4141e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCMethodDecl(SourceLocation(), SourceLocation(),
4151e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                  Selector(), QualType(), 0, 0);
4161e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor}
4171e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor
4183a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidisvoid ObjCMethodDecl::setAsRedeclaration(const ObjCMethodDecl *PrevMethod) {
4193a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis  assert(PrevMethod);
4203a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis  getASTContext().setObjCMethodRedeclaration(PrevMethod, this);
4213a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis  IsRedeclaration = true;
42272b2625aa67c8213acaf4bf6209b67859d60e2cfArgyrios Kyrtzidis  PrevMethod->HasRedeclaration = true;
4233a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis}
4243a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis
425491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidisvoid ObjCMethodDecl::setParamsAndSelLocs(ASTContext &C,
426491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis                                         ArrayRef<ParmVarDecl*> Params,
427491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis                                         ArrayRef<SourceLocation> SelLocs) {
428491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  ParamsAndSelLocs = 0;
429491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  NumParams = Params.size();
430491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  if (Params.empty() && SelLocs.empty())
431491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis    return;
432491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis
433491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  unsigned Size = sizeof(ParmVarDecl *) * NumParams +
434491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis                  sizeof(SourceLocation) * SelLocs.size();
435491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  ParamsAndSelLocs = C.Allocate(Size);
436491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  std::copy(Params.begin(), Params.end(), getParams());
437491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  std::copy(SelLocs.begin(), SelLocs.end(), getStoredSelLocs());
438491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis}
439491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis
440491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidisvoid ObjCMethodDecl::getSelectorLocs(
441491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis                               SmallVectorImpl<SourceLocation> &SelLocs) const {
442491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  for (unsigned i = 0, e = getNumSelectorLocs(); i != e; ++i)
443491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis    SelLocs.push_back(getSelectorLoc(i));
444491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis}
445491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis
446491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidisvoid ObjCMethodDecl::setMethodParams(ASTContext &C,
447491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis                                     ArrayRef<ParmVarDecl*> Params,
448491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis                                     ArrayRef<SourceLocation> SelLocs) {
449491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  assert((!SelLocs.empty() || isImplicit()) &&
450491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis         "No selector locs for non-implicit method");
451491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  if (isImplicit())
452491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis    return setParamsAndSelLocs(C, Params, ArrayRef<SourceLocation>());
453491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis
454491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  SelLocsKind = hasStandardSelectorLocs(getSelector(), SelLocs, Params, EndLoc);
455491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  if (SelLocsKind != SelLoc_NonStandard)
456491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis    return setParamsAndSelLocs(C, Params, ArrayRef<SourceLocation>());
457491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis
458491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis  setParamsAndSelLocs(C, Params, SelLocs);
459491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis}
460491306a83c4f0f49f95a3bcbca8580cb98a91c7aArgyrios Kyrtzidis
46157ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis/// \brief A definition will return its interface declaration.
46257ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis/// An interface declaration will return its definition.
46357ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis/// Otherwise it will return itself.
46457ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios KyrtzidisObjCMethodDecl *ObjCMethodDecl::getNextRedeclaration() {
46557ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis  ASTContext &Ctx = getASTContext();
46672b2625aa67c8213acaf4bf6209b67859d60e2cfArgyrios Kyrtzidis  ObjCMethodDecl *Redecl = 0;
46772b2625aa67c8213acaf4bf6209b67859d60e2cfArgyrios Kyrtzidis  if (HasRedeclaration)
46872b2625aa67c8213acaf4bf6209b67859d60e2cfArgyrios Kyrtzidis    Redecl = const_cast<ObjCMethodDecl*>(Ctx.getObjCMethodRedeclaration(this));
469b40034c2e580ab3b08de9dfb738d8e5d8ef79136Argyrios Kyrtzidis  if (Redecl)
470b40034c2e580ab3b08de9dfb738d8e5d8ef79136Argyrios Kyrtzidis    return Redecl;
471b40034c2e580ab3b08de9dfb738d8e5d8ef79136Argyrios Kyrtzidis
47257ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis  Decl *CtxD = cast<Decl>(getDeclContext());
47357ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis
47457ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis  if (ObjCInterfaceDecl *IFD = dyn_cast<ObjCInterfaceDecl>(CtxD)) {
47557ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis    if (ObjCImplementationDecl *ImplD = Ctx.getObjCImplementation(IFD))
47657ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis      Redecl = ImplD->getMethod(getSelector(), isInstanceMethod());
47757ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis
47857ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis  } else if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(CtxD)) {
47957ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis    if (ObjCCategoryImplDecl *ImplD = Ctx.getObjCImplementation(CD))
48057ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis      Redecl = ImplD->getMethod(getSelector(), isInstanceMethod());
48157ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis
4824292073a858f72769fa405b48390620c8932f8aeArgyrios Kyrtzidis  } else if (ObjCImplementationDecl *ImplD =
4834292073a858f72769fa405b48390620c8932f8aeArgyrios Kyrtzidis               dyn_cast<ObjCImplementationDecl>(CtxD)) {
48457ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis    if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
48557ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis      Redecl = IFD->getMethod(getSelector(), isInstanceMethod());
4864292073a858f72769fa405b48390620c8932f8aeArgyrios Kyrtzidis
4874292073a858f72769fa405b48390620c8932f8aeArgyrios Kyrtzidis  } else if (ObjCCategoryImplDecl *CImplD =
4884292073a858f72769fa405b48390620c8932f8aeArgyrios Kyrtzidis               dyn_cast<ObjCCategoryImplDecl>(CtxD)) {
4890d69b8cc8e90a9364771837cb42d7031b4cbb984Steve Naroff    if (ObjCCategoryDecl *CatD = CImplD->getCategoryDecl())
4904292073a858f72769fa405b48390620c8932f8aeArgyrios Kyrtzidis      Redecl = CatD->getMethod(getSelector(), isInstanceMethod());
49157ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis  }
49257ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis
4933a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis  if (!Redecl && isRedeclaration()) {
4943a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis    // This is the last redeclaration, go back to the first method.
4953a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis    return cast<ObjCContainerDecl>(CtxD)->getMethod(getSelector(),
4963a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis                                                    isInstanceMethod());
4973a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis  }
4983a919e7110407ae7609bb6edc57aac16a5990661Argyrios Kyrtzidis
49957ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis  return Redecl ? Redecl : this;
50057ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis}
50157ea6bee79cc60ba20c7886b453f40f380dce1b1Argyrios Kyrtzidis
502e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios KyrtzidisObjCMethodDecl *ObjCMethodDecl::getCanonicalDecl() {
503e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis  Decl *CtxD = cast<Decl>(getDeclContext());
504e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis
505e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis  if (ObjCImplementationDecl *ImplD = dyn_cast<ObjCImplementationDecl>(CtxD)) {
506e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis    if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
507e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis      if (ObjCMethodDecl *MD = IFD->getMethod(getSelector(),
508e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis                                              isInstanceMethod()))
509e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis        return MD;
510e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis
511e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis  } else if (ObjCCategoryImplDecl *CImplD =
512e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis               dyn_cast<ObjCCategoryImplDecl>(CtxD)) {
5130d69b8cc8e90a9364771837cb42d7031b4cbb984Steve Naroff    if (ObjCCategoryDecl *CatD = CImplD->getCategoryDecl())
514e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis      if (ObjCMethodDecl *MD = CatD->getMethod(getSelector(),
515e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis                                               isInstanceMethod()))
516e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis        return MD;
517e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis  }
518e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis
5196d4740e740f9c109c76c111cc90a63f6a24f6357Argyrios Kyrtzidis  if (isRedeclaration())
5206d4740e740f9c109c76c111cc90a63f6a24f6357Argyrios Kyrtzidis    return cast<ObjCContainerDecl>(CtxD)->getMethod(getSelector(),
5216d4740e740f9c109c76c111cc90a63f6a24f6357Argyrios Kyrtzidis                                                    isInstanceMethod());
5226d4740e740f9c109c76c111cc90a63f6a24f6357Argyrios Kyrtzidis
523e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis  return this;
524e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis}
525e7f9d301a10b4b3223e91d9be4362b44cba0a212Argyrios Kyrtzidis
52685f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCallObjCMethodFamily ObjCMethodDecl::getMethodFamily() const {
52785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  ObjCMethodFamily family = static_cast<ObjCMethodFamily>(Family);
528d976c8e2752bc36c0697d43f985ec55b9450f8c1John McCall  if (family != static_cast<unsigned>(InvalidObjCMethodFamily))
52985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall    return family;
53085f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
531d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall  // Check for an explicit attribute.
532d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall  if (const ObjCMethodFamilyAttr *attr = getAttr<ObjCMethodFamilyAttr>()) {
533d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    // The unfortunate necessity of mapping between enums here is due
534d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    // to the attributes framework.
535d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    switch (attr->getFamily()) {
536d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    case ObjCMethodFamilyAttr::OMF_None: family = OMF_None; break;
537d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    case ObjCMethodFamilyAttr::OMF_alloc: family = OMF_alloc; break;
538d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    case ObjCMethodFamilyAttr::OMF_copy: family = OMF_copy; break;
539d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    case ObjCMethodFamilyAttr::OMF_init: family = OMF_init; break;
540d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    case ObjCMethodFamilyAttr::OMF_mutableCopy: family = OMF_mutableCopy; break;
541d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    case ObjCMethodFamilyAttr::OMF_new: family = OMF_new; break;
542d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    }
543d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    Family = static_cast<unsigned>(family);
544d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall    return family;
545d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall  }
546d5313b0bbf3948fe7c63bf46a7da330c96d07309John McCall
54785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  family = getSelector().getMethodFamily();
54885f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  switch (family) {
54985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_None: break;
55085f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
55185f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // init only has a conventional meaning for an instance method, and
55285f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // it has to return an object.
55385f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_init:
55485f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall    if (!isInstanceMethod() || !getResultType()->isObjCObjectPointerType())
55585f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall      family = OMF_None;
55685f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall    break;
55785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
55885f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // alloc/copy/new have a conventional meaning for both class and
55985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // instance methods, but they require an object return.
56085f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_alloc:
56185f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_copy:
56285f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_mutableCopy:
56385f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_new:
56485f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall    if (!getResultType()->isObjCObjectPointerType())
56585f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall      family = OMF_None;
56685f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall    break;
56785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
56885f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // These selectors have a conventional meaning only for instance methods.
56985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_dealloc:
57080cb6e69d9e85231588ae604e4bc2bc9a07389afNico Weber  case OMF_finalize:
57185f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_retain:
57285f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_release:
57385f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_autorelease:
57485f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  case OMF_retainCount:
575926df6cfabf3eaa4afc990c097fa4619b76a9b57Douglas Gregor  case OMF_self:
57685f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall    if (!isInstanceMethod())
57785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall      family = OMF_None;
57885f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall    break;
5799670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian
5809670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian  case OMF_performSelector:
5819670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian    if (!isInstanceMethod() ||
5829670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian        !getResultType()->isObjCIdType())
5839670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian      family = OMF_None;
5849670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian    else {
5859670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian      unsigned noParams = param_size();
5869670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian      if (noParams < 1 || noParams > 3)
5879670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian        family = OMF_None;
5889670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian      else {
5899670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian        ObjCMethodDecl::arg_type_iterator it = arg_type_begin();
5909670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian        QualType ArgT = (*it);
5919670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian        if (!ArgT->isObjCSelType()) {
5929670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian          family = OMF_None;
5939670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian          break;
5949670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian        }
5959670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian        while (--noParams) {
5969670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian          it++;
5979670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian          ArgT = (*it);
5989670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian          if (!ArgT->isObjCIdType()) {
5999670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian            family = OMF_None;
6009670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian            break;
6019670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian          }
6029670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian        }
6039670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian      }
6049670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian    }
6059670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian    break;
6069670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian
60785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  }
60885f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
60985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // Cache the result.
61085f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  Family = static_cast<unsigned>(family);
61185f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  return family;
61285f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall}
61385f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
6141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpvoid ObjCMethodDecl::createImplicitParams(ASTContext &Context,
615ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner                                          const ObjCInterfaceDecl *OID) {
616ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  QualType selfTy;
617ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  if (isInstanceMethod()) {
618ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    // There may be no interface context due to error in declaration
619ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    // of the interface (which has been reported). Recover gracefully.
620ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    if (OID) {
6213b3a45858c6b2a45114e91902c3bf3c4b7f5f302Daniel Dunbar      selfTy = Context.getObjCInterfaceType(OID);
62214108da7f7fc059772711e4ffee1322a27b152a7Steve Naroff      selfTy = Context.getObjCObjectPointerType(selfTy);
623ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    } else {
624ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner      selfTy = Context.getObjCIdType();
625ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    }
626ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  } else // we have a factory method.
627ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    selfTy = Context.getObjCClassType();
628ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
6297acddacc921cd0b3f813443a8641eeddb82dfbd4John McCall  bool selfIsPseudoStrong = false;
630f85e193739c953358c865005855253af4f68a497John McCall  bool selfIsConsumed = false;
6312bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek
6324e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie  if (Context.getLangOpts().ObjCAutoRefCount) {
6332bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek    if (isInstanceMethod()) {
6342bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      selfIsConsumed = hasAttr<NSConsumesSelfAttr>();
6352bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek
6362bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      // 'self' is always __strong.  It's actually pseudo-strong except
6372bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      // in init methods (or methods labeled ns_consumes_self), though.
6382bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      Qualifiers qs;
6392bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      qs.setObjCLifetime(Qualifiers::OCL_Strong);
6402bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      selfTy = Context.getQualifiedType(selfTy, qs);
6412bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek
6422bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      // In addition, 'self' is const unless this is an init method.
6432bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      if (getMethodFamily() != OMF_init && !selfIsConsumed) {
6442bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek        selfTy = selfTy.withConst();
6452bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek        selfIsPseudoStrong = true;
6462bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      }
6472bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek    }
6482bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek    else {
6492bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      assert(isClassMethod());
6502bbcd5ce370753c86d312d2c72a97476ac35b073Ted Kremenek      // 'self' is always const in class methods.
651f85e193739c953358c865005855253af4f68a497John McCall      selfTy = selfTy.withConst();
6527acddacc921cd0b3f813443a8641eeddb82dfbd4John McCall      selfIsPseudoStrong = true;
6537acddacc921cd0b3f813443a8641eeddb82dfbd4John McCall    }
654f85e193739c953358c865005855253af4f68a497John McCall  }
655f85e193739c953358c865005855253af4f68a497John McCall
656f85e193739c953358c865005855253af4f68a497John McCall  ImplicitParamDecl *self
657f85e193739c953358c865005855253af4f68a497John McCall    = ImplicitParamDecl::Create(Context, this, SourceLocation(),
658f85e193739c953358c865005855253af4f68a497John McCall                                &Context.Idents.get("self"), selfTy);
659f85e193739c953358c865005855253af4f68a497John McCall  setSelfDecl(self);
660f85e193739c953358c865005855253af4f68a497John McCall
661f85e193739c953358c865005855253af4f68a497John McCall  if (selfIsConsumed)
662f85e193739c953358c865005855253af4f68a497John McCall    self->addAttr(new (Context) NSConsumedAttr(SourceLocation(), Context));
663ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
6647acddacc921cd0b3f813443a8641eeddb82dfbd4John McCall  if (selfIsPseudoStrong)
6657acddacc921cd0b3f813443a8641eeddb82dfbd4John McCall    self->setARCPseudoStrong(true);
6667acddacc921cd0b3f813443a8641eeddb82dfbd4John McCall
6671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  setCmdDecl(ImplicitParamDecl::Create(Context, this, SourceLocation(),
6681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                       &Context.Idents.get("_cmd"),
66953c9d8a4b8f0a76cb9dd2fdd8c433ccf110f2eecSteve Naroff                                       Context.getObjCSelType()));
670ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner}
671ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
672ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris LattnerObjCInterfaceDecl *ObjCMethodDecl::getClassInterface() {
673ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  if (ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(getDeclContext()))
674ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    return ID;
675ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(getDeclContext()))
676ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    return CD->getClassInterface();
677a8530375168f578e9039837c58054d55655c981bArgyrios Kyrtzidis  if (ObjCImplDecl *IMD = dyn_cast<ObjCImplDecl>(getDeclContext()))
678ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    return IMD->getClassInterface();
679a8530375168f578e9039837c58054d55655c981bArgyrios Kyrtzidis
680a8530375168f578e9039837c58054d55655c981bArgyrios Kyrtzidis  assert(!isa<ObjCProtocolDecl>(getDeclContext()) && "It's a protocol method");
681b219cfc4d75f0a03630b7c4509ef791b7e97b2c8David Blaikie  llvm_unreachable("unknown method context");
682ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner}
683ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
684ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
685ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCInterfaceDecl
686ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
6870b7ebb3dba0df0a6cbf221e5edbc6a4b8848478cChris Lattner
688a6ea10e22b600d92e084f6b11b9b9a92d0eb2412Douglas GregorObjCInterfaceDecl *ObjCInterfaceDecl::Create(const ASTContext &C,
689d04341000d35c8808a72838b057eed7bf13b7661Douglas Gregor                                             DeclContext *DC,
6900ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner                                             SourceLocation atLoc,
6911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                             IdentifierInfo *Id,
6920af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregor                                             ObjCInterfaceDecl *PrevDecl,
693d6a07aaf62b40cdfbd96f6b874d02b06fc22d015Steve Naroff                                             SourceLocation ClassLoc,
6947723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor                                             bool isInternal){
6950af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregor  ObjCInterfaceDecl *Result = new (C) ObjCInterfaceDecl(DC, atLoc, Id, ClassLoc,
696fd002a7027e47ddd4bc9f32b158b18ab0ebd29c7Douglas Gregor                                                        PrevDecl, isInternal);
6970af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregor  C.getObjCInterfaceType(Result, PrevDecl);
6980af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregor  return Result;
6990af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregor}
7000af550115df1f57f17a4f125ff0e8b34820c65d1Douglas Gregor
7011e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCInterfaceDecl *ObjCInterfaceDecl::CreateDeserialized(ASTContext &C,
7021e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                                         unsigned ID) {
7031e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCInterfaceDecl));
7041e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCInterfaceDecl(0, SourceLocation(), 0, SourceLocation(),
7051e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                     0, false);
7060e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner}
7076c4ae5de0c356777446f823b573821fb95560d91Chris Lattner
7080b7ebb3dba0df0a6cbf221e5edbc6a4b8848478cChris LattnerObjCInterfaceDecl::
7090b7ebb3dba0df0a6cbf221e5edbc6a4b8848478cChris LattnerObjCInterfaceDecl(DeclContext *DC, SourceLocation atLoc, IdentifierInfo *Id,
710fd002a7027e47ddd4bc9f32b158b18ab0ebd29c7Douglas Gregor                  SourceLocation CLoc, ObjCInterfaceDecl *PrevDecl,
711fd002a7027e47ddd4bc9f32b158b18ab0ebd29c7Douglas Gregor                  bool isInternal)
7121711fc91efb36d131f7ba771f73f0154dc1abd1fArgyrios Kyrtzidis  : ObjCContainerDecl(ObjCInterface, DC, Id, CLoc, atLoc),
7137723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor    TypeForDecl(0), Data()
7142e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor{
715fd002a7027e47ddd4bc9f32b158b18ab0ebd29c7Douglas Gregor  setPreviousDeclaration(PrevDecl);
716fd002a7027e47ddd4bc9f32b158b18ab0ebd29c7Douglas Gregor
717fd002a7027e47ddd4bc9f32b158b18ab0ebd29c7Douglas Gregor  // Copy the 'data' pointer over.
718fd002a7027e47ddd4bc9f32b158b18ab0ebd29c7Douglas Gregor  if (PrevDecl)
719fd002a7027e47ddd4bc9f32b158b18ab0ebd29c7Douglas Gregor    Data = PrevDecl->Data;
720fd002a7027e47ddd4bc9f32b158b18ab0ebd29c7Douglas Gregor
72140f57ee2dab3ed3475fa584f83f05bd3c9ed4a00Argyrios Kyrtzidis  setImplicit(isInternal);
722e4f039e01e797a38bc97bf22aff9832ecd18ff5fSteve Naroff}
723e4f039e01e797a38bc97bf22aff9832ecd18ff5fSteve Naroff
72426ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregorvoid ObjCInterfaceDecl::LoadExternalDefinition() const {
7252e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  assert(data().ExternallyCompleted && "Class is not externally completed");
7262e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  data().ExternallyCompleted = false;
72726ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor  getASTContext().getExternalSource()->CompleteType(
72826ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor                                        const_cast<ObjCInterfaceDecl *>(this));
72926ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor}
73026ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor
73126ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregorvoid ObjCInterfaceDecl::setExternallyCompleted() {
73226ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor  assert(getASTContext().getExternalSource() &&
73326ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor         "Class can't be externally completed without an external source");
7342e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  assert(hasDefinition() &&
73526ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor         "Forward declarations can't be externally completed");
7362e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  data().ExternallyCompleted = true;
73726ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor}
73826ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor
7398a1d722f13df383600f36d77f842957c8adb5f1bArgyrios KyrtzidisObjCImplementationDecl *ObjCInterfaceDecl::getImplementation() const {
7407723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor  if (const ObjCInterfaceDecl *Def = getDefinition()) {
7417723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor    if (data().ExternallyCompleted)
7427723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor      LoadExternalDefinition();
7437723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor
7447723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor    return getASTContext().getObjCImplementation(
7457723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor             const_cast<ObjCInterfaceDecl*>(Def));
7467723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor  }
7477723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor
7482e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  // FIXME: Should make sure no callers ever do this.
7497723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor  return 0;
7508a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis}
7518a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis
7528a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidisvoid ObjCInterfaceDecl::setImplementation(ObjCImplementationDecl *ImplD) {
7537723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor  getASTContext().setObjCImplementation(getDefinition(), ImplD);
7548a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis}
7558a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis
7562c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian/// all_declared_ivar_begin - return first ivar declared in this class,
7572c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian/// its extensions and its implementation. Lazily build the list on first
7582c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian/// access.
7592c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz JahanianObjCIvarDecl *ObjCInterfaceDecl::all_declared_ivar_begin() {
7602e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  // FIXME: Should make sure no callers ever do this.
7612e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (!hasDefinition())
7622e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    return 0;
7632e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor
7642e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (data().IvarList)
7652e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    return data().IvarList;
7662c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian
7672c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian  ObjCIvarDecl *curIvar = 0;
7682c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian  if (!ivar_empty()) {
7692c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    ObjCInterfaceDecl::ivar_iterator I = ivar_begin(), E = ivar_end();
7702e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    data().IvarList = (*I); ++I;
7712e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    for (curIvar = data().IvarList; I != E; curIvar = *I, ++I)
7722c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian      curIvar->setNextIvar(*I);
7732c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian  }
7742c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian
7752c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian  for (const ObjCCategoryDecl *CDecl = getFirstClassExtension(); CDecl;
7762c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian       CDecl = CDecl->getNextClassExtension()) {
7772c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    if (!CDecl->ivar_empty()) {
7782c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian      ObjCCategoryDecl::ivar_iterator I = CDecl->ivar_begin(),
7792c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian                                          E = CDecl->ivar_end();
7802e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor      if (!data().IvarList) {
7812e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor        data().IvarList = (*I); ++I;
7822e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor        curIvar = data().IvarList;
7832c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian      }
7842c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian      for ( ;I != E; curIvar = *I, ++I)
7852c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian        curIvar->setNextIvar(*I);
7862c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    }
7872c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian  }
7882c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian
7892c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian  if (ObjCImplementationDecl *ImplDecl = getImplementation()) {
7902c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    if (!ImplDecl->ivar_empty()) {
7912c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian      ObjCImplementationDecl::ivar_iterator I = ImplDecl->ivar_begin(),
7922c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian                                            E = ImplDecl->ivar_end();
7932e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor      if (!data().IvarList) {
7942e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor        data().IvarList = (*I); ++I;
7952e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor        curIvar = data().IvarList;
7962c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian      }
7972c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian      for ( ;I != E; curIvar = *I, ++I)
7982c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian        curIvar->setNextIvar(*I);
7992c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    }
8002c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian  }
8012e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  return data().IvarList;
8022c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian}
8038a779314870760848e61da2c428a78971fe3f1c3Ted Kremenek
804ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// FindCategoryDeclaration - Finds category declaration in the list of
805ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// categories for this class and returns it. Name of the category is passed
806ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// in 'CategoryId'. If category not found, return 0;
807ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner///
808ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris LattnerObjCCategoryDecl *
809ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris LattnerObjCInterfaceDecl::FindCategoryDeclaration(IdentifierInfo *CategoryId) const {
8105a61e0c4229dc82617a37a68ac2fc08d46e00488Argyrios Kyrtzidis  // FIXME: Should make sure no callers ever do this.
8115a61e0c4229dc82617a37a68ac2fc08d46e00488Argyrios Kyrtzidis  if (!hasDefinition())
8125a61e0c4229dc82617a37a68ac2fc08d46e00488Argyrios Kyrtzidis    return 0;
8135a61e0c4229dc82617a37a68ac2fc08d46e00488Argyrios Kyrtzidis
8142e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (data().ExternallyCompleted)
81526ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor    LoadExternalDefinition();
81626ac3f30ecef21749c00a4b1a08dd15d772dd5aaDouglas Gregor
817ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  for (ObjCCategoryDecl *Category = getCategoryList();
818ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner       Category; Category = Category->getNextClassCategory())
819ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    if (Category->getIdentifier() == CategoryId)
820ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner      return Category;
821ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  return 0;
822ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner}
823ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
8241cb35dd4840d21cec58648361180d5688446a9caArgyrios KyrtzidisObjCMethodDecl *
8251cb35dd4840d21cec58648361180d5688446a9caArgyrios KyrtzidisObjCInterfaceDecl::getCategoryInstanceMethod(Selector Sel) const {
8261cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis  for (ObjCCategoryDecl *Category = getCategoryList();
8271cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis       Category; Category = Category->getNextClassCategory())
8281cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis    if (ObjCCategoryImplDecl *Impl = Category->getImplementation())
8291cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis      if (ObjCMethodDecl *MD = Impl->getInstanceMethod(Sel))
8301cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis        return MD;
8311cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis  return 0;
8321cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis}
8331cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis
8341cb35dd4840d21cec58648361180d5688446a9caArgyrios KyrtzidisObjCMethodDecl *ObjCInterfaceDecl::getCategoryClassMethod(Selector Sel) const {
8351cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis  for (ObjCCategoryDecl *Category = getCategoryList();
8361cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis       Category; Category = Category->getNextClassCategory())
8371cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis    if (ObjCCategoryImplDecl *Impl = Category->getImplementation())
8381cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis      if (ObjCMethodDecl *MD = Impl->getClassMethod(Sel))
8391cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis        return MD;
8401cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis  return 0;
8411cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis}
8421cb35dd4840d21cec58648361180d5688446a9caArgyrios Kyrtzidis
8430fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian/// ClassImplementsProtocol - Checks that 'lProto' protocol
8440fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian/// has been implemented in IDecl class, its super class or categories (if
8450fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian/// lookupCategory is true).
8460fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanianbool ObjCInterfaceDecl::ClassImplementsProtocol(ObjCProtocolDecl *lProto,
8470fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian                                    bool lookupCategory,
8480fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian                                    bool RHSIsQualifiedID) {
8492e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor  if (!hasDefinition())
8502e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    return false;
8512e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor
8520fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian  ObjCInterfaceDecl *IDecl = this;
8530fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian  // 1st, look up the class.
854a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis  for (ObjCInterfaceDecl::protocol_iterator
855a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis        PI = IDecl->protocol_begin(), E = IDecl->protocol_end(); PI != E; ++PI){
8560fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian    if (getASTContext().ProtocolCompatibleWithProtocol(lProto, *PI))
8570fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian      return true;
8580fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian    // This is dubious and is added to be compatible with gcc.  In gcc, it is
8590fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian    // also allowed assigning a protocol-qualified 'id' type to a LHS object
8600fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian    // when protocol in qualified LHS is in list of protocols in the rhs 'id'
8610fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian    // object. This IMO, should be a bug.
8620fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian    // FIXME: Treat this as an extension, and flag this as an error when GCC
8630fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian    // extensions are not enabled.
8641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    if (RHSIsQualifiedID &&
8650fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian        getASTContext().ProtocolCompatibleWithProtocol(*PI, lProto))
8660fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian      return true;
8670fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian  }
8681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8690fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian  // 2nd, look up the category.
8700fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian  if (lookupCategory)
8710fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian    for (ObjCCategoryDecl *CDecl = IDecl->getCategoryList(); CDecl;
8720fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian         CDecl = CDecl->getNextClassCategory()) {
8730fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian      for (ObjCCategoryDecl::protocol_iterator PI = CDecl->protocol_begin(),
8740fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian           E = CDecl->protocol_end(); PI != E; ++PI)
8750fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian        if (getASTContext().ProtocolCompatibleWithProtocol(lProto, *PI))
8760fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian          return true;
8770fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian    }
8781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8790fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian  // 3rd, look up the super class(s)
8800fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian  if (IDecl->getSuperClass())
8810fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian    return
8820fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian  IDecl->getSuperClass()->ClassImplementsProtocol(lProto, lookupCategory,
8830fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian                                                  RHSIsQualifiedID);
8841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8850fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian  return false;
8860fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian}
8870fd8904c5f71a11d29f67716c3ebdf7ad1c855fbFariborz Jahanian
888ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
889ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCIvarDecl
890ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
891ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
89299ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCIvarDecl::anchor() { }
89399ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
894a06549226f45d5b72169a3d054415616dd1014a2Daniel DunbarObjCIvarDecl *ObjCIvarDecl::Create(ASTContext &C, ObjCContainerDecl *DC,
895ff676cb48fe8bf7be2feaa251dc7c5fb15af4730Abramo Bagnara                                   SourceLocation StartLoc,
896ff676cb48fe8bf7be2feaa251dc7c5fb15af4730Abramo Bagnara                                   SourceLocation IdLoc, IdentifierInfo *Id,
897a93c934af4fbf97cbe8e649d82e68ccacfe57c95John McCall                                   QualType T, TypeSourceInfo *TInfo,
898ad51e74030a59a8aa4ef0ebca1d7a701602ef53bFariborz Jahanian                                   AccessControl ac, Expr *BW,
899ad51e74030a59a8aa4ef0ebca1d7a701602ef53bFariborz Jahanian                                   bool synthesized) {
900a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar  if (DC) {
901a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    // Ivar's can only appear in interfaces, implementations (via synthesized
902a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    // properties), and class extensions (via direct declaration, or synthesized
903a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    // properties).
904a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    //
905a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    // FIXME: This should really be asserting this:
906a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    //   (isa<ObjCCategoryDecl>(DC) &&
907a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    //    cast<ObjCCategoryDecl>(DC)->IsClassExtension()))
908a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    // but unfortunately we sometimes place ivars into non-class extension
909a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    // categories on error. This breaks an AST invariant, and should not be
910a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    // fixed.
911a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar    assert((isa<ObjCInterfaceDecl>(DC) || isa<ObjCImplementationDecl>(DC) ||
912a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar            isa<ObjCCategoryDecl>(DC)) &&
913a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar           "Invalid ivar decl context!");
9142c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    // Once a new ivar is created in any of class/class-extension/implementation
9152c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    // decl contexts, the previously built IvarList must be rebuilt.
9162c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(DC);
9172c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    if (!ID) {
918000835d0b04345c0014c603fe6339b3bc154050eFariborz Jahanian      if (ObjCImplementationDecl *IM = dyn_cast<ObjCImplementationDecl>(DC)) {
9192c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian        ID = IM->getClassInterface();
920000835d0b04345c0014c603fe6339b3bc154050eFariborz Jahanian        if (BW)
921000835d0b04345c0014c603fe6339b3bc154050eFariborz Jahanian          IM->setHasSynthBitfield(true);
9223060178ad9df29789505c1e6debcfc80a3a13587Chad Rosier      } else {
923000835d0b04345c0014c603fe6339b3bc154050eFariborz Jahanian        ObjCCategoryDecl *CD = cast<ObjCCategoryDecl>(DC);
924000835d0b04345c0014c603fe6339b3bc154050eFariborz Jahanian        ID = CD->getClassInterface();
925000835d0b04345c0014c603fe6339b3bc154050eFariborz Jahanian        if (BW)
926000835d0b04345c0014c603fe6339b3bc154050eFariborz Jahanian          CD->setHasSynthBitfield(true);
927000835d0b04345c0014c603fe6339b3bc154050eFariborz Jahanian      }
9282c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    }
9292c18bb7c9fca66c30b6eabbdcbc6399d24a54fa9Fariborz Jahanian    ID->setIvarList(0);
930a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar  }
931a06549226f45d5b72169a3d054415616dd1014a2Daniel Dunbar
932ff676cb48fe8bf7be2feaa251dc7c5fb15af4730Abramo Bagnara  return new (C) ObjCIvarDecl(DC, StartLoc, IdLoc, Id, T, TInfo,
933ff676cb48fe8bf7be2feaa251dc7c5fb15af4730Abramo Bagnara                              ac, BW, synthesized);
9346c4ae5de0c356777446f823b573821fb95560d91Chris Lattner}
9356c4ae5de0c356777446f823b573821fb95560d91Chris Lattner
9361e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCIvarDecl *ObjCIvarDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
9371e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCIvarDecl));
9381e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCIvarDecl(0, SourceLocation(), SourceLocation(), 0,
9391e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                QualType(), 0, ObjCIvarDecl::None, 0, false);
9401e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor}
9411e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor
94227a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbarconst ObjCInterfaceDecl *ObjCIvarDecl::getContainingInterface() const {
94327a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar  const ObjCContainerDecl *DC = cast<ObjCContainerDecl>(getDeclContext());
94427a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar
94527a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar  switch (DC->getKind()) {
94627a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar  default:
94727a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar  case ObjCCategoryImpl:
94827a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar  case ObjCProtocol:
949b219cfc4d75f0a03630b7c4509ef791b7e97b2c8David Blaikie    llvm_unreachable("invalid ivar container!");
95027a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar
95127a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar    // Ivars can only appear in class extension categories.
95227a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar  case ObjCCategory: {
95327a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar    const ObjCCategoryDecl *CD = cast<ObjCCategoryDecl>(DC);
95427a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar    assert(CD->IsClassExtension() && "invalid container for ivar!");
95527a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar    return CD->getClassInterface();
95627a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar  }
95727a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar
95827a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar  case ObjCImplementation:
95927a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar    return cast<ObjCImplementationDecl>(DC)->getClassInterface();
96001e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
96127a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar  case ObjCInterface:
96227a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar    return cast<ObjCInterfaceDecl>(DC);
96327a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar  }
96427a961a6adab85cfcf7e48485bbec9237719ae96Daniel Dunbar}
965ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
966ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
967ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCAtDefsFieldDecl
968ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
969ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
97099ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCAtDefsFieldDecl::anchor() { }
97199ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
97201e6779faca1e3a3164c697d6e2dfee0881a6981Ted KremenekObjCAtDefsFieldDecl
973ff676cb48fe8bf7be2feaa251dc7c5fb15af4730Abramo Bagnara*ObjCAtDefsFieldDecl::Create(ASTContext &C, DeclContext *DC,
974ff676cb48fe8bf7be2feaa251dc7c5fb15af4730Abramo Bagnara                             SourceLocation StartLoc,  SourceLocation IdLoc,
97501e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek                             IdentifierInfo *Id, QualType T, Expr *BW) {
976ff676cb48fe8bf7be2feaa251dc7c5fb15af4730Abramo Bagnara  return new (C) ObjCAtDefsFieldDecl(DC, StartLoc, IdLoc, Id, T, BW);
97701e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek}
97801e6779faca1e3a3164c697d6e2dfee0881a6981Ted Kremenek
9791e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCAtDefsFieldDecl *ObjCAtDefsFieldDecl::CreateDeserialized(ASTContext &C,
9801e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                                             unsigned ID) {
9811e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCAtDefsFieldDecl));
9821e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCAtDefsFieldDecl(0, SourceLocation(), SourceLocation(),
9831e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                       0, QualType(), 0);
9841e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor}
9851e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor
986ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
987ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCProtocolDecl
988ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
989ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
99099ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCProtocolDecl::anchor() { }
99199ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
99227c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas GregorObjCProtocolDecl::ObjCProtocolDecl(DeclContext *DC, IdentifierInfo *Id,
99327c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor                                   SourceLocation nameLoc,
99427c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor                                   SourceLocation atStartLoc,
995c9d3c7edb513e9b8a6ab65b04133653e71d7a72bDouglas Gregor                                   ObjCProtocolDecl *PrevDecl)
996c9d3c7edb513e9b8a6ab65b04133653e71d7a72bDouglas Gregor  : ObjCContainerDecl(ObjCProtocol, DC, Id, nameLoc, atStartLoc), Data()
99727c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor{
99827c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor  setPreviousDeclaration(PrevDecl);
99927c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor  if (PrevDecl)
100027c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor    Data = PrevDecl->Data;
100127c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor}
100227c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor
1003d04341000d35c8808a72838b057eed7bf13b7661Douglas GregorObjCProtocolDecl *ObjCProtocolDecl::Create(ASTContext &C, DeclContext *DC,
10041711fc91efb36d131f7ba771f73f0154dc1abd1fArgyrios Kyrtzidis                                           IdentifierInfo *Id,
10051711fc91efb36d131f7ba771f73f0154dc1abd1fArgyrios Kyrtzidis                                           SourceLocation nameLoc,
1006b05d7b20171bbd2feb14b059f39332cbe1bf1014Argyrios Kyrtzidis                                           SourceLocation atStartLoc,
1007c9d3c7edb513e9b8a6ab65b04133653e71d7a72bDouglas Gregor                                           ObjCProtocolDecl *PrevDecl) {
100827c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor  ObjCProtocolDecl *Result
1009c9d3c7edb513e9b8a6ab65b04133653e71d7a72bDouglas Gregor    = new (C) ObjCProtocolDecl(DC, Id, nameLoc, atStartLoc, PrevDecl);
101027c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor
101127c6da284f90e32cda0ec8f52a2b6ba5a2613252Douglas Gregor  return Result;
1012cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner}
1013cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner
10141e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCProtocolDecl *ObjCProtocolDecl::CreateDeserialized(ASTContext &C,
10151e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                                       unsigned ID) {
10161e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCProtocolDecl));
10171e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCProtocolDecl(0, 0, SourceLocation(), SourceLocation(),
10181e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                    0);
10191e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor}
10201e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor
102191b0b0cf6b537cbcbca0038c7032f87161a41d31Steve NaroffObjCProtocolDecl *ObjCProtocolDecl::lookupProtocolNamed(IdentifierInfo *Name) {
102291b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff  ObjCProtocolDecl *PDecl = this;
102391b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff
102491b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff  if (Name == getIdentifier())
102591b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff    return PDecl;
102691b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff
102791b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff  for (protocol_iterator I = protocol_begin(), E = protocol_end(); I != E; ++I)
102891b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff    if ((PDecl = (*I)->lookupProtocolNamed(Name)))
102991b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff      return PDecl;
10301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
103191b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff  return NULL;
103291b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff}
103391b0b0cf6b537cbcbca0038c7032f87161a41d31Steve Naroff
1034094e2bb6730d63e0f6919e4839522a43b7644181Argyrios Kyrtzidis// lookupMethod - Lookup a instance/class method in the protocol and protocols
1035ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// it inherited.
1036094e2bb6730d63e0f6919e4839522a43b7644181Argyrios KyrtzidisObjCMethodDecl *ObjCProtocolDecl::lookupMethod(Selector Sel,
1037094e2bb6730d63e0f6919e4839522a43b7644181Argyrios Kyrtzidis                                               bool isInstance) const {
1038ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  ObjCMethodDecl *MethodDecl = NULL;
10391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1040094e2bb6730d63e0f6919e4839522a43b7644181Argyrios Kyrtzidis  if ((MethodDecl = getMethod(Sel, isInstance)))
1041ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    return MethodDecl;
10421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1043ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  for (protocol_iterator I = protocol_begin(), E = protocol_end(); I != E; ++I)
1044094e2bb6730d63e0f6919e4839522a43b7644181Argyrios Kyrtzidis    if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
1045ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner      return MethodDecl;
1046ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  return NULL;
1047ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner}
1048411280e5b25ba7dcd7c8a82a5c23880fe7632a3cChris Lattner
10495e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregorvoid ObjCProtocolDecl::allocateDefinitionData() {
10505e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregor  assert(!Data && "Protocol already has a definition!");
10511d784b277cdfd4eba03680715d2a082b3f28d295Douglas Gregor  Data = new (getASTContext()) DefinitionData;
10521d784b277cdfd4eba03680715d2a082b3f28d295Douglas Gregor  Data->Definition = this;
10535e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregor}
10545e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregor
10555e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregorvoid ObjCProtocolDecl::startDefinition() {
10565e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregor  allocateDefinitionData();
10571d784b277cdfd4eba03680715d2a082b3f28d295Douglas Gregor
10581d784b277cdfd4eba03680715d2a082b3f28d295Douglas Gregor  // Update all of the declarations with a pointer to the definition.
10591d784b277cdfd4eba03680715d2a082b3f28d295Douglas Gregor  for (redecl_iterator RD = redecls_begin(), RDEnd = redecls_end();
10601d784b277cdfd4eba03680715d2a082b3f28d295Douglas Gregor       RD != RDEnd; ++RD)
10611d784b277cdfd4eba03680715d2a082b3f28d295Douglas Gregor    RD->Data = this->Data;
1062ad834d534e9a5db3d3baa09593775f83ceaff1f2Argyrios Kyrtzidis}
1063ad834d534e9a5db3d3baa09593775f83ceaff1f2Argyrios Kyrtzidis
1064ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
1065ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCCategoryDecl
1066ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
1067ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
106899ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCCategoryDecl::anchor() { }
106999ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
1070d04341000d35c8808a72838b057eed7bf13b7661Douglas GregorObjCCategoryDecl *ObjCCategoryDecl::Create(ASTContext &C, DeclContext *DC,
10713db211b617c5073aa70eb25d37ed44ae0dca17c4Douglas Gregor                                           SourceLocation AtLoc,
10723db211b617c5073aa70eb25d37ed44ae0dca17c4Douglas Gregor                                           SourceLocation ClassNameLoc,
10733db211b617c5073aa70eb25d37ed44ae0dca17c4Douglas Gregor                                           SourceLocation CategoryNameLoc,
1074955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis                                           IdentifierInfo *Id,
1075af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian                                           ObjCInterfaceDecl *IDecl,
1076af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian                                           SourceLocation IvarLBraceLoc,
1077af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian                                           SourceLocation IvarRBraceLoc) {
1078955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis  ObjCCategoryDecl *CatDecl = new (C) ObjCCategoryDecl(DC, AtLoc, ClassNameLoc,
1079955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis                                                       CategoryNameLoc, Id,
1080af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian                                                       IDecl,
1081af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian                                                       IvarLBraceLoc, IvarRBraceLoc);
1082955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis  if (IDecl) {
1083955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis    // Link this category into its class's category list.
1084955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis    CatDecl->NextClassCategory = IDecl->getCategoryList();
10852e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    if (IDecl->hasDefinition()) {
10862e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor      IDecl->setCategoryList(CatDecl);
10872e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor      if (ASTMutationListener *L = C.getASTMutationListener())
10882e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor        L->AddedObjCCategoryToInterface(CatDecl, IDecl);
10892e5c15be82f362611c5928ce853d0685ff98c766Douglas Gregor    }
1090955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis  }
1091955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis
1092955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis  return CatDecl;
1093955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis}
1094955fadbdfecfa24a590febe66a86519096787f2dArgyrios Kyrtzidis
10951e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCCategoryDecl *ObjCCategoryDecl::CreateDeserialized(ASTContext &C,
10961e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                                       unsigned ID) {
10971e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCCategoryDecl));
10981e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCCategoryDecl(0, SourceLocation(), SourceLocation(),
10991e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                    SourceLocation(), 0, 0);
110061f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner}
110161f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner
11028a1d722f13df383600f36d77f842957c8adb5f1bArgyrios KyrtzidisObjCCategoryImplDecl *ObjCCategoryDecl::getImplementation() const {
11038a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis  return getASTContext().getObjCImplementation(
11048a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis                                           const_cast<ObjCCategoryDecl*>(this));
11058a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis}
11068a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis
11078a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidisvoid ObjCCategoryDecl::setImplementation(ObjCCategoryImplDecl *ImplD) {
11088a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis  getASTContext().setObjCImplementation(this, ImplD);
11098a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis}
11108a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis
11118a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis
1112ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
1113ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCCategoryImplDecl
1114ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
1115ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner
111699ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCCategoryImplDecl::anchor() { }
111799ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
111875c9cae5f85c72cbb1649e93849e16ede3f07522Chris LattnerObjCCategoryImplDecl *
1119d04341000d35c8808a72838b057eed7bf13b7661Douglas GregorObjCCategoryImplDecl::Create(ASTContext &C, DeclContext *DC,
11201711fc91efb36d131f7ba771f73f0154dc1abd1fArgyrios Kyrtzidis                             IdentifierInfo *Id,
11211711fc91efb36d131f7ba771f73f0154dc1abd1fArgyrios Kyrtzidis                             ObjCInterfaceDecl *ClassInterface,
11221711fc91efb36d131f7ba771f73f0154dc1abd1fArgyrios Kyrtzidis                             SourceLocation nameLoc,
1123c6994005dc9f677c831b8e90bdab483cc2197c29Argyrios Kyrtzidis                             SourceLocation atStartLoc,
1124c6994005dc9f677c831b8e90bdab483cc2197c29Argyrios Kyrtzidis                             SourceLocation CategoryNameLoc) {
1125712ef874534ee1bef41d1aa4664ae36148ec8b12Fariborz Jahanian  if (ClassInterface && ClassInterface->hasDefinition())
1126712ef874534ee1bef41d1aa4664ae36148ec8b12Fariborz Jahanian    ClassInterface = ClassInterface->getDefinition();
11271711fc91efb36d131f7ba771f73f0154dc1abd1fArgyrios Kyrtzidis  return new (C) ObjCCategoryImplDecl(DC, Id, ClassInterface,
1128c6994005dc9f677c831b8e90bdab483cc2197c29Argyrios Kyrtzidis                                      nameLoc, atStartLoc, CategoryNameLoc);
112975c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner}
113075c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner
11311e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCCategoryImplDecl *ObjCCategoryImplDecl::CreateDeserialized(ASTContext &C,
11321e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                                               unsigned ID) {
11331e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCCategoryImplDecl));
11341e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCCategoryImplDecl(0, 0, 0, SourceLocation(),
11351e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                        SourceLocation(), SourceLocation());
11361e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor}
11371e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor
11380d69b8cc8e90a9364771837cb42d7031b4cbb984Steve NaroffObjCCategoryDecl *ObjCCategoryImplDecl::getCategoryDecl() const {
1139ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  // The class interface might be NULL if we are working with invalid code.
1140ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  if (const ObjCInterfaceDecl *ID = getClassInterface())
1141ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek    return ID->FindCategoryDeclaration(getIdentifier());
1142ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  return 0;
11434292073a858f72769fa405b48390620c8932f8aeArgyrios Kyrtzidis}
11444292073a858f72769fa405b48390620c8932f8aeArgyrios Kyrtzidis
1145f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner
114699ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCImplDecl::anchor() { }
114799ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
114817945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidisvoid ObjCImplDecl::addPropertyImplementation(ObjCPropertyImplDecl *property) {
11492c2d43c557beca1b4ba4bd743f33978aecb46a97Douglas Gregor  // FIXME: The context should be correct before we get here.
1150653f1b1bf293a9bd96fd4dd6372e779cc7af1597Douglas Gregor  property->setLexicalDeclContext(this);
115117945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis  addDecl(property);
1152653f1b1bf293a9bd96fd4dd6372e779cc7af1597Douglas Gregor}
1153653f1b1bf293a9bd96fd4dd6372e779cc7af1597Douglas Gregor
11548a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidisvoid ObjCImplDecl::setClassInterface(ObjCInterfaceDecl *IFace) {
11558a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis  ASTContext &Ctx = getASTContext();
11568a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis
11578a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis  if (ObjCImplementationDecl *ImplD
115898f2cca4b2731b5d43da7c1582dd443ecead658dDuncan Sands        = dyn_cast_or_null<ObjCImplementationDecl>(this)) {
11598a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis    if (IFace)
11608a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis      Ctx.setObjCImplementation(IFace, ImplD);
11618a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis
116298f2cca4b2731b5d43da7c1582dd443ecead658dDuncan Sands  } else if (ObjCCategoryImplDecl *ImplD =
11638a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis             dyn_cast_or_null<ObjCCategoryImplDecl>(this)) {
11648a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis    if (ObjCCategoryDecl *CD = IFace->FindCategoryDeclaration(getIdentifier()))
11658a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis      Ctx.setObjCImplementation(CD, ImplD);
11668a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis  }
11678a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis
11688a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis  ClassInterface = IFace;
11698a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis}
11708a1d722f13df383600f36d77f842957c8adb5f1bArgyrios Kyrtzidis
1171ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// FindPropertyImplIvarDecl - This method lookup the ivar in the list of
1172ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// properties implemented in this category @implementation block and returns
1173ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// the implemented property that uses it.
1174aaa63a761c6671a08e3f4f463435b72739fa194bFariborz Jahanian///
11753aa1861bd8b5121e53379b1a00f9d6ad8dead4f6Chris LattnerObjCPropertyImplDecl *ObjCImplDecl::
117617945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios KyrtzidisFindPropertyImplIvarDecl(IdentifierInfo *ivarId) const {
117717945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis  for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i){
1178ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    ObjCPropertyImplDecl *PID = *i;
1179ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    if (PID->getPropertyIvarDecl() &&
1180ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner        PID->getPropertyIvarDecl()->getIdentifier() == ivarId)
1181ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner      return PID;
1182ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  }
11830701bbb228dfd87e1fe82a0a4b7b9facfecb43daSteve Naroff  return 0;
11840701bbb228dfd87e1fe82a0a4b7b9facfecb43daSteve Naroff}
11850701bbb228dfd87e1fe82a0a4b7b9facfecb43daSteve Naroff
1186ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// FindPropertyImplDecl - This method looks up a previous ObjCPropertyImplDecl
1187ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// added to the list of those properties @synthesized/@dynamic in this
1188ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner/// category @implementation block.
1189559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian///
11903aa1861bd8b5121e53379b1a00f9d6ad8dead4f6Chris LattnerObjCPropertyImplDecl *ObjCImplDecl::
119117945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios KyrtzidisFindPropertyImplDecl(IdentifierInfo *Id) const {
119217945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis  for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i){
1193ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    ObjCPropertyImplDecl *PID = *i;
1194ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner    if (PID->getPropertyDecl()->getIdentifier() == Id)
1195ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner      return PID;
1196f034e9cc4dad81d8fe6eb88a84da55b2909a9cddFariborz Jahanian  }
1197559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian  return 0;
1198559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian}
1199559c0c4bbecc017aab0716d546c4fefbcc194687Fariborz Jahanian
12005f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnerraw_ostream &clang::operator<<(raw_ostream &OS,
1201f978059b82db8c0d849c5f992036210b5ca53200Benjamin Kramer                               const ObjCCategoryImplDecl &CID) {
1202f978059b82db8c0d849c5f992036210b5ca53200Benjamin Kramer  OS << CID.getName();
1203900fc6388e803868a34b9483510c345e9b49d7ebBenjamin Kramer  return OS;
1204900fc6388e803868a34b9483510c345e9b49d7ebBenjamin Kramer}
1205900fc6388e803868a34b9483510c345e9b49d7ebBenjamin Kramer
1206ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
1207ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCImplementationDecl
1208ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
12091e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner
121099ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCImplementationDecl::anchor() { }
121199ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
1212ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris LattnerObjCImplementationDecl *
12131eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpObjCImplementationDecl::Create(ASTContext &C, DeclContext *DC,
1214ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner                               ObjCInterfaceDecl *ClassInterface,
12151711fc91efb36d131f7ba771f73f0154dc1abd1fArgyrios Kyrtzidis                               ObjCInterfaceDecl *SuperDecl,
12161711fc91efb36d131f7ba771f73f0154dc1abd1fArgyrios Kyrtzidis                               SourceLocation nameLoc,
1217af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian                               SourceLocation atStartLoc,
1218af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian                               SourceLocation IvarLBraceLoc,
1219af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian                               SourceLocation IvarRBraceLoc) {
1220712ef874534ee1bef41d1aa4664ae36148ec8b12Fariborz Jahanian  if (ClassInterface && ClassInterface->hasDefinition())
1221712ef874534ee1bef41d1aa4664ae36148ec8b12Fariborz Jahanian    ClassInterface = ClassInterface->getDefinition();
12221711fc91efb36d131f7ba771f73f0154dc1abd1fArgyrios Kyrtzidis  return new (C) ObjCImplementationDecl(DC, ClassInterface, SuperDecl,
1223af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian                                        nameLoc, atStartLoc,
1224af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian                                        IvarLBraceLoc, IvarRBraceLoc);
1225ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner}
12261e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner
12271e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCImplementationDecl *
12281e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCImplementationDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
12291e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCImplementationDecl));
12301e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCImplementationDecl(0, 0, 0, SourceLocation(),
12311e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                          SourceLocation());
12321e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor}
12331e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor
1234da6d976b0f2470bb3f854913bc3af3245845ad60John McCallvoid ObjCImplementationDecl::setIvarInitializers(ASTContext &C,
1235da6d976b0f2470bb3f854913bc3af3245845ad60John McCall                                             CXXCtorInitializer ** initializers,
1236da6d976b0f2470bb3f854913bc3af3245845ad60John McCall                                                 unsigned numInitializers) {
1237da6d976b0f2470bb3f854913bc3af3245845ad60John McCall  if (numInitializers > 0) {
1238da6d976b0f2470bb3f854913bc3af3245845ad60John McCall    NumIvarInitializers = numInitializers;
1239da6d976b0f2470bb3f854913bc3af3245845ad60John McCall    CXXCtorInitializer **ivarInitializers =
1240da6d976b0f2470bb3f854913bc3af3245845ad60John McCall    new (C) CXXCtorInitializer*[NumIvarInitializers];
1241da6d976b0f2470bb3f854913bc3af3245845ad60John McCall    memcpy(ivarInitializers, initializers,
1242da6d976b0f2470bb3f854913bc3af3245845ad60John McCall           numInitializers * sizeof(CXXCtorInitializer*));
1243da6d976b0f2470bb3f854913bc3af3245845ad60John McCall    IvarInitializers = ivarInitializers;
1244da6d976b0f2470bb3f854913bc3af3245845ad60John McCall  }
1245da6d976b0f2470bb3f854913bc3af3245845ad60John McCall}
1246da6d976b0f2470bb3f854913bc3af3245845ad60John McCall
12475f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnerraw_ostream &clang::operator<<(raw_ostream &OS,
1248f978059b82db8c0d849c5f992036210b5ca53200Benjamin Kramer                               const ObjCImplementationDecl &ID) {
1249f978059b82db8c0d849c5f992036210b5ca53200Benjamin Kramer  OS << ID.getName();
1250900fc6388e803868a34b9483510c345e9b49d7ebBenjamin Kramer  return OS;
1251900fc6388e803868a34b9483510c345e9b49d7ebBenjamin Kramer}
1252900fc6388e803868a34b9483510c345e9b49d7ebBenjamin Kramer
1253ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
1254ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCCompatibleAliasDecl
1255ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
12561e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner
125799ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCCompatibleAliasDecl::anchor() { }
125899ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
1259ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris LattnerObjCCompatibleAliasDecl *
1260ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris LattnerObjCCompatibleAliasDecl::Create(ASTContext &C, DeclContext *DC,
1261ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner                                SourceLocation L,
12621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                IdentifierInfo *Id,
1263ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner                                ObjCInterfaceDecl* AliasedClass) {
1264ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner  return new (C) ObjCCompatibleAliasDecl(DC, L, Id, AliasedClass);
12651e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner}
12661e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner
12671e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCCompatibleAliasDecl *
12681e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCCompatibleAliasDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
12691e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCCompatibleAliasDecl));
12701e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCCompatibleAliasDecl(0, SourceLocation(), 0, 0);
12711e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor}
12721e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor
1273ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
1274ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCPropertyDecl
1275ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
12761e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner
127799ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikievoid ObjCPropertyDecl::anchor() { }
127899ba9e3bd70671f3441fb974895f226a83ce0e66David Blaikie
1279ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris LattnerObjCPropertyDecl *ObjCPropertyDecl::Create(ASTContext &C, DeclContext *DC,
1280ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner                                           SourceLocation L,
1281ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner                                           IdentifierInfo *Id,
1282d0502407c1b41b2ace326f355d7b7a6876246223Fariborz Jahanian                                           SourceLocation AtLoc,
128377bfb8b43ec3f7dee3a71f6e854b03ad29dab84fFariborz Jahanian                                           SourceLocation LParenLoc,
128483a230c83a54190366138c1a4f4310ef838b88fcJohn McCall                                           TypeSourceInfo *T,
1285ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner                                           PropertyControl propControl) {
128677bfb8b43ec3f7dee3a71f6e854b03ad29dab84fFariborz Jahanian  return new (C) ObjCPropertyDecl(DC, L, Id, AtLoc, LParenLoc, T);
12871e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner}
12881e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner
12891e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCPropertyDecl *ObjCPropertyDecl::CreateDeserialized(ASTContext &C,
12901e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                                       unsigned ID) {
12911e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void * Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCPropertyDecl));
12921e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCPropertyDecl(0, SourceLocation(), 0, SourceLocation(),
129377bfb8b43ec3f7dee3a71f6e854b03ad29dab84fFariborz Jahanian                                    SourceLocation(),
12941e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                    0);
12951e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor}
12961e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor
1297ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
1298ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner// ObjCPropertyImplDecl
1299ab35163a5b80bf1bd49f0eebb708970f2b0e04e9Chris Lattner//===----------------------------------------------------------------------===//
1300f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner
1301628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz JahanianObjCPropertyImplDecl *ObjCPropertyImplDecl::Create(ASTContext &C,
1302d04341000d35c8808a72838b057eed7bf13b7661Douglas Gregor                                                   DeclContext *DC,
1303628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                                   SourceLocation atLoc,
1304628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                                   SourceLocation L,
1305628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian                                                   ObjCPropertyDecl *property,
13069f0afd4e79601d9982072ff9318e6f9a982c770eDaniel Dunbar                                                   Kind PK,
1307a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor                                                   ObjCIvarDecl *ivar,
1308a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor                                                   SourceLocation ivarLoc) {
1309a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  return new (C) ObjCPropertyImplDecl(DC, atLoc, L, property, PK, ivar,
1310a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor                                      ivarLoc);
1311628b96f34e93b643b6e15e75eabb8d96079a7e27Fariborz Jahanian}
1312f4af5154571e0c5eadb19df10e65464766ef6683Chris Lattner
13131e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas GregorObjCPropertyImplDecl *ObjCPropertyImplDecl::CreateDeserialized(ASTContext &C,
13141e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                                               unsigned ID) {
13151e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCPropertyImplDecl));
13161e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor  return new (Mem) ObjCPropertyImplDecl(0, SourceLocation(), SourceLocation(),
13171e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor                                        0, Dynamic, 0, SourceLocation());
13181e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor}
13191e68ecc4fcce12f683c4fd38acfd1a004001b04fDouglas Gregor
1320a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas GregorSourceRange ObjCPropertyImplDecl::getSourceRange() const {
1321a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  SourceLocation EndLoc = getLocation();
1322a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  if (IvarLoc.isValid())
1323a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor    EndLoc = IvarLoc;
13240ed844b04ea4387caa4e1cf3dc375d269657536bChris Lattner
1325a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  return SourceRange(AtLoc, EndLoc);
1326a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor}
1327