Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1 | //===--- StmtPrinter.cpp - Printing implementation for Stmt ASTs ----------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 0bc735f | 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. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
Chris Lattner | 6000dac | 2007-08-08 22:51:59 +0000 | [diff] [blame] | 10 | // This file implements the Stmt::dumpPretty/Stmt::printPretty methods, which |
| 11 | // pretty print the AST back out to C code. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
| 15 | #include "clang/AST/StmtVisitor.h" |
Douglas Gregor | 1a49af9 | 2009-01-06 05:10:23 +0000 | [diff] [blame] | 16 | #include "clang/AST/DeclCXX.h" |
Ted Kremenek | 91d1d7a | 2007-10-17 18:36:42 +0000 | [diff] [blame] | 17 | #include "clang/AST/DeclObjC.h" |
Ted Kremenek | 42a509f | 2007-08-31 21:30:12 +0000 | [diff] [blame] | 18 | #include "clang/AST/PrettyPrinter.h" |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 19 | #include "llvm/Support/Compiler.h" |
Ted Kremenek | 51221ec | 2007-11-26 22:50:46 +0000 | [diff] [blame] | 20 | #include "llvm/Support/Streams.h" |
Ted Kremenek | a95d375 | 2008-09-13 05:16:45 +0000 | [diff] [blame] | 21 | #include "llvm/Support/Format.h" |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 22 | using namespace clang; |
| 23 | |
| 24 | //===----------------------------------------------------------------------===// |
| 25 | // StmtPrinter Visitor |
| 26 | //===----------------------------------------------------------------------===// |
| 27 | |
| 28 | namespace { |
Chris Lattner | c5598cb | 2007-08-21 04:04:25 +0000 | [diff] [blame] | 29 | class VISIBILITY_HIDDEN StmtPrinter : public StmtVisitor<StmtPrinter> { |
Ted Kremenek | a95d375 | 2008-09-13 05:16:45 +0000 | [diff] [blame] | 30 | llvm::raw_ostream &OS; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 31 | unsigned IndentLevel; |
Mike Stump | 071e4da | 2009-02-10 20:16:46 +0000 | [diff] [blame^] | 32 | bool NoIndent; |
Ted Kremenek | 42a509f | 2007-08-31 21:30:12 +0000 | [diff] [blame] | 33 | clang::PrinterHelper* Helper; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 34 | public: |
Mike Stump | 071e4da | 2009-02-10 20:16:46 +0000 | [diff] [blame^] | 35 | StmtPrinter(llvm::raw_ostream &os, PrinterHelper* helper, unsigned I=0, |
| 36 | bool noIndent=false) : |
| 37 | OS(os), IndentLevel(I), NoIndent(noIndent), Helper(helper) {} |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 38 | |
| 39 | void PrintStmt(Stmt *S, int SubIndent = 1) { |
| 40 | IndentLevel += SubIndent; |
| 41 | if (S && isa<Expr>(S)) { |
| 42 | // If this is an expr used in a stmt context, indent and newline it. |
| 43 | Indent(); |
Chris Lattner | c5598cb | 2007-08-21 04:04:25 +0000 | [diff] [blame] | 44 | Visit(S); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 45 | OS << ";\n"; |
| 46 | } else if (S) { |
Chris Lattner | c5598cb | 2007-08-21 04:04:25 +0000 | [diff] [blame] | 47 | Visit(S); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 48 | } else { |
| 49 | Indent() << "<<<NULL STATEMENT>>>\n"; |
| 50 | } |
| 51 | IndentLevel -= SubIndent; |
| 52 | } |
| 53 | |
| 54 | void PrintRawCompoundStmt(CompoundStmt *S); |
| 55 | void PrintRawDecl(Decl *D); |
Ted Kremenek | ecd64c5 | 2008-10-06 18:39:36 +0000 | [diff] [blame] | 56 | void PrintRawDeclStmt(DeclStmt *S); |
Mike Stump | 071e4da | 2009-02-10 20:16:46 +0000 | [diff] [blame^] | 57 | void PrintFieldDecl(FieldDecl *FD); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 58 | void PrintRawIfStmt(IfStmt *If); |
Sebastian Redl | 8351da0 | 2008-12-22 21:35:02 +0000 | [diff] [blame] | 59 | void PrintRawCXXCatchStmt(CXXCatchStmt *Catch); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 60 | |
| 61 | void PrintExpr(Expr *E) { |
| 62 | if (E) |
Chris Lattner | c5598cb | 2007-08-21 04:04:25 +0000 | [diff] [blame] | 63 | Visit(E); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 64 | else |
| 65 | OS << "<null expr>"; |
| 66 | } |
| 67 | |
Mike Stump | 071e4da | 2009-02-10 20:16:46 +0000 | [diff] [blame^] | 68 | llvm::raw_ostream &Indent(int Delta = 0) { |
| 69 | if (!NoIndent) { |
| 70 | for (int i = 0, e = IndentLevel+Delta; i < e; ++i) |
| 71 | OS << " "; |
| 72 | } else NoIndent = false; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 73 | return OS; |
| 74 | } |
| 75 | |
Chris Lattner | 704fe35 | 2007-08-30 17:59:59 +0000 | [diff] [blame] | 76 | bool PrintOffsetOfDesignator(Expr *E); |
| 77 | void VisitUnaryOffsetOf(UnaryOperator *Node); |
| 78 | |
Ted Kremenek | 42a509f | 2007-08-31 21:30:12 +0000 | [diff] [blame] | 79 | void Visit(Stmt* S) { |
| 80 | if (Helper && Helper->handledStmt(S,OS)) |
| 81 | return; |
| 82 | else StmtVisitor<StmtPrinter>::Visit(S); |
| 83 | } |
| 84 | |
Chris Lattner | c5598cb | 2007-08-21 04:04:25 +0000 | [diff] [blame] | 85 | void VisitStmt(Stmt *Node); |
Douglas Gregor | f2cad86 | 2008-11-14 12:46:07 +0000 | [diff] [blame] | 86 | #define STMT(CLASS, PARENT) \ |
Chris Lattner | c5598cb | 2007-08-21 04:04:25 +0000 | [diff] [blame] | 87 | void Visit##CLASS(CLASS *Node); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 88 | #include "clang/AST/StmtNodes.def" |
| 89 | }; |
| 90 | } |
| 91 | |
| 92 | //===----------------------------------------------------------------------===// |
| 93 | // Stmt printing methods. |
| 94 | //===----------------------------------------------------------------------===// |
| 95 | |
| 96 | void StmtPrinter::VisitStmt(Stmt *Node) { |
| 97 | Indent() << "<<unknown stmt type>>\n"; |
| 98 | } |
| 99 | |
| 100 | /// PrintRawCompoundStmt - Print a compound stmt without indenting the {, and |
| 101 | /// with no newline after the }. |
| 102 | void StmtPrinter::PrintRawCompoundStmt(CompoundStmt *Node) { |
| 103 | OS << "{\n"; |
| 104 | for (CompoundStmt::body_iterator I = Node->body_begin(), E = Node->body_end(); |
| 105 | I != E; ++I) |
| 106 | PrintStmt(*I); |
| 107 | |
| 108 | Indent() << "}"; |
| 109 | } |
| 110 | |
| 111 | void StmtPrinter::PrintRawDecl(Decl *D) { |
| 112 | // FIXME: Need to complete/beautify this... this code simply shows the |
| 113 | // nodes are where they need to be. |
| 114 | if (TypedefDecl *localType = dyn_cast<TypedefDecl>(D)) { |
| 115 | OS << "typedef " << localType->getUnderlyingType().getAsString(); |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 116 | OS << " " << localType->getNameAsString(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 117 | } else if (ValueDecl *VD = dyn_cast<ValueDecl>(D)) { |
| 118 | // Emit storage class for vardecls. |
| 119 | if (VarDecl *V = dyn_cast<VarDecl>(VD)) { |
| 120 | switch (V->getStorageClass()) { |
| 121 | default: assert(0 && "Unknown storage class!"); |
| 122 | case VarDecl::None: break; |
| 123 | case VarDecl::Extern: OS << "extern "; break; |
| 124 | case VarDecl::Static: OS << "static "; break; |
| 125 | case VarDecl::Auto: OS << "auto "; break; |
| 126 | case VarDecl::Register: OS << "register "; break; |
| 127 | } |
| 128 | } |
| 129 | |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 130 | std::string Name = VD->getNameAsString(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 131 | VD->getType().getAsStringInternal(Name); |
| 132 | OS << Name; |
| 133 | |
Chris Lattner | 24c3990 | 2007-07-12 00:36:32 +0000 | [diff] [blame] | 134 | // If this is a vardecl with an initializer, emit it. |
| 135 | if (VarDecl *V = dyn_cast<VarDecl>(VD)) { |
| 136 | if (V->getInit()) { |
| 137 | OS << " = "; |
| 138 | PrintExpr(V->getInit()); |
| 139 | } |
| 140 | } |
Steve Naroff | 91578f3 | 2007-11-17 21:21:01 +0000 | [diff] [blame] | 141 | } else if (TagDecl *TD = dyn_cast<TagDecl>(D)) { |
| 142 | // print a free standing tag decl (e.g. "struct x;"). |
| 143 | OS << TD->getKindName(); |
| 144 | OS << " "; |
| 145 | if (const IdentifierInfo *II = TD->getIdentifier()) |
| 146 | OS << II->getName(); |
Mike Stump | 071e4da | 2009-02-10 20:16:46 +0000 | [diff] [blame^] | 147 | if (RecordDecl *RD = dyn_cast<RecordDecl>(TD)) { |
| 148 | OS << "{\n"; |
| 149 | IndentLevel += 1; |
| 150 | for (RecordDecl::field_iterator i = RD->field_begin(); i != RD->field_end(); ++i) { |
| 151 | PrintFieldDecl(*i); |
| 152 | IndentLevel -= 1; |
| 153 | } |
| 154 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 155 | } else { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 156 | assert(0 && "Unexpected decl"); |
| 157 | } |
| 158 | } |
| 159 | |
Mike Stump | 071e4da | 2009-02-10 20:16:46 +0000 | [diff] [blame^] | 160 | void StmtPrinter::PrintFieldDecl(FieldDecl *FD) { |
| 161 | Indent() << FD->getNameAsString() << "\n"; |
| 162 | } |
| 163 | |
Ted Kremenek | ecd64c5 | 2008-10-06 18:39:36 +0000 | [diff] [blame] | 164 | void StmtPrinter::PrintRawDeclStmt(DeclStmt *S) { |
Mike Stump | 071e4da | 2009-02-10 20:16:46 +0000 | [diff] [blame^] | 165 | bool isFirst = true; |
Ted Kremenek | ecd64c5 | 2008-10-06 18:39:36 +0000 | [diff] [blame] | 166 | |
| 167 | for (DeclStmt::decl_iterator I = S->decl_begin(), E = S->decl_end(); |
| 168 | I != E; ++I) { |
| 169 | |
| 170 | if (!isFirst) OS << ", "; |
| 171 | else isFirst = false; |
| 172 | |
| 173 | PrintRawDecl(*I); |
| 174 | } |
| 175 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 176 | |
| 177 | void StmtPrinter::VisitNullStmt(NullStmt *Node) { |
| 178 | Indent() << ";\n"; |
| 179 | } |
| 180 | |
| 181 | void StmtPrinter::VisitDeclStmt(DeclStmt *Node) { |
Ted Kremenek | ecd64c5 | 2008-10-06 18:39:36 +0000 | [diff] [blame] | 182 | for (DeclStmt::decl_iterator I = Node->decl_begin(), E = Node->decl_end(); |
| 183 | I!=E; ++I) { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 184 | Indent(); |
Ted Kremenek | ecd64c5 | 2008-10-06 18:39:36 +0000 | [diff] [blame] | 185 | PrintRawDecl(*I); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 186 | OS << ";\n"; |
| 187 | } |
| 188 | } |
| 189 | |
| 190 | void StmtPrinter::VisitCompoundStmt(CompoundStmt *Node) { |
| 191 | Indent(); |
| 192 | PrintRawCompoundStmt(Node); |
| 193 | OS << "\n"; |
| 194 | } |
| 195 | |
| 196 | void StmtPrinter::VisitCaseStmt(CaseStmt *Node) { |
| 197 | Indent(-1) << "case "; |
| 198 | PrintExpr(Node->getLHS()); |
| 199 | if (Node->getRHS()) { |
| 200 | OS << " ... "; |
| 201 | PrintExpr(Node->getRHS()); |
| 202 | } |
| 203 | OS << ":\n"; |
| 204 | |
| 205 | PrintStmt(Node->getSubStmt(), 0); |
| 206 | } |
| 207 | |
| 208 | void StmtPrinter::VisitDefaultStmt(DefaultStmt *Node) { |
| 209 | Indent(-1) << "default:\n"; |
| 210 | PrintStmt(Node->getSubStmt(), 0); |
| 211 | } |
| 212 | |
| 213 | void StmtPrinter::VisitLabelStmt(LabelStmt *Node) { |
| 214 | Indent(-1) << Node->getName() << ":\n"; |
| 215 | PrintStmt(Node->getSubStmt(), 0); |
| 216 | } |
| 217 | |
| 218 | void StmtPrinter::PrintRawIfStmt(IfStmt *If) { |
Sebastian Redl | bfee9b2 | 2009-02-07 20:05:48 +0000 | [diff] [blame] | 219 | OS << "if ("; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 220 | PrintExpr(If->getCond()); |
Sebastian Redl | bfee9b2 | 2009-02-07 20:05:48 +0000 | [diff] [blame] | 221 | OS << ')'; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 222 | |
| 223 | if (CompoundStmt *CS = dyn_cast<CompoundStmt>(If->getThen())) { |
| 224 | OS << ' '; |
| 225 | PrintRawCompoundStmt(CS); |
| 226 | OS << (If->getElse() ? ' ' : '\n'); |
| 227 | } else { |
| 228 | OS << '\n'; |
| 229 | PrintStmt(If->getThen()); |
| 230 | if (If->getElse()) Indent(); |
| 231 | } |
| 232 | |
| 233 | if (Stmt *Else = If->getElse()) { |
| 234 | OS << "else"; |
| 235 | |
| 236 | if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Else)) { |
| 237 | OS << ' '; |
| 238 | PrintRawCompoundStmt(CS); |
| 239 | OS << '\n'; |
| 240 | } else if (IfStmt *ElseIf = dyn_cast<IfStmt>(Else)) { |
| 241 | OS << ' '; |
| 242 | PrintRawIfStmt(ElseIf); |
| 243 | } else { |
| 244 | OS << '\n'; |
| 245 | PrintStmt(If->getElse()); |
| 246 | } |
| 247 | } |
| 248 | } |
| 249 | |
| 250 | void StmtPrinter::VisitIfStmt(IfStmt *If) { |
| 251 | Indent(); |
| 252 | PrintRawIfStmt(If); |
| 253 | } |
| 254 | |
| 255 | void StmtPrinter::VisitSwitchStmt(SwitchStmt *Node) { |
| 256 | Indent() << "switch ("; |
| 257 | PrintExpr(Node->getCond()); |
| 258 | OS << ")"; |
| 259 | |
| 260 | // Pretty print compoundstmt bodies (very common). |
| 261 | if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) { |
| 262 | OS << " "; |
| 263 | PrintRawCompoundStmt(CS); |
| 264 | OS << "\n"; |
| 265 | } else { |
| 266 | OS << "\n"; |
| 267 | PrintStmt(Node->getBody()); |
| 268 | } |
| 269 | } |
| 270 | |
Anders Carlsson | c1fcb77 | 2007-07-22 07:07:56 +0000 | [diff] [blame] | 271 | void StmtPrinter::VisitSwitchCase(SwitchCase*) { |
| 272 | assert(0 && "SwitchCase is an abstract class"); |
| 273 | } |
| 274 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 275 | void StmtPrinter::VisitWhileStmt(WhileStmt *Node) { |
| 276 | Indent() << "while ("; |
| 277 | PrintExpr(Node->getCond()); |
| 278 | OS << ")\n"; |
| 279 | PrintStmt(Node->getBody()); |
| 280 | } |
| 281 | |
| 282 | void StmtPrinter::VisitDoStmt(DoStmt *Node) { |
Chris Lattner | 8bdcc47 | 2007-09-15 21:49:37 +0000 | [diff] [blame] | 283 | Indent() << "do "; |
| 284 | if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) { |
| 285 | PrintRawCompoundStmt(CS); |
| 286 | OS << " "; |
| 287 | } else { |
| 288 | OS << "\n"; |
| 289 | PrintStmt(Node->getBody()); |
| 290 | Indent(); |
| 291 | } |
| 292 | |
| 293 | OS << "while "; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 294 | PrintExpr(Node->getCond()); |
| 295 | OS << ";\n"; |
| 296 | } |
| 297 | |
| 298 | void StmtPrinter::VisitForStmt(ForStmt *Node) { |
| 299 | Indent() << "for ("; |
| 300 | if (Node->getInit()) { |
| 301 | if (DeclStmt *DS = dyn_cast<DeclStmt>(Node->getInit())) |
Ted Kremenek | ecd64c5 | 2008-10-06 18:39:36 +0000 | [diff] [blame] | 302 | PrintRawDeclStmt(DS); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 303 | else |
| 304 | PrintExpr(cast<Expr>(Node->getInit())); |
| 305 | } |
Chris Lattner | 8bdcc47 | 2007-09-15 21:49:37 +0000 | [diff] [blame] | 306 | OS << ";"; |
| 307 | if (Node->getCond()) { |
| 308 | OS << " "; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 309 | PrintExpr(Node->getCond()); |
Chris Lattner | 8bdcc47 | 2007-09-15 21:49:37 +0000 | [diff] [blame] | 310 | } |
| 311 | OS << ";"; |
| 312 | if (Node->getInc()) { |
| 313 | OS << " "; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 314 | PrintExpr(Node->getInc()); |
Chris Lattner | 8bdcc47 | 2007-09-15 21:49:37 +0000 | [diff] [blame] | 315 | } |
| 316 | OS << ") "; |
| 317 | |
| 318 | if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) { |
| 319 | PrintRawCompoundStmt(CS); |
| 320 | OS << "\n"; |
| 321 | } else { |
| 322 | OS << "\n"; |
| 323 | PrintStmt(Node->getBody()); |
| 324 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 325 | } |
| 326 | |
Ted Kremenek | a526c5c | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 327 | void StmtPrinter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *Node) { |
Fariborz Jahanian | 0196cab | 2008-01-02 22:54:34 +0000 | [diff] [blame] | 328 | Indent() << "for ("; |
| 329 | if (DeclStmt *DS = dyn_cast<DeclStmt>(Node->getElement())) |
Ted Kremenek | ecd64c5 | 2008-10-06 18:39:36 +0000 | [diff] [blame] | 330 | PrintRawDeclStmt(DS); |
Fariborz Jahanian | 0196cab | 2008-01-02 22:54:34 +0000 | [diff] [blame] | 331 | else |
| 332 | PrintExpr(cast<Expr>(Node->getElement())); |
| 333 | OS << " in "; |
| 334 | PrintExpr(Node->getCollection()); |
| 335 | OS << ") "; |
| 336 | |
| 337 | if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) { |
| 338 | PrintRawCompoundStmt(CS); |
| 339 | OS << "\n"; |
| 340 | } else { |
| 341 | OS << "\n"; |
| 342 | PrintStmt(Node->getBody()); |
| 343 | } |
| 344 | } |
| 345 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 346 | void StmtPrinter::VisitGotoStmt(GotoStmt *Node) { |
| 347 | Indent() << "goto " << Node->getLabel()->getName() << ";\n"; |
| 348 | } |
| 349 | |
| 350 | void StmtPrinter::VisitIndirectGotoStmt(IndirectGotoStmt *Node) { |
| 351 | Indent() << "goto *"; |
| 352 | PrintExpr(Node->getTarget()); |
| 353 | OS << ";\n"; |
| 354 | } |
| 355 | |
| 356 | void StmtPrinter::VisitContinueStmt(ContinueStmt *Node) { |
| 357 | Indent() << "continue;\n"; |
| 358 | } |
| 359 | |
| 360 | void StmtPrinter::VisitBreakStmt(BreakStmt *Node) { |
| 361 | Indent() << "break;\n"; |
| 362 | } |
| 363 | |
| 364 | |
| 365 | void StmtPrinter::VisitReturnStmt(ReturnStmt *Node) { |
| 366 | Indent() << "return"; |
| 367 | if (Node->getRetValue()) { |
| 368 | OS << " "; |
| 369 | PrintExpr(Node->getRetValue()); |
| 370 | } |
| 371 | OS << ";\n"; |
| 372 | } |
| 373 | |
Chris Lattner | fe79595 | 2007-10-29 04:04:16 +0000 | [diff] [blame] | 374 | |
| 375 | void StmtPrinter::VisitAsmStmt(AsmStmt *Node) { |
Anders Carlsson | 39c47b5 | 2007-11-23 23:12:25 +0000 | [diff] [blame] | 376 | Indent() << "asm "; |
| 377 | |
| 378 | if (Node->isVolatile()) |
| 379 | OS << "volatile "; |
| 380 | |
| 381 | OS << "("; |
Anders Carlsson | 6a0ef4b | 2007-11-20 19:21:03 +0000 | [diff] [blame] | 382 | VisitStringLiteral(Node->getAsmString()); |
Anders Carlsson | b235fc2 | 2007-11-22 01:36:19 +0000 | [diff] [blame] | 383 | |
| 384 | // Outputs |
| 385 | if (Node->getNumOutputs() != 0 || Node->getNumInputs() != 0 || |
| 386 | Node->getNumClobbers() != 0) |
| 387 | OS << " : "; |
| 388 | |
| 389 | for (unsigned i = 0, e = Node->getNumOutputs(); i != e; ++i) { |
| 390 | if (i != 0) |
| 391 | OS << ", "; |
| 392 | |
| 393 | if (!Node->getOutputName(i).empty()) { |
| 394 | OS << '['; |
| 395 | OS << Node->getOutputName(i); |
| 396 | OS << "] "; |
| 397 | } |
| 398 | |
| 399 | VisitStringLiteral(Node->getOutputConstraint(i)); |
| 400 | OS << " "; |
| 401 | Visit(Node->getOutputExpr(i)); |
| 402 | } |
| 403 | |
| 404 | // Inputs |
| 405 | if (Node->getNumInputs() != 0 || Node->getNumClobbers() != 0) |
| 406 | OS << " : "; |
| 407 | |
| 408 | for (unsigned i = 0, e = Node->getNumInputs(); i != e; ++i) { |
| 409 | if (i != 0) |
| 410 | OS << ", "; |
| 411 | |
| 412 | if (!Node->getInputName(i).empty()) { |
| 413 | OS << '['; |
| 414 | OS << Node->getInputName(i); |
| 415 | OS << "] "; |
| 416 | } |
| 417 | |
| 418 | VisitStringLiteral(Node->getInputConstraint(i)); |
| 419 | OS << " "; |
| 420 | Visit(Node->getInputExpr(i)); |
| 421 | } |
| 422 | |
| 423 | // Clobbers |
| 424 | if (Node->getNumClobbers() != 0) |
| 425 | OS << " : "; |
| 426 | |
| 427 | for (unsigned i = 0, e = Node->getNumClobbers(); i != e; ++i) { |
| 428 | if (i != 0) |
| 429 | OS << ", "; |
| 430 | |
| 431 | VisitStringLiteral(Node->getClobber(i)); |
| 432 | } |
| 433 | |
Anders Carlsson | 6a0ef4b | 2007-11-20 19:21:03 +0000 | [diff] [blame] | 434 | OS << ");\n"; |
Chris Lattner | fe79595 | 2007-10-29 04:04:16 +0000 | [diff] [blame] | 435 | } |
| 436 | |
Ted Kremenek | a526c5c | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 437 | void StmtPrinter::VisitObjCAtTryStmt(ObjCAtTryStmt *Node) { |
Fariborz Jahanian | 7794cb8 | 2007-11-02 18:16:07 +0000 | [diff] [blame] | 438 | Indent() << "@try"; |
| 439 | if (CompoundStmt *TS = dyn_cast<CompoundStmt>(Node->getTryBody())) { |
| 440 | PrintRawCompoundStmt(TS); |
| 441 | OS << "\n"; |
| 442 | } |
| 443 | |
Ted Kremenek | a526c5c | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 444 | for (ObjCAtCatchStmt *catchStmt = |
| 445 | static_cast<ObjCAtCatchStmt *>(Node->getCatchStmts()); |
Fariborz Jahanian | 7794cb8 | 2007-11-02 18:16:07 +0000 | [diff] [blame] | 446 | catchStmt; |
| 447 | catchStmt = |
Ted Kremenek | a526c5c | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 448 | static_cast<ObjCAtCatchStmt *>(catchStmt->getNextCatchStmt())) { |
Fariborz Jahanian | 7794cb8 | 2007-11-02 18:16:07 +0000 | [diff] [blame] | 449 | Indent() << "@catch("; |
| 450 | if (catchStmt->getCatchParamStmt()) { |
| 451 | if (DeclStmt *DS = dyn_cast<DeclStmt>(catchStmt->getCatchParamStmt())) |
Ted Kremenek | ecd64c5 | 2008-10-06 18:39:36 +0000 | [diff] [blame] | 452 | PrintRawDeclStmt(DS); |
Fariborz Jahanian | 7794cb8 | 2007-11-02 18:16:07 +0000 | [diff] [blame] | 453 | } |
| 454 | OS << ")"; |
| 455 | if (CompoundStmt *CS = dyn_cast<CompoundStmt>(catchStmt->getCatchBody())) |
| 456 | { |
| 457 | PrintRawCompoundStmt(CS); |
| 458 | OS << "\n"; |
| 459 | } |
| 460 | } |
| 461 | |
Ted Kremenek | a526c5c | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 462 | if (ObjCAtFinallyStmt *FS =static_cast<ObjCAtFinallyStmt *>( |
Fariborz Jahanian | 1e7eab4 | 2007-11-07 00:46:42 +0000 | [diff] [blame] | 463 | Node->getFinallyStmt())) { |
| 464 | Indent() << "@finally"; |
| 465 | PrintRawCompoundStmt(dyn_cast<CompoundStmt>(FS->getFinallyBody())); |
Fariborz Jahanian | 7794cb8 | 2007-11-02 18:16:07 +0000 | [diff] [blame] | 466 | OS << "\n"; |
| 467 | } |
Fariborz Jahanian | b210bd0 | 2007-11-01 21:12:44 +0000 | [diff] [blame] | 468 | } |
| 469 | |
Ted Kremenek | a526c5c | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 470 | void StmtPrinter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *Node) { |
Fariborz Jahanian | b210bd0 | 2007-11-01 21:12:44 +0000 | [diff] [blame] | 471 | } |
| 472 | |
Ted Kremenek | a526c5c | 2008-01-07 19:49:32 +0000 | [diff] [blame] | 473 | void StmtPrinter::VisitObjCAtCatchStmt (ObjCAtCatchStmt *Node) { |
Fariborz Jahanian | b210bd0 | 2007-11-01 21:12:44 +0000 | [diff] [blame] | 474 | Indent() << "@catch (...) { /* todo */ } \n"; |
| 475 | } |
| 476 | |
Fariborz Jahanian | 78a677b | 2008-01-30 17:38:29 +0000 | [diff] [blame] | 477 | void StmtPrinter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *Node) { |
Fariborz Jahanian | 39f8f15 | 2007-11-07 02:00:49 +0000 | [diff] [blame] | 478 | Indent() << "@throw"; |
| 479 | if (Node->getThrowExpr()) { |
| 480 | OS << " "; |
| 481 | PrintExpr(Node->getThrowExpr()); |
| 482 | } |
| 483 | OS << ";\n"; |
| 484 | } |
| 485 | |
Fariborz Jahanian | 78a677b | 2008-01-30 17:38:29 +0000 | [diff] [blame] | 486 | void StmtPrinter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *Node) { |
Fariborz Jahanian | c385c90 | 2008-01-29 18:21:32 +0000 | [diff] [blame] | 487 | Indent() << "@synchronized ("; |
| 488 | PrintExpr(Node->getSynchExpr()); |
| 489 | OS << ")"; |
Fariborz Jahanian | 78a677b | 2008-01-30 17:38:29 +0000 | [diff] [blame] | 490 | PrintRawCompoundStmt(Node->getSynchBody()); |
| 491 | OS << "\n"; |
Fariborz Jahanian | c385c90 | 2008-01-29 18:21:32 +0000 | [diff] [blame] | 492 | } |
| 493 | |
Sebastian Redl | 8351da0 | 2008-12-22 21:35:02 +0000 | [diff] [blame] | 494 | void StmtPrinter::PrintRawCXXCatchStmt(CXXCatchStmt *Node) { |
| 495 | OS << "catch ("; |
Sebastian Redl | 4b07b29 | 2008-12-22 19:15:10 +0000 | [diff] [blame] | 496 | if (Decl *ExDecl = Node->getExceptionDecl()) |
| 497 | PrintRawDecl(ExDecl); |
| 498 | else |
| 499 | OS << "..."; |
| 500 | OS << ") "; |
| 501 | PrintRawCompoundStmt(cast<CompoundStmt>(Node->getHandlerBlock())); |
Sebastian Redl | 8351da0 | 2008-12-22 21:35:02 +0000 | [diff] [blame] | 502 | } |
| 503 | |
| 504 | void StmtPrinter::VisitCXXCatchStmt(CXXCatchStmt *Node) { |
| 505 | Indent(); |
| 506 | PrintRawCXXCatchStmt(Node); |
| 507 | OS << "\n"; |
| 508 | } |
| 509 | |
| 510 | void StmtPrinter::VisitCXXTryStmt(CXXTryStmt *Node) { |
| 511 | Indent() << "try "; |
| 512 | PrintRawCompoundStmt(Node->getTryBlock()); |
| 513 | for(unsigned i = 0, e = Node->getNumHandlers(); i < e; ++i) { |
| 514 | OS << " "; |
| 515 | PrintRawCXXCatchStmt(Node->getHandler(i)); |
| 516 | } |
Sebastian Redl | 4b07b29 | 2008-12-22 19:15:10 +0000 | [diff] [blame] | 517 | OS << "\n"; |
| 518 | } |
| 519 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 520 | //===----------------------------------------------------------------------===// |
| 521 | // Expr printing methods. |
| 522 | //===----------------------------------------------------------------------===// |
| 523 | |
| 524 | void StmtPrinter::VisitExpr(Expr *Node) { |
| 525 | OS << "<<unknown expr type>>"; |
| 526 | } |
| 527 | |
| 528 | void StmtPrinter::VisitDeclRefExpr(DeclRefExpr *Node) { |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 529 | OS << Node->getDecl()->getNameAsString(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 530 | } |
| 531 | |
Douglas Gregor | 1a49af9 | 2009-01-06 05:10:23 +0000 | [diff] [blame] | 532 | void StmtPrinter::VisitQualifiedDeclRefExpr(QualifiedDeclRefExpr *Node) { |
| 533 | // FIXME: Should we keep enough information in QualifiedDeclRefExpr |
| 534 | // to produce the same qualification that the user wrote? |
| 535 | llvm::SmallVector<DeclContext *, 4> Contexts; |
| 536 | |
| 537 | NamedDecl *D = Node->getDecl(); |
| 538 | |
| 539 | // Build up a stack of contexts. |
Douglas Gregor | 4afa39d | 2009-01-20 01:17:11 +0000 | [diff] [blame] | 540 | DeclContext *Ctx = D->getDeclContext(); |
Douglas Gregor | 1a49af9 | 2009-01-06 05:10:23 +0000 | [diff] [blame] | 541 | for (; Ctx; Ctx = Ctx->getParent()) |
| 542 | if (!Ctx->isTransparentContext()) |
| 543 | Contexts.push_back(Ctx); |
| 544 | |
| 545 | while (!Contexts.empty()) { |
| 546 | DeclContext *Ctx = Contexts.back(); |
| 547 | if (isa<TranslationUnitDecl>(Ctx)) |
| 548 | OS << "::"; |
Douglas Gregor | 4afa39d | 2009-01-20 01:17:11 +0000 | [diff] [blame] | 549 | else if (NamedDecl *ND = dyn_cast<NamedDecl>(Ctx)) |
| 550 | OS << ND->getNameAsString() << "::"; |
Douglas Gregor | 1a49af9 | 2009-01-06 05:10:23 +0000 | [diff] [blame] | 551 | Contexts.pop_back(); |
| 552 | } |
| 553 | |
| 554 | OS << D->getNameAsString(); |
| 555 | } |
| 556 | |
Steve Naroff | 7779db4 | 2007-11-12 14:29:37 +0000 | [diff] [blame] | 557 | void StmtPrinter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *Node) { |
Fariborz Jahanian | 232220c | 2007-11-12 22:29:28 +0000 | [diff] [blame] | 558 | if (Node->getBase()) { |
| 559 | PrintExpr(Node->getBase()); |
| 560 | OS << (Node->isArrow() ? "->" : "."); |
| 561 | } |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 562 | OS << Node->getDecl()->getNameAsString(); |
Steve Naroff | 7779db4 | 2007-11-12 14:29:37 +0000 | [diff] [blame] | 563 | } |
| 564 | |
Steve Naroff | ae78407 | 2008-05-30 00:40:33 +0000 | [diff] [blame] | 565 | void StmtPrinter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *Node) { |
| 566 | if (Node->getBase()) { |
| 567 | PrintExpr(Node->getBase()); |
| 568 | OS << "."; |
| 569 | } |
Steve Naroff | c77a636 | 2008-12-04 16:24:46 +0000 | [diff] [blame] | 570 | OS << Node->getProperty()->getNameAsCString(); |
Steve Naroff | ae78407 | 2008-05-30 00:40:33 +0000 | [diff] [blame] | 571 | } |
| 572 | |
Fariborz Jahanian | 5daf570 | 2008-11-22 18:39:36 +0000 | [diff] [blame] | 573 | void StmtPrinter::VisitObjCKVCRefExpr(ObjCKVCRefExpr *Node) { |
| 574 | if (Node->getBase()) { |
| 575 | PrintExpr(Node->getBase()); |
| 576 | OS << "."; |
| 577 | } |
| 578 | // FIXME: Setter/Getter names |
| 579 | } |
| 580 | |
Chris Lattner | d9f6910 | 2008-08-10 01:53:14 +0000 | [diff] [blame] | 581 | void StmtPrinter::VisitPredefinedExpr(PredefinedExpr *Node) { |
Anders Carlsson | 2274266 | 2007-07-21 05:21:51 +0000 | [diff] [blame] | 582 | switch (Node->getIdentType()) { |
| 583 | default: |
| 584 | assert(0 && "unknown case"); |
Chris Lattner | d9f6910 | 2008-08-10 01:53:14 +0000 | [diff] [blame] | 585 | case PredefinedExpr::Func: |
Anders Carlsson | 2274266 | 2007-07-21 05:21:51 +0000 | [diff] [blame] | 586 | OS << "__func__"; |
| 587 | break; |
Chris Lattner | d9f6910 | 2008-08-10 01:53:14 +0000 | [diff] [blame] | 588 | case PredefinedExpr::Function: |
Anders Carlsson | 2274266 | 2007-07-21 05:21:51 +0000 | [diff] [blame] | 589 | OS << "__FUNCTION__"; |
| 590 | break; |
Chris Lattner | d9f6910 | 2008-08-10 01:53:14 +0000 | [diff] [blame] | 591 | case PredefinedExpr::PrettyFunction: |
Anders Carlsson | 2274266 | 2007-07-21 05:21:51 +0000 | [diff] [blame] | 592 | OS << "__PRETTY_FUNCTION__"; |
| 593 | break; |
| 594 | } |
| 595 | } |
| 596 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 597 | void StmtPrinter::VisitCharacterLiteral(CharacterLiteral *Node) { |
Chris Lattner | b0a721a | 2007-07-13 05:18:11 +0000 | [diff] [blame] | 598 | unsigned value = Node->getValue(); |
Chris Lattner | c250aae | 2008-06-07 22:35:38 +0000 | [diff] [blame] | 599 | if (Node->isWide()) |
| 600 | OS << "L"; |
Chris Lattner | 8bf9f07 | 2007-07-13 23:58:20 +0000 | [diff] [blame] | 601 | switch (value) { |
| 602 | case '\\': |
| 603 | OS << "'\\\\'"; |
| 604 | break; |
| 605 | case '\'': |
| 606 | OS << "'\\''"; |
| 607 | break; |
| 608 | case '\a': |
| 609 | // TODO: K&R: the meaning of '\\a' is different in traditional C |
| 610 | OS << "'\\a'"; |
| 611 | break; |
| 612 | case '\b': |
| 613 | OS << "'\\b'"; |
| 614 | break; |
| 615 | // Nonstandard escape sequence. |
| 616 | /*case '\e': |
| 617 | OS << "'\\e'"; |
| 618 | break;*/ |
| 619 | case '\f': |
| 620 | OS << "'\\f'"; |
| 621 | break; |
| 622 | case '\n': |
| 623 | OS << "'\\n'"; |
| 624 | break; |
| 625 | case '\r': |
| 626 | OS << "'\\r'"; |
| 627 | break; |
| 628 | case '\t': |
| 629 | OS << "'\\t'"; |
| 630 | break; |
| 631 | case '\v': |
| 632 | OS << "'\\v'"; |
| 633 | break; |
| 634 | default: |
Ted Kremenek | 471733d | 2008-02-23 00:52:04 +0000 | [diff] [blame] | 635 | if (value < 256 && isprint(value)) { |
Chris Lattner | 8bf9f07 | 2007-07-13 23:58:20 +0000 | [diff] [blame] | 636 | OS << "'" << (char)value << "'"; |
| 637 | } else if (value < 256) { |
Ted Kremenek | a95d375 | 2008-09-13 05:16:45 +0000 | [diff] [blame] | 638 | OS << "'\\x" << llvm::format("%x", value) << "'"; |
Chris Lattner | 8bf9f07 | 2007-07-13 23:58:20 +0000 | [diff] [blame] | 639 | } else { |
| 640 | // FIXME what to really do here? |
| 641 | OS << value; |
| 642 | } |
Chris Lattner | b0a721a | 2007-07-13 05:18:11 +0000 | [diff] [blame] | 643 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 644 | } |
| 645 | |
| 646 | void StmtPrinter::VisitIntegerLiteral(IntegerLiteral *Node) { |
| 647 | bool isSigned = Node->getType()->isSignedIntegerType(); |
| 648 | OS << Node->getValue().toString(10, isSigned); |
| 649 | |
| 650 | // Emit suffixes. Integer literals are always a builtin integer type. |
Chris Lattner | b77792e | 2008-07-26 22:17:49 +0000 | [diff] [blame] | 651 | switch (Node->getType()->getAsBuiltinType()->getKind()) { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 652 | default: assert(0 && "Unexpected type for integer literal!"); |
| 653 | case BuiltinType::Int: break; // no suffix. |
| 654 | case BuiltinType::UInt: OS << 'U'; break; |
| 655 | case BuiltinType::Long: OS << 'L'; break; |
| 656 | case BuiltinType::ULong: OS << "UL"; break; |
| 657 | case BuiltinType::LongLong: OS << "LL"; break; |
| 658 | case BuiltinType::ULongLong: OS << "ULL"; break; |
| 659 | } |
| 660 | } |
| 661 | void StmtPrinter::VisitFloatingLiteral(FloatingLiteral *Node) { |
Chris Lattner | 86e499d | 2007-08-01 00:23:58 +0000 | [diff] [blame] | 662 | // FIXME: print value more precisely. |
Chris Lattner | da8249e | 2008-06-07 22:13:43 +0000 | [diff] [blame] | 663 | OS << Node->getValueAsApproximateDouble(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 664 | } |
Chris Lattner | 5d66145 | 2007-08-26 03:42:43 +0000 | [diff] [blame] | 665 | |
| 666 | void StmtPrinter::VisitImaginaryLiteral(ImaginaryLiteral *Node) { |
| 667 | PrintExpr(Node->getSubExpr()); |
| 668 | OS << "i"; |
| 669 | } |
| 670 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 671 | void StmtPrinter::VisitStringLiteral(StringLiteral *Str) { |
| 672 | if (Str->isWide()) OS << 'L'; |
| 673 | OS << '"'; |
Anders Carlsson | ee98ac5 | 2007-10-15 02:50:23 +0000 | [diff] [blame] | 674 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 675 | // FIXME: this doesn't print wstrings right. |
| 676 | for (unsigned i = 0, e = Str->getByteLength(); i != e; ++i) { |
Chris Lattner | 9a81c87 | 2009-01-16 19:25:18 +0000 | [diff] [blame] | 677 | unsigned char Char = Str->getStrData()[i]; |
| 678 | |
| 679 | switch (Char) { |
| 680 | default: |
| 681 | if (isprint(Char)) |
| 682 | OS << (char)Char; |
| 683 | else // Output anything hard as an octal escape. |
| 684 | OS << '\\' |
| 685 | << (char)('0'+ ((Char >> 6) & 7)) |
| 686 | << (char)('0'+ ((Char >> 3) & 7)) |
| 687 | << (char)('0'+ ((Char >> 0) & 7)); |
| 688 | break; |
| 689 | // Handle some common non-printable cases to make dumps prettier. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 690 | case '\\': OS << "\\\\"; break; |
| 691 | case '"': OS << "\\\""; break; |
| 692 | case '\n': OS << "\\n"; break; |
| 693 | case '\t': OS << "\\t"; break; |
| 694 | case '\a': OS << "\\a"; break; |
| 695 | case '\b': OS << "\\b"; break; |
| 696 | } |
| 697 | } |
| 698 | OS << '"'; |
| 699 | } |
| 700 | void StmtPrinter::VisitParenExpr(ParenExpr *Node) { |
| 701 | OS << "("; |
| 702 | PrintExpr(Node->getSubExpr()); |
| 703 | OS << ")"; |
| 704 | } |
| 705 | void StmtPrinter::VisitUnaryOperator(UnaryOperator *Node) { |
Chris Lattner | 296bf19 | 2007-08-23 21:46:40 +0000 | [diff] [blame] | 706 | if (!Node->isPostfix()) { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 707 | OS << UnaryOperator::getOpcodeStr(Node->getOpcode()); |
Chris Lattner | 296bf19 | 2007-08-23 21:46:40 +0000 | [diff] [blame] | 708 | |
Sebastian Redl | 0518999 | 2008-11-11 17:56:53 +0000 | [diff] [blame] | 709 | // Print a space if this is an "identifier operator" like __real. |
Chris Lattner | 296bf19 | 2007-08-23 21:46:40 +0000 | [diff] [blame] | 710 | switch (Node->getOpcode()) { |
| 711 | default: break; |
Chris Lattner | 296bf19 | 2007-08-23 21:46:40 +0000 | [diff] [blame] | 712 | case UnaryOperator::Real: |
| 713 | case UnaryOperator::Imag: |
| 714 | case UnaryOperator::Extension: |
| 715 | OS << ' '; |
| 716 | break; |
| 717 | } |
| 718 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 719 | PrintExpr(Node->getSubExpr()); |
| 720 | |
| 721 | if (Node->isPostfix()) |
| 722 | OS << UnaryOperator::getOpcodeStr(Node->getOpcode()); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 723 | } |
Chris Lattner | 704fe35 | 2007-08-30 17:59:59 +0000 | [diff] [blame] | 724 | |
| 725 | bool StmtPrinter::PrintOffsetOfDesignator(Expr *E) { |
| 726 | if (isa<CompoundLiteralExpr>(E)) { |
| 727 | // Base case, print the type and comma. |
| 728 | OS << E->getType().getAsString() << ", "; |
| 729 | return true; |
| 730 | } else if (ArraySubscriptExpr *ASE = dyn_cast<ArraySubscriptExpr>(E)) { |
| 731 | PrintOffsetOfDesignator(ASE->getLHS()); |
| 732 | OS << "["; |
| 733 | PrintExpr(ASE->getRHS()); |
| 734 | OS << "]"; |
| 735 | return false; |
| 736 | } else { |
| 737 | MemberExpr *ME = cast<MemberExpr>(E); |
| 738 | bool IsFirst = PrintOffsetOfDesignator(ME->getBase()); |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 739 | OS << (IsFirst ? "" : ".") << ME->getMemberDecl()->getNameAsString(); |
Chris Lattner | 704fe35 | 2007-08-30 17:59:59 +0000 | [diff] [blame] | 740 | return false; |
| 741 | } |
| 742 | } |
| 743 | |
| 744 | void StmtPrinter::VisitUnaryOffsetOf(UnaryOperator *Node) { |
| 745 | OS << "__builtin_offsetof("; |
| 746 | PrintOffsetOfDesignator(Node->getSubExpr()); |
| 747 | OS << ")"; |
| 748 | } |
| 749 | |
Sebastian Redl | 0518999 | 2008-11-11 17:56:53 +0000 | [diff] [blame] | 750 | void StmtPrinter::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *Node) { |
| 751 | OS << (Node->isSizeOf() ? "sizeof" : "__alignof"); |
| 752 | if (Node->isArgumentType()) |
| 753 | OS << "(" << Node->getArgumentType().getAsString() << ")"; |
| 754 | else { |
| 755 | OS << " "; |
| 756 | PrintExpr(Node->getArgumentExpr()); |
| 757 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 758 | } |
| 759 | void StmtPrinter::VisitArraySubscriptExpr(ArraySubscriptExpr *Node) { |
Ted Kremenek | 2324512 | 2007-08-20 16:18:38 +0000 | [diff] [blame] | 760 | PrintExpr(Node->getLHS()); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 761 | OS << "["; |
Ted Kremenek | 2324512 | 2007-08-20 16:18:38 +0000 | [diff] [blame] | 762 | PrintExpr(Node->getRHS()); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 763 | OS << "]"; |
| 764 | } |
| 765 | |
| 766 | void StmtPrinter::VisitCallExpr(CallExpr *Call) { |
| 767 | PrintExpr(Call->getCallee()); |
| 768 | OS << "("; |
| 769 | for (unsigned i = 0, e = Call->getNumArgs(); i != e; ++i) { |
Chris Lattner | 0442108 | 2008-04-08 04:40:51 +0000 | [diff] [blame] | 770 | if (isa<CXXDefaultArgExpr>(Call->getArg(i))) { |
| 771 | // Don't print any defaulted arguments |
| 772 | break; |
| 773 | } |
| 774 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 775 | if (i) OS << ", "; |
| 776 | PrintExpr(Call->getArg(i)); |
| 777 | } |
| 778 | OS << ")"; |
| 779 | } |
| 780 | void StmtPrinter::VisitMemberExpr(MemberExpr *Node) { |
Douglas Gregor | b3eef68 | 2009-01-08 22:45:41 +0000 | [diff] [blame] | 781 | // FIXME: Suppress printing implicit bases (like "this") |
| 782 | PrintExpr(Node->getBase()); |
| 783 | OS << (Node->isArrow() ? "->" : "."); |
| 784 | // FIXME: Suppress printing references to unnamed objects |
| 785 | // representing anonymous unions/structs |
Douglas Gregor | 86f1940 | 2008-12-20 23:49:58 +0000 | [diff] [blame] | 786 | OS << Node->getMemberDecl()->getNameAsString(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 787 | } |
Nate Begeman | 213541a | 2008-04-18 23:10:10 +0000 | [diff] [blame] | 788 | void StmtPrinter::VisitExtVectorElementExpr(ExtVectorElementExpr *Node) { |
Steve Naroff | 31a4584 | 2007-07-28 23:10:27 +0000 | [diff] [blame] | 789 | PrintExpr(Node->getBase()); |
| 790 | OS << "."; |
| 791 | OS << Node->getAccessor().getName(); |
| 792 | } |
Argyrios Kyrtzidis | 0835a3c | 2008-08-18 23:01:59 +0000 | [diff] [blame] | 793 | void StmtPrinter::VisitCastExpr(CastExpr *) { |
| 794 | assert(0 && "CastExpr is an abstract class"); |
| 795 | } |
Douglas Gregor | 49badde | 2008-10-27 19:41:14 +0000 | [diff] [blame] | 796 | void StmtPrinter::VisitExplicitCastExpr(ExplicitCastExpr *) { |
| 797 | assert(0 && "ExplicitCastExpr is an abstract class"); |
| 798 | } |
Douglas Gregor | 6eec8e8 | 2008-10-28 15:36:24 +0000 | [diff] [blame] | 799 | void StmtPrinter::VisitCStyleCastExpr(CStyleCastExpr *Node) { |
Chris Lattner | 26dc7b3 | 2007-07-15 23:54:50 +0000 | [diff] [blame] | 800 | OS << "(" << Node->getType().getAsString() << ")"; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 801 | PrintExpr(Node->getSubExpr()); |
| 802 | } |
Steve Naroff | aff1edd | 2007-07-19 21:32:11 +0000 | [diff] [blame] | 803 | void StmtPrinter::VisitCompoundLiteralExpr(CompoundLiteralExpr *Node) { |
| 804 | OS << "(" << Node->getType().getAsString() << ")"; |
| 805 | PrintExpr(Node->getInitializer()); |
| 806 | } |
Steve Naroff | 49b4526 | 2007-07-13 16:58:59 +0000 | [diff] [blame] | 807 | void StmtPrinter::VisitImplicitCastExpr(ImplicitCastExpr *Node) { |
Steve Naroff | 90045e8 | 2007-07-13 23:32:42 +0000 | [diff] [blame] | 808 | // No need to print anything, simply forward to the sub expression. |
| 809 | PrintExpr(Node->getSubExpr()); |
Steve Naroff | 49b4526 | 2007-07-13 16:58:59 +0000 | [diff] [blame] | 810 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 811 | void StmtPrinter::VisitBinaryOperator(BinaryOperator *Node) { |
| 812 | PrintExpr(Node->getLHS()); |
| 813 | OS << " " << BinaryOperator::getOpcodeStr(Node->getOpcode()) << " "; |
| 814 | PrintExpr(Node->getRHS()); |
| 815 | } |
Chris Lattner | eb14fe8 | 2007-08-25 02:00:02 +0000 | [diff] [blame] | 816 | void StmtPrinter::VisitCompoundAssignOperator(CompoundAssignOperator *Node) { |
| 817 | PrintExpr(Node->getLHS()); |
| 818 | OS << " " << BinaryOperator::getOpcodeStr(Node->getOpcode()) << " "; |
| 819 | PrintExpr(Node->getRHS()); |
| 820 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 821 | void StmtPrinter::VisitConditionalOperator(ConditionalOperator *Node) { |
| 822 | PrintExpr(Node->getCond()); |
Ted Kremenek | 8e911c4 | 2007-11-26 18:27:54 +0000 | [diff] [blame] | 823 | |
| 824 | if (Node->getLHS()) { |
| 825 | OS << " ? "; |
| 826 | PrintExpr(Node->getLHS()); |
| 827 | OS << " : "; |
| 828 | } |
| 829 | else { // Handle GCC extention where LHS can be NULL. |
| 830 | OS << " ?: "; |
| 831 | } |
| 832 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 833 | PrintExpr(Node->getRHS()); |
| 834 | } |
| 835 | |
| 836 | // GNU extensions. |
| 837 | |
Chris Lattner | 6481a57 | 2007-08-03 17:31:20 +0000 | [diff] [blame] | 838 | void StmtPrinter::VisitAddrLabelExpr(AddrLabelExpr *Node) { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 839 | OS << "&&" << Node->getLabel()->getName(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 840 | } |
| 841 | |
Chris Lattner | ab18c4c | 2007-07-24 16:58:17 +0000 | [diff] [blame] | 842 | void StmtPrinter::VisitStmtExpr(StmtExpr *E) { |
| 843 | OS << "("; |
| 844 | PrintRawCompoundStmt(E->getSubStmt()); |
| 845 | OS << ")"; |
| 846 | } |
| 847 | |
Steve Naroff | d34e915 | 2007-08-01 22:05:33 +0000 | [diff] [blame] | 848 | void StmtPrinter::VisitTypesCompatibleExpr(TypesCompatibleExpr *Node) { |
| 849 | OS << "__builtin_types_compatible_p("; |
| 850 | OS << Node->getArgType1().getAsString() << ","; |
| 851 | OS << Node->getArgType2().getAsString() << ")"; |
| 852 | } |
| 853 | |
Steve Naroff | d04fdd5 | 2007-08-03 21:21:27 +0000 | [diff] [blame] | 854 | void StmtPrinter::VisitChooseExpr(ChooseExpr *Node) { |
| 855 | OS << "__builtin_choose_expr("; |
| 856 | PrintExpr(Node->getCond()); |
Chris Lattner | 94f05e3 | 2007-08-04 00:20:15 +0000 | [diff] [blame] | 857 | OS << ", "; |
Steve Naroff | d04fdd5 | 2007-08-03 21:21:27 +0000 | [diff] [blame] | 858 | PrintExpr(Node->getLHS()); |
Chris Lattner | 94f05e3 | 2007-08-04 00:20:15 +0000 | [diff] [blame] | 859 | OS << ", "; |
Steve Naroff | d04fdd5 | 2007-08-03 21:21:27 +0000 | [diff] [blame] | 860 | PrintExpr(Node->getRHS()); |
| 861 | OS << ")"; |
| 862 | } |
Chris Lattner | ab18c4c | 2007-07-24 16:58:17 +0000 | [diff] [blame] | 863 | |
Douglas Gregor | 2d8b273 | 2008-11-29 04:51:27 +0000 | [diff] [blame] | 864 | void StmtPrinter::VisitGNUNullExpr(GNUNullExpr *) { |
| 865 | OS << "__null"; |
| 866 | } |
| 867 | |
Nate Begeman | e2ce1d9 | 2008-01-17 17:46:27 +0000 | [diff] [blame] | 868 | void StmtPrinter::VisitOverloadExpr(OverloadExpr *Node) { |
| 869 | OS << "__builtin_overload("; |
Nate Begeman | 67295d0 | 2008-01-30 20:50:20 +0000 | [diff] [blame] | 870 | for (unsigned i = 0, e = Node->getNumSubExprs(); i != e; ++i) { |
Nate Begeman | e2ce1d9 | 2008-01-17 17:46:27 +0000 | [diff] [blame] | 871 | if (i) OS << ", "; |
Nate Begeman | 67295d0 | 2008-01-30 20:50:20 +0000 | [diff] [blame] | 872 | PrintExpr(Node->getExpr(i)); |
Nate Begeman | e2ce1d9 | 2008-01-17 17:46:27 +0000 | [diff] [blame] | 873 | } |
| 874 | OS << ")"; |
| 875 | } |
| 876 | |
Eli Friedman | d38617c | 2008-05-14 19:38:39 +0000 | [diff] [blame] | 877 | void StmtPrinter::VisitShuffleVectorExpr(ShuffleVectorExpr *Node) { |
| 878 | OS << "__builtin_shufflevector("; |
| 879 | for (unsigned i = 0, e = Node->getNumSubExprs(); i != e; ++i) { |
| 880 | if (i) OS << ", "; |
| 881 | PrintExpr(Node->getExpr(i)); |
| 882 | } |
| 883 | OS << ")"; |
| 884 | } |
| 885 | |
Anders Carlsson | 66b5a8a | 2007-08-31 04:56:16 +0000 | [diff] [blame] | 886 | void StmtPrinter::VisitInitListExpr(InitListExpr* Node) { |
| 887 | OS << "{ "; |
| 888 | for (unsigned i = 0, e = Node->getNumInits(); i != e; ++i) { |
| 889 | if (i) OS << ", "; |
Douglas Gregor | 4c67834 | 2009-01-28 21:54:33 +0000 | [diff] [blame] | 890 | if (Node->getInit(i)) |
| 891 | PrintExpr(Node->getInit(i)); |
| 892 | else |
| 893 | OS << "0"; |
Anders Carlsson | 66b5a8a | 2007-08-31 04:56:16 +0000 | [diff] [blame] | 894 | } |
| 895 | OS << " }"; |
| 896 | } |
| 897 | |
Douglas Gregor | 05c13a3 | 2009-01-22 00:58:24 +0000 | [diff] [blame] | 898 | void StmtPrinter::VisitDesignatedInitExpr(DesignatedInitExpr *Node) { |
Douglas Gregor | 4c67834 | 2009-01-28 21:54:33 +0000 | [diff] [blame] | 899 | for (DesignatedInitExpr::designators_iterator D = Node->designators_begin(), |
| 900 | DEnd = Node->designators_end(); |
| 901 | D != DEnd; ++D) { |
| 902 | if (D->isFieldDesignator()) { |
| 903 | if (D->getDotLoc().isInvalid()) |
| 904 | OS << D->getFieldName()->getName() << ":"; |
| 905 | else |
| 906 | OS << "." << D->getFieldName()->getName(); |
| 907 | } else { |
| 908 | OS << "["; |
| 909 | if (D->isArrayDesignator()) { |
| 910 | PrintExpr(Node->getArrayIndex(*D)); |
| 911 | } else { |
| 912 | PrintExpr(Node->getArrayRangeStart(*D)); |
| 913 | OS << " ... "; |
| 914 | PrintExpr(Node->getArrayRangeEnd(*D)); |
| 915 | } |
| 916 | OS << "]"; |
| 917 | } |
| 918 | } |
| 919 | |
| 920 | OS << " = "; |
| 921 | PrintExpr(Node->getInit()); |
Douglas Gregor | 05c13a3 | 2009-01-22 00:58:24 +0000 | [diff] [blame] | 922 | } |
| 923 | |
Douglas Gregor | 3498bdb | 2009-01-29 17:44:32 +0000 | [diff] [blame] | 924 | void StmtPrinter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *Node) { |
| 925 | OS << "/*implicit*/" << Node->getType().getAsString() << "()"; |
| 926 | } |
| 927 | |
Anders Carlsson | 7c50aca | 2007-10-15 20:28:48 +0000 | [diff] [blame] | 928 | void StmtPrinter::VisitVAArgExpr(VAArgExpr *Node) { |
| 929 | OS << "va_arg("; |
| 930 | PrintExpr(Node->getSubExpr()); |
| 931 | OS << ", "; |
| 932 | OS << Node->getType().getAsString(); |
| 933 | OS << ")"; |
| 934 | } |
| 935 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 936 | // C++ |
Douglas Gregor | b460980 | 2008-11-14 16:09:21 +0000 | [diff] [blame] | 937 | void StmtPrinter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *Node) { |
| 938 | const char *OpStrings[NUM_OVERLOADED_OPERATORS] = { |
| 939 | "", |
| 940 | #define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \ |
| 941 | Spelling, |
| 942 | #include "clang/Basic/OperatorKinds.def" |
| 943 | }; |
| 944 | |
| 945 | OverloadedOperatorKind Kind = Node->getOperator(); |
| 946 | if (Kind == OO_PlusPlus || Kind == OO_MinusMinus) { |
| 947 | if (Node->getNumArgs() == 1) { |
| 948 | OS << OpStrings[Kind] << ' '; |
| 949 | PrintExpr(Node->getArg(0)); |
| 950 | } else { |
| 951 | PrintExpr(Node->getArg(0)); |
| 952 | OS << ' ' << OpStrings[Kind]; |
| 953 | } |
| 954 | } else if (Kind == OO_Call) { |
| 955 | PrintExpr(Node->getArg(0)); |
| 956 | OS << '('; |
| 957 | for (unsigned ArgIdx = 1; ArgIdx < Node->getNumArgs(); ++ArgIdx) { |
| 958 | if (ArgIdx > 1) |
| 959 | OS << ", "; |
| 960 | if (!isa<CXXDefaultArgExpr>(Node->getArg(ArgIdx))) |
| 961 | PrintExpr(Node->getArg(ArgIdx)); |
| 962 | } |
| 963 | OS << ')'; |
| 964 | } else if (Kind == OO_Subscript) { |
| 965 | PrintExpr(Node->getArg(0)); |
| 966 | OS << '['; |
| 967 | PrintExpr(Node->getArg(1)); |
| 968 | OS << ']'; |
| 969 | } else if (Node->getNumArgs() == 1) { |
| 970 | OS << OpStrings[Kind] << ' '; |
| 971 | PrintExpr(Node->getArg(0)); |
| 972 | } else if (Node->getNumArgs() == 2) { |
| 973 | PrintExpr(Node->getArg(0)); |
| 974 | OS << ' ' << OpStrings[Kind] << ' '; |
| 975 | PrintExpr(Node->getArg(1)); |
| 976 | } else { |
| 977 | assert(false && "unknown overloaded operator"); |
| 978 | } |
| 979 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 980 | |
Douglas Gregor | 88a3514 | 2008-12-22 05:46:06 +0000 | [diff] [blame] | 981 | void StmtPrinter::VisitCXXMemberCallExpr(CXXMemberCallExpr *Node) { |
| 982 | VisitCallExpr(cast<CallExpr>(Node)); |
| 983 | } |
| 984 | |
Douglas Gregor | 49badde | 2008-10-27 19:41:14 +0000 | [diff] [blame] | 985 | void StmtPrinter::VisitCXXNamedCastExpr(CXXNamedCastExpr *Node) { |
| 986 | OS << Node->getCastName() << '<'; |
| 987 | OS << Node->getTypeAsWritten().getAsString() << ">("; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 988 | PrintExpr(Node->getSubExpr()); |
| 989 | OS << ")"; |
| 990 | } |
| 991 | |
Douglas Gregor | 49badde | 2008-10-27 19:41:14 +0000 | [diff] [blame] | 992 | void StmtPrinter::VisitCXXStaticCastExpr(CXXStaticCastExpr *Node) { |
| 993 | VisitCXXNamedCastExpr(Node); |
| 994 | } |
| 995 | |
| 996 | void StmtPrinter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *Node) { |
| 997 | VisitCXXNamedCastExpr(Node); |
| 998 | } |
| 999 | |
| 1000 | void StmtPrinter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *Node) { |
| 1001 | VisitCXXNamedCastExpr(Node); |
| 1002 | } |
| 1003 | |
| 1004 | void StmtPrinter::VisitCXXConstCastExpr(CXXConstCastExpr *Node) { |
| 1005 | VisitCXXNamedCastExpr(Node); |
| 1006 | } |
| 1007 | |
Sebastian Redl | c42e118 | 2008-11-11 11:37:55 +0000 | [diff] [blame] | 1008 | void StmtPrinter::VisitCXXTypeidExpr(CXXTypeidExpr *Node) { |
| 1009 | OS << "typeid("; |
| 1010 | if (Node->isTypeOperand()) { |
| 1011 | OS << Node->getTypeOperand().getAsString(); |
| 1012 | } else { |
| 1013 | PrintExpr(Node->getExprOperand()); |
| 1014 | } |
| 1015 | OS << ")"; |
| 1016 | } |
| 1017 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1018 | void StmtPrinter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *Node) { |
| 1019 | OS << (Node->getValue() ? "true" : "false"); |
| 1020 | } |
| 1021 | |
Douglas Gregor | 796da18 | 2008-11-04 14:32:21 +0000 | [diff] [blame] | 1022 | void StmtPrinter::VisitCXXThisExpr(CXXThisExpr *Node) { |
| 1023 | OS << "this"; |
| 1024 | } |
| 1025 | |
Chris Lattner | 50dd289 | 2008-02-26 00:51:44 +0000 | [diff] [blame] | 1026 | void StmtPrinter::VisitCXXThrowExpr(CXXThrowExpr *Node) { |
| 1027 | if (Node->getSubExpr() == 0) |
| 1028 | OS << "throw"; |
| 1029 | else { |
| 1030 | OS << "throw "; |
| 1031 | PrintExpr(Node->getSubExpr()); |
| 1032 | } |
| 1033 | } |
| 1034 | |
Chris Lattner | 0442108 | 2008-04-08 04:40:51 +0000 | [diff] [blame] | 1035 | void StmtPrinter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *Node) { |
| 1036 | // Nothing to print: we picked up the default argument |
| 1037 | } |
| 1038 | |
Argyrios Kyrtzidis | 987a14b | 2008-08-22 15:38:55 +0000 | [diff] [blame] | 1039 | void StmtPrinter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *Node) { |
| 1040 | OS << Node->getType().getAsString(); |
| 1041 | OS << "("; |
| 1042 | PrintExpr(Node->getSubExpr()); |
| 1043 | OS << ")"; |
| 1044 | } |
| 1045 | |
Douglas Gregor | 506ae41 | 2009-01-16 18:33:17 +0000 | [diff] [blame] | 1046 | void StmtPrinter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *Node) { |
| 1047 | OS << Node->getType().getAsString(); |
| 1048 | OS << "("; |
| 1049 | for (CXXTemporaryObjectExpr::arg_iterator Arg = Node->arg_begin(), |
| 1050 | ArgEnd = Node->arg_end(); |
| 1051 | Arg != ArgEnd; ++Arg) { |
| 1052 | if (Arg != Node->arg_begin()) |
| 1053 | OS << ", "; |
| 1054 | PrintExpr(*Arg); |
| 1055 | } |
| 1056 | OS << ")"; |
| 1057 | } |
| 1058 | |
Argyrios Kyrtzidis | 987a14b | 2008-08-22 15:38:55 +0000 | [diff] [blame] | 1059 | void StmtPrinter::VisitCXXZeroInitValueExpr(CXXZeroInitValueExpr *Node) { |
| 1060 | OS << Node->getType().getAsString() << "()"; |
| 1061 | } |
| 1062 | |
Argyrios Kyrtzidis | 9e922b1 | 2008-09-09 23:47:53 +0000 | [diff] [blame] | 1063 | void |
| 1064 | StmtPrinter::VisitCXXConditionDeclExpr(CXXConditionDeclExpr *E) { |
| 1065 | PrintRawDecl(E->getVarDecl()); |
| 1066 | } |
| 1067 | |
Sebastian Redl | 4c5d320 | 2008-11-21 19:14:01 +0000 | [diff] [blame] | 1068 | void StmtPrinter::VisitCXXNewExpr(CXXNewExpr *E) { |
| 1069 | if (E->isGlobalNew()) |
| 1070 | OS << "::"; |
| 1071 | OS << "new "; |
| 1072 | unsigned NumPlace = E->getNumPlacementArgs(); |
| 1073 | if (NumPlace > 0) { |
| 1074 | OS << "("; |
| 1075 | PrintExpr(E->getPlacementArg(0)); |
| 1076 | for (unsigned i = 1; i < NumPlace; ++i) { |
| 1077 | OS << ", "; |
| 1078 | PrintExpr(E->getPlacementArg(i)); |
| 1079 | } |
| 1080 | OS << ") "; |
| 1081 | } |
| 1082 | if (E->isParenTypeId()) |
| 1083 | OS << "("; |
Sebastian Redl | 6fec648 | 2008-12-02 22:08:59 +0000 | [diff] [blame] | 1084 | std::string TypeS; |
| 1085 | if (Expr *Size = E->getArraySize()) { |
| 1086 | llvm::raw_string_ostream s(TypeS); |
| 1087 | Size->printPretty(s); |
| 1088 | s.flush(); |
| 1089 | TypeS = "[" + TypeS + "]"; |
| 1090 | } |
| 1091 | E->getAllocatedType().getAsStringInternal(TypeS); |
| 1092 | OS << TypeS; |
Sebastian Redl | 4c5d320 | 2008-11-21 19:14:01 +0000 | [diff] [blame] | 1093 | if (E->isParenTypeId()) |
| 1094 | OS << ")"; |
| 1095 | |
| 1096 | if (E->hasInitializer()) { |
| 1097 | OS << "("; |
| 1098 | unsigned NumCons = E->getNumConstructorArgs(); |
| 1099 | if (NumCons > 0) { |
| 1100 | PrintExpr(E->getConstructorArg(0)); |
| 1101 | for (unsigned i = 1; i < NumCons; ++i) { |
| 1102 | OS << ", "; |
| 1103 | PrintExpr(E->getConstructorArg(i)); |
| 1104 | } |
| 1105 | } |
| 1106 | OS << ")"; |
| 1107 | } |
| 1108 | } |
| 1109 | |
| 1110 | void StmtPrinter::VisitCXXDeleteExpr(CXXDeleteExpr *E) { |
| 1111 | if (E->isGlobalDelete()) |
| 1112 | OS << "::"; |
| 1113 | OS << "delete "; |
| 1114 | if (E->isArrayForm()) |
| 1115 | OS << "[] "; |
| 1116 | PrintExpr(E->getArgument()); |
| 1117 | } |
| 1118 | |
Douglas Gregor | 1733001 | 2009-02-04 15:01:18 +0000 | [diff] [blame] | 1119 | void StmtPrinter::VisitUnresolvedFunctionNameExpr(UnresolvedFunctionNameExpr *E) { |
| 1120 | OS << E->getName().getAsString(); |
Douglas Gregor | 5c37de7 | 2008-12-06 00:22:45 +0000 | [diff] [blame] | 1121 | } |
| 1122 | |
Sebastian Redl | 64b45f7 | 2009-01-05 20:52:13 +0000 | [diff] [blame] | 1123 | static const char *getTypeTraitName(UnaryTypeTrait UTT) { |
| 1124 | switch (UTT) { |
| 1125 | default: assert(false && "Unknown type trait"); |
| 1126 | case UTT_HasNothrowAssign: return "__has_nothrow_assign"; |
| 1127 | case UTT_HasNothrowCopy: return "__has_nothrow_copy"; |
| 1128 | case UTT_HasNothrowConstructor: return "__has_nothrow_constructor"; |
| 1129 | case UTT_HasTrivialAssign: return "__has_trivial_assign"; |
| 1130 | case UTT_HasTrivialCopy: return "__has_trivial_copy"; |
| 1131 | case UTT_HasTrivialConstructor: return "__has_trivial_constructor"; |
| 1132 | case UTT_HasTrivialDestructor: return "__has_trivial_destructor"; |
| 1133 | case UTT_HasVirtualDestructor: return "__has_virtual_destructor"; |
| 1134 | case UTT_IsAbstract: return "__is_abstract"; |
| 1135 | case UTT_IsClass: return "__is_class"; |
| 1136 | case UTT_IsEmpty: return "__is_empty"; |
| 1137 | case UTT_IsEnum: return "__is_enum"; |
| 1138 | case UTT_IsPOD: return "__is_pod"; |
| 1139 | case UTT_IsPolymorphic: return "__is_polymorphic"; |
| 1140 | case UTT_IsUnion: return "__is_union"; |
| 1141 | } |
| 1142 | } |
| 1143 | |
| 1144 | void StmtPrinter::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) { |
| 1145 | OS << getTypeTraitName(E->getTrait()) << "(" |
| 1146 | << E->getQueriedType().getAsString() << ")"; |
| 1147 | } |
| 1148 | |
Anders Carlsson | 5508518 | 2007-08-21 17:43:55 +0000 | [diff] [blame] | 1149 | // Obj-C |
| 1150 | |
| 1151 | void StmtPrinter::VisitObjCStringLiteral(ObjCStringLiteral *Node) { |
| 1152 | OS << "@"; |
| 1153 | VisitStringLiteral(Node->getString()); |
| 1154 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1155 | |
Anders Carlsson | f9bcf01 | 2007-08-22 15:14:15 +0000 | [diff] [blame] | 1156 | void StmtPrinter::VisitObjCEncodeExpr(ObjCEncodeExpr *Node) { |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 1157 | OS << "@encode(" << Node->getEncodedType().getAsString() << ')'; |
Anders Carlsson | f9bcf01 | 2007-08-22 15:14:15 +0000 | [diff] [blame] | 1158 | } |
| 1159 | |
Fariborz Jahanian | b62f681 | 2007-10-16 20:40:23 +0000 | [diff] [blame] | 1160 | void StmtPrinter::VisitObjCSelectorExpr(ObjCSelectorExpr *Node) { |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 1161 | OS << "@selector(" << Node->getSelector().getAsString() << ')'; |
Fariborz Jahanian | b62f681 | 2007-10-16 20:40:23 +0000 | [diff] [blame] | 1162 | } |
| 1163 | |
Fariborz Jahanian | 390d50a | 2007-10-17 16:58:11 +0000 | [diff] [blame] | 1164 | void StmtPrinter::VisitObjCProtocolExpr(ObjCProtocolExpr *Node) { |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 1165 | OS << "@protocol(" << Node->getProtocol()->getNameAsString() << ')'; |
Fariborz Jahanian | 390d50a | 2007-10-17 16:58:11 +0000 | [diff] [blame] | 1166 | } |
| 1167 | |
Steve Naroff | 563477d | 2007-09-18 23:55:05 +0000 | [diff] [blame] | 1168 | void StmtPrinter::VisitObjCMessageExpr(ObjCMessageExpr *Mess) { |
| 1169 | OS << "["; |
Steve Naroff | 6a8a9a4 | 2007-10-02 20:01:56 +0000 | [diff] [blame] | 1170 | Expr *receiver = Mess->getReceiver(); |
| 1171 | if (receiver) PrintExpr(receiver); |
| 1172 | else OS << Mess->getClassName()->getName(); |
Ted Kremenek | c29efd8 | 2008-05-02 17:32:38 +0000 | [diff] [blame] | 1173 | OS << ' '; |
Ted Kremenek | 97b7f26 | 2008-04-16 04:30:16 +0000 | [diff] [blame] | 1174 | Selector selector = Mess->getSelector(); |
Steve Naroff | 6a8a9a4 | 2007-10-02 20:01:56 +0000 | [diff] [blame] | 1175 | if (selector.isUnarySelector()) { |
Ted Kremenek | c29efd8 | 2008-05-02 17:32:38 +0000 | [diff] [blame] | 1176 | OS << selector.getIdentifierInfoForSlot(0)->getName(); |
Steve Naroff | 6a8a9a4 | 2007-10-02 20:01:56 +0000 | [diff] [blame] | 1177 | } else { |
| 1178 | for (unsigned i = 0, e = Mess->getNumArgs(); i != e; ++i) { |
Ted Kremenek | c29efd8 | 2008-05-02 17:32:38 +0000 | [diff] [blame] | 1179 | if (i < selector.getNumArgs()) { |
| 1180 | if (i > 0) OS << ' '; |
| 1181 | if (selector.getIdentifierInfoForSlot(i)) |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 1182 | OS << selector.getIdentifierInfoForSlot(i)->getName() << ':'; |
Ted Kremenek | c29efd8 | 2008-05-02 17:32:38 +0000 | [diff] [blame] | 1183 | else |
| 1184 | OS << ":"; |
| 1185 | } |
| 1186 | else OS << ", "; // Handle variadic methods. |
| 1187 | |
Steve Naroff | 6a8a9a4 | 2007-10-02 20:01:56 +0000 | [diff] [blame] | 1188 | PrintExpr(Mess->getArg(i)); |
| 1189 | } |
Steve Naroff | 563477d | 2007-09-18 23:55:05 +0000 | [diff] [blame] | 1190 | } |
| 1191 | OS << "]"; |
| 1192 | } |
| 1193 | |
Douglas Gregor | cd9b46e | 2008-11-04 14:56:14 +0000 | [diff] [blame] | 1194 | void StmtPrinter::VisitObjCSuperExpr(ObjCSuperExpr *) { |
| 1195 | OS << "super"; |
| 1196 | } |
| 1197 | |
Steve Naroff | 4eb206b | 2008-09-03 18:15:37 +0000 | [diff] [blame] | 1198 | void StmtPrinter::VisitBlockExpr(BlockExpr *Node) { |
Steve Naroff | 56ee689 | 2008-10-08 17:01:13 +0000 | [diff] [blame] | 1199 | BlockDecl *BD = Node->getBlockDecl(); |
Steve Naroff | 4eb206b | 2008-09-03 18:15:37 +0000 | [diff] [blame] | 1200 | OS << "^"; |
| 1201 | |
| 1202 | const FunctionType *AFT = Node->getFunctionType(); |
| 1203 | |
| 1204 | if (isa<FunctionTypeNoProto>(AFT)) { |
| 1205 | OS << "()"; |
Steve Naroff | 56ee689 | 2008-10-08 17:01:13 +0000 | [diff] [blame] | 1206 | } else if (!BD->param_empty() || cast<FunctionTypeProto>(AFT)->isVariadic()) { |
Steve Naroff | 4eb206b | 2008-09-03 18:15:37 +0000 | [diff] [blame] | 1207 | OS << '('; |
| 1208 | std::string ParamStr; |
Steve Naroff | 56ee689 | 2008-10-08 17:01:13 +0000 | [diff] [blame] | 1209 | for (BlockDecl::param_iterator AI = BD->param_begin(), |
| 1210 | E = BD->param_end(); AI != E; ++AI) { |
| 1211 | if (AI != BD->param_begin()) OS << ", "; |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 1212 | ParamStr = (*AI)->getNameAsString(); |
Steve Naroff | 4eb206b | 2008-09-03 18:15:37 +0000 | [diff] [blame] | 1213 | (*AI)->getType().getAsStringInternal(ParamStr); |
| 1214 | OS << ParamStr; |
| 1215 | } |
| 1216 | |
Steve Naroff | 56ee689 | 2008-10-08 17:01:13 +0000 | [diff] [blame] | 1217 | const FunctionTypeProto *FT = cast<FunctionTypeProto>(AFT); |
Steve Naroff | 4eb206b | 2008-09-03 18:15:37 +0000 | [diff] [blame] | 1218 | if (FT->isVariadic()) { |
Steve Naroff | 56ee689 | 2008-10-08 17:01:13 +0000 | [diff] [blame] | 1219 | if (!BD->param_empty()) OS << ", "; |
Steve Naroff | 4eb206b | 2008-09-03 18:15:37 +0000 | [diff] [blame] | 1220 | OS << "..."; |
| 1221 | } |
| 1222 | OS << ')'; |
| 1223 | } |
| 1224 | } |
| 1225 | |
Steve Naroff | 4eb206b | 2008-09-03 18:15:37 +0000 | [diff] [blame] | 1226 | void StmtPrinter::VisitBlockDeclRefExpr(BlockDeclRefExpr *Node) { |
Chris Lattner | 39f34e9 | 2008-11-24 04:00:27 +0000 | [diff] [blame] | 1227 | OS << Node->getDecl()->getNameAsString(); |
Steve Naroff | 4eb206b | 2008-09-03 18:15:37 +0000 | [diff] [blame] | 1228 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1229 | //===----------------------------------------------------------------------===// |
| 1230 | // Stmt method implementations |
| 1231 | //===----------------------------------------------------------------------===// |
| 1232 | |
Chris Lattner | 6000dac | 2007-08-08 22:51:59 +0000 | [diff] [blame] | 1233 | void Stmt::dumpPretty() const { |
Douglas Gregor | 4c67834 | 2009-01-28 21:54:33 +0000 | [diff] [blame] | 1234 | llvm::raw_ostream &OS = llvm::errs(); |
| 1235 | printPretty(OS); |
| 1236 | OS.flush(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1237 | } |
| 1238 | |
Mike Stump | 071e4da | 2009-02-10 20:16:46 +0000 | [diff] [blame^] | 1239 | void Stmt::printPretty(llvm::raw_ostream &OS, PrinterHelper* Helper, |
| 1240 | unsigned I, bool NoIndent) const { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1241 | if (this == 0) { |
| 1242 | OS << "<NULL>"; |
| 1243 | return; |
| 1244 | } |
| 1245 | |
Mike Stump | 071e4da | 2009-02-10 20:16:46 +0000 | [diff] [blame^] | 1246 | StmtPrinter P(OS, Helper, I, NoIndent); |
Chris Lattner | c5598cb | 2007-08-21 04:04:25 +0000 | [diff] [blame] | 1247 | P.Visit(const_cast<Stmt*>(this)); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1248 | } |
Ted Kremenek | 42a509f | 2007-08-31 21:30:12 +0000 | [diff] [blame] | 1249 | |
| 1250 | //===----------------------------------------------------------------------===// |
| 1251 | // PrinterHelper |
| 1252 | //===----------------------------------------------------------------------===// |
| 1253 | |
| 1254 | // Implement virtual destructor. |
Gabor Greif | 8467583 | 2007-09-11 15:32:40 +0000 | [diff] [blame] | 1255 | PrinterHelper::~PrinterHelper() {} |