blob: 4d48ad8e4f5ac888de4e816738c22bd7a10c3bd4 [file] [log] [blame]
Chris Lattner1e03a562008-03-16 00:19:01 +00001//===--- DeclObjC.cpp - ObjC Declaration AST Node Implementation ----------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the Objective-C related Decl classes.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/AST/DeclObjC.h"
15#include "clang/AST/ASTContext.h"
Daniel Dunbare91593e2008-08-11 04:54:23 +000016#include "clang/AST/Stmt.h"
Argyrios Kyrtzidise6b8d682011-09-01 00:58:55 +000017#include "clang/AST/ASTMutationListener.h"
Steve Naroff0de21fd2009-02-22 19:35:57 +000018#include "llvm/ADT/STLExtras.h"
Chris Lattner1e03a562008-03-16 00:19:01 +000019using namespace clang;
20
Chris Lattner6c4ae5d2008-03-16 00:49:28 +000021//===----------------------------------------------------------------------===//
Chris Lattner11e1e1a2009-02-20 21:16:26 +000022// ObjCListBase
23//===----------------------------------------------------------------------===//
24
Chris Lattner38af2de2009-02-20 21:35:13 +000025void ObjCListBase::set(void *const* InList, unsigned Elts, ASTContext &Ctx) {
Douglas Gregorff331c12010-07-25 18:17:45 +000026 List = 0;
Chris Lattner11e1e1a2009-02-20 21:16:26 +000027 if (Elts == 0) return; // Setting to an empty list is a noop.
Mike Stump1eb44332009-09-09 15:08:12 +000028
29
Chris Lattner4ee413b2009-02-20 21:44:01 +000030 List = new (Ctx) void*[Elts];
Chris Lattner11e1e1a2009-02-20 21:16:26 +000031 NumElts = Elts;
32 memcpy(List, InList, sizeof(void*)*Elts);
33}
34
Douglas Gregor18df52b2010-01-16 15:02:53 +000035void ObjCProtocolList::set(ObjCProtocolDecl* const* InList, unsigned Elts,
36 const SourceLocation *Locs, ASTContext &Ctx) {
37 if (Elts == 0)
38 return;
39
40 Locations = new (Ctx) SourceLocation[Elts];
41 memcpy(Locations, Locs, sizeof(SourceLocation) * Elts);
42 set(InList, Elts, Ctx);
43}
44
Chris Lattner11e1e1a2009-02-20 21:16:26 +000045//===----------------------------------------------------------------------===//
Chris Lattnerab351632009-02-20 20:59:54 +000046// ObjCInterfaceDecl
Chris Lattner6c4ae5d2008-03-16 00:49:28 +000047//===----------------------------------------------------------------------===//
48
David Blaikie99ba9e32011-12-20 02:48:34 +000049void ObjCContainerDecl::anchor() { }
50
Fariborz Jahanian496b5a82009-06-05 18:16:35 +000051/// getIvarDecl - This method looks up an ivar in this ContextDecl.
52///
53ObjCIvarDecl *
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +000054ObjCContainerDecl::getIvarDecl(IdentifierInfo *Id) const {
Fariborz Jahanian496b5a82009-06-05 18:16:35 +000055 lookup_const_iterator Ivar, IvarEnd;
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +000056 for (llvm::tie(Ivar, IvarEnd) = lookup(Id); Ivar != IvarEnd; ++Ivar) {
Fariborz Jahanian496b5a82009-06-05 18:16:35 +000057 if (ObjCIvarDecl *ivar = dyn_cast<ObjCIvarDecl>(*Ivar))
58 return ivar;
59 }
60 return 0;
61}
62
Argyrios Kyrtzidis467c0b12009-07-25 22:15:22 +000063// Get the local instance/class method declared in this interface.
Douglas Gregor6ab35242009-04-09 21:40:53 +000064ObjCMethodDecl *
Argyrios Kyrtzidis467c0b12009-07-25 22:15:22 +000065ObjCContainerDecl::getMethod(Selector Sel, bool isInstance) const {
Steve Naroff0de21fd2009-02-22 19:35:57 +000066 // Since instance & class methods can have the same name, the loop below
67 // ensures we get the correct method.
68 //
69 // @interface Whatever
70 // - (int) class_method;
71 // + (float) class_method;
72 // @end
73 //
74 lookup_const_iterator Meth, MethEnd;
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +000075 for (llvm::tie(Meth, MethEnd) = lookup(Sel); Meth != MethEnd; ++Meth) {
Steve Naroff0de21fd2009-02-22 19:35:57 +000076 ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(*Meth);
Argyrios Kyrtzidis467c0b12009-07-25 22:15:22 +000077 if (MD && MD->isInstanceMethod() == isInstance)
Steve Naroff0de21fd2009-02-22 19:35:57 +000078 return MD;
79 }
Steve Naroff0701bbb2009-01-08 17:28:14 +000080 return 0;
81}
82
Ted Kremenek9f550ff2010-03-15 20:11:46 +000083ObjCPropertyDecl *
Ted Kremenekde09d0c2010-03-15 20:11:53 +000084ObjCPropertyDecl::findPropertyDecl(const DeclContext *DC,
Ted Kremenek9f550ff2010-03-15 20:11:46 +000085 IdentifierInfo *propertyID) {
86
Ted Kremenekde09d0c2010-03-15 20:11:53 +000087 DeclContext::lookup_const_iterator I, E;
Ted Kremenek9f550ff2010-03-15 20:11:46 +000088 llvm::tie(I, E) = DC->lookup(propertyID);
89 for ( ; I != E; ++I)
90 if (ObjCPropertyDecl *PD = dyn_cast<ObjCPropertyDecl>(*I))
91 return PD;
92
93 return 0;
94}
95
Fariborz Jahanian559c0c42008-04-21 19:04:53 +000096/// FindPropertyDeclaration - Finds declaration of the property given its name
97/// in 'PropertyId' and returns it. It returns 0, if not found.
Fariborz Jahanian559c0c42008-04-21 19:04:53 +000098ObjCPropertyDecl *
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +000099ObjCContainerDecl::FindPropertyDeclaration(IdentifierInfo *PropertyId) const {
Mike Stump1eb44332009-09-09 15:08:12 +0000100
Ted Kremenekde09d0c2010-03-15 20:11:53 +0000101 if (ObjCPropertyDecl *PD =
102 ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(this), PropertyId))
103 return PD;
Mike Stump1eb44332009-09-09 15:08:12 +0000104
Ted Kremenekde09d0c2010-03-15 20:11:53 +0000105 switch (getKind()) {
106 default:
107 break;
108 case Decl::ObjCProtocol: {
109 const ObjCProtocolDecl *PID = cast<ObjCProtocolDecl>(this);
110 for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
111 E = PID->protocol_end(); I != E; ++I)
Fariborz Jahanian25760612010-02-15 21:55:26 +0000112 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
113 return P;
Ted Kremenekde09d0c2010-03-15 20:11:53 +0000114 break;
115 }
116 case Decl::ObjCInterface: {
117 const ObjCInterfaceDecl *OID = cast<ObjCInterfaceDecl>(this);
118 // Look through categories.
119 for (ObjCCategoryDecl *Cat = OID->getCategoryList();
120 Cat; Cat = Cat->getNextClassCategory())
121 if (!Cat->IsClassExtension())
122 if (ObjCPropertyDecl *P = Cat->FindPropertyDeclaration(PropertyId))
123 return P;
124
125 // Look through protocols.
Ted Kremenek53b94412010-09-01 01:21:15 +0000126 for (ObjCInterfaceDecl::all_protocol_iterator
127 I = OID->all_referenced_protocol_begin(),
128 E = OID->all_referenced_protocol_end(); I != E; ++I)
Ted Kremenekde09d0c2010-03-15 20:11:53 +0000129 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
130 return P;
131
132 // Finally, check the super class.
133 if (const ObjCInterfaceDecl *superClass = OID->getSuperClass())
134 return superClass->FindPropertyDeclaration(PropertyId);
135 break;
136 }
137 case Decl::ObjCCategory: {
138 const ObjCCategoryDecl *OCD = cast<ObjCCategoryDecl>(this);
139 // Look through protocols.
140 if (!OCD->IsClassExtension())
141 for (ObjCCategoryDecl::protocol_iterator
142 I = OCD->protocol_begin(), E = OCD->protocol_end(); I != E; ++I)
143 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
144 return P;
145
146 break;
Fariborz Jahanianf034e9c2009-01-19 18:16:19 +0000147 }
148 }
Steve Naroff3d2c22b2008-06-05 13:55:23 +0000149 return 0;
150}
151
David Blaikie99ba9e32011-12-20 02:48:34 +0000152void ObjCInterfaceDecl::anchor() { }
153
Fariborz Jahaniana6f14e12009-11-02 22:45:15 +0000154/// FindPropertyVisibleInPrimaryClass - Finds declaration of the property
155/// with name 'PropertyId' in the primary class; including those in protocols
Ted Kremenek37cafb02010-03-15 20:30:07 +0000156/// (direct or indirect) used by the primary class.
Fariborz Jahaniana6f14e12009-11-02 22:45:15 +0000157///
158ObjCPropertyDecl *
Ted Kremenek37cafb02010-03-15 20:30:07 +0000159ObjCInterfaceDecl::FindPropertyVisibleInPrimaryClass(
Fariborz Jahaniana6f14e12009-11-02 22:45:15 +0000160 IdentifierInfo *PropertyId) const {
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000161 // FIXME: Should make sure no callers ever do this.
162 if (!hasDefinition())
163 return 0;
164
165 if (data().ExternallyCompleted)
Douglas Gregor26ac3f32010-12-01 23:49:52 +0000166 LoadExternalDefinition();
167
Ted Kremenek37cafb02010-03-15 20:30:07 +0000168 if (ObjCPropertyDecl *PD =
169 ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(this), PropertyId))
170 return PD;
171
Fariborz Jahaniana6f14e12009-11-02 22:45:15 +0000172 // Look through protocols.
Ted Kremenek53b94412010-09-01 01:21:15 +0000173 for (ObjCInterfaceDecl::all_protocol_iterator
174 I = all_referenced_protocol_begin(),
175 E = all_referenced_protocol_end(); I != E; ++I)
Fariborz Jahaniana6f14e12009-11-02 22:45:15 +0000176 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
177 return P;
Ted Kremenek37cafb02010-03-15 20:30:07 +0000178
Fariborz Jahaniana6f14e12009-11-02 22:45:15 +0000179 return 0;
180}
181
Fariborz Jahanian339798e2009-10-05 20:41:32 +0000182void ObjCInterfaceDecl::mergeClassExtensionProtocolList(
183 ObjCProtocolDecl *const* ExtList, unsigned ExtNum,
184 ASTContext &C)
185{
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000186 if (data().ExternallyCompleted)
Douglas Gregor26ac3f32010-12-01 23:49:52 +0000187 LoadExternalDefinition();
188
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000189 if (data().AllReferencedProtocols.empty() &&
190 data().ReferencedProtocols.empty()) {
191 data().AllReferencedProtocols.set(ExtList, ExtNum, C);
Fariborz Jahanian339798e2009-10-05 20:41:32 +0000192 return;
193 }
Ted Kremenek53b94412010-09-01 01:21:15 +0000194
Fariborz Jahanian339798e2009-10-05 20:41:32 +0000195 // Check for duplicate protocol in class's protocol list.
Ted Kremenek53b94412010-09-01 01:21:15 +0000196 // This is O(n*m). But it is extremely rare and number of protocols in
Fariborz Jahanian339798e2009-10-05 20:41:32 +0000197 // class or its extension are very few.
Chris Lattner5f9e2722011-07-23 10:55:15 +0000198 SmallVector<ObjCProtocolDecl*, 8> ProtocolRefs;
Fariborz Jahanian339798e2009-10-05 20:41:32 +0000199 for (unsigned i = 0; i < ExtNum; i++) {
200 bool protocolExists = false;
201 ObjCProtocolDecl *ProtoInExtension = ExtList[i];
Ted Kremenek53b94412010-09-01 01:21:15 +0000202 for (all_protocol_iterator
203 p = all_referenced_protocol_begin(),
204 e = all_referenced_protocol_end(); p != e; ++p) {
Fariborz Jahanian339798e2009-10-05 20:41:32 +0000205 ObjCProtocolDecl *Proto = (*p);
206 if (C.ProtocolCompatibleWithProtocol(ProtoInExtension, Proto)) {
207 protocolExists = true;
208 break;
209 }
210 }
211 // Do we want to warn on a protocol in extension class which
212 // already exist in the class? Probably not.
Ted Kremenek53b94412010-09-01 01:21:15 +0000213 if (!protocolExists)
Fariborz Jahanian339798e2009-10-05 20:41:32 +0000214 ProtocolRefs.push_back(ProtoInExtension);
215 }
Ted Kremenek53b94412010-09-01 01:21:15 +0000216
Fariborz Jahanian339798e2009-10-05 20:41:32 +0000217 if (ProtocolRefs.empty())
218 return;
Ted Kremenek53b94412010-09-01 01:21:15 +0000219
Fariborz Jahanianb106fc62009-10-05 21:32:49 +0000220 // Merge ProtocolRefs into class's protocol list;
Ted Kremenek53b94412010-09-01 01:21:15 +0000221 for (all_protocol_iterator p = all_referenced_protocol_begin(),
222 e = all_referenced_protocol_end(); p != e; ++p) {
Fariborz Jahanian339798e2009-10-05 20:41:32 +0000223 ProtocolRefs.push_back(*p);
Douglas Gregor18df52b2010-01-16 15:02:53 +0000224 }
Ted Kremenek53b94412010-09-01 01:21:15 +0000225
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000226 data().AllReferencedProtocols.set(ProtocolRefs.data(), ProtocolRefs.size(),C);
Fariborz Jahanian339798e2009-10-05 20:41:32 +0000227}
228
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000229void ObjCInterfaceDecl::allocateDefinitionData() {
230 assert(!hasDefinition() && "ObjC class already has a definition");
Douglas Gregor26fec632011-12-15 18:17:27 +0000231 Data = new (getASTContext()) DefinitionData();
Douglas Gregor8d2dbbf2011-12-16 16:34:57 +0000232 Data->Definition = this;
233
234 // Make the type point at the definition, now that we have one.
235 if (TypeForDecl)
236 cast<ObjCInterfaceType>(TypeForDecl)->Decl = this;
Douglas Gregor0af55012011-12-16 03:12:41 +0000237}
238
239void ObjCInterfaceDecl::startDefinition() {
240 allocateDefinitionData();
241
Douglas Gregor53df7a12011-12-15 18:03:09 +0000242 // Update all of the declarations with a pointer to the definition.
243 for (redecl_iterator RD = redecls_begin(), RDEnd = redecls_end();
244 RD != RDEnd; ++RD) {
245 if (*RD != this)
Douglas Gregor26fec632011-12-15 18:17:27 +0000246 RD->Data = Data;
Douglas Gregor53df7a12011-12-15 18:03:09 +0000247 }
Argyrios Kyrtzidisad834d52011-11-12 21:07:46 +0000248}
249
Fariborz Jahanian80aa1cd2010-06-22 23:20:40 +0000250/// getFirstClassExtension - Find first class extension of the given class.
251ObjCCategoryDecl* ObjCInterfaceDecl::getFirstClassExtension() const {
252 for (ObjCCategoryDecl *CDecl = getCategoryList(); CDecl;
Fariborz Jahanian0e5ad252010-02-23 01:26:30 +0000253 CDecl = CDecl->getNextClassCategory())
254 if (CDecl->IsClassExtension())
255 return CDecl;
256 return 0;
257}
258
Fariborz Jahanian80aa1cd2010-06-22 23:20:40 +0000259/// getNextClassCategory - Find next class extension in list of categories.
260const ObjCCategoryDecl* ObjCCategoryDecl::getNextClassExtension() const {
261 for (const ObjCCategoryDecl *CDecl = getNextClassCategory(); CDecl;
262 CDecl = CDecl->getNextClassCategory())
263 if (CDecl->IsClassExtension())
264 return CDecl;
265 return 0;
266}
267
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000268ObjCIvarDecl *ObjCInterfaceDecl::lookupInstanceVariable(IdentifierInfo *ID,
269 ObjCInterfaceDecl *&clsDeclared) {
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000270 // FIXME: Should make sure no callers ever do this.
271 if (!hasDefinition())
272 return 0;
273
274 if (data().ExternallyCompleted)
Argyrios Kyrtzidis7c81c2a2011-10-19 02:25:16 +0000275 LoadExternalDefinition();
276
Chris Lattner1e03a562008-03-16 00:19:01 +0000277 ObjCInterfaceDecl* ClassDecl = this;
278 while (ClassDecl != NULL) {
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000279 if (ObjCIvarDecl *I = ClassDecl->getIvarDecl(ID)) {
Fariborz Jahanian496b5a82009-06-05 18:16:35 +0000280 clsDeclared = ClassDecl;
281 return I;
Chris Lattner1e03a562008-03-16 00:19:01 +0000282 }
Fariborz Jahanian80aa1cd2010-06-22 23:20:40 +0000283 for (const ObjCCategoryDecl *CDecl = ClassDecl->getFirstClassExtension();
284 CDecl; CDecl = CDecl->getNextClassExtension()) {
Fariborz Jahanian0e5ad252010-02-23 01:26:30 +0000285 if (ObjCIvarDecl *I = CDecl->getIvarDecl(ID)) {
286 clsDeclared = ClassDecl;
287 return I;
288 }
Fariborz Jahanian80aa1cd2010-06-22 23:20:40 +0000289 }
Fariborz Jahanian0e5ad252010-02-23 01:26:30 +0000290
Chris Lattner1e03a562008-03-16 00:19:01 +0000291 ClassDecl = ClassDecl->getSuperClass();
292 }
293 return NULL;
294}
295
Fariborz Jahaniancd187622009-05-22 17:12:32 +0000296/// lookupInheritedClass - This method returns ObjCInterfaceDecl * of the super
297/// class whose name is passed as argument. If it is not one of the super classes
298/// the it returns NULL.
299ObjCInterfaceDecl *ObjCInterfaceDecl::lookupInheritedClass(
300 const IdentifierInfo*ICName) {
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000301 // FIXME: Should make sure no callers ever do this.
302 if (!hasDefinition())
303 return 0;
304
305 if (data().ExternallyCompleted)
Argyrios Kyrtzidis7c81c2a2011-10-19 02:25:16 +0000306 LoadExternalDefinition();
307
Fariborz Jahaniancd187622009-05-22 17:12:32 +0000308 ObjCInterfaceDecl* ClassDecl = this;
309 while (ClassDecl != NULL) {
310 if (ClassDecl->getIdentifier() == ICName)
311 return ClassDecl;
312 ClassDecl = ClassDecl->getSuperClass();
313 }
314 return NULL;
315}
316
Argyrios Kyrtzidisaa5420c2009-07-25 22:15:51 +0000317/// lookupMethod - This method returns an instance/class method by looking in
Chris Lattner1e03a562008-03-16 00:19:01 +0000318/// the class, its categories, and its super classes (using a linear search).
Fariborz Jahanianbf393be2012-04-05 22:14:12 +0000319ObjCMethodDecl *ObjCInterfaceDecl::lookupMethod(Selector Sel,
320 bool isInstance,
321 bool shallowCategoryLookup) const {
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000322 // FIXME: Should make sure no callers ever do this.
323 if (!hasDefinition())
324 return 0;
325
Argyrios Kyrtzidisaa5420c2009-07-25 22:15:51 +0000326 const ObjCInterfaceDecl* ClassDecl = this;
Chris Lattner1e03a562008-03-16 00:19:01 +0000327 ObjCMethodDecl *MethodDecl = 0;
Mike Stump1eb44332009-09-09 15:08:12 +0000328
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000329 if (data().ExternallyCompleted)
Douglas Gregor26ac3f32010-12-01 23:49:52 +0000330 LoadExternalDefinition();
331
Chris Lattner1e03a562008-03-16 00:19:01 +0000332 while (ClassDecl != NULL) {
Argyrios Kyrtzidisaa5420c2009-07-25 22:15:51 +0000333 if ((MethodDecl = ClassDecl->getMethod(Sel, isInstance)))
Chris Lattner1e03a562008-03-16 00:19:01 +0000334 return MethodDecl;
Mike Stump1eb44332009-09-09 15:08:12 +0000335
Chris Lattner1e03a562008-03-16 00:19:01 +0000336 // Didn't find one yet - look through protocols.
Argyrios Kyrtzidisa5f44412012-03-13 01:09:41 +0000337 for (ObjCInterfaceDecl::protocol_iterator I = ClassDecl->protocol_begin(),
338 E = ClassDecl->protocol_end();
339 I != E; ++I)
Argyrios Kyrtzidisaa5420c2009-07-25 22:15:51 +0000340 if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
Chris Lattner1e03a562008-03-16 00:19:01 +0000341 return MethodDecl;
Fariborz Jahanianbf393be2012-04-05 22:14:12 +0000342
343 // Didn't find one yet - now look through categories.
344 ObjCCategoryDecl *CatDecl = ClassDecl->getCategoryList();
345 while (CatDecl) {
346 if ((MethodDecl = CatDecl->getMethod(Sel, isInstance)))
347 return MethodDecl;
Fariborz Jahanianbb3d14e2012-02-09 21:30:24 +0000348
Fariborz Jahanianbf393be2012-04-05 22:14:12 +0000349 if (!shallowCategoryLookup) {
Fariborz Jahanianbb3d14e2012-02-09 21:30:24 +0000350 // Didn't find one yet - look through protocols.
351 const ObjCList<ObjCProtocolDecl> &Protocols =
352 CatDecl->getReferencedProtocols();
353 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
354 E = Protocols.end(); I != E; ++I)
355 if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
356 return MethodDecl;
Fariborz Jahanianbb3d14e2012-02-09 21:30:24 +0000357 }
Fariborz Jahanianbf393be2012-04-05 22:14:12 +0000358 CatDecl = CatDecl->getNextClassCategory();
Chris Lattner1e03a562008-03-16 00:19:01 +0000359 }
Fariborz Jahanianbf393be2012-04-05 22:14:12 +0000360
Chris Lattner1e03a562008-03-16 00:19:01 +0000361 ClassDecl = ClassDecl->getSuperClass();
362 }
363 return NULL;
364}
365
Anna Zakse61354b2012-07-27 19:07:44 +0000366// Will search "local" class/category implementations for a method decl.
367// If failed, then we search in class's root for an instance method.
368// Returns 0 if no method is found.
Fariborz Jahanian74b27562010-12-03 23:37:08 +0000369ObjCMethodDecl *ObjCInterfaceDecl::lookupPrivateMethod(
370 const Selector &Sel,
Anna Zaksca93ee72012-07-30 20:31:21 +0000371 bool Instance) const {
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000372 // FIXME: Should make sure no callers ever do this.
373 if (!hasDefinition())
374 return 0;
375
376 if (data().ExternallyCompleted)
Argyrios Kyrtzidis7c81c2a2011-10-19 02:25:16 +0000377 LoadExternalDefinition();
378
Steve Naroffd789d3d2009-10-01 23:46:04 +0000379 ObjCMethodDecl *Method = 0;
380 if (ObjCImplementationDecl *ImpDecl = getImplementation())
Fariborz Jahanian74b27562010-12-03 23:37:08 +0000381 Method = Instance ? ImpDecl->getInstanceMethod(Sel)
382 : ImpDecl->getClassMethod(Sel);
Anna Zakse61354b2012-07-27 19:07:44 +0000383
384 // Look through local category implementations associated with the class.
385 if (!Method)
386 Method = Instance ? getCategoryInstanceMethod(Sel)
387 : getCategoryClassMethod(Sel);
388
389 // Before we give up, check if the selector is an instance method.
390 // But only in the root. This matches gcc's behavior and what the
391 // runtime expects.
392 if (!Instance && !Method && !getSuperClass()) {
393 Method = lookupInstanceMethod(Sel);
394 // Look through local category implementations associated
395 // with the root class.
396 if (!Method)
397 Method = lookupPrivateMethod(Sel, true);
398 }
399
Steve Naroffd789d3d2009-10-01 23:46:04 +0000400 if (!Method && getSuperClass())
Fariborz Jahanian74b27562010-12-03 23:37:08 +0000401 return getSuperClass()->lookupPrivateMethod(Sel, Instance);
Steve Naroffd789d3d2009-10-01 23:46:04 +0000402 return Method;
403}
Chris Lattnerab351632009-02-20 20:59:54 +0000404
405//===----------------------------------------------------------------------===//
406// ObjCMethodDecl
407//===----------------------------------------------------------------------===//
408
409ObjCMethodDecl *ObjCMethodDecl::Create(ASTContext &C,
Mike Stump1eb44332009-09-09 15:08:12 +0000410 SourceLocation beginLoc,
Chris Lattnerab351632009-02-20 20:59:54 +0000411 SourceLocation endLoc,
412 Selector SelInfo, QualType T,
Douglas Gregor4bc1cb62010-03-08 14:59:44 +0000413 TypeSourceInfo *ResultTInfo,
Chris Lattnerab351632009-02-20 20:59:54 +0000414 DeclContext *contextDecl,
415 bool isInstance,
416 bool isVariadic,
417 bool isSynthesized,
Argyrios Kyrtzidis75cf3e82011-08-17 19:25:08 +0000418 bool isImplicitlyDeclared,
Fariborz Jahanian3fe10412010-07-22 18:24:20 +0000419 bool isDefined,
Fariborz Jahanian7732cc92010-04-08 21:29:11 +0000420 ImplementationControl impControl,
Argyrios Kyrtzidisda92a7f2011-10-03 06:36:29 +0000421 bool HasRelatedResultType) {
Chris Lattnerab351632009-02-20 20:59:54 +0000422 return new (C) ObjCMethodDecl(beginLoc, endLoc,
Douglas Gregor4bc1cb62010-03-08 14:59:44 +0000423 SelInfo, T, ResultTInfo, contextDecl,
424 isInstance,
Argyrios Kyrtzidis75cf3e82011-08-17 19:25:08 +0000425 isVariadic, isSynthesized, isImplicitlyDeclared,
426 isDefined,
Fariborz Jahanian3fe10412010-07-22 18:24:20 +0000427 impControl,
Argyrios Kyrtzidisda92a7f2011-10-03 06:36:29 +0000428 HasRelatedResultType);
Chris Lattner1e03a562008-03-16 00:19:01 +0000429}
430
Douglas Gregor1e68ecc2012-01-05 21:55:30 +0000431ObjCMethodDecl *ObjCMethodDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
432 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCMethodDecl));
433 return new (Mem) ObjCMethodDecl(SourceLocation(), SourceLocation(),
434 Selector(), QualType(), 0, 0);
435}
436
Argyrios Kyrtzidis3a919e72011-10-14 08:02:31 +0000437void ObjCMethodDecl::setAsRedeclaration(const ObjCMethodDecl *PrevMethod) {
438 assert(PrevMethod);
439 getASTContext().setObjCMethodRedeclaration(PrevMethod, this);
440 IsRedeclaration = true;
Argyrios Kyrtzidis72b26252011-10-14 17:41:52 +0000441 PrevMethod->HasRedeclaration = true;
Argyrios Kyrtzidis3a919e72011-10-14 08:02:31 +0000442}
443
Argyrios Kyrtzidis491306a2011-10-03 06:37:04 +0000444void ObjCMethodDecl::setParamsAndSelLocs(ASTContext &C,
445 ArrayRef<ParmVarDecl*> Params,
446 ArrayRef<SourceLocation> SelLocs) {
447 ParamsAndSelLocs = 0;
448 NumParams = Params.size();
449 if (Params.empty() && SelLocs.empty())
450 return;
451
452 unsigned Size = sizeof(ParmVarDecl *) * NumParams +
453 sizeof(SourceLocation) * SelLocs.size();
454 ParamsAndSelLocs = C.Allocate(Size);
455 std::copy(Params.begin(), Params.end(), getParams());
456 std::copy(SelLocs.begin(), SelLocs.end(), getStoredSelLocs());
457}
458
459void ObjCMethodDecl::getSelectorLocs(
460 SmallVectorImpl<SourceLocation> &SelLocs) const {
461 for (unsigned i = 0, e = getNumSelectorLocs(); i != e; ++i)
462 SelLocs.push_back(getSelectorLoc(i));
463}
464
465void ObjCMethodDecl::setMethodParams(ASTContext &C,
466 ArrayRef<ParmVarDecl*> Params,
467 ArrayRef<SourceLocation> SelLocs) {
468 assert((!SelLocs.empty() || isImplicit()) &&
469 "No selector locs for non-implicit method");
470 if (isImplicit())
471 return setParamsAndSelLocs(C, Params, ArrayRef<SourceLocation>());
472
Argyrios Kyrtzidisa0cff722012-06-16 00:46:02 +0000473 SelLocsKind = hasStandardSelectorLocs(getSelector(), SelLocs, Params,
474 DeclEndLoc);
Argyrios Kyrtzidis491306a2011-10-03 06:37:04 +0000475 if (SelLocsKind != SelLoc_NonStandard)
476 return setParamsAndSelLocs(C, Params, ArrayRef<SourceLocation>());
477
478 setParamsAndSelLocs(C, Params, SelLocs);
479}
480
Argyrios Kyrtzidis57ea6be2009-07-21 00:06:36 +0000481/// \brief A definition will return its interface declaration.
482/// An interface declaration will return its definition.
483/// Otherwise it will return itself.
484ObjCMethodDecl *ObjCMethodDecl::getNextRedeclaration() {
485 ASTContext &Ctx = getASTContext();
Argyrios Kyrtzidis72b26252011-10-14 17:41:52 +0000486 ObjCMethodDecl *Redecl = 0;
487 if (HasRedeclaration)
488 Redecl = const_cast<ObjCMethodDecl*>(Ctx.getObjCMethodRedeclaration(this));
Argyrios Kyrtzidisb40034c2011-10-14 06:48:06 +0000489 if (Redecl)
490 return Redecl;
491
Argyrios Kyrtzidis57ea6be2009-07-21 00:06:36 +0000492 Decl *CtxD = cast<Decl>(getDeclContext());
493
494 if (ObjCInterfaceDecl *IFD = dyn_cast<ObjCInterfaceDecl>(CtxD)) {
495 if (ObjCImplementationDecl *ImplD = Ctx.getObjCImplementation(IFD))
496 Redecl = ImplD->getMethod(getSelector(), isInstanceMethod());
497
498 } else if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(CtxD)) {
499 if (ObjCCategoryImplDecl *ImplD = Ctx.getObjCImplementation(CD))
500 Redecl = ImplD->getMethod(getSelector(), isInstanceMethod());
501
Argyrios Kyrtzidis42920732009-07-28 05:11:05 +0000502 } else if (ObjCImplementationDecl *ImplD =
503 dyn_cast<ObjCImplementationDecl>(CtxD)) {
Argyrios Kyrtzidis57ea6be2009-07-21 00:06:36 +0000504 if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
505 Redecl = IFD->getMethod(getSelector(), isInstanceMethod());
Argyrios Kyrtzidis42920732009-07-28 05:11:05 +0000506
507 } else if (ObjCCategoryImplDecl *CImplD =
508 dyn_cast<ObjCCategoryImplDecl>(CtxD)) {
Steve Naroff0d69b8c2009-10-29 21:11:04 +0000509 if (ObjCCategoryDecl *CatD = CImplD->getCategoryDecl())
Argyrios Kyrtzidis42920732009-07-28 05:11:05 +0000510 Redecl = CatD->getMethod(getSelector(), isInstanceMethod());
Argyrios Kyrtzidis57ea6be2009-07-21 00:06:36 +0000511 }
512
Argyrios Kyrtzidis3a919e72011-10-14 08:02:31 +0000513 if (!Redecl && isRedeclaration()) {
514 // This is the last redeclaration, go back to the first method.
515 return cast<ObjCContainerDecl>(CtxD)->getMethod(getSelector(),
516 isInstanceMethod());
517 }
518
Argyrios Kyrtzidis57ea6be2009-07-21 00:06:36 +0000519 return Redecl ? Redecl : this;
520}
521
Argyrios Kyrtzidise7f9d302009-07-28 05:11:17 +0000522ObjCMethodDecl *ObjCMethodDecl::getCanonicalDecl() {
523 Decl *CtxD = cast<Decl>(getDeclContext());
524
525 if (ObjCImplementationDecl *ImplD = dyn_cast<ObjCImplementationDecl>(CtxD)) {
526 if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
527 if (ObjCMethodDecl *MD = IFD->getMethod(getSelector(),
528 isInstanceMethod()))
529 return MD;
530
531 } else if (ObjCCategoryImplDecl *CImplD =
532 dyn_cast<ObjCCategoryImplDecl>(CtxD)) {
Steve Naroff0d69b8c2009-10-29 21:11:04 +0000533 if (ObjCCategoryDecl *CatD = CImplD->getCategoryDecl())
Argyrios Kyrtzidise7f9d302009-07-28 05:11:17 +0000534 if (ObjCMethodDecl *MD = CatD->getMethod(getSelector(),
535 isInstanceMethod()))
536 return MD;
537 }
538
Argyrios Kyrtzidis6d4740e2011-10-17 19:48:09 +0000539 if (isRedeclaration())
540 return cast<ObjCContainerDecl>(CtxD)->getMethod(getSelector(),
541 isInstanceMethod());
542
Argyrios Kyrtzidise7f9d302009-07-28 05:11:17 +0000543 return this;
544}
545
Argyrios Kyrtzidisa0cff722012-06-16 00:46:02 +0000546SourceLocation ObjCMethodDecl::getLocEnd() const {
547 if (Stmt *Body = getBody())
548 return Body->getLocEnd();
549 return DeclEndLoc;
550}
551
John McCall85f3d762011-03-02 01:50:55 +0000552ObjCMethodFamily ObjCMethodDecl::getMethodFamily() const {
553 ObjCMethodFamily family = static_cast<ObjCMethodFamily>(Family);
John McCalld976c8e2011-03-02 21:01:41 +0000554 if (family != static_cast<unsigned>(InvalidObjCMethodFamily))
John McCall85f3d762011-03-02 01:50:55 +0000555 return family;
556
John McCalld5313b02011-03-02 11:33:24 +0000557 // Check for an explicit attribute.
558 if (const ObjCMethodFamilyAttr *attr = getAttr<ObjCMethodFamilyAttr>()) {
559 // The unfortunate necessity of mapping between enums here is due
560 // to the attributes framework.
561 switch (attr->getFamily()) {
562 case ObjCMethodFamilyAttr::OMF_None: family = OMF_None; break;
563 case ObjCMethodFamilyAttr::OMF_alloc: family = OMF_alloc; break;
564 case ObjCMethodFamilyAttr::OMF_copy: family = OMF_copy; break;
565 case ObjCMethodFamilyAttr::OMF_init: family = OMF_init; break;
566 case ObjCMethodFamilyAttr::OMF_mutableCopy: family = OMF_mutableCopy; break;
567 case ObjCMethodFamilyAttr::OMF_new: family = OMF_new; break;
568 }
569 Family = static_cast<unsigned>(family);
570 return family;
571 }
572
John McCall85f3d762011-03-02 01:50:55 +0000573 family = getSelector().getMethodFamily();
574 switch (family) {
575 case OMF_None: break;
576
577 // init only has a conventional meaning for an instance method, and
578 // it has to return an object.
579 case OMF_init:
580 if (!isInstanceMethod() || !getResultType()->isObjCObjectPointerType())
581 family = OMF_None;
582 break;
583
584 // alloc/copy/new have a conventional meaning for both class and
585 // instance methods, but they require an object return.
586 case OMF_alloc:
587 case OMF_copy:
588 case OMF_mutableCopy:
589 case OMF_new:
590 if (!getResultType()->isObjCObjectPointerType())
591 family = OMF_None;
592 break;
593
594 // These selectors have a conventional meaning only for instance methods.
595 case OMF_dealloc:
Nico Weber80cb6e62011-08-28 22:35:17 +0000596 case OMF_finalize:
John McCall85f3d762011-03-02 01:50:55 +0000597 case OMF_retain:
598 case OMF_release:
599 case OMF_autorelease:
600 case OMF_retainCount:
Douglas Gregor926df6c2011-06-11 01:09:30 +0000601 case OMF_self:
John McCall85f3d762011-03-02 01:50:55 +0000602 if (!isInstanceMethod())
603 family = OMF_None;
604 break;
Fariborz Jahanian9670e172011-07-05 22:38:59 +0000605
606 case OMF_performSelector:
607 if (!isInstanceMethod() ||
608 !getResultType()->isObjCIdType())
609 family = OMF_None;
610 else {
611 unsigned noParams = param_size();
612 if (noParams < 1 || noParams > 3)
613 family = OMF_None;
614 else {
615 ObjCMethodDecl::arg_type_iterator it = arg_type_begin();
616 QualType ArgT = (*it);
617 if (!ArgT->isObjCSelType()) {
618 family = OMF_None;
619 break;
620 }
621 while (--noParams) {
622 it++;
623 ArgT = (*it);
624 if (!ArgT->isObjCIdType()) {
625 family = OMF_None;
626 break;
627 }
628 }
629 }
630 }
631 break;
632
John McCall85f3d762011-03-02 01:50:55 +0000633 }
634
635 // Cache the result.
636 Family = static_cast<unsigned>(family);
637 return family;
638}
639
Mike Stump1eb44332009-09-09 15:08:12 +0000640void ObjCMethodDecl::createImplicitParams(ASTContext &Context,
Chris Lattnerab351632009-02-20 20:59:54 +0000641 const ObjCInterfaceDecl *OID) {
642 QualType selfTy;
643 if (isInstanceMethod()) {
644 // There may be no interface context due to error in declaration
645 // of the interface (which has been reported). Recover gracefully.
646 if (OID) {
Daniel Dunbar3b3a4582009-04-22 04:34:53 +0000647 selfTy = Context.getObjCInterfaceType(OID);
Steve Naroff14108da2009-07-10 23:34:53 +0000648 selfTy = Context.getObjCObjectPointerType(selfTy);
Chris Lattnerab351632009-02-20 20:59:54 +0000649 } else {
650 selfTy = Context.getObjCIdType();
651 }
652 } else // we have a factory method.
653 selfTy = Context.getObjCClassType();
654
John McCall7acddac2011-06-17 06:42:21 +0000655 bool selfIsPseudoStrong = false;
John McCallf85e1932011-06-15 23:02:42 +0000656 bool selfIsConsumed = false;
Ted Kremenek2bbcd5c2011-11-14 21:59:25 +0000657
David Blaikie4e4d0842012-03-11 07:00:24 +0000658 if (Context.getLangOpts().ObjCAutoRefCount) {
Ted Kremenek2bbcd5c2011-11-14 21:59:25 +0000659 if (isInstanceMethod()) {
660 selfIsConsumed = hasAttr<NSConsumesSelfAttr>();
John McCallf85e1932011-06-15 23:02:42 +0000661
Ted Kremenek2bbcd5c2011-11-14 21:59:25 +0000662 // 'self' is always __strong. It's actually pseudo-strong except
663 // in init methods (or methods labeled ns_consumes_self), though.
664 Qualifiers qs;
665 qs.setObjCLifetime(Qualifiers::OCL_Strong);
666 selfTy = Context.getQualifiedType(selfTy, qs);
John McCallf85e1932011-06-15 23:02:42 +0000667
Ted Kremenek2bbcd5c2011-11-14 21:59:25 +0000668 // In addition, 'self' is const unless this is an init method.
669 if (getMethodFamily() != OMF_init && !selfIsConsumed) {
670 selfTy = selfTy.withConst();
671 selfIsPseudoStrong = true;
672 }
673 }
674 else {
675 assert(isClassMethod());
676 // 'self' is always const in class methods.
John McCallf85e1932011-06-15 23:02:42 +0000677 selfTy = selfTy.withConst();
John McCall7acddac2011-06-17 06:42:21 +0000678 selfIsPseudoStrong = true;
679 }
John McCallf85e1932011-06-15 23:02:42 +0000680 }
681
682 ImplicitParamDecl *self
683 = ImplicitParamDecl::Create(Context, this, SourceLocation(),
684 &Context.Idents.get("self"), selfTy);
685 setSelfDecl(self);
686
687 if (selfIsConsumed)
688 self->addAttr(new (Context) NSConsumedAttr(SourceLocation(), Context));
Chris Lattnerab351632009-02-20 20:59:54 +0000689
John McCall7acddac2011-06-17 06:42:21 +0000690 if (selfIsPseudoStrong)
691 self->setARCPseudoStrong(true);
692
Mike Stump1eb44332009-09-09 15:08:12 +0000693 setCmdDecl(ImplicitParamDecl::Create(Context, this, SourceLocation(),
694 &Context.Idents.get("_cmd"),
Steve Naroff53c9d8a2009-04-20 15:06:07 +0000695 Context.getObjCSelType()));
Chris Lattnerab351632009-02-20 20:59:54 +0000696}
697
Chris Lattnerab351632009-02-20 20:59:54 +0000698ObjCInterfaceDecl *ObjCMethodDecl::getClassInterface() {
699 if (ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(getDeclContext()))
700 return ID;
701 if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(getDeclContext()))
702 return CD->getClassInterface();
Argyrios Kyrtzidisa8530372009-07-28 05:10:52 +0000703 if (ObjCImplDecl *IMD = dyn_cast<ObjCImplDecl>(getDeclContext()))
Chris Lattnerab351632009-02-20 20:59:54 +0000704 return IMD->getClassInterface();
Argyrios Kyrtzidisa8530372009-07-28 05:10:52 +0000705
706 assert(!isa<ObjCProtocolDecl>(getDeclContext()) && "It's a protocol method");
David Blaikieb219cfc2011-09-23 05:06:16 +0000707 llvm_unreachable("unknown method context");
Fariborz Jahanianae6f6fd2008-12-05 22:32:48 +0000708}
709
Chris Lattnerab351632009-02-20 20:59:54 +0000710//===----------------------------------------------------------------------===//
711// ObjCInterfaceDecl
712//===----------------------------------------------------------------------===//
713
Douglas Gregora6ea10e2012-01-17 18:09:05 +0000714ObjCInterfaceDecl *ObjCInterfaceDecl::Create(const ASTContext &C,
Chris Lattnerab351632009-02-20 20:59:54 +0000715 DeclContext *DC,
716 SourceLocation atLoc,
Mike Stump1eb44332009-09-09 15:08:12 +0000717 IdentifierInfo *Id,
Douglas Gregor0af55012011-12-16 03:12:41 +0000718 ObjCInterfaceDecl *PrevDecl,
Chris Lattnerab351632009-02-20 20:59:54 +0000719 SourceLocation ClassLoc,
Douglas Gregor7723fec2011-12-15 20:29:51 +0000720 bool isInternal){
Douglas Gregor0af55012011-12-16 03:12:41 +0000721 ObjCInterfaceDecl *Result = new (C) ObjCInterfaceDecl(DC, atLoc, Id, ClassLoc,
Douglas Gregorfd002a72011-12-16 22:37:11 +0000722 PrevDecl, isInternal);
Douglas Gregor0af55012011-12-16 03:12:41 +0000723 C.getObjCInterfaceType(Result, PrevDecl);
Douglas Gregor0af55012011-12-16 03:12:41 +0000724 return Result;
725}
726
Douglas Gregor1e68ecc2012-01-05 21:55:30 +0000727ObjCInterfaceDecl *ObjCInterfaceDecl::CreateDeserialized(ASTContext &C,
728 unsigned ID) {
729 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCInterfaceDecl));
730 return new (Mem) ObjCInterfaceDecl(0, SourceLocation(), 0, SourceLocation(),
731 0, false);
Chris Lattnerab351632009-02-20 20:59:54 +0000732}
733
734ObjCInterfaceDecl::
735ObjCInterfaceDecl(DeclContext *DC, SourceLocation atLoc, IdentifierInfo *Id,
Douglas Gregorfd002a72011-12-16 22:37:11 +0000736 SourceLocation CLoc, ObjCInterfaceDecl *PrevDecl,
737 bool isInternal)
Argyrios Kyrtzidis1711fc92011-10-04 04:48:02 +0000738 : ObjCContainerDecl(ObjCInterface, DC, Id, CLoc, atLoc),
Douglas Gregor7723fec2011-12-15 20:29:51 +0000739 TypeForDecl(0), Data()
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000740{
Douglas Gregorfd002a72011-12-16 22:37:11 +0000741 setPreviousDeclaration(PrevDecl);
742
743 // Copy the 'data' pointer over.
744 if (PrevDecl)
745 Data = PrevDecl->Data;
746
Argyrios Kyrtzidis40f57ee2011-11-15 06:20:21 +0000747 setImplicit(isInternal);
Chris Lattnerab351632009-02-20 20:59:54 +0000748}
749
Douglas Gregor26ac3f32010-12-01 23:49:52 +0000750void ObjCInterfaceDecl::LoadExternalDefinition() const {
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000751 assert(data().ExternallyCompleted && "Class is not externally completed");
752 data().ExternallyCompleted = false;
Douglas Gregor26ac3f32010-12-01 23:49:52 +0000753 getASTContext().getExternalSource()->CompleteType(
754 const_cast<ObjCInterfaceDecl *>(this));
755}
756
757void ObjCInterfaceDecl::setExternallyCompleted() {
758 assert(getASTContext().getExternalSource() &&
759 "Class can't be externally completed without an external source");
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000760 assert(hasDefinition() &&
Douglas Gregor26ac3f32010-12-01 23:49:52 +0000761 "Forward declarations can't be externally completed");
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000762 data().ExternallyCompleted = true;
Douglas Gregor26ac3f32010-12-01 23:49:52 +0000763}
764
Argyrios Kyrtzidis8a1d7222009-07-21 00:05:53 +0000765ObjCImplementationDecl *ObjCInterfaceDecl::getImplementation() const {
Douglas Gregor7723fec2011-12-15 20:29:51 +0000766 if (const ObjCInterfaceDecl *Def = getDefinition()) {
767 if (data().ExternallyCompleted)
768 LoadExternalDefinition();
769
770 return getASTContext().getObjCImplementation(
771 const_cast<ObjCInterfaceDecl*>(Def));
772 }
773
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000774 // FIXME: Should make sure no callers ever do this.
Douglas Gregor7723fec2011-12-15 20:29:51 +0000775 return 0;
Argyrios Kyrtzidis8a1d7222009-07-21 00:05:53 +0000776}
777
778void ObjCInterfaceDecl::setImplementation(ObjCImplementationDecl *ImplD) {
Douglas Gregor7723fec2011-12-15 20:29:51 +0000779 getASTContext().setObjCImplementation(getDefinition(), ImplD);
Argyrios Kyrtzidis8a1d7222009-07-21 00:05:53 +0000780}
781
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000782/// all_declared_ivar_begin - return first ivar declared in this class,
783/// its extensions and its implementation. Lazily build the list on first
784/// access.
785ObjCIvarDecl *ObjCInterfaceDecl::all_declared_ivar_begin() {
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000786 // FIXME: Should make sure no callers ever do this.
787 if (!hasDefinition())
788 return 0;
789
790 if (data().IvarList)
791 return data().IvarList;
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000792
793 ObjCIvarDecl *curIvar = 0;
794 if (!ivar_empty()) {
795 ObjCInterfaceDecl::ivar_iterator I = ivar_begin(), E = ivar_end();
David Blaikie581deb32012-06-06 20:45:41 +0000796 data().IvarList = *I; ++I;
797 for (curIvar = data().IvarList; I != E; curIvar = *I, ++I)
798 curIvar->setNextIvar(*I);
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000799 }
800
801 for (const ObjCCategoryDecl *CDecl = getFirstClassExtension(); CDecl;
802 CDecl = CDecl->getNextClassExtension()) {
803 if (!CDecl->ivar_empty()) {
804 ObjCCategoryDecl::ivar_iterator I = CDecl->ivar_begin(),
805 E = CDecl->ivar_end();
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000806 if (!data().IvarList) {
David Blaikie581deb32012-06-06 20:45:41 +0000807 data().IvarList = *I; ++I;
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000808 curIvar = data().IvarList;
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000809 }
David Blaikie581deb32012-06-06 20:45:41 +0000810 for ( ;I != E; curIvar = *I, ++I)
811 curIvar->setNextIvar(*I);
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000812 }
813 }
814
815 if (ObjCImplementationDecl *ImplDecl = getImplementation()) {
816 if (!ImplDecl->ivar_empty()) {
817 ObjCImplementationDecl::ivar_iterator I = ImplDecl->ivar_begin(),
818 E = ImplDecl->ivar_end();
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000819 if (!data().IvarList) {
David Blaikie581deb32012-06-06 20:45:41 +0000820 data().IvarList = *I; ++I;
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000821 curIvar = data().IvarList;
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000822 }
David Blaikie581deb32012-06-06 20:45:41 +0000823 for ( ;I != E; curIvar = *I, ++I)
824 curIvar->setNextIvar(*I);
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000825 }
826 }
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000827 return data().IvarList;
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000828}
Chris Lattnerab351632009-02-20 20:59:54 +0000829
830/// FindCategoryDeclaration - Finds category declaration in the list of
831/// categories for this class and returns it. Name of the category is passed
832/// in 'CategoryId'. If category not found, return 0;
Fariborz Jahanianae6f6fd2008-12-05 22:32:48 +0000833///
Chris Lattnerab351632009-02-20 20:59:54 +0000834ObjCCategoryDecl *
835ObjCInterfaceDecl::FindCategoryDeclaration(IdentifierInfo *CategoryId) const {
Argyrios Kyrtzidis5a61e0c2012-03-02 19:14:29 +0000836 // FIXME: Should make sure no callers ever do this.
837 if (!hasDefinition())
838 return 0;
839
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000840 if (data().ExternallyCompleted)
Douglas Gregor26ac3f32010-12-01 23:49:52 +0000841 LoadExternalDefinition();
842
Chris Lattnerab351632009-02-20 20:59:54 +0000843 for (ObjCCategoryDecl *Category = getCategoryList();
844 Category; Category = Category->getNextClassCategory())
845 if (Category->getIdentifier() == CategoryId)
846 return Category;
Fariborz Jahanianae6f6fd2008-12-05 22:32:48 +0000847 return 0;
848}
849
Argyrios Kyrtzidis1cb35dd2009-07-21 00:06:20 +0000850ObjCMethodDecl *
851ObjCInterfaceDecl::getCategoryInstanceMethod(Selector Sel) const {
852 for (ObjCCategoryDecl *Category = getCategoryList();
853 Category; Category = Category->getNextClassCategory())
854 if (ObjCCategoryImplDecl *Impl = Category->getImplementation())
855 if (ObjCMethodDecl *MD = Impl->getInstanceMethod(Sel))
856 return MD;
857 return 0;
858}
859
860ObjCMethodDecl *ObjCInterfaceDecl::getCategoryClassMethod(Selector Sel) const {
861 for (ObjCCategoryDecl *Category = getCategoryList();
862 Category; Category = Category->getNextClassCategory())
863 if (ObjCCategoryImplDecl *Impl = Category->getImplementation())
864 if (ObjCMethodDecl *MD = Impl->getClassMethod(Sel))
865 return MD;
866 return 0;
867}
868
Fariborz Jahanian0fd89042009-08-11 22:02:25 +0000869/// ClassImplementsProtocol - Checks that 'lProto' protocol
870/// has been implemented in IDecl class, its super class or categories (if
871/// lookupCategory is true).
872bool ObjCInterfaceDecl::ClassImplementsProtocol(ObjCProtocolDecl *lProto,
873 bool lookupCategory,
874 bool RHSIsQualifiedID) {
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000875 if (!hasDefinition())
876 return false;
877
Fariborz Jahanian0fd89042009-08-11 22:02:25 +0000878 ObjCInterfaceDecl *IDecl = this;
879 // 1st, look up the class.
Argyrios Kyrtzidisa5f44412012-03-13 01:09:41 +0000880 for (ObjCInterfaceDecl::protocol_iterator
881 PI = IDecl->protocol_begin(), E = IDecl->protocol_end(); PI != E; ++PI){
Fariborz Jahanian0fd89042009-08-11 22:02:25 +0000882 if (getASTContext().ProtocolCompatibleWithProtocol(lProto, *PI))
883 return true;
884 // This is dubious and is added to be compatible with gcc. In gcc, it is
885 // also allowed assigning a protocol-qualified 'id' type to a LHS object
886 // when protocol in qualified LHS is in list of protocols in the rhs 'id'
887 // object. This IMO, should be a bug.
888 // FIXME: Treat this as an extension, and flag this as an error when GCC
889 // extensions are not enabled.
Mike Stump1eb44332009-09-09 15:08:12 +0000890 if (RHSIsQualifiedID &&
Fariborz Jahanian0fd89042009-08-11 22:02:25 +0000891 getASTContext().ProtocolCompatibleWithProtocol(*PI, lProto))
892 return true;
893 }
Mike Stump1eb44332009-09-09 15:08:12 +0000894
Fariborz Jahanian0fd89042009-08-11 22:02:25 +0000895 // 2nd, look up the category.
896 if (lookupCategory)
897 for (ObjCCategoryDecl *CDecl = IDecl->getCategoryList(); CDecl;
898 CDecl = CDecl->getNextClassCategory()) {
899 for (ObjCCategoryDecl::protocol_iterator PI = CDecl->protocol_begin(),
900 E = CDecl->protocol_end(); PI != E; ++PI)
901 if (getASTContext().ProtocolCompatibleWithProtocol(lProto, *PI))
902 return true;
903 }
Mike Stump1eb44332009-09-09 15:08:12 +0000904
Fariborz Jahanian0fd89042009-08-11 22:02:25 +0000905 // 3rd, look up the super class(s)
906 if (IDecl->getSuperClass())
907 return
908 IDecl->getSuperClass()->ClassImplementsProtocol(lProto, lookupCategory,
909 RHSIsQualifiedID);
Mike Stump1eb44332009-09-09 15:08:12 +0000910
Fariborz Jahanian0fd89042009-08-11 22:02:25 +0000911 return false;
912}
913
Chris Lattnerab351632009-02-20 20:59:54 +0000914//===----------------------------------------------------------------------===//
915// ObjCIvarDecl
916//===----------------------------------------------------------------------===//
917
David Blaikie99ba9e32011-12-20 02:48:34 +0000918void ObjCIvarDecl::anchor() { }
919
Daniel Dunbara0654922010-04-02 20:10:03 +0000920ObjCIvarDecl *ObjCIvarDecl::Create(ASTContext &C, ObjCContainerDecl *DC,
Abramo Bagnaraff676cb2011-03-08 08:55:46 +0000921 SourceLocation StartLoc,
922 SourceLocation IdLoc, IdentifierInfo *Id,
John McCalla93c9342009-12-07 02:54:59 +0000923 QualType T, TypeSourceInfo *TInfo,
Fariborz Jahanianad51e742010-07-17 00:59:30 +0000924 AccessControl ac, Expr *BW,
925 bool synthesized) {
Daniel Dunbara0654922010-04-02 20:10:03 +0000926 if (DC) {
927 // Ivar's can only appear in interfaces, implementations (via synthesized
928 // properties), and class extensions (via direct declaration, or synthesized
929 // properties).
930 //
931 // FIXME: This should really be asserting this:
932 // (isa<ObjCCategoryDecl>(DC) &&
933 // cast<ObjCCategoryDecl>(DC)->IsClassExtension()))
934 // but unfortunately we sometimes place ivars into non-class extension
935 // categories on error. This breaks an AST invariant, and should not be
936 // fixed.
937 assert((isa<ObjCInterfaceDecl>(DC) || isa<ObjCImplementationDecl>(DC) ||
938 isa<ObjCCategoryDecl>(DC)) &&
939 "Invalid ivar decl context!");
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000940 // Once a new ivar is created in any of class/class-extension/implementation
941 // decl contexts, the previously built IvarList must be rebuilt.
942 ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(DC);
943 if (!ID) {
Eric Christopherffb0c3a2012-07-19 22:22:55 +0000944 if (ObjCImplementationDecl *IM = dyn_cast<ObjCImplementationDecl>(DC))
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000945 ID = IM->getClassInterface();
Eric Christopherffb0c3a2012-07-19 22:22:55 +0000946 else
947 ID = cast<ObjCCategoryDecl>(DC)->getClassInterface();
Fariborz Jahanian2c18bb72010-08-20 21:21:08 +0000948 }
949 ID->setIvarList(0);
Daniel Dunbara0654922010-04-02 20:10:03 +0000950 }
951
Abramo Bagnaraff676cb2011-03-08 08:55:46 +0000952 return new (C) ObjCIvarDecl(DC, StartLoc, IdLoc, Id, T, TInfo,
953 ac, BW, synthesized);
Chris Lattnerab351632009-02-20 20:59:54 +0000954}
955
Douglas Gregor1e68ecc2012-01-05 21:55:30 +0000956ObjCIvarDecl *ObjCIvarDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
957 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCIvarDecl));
958 return new (Mem) ObjCIvarDecl(0, SourceLocation(), SourceLocation(), 0,
959 QualType(), 0, ObjCIvarDecl::None, 0, false);
960}
961
Daniel Dunbar27a961a2010-04-02 21:13:59 +0000962const ObjCInterfaceDecl *ObjCIvarDecl::getContainingInterface() const {
963 const ObjCContainerDecl *DC = cast<ObjCContainerDecl>(getDeclContext());
Chris Lattnerab351632009-02-20 20:59:54 +0000964
Daniel Dunbar27a961a2010-04-02 21:13:59 +0000965 switch (DC->getKind()) {
966 default:
967 case ObjCCategoryImpl:
968 case ObjCProtocol:
David Blaikieb219cfc2011-09-23 05:06:16 +0000969 llvm_unreachable("invalid ivar container!");
Daniel Dunbar27a961a2010-04-02 21:13:59 +0000970
971 // Ivars can only appear in class extension categories.
972 case ObjCCategory: {
973 const ObjCCategoryDecl *CD = cast<ObjCCategoryDecl>(DC);
974 assert(CD->IsClassExtension() && "invalid container for ivar!");
975 return CD->getClassInterface();
976 }
977
978 case ObjCImplementation:
979 return cast<ObjCImplementationDecl>(DC)->getClassInterface();
980
981 case ObjCInterface:
982 return cast<ObjCInterfaceDecl>(DC);
983 }
984}
Chris Lattnerab351632009-02-20 20:59:54 +0000985
986//===----------------------------------------------------------------------===//
987// ObjCAtDefsFieldDecl
988//===----------------------------------------------------------------------===//
989
David Blaikie99ba9e32011-12-20 02:48:34 +0000990void ObjCAtDefsFieldDecl::anchor() { }
991
Chris Lattnerab351632009-02-20 20:59:54 +0000992ObjCAtDefsFieldDecl
Abramo Bagnaraff676cb2011-03-08 08:55:46 +0000993*ObjCAtDefsFieldDecl::Create(ASTContext &C, DeclContext *DC,
994 SourceLocation StartLoc, SourceLocation IdLoc,
Chris Lattnerab351632009-02-20 20:59:54 +0000995 IdentifierInfo *Id, QualType T, Expr *BW) {
Abramo Bagnaraff676cb2011-03-08 08:55:46 +0000996 return new (C) ObjCAtDefsFieldDecl(DC, StartLoc, IdLoc, Id, T, BW);
Chris Lattnerab351632009-02-20 20:59:54 +0000997}
998
Douglas Gregor1e68ecc2012-01-05 21:55:30 +0000999ObjCAtDefsFieldDecl *ObjCAtDefsFieldDecl::CreateDeserialized(ASTContext &C,
1000 unsigned ID) {
1001 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCAtDefsFieldDecl));
1002 return new (Mem) ObjCAtDefsFieldDecl(0, SourceLocation(), SourceLocation(),
1003 0, QualType(), 0);
1004}
1005
Chris Lattnerab351632009-02-20 20:59:54 +00001006//===----------------------------------------------------------------------===//
1007// ObjCProtocolDecl
1008//===----------------------------------------------------------------------===//
1009
David Blaikie99ba9e32011-12-20 02:48:34 +00001010void ObjCProtocolDecl::anchor() { }
1011
Douglas Gregor27c6da22012-01-01 20:30:41 +00001012ObjCProtocolDecl::ObjCProtocolDecl(DeclContext *DC, IdentifierInfo *Id,
1013 SourceLocation nameLoc,
1014 SourceLocation atStartLoc,
Douglas Gregorc9d3c7e2012-01-01 22:06:18 +00001015 ObjCProtocolDecl *PrevDecl)
1016 : ObjCContainerDecl(ObjCProtocol, DC, Id, nameLoc, atStartLoc), Data()
Douglas Gregor27c6da22012-01-01 20:30:41 +00001017{
1018 setPreviousDeclaration(PrevDecl);
1019 if (PrevDecl)
1020 Data = PrevDecl->Data;
1021}
1022
Chris Lattnerab351632009-02-20 20:59:54 +00001023ObjCProtocolDecl *ObjCProtocolDecl::Create(ASTContext &C, DeclContext *DC,
Argyrios Kyrtzidis1711fc92011-10-04 04:48:02 +00001024 IdentifierInfo *Id,
1025 SourceLocation nameLoc,
Argyrios Kyrtzidisb05d7b22011-10-17 19:48:06 +00001026 SourceLocation atStartLoc,
Douglas Gregorc9d3c7e2012-01-01 22:06:18 +00001027 ObjCProtocolDecl *PrevDecl) {
Douglas Gregor27c6da22012-01-01 20:30:41 +00001028 ObjCProtocolDecl *Result
Douglas Gregorc9d3c7e2012-01-01 22:06:18 +00001029 = new (C) ObjCProtocolDecl(DC, Id, nameLoc, atStartLoc, PrevDecl);
Douglas Gregor27c6da22012-01-01 20:30:41 +00001030
1031 return Result;
Chris Lattnerab351632009-02-20 20:59:54 +00001032}
1033
Douglas Gregor1e68ecc2012-01-05 21:55:30 +00001034ObjCProtocolDecl *ObjCProtocolDecl::CreateDeserialized(ASTContext &C,
1035 unsigned ID) {
1036 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCProtocolDecl));
1037 return new (Mem) ObjCProtocolDecl(0, 0, SourceLocation(), SourceLocation(),
1038 0);
1039}
1040
Steve Naroff91b0b0c2009-03-01 16:12:44 +00001041ObjCProtocolDecl *ObjCProtocolDecl::lookupProtocolNamed(IdentifierInfo *Name) {
1042 ObjCProtocolDecl *PDecl = this;
1043
1044 if (Name == getIdentifier())
1045 return PDecl;
1046
1047 for (protocol_iterator I = protocol_begin(), E = protocol_end(); I != E; ++I)
1048 if ((PDecl = (*I)->lookupProtocolNamed(Name)))
1049 return PDecl;
Mike Stump1eb44332009-09-09 15:08:12 +00001050
Steve Naroff91b0b0c2009-03-01 16:12:44 +00001051 return NULL;
1052}
1053
Argyrios Kyrtzidis094e2bb2009-07-25 22:15:38 +00001054// lookupMethod - Lookup a instance/class method in the protocol and protocols
Chris Lattnerab351632009-02-20 20:59:54 +00001055// it inherited.
Argyrios Kyrtzidis094e2bb2009-07-25 22:15:38 +00001056ObjCMethodDecl *ObjCProtocolDecl::lookupMethod(Selector Sel,
1057 bool isInstance) const {
Chris Lattnerab351632009-02-20 20:59:54 +00001058 ObjCMethodDecl *MethodDecl = NULL;
Mike Stump1eb44332009-09-09 15:08:12 +00001059
Argyrios Kyrtzidis094e2bb2009-07-25 22:15:38 +00001060 if ((MethodDecl = getMethod(Sel, isInstance)))
Chris Lattnerab351632009-02-20 20:59:54 +00001061 return MethodDecl;
Mike Stump1eb44332009-09-09 15:08:12 +00001062
Chris Lattnerab351632009-02-20 20:59:54 +00001063 for (protocol_iterator I = protocol_begin(), E = protocol_end(); I != E; ++I)
Argyrios Kyrtzidis094e2bb2009-07-25 22:15:38 +00001064 if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
Chris Lattnerab351632009-02-20 20:59:54 +00001065 return MethodDecl;
1066 return NULL;
1067}
1068
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +00001069void ObjCProtocolDecl::allocateDefinitionData() {
1070 assert(!Data && "Protocol already has a definition!");
Douglas Gregor1d784b22012-01-01 19:51:50 +00001071 Data = new (getASTContext()) DefinitionData;
1072 Data->Definition = this;
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +00001073}
1074
1075void ObjCProtocolDecl::startDefinition() {
1076 allocateDefinitionData();
Douglas Gregor1d784b22012-01-01 19:51:50 +00001077
1078 // Update all of the declarations with a pointer to the definition.
1079 for (redecl_iterator RD = redecls_begin(), RDEnd = redecls_end();
1080 RD != RDEnd; ++RD)
1081 RD->Data = this->Data;
Argyrios Kyrtzidisad834d52011-11-12 21:07:46 +00001082}
1083
Chris Lattnerab351632009-02-20 20:59:54 +00001084//===----------------------------------------------------------------------===//
Chris Lattnerab351632009-02-20 20:59:54 +00001085// ObjCCategoryDecl
1086//===----------------------------------------------------------------------===//
1087
David Blaikie99ba9e32011-12-20 02:48:34 +00001088void ObjCCategoryDecl::anchor() { }
1089
Chris Lattnerab351632009-02-20 20:59:54 +00001090ObjCCategoryDecl *ObjCCategoryDecl::Create(ASTContext &C, DeclContext *DC,
Douglas Gregor3db211b2010-01-16 16:38:58 +00001091 SourceLocation AtLoc,
1092 SourceLocation ClassNameLoc,
1093 SourceLocation CategoryNameLoc,
Argyrios Kyrtzidis955fadb2011-08-30 19:43:26 +00001094 IdentifierInfo *Id,
Fariborz Jahanianaf300292012-02-20 20:09:20 +00001095 ObjCInterfaceDecl *IDecl,
1096 SourceLocation IvarLBraceLoc,
1097 SourceLocation IvarRBraceLoc) {
Argyrios Kyrtzidis955fadb2011-08-30 19:43:26 +00001098 ObjCCategoryDecl *CatDecl = new (C) ObjCCategoryDecl(DC, AtLoc, ClassNameLoc,
1099 CategoryNameLoc, Id,
Fariborz Jahanianaf300292012-02-20 20:09:20 +00001100 IDecl,
1101 IvarLBraceLoc, IvarRBraceLoc);
Argyrios Kyrtzidis955fadb2011-08-30 19:43:26 +00001102 if (IDecl) {
1103 // Link this category into its class's category list.
1104 CatDecl->NextClassCategory = IDecl->getCategoryList();
Douglas Gregor2e5c15b2011-12-15 05:27:12 +00001105 if (IDecl->hasDefinition()) {
1106 IDecl->setCategoryList(CatDecl);
1107 if (ASTMutationListener *L = C.getASTMutationListener())
1108 L->AddedObjCCategoryToInterface(CatDecl, IDecl);
1109 }
Argyrios Kyrtzidis955fadb2011-08-30 19:43:26 +00001110 }
1111
1112 return CatDecl;
1113}
1114
Douglas Gregor1e68ecc2012-01-05 21:55:30 +00001115ObjCCategoryDecl *ObjCCategoryDecl::CreateDeserialized(ASTContext &C,
1116 unsigned ID) {
1117 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCCategoryDecl));
1118 return new (Mem) ObjCCategoryDecl(0, SourceLocation(), SourceLocation(),
1119 SourceLocation(), 0, 0);
Chris Lattnerab351632009-02-20 20:59:54 +00001120}
1121
Argyrios Kyrtzidis8a1d7222009-07-21 00:05:53 +00001122ObjCCategoryImplDecl *ObjCCategoryDecl::getImplementation() const {
1123 return getASTContext().getObjCImplementation(
1124 const_cast<ObjCCategoryDecl*>(this));
1125}
1126
1127void ObjCCategoryDecl::setImplementation(ObjCCategoryImplDecl *ImplD) {
1128 getASTContext().setObjCImplementation(this, ImplD);
1129}
1130
1131
Chris Lattnerab351632009-02-20 20:59:54 +00001132//===----------------------------------------------------------------------===//
1133// ObjCCategoryImplDecl
1134//===----------------------------------------------------------------------===//
1135
David Blaikie99ba9e32011-12-20 02:48:34 +00001136void ObjCCategoryImplDecl::anchor() { }
1137
Chris Lattnerab351632009-02-20 20:59:54 +00001138ObjCCategoryImplDecl *
1139ObjCCategoryImplDecl::Create(ASTContext &C, DeclContext *DC,
Argyrios Kyrtzidis1711fc92011-10-04 04:48:02 +00001140 IdentifierInfo *Id,
1141 ObjCInterfaceDecl *ClassInterface,
1142 SourceLocation nameLoc,
Argyrios Kyrtzidisc6994002011-12-09 00:31:40 +00001143 SourceLocation atStartLoc,
1144 SourceLocation CategoryNameLoc) {
Fariborz Jahanian712ef872011-12-23 00:31:02 +00001145 if (ClassInterface && ClassInterface->hasDefinition())
1146 ClassInterface = ClassInterface->getDefinition();
Argyrios Kyrtzidis1711fc92011-10-04 04:48:02 +00001147 return new (C) ObjCCategoryImplDecl(DC, Id, ClassInterface,
Argyrios Kyrtzidisc6994002011-12-09 00:31:40 +00001148 nameLoc, atStartLoc, CategoryNameLoc);
Chris Lattnerab351632009-02-20 20:59:54 +00001149}
1150
Douglas Gregor1e68ecc2012-01-05 21:55:30 +00001151ObjCCategoryImplDecl *ObjCCategoryImplDecl::CreateDeserialized(ASTContext &C,
1152 unsigned ID) {
1153 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCCategoryImplDecl));
1154 return new (Mem) ObjCCategoryImplDecl(0, 0, 0, SourceLocation(),
1155 SourceLocation(), SourceLocation());
1156}
1157
Steve Naroff0d69b8c2009-10-29 21:11:04 +00001158ObjCCategoryDecl *ObjCCategoryImplDecl::getCategoryDecl() const {
Ted Kremenekebfa3392010-03-19 20:39:03 +00001159 // The class interface might be NULL if we are working with invalid code.
1160 if (const ObjCInterfaceDecl *ID = getClassInterface())
1161 return ID->FindCategoryDeclaration(getIdentifier());
1162 return 0;
Argyrios Kyrtzidis42920732009-07-28 05:11:05 +00001163}
1164
Chris Lattnerab351632009-02-20 20:59:54 +00001165
David Blaikie99ba9e32011-12-20 02:48:34 +00001166void ObjCImplDecl::anchor() { }
1167
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001168void ObjCImplDecl::addPropertyImplementation(ObjCPropertyImplDecl *property) {
Douglas Gregor2c2d43c2009-04-23 02:42:49 +00001169 // FIXME: The context should be correct before we get here.
Douglas Gregor653f1b12009-04-23 01:02:12 +00001170 property->setLexicalDeclContext(this);
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001171 addDecl(property);
Douglas Gregor653f1b12009-04-23 01:02:12 +00001172}
1173
Argyrios Kyrtzidis8a1d7222009-07-21 00:05:53 +00001174void ObjCImplDecl::setClassInterface(ObjCInterfaceDecl *IFace) {
1175 ASTContext &Ctx = getASTContext();
1176
1177 if (ObjCImplementationDecl *ImplD
Duncan Sands98f2cca2009-07-21 07:56:29 +00001178 = dyn_cast_or_null<ObjCImplementationDecl>(this)) {
Argyrios Kyrtzidis8a1d7222009-07-21 00:05:53 +00001179 if (IFace)
1180 Ctx.setObjCImplementation(IFace, ImplD);
1181
Duncan Sands98f2cca2009-07-21 07:56:29 +00001182 } else if (ObjCCategoryImplDecl *ImplD =
Argyrios Kyrtzidis8a1d7222009-07-21 00:05:53 +00001183 dyn_cast_or_null<ObjCCategoryImplDecl>(this)) {
1184 if (ObjCCategoryDecl *CD = IFace->FindCategoryDeclaration(getIdentifier()))
1185 Ctx.setObjCImplementation(CD, ImplD);
1186 }
1187
1188 ClassInterface = IFace;
1189}
1190
Fariborz Jahanianae6f6fd2008-12-05 22:32:48 +00001191/// FindPropertyImplIvarDecl - This method lookup the ivar in the list of
James Dennett1c3a46a2012-06-15 22:30:14 +00001192/// properties implemented in this category \@implementation block and returns
Chris Lattnerd6eed1c2009-02-16 19:24:31 +00001193/// the implemented property that uses it.
Fariborz Jahanianae6f6fd2008-12-05 22:32:48 +00001194///
Chris Lattner3aa18612009-02-28 18:42:10 +00001195ObjCPropertyImplDecl *ObjCImplDecl::
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001196FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const {
1197 for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i){
David Blaikie581deb32012-06-06 20:45:41 +00001198 ObjCPropertyImplDecl *PID = *i;
Fariborz Jahanianae6f6fd2008-12-05 22:32:48 +00001199 if (PID->getPropertyIvarDecl() &&
1200 PID->getPropertyIvarDecl()->getIdentifier() == ivarId)
1201 return PID;
1202 }
1203 return 0;
1204}
1205
1206/// FindPropertyImplDecl - This method looks up a previous ObjCPropertyImplDecl
James Dennett1c3a46a2012-06-15 22:30:14 +00001207/// added to the list of those properties \@synthesized/\@dynamic in this
1208/// category \@implementation block.
Fariborz Jahanianae6f6fd2008-12-05 22:32:48 +00001209///
Chris Lattner3aa18612009-02-28 18:42:10 +00001210ObjCPropertyImplDecl *ObjCImplDecl::
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001211FindPropertyImplDecl(IdentifierInfo *Id) const {
1212 for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i){
David Blaikie581deb32012-06-06 20:45:41 +00001213 ObjCPropertyImplDecl *PID = *i;
Fariborz Jahanianae6f6fd2008-12-05 22:32:48 +00001214 if (PID->getPropertyDecl()->getIdentifier() == Id)
1215 return PID;
1216 }
1217 return 0;
1218}
1219
Chris Lattner5f9e2722011-07-23 10:55:15 +00001220raw_ostream &clang::operator<<(raw_ostream &OS,
Benjamin Kramerf9780592012-02-07 11:57:45 +00001221 const ObjCCategoryImplDecl &CID) {
1222 OS << CID.getName();
Benjamin Kramer900fc632010-04-17 09:33:03 +00001223 return OS;
1224}
1225
Chris Lattnerab351632009-02-20 20:59:54 +00001226//===----------------------------------------------------------------------===//
1227// ObjCImplementationDecl
1228//===----------------------------------------------------------------------===//
1229
David Blaikie99ba9e32011-12-20 02:48:34 +00001230void ObjCImplementationDecl::anchor() { }
1231
Chris Lattnerab351632009-02-20 20:59:54 +00001232ObjCImplementationDecl *
Mike Stump1eb44332009-09-09 15:08:12 +00001233ObjCImplementationDecl::Create(ASTContext &C, DeclContext *DC,
Chris Lattnerab351632009-02-20 20:59:54 +00001234 ObjCInterfaceDecl *ClassInterface,
Argyrios Kyrtzidis1711fc92011-10-04 04:48:02 +00001235 ObjCInterfaceDecl *SuperDecl,
1236 SourceLocation nameLoc,
Fariborz Jahanianaf300292012-02-20 20:09:20 +00001237 SourceLocation atStartLoc,
1238 SourceLocation IvarLBraceLoc,
1239 SourceLocation IvarRBraceLoc) {
Fariborz Jahanian712ef872011-12-23 00:31:02 +00001240 if (ClassInterface && ClassInterface->hasDefinition())
1241 ClassInterface = ClassInterface->getDefinition();
Argyrios Kyrtzidis1711fc92011-10-04 04:48:02 +00001242 return new (C) ObjCImplementationDecl(DC, ClassInterface, SuperDecl,
Fariborz Jahanianaf300292012-02-20 20:09:20 +00001243 nameLoc, atStartLoc,
1244 IvarLBraceLoc, IvarRBraceLoc);
Chris Lattnerab351632009-02-20 20:59:54 +00001245}
1246
Douglas Gregor1e68ecc2012-01-05 21:55:30 +00001247ObjCImplementationDecl *
1248ObjCImplementationDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
1249 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCImplementationDecl));
1250 return new (Mem) ObjCImplementationDecl(0, 0, 0, SourceLocation(),
1251 SourceLocation());
1252}
1253
John McCallda6d9762011-07-22 04:15:06 +00001254void ObjCImplementationDecl::setIvarInitializers(ASTContext &C,
1255 CXXCtorInitializer ** initializers,
1256 unsigned numInitializers) {
1257 if (numInitializers > 0) {
1258 NumIvarInitializers = numInitializers;
1259 CXXCtorInitializer **ivarInitializers =
1260 new (C) CXXCtorInitializer*[NumIvarInitializers];
1261 memcpy(ivarInitializers, initializers,
1262 numInitializers * sizeof(CXXCtorInitializer*));
1263 IvarInitializers = ivarInitializers;
1264 }
1265}
1266
Chris Lattner5f9e2722011-07-23 10:55:15 +00001267raw_ostream &clang::operator<<(raw_ostream &OS,
Benjamin Kramerf9780592012-02-07 11:57:45 +00001268 const ObjCImplementationDecl &ID) {
1269 OS << ID.getName();
Benjamin Kramer900fc632010-04-17 09:33:03 +00001270 return OS;
1271}
1272
Chris Lattnerab351632009-02-20 20:59:54 +00001273//===----------------------------------------------------------------------===//
1274// ObjCCompatibleAliasDecl
1275//===----------------------------------------------------------------------===//
1276
David Blaikie99ba9e32011-12-20 02:48:34 +00001277void ObjCCompatibleAliasDecl::anchor() { }
1278
Chris Lattnerab351632009-02-20 20:59:54 +00001279ObjCCompatibleAliasDecl *
1280ObjCCompatibleAliasDecl::Create(ASTContext &C, DeclContext *DC,
1281 SourceLocation L,
Mike Stump1eb44332009-09-09 15:08:12 +00001282 IdentifierInfo *Id,
Chris Lattnerab351632009-02-20 20:59:54 +00001283 ObjCInterfaceDecl* AliasedClass) {
1284 return new (C) ObjCCompatibleAliasDecl(DC, L, Id, AliasedClass);
1285}
1286
Douglas Gregor1e68ecc2012-01-05 21:55:30 +00001287ObjCCompatibleAliasDecl *
1288ObjCCompatibleAliasDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
1289 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCCompatibleAliasDecl));
1290 return new (Mem) ObjCCompatibleAliasDecl(0, SourceLocation(), 0, 0);
1291}
1292
Chris Lattnerab351632009-02-20 20:59:54 +00001293//===----------------------------------------------------------------------===//
1294// ObjCPropertyDecl
1295//===----------------------------------------------------------------------===//
1296
David Blaikie99ba9e32011-12-20 02:48:34 +00001297void ObjCPropertyDecl::anchor() { }
1298
Chris Lattnerab351632009-02-20 20:59:54 +00001299ObjCPropertyDecl *ObjCPropertyDecl::Create(ASTContext &C, DeclContext *DC,
1300 SourceLocation L,
1301 IdentifierInfo *Id,
Fariborz Jahaniand0502402010-01-21 17:36:00 +00001302 SourceLocation AtLoc,
Fariborz Jahanian77bfb8b2012-02-29 22:18:55 +00001303 SourceLocation LParenLoc,
John McCall83a230c2010-06-04 20:50:08 +00001304 TypeSourceInfo *T,
Chris Lattnerab351632009-02-20 20:59:54 +00001305 PropertyControl propControl) {
Fariborz Jahanian77bfb8b2012-02-29 22:18:55 +00001306 return new (C) ObjCPropertyDecl(DC, L, Id, AtLoc, LParenLoc, T);
Chris Lattnerab351632009-02-20 20:59:54 +00001307}
1308
Douglas Gregor1e68ecc2012-01-05 21:55:30 +00001309ObjCPropertyDecl *ObjCPropertyDecl::CreateDeserialized(ASTContext &C,
1310 unsigned ID) {
1311 void * Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCPropertyDecl));
1312 return new (Mem) ObjCPropertyDecl(0, SourceLocation(), 0, SourceLocation(),
Fariborz Jahanian77bfb8b2012-02-29 22:18:55 +00001313 SourceLocation(),
Douglas Gregor1e68ecc2012-01-05 21:55:30 +00001314 0);
1315}
1316
Chris Lattnerab351632009-02-20 20:59:54 +00001317//===----------------------------------------------------------------------===//
1318// ObjCPropertyImplDecl
1319//===----------------------------------------------------------------------===//
1320
Fariborz Jahanian628b96f2008-04-23 00:06:01 +00001321ObjCPropertyImplDecl *ObjCPropertyImplDecl::Create(ASTContext &C,
Douglas Gregord0434102009-01-09 00:49:46 +00001322 DeclContext *DC,
Fariborz Jahanian628b96f2008-04-23 00:06:01 +00001323 SourceLocation atLoc,
1324 SourceLocation L,
1325 ObjCPropertyDecl *property,
Daniel Dunbar9f0afd42008-08-26 04:47:31 +00001326 Kind PK,
Douglas Gregora4ffd852010-11-17 01:03:52 +00001327 ObjCIvarDecl *ivar,
1328 SourceLocation ivarLoc) {
1329 return new (C) ObjCPropertyImplDecl(DC, atLoc, L, property, PK, ivar,
1330 ivarLoc);
Fariborz Jahanian628b96f2008-04-23 00:06:01 +00001331}
Chris Lattnerf4af5152008-03-17 01:19:02 +00001332
Douglas Gregor1e68ecc2012-01-05 21:55:30 +00001333ObjCPropertyImplDecl *ObjCPropertyImplDecl::CreateDeserialized(ASTContext &C,
1334 unsigned ID) {
1335 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCPropertyImplDecl));
1336 return new (Mem) ObjCPropertyImplDecl(0, SourceLocation(), SourceLocation(),
1337 0, Dynamic, 0, SourceLocation());
1338}
1339
Douglas Gregora4ffd852010-11-17 01:03:52 +00001340SourceRange ObjCPropertyImplDecl::getSourceRange() const {
1341 SourceLocation EndLoc = getLocation();
1342 if (IvarLoc.isValid())
1343 EndLoc = IvarLoc;
Chris Lattner0ed844b2008-04-04 06:12:32 +00001344
Douglas Gregora4ffd852010-11-17 01:03:52 +00001345 return SourceRange(AtLoc, EndLoc);
1346}