blob: 125c94194cbda861ef0d46fb74d043febd53c1b1 [file] [log] [blame]
Chris Lattner10318b82008-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 Dunbarde300732008-08-11 04:54:23 +000016#include "clang/AST/Stmt.h"
Steve Naroffd85ba922009-02-22 19:35:57 +000017#include "llvm/ADT/STLExtras.h"
Chris Lattner10318b82008-03-16 00:19:01 +000018using namespace clang;
19
Chris Lattner114add62008-03-16 00:49:28 +000020//===----------------------------------------------------------------------===//
Chris Lattner217eb8c2009-02-20 21:16:26 +000021// ObjCListBase
22//===----------------------------------------------------------------------===//
23
Chris Lattnerdf6133b2009-02-20 21:35:13 +000024void ObjCListBase::Destroy(ASTContext &Ctx) {
Chris Lattnercf6c8282009-02-20 21:44:01 +000025 Ctx.Deallocate(List);
Chris Lattner217eb8c2009-02-20 21:16:26 +000026 NumElts = 0;
27 List = 0;
28}
29
Chris Lattnerdf6133b2009-02-20 21:35:13 +000030void ObjCListBase::set(void *const* InList, unsigned Elts, ASTContext &Ctx) {
Chris Lattner217eb8c2009-02-20 21:16:26 +000031 assert(List == 0 && "Elements already set!");
32 if (Elts == 0) return; // Setting to an empty list is a noop.
33
Chris Lattnercf6c8282009-02-20 21:44:01 +000034
35 List = new (Ctx) void*[Elts];
Chris Lattner217eb8c2009-02-20 21:16:26 +000036 NumElts = Elts;
37 memcpy(List, InList, sizeof(void*)*Elts);
38}
39
40
41//===----------------------------------------------------------------------===//
Chris Lattner13f63602009-02-20 20:59:54 +000042// ObjCInterfaceDecl
Chris Lattner114add62008-03-16 00:49:28 +000043//===----------------------------------------------------------------------===//
44
Fariborz Jahanian074cf972009-06-05 18:16:35 +000045/// getIvarDecl - This method looks up an ivar in this ContextDecl.
46///
47ObjCIvarDecl *
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +000048ObjCContainerDecl::getIvarDecl(IdentifierInfo *Id) const {
Fariborz Jahanian074cf972009-06-05 18:16:35 +000049 lookup_const_iterator Ivar, IvarEnd;
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +000050 for (llvm::tie(Ivar, IvarEnd) = lookup(Id); Ivar != IvarEnd; ++Ivar) {
Fariborz Jahanian074cf972009-06-05 18:16:35 +000051 if (ObjCIvarDecl *ivar = dyn_cast<ObjCIvarDecl>(*Ivar))
52 return ivar;
53 }
54 return 0;
55}
56
Argiris Kirtzidis7a7757b2009-07-25 22:15:22 +000057// Get the local instance/class method declared in this interface.
Douglas Gregorc55b0b02009-04-09 21:40:53 +000058ObjCMethodDecl *
Argiris Kirtzidis7a7757b2009-07-25 22:15:22 +000059ObjCContainerDecl::getMethod(Selector Sel, bool isInstance) const {
Steve Naroffd85ba922009-02-22 19:35:57 +000060 // Since instance & class methods can have the same name, the loop below
61 // ensures we get the correct method.
62 //
63 // @interface Whatever
64 // - (int) class_method;
65 // + (float) class_method;
66 // @end
67 //
68 lookup_const_iterator Meth, MethEnd;
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +000069 for (llvm::tie(Meth, MethEnd) = lookup(Sel); Meth != MethEnd; ++Meth) {
Steve Naroffd85ba922009-02-22 19:35:57 +000070 ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(*Meth);
Argiris Kirtzidis7a7757b2009-07-25 22:15:22 +000071 if (MD && MD->isInstanceMethod() == isInstance)
Steve Naroffd85ba922009-02-22 19:35:57 +000072 return MD;
73 }
Steve Naroffab63fd62009-01-08 17:28:14 +000074 return 0;
75}
76
Fariborz Jahanianef8a3df2008-04-21 19:04:53 +000077/// FindPropertyDeclaration - Finds declaration of the property given its name
78/// in 'PropertyId' and returns it. It returns 0, if not found.
Steve Naroffdcf1e842009-01-11 12:47:58 +000079/// FIXME: Convert to DeclContext lookup...
Fariborz Jahanianef8a3df2008-04-21 19:04:53 +000080///
81ObjCPropertyDecl *
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +000082ObjCContainerDecl::FindPropertyDeclaration(IdentifierInfo *PropertyId) const {
83 for (prop_iterator I = prop_begin(), E = prop_end(); I != E; ++I)
Chris Lattner13f63602009-02-20 20:59:54 +000084 if ((*I)->getIdentifier() == PropertyId)
85 return *I;
86
Fariborz Jahanianfaca5e22009-01-09 21:04:52 +000087 const ObjCProtocolDecl *PID = dyn_cast<ObjCProtocolDecl>(this);
88 if (PID) {
Chris Lattner13f63602009-02-20 20:59:54 +000089 for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
90 E = PID->protocol_end(); I != E; ++I)
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +000091 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
Chris Lattner13f63602009-02-20 20:59:54 +000092 return P;
Fariborz Jahanianfaca5e22009-01-09 21:04:52 +000093 }
94
Fariborz Jahanian6b0ed6f2009-01-19 18:16:19 +000095 if (const ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(this)) {
Steve Naroff451f83c2009-01-09 15:36:25 +000096 // Look through categories.
97 for (ObjCCategoryDecl *Category = OID->getCategoryList();
98 Category; Category = Category->getNextClassCategory()) {
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +000099 if (ObjCPropertyDecl *P = Category->FindPropertyDeclaration(PropertyId))
Chris Lattner13f63602009-02-20 20:59:54 +0000100 return P;
Steve Naroff451f83c2009-01-09 15:36:25 +0000101 }
102 // Look through protocols.
103 for (ObjCInterfaceDecl::protocol_iterator I = OID->protocol_begin(),
104 E = OID->protocol_end(); I != E; ++I) {
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +0000105 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
Chris Lattner13f63602009-02-20 20:59:54 +0000106 return P;
Steve Naroff451f83c2009-01-09 15:36:25 +0000107 }
108 if (OID->getSuperClass())
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +0000109 return OID->getSuperClass()->FindPropertyDeclaration(PropertyId);
Chris Lattner13f63602009-02-20 20:59:54 +0000110 } else if (const ObjCCategoryDecl *OCD = dyn_cast<ObjCCategoryDecl>(this)) {
Fariborz Jahanian6b0ed6f2009-01-19 18:16:19 +0000111 // Look through protocols.
112 for (ObjCInterfaceDecl::protocol_iterator I = OCD->protocol_begin(),
113 E = OCD->protocol_end(); I != E; ++I) {
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +0000114 if (ObjCPropertyDecl *P = (*I)->FindPropertyDeclaration(PropertyId))
Chris Lattner13f63602009-02-20 20:59:54 +0000115 return P;
Fariborz Jahanian6b0ed6f2009-01-19 18:16:19 +0000116 }
117 }
Steve Naroffd1f0eb42008-06-05 13:55:23 +0000118 return 0;
119}
120
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +0000121ObjCIvarDecl *ObjCInterfaceDecl::lookupInstanceVariable(IdentifierInfo *ID,
122 ObjCInterfaceDecl *&clsDeclared) {
Chris Lattner10318b82008-03-16 00:19:01 +0000123 ObjCInterfaceDecl* ClassDecl = this;
124 while (ClassDecl != NULL) {
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +0000125 if (ObjCIvarDecl *I = ClassDecl->getIvarDecl(ID)) {
Fariborz Jahanian074cf972009-06-05 18:16:35 +0000126 clsDeclared = ClassDecl;
127 return I;
Chris Lattner10318b82008-03-16 00:19:01 +0000128 }
129 ClassDecl = ClassDecl->getSuperClass();
130 }
131 return NULL;
132}
133
Fariborz Jahanianc0672e42009-05-22 17:12:32 +0000134/// lookupInheritedClass - This method returns ObjCInterfaceDecl * of the super
135/// class whose name is passed as argument. If it is not one of the super classes
136/// the it returns NULL.
137ObjCInterfaceDecl *ObjCInterfaceDecl::lookupInheritedClass(
138 const IdentifierInfo*ICName) {
139 ObjCInterfaceDecl* ClassDecl = this;
140 while (ClassDecl != NULL) {
141 if (ClassDecl->getIdentifier() == ICName)
142 return ClassDecl;
143 ClassDecl = ClassDecl->getSuperClass();
144 }
145 return NULL;
146}
147
Argiris Kirtzidis9ac39f32009-07-25 22:15:51 +0000148/// lookupMethod - This method returns an instance/class method by looking in
Chris Lattner10318b82008-03-16 00:19:01 +0000149/// the class, its categories, and its super classes (using a linear search).
Argiris Kirtzidis9ac39f32009-07-25 22:15:51 +0000150ObjCMethodDecl *ObjCInterfaceDecl::lookupMethod(Selector Sel,
151 bool isInstance) const {
152 const ObjCInterfaceDecl* ClassDecl = this;
Chris Lattner10318b82008-03-16 00:19:01 +0000153 ObjCMethodDecl *MethodDecl = 0;
154
155 while (ClassDecl != NULL) {
Argiris Kirtzidis9ac39f32009-07-25 22:15:51 +0000156 if ((MethodDecl = ClassDecl->getMethod(Sel, isInstance)))
Chris Lattner10318b82008-03-16 00:19:01 +0000157 return MethodDecl;
158
159 // Didn't find one yet - look through protocols.
Chris Lattner8bcb5252008-07-21 18:19:38 +0000160 const ObjCList<ObjCProtocolDecl> &Protocols =
161 ClassDecl->getReferencedProtocols();
162 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
163 E = Protocols.end(); I != E; ++I)
Argiris Kirtzidis9ac39f32009-07-25 22:15:51 +0000164 if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
Chris Lattner10318b82008-03-16 00:19:01 +0000165 return MethodDecl;
Chris Lattner8bcb5252008-07-21 18:19:38 +0000166
Chris Lattner10318b82008-03-16 00:19:01 +0000167 // Didn't find one yet - now look through categories.
168 ObjCCategoryDecl *CatDecl = ClassDecl->getCategoryList();
169 while (CatDecl) {
Argiris Kirtzidis9ac39f32009-07-25 22:15:51 +0000170 if ((MethodDecl = CatDecl->getMethod(Sel, isInstance)))
Chris Lattner10318b82008-03-16 00:19:01 +0000171 return MethodDecl;
Steve Naroffaf151802008-12-08 20:57:28 +0000172
173 // Didn't find one yet - look through protocols.
174 const ObjCList<ObjCProtocolDecl> &Protocols =
175 CatDecl->getReferencedProtocols();
176 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
177 E = Protocols.end(); I != E; ++I)
Argiris Kirtzidis9ac39f32009-07-25 22:15:51 +0000178 if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
Steve Naroff0ba24842009-02-26 11:32:02 +0000179 return MethodDecl;
Chris Lattner10318b82008-03-16 00:19:01 +0000180 CatDecl = CatDecl->getNextClassCategory();
181 }
182 ClassDecl = ClassDecl->getSuperClass();
183 }
184 return NULL;
185}
186
Chris Lattner13f63602009-02-20 20:59:54 +0000187
188
189//===----------------------------------------------------------------------===//
190// ObjCMethodDecl
191//===----------------------------------------------------------------------===//
192
193ObjCMethodDecl *ObjCMethodDecl::Create(ASTContext &C,
194 SourceLocation beginLoc,
195 SourceLocation endLoc,
196 Selector SelInfo, QualType T,
197 DeclContext *contextDecl,
198 bool isInstance,
199 bool isVariadic,
200 bool isSynthesized,
201 ImplementationControl impControl) {
202 return new (C) ObjCMethodDecl(beginLoc, endLoc,
203 SelInfo, T, contextDecl,
204 isInstance,
205 isVariadic, isSynthesized, impControl);
Chris Lattner10318b82008-03-16 00:19:01 +0000206}
207
Chris Lattnerdf6133b2009-02-20 21:35:13 +0000208void ObjCMethodDecl::Destroy(ASTContext &C) {
Chris Lattner13f63602009-02-20 20:59:54 +0000209 if (Body) Body->Destroy(C);
210 if (SelfDecl) SelfDecl->Destroy(C);
211
212 for (param_iterator I=param_begin(), E=param_end(); I!=E; ++I)
213 if (*I) (*I)->Destroy(C);
214
Chris Lattnerdf6133b2009-02-20 21:35:13 +0000215 ParamInfo.Destroy(C);
Chris Lattner13f63602009-02-20 20:59:54 +0000216
217 Decl::Destroy(C);
Chris Lattner10318b82008-03-16 00:19:01 +0000218}
219
Argiris Kirtzidis29c33872009-07-21 00:06:36 +0000220/// \brief A definition will return its interface declaration.
221/// An interface declaration will return its definition.
222/// Otherwise it will return itself.
223ObjCMethodDecl *ObjCMethodDecl::getNextRedeclaration() {
224 ASTContext &Ctx = getASTContext();
225 ObjCMethodDecl *Redecl = 0;
226 Decl *CtxD = cast<Decl>(getDeclContext());
227
228 if (ObjCInterfaceDecl *IFD = dyn_cast<ObjCInterfaceDecl>(CtxD)) {
229 if (ObjCImplementationDecl *ImplD = Ctx.getObjCImplementation(IFD))
230 Redecl = ImplD->getMethod(getSelector(), isInstanceMethod());
231
232 } else if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(CtxD)) {
233 if (ObjCCategoryImplDecl *ImplD = Ctx.getObjCImplementation(CD))
234 Redecl = ImplD->getMethod(getSelector(), isInstanceMethod());
235
Argiris Kirtzidis0f3b8422009-07-28 05:11:05 +0000236 } else if (ObjCImplementationDecl *ImplD =
237 dyn_cast<ObjCImplementationDecl>(CtxD)) {
Argiris Kirtzidis29c33872009-07-21 00:06:36 +0000238 if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
239 Redecl = IFD->getMethod(getSelector(), isInstanceMethod());
Argiris Kirtzidis0f3b8422009-07-28 05:11:05 +0000240
241 } else if (ObjCCategoryImplDecl *CImplD =
242 dyn_cast<ObjCCategoryImplDecl>(CtxD)) {
243 if (ObjCCategoryDecl *CatD = CImplD->getCategoryClass())
244 Redecl = CatD->getMethod(getSelector(), isInstanceMethod());
Argiris Kirtzidis29c33872009-07-21 00:06:36 +0000245 }
246
247 return Redecl ? Redecl : this;
248}
249
Argiris Kirtzidis029f7872009-07-28 05:11:17 +0000250ObjCMethodDecl *ObjCMethodDecl::getCanonicalDecl() {
251 Decl *CtxD = cast<Decl>(getDeclContext());
252
253 if (ObjCImplementationDecl *ImplD = dyn_cast<ObjCImplementationDecl>(CtxD)) {
254 if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
255 if (ObjCMethodDecl *MD = IFD->getMethod(getSelector(),
256 isInstanceMethod()))
257 return MD;
258
259 } else if (ObjCCategoryImplDecl *CImplD =
260 dyn_cast<ObjCCategoryImplDecl>(CtxD)) {
261 if (ObjCCategoryDecl *CatD = CImplD->getCategoryClass())
262 if (ObjCMethodDecl *MD = CatD->getMethod(getSelector(),
263 isInstanceMethod()))
264 return MD;
265 }
266
267 return this;
268}
269
Chris Lattner13f63602009-02-20 20:59:54 +0000270void ObjCMethodDecl::createImplicitParams(ASTContext &Context,
271 const ObjCInterfaceDecl *OID) {
272 QualType selfTy;
273 if (isInstanceMethod()) {
274 // There may be no interface context due to error in declaration
275 // of the interface (which has been reported). Recover gracefully.
276 if (OID) {
Daniel Dunbarbe1ff272009-04-22 04:34:53 +0000277 selfTy = Context.getObjCInterfaceType(OID);
Steve Naroff329ec222009-07-10 23:34:53 +0000278 selfTy = Context.getObjCObjectPointerType(selfTy);
Chris Lattner13f63602009-02-20 20:59:54 +0000279 } else {
280 selfTy = Context.getObjCIdType();
281 }
282 } else // we have a factory method.
283 selfTy = Context.getObjCClassType();
284
Steve Naroff79ea0e02009-04-20 15:06:07 +0000285 setSelfDecl(ImplicitParamDecl::Create(Context, this, SourceLocation(),
286 &Context.Idents.get("self"), selfTy));
Chris Lattner13f63602009-02-20 20:59:54 +0000287
Steve Naroff79ea0e02009-04-20 15:06:07 +0000288 setCmdDecl(ImplicitParamDecl::Create(Context, this, SourceLocation(),
289 &Context.Idents.get("_cmd"),
290 Context.getObjCSelType()));
Chris Lattner13f63602009-02-20 20:59:54 +0000291}
292
293
294
295/// getSynthesizedMethodSize - Compute size of synthesized method name
296/// as done be the rewrite.
Fariborz Jahanian68342282008-12-05 22:32:48 +0000297///
Chris Lattner13f63602009-02-20 20:59:54 +0000298unsigned ObjCMethodDecl::getSynthesizedMethodSize() const {
299 // syntesized method name is a concatenation of -/+[class-name selector]
300 // Get length of this name.
301 unsigned length = 3; // _I_ or _C_
302 length += getClassInterface()->getNameAsString().size()+1; // extra for _
303 if (const ObjCCategoryImplDecl *CID =
304 dyn_cast<ObjCCategoryImplDecl>(getDeclContext()))
305 length += CID->getNameAsString().size()+1;
306 length += getSelector().getAsString().size(); // selector name
307 return length;
308}
309
310ObjCInterfaceDecl *ObjCMethodDecl::getClassInterface() {
311 if (ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(getDeclContext()))
312 return ID;
313 if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(getDeclContext()))
314 return CD->getClassInterface();
Argiris Kirtzidisa06e3ba2009-07-28 05:10:52 +0000315 if (ObjCImplDecl *IMD = dyn_cast<ObjCImplDecl>(getDeclContext()))
Chris Lattner13f63602009-02-20 20:59:54 +0000316 return IMD->getClassInterface();
Argiris Kirtzidisa06e3ba2009-07-28 05:10:52 +0000317
318 assert(!isa<ObjCProtocolDecl>(getDeclContext()) && "It's a protocol method");
Chris Lattner13f63602009-02-20 20:59:54 +0000319 assert(false && "unknown method context");
Fariborz Jahanian68342282008-12-05 22:32:48 +0000320 return 0;
321}
322
Chris Lattner13f63602009-02-20 20:59:54 +0000323//===----------------------------------------------------------------------===//
324// ObjCInterfaceDecl
325//===----------------------------------------------------------------------===//
326
327ObjCInterfaceDecl *ObjCInterfaceDecl::Create(ASTContext &C,
328 DeclContext *DC,
329 SourceLocation atLoc,
330 IdentifierInfo *Id,
331 SourceLocation ClassLoc,
332 bool ForwardDecl, bool isInternal){
333 return new (C) ObjCInterfaceDecl(DC, atLoc, Id, ClassLoc, ForwardDecl,
334 isInternal);
335}
336
337ObjCInterfaceDecl::
338ObjCInterfaceDecl(DeclContext *DC, SourceLocation atLoc, IdentifierInfo *Id,
339 SourceLocation CLoc, bool FD, bool isInternal)
340 : ObjCContainerDecl(ObjCInterface, DC, atLoc, Id),
341 TypeForDecl(0), SuperClass(0),
342 CategoryList(0), ForwardDecl(FD), InternalInterface(isInternal),
343 ClassLoc(CLoc) {
344}
345
346void ObjCInterfaceDecl::Destroy(ASTContext &C) {
Chris Lattnereff36ed2009-03-31 08:36:08 +0000347 for (ivar_iterator I = ivar_begin(), E = ivar_end(); I != E; ++I)
Chris Lattner13f63602009-02-20 20:59:54 +0000348 if (*I) (*I)->Destroy(C);
349
Chris Lattnerdf6133b2009-02-20 21:35:13 +0000350 IVars.Destroy(C);
Chris Lattner13f63602009-02-20 20:59:54 +0000351 // FIXME: CategoryList?
352
353 // FIXME: Because there is no clear ownership
354 // role between ObjCInterfaceDecls and the ObjCPropertyDecls that they
355 // reference, we destroy ObjCPropertyDecls in ~TranslationUnit.
356 Decl::Destroy(C);
357}
358
Argiris Kirtzidis3a4d9832009-07-21 00:05:53 +0000359ObjCImplementationDecl *ObjCInterfaceDecl::getImplementation() const {
360 return getASTContext().getObjCImplementation(
361 const_cast<ObjCInterfaceDecl*>(this));
362}
363
364void ObjCInterfaceDecl::setImplementation(ObjCImplementationDecl *ImplD) {
365 getASTContext().setObjCImplementation(this, ImplD);
366}
367
Chris Lattner13f63602009-02-20 20:59:54 +0000368
369/// FindCategoryDeclaration - Finds category declaration in the list of
370/// categories for this class and returns it. Name of the category is passed
371/// in 'CategoryId'. If category not found, return 0;
Fariborz Jahanian68342282008-12-05 22:32:48 +0000372///
Chris Lattner13f63602009-02-20 20:59:54 +0000373ObjCCategoryDecl *
374ObjCInterfaceDecl::FindCategoryDeclaration(IdentifierInfo *CategoryId) const {
375 for (ObjCCategoryDecl *Category = getCategoryList();
376 Category; Category = Category->getNextClassCategory())
377 if (Category->getIdentifier() == CategoryId)
378 return Category;
Fariborz Jahanian68342282008-12-05 22:32:48 +0000379 return 0;
380}
381
Argiris Kirtzidis20096862009-07-21 00:06:20 +0000382ObjCMethodDecl *
383ObjCInterfaceDecl::getCategoryInstanceMethod(Selector Sel) const {
384 for (ObjCCategoryDecl *Category = getCategoryList();
385 Category; Category = Category->getNextClassCategory())
386 if (ObjCCategoryImplDecl *Impl = Category->getImplementation())
387 if (ObjCMethodDecl *MD = Impl->getInstanceMethod(Sel))
388 return MD;
389 return 0;
390}
391
392ObjCMethodDecl *ObjCInterfaceDecl::getCategoryClassMethod(Selector Sel) const {
393 for (ObjCCategoryDecl *Category = getCategoryList();
394 Category; Category = Category->getNextClassCategory())
395 if (ObjCCategoryImplDecl *Impl = Category->getImplementation())
396 if (ObjCMethodDecl *MD = Impl->getClassMethod(Sel))
397 return MD;
398 return 0;
399}
400
Chris Lattner13f63602009-02-20 20:59:54 +0000401//===----------------------------------------------------------------------===//
402// ObjCIvarDecl
403//===----------------------------------------------------------------------===//
404
405ObjCIvarDecl *ObjCIvarDecl::Create(ASTContext &C, DeclContext *DC,
406 SourceLocation L, IdentifierInfo *Id,
407 QualType T, AccessControl ac, Expr *BW) {
408 return new (C) ObjCIvarDecl(DC, L, Id, T, ac, BW);
409}
410
411
412
413//===----------------------------------------------------------------------===//
414// ObjCAtDefsFieldDecl
415//===----------------------------------------------------------------------===//
416
417ObjCAtDefsFieldDecl
418*ObjCAtDefsFieldDecl::Create(ASTContext &C, DeclContext *DC, SourceLocation L,
419 IdentifierInfo *Id, QualType T, Expr *BW) {
420 return new (C) ObjCAtDefsFieldDecl(DC, L, Id, T, BW);
421}
422
423void ObjCAtDefsFieldDecl::Destroy(ASTContext& C) {
424 this->~ObjCAtDefsFieldDecl();
425 C.Deallocate((void *)this);
426}
427
428//===----------------------------------------------------------------------===//
429// ObjCProtocolDecl
430//===----------------------------------------------------------------------===//
431
432ObjCProtocolDecl *ObjCProtocolDecl::Create(ASTContext &C, DeclContext *DC,
433 SourceLocation L,
434 IdentifierInfo *Id) {
435 return new (C) ObjCProtocolDecl(DC, L, Id);
436}
437
438void ObjCProtocolDecl::Destroy(ASTContext &C) {
Chris Lattnerdf6133b2009-02-20 21:35:13 +0000439 ReferencedProtocols.Destroy(C);
Chris Lattner13f63602009-02-20 20:59:54 +0000440 ObjCContainerDecl::Destroy(C);
441}
442
Steve Naroff98e71b82009-03-01 16:12:44 +0000443ObjCProtocolDecl *ObjCProtocolDecl::lookupProtocolNamed(IdentifierInfo *Name) {
444 ObjCProtocolDecl *PDecl = this;
445
446 if (Name == getIdentifier())
447 return PDecl;
448
449 for (protocol_iterator I = protocol_begin(), E = protocol_end(); I != E; ++I)
450 if ((PDecl = (*I)->lookupProtocolNamed(Name)))
451 return PDecl;
452
453 return NULL;
454}
455
Argiris Kirtzidisaba400c2009-07-25 22:15:38 +0000456// lookupMethod - Lookup a instance/class method in the protocol and protocols
Chris Lattner13f63602009-02-20 20:59:54 +0000457// it inherited.
Argiris Kirtzidisaba400c2009-07-25 22:15:38 +0000458ObjCMethodDecl *ObjCProtocolDecl::lookupMethod(Selector Sel,
459 bool isInstance) const {
Chris Lattner13f63602009-02-20 20:59:54 +0000460 ObjCMethodDecl *MethodDecl = NULL;
461
Argiris Kirtzidisaba400c2009-07-25 22:15:38 +0000462 if ((MethodDecl = getMethod(Sel, isInstance)))
Chris Lattner13f63602009-02-20 20:59:54 +0000463 return MethodDecl;
464
465 for (protocol_iterator I = protocol_begin(), E = protocol_end(); I != E; ++I)
Argiris Kirtzidisaba400c2009-07-25 22:15:38 +0000466 if ((MethodDecl = (*I)->lookupMethod(Sel, isInstance)))
Chris Lattner13f63602009-02-20 20:59:54 +0000467 return MethodDecl;
468 return NULL;
469}
470
471//===----------------------------------------------------------------------===//
472// ObjCClassDecl
473//===----------------------------------------------------------------------===//
474
Chris Lattnerdf6133b2009-02-20 21:35:13 +0000475ObjCClassDecl::ObjCClassDecl(DeclContext *DC, SourceLocation L,
476 ObjCInterfaceDecl *const *Elts, unsigned nElts,
477 ASTContext &C)
478 : Decl(ObjCClass, DC, L) {
479 ForwardDecls.set(Elts, nElts, C);
480}
481
482
Chris Lattner13f63602009-02-20 20:59:54 +0000483ObjCClassDecl *ObjCClassDecl::Create(ASTContext &C, DeclContext *DC,
484 SourceLocation L,
485 ObjCInterfaceDecl *const *Elts,
486 unsigned nElts) {
Chris Lattnerdf6133b2009-02-20 21:35:13 +0000487 return new (C) ObjCClassDecl(DC, L, Elts, nElts, C);
Chris Lattner13f63602009-02-20 20:59:54 +0000488}
489
490void ObjCClassDecl::Destroy(ASTContext &C) {
491
492 // FIXME: There is no clear ownership policy now for referenced
493 // ObjCInterfaceDecls. Some of them can be forward declarations that
494 // are never later defined (in which case the ObjCClassDecl owns them)
495 // or the ObjCInterfaceDecl later becomes a real definition later. Ideally
496 // we should have separate objects for forward declarations and definitions,
497 // obviating this problem. Because of this situation, referenced
498 // ObjCInterfaceDecls are destroyed in ~TranslationUnit.
499
Chris Lattnerdf6133b2009-02-20 21:35:13 +0000500 ForwardDecls.Destroy(C);
Chris Lattner13f63602009-02-20 20:59:54 +0000501 Decl::Destroy(C);
502}
503
504//===----------------------------------------------------------------------===//
505// ObjCForwardProtocolDecl
506//===----------------------------------------------------------------------===//
507
Chris Lattnerdf6133b2009-02-20 21:35:13 +0000508ObjCForwardProtocolDecl::
509ObjCForwardProtocolDecl(DeclContext *DC, SourceLocation L,
510 ObjCProtocolDecl *const *Elts, unsigned nElts,
511 ASTContext &C)
512: Decl(ObjCForwardProtocol, DC, L) {
513 ReferencedProtocols.set(Elts, nElts, C);
514}
515
516
Chris Lattner13f63602009-02-20 20:59:54 +0000517ObjCForwardProtocolDecl *
518ObjCForwardProtocolDecl::Create(ASTContext &C, DeclContext *DC,
519 SourceLocation L,
520 ObjCProtocolDecl *const *Elts,
521 unsigned NumElts) {
Chris Lattnerdf6133b2009-02-20 21:35:13 +0000522 return new (C) ObjCForwardProtocolDecl(DC, L, Elts, NumElts, C);
Chris Lattner13f63602009-02-20 20:59:54 +0000523}
524
525void ObjCForwardProtocolDecl::Destroy(ASTContext &C) {
Chris Lattnerdf6133b2009-02-20 21:35:13 +0000526 ReferencedProtocols.Destroy(C);
Chris Lattner13f63602009-02-20 20:59:54 +0000527 Decl::Destroy(C);
528}
529
530//===----------------------------------------------------------------------===//
531// ObjCCategoryDecl
532//===----------------------------------------------------------------------===//
533
534ObjCCategoryDecl *ObjCCategoryDecl::Create(ASTContext &C, DeclContext *DC,
535 SourceLocation L,
536 IdentifierInfo *Id) {
537 return new (C) ObjCCategoryDecl(DC, L, Id);
538}
539
Argiris Kirtzidis3a4d9832009-07-21 00:05:53 +0000540ObjCCategoryImplDecl *ObjCCategoryDecl::getImplementation() const {
541 return getASTContext().getObjCImplementation(
542 const_cast<ObjCCategoryDecl*>(this));
543}
544
545void ObjCCategoryDecl::setImplementation(ObjCCategoryImplDecl *ImplD) {
546 getASTContext().setObjCImplementation(this, ImplD);
547}
548
549
Chris Lattner13f63602009-02-20 20:59:54 +0000550//===----------------------------------------------------------------------===//
551// ObjCCategoryImplDecl
552//===----------------------------------------------------------------------===//
553
554ObjCCategoryImplDecl *
555ObjCCategoryImplDecl::Create(ASTContext &C, DeclContext *DC,
556 SourceLocation L,IdentifierInfo *Id,
557 ObjCInterfaceDecl *ClassInterface) {
558 return new (C) ObjCCategoryImplDecl(DC, L, Id, ClassInterface);
559}
560
Argiris Kirtzidis0f3b8422009-07-28 05:11:05 +0000561ObjCCategoryDecl *ObjCCategoryImplDecl::getCategoryClass() const {
562 return getClassInterface()->FindCategoryDeclaration(getIdentifier());
563}
564
Chris Lattner13f63602009-02-20 20:59:54 +0000565
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +0000566void ObjCImplDecl::addPropertyImplementation(ObjCPropertyImplDecl *property) {
Douglas Gregorbd336c52009-04-23 02:42:49 +0000567 // FIXME: The context should be correct before we get here.
Douglas Gregorcd19b572009-04-23 01:02:12 +0000568 property->setLexicalDeclContext(this);
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +0000569 addDecl(property);
Douglas Gregorcd19b572009-04-23 01:02:12 +0000570}
571
Argiris Kirtzidis3a4d9832009-07-21 00:05:53 +0000572void ObjCImplDecl::setClassInterface(ObjCInterfaceDecl *IFace) {
573 ASTContext &Ctx = getASTContext();
574
575 if (ObjCImplementationDecl *ImplD
Duncan Sands57f4dfe2009-07-21 07:56:29 +0000576 = dyn_cast_or_null<ObjCImplementationDecl>(this)) {
Argiris Kirtzidis3a4d9832009-07-21 00:05:53 +0000577 if (IFace)
578 Ctx.setObjCImplementation(IFace, ImplD);
579
Duncan Sands57f4dfe2009-07-21 07:56:29 +0000580 } else if (ObjCCategoryImplDecl *ImplD =
Argiris Kirtzidis3a4d9832009-07-21 00:05:53 +0000581 dyn_cast_or_null<ObjCCategoryImplDecl>(this)) {
582 if (ObjCCategoryDecl *CD = IFace->FindCategoryDeclaration(getIdentifier()))
583 Ctx.setObjCImplementation(CD, ImplD);
584 }
585
586 ClassInterface = IFace;
587}
588
Fariborz Jahanian68342282008-12-05 22:32:48 +0000589/// FindPropertyImplIvarDecl - This method lookup the ivar in the list of
Chris Lattner715e8482009-02-16 19:24:31 +0000590/// properties implemented in this category @implementation block and returns
591/// the implemented property that uses it.
Fariborz Jahanian68342282008-12-05 22:32:48 +0000592///
Chris Lattner649b3da2009-02-28 18:42:10 +0000593ObjCPropertyImplDecl *ObjCImplDecl::
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +0000594FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const {
595 for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i){
Fariborz Jahanian68342282008-12-05 22:32:48 +0000596 ObjCPropertyImplDecl *PID = *i;
597 if (PID->getPropertyIvarDecl() &&
598 PID->getPropertyIvarDecl()->getIdentifier() == ivarId)
599 return PID;
600 }
601 return 0;
602}
603
604/// FindPropertyImplDecl - This method looks up a previous ObjCPropertyImplDecl
605/// added to the list of those properties @synthesized/@dynamic in this
606/// category @implementation block.
607///
Chris Lattner649b3da2009-02-28 18:42:10 +0000608ObjCPropertyImplDecl *ObjCImplDecl::
Argiris Kirtzidisab6e38a2009-06-30 02:36:12 +0000609FindPropertyImplDecl(IdentifierInfo *Id) const {
610 for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i){
Fariborz Jahanian68342282008-12-05 22:32:48 +0000611 ObjCPropertyImplDecl *PID = *i;
612 if (PID->getPropertyDecl()->getIdentifier() == Id)
613 return PID;
614 }
615 return 0;
616}
617
Chris Lattner13f63602009-02-20 20:59:54 +0000618//===----------------------------------------------------------------------===//
619// ObjCImplementationDecl
620//===----------------------------------------------------------------------===//
621
622ObjCImplementationDecl *
623ObjCImplementationDecl::Create(ASTContext &C, DeclContext *DC,
624 SourceLocation L,
625 ObjCInterfaceDecl *ClassInterface,
626 ObjCInterfaceDecl *SuperDecl) {
627 return new (C) ObjCImplementationDecl(DC, L, ClassInterface, SuperDecl);
628}
629
Chris Lattner13f63602009-02-20 20:59:54 +0000630//===----------------------------------------------------------------------===//
631// ObjCCompatibleAliasDecl
632//===----------------------------------------------------------------------===//
633
634ObjCCompatibleAliasDecl *
635ObjCCompatibleAliasDecl::Create(ASTContext &C, DeclContext *DC,
636 SourceLocation L,
637 IdentifierInfo *Id,
638 ObjCInterfaceDecl* AliasedClass) {
639 return new (C) ObjCCompatibleAliasDecl(DC, L, Id, AliasedClass);
640}
641
642//===----------------------------------------------------------------------===//
643// ObjCPropertyDecl
644//===----------------------------------------------------------------------===//
645
646ObjCPropertyDecl *ObjCPropertyDecl::Create(ASTContext &C, DeclContext *DC,
647 SourceLocation L,
648 IdentifierInfo *Id,
649 QualType T,
650 PropertyControl propControl) {
651 return new (C) ObjCPropertyDecl(DC, L, Id, T);
652}
653
654
655//===----------------------------------------------------------------------===//
656// ObjCPropertyImplDecl
657//===----------------------------------------------------------------------===//
658
Fariborz Jahaniandc0569e2008-04-23 00:06:01 +0000659ObjCPropertyImplDecl *ObjCPropertyImplDecl::Create(ASTContext &C,
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +0000660 DeclContext *DC,
Fariborz Jahaniandc0569e2008-04-23 00:06:01 +0000661 SourceLocation atLoc,
662 SourceLocation L,
663 ObjCPropertyDecl *property,
Daniel Dunbar14117fc2008-08-26 04:47:31 +0000664 Kind PK,
Fariborz Jahaniandc0569e2008-04-23 00:06:01 +0000665 ObjCIvarDecl *ivar) {
Steve Naroff5abb0282009-01-27 21:25:57 +0000666 return new (C) ObjCPropertyImplDecl(DC, atLoc, L, property, PK, ivar);
Fariborz Jahaniandc0569e2008-04-23 00:06:01 +0000667}
Chris Lattner44859612008-03-17 01:19:02 +0000668
Chris Lattnereee57c02008-04-04 06:12:32 +0000669