Ted Kremenek | d2fa566 | 2009-08-26 22:36:44 +0000 | [diff] [blame] | 1 | //===- CIndex.cpp - Clang-C Source Indexing Library -----------------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file implements the Clang-C Source Indexing library. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
| 14 | #include "clang-c/Index.h" |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 15 | #include "clang/Index/Program.h" |
| 16 | #include "clang/Index/Indexer.h" |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 17 | #include "clang/Index/ASTLocation.h" |
| 18 | #include "clang/Index/Utils.h" |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 19 | #include "clang/AST/DeclVisitor.h" |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 20 | #include "clang/AST/Decl.h" |
Benjamin Kramer | d01a0bc | 2009-08-29 12:56:35 +0000 | [diff] [blame] | 21 | #include "clang/Basic/FileManager.h" |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 22 | #include "clang/Basic/SourceManager.h" |
Benjamin Kramer | d01a0bc | 2009-08-29 12:56:35 +0000 | [diff] [blame] | 23 | #include "clang/Frontend/ASTUnit.h" |
| 24 | #include <cstdio> |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 25 | using namespace clang; |
| 26 | using namespace idx; |
| 27 | |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 28 | namespace { |
| 29 | |
| 30 | // Translation Unit Visitor. |
| 31 | class TUVisitor : public DeclVisitor<TUVisitor> { |
| 32 | CXTranslationUnit TUnit; |
| 33 | CXTranslationUnitIterator Callback; |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 34 | CXClientData CData; |
| 35 | |
| 36 | void Call(enum CXCursorKind CK, NamedDecl *ND) { |
| 37 | CXCursor C = { CK, ND }; |
| 38 | Callback(TUnit, C, CData); |
| 39 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 40 | public: |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 41 | TUVisitor(CXTranslationUnit CTU, |
| 42 | CXTranslationUnitIterator cback, CXClientData D) : |
| 43 | TUnit(CTU), Callback(cback), CData(D) {} |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 44 | |
| 45 | void VisitTranslationUnitDecl(TranslationUnitDecl *D) { |
| 46 | VisitDeclContext(dyn_cast<DeclContext>(D)); |
| 47 | } |
| 48 | void VisitDeclContext(DeclContext *DC) { |
| 49 | for (DeclContext::decl_iterator |
| 50 | I = DC->decls_begin(), E = DC->decls_end(); I != E; ++I) |
| 51 | Visit(*I); |
| 52 | } |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 53 | void VisitTypedefDecl(TypedefDecl *ND) { |
| 54 | Call(CXCursor_TypedefDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 55 | } |
| 56 | void VisitTagDecl(TagDecl *ND) { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 57 | switch (ND->getTagKind()) { |
| 58 | case TagDecl::TK_struct: |
| 59 | Call(CXCursor_StructDecl, ND); |
| 60 | break; |
| 61 | case TagDecl::TK_class: |
| 62 | Call(CXCursor_ClassDecl, ND); |
| 63 | break; |
| 64 | case TagDecl::TK_union: |
| 65 | Call(CXCursor_UnionDecl, ND); |
| 66 | break; |
| 67 | case TagDecl::TK_enum: |
| 68 | Call(CXCursor_EnumDecl, ND); |
| 69 | break; |
| 70 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 71 | } |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 72 | void VisitVarDecl(VarDecl *ND) { |
| 73 | Call(CXCursor_VarDecl, ND); |
| 74 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 75 | void VisitFunctionDecl(FunctionDecl *ND) { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 76 | Call(ND->isThisDeclarationADefinition() ? CXCursor_FunctionDefn |
| 77 | : CXCursor_FunctionDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 78 | } |
| 79 | void VisitObjCInterfaceDecl(ObjCInterfaceDecl *ND) { |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 80 | Call(CXCursor_ObjCInterfaceDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 81 | } |
| 82 | void VisitObjCCategoryDecl(ObjCCategoryDecl *ND) { |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 83 | Call(CXCursor_ObjCCategoryDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 84 | } |
| 85 | void VisitObjCProtocolDecl(ObjCProtocolDecl *ND) { |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 86 | Call(CXCursor_ObjCProtocolDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 87 | } |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 88 | void VisitObjCImplementationDecl(ObjCImplementationDecl *ND) { |
| 89 | Call(CXCursor_ObjCClassDefn, ND); |
| 90 | } |
| 91 | void VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *ND) { |
| 92 | Call(CXCursor_ObjCCategoryDefn, ND); |
| 93 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 94 | }; |
| 95 | |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 96 | // Declaration visitor. |
| 97 | class CDeclVisitor : public DeclVisitor<CDeclVisitor> { |
| 98 | CXDecl CDecl; |
| 99 | CXDeclIterator Callback; |
| 100 | CXClientData CData; |
| 101 | |
| 102 | void Call(enum CXCursorKind CK, NamedDecl *ND) { |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 103 | // Disable the callback when the context is equal to the visiting decl. |
| 104 | if (CDecl == ND && !clang_isReference(CK)) |
| 105 | return; |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 106 | CXCursor C = { CK, ND }; |
| 107 | Callback(CDecl, C, CData); |
| 108 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 109 | public: |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 110 | CDeclVisitor(CXDecl C, CXDeclIterator cback, CXClientData D) : |
| 111 | CDecl(C), Callback(cback), CData(D) {} |
| 112 | |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 113 | void VisitObjCCategoryDecl(ObjCCategoryDecl *ND) { |
| 114 | // Issue callbacks for the containing class. |
| 115 | Call(CXCursor_ObjCClassRef, ND); |
| 116 | // FIXME: Issue callbacks for protocol refs. |
| 117 | VisitDeclContext(dyn_cast<DeclContext>(ND)); |
| 118 | } |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 119 | void VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) { |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 120 | // Issue callbacks for super class. |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 121 | if (D->getSuperClass()) |
| 122 | Call(CXCursor_ObjCSuperClassRef, D); |
| 123 | |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 124 | for (ObjCProtocolDecl::protocol_iterator I = D->protocol_begin(), |
| 125 | E = D->protocol_end(); I != E; ++I) |
| 126 | Call(CXCursor_ObjCProtocolRef, *I); |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 127 | VisitDeclContext(dyn_cast<DeclContext>(D)); |
| 128 | } |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 129 | void VisitObjCProtocolDecl(ObjCProtocolDecl *PID) { |
| 130 | for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(), |
| 131 | E = PID->protocol_end(); I != E; ++I) |
| 132 | Call(CXCursor_ObjCProtocolRef, *I); |
| 133 | |
| 134 | VisitDeclContext(dyn_cast<DeclContext>(PID)); |
| 135 | } |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 136 | void VisitTagDecl(TagDecl *D) { |
| 137 | VisitDeclContext(dyn_cast<DeclContext>(D)); |
| 138 | } |
| 139 | void VisitObjCImplementationDecl(ObjCImplementationDecl *D) { |
| 140 | VisitDeclContext(dyn_cast<DeclContext>(D)); |
| 141 | } |
| 142 | void VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) { |
| 143 | VisitDeclContext(dyn_cast<DeclContext>(D)); |
| 144 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 145 | void VisitDeclContext(DeclContext *DC) { |
| 146 | for (DeclContext::decl_iterator |
| 147 | I = DC->decls_begin(), E = DC->decls_end(); I != E; ++I) |
| 148 | Visit(*I); |
| 149 | } |
| 150 | void VisitEnumConstantDecl(EnumConstantDecl *ND) { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 151 | Call(CXCursor_EnumConstantDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 152 | } |
| 153 | void VisitFieldDecl(FieldDecl *ND) { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 154 | Call(CXCursor_FieldDecl, ND); |
| 155 | } |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 156 | void VisitVarDecl(VarDecl *ND) { |
| 157 | Call(CXCursor_VarDecl, ND); |
| 158 | } |
| 159 | void VisitParmVarDecl(ParmVarDecl *ND) { |
| 160 | Call(CXCursor_ParmDecl, ND); |
| 161 | } |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 162 | void VisitObjCPropertyDecl(ObjCPropertyDecl *ND) { |
| 163 | Call(CXCursor_ObjCPropertyDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 164 | } |
| 165 | void VisitObjCIvarDecl(ObjCIvarDecl *ND) { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 166 | Call(CXCursor_ObjCIvarDecl, ND); |
| 167 | } |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 168 | void VisitFunctionDecl(FunctionDecl *ND) { |
| 169 | if (ND->isThisDeclarationADefinition()) { |
| 170 | VisitDeclContext(dyn_cast<DeclContext>(ND)); |
| 171 | } |
| 172 | } |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 173 | void VisitObjCMethodDecl(ObjCMethodDecl *ND) { |
| 174 | if (ND->getBody()) { |
| 175 | Call(ND->isInstanceMethod() ? CXCursor_ObjCInstanceMethodDefn |
| 176 | : CXCursor_ObjCClassMethodDefn, ND); |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 177 | VisitDeclContext(dyn_cast<DeclContext>(ND)); |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 178 | } else |
| 179 | Call(ND->isInstanceMethod() ? CXCursor_ObjCInstanceMethodDecl |
| 180 | : CXCursor_ObjCClassMethodDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 181 | } |
| 182 | }; |
| 183 | |
| 184 | } |
| 185 | |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 186 | extern "C" { |
Ted Kremenek | d2fa566 | 2009-08-26 22:36:44 +0000 | [diff] [blame] | 187 | |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 188 | CXIndex clang_createIndex() |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 189 | { |
Daniel Dunbar | a390759 | 2009-09-21 03:03:22 +0000 | [diff] [blame^] | 190 | // FIXME: Program is leaked. |
| 191 | return new Indexer(*new Program()); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 192 | } |
| 193 | |
Steve Naroff | 2bd6b9f | 2009-09-17 18:33:27 +0000 | [diff] [blame] | 194 | void clang_disposeIndex(CXIndex CIdx) |
| 195 | { |
| 196 | assert(CIdx && "Passed null CXIndex"); |
| 197 | delete static_cast<Indexer *>(CIdx); |
| 198 | } |
| 199 | |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 200 | // FIXME: need to pass back error info. |
| 201 | CXTranslationUnit clang_createTranslationUnit( |
| 202 | CXIndex CIdx, const char *ast_filename) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 203 | { |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 204 | assert(CIdx && "Passed null CXIndex"); |
| 205 | Indexer *CXXIdx = static_cast<Indexer *>(CIdx); |
| 206 | std::string astName(ast_filename); |
| 207 | std::string ErrMsg; |
| 208 | |
| 209 | return ASTUnit::LoadFromPCHFile(astName, CXXIdx->getFileManager(), &ErrMsg); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 210 | } |
| 211 | |
Steve Naroff | 2bd6b9f | 2009-09-17 18:33:27 +0000 | [diff] [blame] | 212 | void clang_disposeTranslationUnit( |
| 213 | CXTranslationUnit CTUnit) |
| 214 | { |
| 215 | assert(CTUnit && "Passed null CXTranslationUnit"); |
| 216 | delete static_cast<ASTUnit *>(CTUnit); |
| 217 | } |
| 218 | |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 219 | const char *clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) |
| 220 | { |
| 221 | assert(CTUnit && "Passed null CXTranslationUnit"); |
Steve Naroff | 77accc1 | 2009-09-03 18:19:54 +0000 | [diff] [blame] | 222 | ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit); |
| 223 | return CXXUnit->getOriginalSourceFileName().c_str(); |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 224 | } |
Daniel Dunbar | 1eb79b5 | 2009-08-28 16:30:07 +0000 | [diff] [blame] | 225 | |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 226 | void clang_loadTranslationUnit(CXTranslationUnit CTUnit, |
| 227 | CXTranslationUnitIterator callback, |
| 228 | CXClientData CData) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 229 | { |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 230 | assert(CTUnit && "Passed null CXTranslationUnit"); |
| 231 | ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit); |
| 232 | ASTContext &Ctx = CXXUnit->getASTContext(); |
| 233 | |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 234 | TUVisitor DVisit(CTUnit, callback, CData); |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 235 | DVisit.Visit(Ctx.getTranslationUnitDecl()); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 236 | } |
| 237 | |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 238 | void clang_loadDeclaration(CXDecl Dcl, |
| 239 | CXDeclIterator callback, |
| 240 | CXClientData CData) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 241 | { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 242 | assert(Dcl && "Passed null CXDecl"); |
| 243 | |
| 244 | CDeclVisitor DVisit(Dcl, callback, CData); |
| 245 | DVisit.Visit(static_cast<Decl *>(Dcl)); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 246 | } |
| 247 | |
Steve Naroff | 7e8f818 | 2009-08-28 12:07:44 +0000 | [diff] [blame] | 248 | // Some notes on CXEntity: |
| 249 | // |
| 250 | // - Since the 'ordinary' namespace includes functions, data, typedefs, |
| 251 | // ObjC interfaces, thecurrent algorithm is a bit naive (resulting in one |
| 252 | // entity for 2 different types). For example: |
| 253 | // |
| 254 | // module1.m: @interface Foo @end Foo *x; |
| 255 | // module2.m: void Foo(int); |
| 256 | // |
| 257 | // - Since the unique name spans translation units, static data/functions |
| 258 | // within a CXTranslationUnit are *not* currently represented by entities. |
| 259 | // As a result, there will be no entity for the following: |
| 260 | // |
| 261 | // module.m: static void Foo() { } |
| 262 | // |
| 263 | |
| 264 | |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 265 | const char *clang_getDeclarationName(CXEntity) |
| 266 | { |
| 267 | return ""; |
| 268 | } |
| 269 | const char *clang_getURI(CXEntity) |
| 270 | { |
| 271 | return ""; |
| 272 | } |
| 273 | |
| 274 | CXEntity clang_getEntity(const char *URI) |
| 275 | { |
| 276 | return 0; |
| 277 | } |
| 278 | |
| 279 | // |
| 280 | // CXDecl Operations. |
| 281 | // |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 282 | CXEntity clang_getEntityFromDecl(CXDecl) |
| 283 | { |
| 284 | return 0; |
| 285 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 286 | const char *clang_getDeclSpelling(CXDecl AnonDecl) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 287 | { |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 288 | assert(AnonDecl && "Passed null CXDecl"); |
| 289 | NamedDecl *ND = static_cast<NamedDecl *>(AnonDecl); |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 290 | |
| 291 | if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND)) { |
| 292 | return OMD->getSelector().getAsString().c_str(); |
| 293 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 294 | if (ND->getIdentifier()) |
| 295 | return ND->getIdentifier()->getName(); |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 296 | else |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 297 | return ""; |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 298 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 299 | |
| 300 | const char *clang_getCursorSpelling(CXCursor C) |
| 301 | { |
| 302 | assert(C.decl && "CXCursor has null decl"); |
| 303 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
| 304 | |
| 305 | if (clang_isReference(C.kind)) { |
| 306 | switch (C.kind) { |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 307 | case CXCursor_ObjCSuperClassRef: |
| 308 | { |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 309 | ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(ND); |
| 310 | assert(OID && "clang_getCursorLine(): Missing interface decl"); |
| 311 | return OID->getSuperClass()->getIdentifier()->getName(); |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 312 | } |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 313 | case CXCursor_ObjCClassRef: |
| 314 | { |
| 315 | ObjCCategoryDecl *OID = dyn_cast<ObjCCategoryDecl>(ND); |
| 316 | assert(OID && "clang_getCursorLine(): Missing category decl"); |
| 317 | return OID->getClassInterface()->getIdentifier()->getName(); |
| 318 | } |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 319 | case CXCursor_ObjCProtocolRef: |
| 320 | { |
| 321 | ObjCProtocolDecl *OID = dyn_cast<ObjCProtocolDecl>(ND); |
| 322 | assert(OID && "clang_getCursorLine(): Missing protocol decl"); |
| 323 | return OID->getIdentifier()->getName(); |
| 324 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 325 | default: |
| 326 | return "<not implemented>"; |
| 327 | } |
| 328 | } |
| 329 | return clang_getDeclSpelling(C.decl); |
| 330 | } |
| 331 | |
| 332 | const char *clang_getCursorKindSpelling(enum CXCursorKind Kind) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 333 | { |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 334 | switch (Kind) { |
| 335 | case CXCursor_FunctionDecl: return "FunctionDecl"; |
| 336 | case CXCursor_TypedefDecl: return "TypedefDecl"; |
| 337 | case CXCursor_EnumDecl: return "EnumDecl"; |
| 338 | case CXCursor_EnumConstantDecl: return "EnumConstantDecl"; |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 339 | case CXCursor_StructDecl: return "StructDecl"; |
| 340 | case CXCursor_UnionDecl: return "UnionDecl"; |
| 341 | case CXCursor_ClassDecl: return "ClassDecl"; |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 342 | case CXCursor_FieldDecl: return "FieldDecl"; |
| 343 | case CXCursor_VarDecl: return "VarDecl"; |
| 344 | case CXCursor_ParmDecl: return "ParmDecl"; |
| 345 | case CXCursor_ObjCInterfaceDecl: return "ObjCInterfaceDecl"; |
| 346 | case CXCursor_ObjCCategoryDecl: return "ObjCCategoryDecl"; |
| 347 | case CXCursor_ObjCProtocolDecl: return "ObjCProtocolDecl"; |
| 348 | case CXCursor_ObjCPropertyDecl: return "ObjCPropertyDecl"; |
| 349 | case CXCursor_ObjCIvarDecl: return "ObjCIvarDecl"; |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 350 | case CXCursor_ObjCInstanceMethodDecl: return "ObjCInstanceMethodDecl"; |
| 351 | case CXCursor_ObjCClassMethodDecl: return "ObjCClassMethodDecl"; |
| 352 | case CXCursor_FunctionDefn: return "FunctionDefn"; |
| 353 | case CXCursor_ObjCInstanceMethodDefn: return "ObjCInstanceMethodDefn"; |
| 354 | case CXCursor_ObjCClassMethodDefn: return "ObjCClassMethodDefn"; |
| 355 | case CXCursor_ObjCClassDefn: return "ObjCClassDefn"; |
| 356 | case CXCursor_ObjCCategoryDefn: return "ObjCCategoryDefn"; |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 357 | case CXCursor_ObjCSuperClassRef: return "ObjCSuperClassRef"; |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 358 | case CXCursor_ObjCProtocolRef: return "ObjCProtocolRef"; |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 359 | case CXCursor_ObjCClassRef: return "ObjCClassRef"; |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 360 | case CXCursor_InvalidFile: return "InvalidFile"; |
| 361 | case CXCursor_NoDeclFound: return "NoDeclFound"; |
| 362 | case CXCursor_NotImplemented: return "NotImplemented"; |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 363 | default: return "<not implemented>"; |
| 364 | } |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 365 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 366 | |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 367 | static enum CXCursorKind TranslateKind(Decl *D) { |
| 368 | switch (D->getKind()) { |
| 369 | case Decl::Function: return CXCursor_FunctionDecl; |
| 370 | case Decl::Typedef: return CXCursor_TypedefDecl; |
| 371 | case Decl::Enum: return CXCursor_EnumDecl; |
| 372 | case Decl::EnumConstant: return CXCursor_EnumConstantDecl; |
| 373 | case Decl::Record: return CXCursor_StructDecl; // FIXME: union/class |
| 374 | case Decl::Field: return CXCursor_FieldDecl; |
| 375 | case Decl::Var: return CXCursor_VarDecl; |
| 376 | case Decl::ParmVar: return CXCursor_ParmDecl; |
| 377 | case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl; |
| 378 | case Decl::ObjCMethod: { |
| 379 | ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D); |
| 380 | if (MD->isInstanceMethod()) |
| 381 | return CXCursor_ObjCInstanceMethodDecl; |
| 382 | return CXCursor_ObjCClassMethodDecl; |
| 383 | } |
| 384 | default: break; |
| 385 | } |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 386 | return CXCursor_NotImplemented; |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 387 | } |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 388 | // |
| 389 | // CXCursor Operations. |
| 390 | // |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 391 | CXCursor clang_getCursor(CXTranslationUnit CTUnit, const char *source_name, |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 392 | unsigned line, unsigned column) |
| 393 | { |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 394 | assert(CTUnit && "Passed null CXTranslationUnit"); |
| 395 | ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit); |
| 396 | |
| 397 | FileManager &FMgr = CXXUnit->getFileManager(); |
| 398 | const FileEntry *File = FMgr.getFile(source_name, |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 399 | source_name+strlen(source_name)); |
| 400 | if (!File) { |
| 401 | CXCursor C = { CXCursor_InvalidFile, 0 }; |
| 402 | return C; |
| 403 | } |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 404 | SourceLocation SLoc = |
| 405 | CXXUnit->getSourceManager().getLocation(File, line, column); |
| 406 | |
| 407 | ASTLocation ALoc = ResolveLocationInAST(CXXUnit->getASTContext(), SLoc); |
| 408 | |
| 409 | Decl *Dcl = ALoc.getDecl(); |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 410 | if (Dcl) { |
| 411 | CXCursor C = { TranslateKind(Dcl), Dcl }; |
| 412 | return C; |
| 413 | } |
| 414 | CXCursor C = { CXCursor_NoDeclFound, 0 }; |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 415 | return C; |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 416 | } |
| 417 | |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 418 | CXCursor clang_getCursorFromDecl(CXDecl AnonDecl) |
| 419 | { |
| 420 | assert(AnonDecl && "Passed null CXDecl"); |
| 421 | NamedDecl *ND = static_cast<NamedDecl *>(AnonDecl); |
| 422 | |
| 423 | CXCursor C = { TranslateKind(ND), ND }; |
| 424 | return C; |
| 425 | } |
| 426 | |
| 427 | unsigned clang_isInvalid(enum CXCursorKind K) |
| 428 | { |
| 429 | return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid; |
| 430 | } |
| 431 | |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 432 | unsigned clang_isDeclaration(enum CXCursorKind K) |
| 433 | { |
| 434 | return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl; |
| 435 | } |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 436 | |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 437 | unsigned clang_isReference(enum CXCursorKind K) |
| 438 | { |
| 439 | return K >= CXCursor_FirstRef && K <= CXCursor_LastRef; |
| 440 | } |
| 441 | |
| 442 | unsigned clang_isDefinition(enum CXCursorKind K) |
| 443 | { |
| 444 | return K >= CXCursor_FirstDefn && K <= CXCursor_LastDefn; |
| 445 | } |
| 446 | |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 447 | CXCursorKind clang_getCursorKind(CXCursor C) |
| 448 | { |
| 449 | return C.kind; |
| 450 | } |
| 451 | |
| 452 | CXDecl clang_getCursorDecl(CXCursor C) |
| 453 | { |
| 454 | return C.decl; |
| 455 | } |
| 456 | |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 457 | static SourceLocation getLocationFromCursor(CXCursor C, |
| 458 | SourceManager &SourceMgr, |
| 459 | NamedDecl *ND) { |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 460 | if (clang_isReference(C.kind)) { |
| 461 | switch (C.kind) { |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 462 | case CXCursor_ObjCSuperClassRef: |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 463 | { |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 464 | ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(ND); |
| 465 | assert(OID && "clang_getCursorLine(): Missing interface decl"); |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 466 | return OID->getSuperClassLoc(); |
| 467 | } |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 468 | case CXCursor_ObjCProtocolRef: |
| 469 | { |
| 470 | ObjCProtocolDecl *OID = dyn_cast<ObjCProtocolDecl>(ND); |
| 471 | assert(OID && "clang_getCursorLine(): Missing protocol decl"); |
| 472 | return OID->getLocation(); |
| 473 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 474 | default: |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 475 | return SourceLocation(); |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 476 | } |
| 477 | } else { // We have a declaration or a definition. |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 478 | SourceLocation SLoc; |
| 479 | switch (ND->getKind()) { |
| 480 | case Decl::ObjCInterface: |
| 481 | { |
| 482 | SLoc = dyn_cast<ObjCInterfaceDecl>(ND)->getClassLoc(); |
| 483 | break; |
| 484 | } |
| 485 | case Decl::ObjCProtocol: |
| 486 | { |
| 487 | SLoc = ND->getLocation(); /* FIXME: need to get the name location. */ |
| 488 | break; |
| 489 | } |
| 490 | default: |
| 491 | { |
| 492 | SLoc = ND->getLocation(); |
| 493 | break; |
| 494 | } |
| 495 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 496 | if (SLoc.isInvalid()) |
| 497 | return SourceLocation(); |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 498 | return SourceMgr.getSpellingLoc(SLoc); // handles macro instantiations. |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 499 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 500 | } |
| 501 | |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 502 | unsigned clang_getCursorLine(CXCursor C) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 503 | { |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 504 | assert(C.decl && "CXCursor has null decl"); |
| 505 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 506 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 507 | |
| 508 | SourceLocation SLoc = getLocationFromCursor(C, SourceMgr, ND); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 509 | return SourceMgr.getSpellingLineNumber(SLoc); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 510 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 511 | |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 512 | unsigned clang_getCursorColumn(CXCursor C) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 513 | { |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 514 | assert(C.decl && "CXCursor has null decl"); |
| 515 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 516 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 517 | |
| 518 | SourceLocation SLoc = getLocationFromCursor(C, SourceMgr, ND); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 519 | return SourceMgr.getSpellingColumnNumber(SLoc); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 520 | } |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 521 | const char *clang_getCursorSource(CXCursor C) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 522 | { |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 523 | assert(C.decl && "CXCursor has null decl"); |
| 524 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 525 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 526 | |
| 527 | SourceLocation SLoc = getLocationFromCursor(C, SourceMgr, ND); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 528 | return SourceMgr.getBufferName(SLoc); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 529 | } |
| 530 | |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 531 | } // end extern "C" |