blob: c344b41cecc1d459be85aaddf37ee869a2f774bd [file] [log] [blame]
Ted Kremenek16c440a2010-01-15 20:35:54 +00001//===- CXCursor.cpp - Routines for manipulating CXCursors -----------------===//
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//
Douglas Gregor2e331b92010-01-16 14:00:32 +000010// This file defines routines for manipulating CXCursors. It should be the
11// only file that has internal knowledge of the encoding of the data in
12// CXCursor.
Ted Kremenek16c440a2010-01-15 20:35:54 +000013//
14//===----------------------------------------------------------------------===//
15
Ted Kremenek0a90d322010-11-17 23:24:11 +000016#include "CXTranslationUnit.h"
Ted Kremenek16c440a2010-01-15 20:35:54 +000017#include "CXCursor.h"
Ted Kremeneked122732010-11-16 01:56:27 +000018#include "CXString.h"
Douglas Gregor7eaa8ae2010-01-20 00:23:15 +000019#include "clang/Frontend/ASTUnit.h"
Ted Kremenek16c440a2010-01-15 20:35:54 +000020#include "clang/AST/Decl.h"
Douglas Gregor69319002010-08-31 23:48:11 +000021#include "clang/AST/DeclCXX.h"
Douglas Gregor283cae32010-01-15 21:56:13 +000022#include "clang/AST/DeclObjC.h"
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +000023#include "clang/AST/DeclTemplate.h"
Douglas Gregor283cae32010-01-15 21:56:13 +000024#include "clang/AST/Expr.h"
Douglas Gregor1f60d9e2010-09-13 22:52:57 +000025#include "clang/AST/ExprCXX.h"
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +000026#include "clang/AST/ExprObjC.h"
Ted Kremenek007a7c92010-11-01 23:26:51 +000027#include "clang-c/Index.h"
Ted Kremenekedc8aa62010-01-16 00:36:30 +000028#include "llvm/Support/ErrorHandling.h"
Ted Kremenek16c440a2010-01-15 20:35:54 +000029
30using namespace clang;
Douglas Gregor1f60d9e2010-09-13 22:52:57 +000031using namespace cxcursor;
Ted Kremenek16c440a2010-01-15 20:35:54 +000032
Ted Kremenekbbf66ca2012-04-30 19:06:49 +000033CXCursor cxcursor::MakeCXCursorInvalid(CXCursorKind K, CXTranslationUnit TU) {
Douglas Gregor5bfb8c12010-01-20 23:34:41 +000034 assert(K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid);
Ted Kremenekbbf66ca2012-04-30 19:06:49 +000035 CXCursor C = { K, 0, { 0, 0, TU } };
Douglas Gregor5bfb8c12010-01-20 23:34:41 +000036 return C;
Ted Kremenek16c440a2010-01-15 20:35:54 +000037}
38
Ted Kremeneke77f4432010-02-18 03:09:07 +000039static CXCursorKind GetCursorKind(const Attr *A) {
40 assert(A && "Invalid arguments!");
41 switch (A->getKind()) {
42 default: break;
Sean Hunt387475d2010-06-16 23:43:53 +000043 case attr::IBAction: return CXCursor_IBActionAttr;
44 case attr::IBOutlet: return CXCursor_IBOutletAttr;
45 case attr::IBOutletCollection: return CXCursor_IBOutletCollectionAttr;
Argyrios Kyrtzidis6639e922011-09-13 17:39:31 +000046 case attr::Final: return CXCursor_CXXFinalAttr;
47 case attr::Override: return CXCursor_CXXOverrideAttr;
Erik Verbruggen5f1c8222011-10-13 09:41:32 +000048 case attr::Annotate: return CXCursor_AnnotateAttr;
Argyrios Kyrtzidis84b79642011-12-06 22:05:01 +000049 case attr::AsmLabel: return CXCursor_AsmLabelAttr;
Ted Kremeneke77f4432010-02-18 03:09:07 +000050 }
51
52 return CXCursor_UnexposedAttr;
53}
54
Ted Kremeneka60ed472010-11-16 08:15:36 +000055CXCursor cxcursor::MakeCXCursor(const Attr *A, Decl *Parent,
56 CXTranslationUnit TU) {
Ted Kremeneke77f4432010-02-18 03:09:07 +000057 assert(A && Parent && TU && "Invalid arguments!");
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +000058 CXCursor C = { GetCursorKind(A), 0, { Parent, (void*)A, TU } };
Ted Kremeneke77f4432010-02-18 03:09:07 +000059 return C;
60}
61
Ted Kremeneka60ed472010-11-16 08:15:36 +000062CXCursor cxcursor::MakeCXCursor(Decl *D, CXTranslationUnit TU,
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +000063 SourceRange RegionOfInterest,
Ted Kremenek007a7c92010-11-01 23:26:51 +000064 bool FirstInDeclGroup) {
Daniel Dunbar54d67ca2010-01-25 00:40:30 +000065 assert(D && TU && "Invalid arguments!");
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +000066
67 CXCursorKind K = getCursorKindForDecl(D);
68
69 if (K == CXCursor_ObjCClassMethodDecl ||
70 K == CXCursor_ObjCInstanceMethodDecl) {
71 int SelectorIdIndex = -1;
72 // Check if cursor points to a selector id.
73 if (RegionOfInterest.isValid() &&
74 RegionOfInterest.getBegin() == RegionOfInterest.getEnd()) {
75 SmallVector<SourceLocation, 16> SelLocs;
76 cast<ObjCMethodDecl>(D)->getSelectorLocs(SelLocs);
77 SmallVector<SourceLocation, 16>::iterator
78 I=std::find(SelLocs.begin(), SelLocs.end(),RegionOfInterest.getBegin());
79 if (I != SelLocs.end())
80 SelectorIdIndex = I - SelLocs.begin();
81 }
82 CXCursor C = { K, SelectorIdIndex,
83 { D, (void*)(intptr_t) (FirstInDeclGroup ? 1 : 0), TU }};
84 return C;
85 }
86
87 CXCursor C = { K, 0, { D, (void*)(intptr_t) (FirstInDeclGroup ? 1 : 0), TU }};
Douglas Gregor5bfb8c12010-01-20 23:34:41 +000088 return C;
Ted Kremenekedc8aa62010-01-16 00:36:30 +000089}
90
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +000091CXCursor cxcursor::MakeCXCursor(Stmt *S, Decl *Parent, CXTranslationUnit TU,
92 SourceRange RegionOfInterest) {
Daniel Dunbar54d67ca2010-01-25 00:40:30 +000093 assert(S && TU && "Invalid arguments!");
Douglas Gregor97b98722010-01-19 23:20:36 +000094 CXCursorKind K = CXCursor_NotImplemented;
95
96 switch (S->getStmtClass()) {
97 case Stmt::NoStmtClass:
98 break;
Douglas Gregor42b29842011-10-05 19:00:14 +000099
Douglas Gregor97b98722010-01-19 23:20:36 +0000100 case Stmt::CaseStmtClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000101 K = CXCursor_CaseStmt;
102 break;
103
Douglas Gregor97b98722010-01-19 23:20:36 +0000104 case Stmt::DefaultStmtClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000105 K = CXCursor_DefaultStmt;
106 break;
107
108 case Stmt::IfStmtClass:
109 K = CXCursor_IfStmt;
110 break;
111
112 case Stmt::SwitchStmtClass:
113 K = CXCursor_SwitchStmt;
114 break;
115
116 case Stmt::WhileStmtClass:
117 K = CXCursor_WhileStmt;
118 break;
119
120 case Stmt::DoStmtClass:
121 K = CXCursor_DoStmt;
122 break;
123
124 case Stmt::ForStmtClass:
125 K = CXCursor_ForStmt;
126 break;
127
128 case Stmt::GotoStmtClass:
129 K = CXCursor_GotoStmt;
130 break;
131
Douglas Gregor97b98722010-01-19 23:20:36 +0000132 case Stmt::IndirectGotoStmtClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000133 K = CXCursor_IndirectGotoStmt;
134 break;
135
136 case Stmt::ContinueStmtClass:
137 K = CXCursor_ContinueStmt;
138 break;
139
140 case Stmt::BreakStmtClass:
141 K = CXCursor_BreakStmt;
142 break;
143
144 case Stmt::ReturnStmtClass:
145 K = CXCursor_ReturnStmt;
146 break;
147
148 case Stmt::AsmStmtClass:
149 K = CXCursor_AsmStmt;
150 break;
151
152 case Stmt::ObjCAtTryStmtClass:
153 K = CXCursor_ObjCAtTryStmt;
154 break;
155
156 case Stmt::ObjCAtCatchStmtClass:
157 K = CXCursor_ObjCAtCatchStmt;
158 break;
159
160 case Stmt::ObjCAtFinallyStmtClass:
161 K = CXCursor_ObjCAtFinallyStmt;
162 break;
163
164 case Stmt::ObjCAtThrowStmtClass:
165 K = CXCursor_ObjCAtThrowStmt;
166 break;
167
168 case Stmt::ObjCAtSynchronizedStmtClass:
169 K = CXCursor_ObjCAtSynchronizedStmt;
170 break;
171
172 case Stmt::ObjCAutoreleasePoolStmtClass:
173 K = CXCursor_ObjCAutoreleasePoolStmt;
174 break;
175
Douglas Gregor97b98722010-01-19 23:20:36 +0000176 case Stmt::ObjCForCollectionStmtClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000177 K = CXCursor_ObjCForCollectionStmt;
178 break;
179
Douglas Gregor97b98722010-01-19 23:20:36 +0000180 case Stmt::CXXCatchStmtClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000181 K = CXCursor_CXXCatchStmt;
182 break;
183
184 case Stmt::CXXTryStmtClass:
185 K = CXCursor_CXXTryStmt;
186 break;
187
188 case Stmt::CXXForRangeStmtClass:
189 K = CXCursor_CXXForRangeStmt;
190 break;
191
John Wiegley28bbe4b2011-04-28 01:08:34 +0000192 case Stmt::SEHTryStmtClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000193 K = CXCursor_SEHTryStmt;
194 break;
195
John Wiegley28bbe4b2011-04-28 01:08:34 +0000196 case Stmt::SEHExceptStmtClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000197 K = CXCursor_SEHExceptStmt;
198 break;
199
John Wiegley28bbe4b2011-04-28 01:08:34 +0000200 case Stmt::SEHFinallyStmtClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000201 K = CXCursor_SEHFinallyStmt;
Douglas Gregor97b98722010-01-19 23:20:36 +0000202 break;
Douglas Gregor42b29842011-10-05 19:00:14 +0000203
John Wiegley21ff2e52011-04-28 00:16:57 +0000204 case Stmt::ArrayTypeTraitExprClass:
Tanya Lattner61eee0c2011-06-04 00:47:47 +0000205 case Stmt::AsTypeExprClass:
Eli Friedman276b0612011-10-11 02:20:01 +0000206 case Stmt::AtomicExprClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000207 case Stmt::BinaryConditionalOperatorClass:
208 case Stmt::BinaryTypeTraitExprClass:
Douglas Gregor4ca8ac22012-02-24 07:38:34 +0000209 case Stmt::TypeTraitExprClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000210 case Stmt::CXXBindTemporaryExprClass:
211 case Stmt::CXXDefaultArgExprClass:
212 case Stmt::CXXScalarValueInitExprClass:
213 case Stmt::CXXUuidofExprClass:
214 case Stmt::ChooseExprClass:
215 case Stmt::DesignatedInitExprClass:
216 case Stmt::ExprWithCleanupsClass:
217 case Stmt::ExpressionTraitExprClass:
218 case Stmt::ExtVectorElementExprClass:
219 case Stmt::ImplicitCastExprClass:
220 case Stmt::ImplicitValueInitExprClass:
221 case Stmt::MaterializeTemporaryExprClass:
222 case Stmt::ObjCIndirectCopyRestoreExprClass:
223 case Stmt::OffsetOfExprClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000224 case Stmt::ParenListExprClass:
225 case Stmt::PredefinedExprClass:
226 case Stmt::ShuffleVectorExprClass:
227 case Stmt::UnaryExprOrTypeTraitExprClass:
228 case Stmt::UnaryTypeTraitExprClass:
229 case Stmt::VAArgExprClass:
Ted Kremenekb3f75422012-03-06 20:06:06 +0000230 case Stmt::ObjCArrayLiteralClass:
231 case Stmt::ObjCDictionaryLiteralClass:
Patrick Beardeb382ec2012-04-19 00:25:12 +0000232 case Stmt::ObjCBoxedExprClass:
Ted Kremenekb3f75422012-03-06 20:06:06 +0000233 case Stmt::ObjCSubscriptRefExprClass:
Douglas Gregor97b98722010-01-19 23:20:36 +0000234 K = CXCursor_UnexposedExpr;
235 break;
Douglas Gregor42b29842011-10-05 19:00:14 +0000236
John McCall4b9c2d22011-11-06 09:01:30 +0000237 case Stmt::OpaqueValueExprClass:
238 if (Expr *Src = cast<OpaqueValueExpr>(S)->getSourceExpr())
239 return MakeCXCursor(Src, Parent, TU, RegionOfInterest);
240 K = CXCursor_UnexposedExpr;
241 break;
242
243 case Stmt::PseudoObjectExprClass:
244 return MakeCXCursor(cast<PseudoObjectExpr>(S)->getSyntacticForm(),
245 Parent, TU, RegionOfInterest);
246
Douglas Gregor42b29842011-10-05 19:00:14 +0000247 case Stmt::CompoundStmtClass:
248 K = CXCursor_CompoundStmt;
249 break;
Richard Smith534986f2012-04-14 00:33:13 +0000250
Douglas Gregor42b29842011-10-05 19:00:14 +0000251 case Stmt::NullStmtClass:
252 K = CXCursor_NullStmt;
253 break;
Richard Smith534986f2012-04-14 00:33:13 +0000254
Douglas Gregor42b29842011-10-05 19:00:14 +0000255 case Stmt::LabelStmtClass:
256 K = CXCursor_LabelStmt;
257 break;
Richard Smith534986f2012-04-14 00:33:13 +0000258
259 case Stmt::AttributedStmtClass:
260 K = CXCursor_UnexposedStmt;
261 break;
262
Douglas Gregor42b29842011-10-05 19:00:14 +0000263 case Stmt::DeclStmtClass:
264 K = CXCursor_DeclStmt;
265 break;
Richard Smith534986f2012-04-14 00:33:13 +0000266
Douglas Gregor42b29842011-10-05 19:00:14 +0000267 case Stmt::IntegerLiteralClass:
268 K = CXCursor_IntegerLiteral;
269 break;
270
271 case Stmt::FloatingLiteralClass:
272 K = CXCursor_FloatingLiteral;
273 break;
274
275 case Stmt::ImaginaryLiteralClass:
276 K = CXCursor_ImaginaryLiteral;
277 break;
278
279 case Stmt::StringLiteralClass:
280 K = CXCursor_StringLiteral;
281 break;
282
283 case Stmt::CharacterLiteralClass:
284 K = CXCursor_CharacterLiteral;
285 break;
286
287 case Stmt::ParenExprClass:
288 K = CXCursor_ParenExpr;
289 break;
290
291 case Stmt::UnaryOperatorClass:
292 K = CXCursor_UnaryOperator;
293 break;
Richard Smith534986f2012-04-14 00:33:13 +0000294
Douglas Gregor42b29842011-10-05 19:00:14 +0000295 case Stmt::CXXNoexceptExprClass:
296 K = CXCursor_UnaryExpr;
297 break;
298
299 case Stmt::ArraySubscriptExprClass:
300 K = CXCursor_ArraySubscriptExpr;
301 break;
302
303 case Stmt::BinaryOperatorClass:
304 K = CXCursor_BinaryOperator;
305 break;
306
307 case Stmt::CompoundAssignOperatorClass:
308 K = CXCursor_CompoundAssignOperator;
309 break;
310
311 case Stmt::ConditionalOperatorClass:
312 K = CXCursor_ConditionalOperator;
313 break;
314
315 case Stmt::CStyleCastExprClass:
316 K = CXCursor_CStyleCastExpr;
317 break;
318
319 case Stmt::CompoundLiteralExprClass:
320 K = CXCursor_CompoundLiteralExpr;
321 break;
322
323 case Stmt::InitListExprClass:
324 K = CXCursor_InitListExpr;
325 break;
326
327 case Stmt::AddrLabelExprClass:
328 K = CXCursor_AddrLabelExpr;
329 break;
330
331 case Stmt::StmtExprClass:
332 K = CXCursor_StmtExpr;
333 break;
334
335 case Stmt::GenericSelectionExprClass:
336 K = CXCursor_GenericSelectionExpr;
337 break;
338
339 case Stmt::GNUNullExprClass:
340 K = CXCursor_GNUNullExpr;
341 break;
342
343 case Stmt::CXXStaticCastExprClass:
344 K = CXCursor_CXXStaticCastExpr;
345 break;
346
347 case Stmt::CXXDynamicCastExprClass:
348 K = CXCursor_CXXDynamicCastExpr;
349 break;
350
351 case Stmt::CXXReinterpretCastExprClass:
352 K = CXCursor_CXXReinterpretCastExpr;
353 break;
354
355 case Stmt::CXXConstCastExprClass:
356 K = CXCursor_CXXConstCastExpr;
357 break;
358
359 case Stmt::CXXFunctionalCastExprClass:
360 K = CXCursor_CXXFunctionalCastExpr;
361 break;
362
363 case Stmt::CXXTypeidExprClass:
364 K = CXCursor_CXXTypeidExpr;
365 break;
366
367 case Stmt::CXXBoolLiteralExprClass:
368 K = CXCursor_CXXBoolLiteralExpr;
369 break;
370
371 case Stmt::CXXNullPtrLiteralExprClass:
372 K = CXCursor_CXXNullPtrLiteralExpr;
373 break;
374
375 case Stmt::CXXThisExprClass:
376 K = CXCursor_CXXThisExpr;
377 break;
378
379 case Stmt::CXXThrowExprClass:
380 K = CXCursor_CXXThrowExpr;
381 break;
382
383 case Stmt::CXXNewExprClass:
384 K = CXCursor_CXXNewExpr;
385 break;
386
387 case Stmt::CXXDeleteExprClass:
388 K = CXCursor_CXXDeleteExpr;
389 break;
390
391 case Stmt::ObjCStringLiteralClass:
392 K = CXCursor_ObjCStringLiteral;
393 break;
394
395 case Stmt::ObjCEncodeExprClass:
396 K = CXCursor_ObjCEncodeExpr;
397 break;
398
399 case Stmt::ObjCSelectorExprClass:
400 K = CXCursor_ObjCSelectorExpr;
401 break;
402
403 case Stmt::ObjCProtocolExprClass:
404 K = CXCursor_ObjCProtocolExpr;
405 break;
Ted Kremenekb3f75422012-03-06 20:06:06 +0000406
407 case Stmt::ObjCBoolLiteralExprClass:
408 K = CXCursor_ObjCBoolLiteralExpr;
409 break;
410
Douglas Gregor42b29842011-10-05 19:00:14 +0000411 case Stmt::ObjCBridgedCastExprClass:
412 K = CXCursor_ObjCBridgedCastExpr;
413 break;
414
415 case Stmt::BlockExprClass:
416 K = CXCursor_BlockExpr;
417 break;
418
419 case Stmt::PackExpansionExprClass:
420 K = CXCursor_PackExpansionExpr;
421 break;
422
423 case Stmt::SizeOfPackExprClass:
424 K = CXCursor_SizeOfPackExpr;
425 break;
426
Douglas Gregor42b29842011-10-05 19:00:14 +0000427 case Stmt::DeclRefExprClass:
428 case Stmt::DependentScopeDeclRefExprClass:
John McCall91a57552011-07-15 05:09:51 +0000429 case Stmt::SubstNonTypeTemplateParmExprClass:
Douglas Gregorc7793c72011-01-15 01:15:58 +0000430 case Stmt::SubstNonTypeTemplateParmPackExprClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000431 case Stmt::UnresolvedLookupExprClass:
Douglas Gregor97b98722010-01-19 23:20:36 +0000432 K = CXCursor_DeclRefExpr;
433 break;
434
Douglas Gregor42b29842011-10-05 19:00:14 +0000435 case Stmt::CXXDependentScopeMemberExprClass:
436 case Stmt::CXXPseudoDestructorExprClass:
Douglas Gregor97b98722010-01-19 23:20:36 +0000437 case Stmt::MemberExprClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000438 case Stmt::ObjCIsaExprClass:
Douglas Gregor97b98722010-01-19 23:20:36 +0000439 case Stmt::ObjCIvarRefExprClass:
440 case Stmt::ObjCPropertyRefExprClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000441 case Stmt::UnresolvedMemberExprClass:
Douglas Gregor97b98722010-01-19 23:20:36 +0000442 K = CXCursor_MemberRefExpr;
443 break;
444
445 case Stmt::CallExprClass:
446 case Stmt::CXXOperatorCallExprClass:
447 case Stmt::CXXMemberCallExprClass:
Peter Collingbournee08ce652011-02-09 21:07:24 +0000448 case Stmt::CUDAKernelCallExprClass:
Douglas Gregor97b98722010-01-19 23:20:36 +0000449 case Stmt::CXXConstructExprClass:
450 case Stmt::CXXTemporaryObjectExprClass:
Douglas Gregor42b29842011-10-05 19:00:14 +0000451 case Stmt::CXXUnresolvedConstructExprClass:
Richard Smith9fcce652012-03-07 08:35:16 +0000452 case Stmt::UserDefinedLiteralClass:
Douglas Gregor97b98722010-01-19 23:20:36 +0000453 K = CXCursor_CallExpr;
454 break;
455
Douglas Gregor011d8b92012-02-15 00:54:55 +0000456 case Stmt::LambdaExprClass:
457 K = CXCursor_LambdaExpr;
458 break;
459
Douglas Gregorba0513d2011-10-25 01:33:02 +0000460 case Stmt::ObjCMessageExprClass: {
Douglas Gregor97b98722010-01-19 23:20:36 +0000461 K = CXCursor_ObjCMessageExpr;
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000462 int SelectorIdIndex = -1;
463 // Check if cursor points to a selector id.
464 if (RegionOfInterest.isValid() &&
465 RegionOfInterest.getBegin() == RegionOfInterest.getEnd()) {
466 SmallVector<SourceLocation, 16> SelLocs;
467 cast<ObjCMessageExpr>(S)->getSelectorLocs(SelLocs);
468 SmallVector<SourceLocation, 16>::iterator
469 I=std::find(SelLocs.begin(), SelLocs.end(),RegionOfInterest.getBegin());
470 if (I != SelLocs.end())
471 SelectorIdIndex = I - SelLocs.begin();
472 }
473 CXCursor C = { K, 0, { Parent, S, TU } };
474 return getSelectorIdentifierCursor(SelectorIdIndex, C);
Douglas Gregor97b98722010-01-19 23:20:36 +0000475 }
Douglas Gregorba0513d2011-10-25 01:33:02 +0000476
477 case Stmt::MSDependentExistsStmtClass:
478 K = CXCursor_UnexposedStmt;
479 break;
480 }
Douglas Gregor97b98722010-01-19 23:20:36 +0000481
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000482 CXCursor C = { K, 0, { Parent, S, TU } };
Douglas Gregor97b98722010-01-19 23:20:36 +0000483 return C;
484}
485
Douglas Gregor2e331b92010-01-16 14:00:32 +0000486CXCursor cxcursor::MakeCursorObjCSuperClassRef(ObjCInterfaceDecl *Super,
Douglas Gregorb2cd4872010-01-20 23:57:43 +0000487 SourceLocation Loc,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000488 CXTranslationUnit TU) {
Daniel Dunbar54d67ca2010-01-25 00:40:30 +0000489 assert(Super && TU && "Invalid arguments!");
Douglas Gregor2e331b92010-01-16 14:00:32 +0000490 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000491 CXCursor C = { CXCursor_ObjCSuperClassRef, 0, { Super, RawLoc, TU } };
Douglas Gregor2e331b92010-01-16 14:00:32 +0000492 return C;
493}
494
495std::pair<ObjCInterfaceDecl *, SourceLocation>
496cxcursor::getCursorObjCSuperClassRef(CXCursor C) {
497 assert(C.kind == CXCursor_ObjCSuperClassRef);
498 return std::make_pair(static_cast<ObjCInterfaceDecl *>(C.data[0]),
499 SourceLocation::getFromRawEncoding(
500 reinterpret_cast<uintptr_t>(C.data[1])));
501}
502
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000503CXCursor cxcursor::MakeCursorObjCProtocolRef(const ObjCProtocolDecl *Proto,
Douglas Gregorb2cd4872010-01-20 23:57:43 +0000504 SourceLocation Loc,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000505 CXTranslationUnit TU) {
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000506 assert(Proto && TU && "Invalid arguments!");
Douglas Gregor78db0cd2010-01-16 15:44:18 +0000507 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000508 CXCursor C = { CXCursor_ObjCProtocolRef, 0, { (void*)Proto, RawLoc, TU } };
Douglas Gregor78db0cd2010-01-16 15:44:18 +0000509 return C;
510}
511
512std::pair<ObjCProtocolDecl *, SourceLocation>
513cxcursor::getCursorObjCProtocolRef(CXCursor C) {
514 assert(C.kind == CXCursor_ObjCProtocolRef);
515 return std::make_pair(static_cast<ObjCProtocolDecl *>(C.data[0]),
516 SourceLocation::getFromRawEncoding(
517 reinterpret_cast<uintptr_t>(C.data[1])));
518}
519
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000520CXCursor cxcursor::MakeCursorObjCClassRef(const ObjCInterfaceDecl *Class,
Douglas Gregorb2cd4872010-01-20 23:57:43 +0000521 SourceLocation Loc,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000522 CXTranslationUnit TU) {
Ted Kremenekebfa3392010-03-19 20:39:03 +0000523 // 'Class' can be null for invalid code.
524 if (!Class)
525 return MakeCXCursorInvalid(CXCursor_InvalidCode);
526 assert(TU && "Invalid arguments!");
Douglas Gregor1adb0822010-01-16 17:14:40 +0000527 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000528 CXCursor C = { CXCursor_ObjCClassRef, 0, { (void*)Class, RawLoc, TU } };
Douglas Gregor1adb0822010-01-16 17:14:40 +0000529 return C;
530}
531
532std::pair<ObjCInterfaceDecl *, SourceLocation>
533cxcursor::getCursorObjCClassRef(CXCursor C) {
534 assert(C.kind == CXCursor_ObjCClassRef);
535 return std::make_pair(static_cast<ObjCInterfaceDecl *>(C.data[0]),
536 SourceLocation::getFromRawEncoding(
537 reinterpret_cast<uintptr_t>(C.data[1])));
538}
539
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000540CXCursor cxcursor::MakeCursorTypeRef(const TypeDecl *Type, SourceLocation Loc,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000541 CXTranslationUnit TU) {
Daniel Dunbar54d67ca2010-01-25 00:40:30 +0000542 assert(Type && TU && "Invalid arguments!");
Douglas Gregor7d0d40e2010-01-21 16:28:34 +0000543 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
Argyrios Kyrtzidis4e7064f2011-10-17 19:48:19 +0000544 CXCursor C = { CXCursor_TypeRef, 0, { (void*)Type, RawLoc, TU } };
Douglas Gregor7d0d40e2010-01-21 16:28:34 +0000545 return C;
546}
547
548std::pair<TypeDecl *, SourceLocation>
549cxcursor::getCursorTypeRef(CXCursor C) {
550 assert(C.kind == CXCursor_TypeRef);
551 return std::make_pair(static_cast<TypeDecl *>(C.data[0]),
552 SourceLocation::getFromRawEncoding(
553 reinterpret_cast<uintptr_t>(C.data[1])));
554}
555
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000556CXCursor cxcursor::MakeCursorTemplateRef(const TemplateDecl *Template,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000557 SourceLocation Loc,
558 CXTranslationUnit TU) {
Douglas Gregor0b36e612010-08-31 20:37:03 +0000559 assert(Template && TU && "Invalid arguments!");
560 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000561 CXCursor C = { CXCursor_TemplateRef, 0, { (void*)Template, RawLoc, TU } };
Douglas Gregor0b36e612010-08-31 20:37:03 +0000562 return C;
563}
564
565std::pair<TemplateDecl *, SourceLocation>
566cxcursor::getCursorTemplateRef(CXCursor C) {
567 assert(C.kind == CXCursor_TemplateRef);
568 return std::make_pair(static_cast<TemplateDecl *>(C.data[0]),
569 SourceLocation::getFromRawEncoding(
570 reinterpret_cast<uintptr_t>(C.data[1])));
571}
572
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000573CXCursor cxcursor::MakeCursorNamespaceRef(const NamedDecl *NS,
574 SourceLocation Loc,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000575 CXTranslationUnit TU) {
Douglas Gregor69319002010-08-31 23:48:11 +0000576
577 assert(NS && (isa<NamespaceDecl>(NS) || isa<NamespaceAliasDecl>(NS)) && TU &&
578 "Invalid arguments!");
579 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000580 CXCursor C = { CXCursor_NamespaceRef, 0, { (void*)NS, RawLoc, TU } };
Douglas Gregor69319002010-08-31 23:48:11 +0000581 return C;
582}
583
584std::pair<NamedDecl *, SourceLocation>
585cxcursor::getCursorNamespaceRef(CXCursor C) {
586 assert(C.kind == CXCursor_NamespaceRef);
587 return std::make_pair(static_cast<NamedDecl *>(C.data[0]),
588 SourceLocation::getFromRawEncoding(
589 reinterpret_cast<uintptr_t>(C.data[1])));
590}
591
Douglas Gregor011d8b92012-02-15 00:54:55 +0000592CXCursor cxcursor::MakeCursorVariableRef(const VarDecl *Var, SourceLocation Loc,
593 CXTranslationUnit TU) {
594
595 assert(Var && TU && "Invalid arguments!");
596 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
597 CXCursor C = { CXCursor_VariableRef, 0, { (void*)Var, RawLoc, TU } };
598 return C;
599}
600
601std::pair<VarDecl *, SourceLocation>
602cxcursor::getCursorVariableRef(CXCursor C) {
603 assert(C.kind == CXCursor_VariableRef);
604 return std::make_pair(static_cast<VarDecl *>(C.data[0]),
605 SourceLocation::getFromRawEncoding(
606 reinterpret_cast<uintptr_t>(C.data[1])));
607}
608
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000609CXCursor cxcursor::MakeCursorMemberRef(const FieldDecl *Field, SourceLocation Loc,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000610 CXTranslationUnit TU) {
Douglas Gregora67e03f2010-09-09 21:42:20 +0000611
612 assert(Field && TU && "Invalid arguments!");
613 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
Argyrios Kyrtzidisb395c632011-11-18 00:26:51 +0000614 CXCursor C = { CXCursor_MemberRef, 0, { (void*)Field, RawLoc, TU } };
Douglas Gregora67e03f2010-09-09 21:42:20 +0000615 return C;
616}
617
618std::pair<FieldDecl *, SourceLocation>
619cxcursor::getCursorMemberRef(CXCursor C) {
620 assert(C.kind == CXCursor_MemberRef);
621 return std::make_pair(static_cast<FieldDecl *>(C.data[0]),
622 SourceLocation::getFromRawEncoding(
623 reinterpret_cast<uintptr_t>(C.data[1])));
624}
625
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000626CXCursor cxcursor::MakeCursorCXXBaseSpecifier(const CXXBaseSpecifier *B,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000627 CXTranslationUnit TU){
Argyrios Kyrtzidis2957e6f2011-11-22 07:24:51 +0000628 CXCursor C = { CXCursor_CXXBaseSpecifier, 0, { (void*)B, 0, TU } };
Ted Kremenek3064ef92010-08-27 21:34:58 +0000629 return C;
630}
631
632CXXBaseSpecifier *cxcursor::getCursorCXXBaseSpecifier(CXCursor C) {
633 assert(C.kind == CXCursor_CXXBaseSpecifier);
634 return static_cast<CXXBaseSpecifier*>(C.data[0]);
635}
636
Douglas Gregor9f1e3ff2010-03-18 00:42:48 +0000637CXCursor cxcursor::MakePreprocessingDirectiveCursor(SourceRange Range,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000638 CXTranslationUnit TU) {
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000639 CXCursor C = { CXCursor_PreprocessingDirective, 0,
Douglas Gregor9f1e3ff2010-03-18 00:42:48 +0000640 { reinterpret_cast<void *>(Range.getBegin().getRawEncoding()),
641 reinterpret_cast<void *>(Range.getEnd().getRawEncoding()),
642 TU }
643 };
644 return C;
645}
646
647SourceRange cxcursor::getCursorPreprocessingDirective(CXCursor C) {
648 assert(C.kind == CXCursor_PreprocessingDirective);
Argyrios Kyrtzidisee0f84f2011-09-26 08:01:41 +0000649 SourceRange Range = SourceRange(SourceLocation::getFromRawEncoding(
Douglas Gregor9f1e3ff2010-03-18 00:42:48 +0000650 reinterpret_cast<uintptr_t> (C.data[0])),
651 SourceLocation::getFromRawEncoding(
652 reinterpret_cast<uintptr_t> (C.data[1])));
Argyrios Kyrtzidisee0f84f2011-09-26 08:01:41 +0000653 ASTUnit *TU = getCursorASTUnit(C);
654 return TU->mapRangeFromPreamble(Range);
Douglas Gregor9f1e3ff2010-03-18 00:42:48 +0000655}
656
Ted Kremeneka60ed472010-11-16 08:15:36 +0000657CXCursor cxcursor::MakeMacroDefinitionCursor(MacroDefinition *MI,
658 CXTranslationUnit TU) {
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000659 CXCursor C = { CXCursor_MacroDefinition, 0, { MI, 0, TU } };
Douglas Gregor572feb22010-03-18 18:04:21 +0000660 return C;
661}
662
663MacroDefinition *cxcursor::getCursorMacroDefinition(CXCursor C) {
664 assert(C.kind == CXCursor_MacroDefinition);
665 return static_cast<MacroDefinition *>(C.data[0]);
666}
667
Chandler Carruth9e5bb852011-07-14 08:20:46 +0000668CXCursor cxcursor::MakeMacroExpansionCursor(MacroExpansion *MI,
669 CXTranslationUnit TU) {
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000670 CXCursor C = { CXCursor_MacroExpansion, 0, { MI, 0, TU } };
Douglas Gregor48072312010-03-18 15:23:44 +0000671 return C;
672}
673
Chandler Carruth9e5bb852011-07-14 08:20:46 +0000674MacroExpansion *cxcursor::getCursorMacroExpansion(CXCursor C) {
Chandler Carruth9b2a0ac2011-07-14 08:41:15 +0000675 assert(C.kind == CXCursor_MacroExpansion);
Chandler Carruth9e5bb852011-07-14 08:20:46 +0000676 return static_cast<MacroExpansion *>(C.data[0]);
Douglas Gregor48072312010-03-18 15:23:44 +0000677}
678
Douglas Gregorecdcb882010-10-20 22:00:55 +0000679CXCursor cxcursor::MakeInclusionDirectiveCursor(InclusionDirective *ID,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000680 CXTranslationUnit TU) {
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000681 CXCursor C = { CXCursor_InclusionDirective, 0, { ID, 0, TU } };
Douglas Gregorecdcb882010-10-20 22:00:55 +0000682 return C;
683}
684
685InclusionDirective *cxcursor::getCursorInclusionDirective(CXCursor C) {
686 assert(C.kind == CXCursor_InclusionDirective);
687 return static_cast<InclusionDirective *>(C.data[0]);
688}
689
Douglas Gregor36897b02010-09-10 00:22:18 +0000690CXCursor cxcursor::MakeCursorLabelRef(LabelStmt *Label, SourceLocation Loc,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000691 CXTranslationUnit TU) {
Douglas Gregor36897b02010-09-10 00:22:18 +0000692
693 assert(Label && TU && "Invalid arguments!");
694 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000695 CXCursor C = { CXCursor_LabelRef, 0, { Label, RawLoc, TU } };
Douglas Gregor36897b02010-09-10 00:22:18 +0000696 return C;
697}
698
699std::pair<LabelStmt*, SourceLocation>
700cxcursor::getCursorLabelRef(CXCursor C) {
701 assert(C.kind == CXCursor_LabelRef);
702 return std::make_pair(static_cast<LabelStmt *>(C.data[0]),
703 SourceLocation::getFromRawEncoding(
704 reinterpret_cast<uintptr_t>(C.data[1])));
705}
706
Douglas Gregor1f60d9e2010-09-13 22:52:57 +0000707CXCursor cxcursor::MakeCursorOverloadedDeclRef(OverloadExpr *E,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000708 CXTranslationUnit TU) {
Douglas Gregor1f60d9e2010-09-13 22:52:57 +0000709 assert(E && TU && "Invalid arguments!");
710 OverloadedDeclRefStorage Storage(E);
711 void *RawLoc = reinterpret_cast<void *>(E->getNameLoc().getRawEncoding());
712 CXCursor C = {
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000713 CXCursor_OverloadedDeclRef, 0,
Douglas Gregor1f60d9e2010-09-13 22:52:57 +0000714 { Storage.getOpaqueValue(), RawLoc, TU }
715 };
716 return C;
717}
718
719CXCursor cxcursor::MakeCursorOverloadedDeclRef(Decl *D,
720 SourceLocation Loc,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000721 CXTranslationUnit TU) {
Douglas Gregor1f60d9e2010-09-13 22:52:57 +0000722 assert(D && TU && "Invalid arguments!");
723 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
724 OverloadedDeclRefStorage Storage(D);
725 CXCursor C = {
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000726 CXCursor_OverloadedDeclRef, 0,
Douglas Gregor1f60d9e2010-09-13 22:52:57 +0000727 { Storage.getOpaqueValue(), RawLoc, TU }
728 };
729 return C;
730}
731
732CXCursor cxcursor::MakeCursorOverloadedDeclRef(TemplateName Name,
733 SourceLocation Loc,
Ted Kremeneka60ed472010-11-16 08:15:36 +0000734 CXTranslationUnit TU) {
Douglas Gregor1f60d9e2010-09-13 22:52:57 +0000735 assert(Name.getAsOverloadedTemplate() && TU && "Invalid arguments!");
736 void *RawLoc = reinterpret_cast<void *>(Loc.getRawEncoding());
737 OverloadedDeclRefStorage Storage(Name.getAsOverloadedTemplate());
738 CXCursor C = {
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000739 CXCursor_OverloadedDeclRef, 0,
Douglas Gregor1f60d9e2010-09-13 22:52:57 +0000740 { Storage.getOpaqueValue(), RawLoc, TU }
741 };
742 return C;
743}
744
745std::pair<cxcursor::OverloadedDeclRefStorage, SourceLocation>
746cxcursor::getCursorOverloadedDeclRef(CXCursor C) {
747 assert(C.kind == CXCursor_OverloadedDeclRef);
748 return std::make_pair(OverloadedDeclRefStorage::getFromOpaqueValue(C.data[0]),
749 SourceLocation::getFromRawEncoding(
750 reinterpret_cast<uintptr_t>(C.data[1])));
751}
752
Douglas Gregor283cae32010-01-15 21:56:13 +0000753Decl *cxcursor::getCursorDecl(CXCursor Cursor) {
754 return (Decl *)Cursor.data[0];
755}
756
757Expr *cxcursor::getCursorExpr(CXCursor Cursor) {
758 return dyn_cast_or_null<Expr>(getCursorStmt(Cursor));
759}
760
761Stmt *cxcursor::getCursorStmt(CXCursor Cursor) {
Douglas Gregor78db0cd2010-01-16 15:44:18 +0000762 if (Cursor.kind == CXCursor_ObjCSuperClassRef ||
Douglas Gregor1adb0822010-01-16 17:14:40 +0000763 Cursor.kind == CXCursor_ObjCProtocolRef ||
764 Cursor.kind == CXCursor_ObjCClassRef)
Douglas Gregor2e331b92010-01-16 14:00:32 +0000765 return 0;
766
Douglas Gregor283cae32010-01-15 21:56:13 +0000767 return (Stmt *)Cursor.data[1];
768}
769
Ted Kremenek95f33552010-08-26 01:42:22 +0000770Attr *cxcursor::getCursorAttr(CXCursor Cursor) {
771 return (Attr *)Cursor.data[1];
772}
773
Argyrios Kyrtzidis8ccac3d2011-06-29 22:20:07 +0000774Decl *cxcursor::getCursorParentDecl(CXCursor Cursor) {
775 return (Decl *)Cursor.data[0];
776}
777
Douglas Gregorf46034a2010-01-18 23:41:10 +0000778ASTContext &cxcursor::getCursorContext(CXCursor Cursor) {
Douglas Gregorb2cd4872010-01-20 23:57:43 +0000779 return getCursorASTUnit(Cursor)->getASTContext();
780}
Douglas Gregorf46034a2010-01-18 23:41:10 +0000781
Douglas Gregorb2cd4872010-01-20 23:57:43 +0000782ASTUnit *cxcursor::getCursorASTUnit(CXCursor Cursor) {
Argyrios Kyrtzidis44517462011-12-09 00:17:49 +0000783 CXTranslationUnit TU = static_cast<CXTranslationUnit>(Cursor.data[2]);
784 if (!TU)
785 return 0;
786 return static_cast<ASTUnit *>(TU->TUData);
Ted Kremeneka60ed472010-11-16 08:15:36 +0000787}
788
789CXTranslationUnit cxcursor::getCursorTU(CXCursor Cursor) {
790 return static_cast<CXTranslationUnit>(Cursor.data[2]);
Douglas Gregor283cae32010-01-15 21:56:13 +0000791}
792
Argyrios Kyrtzidis15f4c982012-04-10 21:01:03 +0000793static void CollectOverriddenMethodsRecurse(CXTranslationUnit TU,
794 ObjCContainerDecl *Container,
Argyrios Kyrtzidisb11be042011-10-06 07:00:46 +0000795 ObjCMethodDecl *Method,
Argyrios Kyrtzidis15f4c982012-04-10 21:01:03 +0000796 SmallVectorImpl<CXCursor> &Methods,
797 bool MovedToSuper) {
Argyrios Kyrtzidisb11be042011-10-06 07:00:46 +0000798 if (!Container)
799 return;
800
Argyrios Kyrtzidis044e6452012-03-08 00:20:03 +0000801 // In categories look for overriden methods from protocols. A method from
802 // category is not "overriden" since it is considered as the "same" method
803 // (same USR) as the one from the interface.
804 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
Argyrios Kyrtzidis15f4c982012-04-10 21:01:03 +0000805 // Check whether we have a matching method at this category but only if we
806 // are at the super class level.
807 if (MovedToSuper)
808 if (ObjCMethodDecl *
809 Overridden = Container->getMethod(Method->getSelector(),
810 Method->isInstanceMethod()))
811 if (Method != Overridden) {
812 // We found an override at this category; there is no need to look
813 // into its protocols.
814 Methods.push_back(MakeCXCursor(Overridden, TU));
815 return;
816 }
817
Argyrios Kyrtzidis044e6452012-03-08 00:20:03 +0000818 for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
819 PEnd = Category->protocol_end();
820 P != PEnd; ++P)
Argyrios Kyrtzidis15f4c982012-04-10 21:01:03 +0000821 CollectOverriddenMethodsRecurse(TU, *P, Method, Methods, MovedToSuper);
Argyrios Kyrtzidis044e6452012-03-08 00:20:03 +0000822 return;
823 }
824
Argyrios Kyrtzidisb11be042011-10-06 07:00:46 +0000825 // Check whether we have a matching method at this level.
826 if (ObjCMethodDecl *Overridden = Container->getMethod(Method->getSelector(),
827 Method->isInstanceMethod()))
828 if (Method != Overridden) {
829 // We found an override at this level; there is no need to look
830 // into other protocols or categories.
831 Methods.push_back(MakeCXCursor(Overridden, TU));
832 return;
833 }
834
835 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
836 for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
837 PEnd = Protocol->protocol_end();
838 P != PEnd; ++P)
Argyrios Kyrtzidis15f4c982012-04-10 21:01:03 +0000839 CollectOverriddenMethodsRecurse(TU, *P, Method, Methods, MovedToSuper);
Argyrios Kyrtzidisb11be042011-10-06 07:00:46 +0000840 }
841
Argyrios Kyrtzidisb11be042011-10-06 07:00:46 +0000842 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
843 for (ObjCInterfaceDecl::protocol_iterator P = Interface->protocol_begin(),
844 PEnd = Interface->protocol_end();
845 P != PEnd; ++P)
Argyrios Kyrtzidis15f4c982012-04-10 21:01:03 +0000846 CollectOverriddenMethodsRecurse(TU, *P, Method, Methods, MovedToSuper);
Argyrios Kyrtzidisb11be042011-10-06 07:00:46 +0000847
848 for (ObjCCategoryDecl *Category = Interface->getCategoryList();
849 Category; Category = Category->getNextClassCategory())
Argyrios Kyrtzidis15f4c982012-04-10 21:01:03 +0000850 CollectOverriddenMethodsRecurse(TU, Category, Method, Methods,
851 MovedToSuper);
Argyrios Kyrtzidisb11be042011-10-06 07:00:46 +0000852
Argyrios Kyrtzidis044e6452012-03-08 00:20:03 +0000853 if (ObjCInterfaceDecl *Super = Interface->getSuperClass())
Argyrios Kyrtzidis15f4c982012-04-10 21:01:03 +0000854 return CollectOverriddenMethodsRecurse(TU, Super, Method, Methods,
855 /*MovedToSuper=*/true);
Argyrios Kyrtzidisb11be042011-10-06 07:00:46 +0000856 }
857}
858
Argyrios Kyrtzidis15f4c982012-04-10 21:01:03 +0000859static inline void CollectOverriddenMethods(CXTranslationUnit TU,
860 ObjCContainerDecl *Container,
861 ObjCMethodDecl *Method,
862 SmallVectorImpl<CXCursor> &Methods) {
863 CollectOverriddenMethodsRecurse(TU, Container, Method, Methods,
864 /*MovedToSuper=*/false);
865}
866
Argyrios Kyrtzidise15db6f2012-05-09 16:12:57 +0000867static void collectOverriddenMethodsSlow(CXTranslationUnit TU,
868 ObjCMethodDecl *Method,
869 SmallVectorImpl<CXCursor> &overridden) {
870 assert(Method->isOverriding());
Argyrios Kyrtzidisb11be042011-10-06 07:00:46 +0000871
Argyrios Kyrtzidis044e6452012-03-08 00:20:03 +0000872 if (ObjCProtocolDecl *
873 ProtD = dyn_cast<ObjCProtocolDecl>(Method->getDeclContext())) {
874 CollectOverriddenMethods(TU, ProtD, Method, overridden);
875
876 } else if (ObjCImplDecl *
877 IMD = dyn_cast<ObjCImplDecl>(Method->getDeclContext())) {
878 ObjCInterfaceDecl *ID = IMD->getClassInterface();
879 if (!ID)
880 return;
881 // Start searching for overridden methods using the method from the
882 // interface as starting point.
883 if (ObjCMethodDecl *IFaceMeth = ID->getMethod(Method->getSelector(),
884 Method->isInstanceMethod()))
885 Method = IFaceMeth;
886 CollectOverriddenMethods(TU, ID, Method, overridden);
887
888 } else if (ObjCCategoryDecl *
889 CatD = dyn_cast<ObjCCategoryDecl>(Method->getDeclContext())) {
890 ObjCInterfaceDecl *ID = CatD->getClassInterface();
891 if (!ID)
892 return;
893 // Start searching for overridden methods using the method from the
894 // interface as starting point.
895 if (ObjCMethodDecl *IFaceMeth = ID->getMethod(Method->getSelector(),
896 Method->isInstanceMethod()))
897 Method = IFaceMeth;
898 CollectOverriddenMethods(TU, ID, Method, overridden);
899
900 } else {
Argyrios Kyrtzidis15f4c982012-04-10 21:01:03 +0000901 CollectOverriddenMethods(TU,
902 dyn_cast_or_null<ObjCContainerDecl>(Method->getDeclContext()),
903 Method, overridden);
Argyrios Kyrtzidis044e6452012-03-08 00:20:03 +0000904 }
Argyrios Kyrtzidisb11be042011-10-06 07:00:46 +0000905}
906
Argyrios Kyrtzidise15db6f2012-05-09 16:12:57 +0000907static void collectOnCategoriesAfterLocation(SourceLocation Loc,
908 ObjCInterfaceDecl *Class,
909 CXTranslationUnit TU,
910 ObjCMethodDecl *Method,
911 SmallVectorImpl<CXCursor> &Methods) {
912 if (!Class)
913 return;
914
915 SourceManager &SM = static_cast<ASTUnit *>(TU->TUData)->getSourceManager();
916 for (ObjCCategoryDecl *Category = Class->getCategoryList();
917 Category; Category = Category->getNextClassCategory())
918 if (SM.isBeforeInTranslationUnit(Loc, Category->getLocation()))
919 CollectOverriddenMethodsRecurse(TU, Category, Method, Methods, true);
920
921 collectOnCategoriesAfterLocation(Loc, Class->getSuperClass(), TU,
922 Method, Methods);
923}
924
925/// \brief Faster collection that is enabled when ObjCMethodDecl::isOverriding()
926/// returns false.
927/// You'd think that in that case there are no overrides but categories can
928/// "introduce" new overridden methods that are missed by Sema because the
929/// overrides lookup that it does for methods, inside implementations, will
930/// stop at the interface level (if there is a method there) and not look
931/// further in super classes.
932static void collectOverriddenMethodsFast(CXTranslationUnit TU,
933 ObjCMethodDecl *Method,
934 SmallVectorImpl<CXCursor> &Methods) {
935 assert(!Method->isOverriding());
936
937 ObjCContainerDecl *ContD = cast<ObjCContainerDecl>(Method->getDeclContext());
938 if (isa<ObjCInterfaceDecl>(ContD) || isa<ObjCProtocolDecl>(ContD))
939 return;
940 ObjCInterfaceDecl *Class = Method->getClassInterface();
941 if (!Class)
942 return;
943
944 collectOnCategoriesAfterLocation(Class->getLocation(), Class->getSuperClass(),
945 TU, Method, Methods);
946}
947
948void cxcursor::getOverriddenCursors(CXCursor cursor,
949 SmallVectorImpl<CXCursor> &overridden) {
950 assert(clang_isDeclaration(cursor.kind));
951 Decl *D = getCursorDecl(cursor);
952 if (!D)
953 return;
954
955 // Handle C++ member functions.
956 CXTranslationUnit TU = getCursorTU(cursor);
957 if (CXXMethodDecl *CXXMethod = dyn_cast<CXXMethodDecl>(D)) {
958 for (CXXMethodDecl::method_iterator
959 M = CXXMethod->begin_overridden_methods(),
960 MEnd = CXXMethod->end_overridden_methods();
961 M != MEnd; ++M)
962 overridden.push_back(MakeCXCursor(const_cast<CXXMethodDecl*>(*M), TU));
963 return;
964 }
965
966 ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(D);
967 if (!Method)
968 return;
969
970 if (Method->isRedeclaration()) {
971 Method = cast<ObjCContainerDecl>(Method->getDeclContext())->
972 getMethod(Method->getSelector(), Method->isInstanceMethod());
973 }
974
975 if (!Method->isOverriding()) {
976 collectOverriddenMethodsFast(TU, Method, overridden);
977 } else {
978 collectOverriddenMethodsSlow(TU, Method, overridden);
979 assert(!overridden.empty() &&
980 "ObjCMethodDecl's overriding bit is not as expected");
981 }
982}
983
Argyrios Kyrtzidisaed123e2011-10-06 07:00:54 +0000984std::pair<int, SourceLocation>
985cxcursor::getSelectorIdentifierIndexAndLoc(CXCursor cursor) {
986 if (cursor.kind == CXCursor_ObjCMessageExpr) {
987 if (cursor.xdata != -1)
988 return std::make_pair(cursor.xdata,
989 cast<ObjCMessageExpr>(getCursorExpr(cursor))
990 ->getSelectorLoc(cursor.xdata));
991 } else if (cursor.kind == CXCursor_ObjCClassMethodDecl ||
992 cursor.kind == CXCursor_ObjCInstanceMethodDecl) {
993 if (cursor.xdata != -1)
994 return std::make_pair(cursor.xdata,
995 cast<ObjCMethodDecl>(getCursorDecl(cursor))
996 ->getSelectorLoc(cursor.xdata));
997 }
998
999 return std::make_pair(-1, SourceLocation());
1000}
1001
1002CXCursor cxcursor::getSelectorIdentifierCursor(int SelIdx, CXCursor cursor) {
1003 CXCursor newCursor = cursor;
1004
1005 if (cursor.kind == CXCursor_ObjCMessageExpr) {
1006 if (SelIdx == -1 ||
1007 unsigned(SelIdx) >= cast<ObjCMessageExpr>(getCursorExpr(cursor))
1008 ->getNumSelectorLocs())
1009 newCursor.xdata = -1;
1010 else
1011 newCursor.xdata = SelIdx;
1012 } else if (cursor.kind == CXCursor_ObjCClassMethodDecl ||
1013 cursor.kind == CXCursor_ObjCInstanceMethodDecl) {
1014 if (SelIdx == -1 ||
1015 unsigned(SelIdx) >= cast<ObjCMethodDecl>(getCursorDecl(cursor))
1016 ->getNumSelectorLocs())
1017 newCursor.xdata = -1;
1018 else
1019 newCursor.xdata = SelIdx;
1020 }
1021
1022 return newCursor;
1023}
1024
1025CXCursor cxcursor::getTypeRefCursor(CXCursor cursor) {
1026 if (cursor.kind != CXCursor_CallExpr)
1027 return cursor;
1028
1029 if (cursor.xdata == 0)
1030 return cursor;
1031
1032 Expr *E = getCursorExpr(cursor);
1033 TypeSourceInfo *Type = 0;
1034 if (CXXUnresolvedConstructExpr *
1035 UnCtor = dyn_cast<CXXUnresolvedConstructExpr>(E)) {
1036 Type = UnCtor->getTypeSourceInfo();
1037 } else if (CXXTemporaryObjectExpr *Tmp = dyn_cast<CXXTemporaryObjectExpr>(E)){
1038 Type = Tmp->getTypeSourceInfo();
1039 }
1040
1041 if (!Type)
1042 return cursor;
1043
1044 CXTranslationUnit TU = getCursorTU(cursor);
1045 QualType Ty = Type->getType();
1046 TypeLoc TL = Type->getTypeLoc();
1047 SourceLocation Loc = TL.getBeginLoc();
1048
1049 if (const ElaboratedType *ElabT = Ty->getAs<ElaboratedType>()) {
1050 Ty = ElabT->getNamedType();
1051 ElaboratedTypeLoc ElabTL = cast<ElaboratedTypeLoc>(TL);
1052 Loc = ElabTL.getNamedTypeLoc().getBeginLoc();
1053 }
1054
1055 if (const TypedefType *Typedef = Ty->getAs<TypedefType>())
1056 return MakeCursorTypeRef(Typedef->getDecl(), Loc, TU);
1057 if (const TagType *Tag = Ty->getAs<TagType>())
1058 return MakeCursorTypeRef(Tag->getDecl(), Loc, TU);
1059 if (const TemplateTypeParmType *TemplP = Ty->getAs<TemplateTypeParmType>())
1060 return MakeCursorTypeRef(TemplP->getDecl(), Loc, TU);
1061
1062 return cursor;
1063}
1064
Douglas Gregor283cae32010-01-15 21:56:13 +00001065bool cxcursor::operator==(CXCursor X, CXCursor Y) {
1066 return X.kind == Y.kind && X.data[0] == Y.data[0] && X.data[1] == Y.data[1] &&
1067 X.data[2] == Y.data[2];
Douglas Gregor2e331b92010-01-16 14:00:32 +00001068}
Ted Kremenek007a7c92010-11-01 23:26:51 +00001069
1070// FIXME: Remove once we can model DeclGroups and their appropriate ranges
1071// properly in the ASTs.
1072bool cxcursor::isFirstInDeclGroup(CXCursor C) {
1073 assert(clang_isDeclaration(C.kind));
1074 return ((uintptr_t) (C.data[1])) != 0;
1075}
1076
Ted Kremenekeca099b2010-12-08 23:43:14 +00001077//===----------------------------------------------------------------------===//
Argyrios Kyrtzidisb0d6eaa2011-09-27 00:30:30 +00001078// libclang CXCursor APIs
1079//===----------------------------------------------------------------------===//
1080
Argyrios Kyrtzidisfa865df2011-09-27 04:14:36 +00001081extern "C" {
1082
1083int clang_Cursor_isNull(CXCursor cursor) {
1084 return clang_equalCursors(cursor, clang_getNullCursor());
1085}
1086
Argyrios Kyrtzidisb0d6eaa2011-09-27 00:30:30 +00001087CXTranslationUnit clang_Cursor_getTranslationUnit(CXCursor cursor) {
1088 return getCursorTU(cursor);
1089}
1090
Argyrios Kyrtzidisd98ef9a2012-04-11 19:32:19 +00001091int clang_Cursor_getNumArguments(CXCursor C) {
1092 if (clang_isDeclaration(C.kind)) {
1093 Decl *D = cxcursor::getCursorDecl(C);
1094 if (const ObjCMethodDecl *MD = dyn_cast_or_null<ObjCMethodDecl>(D))
1095 return MD->param_size();
1096 if (const FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(D))
1097 return FD->param_size();
1098 }
1099
1100 return -1;
1101}
1102
1103CXCursor clang_Cursor_getArgument(CXCursor C, unsigned i) {
1104 if (clang_isDeclaration(C.kind)) {
1105 Decl *D = cxcursor::getCursorDecl(C);
1106 if (ObjCMethodDecl *MD = dyn_cast_or_null<ObjCMethodDecl>(D)) {
1107 if (i < MD->param_size())
1108 return cxcursor::MakeCXCursor(MD->param_begin()[i],
1109 cxcursor::getCursorTU(C));
1110 } else if (FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(D)) {
1111 if (i < FD->param_size())
1112 return cxcursor::MakeCXCursor(FD->param_begin()[i],
1113 cxcursor::getCursorTU(C));
1114 }
1115 }
1116
1117 return clang_getNullCursor();
1118}
1119
Argyrios Kyrtzidisfa865df2011-09-27 04:14:36 +00001120} // end: extern "C"
1121
Argyrios Kyrtzidisb0d6eaa2011-09-27 00:30:30 +00001122//===----------------------------------------------------------------------===//
Ted Kremenekeca099b2010-12-08 23:43:14 +00001123// CXCursorSet.
1124//===----------------------------------------------------------------------===//
1125
1126typedef llvm::DenseMap<CXCursor, unsigned> CXCursorSet_Impl;
1127
1128static inline CXCursorSet packCXCursorSet(CXCursorSet_Impl *setImpl) {
1129 return (CXCursorSet) setImpl;
1130}
1131static inline CXCursorSet_Impl *unpackCXCursorSet(CXCursorSet set) {
1132 return (CXCursorSet_Impl*) set;
1133}
1134namespace llvm {
Ted Kremenekda6fb692010-12-09 00:33:41 +00001135template<> struct DenseMapInfo<CXCursor> {
Ted Kremenekeca099b2010-12-08 23:43:14 +00001136public:
1137 static inline CXCursor getEmptyKey() {
1138 return MakeCXCursorInvalid(CXCursor_InvalidFile);
1139 }
1140 static inline CXCursor getTombstoneKey() {
1141 return MakeCXCursorInvalid(CXCursor_NoDeclFound);
1142 }
1143 static inline unsigned getHashValue(const CXCursor &cursor) {
1144 return llvm::DenseMapInfo<std::pair<void*,void*> >
1145 ::getHashValue(std::make_pair(cursor.data[0], cursor.data[1]));
1146 }
1147 static inline bool isEqual(const CXCursor &x, const CXCursor &y) {
1148 return x.kind == y.kind &&
1149 x.data[0] == y.data[0] &&
1150 x.data[1] == y.data[1];
1151 }
1152};
1153}
1154
1155extern "C" {
1156CXCursorSet clang_createCXCursorSet() {
1157 return packCXCursorSet(new CXCursorSet_Impl());
1158}
1159
1160void clang_disposeCXCursorSet(CXCursorSet set) {
1161 delete unpackCXCursorSet(set);
1162}
1163
1164unsigned clang_CXCursorSet_contains(CXCursorSet set, CXCursor cursor) {
1165 CXCursorSet_Impl *setImpl = unpackCXCursorSet(set);
1166 if (!setImpl)
1167 return 0;
1168 return setImpl->find(cursor) == setImpl->end();
1169}
1170
Anders Carlssone8b3de02010-12-09 01:00:12 +00001171unsigned clang_CXCursorSet_insert(CXCursorSet set, CXCursor cursor) {
Ted Kremenekeca099b2010-12-08 23:43:14 +00001172 // Do not insert invalid cursors into the set.
1173 if (cursor.kind >= CXCursor_FirstInvalid &&
1174 cursor.kind <= CXCursor_LastInvalid)
1175 return 1;
1176
1177 CXCursorSet_Impl *setImpl = unpackCXCursorSet(set);
1178 if (!setImpl)
1179 return 1;
1180 unsigned &entry = (*setImpl)[cursor];
1181 unsigned flag = entry == 0 ? 1 : 0;
1182 entry = 1;
1183 return flag;
1184}
Douglas Gregor8fa0a802011-08-04 20:04:59 +00001185
1186CXCompletionString clang_getCursorCompletionString(CXCursor cursor) {
1187 enum CXCursorKind kind = clang_getCursorKind(cursor);
1188 if (clang_isDeclaration(kind)) {
1189 Decl *decl = getCursorDecl(cursor);
Argyrios Kyrtzidis16ed0e62011-12-10 02:36:25 +00001190 if (NamedDecl *namedDecl = dyn_cast_or_null<NamedDecl>(decl)) {
Douglas Gregor8fa0a802011-08-04 20:04:59 +00001191 ASTUnit *unit = getCursorASTUnit(cursor);
Argyrios Kyrtzidis5e192a72012-01-17 02:15:54 +00001192 CodeCompletionResult Result(namedDecl);
1193 CodeCompletionString *String
1194 = Result.CreateCodeCompletionString(unit->getASTContext(),
1195 unit->getPreprocessor(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00001196 unit->getCodeCompletionTUInfo().getAllocator(),
1197 unit->getCodeCompletionTUInfo());
Argyrios Kyrtzidis5e192a72012-01-17 02:15:54 +00001198 return String;
Douglas Gregor8fa0a802011-08-04 20:04:59 +00001199 }
1200 }
1201 else if (kind == CXCursor_MacroDefinition) {
1202 MacroDefinition *definition = getCursorMacroDefinition(cursor);
1203 const IdentifierInfo *MacroInfo = definition->getName();
1204 ASTUnit *unit = getCursorASTUnit(cursor);
Argyrios Kyrtzidis5e192a72012-01-17 02:15:54 +00001205 CodeCompletionResult Result(const_cast<IdentifierInfo *>(MacroInfo));
1206 CodeCompletionString *String
1207 = Result.CreateCodeCompletionString(unit->getASTContext(),
1208 unit->getPreprocessor(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00001209 unit->getCodeCompletionTUInfo().getAllocator(),
1210 unit->getCodeCompletionTUInfo());
Argyrios Kyrtzidis5e192a72012-01-17 02:15:54 +00001211 return String;
Douglas Gregor8fa0a802011-08-04 20:04:59 +00001212 }
1213 return NULL;
1214}
Ted Kremenek8eece462012-04-30 19:33:45 +00001215} // end: extern C.
Ted Kremenekbbf66ca2012-04-30 19:06:49 +00001216
1217namespace {
1218 struct OverridenCursorsPool {
1219 typedef llvm::SmallVector<CXCursor, 2> CursorVec;
1220 std::vector<CursorVec*> AllCursors;
1221 std::vector<CursorVec*> AvailableCursors;
1222
1223 ~OverridenCursorsPool() {
1224 for (std::vector<CursorVec*>::iterator I = AllCursors.begin(),
1225 E = AllCursors.end(); I != E; ++I) {
1226 delete *I;
1227 }
1228 }
1229 };
1230}
1231
1232void *cxcursor::createOverridenCXCursorsPool() {
1233 return new OverridenCursorsPool();
1234}
1235
1236void cxcursor::disposeOverridenCXCursorsPool(void *pool) {
1237 delete static_cast<OverridenCursorsPool*>(pool);
1238}
Ted Kremenek8eece462012-04-30 19:33:45 +00001239
1240extern "C" {
Ted Kremenekbbf66ca2012-04-30 19:06:49 +00001241void clang_getOverriddenCursors(CXCursor cursor,
1242 CXCursor **overridden,
1243 unsigned *num_overridden) {
1244 if (overridden)
1245 *overridden = 0;
1246 if (num_overridden)
1247 *num_overridden = 0;
1248
1249 CXTranslationUnit TU = cxcursor::getCursorTU(cursor);
1250
1251 if (!overridden || !num_overridden || !TU)
1252 return;
1253
1254 if (!clang_isDeclaration(cursor.kind))
1255 return;
1256
1257 OverridenCursorsPool &pool =
1258 *static_cast<OverridenCursorsPool*>(TU->OverridenCursorsPool);
1259
1260 OverridenCursorsPool::CursorVec *Vec = 0;
1261
1262 if (!pool.AvailableCursors.empty()) {
1263 Vec = pool.AvailableCursors.back();
1264 pool.AvailableCursors.pop_back();
1265 }
1266 else {
1267 Vec = new OverridenCursorsPool::CursorVec();
1268 pool.AllCursors.push_back(Vec);
1269 }
1270
1271 // Clear out the vector, but don't free the memory contents. This
1272 // reduces malloc() traffic.
1273 Vec->clear();
1274
1275 // Use the first entry to contain a back reference to the vector.
1276 // This is a complete hack.
1277 CXCursor backRefCursor = MakeCXCursorInvalid(CXCursor_InvalidFile, TU);
1278 backRefCursor.data[0] = Vec;
1279 assert(cxcursor::getCursorTU(backRefCursor) == TU);
1280 Vec->push_back(backRefCursor);
1281
1282 // Get the overriden cursors.
1283 cxcursor::getOverriddenCursors(cursor, *Vec);
1284
1285 // Did we get any overriden cursors? If not, return Vec to the pool
1286 // of available cursor vectors.
1287 if (Vec->size() == 1) {
1288 pool.AvailableCursors.push_back(Vec);
1289 return;
1290 }
1291
1292 // Now tell the caller about the overriden cursors.
1293 assert(Vec->size() > 1);
1294 *overridden = &((*Vec)[1]);
1295 *num_overridden = Vec->size() - 1;
1296}
1297
1298void clang_disposeOverriddenCursors(CXCursor *overridden) {
1299 if (!overridden)
1300 return;
1301
1302 // Use pointer arithmetic to get back the first faux entry
1303 // which has a back-reference to the TU and the vector.
1304 --overridden;
1305 OverridenCursorsPool::CursorVec *Vec =
1306 static_cast<OverridenCursorsPool::CursorVec*>(overridden->data[0]);
1307 CXTranslationUnit TU = getCursorTU(*overridden);
1308
1309 assert(Vec && TU);
1310
1311 OverridenCursorsPool &pool =
1312 *static_cast<OverridenCursorsPool*>(TU->OverridenCursorsPool);
1313
1314 pool.AvailableCursors.push_back(Vec);
1315}
Douglas Gregor8fa0a802011-08-04 20:04:59 +00001316
Ted Kremenekeca099b2010-12-08 23:43:14 +00001317} // end: extern "C"