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 | { |
| 190 | return new Indexer(*new Program(), *new FileManager()); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 191 | } |
| 192 | |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 193 | // FIXME: need to pass back error info. |
| 194 | CXTranslationUnit clang_createTranslationUnit( |
| 195 | CXIndex CIdx, const char *ast_filename) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 196 | { |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 197 | assert(CIdx && "Passed null CXIndex"); |
| 198 | Indexer *CXXIdx = static_cast<Indexer *>(CIdx); |
| 199 | std::string astName(ast_filename); |
| 200 | std::string ErrMsg; |
| 201 | |
| 202 | return ASTUnit::LoadFromPCHFile(astName, CXXIdx->getFileManager(), &ErrMsg); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 203 | } |
| 204 | |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 205 | const char *clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) |
| 206 | { |
| 207 | assert(CTUnit && "Passed null CXTranslationUnit"); |
Steve Naroff | 77accc1 | 2009-09-03 18:19:54 +0000 | [diff] [blame] | 208 | ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit); |
| 209 | return CXXUnit->getOriginalSourceFileName().c_str(); |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 210 | } |
Daniel Dunbar | 1eb79b5 | 2009-08-28 16:30:07 +0000 | [diff] [blame] | 211 | |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 212 | void clang_loadTranslationUnit(CXTranslationUnit CTUnit, |
| 213 | CXTranslationUnitIterator callback, |
| 214 | CXClientData CData) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 215 | { |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 216 | assert(CTUnit && "Passed null CXTranslationUnit"); |
| 217 | ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit); |
| 218 | ASTContext &Ctx = CXXUnit->getASTContext(); |
| 219 | |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 220 | TUVisitor DVisit(CTUnit, callback, CData); |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 221 | DVisit.Visit(Ctx.getTranslationUnitDecl()); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 222 | } |
| 223 | |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 224 | void clang_loadDeclaration(CXDecl Dcl, |
| 225 | CXDeclIterator callback, |
| 226 | CXClientData CData) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 227 | { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 228 | assert(Dcl && "Passed null CXDecl"); |
| 229 | |
| 230 | CDeclVisitor DVisit(Dcl, callback, CData); |
| 231 | DVisit.Visit(static_cast<Decl *>(Dcl)); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 232 | } |
| 233 | |
Steve Naroff | 7e8f818 | 2009-08-28 12:07:44 +0000 | [diff] [blame] | 234 | // Some notes on CXEntity: |
| 235 | // |
| 236 | // - Since the 'ordinary' namespace includes functions, data, typedefs, |
| 237 | // ObjC interfaces, thecurrent algorithm is a bit naive (resulting in one |
| 238 | // entity for 2 different types). For example: |
| 239 | // |
| 240 | // module1.m: @interface Foo @end Foo *x; |
| 241 | // module2.m: void Foo(int); |
| 242 | // |
| 243 | // - Since the unique name spans translation units, static data/functions |
| 244 | // within a CXTranslationUnit are *not* currently represented by entities. |
| 245 | // As a result, there will be no entity for the following: |
| 246 | // |
| 247 | // module.m: static void Foo() { } |
| 248 | // |
| 249 | |
| 250 | |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 251 | const char *clang_getDeclarationName(CXEntity) |
| 252 | { |
| 253 | return ""; |
| 254 | } |
| 255 | const char *clang_getURI(CXEntity) |
| 256 | { |
| 257 | return ""; |
| 258 | } |
| 259 | |
| 260 | CXEntity clang_getEntity(const char *URI) |
| 261 | { |
| 262 | return 0; |
| 263 | } |
| 264 | |
| 265 | // |
| 266 | // CXDecl Operations. |
| 267 | // |
| 268 | CXCursor clang_getCursorFromDecl(CXDecl) |
| 269 | { |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 270 | return CXCursor(); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 271 | } |
| 272 | CXEntity clang_getEntityFromDecl(CXDecl) |
| 273 | { |
| 274 | return 0; |
| 275 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 276 | const char *clang_getDeclSpelling(CXDecl AnonDecl) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 277 | { |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 278 | assert(AnonDecl && "Passed null CXDecl"); |
| 279 | NamedDecl *ND = static_cast<NamedDecl *>(AnonDecl); |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 280 | |
| 281 | if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND)) { |
| 282 | return OMD->getSelector().getAsString().c_str(); |
| 283 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 284 | if (ND->getIdentifier()) |
| 285 | return ND->getIdentifier()->getName(); |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 286 | else |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 287 | return ""; |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 288 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 289 | |
| 290 | const char *clang_getCursorSpelling(CXCursor C) |
| 291 | { |
| 292 | assert(C.decl && "CXCursor has null decl"); |
| 293 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
| 294 | |
| 295 | if (clang_isReference(C.kind)) { |
| 296 | switch (C.kind) { |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 297 | case CXCursor_ObjCSuperClassRef: |
| 298 | { |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 299 | ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(ND); |
| 300 | assert(OID && "clang_getCursorLine(): Missing interface decl"); |
| 301 | return OID->getSuperClass()->getIdentifier()->getName(); |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 302 | } |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 303 | case CXCursor_ObjCClassRef: |
| 304 | { |
| 305 | ObjCCategoryDecl *OID = dyn_cast<ObjCCategoryDecl>(ND); |
| 306 | assert(OID && "clang_getCursorLine(): Missing category decl"); |
| 307 | return OID->getClassInterface()->getIdentifier()->getName(); |
| 308 | } |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 309 | case CXCursor_ObjCProtocolRef: |
| 310 | { |
| 311 | ObjCProtocolDecl *OID = dyn_cast<ObjCProtocolDecl>(ND); |
| 312 | assert(OID && "clang_getCursorLine(): Missing protocol decl"); |
| 313 | return OID->getIdentifier()->getName(); |
| 314 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 315 | default: |
| 316 | return "<not implemented>"; |
| 317 | } |
| 318 | } |
| 319 | return clang_getDeclSpelling(C.decl); |
| 320 | } |
| 321 | |
| 322 | const char *clang_getCursorKindSpelling(enum CXCursorKind Kind) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 323 | { |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 324 | switch (Kind) { |
| 325 | case CXCursor_FunctionDecl: return "FunctionDecl"; |
| 326 | case CXCursor_TypedefDecl: return "TypedefDecl"; |
| 327 | case CXCursor_EnumDecl: return "EnumDecl"; |
| 328 | case CXCursor_EnumConstantDecl: return "EnumConstantDecl"; |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 329 | case CXCursor_StructDecl: return "StructDecl"; |
| 330 | case CXCursor_UnionDecl: return "UnionDecl"; |
| 331 | case CXCursor_ClassDecl: return "ClassDecl"; |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 332 | case CXCursor_FieldDecl: return "FieldDecl"; |
| 333 | case CXCursor_VarDecl: return "VarDecl"; |
| 334 | case CXCursor_ParmDecl: return "ParmDecl"; |
| 335 | case CXCursor_ObjCInterfaceDecl: return "ObjCInterfaceDecl"; |
| 336 | case CXCursor_ObjCCategoryDecl: return "ObjCCategoryDecl"; |
| 337 | case CXCursor_ObjCProtocolDecl: return "ObjCProtocolDecl"; |
| 338 | case CXCursor_ObjCPropertyDecl: return "ObjCPropertyDecl"; |
| 339 | case CXCursor_ObjCIvarDecl: return "ObjCIvarDecl"; |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 340 | case CXCursor_ObjCInstanceMethodDecl: return "ObjCInstanceMethodDecl"; |
| 341 | case CXCursor_ObjCClassMethodDecl: return "ObjCClassMethodDecl"; |
| 342 | case CXCursor_FunctionDefn: return "FunctionDefn"; |
| 343 | case CXCursor_ObjCInstanceMethodDefn: return "ObjCInstanceMethodDefn"; |
| 344 | case CXCursor_ObjCClassMethodDefn: return "ObjCClassMethodDefn"; |
| 345 | case CXCursor_ObjCClassDefn: return "ObjCClassDefn"; |
| 346 | case CXCursor_ObjCCategoryDefn: return "ObjCCategoryDefn"; |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 347 | case CXCursor_ObjCSuperClassRef: return "ObjCSuperClassRef"; |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 348 | case CXCursor_ObjCProtocolRef: return "ObjCProtocolRef"; |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 349 | case CXCursor_ObjCClassRef: return "ObjCClassRef"; |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 350 | default: return "<not implemented>"; |
| 351 | } |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 352 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 353 | |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 354 | static enum CXCursorKind TranslateKind(Decl *D) { |
| 355 | switch (D->getKind()) { |
| 356 | case Decl::Function: return CXCursor_FunctionDecl; |
| 357 | case Decl::Typedef: return CXCursor_TypedefDecl; |
| 358 | case Decl::Enum: return CXCursor_EnumDecl; |
| 359 | case Decl::EnumConstant: return CXCursor_EnumConstantDecl; |
| 360 | case Decl::Record: return CXCursor_StructDecl; // FIXME: union/class |
| 361 | case Decl::Field: return CXCursor_FieldDecl; |
| 362 | case Decl::Var: return CXCursor_VarDecl; |
| 363 | case Decl::ParmVar: return CXCursor_ParmDecl; |
| 364 | case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl; |
| 365 | case Decl::ObjCMethod: { |
| 366 | ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D); |
| 367 | if (MD->isInstanceMethod()) |
| 368 | return CXCursor_ObjCInstanceMethodDecl; |
| 369 | return CXCursor_ObjCClassMethodDecl; |
| 370 | } |
| 371 | default: break; |
| 372 | } |
| 373 | return CXCursor_Invalid; |
| 374 | } |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 375 | // |
| 376 | // CXCursor Operations. |
| 377 | // |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 378 | CXCursor clang_getCursor(CXTranslationUnit CTUnit, const char *source_name, |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 379 | unsigned line, unsigned column) |
| 380 | { |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 381 | assert(CTUnit && "Passed null CXTranslationUnit"); |
| 382 | ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit); |
| 383 | |
| 384 | FileManager &FMgr = CXXUnit->getFileManager(); |
| 385 | const FileEntry *File = FMgr.getFile(source_name, |
| 386 | source_name+strlen(source_name)); |
| 387 | assert(File && "clang_getCursor(): FileManager returned 0"); |
| 388 | |
| 389 | SourceLocation SLoc = |
| 390 | CXXUnit->getSourceManager().getLocation(File, line, column); |
| 391 | |
| 392 | ASTLocation ALoc = ResolveLocationInAST(CXXUnit->getASTContext(), SLoc); |
| 393 | |
| 394 | Decl *Dcl = ALoc.getDecl(); |
| 395 | assert(Dcl && "clang_getCursor(): ASTLocation has a null decl"); |
| 396 | |
| 397 | CXCursor C = { TranslateKind(Dcl), Dcl }; |
| 398 | return C; |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 399 | } |
| 400 | |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 401 | unsigned clang_isDeclaration(enum CXCursorKind K) |
| 402 | { |
| 403 | return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl; |
| 404 | } |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 405 | |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 406 | unsigned clang_isReference(enum CXCursorKind K) |
| 407 | { |
| 408 | return K >= CXCursor_FirstRef && K <= CXCursor_LastRef; |
| 409 | } |
| 410 | |
| 411 | unsigned clang_isDefinition(enum CXCursorKind K) |
| 412 | { |
| 413 | return K >= CXCursor_FirstDefn && K <= CXCursor_LastDefn; |
| 414 | } |
| 415 | |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 416 | CXCursorKind clang_getCursorKind(CXCursor C) |
| 417 | { |
| 418 | return C.kind; |
| 419 | } |
| 420 | |
| 421 | CXDecl clang_getCursorDecl(CXCursor C) |
| 422 | { |
| 423 | return C.decl; |
| 424 | } |
| 425 | |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 426 | static SourceLocation getLocationFromCursor(CXCursor C, |
| 427 | SourceManager &SourceMgr, |
| 428 | NamedDecl *ND) { |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 429 | if (clang_isReference(C.kind)) { |
| 430 | switch (C.kind) { |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 431 | case CXCursor_ObjCSuperClassRef: |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 432 | { |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 433 | ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(ND); |
| 434 | assert(OID && "clang_getCursorLine(): Missing interface decl"); |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 435 | return OID->getSuperClassLoc(); |
| 436 | } |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 437 | case CXCursor_ObjCProtocolRef: |
| 438 | { |
| 439 | ObjCProtocolDecl *OID = dyn_cast<ObjCProtocolDecl>(ND); |
| 440 | assert(OID && "clang_getCursorLine(): Missing protocol decl"); |
| 441 | return OID->getLocation(); |
| 442 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 443 | default: |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 444 | return SourceLocation(); |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 445 | } |
| 446 | } else { // We have a declaration or a definition. |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 447 | SourceLocation SLoc; |
| 448 | switch (ND->getKind()) { |
| 449 | case Decl::ObjCInterface: |
| 450 | { |
| 451 | SLoc = dyn_cast<ObjCInterfaceDecl>(ND)->getClassLoc(); |
| 452 | break; |
| 453 | } |
| 454 | case Decl::ObjCProtocol: |
| 455 | { |
| 456 | SLoc = ND->getLocation(); /* FIXME: need to get the name location. */ |
| 457 | break; |
| 458 | } |
| 459 | default: |
| 460 | { |
| 461 | SLoc = ND->getLocation(); |
| 462 | break; |
| 463 | } |
| 464 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 465 | if (SLoc.isInvalid()) |
| 466 | return SourceLocation(); |
Steve Naroff | 1164d85 | 2009-09-02 18:58:52 +0000 | [diff] [blame] | 467 | return SourceMgr.getSpellingLoc(SLoc); // handles macro instantiations. |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 468 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 469 | } |
| 470 | |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 471 | unsigned clang_getCursorLine(CXCursor C) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 472 | { |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 473 | assert(C.decl && "CXCursor has null decl"); |
| 474 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 475 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 476 | |
| 477 | SourceLocation SLoc = getLocationFromCursor(C, SourceMgr, ND); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 478 | return SourceMgr.getSpellingLineNumber(SLoc); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 479 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 480 | |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 481 | unsigned clang_getCursorColumn(CXCursor C) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 482 | { |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 483 | assert(C.decl && "CXCursor has null decl"); |
| 484 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 485 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 486 | |
| 487 | SourceLocation SLoc = getLocationFromCursor(C, SourceMgr, ND); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 488 | return SourceMgr.getSpellingColumnNumber(SLoc); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 489 | } |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 490 | const char *clang_getCursorSource(CXCursor C) |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 491 | { |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 492 | assert(C.decl && "CXCursor has null decl"); |
| 493 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 494 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 495 | |
| 496 | SourceLocation SLoc = getLocationFromCursor(C, SourceMgr, ND); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 497 | return SourceMgr.getBufferName(SLoc); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 498 | } |
| 499 | |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 500 | } // end extern "C" |