Chris Lattner | eb8c963 | 2007-10-07 06:04:32 +0000 | [diff] [blame] | 1 | //===--- ASTConsumers.cpp - ASTConsumer implementations -------------------===// |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 959e5be | 2007-12-29 19:59:25 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
Chris Lattner | eb8c963 | 2007-10-07 06:04:32 +0000 | [diff] [blame] | 10 | // AST Consumer Implementations. |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
Chris Lattner | eb8c963 | 2007-10-07 06:04:32 +0000 | [diff] [blame] | 14 | #include "ASTConsumers.h" |
Ted Kremenek | dd0126b | 2008-03-31 18:26:32 +0000 | [diff] [blame] | 15 | #include "HTMLDiagnostics.h" |
Ted Kremenek | ac88193 | 2007-12-18 21:34:28 +0000 | [diff] [blame] | 16 | #include "clang/AST/TranslationUnit.h" |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 17 | #include "clang/Basic/SourceManager.h" |
| 18 | #include "clang/Basic/FileManager.h" |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 19 | #include "clang/AST/AST.h" |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 20 | #include "clang/AST/ASTConsumer.h" |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 21 | #include "llvm/Support/Streams.h" |
Ted Kremenek | 0118bb5 | 2008-02-18 21:21:23 +0000 | [diff] [blame] | 22 | #include "llvm/Support/Timer.h" |
Ted Kremenek | dd0126b | 2008-03-31 18:26:32 +0000 | [diff] [blame] | 23 | #include "llvm/ADT/OwningPtr.h" |
| 24 | |
Chris Lattner | 9557878 | 2007-08-08 22:51:59 +0000 | [diff] [blame] | 25 | using namespace clang; |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 26 | |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 27 | //===----------------------------------------------------------------------===// |
| 28 | /// DeclPrinter - Utility class for printing top-level decls. |
Chris Lattner | 9557878 | 2007-08-08 22:51:59 +0000 | [diff] [blame] | 29 | |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 30 | namespace { |
| 31 | class DeclPrinter { |
| 32 | public: |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 33 | std::ostream& Out; |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 34 | |
Chris Lattner | 216012f | 2008-01-10 01:43:14 +0000 | [diff] [blame] | 35 | DeclPrinter(std::ostream* out) : Out(out ? *out : *llvm::cerr.stream()) {} |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 36 | DeclPrinter() : Out(*llvm::cerr.stream()) {} |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 37 | |
Chris Lattner | 1c1aabb | 2008-01-02 21:04:16 +0000 | [diff] [blame] | 38 | void PrintDecl(Decl *D); |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 39 | void PrintFunctionDeclStart(FunctionDecl *FD); |
| 40 | void PrintTypeDefDecl(TypedefDecl *TD); |
Chris Lattner | 806a5f5 | 2008-01-12 07:05:38 +0000 | [diff] [blame] | 41 | void PrintLinkageSpec(LinkageSpecDecl *LS); |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 42 | void PrintObjCMethodDecl(ObjCMethodDecl *OMD); |
| 43 | void PrintObjCImplementationDecl(ObjCImplementationDecl *OID); |
| 44 | void PrintObjCInterfaceDecl(ObjCInterfaceDecl *OID); |
| 45 | void PrintObjCProtocolDecl(ObjCProtocolDecl *PID); |
| 46 | void PrintObjCCategoryImplDecl(ObjCCategoryImplDecl *PID); |
| 47 | void PrintObjCCategoryDecl(ObjCCategoryDecl *PID); |
| 48 | void PrintObjCCompatibleAliasDecl(ObjCCompatibleAliasDecl *AID); |
Fariborz Jahanian | 8516e9a | 2008-04-17 18:25:18 +0000 | [diff] [blame] | 49 | void PrintObjCPropertyDecl(ObjCPropertyDecl *PD); |
Fariborz Jahanian | dc0569e | 2008-04-23 00:06:01 +0000 | [diff] [blame] | 50 | void PrintObjCPropertyImplDecl(ObjCPropertyImplDecl *PID); |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 51 | }; |
| 52 | } // end anonymous namespace |
| 53 | |
Chris Lattner | 1c1aabb | 2008-01-02 21:04:16 +0000 | [diff] [blame] | 54 | void DeclPrinter:: PrintDecl(Decl *D) { |
| 55 | if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) { |
| 56 | PrintFunctionDeclStart(FD); |
| 57 | |
| 58 | if (FD->getBody()) { |
| 59 | Out << ' '; |
| 60 | FD->getBody()->printPretty(Out); |
| 61 | Out << '\n'; |
| 62 | } |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 63 | } else if (isa<ObjCMethodDecl>(D)) { |
Chris Lattner | 1c1aabb | 2008-01-02 21:04:16 +0000 | [diff] [blame] | 64 | // Do nothing, methods definitions are printed in |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 65 | // PrintObjCImplementationDecl. |
Chris Lattner | 1c1aabb | 2008-01-02 21:04:16 +0000 | [diff] [blame] | 66 | } else if (TypedefDecl *TD = dyn_cast<TypedefDecl>(D)) { |
| 67 | PrintTypeDefDecl(TD); |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 68 | } else if (ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(D)) { |
| 69 | PrintObjCInterfaceDecl(OID); |
| 70 | } else if (ObjCProtocolDecl *PID = dyn_cast<ObjCProtocolDecl>(D)) { |
| 71 | PrintObjCProtocolDecl(PID); |
| 72 | } else if (ObjCForwardProtocolDecl *OFPD = |
Chris Lattner | 43b885f | 2008-02-25 21:04:36 +0000 | [diff] [blame] | 73 | dyn_cast<ObjCForwardProtocolDecl>(D)) { |
Chris Lattner | 1c1aabb | 2008-01-02 21:04:16 +0000 | [diff] [blame] | 74 | Out << "@protocol "; |
| 75 | for (unsigned i = 0, e = OFPD->getNumForwardDecls(); i != e; ++i) { |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 76 | const ObjCProtocolDecl *D = OFPD->getForwardProtocolDecl(i); |
Chris Lattner | 1c1aabb | 2008-01-02 21:04:16 +0000 | [diff] [blame] | 77 | if (i) Out << ", "; |
| 78 | Out << D->getName(); |
| 79 | } |
| 80 | Out << ";\n"; |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 81 | } else if (ObjCImplementationDecl *OID = |
Chris Lattner | 43b885f | 2008-02-25 21:04:36 +0000 | [diff] [blame] | 82 | dyn_cast<ObjCImplementationDecl>(D)) { |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 83 | PrintObjCImplementationDecl(OID); |
| 84 | } else if (ObjCCategoryImplDecl *OID = |
Chris Lattner | 43b885f | 2008-02-25 21:04:36 +0000 | [diff] [blame] | 85 | dyn_cast<ObjCCategoryImplDecl>(D)) { |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 86 | PrintObjCCategoryImplDecl(OID); |
| 87 | } else if (ObjCCategoryDecl *OID = |
Chris Lattner | 43b885f | 2008-02-25 21:04:36 +0000 | [diff] [blame] | 88 | dyn_cast<ObjCCategoryDecl>(D)) { |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 89 | PrintObjCCategoryDecl(OID); |
| 90 | } else if (ObjCCompatibleAliasDecl *OID = |
Chris Lattner | 43b885f | 2008-02-25 21:04:36 +0000 | [diff] [blame] | 91 | dyn_cast<ObjCCompatibleAliasDecl>(D)) { |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 92 | PrintObjCCompatibleAliasDecl(OID); |
Chris Lattner | e134914 | 2008-06-21 21:40:20 +0000 | [diff] [blame] | 93 | } else if (ObjCClassDecl *OFCD = dyn_cast<ObjCClassDecl>(D)) { |
| 94 | Out << "@class "; |
| 95 | ObjCInterfaceDecl **ForwardDecls = OFCD->getForwardDecls(); |
| 96 | for (unsigned i = 0, e = OFCD->getNumForwardDecls(); i != e; ++i) { |
| 97 | const ObjCInterfaceDecl *D = ForwardDecls[i]; |
| 98 | if (i) Out << ", "; |
| 99 | Out << D->getName(); |
| 100 | } |
| 101 | Out << ";\n"; |
Chris Lattner | 1c1aabb | 2008-01-02 21:04:16 +0000 | [diff] [blame] | 102 | } else if (TagDecl *TD = dyn_cast<TagDecl>(D)) { |
| 103 | Out << "Read top-level tag decl: '" << TD->getName() << "'\n"; |
| 104 | } else if (ScopedDecl *SD = dyn_cast<ScopedDecl>(D)) { |
| 105 | Out << "Read top-level variable decl: '" << SD->getName() << "'\n"; |
Chris Lattner | 806a5f5 | 2008-01-12 07:05:38 +0000 | [diff] [blame] | 106 | } else if (LinkageSpecDecl *LSD = dyn_cast<LinkageSpecDecl>(D)) { |
| 107 | PrintLinkageSpec(LSD); |
Anders Carlsson | 4f7f441 | 2008-02-08 00:33:21 +0000 | [diff] [blame] | 108 | } else if (FileScopeAsmDecl *AD = dyn_cast<FileScopeAsmDecl>(D)) { |
| 109 | Out << "asm("; |
| 110 | AD->getAsmString()->printPretty(Out); |
| 111 | Out << ")\n"; |
Chris Lattner | 1c1aabb | 2008-01-02 21:04:16 +0000 | [diff] [blame] | 112 | } else { |
| 113 | assert(0 && "Unknown decl type!"); |
| 114 | } |
| 115 | } |
| 116 | |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 117 | void DeclPrinter::PrintFunctionDeclStart(FunctionDecl *FD) { |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 118 | bool HasBody = FD->getBody(); |
| 119 | |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 120 | Out << '\n'; |
Chris Lattner | 987058a | 2007-08-26 04:02:13 +0000 | [diff] [blame] | 121 | |
| 122 | switch (FD->getStorageClass()) { |
| 123 | default: assert(0 && "Unknown storage class"); |
| 124 | case FunctionDecl::None: break; |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 125 | case FunctionDecl::Extern: Out << "extern "; break; |
| 126 | case FunctionDecl::Static: Out << "static "; break; |
Ted Kremenek | c6d26e0 | 2008-04-15 03:57:09 +0000 | [diff] [blame] | 127 | case FunctionDecl::PrivateExtern: Out << "__private_extern__ "; break; |
Chris Lattner | 987058a | 2007-08-26 04:02:13 +0000 | [diff] [blame] | 128 | } |
| 129 | |
| 130 | if (FD->isInline()) |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 131 | Out << "inline "; |
Chris Lattner | 987058a | 2007-08-26 04:02:13 +0000 | [diff] [blame] | 132 | |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 133 | std::string Proto = FD->getName(); |
Chris Lattner | 934fff6 | 2007-12-03 21:43:25 +0000 | [diff] [blame] | 134 | const FunctionType *AFT = FD->getType()->getAsFunctionType(); |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 135 | |
Chris Lattner | 934fff6 | 2007-12-03 21:43:25 +0000 | [diff] [blame] | 136 | if (const FunctionTypeProto *FT = dyn_cast<FunctionTypeProto>(AFT)) { |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 137 | Proto += "("; |
| 138 | for (unsigned i = 0, e = FD->getNumParams(); i != e; ++i) { |
| 139 | if (i) Proto += ", "; |
| 140 | std::string ParamStr; |
| 141 | if (HasBody) ParamStr = FD->getParamDecl(i)->getName(); |
| 142 | |
| 143 | FT->getArgType(i).getAsStringInternal(ParamStr); |
| 144 | Proto += ParamStr; |
| 145 | } |
| 146 | |
| 147 | if (FT->isVariadic()) { |
| 148 | if (FD->getNumParams()) Proto += ", "; |
| 149 | Proto += "..."; |
| 150 | } |
| 151 | Proto += ")"; |
| 152 | } else { |
| 153 | assert(isa<FunctionTypeNoProto>(AFT)); |
| 154 | Proto += "()"; |
| 155 | } |
| 156 | |
| 157 | AFT->getResultType().getAsStringInternal(Proto); |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 158 | Out << Proto; |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 159 | |
Chris Lattner | 9557878 | 2007-08-08 22:51:59 +0000 | [diff] [blame] | 160 | if (!FD->getBody()) |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 161 | Out << ";\n"; |
Chris Lattner | 9557878 | 2007-08-08 22:51:59 +0000 | [diff] [blame] | 162 | // Doesn't print the body. |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 163 | } |
| 164 | |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 165 | void DeclPrinter::PrintTypeDefDecl(TypedefDecl *TD) { |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 166 | std::string S = TD->getName(); |
| 167 | TD->getUnderlyingType().getAsStringInternal(S); |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 168 | Out << "typedef " << S << ";\n"; |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 169 | } |
| 170 | |
Chris Lattner | 806a5f5 | 2008-01-12 07:05:38 +0000 | [diff] [blame] | 171 | void DeclPrinter::PrintLinkageSpec(LinkageSpecDecl *LS) { |
| 172 | const char *l; |
| 173 | if (LS->getLanguage() == LinkageSpecDecl::lang_c) |
| 174 | l = "C"; |
Chris Lattner | 690c287 | 2008-04-08 05:52:18 +0000 | [diff] [blame] | 175 | else { |
| 176 | assert(LS->getLanguage() == LinkageSpecDecl::lang_cxx && |
| 177 | "unknown language in linkage specification"); |
Chris Lattner | 806a5f5 | 2008-01-12 07:05:38 +0000 | [diff] [blame] | 178 | l = "C++"; |
Chris Lattner | 690c287 | 2008-04-08 05:52:18 +0000 | [diff] [blame] | 179 | } |
Chris Lattner | 806a5f5 | 2008-01-12 07:05:38 +0000 | [diff] [blame] | 180 | Out << "extern \"" << l << "\" { "; |
| 181 | PrintDecl(LS->getDecl()); |
| 182 | Out << "}\n"; |
| 183 | } |
| 184 | |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 185 | void DeclPrinter::PrintObjCMethodDecl(ObjCMethodDecl *OMD) { |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 186 | if (OMD->isInstance()) |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 187 | Out << "\n- "; |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 188 | else |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 189 | Out << "\n+ "; |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 190 | if (!OMD->getResultType().isNull()) |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 191 | Out << '(' << OMD->getResultType().getAsString() << ") "; |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 192 | // FIXME: just print original selector name! |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 193 | Out << OMD->getSelector().getName(); |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 194 | |
Chris Lattner | 685d792 | 2008-03-16 01:07:14 +0000 | [diff] [blame] | 195 | for (unsigned i = 0, e = OMD->getNumParams(); i != e; ++i) { |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 196 | ParmVarDecl *PDecl = OMD->getParamDecl(i); |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 197 | // FIXME: selector is missing here! |
| 198 | Out << " :(" << PDecl->getType().getAsString() << ") " << PDecl->getName(); |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 199 | } |
| 200 | } |
| 201 | |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 202 | void DeclPrinter::PrintObjCImplementationDecl(ObjCImplementationDecl *OID) { |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 203 | std::string I = OID->getName(); |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 204 | ObjCInterfaceDecl *SID = OID->getSuperClass(); |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 205 | |
| 206 | if (SID) |
| 207 | Out << "@implementation " << I << " : " << SID->getName(); |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 208 | else |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 209 | Out << "@implementation " << I; |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 210 | |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 211 | for (ObjCImplementationDecl::instmeth_iterator I = OID->instmeth_begin(), |
Chris Lattner | dea5bec | 2007-12-12 07:46:12 +0000 | [diff] [blame] | 212 | E = OID->instmeth_end(); I != E; ++I) { |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 213 | ObjCMethodDecl *OMD = *I; |
| 214 | PrintObjCMethodDecl(OMD); |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 215 | if (OMD->getBody()) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 216 | Out << ' '; |
| 217 | OMD->getBody()->printPretty(Out); |
| 218 | Out << '\n'; |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 219 | } |
| 220 | } |
| 221 | |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 222 | for (ObjCImplementationDecl::classmeth_iterator I = OID->classmeth_begin(), |
Chris Lattner | dea5bec | 2007-12-12 07:46:12 +0000 | [diff] [blame] | 223 | E = OID->classmeth_end(); I != E; ++I) { |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 224 | ObjCMethodDecl *OMD = *I; |
| 225 | PrintObjCMethodDecl(OMD); |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 226 | if (OMD->getBody()) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 227 | Out << ' '; |
| 228 | OMD->getBody()->printPretty(Out); |
| 229 | Out << '\n'; |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 230 | } |
| 231 | } |
| 232 | |
Fariborz Jahanian | dc0569e | 2008-04-23 00:06:01 +0000 | [diff] [blame] | 233 | for (ObjCImplementationDecl::propimpl_iterator I = OID->propimpl_begin(), |
| 234 | E = OID->propimpl_end(); I != E; ++I) |
| 235 | PrintObjCPropertyImplDecl(*I); |
| 236 | |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 237 | Out << "@end\n"; |
Fariborz Jahanian | 83ddf82 | 2007-11-10 20:59:13 +0000 | [diff] [blame] | 238 | } |
| 239 | |
| 240 | |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 241 | void DeclPrinter::PrintObjCInterfaceDecl(ObjCInterfaceDecl *OID) { |
Fariborz Jahanian | c04aff1 | 2007-10-08 23:06:41 +0000 | [diff] [blame] | 242 | std::string I = OID->getName(); |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 243 | ObjCInterfaceDecl *SID = OID->getSuperClass(); |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 244 | |
| 245 | if (SID) |
| 246 | Out << "@interface " << I << " : " << SID->getName(); |
Fariborz Jahanian | c04aff1 | 2007-10-08 23:06:41 +0000 | [diff] [blame] | 247 | else |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 248 | Out << "@interface " << I; |
| 249 | |
Fariborz Jahanian | c04aff1 | 2007-10-08 23:06:41 +0000 | [diff] [blame] | 250 | // Protocols? |
| 251 | int count = OID->getNumIntfRefProtocols(); |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 252 | |
Fariborz Jahanian | c04aff1 | 2007-10-08 23:06:41 +0000 | [diff] [blame] | 253 | if (count > 0) { |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 254 | ObjCProtocolDecl **refProtocols = OID->getReferencedProtocols(); |
Fariborz Jahanian | c04aff1 | 2007-10-08 23:06:41 +0000 | [diff] [blame] | 255 | for (int i = 0; i < count; i++) |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 256 | Out << (i == 0 ? '<' : ',') << refProtocols[i]->getName(); |
Fariborz Jahanian | c04aff1 | 2007-10-08 23:06:41 +0000 | [diff] [blame] | 257 | } |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 258 | |
Fariborz Jahanian | c04aff1 | 2007-10-08 23:06:41 +0000 | [diff] [blame] | 259 | if (count > 0) |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 260 | Out << ">\n"; |
Fariborz Jahanian | c04aff1 | 2007-10-08 23:06:41 +0000 | [diff] [blame] | 261 | else |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 262 | Out << '\n'; |
Fariborz Jahanian | f468b31 | 2007-10-26 16:29:12 +0000 | [diff] [blame] | 263 | |
Chris Lattner | ec4979b | 2008-03-16 21:08:55 +0000 | [diff] [blame] | 264 | if (OID->ivar_size() > 0) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 265 | Out << '{'; |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 266 | for (ObjCInterfaceDecl::ivar_iterator I = OID->ivar_begin(), |
Chris Lattner | c7b0675 | 2007-12-12 07:56:42 +0000 | [diff] [blame] | 267 | E = OID->ivar_end(); I != E; ++I) { |
| 268 | Out << '\t' << (*I)->getType().getAsString() |
| 269 | << ' ' << (*I)->getName() << ";\n"; |
Fariborz Jahanian | f468b31 | 2007-10-26 16:29:12 +0000 | [diff] [blame] | 270 | } |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 271 | Out << "}\n"; |
Fariborz Jahanian | f468b31 | 2007-10-26 16:29:12 +0000 | [diff] [blame] | 272 | } |
Fariborz Jahanian | d8df6d8 | 2007-11-06 22:01:00 +0000 | [diff] [blame] | 273 | |
Fariborz Jahanian | 8516e9a | 2008-04-17 18:25:18 +0000 | [diff] [blame] | 274 | for (ObjCInterfaceDecl::classprop_iterator I = OID->classprop_begin(), |
| 275 | E = OID->classprop_end(); I != E; ++I) |
| 276 | PrintObjCPropertyDecl(*I); |
Fariborz Jahanian | e4534e7 | 2008-05-07 17:43:59 +0000 | [diff] [blame] | 277 | bool eol_needed = false; |
Fariborz Jahanian | deeae05 | 2008-05-06 23:14:25 +0000 | [diff] [blame] | 278 | for (ObjCInterfaceDecl::classmeth_iterator I = OID->classmeth_begin(), |
| 279 | E = OID->classmeth_end(); I != E; ++I) |
Fariborz Jahanian | e4534e7 | 2008-05-07 17:43:59 +0000 | [diff] [blame] | 280 | eol_needed = true, PrintObjCMethodDecl(*I); |
Fariborz Jahanian | deeae05 | 2008-05-06 23:14:25 +0000 | [diff] [blame] | 281 | |
| 282 | for (ObjCInterfaceDecl::instmeth_iterator I = OID->instmeth_begin(), |
| 283 | E = OID->instmeth_end(); I != E; ++I) |
Fariborz Jahanian | e4534e7 | 2008-05-07 17:43:59 +0000 | [diff] [blame] | 284 | eol_needed = true, PrintObjCMethodDecl(*I); |
Fariborz Jahanian | deeae05 | 2008-05-06 23:14:25 +0000 | [diff] [blame] | 285 | |
Fariborz Jahanian | e4534e7 | 2008-05-07 17:43:59 +0000 | [diff] [blame] | 286 | Out << (eol_needed ? "\n@end\n" : "@end\n"); |
Steve Naroff | faed3bf | 2007-09-10 20:51:04 +0000 | [diff] [blame] | 287 | // FIXME: implement the rest... |
| 288 | } |
| 289 | |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 290 | void DeclPrinter::PrintObjCProtocolDecl(ObjCProtocolDecl *PID) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 291 | Out << "@protocol " << PID->getName() << '\n'; |
Fariborz Jahanian | 8516e9a | 2008-04-17 18:25:18 +0000 | [diff] [blame] | 292 | |
| 293 | for (ObjCProtocolDecl::classprop_iterator I = PID->classprop_begin(), |
| 294 | E = PID->classprop_end(); I != E; ++I) |
| 295 | PrintObjCPropertyDecl(*I); |
| 296 | Out << "@end\n"; |
Fariborz Jahanian | ac20be2 | 2007-10-08 18:53:38 +0000 | [diff] [blame] | 297 | // FIXME: implement the rest... |
| 298 | } |
| 299 | |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 300 | void DeclPrinter::PrintObjCCategoryImplDecl(ObjCCategoryImplDecl *PID) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 301 | Out << "@implementation " |
| 302 | << PID->getClassInterface()->getName() |
| 303 | << '(' << PID->getName() << ");\n"; |
Fariborz Jahanian | dc0569e | 2008-04-23 00:06:01 +0000 | [diff] [blame] | 304 | for (ObjCCategoryImplDecl::propimpl_iterator I = PID->propimpl_begin(), |
| 305 | E = PID->propimpl_end(); I != E; ++I) |
| 306 | PrintObjCPropertyImplDecl(*I); |
| 307 | Out << "@end\n"; |
Fariborz Jahanian | ac20be2 | 2007-10-08 18:53:38 +0000 | [diff] [blame] | 308 | // FIXME: implement the rest... |
| 309 | } |
| 310 | |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 311 | void DeclPrinter::PrintObjCCategoryDecl(ObjCCategoryDecl *PID) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 312 | Out << "@interface " |
| 313 | << PID->getClassInterface()->getName() |
| 314 | << '(' << PID->getName() << ");\n"; |
Fariborz Jahanian | 52ff844 | 2008-04-16 21:08:45 +0000 | [diff] [blame] | 315 | // Output property declarations. |
Fariborz Jahanian | 8516e9a | 2008-04-17 18:25:18 +0000 | [diff] [blame] | 316 | for (ObjCCategoryDecl::classprop_iterator I = PID->classprop_begin(), |
| 317 | E = PID->classprop_end(); I != E; ++I) |
| 318 | PrintObjCPropertyDecl(*I); |
Fariborz Jahanian | 52ff844 | 2008-04-16 21:08:45 +0000 | [diff] [blame] | 319 | Out << "@end\n"; |
| 320 | |
Fariborz Jahanian | ac20be2 | 2007-10-08 18:53:38 +0000 | [diff] [blame] | 321 | // FIXME: implement the rest... |
| 322 | } |
| 323 | |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 324 | void DeclPrinter::PrintObjCCompatibleAliasDecl(ObjCCompatibleAliasDecl *AID) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 325 | Out << "@compatibility_alias " << AID->getName() |
| 326 | << ' ' << AID->getClassInterface()->getName() << ";\n"; |
Fariborz Jahanian | 05d212a | 2007-10-11 23:42:27 +0000 | [diff] [blame] | 327 | } |
| 328 | |
Fariborz Jahanian | 8516e9a | 2008-04-17 18:25:18 +0000 | [diff] [blame] | 329 | /// PrintObjCPropertyDecl - print a property declaration. |
| 330 | /// |
| 331 | void DeclPrinter::PrintObjCPropertyDecl(ObjCPropertyDecl *PDecl) { |
Fariborz Jahanian | 4aa72a7 | 2008-05-05 18:51:55 +0000 | [diff] [blame] | 332 | if (PDecl->getPropertyImplementation() == ObjCPropertyDecl::Required) |
| 333 | Out << "@required\n"; |
| 334 | else if (PDecl->getPropertyImplementation() == ObjCPropertyDecl::Optional) |
| 335 | Out << "@optional\n"; |
| 336 | |
Fariborz Jahanian | 8516e9a | 2008-04-17 18:25:18 +0000 | [diff] [blame] | 337 | Out << "@property"; |
| 338 | if (PDecl->getPropertyAttributes() != ObjCPropertyDecl::OBJC_PR_noattr) { |
| 339 | bool first = true; |
| 340 | Out << " ("; |
| 341 | if (PDecl->getPropertyAttributes() & |
| 342 | ObjCPropertyDecl::OBJC_PR_readonly) { |
| 343 | Out << (first ? ' ' : ',') << "readonly"; |
| 344 | first = false; |
| 345 | } |
| 346 | |
| 347 | if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_getter) { |
| 348 | Out << (first ? ' ' : ',') << "getter = " |
Fariborz Jahanian | b7080ae | 2008-05-06 18:09:04 +0000 | [diff] [blame] | 349 | << PDecl->getGetterName().getName(); |
Fariborz Jahanian | 8516e9a | 2008-04-17 18:25:18 +0000 | [diff] [blame] | 350 | first = false; |
| 351 | } |
| 352 | if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_setter) { |
| 353 | Out << (first ? ' ' : ',') << "setter = " |
Fariborz Jahanian | b7080ae | 2008-05-06 18:09:04 +0000 | [diff] [blame] | 354 | << PDecl->getSetterName().getName(); |
Fariborz Jahanian | 8516e9a | 2008-04-17 18:25:18 +0000 | [diff] [blame] | 355 | first = false; |
| 356 | } |
| 357 | |
| 358 | if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_assign) { |
| 359 | Out << (first ? ' ' : ',') << "assign"; |
| 360 | first = false; |
| 361 | } |
| 362 | |
| 363 | if (PDecl->getPropertyAttributes() & |
| 364 | ObjCPropertyDecl::OBJC_PR_readwrite) { |
| 365 | Out << (first ? ' ' : ',') << "readwrite"; |
| 366 | first = false; |
| 367 | } |
| 368 | |
| 369 | if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_retain) { |
| 370 | Out << (first ? ' ' : ',') << "retain"; |
| 371 | first = false; |
| 372 | } |
| 373 | |
| 374 | if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_copy) { |
| 375 | Out << (first ? ' ' : ',') << "copy"; |
| 376 | first = false; |
| 377 | } |
| 378 | |
| 379 | if (PDecl->getPropertyAttributes() & |
| 380 | ObjCPropertyDecl::OBJC_PR_nonatomic) { |
| 381 | Out << (first ? ' ' : ',') << "nonatomic"; |
| 382 | first = false; |
| 383 | } |
| 384 | Out << " )"; |
| 385 | } |
| 386 | Out << ' ' << PDecl->getType().getAsString() |
| 387 | << ' ' << PDecl->getName(); |
| 388 | |
| 389 | Out << ";\n"; |
| 390 | } |
Fariborz Jahanian | dc0569e | 2008-04-23 00:06:01 +0000 | [diff] [blame] | 391 | |
| 392 | /// PrintObjCPropertyImplDecl - Print an objective-c property implementation |
| 393 | /// declaration syntax. |
| 394 | /// |
| 395 | void DeclPrinter::PrintObjCPropertyImplDecl(ObjCPropertyImplDecl *PID) { |
| 396 | if (PID->getPropertyImplementation() == |
| 397 | ObjCPropertyImplDecl::OBJC_PR_IMPL_SYNTHSIZE) |
| 398 | Out << "\n@synthesize "; |
| 399 | else |
| 400 | Out << "\n@dynamic "; |
| 401 | Out << PID->getPropertyDecl()->getName(); |
| 402 | if (PID->getPropertyIvarDecl()) |
| 403 | Out << "=" << PID->getPropertyIvarDecl()->getName(); |
| 404 | Out << ";\n"; |
| 405 | } |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 406 | //===----------------------------------------------------------------------===// |
| 407 | /// ASTPrinter - Pretty-printer of ASTs |
| 408 | |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 409 | namespace { |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 410 | class ASTPrinter : public ASTConsumer, public DeclPrinter { |
| 411 | public: |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 412 | ASTPrinter(std::ostream* o = NULL) : DeclPrinter(o) {} |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 413 | |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 414 | virtual void HandleTopLevelDecl(Decl *D) { |
Chris Lattner | 1c1aabb | 2008-01-02 21:04:16 +0000 | [diff] [blame] | 415 | PrintDecl(D); |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 416 | } |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 417 | }; |
Chris Lattner | 4b00965 | 2007-07-25 00:24:17 +0000 | [diff] [blame] | 418 | } |
Chris Lattner | 9557878 | 2007-08-08 22:51:59 +0000 | [diff] [blame] | 419 | |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 420 | ASTConsumer *clang::CreateASTPrinter(std::ostream* out) { |
| 421 | return new ASTPrinter(out); |
| 422 | } |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 423 | |
| 424 | //===----------------------------------------------------------------------===// |
| 425 | /// ASTDumper - Low-level dumper of ASTs |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 426 | |
| 427 | namespace { |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 428 | class ASTDumper : public ASTConsumer, public DeclPrinter { |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 429 | SourceManager *SM; |
| 430 | public: |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 431 | ASTDumper() : DeclPrinter() {} |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 432 | |
Ted Kremenek | 17861c5 | 2007-12-19 22:51:13 +0000 | [diff] [blame] | 433 | void Initialize(ASTContext &Context) { |
Ted Kremenek | b3ee193 | 2007-12-11 21:27:55 +0000 | [diff] [blame] | 434 | SM = &Context.getSourceManager(); |
Chris Lattner | 9557878 | 2007-08-08 22:51:59 +0000 | [diff] [blame] | 435 | } |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 436 | |
| 437 | virtual void HandleTopLevelDecl(Decl *D) { |
| 438 | if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) { |
| 439 | PrintFunctionDeclStart(FD); |
| 440 | |
| 441 | if (FD->getBody()) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 442 | Out << '\n'; |
| 443 | // FIXME: convert dumper to use std::ostream? |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 444 | FD->getBody()->dumpAll(*SM); |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 445 | Out << '\n'; |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 446 | } |
| 447 | } else if (TypedefDecl *TD = dyn_cast<TypedefDecl>(D)) { |
| 448 | PrintTypeDefDecl(TD); |
| 449 | } else if (ScopedDecl *SD = dyn_cast<ScopedDecl>(D)) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 450 | Out << "Read top-level variable decl: '" << SD->getName() << "'\n"; |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 451 | } else if (ObjCInterfaceDecl *OID = dyn_cast<ObjCInterfaceDecl>(D)) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 452 | Out << "Read objc interface '" << OID->getName() << "'\n"; |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 453 | } else if (ObjCProtocolDecl *OPD = dyn_cast<ObjCProtocolDecl>(D)) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 454 | Out << "Read objc protocol '" << OPD->getName() << "'\n"; |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 455 | } else if (ObjCCategoryDecl *OCD = dyn_cast<ObjCCategoryDecl>(D)) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 456 | Out << "Read objc category '" << OCD->getName() << "'\n"; |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 457 | } else if (isa<ObjCForwardProtocolDecl>(D)) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 458 | Out << "Read objc fwd protocol decl\n"; |
Ted Kremenek | 42730c5 | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 459 | } else if (isa<ObjCClassDecl>(D)) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 460 | Out << "Read objc fwd class decl\n"; |
Anders Carlsson | 4f7f441 | 2008-02-08 00:33:21 +0000 | [diff] [blame] | 461 | } else if (isa<FileScopeAsmDecl>(D)) { |
| 462 | Out << "Read file scope asm decl\n"; |
Ted Kremenek | 5d257d4 | 2008-03-14 17:31:00 +0000 | [diff] [blame] | 463 | } else if (ObjCMethodDecl* MD = dyn_cast<ObjCMethodDecl>(D)) { |
| 464 | Out << "Read objc method decl: '" << MD->getSelector().getName() |
| 465 | << "'\n"; |
Steve Naroff | 045e0e0 | 2008-05-23 18:50:58 +0000 | [diff] [blame] | 466 | if (MD->getBody()) { |
| 467 | // FIXME: convert dumper to use std::ostream? |
| 468 | MD->getBody()->dumpAll(*SM); |
| 469 | Out << '\n'; |
| 470 | } |
Ted Kremenek | 5d257d4 | 2008-03-14 17:31:00 +0000 | [diff] [blame] | 471 | } else if (isa<ObjCImplementationDecl>(D)) { |
| 472 | Out << "Read objc implementation decl\n"; |
| 473 | } |
| 474 | else { |
Chris Lattner | d5c9d3d | 2007-10-06 18:52:10 +0000 | [diff] [blame] | 475 | assert(0 && "Unknown decl type!"); |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 476 | } |
| 477 | } |
| 478 | }; |
Chris Lattner | 9557878 | 2007-08-08 22:51:59 +0000 | [diff] [blame] | 479 | } |
| 480 | |
Chris Lattner | b73abd5 | 2007-09-15 23:02:28 +0000 | [diff] [blame] | 481 | ASTConsumer *clang::CreateASTDumper() { return new ASTDumper(); } |
| 482 | |
Ted Kremenek | e09391a | 2007-11-27 21:46:50 +0000 | [diff] [blame] | 483 | //===----------------------------------------------------------------------===// |
| 484 | /// ASTViewer - AST Visualization |
| 485 | |
Ted Kremenek | b6976a2 | 2007-09-19 21:29:43 +0000 | [diff] [blame] | 486 | namespace { |
| 487 | class ASTViewer : public ASTConsumer { |
| 488 | SourceManager *SM; |
| 489 | public: |
Ted Kremenek | 17861c5 | 2007-12-19 22:51:13 +0000 | [diff] [blame] | 490 | void Initialize(ASTContext &Context) { |
Ted Kremenek | b3ee193 | 2007-12-11 21:27:55 +0000 | [diff] [blame] | 491 | SM = &Context.getSourceManager(); |
Ted Kremenek | b6976a2 | 2007-09-19 21:29:43 +0000 | [diff] [blame] | 492 | } |
| 493 | |
| 494 | virtual void HandleTopLevelDecl(Decl *D) { |
| 495 | if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 496 | DeclPrinter().PrintFunctionDeclStart(FD); |
Ted Kremenek | b6976a2 | 2007-09-19 21:29:43 +0000 | [diff] [blame] | 497 | |
| 498 | if (FD->getBody()) { |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 499 | llvm::cerr << '\n'; |
Ted Kremenek | b6976a2 | 2007-09-19 21:29:43 +0000 | [diff] [blame] | 500 | FD->getBody()->viewAST(); |
Ted Kremenek | be2ea3b | 2007-11-28 21:32:21 +0000 | [diff] [blame] | 501 | llvm::cerr << '\n'; |
Ted Kremenek | b6976a2 | 2007-09-19 21:29:43 +0000 | [diff] [blame] | 502 | } |
| 503 | } |
Ted Kremenek | 5d257d4 | 2008-03-14 17:31:00 +0000 | [diff] [blame] | 504 | else if (ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D)) { |
| 505 | DeclPrinter().PrintObjCMethodDecl(MD); |
| 506 | |
| 507 | if (MD->getBody()) { |
| 508 | llvm::cerr << '\n'; |
| 509 | MD->getBody()->viewAST(); |
| 510 | llvm::cerr << '\n'; |
| 511 | } |
| 512 | } |
Ted Kremenek | b6976a2 | 2007-09-19 21:29:43 +0000 | [diff] [blame] | 513 | } |
| 514 | }; |
| 515 | } |
| 516 | |
| 517 | ASTConsumer *clang::CreateASTViewer() { return new ASTViewer(); } |
| 518 | |
Ted Kremenek | 1e3c202 | 2007-09-07 23:47:56 +0000 | [diff] [blame] | 519 | //===----------------------------------------------------------------------===// |
Ted Kremenek | 397de01 | 2007-12-13 00:37:31 +0000 | [diff] [blame] | 520 | // AST Serializer |
| 521 | |
| 522 | namespace { |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 523 | |
| 524 | class ASTSerializer : public ASTConsumer { |
| 525 | protected: |
Ted Kremenek | 863b01f | 2008-04-23 16:25:39 +0000 | [diff] [blame] | 526 | TranslationUnit* TU; |
| 527 | |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 528 | public: |
Eli Friedman | b456369 | 2008-06-09 20:02:51 +0000 | [diff] [blame] | 529 | ASTSerializer() : TU(0) {} |
| 530 | |
| 531 | virtual void InitializeTU(TranslationUnit &tu) { |
| 532 | TU = &tu; |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 533 | } |
Eli Friedman | b456369 | 2008-06-09 20:02:51 +0000 | [diff] [blame] | 534 | |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 535 | }; |
Eli Friedman | b456369 | 2008-06-09 20:02:51 +0000 | [diff] [blame] | 536 | |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 537 | class SingleFileSerializer : public ASTSerializer { |
| 538 | const llvm::sys::Path FName; |
| 539 | public: |
Eli Friedman | b456369 | 2008-06-09 20:02:51 +0000 | [diff] [blame] | 540 | SingleFileSerializer(const llvm::sys::Path& F) : FName(F) {} |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 541 | |
| 542 | ~SingleFileSerializer() { |
Ted Kremenek | 863b01f | 2008-04-23 16:25:39 +0000 | [diff] [blame] | 543 | EmitASTBitcodeFile(TU, FName); |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 544 | } |
| 545 | }; |
| 546 | |
| 547 | class BuildSerializer : public ASTSerializer { |
| 548 | llvm::sys::Path EmitDir; |
| 549 | public: |
Eli Friedman | b456369 | 2008-06-09 20:02:51 +0000 | [diff] [blame] | 550 | BuildSerializer(const llvm::sys::Path& dir) : EmitDir(dir) {} |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 551 | |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 552 | ~BuildSerializer() { |
Ted Kremenek | 863b01f | 2008-04-23 16:25:39 +0000 | [diff] [blame] | 553 | |
| 554 | if (!TU) |
| 555 | return; |
| 556 | |
| 557 | SourceManager& SourceMgr = TU->getContext().getSourceManager(); |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 558 | unsigned ID = SourceMgr.getMainFileID(); |
| 559 | assert (ID && "MainFileID not set!"); |
| 560 | const FileEntry* FE = SourceMgr.getFileEntryForID(ID); |
| 561 | assert (FE && "No FileEntry for main file."); |
| 562 | |
| 563 | // FIXME: This is not portable to Windows. |
| 564 | // FIXME: This logic should probably be moved elsewhere later. |
| 565 | |
Ted Kremenek | 0c7cd7a | 2007-12-20 19:47:16 +0000 | [diff] [blame] | 566 | llvm::sys::Path FName(EmitDir); |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 567 | |
| 568 | std::vector<char> buf; |
| 569 | buf.reserve(strlen(FE->getName())+100); |
| 570 | |
| 571 | sprintf(&buf[0], "dev_%llx", (uint64_t) FE->getDevice()); |
Ted Kremenek | 0c7cd7a | 2007-12-20 19:47:16 +0000 | [diff] [blame] | 572 | FName.appendComponent(&buf[0]); |
| 573 | FName.createDirectoryOnDisk(true); |
| 574 | if (!FName.canWrite() || !FName.isDirectory()) { |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 575 | assert (false && "Could not create 'device' serialization directory."); |
| 576 | return; |
| 577 | } |
Ted Kremenek | 0c7cd7a | 2007-12-20 19:47:16 +0000 | [diff] [blame] | 578 | |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 579 | sprintf(&buf[0], "%s-%llX.ast", FE->getName(), (uint64_t) FE->getInode()); |
Ted Kremenek | 0c7cd7a | 2007-12-20 19:47:16 +0000 | [diff] [blame] | 580 | FName.appendComponent(&buf[0]); |
Ted Kremenek | 863b01f | 2008-04-23 16:25:39 +0000 | [diff] [blame] | 581 | EmitASTBitcodeFile(TU, FName); |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 582 | |
Ted Kremenek | 0c7cd7a | 2007-12-20 19:47:16 +0000 | [diff] [blame] | 583 | // Now emit the sources. |
| 584 | |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 585 | } |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 586 | }; |
| 587 | |
| 588 | |
Ted Kremenek | 397de01 | 2007-12-13 00:37:31 +0000 | [diff] [blame] | 589 | } // end anonymous namespace |
| 590 | |
| 591 | |
Ted Kremenek | d890f6a | 2007-12-19 22:24:34 +0000 | [diff] [blame] | 592 | ASTConsumer* clang::CreateASTSerializer(const std::string& InFile, |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 593 | const std::string& OutputFile, |
Ted Kremenek | 842126e | 2008-06-04 15:55:15 +0000 | [diff] [blame] | 594 | Diagnostic &Diags) { |
Ted Kremenek | bde3033 | 2007-12-19 17:25:59 +0000 | [diff] [blame] | 595 | |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 596 | if (OutputFile.size()) { |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 597 | if (InFile == "-") { |
| 598 | llvm::cerr << |
| 599 | "error: Cannot use --serialize with -o for source read from STDIN.\n"; |
| 600 | return NULL; |
| 601 | } |
| 602 | |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 603 | // The user specified an AST-emission directory. Determine if the path |
| 604 | // is absolute. |
| 605 | llvm::sys::Path EmitDir(OutputFile); |
| 606 | |
| 607 | if (!EmitDir.isAbsolute()) { |
| 608 | llvm::cerr << |
| 609 | "error: Output directory for --serialize must be an absolute path.\n"; |
| 610 | |
| 611 | return NULL; |
| 612 | } |
| 613 | |
| 614 | // Create the directory if it does not exist. |
| 615 | EmitDir.createDirectoryOnDisk(true); |
| 616 | if (!EmitDir.canWrite() || !EmitDir.isDirectory()) { |
| 617 | llvm::cerr << |
| 618 | "error: Could not create output directory for --serialize.\n"; |
| 619 | |
| 620 | return NULL; |
| 621 | } |
| 622 | |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 623 | // FIXME: We should probably only allow using BuildSerializer when |
| 624 | // the ASTs come from parsed source files, and not from .ast files. |
Eli Friedman | b456369 | 2008-06-09 20:02:51 +0000 | [diff] [blame] | 625 | return new BuildSerializer(EmitDir); |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 626 | } |
| 627 | |
| 628 | // The user did not specify an output directory for serialized ASTs. |
| 629 | // Serialize the translation to a single file whose name is the same |
| 630 | // as the input file with the ".ast" extension appended. |
Ted Kremenek | ab74937 | 2007-12-19 19:27:38 +0000 | [diff] [blame] | 631 | |
Ted Kremenek | 2118901 | 2007-12-19 23:49:37 +0000 | [diff] [blame] | 632 | llvm::sys::Path FName(InFile.c_str()); |
Ted Kremenek | fc17b8a | 2007-12-20 00:34:58 +0000 | [diff] [blame] | 633 | FName.appendSuffix("ast"); |
Eli Friedman | b456369 | 2008-06-09 20:02:51 +0000 | [diff] [blame] | 634 | return new SingleFileSerializer(FName); |
Ted Kremenek | 397de01 | 2007-12-13 00:37:31 +0000 | [diff] [blame] | 635 | } |