blob: b0d36dbab8db6fab0af21db614f6666f00642b19 [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"
Chris Lattner10318b82008-03-16 00:19:01 +000017using namespace clang;
18
Chris Lattner114add62008-03-16 00:49:28 +000019//===----------------------------------------------------------------------===//
20// ObjC Decl Allocation/Deallocation Method Implementations
21//===----------------------------------------------------------------------===//
22
Chris Lattnereee57c02008-04-04 06:12:32 +000023ObjCMethodDecl *ObjCMethodDecl::Create(ASTContext &C,
24 SourceLocation beginLoc,
Chris Lattner114add62008-03-16 00:49:28 +000025 SourceLocation endLoc,
26 Selector SelInfo, QualType T,
Steve Naroffab63fd62009-01-08 17:28:14 +000027 DeclContext *contextDecl,
Daniel Dunbard8bd6822008-08-20 18:02:42 +000028 bool isInstance,
Chris Lattner114add62008-03-16 00:49:28 +000029 bool isVariadic,
Fariborz Jahanian5f2e2242008-05-07 20:53:44 +000030 bool isSynthesized,
Chris Lattnerf7355832008-03-16 00:58:16 +000031 ImplementationControl impControl) {
Steve Naroff5abb0282009-01-27 21:25:57 +000032 return new (C) ObjCMethodDecl(beginLoc, endLoc,
Chris Lattnereee57c02008-04-04 06:12:32 +000033 SelInfo, T, contextDecl,
Daniel Dunbard8bd6822008-08-20 18:02:42 +000034 isInstance,
Fariborz Jahanian5f2e2242008-05-07 20:53:44 +000035 isVariadic, isSynthesized, impControl);
Chris Lattner0db541b2008-03-16 01:15:50 +000036}
Chris Lattner114add62008-03-16 00:49:28 +000037
Ted Kremenek8c945b12008-06-06 16:45:15 +000038void ObjCMethodDecl::Destroy(ASTContext& C) {
39 if (Body) Body->Destroy(C);
40 if (SelfDecl) SelfDecl->Destroy(C);
41
42 for (param_iterator I=param_begin(), E=param_end(); I!=E; ++I)
43 if (*I) (*I)->Destroy(C);
Chris Lattnerae538882009-02-20 05:54:35 +000044
Chris Lattner9408eb12009-02-20 06:23:21 +000045 ParamInfo.clear();
Chris Lattnerae538882009-02-20 05:54:35 +000046
Ted Kremenek8c945b12008-06-06 16:45:15 +000047 Decl::Destroy(C);
48}
49
Chris Lattner3e27e632009-02-20 06:03:09 +000050
Chris Lattnereee57c02008-04-04 06:12:32 +000051ObjCInterfaceDecl *ObjCInterfaceDecl::Create(ASTContext &C,
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +000052 DeclContext *DC,
Chris Lattnereee57c02008-04-04 06:12:32 +000053 SourceLocation atLoc,
Steve Naroff7c371742008-04-11 19:35:35 +000054 IdentifierInfo *Id,
55 SourceLocation ClassLoc,
Chris Lattner0db541b2008-03-16 01:15:50 +000056 bool ForwardDecl, bool isInternal){
Steve Naroff5abb0282009-01-27 21:25:57 +000057 return new (C) ObjCInterfaceDecl(DC, atLoc, Id, ClassLoc, ForwardDecl,
Chris Lattner0db541b2008-03-16 01:15:50 +000058 isInternal);
59}
60
Chris Lattner3e27e632009-02-20 06:03:09 +000061ObjCInterfaceDecl::
62ObjCInterfaceDecl(DeclContext *DC, SourceLocation atLoc, IdentifierInfo *Id,
63 SourceLocation CLoc, bool FD, bool isInternal)
64 : ObjCContainerDecl(ObjCInterface, DC, atLoc, Id),
Chris Lattner922132e2009-02-20 06:10:45 +000065 TypeForDecl(0), SuperClass(0),
Chris Lattner3e27e632009-02-20 06:03:09 +000066 CategoryList(0), ForwardDecl(FD), InternalInterface(isInternal),
67 ClassLoc(CLoc) {
Steve Naroffd9d4d502009-01-07 17:57:40 +000068}
69
Chris Lattner3e27e632009-02-20 06:03:09 +000070void ObjCInterfaceDecl::Destroy(ASTContext &C) {
Ted Kremenek8c945b12008-06-06 16:45:15 +000071 for (ivar_iterator I=ivar_begin(), E=ivar_end(); I!=E; ++I)
72 if (*I) (*I)->Destroy(C);
73
Chris Lattner922132e2009-02-20 06:10:45 +000074 IVars.clear();
Chris Lattner3e27e632009-02-20 06:03:09 +000075 // FIXME: CategoryList?
76
Ted Kremenek22db1602008-06-06 17:21:42 +000077 // FIXME: Because there is no clear ownership
78 // role between ObjCInterfaceDecls and the ObjCPropertyDecls that they
79 // reference, we destroy ObjCPropertyDecls in ~TranslationUnit.
Ted Kremenek8c945b12008-06-06 16:45:15 +000080 Decl::Destroy(C);
81}
82
83
Argiris Kirtzidis96e79bf2009-02-17 20:20:37 +000084ObjCIvarDecl *ObjCIvarDecl::Create(ASTContext &C, DeclContext *DC,
85 SourceLocation L, IdentifierInfo *Id,
86 QualType T, AccessControl ac, Expr *BW) {
87 return new (C) ObjCIvarDecl(DC, L, Id, T, ac, BW);
Chris Lattner114add62008-03-16 00:49:28 +000088}
89
Ted Kremeneke5bedfe2008-08-20 03:26:33 +000090
91ObjCAtDefsFieldDecl
Douglas Gregor8acb7272008-12-11 16:49:14 +000092*ObjCAtDefsFieldDecl::Create(ASTContext &C, DeclContext *DC, SourceLocation L,
Ted Kremeneke5bedfe2008-08-20 03:26:33 +000093 IdentifierInfo *Id, QualType T, Expr *BW) {
Steve Naroff5abb0282009-01-27 21:25:57 +000094 return new (C) ObjCAtDefsFieldDecl(DC, L, Id, T, BW);
Ted Kremeneke5bedfe2008-08-20 03:26:33 +000095}
96
97void ObjCAtDefsFieldDecl::Destroy(ASTContext& C) {
98 this->~ObjCAtDefsFieldDecl();
Steve Naroff5abb0282009-01-27 21:25:57 +000099 C.Deallocate((void *)this);
Ted Kremeneke5bedfe2008-08-20 03:26:33 +0000100}
101
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +0000102ObjCProtocolDecl *ObjCProtocolDecl::Create(ASTContext &C, DeclContext *DC,
Chris Lattnereee57c02008-04-04 06:12:32 +0000103 SourceLocation L,
Chris Lattner7afba9c2008-03-16 20:19:15 +0000104 IdentifierInfo *Id) {
Steve Naroff5abb0282009-01-27 21:25:57 +0000105 return new (C) ObjCProtocolDecl(DC, L, Id);
Chris Lattner180f7e22008-03-16 01:23:04 +0000106}
107
Chris Lattnerae538882009-02-20 05:54:35 +0000108void ObjCProtocolDecl::Destroy(ASTContext &C) {
Chris Lattner38d2ff22009-02-20 17:53:35 +0000109 ReferencedProtocols.clear();
Chris Lattnerae538882009-02-20 05:54:35 +0000110 ObjCContainerDecl::Destroy(C);
Ted Kremenekc906e5e2008-06-06 19:48:57 +0000111}
112
Ted Kremenekc906e5e2008-06-06 19:48:57 +0000113
Chris Lattnerae538882009-02-20 05:54:35 +0000114
115
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +0000116ObjCClassDecl *ObjCClassDecl::Create(ASTContext &C, DeclContext *DC,
Chris Lattnereee57c02008-04-04 06:12:32 +0000117 SourceLocation L,
Chris Lattner6a028742009-02-20 18:04:31 +0000118 ObjCInterfaceDecl *const *Elts,
119 unsigned nElts) {
Steve Naroff5abb0282009-01-27 21:25:57 +0000120 return new (C) ObjCClassDecl(DC, L, Elts, nElts);
Chris Lattnere29dc832008-03-16 20:34:23 +0000121}
122
Chris Lattner3e27e632009-02-20 06:03:09 +0000123void ObjCClassDecl::Destroy(ASTContext &C) {
Ted Kremenekff291622008-06-06 20:11:53 +0000124
125 // FIXME: There is no clear ownership policy now for referenced
126 // ObjCInterfaceDecls. Some of them can be forward declarations that
127 // are never later defined (in which case the ObjCClassDecl owns them)
128 // or the ObjCInterfaceDecl later becomes a real definition later. Ideally
129 // we should have separate objects for forward declarations and definitions,
130 // obviating this problem. Because of this situation, referenced
131 // ObjCInterfaceDecls are destroyed in ~TranslationUnit.
132
Chris Lattner6a028742009-02-20 18:04:31 +0000133 ForwardDecls.clear();
Ted Kremenekff291622008-06-06 20:11:53 +0000134 Decl::Destroy(C);
135}
136
Chris Lattnere29dc832008-03-16 20:34:23 +0000137ObjCForwardProtocolDecl *
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +0000138ObjCForwardProtocolDecl::Create(ASTContext &C, DeclContext *DC,
Chris Lattnereee57c02008-04-04 06:12:32 +0000139 SourceLocation L,
Chris Lattnere29dc832008-03-16 20:34:23 +0000140 ObjCProtocolDecl **Elts, unsigned NumElts) {
Steve Naroff5abb0282009-01-27 21:25:57 +0000141 return new (C) ObjCForwardProtocolDecl(DC, L, Elts, NumElts);
Chris Lattnere29dc832008-03-16 20:34:23 +0000142}
143
Chris Lattner3e27e632009-02-20 06:03:09 +0000144ObjCForwardProtocolDecl::
145ObjCForwardProtocolDecl(DeclContext *DC, SourceLocation L,
146 ObjCProtocolDecl **Elts, unsigned nElts)
147 : Decl(ObjCForwardProtocol, DC, L) {
148 NumReferencedProtocols = nElts;
149 if (nElts) {
150 ReferencedProtocols = new ObjCProtocolDecl*[nElts];
151 memcpy(ReferencedProtocols, Elts, nElts*sizeof(ObjCProtocolDecl*));
152 } else {
153 ReferencedProtocols = 0;
154 }
155}
156
157void ObjCForwardProtocolDecl::Destroy(ASTContext &C) {
Ted Kremenek1e5e0bf2008-06-06 21:05:33 +0000158 delete [] ReferencedProtocols;
Chris Lattner3e27e632009-02-20 06:03:09 +0000159 ReferencedProtocols = 0;
Ted Kremenek1e5e0bf2008-06-06 21:05:33 +0000160}
161
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +0000162ObjCCategoryDecl *ObjCCategoryDecl::Create(ASTContext &C, DeclContext *DC,
Chris Lattnereee57c02008-04-04 06:12:32 +0000163 SourceLocation L,
Chris Lattnere29dc832008-03-16 20:34:23 +0000164 IdentifierInfo *Id) {
Steve Naroff5abb0282009-01-27 21:25:57 +0000165 return new (C) ObjCCategoryDecl(DC, L, Id);
Chris Lattnere29dc832008-03-16 20:34:23 +0000166}
167
Chris Lattner1b6de332008-03-16 20:53:07 +0000168ObjCCategoryImplDecl *
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +0000169ObjCCategoryImplDecl::Create(ASTContext &C, DeclContext *DC,
Chris Lattnereee57c02008-04-04 06:12:32 +0000170 SourceLocation L,IdentifierInfo *Id,
Chris Lattner1b6de332008-03-16 20:53:07 +0000171 ObjCInterfaceDecl *ClassInterface) {
Steve Naroff5abb0282009-01-27 21:25:57 +0000172 return new (C) ObjCCategoryImplDecl(DC, L, Id, ClassInterface);
Chris Lattner1b6de332008-03-16 20:53:07 +0000173}
174
175ObjCImplementationDecl *
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +0000176ObjCImplementationDecl::Create(ASTContext &C, DeclContext *DC,
Chris Lattnereee57c02008-04-04 06:12:32 +0000177 SourceLocation L,
Chris Lattner1b6de332008-03-16 20:53:07 +0000178 ObjCInterfaceDecl *ClassInterface,
179 ObjCInterfaceDecl *SuperDecl) {
Steve Naroff5abb0282009-01-27 21:25:57 +0000180 return new (C) ObjCImplementationDecl(DC, L, ClassInterface, SuperDecl);
Chris Lattner1b6de332008-03-16 20:53:07 +0000181}
Chris Lattner10318b82008-03-16 00:19:01 +0000182
Chris Lattner2d1c4312008-03-16 21:17:37 +0000183ObjCCompatibleAliasDecl *
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +0000184ObjCCompatibleAliasDecl::Create(ASTContext &C, DeclContext *DC,
Chris Lattnereee57c02008-04-04 06:12:32 +0000185 SourceLocation L,
Chris Lattner2d1c4312008-03-16 21:17:37 +0000186 IdentifierInfo *Id,
187 ObjCInterfaceDecl* AliasedClass) {
Steve Naroff5abb0282009-01-27 21:25:57 +0000188 return new (C) ObjCCompatibleAliasDecl(DC, L, Id, AliasedClass);
Chris Lattner2d1c4312008-03-16 21:17:37 +0000189}
190
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +0000191ObjCPropertyDecl *ObjCPropertyDecl::Create(ASTContext &C, DeclContext *DC,
Fariborz Jahaniane7071722008-04-11 23:40:25 +0000192 SourceLocation L,
Fariborz Jahanian0ceb4be2008-04-14 23:36:35 +0000193 IdentifierInfo *Id,
Fariborz Jahanian4aa72a72008-05-05 18:51:55 +0000194 QualType T,
195 PropertyControl propControl) {
Steve Naroff5abb0282009-01-27 21:25:57 +0000196 return new (C) ObjCPropertyDecl(DC, L, Id, T);
Chris Lattner2d1c4312008-03-16 21:17:37 +0000197}
198
Chris Lattner10318b82008-03-16 00:19:01 +0000199//===----------------------------------------------------------------------===//
200// Objective-C Decl Implementation
201//===----------------------------------------------------------------------===//
202
Fariborz Jahanian91dd9d32008-12-09 20:23:04 +0000203void ObjCMethodDecl::createImplicitParams(ASTContext &Context,
204 const ObjCInterfaceDecl *OID) {
Daniel Dunbar88116372008-08-26 06:08:30 +0000205 QualType selfTy;
Douglas Gregor5d764842009-01-09 17:18:27 +0000206 if (isInstanceMethod()) {
Daniel Dunbar88116372008-08-26 06:08:30 +0000207 // There may be no interface context due to error in declaration
208 // of the interface (which has been reported). Recover gracefully.
Fariborz Jahanian91dd9d32008-12-09 20:23:04 +0000209 if (OID) {
210 selfTy = Context.getObjCInterfaceType(const_cast<ObjCInterfaceDecl *>(OID));
Daniel Dunbar88116372008-08-26 06:08:30 +0000211 selfTy = Context.getPointerType(selfTy);
212 } else {
213 selfTy = Context.getObjCIdType();
214 }
215 } else // we have a factory method.
216 selfTy = Context.getObjCClassType();
217
218 SelfDecl = ImplicitParamDecl::Create(Context, this,
219 SourceLocation(),
220 &Context.Idents.get("self"),
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000221 selfTy);
Daniel Dunbar88116372008-08-26 06:08:30 +0000222
223 CmdDecl = ImplicitParamDecl::Create(Context, this,
224 SourceLocation(),
225 &Context.Idents.get("_cmd"),
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000226 Context.getObjCSelType());
Daniel Dunbar88116372008-08-26 06:08:30 +0000227}
228
Fariborz Jahanianef8a3df2008-04-21 19:04:53 +0000229/// FindCategoryDeclaration - Finds category declaration in the list of
230/// categories for this class and returns it. Name of the category is passed
231/// in 'CategoryId'. If category not found, return 0;
232///
233ObjCCategoryDecl *
234 ObjCInterfaceDecl::FindCategoryDeclaration(IdentifierInfo *CategoryId) const {
235 for (ObjCCategoryDecl *Category = getCategoryList();
236 Category; Category = Category->getNextClassCategory())
237 if (Category->getIdentifier() == CategoryId)
238 return Category;
239 return 0;
240}
241
Fariborz Jahanian09772392008-12-13 22:20:28 +0000242/// lookupFieldDeclForIvar - looks up a field decl' in the laid out
243/// storage which matches this 'ivar'.
244///
245FieldDecl *ObjCInterfaceDecl::lookupFieldDeclForIvar(ASTContext &Context,
Fariborz Jahanian86008c02008-12-15 20:35:07 +0000246 const ObjCIvarDecl *ivar) {
Fariborz Jahanian0556b152008-12-17 21:40:49 +0000247 const RecordDecl *RecordForDecl = Context.addRecordToClass(this);
Fariborz Jahanian09772392008-12-13 22:20:28 +0000248 assert(RecordForDecl && "lookupFieldDeclForIvar no storage for class");
249 DeclarationName Member = ivar->getDeclName();
Fariborz Jahanian0556b152008-12-17 21:40:49 +0000250 DeclContext::lookup_result Lookup = (const_cast< RecordDecl *>(RecordForDecl))
Steve Naroffab63fd62009-01-08 17:28:14 +0000251 ->lookup(Member);
Fariborz Jahanian09772392008-12-13 22:20:28 +0000252 assert((Lookup.first != Lookup.second) && "field decl not found");
253 FieldDecl *MemberDecl = dyn_cast<FieldDecl>(*Lookup.first);
254 assert(MemberDecl && "field decl not found");
255 return MemberDecl;
256}
257
Chris Lattner10318b82008-03-16 00:19:01 +0000258/// ObjCAddInstanceVariablesToClassImpl - Checks for correctness of Instance
259/// Variables (Ivars) relative to what declared in @implementation;s class.
260/// Ivars into ObjCImplementationDecl's fields.
261///
262void ObjCImplementationDecl::ObjCAddInstanceVariablesToClassImpl(
263 ObjCIvarDecl **ivars, unsigned numIvars) {
264 NumIvars = numIvars;
265 if (numIvars) {
266 Ivars = new ObjCIvarDecl*[numIvars];
267 memcpy(Ivars, ivars, numIvars*sizeof(ObjCIvarDecl*));
268 }
269}
270
Steve Naroffab63fd62009-01-08 17:28:14 +0000271// Get the local instance method declared in this interface.
272// FIXME: handle overloading, instance & class methods can have the same name.
273ObjCMethodDecl *ObjCContainerDecl::getInstanceMethod(Selector Sel) const {
274 lookup_const_result MethodResult = lookup(Sel);
275 if (MethodResult.first)
276 return const_cast<ObjCMethodDecl*>(
277 dyn_cast<ObjCMethodDecl>(*MethodResult.first));
278 return 0;
279}
280
281// Get the local class method declared in this interface.
282ObjCMethodDecl *ObjCContainerDecl::getClassMethod(Selector Sel) const {
283 lookup_const_result MethodResult = lookup(Sel);
284 if (MethodResult.first)
285 return const_cast<ObjCMethodDecl*>(
286 dyn_cast<ObjCMethodDecl>(*MethodResult.first));
287 return 0;
288}
289
290unsigned ObjCContainerDecl::getNumInstanceMethods() const {
291 unsigned sum = 0;
292 for (instmeth_iterator I=instmeth_begin(), E=instmeth_end(); I != E; ++I)
293 sum++;
294 return sum;
295}
296unsigned ObjCContainerDecl::getNumClassMethods() const {
297 unsigned sum = 0;
298 for (classmeth_iterator I=classmeth_begin(), E=classmeth_end(); I != E; ++I)
299 sum++;
300 return sum;
301}
Steve Naroff56af2002009-01-11 01:06:09 +0000302unsigned ObjCContainerDecl::getNumProperties() const {
303 unsigned sum = 0;
304 for (prop_iterator I=prop_begin(), E=prop_end(); I != E; ++I)
305 sum++;
306 return sum;
307}
Steve Naroffab63fd62009-01-08 17:28:14 +0000308
Fariborz Jahanianef8a3df2008-04-21 19:04:53 +0000309/// FindPropertyDeclaration - Finds declaration of the property given its name
310/// in 'PropertyId' and returns it. It returns 0, if not found.
Steve Naroffdcf1e842009-01-11 12:47:58 +0000311/// FIXME: Convert to DeclContext lookup...
Fariborz Jahanianef8a3df2008-04-21 19:04:53 +0000312///
313ObjCPropertyDecl *
Steve Naroff451f83c2009-01-09 15:36:25 +0000314ObjCContainerDecl::FindPropertyDeclaration(IdentifierInfo *PropertyId) const {
315 for (prop_iterator I = prop_begin(), E = prop_end(); I != E; ++I) {
Fariborz Jahanianef8a3df2008-04-21 19:04:53 +0000316 ObjCPropertyDecl *property = *I;
317 if (property->getIdentifier() == PropertyId)
318 return property;
319 }
Fariborz Jahanianfaca5e22009-01-09 21:04:52 +0000320 const ObjCProtocolDecl *PID = dyn_cast<ObjCProtocolDecl>(this);
321 if (PID) {
322 for (ObjCProtocolDecl::protocol_iterator P = PID->protocol_begin(),
323 E = PID->protocol_end();
324 P != E; ++P)
325 if (ObjCPropertyDecl *property =
326 (*P)->FindPropertyDeclaration(PropertyId))
327 return property;
328 }
329
Fariborz Jahanian6b0ed6f2009-01-19 18:16:19 +0000330 if (const ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(this)) {
Steve Naroff451f83c2009-01-09 15:36:25 +0000331 // Look through categories.
332 for (ObjCCategoryDecl *Category = OID->getCategoryList();
333 Category; Category = Category->getNextClassCategory()) {
334 ObjCPropertyDecl *property = Category->FindPropertyDeclaration(PropertyId);
335 if (property)
336 return property;
337 }
338 // Look through protocols.
339 for (ObjCInterfaceDecl::protocol_iterator I = OID->protocol_begin(),
340 E = OID->protocol_end(); I != E; ++I) {
341 ObjCProtocolDecl *Protocol = *I;
342 ObjCPropertyDecl *property = Protocol->FindPropertyDeclaration(PropertyId);
343 if (property)
344 return property;
345 }
346 if (OID->getSuperClass())
347 return OID->getSuperClass()->FindPropertyDeclaration(PropertyId);
Steve Naroffd1f0eb42008-06-05 13:55:23 +0000348 }
Fariborz Jahanian6b0ed6f2009-01-19 18:16:19 +0000349 else if (const ObjCCategoryDecl *OCD = dyn_cast<ObjCCategoryDecl>(this)) {
350 // Look through protocols.
351 for (ObjCInterfaceDecl::protocol_iterator I = OCD->protocol_begin(),
352 E = OCD->protocol_end(); I != E; ++I) {
353 ObjCProtocolDecl *Protocol = *I;
354 ObjCPropertyDecl *property = Protocol->FindPropertyDeclaration(PropertyId);
355 if (property)
356 return property;
357 }
358 }
Steve Naroffd1f0eb42008-06-05 13:55:23 +0000359 return 0;
360}
361
Chris Lattner10318b82008-03-16 00:19:01 +0000362ObjCIvarDecl *ObjCInterfaceDecl::lookupInstanceVariable(
363 IdentifierInfo *ID, ObjCInterfaceDecl *&clsDeclared) {
364 ObjCInterfaceDecl* ClassDecl = this;
365 while (ClassDecl != NULL) {
366 for (ivar_iterator I = ClassDecl->ivar_begin(), E = ClassDecl->ivar_end();
367 I != E; ++I) {
368 if ((*I)->getIdentifier() == ID) {
369 clsDeclared = ClassDecl;
370 return *I;
371 }
372 }
373 ClassDecl = ClassDecl->getSuperClass();
374 }
375 return NULL;
376}
377
378/// lookupInstanceMethod - This method returns an instance method by looking in
379/// the class, its categories, and its super classes (using a linear search).
380ObjCMethodDecl *ObjCInterfaceDecl::lookupInstanceMethod(Selector Sel) {
381 ObjCInterfaceDecl* ClassDecl = this;
382 ObjCMethodDecl *MethodDecl = 0;
383
384 while (ClassDecl != NULL) {
385 if ((MethodDecl = ClassDecl->getInstanceMethod(Sel)))
386 return MethodDecl;
387
388 // Didn't find one yet - look through protocols.
Chris Lattner8bcb5252008-07-21 18:19:38 +0000389 const ObjCList<ObjCProtocolDecl> &Protocols =
390 ClassDecl->getReferencedProtocols();
391 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
392 E = Protocols.end(); I != E; ++I)
393 if ((MethodDecl = (*I)->getInstanceMethod(Sel)))
Chris Lattner10318b82008-03-16 00:19:01 +0000394 return MethodDecl;
Chris Lattner8bcb5252008-07-21 18:19:38 +0000395
Chris Lattner10318b82008-03-16 00:19:01 +0000396 // Didn't find one yet - now look through categories.
397 ObjCCategoryDecl *CatDecl = ClassDecl->getCategoryList();
398 while (CatDecl) {
399 if ((MethodDecl = CatDecl->getInstanceMethod(Sel)))
400 return MethodDecl;
Steve Naroffaf151802008-12-08 20:57:28 +0000401
402 // Didn't find one yet - look through protocols.
403 const ObjCList<ObjCProtocolDecl> &Protocols =
404 CatDecl->getReferencedProtocols();
405 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
406 E = Protocols.end(); I != E; ++I)
407 if ((MethodDecl = (*I)->getInstanceMethod(Sel)))
408 return MethodDecl;
Chris Lattner10318b82008-03-16 00:19:01 +0000409 CatDecl = CatDecl->getNextClassCategory();
410 }
411 ClassDecl = ClassDecl->getSuperClass();
412 }
413 return NULL;
414}
415
416// lookupClassMethod - This method returns a class method by looking in the
417// class, its categories, and its super classes (using a linear search).
418ObjCMethodDecl *ObjCInterfaceDecl::lookupClassMethod(Selector Sel) {
419 ObjCInterfaceDecl* ClassDecl = this;
420 ObjCMethodDecl *MethodDecl = 0;
421
422 while (ClassDecl != NULL) {
423 if ((MethodDecl = ClassDecl->getClassMethod(Sel)))
424 return MethodDecl;
425
426 // Didn't find one yet - look through protocols.
Chris Lattner0be08822008-07-21 21:32:27 +0000427 for (ObjCInterfaceDecl::protocol_iterator I = ClassDecl->protocol_begin(),
428 E = ClassDecl->protocol_end(); I != E; ++I)
Chris Lattner8bcb5252008-07-21 18:19:38 +0000429 if ((MethodDecl = (*I)->getClassMethod(Sel)))
Chris Lattner10318b82008-03-16 00:19:01 +0000430 return MethodDecl;
Chris Lattner0be08822008-07-21 21:32:27 +0000431
Chris Lattner10318b82008-03-16 00:19:01 +0000432 // Didn't find one yet - now look through categories.
433 ObjCCategoryDecl *CatDecl = ClassDecl->getCategoryList();
434 while (CatDecl) {
435 if ((MethodDecl = CatDecl->getClassMethod(Sel)))
436 return MethodDecl;
437 CatDecl = CatDecl->getNextClassCategory();
438 }
439 ClassDecl = ClassDecl->getSuperClass();
440 }
441 return NULL;
442}
443
Daniel Dunbardd24b9a2008-08-26 06:53:45 +0000444/// getInstanceMethod - This method returns an instance method by
445/// looking in the class implementation. Unlike interfaces, we don't
446/// look outside the implementation.
447ObjCMethodDecl *ObjCImplementationDecl::getInstanceMethod(Selector Sel) const {
Chris Lattner10318b82008-03-16 00:19:01 +0000448 for (instmeth_iterator I = instmeth_begin(), E = instmeth_end(); I != E; ++I)
449 if ((*I)->getSelector() == Sel)
450 return *I;
451 return NULL;
452}
453
Daniel Dunbardd24b9a2008-08-26 06:53:45 +0000454/// getClassMethod - This method returns a class method by looking in
455/// the class implementation. Unlike interfaces, we don't look outside
456/// the implementation.
457ObjCMethodDecl *ObjCImplementationDecl::getClassMethod(Selector Sel) const {
Chris Lattner10318b82008-03-16 00:19:01 +0000458 for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
459 I != E; ++I)
460 if ((*I)->getSelector() == Sel)
461 return *I;
462 return NULL;
463}
464
Fariborz Jahanian68342282008-12-05 22:32:48 +0000465/// FindPropertyImplDecl - This method looks up a previous ObjCPropertyImplDecl
466/// added to the list of those properties @synthesized/@dynamic in this
467/// @implementation block.
468///
469ObjCPropertyImplDecl *ObjCImplementationDecl::FindPropertyImplDecl(IdentifierInfo *Id) const {
470 for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i) {
471 ObjCPropertyImplDecl *PID = *i;
472 if (PID->getPropertyDecl()->getIdentifier() == Id)
473 return PID;
474 }
475 return 0;
476}
477
478/// FindPropertyImplIvarDecl - This method lookup the ivar in the list of
Fariborz Jahanian17eb8882008-12-05 22:36:19 +0000479/// properties implemented in this @implementation block and returns the
480/// implemented property that uses it.
Fariborz Jahanian68342282008-12-05 22:32:48 +0000481///
482ObjCPropertyImplDecl *ObjCImplementationDecl::FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const {
483 for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i) {
484 ObjCPropertyImplDecl *PID = *i;
485 if (PID->getPropertyIvarDecl() &&
486 PID->getPropertyIvarDecl()->getIdentifier() == ivarId)
487 return PID;
488 }
489 return 0;
490}
491
492/// FindPropertyImplIvarDecl - This method lookup the ivar in the list of
Chris Lattner715e8482009-02-16 19:24:31 +0000493/// properties implemented in this category @implementation block and returns
494/// the implemented property that uses it.
Fariborz Jahanian68342282008-12-05 22:32:48 +0000495///
Chris Lattner715e8482009-02-16 19:24:31 +0000496ObjCPropertyImplDecl *ObjCCategoryImplDecl::
497FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const {
498 for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i){
Fariborz Jahanian68342282008-12-05 22:32:48 +0000499 ObjCPropertyImplDecl *PID = *i;
500 if (PID->getPropertyIvarDecl() &&
501 PID->getPropertyIvarDecl()->getIdentifier() == ivarId)
502 return PID;
503 }
504 return 0;
505}
506
507/// FindPropertyImplDecl - This method looks up a previous ObjCPropertyImplDecl
508/// added to the list of those properties @synthesized/@dynamic in this
509/// category @implementation block.
510///
Chris Lattner715e8482009-02-16 19:24:31 +0000511ObjCPropertyImplDecl *ObjCCategoryImplDecl::
512FindPropertyImplDecl(IdentifierInfo *Id) const {
513 for (propimpl_iterator i = propimpl_begin(), e = propimpl_end(); i != e; ++i){
Fariborz Jahanian68342282008-12-05 22:32:48 +0000514 ObjCPropertyImplDecl *PID = *i;
515 if (PID->getPropertyDecl()->getIdentifier() == Id)
516 return PID;
517 }
518 return 0;
519}
520
Chris Lattner10318b82008-03-16 00:19:01 +0000521// lookupInstanceMethod - This method returns an instance method by looking in
522// the class implementation. Unlike interfaces, we don't look outside the
523// implementation.
Daniel Dunbardd24b9a2008-08-26 06:53:45 +0000524ObjCMethodDecl *ObjCCategoryImplDecl::getInstanceMethod(Selector Sel) const {
Chris Lattner10318b82008-03-16 00:19:01 +0000525 for (instmeth_iterator I = instmeth_begin(), E = instmeth_end(); I != E; ++I)
526 if ((*I)->getSelector() == Sel)
527 return *I;
528 return NULL;
529}
530
531// lookupClassMethod - This method returns an instance method by looking in
532// the class implementation. Unlike interfaces, we don't look outside the
533// implementation.
Daniel Dunbardd24b9a2008-08-26 06:53:45 +0000534ObjCMethodDecl *ObjCCategoryImplDecl::getClassMethod(Selector Sel) const {
Chris Lattner10318b82008-03-16 00:19:01 +0000535 for (classmeth_iterator I = classmeth_begin(), E = classmeth_end();
536 I != E; ++I)
537 if ((*I)->getSelector() == Sel)
538 return *I;
539 return NULL;
540}
541
542// lookupInstanceMethod - Lookup a instance method in the protocol and protocols
543// it inherited.
544ObjCMethodDecl *ObjCProtocolDecl::lookupInstanceMethod(Selector Sel) {
545 ObjCMethodDecl *MethodDecl = NULL;
546
547 if ((MethodDecl = getInstanceMethod(Sel)))
548 return MethodDecl;
549
Chris Lattner0be08822008-07-21 21:32:27 +0000550 for (protocol_iterator I = protocol_begin(), E = protocol_end(); I != E; ++I)
Steve Naroff30fb1da2008-12-18 15:50:41 +0000551 if ((MethodDecl = (*I)->lookupInstanceMethod(Sel)))
Chris Lattner0be08822008-07-21 21:32:27 +0000552 return MethodDecl;
Chris Lattner10318b82008-03-16 00:19:01 +0000553 return NULL;
554}
555
556// lookupInstanceMethod - Lookup a class method in the protocol and protocols
557// it inherited.
558ObjCMethodDecl *ObjCProtocolDecl::lookupClassMethod(Selector Sel) {
559 ObjCMethodDecl *MethodDecl = NULL;
560
561 if ((MethodDecl = getClassMethod(Sel)))
562 return MethodDecl;
563
Chris Lattner0be08822008-07-21 21:32:27 +0000564 for (protocol_iterator I = protocol_begin(), E = protocol_end(); I != E; ++I)
Steve Naroff30fb1da2008-12-18 15:50:41 +0000565 if ((MethodDecl = (*I)->lookupClassMethod(Sel)))
Chris Lattner0be08822008-07-21 21:32:27 +0000566 return MethodDecl;
Chris Lattner10318b82008-03-16 00:19:01 +0000567 return NULL;
568}
569
570/// getSynthesizedMethodSize - Compute size of synthesized method name
571/// as done be the rewrite.
572///
573unsigned ObjCMethodDecl::getSynthesizedMethodSize() const {
574 // syntesized method name is a concatenation of -/+[class-name selector]
575 // Get length of this name.
576 unsigned length = 3; // _I_ or _C_
Chris Lattnerd120b9e2008-11-24 03:54:41 +0000577 length += getClassInterface()->getNameAsString().size()+1; // extra for _
Steve Naroff438be772009-01-08 19:41:02 +0000578 if (const ObjCCategoryImplDecl *CID =
579 dyn_cast<ObjCCategoryImplDecl>(getDeclContext()))
Chris Lattnerd120b9e2008-11-24 03:54:41 +0000580 length += CID->getNameAsString().size()+1;
Chris Lattner3a8f2942008-11-24 03:33:13 +0000581 length += getSelector().getAsString().size(); // selector name
Chris Lattner10318b82008-03-16 00:19:01 +0000582 return length;
583}
584
Chris Lattner052fbcd2008-04-06 05:25:03 +0000585ObjCInterfaceDecl *ObjCMethodDecl::getClassInterface() {
Steve Naroff438be772009-01-08 19:41:02 +0000586 if (ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(getDeclContext()))
Chris Lattner10318b82008-03-16 00:19:01 +0000587 return ID;
Steve Naroff438be772009-01-08 19:41:02 +0000588 if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(getDeclContext()))
Chris Lattner10318b82008-03-16 00:19:01 +0000589 return CD->getClassInterface();
590 if (ObjCImplementationDecl *IMD =
Steve Naroff438be772009-01-08 19:41:02 +0000591 dyn_cast<ObjCImplementationDecl>(getDeclContext()))
Chris Lattner10318b82008-03-16 00:19:01 +0000592 return IMD->getClassInterface();
Steve Naroff438be772009-01-08 19:41:02 +0000593 if (ObjCCategoryImplDecl *CID =
594 dyn_cast<ObjCCategoryImplDecl>(getDeclContext()))
Chris Lattner10318b82008-03-16 00:19:01 +0000595 return CID->getClassInterface();
596 assert(false && "unknown method context");
597 return 0;
598}
Chris Lattner44859612008-03-17 01:19:02 +0000599
Fariborz Jahaniandc0569e2008-04-23 00:06:01 +0000600ObjCPropertyImplDecl *ObjCPropertyImplDecl::Create(ASTContext &C,
Douglas Gregor6e4fa2c2009-01-09 00:49:46 +0000601 DeclContext *DC,
Fariborz Jahaniandc0569e2008-04-23 00:06:01 +0000602 SourceLocation atLoc,
603 SourceLocation L,
604 ObjCPropertyDecl *property,
Daniel Dunbar14117fc2008-08-26 04:47:31 +0000605 Kind PK,
Fariborz Jahaniandc0569e2008-04-23 00:06:01 +0000606 ObjCIvarDecl *ivar) {
Steve Naroff5abb0282009-01-27 21:25:57 +0000607 return new (C) ObjCPropertyImplDecl(DC, atLoc, L, property, PK, ivar);
Fariborz Jahaniandc0569e2008-04-23 00:06:01 +0000608}
Chris Lattner44859612008-03-17 01:19:02 +0000609
Chris Lattnereee57c02008-04-04 06:12:32 +0000610