blob: befdce90459681ff16bf7f81b21669752898b541 [file] [log] [blame]
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +00001//===- CIndexHigh.cpp - Higher level API functions ------------------------===//
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#include "IndexingContext.h"
11#include "CXTranslationUnit.h"
12#include "CIndexDiagnostic.h"
13
14#include "clang/Frontend/ASTUnit.h"
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +000015#include "clang/AST/DeclCXX.h"
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +000016#include "clang/AST/DeclTemplate.h"
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +000017
18using namespace clang;
19using namespace cxindex;
20using namespace cxcursor;
21
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +000022IndexingContext::ObjCProtocolListInfo::ObjCProtocolListInfo(
23 const ObjCProtocolList &ProtList,
24 IndexingContext &IdxCtx,
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +000025 ScratchAlloc &SA) {
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +000026 ObjCInterfaceDecl::protocol_loc_iterator LI = ProtList.loc_begin();
27 for (ObjCInterfaceDecl::protocol_iterator
28 I = ProtList.begin(), E = ProtList.end(); I != E; ++I, ++LI) {
29 SourceLocation Loc = *LI;
30 ObjCProtocolDecl *PD = *I;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +000031 ProtEntities.push_back(EntityInfo());
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +000032 IdxCtx.getEntityInfo(PD, ProtEntities.back(), SA);
33 CXIdxObjCProtocolRefInfo ProtInfo = { 0,
34 MakeCursorObjCProtocolRef(PD, Loc, IdxCtx.CXTU),
35 IdxCtx.getIndexLoc(Loc) };
36 ProtInfos.push_back(ProtInfo);
Argyrios Kyrtzidis90fd67a2011-12-15 00:04:56 +000037
38 if (IdxCtx.suppressRefs())
39 IdxCtx.markEntityOccurrenceInFile(PD, Loc);
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +000040 }
41
42 for (unsigned i = 0, e = ProtInfos.size(); i != e; ++i)
43 ProtInfos[i].protocol = &ProtEntities[i];
44
45 for (unsigned i = 0, e = ProtInfos.size(); i != e; ++i)
46 Prots.push_back(&ProtInfos[i]);
47}
48
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +000049
50IBOutletCollectionInfo::IBOutletCollectionInfo(
51 const IBOutletCollectionInfo &other)
52 : AttrInfo(CXIdxAttr_IBOutletCollection, other.cursor, other.loc, A) {
53
54 IBCollInfo.attrInfo = this;
55 IBCollInfo.classCursor = other.IBCollInfo.classCursor;
56 IBCollInfo.classLoc = other.IBCollInfo.classLoc;
57 if (other.IBCollInfo.objcClass) {
58 ClassInfo = other.ClassInfo;
59 IBCollInfo.objcClass = &ClassInfo;
60 } else
61 IBCollInfo.objcClass = 0;
62}
63
64AttrListInfo::AttrListInfo(const Decl *D,
65 IndexingContext &IdxCtx,
66 ScratchAlloc &SA) : ref_cnt(0) {
67 if (!D->hasAttrs())
68 return;
69
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +000070 for (AttrVec::const_iterator AttrI = D->attr_begin(), AttrE = D->attr_end();
71 AttrI != AttrE; ++AttrI) {
72 const Attr *A = *AttrI;
73 CXCursor C = MakeCXCursor(A, const_cast<Decl *>(D), IdxCtx.CXTU);
74 CXIdxLoc Loc = IdxCtx.getIndexLoc(A->getLocation());
75 switch (C.kind) {
76 default:
77 Attrs.push_back(AttrInfo(CXIdxAttr_Unexposed, C, Loc, A));
78 break;
79 case CXCursor_IBActionAttr:
80 Attrs.push_back(AttrInfo(CXIdxAttr_IBAction, C, Loc, A));
81 break;
82 case CXCursor_IBOutletAttr:
83 Attrs.push_back(AttrInfo(CXIdxAttr_IBOutlet, C, Loc, A));
84 break;
85 case CXCursor_IBOutletCollectionAttr:
86 IBCollAttrs.push_back(IBOutletCollectionInfo(C, Loc, A));
87 break;
88 }
89 }
90
91 for (unsigned i = 0, e = IBCollAttrs.size(); i != e; ++i) {
92 IBOutletCollectionInfo &IBInfo = IBCollAttrs[i];
93 CXAttrs.push_back(&IBInfo);
94
95 const IBOutletCollectionAttr *
96 IBAttr = cast<IBOutletCollectionAttr>(IBInfo.A);
97 IBInfo.IBCollInfo.attrInfo = &IBInfo;
98 IBInfo.IBCollInfo.classLoc = IdxCtx.getIndexLoc(IBAttr->getInterfaceLoc());
99 IBInfo.IBCollInfo.objcClass = 0;
100 IBInfo.IBCollInfo.classCursor = clang_getNullCursor();
101 QualType Ty = IBAttr->getInterface();
102 if (const ObjCInterfaceType *InterTy = Ty->getAs<ObjCInterfaceType>()) {
103 if (const ObjCInterfaceDecl *InterD = InterTy->getInterface()) {
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000104 IdxCtx.getEntityInfo(InterD, IBInfo.ClassInfo, SA);
105 IBInfo.IBCollInfo.objcClass = &IBInfo.ClassInfo;
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000106 IBInfo.IBCollInfo.classCursor = MakeCursorObjCClassRef(InterD,
107 IBAttr->getInterfaceLoc(), IdxCtx.CXTU);
108 }
109 }
110 }
111
112 for (unsigned i = 0, e = Attrs.size(); i != e; ++i)
113 CXAttrs.push_back(&Attrs[i]);
114}
115
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000116AttrListInfo::AttrListInfo(const AttrListInfo &other) {
117 assert(other.ref_cnt == 0 &&
118 "Should not copy an AttrListInfo that is ref-counted");
119 ref_cnt = 0;
120
121 Attrs = other.Attrs;
122 IBCollAttrs = other.IBCollAttrs;
123
124 for (unsigned i = 0, e = IBCollAttrs.size(); i != e; ++i)
125 CXAttrs.push_back(&IBCollAttrs[i]);
126
127 for (unsigned i = 0, e = Attrs.size(); i != e; ++i)
128 CXAttrs.push_back(&Attrs[i]);
129}
130
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000131IndexingContext::CXXBasesListInfo::CXXBasesListInfo(const CXXRecordDecl *D,
132 IndexingContext &IdxCtx,
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000133 ScratchAlloc &SA) {
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000134 for (CXXRecordDecl::base_class_const_iterator
135 I = D->bases_begin(), E = D->bases_end(); I != E; ++I) {
136 const CXXBaseSpecifier &Base = *I;
137 BaseEntities.push_back(EntityInfo());
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000138 const NamedDecl *BaseD = 0;
Argyrios Kyrtzidisb526a872011-12-07 20:44:15 +0000139 QualType T = Base.getType();
140 SourceLocation Loc = getBaseLoc(Base);
141
142 if (const TypedefType *TDT = T->getAs<TypedefType>()) {
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000143 BaseD = TDT->getDecl();
Argyrios Kyrtzidisb526a872011-12-07 20:44:15 +0000144 } else if (const TemplateSpecializationType *
145 TST = T->getAs<TemplateSpecializationType>()) {
146 BaseD = TST->getTemplateName().getAsTemplateDecl();
147 } else if (const RecordType *RT = T->getAs<RecordType>()) {
148 BaseD = RT->getDecl();
149 }
150
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000151 if (BaseD)
152 IdxCtx.getEntityInfo(BaseD, BaseEntities.back(), SA);
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000153 CXIdxBaseClassInfo BaseInfo = { 0,
154 MakeCursorCXXBaseSpecifier(&Base, IdxCtx.CXTU),
Argyrios Kyrtzidisb526a872011-12-07 20:44:15 +0000155 IdxCtx.getIndexLoc(Loc) };
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000156 BaseInfos.push_back(BaseInfo);
157 }
158
159 for (unsigned i = 0, e = BaseInfos.size(); i != e; ++i) {
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000160 if (BaseEntities[i].name && BaseEntities[i].USR)
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000161 BaseInfos[i].base = &BaseEntities[i];
162 }
163
164 for (unsigned i = 0, e = BaseInfos.size(); i != e; ++i)
165 CXBases.push_back(&BaseInfos[i]);
166}
167
Argyrios Kyrtzidisb526a872011-12-07 20:44:15 +0000168SourceLocation IndexingContext::CXXBasesListInfo::getBaseLoc(
169 const CXXBaseSpecifier &Base) const {
170 SourceLocation Loc = Base.getSourceRange().getBegin();
171 TypeLoc TL;
172 if (Base.getTypeSourceInfo())
173 TL = Base.getTypeSourceInfo()->getTypeLoc();
174 if (TL.isNull())
175 return Loc;
176
177 if (const QualifiedTypeLoc *QL = dyn_cast<QualifiedTypeLoc>(&TL))
178 TL = QL->getUnqualifiedLoc();
179
180 if (const ElaboratedTypeLoc *EL = dyn_cast<ElaboratedTypeLoc>(&TL))
181 return EL->getNamedTypeLoc().getBeginLoc();
182 if (const DependentNameTypeLoc *DL = dyn_cast<DependentNameTypeLoc>(&TL))
183 return DL->getNameLoc();
184 if (const DependentTemplateSpecializationTypeLoc *
185 DTL = dyn_cast<DependentTemplateSpecializationTypeLoc>(&TL))
186 return DTL->getNameLoc();
187
188 return Loc;
189}
190
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000191const char *ScratchAlloc::toCStr(StringRef Str) {
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000192 if (Str.empty())
193 return "";
194 if (Str.data()[Str.size()] == '\0')
195 return Str.data();
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000196 return copyCStr(Str);
197}
198
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000199const char *ScratchAlloc::copyCStr(StringRef Str) {
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000200 char *buf = IdxCtx.StrScratch.Allocate<char>(Str.size() + 1);
201 std::uninitialized_copy(Str.begin(), Str.end(), buf);
202 buf[Str.size()] = '\0';
203 return buf;
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000204}
205
206void IndexingContext::setASTContext(ASTContext &ctx) {
207 Ctx = &ctx;
208 static_cast<ASTUnit*>(CXTU->TUData)->setASTContext(&ctx);
209}
210
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000211bool IndexingContext::shouldAbort() {
212 if (!CB.abortQuery)
213 return false;
214 return CB.abortQuery(ClientData, 0);
215}
216
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000217void IndexingContext::enteredMainFile(const FileEntry *File) {
218 if (File && CB.enteredMainFile) {
219 CXIdxClientFile idxFile = CB.enteredMainFile(ClientData, (CXFile)File, 0);
220 FileMap[File] = idxFile;
221 }
222}
223
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000224void IndexingContext::ppIncludedFile(SourceLocation hashLoc,
225 StringRef filename,
226 const FileEntry *File,
227 bool isImport, bool isAngled) {
228 if (!CB.ppIncludedFile)
229 return;
230
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000231 ScratchAlloc SA(*this);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000232 CXIdxIncludedFileInfo Info = { getIndexLoc(hashLoc),
233 SA.toCStr(filename),
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000234 (CXFile)File,
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000235 isImport, isAngled };
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000236 CXIdxClientFile idxFile = CB.ppIncludedFile(ClientData, &Info);
237 FileMap[File] = idxFile;
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000238}
239
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000240void IndexingContext::startedTranslationUnit() {
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000241 CXIdxClientContainer idxCont = 0;
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000242 if (CB.startedTranslationUnit)
243 idxCont = CB.startedTranslationUnit(ClientData, 0);
244 addContainerInMap(Ctx->getTranslationUnitDecl(), idxCont);
245}
246
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000247void IndexingContext::handleDiagnosticSet(CXDiagnostic CXDiagSet) {
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000248 if (!CB.diagnostic)
249 return;
250
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000251 CB.diagnostic(ClientData, CXDiagSet, 0);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000252}
253
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000254bool IndexingContext::handleDecl(const NamedDecl *D,
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000255 SourceLocation Loc, CXCursor Cursor,
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000256 DeclInfo &DInfo) {
Argyrios Kyrtzidis144b6c02011-11-18 00:26:46 +0000257 if (!CB.indexDeclaration || !D)
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000258 return false;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000259 if (D->isImplicit() && shouldIgnoreIfImplicit(D))
260 return false;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000261
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000262 ScratchAlloc SA(*this);
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000263 getEntityInfo(D, DInfo.EntInfo, SA);
Argyrios Kyrtzidisdb4d7a52012-01-14 02:05:51 +0000264 if ((!indexFunctionLocalSymbols() && !DInfo.EntInfo.USR)
265 || Loc.isInvalid())
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000266 return false;
Argyrios Kyrtzidis144b6c02011-11-18 00:26:46 +0000267
Argyrios Kyrtzidisc2be04e2011-12-13 18:47:35 +0000268 if (suppressRefs())
269 markEntityOccurrenceInFile(D, Loc);
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000270
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000271 DInfo.entityInfo = &DInfo.EntInfo;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000272 DInfo.cursor = Cursor;
273 DInfo.loc = getIndexLoc(Loc);
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000274 DInfo.isImplicit = D->isImplicit();
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000275
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000276 AttrListInfo AttrList(D, *this, SA);
277 DInfo.attributes = AttrList.getAttrs();
278 DInfo.numAttributes = AttrList.getNumAttrs();
279
Argyrios Kyrtzidisb1febb62011-12-07 20:44:19 +0000280 getContainerInfo(D->getDeclContext(), DInfo.SemanticContainer);
281 getContainerInfo(D->getLexicalDeclContext(), DInfo.LexicalContainer);
282 DInfo.semanticContainer = &DInfo.SemanticContainer;
283 DInfo.lexicalContainer = &DInfo.LexicalContainer;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000284 if (DInfo.isContainer) {
285 getContainerInfo(getEntityContainer(D), DInfo.DeclAsContainer);
286 DInfo.declAsContainer = &DInfo.DeclAsContainer;
287 }
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000288
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000289 CB.indexDeclaration(ClientData, &DInfo);
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000290 return true;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000291}
292
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000293bool IndexingContext::handleObjCContainer(const ObjCContainerDecl *D,
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000294 SourceLocation Loc, CXCursor Cursor,
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000295 ObjCContainerDeclInfo &ContDInfo) {
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000296 ContDInfo.ObjCContDeclInfo.declInfo = &ContDInfo;
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000297 return handleDecl(D, Loc, Cursor, ContDInfo);
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000298}
299
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000300bool IndexingContext::handleFunction(const FunctionDecl *D) {
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000301 DeclInfo DInfo(!D->isFirstDeclaration(), D->isThisDeclarationADefinition(),
302 D->isThisDeclarationADefinition());
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000303 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000304}
305
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000306bool IndexingContext::handleVar(const VarDecl *D) {
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000307 DeclInfo DInfo(!D->isFirstDeclaration(), D->isThisDeclarationADefinition(),
308 /*isContainer=*/false);
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000309 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000310}
311
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000312bool IndexingContext::handleField(const FieldDecl *D) {
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000313 DeclInfo DInfo(/*isRedeclaration=*/false, /*isDefinition=*/true,
314 /*isContainer=*/false);
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000315 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000316}
317
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000318bool IndexingContext::handleEnumerator(const EnumConstantDecl *D) {
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000319 DeclInfo DInfo(/*isRedeclaration=*/false, /*isDefinition=*/true,
320 /*isContainer=*/false);
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000321 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000322}
323
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000324bool IndexingContext::handleTagDecl(const TagDecl *D) {
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000325 if (const CXXRecordDecl *CXXRD = dyn_cast<CXXRecordDecl>(D))
326 return handleCXXRecordDecl(CXXRD, D);
327
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000328 DeclInfo DInfo(!D->isFirstDeclaration(), D->isThisDeclarationADefinition(),
329 D->isThisDeclarationADefinition());
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000330 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000331}
332
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000333bool IndexingContext::handleTypedefName(const TypedefNameDecl *D) {
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000334 DeclInfo DInfo(!D->isFirstDeclaration(), /*isDefinition=*/true,
335 /*isContainer=*/false);
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000336 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000337}
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000338
Douglas Gregor375bb142011-12-27 22:43:10 +0000339bool IndexingContext::handleObjCInterface(const ObjCInterfaceDecl *D) {
Argyrios Kyrtzidisc2be04e2011-12-13 18:47:35 +0000340 // For @class forward declarations, suppress them the same way as references.
Douglas Gregor375bb142011-12-27 22:43:10 +0000341 if (!D->isThisDeclarationADefinition()) {
342 if (suppressRefs() && markEntityOccurrenceInFile(D, D->getLocation()))
Argyrios Kyrtzidisc2be04e2011-12-13 18:47:35 +0000343 return false; // already occurred.
Douglas Gregor375bb142011-12-27 22:43:10 +0000344
345 // FIXME: This seems like the wrong definition for redeclaration.
Douglas Gregoref96ee02012-01-14 16:38:05 +0000346 bool isRedeclaration = D->hasDefinition() || D->getPreviousDecl();
Douglas Gregor375bb142011-12-27 22:43:10 +0000347 ObjCContainerDeclInfo ContDInfo(/*isForwardRef=*/true, isRedeclaration,
348 /*isImplementation=*/false);
349 return handleObjCContainer(D, D->getLocation(),
350 MakeCursorObjCClassRef(D, D->getLocation(),
351 CXTU),
352 ContDInfo);
Argyrios Kyrtzidisc2be04e2011-12-13 18:47:35 +0000353 }
354
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000355 ScratchAlloc SA(*this);
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000356
357 CXIdxBaseClassInfo BaseClass;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000358 EntityInfo BaseEntity;
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000359 BaseClass.cursor = clang_getNullCursor();
360 if (ObjCInterfaceDecl *SuperD = D->getSuperClass()) {
361 getEntityInfo(SuperD, BaseEntity, SA);
362 SourceLocation SuperLoc = D->getSuperClassLoc();
363 BaseClass.base = &BaseEntity;
364 BaseClass.cursor = MakeCursorObjCSuperClassRef(SuperD, SuperLoc, CXTU);
365 BaseClass.loc = getIndexLoc(SuperLoc);
Argyrios Kyrtzidis90fd67a2011-12-15 00:04:56 +0000366
367 if (suppressRefs())
368 markEntityOccurrenceInFile(SuperD, SuperLoc);
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000369 }
370
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000371 ObjCProtocolList EmptyProtoList;
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +0000372 ObjCProtocolListInfo ProtInfo(D->isThisDeclarationADefinition()
373 ? D->getReferencedProtocols()
374 : EmptyProtoList,
Douglas Gregor2e5c15b2011-12-15 05:27:12 +0000375 *this, SA);
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000376
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000377 ObjCInterfaceDeclInfo InterInfo(D);
378 InterInfo.ObjCProtoListInfo = ProtInfo.getListInfo();
379 InterInfo.ObjCInterDeclInfo.containerInfo = &InterInfo.ObjCContDeclInfo;
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000380 InterInfo.ObjCInterDeclInfo.superInfo = D->getSuperClass() ? &BaseClass : 0;
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000381 InterInfo.ObjCInterDeclInfo.protocols = &InterInfo.ObjCProtoListInfo;
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000382
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000383 return handleObjCContainer(D, D->getLocation(), getCursor(D), InterInfo);
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000384}
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000385
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000386bool IndexingContext::handleObjCImplementation(
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000387 const ObjCImplementationDecl *D) {
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000388 ObjCContainerDeclInfo ContDInfo(/*isForwardRef=*/false,
Argyrios Kyrtzidise7bbab92011-11-15 06:20:24 +0000389 /*isRedeclaration=*/true,
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000390 /*isImplementation=*/true);
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000391 return handleObjCContainer(D, D->getLocation(), getCursor(D), ContDInfo);
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000392}
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000393
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000394bool IndexingContext::handleObjCProtocol(const ObjCProtocolDecl *D) {
Douglas Gregorbd9482d2012-01-01 21:23:57 +0000395 if (!D->isThisDeclarationADefinition()) {
396 if (suppressRefs() && markEntityOccurrenceInFile(D, D->getLocation()))
397 return false; // already occurred.
398
399 // FIXME: This seems like the wrong definition for redeclaration.
Douglas Gregoref96ee02012-01-14 16:38:05 +0000400 bool isRedeclaration = D->hasDefinition() || D->getPreviousDecl();
Douglas Gregorbd9482d2012-01-01 21:23:57 +0000401 ObjCContainerDeclInfo ContDInfo(/*isForwardRef=*/true,
402 isRedeclaration,
403 /*isImplementation=*/false);
404 return handleObjCContainer(D, D->getLocation(),
405 MakeCursorObjCProtocolRef(D, D->getLocation(),
406 CXTU),
407 ContDInfo);
408 }
409
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000410 ScratchAlloc SA(*this);
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +0000411 ObjCProtocolList EmptyProtoList;
412 ObjCProtocolListInfo ProtListInfo(D->isThisDeclarationADefinition()
413 ? D->getReferencedProtocols()
414 : EmptyProtoList,
415 *this, SA);
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000416
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000417 ObjCProtocolDeclInfo ProtInfo(D);
418 ProtInfo.ObjCProtoRefListInfo = ProtListInfo.getListInfo();
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000419
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000420 return handleObjCContainer(D, D->getLocation(), getCursor(D), ProtInfo);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000421}
422
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000423bool IndexingContext::handleObjCCategory(const ObjCCategoryDecl *D) {
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000424 ObjCCategoryDeclInfo CatDInfo(/*isImplementation=*/false);
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000425 EntityInfo ClassEntity;
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000426 ScratchAlloc SA(*this);
Argyrios Kyrtzidis21ee5702011-11-15 06:20:16 +0000427 const ObjCInterfaceDecl *IFaceD = D->getClassInterface();
428 SourceLocation ClassLoc = D->getLocation();
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000429 SourceLocation CategoryLoc = D->IsClassExtension() ? ClassLoc
430 : D->getCategoryNameLoc();
Argyrios Kyrtzidisc6b4a502011-11-16 02:34:59 +0000431 getEntityInfo(IFaceD, ClassEntity, SA);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000432
Argyrios Kyrtzidisc2be04e2011-12-13 18:47:35 +0000433 if (suppressRefs())
434 markEntityOccurrenceInFile(IFaceD, ClassLoc);
435
Argyrios Kyrtzidisc10a4c82011-12-13 18:47:45 +0000436 ObjCProtocolListInfo ProtInfo(D->getReferencedProtocols(), *this, SA);
437
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000438 CatDInfo.ObjCCatDeclInfo.containerInfo = &CatDInfo.ObjCContDeclInfo;
Argyrios Kyrtzidisc6b4a502011-11-16 02:34:59 +0000439 if (IFaceD) {
440 CatDInfo.ObjCCatDeclInfo.objcClass = &ClassEntity;
441 CatDInfo.ObjCCatDeclInfo.classCursor =
442 MakeCursorObjCClassRef(IFaceD, ClassLoc, CXTU);
443 } else {
444 CatDInfo.ObjCCatDeclInfo.objcClass = 0;
445 CatDInfo.ObjCCatDeclInfo.classCursor = clang_getNullCursor();
446 }
Argyrios Kyrtzidis21ee5702011-11-15 06:20:16 +0000447 CatDInfo.ObjCCatDeclInfo.classLoc = getIndexLoc(ClassLoc);
Argyrios Kyrtzidisc10a4c82011-12-13 18:47:45 +0000448 CatDInfo.ObjCProtoListInfo = ProtInfo.getListInfo();
449 CatDInfo.ObjCCatDeclInfo.protocols = &CatDInfo.ObjCProtoListInfo;
450
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000451 return handleObjCContainer(D, CategoryLoc, getCursor(D), CatDInfo);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000452}
453
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000454bool IndexingContext::handleObjCCategoryImpl(const ObjCCategoryImplDecl *D) {
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000455 const ObjCCategoryDecl *CatD = D->getCategoryDecl();
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000456 ObjCCategoryDeclInfo CatDInfo(/*isImplementation=*/true);
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000457 EntityInfo ClassEntity;
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000458 ScratchAlloc SA(*this);
Argyrios Kyrtzidis3e340a62011-11-16 02:35:05 +0000459 const ObjCInterfaceDecl *IFaceD = CatD->getClassInterface();
460 SourceLocation ClassLoc = D->getLocation();
Argyrios Kyrtzidisc6994002011-12-09 00:31:40 +0000461 SourceLocation CategoryLoc = D->getCategoryNameLoc();
Argyrios Kyrtzidis3e340a62011-11-16 02:35:05 +0000462 getEntityInfo(IFaceD, ClassEntity, SA);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000463
Argyrios Kyrtzidis6ec43ad2011-11-12 02:16:30 +0000464 CatDInfo.ObjCCatDeclInfo.containerInfo = &CatDInfo.ObjCContDeclInfo;
Argyrios Kyrtzidis3e340a62011-11-16 02:35:05 +0000465 if (IFaceD) {
466 CatDInfo.ObjCCatDeclInfo.objcClass = &ClassEntity;
467 CatDInfo.ObjCCatDeclInfo.classCursor =
468 MakeCursorObjCClassRef(IFaceD, ClassLoc, CXTU);
469 } else {
470 CatDInfo.ObjCCatDeclInfo.objcClass = 0;
471 CatDInfo.ObjCCatDeclInfo.classCursor = clang_getNullCursor();
472 }
473 CatDInfo.ObjCCatDeclInfo.classLoc = getIndexLoc(ClassLoc);
Argyrios Kyrtzidisc10a4c82011-12-13 18:47:45 +0000474 CatDInfo.ObjCCatDeclInfo.protocols = 0;
475
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000476 return handleObjCContainer(D, CategoryLoc, getCursor(D), CatDInfo);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000477}
478
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000479bool IndexingContext::handleObjCMethod(const ObjCMethodDecl *D) {
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000480 DeclInfo DInfo(!D->isCanonicalDecl(), D->isThisDeclarationADefinition(),
481 D->isThisDeclarationADefinition());
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000482 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000483}
484
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000485bool IndexingContext::handleSynthesizedObjCProperty(
486 const ObjCPropertyImplDecl *D) {
487 ObjCPropertyDecl *PD = D->getPropertyDecl();
488 return handleReference(PD, D->getLocation(), getCursor(D), 0, D->getDeclContext());
489}
490
491bool IndexingContext::handleSynthesizedObjCMethod(const ObjCMethodDecl *D,
492 SourceLocation Loc) {
493 DeclInfo DInfo(/*isRedeclaration=*/true, /*isDefinition=*/true,
494 /*isContainer=*/false);
495 return handleDecl(D, Loc, getCursor(D), DInfo);
496}
497
498bool IndexingContext::handleObjCProperty(const ObjCPropertyDecl *D) {
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000499 DeclInfo DInfo(/*isRedeclaration=*/false, /*isDefinition=*/false,
500 /*isContainer=*/false);
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000501 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000502}
503
Argyrios Kyrtzidis68478b02011-12-07 05:52:06 +0000504bool IndexingContext::handleNamespace(const NamespaceDecl *D) {
505 DeclInfo DInfo(/*isRedeclaration=*/!D->isOriginalNamespace(),
506 /*isDefinition=*/true,
507 /*isContainer=*/true);
508 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
509}
510
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000511bool IndexingContext::handleClassTemplate(const ClassTemplateDecl *D) {
512 return handleCXXRecordDecl(D->getTemplatedDecl(), D);
513}
514
515bool IndexingContext::handleFunctionTemplate(const FunctionTemplateDecl *D) {
516 DeclInfo DInfo(/*isRedeclaration=*/!D->isCanonicalDecl(),
517 /*isDefinition=*/D->isThisDeclarationADefinition(),
518 /*isContainer=*/D->isThisDeclarationADefinition());
519 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
520}
521
522bool IndexingContext::handleTypeAliasTemplate(const TypeAliasTemplateDecl *D) {
523 DeclInfo DInfo(/*isRedeclaration=*/!D->isCanonicalDecl(),
524 /*isDefinition=*/true, /*isContainer=*/false);
525 return handleDecl(D, D->getLocation(), getCursor(D), DInfo);
526}
527
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000528bool IndexingContext::handleReference(const NamedDecl *D, SourceLocation Loc,
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000529 const NamedDecl *Parent,
530 const DeclContext *DC,
Argyrios Kyrtzidisaca19be2011-10-18 15:50:50 +0000531 const Expr *E,
532 CXIdxEntityRefKind Kind) {
Argyrios Kyrtzidisd6c82092011-11-16 02:35:01 +0000533 if (!D)
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000534 return false;
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000535
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000536 CXCursor Cursor = E ? MakeCXCursor(const_cast<Expr*>(E),
537 const_cast<Decl*>(cast<Decl>(DC)), CXTU)
538 : getRefCursor(D, Loc);
539 return handleReference(D, Loc, Cursor, Parent, DC, E, Kind);
540}
541
542bool IndexingContext::handleReference(const NamedDecl *D, SourceLocation Loc,
543 CXCursor Cursor,
544 const NamedDecl *Parent,
545 const DeclContext *DC,
546 const Expr *E,
547 CXIdxEntityRefKind Kind) {
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000548 if (!CB.indexEntityReference)
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000549 return false;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000550
551 if (!D)
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000552 return false;
553 if (Loc.isInvalid())
554 return false;
Argyrios Kyrtzidis22490742012-01-14 00:11:49 +0000555 if (!indexFunctionLocalSymbols() && D->getParentFunctionOrMethod())
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000556 return false;
557 if (isNotFromSourceFile(D->getLocation()))
558 return false;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000559 if (D->isImplicit() && shouldIgnoreIfImplicit(D))
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000560 return false;
Argyrios Kyrtzidis144b6c02011-11-18 00:26:46 +0000561
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000562 if (suppressRefs()) {
563 if (markEntityOccurrenceInFile(D, Loc))
564 return false; // already occurred.
Argyrios Kyrtzidisc6b4a502011-11-16 02:34:59 +0000565 }
566
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000567 ScratchAlloc SA(*this);
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000568 EntityInfo RefEntity, ParentEntity;
569 getEntityInfo(D, RefEntity, SA);
570 if (!RefEntity.USR)
571 return false;
572
573 getEntityInfo(Parent, ParentEntity, SA);
574
575 ContainerInfo Container;
576 getContainerInfo(DC, Container);
577
Argyrios Kyrtzidisb1febb62011-12-07 20:44:19 +0000578 CXIdxEntityRefInfo Info = { Kind,
579 Cursor,
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000580 getIndexLoc(Loc),
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000581 &RefEntity,
Argyrios Kyrtzidisc6b4a502011-11-16 02:34:59 +0000582 Parent ? &ParentEntity : 0,
Argyrios Kyrtzidisb1febb62011-12-07 20:44:19 +0000583 &Container };
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000584 CB.indexEntityReference(ClientData, &Info);
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000585 return true;
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000586}
587
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000588bool IndexingContext::isNotFromSourceFile(SourceLocation Loc) const {
589 if (Loc.isInvalid())
590 return true;
591 SourceManager &SM = Ctx->getSourceManager();
592 SourceLocation FileLoc = SM.getFileLoc(Loc);
593 FileID FID = SM.getFileID(FileLoc);
594 return SM.getFileEntryForID(FID) == 0;
595}
596
597void IndexingContext::addContainerInMap(const DeclContext *DC,
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000598 CXIdxClientContainer container) {
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000599 if (!DC)
600 return;
601
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000602 ContainerMapTy::iterator I = ContainerMap.find(DC);
603 if (I == ContainerMap.end()) {
604 if (container)
605 ContainerMap[DC] = container;
606 return;
607 }
608 // Allow changing the container of a previously seen DeclContext so we
609 // can handle invalid user code, like a function re-definition.
610 if (container)
611 I->second = container;
612 else
613 ContainerMap.erase(I);
614}
615
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000616CXIdxClientEntity IndexingContext::getClientEntity(const Decl *D) const {
617 if (!D)
618 return 0;
619 EntityMapTy::const_iterator I = EntityMap.find(D);
620 if (I == EntityMap.end())
621 return 0;
622 return I->second;
623}
624
625void IndexingContext::setClientEntity(const Decl *D, CXIdxClientEntity client) {
626 if (!D)
627 return;
628 EntityMap[D] = client;
629}
630
631bool IndexingContext::handleCXXRecordDecl(const CXXRecordDecl *RD,
632 const NamedDecl *OrigD) {
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000633 if (RD->isThisDeclarationADefinition()) {
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000634 ScratchAlloc SA(*this);
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000635 CXXClassDeclInfo CXXDInfo(/*isRedeclaration=*/!OrigD->isCanonicalDecl(),
636 /*isDefinition=*/RD->isThisDeclarationADefinition());
637 CXXBasesListInfo BaseList(RD, *this, SA);
638 CXXDInfo.CXXClassInfo.declInfo = &CXXDInfo;
639 CXXDInfo.CXXClassInfo.bases = BaseList.getBases();
640 CXXDInfo.CXXClassInfo.numBases = BaseList.getNumBases();
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000641
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000642 return handleDecl(OrigD, OrigD->getLocation(), getCursor(OrigD), CXXDInfo);
643 }
644
645 DeclInfo DInfo(/*isRedeclaration=*/!OrigD->isCanonicalDecl(),
646 /*isDefinition=*/RD->isThisDeclarationADefinition(),
647 /*isContainer=*/RD->isThisDeclarationADefinition());
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000648 return handleDecl(OrigD, OrigD->getLocation(), getCursor(OrigD), DInfo);
649}
650
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000651bool IndexingContext::markEntityOccurrenceInFile(const NamedDecl *D,
652 SourceLocation Loc) {
Argyrios Kyrtzidisc2be04e2011-12-13 18:47:35 +0000653 if (!D || Loc.isInvalid())
654 return true;
655
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000656 SourceManager &SM = Ctx->getSourceManager();
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000657 D = getEntityDecl(D);
658
Argyrios Kyrtzidisc2be04e2011-12-13 18:47:35 +0000659 std::pair<FileID, unsigned> LocInfo = SM.getDecomposedLoc(SM.getFileLoc(Loc));
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000660 FileID FID = LocInfo.first;
661 if (FID.isInvalid())
662 return true;
663
664 const FileEntry *FE = SM.getFileEntryForID(FID);
665 if (!FE)
666 return true;
667 RefFileOccurence RefOccur(FE, D);
668 std::pair<llvm::DenseSet<RefFileOccurence>::iterator, bool>
669 res = RefFileOccurences.insert(RefOccur);
670 if (!res.second)
671 return true; // already in map.
672
673 return false;
674}
675
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000676const NamedDecl *IndexingContext::getEntityDecl(const NamedDecl *D) const {
677 assert(D);
678 D = cast<NamedDecl>(D->getCanonicalDecl());
679
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000680 if (const ObjCImplementationDecl *
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000681 ImplD = dyn_cast<ObjCImplementationDecl>(D)) {
682 return getEntityDecl(ImplD->getClassInterface());
683
684 } else if (const ObjCCategoryImplDecl *
685 CatImplD = dyn_cast<ObjCCategoryImplDecl>(D)) {
686 return getEntityDecl(CatImplD->getCategoryDecl());
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000687 } else if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
688 if (FunctionTemplateDecl *TemplD = FD->getDescribedFunctionTemplate())
689 return getEntityDecl(TemplD);
690 } else if (const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D)) {
691 if (ClassTemplateDecl *TemplD = RD->getDescribedClassTemplate())
692 return getEntityDecl(TemplD);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000693 }
694
695 return D;
696}
697
698const DeclContext *
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000699IndexingContext::getEntityContainer(const Decl *D) const {
700 const DeclContext *DC = dyn_cast<DeclContext>(D);
701 if (DC)
702 return DC;
703
704 if (const ClassTemplateDecl *ClassTempl = dyn_cast<ClassTemplateDecl>(D)) {
705 DC = ClassTempl->getTemplatedDecl();
706 } if (const FunctionTemplateDecl *
707 FuncTempl = dyn_cast<FunctionTemplateDecl>(D)) {
708 DC = FuncTempl->getTemplatedDecl();
709 }
710
711 return DC;
712}
713
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000714CXIdxClientContainer
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000715IndexingContext::getClientContainerForDC(const DeclContext *DC) const {
716 if (!DC)
717 return 0;
718
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000719 ContainerMapTy::const_iterator I = ContainerMap.find(DC);
Argyrios Kyrtzidis3e340a62011-11-16 02:35:05 +0000720 if (I == ContainerMap.end())
721 return 0;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000722
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000723 return I->second;
724}
725
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000726CXIdxClientFile IndexingContext::getIndexFile(const FileEntry *File) {
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000727 if (!File)
728 return 0;
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000729
730 FileMapTy::iterator FI = FileMap.find(File);
731 if (FI != FileMap.end())
732 return FI->second;
733
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000734 return 0;
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000735}
736
737CXIdxLoc IndexingContext::getIndexLoc(SourceLocation Loc) const {
738 CXIdxLoc idxLoc = { {0, 0}, 0 };
739 if (Loc.isInvalid())
740 return idxLoc;
741
742 idxLoc.ptr_data[0] = (void*)this;
743 idxLoc.int_data = Loc.getRawEncoding();
744 return idxLoc;
745}
746
747void IndexingContext::translateLoc(SourceLocation Loc,
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000748 CXIdxClientFile *indexFile, CXFile *file,
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000749 unsigned *line, unsigned *column,
750 unsigned *offset) {
751 if (Loc.isInvalid())
752 return;
753
754 SourceManager &SM = Ctx->getSourceManager();
755 Loc = SM.getFileLoc(Loc);
756
757 std::pair<FileID, unsigned> LocInfo = SM.getDecomposedLoc(Loc);
758 FileID FID = LocInfo.first;
759 unsigned FileOffset = LocInfo.second;
760
761 if (FID.isInvalid())
762 return;
763
764 const FileEntry *FE = SM.getFileEntryForID(FID);
765 if (indexFile)
766 *indexFile = getIndexFile(FE);
767 if (file)
768 *file = (void *)FE;
769 if (line)
770 *line = SM.getLineNumber(FID, FileOffset);
771 if (column)
772 *column = SM.getColumnNumber(FID, FileOffset);
773 if (offset)
774 *offset = FileOffset;
775}
776
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000777void IndexingContext::getEntityInfo(const NamedDecl *D,
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000778 EntityInfo &EntityInfo,
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000779 ScratchAlloc &SA) {
Argyrios Kyrtzidisc6b4a502011-11-16 02:34:59 +0000780 if (!D)
781 return;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000782
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000783 D = getEntityDecl(D);
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000784 EntityInfo.cursor = getCursor(D);
785 EntityInfo.Dcl = D;
786 EntityInfo.IndexCtx = this;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000787 EntityInfo.kind = CXIdxEntity_Unexposed;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000788 EntityInfo.templateKind = CXIdxEntity_NonTemplate;
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000789 EntityInfo.lang = CXIdxEntityLang_C;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000790
Argyrios Kyrtzidis643d3ce2011-12-15 00:05:00 +0000791 if (D->hasAttrs()) {
792 AttrListInfo *attrs = SA.allocate<AttrListInfo>();
793 new (attrs) AttrListInfo(D, *this, SA);
794 EntityInfo.AttrList = attrs;
795 EntityInfo.attributes = attrs->getAttrs();
796 EntityInfo.numAttributes = attrs->getNumAttrs();
797 }
798
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000799 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
800 switch (TD->getTagKind()) {
801 case TTK_Struct:
802 EntityInfo.kind = CXIdxEntity_Struct; break;
803 case TTK_Union:
804 EntityInfo.kind = CXIdxEntity_Union; break;
805 case TTK_Class:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000806 EntityInfo.kind = CXIdxEntity_CXXClass;
807 EntityInfo.lang = CXIdxEntityLang_CXX;
808 break;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000809 case TTK_Enum:
810 EntityInfo.kind = CXIdxEntity_Enum; break;
811 }
812
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000813 if (const CXXRecordDecl *CXXRec = dyn_cast<CXXRecordDecl>(D)) {
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000814 // FIXME: isPOD check is not sufficient, a POD can contain methods,
815 // we want a isCStructLike check.
816 if (CXXRec->hasDefinition() && !CXXRec->isPOD())
817 EntityInfo.lang = CXIdxEntityLang_CXX;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000818 }
819
820 if (isa<ClassTemplatePartialSpecializationDecl>(D)) {
821 EntityInfo.templateKind = CXIdxEntity_TemplatePartialSpecialization;
822 } else if (isa<ClassTemplateSpecializationDecl>(D)) {
823 EntityInfo.templateKind = CXIdxEntity_TemplateSpecialization;
824 }
825
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000826 } else {
827 switch (D->getKind()) {
828 case Decl::Typedef:
829 EntityInfo.kind = CXIdxEntity_Typedef; break;
830 case Decl::Function:
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000831 EntityInfo.kind = CXIdxEntity_Function;
832 break;
Argyrios Kyrtzidisdb4d7a52012-01-14 02:05:51 +0000833 case Decl::ParmVar:
834 EntityInfo.kind = CXIdxEntity_Variable;
835 break;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000836 case Decl::Var:
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000837 EntityInfo.kind = CXIdxEntity_Variable;
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000838 if (isa<CXXRecordDecl>(D->getDeclContext())) {
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000839 EntityInfo.kind = CXIdxEntity_CXXStaticVariable;
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000840 EntityInfo.lang = CXIdxEntityLang_CXX;
841 }
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000842 break;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000843 case Decl::Field:
Argyrios Kyrtzidisda5f2d22011-12-05 22:05:28 +0000844 EntityInfo.kind = CXIdxEntity_Field;
845 if (const CXXRecordDecl *
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000846 CXXRec = dyn_cast<CXXRecordDecl>(D->getDeclContext())) {
847 // FIXME: isPOD check is not sufficient, a POD can contain methods,
848 // we want a isCStructLike check.
Argyrios Kyrtzidisda5f2d22011-12-05 22:05:28 +0000849 if (!CXXRec->isPOD())
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000850 EntityInfo.lang = CXIdxEntityLang_CXX;
851 }
Argyrios Kyrtzidisda5f2d22011-12-05 22:05:28 +0000852 break;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000853 case Decl::EnumConstant:
854 EntityInfo.kind = CXIdxEntity_EnumConstant; break;
855 case Decl::ObjCInterface:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000856 EntityInfo.kind = CXIdxEntity_ObjCClass;
857 EntityInfo.lang = CXIdxEntityLang_ObjC;
858 break;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000859 case Decl::ObjCProtocol:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000860 EntityInfo.kind = CXIdxEntity_ObjCProtocol;
861 EntityInfo.lang = CXIdxEntityLang_ObjC;
862 break;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000863 case Decl::ObjCCategory:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000864 EntityInfo.kind = CXIdxEntity_ObjCCategory;
865 EntityInfo.lang = CXIdxEntityLang_ObjC;
866 break;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000867 case Decl::ObjCMethod:
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000868 if (cast<ObjCMethodDecl>(D)->isInstanceMethod())
869 EntityInfo.kind = CXIdxEntity_ObjCInstanceMethod;
870 else
871 EntityInfo.kind = CXIdxEntity_ObjCClassMethod;
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000872 EntityInfo.lang = CXIdxEntityLang_ObjC;
Argyrios Kyrtzidisc71d5542011-11-14 22:39:19 +0000873 break;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000874 case Decl::ObjCProperty:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000875 EntityInfo.kind = CXIdxEntity_ObjCProperty;
876 EntityInfo.lang = CXIdxEntityLang_ObjC;
877 break;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000878 case Decl::ObjCIvar:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000879 EntityInfo.kind = CXIdxEntity_ObjCIvar;
880 EntityInfo.lang = CXIdxEntityLang_ObjC;
881 break;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000882 case Decl::Namespace:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000883 EntityInfo.kind = CXIdxEntity_CXXNamespace;
884 EntityInfo.lang = CXIdxEntityLang_CXX;
885 break;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000886 case Decl::NamespaceAlias:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000887 EntityInfo.kind = CXIdxEntity_CXXNamespaceAlias;
888 EntityInfo.lang = CXIdxEntityLang_CXX;
889 break;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000890 case Decl::CXXConstructor:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000891 EntityInfo.kind = CXIdxEntity_CXXConstructor;
892 EntityInfo.lang = CXIdxEntityLang_CXX;
893 break;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000894 case Decl::CXXDestructor:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000895 EntityInfo.kind = CXIdxEntity_CXXDestructor;
896 EntityInfo.lang = CXIdxEntityLang_CXX;
897 break;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000898 case Decl::CXXConversion:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000899 EntityInfo.kind = CXIdxEntity_CXXConversionFunction;
900 EntityInfo.lang = CXIdxEntityLang_CXX;
901 break;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000902 case Decl::CXXMethod: {
903 const CXXMethodDecl *MD = cast<CXXMethodDecl>(D);
904 if (MD->isStatic())
905 EntityInfo.kind = CXIdxEntity_CXXStaticMethod;
906 else
907 EntityInfo.kind = CXIdxEntity_CXXInstanceMethod;
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000908 EntityInfo.lang = CXIdxEntityLang_CXX;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000909 break;
910 }
911 case Decl::ClassTemplate:
912 EntityInfo.kind = CXIdxEntity_CXXClass;
913 EntityInfo.templateKind = CXIdxEntity_Template;
914 break;
915 case Decl::FunctionTemplate:
916 EntityInfo.kind = CXIdxEntity_Function;
917 EntityInfo.templateKind = CXIdxEntity_Template;
918 if (const CXXMethodDecl *MD = dyn_cast_or_null<CXXMethodDecl>(
919 cast<FunctionTemplateDecl>(D)->getTemplatedDecl())) {
920 if (isa<CXXConstructorDecl>(MD))
921 EntityInfo.kind = CXIdxEntity_CXXConstructor;
922 else if (isa<CXXDestructorDecl>(MD))
923 EntityInfo.kind = CXIdxEntity_CXXDestructor;
924 else if (isa<CXXConversionDecl>(MD))
925 EntityInfo.kind = CXIdxEntity_CXXConversionFunction;
926 else {
927 if (MD->isStatic())
928 EntityInfo.kind = CXIdxEntity_CXXStaticMethod;
929 else
930 EntityInfo.kind = CXIdxEntity_CXXInstanceMethod;
931 }
932 }
933 break;
934 case Decl::TypeAliasTemplate:
935 EntityInfo.kind = CXIdxEntity_CXXTypeAlias;
936 EntityInfo.templateKind = CXIdxEntity_Template;
937 break;
938 case Decl::TypeAlias:
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000939 EntityInfo.kind = CXIdxEntity_CXXTypeAlias;
940 EntityInfo.lang = CXIdxEntityLang_CXX;
941 break;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000942 default:
943 break;
944 }
945 }
946
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000947 if (EntityInfo.kind == CXIdxEntity_Unexposed)
948 return;
949
950 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
951 if (FD->getTemplatedKind() ==
952 FunctionDecl::TK_FunctionTemplateSpecialization)
953 EntityInfo.templateKind = CXIdxEntity_TemplateSpecialization;
954 }
955
Argyrios Kyrtzidis838d3c22011-12-07 20:44:12 +0000956 if (EntityInfo.templateKind != CXIdxEntity_NonTemplate)
957 EntityInfo.lang = CXIdxEntityLang_CXX;
958
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000959 if (IdentifierInfo *II = D->getIdentifier()) {
960 EntityInfo.name = SA.toCStr(II->getName());
961
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000962 } else if (isa<TagDecl>(D) || isa<FieldDecl>(D) || isa<NamespaceDecl>(D)) {
963 EntityInfo.name = 0; // anonymous tag/field/namespace.
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000964
965 } else {
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000966 llvm::SmallString<256> StrBuf;
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000967 {
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000968 llvm::raw_svector_ostream OS(StrBuf);
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000969 D->printName(OS);
970 }
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000971 EntityInfo.name = SA.copyCStr(StrBuf.str());
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000972 }
973
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000974 {
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000975 llvm::SmallString<512> StrBuf;
976 bool Ignore = getDeclCursorUSR(D, StrBuf);
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000977 if (Ignore) {
Argyrios Kyrtzidis144b6c02011-11-18 00:26:46 +0000978 EntityInfo.USR = 0;
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000979 } else {
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +0000980 EntityInfo.USR = SA.copyCStr(StrBuf.str());
Argyrios Kyrtzidisdd93c592011-11-11 00:23:36 +0000981 }
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000982 }
983}
984
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000985void IndexingContext::getContainerInfo(const DeclContext *DC,
986 ContainerInfo &ContInfo) {
987 ContInfo.cursor = getCursor(cast<Decl>(DC));
988 ContInfo.DC = DC;
989 ContInfo.IndexCtx = this;
990}
991
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000992CXCursor IndexingContext::getRefCursor(const NamedDecl *D, SourceLocation Loc) {
993 if (const TypeDecl *TD = dyn_cast<TypeDecl>(D))
994 return MakeCursorTypeRef(TD, Loc, CXTU);
995 if (const ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(D))
996 return MakeCursorObjCClassRef(ID, Loc, CXTU);
997 if (const ObjCProtocolDecl *PD = dyn_cast<ObjCProtocolDecl>(D))
998 return MakeCursorObjCProtocolRef(PD, Loc, CXTU);
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000999 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(D))
1000 return MakeCursorTemplateRef(Template, Loc, CXTU);
1001 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(D))
1002 return MakeCursorNamespaceRef(Namespace, Loc, CXTU);
1003 if (const NamespaceAliasDecl *Namespace = dyn_cast<NamespaceAliasDecl>(D))
1004 return MakeCursorNamespaceRef(Namespace, Loc, CXTU);
1005 if (const FieldDecl *Field = dyn_cast<FieldDecl>(D))
1006 return MakeCursorMemberRef(Field, Loc, CXTU);
1007
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +00001008 return clang_getNullCursor();
1009}
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +00001010
1011bool IndexingContext::shouldIgnoreIfImplicit(const NamedDecl *D) {
Argyrios Kyrtzidis37f40572011-11-23 20:27:26 +00001012 if (isa<ObjCInterfaceDecl>(D))
1013 return false;
1014 if (isa<ObjCCategoryDecl>(D))
1015 return false;
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +00001016 if (isa<ObjCIvarDecl>(D))
1017 return false;
1018 if (isa<ObjCMethodDecl>(D))
1019 return false;
1020 return true;
1021}