blob: 45c57d6591009c0b1c781091d1671869d65ea287 [file] [log] [blame]
Chris Lattner89375192008-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 Dunbar221fa942008-08-11 04:54:23 +000016#include "clang/AST/Stmt.h"
Argyrios Kyrtzidis7d847c92011-09-01 00:58:55 +000017#include "clang/AST/ASTMutationListener.h"
Steve Naroffc4173fa2009-02-22 19:35:57 +000018#include "llvm/ADT/STLExtras.h"
Anna Zaks454477c2012-09-27 19:45:11 +000019#include "llvm/ADT/SmallString.h"
Chris Lattner89375192008-03-16 00:19:01 +000020using namespace clang;
21
Chris Lattner8d8829e2008-03-16 00:49:28 +000022//===----------------------------------------------------------------------===//
Chris Lattner4d1eb762009-02-20 21:16:26 +000023// ObjCListBase
24//===----------------------------------------------------------------------===//
25
Chris Lattner22298722009-02-20 21:35:13 +000026void ObjCListBase::set(void *const* InList, unsigned Elts, ASTContext &Ctx) {
Douglas Gregorb412e172010-07-25 18:17:45 +000027 List = 0;
Chris Lattner4d1eb762009-02-20 21:16:26 +000028 if (Elts == 0) return; // Setting to an empty list is a noop.
Mike Stump11289f42009-09-09 15:08:12 +000029
30
Chris Lattner7c981a72009-02-20 21:44:01 +000031 List = new (Ctx) void*[Elts];
Chris Lattner4d1eb762009-02-20 21:16:26 +000032 NumElts = Elts;
33 memcpy(List, InList, sizeof(void*)*Elts);
34}
35
Douglas Gregor002b6712010-01-16 15:02:53 +000036void ObjCProtocolList::set(ObjCProtocolDecl* const* InList, unsigned Elts,
37 const SourceLocation *Locs, ASTContext &Ctx) {
38 if (Elts == 0)
39 return;
40
41 Locations = new (Ctx) SourceLocation[Elts];
42 memcpy(Locations, Locs, sizeof(SourceLocation) * Elts);
43 set(InList, Elts, Ctx);
44}
45
Chris Lattner4d1eb762009-02-20 21:16:26 +000046//===----------------------------------------------------------------------===//
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +000047// ObjCInterfaceDecl
Chris Lattner8d8829e2008-03-16 00:49:28 +000048//===----------------------------------------------------------------------===//
49
David Blaikie68e081d2011-12-20 02:48:34 +000050void ObjCContainerDecl::anchor() { }
51
Fariborz Jahanian68453832009-06-05 18:16:35 +000052/// getIvarDecl - This method looks up an ivar in this ContextDecl.
53///
54ObjCIvarDecl *
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +000055ObjCContainerDecl::getIvarDecl(IdentifierInfo *Id) const {
Fariborz Jahanian68453832009-06-05 18:16:35 +000056 lookup_const_iterator Ivar, IvarEnd;
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +000057 for (llvm::tie(Ivar, IvarEnd) = lookup(Id); Ivar != IvarEnd; ++Ivar) {
Fariborz Jahanian68453832009-06-05 18:16:35 +000058 if (ObjCIvarDecl *ivar = dyn_cast<ObjCIvarDecl>(*Ivar))
59 return ivar;
60 }
61 return 0;
62}
63
Argyrios Kyrtzidis6de05602009-07-25 22:15:22 +000064// Get the local instance/class method declared in this interface.
Douglas Gregorbcced4e2009-04-09 21:40:53 +000065ObjCMethodDecl *
Argyrios Kyrtzidis6de05602009-07-25 22:15:22 +000066ObjCContainerDecl::getMethod(Selector Sel, bool isInstance) const {
Steve Naroffc4173fa2009-02-22 19:35:57 +000067 // Since instance & class methods can have the same name, the loop below
68 // ensures we get the correct method.
69 //
70 // @interface Whatever
71 // - (int) class_method;
72 // + (float) class_method;
73 // @end
74 //
75 lookup_const_iterator Meth, MethEnd;
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +000076 for (llvm::tie(Meth, MethEnd) = lookup(Sel); Meth != MethEnd; ++Meth) {
Steve Naroffc4173fa2009-02-22 19:35:57 +000077 ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(*Meth);
Argyrios Kyrtzidis6de05602009-07-25 22:15:22 +000078 if (MD && MD->isInstanceMethod() == isInstance)
Steve Naroffc4173fa2009-02-22 19:35:57 +000079 return MD;
80 }
Steve Naroff35c62ae2009-01-08 17:28:14 +000081 return 0;
82}
83
Ted Kremenek4fb821e2010-03-15 20:11:46 +000084ObjCPropertyDecl *
Ted Kremenekddcd1092010-03-15 20:11:53 +000085ObjCPropertyDecl::findPropertyDecl(const DeclContext *DC,
Ted Kremenek4fb821e2010-03-15 20:11:46 +000086 IdentifierInfo *propertyID) {
87
Ted Kremenekddcd1092010-03-15 20:11:53 +000088 DeclContext::lookup_const_iterator I, E;
Ted Kremenek4fb821e2010-03-15 20:11:46 +000089 llvm::tie(I, E) = DC->lookup(propertyID);
90 for ( ; I != E; ++I)
91 if (ObjCPropertyDecl *PD = dyn_cast<ObjCPropertyDecl>(*I))
92 return PD;
93
94 return 0;
95}
96
Anna Zaks454477c2012-09-27 19:45:11 +000097IdentifierInfo *
98ObjCPropertyDecl::getDefaultSynthIvarName(ASTContext &Ctx) const {
99 SmallString<128> ivarName;
100 {
101 llvm::raw_svector_ostream os(ivarName);
102 os << '_' << getIdentifier()->getName();
103 }
104 return &Ctx.Idents.get(ivarName.str());
105}
106
Fariborz Jahaniana054e992008-04-21 19:04:53 +0000107/// FindPropertyDeclaration - Finds declaration of the property given its name
108/// in 'PropertyId' and returns it. It returns 0, if not found.
Fariborz Jahaniana054e992008-04-21 19:04:53 +0000109ObjCPropertyDecl *
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +0000110ObjCContainerDecl::FindPropertyDeclaration(IdentifierInfo *PropertyId) const {
Mike Stump11289f42009-09-09 15:08:12 +0000111
Ted Kremenekddcd1092010-03-15 20:11:53 +0000112 if (ObjCPropertyDecl *PD =
113 ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(this), PropertyId))
114 return PD;
Mike Stump11289f42009-09-09 15:08:12 +0000115
Ted Kremenekddcd1092010-03-15 20:11:53 +0000116 switch (getKind()) {
117 default:
118 break;
119 case Decl::ObjCProtocol: {
120 const ObjCProtocolDecl *PID = cast<ObjCProtocolDecl>(this);
121 for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
122 E = PID->protocol_end(); I != E; ++I)
Fariborz Jahanian30a42922010-02-15 21:55:26 +0000123 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
124 return P;
Ted Kremenekddcd1092010-03-15 20:11:53 +0000125 break;
126 }
127 case Decl::ObjCInterface: {
128 const ObjCInterfaceDecl *OID = cast<ObjCInterfaceDecl>(this);
129 // Look through categories.
130 for (ObjCCategoryDecl *Cat = OID->getCategoryList();
131 Cat; Cat = Cat->getNextClassCategory())
132 if (!Cat->IsClassExtension())
133 if (ObjCPropertyDecl *P = Cat->FindPropertyDeclaration(PropertyId))
134 return P;
135
136 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +0000137 for (ObjCInterfaceDecl::all_protocol_iterator
138 I = OID->all_referenced_protocol_begin(),
139 E = OID->all_referenced_protocol_end(); I != E; ++I)
Ted Kremenekddcd1092010-03-15 20:11:53 +0000140 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
141 return P;
142
143 // Finally, check the super class.
144 if (const ObjCInterfaceDecl *superClass = OID->getSuperClass())
145 return superClass->FindPropertyDeclaration(PropertyId);
146 break;
147 }
148 case Decl::ObjCCategory: {
149 const ObjCCategoryDecl *OCD = cast<ObjCCategoryDecl>(this);
150 // Look through protocols.
151 if (!OCD->IsClassExtension())
152 for (ObjCCategoryDecl::protocol_iterator
153 I = OCD->protocol_begin(), E = OCD->protocol_end(); I != E; ++I)
154 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
155 return P;
156
157 break;
Fariborz Jahaniandab04842009-01-19 18:16:19 +0000158 }
159 }
Steve Narofff9c65242008-06-05 13:55:23 +0000160 return 0;
161}
162
David Blaikie68e081d2011-12-20 02:48:34 +0000163void ObjCInterfaceDecl::anchor() { }
164
Fariborz Jahaniande8db162009-11-02 22:45:15 +0000165/// FindPropertyVisibleInPrimaryClass - Finds declaration of the property
166/// with name 'PropertyId' in the primary class; including those in protocols
Ted Kremenekd133a862010-03-15 20:30:07 +0000167/// (direct or indirect) used by the primary class.
Fariborz Jahaniande8db162009-11-02 22:45:15 +0000168///
169ObjCPropertyDecl *
Ted Kremenekd133a862010-03-15 20:30:07 +0000170ObjCInterfaceDecl::FindPropertyVisibleInPrimaryClass(
Fariborz Jahaniande8db162009-11-02 22:45:15 +0000171 IdentifierInfo *PropertyId) const {
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000172 // FIXME: Should make sure no callers ever do this.
173 if (!hasDefinition())
174 return 0;
175
176 if (data().ExternallyCompleted)
Douglas Gregor73693022010-12-01 23:49:52 +0000177 LoadExternalDefinition();
178
Ted Kremenekd133a862010-03-15 20:30:07 +0000179 if (ObjCPropertyDecl *PD =
180 ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(this), PropertyId))
181 return PD;
182
Fariborz Jahaniande8db162009-11-02 22:45:15 +0000183 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +0000184 for (ObjCInterfaceDecl::all_protocol_iterator
185 I = all_referenced_protocol_begin(),
186 E = all_referenced_protocol_end(); I != E; ++I)
Fariborz Jahaniande8db162009-11-02 22:45:15 +0000187 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
188 return P;
Ted Kremenekd133a862010-03-15 20:30:07 +0000189
Fariborz Jahaniande8db162009-11-02 22:45:15 +0000190 return 0;
191}
192
Fariborz Jahanian092cd6e2009-10-05 20:41:32 +0000193void ObjCInterfaceDecl::mergeClassExtensionProtocolList(
194 ObjCProtocolDecl *const* ExtList, unsigned ExtNum,
195 ASTContext &C)
196{
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000197 if (data().ExternallyCompleted)
Douglas Gregor73693022010-12-01 23:49:52 +0000198 LoadExternalDefinition();
199
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000200 if (data().AllReferencedProtocols.empty() &&
201 data().ReferencedProtocols.empty()) {
202 data().AllReferencedProtocols.set(ExtList, ExtNum, C);
Fariborz Jahanian092cd6e2009-10-05 20:41:32 +0000203 return;
204 }
Ted Kremenek0ef508d2010-09-01 01:21:15 +0000205
Fariborz Jahanian092cd6e2009-10-05 20:41:32 +0000206 // Check for duplicate protocol in class's protocol list.
Ted Kremenek0ef508d2010-09-01 01:21:15 +0000207 // This is O(n*m). But it is extremely rare and number of protocols in
Fariborz Jahanian092cd6e2009-10-05 20:41:32 +0000208 // class or its extension are very few.
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000209 SmallVector<ObjCProtocolDecl*, 8> ProtocolRefs;
Fariborz Jahanian092cd6e2009-10-05 20:41:32 +0000210 for (unsigned i = 0; i < ExtNum; i++) {
211 bool protocolExists = false;
212 ObjCProtocolDecl *ProtoInExtension = ExtList[i];
Ted Kremenek0ef508d2010-09-01 01:21:15 +0000213 for (all_protocol_iterator
214 p = all_referenced_protocol_begin(),
215 e = all_referenced_protocol_end(); p != e; ++p) {
Fariborz Jahanian092cd6e2009-10-05 20:41:32 +0000216 ObjCProtocolDecl *Proto = (*p);
217 if (C.ProtocolCompatibleWithProtocol(ProtoInExtension, Proto)) {
218 protocolExists = true;
219 break;
220 }
221 }
222 // Do we want to warn on a protocol in extension class which
223 // already exist in the class? Probably not.
Ted Kremenek0ef508d2010-09-01 01:21:15 +0000224 if (!protocolExists)
Fariborz Jahanian092cd6e2009-10-05 20:41:32 +0000225 ProtocolRefs.push_back(ProtoInExtension);
226 }
Ted Kremenek0ef508d2010-09-01 01:21:15 +0000227
Fariborz Jahanian092cd6e2009-10-05 20:41:32 +0000228 if (ProtocolRefs.empty())
229 return;
Ted Kremenek0ef508d2010-09-01 01:21:15 +0000230
Fariborz Jahanian8764c742009-10-05 21:32:49 +0000231 // Merge ProtocolRefs into class's protocol list;
Ted Kremenek0ef508d2010-09-01 01:21:15 +0000232 for (all_protocol_iterator p = all_referenced_protocol_begin(),
233 e = all_referenced_protocol_end(); p != e; ++p) {
Fariborz Jahanian092cd6e2009-10-05 20:41:32 +0000234 ProtocolRefs.push_back(*p);
Douglas Gregor002b6712010-01-16 15:02:53 +0000235 }
Ted Kremenek0ef508d2010-09-01 01:21:15 +0000236
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000237 data().AllReferencedProtocols.set(ProtocolRefs.data(), ProtocolRefs.size(),C);
Fariborz Jahanian092cd6e2009-10-05 20:41:32 +0000238}
239
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000240void ObjCInterfaceDecl::allocateDefinitionData() {
241 assert(!hasDefinition() && "ObjC class already has a definition");
Douglas Gregora323c4c2011-12-15 18:17:27 +0000242 Data = new (getASTContext()) DefinitionData();
Douglas Gregor7671e532011-12-16 16:34:57 +0000243 Data->Definition = this;
244
245 // Make the type point at the definition, now that we have one.
246 if (TypeForDecl)
247 cast<ObjCInterfaceType>(TypeForDecl)->Decl = this;
Douglas Gregorab1ec82e2011-12-16 03:12:41 +0000248}
249
250void ObjCInterfaceDecl::startDefinition() {
251 allocateDefinitionData();
252
Douglas Gregor66b310c2011-12-15 18:03:09 +0000253 // Update all of the declarations with a pointer to the definition.
254 for (redecl_iterator RD = redecls_begin(), RDEnd = redecls_end();
255 RD != RDEnd; ++RD) {
256 if (*RD != this)
Douglas Gregora323c4c2011-12-15 18:17:27 +0000257 RD->Data = Data;
Douglas Gregor66b310c2011-12-15 18:03:09 +0000258 }
Argyrios Kyrtzidisb97a4022011-11-12 21:07:46 +0000259}
260
Fariborz Jahanian3bf0ded2010-06-22 23:20:40 +0000261/// getFirstClassExtension - Find first class extension of the given class.
262ObjCCategoryDecl* ObjCInterfaceDecl::getFirstClassExtension() const {
263 for (ObjCCategoryDecl *CDecl = getCategoryList(); CDecl;
Fariborz Jahanianafe13862010-02-23 01:26:30 +0000264 CDecl = CDecl->getNextClassCategory())
265 if (CDecl->IsClassExtension())
266 return CDecl;
267 return 0;
268}
269
Fariborz Jahanian3bf0ded2010-06-22 23:20:40 +0000270/// getNextClassCategory - Find next class extension in list of categories.
271const ObjCCategoryDecl* ObjCCategoryDecl::getNextClassExtension() const {
272 for (const ObjCCategoryDecl *CDecl = getNextClassCategory(); CDecl;
273 CDecl = CDecl->getNextClassCategory())
274 if (CDecl->IsClassExtension())
275 return CDecl;
276 return 0;
277}
278
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +0000279ObjCIvarDecl *ObjCInterfaceDecl::lookupInstanceVariable(IdentifierInfo *ID,
280 ObjCInterfaceDecl *&clsDeclared) {
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000281 // FIXME: Should make sure no callers ever do this.
282 if (!hasDefinition())
283 return 0;
284
285 if (data().ExternallyCompleted)
Argyrios Kyrtzidis4e8b1362011-10-19 02:25:16 +0000286 LoadExternalDefinition();
287
Chris Lattner89375192008-03-16 00:19:01 +0000288 ObjCInterfaceDecl* ClassDecl = this;
289 while (ClassDecl != NULL) {
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +0000290 if (ObjCIvarDecl *I = ClassDecl->getIvarDecl(ID)) {
Fariborz Jahanian68453832009-06-05 18:16:35 +0000291 clsDeclared = ClassDecl;
292 return I;
Chris Lattner89375192008-03-16 00:19:01 +0000293 }
Fariborz Jahanian3bf0ded2010-06-22 23:20:40 +0000294 for (const ObjCCategoryDecl *CDecl = ClassDecl->getFirstClassExtension();
295 CDecl; CDecl = CDecl->getNextClassExtension()) {
Fariborz Jahanianafe13862010-02-23 01:26:30 +0000296 if (ObjCIvarDecl *I = CDecl->getIvarDecl(ID)) {
297 clsDeclared = ClassDecl;
298 return I;
299 }
Fariborz Jahanian3bf0ded2010-06-22 23:20:40 +0000300 }
Fariborz Jahanianafe13862010-02-23 01:26:30 +0000301
Chris Lattner89375192008-03-16 00:19:01 +0000302 ClassDecl = ClassDecl->getSuperClass();
303 }
304 return NULL;
305}
306
Fariborz Jahaniandb3a4c12009-05-22 17:12:32 +0000307/// lookupInheritedClass - This method returns ObjCInterfaceDecl * of the super
308/// class whose name is passed as argument. If it is not one of the super classes
309/// the it returns NULL.
310ObjCInterfaceDecl *ObjCInterfaceDecl::lookupInheritedClass(
311 const IdentifierInfo*ICName) {
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000312 // FIXME: Should make sure no callers ever do this.
313 if (!hasDefinition())
314 return 0;
315
316 if (data().ExternallyCompleted)
Argyrios Kyrtzidis4e8b1362011-10-19 02:25:16 +0000317 LoadExternalDefinition();
318
Fariborz Jahaniandb3a4c12009-05-22 17:12:32 +0000319 ObjCInterfaceDecl* ClassDecl = this;
320 while (ClassDecl != NULL) {
321 if (ClassDecl->getIdentifier() == ICName)
322 return ClassDecl;
323 ClassDecl = ClassDecl->getSuperClass();
324 }
325 return NULL;
326}
327
Argyrios Kyrtzidis553376b2009-07-25 22:15:51 +0000328/// lookupMethod - This method returns an instance/class method by looking in
Chris Lattner89375192008-03-16 00:19:01 +0000329/// the class, its categories, and its super classes (using a linear search).
Fariborz Jahanianc806b902012-04-05 22:14:12 +0000330ObjCMethodDecl *ObjCInterfaceDecl::lookupMethod(Selector Sel,
331 bool isInstance,
332 bool shallowCategoryLookup) const {
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000333 // FIXME: Should make sure no callers ever do this.
334 if (!hasDefinition())
335 return 0;
336
Argyrios Kyrtzidis553376b2009-07-25 22:15:51 +0000337 const ObjCInterfaceDecl* ClassDecl = this;
Chris Lattner89375192008-03-16 00:19:01 +0000338 ObjCMethodDecl *MethodDecl = 0;
Mike Stump11289f42009-09-09 15:08:12 +0000339
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000340 if (data().ExternallyCompleted)
Douglas Gregor73693022010-12-01 23:49:52 +0000341 LoadExternalDefinition();
342
Chris Lattner89375192008-03-16 00:19:01 +0000343 while (ClassDecl != NULL) {
Argyrios Kyrtzidis553376b2009-07-25 22:15:51 +0000344 if ((MethodDecl = ClassDecl->getMethod(Sel, isInstance)))
Chris Lattner89375192008-03-16 00:19:01 +0000345 return MethodDecl;
Mike Stump11289f42009-09-09 15:08:12 +0000346
Chris Lattner89375192008-03-16 00:19:01 +0000347 // Didn't find one yet - look through protocols.
Argyrios Kyrtzidise7f3ef32012-03-13 01:09:41 +0000348 for (ObjCInterfaceDecl::protocol_iterator I = ClassDecl->protocol_begin(),
349 E = ClassDecl->protocol_end();
350 I != E; ++I)
Argyrios Kyrtzidis553376b2009-07-25 22:15:51 +0000351 if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
Chris Lattner89375192008-03-16 00:19:01 +0000352 return MethodDecl;
Fariborz Jahanianc806b902012-04-05 22:14:12 +0000353
354 // Didn't find one yet - now look through categories.
355 ObjCCategoryDecl *CatDecl = ClassDecl->getCategoryList();
356 while (CatDecl) {
357 if ((MethodDecl = CatDecl->getMethod(Sel, isInstance)))
358 return MethodDecl;
Fariborz Jahanian29082a52012-02-09 21:30:24 +0000359
Fariborz Jahanianc806b902012-04-05 22:14:12 +0000360 if (!shallowCategoryLookup) {
Fariborz Jahanian29082a52012-02-09 21:30:24 +0000361 // Didn't find one yet - look through protocols.
362 const ObjCList<ObjCProtocolDecl> &Protocols =
363 CatDecl->getReferencedProtocols();
364 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
365 E = Protocols.end(); I != E; ++I)
366 if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
367 return MethodDecl;
Fariborz Jahanian29082a52012-02-09 21:30:24 +0000368 }
Fariborz Jahanianc806b902012-04-05 22:14:12 +0000369 CatDecl = CatDecl->getNextClassCategory();
Chris Lattner89375192008-03-16 00:19:01 +0000370 }
Fariborz Jahanianc806b902012-04-05 22:14:12 +0000371
Chris Lattner89375192008-03-16 00:19:01 +0000372 ClassDecl = ClassDecl->getSuperClass();
373 }
374 return NULL;
375}
376
Anna Zaksc77a3b12012-07-27 19:07:44 +0000377// Will search "local" class/category implementations for a method decl.
378// If failed, then we search in class's root for an instance method.
379// Returns 0 if no method is found.
Fariborz Jahanianecbbb6e2010-12-03 23:37:08 +0000380ObjCMethodDecl *ObjCInterfaceDecl::lookupPrivateMethod(
381 const Selector &Sel,
Anna Zaks7044adc2012-07-30 20:31:21 +0000382 bool Instance) const {
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000383 // FIXME: Should make sure no callers ever do this.
384 if (!hasDefinition())
385 return 0;
386
387 if (data().ExternallyCompleted)
Argyrios Kyrtzidis4e8b1362011-10-19 02:25:16 +0000388 LoadExternalDefinition();
389
Steve Naroffbb69c942009-10-01 23:46:04 +0000390 ObjCMethodDecl *Method = 0;
391 if (ObjCImplementationDecl *ImpDecl = getImplementation())
Fariborz Jahanianecbbb6e2010-12-03 23:37:08 +0000392 Method = Instance ? ImpDecl->getInstanceMethod(Sel)
393 : ImpDecl->getClassMethod(Sel);
Anna Zaksc77a3b12012-07-27 19:07:44 +0000394
395 // Look through local category implementations associated with the class.
396 if (!Method)
397 Method = Instance ? getCategoryInstanceMethod(Sel)
398 : getCategoryClassMethod(Sel);
399
400 // Before we give up, check if the selector is an instance method.
401 // But only in the root. This matches gcc's behavior and what the
402 // runtime expects.
403 if (!Instance && !Method && !getSuperClass()) {
404 Method = lookupInstanceMethod(Sel);
405 // Look through local category implementations associated
406 // with the root class.
407 if (!Method)
408 Method = lookupPrivateMethod(Sel, true);
409 }
410
Steve Naroffbb69c942009-10-01 23:46:04 +0000411 if (!Method && getSuperClass())
Fariborz Jahanianecbbb6e2010-12-03 23:37:08 +0000412 return getSuperClass()->lookupPrivateMethod(Sel, Instance);
Steve Naroffbb69c942009-10-01 23:46:04 +0000413 return Method;
414}
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000415
416//===----------------------------------------------------------------------===//
417// ObjCMethodDecl
418//===----------------------------------------------------------------------===//
419
420ObjCMethodDecl *ObjCMethodDecl::Create(ASTContext &C,
Mike Stump11289f42009-09-09 15:08:12 +0000421 SourceLocation beginLoc,
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000422 SourceLocation endLoc,
423 Selector SelInfo, QualType T,
Douglas Gregor12852d92010-03-08 14:59:44 +0000424 TypeSourceInfo *ResultTInfo,
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000425 DeclContext *contextDecl,
426 bool isInstance,
427 bool isVariadic,
428 bool isSynthesized,
Argyrios Kyrtzidis004df6e2011-08-17 19:25:08 +0000429 bool isImplicitlyDeclared,
Fariborz Jahanian6e7e8cc2010-07-22 18:24:20 +0000430 bool isDefined,
Fariborz Jahaniand9235db2010-04-08 21:29:11 +0000431 ImplementationControl impControl,
Argyrios Kyrtzidis38493942011-10-03 06:36:29 +0000432 bool HasRelatedResultType) {
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000433 return new (C) ObjCMethodDecl(beginLoc, endLoc,
Douglas Gregor12852d92010-03-08 14:59:44 +0000434 SelInfo, T, ResultTInfo, contextDecl,
435 isInstance,
Argyrios Kyrtzidis004df6e2011-08-17 19:25:08 +0000436 isVariadic, isSynthesized, isImplicitlyDeclared,
437 isDefined,
Fariborz Jahanian6e7e8cc2010-07-22 18:24:20 +0000438 impControl,
Argyrios Kyrtzidis38493942011-10-03 06:36:29 +0000439 HasRelatedResultType);
Chris Lattner89375192008-03-16 00:19:01 +0000440}
441
Douglas Gregor72172e92012-01-05 21:55:30 +0000442ObjCMethodDecl *ObjCMethodDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
443 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCMethodDecl));
444 return new (Mem) ObjCMethodDecl(SourceLocation(), SourceLocation(),
445 Selector(), QualType(), 0, 0);
446}
447
Douglas Gregora6017bb2012-10-09 17:21:28 +0000448Stmt *ObjCMethodDecl::getBody() const {
449 return Body.get(getASTContext().getExternalSource());
450}
451
Argyrios Kyrtzidisdcaaa212011-10-14 08:02:31 +0000452void ObjCMethodDecl::setAsRedeclaration(const ObjCMethodDecl *PrevMethod) {
453 assert(PrevMethod);
454 getASTContext().setObjCMethodRedeclaration(PrevMethod, this);
455 IsRedeclaration = true;
Argyrios Kyrtzidisdb215962011-10-14 17:41:52 +0000456 PrevMethod->HasRedeclaration = true;
Argyrios Kyrtzidisdcaaa212011-10-14 08:02:31 +0000457}
458
Argyrios Kyrtzidisb8c3aaf2011-10-03 06:37:04 +0000459void ObjCMethodDecl::setParamsAndSelLocs(ASTContext &C,
460 ArrayRef<ParmVarDecl*> Params,
461 ArrayRef<SourceLocation> SelLocs) {
462 ParamsAndSelLocs = 0;
463 NumParams = Params.size();
464 if (Params.empty() && SelLocs.empty())
465 return;
466
467 unsigned Size = sizeof(ParmVarDecl *) * NumParams +
468 sizeof(SourceLocation) * SelLocs.size();
469 ParamsAndSelLocs = C.Allocate(Size);
470 std::copy(Params.begin(), Params.end(), getParams());
471 std::copy(SelLocs.begin(), SelLocs.end(), getStoredSelLocs());
472}
473
474void ObjCMethodDecl::getSelectorLocs(
475 SmallVectorImpl<SourceLocation> &SelLocs) const {
476 for (unsigned i = 0, e = getNumSelectorLocs(); i != e; ++i)
477 SelLocs.push_back(getSelectorLoc(i));
478}
479
480void ObjCMethodDecl::setMethodParams(ASTContext &C,
481 ArrayRef<ParmVarDecl*> Params,
482 ArrayRef<SourceLocation> SelLocs) {
483 assert((!SelLocs.empty() || isImplicit()) &&
484 "No selector locs for non-implicit method");
485 if (isImplicit())
486 return setParamsAndSelLocs(C, Params, ArrayRef<SourceLocation>());
487
Argyrios Kyrtzidis33b4bfc2012-06-16 00:46:02 +0000488 SelLocsKind = hasStandardSelectorLocs(getSelector(), SelLocs, Params,
489 DeclEndLoc);
Argyrios Kyrtzidisb8c3aaf2011-10-03 06:37:04 +0000490 if (SelLocsKind != SelLoc_NonStandard)
491 return setParamsAndSelLocs(C, Params, ArrayRef<SourceLocation>());
492
493 setParamsAndSelLocs(C, Params, SelLocs);
494}
495
Argyrios Kyrtzidisa8cf0be2009-07-21 00:06:36 +0000496/// \brief A definition will return its interface declaration.
497/// An interface declaration will return its definition.
498/// Otherwise it will return itself.
499ObjCMethodDecl *ObjCMethodDecl::getNextRedeclaration() {
500 ASTContext &Ctx = getASTContext();
Argyrios Kyrtzidisdb215962011-10-14 17:41:52 +0000501 ObjCMethodDecl *Redecl = 0;
502 if (HasRedeclaration)
503 Redecl = const_cast<ObjCMethodDecl*>(Ctx.getObjCMethodRedeclaration(this));
Argyrios Kyrtzidisc5e829c2011-10-14 06:48:06 +0000504 if (Redecl)
505 return Redecl;
506
Argyrios Kyrtzidisa8cf0be2009-07-21 00:06:36 +0000507 Decl *CtxD = cast<Decl>(getDeclContext());
508
509 if (ObjCInterfaceDecl *IFD = dyn_cast<ObjCInterfaceDecl>(CtxD)) {
510 if (ObjCImplementationDecl *ImplD = Ctx.getObjCImplementation(IFD))
511 Redecl = ImplD->getMethod(getSelector(), isInstanceMethod());
512
513 } else if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(CtxD)) {
514 if (ObjCCategoryImplDecl *ImplD = Ctx.getObjCImplementation(CD))
515 Redecl = ImplD->getMethod(getSelector(), isInstanceMethod());
516
Argyrios Kyrtzidisa56fa192009-07-28 05:11:05 +0000517 } else if (ObjCImplementationDecl *ImplD =
518 dyn_cast<ObjCImplementationDecl>(CtxD)) {
Argyrios Kyrtzidisa8cf0be2009-07-21 00:06:36 +0000519 if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
520 Redecl = IFD->getMethod(getSelector(), isInstanceMethod());
Argyrios Kyrtzidisa56fa192009-07-28 05:11:05 +0000521
522 } else if (ObjCCategoryImplDecl *CImplD =
523 dyn_cast<ObjCCategoryImplDecl>(CtxD)) {
Steve Narofff406f4d2009-10-29 21:11:04 +0000524 if (ObjCCategoryDecl *CatD = CImplD->getCategoryDecl())
Argyrios Kyrtzidisa56fa192009-07-28 05:11:05 +0000525 Redecl = CatD->getMethod(getSelector(), isInstanceMethod());
Argyrios Kyrtzidisa8cf0be2009-07-21 00:06:36 +0000526 }
527
Argyrios Kyrtzidisdcaaa212011-10-14 08:02:31 +0000528 if (!Redecl && isRedeclaration()) {
529 // This is the last redeclaration, go back to the first method.
530 return cast<ObjCContainerDecl>(CtxD)->getMethod(getSelector(),
531 isInstanceMethod());
532 }
533
Argyrios Kyrtzidisa8cf0be2009-07-21 00:06:36 +0000534 return Redecl ? Redecl : this;
535}
536
Argyrios Kyrtzidisf390c432009-07-28 05:11:17 +0000537ObjCMethodDecl *ObjCMethodDecl::getCanonicalDecl() {
538 Decl *CtxD = cast<Decl>(getDeclContext());
539
540 if (ObjCImplementationDecl *ImplD = dyn_cast<ObjCImplementationDecl>(CtxD)) {
541 if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
542 if (ObjCMethodDecl *MD = IFD->getMethod(getSelector(),
543 isInstanceMethod()))
544 return MD;
545
546 } else if (ObjCCategoryImplDecl *CImplD =
547 dyn_cast<ObjCCategoryImplDecl>(CtxD)) {
Steve Narofff406f4d2009-10-29 21:11:04 +0000548 if (ObjCCategoryDecl *CatD = CImplD->getCategoryDecl())
Argyrios Kyrtzidisf390c432009-07-28 05:11:17 +0000549 if (ObjCMethodDecl *MD = CatD->getMethod(getSelector(),
550 isInstanceMethod()))
551 return MD;
552 }
553
Argyrios Kyrtzidis690dccd2011-10-17 19:48:09 +0000554 if (isRedeclaration())
555 return cast<ObjCContainerDecl>(CtxD)->getMethod(getSelector(),
556 isInstanceMethod());
557
Argyrios Kyrtzidisf390c432009-07-28 05:11:17 +0000558 return this;
559}
560
Argyrios Kyrtzidis33b4bfc2012-06-16 00:46:02 +0000561SourceLocation ObjCMethodDecl::getLocEnd() const {
562 if (Stmt *Body = getBody())
563 return Body->getLocEnd();
564 return DeclEndLoc;
565}
566
John McCallb4526252011-03-02 01:50:55 +0000567ObjCMethodFamily ObjCMethodDecl::getMethodFamily() const {
568 ObjCMethodFamily family = static_cast<ObjCMethodFamily>(Family);
John McCallfb55f852011-03-02 21:01:41 +0000569 if (family != static_cast<unsigned>(InvalidObjCMethodFamily))
John McCallb4526252011-03-02 01:50:55 +0000570 return family;
571
John McCall86bc21f2011-03-02 11:33:24 +0000572 // Check for an explicit attribute.
573 if (const ObjCMethodFamilyAttr *attr = getAttr<ObjCMethodFamilyAttr>()) {
574 // The unfortunate necessity of mapping between enums here is due
575 // to the attributes framework.
576 switch (attr->getFamily()) {
577 case ObjCMethodFamilyAttr::OMF_None: family = OMF_None; break;
578 case ObjCMethodFamilyAttr::OMF_alloc: family = OMF_alloc; break;
579 case ObjCMethodFamilyAttr::OMF_copy: family = OMF_copy; break;
580 case ObjCMethodFamilyAttr::OMF_init: family = OMF_init; break;
581 case ObjCMethodFamilyAttr::OMF_mutableCopy: family = OMF_mutableCopy; break;
582 case ObjCMethodFamilyAttr::OMF_new: family = OMF_new; break;
583 }
584 Family = static_cast<unsigned>(family);
585 return family;
586 }
587
John McCallb4526252011-03-02 01:50:55 +0000588 family = getSelector().getMethodFamily();
589 switch (family) {
590 case OMF_None: break;
591
592 // init only has a conventional meaning for an instance method, and
593 // it has to return an object.
594 case OMF_init:
595 if (!isInstanceMethod() || !getResultType()->isObjCObjectPointerType())
596 family = OMF_None;
597 break;
598
599 // alloc/copy/new have a conventional meaning for both class and
600 // instance methods, but they require an object return.
601 case OMF_alloc:
602 case OMF_copy:
603 case OMF_mutableCopy:
604 case OMF_new:
605 if (!getResultType()->isObjCObjectPointerType())
606 family = OMF_None;
607 break;
608
609 // These selectors have a conventional meaning only for instance methods.
610 case OMF_dealloc:
Nico Weber1fb82662011-08-28 22:35:17 +0000611 case OMF_finalize:
John McCallb4526252011-03-02 01:50:55 +0000612 case OMF_retain:
613 case OMF_release:
614 case OMF_autorelease:
615 case OMF_retainCount:
Douglas Gregor33823722011-06-11 01:09:30 +0000616 case OMF_self:
John McCallb4526252011-03-02 01:50:55 +0000617 if (!isInstanceMethod())
618 family = OMF_None;
619 break;
Fariborz Jahanianb7a77362011-07-05 22:38:59 +0000620
621 case OMF_performSelector:
622 if (!isInstanceMethod() ||
623 !getResultType()->isObjCIdType())
624 family = OMF_None;
625 else {
626 unsigned noParams = param_size();
627 if (noParams < 1 || noParams > 3)
628 family = OMF_None;
629 else {
630 ObjCMethodDecl::arg_type_iterator it = arg_type_begin();
631 QualType ArgT = (*it);
632 if (!ArgT->isObjCSelType()) {
633 family = OMF_None;
634 break;
635 }
636 while (--noParams) {
637 it++;
638 ArgT = (*it);
639 if (!ArgT->isObjCIdType()) {
640 family = OMF_None;
641 break;
642 }
643 }
644 }
645 }
646 break;
647
John McCallb4526252011-03-02 01:50:55 +0000648 }
649
650 // Cache the result.
651 Family = static_cast<unsigned>(family);
652 return family;
653}
654
Mike Stump11289f42009-09-09 15:08:12 +0000655void ObjCMethodDecl::createImplicitParams(ASTContext &Context,
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000656 const ObjCInterfaceDecl *OID) {
657 QualType selfTy;
658 if (isInstanceMethod()) {
659 // There may be no interface context due to error in declaration
660 // of the interface (which has been reported). Recover gracefully.
661 if (OID) {
Daniel Dunbaraefc2b92009-04-22 04:34:53 +0000662 selfTy = Context.getObjCInterfaceType(OID);
Steve Naroff7cae42b2009-07-10 23:34:53 +0000663 selfTy = Context.getObjCObjectPointerType(selfTy);
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000664 } else {
665 selfTy = Context.getObjCIdType();
666 }
667 } else // we have a factory method.
668 selfTy = Context.getObjCClassType();
669
John McCalld4631322011-06-17 06:42:21 +0000670 bool selfIsPseudoStrong = false;
John McCall31168b02011-06-15 23:02:42 +0000671 bool selfIsConsumed = false;
Ted Kremenek1fcdaa92011-11-14 21:59:25 +0000672
David Blaikiebbafb8a2012-03-11 07:00:24 +0000673 if (Context.getLangOpts().ObjCAutoRefCount) {
Ted Kremenek1fcdaa92011-11-14 21:59:25 +0000674 if (isInstanceMethod()) {
675 selfIsConsumed = hasAttr<NSConsumesSelfAttr>();
John McCall31168b02011-06-15 23:02:42 +0000676
Ted Kremenek1fcdaa92011-11-14 21:59:25 +0000677 // 'self' is always __strong. It's actually pseudo-strong except
678 // in init methods (or methods labeled ns_consumes_self), though.
679 Qualifiers qs;
680 qs.setObjCLifetime(Qualifiers::OCL_Strong);
681 selfTy = Context.getQualifiedType(selfTy, qs);
John McCall31168b02011-06-15 23:02:42 +0000682
Ted Kremenek1fcdaa92011-11-14 21:59:25 +0000683 // In addition, 'self' is const unless this is an init method.
684 if (getMethodFamily() != OMF_init && !selfIsConsumed) {
685 selfTy = selfTy.withConst();
686 selfIsPseudoStrong = true;
687 }
688 }
689 else {
690 assert(isClassMethod());
691 // 'self' is always const in class methods.
John McCall31168b02011-06-15 23:02:42 +0000692 selfTy = selfTy.withConst();
John McCalld4631322011-06-17 06:42:21 +0000693 selfIsPseudoStrong = true;
694 }
John McCall31168b02011-06-15 23:02:42 +0000695 }
696
697 ImplicitParamDecl *self
698 = ImplicitParamDecl::Create(Context, this, SourceLocation(),
699 &Context.Idents.get("self"), selfTy);
700 setSelfDecl(self);
701
702 if (selfIsConsumed)
703 self->addAttr(new (Context) NSConsumedAttr(SourceLocation(), Context));
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000704
John McCalld4631322011-06-17 06:42:21 +0000705 if (selfIsPseudoStrong)
706 self->setARCPseudoStrong(true);
707
Mike Stump11289f42009-09-09 15:08:12 +0000708 setCmdDecl(ImplicitParamDecl::Create(Context, this, SourceLocation(),
709 &Context.Idents.get("_cmd"),
Steve Naroff04f2d142009-04-20 15:06:07 +0000710 Context.getObjCSelType()));
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000711}
712
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000713ObjCInterfaceDecl *ObjCMethodDecl::getClassInterface() {
714 if (ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(getDeclContext()))
715 return ID;
716 if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(getDeclContext()))
717 return CD->getClassInterface();
Argyrios Kyrtzidis2cee40d2009-07-28 05:10:52 +0000718 if (ObjCImplDecl *IMD = dyn_cast<ObjCImplDecl>(getDeclContext()))
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000719 return IMD->getClassInterface();
Argyrios Kyrtzidis2cee40d2009-07-28 05:10:52 +0000720
721 assert(!isa<ObjCProtocolDecl>(getDeclContext()) && "It's a protocol method");
David Blaikie83d382b2011-09-23 05:06:16 +0000722 llvm_unreachable("unknown method context");
Fariborz Jahanianfbbaf6a2008-12-05 22:32:48 +0000723}
724
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000725//===----------------------------------------------------------------------===//
726// ObjCInterfaceDecl
727//===----------------------------------------------------------------------===//
728
Douglas Gregord53ae832012-01-17 18:09:05 +0000729ObjCInterfaceDecl *ObjCInterfaceDecl::Create(const ASTContext &C,
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000730 DeclContext *DC,
731 SourceLocation atLoc,
Mike Stump11289f42009-09-09 15:08:12 +0000732 IdentifierInfo *Id,
Douglas Gregorab1ec82e2011-12-16 03:12:41 +0000733 ObjCInterfaceDecl *PrevDecl,
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000734 SourceLocation ClassLoc,
Douglas Gregordc9166c2011-12-15 20:29:51 +0000735 bool isInternal){
Douglas Gregorab1ec82e2011-12-16 03:12:41 +0000736 ObjCInterfaceDecl *Result = new (C) ObjCInterfaceDecl(DC, atLoc, Id, ClassLoc,
Douglas Gregor81252352011-12-16 22:37:11 +0000737 PrevDecl, isInternal);
Douglas Gregorab1ec82e2011-12-16 03:12:41 +0000738 C.getObjCInterfaceType(Result, PrevDecl);
Douglas Gregorab1ec82e2011-12-16 03:12:41 +0000739 return Result;
740}
741
Douglas Gregor72172e92012-01-05 21:55:30 +0000742ObjCInterfaceDecl *ObjCInterfaceDecl::CreateDeserialized(ASTContext &C,
743 unsigned ID) {
744 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCInterfaceDecl));
745 return new (Mem) ObjCInterfaceDecl(0, SourceLocation(), 0, SourceLocation(),
746 0, false);
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000747}
748
749ObjCInterfaceDecl::
750ObjCInterfaceDecl(DeclContext *DC, SourceLocation atLoc, IdentifierInfo *Id,
Douglas Gregor81252352011-12-16 22:37:11 +0000751 SourceLocation CLoc, ObjCInterfaceDecl *PrevDecl,
752 bool isInternal)
Argyrios Kyrtzidis52f53fb2011-10-04 04:48:02 +0000753 : ObjCContainerDecl(ObjCInterface, DC, Id, CLoc, atLoc),
Douglas Gregordc9166c2011-12-15 20:29:51 +0000754 TypeForDecl(0), Data()
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000755{
Douglas Gregor81252352011-12-16 22:37:11 +0000756 setPreviousDeclaration(PrevDecl);
757
758 // Copy the 'data' pointer over.
759 if (PrevDecl)
760 Data = PrevDecl->Data;
761
Argyrios Kyrtzidisae8e7922011-11-15 06:20:21 +0000762 setImplicit(isInternal);
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000763}
764
Douglas Gregor73693022010-12-01 23:49:52 +0000765void ObjCInterfaceDecl::LoadExternalDefinition() const {
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000766 assert(data().ExternallyCompleted && "Class is not externally completed");
767 data().ExternallyCompleted = false;
Douglas Gregor73693022010-12-01 23:49:52 +0000768 getASTContext().getExternalSource()->CompleteType(
769 const_cast<ObjCInterfaceDecl *>(this));
770}
771
772void ObjCInterfaceDecl::setExternallyCompleted() {
773 assert(getASTContext().getExternalSource() &&
774 "Class can't be externally completed without an external source");
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000775 assert(hasDefinition() &&
Douglas Gregor73693022010-12-01 23:49:52 +0000776 "Forward declarations can't be externally completed");
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000777 data().ExternallyCompleted = true;
Douglas Gregor73693022010-12-01 23:49:52 +0000778}
779
Argyrios Kyrtzidis6d9fab72009-07-21 00:05:53 +0000780ObjCImplementationDecl *ObjCInterfaceDecl::getImplementation() const {
Douglas Gregordc9166c2011-12-15 20:29:51 +0000781 if (const ObjCInterfaceDecl *Def = getDefinition()) {
782 if (data().ExternallyCompleted)
783 LoadExternalDefinition();
784
785 return getASTContext().getObjCImplementation(
786 const_cast<ObjCInterfaceDecl*>(Def));
787 }
788
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000789 // FIXME: Should make sure no callers ever do this.
Douglas Gregordc9166c2011-12-15 20:29:51 +0000790 return 0;
Argyrios Kyrtzidis6d9fab72009-07-21 00:05:53 +0000791}
792
793void ObjCInterfaceDecl::setImplementation(ObjCImplementationDecl *ImplD) {
Douglas Gregordc9166c2011-12-15 20:29:51 +0000794 getASTContext().setObjCImplementation(getDefinition(), ImplD);
Argyrios Kyrtzidis6d9fab72009-07-21 00:05:53 +0000795}
796
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000797/// all_declared_ivar_begin - return first ivar declared in this class,
798/// its extensions and its implementation. Lazily build the list on first
799/// access.
800ObjCIvarDecl *ObjCInterfaceDecl::all_declared_ivar_begin() {
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000801 // FIXME: Should make sure no callers ever do this.
802 if (!hasDefinition())
803 return 0;
804
805 if (data().IvarList)
806 return data().IvarList;
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000807
808 ObjCIvarDecl *curIvar = 0;
809 if (!ivar_empty()) {
810 ObjCInterfaceDecl::ivar_iterator I = ivar_begin(), E = ivar_end();
David Blaikie40ed2972012-06-06 20:45:41 +0000811 data().IvarList = *I; ++I;
812 for (curIvar = data().IvarList; I != E; curIvar = *I, ++I)
813 curIvar->setNextIvar(*I);
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000814 }
815
816 for (const ObjCCategoryDecl *CDecl = getFirstClassExtension(); CDecl;
817 CDecl = CDecl->getNextClassExtension()) {
818 if (!CDecl->ivar_empty()) {
819 ObjCCategoryDecl::ivar_iterator I = CDecl->ivar_begin(),
820 E = CDecl->ivar_end();
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000821 if (!data().IvarList) {
David Blaikie40ed2972012-06-06 20:45:41 +0000822 data().IvarList = *I; ++I;
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000823 curIvar = data().IvarList;
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000824 }
David Blaikie40ed2972012-06-06 20:45:41 +0000825 for ( ;I != E; curIvar = *I, ++I)
826 curIvar->setNextIvar(*I);
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000827 }
828 }
829
830 if (ObjCImplementationDecl *ImplDecl = getImplementation()) {
831 if (!ImplDecl->ivar_empty()) {
832 ObjCImplementationDecl::ivar_iterator I = ImplDecl->ivar_begin(),
833 E = ImplDecl->ivar_end();
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000834 if (!data().IvarList) {
David Blaikie40ed2972012-06-06 20:45:41 +0000835 data().IvarList = *I; ++I;
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000836 curIvar = data().IvarList;
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000837 }
David Blaikie40ed2972012-06-06 20:45:41 +0000838 for ( ;I != E; curIvar = *I, ++I)
839 curIvar->setNextIvar(*I);
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000840 }
841 }
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000842 return data().IvarList;
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000843}
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000844
845/// FindCategoryDeclaration - Finds category declaration in the list of
846/// categories for this class and returns it. Name of the category is passed
847/// in 'CategoryId'. If category not found, return 0;
Fariborz Jahanianfbbaf6a2008-12-05 22:32:48 +0000848///
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000849ObjCCategoryDecl *
850ObjCInterfaceDecl::FindCategoryDeclaration(IdentifierInfo *CategoryId) const {
Argyrios Kyrtzidis4af2cb32012-03-02 19:14:29 +0000851 // FIXME: Should make sure no callers ever do this.
852 if (!hasDefinition())
853 return 0;
854
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000855 if (data().ExternallyCompleted)
Douglas Gregor73693022010-12-01 23:49:52 +0000856 LoadExternalDefinition();
857
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000858 for (ObjCCategoryDecl *Category = getCategoryList();
859 Category; Category = Category->getNextClassCategory())
860 if (Category->getIdentifier() == CategoryId)
861 return Category;
Fariborz Jahanianfbbaf6a2008-12-05 22:32:48 +0000862 return 0;
863}
864
Argyrios Kyrtzidis1559d67b2009-07-21 00:06:20 +0000865ObjCMethodDecl *
866ObjCInterfaceDecl::getCategoryInstanceMethod(Selector Sel) const {
867 for (ObjCCategoryDecl *Category = getCategoryList();
868 Category; Category = Category->getNextClassCategory())
869 if (ObjCCategoryImplDecl *Impl = Category->getImplementation())
870 if (ObjCMethodDecl *MD = Impl->getInstanceMethod(Sel))
871 return MD;
872 return 0;
873}
874
875ObjCMethodDecl *ObjCInterfaceDecl::getCategoryClassMethod(Selector Sel) const {
876 for (ObjCCategoryDecl *Category = getCategoryList();
877 Category; Category = Category->getNextClassCategory())
878 if (ObjCCategoryImplDecl *Impl = Category->getImplementation())
879 if (ObjCMethodDecl *MD = Impl->getClassMethod(Sel))
880 return MD;
881 return 0;
882}
883
Fariborz Jahanian3f8917a2009-08-11 22:02:25 +0000884/// ClassImplementsProtocol - Checks that 'lProto' protocol
885/// has been implemented in IDecl class, its super class or categories (if
886/// lookupCategory is true).
887bool ObjCInterfaceDecl::ClassImplementsProtocol(ObjCProtocolDecl *lProto,
888 bool lookupCategory,
889 bool RHSIsQualifiedID) {
Douglas Gregorc0ac7d62011-12-15 05:27:12 +0000890 if (!hasDefinition())
891 return false;
892
Fariborz Jahanian3f8917a2009-08-11 22:02:25 +0000893 ObjCInterfaceDecl *IDecl = this;
894 // 1st, look up the class.
Argyrios Kyrtzidise7f3ef32012-03-13 01:09:41 +0000895 for (ObjCInterfaceDecl::protocol_iterator
896 PI = IDecl->protocol_begin(), E = IDecl->protocol_end(); PI != E; ++PI){
Fariborz Jahanian3f8917a2009-08-11 22:02:25 +0000897 if (getASTContext().ProtocolCompatibleWithProtocol(lProto, *PI))
898 return true;
899 // This is dubious and is added to be compatible with gcc. In gcc, it is
900 // also allowed assigning a protocol-qualified 'id' type to a LHS object
901 // when protocol in qualified LHS is in list of protocols in the rhs 'id'
902 // object. This IMO, should be a bug.
903 // FIXME: Treat this as an extension, and flag this as an error when GCC
904 // extensions are not enabled.
Mike Stump11289f42009-09-09 15:08:12 +0000905 if (RHSIsQualifiedID &&
Fariborz Jahanian3f8917a2009-08-11 22:02:25 +0000906 getASTContext().ProtocolCompatibleWithProtocol(*PI, lProto))
907 return true;
908 }
Mike Stump11289f42009-09-09 15:08:12 +0000909
Fariborz Jahanian3f8917a2009-08-11 22:02:25 +0000910 // 2nd, look up the category.
911 if (lookupCategory)
912 for (ObjCCategoryDecl *CDecl = IDecl->getCategoryList(); CDecl;
913 CDecl = CDecl->getNextClassCategory()) {
914 for (ObjCCategoryDecl::protocol_iterator PI = CDecl->protocol_begin(),
915 E = CDecl->protocol_end(); PI != E; ++PI)
916 if (getASTContext().ProtocolCompatibleWithProtocol(lProto, *PI))
917 return true;
918 }
Mike Stump11289f42009-09-09 15:08:12 +0000919
Fariborz Jahanian3f8917a2009-08-11 22:02:25 +0000920 // 3rd, look up the super class(s)
921 if (IDecl->getSuperClass())
922 return
923 IDecl->getSuperClass()->ClassImplementsProtocol(lProto, lookupCategory,
924 RHSIsQualifiedID);
Mike Stump11289f42009-09-09 15:08:12 +0000925
Fariborz Jahanian3f8917a2009-08-11 22:02:25 +0000926 return false;
927}
928
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000929//===----------------------------------------------------------------------===//
930// ObjCIvarDecl
931//===----------------------------------------------------------------------===//
932
David Blaikie68e081d2011-12-20 02:48:34 +0000933void ObjCIvarDecl::anchor() { }
934
Daniel Dunbarfe3ead72010-04-02 20:10:03 +0000935ObjCIvarDecl *ObjCIvarDecl::Create(ASTContext &C, ObjCContainerDecl *DC,
Abramo Bagnaradff19302011-03-08 08:55:46 +0000936 SourceLocation StartLoc,
937 SourceLocation IdLoc, IdentifierInfo *Id,
John McCallbcd03502009-12-07 02:54:59 +0000938 QualType T, TypeSourceInfo *TInfo,
Fariborz Jahanian18722982010-07-17 00:59:30 +0000939 AccessControl ac, Expr *BW,
940 bool synthesized) {
Daniel Dunbarfe3ead72010-04-02 20:10:03 +0000941 if (DC) {
942 // Ivar's can only appear in interfaces, implementations (via synthesized
943 // properties), and class extensions (via direct declaration, or synthesized
944 // properties).
945 //
946 // FIXME: This should really be asserting this:
947 // (isa<ObjCCategoryDecl>(DC) &&
948 // cast<ObjCCategoryDecl>(DC)->IsClassExtension()))
949 // but unfortunately we sometimes place ivars into non-class extension
950 // categories on error. This breaks an AST invariant, and should not be
951 // fixed.
952 assert((isa<ObjCInterfaceDecl>(DC) || isa<ObjCImplementationDecl>(DC) ||
953 isa<ObjCCategoryDecl>(DC)) &&
954 "Invalid ivar decl context!");
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000955 // Once a new ivar is created in any of class/class-extension/implementation
956 // decl contexts, the previously built IvarList must be rebuilt.
957 ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(DC);
958 if (!ID) {
Eric Christopherf8378ca2012-07-19 22:22:55 +0000959 if (ObjCImplementationDecl *IM = dyn_cast<ObjCImplementationDecl>(DC))
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000960 ID = IM->getClassInterface();
Eric Christopherf8378ca2012-07-19 22:22:55 +0000961 else
962 ID = cast<ObjCCategoryDecl>(DC)->getClassInterface();
Fariborz Jahaniana50b3a22010-08-20 21:21:08 +0000963 }
964 ID->setIvarList(0);
Daniel Dunbarfe3ead72010-04-02 20:10:03 +0000965 }
966
Abramo Bagnaradff19302011-03-08 08:55:46 +0000967 return new (C) ObjCIvarDecl(DC, StartLoc, IdLoc, Id, T, TInfo,
968 ac, BW, synthesized);
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000969}
970
Douglas Gregor72172e92012-01-05 21:55:30 +0000971ObjCIvarDecl *ObjCIvarDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
972 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCIvarDecl));
973 return new (Mem) ObjCIvarDecl(0, SourceLocation(), SourceLocation(), 0,
974 QualType(), 0, ObjCIvarDecl::None, 0, false);
975}
976
Daniel Dunbar89947ea2010-04-02 21:13:59 +0000977const ObjCInterfaceDecl *ObjCIvarDecl::getContainingInterface() const {
978 const ObjCContainerDecl *DC = cast<ObjCContainerDecl>(getDeclContext());
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +0000979
Daniel Dunbar89947ea2010-04-02 21:13:59 +0000980 switch (DC->getKind()) {
981 default:
982 case ObjCCategoryImpl:
983 case ObjCProtocol:
David Blaikie83d382b2011-09-23 05:06:16 +0000984 llvm_unreachable("invalid ivar container!");
Daniel Dunbar89947ea2010-04-02 21:13:59 +0000985
986 // Ivars can only appear in class extension categories.
987 case ObjCCategory: {
988 const ObjCCategoryDecl *CD = cast<ObjCCategoryDecl>(DC);
989 assert(CD->IsClassExtension() && "invalid container for ivar!");
990 return CD->getClassInterface();
991 }
992
993 case ObjCImplementation:
994 return cast<ObjCImplementationDecl>(DC)->getClassInterface();
995
996 case ObjCInterface:
997 return cast<ObjCInterfaceDecl>(DC);
998 }
999}
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001000
1001//===----------------------------------------------------------------------===//
1002// ObjCAtDefsFieldDecl
1003//===----------------------------------------------------------------------===//
1004
David Blaikie68e081d2011-12-20 02:48:34 +00001005void ObjCAtDefsFieldDecl::anchor() { }
1006
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001007ObjCAtDefsFieldDecl
Abramo Bagnaradff19302011-03-08 08:55:46 +00001008*ObjCAtDefsFieldDecl::Create(ASTContext &C, DeclContext *DC,
1009 SourceLocation StartLoc, SourceLocation IdLoc,
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001010 IdentifierInfo *Id, QualType T, Expr *BW) {
Abramo Bagnaradff19302011-03-08 08:55:46 +00001011 return new (C) ObjCAtDefsFieldDecl(DC, StartLoc, IdLoc, Id, T, BW);
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001012}
1013
Douglas Gregor72172e92012-01-05 21:55:30 +00001014ObjCAtDefsFieldDecl *ObjCAtDefsFieldDecl::CreateDeserialized(ASTContext &C,
1015 unsigned ID) {
1016 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCAtDefsFieldDecl));
1017 return new (Mem) ObjCAtDefsFieldDecl(0, SourceLocation(), SourceLocation(),
1018 0, QualType(), 0);
1019}
1020
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001021//===----------------------------------------------------------------------===//
1022// ObjCProtocolDecl
1023//===----------------------------------------------------------------------===//
1024
David Blaikie68e081d2011-12-20 02:48:34 +00001025void ObjCProtocolDecl::anchor() { }
1026
Douglas Gregor32c17572012-01-01 20:30:41 +00001027ObjCProtocolDecl::ObjCProtocolDecl(DeclContext *DC, IdentifierInfo *Id,
1028 SourceLocation nameLoc,
1029 SourceLocation atStartLoc,
Douglas Gregor05a1f4d2012-01-01 22:06:18 +00001030 ObjCProtocolDecl *PrevDecl)
1031 : ObjCContainerDecl(ObjCProtocol, DC, Id, nameLoc, atStartLoc), Data()
Douglas Gregor32c17572012-01-01 20:30:41 +00001032{
1033 setPreviousDeclaration(PrevDecl);
1034 if (PrevDecl)
1035 Data = PrevDecl->Data;
1036}
1037
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001038ObjCProtocolDecl *ObjCProtocolDecl::Create(ASTContext &C, DeclContext *DC,
Argyrios Kyrtzidis52f53fb2011-10-04 04:48:02 +00001039 IdentifierInfo *Id,
1040 SourceLocation nameLoc,
Argyrios Kyrtzidis1f4bee52011-10-17 19:48:06 +00001041 SourceLocation atStartLoc,
Douglas Gregor05a1f4d2012-01-01 22:06:18 +00001042 ObjCProtocolDecl *PrevDecl) {
Douglas Gregor32c17572012-01-01 20:30:41 +00001043 ObjCProtocolDecl *Result
Douglas Gregor05a1f4d2012-01-01 22:06:18 +00001044 = new (C) ObjCProtocolDecl(DC, Id, nameLoc, atStartLoc, PrevDecl);
Douglas Gregor32c17572012-01-01 20:30:41 +00001045
1046 return Result;
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001047}
1048
Douglas Gregor72172e92012-01-05 21:55:30 +00001049ObjCProtocolDecl *ObjCProtocolDecl::CreateDeserialized(ASTContext &C,
1050 unsigned ID) {
1051 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCProtocolDecl));
1052 return new (Mem) ObjCProtocolDecl(0, 0, SourceLocation(), SourceLocation(),
1053 0);
1054}
1055
Steve Naroff114aecb2009-03-01 16:12:44 +00001056ObjCProtocolDecl *ObjCProtocolDecl::lookupProtocolNamed(IdentifierInfo *Name) {
1057 ObjCProtocolDecl *PDecl = this;
1058
1059 if (Name == getIdentifier())
1060 return PDecl;
1061
1062 for (protocol_iterator I = protocol_begin(), E = protocol_end(); I != E; ++I)
1063 if ((PDecl = (*I)->lookupProtocolNamed(Name)))
1064 return PDecl;
Mike Stump11289f42009-09-09 15:08:12 +00001065
Steve Naroff114aecb2009-03-01 16:12:44 +00001066 return NULL;
1067}
1068
Argyrios Kyrtzidise6ed65b2009-07-25 22:15:38 +00001069// lookupMethod - Lookup a instance/class method in the protocol and protocols
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001070// it inherited.
Argyrios Kyrtzidise6ed65b2009-07-25 22:15:38 +00001071ObjCMethodDecl *ObjCProtocolDecl::lookupMethod(Selector Sel,
1072 bool isInstance) const {
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001073 ObjCMethodDecl *MethodDecl = NULL;
Mike Stump11289f42009-09-09 15:08:12 +00001074
Argyrios Kyrtzidise6ed65b2009-07-25 22:15:38 +00001075 if ((MethodDecl = getMethod(Sel, isInstance)))
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001076 return MethodDecl;
Mike Stump11289f42009-09-09 15:08:12 +00001077
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001078 for (protocol_iterator I = protocol_begin(), E = protocol_end(); I != E; ++I)
Argyrios Kyrtzidise6ed65b2009-07-25 22:15:38 +00001079 if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001080 return MethodDecl;
1081 return NULL;
1082}
1083
Douglas Gregore6e48b12012-01-01 19:29:29 +00001084void ObjCProtocolDecl::allocateDefinitionData() {
1085 assert(!Data && "Protocol already has a definition!");
Douglas Gregora715bff2012-01-01 19:51:50 +00001086 Data = new (getASTContext()) DefinitionData;
1087 Data->Definition = this;
Douglas Gregore6e48b12012-01-01 19:29:29 +00001088}
1089
1090void ObjCProtocolDecl::startDefinition() {
1091 allocateDefinitionData();
Douglas Gregora715bff2012-01-01 19:51:50 +00001092
1093 // Update all of the declarations with a pointer to the definition.
1094 for (redecl_iterator RD = redecls_begin(), RDEnd = redecls_end();
1095 RD != RDEnd; ++RD)
1096 RD->Data = this->Data;
Argyrios Kyrtzidisb97a4022011-11-12 21:07:46 +00001097}
1098
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001099//===----------------------------------------------------------------------===//
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001100// ObjCCategoryDecl
1101//===----------------------------------------------------------------------===//
1102
David Blaikie68e081d2011-12-20 02:48:34 +00001103void ObjCCategoryDecl::anchor() { }
1104
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001105ObjCCategoryDecl *ObjCCategoryDecl::Create(ASTContext &C, DeclContext *DC,
Douglas Gregor071676f2010-01-16 16:38:58 +00001106 SourceLocation AtLoc,
1107 SourceLocation ClassNameLoc,
1108 SourceLocation CategoryNameLoc,
Argyrios Kyrtzidis3a5094b2011-08-30 19:43:26 +00001109 IdentifierInfo *Id,
Fariborz Jahaniana7765fe2012-02-20 20:09:20 +00001110 ObjCInterfaceDecl *IDecl,
1111 SourceLocation IvarLBraceLoc,
1112 SourceLocation IvarRBraceLoc) {
Argyrios Kyrtzidis3a5094b2011-08-30 19:43:26 +00001113 ObjCCategoryDecl *CatDecl = new (C) ObjCCategoryDecl(DC, AtLoc, ClassNameLoc,
1114 CategoryNameLoc, Id,
Fariborz Jahaniana7765fe2012-02-20 20:09:20 +00001115 IDecl,
1116 IvarLBraceLoc, IvarRBraceLoc);
Argyrios Kyrtzidis3a5094b2011-08-30 19:43:26 +00001117 if (IDecl) {
1118 // Link this category into its class's category list.
1119 CatDecl->NextClassCategory = IDecl->getCategoryList();
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00001120 if (IDecl->hasDefinition()) {
1121 IDecl->setCategoryList(CatDecl);
1122 if (ASTMutationListener *L = C.getASTMutationListener())
1123 L->AddedObjCCategoryToInterface(CatDecl, IDecl);
1124 }
Argyrios Kyrtzidis3a5094b2011-08-30 19:43:26 +00001125 }
1126
1127 return CatDecl;
1128}
1129
Douglas Gregor72172e92012-01-05 21:55:30 +00001130ObjCCategoryDecl *ObjCCategoryDecl::CreateDeserialized(ASTContext &C,
1131 unsigned ID) {
1132 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCCategoryDecl));
1133 return new (Mem) ObjCCategoryDecl(0, SourceLocation(), SourceLocation(),
1134 SourceLocation(), 0, 0);
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001135}
1136
Argyrios Kyrtzidis6d9fab72009-07-21 00:05:53 +00001137ObjCCategoryImplDecl *ObjCCategoryDecl::getImplementation() const {
1138 return getASTContext().getObjCImplementation(
1139 const_cast<ObjCCategoryDecl*>(this));
1140}
1141
1142void ObjCCategoryDecl::setImplementation(ObjCCategoryImplDecl *ImplD) {
1143 getASTContext().setObjCImplementation(this, ImplD);
1144}
1145
1146
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001147//===----------------------------------------------------------------------===//
1148// ObjCCategoryImplDecl
1149//===----------------------------------------------------------------------===//
1150
David Blaikie68e081d2011-12-20 02:48:34 +00001151void ObjCCategoryImplDecl::anchor() { }
1152
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001153ObjCCategoryImplDecl *
1154ObjCCategoryImplDecl::Create(ASTContext &C, DeclContext *DC,
Argyrios Kyrtzidis52f53fb2011-10-04 04:48:02 +00001155 IdentifierInfo *Id,
1156 ObjCInterfaceDecl *ClassInterface,
1157 SourceLocation nameLoc,
Argyrios Kyrtzidis4996f5f2011-12-09 00:31:40 +00001158 SourceLocation atStartLoc,
1159 SourceLocation CategoryNameLoc) {
Fariborz Jahanian87b4ae6c2011-12-23 00:31:02 +00001160 if (ClassInterface && ClassInterface->hasDefinition())
1161 ClassInterface = ClassInterface->getDefinition();
Argyrios Kyrtzidis52f53fb2011-10-04 04:48:02 +00001162 return new (C) ObjCCategoryImplDecl(DC, Id, ClassInterface,
Argyrios Kyrtzidis4996f5f2011-12-09 00:31:40 +00001163 nameLoc, atStartLoc, CategoryNameLoc);
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001164}
1165
Douglas Gregor72172e92012-01-05 21:55:30 +00001166ObjCCategoryImplDecl *ObjCCategoryImplDecl::CreateDeserialized(ASTContext &C,
1167 unsigned ID) {
1168 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCCategoryImplDecl));
1169 return new (Mem) ObjCCategoryImplDecl(0, 0, 0, SourceLocation(),
1170 SourceLocation(), SourceLocation());
1171}
1172
Steve Narofff406f4d2009-10-29 21:11:04 +00001173ObjCCategoryDecl *ObjCCategoryImplDecl::getCategoryDecl() const {
Ted Kremeneke184ac52010-03-19 20:39:03 +00001174 // The class interface might be NULL if we are working with invalid code.
1175 if (const ObjCInterfaceDecl *ID = getClassInterface())
1176 return ID->FindCategoryDeclaration(getIdentifier());
1177 return 0;
Argyrios Kyrtzidisa56fa192009-07-28 05:11:05 +00001178}
1179
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001180
David Blaikie68e081d2011-12-20 02:48:34 +00001181void ObjCImplDecl::anchor() { }
1182
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00001183void ObjCImplDecl::addPropertyImplementation(ObjCPropertyImplDecl *property) {
Douglas Gregor9a13efd2009-04-23 02:42:49 +00001184 // FIXME: The context should be correct before we get here.
Douglas Gregor29bd76f2009-04-23 01:02:12 +00001185 property->setLexicalDeclContext(this);
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00001186 addDecl(property);
Douglas Gregor29bd76f2009-04-23 01:02:12 +00001187}
1188
Argyrios Kyrtzidis6d9fab72009-07-21 00:05:53 +00001189void ObjCImplDecl::setClassInterface(ObjCInterfaceDecl *IFace) {
1190 ASTContext &Ctx = getASTContext();
1191
1192 if (ObjCImplementationDecl *ImplD
Duncan Sands49c29ee2009-07-21 07:56:29 +00001193 = dyn_cast_or_null<ObjCImplementationDecl>(this)) {
Argyrios Kyrtzidis6d9fab72009-07-21 00:05:53 +00001194 if (IFace)
1195 Ctx.setObjCImplementation(IFace, ImplD);
1196
Duncan Sands49c29ee2009-07-21 07:56:29 +00001197 } else if (ObjCCategoryImplDecl *ImplD =
Argyrios Kyrtzidis6d9fab72009-07-21 00:05:53 +00001198 dyn_cast_or_null<ObjCCategoryImplDecl>(this)) {
1199 if (ObjCCategoryDecl *CD = IFace->FindCategoryDeclaration(getIdentifier()))
1200 Ctx.setObjCImplementation(CD, ImplD);
1201 }
1202
1203 ClassInterface = IFace;
1204}
1205
Fariborz Jahanianfbbaf6a2008-12-05 22:32:48 +00001206/// FindPropertyImplIvarDecl - This method lookup the ivar in the list of
James Dennett5207a1c2012-06-15 22:30:14 +00001207/// properties implemented in this category \@implementation block and returns
Chris Lattneraab70d22009-02-16 19:24:31 +00001208/// the implemented property that uses it.
Fariborz Jahanianfbbaf6a2008-12-05 22:32:48 +00001209///
Chris Lattnera9ca0522009-02-28 18:42:10 +00001210ObjCPropertyImplDecl *ObjCImplDecl::
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00001211FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const {
1212 for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i){
David Blaikie40ed2972012-06-06 20:45:41 +00001213 ObjCPropertyImplDecl *PID = *i;
Fariborz Jahanianfbbaf6a2008-12-05 22:32:48 +00001214 if (PID->getPropertyIvarDecl() &&
1215 PID->getPropertyIvarDecl()->getIdentifier() == ivarId)
1216 return PID;
1217 }
1218 return 0;
1219}
1220
1221/// FindPropertyImplDecl - This method looks up a previous ObjCPropertyImplDecl
James Dennett5207a1c2012-06-15 22:30:14 +00001222/// added to the list of those properties \@synthesized/\@dynamic in this
1223/// category \@implementation block.
Fariborz Jahanianfbbaf6a2008-12-05 22:32:48 +00001224///
Chris Lattnera9ca0522009-02-28 18:42:10 +00001225ObjCPropertyImplDecl *ObjCImplDecl::
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00001226FindPropertyImplDecl(IdentifierInfo *Id) const {
1227 for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i){
David Blaikie40ed2972012-06-06 20:45:41 +00001228 ObjCPropertyImplDecl *PID = *i;
Fariborz Jahanianfbbaf6a2008-12-05 22:32:48 +00001229 if (PID->getPropertyDecl()->getIdentifier() == Id)
1230 return PID;
1231 }
1232 return 0;
1233}
1234
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001235raw_ostream &clang::operator<<(raw_ostream &OS,
Benjamin Kramer2f569922012-02-07 11:57:45 +00001236 const ObjCCategoryImplDecl &CID) {
1237 OS << CID.getName();
Benjamin Kramerb11416d2010-04-17 09:33:03 +00001238 return OS;
1239}
1240
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001241//===----------------------------------------------------------------------===//
1242// ObjCImplementationDecl
1243//===----------------------------------------------------------------------===//
1244
David Blaikie68e081d2011-12-20 02:48:34 +00001245void ObjCImplementationDecl::anchor() { }
1246
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001247ObjCImplementationDecl *
Mike Stump11289f42009-09-09 15:08:12 +00001248ObjCImplementationDecl::Create(ASTContext &C, DeclContext *DC,
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001249 ObjCInterfaceDecl *ClassInterface,
Argyrios Kyrtzidis52f53fb2011-10-04 04:48:02 +00001250 ObjCInterfaceDecl *SuperDecl,
1251 SourceLocation nameLoc,
Fariborz Jahaniana7765fe2012-02-20 20:09:20 +00001252 SourceLocation atStartLoc,
1253 SourceLocation IvarLBraceLoc,
1254 SourceLocation IvarRBraceLoc) {
Fariborz Jahanian87b4ae6c2011-12-23 00:31:02 +00001255 if (ClassInterface && ClassInterface->hasDefinition())
1256 ClassInterface = ClassInterface->getDefinition();
Argyrios Kyrtzidis52f53fb2011-10-04 04:48:02 +00001257 return new (C) ObjCImplementationDecl(DC, ClassInterface, SuperDecl,
Fariborz Jahaniana7765fe2012-02-20 20:09:20 +00001258 nameLoc, atStartLoc,
1259 IvarLBraceLoc, IvarRBraceLoc);
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001260}
1261
Douglas Gregor72172e92012-01-05 21:55:30 +00001262ObjCImplementationDecl *
1263ObjCImplementationDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
1264 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCImplementationDecl));
1265 return new (Mem) ObjCImplementationDecl(0, 0, 0, SourceLocation(),
1266 SourceLocation());
1267}
1268
John McCall0410e572011-07-22 04:15:06 +00001269void ObjCImplementationDecl::setIvarInitializers(ASTContext &C,
1270 CXXCtorInitializer ** initializers,
1271 unsigned numInitializers) {
1272 if (numInitializers > 0) {
1273 NumIvarInitializers = numInitializers;
1274 CXXCtorInitializer **ivarInitializers =
1275 new (C) CXXCtorInitializer*[NumIvarInitializers];
1276 memcpy(ivarInitializers, initializers,
1277 numInitializers * sizeof(CXXCtorInitializer*));
1278 IvarInitializers = ivarInitializers;
1279 }
1280}
1281
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001282raw_ostream &clang::operator<<(raw_ostream &OS,
Benjamin Kramer2f569922012-02-07 11:57:45 +00001283 const ObjCImplementationDecl &ID) {
1284 OS << ID.getName();
Benjamin Kramerb11416d2010-04-17 09:33:03 +00001285 return OS;
1286}
1287
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001288//===----------------------------------------------------------------------===//
1289// ObjCCompatibleAliasDecl
1290//===----------------------------------------------------------------------===//
1291
David Blaikie68e081d2011-12-20 02:48:34 +00001292void ObjCCompatibleAliasDecl::anchor() { }
1293
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001294ObjCCompatibleAliasDecl *
1295ObjCCompatibleAliasDecl::Create(ASTContext &C, DeclContext *DC,
1296 SourceLocation L,
Mike Stump11289f42009-09-09 15:08:12 +00001297 IdentifierInfo *Id,
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001298 ObjCInterfaceDecl* AliasedClass) {
1299 return new (C) ObjCCompatibleAliasDecl(DC, L, Id, AliasedClass);
1300}
1301
Douglas Gregor72172e92012-01-05 21:55:30 +00001302ObjCCompatibleAliasDecl *
1303ObjCCompatibleAliasDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
1304 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCCompatibleAliasDecl));
1305 return new (Mem) ObjCCompatibleAliasDecl(0, SourceLocation(), 0, 0);
1306}
1307
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001308//===----------------------------------------------------------------------===//
1309// ObjCPropertyDecl
1310//===----------------------------------------------------------------------===//
1311
David Blaikie68e081d2011-12-20 02:48:34 +00001312void ObjCPropertyDecl::anchor() { }
1313
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001314ObjCPropertyDecl *ObjCPropertyDecl::Create(ASTContext &C, DeclContext *DC,
1315 SourceLocation L,
1316 IdentifierInfo *Id,
Fariborz Jahanianda8ec2b2010-01-21 17:36:00 +00001317 SourceLocation AtLoc,
Fariborz Jahanian86c2f5c2012-02-29 22:18:55 +00001318 SourceLocation LParenLoc,
John McCall339bb662010-06-04 20:50:08 +00001319 TypeSourceInfo *T,
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001320 PropertyControl propControl) {
Fariborz Jahanian86c2f5c2012-02-29 22:18:55 +00001321 return new (C) ObjCPropertyDecl(DC, L, Id, AtLoc, LParenLoc, T);
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001322}
1323
Douglas Gregor72172e92012-01-05 21:55:30 +00001324ObjCPropertyDecl *ObjCPropertyDecl::CreateDeserialized(ASTContext &C,
1325 unsigned ID) {
1326 void * Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCPropertyDecl));
1327 return new (Mem) ObjCPropertyDecl(0, SourceLocation(), 0, SourceLocation(),
Fariborz Jahanian86c2f5c2012-02-29 22:18:55 +00001328 SourceLocation(),
Douglas Gregor72172e92012-01-05 21:55:30 +00001329 0);
1330}
1331
Chris Lattnerf1ccb0c2009-02-20 20:59:54 +00001332//===----------------------------------------------------------------------===//
1333// ObjCPropertyImplDecl
1334//===----------------------------------------------------------------------===//
1335
Fariborz Jahanian6efdf1d2008-04-23 00:06:01 +00001336ObjCPropertyImplDecl *ObjCPropertyImplDecl::Create(ASTContext &C,
Douglas Gregorc25d7a72009-01-09 00:49:46 +00001337 DeclContext *DC,
Fariborz Jahanian6efdf1d2008-04-23 00:06:01 +00001338 SourceLocation atLoc,
1339 SourceLocation L,
1340 ObjCPropertyDecl *property,
Daniel Dunbar3b4fdb02008-08-26 04:47:31 +00001341 Kind PK,
Douglas Gregorb1b71e52010-11-17 01:03:52 +00001342 ObjCIvarDecl *ivar,
1343 SourceLocation ivarLoc) {
1344 return new (C) ObjCPropertyImplDecl(DC, atLoc, L, property, PK, ivar,
1345 ivarLoc);
Fariborz Jahanian6efdf1d2008-04-23 00:06:01 +00001346}
Chris Lattnered0e1642008-03-17 01:19:02 +00001347
Douglas Gregor72172e92012-01-05 21:55:30 +00001348ObjCPropertyImplDecl *ObjCPropertyImplDecl::CreateDeserialized(ASTContext &C,
1349 unsigned ID) {
1350 void *Mem = AllocateDeserializedDecl(C, ID, sizeof(ObjCPropertyImplDecl));
1351 return new (Mem) ObjCPropertyImplDecl(0, SourceLocation(), SourceLocation(),
1352 0, Dynamic, 0, SourceLocation());
1353}
1354
Douglas Gregorb1b71e52010-11-17 01:03:52 +00001355SourceRange ObjCPropertyImplDecl::getSourceRange() const {
1356 SourceLocation EndLoc = getLocation();
1357 if (IvarLoc.isValid())
1358 EndLoc = IvarLoc;
Chris Lattnerc5ffed42008-04-04 06:12:32 +00001359
Douglas Gregorb1b71e52010-11-17 01:03:52 +00001360 return SourceRange(AtLoc, EndLoc);
1361}