blob: 0abe047a80c883c2c3cf889a2e98c4b25b9f140e [file] [log] [blame]
Eli Friedmana8a09ac2008-06-07 16:52:53 +00001//===--- DeclBase.cpp - 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 Decl and DeclContext classes.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/AST/DeclBase.h"
Douglas Gregor469fc9a2009-02-02 23:39:07 +000015#include "clang/AST/Decl.h"
Argiris Kirtzidis6df1fc02008-06-09 21:05:31 +000016#include "clang/AST/DeclCXX.h"
Douglas Gregor279272e2009-02-04 19:02:06 +000017#include "clang/AST/DeclObjC.h"
18#include "clang/AST/DeclTemplate.h"
Eli Friedmana8a09ac2008-06-07 16:52:53 +000019#include "clang/AST/ASTContext.h"
Douglas Gregor8acb7272008-12-11 16:49:14 +000020#include "clang/AST/Type.h"
Eli Friedmana8a09ac2008-06-07 16:52:53 +000021#include "llvm/ADT/DenseMap.h"
Chris Lattnerc309ade2009-03-05 08:00:35 +000022#include "llvm/Support/raw_ostream.h"
Douglas Gregor6e71edc2008-12-23 21:05:05 +000023#include <algorithm>
Chris Lattner7d6220c2009-03-02 22:20:04 +000024#include <cstdio>
Douglas Gregor6e71edc2008-12-23 21:05:05 +000025#include <functional>
Douglas Gregorddfd9d52008-12-23 00:26:44 +000026#include <vector>
Eli Friedmana8a09ac2008-06-07 16:52:53 +000027using namespace clang;
28
29//===----------------------------------------------------------------------===//
30// Statistics
31//===----------------------------------------------------------------------===//
32
Douglas Gregor469fc9a2009-02-02 23:39:07 +000033#define DECL(Derived, Base) static int n##Derived##s = 0;
34#include "clang/AST/DeclNodes.def"
Eli Friedmana8a09ac2008-06-07 16:52:53 +000035
36static bool StatSwitch = false;
37
38// This keeps track of all decl attributes. Since so few decls have attrs, we
39// keep them in a hash map instead of wasting space in the Decl class.
40typedef llvm::DenseMap<const Decl*, Attr*> DeclAttrMapTy;
41
42static DeclAttrMapTy *DeclAttrs = 0;
43
44const char *Decl::getDeclKindName() const {
45 switch (DeclKind) {
Douglas Gregor469fc9a2009-02-02 23:39:07 +000046 default: assert(0 && "Declaration not in DeclNodes.def!");
47#define DECL(Derived, Base) case Derived: return #Derived;
48#include "clang/AST/DeclNodes.def"
Eli Friedmana8a09ac2008-06-07 16:52:53 +000049 }
50}
51
Steve Naroffe5f128a2009-01-20 19:53:53 +000052const char *DeclContext::getDeclKindName() const {
53 switch (DeclKind) {
Douglas Gregor469fc9a2009-02-02 23:39:07 +000054 default: assert(0 && "Declaration context not in DeclNodes.def!");
Argiris Kirtzidis2d9b7612009-02-16 14:28:33 +000055#define DECL(Derived, Base) case Decl::Derived: return #Derived;
Douglas Gregor469fc9a2009-02-02 23:39:07 +000056#include "clang/AST/DeclNodes.def"
Steve Naroffe5f128a2009-01-20 19:53:53 +000057 }
58}
59
Eli Friedmana8a09ac2008-06-07 16:52:53 +000060bool Decl::CollectingStats(bool Enable) {
61 if (Enable)
62 StatSwitch = true;
63 return StatSwitch;
64}
65
66void Decl::PrintStats() {
67 fprintf(stderr, "*** Decl Stats:\n");
Eli Friedmana8a09ac2008-06-07 16:52:53 +000068
Douglas Gregor469fc9a2009-02-02 23:39:07 +000069 int totalDecls = 0;
70#define DECL(Derived, Base) totalDecls += n##Derived##s;
71#include "clang/AST/DeclNodes.def"
72 fprintf(stderr, " %d decls total.\n", totalDecls);
73
74 int totalBytes = 0;
75#define DECL(Derived, Base) \
76 if (n##Derived##s > 0) { \
77 totalBytes += (int)(n##Derived##s * sizeof(Derived##Decl)); \
78 fprintf(stderr, " %d " #Derived " decls, %d each (%d bytes)\n", \
79 n##Derived##s, (int)sizeof(Derived##Decl), \
80 (int)(n##Derived##s * sizeof(Derived##Decl))); \
81 }
82#include "clang/AST/DeclNodes.def"
Eli Friedmana8a09ac2008-06-07 16:52:53 +000083
Douglas Gregor469fc9a2009-02-02 23:39:07 +000084 fprintf(stderr, "Total bytes = %d\n", totalBytes);
Eli Friedmana8a09ac2008-06-07 16:52:53 +000085}
86
87void Decl::addDeclKind(Kind k) {
88 switch (k) {
Douglas Gregor469fc9a2009-02-02 23:39:07 +000089 default: assert(0 && "Declaration not in DeclNodes.def!");
90#define DECL(Derived, Base) case Derived: ++n##Derived##s; break;
91#include "clang/AST/DeclNodes.def"
Eli Friedmana8a09ac2008-06-07 16:52:53 +000092 }
93}
94
95//===----------------------------------------------------------------------===//
Chris Lattnerc309ade2009-03-05 08:00:35 +000096// PrettyStackTraceDecl Implementation
97//===----------------------------------------------------------------------===//
98
99void PrettyStackTraceDecl::print(llvm::raw_ostream &OS) const {
100 SourceLocation TheLoc = Loc;
101 if (TheLoc.isInvalid() && TheDecl)
102 TheLoc = TheDecl->getLocation();
103
104 if (TheLoc.isValid()) {
105 TheLoc.print(OS, SM);
106 OS << ": ";
107 }
108
109 OS << Message;
110
111 if (NamedDecl *DN = dyn_cast_or_null<NamedDecl>(TheDecl))
112 OS << " '" << DN->getQualifiedNameAsString() << '\'';
113 OS << '\n';
114}
115
116//===----------------------------------------------------------------------===//
Eli Friedmana8a09ac2008-06-07 16:52:53 +0000117// Decl Implementation
118//===----------------------------------------------------------------------===//
119
Chris Lattner8752fe02009-03-27 20:18:19 +0000120// Out-of-line virtual method providing a home for Decl.
121Decl::~Decl() {
122 if (isOutOfSemaDC())
123 delete getMultipleDC();
124
125 assert(!HasAttrs && "attributes should have been freed by Destroy");
126}
127
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000128void Decl::setDeclContext(DeclContext *DC) {
129 if (isOutOfSemaDC())
130 delete getMultipleDC();
131
Chris Lattner5ce4f6d2009-03-29 06:06:59 +0000132 DeclCtx = DC;
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000133}
134
135void Decl::setLexicalDeclContext(DeclContext *DC) {
136 if (DC == getLexicalDeclContext())
137 return;
138
139 if (isInSemaDC()) {
140 MultipleDC *MDC = new MultipleDC();
141 MDC->SemanticDC = getDeclContext();
142 MDC->LexicalDC = DC;
Chris Lattner5ce4f6d2009-03-29 06:06:59 +0000143 DeclCtx = MDC;
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000144 } else {
145 getMultipleDC()->LexicalDC = DC;
146 }
147}
148
Chris Lattner8752fe02009-03-27 20:18:19 +0000149unsigned Decl::getIdentifierNamespaceForKind(Kind DeclKind) {
150 switch (DeclKind) {
151 default:
152 if (DeclKind >= FunctionFirst && DeclKind <= FunctionLast)
153 return IDNS_Ordinary;
154 assert(0 && "Unknown decl kind!");
155 case OverloadedFunction:
156 case Typedef:
157 case EnumConstant:
158 case Var:
159 case ImplicitParam:
160 case ParmVar:
161 case OriginalParmVar:
162 case NonTypeTemplateParm:
163 case ObjCMethod:
164 case ObjCContainer:
165 case ObjCCategory:
166 case ObjCInterface:
167 case ObjCCategoryImpl:
168 case ObjCProperty:
169 case ObjCCompatibleAlias:
170 return IDNS_Ordinary;
171
172 case ObjCProtocol:
173 return IDNS_Protocol;
174
175 case Field:
176 case ObjCAtDefsField:
177 case ObjCIvar:
178 return IDNS_Member;
179
180 case Record:
181 case CXXRecord:
182 case Enum:
183 case TemplateTypeParm:
184 return IDNS_Tag;
185
186 case Namespace:
187 case Template:
188 case FunctionTemplate:
189 case ClassTemplate:
190 case TemplateTemplateParm:
Anders Carlsson1bf91fb2009-03-28 23:02:53 +0000191 case NamespaceAlias:
Chris Lattner8752fe02009-03-27 20:18:19 +0000192 return IDNS_Tag | IDNS_Ordinary;
193
194 // Never have names.
195 case LinkageSpec:
196 case FileScopeAsm:
197 case StaticAssert:
198 case ObjCClass:
199 case ObjCImplementation:
200 case ObjCPropertyImpl:
201 case ObjCForwardProtocol:
202 case Block:
203 case TranslationUnit:
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000204
Chris Lattner8752fe02009-03-27 20:18:19 +0000205 // Aren't looked up?
206 case UsingDirective:
207 case ClassTemplateSpecialization:
208 return 0;
209 }
Eli Friedmana8a09ac2008-06-07 16:52:53 +0000210}
211
212void Decl::addAttr(Attr *NewAttr) {
213 if (!DeclAttrs)
214 DeclAttrs = new DeclAttrMapTy();
215
216 Attr *&ExistingAttr = (*DeclAttrs)[this];
217
218 NewAttr->setNext(ExistingAttr);
219 ExistingAttr = NewAttr;
220
221 HasAttrs = true;
222}
223
224void Decl::invalidateAttrs() {
225 if (!HasAttrs) return;
226
227 HasAttrs = false;
228 (*DeclAttrs)[this] = 0;
229 DeclAttrs->erase(this);
230
231 if (DeclAttrs->empty()) {
232 delete DeclAttrs;
233 DeclAttrs = 0;
234 }
235}
236
Chris Lattner004a9bb2009-03-21 06:27:31 +0000237const Attr *Decl::getAttrsImpl() const {
238 assert(HasAttrs && "getAttrs() should verify this!");
Eli Friedmana8a09ac2008-06-07 16:52:53 +0000239 return (*DeclAttrs)[this];
240}
241
242void Decl::swapAttrs(Decl *RHS) {
243 bool HasLHSAttr = this->HasAttrs;
244 bool HasRHSAttr = RHS->HasAttrs;
245
246 // Usually, neither decl has attrs, nothing to do.
247 if (!HasLHSAttr && !HasRHSAttr) return;
248
249 // If 'this' has no attrs, swap the other way.
250 if (!HasLHSAttr)
251 return RHS->swapAttrs(this);
252
253 // Handle the case when both decls have attrs.
254 if (HasRHSAttr) {
255 std::swap((*DeclAttrs)[this], (*DeclAttrs)[RHS]);
256 return;
257 }
258
259 // Otherwise, LHS has an attr and RHS doesn't.
260 (*DeclAttrs)[RHS] = (*DeclAttrs)[this];
261 (*DeclAttrs).erase(this);
262 this->HasAttrs = false;
263 RHS->HasAttrs = true;
264}
265
266
Chris Lattner0fc6d642009-03-04 06:05:19 +0000267void Decl::Destroy(ASTContext &C) {
268 // Free attributes for this decl.
269 if (HasAttrs) {
270 DeclAttrMapTy::iterator it = DeclAttrs->find(this);
271 assert(it != DeclAttrs->end() && "No attrs found but HasAttrs is true!");
272
273 // release attributes.
274 it->second->Destroy(C);
275 invalidateAttrs();
276 HasAttrs = false;
277 }
278
Douglas Gregor8314d742009-01-13 19:47:12 +0000279#if 0
Douglas Gregorf4006be2009-01-20 04:25:11 +0000280 // FIXME: Once ownership is fully understood, we can enable this code
281 if (DeclContext *DC = dyn_cast<DeclContext>(this))
282 DC->decls_begin()->Destroy(C);
Eli Friedmana8a09ac2008-06-07 16:52:53 +0000283
Chris Lattnerfb8416a2009-03-28 06:04:26 +0000284 // Observe the unrolled recursion. By setting N->NextDeclInContext = 0x0
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000285 // within the loop, only the Destroy method for the first Decl
286 // will deallocate all of the Decls in a chain.
287
Chris Lattnerfb8416a2009-03-28 06:04:26 +0000288 Decl* N = getNextDeclInContext();
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000289
290 while (N) {
Chris Lattnerfb8416a2009-03-28 06:04:26 +0000291 Decl* Tmp = N->getNextDeclInContext();
292 N->NextDeclInContext = 0;
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000293 N->Destroy(C);
294 N = Tmp;
Eli Friedmana8a09ac2008-06-07 16:52:53 +0000295 }
Douglas Gregor8314d742009-01-13 19:47:12 +0000296
Eli Friedmana8a09ac2008-06-07 16:52:53 +0000297 this->~Decl();
Steve Naroff5abb0282009-01-27 21:25:57 +0000298 C.Deallocate((void *)this);
Douglas Gregorf4006be2009-01-20 04:25:11 +0000299#endif
Eli Friedmana8a09ac2008-06-07 16:52:53 +0000300}
301
Argiris Kirtzidis4b662ea2008-10-12 16:14:48 +0000302Decl *Decl::castFromDeclContext (const DeclContext *D) {
Argiris Kirtzidis981aa132009-02-16 14:29:28 +0000303 Decl::Kind DK = D->getDeclKind();
304 switch(DK) {
305#define DECL_CONTEXT(Name) \
306 case Decl::Name: \
307 return static_cast<Name##Decl*>(const_cast<DeclContext*>(D));
308#define DECL_CONTEXT_BASE(Name)
309#include "clang/AST/DeclNodes.def"
310 default:
311#define DECL_CONTEXT_BASE(Name) \
312 if (DK >= Decl::Name##First && DK <= Decl::Name##Last) \
313 return static_cast<Name##Decl*>(const_cast<DeclContext*>(D));
314#include "clang/AST/DeclNodes.def"
315 assert(false && "a decl that inherits DeclContext isn't handled");
316 return 0;
317 }
Argiris Kirtzidis4b662ea2008-10-12 16:14:48 +0000318}
319
320DeclContext *Decl::castToDeclContext(const Decl *D) {
Argiris Kirtzidis981aa132009-02-16 14:29:28 +0000321 Decl::Kind DK = D->getKind();
322 switch(DK) {
323#define DECL_CONTEXT(Name) \
324 case Decl::Name: \
325 return static_cast<Name##Decl*>(const_cast<Decl*>(D));
326#define DECL_CONTEXT_BASE(Name)
327#include "clang/AST/DeclNodes.def"
328 default:
329#define DECL_CONTEXT_BASE(Name) \
330 if (DK >= Decl::Name##First && DK <= Decl::Name##Last) \
331 return static_cast<Name##Decl*>(const_cast<Decl*>(D));
332#include "clang/AST/DeclNodes.def"
333 assert(false && "a decl that inherits DeclContext isn't handled");
334 return 0;
335 }
Argiris Kirtzidis4b662ea2008-10-12 16:14:48 +0000336}
337
Anders Carlsson8ea6a322009-03-25 23:38:06 +0000338#ifndef NDEBUG
339void Decl::CheckAccessDeclContext() const {
340 assert((Access != AS_none || !isa<CXXRecordDecl>(getDeclContext())) &&
341 "Access specifier is AS_none inside a record decl");
342}
343
344#endif
345
Eli Friedmana8a09ac2008-06-07 16:52:53 +0000346//===----------------------------------------------------------------------===//
347// DeclContext Implementation
348//===----------------------------------------------------------------------===//
349
Argiris Kirtzidis981aa132009-02-16 14:29:28 +0000350bool DeclContext::classof(const Decl *D) {
351 switch (D->getKind()) {
352#define DECL_CONTEXT(Name) case Decl::Name:
353#define DECL_CONTEXT_BASE(Name)
354#include "clang/AST/DeclNodes.def"
355 return true;
356 default:
357#define DECL_CONTEXT_BASE(Name) \
358 if (D->getKind() >= Decl::Name##First && \
359 D->getKind() <= Decl::Name##Last) \
360 return true;
361#include "clang/AST/DeclNodes.def"
362 return false;
363 }
364}
365
Chris Lattner77820d52009-02-20 01:44:05 +0000366/// StoredDeclsList - This is an array of decls optimized a common case of only
367/// containing one entry.
368struct StoredDeclsList {
Chris Lattner77820d52009-02-20 01:44:05 +0000369 /// VectorTy - When in vector form, this is what the Data pointer points to.
370 typedef llvm::SmallVector<NamedDecl*, 4> VectorTy;
Chris Lattner655863f2009-03-29 06:43:22 +0000371
372 /// Data - Union of NamedDecl*/VectorTy*.
373 llvm::PointerUnion<NamedDecl*, VectorTy*> Data;
Chris Lattner77820d52009-02-20 01:44:05 +0000374public:
375 StoredDeclsList() {}
376 StoredDeclsList(const StoredDeclsList &RHS) : Data(RHS.Data) {
377 if (isVector())
Chris Lattner655863f2009-03-29 06:43:22 +0000378 Data = new VectorTy(*Data.get<VectorTy*>());
Chris Lattner77820d52009-02-20 01:44:05 +0000379 }
380
381 ~StoredDeclsList() {
382 // If this is a vector-form, free the vector.
383 if (isVector())
Chris Lattner655863f2009-03-29 06:43:22 +0000384 delete Data.get<VectorTy*>();
Chris Lattner77820d52009-02-20 01:44:05 +0000385 }
386
Chris Lattner51e57e62009-02-23 18:17:44 +0000387 StoredDeclsList &operator=(const StoredDeclsList &RHS) {
388 if (isVector())
Chris Lattner655863f2009-03-29 06:43:22 +0000389 delete Data.get<VectorTy*>();
Chris Lattner51e57e62009-02-23 18:17:44 +0000390 Data = RHS.Data;
391 if (isVector())
Chris Lattner655863f2009-03-29 06:43:22 +0000392 Data = new VectorTy(*Data.get<VectorTy*>());
Chris Lattner51e57e62009-02-23 18:17:44 +0000393 return *this;
394 }
395
Chris Lattner655863f2009-03-29 06:43:22 +0000396 bool isVector() const { return Data.is<VectorTy*>(); }
397 bool isInline() const { return Data.is<NamedDecl*>(); }
398 bool isNull() const { return Data.isNull(); }
Chris Lattner77820d52009-02-20 01:44:05 +0000399
400 void setOnlyValue(NamedDecl *ND) {
401 assert(isInline() && "Not inline");
Chris Lattner655863f2009-03-29 06:43:22 +0000402 Data = ND;
Chris Lattner77820d52009-02-20 01:44:05 +0000403 }
404
405 /// getLookupResult - Return an array of all the decls that this list
406 /// represents.
407 DeclContext::lookup_result getLookupResult() {
408 // If we have a single inline unit, return it.
409 if (isInline()) {
410 assert(!isNull() && "Empty list isn't allowed");
411
412 // Data is a raw pointer to a NamedDecl*, return it.
413 void *Ptr = &Data;
414 return DeclContext::lookup_result((NamedDecl**)Ptr, (NamedDecl**)Ptr+1);
415 }
416
417 // Otherwise, we have a range result.
Chris Lattner655863f2009-03-29 06:43:22 +0000418 VectorTy &V = *Data.get<VectorTy*>();
Chris Lattner77820d52009-02-20 01:44:05 +0000419 return DeclContext::lookup_result(&V[0], &V[0]+V.size());
420 }
421
422 /// HandleRedeclaration - If this is a redeclaration of an existing decl,
423 /// replace the old one with D and return true. Otherwise return false.
424 bool HandleRedeclaration(NamedDecl *D) {
425 // Most decls only have one entry in their list, special case it.
426 if (isInline()) {
Chris Lattner655863f2009-03-29 06:43:22 +0000427 if (!D->declarationReplaces(Data.get<NamedDecl*>()))
Chris Lattner77820d52009-02-20 01:44:05 +0000428 return false;
429 setOnlyValue(D);
430 return true;
431 }
432
433 // Determine if this declaration is actually a redeclaration.
Chris Lattner655863f2009-03-29 06:43:22 +0000434 VectorTy &Vec = *Data.get<VectorTy*>();
Chris Lattner77820d52009-02-20 01:44:05 +0000435 VectorTy::iterator RDI
436 = std::find_if(Vec.begin(), Vec.end(),
437 std::bind1st(std::mem_fun(&NamedDecl::declarationReplaces),
438 D));
439 if (RDI == Vec.end())
440 return false;
441 *RDI = D;
442 return true;
443 }
444
445 /// AddSubsequentDecl - This is called on the second and later decl when it is
446 /// not a redeclaration to merge it into the appropriate place in our list.
447 ///
448 void AddSubsequentDecl(NamedDecl *D) {
449 // If this is the second decl added to the list, convert this to vector
450 // form.
451 if (isInline()) {
Chris Lattner655863f2009-03-29 06:43:22 +0000452 NamedDecl *OldD = Data.get<NamedDecl*>();
Chris Lattner77820d52009-02-20 01:44:05 +0000453 VectorTy *VT = new VectorTy();
454 VT->push_back(OldD);
Chris Lattner655863f2009-03-29 06:43:22 +0000455 Data = VT;
Chris Lattner77820d52009-02-20 01:44:05 +0000456 }
457
Chris Lattner655863f2009-03-29 06:43:22 +0000458 VectorTy &Vec = *Data.get<VectorTy*>();
Chris Lattner77820d52009-02-20 01:44:05 +0000459 if (isa<UsingDirectiveDecl>(D) ||
460 D->getIdentifierNamespace() == Decl::IDNS_Tag)
461 Vec.push_back(D);
462 else if (Vec.back()->getIdentifierNamespace() == Decl::IDNS_Tag) {
463 NamedDecl *TagD = Vec.back();
464 Vec.back() = D;
465 Vec.push_back(TagD);
466 } else
467 Vec.push_back(D);
468 }
Chris Lattner77820d52009-02-20 01:44:05 +0000469};
470
471
472
473typedef llvm::DenseMap<DeclarationName, StoredDeclsList> StoredDeclsMap;
Douglas Gregor8acb7272008-12-11 16:49:14 +0000474
475DeclContext::~DeclContext() {
Chris Lattner8752fe02009-03-27 20:18:19 +0000476 if (isLookupMap())
Chris Lattner265f01d2009-02-20 00:55:03 +0000477 delete static_cast<StoredDeclsMap*>(LookupPtr.getPointer());
478 else
479 delete [] static_cast<NamedDecl**>(LookupPtr.getPointer());
Douglas Gregor8acb7272008-12-11 16:49:14 +0000480}
481
482void DeclContext::DestroyDecls(ASTContext &C) {
Douglas Gregorf4006be2009-01-20 04:25:11 +0000483 for (decl_iterator D = decls_begin(); D != decls_end(); )
484 (*D++)->Destroy(C);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000485}
486
Douglas Gregord8028382009-01-05 19:45:36 +0000487bool DeclContext::isTransparentContext() const {
488 if (DeclKind == Decl::Enum)
489 return true; // FIXME: Check for C++0x scoped enums
490 else if (DeclKind == Decl::LinkageSpec)
491 return true;
Douglas Gregor1db12932009-02-26 00:02:51 +0000492 else if (DeclKind >= Decl::RecordFirst && DeclKind <= Decl::RecordLast)
Douglas Gregor723d3332009-01-07 00:43:41 +0000493 return cast<RecordDecl>(this)->isAnonymousStructOrUnion();
Douglas Gregord8028382009-01-05 19:45:36 +0000494 else if (DeclKind == Decl::Namespace)
495 return false; // FIXME: Check for C++0x inline namespaces
496
497 return false;
498}
499
Steve Naroffab63fd62009-01-08 17:28:14 +0000500DeclContext *DeclContext::getPrimaryContext() {
Douglas Gregor8acb7272008-12-11 16:49:14 +0000501 switch (DeclKind) {
Douglas Gregor8acb7272008-12-11 16:49:14 +0000502 case Decl::TranslationUnit:
Douglas Gregord8028382009-01-05 19:45:36 +0000503 case Decl::LinkageSpec:
504 case Decl::Block:
Douglas Gregor8acb7272008-12-11 16:49:14 +0000505 // There is only one DeclContext for these entities.
506 return this;
507
508 case Decl::Namespace:
509 // The original namespace is our primary context.
510 return static_cast<NamespaceDecl*>(this)->getOriginalNamespace();
511
Douglas Gregor8acb7272008-12-11 16:49:14 +0000512 case Decl::ObjCMethod:
513 return this;
514
515 case Decl::ObjCInterface:
Steve Naroffab63fd62009-01-08 17:28:14 +0000516 case Decl::ObjCProtocol:
517 case Decl::ObjCCategory:
Douglas Gregor8acb7272008-12-11 16:49:14 +0000518 // FIXME: Can Objective-C interfaces be forward-declared?
519 return this;
520
Steve Naroffab63fd62009-01-08 17:28:14 +0000521 case Decl::ObjCImplementation:
522 case Decl::ObjCCategoryImpl:
523 return this;
524
Douglas Gregor8acb7272008-12-11 16:49:14 +0000525 default:
Douglas Gregora08b6c72009-02-17 23:15:12 +0000526 if (DeclKind >= Decl::TagFirst && DeclKind <= Decl::TagLast) {
527 // If this is a tag type that has a definition or is currently
528 // being defined, that definition is our primary context.
Chris Lattnerfb8416a2009-03-28 06:04:26 +0000529 if (const TagType *TagT =cast<TagDecl>(this)->TypeForDecl->getAsTagType())
Douglas Gregora08b6c72009-02-17 23:15:12 +0000530 if (TagT->isBeingDefined() ||
531 (TagT->getDecl() && TagT->getDecl()->isDefinition()))
532 return TagT->getDecl();
533 return this;
534 }
535
Douglas Gregor8acb7272008-12-11 16:49:14 +0000536 assert(DeclKind >= Decl::FunctionFirst && DeclKind <= Decl::FunctionLast &&
537 "Unknown DeclContext kind");
538 return this;
539 }
540}
541
542DeclContext *DeclContext::getNextContext() {
543 switch (DeclKind) {
Douglas Gregor8acb7272008-12-11 16:49:14 +0000544 case Decl::Namespace:
545 // Return the next namespace
546 return static_cast<NamespaceDecl*>(this)->getNextNamespace();
547
548 default:
Douglas Gregor8acb7272008-12-11 16:49:14 +0000549 return 0;
550 }
551}
552
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000553void DeclContext::addDecl(Decl *D) {
Chris Lattner53602072009-02-20 00:56:18 +0000554 assert(D->getLexicalDeclContext() == this &&
555 "Decl inserted into wrong lexical context");
Chris Lattnerfb8416a2009-03-28 06:04:26 +0000556 assert(!D->getNextDeclInContext() && D != LastDecl &&
Douglas Gregord1675382009-01-09 19:42:16 +0000557 "Decl already inserted into a DeclContext");
558
559 if (FirstDecl) {
Chris Lattnerfb8416a2009-03-28 06:04:26 +0000560 LastDecl->NextDeclInContext = D;
Douglas Gregord1675382009-01-09 19:42:16 +0000561 LastDecl = D;
562 } else {
563 FirstDecl = LastDecl = D;
564 }
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000565
566 if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
Douglas Gregor9ac66d22009-01-20 16:54:50 +0000567 ND->getDeclContext()->makeDeclVisibleInContext(ND);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000568}
569
Douglas Gregord8028382009-01-05 19:45:36 +0000570/// buildLookup - Build the lookup data structure with all of the
571/// declarations in DCtx (and any other contexts linked to it or
572/// transparent contexts nested within it).
Steve Naroffab63fd62009-01-08 17:28:14 +0000573void DeclContext::buildLookup(DeclContext *DCtx) {
Douglas Gregord8028382009-01-05 19:45:36 +0000574 for (; DCtx; DCtx = DCtx->getNextContext()) {
Douglas Gregorf0464ec2009-01-06 07:17:58 +0000575 for (decl_iterator D = DCtx->decls_begin(), DEnd = DCtx->decls_end();
576 D != DEnd; ++D) {
Douglas Gregord8028382009-01-05 19:45:36 +0000577 // Insert this declaration into the lookup structure
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000578 if (NamedDecl *ND = dyn_cast<NamedDecl>(*D))
Douglas Gregor9ac66d22009-01-20 16:54:50 +0000579 makeDeclVisibleInContextImpl(ND);
Douglas Gregord8028382009-01-05 19:45:36 +0000580
581 // If this declaration is itself a transparent declaration context,
582 // add its members (recursively).
583 if (DeclContext *InnerCtx = dyn_cast<DeclContext>(*D))
584 if (InnerCtx->isTransparentContext())
Steve Naroffab63fd62009-01-08 17:28:14 +0000585 buildLookup(InnerCtx->getPrimaryContext());
Douglas Gregord8028382009-01-05 19:45:36 +0000586 }
587 }
588}
589
Douglas Gregor8acb7272008-12-11 16:49:14 +0000590DeclContext::lookup_result
Steve Naroffab63fd62009-01-08 17:28:14 +0000591DeclContext::lookup(DeclarationName Name) {
592 DeclContext *PrimaryContext = getPrimaryContext();
Douglas Gregor8acb7272008-12-11 16:49:14 +0000593 if (PrimaryContext != this)
Steve Naroffab63fd62009-01-08 17:28:14 +0000594 return PrimaryContext->lookup(Name);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000595
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000596 /// If there is no lookup data structure, build one now by walking
Douglas Gregor8acb7272008-12-11 16:49:14 +0000597 /// all of the linked DeclContexts (in declaration order!) and
598 /// inserting their values.
Douglas Gregord8028382009-01-05 19:45:36 +0000599 if (LookupPtr.getPointer() == 0)
Steve Naroffab63fd62009-01-08 17:28:14 +0000600 buildLookup(this);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000601
Douglas Gregor8acb7272008-12-11 16:49:14 +0000602 if (isLookupMap()) {
603 StoredDeclsMap *Map = static_cast<StoredDeclsMap*>(LookupPtr.getPointer());
604 StoredDeclsMap::iterator Pos = Map->find(Name);
Chris Lattner265f01d2009-02-20 00:55:03 +0000605 if (Pos == Map->end())
606 return lookup_result(0, 0);
Chris Lattner77820d52009-02-20 01:44:05 +0000607 return Pos->second.getLookupResult();
Douglas Gregor8acb7272008-12-11 16:49:14 +0000608 }
609
610 // We have a small array. Look into it.
611 unsigned Size = LookupPtr.getInt();
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000612 NamedDecl **Array = static_cast<NamedDecl**>(LookupPtr.getPointer());
Douglas Gregor39677622008-12-11 20:41:00 +0000613 for (unsigned Idx = 0; Idx != Size; ++Idx)
Douglas Gregor8acb7272008-12-11 16:49:14 +0000614 if (Array[Idx]->getDeclName() == Name) {
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000615 unsigned Last = Idx + 1;
616 while (Last != Size && Array[Last]->getDeclName() == Name)
617 ++Last;
618 return lookup_result(&Array[Idx], &Array[Last]);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000619 }
620
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000621 return lookup_result(0, 0);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000622}
623
624DeclContext::lookup_const_result
Steve Naroffab63fd62009-01-08 17:28:14 +0000625DeclContext::lookup(DeclarationName Name) const {
626 return const_cast<DeclContext*>(this)->lookup(Name);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000627}
628
Chris Lattner9a502bd2009-03-27 19:19:59 +0000629DeclContext *DeclContext::getLookupContext() {
630 DeclContext *Ctx = this;
Douglas Gregordb568cf2009-01-08 20:45:30 +0000631 // Skip through transparent contexts.
Douglas Gregor69e781f2009-01-06 23:51:29 +0000632 while (Ctx->isTransparentContext())
633 Ctx = Ctx->getParent();
634 return Ctx;
635}
636
Douglas Gregor0d93f692009-02-25 22:02:03 +0000637DeclContext *DeclContext::getEnclosingNamespaceContext() {
638 DeclContext *Ctx = this;
639 // Skip through non-namespace, non-translation-unit contexts.
640 while (!Ctx->isFileContext() || Ctx->isTransparentContext())
641 Ctx = Ctx->getParent();
642 return Ctx->getPrimaryContext();
643}
644
Douglas Gregor9ac66d22009-01-20 16:54:50 +0000645void DeclContext::makeDeclVisibleInContext(NamedDecl *D) {
Douglas Gregora08b6c72009-02-17 23:15:12 +0000646 // FIXME: This feels like a hack. Should DeclarationName support
647 // template-ids, or is there a better way to keep specializations
648 // from being visible?
649 if (isa<ClassTemplateSpecializationDecl>(D))
650 return;
651
Steve Naroffab63fd62009-01-08 17:28:14 +0000652 DeclContext *PrimaryContext = getPrimaryContext();
Douglas Gregor8acb7272008-12-11 16:49:14 +0000653 if (PrimaryContext != this) {
Douglas Gregor9ac66d22009-01-20 16:54:50 +0000654 PrimaryContext->makeDeclVisibleInContext(D);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000655 return;
656 }
657
658 // If we already have a lookup data structure, perform the insertion
659 // into it. Otherwise, be lazy and don't build that structure until
660 // someone asks for it.
661 if (LookupPtr.getPointer())
Douglas Gregor9ac66d22009-01-20 16:54:50 +0000662 makeDeclVisibleInContextImpl(D);
Douglas Gregord8028382009-01-05 19:45:36 +0000663
Douglas Gregord8028382009-01-05 19:45:36 +0000664 // If we are a transparent context, insert into our parent context,
665 // too. This operation is recursive.
666 if (isTransparentContext())
Douglas Gregor9ac66d22009-01-20 16:54:50 +0000667 getParent()->makeDeclVisibleInContext(D);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000668}
669
Douglas Gregor9ac66d22009-01-20 16:54:50 +0000670void DeclContext::makeDeclVisibleInContextImpl(NamedDecl *D) {
Douglas Gregord8028382009-01-05 19:45:36 +0000671 // Skip unnamed declarations.
672 if (!D->getDeclName())
673 return;
674
Douglas Gregora08b6c72009-02-17 23:15:12 +0000675 // FIXME: This feels like a hack. Should DeclarationName support
676 // template-ids, or is there a better way to keep specializations
677 // from being visible?
678 if (isa<ClassTemplateSpecializationDecl>(D))
679 return;
680
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000681 bool MayBeRedeclaration = true;
682
Douglas Gregor8acb7272008-12-11 16:49:14 +0000683 if (!isLookupMap()) {
684 unsigned Size = LookupPtr.getInt();
685
686 // The lookup data is stored as an array. Search through the array
687 // to find the insertion location.
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000688 NamedDecl **Array;
Douglas Gregor8acb7272008-12-11 16:49:14 +0000689 if (Size == 0) {
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000690 Array = new NamedDecl*[LookupIsMap - 1];
Douglas Gregor8acb7272008-12-11 16:49:14 +0000691 LookupPtr.setPointer(Array);
692 } else {
Douglas Gregoraf8ad2b2009-01-20 01:17:11 +0000693 Array = static_cast<NamedDecl **>(LookupPtr.getPointer());
Douglas Gregor8acb7272008-12-11 16:49:14 +0000694 }
695
696 // We always keep declarations of the same name next to each other
697 // in the array, so that it is easy to return multiple results
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000698 // from lookup().
699 unsigned FirstMatch;
700 for (FirstMatch = 0; FirstMatch != Size; ++FirstMatch)
701 if (Array[FirstMatch]->getDeclName() == D->getDeclName())
Douglas Gregor39677622008-12-11 20:41:00 +0000702 break;
Douglas Gregor8acb7272008-12-11 16:49:14 +0000703
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000704 unsigned InsertPos = FirstMatch;
705 if (FirstMatch != Size) {
706 // We found another declaration with the same name. First
707 // determine whether this is a redeclaration of an existing
708 // declaration in this scope, in which case we will replace the
709 // existing declaration.
710 unsigned LastMatch = FirstMatch;
711 for (; LastMatch != Size; ++LastMatch) {
712 if (Array[LastMatch]->getDeclName() != D->getDeclName())
713 break;
714
Douglas Gregor6e71edc2008-12-23 21:05:05 +0000715 if (D->declarationReplaces(Array[LastMatch])) {
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000716 // D is a redeclaration of an existing element in the
717 // array. Replace that element with D.
718 Array[LastMatch] = D;
719 return;
720 }
Douglas Gregor8acb7272008-12-11 16:49:14 +0000721 }
722
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000723 // [FirstMatch, LastMatch) contains the set of declarations that
724 // have the same name as this declaration. Determine where the
Douglas Gregor7a7be652009-02-03 19:21:40 +0000725 // declaration D will be inserted into this range.
726 if (D->getKind() == Decl::UsingDirective ||
727 D->getIdentifierNamespace() == Decl::IDNS_Tag)
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000728 InsertPos = LastMatch;
729 else if (Array[LastMatch-1]->getIdentifierNamespace() == Decl::IDNS_Tag)
730 InsertPos = LastMatch - 1;
731 else
732 InsertPos = LastMatch;
Douglas Gregor8acb7272008-12-11 16:49:14 +0000733 }
734
735 if (Size < LookupIsMap - 1) {
736 // The new declaration will fit in the array. Insert the new
737 // declaration at the position Match in the array.
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000738 for (unsigned Idx = Size; Idx > InsertPos; --Idx)
Douglas Gregor8acb7272008-12-11 16:49:14 +0000739 Array[Idx] = Array[Idx-1];
740
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000741 Array[InsertPos] = D;
Douglas Gregor8acb7272008-12-11 16:49:14 +0000742 LookupPtr.setInt(Size + 1);
743 return;
744 }
745
746 // We've reached capacity in this array. Create a map and copy in
747 // all of the declarations that were stored in the array.
748 StoredDeclsMap *Map = new StoredDeclsMap(16);
749 LookupPtr.setPointer(Map);
750 LookupPtr.setInt(LookupIsMap);
Douglas Gregor39677622008-12-11 20:41:00 +0000751 for (unsigned Idx = 0; Idx != LookupIsMap - 1; ++Idx)
Douglas Gregor9ac66d22009-01-20 16:54:50 +0000752 makeDeclVisibleInContextImpl(Array[Idx]);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000753 delete [] Array;
754
755 // Fall through to perform insertion into the map.
Douglas Gregorddfd9d52008-12-23 00:26:44 +0000756 MayBeRedeclaration = false;
757 }
Douglas Gregor8acb7272008-12-11 16:49:14 +0000758
759 // Insert this declaration into the map.
Chris Lattner77820d52009-02-20 01:44:05 +0000760 StoredDeclsMap &Map = *static_cast<StoredDeclsMap*>(LookupPtr.getPointer());
761 StoredDeclsList &DeclNameEntries = Map[D->getDeclName()];
762 if (DeclNameEntries.isNull()) {
763 DeclNameEntries.setOnlyValue(D);
Chris Lattner6719b1f2009-02-19 07:00:44 +0000764 return;
Douglas Gregor8acb7272008-12-11 16:49:14 +0000765 }
Chris Lattner265f01d2009-02-20 00:55:03 +0000766
Chris Lattnerba589d42009-02-20 01:10:07 +0000767 // If it is possible that this is a redeclaration, check to see if there is
768 // already a decl for which declarationReplaces returns true. If there is
769 // one, just replace it and return.
Chris Lattner77820d52009-02-20 01:44:05 +0000770 if (MayBeRedeclaration && DeclNameEntries.HandleRedeclaration(D))
771 return;
Chris Lattner265f01d2009-02-20 00:55:03 +0000772
Chris Lattner6719b1f2009-02-19 07:00:44 +0000773 // Put this declaration into the appropriate slot.
Chris Lattner77820d52009-02-20 01:44:05 +0000774 DeclNameEntries.AddSubsequentDecl(D);
Douglas Gregor8acb7272008-12-11 16:49:14 +0000775}
Douglas Gregor7a7be652009-02-03 19:21:40 +0000776
777/// Returns iterator range [First, Last) of UsingDirectiveDecls stored within
778/// this context.
779DeclContext::udir_iterator_range DeclContext::getUsingDirectives() const {
780 lookup_const_result Result = lookup(UsingDirectiveDecl::getName());
781 return udir_iterator_range(reinterpret_cast<udir_iterator>(Result.first),
782 reinterpret_cast<udir_iterator>(Result.second));
783}
784