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. |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 7 | // |
Ted Kremenek | d2fa566 | 2009-08-26 22:36:44 +0000 | [diff] [blame] | 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" |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 19 | #include "clang/Sema/CodeCompleteConsumer.h" |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 20 | #include "clang/AST/DeclVisitor.h" |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 21 | #include "clang/AST/StmtVisitor.h" |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 22 | #include "clang/AST/Decl.h" |
Benjamin Kramer | d01a0bc | 2009-08-29 12:56:35 +0000 | [diff] [blame] | 23 | #include "clang/Basic/FileManager.h" |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 24 | #include "clang/Basic/SourceManager.h" |
Benjamin Kramer | d01a0bc | 2009-08-29 12:56:35 +0000 | [diff] [blame] | 25 | #include "clang/Frontend/ASTUnit.h" |
Daniel Dunbar | 5262fda | 2009-12-03 01:45:44 +0000 | [diff] [blame^] | 26 | #include "clang/Frontend/CompilerInstance.h" |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 27 | #include "llvm/ADT/StringExtras.h" |
Benjamin Kramer | 20d7581 | 2009-10-18 16:13:48 +0000 | [diff] [blame] | 28 | #include "llvm/Config/config.h" |
Ted Kremenek | fc06221 | 2009-10-19 21:44:57 +0000 | [diff] [blame] | 29 | #include "llvm/Support/Compiler.h" |
Douglas Gregor | 0246575 | 2009-10-16 21:24:31 +0000 | [diff] [blame] | 30 | #include "llvm/Support/MemoryBuffer.h" |
| 31 | #include "llvm/System/Path.h" |
Benjamin Kramer | 0829a83 | 2009-10-18 11:19:36 +0000 | [diff] [blame] | 32 | #include "llvm/System/Program.h" |
Ted Kremenek | 379afec | 2009-10-22 03:24:01 +0000 | [diff] [blame] | 33 | #include "llvm/Support/raw_ostream.h" |
Ted Kremenek | fc06221 | 2009-10-19 21:44:57 +0000 | [diff] [blame] | 34 | |
Benjamin Kramer | d01a0bc | 2009-08-29 12:56:35 +0000 | [diff] [blame] | 35 | #include <cstdio> |
Ted Kremenek | 49358d8 | 2009-10-19 21:17:25 +0000 | [diff] [blame] | 36 | #include <vector> |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 37 | #include <sstream> |
Ted Kremenek | 49358d8 | 2009-10-19 21:17:25 +0000 | [diff] [blame] | 38 | |
Benjamin Kramer | 20d7581 | 2009-10-18 16:13:48 +0000 | [diff] [blame] | 39 | #ifdef LLVM_ON_WIN32 |
| 40 | #define WIN32_LEAN_AND_MEAN |
| 41 | #include <windows.h> |
| 42 | #else |
Steve Naroff | 5b7d8e2 | 2009-10-15 20:04:39 +0000 | [diff] [blame] | 43 | #include <dlfcn.h> |
Daniel Dunbar | a47dd19 | 2009-10-17 23:53:11 +0000 | [diff] [blame] | 44 | #endif |
Steve Naroff | 5b7d8e2 | 2009-10-15 20:04:39 +0000 | [diff] [blame] | 45 | |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 46 | using namespace clang; |
| 47 | using namespace idx; |
| 48 | |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 49 | namespace { |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 50 | static enum CXCursorKind TranslateDeclRefExpr(DeclRefExpr *DRE) |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 51 | { |
| 52 | NamedDecl *D = DRE->getDecl(); |
| 53 | if (isa<VarDecl>(D)) |
| 54 | return CXCursor_VarRef; |
| 55 | else if (isa<FunctionDecl>(D)) |
| 56 | return CXCursor_FunctionRef; |
| 57 | else if (isa<EnumConstantDecl>(D)) |
| 58 | return CXCursor_EnumConstantRef; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 59 | else |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 60 | return CXCursor_NotImplemented; |
| 61 | } |
| 62 | |
| 63 | #if 0 |
| 64 | // Will be useful one day. |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 65 | class CRefVisitor : public StmtVisitor<CRefVisitor> { |
| 66 | CXDecl CDecl; |
| 67 | CXDeclIterator Callback; |
| 68 | CXClientData CData; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 69 | |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 70 | void Call(enum CXCursorKind CK, Stmt *SRef) { |
| 71 | CXCursor C = { CK, CDecl, SRef }; |
| 72 | Callback(CDecl, C, CData); |
| 73 | } |
| 74 | |
| 75 | public: |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 76 | CRefVisitor(CXDecl C, CXDeclIterator cback, CXClientData D) : |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 77 | CDecl(C), Callback(cback), CData(D) {} |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 78 | |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 79 | void VisitStmt(Stmt *S) { |
| 80 | for (Stmt::child_iterator C = S->child_begin(), CEnd = S->child_end(); |
| 81 | C != CEnd; ++C) |
| 82 | Visit(*C); |
| 83 | } |
| 84 | void VisitDeclRefExpr(DeclRefExpr *Node) { |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 85 | Call(TranslateDeclRefExpr(Node), Node); |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 86 | } |
| 87 | void VisitMemberExpr(MemberExpr *Node) { |
| 88 | Call(CXCursor_MemberRef, Node); |
| 89 | } |
| 90 | void VisitObjCMessageExpr(ObjCMessageExpr *Node) { |
| 91 | Call(CXCursor_ObjCSelectorRef, Node); |
| 92 | } |
| 93 | void VisitObjCIvarRefExpr(ObjCIvarRefExpr *Node) { |
| 94 | Call(CXCursor_ObjCIvarRef, Node); |
| 95 | } |
| 96 | }; |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 97 | #endif |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 98 | |
Ted Kremenek | fc06221 | 2009-10-19 21:44:57 +0000 | [diff] [blame] | 99 | /// IgnoreDiagnosticsClient - A DiagnosticsClient that just ignores emitted |
| 100 | /// warnings and errors. |
| 101 | class VISIBILITY_HIDDEN IgnoreDiagnosticsClient : public DiagnosticClient { |
| 102 | public: |
| 103 | virtual ~IgnoreDiagnosticsClient() {} |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 104 | virtual void HandleDiagnostic(Diagnostic::Level, const DiagnosticInfo &) {} |
Ted Kremenek | fc06221 | 2009-10-19 21:44:57 +0000 | [diff] [blame] | 105 | }; |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 106 | |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 107 | // Translation Unit Visitor. |
| 108 | class TUVisitor : public DeclVisitor<TUVisitor> { |
| 109 | CXTranslationUnit TUnit; |
| 110 | CXTranslationUnitIterator Callback; |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 111 | CXClientData CData; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 112 | |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 113 | // MaxPCHLevel - the maximum PCH level of declarations that we will pass on |
| 114 | // to the visitor. Declarations with a PCH level greater than this value will |
| 115 | // be suppressed. |
| 116 | unsigned MaxPCHLevel; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 117 | |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 118 | void Call(enum CXCursorKind CK, NamedDecl *ND) { |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 119 | // Filter any declarations that have a PCH level greater than what we allow. |
| 120 | if (ND->getPCHLevel() > MaxPCHLevel) |
| 121 | return; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 122 | |
Steve Naroff | f96b524 | 2009-10-28 20:44:47 +0000 | [diff] [blame] | 123 | // Filter any implicit declarations (since the source info will be bogus). |
| 124 | if (ND->isImplicit()) |
| 125 | return; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 126 | |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 127 | CXCursor C = { CK, ND, 0 }; |
Steve Naroff | 2b8ee6c | 2009-09-01 15:55:40 +0000 | [diff] [blame] | 128 | Callback(TUnit, C, CData); |
| 129 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 130 | public: |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 131 | TUVisitor(CXTranslationUnit CTU, |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 132 | CXTranslationUnitIterator cback, CXClientData D, |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 133 | unsigned MaxPCHLevel) : |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 134 | TUnit(CTU), Callback(cback), CData(D), MaxPCHLevel(MaxPCHLevel) {} |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 135 | |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 136 | void VisitTranslationUnitDecl(TranslationUnitDecl *D) { |
| 137 | VisitDeclContext(dyn_cast<DeclContext>(D)); |
| 138 | } |
| 139 | void VisitDeclContext(DeclContext *DC) { |
| 140 | for (DeclContext::decl_iterator |
| 141 | I = DC->decls_begin(), E = DC->decls_end(); I != E; ++I) |
| 142 | Visit(*I); |
| 143 | } |
Ted Kremenek | ef7fdc6 | 2009-11-17 07:02:15 +0000 | [diff] [blame] | 144 | |
| 145 | void VisitFunctionDecl(FunctionDecl *ND) { |
| 146 | Call(ND->isThisDeclarationADefinition() ? CXCursor_FunctionDefn |
| 147 | : CXCursor_FunctionDecl, ND); |
| 148 | } |
| 149 | void VisitObjCCategoryDecl(ObjCCategoryDecl *ND) { |
| 150 | Call(CXCursor_ObjCCategoryDecl, ND); |
| 151 | } |
| 152 | void VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *ND) { |
| 153 | Call(CXCursor_ObjCCategoryDefn, ND); |
| 154 | } |
| 155 | void VisitObjCImplementationDecl(ObjCImplementationDecl *ND) { |
| 156 | Call(CXCursor_ObjCClassDefn, ND); |
| 157 | } |
| 158 | void VisitObjCInterfaceDecl(ObjCInterfaceDecl *ND) { |
| 159 | Call(CXCursor_ObjCInterfaceDecl, ND); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 160 | } |
Ted Kremenek | ef7fdc6 | 2009-11-17 07:02:15 +0000 | [diff] [blame] | 161 | void VisitObjCProtocolDecl(ObjCProtocolDecl *ND) { |
| 162 | Call(CXCursor_ObjCProtocolDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 163 | } |
| 164 | void VisitTagDecl(TagDecl *ND) { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 165 | switch (ND->getTagKind()) { |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 166 | case TagDecl::TK_struct: |
| 167 | Call(CXCursor_StructDecl, ND); |
| 168 | break; |
| 169 | case TagDecl::TK_class: |
| 170 | Call(CXCursor_ClassDecl, ND); |
| 171 | break; |
| 172 | case TagDecl::TK_union: |
| 173 | Call(CXCursor_UnionDecl, ND); |
| 174 | break; |
| 175 | case TagDecl::TK_enum: |
| 176 | Call(CXCursor_EnumDecl, ND); |
| 177 | break; |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 178 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 179 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 180 | void VisitTypedefDecl(TypedefDecl *ND) { |
| 181 | Call(CXCursor_TypedefDecl, ND); |
| 182 | } |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 183 | void VisitVarDecl(VarDecl *ND) { |
| 184 | Call(CXCursor_VarDecl, ND); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 185 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 186 | }; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 187 | |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 188 | |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 189 | // Declaration visitor. |
| 190 | class CDeclVisitor : public DeclVisitor<CDeclVisitor> { |
| 191 | CXDecl CDecl; |
| 192 | CXDeclIterator Callback; |
| 193 | CXClientData CData; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 194 | |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 195 | // MaxPCHLevel - the maximum PCH level of declarations that we will pass on |
| 196 | // to the visitor. Declarations with a PCH level greater than this value will |
| 197 | // be suppressed. |
| 198 | unsigned MaxPCHLevel; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 199 | |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 200 | void Call(enum CXCursorKind CK, NamedDecl *ND) { |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 201 | // Disable the callback when the context is equal to the visiting decl. |
| 202 | if (CDecl == ND && !clang_isReference(CK)) |
| 203 | return; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 204 | |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 205 | // Filter any declarations that have a PCH level greater than what we allow. |
| 206 | if (ND->getPCHLevel() > MaxPCHLevel) |
| 207 | return; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 208 | |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 209 | CXCursor C = { CK, ND, 0 }; |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 210 | Callback(CDecl, C, CData); |
| 211 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 212 | public: |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 213 | CDeclVisitor(CXDecl C, CXDeclIterator cback, CXClientData D, |
| 214 | unsigned MaxPCHLevel) : |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 215 | CDecl(C), Callback(cback), CData(D), MaxPCHLevel(MaxPCHLevel) {} |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 216 | |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 217 | void VisitObjCCategoryDecl(ObjCCategoryDecl *ND) { |
| 218 | // Issue callbacks for the containing class. |
| 219 | Call(CXCursor_ObjCClassRef, ND); |
| 220 | // FIXME: Issue callbacks for protocol refs. |
| 221 | VisitDeclContext(dyn_cast<DeclContext>(ND)); |
| 222 | } |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 223 | void VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) { |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 224 | // Issue callbacks for super class. |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 225 | if (D->getSuperClass()) |
| 226 | Call(CXCursor_ObjCSuperClassRef, D); |
| 227 | |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 228 | for (ObjCProtocolDecl::protocol_iterator I = D->protocol_begin(), |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 229 | E = D->protocol_end(); I != E; ++I) |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 230 | Call(CXCursor_ObjCProtocolRef, *I); |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 231 | VisitDeclContext(dyn_cast<DeclContext>(D)); |
| 232 | } |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 233 | void VisitObjCProtocolDecl(ObjCProtocolDecl *PID) { |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 234 | for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(), |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 235 | E = PID->protocol_end(); I != E; ++I) |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 236 | Call(CXCursor_ObjCProtocolRef, *I); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 237 | |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 238 | VisitDeclContext(dyn_cast<DeclContext>(PID)); |
| 239 | } |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 240 | void VisitTagDecl(TagDecl *D) { |
| 241 | VisitDeclContext(dyn_cast<DeclContext>(D)); |
| 242 | } |
| 243 | void VisitObjCImplementationDecl(ObjCImplementationDecl *D) { |
| 244 | VisitDeclContext(dyn_cast<DeclContext>(D)); |
| 245 | } |
| 246 | void VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) { |
| 247 | VisitDeclContext(dyn_cast<DeclContext>(D)); |
| 248 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 249 | void VisitDeclContext(DeclContext *DC) { |
| 250 | for (DeclContext::decl_iterator |
| 251 | I = DC->decls_begin(), E = DC->decls_end(); I != E; ++I) |
| 252 | Visit(*I); |
| 253 | } |
| 254 | void VisitEnumConstantDecl(EnumConstantDecl *ND) { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 255 | Call(CXCursor_EnumConstantDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 256 | } |
| 257 | void VisitFieldDecl(FieldDecl *ND) { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 258 | Call(CXCursor_FieldDecl, ND); |
| 259 | } |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 260 | void VisitVarDecl(VarDecl *ND) { |
| 261 | Call(CXCursor_VarDecl, ND); |
| 262 | } |
| 263 | void VisitParmVarDecl(ParmVarDecl *ND) { |
| 264 | Call(CXCursor_ParmDecl, ND); |
| 265 | } |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 266 | void VisitObjCPropertyDecl(ObjCPropertyDecl *ND) { |
| 267 | Call(CXCursor_ObjCPropertyDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 268 | } |
| 269 | void VisitObjCIvarDecl(ObjCIvarDecl *ND) { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 270 | Call(CXCursor_ObjCIvarDecl, ND); |
| 271 | } |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 272 | void VisitFunctionDecl(FunctionDecl *ND) { |
| 273 | if (ND->isThisDeclarationADefinition()) { |
| 274 | VisitDeclContext(dyn_cast<DeclContext>(ND)); |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 275 | #if 0 |
| 276 | // Not currently needed. |
| 277 | CompoundStmt *Body = dyn_cast<CompoundStmt>(ND->getBody()); |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 278 | CRefVisitor RVisit(CDecl, Callback, CData); |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 279 | RVisit.Visit(Body); |
| 280 | #endif |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 281 | } |
| 282 | } |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 283 | void VisitObjCMethodDecl(ObjCMethodDecl *ND) { |
| 284 | if (ND->getBody()) { |
| 285 | Call(ND->isInstanceMethod() ? CXCursor_ObjCInstanceMethodDefn |
| 286 | : CXCursor_ObjCClassMethodDefn, ND); |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 287 | VisitDeclContext(dyn_cast<DeclContext>(ND)); |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 288 | } else |
| 289 | Call(ND->isInstanceMethod() ? CXCursor_ObjCInstanceMethodDecl |
| 290 | : CXCursor_ObjCClassMethodDecl, ND); |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 291 | } |
| 292 | }; |
| 293 | |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 294 | class CIndexer : public Indexer { |
Daniel Dunbar | 5262fda | 2009-12-03 01:45:44 +0000 | [diff] [blame^] | 295 | IgnoreDiagnosticsClient IgnoreDiagClient; |
| 296 | llvm::OwningPtr<Diagnostic> TextDiags; |
| 297 | Diagnostic IgnoreDiags; |
| 298 | bool UseExternalASTGeneration; |
| 299 | bool OnlyLocalDecls; |
| 300 | bool DisplayDiagnostics; |
| 301 | |
| 302 | llvm::sys::Path ClangPath; |
| 303 | |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 304 | public: |
| 305 | explicit CIndexer(Program *prog) : Indexer(*prog), |
Daniel Dunbar | 5262fda | 2009-12-03 01:45:44 +0000 | [diff] [blame^] | 306 | IgnoreDiags(&IgnoreDiagClient), |
| 307 | UseExternalASTGeneration(false), |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 308 | OnlyLocalDecls(false), |
Daniel Dunbar | 5262fda | 2009-12-03 01:45:44 +0000 | [diff] [blame^] | 309 | DisplayDiagnostics(false) { |
| 310 | TextDiags.reset( |
| 311 | CompilerInstance::createDiagnostics(DiagnosticOptions(), 0, 0)); |
| 312 | } |
Ted Kremenek | dff7689 | 2009-10-17 06:21:47 +0000 | [diff] [blame] | 313 | |
| 314 | virtual ~CIndexer() { delete &getProgram(); } |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 315 | |
| 316 | /// \brief Whether we only want to see "local" declarations (that did not |
| 317 | /// come from a previous precompiled header). If false, we want to see all |
| 318 | /// declarations. |
| 319 | bool getOnlyLocalDecls() const { return OnlyLocalDecls; } |
| 320 | void setOnlyLocalDecls(bool Local = true) { OnlyLocalDecls = Local; } |
Benjamin Kramer | 9670762 | 2009-10-18 11:10:55 +0000 | [diff] [blame] | 321 | |
Daniel Dunbar | 5262fda | 2009-12-03 01:45:44 +0000 | [diff] [blame^] | 322 | bool getDisplayDiagnostics() const { return DisplayDiagnostics; } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 323 | void setDisplayDiagnostics(bool Display = true) { |
Steve Naroff | e56b4ba | 2009-10-20 14:46:24 +0000 | [diff] [blame] | 324 | DisplayDiagnostics = Display; |
| 325 | } |
Daniel Dunbar | 5262fda | 2009-12-03 01:45:44 +0000 | [diff] [blame^] | 326 | |
| 327 | Diagnostic &getDiags() { |
| 328 | return DisplayDiagnostics ? *TextDiags : IgnoreDiags; |
| 329 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 330 | |
Benjamin Kramer | 5e4bc59 | 2009-10-18 16:11:04 +0000 | [diff] [blame] | 331 | /// \brief Get the path of the clang binary. |
Benjamin Kramer | c5a9e95 | 2009-10-19 10:20:24 +0000 | [diff] [blame] | 332 | const llvm::sys::Path& getClangPath(); |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 333 | }; |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 334 | |
Benjamin Kramer | 5e4bc59 | 2009-10-18 16:11:04 +0000 | [diff] [blame] | 335 | const llvm::sys::Path& CIndexer::getClangPath() { |
| 336 | // Did we already compute the path? |
| 337 | if (!ClangPath.empty()) |
| 338 | return ClangPath; |
| 339 | |
Steve Naroff | 5b7d8e2 | 2009-10-15 20:04:39 +0000 | [diff] [blame] | 340 | // Find the location where this library lives (libCIndex.dylib). |
Benjamin Kramer | 20d7581 | 2009-10-18 16:13:48 +0000 | [diff] [blame] | 341 | #ifdef LLVM_ON_WIN32 |
| 342 | MEMORY_BASIC_INFORMATION mbi; |
| 343 | char path[MAX_PATH]; |
Benjamin Kramer | edcd828 | 2009-10-18 16:20:58 +0000 | [diff] [blame] | 344 | VirtualQuery((void *)(uintptr_t)clang_createTranslationUnit, &mbi, |
Benjamin Kramer | 20d7581 | 2009-10-18 16:13:48 +0000 | [diff] [blame] | 345 | sizeof(mbi)); |
| 346 | GetModuleFileNameA((HINSTANCE)mbi.AllocationBase, path, MAX_PATH); |
| 347 | |
| 348 | llvm::sys::Path CIndexPath(path); |
John Thompson | 6a6742a | 2009-11-11 23:11:14 +0000 | [diff] [blame] | 349 | |
| 350 | CIndexPath.eraseComponent(); |
| 351 | CIndexPath.appendComponent("clang"); |
| 352 | CIndexPath.appendSuffix("exe"); |
| 353 | CIndexPath.makeAbsolute(); |
Benjamin Kramer | 20d7581 | 2009-10-18 16:13:48 +0000 | [diff] [blame] | 354 | #else |
Steve Naroff | 5b7d8e2 | 2009-10-15 20:04:39 +0000 | [diff] [blame] | 355 | // This silly cast below avoids a C++ warning. |
| 356 | Dl_info info; |
| 357 | if (dladdr((void *)(uintptr_t)clang_createTranslationUnit, &info) == 0) |
| 358 | assert(0 && "Call to dladdr() failed"); |
| 359 | |
| 360 | llvm::sys::Path CIndexPath(info.dli_fname); |
Benjamin Kramer | 5e4bc59 | 2009-10-18 16:11:04 +0000 | [diff] [blame] | 361 | |
Steve Naroff | 5b7d8e2 | 2009-10-15 20:04:39 +0000 | [diff] [blame] | 362 | // We now have the CIndex directory, locate clang relative to it. |
Benjamin Kramer | 5e4bc59 | 2009-10-18 16:11:04 +0000 | [diff] [blame] | 363 | CIndexPath.eraseComponent(); |
| 364 | CIndexPath.eraseComponent(); |
| 365 | CIndexPath.appendComponent("bin"); |
| 366 | CIndexPath.appendComponent("clang"); |
John Thompson | 6a6742a | 2009-11-11 23:11:14 +0000 | [diff] [blame] | 367 | #endif |
Benjamin Kramer | 5e4bc59 | 2009-10-18 16:11:04 +0000 | [diff] [blame] | 368 | |
| 369 | // Cache our result. |
| 370 | ClangPath = CIndexPath; |
| 371 | return ClangPath; |
| 372 | } |
| 373 | |
| 374 | } |
| 375 | |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 376 | static SourceLocation getLocationFromCursor(CXCursor C, |
Daniel Dunbar | c619033 | 2009-11-08 04:13:53 +0000 | [diff] [blame] | 377 | SourceManager &SourceMgr, |
| 378 | NamedDecl *ND) { |
| 379 | if (clang_isReference(C.kind)) { |
| 380 | switch (C.kind) { |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 381 | case CXCursor_ObjCClassRef: { |
| 382 | if (isa<ObjCInterfaceDecl>(ND)) { |
| 383 | // FIXME: This is a hack (storing the parent decl in the stmt slot). |
| 384 | NamedDecl *parentDecl = static_cast<NamedDecl *>(C.stmt); |
| 385 | return parentDecl->getLocation(); |
Daniel Dunbar | c619033 | 2009-11-08 04:13:53 +0000 | [diff] [blame] | 386 | } |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 387 | ObjCCategoryDecl *OID = dyn_cast<ObjCCategoryDecl>(ND); |
| 388 | assert(OID && "clang_getCursorLine(): Missing category decl"); |
| 389 | return OID->getClassInterface()->getLocation(); |
| 390 | } |
| 391 | case CXCursor_ObjCSuperClassRef: { |
| 392 | ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(ND); |
| 393 | assert(OID && "clang_getCursorLine(): Missing interface decl"); |
| 394 | return OID->getSuperClassLoc(); |
| 395 | } |
| 396 | case CXCursor_ObjCProtocolRef: { |
| 397 | ObjCProtocolDecl *OID = dyn_cast<ObjCProtocolDecl>(ND); |
| 398 | assert(OID && "clang_getCursorLine(): Missing protocol decl"); |
| 399 | return OID->getLocation(); |
| 400 | } |
| 401 | case CXCursor_ObjCSelectorRef: { |
| 402 | ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>( |
| 403 | static_cast<Stmt *>(C.stmt)); |
| 404 | assert(OME && "clang_getCursorLine(): Missing message expr"); |
| 405 | return OME->getLeftLoc(); /* FIXME: should be a range */ |
| 406 | } |
| 407 | case CXCursor_VarRef: |
| 408 | case CXCursor_FunctionRef: |
| 409 | case CXCursor_EnumConstantRef: { |
| 410 | DeclRefExpr *DRE = dyn_cast<DeclRefExpr>( |
| 411 | static_cast<Stmt *>(C.stmt)); |
| 412 | assert(DRE && "clang_getCursorLine(): Missing decl ref expr"); |
| 413 | return DRE->getLocation(); |
| 414 | } |
| 415 | default: |
| 416 | return SourceLocation(); |
Daniel Dunbar | c619033 | 2009-11-08 04:13:53 +0000 | [diff] [blame] | 417 | } |
| 418 | } else { // We have a declaration or a definition. |
| 419 | SourceLocation SLoc; |
| 420 | switch (ND->getKind()) { |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 421 | case Decl::ObjCInterface: { |
| 422 | SLoc = dyn_cast<ObjCInterfaceDecl>(ND)->getClassLoc(); |
| 423 | break; |
| 424 | } |
| 425 | case Decl::ObjCProtocol: { |
| 426 | SLoc = ND->getLocation(); /* FIXME: need to get the name location. */ |
| 427 | break; |
| 428 | } |
| 429 | default: { |
| 430 | SLoc = ND->getLocation(); |
| 431 | break; |
| 432 | } |
Daniel Dunbar | c619033 | 2009-11-08 04:13:53 +0000 | [diff] [blame] | 433 | } |
| 434 | if (SLoc.isInvalid()) |
| 435 | return SourceLocation(); |
| 436 | return SourceMgr.getSpellingLoc(SLoc); // handles macro instantiations. |
| 437 | } |
| 438 | } |
| 439 | |
Benjamin Kramer | 62cf322 | 2009-11-09 19:13:48 +0000 | [diff] [blame] | 440 | static CXString createCXString(const char *String, bool DupString = false) { |
| 441 | CXString Str; |
| 442 | if (DupString) { |
| 443 | Str.Spelling = strdup(String); |
| 444 | Str.MustFreeString = 1; |
| 445 | } else { |
| 446 | Str.Spelling = String; |
| 447 | Str.MustFreeString = 0; |
| 448 | } |
| 449 | return Str; |
| 450 | } |
| 451 | |
Benjamin Kramer | 5e4bc59 | 2009-10-18 16:11:04 +0000 | [diff] [blame] | 452 | extern "C" { |
| 453 | |
Steve Naroff | e56b4ba | 2009-10-20 14:46:24 +0000 | [diff] [blame] | 454 | CXIndex clang_createIndex(int excludeDeclarationsFromPCH, |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 455 | int displayDiagnostics) { |
Steve Naroff | e56b4ba | 2009-10-20 14:46:24 +0000 | [diff] [blame] | 456 | CIndexer *CIdxr = new CIndexer(new Program()); |
| 457 | if (excludeDeclarationsFromPCH) |
| 458 | CIdxr->setOnlyLocalDecls(); |
| 459 | if (displayDiagnostics) |
| 460 | CIdxr->setDisplayDiagnostics(); |
| 461 | return CIdxr; |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 462 | } |
| 463 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 464 | void clang_disposeIndex(CXIndex CIdx) { |
Steve Naroff | 2bd6b9f | 2009-09-17 18:33:27 +0000 | [diff] [blame] | 465 | assert(CIdx && "Passed null CXIndex"); |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 466 | delete static_cast<CIndexer *>(CIdx); |
Steve Naroff | 2bd6b9f | 2009-09-17 18:33:27 +0000 | [diff] [blame] | 467 | } |
| 468 | |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 469 | // FIXME: need to pass back error info. |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 470 | CXTranslationUnit clang_createTranslationUnit(CXIndex CIdx, |
| 471 | const char *ast_filename) { |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 472 | assert(CIdx && "Passed null CXIndex"); |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 473 | CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 474 | |
Daniel Dunbar | 5262fda | 2009-12-03 01:45:44 +0000 | [diff] [blame^] | 475 | return ASTUnit::LoadFromPCHFile(ast_filename, CXXIdx->getDiags(), |
| 476 | CXXIdx->getOnlyLocalDecls(), |
| 477 | /* UseBumpAllocator = */ true); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 478 | } |
| 479 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 480 | CXTranslationUnit |
| 481 | clang_createTranslationUnitFromSourceFile(CXIndex CIdx, |
| 482 | const char *source_filename, |
| 483 | int num_command_line_args, |
| 484 | const char **command_line_args) { |
Steve Naroff | e56b4ba | 2009-10-20 14:46:24 +0000 | [diff] [blame] | 485 | assert(CIdx && "Passed null CXIndex"); |
| 486 | CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx); |
| 487 | |
Ted Kremenek | 139ba86 | 2009-10-22 00:03:57 +0000 | [diff] [blame] | 488 | // Build up the arguments for invoking 'clang'. |
Ted Kremenek | 74cd069 | 2009-10-15 23:21:22 +0000 | [diff] [blame] | 489 | std::vector<const char *> argv; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 490 | |
Ted Kremenek | 139ba86 | 2009-10-22 00:03:57 +0000 | [diff] [blame] | 491 | // First add the complete path to the 'clang' executable. |
| 492 | llvm::sys::Path ClangPath = static_cast<CIndexer *>(CIdx)->getClangPath(); |
Benjamin Kramer | 5e4bc59 | 2009-10-18 16:11:04 +0000 | [diff] [blame] | 493 | argv.push_back(ClangPath.c_str()); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 494 | |
Ted Kremenek | 139ba86 | 2009-10-22 00:03:57 +0000 | [diff] [blame] | 495 | // Add the '-emit-ast' option as our execution mode for 'clang'. |
Ted Kremenek | 74cd069 | 2009-10-15 23:21:22 +0000 | [diff] [blame] | 496 | argv.push_back("-emit-ast"); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 497 | |
Ted Kremenek | 139ba86 | 2009-10-22 00:03:57 +0000 | [diff] [blame] | 498 | // The 'source_filename' argument is optional. If the caller does not |
| 499 | // specify it then it is assumed that the source file is specified |
| 500 | // in the actual argument list. |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 501 | if (source_filename) |
| 502 | argv.push_back(source_filename); |
Ted Kremenek | 139ba86 | 2009-10-22 00:03:57 +0000 | [diff] [blame] | 503 | |
Steve Naroff | 37b5ac2 | 2009-10-15 20:50:09 +0000 | [diff] [blame] | 504 | // Generate a temporary name for the AST file. |
Ted Kremenek | 139ba86 | 2009-10-22 00:03:57 +0000 | [diff] [blame] | 505 | argv.push_back("-o"); |
Steve Naroff | 37b5ac2 | 2009-10-15 20:50:09 +0000 | [diff] [blame] | 506 | char astTmpFile[L_tmpnam]; |
Ted Kremenek | 74cd069 | 2009-10-15 23:21:22 +0000 | [diff] [blame] | 507 | argv.push_back(tmpnam(astTmpFile)); |
Ted Kremenek | 139ba86 | 2009-10-22 00:03:57 +0000 | [diff] [blame] | 508 | |
| 509 | // Process the compiler options, stripping off '-o', '-c', '-fsyntax-only'. |
| 510 | for (int i = 0; i < num_command_line_args; ++i) |
| 511 | if (const char *arg = command_line_args[i]) { |
| 512 | if (strcmp(arg, "-o") == 0) { |
| 513 | ++i; // Also skip the matching argument. |
| 514 | continue; |
| 515 | } |
| 516 | if (strcmp(arg, "-emit-ast") == 0 || |
| 517 | strcmp(arg, "-c") == 0 || |
| 518 | strcmp(arg, "-fsyntax-only") == 0) { |
| 519 | continue; |
| 520 | } |
| 521 | |
| 522 | // Keep the argument. |
| 523 | argv.push_back(arg); |
Steve Naroff | e56b4ba | 2009-10-20 14:46:24 +0000 | [diff] [blame] | 524 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 525 | |
Ted Kremenek | 139ba86 | 2009-10-22 00:03:57 +0000 | [diff] [blame] | 526 | // Add the null terminator. |
Ted Kremenek | 74cd069 | 2009-10-15 23:21:22 +0000 | [diff] [blame] | 527 | argv.push_back(NULL); |
| 528 | |
Ted Kremenek | feb15e3 | 2009-10-26 22:14:08 +0000 | [diff] [blame] | 529 | // Invoke 'clang'. |
| 530 | llvm::sys::Path DevNull; // leave empty, causes redirection to /dev/null |
| 531 | // on Unix or NUL (Windows). |
Ted Kremenek | 379afec | 2009-10-22 03:24:01 +0000 | [diff] [blame] | 532 | std::string ErrMsg; |
Ted Kremenek | c46e463 | 2009-10-19 22:27:32 +0000 | [diff] [blame] | 533 | const llvm::sys::Path *Redirects[] = { &DevNull, &DevNull, &DevNull, NULL }; |
Ted Kremenek | 379afec | 2009-10-22 03:24:01 +0000 | [diff] [blame] | 534 | llvm::sys::Program::ExecuteAndWait(ClangPath, &argv[0], /* env */ NULL, |
| 535 | /* redirects */ !CXXIdx->getDisplayDiagnostics() ? &Redirects[0] : NULL, |
| 536 | /* secondsToWait */ 0, /* memoryLimits */ 0, &ErrMsg); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 537 | |
Ted Kremenek | 0854d70 | 2009-11-10 19:18:52 +0000 | [diff] [blame] | 538 | if (CXXIdx->getDisplayDiagnostics() && !ErrMsg.empty()) { |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 539 | llvm::errs() << "clang_createTranslationUnitFromSourceFile: " << ErrMsg |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 540 | << '\n' << "Arguments: \n"; |
Ted Kremenek | 379afec | 2009-10-22 03:24:01 +0000 | [diff] [blame] | 541 | for (std::vector<const char*>::iterator I = argv.begin(), E = argv.end(); |
Ted Kremenek | 779e5f4 | 2009-10-26 22:08:39 +0000 | [diff] [blame] | 542 | I!=E; ++I) { |
| 543 | if (*I) |
| 544 | llvm::errs() << ' ' << *I << '\n'; |
| 545 | } |
| 546 | llvm::errs() << '\n'; |
Ted Kremenek | 379afec | 2009-10-22 03:24:01 +0000 | [diff] [blame] | 547 | } |
Benjamin Kramer | 0829a83 | 2009-10-18 11:19:36 +0000 | [diff] [blame] | 548 | |
Steve Naroff | 37b5ac2 | 2009-10-15 20:50:09 +0000 | [diff] [blame] | 549 | // Finally, we create the translation unit from the ast file. |
Steve Naroff | e19944c | 2009-10-15 22:23:48 +0000 | [diff] [blame] | 550 | ASTUnit *ATU = static_cast<ASTUnit *>( |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 551 | clang_createTranslationUnit(CIdx, astTmpFile)); |
Steve Naroff | e56b4ba | 2009-10-20 14:46:24 +0000 | [diff] [blame] | 552 | if (ATU) |
| 553 | ATU->unlinkTemporaryFile(); |
Steve Naroff | e19944c | 2009-10-15 22:23:48 +0000 | [diff] [blame] | 554 | return ATU; |
Steve Naroff | 5b7d8e2 | 2009-10-15 20:04:39 +0000 | [diff] [blame] | 555 | } |
| 556 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 557 | void clang_disposeTranslationUnit(CXTranslationUnit CTUnit) { |
Steve Naroff | 2bd6b9f | 2009-09-17 18:33:27 +0000 | [diff] [blame] | 558 | assert(CTUnit && "Passed null CXTranslationUnit"); |
| 559 | delete static_cast<ASTUnit *>(CTUnit); |
| 560 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 561 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 562 | CXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) { |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 563 | assert(CTUnit && "Passed null CXTranslationUnit"); |
Steve Naroff | 77accc1 | 2009-09-03 18:19:54 +0000 | [diff] [blame] | 564 | ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit); |
Benjamin Kramer | 62cf322 | 2009-11-09 19:13:48 +0000 | [diff] [blame] | 565 | return createCXString(CXXUnit->getOriginalSourceFileName().c_str(), true); |
Steve Naroff | af08ddc | 2009-09-03 15:49:00 +0000 | [diff] [blame] | 566 | } |
Daniel Dunbar | 1eb79b5 | 2009-08-28 16:30:07 +0000 | [diff] [blame] | 567 | |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 568 | void clang_loadTranslationUnit(CXTranslationUnit CTUnit, |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 569 | CXTranslationUnitIterator callback, |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 570 | CXClientData CData) { |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 571 | assert(CTUnit && "Passed null CXTranslationUnit"); |
| 572 | ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit); |
| 573 | ASTContext &Ctx = CXXUnit->getASTContext(); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 574 | |
| 575 | TUVisitor DVisit(CTUnit, callback, CData, |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 576 | CXXUnit->getOnlyLocalDecls()? 1 : Decl::MaxPCHLevel); |
Steve Naroff | 5039819 | 2009-08-28 15:28:48 +0000 | [diff] [blame] | 577 | DVisit.Visit(Ctx.getTranslationUnitDecl()); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 578 | } |
| 579 | |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 580 | void clang_loadDeclaration(CXDecl Dcl, |
| 581 | CXDeclIterator callback, |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 582 | CXClientData CData) { |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 583 | assert(Dcl && "Passed null CXDecl"); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 584 | |
Douglas Gregor | 7d1d49d | 2009-10-16 20:01:17 +0000 | [diff] [blame] | 585 | CDeclVisitor DVisit(Dcl, callback, CData, |
| 586 | static_cast<Decl *>(Dcl)->getPCHLevel()); |
Steve Naroff | c857ea4 | 2009-09-02 13:28:54 +0000 | [diff] [blame] | 587 | DVisit.Visit(static_cast<Decl *>(Dcl)); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 588 | } |
| 589 | |
Steve Naroff | 7e8f818 | 2009-08-28 12:07:44 +0000 | [diff] [blame] | 590 | // Some notes on CXEntity: |
| 591 | // |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 592 | // - Since the 'ordinary' namespace includes functions, data, typedefs, |
| 593 | // ObjC interfaces, thecurrent algorithm is a bit naive (resulting in one |
Steve Naroff | 7e8f818 | 2009-08-28 12:07:44 +0000 | [diff] [blame] | 594 | // entity for 2 different types). For example: |
| 595 | // |
| 596 | // module1.m: @interface Foo @end Foo *x; |
| 597 | // module2.m: void Foo(int); |
| 598 | // |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 599 | // - Since the unique name spans translation units, static data/functions |
Steve Naroff | 7e8f818 | 2009-08-28 12:07:44 +0000 | [diff] [blame] | 600 | // within a CXTranslationUnit are *not* currently represented by entities. |
| 601 | // As a result, there will be no entity for the following: |
| 602 | // |
| 603 | // module.m: static void Foo() { } |
| 604 | // |
| 605 | |
| 606 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 607 | const char *clang_getDeclarationName(CXEntity) { |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 608 | return ""; |
| 609 | } |
| 610 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 611 | const char *clang_getURI(CXEntity) { |
| 612 | return ""; |
| 613 | } |
| 614 | |
| 615 | CXEntity clang_getEntity(const char *URI) { |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 616 | return 0; |
| 617 | } |
| 618 | |
| 619 | // |
| 620 | // CXDecl Operations. |
| 621 | // |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 622 | |
| 623 | CXEntity clang_getEntityFromDecl(CXDecl) { |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 624 | return 0; |
| 625 | } |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 626 | |
| 627 | CXString clang_getDeclSpelling(CXDecl AnonDecl) { |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 628 | assert(AnonDecl && "Passed null CXDecl"); |
| 629 | NamedDecl *ND = static_cast<NamedDecl *>(AnonDecl); |
Steve Naroff | ef0cef6 | 2009-11-09 17:45:52 +0000 | [diff] [blame] | 630 | |
Benjamin Kramer | 62cf322 | 2009-11-09 19:13:48 +0000 | [diff] [blame] | 631 | if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND)) |
| 632 | return createCXString(OMD->getSelector().getAsString().c_str(), true); |
| 633 | |
| 634 | if (ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND)) |
Steve Naroff | 0d69b8c | 2009-10-29 21:11:04 +0000 | [diff] [blame] | 635 | // No, this isn't the same as the code below. getIdentifier() is non-virtual |
| 636 | // and returns different names. NamedDecl returns the class name and |
| 637 | // ObjCCategoryImplDecl returns the category name. |
Benjamin Kramer | 62cf322 | 2009-11-09 19:13:48 +0000 | [diff] [blame] | 638 | return createCXString(CIMP->getIdentifier()->getNameStart()); |
| 639 | |
| 640 | if (ND->getIdentifier()) |
| 641 | return createCXString(ND->getIdentifier()->getNameStart()); |
| 642 | |
| 643 | return createCXString(""); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 644 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 645 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 646 | unsigned clang_getDeclLine(CXDecl AnonDecl) { |
Steve Naroff | 699a07d | 2009-09-25 21:32:34 +0000 | [diff] [blame] | 647 | assert(AnonDecl && "Passed null CXDecl"); |
| 648 | NamedDecl *ND = static_cast<NamedDecl *>(AnonDecl); |
| 649 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
| 650 | return SourceMgr.getSpellingLineNumber(ND->getLocation()); |
| 651 | } |
| 652 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 653 | unsigned clang_getDeclColumn(CXDecl AnonDecl) { |
Steve Naroff | 699a07d | 2009-09-25 21:32:34 +0000 | [diff] [blame] | 654 | assert(AnonDecl && "Passed null CXDecl"); |
| 655 | NamedDecl *ND = static_cast<NamedDecl *>(AnonDecl); |
| 656 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Steve Naroff | 7416524 | 2009-09-25 22:15:54 +0000 | [diff] [blame] | 657 | return SourceMgr.getSpellingColumnNumber(ND->getLocation()); |
Steve Naroff | 699a07d | 2009-09-25 21:32:34 +0000 | [diff] [blame] | 658 | } |
| 659 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 660 | const char *clang_getDeclSource(CXDecl AnonDecl) { |
Steve Naroff | ee9405e | 2009-09-25 21:45:39 +0000 | [diff] [blame] | 661 | assert(AnonDecl && "Passed null CXDecl"); |
Steve Naroff | 8814503 | 2009-10-27 14:35:18 +0000 | [diff] [blame] | 662 | FileEntry *FEnt = static_cast<FileEntry *>(clang_getDeclSourceFile(AnonDecl)); |
| 663 | assert (FEnt && "Cannot find FileEntry for Decl"); |
| 664 | return clang_getFileName(FEnt); |
| 665 | } |
| 666 | |
| 667 | static const FileEntry *getFileEntryFromSourceLocation(SourceManager &SMgr, |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 668 | SourceLocation SLoc) { |
Steve Naroff | 8814503 | 2009-10-27 14:35:18 +0000 | [diff] [blame] | 669 | FileID FID; |
| 670 | if (SLoc.isFileID()) |
| 671 | FID = SMgr.getFileID(SLoc); |
| 672 | else |
| 673 | FID = SMgr.getDecomposedSpellingLoc(SLoc).first; |
| 674 | return SMgr.getFileEntryForID(FID); |
| 675 | } |
| 676 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 677 | CXFile clang_getDeclSourceFile(CXDecl AnonDecl) { |
Steve Naroff | 8814503 | 2009-10-27 14:35:18 +0000 | [diff] [blame] | 678 | assert(AnonDecl && "Passed null CXDecl"); |
Steve Naroff | ee9405e | 2009-09-25 21:45:39 +0000 | [diff] [blame] | 679 | NamedDecl *ND = static_cast<NamedDecl *>(AnonDecl); |
| 680 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Steve Naroff | 8814503 | 2009-10-27 14:35:18 +0000 | [diff] [blame] | 681 | return (void *)getFileEntryFromSourceLocation(SourceMgr, ND->getLocation()); |
| 682 | } |
| 683 | |
| 684 | const char *clang_getFileName(CXFile SFile) { |
| 685 | assert(SFile && "Passed null CXFile"); |
| 686 | FileEntry *FEnt = static_cast<FileEntry *>(SFile); |
| 687 | return FEnt->getName(); |
| 688 | } |
| 689 | |
| 690 | time_t clang_getFileTime(CXFile SFile) { |
| 691 | assert(SFile && "Passed null CXFile"); |
| 692 | FileEntry *FEnt = static_cast<FileEntry *>(SFile); |
| 693 | return FEnt->getModificationTime(); |
Steve Naroff | ee9405e | 2009-09-25 21:45:39 +0000 | [diff] [blame] | 694 | } |
| 695 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 696 | CXString clang_getCursorSpelling(CXCursor C) { |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 697 | assert(C.decl && "CXCursor has null decl"); |
| 698 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
Steve Naroff | ef0cef6 | 2009-11-09 17:45:52 +0000 | [diff] [blame] | 699 | |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 700 | if (clang_isReference(C.kind)) { |
| 701 | switch (C.kind) { |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 702 | case CXCursor_ObjCSuperClassRef: { |
| 703 | ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(ND); |
| 704 | assert(OID && "clang_getCursorLine(): Missing interface decl"); |
| 705 | return createCXString(OID->getSuperClass()->getIdentifier() |
| 706 | ->getNameStart()); |
| 707 | } |
| 708 | case CXCursor_ObjCClassRef: { |
| 709 | if (ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(ND)) |
Benjamin Kramer | 62cf322 | 2009-11-09 19:13:48 +0000 | [diff] [blame] | 710 | return createCXString(OID->getIdentifier()->getNameStart()); |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 711 | |
| 712 | ObjCCategoryDecl *OCD = dyn_cast<ObjCCategoryDecl>(ND); |
| 713 | assert(OCD && "clang_getCursorLine(): Missing category decl"); |
| 714 | return createCXString(OCD->getClassInterface()->getIdentifier() |
| 715 | ->getNameStart()); |
| 716 | } |
| 717 | case CXCursor_ObjCProtocolRef: { |
| 718 | ObjCProtocolDecl *OID = dyn_cast<ObjCProtocolDecl>(ND); |
| 719 | assert(OID && "clang_getCursorLine(): Missing protocol decl"); |
| 720 | return createCXString(OID->getIdentifier()->getNameStart()); |
| 721 | } |
| 722 | case CXCursor_ObjCSelectorRef: { |
| 723 | ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>( |
| 724 | static_cast<Stmt *>(C.stmt)); |
| 725 | assert(OME && "clang_getCursorLine(): Missing message expr"); |
| 726 | return createCXString(OME->getSelector().getAsString().c_str(), true); |
| 727 | } |
| 728 | case CXCursor_VarRef: |
| 729 | case CXCursor_FunctionRef: |
| 730 | case CXCursor_EnumConstantRef: { |
| 731 | DeclRefExpr *DRE = dyn_cast<DeclRefExpr>( |
| 732 | static_cast<Stmt *>(C.stmt)); |
| 733 | assert(DRE && "clang_getCursorLine(): Missing decl ref expr"); |
| 734 | return createCXString(DRE->getDecl()->getIdentifier()->getNameStart()); |
| 735 | } |
| 736 | default: |
| 737 | return createCXString("<not implemented>"); |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 738 | } |
| 739 | } |
| 740 | return clang_getDeclSpelling(C.decl); |
| 741 | } |
| 742 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 743 | const char *clang_getCursorKindSpelling(enum CXCursorKind Kind) { |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 744 | switch (Kind) { |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 745 | case CXCursor_FunctionDecl: return "FunctionDecl"; |
| 746 | case CXCursor_TypedefDecl: return "TypedefDecl"; |
| 747 | case CXCursor_EnumDecl: return "EnumDecl"; |
| 748 | case CXCursor_EnumConstantDecl: return "EnumConstantDecl"; |
| 749 | case CXCursor_StructDecl: return "StructDecl"; |
| 750 | case CXCursor_UnionDecl: return "UnionDecl"; |
| 751 | case CXCursor_ClassDecl: return "ClassDecl"; |
| 752 | case CXCursor_FieldDecl: return "FieldDecl"; |
| 753 | case CXCursor_VarDecl: return "VarDecl"; |
| 754 | case CXCursor_ParmDecl: return "ParmDecl"; |
| 755 | case CXCursor_ObjCInterfaceDecl: return "ObjCInterfaceDecl"; |
| 756 | case CXCursor_ObjCCategoryDecl: return "ObjCCategoryDecl"; |
| 757 | case CXCursor_ObjCProtocolDecl: return "ObjCProtocolDecl"; |
| 758 | case CXCursor_ObjCPropertyDecl: return "ObjCPropertyDecl"; |
| 759 | case CXCursor_ObjCIvarDecl: return "ObjCIvarDecl"; |
| 760 | case CXCursor_ObjCInstanceMethodDecl: return "ObjCInstanceMethodDecl"; |
| 761 | case CXCursor_ObjCClassMethodDecl: return "ObjCClassMethodDecl"; |
| 762 | case CXCursor_FunctionDefn: return "FunctionDefn"; |
| 763 | case CXCursor_ObjCInstanceMethodDefn: return "ObjCInstanceMethodDefn"; |
| 764 | case CXCursor_ObjCClassMethodDefn: return "ObjCClassMethodDefn"; |
| 765 | case CXCursor_ObjCClassDefn: return "ObjCClassDefn"; |
| 766 | case CXCursor_ObjCCategoryDefn: return "ObjCCategoryDefn"; |
| 767 | case CXCursor_ObjCSuperClassRef: return "ObjCSuperClassRef"; |
| 768 | case CXCursor_ObjCProtocolRef: return "ObjCProtocolRef"; |
| 769 | case CXCursor_ObjCClassRef: return "ObjCClassRef"; |
| 770 | case CXCursor_ObjCSelectorRef: return "ObjCSelectorRef"; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 771 | |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 772 | case CXCursor_VarRef: return "VarRef"; |
| 773 | case CXCursor_FunctionRef: return "FunctionRef"; |
| 774 | case CXCursor_EnumConstantRef: return "EnumConstantRef"; |
| 775 | case CXCursor_MemberRef: return "MemberRef"; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 776 | |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 777 | case CXCursor_InvalidFile: return "InvalidFile"; |
| 778 | case CXCursor_NoDeclFound: return "NoDeclFound"; |
| 779 | case CXCursor_NotImplemented: return "NotImplemented"; |
| 780 | default: return "<not implemented>"; |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 781 | } |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 782 | } |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 783 | |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 784 | static enum CXCursorKind TranslateKind(Decl *D) { |
| 785 | switch (D->getKind()) { |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 786 | case Decl::Function: return CXCursor_FunctionDecl; |
| 787 | case Decl::Typedef: return CXCursor_TypedefDecl; |
| 788 | case Decl::Enum: return CXCursor_EnumDecl; |
| 789 | case Decl::EnumConstant: return CXCursor_EnumConstantDecl; |
| 790 | case Decl::Record: return CXCursor_StructDecl; // FIXME: union/class |
| 791 | case Decl::Field: return CXCursor_FieldDecl; |
| 792 | case Decl::Var: return CXCursor_VarDecl; |
| 793 | case Decl::ParmVar: return CXCursor_ParmDecl; |
| 794 | case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl; |
| 795 | case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl; |
| 796 | case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl; |
| 797 | case Decl::ObjCMethod: { |
| 798 | ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D); |
| 799 | if (MD->isInstanceMethod()) |
| 800 | return CXCursor_ObjCInstanceMethodDecl; |
| 801 | return CXCursor_ObjCClassMethodDecl; |
| 802 | } |
| 803 | default: break; |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 804 | } |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 805 | return CXCursor_NotImplemented; |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 806 | } |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 807 | // |
| 808 | // CXCursor Operations. |
| 809 | // |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 810 | |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 811 | CXCursor clang_getCursor(CXTranslationUnit CTUnit, const char *source_name, |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 812 | unsigned line, unsigned column) { |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 813 | assert(CTUnit && "Passed null CXTranslationUnit"); |
| 814 | ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 815 | |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 816 | FileManager &FMgr = CXXUnit->getFileManager(); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 817 | const FileEntry *File = FMgr.getFile(source_name, |
| 818 | source_name+strlen(source_name)); |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 819 | if (!File) { |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 820 | CXCursor C = { CXCursor_InvalidFile, 0, 0 }; |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 821 | return C; |
| 822 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 823 | SourceLocation SLoc = |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 824 | CXXUnit->getSourceManager().getLocation(File, line, column); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 825 | |
Steve Naroff | f96b524 | 2009-10-28 20:44:47 +0000 | [diff] [blame] | 826 | ASTLocation LastLoc = CXXUnit->getLastASTLocation(); |
| 827 | |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 828 | ASTLocation ALoc = ResolveLocationInAST(CXXUnit->getASTContext(), SLoc, |
Steve Naroff | f96b524 | 2009-10-28 20:44:47 +0000 | [diff] [blame] | 829 | &LastLoc); |
| 830 | if (ALoc.isValid()) |
| 831 | CXXUnit->setLastASTLocation(ALoc); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 832 | |
Argyrios Kyrtzidis | f4526e3 | 2009-09-29 19:44:27 +0000 | [diff] [blame] | 833 | Decl *Dcl = ALoc.getParentDecl(); |
Argyrios Kyrtzidis | 05a7651 | 2009-09-29 19:45:58 +0000 | [diff] [blame] | 834 | if (ALoc.isNamedRef()) |
| 835 | Dcl = ALoc.AsNamedRef().ND; |
Argyrios Kyrtzidis | f4526e3 | 2009-09-29 19:44:27 +0000 | [diff] [blame] | 836 | Stmt *Stm = ALoc.dyn_AsStmt(); |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 837 | if (Dcl) { |
| 838 | if (Stm) { |
| 839 | if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Stm)) { |
| 840 | CXCursor C = { TranslateDeclRefExpr(DRE), Dcl, Stm }; |
| 841 | return C; |
| 842 | } else if (ObjCMessageExpr *MExp = dyn_cast<ObjCMessageExpr>(Stm)) { |
| 843 | CXCursor C = { CXCursor_ObjCSelectorRef, Dcl, MExp }; |
| 844 | return C; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 845 | } |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 846 | // Fall through...treat as a decl, not a ref. |
| 847 | } |
Steve Naroff | 85e2db7 | 2009-10-01 00:31:07 +0000 | [diff] [blame] | 848 | if (ALoc.isNamedRef()) { |
| 849 | if (isa<ObjCInterfaceDecl>(Dcl)) { |
| 850 | CXCursor C = { CXCursor_ObjCClassRef, Dcl, ALoc.getParentDecl() }; |
| 851 | return C; |
| 852 | } |
| 853 | if (isa<ObjCProtocolDecl>(Dcl)) { |
| 854 | CXCursor C = { CXCursor_ObjCProtocolRef, Dcl, ALoc.getParentDecl() }; |
| 855 | return C; |
| 856 | } |
| 857 | } |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 858 | CXCursor C = { TranslateKind(Dcl), Dcl, 0 }; |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 859 | return C; |
| 860 | } |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 861 | CXCursor C = { CXCursor_NoDeclFound, 0, 0 }; |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 862 | return C; |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 863 | } |
| 864 | |
Ted Kremenek | 7388555 | 2009-11-17 19:28:59 +0000 | [diff] [blame] | 865 | CXCursor clang_getNullCursor(void) { |
| 866 | CXCursor C; |
| 867 | C.kind = CXCursor_InvalidFile; |
| 868 | C.decl = NULL; |
| 869 | C.stmt = NULL; |
| 870 | return C; |
| 871 | } |
| 872 | |
| 873 | unsigned clang_equalCursors(CXCursor X, CXCursor Y) { |
| 874 | return X.kind == Y.kind && X.decl == Y.decl && X.stmt == Y.stmt; |
| 875 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 876 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 877 | CXCursor clang_getCursorFromDecl(CXDecl AnonDecl) { |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 878 | assert(AnonDecl && "Passed null CXDecl"); |
| 879 | NamedDecl *ND = static_cast<NamedDecl *>(AnonDecl); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 880 | |
Steve Naroff | fb57042 | 2009-09-22 19:25:29 +0000 | [diff] [blame] | 881 | CXCursor C = { TranslateKind(ND), ND, 0 }; |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 882 | return C; |
| 883 | } |
| 884 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 885 | unsigned clang_isInvalid(enum CXCursorKind K) { |
Steve Naroff | 77128dd | 2009-09-15 20:25:34 +0000 | [diff] [blame] | 886 | return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid; |
| 887 | } |
| 888 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 889 | unsigned clang_isDeclaration(enum CXCursorKind K) { |
Steve Naroff | 89922f8 | 2009-08-31 00:59:03 +0000 | [diff] [blame] | 890 | return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl; |
| 891 | } |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 892 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 893 | unsigned clang_isReference(enum CXCursorKind K) { |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 894 | return K >= CXCursor_FirstRef && K <= CXCursor_LastRef; |
| 895 | } |
| 896 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 897 | unsigned clang_isDefinition(enum CXCursorKind K) { |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 898 | return K >= CXCursor_FirstDefn && K <= CXCursor_LastDefn; |
| 899 | } |
| 900 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 901 | CXCursorKind clang_getCursorKind(CXCursor C) { |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 902 | return C.kind; |
| 903 | } |
| 904 | |
Steve Naroff | 699a07d | 2009-09-25 21:32:34 +0000 | [diff] [blame] | 905 | static Decl *getDeclFromExpr(Stmt *E) { |
| 906 | if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E)) |
| 907 | return RefExpr->getDecl(); |
| 908 | if (MemberExpr *ME = dyn_cast<MemberExpr>(E)) |
| 909 | return ME->getMemberDecl(); |
| 910 | if (ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E)) |
| 911 | return RE->getDecl(); |
| 912 | |
| 913 | if (CallExpr *CE = dyn_cast<CallExpr>(E)) |
| 914 | return getDeclFromExpr(CE->getCallee()); |
| 915 | if (CastExpr *CE = dyn_cast<CastExpr>(E)) |
| 916 | return getDeclFromExpr(CE->getSubExpr()); |
| 917 | if (ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E)) |
| 918 | return OME->getMethodDecl(); |
| 919 | |
| 920 | return 0; |
| 921 | } |
| 922 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 923 | CXDecl clang_getCursorDecl(CXCursor C) { |
Steve Naroff | 699a07d | 2009-09-25 21:32:34 +0000 | [diff] [blame] | 924 | if (clang_isDeclaration(C.kind)) |
| 925 | return C.decl; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 926 | |
Steve Naroff | 699a07d | 2009-09-25 21:32:34 +0000 | [diff] [blame] | 927 | if (clang_isReference(C.kind)) { |
Steve Naroff | 85e2db7 | 2009-10-01 00:31:07 +0000 | [diff] [blame] | 928 | if (C.stmt) { |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 929 | if (C.kind == CXCursor_ObjCClassRef || |
Steve Naroff | f9adf8f | 2009-10-05 17:58:19 +0000 | [diff] [blame] | 930 | C.kind == CXCursor_ObjCProtocolRef) |
Steve Naroff | 85e2db7 | 2009-10-01 00:31:07 +0000 | [diff] [blame] | 931 | return static_cast<Stmt *>(C.stmt); |
| 932 | else |
| 933 | return getDeclFromExpr(static_cast<Stmt *>(C.stmt)); |
| 934 | } else |
Steve Naroff | 699a07d | 2009-09-25 21:32:34 +0000 | [diff] [blame] | 935 | return C.decl; |
| 936 | } |
| 937 | return 0; |
Steve Naroff | 9efa767 | 2009-09-04 15:44:05 +0000 | [diff] [blame] | 938 | } |
| 939 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 940 | unsigned clang_getCursorLine(CXCursor C) { |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 941 | assert(C.decl && "CXCursor has null decl"); |
| 942 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 943 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 944 | |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 945 | SourceLocation SLoc = getLocationFromCursor(C, SourceMgr, ND); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 946 | return SourceMgr.getSpellingLineNumber(SLoc); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 947 | } |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 948 | |
Steve Naroff | ef0cef6 | 2009-11-09 17:45:52 +0000 | [diff] [blame] | 949 | const char *clang_getCString(CXString string) { |
| 950 | return string.Spelling; |
| 951 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 952 | |
Steve Naroff | ef0cef6 | 2009-11-09 17:45:52 +0000 | [diff] [blame] | 953 | void clang_disposeString(CXString string) { |
Benjamin Kramer | 858e5de | 2009-11-09 18:24:53 +0000 | [diff] [blame] | 954 | if (string.MustFreeString) |
| 955 | free((void*)string.Spelling); |
Steve Naroff | ef0cef6 | 2009-11-09 17:45:52 +0000 | [diff] [blame] | 956 | } |
| 957 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 958 | unsigned clang_getCursorColumn(CXCursor C) { |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 959 | assert(C.decl && "CXCursor has null decl"); |
| 960 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 961 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 962 | |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 963 | SourceLocation SLoc = getLocationFromCursor(C, SourceMgr, ND); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 964 | return SourceMgr.getSpellingColumnNumber(SLoc); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 965 | } |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 966 | |
| 967 | const char *clang_getCursorSource(CXCursor C) { |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 968 | assert(C.decl && "CXCursor has null decl"); |
| 969 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
Steve Naroff | 2d4d629 | 2009-08-31 14:26:51 +0000 | [diff] [blame] | 970 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 971 | |
Steve Naroff | f334b4e | 2009-09-02 18:26:48 +0000 | [diff] [blame] | 972 | SourceLocation SLoc = getLocationFromCursor(C, SourceMgr, ND); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 973 | |
Ted Kremenek | 9298cfc | 2009-11-17 05:31:58 +0000 | [diff] [blame] | 974 | if (SLoc.isFileID()) { |
| 975 | const char *bufferName = SourceMgr.getBufferName(SLoc); |
| 976 | return bufferName[0] == '<' ? NULL : bufferName; |
| 977 | } |
Douglas Gregor | 0246575 | 2009-10-16 21:24:31 +0000 | [diff] [blame] | 978 | |
| 979 | // Retrieve the file in which the macro was instantiated, then provide that |
| 980 | // buffer name. |
| 981 | // FIXME: Do we want to give specific macro-instantiation information? |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 982 | const llvm::MemoryBuffer *Buffer |
Douglas Gregor | 0246575 | 2009-10-16 21:24:31 +0000 | [diff] [blame] | 983 | = SourceMgr.getBuffer(SourceMgr.getDecomposedSpellingLoc(SLoc).first); |
| 984 | if (!Buffer) |
| 985 | return 0; |
| 986 | |
| 987 | return Buffer->getBufferIdentifier(); |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 988 | } |
| 989 | |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 990 | CXFile clang_getCursorSourceFile(CXCursor C) { |
Steve Naroff | 8814503 | 2009-10-27 14:35:18 +0000 | [diff] [blame] | 991 | assert(C.decl && "CXCursor has null decl"); |
| 992 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
| 993 | SourceManager &SourceMgr = ND->getASTContext().getSourceManager(); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 994 | |
Steve Naroff | 8814503 | 2009-10-27 14:35:18 +0000 | [diff] [blame] | 995 | return (void *)getFileEntryFromSourceLocation(SourceMgr, |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 996 | getLocationFromCursor(C,SourceMgr, ND)); |
Steve Naroff | 8814503 | 2009-10-27 14:35:18 +0000 | [diff] [blame] | 997 | } |
| 998 | |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 999 | void clang_getDefinitionSpellingAndExtent(CXCursor C, |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 1000 | const char **startBuf, |
| 1001 | const char **endBuf, |
| 1002 | unsigned *startLine, |
| 1003 | unsigned *startColumn, |
| 1004 | unsigned *endLine, |
Daniel Dunbar | 9ebfa31 | 2009-12-01 03:14:51 +0000 | [diff] [blame] | 1005 | unsigned *endColumn) { |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 1006 | assert(C.decl && "CXCursor has null decl"); |
| 1007 | NamedDecl *ND = static_cast<NamedDecl *>(C.decl); |
| 1008 | FunctionDecl *FD = dyn_cast<FunctionDecl>(ND); |
| 1009 | CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody()); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1010 | |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 1011 | SourceManager &SM = FD->getASTContext().getSourceManager(); |
| 1012 | *startBuf = SM.getCharacterData(Body->getLBracLoc()); |
| 1013 | *endBuf = SM.getCharacterData(Body->getRBracLoc()); |
| 1014 | *startLine = SM.getSpellingLineNumber(Body->getLBracLoc()); |
| 1015 | *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc()); |
| 1016 | *endLine = SM.getSpellingLineNumber(Body->getRBracLoc()); |
| 1017 | *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc()); |
| 1018 | } |
| 1019 | |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1020 | enum CXCompletionChunkKind |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1021 | clang_getCompletionChunkKind(CXCompletionString completion_string, |
| 1022 | unsigned chunk_number) { |
| 1023 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
| 1024 | if (!CCStr || chunk_number >= CCStr->size()) |
| 1025 | return CXCompletionChunk_Text; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1026 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1027 | switch ((*CCStr)[chunk_number].Kind) { |
| 1028 | case CodeCompletionString::CK_TypedText: |
| 1029 | return CXCompletionChunk_TypedText; |
| 1030 | case CodeCompletionString::CK_Text: |
| 1031 | return CXCompletionChunk_Text; |
| 1032 | case CodeCompletionString::CK_Optional: |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1033 | return CXCompletionChunk_Optional; |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1034 | case CodeCompletionString::CK_Placeholder: |
| 1035 | return CXCompletionChunk_Placeholder; |
| 1036 | case CodeCompletionString::CK_Informative: |
| 1037 | return CXCompletionChunk_Informative; |
| 1038 | case CodeCompletionString::CK_CurrentParameter: |
| 1039 | return CXCompletionChunk_CurrentParameter; |
| 1040 | case CodeCompletionString::CK_LeftParen: |
| 1041 | return CXCompletionChunk_LeftParen; |
| 1042 | case CodeCompletionString::CK_RightParen: |
| 1043 | return CXCompletionChunk_RightParen; |
| 1044 | case CodeCompletionString::CK_LeftBracket: |
| 1045 | return CXCompletionChunk_LeftBracket; |
| 1046 | case CodeCompletionString::CK_RightBracket: |
| 1047 | return CXCompletionChunk_RightBracket; |
| 1048 | case CodeCompletionString::CK_LeftBrace: |
| 1049 | return CXCompletionChunk_LeftBrace; |
| 1050 | case CodeCompletionString::CK_RightBrace: |
| 1051 | return CXCompletionChunk_RightBrace; |
| 1052 | case CodeCompletionString::CK_LeftAngle: |
| 1053 | return CXCompletionChunk_LeftAngle; |
| 1054 | case CodeCompletionString::CK_RightAngle: |
| 1055 | return CXCompletionChunk_RightAngle; |
| 1056 | case CodeCompletionString::CK_Comma: |
| 1057 | return CXCompletionChunk_Comma; |
| 1058 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1059 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1060 | // Should be unreachable, but let's be careful. |
| 1061 | return CXCompletionChunk_Text; |
| 1062 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1063 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1064 | const char *clang_getCompletionChunkText(CXCompletionString completion_string, |
| 1065 | unsigned chunk_number) { |
| 1066 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
| 1067 | if (!CCStr || chunk_number >= CCStr->size()) |
| 1068 | return 0; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1069 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1070 | switch ((*CCStr)[chunk_number].Kind) { |
| 1071 | case CodeCompletionString::CK_TypedText: |
| 1072 | case CodeCompletionString::CK_Text: |
| 1073 | case CodeCompletionString::CK_Placeholder: |
| 1074 | case CodeCompletionString::CK_CurrentParameter: |
| 1075 | case CodeCompletionString::CK_Informative: |
| 1076 | case CodeCompletionString::CK_LeftParen: |
| 1077 | case CodeCompletionString::CK_RightParen: |
| 1078 | case CodeCompletionString::CK_LeftBracket: |
| 1079 | case CodeCompletionString::CK_RightBracket: |
| 1080 | case CodeCompletionString::CK_LeftBrace: |
| 1081 | case CodeCompletionString::CK_RightBrace: |
| 1082 | case CodeCompletionString::CK_LeftAngle: |
| 1083 | case CodeCompletionString::CK_RightAngle: |
| 1084 | case CodeCompletionString::CK_Comma: |
| 1085 | return (*CCStr)[chunk_number].Text; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1086 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1087 | case CodeCompletionString::CK_Optional: |
| 1088 | // Note: treated as an empty text block. |
| 1089 | return 0; |
| 1090 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1091 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1092 | // Should be unreachable, but let's be careful. |
| 1093 | return 0; |
| 1094 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1095 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1096 | CXCompletionString |
| 1097 | clang_getCompletionChunkCompletionString(CXCompletionString completion_string, |
| 1098 | unsigned chunk_number) { |
| 1099 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
| 1100 | if (!CCStr || chunk_number >= CCStr->size()) |
| 1101 | return 0; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1102 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1103 | switch ((*CCStr)[chunk_number].Kind) { |
| 1104 | case CodeCompletionString::CK_TypedText: |
| 1105 | case CodeCompletionString::CK_Text: |
| 1106 | case CodeCompletionString::CK_Placeholder: |
| 1107 | case CodeCompletionString::CK_CurrentParameter: |
| 1108 | case CodeCompletionString::CK_Informative: |
| 1109 | case CodeCompletionString::CK_LeftParen: |
| 1110 | case CodeCompletionString::CK_RightParen: |
| 1111 | case CodeCompletionString::CK_LeftBracket: |
| 1112 | case CodeCompletionString::CK_RightBracket: |
| 1113 | case CodeCompletionString::CK_LeftBrace: |
| 1114 | case CodeCompletionString::CK_RightBrace: |
| 1115 | case CodeCompletionString::CK_LeftAngle: |
| 1116 | case CodeCompletionString::CK_RightAngle: |
| 1117 | case CodeCompletionString::CK_Comma: |
| 1118 | return 0; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1119 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1120 | case CodeCompletionString::CK_Optional: |
| 1121 | // Note: treated as an empty text block. |
| 1122 | return (*CCStr)[chunk_number].Optional; |
| 1123 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1124 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1125 | // Should be unreachable, but let's be careful. |
| 1126 | return 0; |
| 1127 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1128 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1129 | unsigned clang_getNumCompletionChunks(CXCompletionString completion_string) { |
| 1130 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
| 1131 | return CCStr? CCStr->size() : 0; |
| 1132 | } |
Steve Naroff | 4ade6d6 | 2009-09-23 17:52:52 +0000 | [diff] [blame] | 1133 | |
Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 1134 | static bool ReadUnsigned(const char *&Memory, const char *MemoryEnd, |
| 1135 | unsigned &Value) { |
| 1136 | if (Memory + sizeof(unsigned) > MemoryEnd) |
| 1137 | return true; |
| 1138 | |
| 1139 | memmove(&Value, Memory, sizeof(unsigned)); |
| 1140 | Memory += sizeof(unsigned); |
| 1141 | return false; |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1142 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1143 | |
| 1144 | void clang_codeComplete(CXIndex CIdx, |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1145 | const char *source_filename, |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1146 | int num_command_line_args, |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1147 | const char **command_line_args, |
Douglas Gregor | 735df88 | 2009-12-02 09:21:34 +0000 | [diff] [blame] | 1148 | unsigned num_unsaved_files, |
| 1149 | struct CXUnsavedFile *unsaved_files, |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1150 | const char *complete_filename, |
| 1151 | unsigned complete_line, |
| 1152 | unsigned complete_column, |
| 1153 | CXCompletionIterator completion_iterator, |
| 1154 | CXClientData client_data) { |
| 1155 | // The indexer, which is mainly used to determine where diagnostics go. |
| 1156 | CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1157 | |
Douglas Gregor | 735df88 | 2009-12-02 09:21:34 +0000 | [diff] [blame] | 1158 | // The set of temporary files that we've built. |
| 1159 | std::vector<llvm::sys::Path> TemporaryFiles; |
| 1160 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1161 | // Build up the arguments for invoking 'clang'. |
| 1162 | std::vector<const char *> argv; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1163 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1164 | // First add the complete path to the 'clang' executable. |
| 1165 | llvm::sys::Path ClangPath = CXXIdx->getClangPath(); |
| 1166 | argv.push_back(ClangPath.c_str()); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1167 | |
| 1168 | // Add the '-fsyntax-only' argument so that we only perform a basic |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1169 | // syntax check of the code. |
| 1170 | argv.push_back("-fsyntax-only"); |
| 1171 | |
| 1172 | // Add the appropriate '-code-completion-at=file:line:column' argument |
| 1173 | // to perform code completion, with an "-Xclang" preceding it. |
| 1174 | std::string code_complete_at; |
| 1175 | code_complete_at += "-code-completion-at="; |
| 1176 | code_complete_at += complete_filename; |
| 1177 | code_complete_at += ":"; |
| 1178 | code_complete_at += llvm::utostr(complete_line); |
| 1179 | code_complete_at += ":"; |
| 1180 | code_complete_at += llvm::utostr(complete_column); |
| 1181 | argv.push_back("-Xclang"); |
| 1182 | argv.push_back(code_complete_at.c_str()); |
| 1183 | argv.push_back("-Xclang"); |
Daniel Dunbar | 4db166b | 2009-11-19 05:32:09 +0000 | [diff] [blame] | 1184 | argv.push_back("-no-code-completion-debug-printer"); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1185 | |
Douglas Gregor | 735df88 | 2009-12-02 09:21:34 +0000 | [diff] [blame] | 1186 | std::vector<std::string> RemapArgs; |
| 1187 | for (unsigned i = 0; i != num_unsaved_files; ++i) { |
| 1188 | char tmpFile[L_tmpnam]; |
| 1189 | char *tmpFileName = tmpnam(tmpFile); |
| 1190 | |
| 1191 | // Write the contents of this unsaved file into the temporary file. |
| 1192 | llvm::sys::Path SavedFile(tmpFileName); |
| 1193 | std::string ErrorInfo; |
| 1194 | llvm::raw_fd_ostream OS(SavedFile.c_str(), ErrorInfo); |
| 1195 | if (!ErrorInfo.empty()) |
| 1196 | continue; |
| 1197 | |
| 1198 | OS.write(unsaved_files[i].Contents, unsaved_files[i].Length); |
| 1199 | OS.close(); |
| 1200 | if (OS.has_error()) { |
| 1201 | SavedFile.eraseFromDisk(); |
| 1202 | continue; |
| 1203 | } |
| 1204 | |
| 1205 | // Remap the file. |
| 1206 | std::string RemapArg = "-remap-file="; |
| 1207 | RemapArg += unsaved_files[i].Filename; |
| 1208 | RemapArg += ';'; |
| 1209 | RemapArg += tmpFileName; |
| 1210 | RemapArgs.push_back("-Xclang"); |
| 1211 | RemapArgs.push_back(RemapArg); |
| 1212 | TemporaryFiles.push_back(SavedFile); |
| 1213 | } |
| 1214 | |
| 1215 | // The pointers into the elements of RemapArgs are stable because we |
| 1216 | // won't be adding anything to RemapArgs after this point. |
| 1217 | for (unsigned i = 0, e = RemapArgs.size(); i != e; ++i) |
| 1218 | argv.push_back(RemapArgs[i].c_str()); |
| 1219 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1220 | // Add the source file name (FIXME: later, we'll want to build temporary |
| 1221 | // file from the buffer, or just feed the source text via standard input). |
Ted Kremenek | 4633d1b | 2009-11-17 18:18:02 +0000 | [diff] [blame] | 1222 | if (source_filename) |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1223 | argv.push_back(source_filename); |
| 1224 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1225 | // Process the compiler options, stripping off '-o', '-c', '-fsyntax-only'. |
| 1226 | for (int i = 0; i < num_command_line_args; ++i) |
| 1227 | if (const char *arg = command_line_args[i]) { |
| 1228 | if (strcmp(arg, "-o") == 0) { |
| 1229 | ++i; // Also skip the matching argument. |
| 1230 | continue; |
| 1231 | } |
| 1232 | if (strcmp(arg, "-emit-ast") == 0 || |
| 1233 | strcmp(arg, "-c") == 0 || |
| 1234 | strcmp(arg, "-fsyntax-only") == 0) { |
| 1235 | continue; |
| 1236 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1237 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1238 | // Keep the argument. |
| 1239 | argv.push_back(arg); |
| 1240 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1241 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1242 | // Add the null terminator. |
| 1243 | argv.push_back(NULL); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1244 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1245 | // Generate a temporary name for the AST file. |
| 1246 | char tmpFile[L_tmpnam]; |
| 1247 | char *tmpFileName = tmpnam(tmpFile); |
| 1248 | llvm::sys::Path ResultsFile(tmpFileName); |
Douglas Gregor | 735df88 | 2009-12-02 09:21:34 +0000 | [diff] [blame] | 1249 | TemporaryFiles.push_back(ResultsFile); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1250 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1251 | // Invoke 'clang'. |
| 1252 | llvm::sys::Path DevNull; // leave empty, causes redirection to /dev/null |
| 1253 | // on Unix or NUL (Windows). |
| 1254 | std::string ErrMsg; |
| 1255 | const llvm::sys::Path *Redirects[] = { &DevNull, &ResultsFile, &DevNull, 0 }; |
| 1256 | llvm::sys::Program::ExecuteAndWait(ClangPath, &argv[0], /* env */ NULL, |
| 1257 | /* redirects */ &Redirects[0], |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1258 | /* secondsToWait */ 0, |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1259 | /* memoryLimits */ 0, &ErrMsg); |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1260 | |
Ted Kremenek | 0854d70 | 2009-11-10 19:18:52 +0000 | [diff] [blame] | 1261 | if (CXXIdx->getDisplayDiagnostics() && !ErrMsg.empty()) { |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1262 | llvm::errs() << "clang_codeComplete: " << ErrMsg |
Daniel Dunbar | acca725 | 2009-11-30 20:42:49 +0000 | [diff] [blame] | 1263 | << '\n' << "Arguments: \n"; |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1264 | for (std::vector<const char*>::iterator I = argv.begin(), E = argv.end(); |
| 1265 | I!=E; ++I) { |
| 1266 | if (*I) |
| 1267 | llvm::errs() << ' ' << *I << '\n'; |
| 1268 | } |
| 1269 | llvm::errs() << '\n'; |
| 1270 | } |
| 1271 | |
| 1272 | // Parse the resulting source file to find code-completion results. |
| 1273 | using llvm::MemoryBuffer; |
| 1274 | using llvm::StringRef; |
| 1275 | if (MemoryBuffer *F = MemoryBuffer::getFile(ResultsFile.c_str())) { |
| 1276 | StringRef Buffer = F->getBuffer(); |
Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 1277 | for (const char *Str = Buffer.data(), *StrEnd = Str + Buffer.size(); |
| 1278 | Str < StrEnd;) { |
| 1279 | unsigned KindValue; |
| 1280 | if (ReadUnsigned(Str, StrEnd, KindValue)) |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1281 | break; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1282 | |
Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 1283 | CodeCompletionString *CCStr |
| 1284 | = CodeCompletionString::Deserialize(Str, StrEnd); |
| 1285 | if (!CCStr) |
| 1286 | continue; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1287 | |
Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 1288 | if (!CCStr->empty()) { |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1289 | // Vend the code-completion result to the caller. |
| 1290 | CXCompletionResult Result; |
Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 1291 | Result.CursorKind = (CXCursorKind)KindValue; |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1292 | Result.CompletionString = CCStr; |
| 1293 | if (completion_iterator) |
| 1294 | completion_iterator(&Result, client_data); |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1295 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1296 | |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1297 | delete CCStr; |
Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 1298 | }; |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1299 | delete F; |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1300 | } |
| 1301 | |
Douglas Gregor | 735df88 | 2009-12-02 09:21:34 +0000 | [diff] [blame] | 1302 | for (unsigned i = 0, e = TemporaryFiles.size(); i != e; ++i) |
| 1303 | TemporaryFiles[i].eraseFromDisk(); |
Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1304 | } |
Daniel Dunbar | 0d7dd22 | 2009-11-30 20:42:43 +0000 | [diff] [blame] | 1305 | |
Steve Naroff | 600866c | 2009-08-27 19:51:58 +0000 | [diff] [blame] | 1306 | } // end extern "C" |