external/bsd/llvm/dist/clang/lib/AST/StmtPrinter.cpp

   1 //===--- StmtPrinter.cpp - Printing implementation for Stmt ASTs ----------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements the Stmt::dumpPretty/Stmt::printPretty methods, which
  11 // pretty print the AST back out to C code.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "clang/AST/ASTContext.h"
  16 #include "clang/AST/Attr.h"
  17 #include "clang/AST/DeclCXX.h"
  18 #include "clang/AST/DeclObjC.h"
  19 #include "clang/AST/DeclTemplate.h"
  20 #include "clang/AST/Expr.h"
  21 #include "clang/AST/ExprCXX.h"
  22 #include "clang/AST/PrettyPrinter.h"
  23 #include "clang/AST/StmtVisitor.h"
  24 #include "clang/Basic/CharInfo.h"
  25 #include "llvm/ADT/SmallString.h"
  26 #include "llvm/Support/Format.h"
  27 using namespace clang;
  28
  29 //===----------------------------------------------------------------------===//
  30 // StmtPrinter Visitor
  31 //===----------------------------------------------------------------------===//
  32
  33 namespace  {
  34   class StmtPrinter : public StmtVisitor<StmtPrinter> {
  35     raw_ostream &OS;
  36     unsigned IndentLevel;
  37     clang::PrinterHelper* Helper;
  38     PrintingPolicy Policy;
  39
  40   public:
  41     StmtPrinter(raw_ostream &os, PrinterHelper* helper,
  42                 const PrintingPolicy &Policy,
  43                 unsigned Indentation = 0)
  44       : OS(os), IndentLevel(Indentation), Helper(helper), Policy(Policy) {}
  45
  46     void PrintStmt(Stmt *S) {
  47       PrintStmt(S, Policy.Indentation);
  48     }
  49
  50     void PrintStmt(Stmt *S, int SubIndent) {
  51       IndentLevel += SubIndent;
  52       if (S && isa<Expr>(S)) {
  53         // If this is an expr used in a stmt context, indent and newline it.
  54         Indent();
  55         Visit(S);
  56         OS << ";\n";
  57       } else if (S) {
  58         Visit(S);
  59       } else {
  60         Indent() << "<<<NULL STATEMENT>>>\n";
  61       }
  62       IndentLevel -= SubIndent;
  63     }
  64
  65     void PrintRawCompoundStmt(CompoundStmt *S);
  66     void PrintRawDecl(Decl *D);
  67     void PrintRawDeclStmt(const DeclStmt *S);
  68     void PrintRawIfStmt(IfStmt *If);
  69     void PrintRawCXXCatchStmt(CXXCatchStmt *Catch);
  70     void PrintCallArgs(CallExpr *E);
  71     void PrintRawSEHExceptHandler(SEHExceptStmt *S);
  72     void PrintRawSEHFinallyStmt(SEHFinallyStmt *S);
  73     void PrintOMPExecutableDirective(OMPExecutableDirective *S);
  74
  75     void PrintExpr(Expr *E) {
  76       if (E)
  77         Visit(E);
  78       else
  79         OS << "<null expr>";
  80     }
  81
  82     raw_ostream &Indent(int Delta = 0) {
  83       for (int i = 0, e = IndentLevel+Delta; i < e; ++i)
  84         OS << "  ";
  85       return OS;
  86     }
  87
  88     void Visit(Stmt* S) {
  89       if (Helper && Helper->handledStmt(S,OS))
  90           return;
  91       else StmtVisitor<StmtPrinter>::Visit(S);
  92     }
  93
  94     void VisitStmt(Stmt *Node) LLVM_ATTRIBUTE_UNUSED {
  95       Indent() << "<<unknown stmt type>>\n";
  96     }
  97     void VisitExpr(Expr *Node) LLVM_ATTRIBUTE_UNUSED {
  98       OS << "<<unknown expr type>>";
  99     }
 100     void VisitCXXNamedCastExpr(CXXNamedCastExpr *Node);
 101
 102 #define ABSTRACT_STMT(CLASS)
 103 #define STMT(CLASS, PARENT) \
 104     void Visit##CLASS(CLASS *Node);
 105 #include "clang/AST/StmtNodes.inc"
 106   };
 107 }
 108
 109 //===----------------------------------------------------------------------===//
 110 //  Stmt printing methods.
 111 //===----------------------------------------------------------------------===//
 112
 113 /// PrintRawCompoundStmt - Print a compound stmt without indenting the {, and
 114 /// with no newline after the }.
 115 void StmtPrinter::PrintRawCompoundStmt(CompoundStmt *Node) {
 116   OS << "{\n";
 117   for (auto *I : Node->body())
 118     PrintStmt(I);
 119
 120   Indent() << "}";
 121 }
 122
 123 void StmtPrinter::PrintRawDecl(Decl *D) {
 124   D->print(OS, Policy, IndentLevel);
 125 }
 126
 127 void StmtPrinter::PrintRawDeclStmt(const DeclStmt *S) {
 128   SmallVector<Decl*, 2> Decls(S->decls());
 129   Decl::printGroup(Decls.data(), Decls.size(), OS, Policy, IndentLevel);
 130 }
 131
 132 void StmtPrinter::VisitNullStmt(NullStmt *Node) {
 133   Indent() << ";\n";
 134 }
 135
 136 void StmtPrinter::VisitDeclStmt(DeclStmt *Node) {
 137   Indent();
 138   PrintRawDeclStmt(Node);
 139   OS << ";\n";
 140 }
 141
 142 void StmtPrinter::VisitCompoundStmt(CompoundStmt *Node) {
 143   Indent();
 144   PrintRawCompoundStmt(Node);
 145   OS << "\n";
 146 }
 147
 148 void StmtPrinter::VisitCaseStmt(CaseStmt *Node) {
 149   Indent(-1) << "case ";
 150   PrintExpr(Node->getLHS());
 151   if (Node->getRHS()) {
 152     OS << " ... ";
 153     PrintExpr(Node->getRHS());
 154   }
 155   OS << ":\n";
 156
 157   PrintStmt(Node->getSubStmt(), 0);
 158 }
 159
 160 void StmtPrinter::VisitDefaultStmt(DefaultStmt *Node) {
 161   Indent(-1) << "default:\n";
 162   PrintStmt(Node->getSubStmt(), 0);
 163 }
 164
 165 void StmtPrinter::VisitLabelStmt(LabelStmt *Node) {
 166   Indent(-1) << Node->getName() << ":\n";
 167   PrintStmt(Node->getSubStmt(), 0);
 168 }
 169
 170 void StmtPrinter::VisitAttributedStmt(AttributedStmt *Node) {
 171   for (const auto *Attr : Node->getAttrs()) {
 172     Attr->printPretty(OS, Policy);
 173   }
 174
 175   PrintStmt(Node->getSubStmt(), 0);
 176 }
 177
 178 void StmtPrinter::PrintRawIfStmt(IfStmt *If) {
 179   OS << "if (";
 180   if (const DeclStmt *DS = If->getConditionVariableDeclStmt())
 181     PrintRawDeclStmt(DS);
 182   else
 183     PrintExpr(If->getCond());
 184   OS << ')';
 185
 186   if (CompoundStmt *CS = dyn_cast<CompoundStmt>(If->getThen())) {
 187     OS << ' ';
 188     PrintRawCompoundStmt(CS);
 189     OS << (If->getElse() ? ' ' : '\n');
 190   } else {
 191     OS << '\n';
 192     PrintStmt(If->getThen());
 193     if (If->getElse()) Indent();
 194   }
 195
 196   if (Stmt *Else = If->getElse()) {
 197     OS << "else";
 198
 199     if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Else)) {
 200       OS << ' ';
 201       PrintRawCompoundStmt(CS);
 202       OS << '\n';
 203     } else if (IfStmt *ElseIf = dyn_cast<IfStmt>(Else)) {
 204       OS << ' ';
 205       PrintRawIfStmt(ElseIf);
 206     } else {
 207       OS << '\n';
 208       PrintStmt(If->getElse());
 209     }
 210   }
 211 }
 212
 213 void StmtPrinter::VisitIfStmt(IfStmt *If) {
 214   Indent();
 215   PrintRawIfStmt(If);
 216 }
 217
 218 void StmtPrinter::VisitSwitchStmt(SwitchStmt *Node) {
 219   Indent() << "switch (";
 220   if (const DeclStmt *DS = Node->getConditionVariableDeclStmt())
 221     PrintRawDeclStmt(DS);
 222   else
 223     PrintExpr(Node->getCond());
 224   OS << ")";
 225
 226   // Pretty print compoundstmt bodies (very common).
 227   if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
 228     OS << " ";
 229     PrintRawCompoundStmt(CS);
 230     OS << "\n";
 231   } else {
 232     OS << "\n";
 233     PrintStmt(Node->getBody());
 234   }
 235 }
 236
 237 void StmtPrinter::VisitWhileStmt(WhileStmt *Node) {
 238   Indent() << "while (";
 239   if (const DeclStmt *DS = Node->getConditionVariableDeclStmt())
 240     PrintRawDeclStmt(DS);
 241   else
 242     PrintExpr(Node->getCond());
 243   OS << ")\n";
 244   PrintStmt(Node->getBody());
 245 }
 246
 247 void StmtPrinter::VisitDoStmt(DoStmt *Node) {
 248   Indent() << "do ";
 249   if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
 250     PrintRawCompoundStmt(CS);
 251     OS << " ";
 252   } else {
 253     OS << "\n";
 254     PrintStmt(Node->getBody());
 255     Indent();
 256   }
 257
 258   OS << "while (";
 259   PrintExpr(Node->getCond());
 260   OS << ");\n";
 261 }
 262
 263 void StmtPrinter::VisitForStmt(ForStmt *Node) {
 264   Indent() << "for (";
 265   if (Node->getInit()) {
 266     if (DeclStmt *DS = dyn_cast<DeclStmt>(Node->getInit()))
 267       PrintRawDeclStmt(DS);
 268     else
 269       PrintExpr(cast<Expr>(Node->getInit()));
 270   }
 271   OS << ";";
 272   if (Node->getCond()) {
 273     OS << " ";
 274     PrintExpr(Node->getCond());
 275   }
 276   OS << ";";
 277   if (Node->getInc()) {
 278     OS << " ";
 279     PrintExpr(Node->getInc());
 280   }
 281   OS << ") ";
 282
 283   if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
 284     PrintRawCompoundStmt(CS);
 285     OS << "\n";
 286   } else {
 287     OS << "\n";
 288     PrintStmt(Node->getBody());
 289   }
 290 }
 291
 292 void StmtPrinter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *Node) {
 293   Indent() << "for (";
 294   if (DeclStmt *DS = dyn_cast<DeclStmt>(Node->getElement()))
 295     PrintRawDeclStmt(DS);
 296   else
 297     PrintExpr(cast<Expr>(Node->getElement()));
 298   OS << " in ";
 299   PrintExpr(Node->getCollection());
 300   OS << ") ";
 301
 302   if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
 303     PrintRawCompoundStmt(CS);
 304     OS << "\n";
 305   } else {
 306     OS << "\n";
 307     PrintStmt(Node->getBody());
 308   }
 309 }
 310
 311 void StmtPrinter::VisitCXXForRangeStmt(CXXForRangeStmt *Node) {
 312   Indent() << "for (";
 313   PrintingPolicy SubPolicy(Policy);
 314   SubPolicy.SuppressInitializers = true;
 315   Node->getLoopVariable()->print(OS, SubPolicy, IndentLevel);
 316   OS << " : ";
 317   PrintExpr(Node->getRangeInit());
 318   OS << ") {\n";
 319   PrintStmt(Node->getBody());
 320   Indent() << "}";
 321   if (Policy.IncludeNewlines) OS << "\n";
 322 }
 323
 324 void StmtPrinter::VisitMSDependentExistsStmt(MSDependentExistsStmt *Node) {
 325   Indent();
 326   if (Node->isIfExists())
 327     OS << "__if_exists (";
 328   else
 329     OS << "__if_not_exists (";
 330
 331   if (NestedNameSpecifier *Qualifier
 332         = Node->getQualifierLoc().getNestedNameSpecifier())
 333     Qualifier->print(OS, Policy);
 334
 335   OS << Node->getNameInfo() << ") ";
 336
 337   PrintRawCompoundStmt(Node->getSubStmt());
 338 }
 339
 340 void StmtPrinter::VisitGotoStmt(GotoStmt *Node) {
 341   Indent() << "goto " << Node->getLabel()->getName() << ";";
 342   if (Policy.IncludeNewlines) OS << "\n";
 343 }
 344
 345 void StmtPrinter::VisitIndirectGotoStmt(IndirectGotoStmt *Node) {
 346   Indent() << "goto *";
 347   PrintExpr(Node->getTarget());
 348   OS << ";";
 349   if (Policy.IncludeNewlines) OS << "\n";
 350 }
 351
 352 void StmtPrinter::VisitContinueStmt(ContinueStmt *Node) {
 353   Indent() << "continue;";
 354   if (Policy.IncludeNewlines) OS << "\n";
 355 }
 356
 357 void StmtPrinter::VisitBreakStmt(BreakStmt *Node) {
 358   Indent() << "break;";
 359   if (Policy.IncludeNewlines) OS << "\n";
 360 }
 361
 362
 363 void StmtPrinter::VisitReturnStmt(ReturnStmt *Node) {
 364   Indent() << "return";
 365   if (Node->getRetValue()) {
 366     OS << " ";
 367     PrintExpr(Node->getRetValue());
 368   }
 369   OS << ";";
 370   if (Policy.IncludeNewlines) OS << "\n";
 371 }
 372
 373
 374 void StmtPrinter::VisitGCCAsmStmt(GCCAsmStmt *Node) {
 375   Indent() << "asm ";
 376
 377   if (Node->isVolatile())
 378     OS << "volatile ";
 379
 380   OS << "(";
 381   VisitStringLiteral(Node->getAsmString());
 382
 383   // Outputs
 384   if (Node->getNumOutputs() != 0 || Node->getNumInputs() != 0 ||
 385       Node->getNumClobbers() != 0)
 386     OS << " : ";
 387
 388   for (unsigned i = 0, e = Node->getNumOutputs(); i != e; ++i) {
 389     if (i != 0)
 390       OS << ", ";
 391
 392     if (!Node->getOutputName(i).empty()) {
 393       OS << '[';
 394       OS << Node->getOutputName(i);
 395       OS << "] ";
 396     }
 397
 398     VisitStringLiteral(Node->getOutputConstraintLiteral(i));
 399     OS << " ";
 400     Visit(Node->getOutputExpr(i));
 401   }
 402
 403   // Inputs
 404   if (Node->getNumInputs() != 0 || Node->getNumClobbers() != 0)
 405     OS << " : ";
 406
 407   for (unsigned i = 0, e = Node->getNumInputs(); i != e; ++i) {
 408     if (i != 0)
 409       OS << ", ";
 410
 411     if (!Node->getInputName(i).empty()) {
 412       OS << '[';
 413       OS << Node->getInputName(i);
 414       OS << "] ";
 415     }
 416
 417     VisitStringLiteral(Node->getInputConstraintLiteral(i));
 418     OS << " ";
 419     Visit(Node->getInputExpr(i));
 420   }
 421
 422   // Clobbers
 423   if (Node->getNumClobbers() != 0)
 424     OS << " : ";
 425
 426   for (unsigned i = 0, e = Node->getNumClobbers(); i != e; ++i) {
 427     if (i != 0)
 428       OS << ", ";
 429
 430     VisitStringLiteral(Node->getClobberStringLiteral(i));
 431   }
 432
 433   OS << ");";
 434   if (Policy.IncludeNewlines) OS << "\n";
 435 }
 436
 437 void StmtPrinter::VisitMSAsmStmt(MSAsmStmt *Node) {
 438   // FIXME: Implement MS style inline asm statement printer.
 439   Indent() << "__asm ";
 440   if (Node->hasBraces())
 441     OS << "{\n";
 442   OS << Node->getAsmString() << "\n";
 443   if (Node->hasBraces())
 444     Indent() << "}\n";
 445 }
 446
 447 void StmtPrinter::VisitCapturedStmt(CapturedStmt *Node) {
 448   PrintStmt(Node->getCapturedDecl()->getBody());
 449 }
 450
 451 void StmtPrinter::VisitObjCAtTryStmt(ObjCAtTryStmt *Node) {
 452   Indent() << "@try";
 453   if (CompoundStmt *TS = dyn_cast<CompoundStmt>(Node->getTryBody())) {
 454     PrintRawCompoundStmt(TS);
 455     OS << "\n";
 456   }
 457
 458   for (unsigned I = 0, N = Node->getNumCatchStmts(); I != N; ++I) {
 459     ObjCAtCatchStmt *catchStmt = Node->getCatchStmt(I);
 460     Indent() << "@catch(";
 461     if (catchStmt->getCatchParamDecl()) {
 462       if (Decl *DS = catchStmt->getCatchParamDecl())
 463         PrintRawDecl(DS);
 464     }
 465     OS << ")";
 466     if (CompoundStmt *CS = dyn_cast<CompoundStmt>(catchStmt->getCatchBody())) {
 467       PrintRawCompoundStmt(CS);
 468       OS << "\n";
 469     }
 470   }
 471
 472   if (ObjCAtFinallyStmt *FS = static_cast<ObjCAtFinallyStmt *>(
 473         Node->getFinallyStmt())) {
 474     Indent() << "@finally";
 475     PrintRawCompoundStmt(dyn_cast<CompoundStmt>(FS->getFinallyBody()));
 476     OS << "\n";
 477   }
 478 }
 479
 480 void StmtPrinter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *Node) {
 481 }
 482
 483 void StmtPrinter::VisitObjCAtCatchStmt (ObjCAtCatchStmt *Node) {
 484   Indent() << "@catch (...) { /* todo */ } \n";
 485 }
 486
 487 void StmtPrinter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *Node) {
 488   Indent() << "@throw";
 489   if (Node->getThrowExpr()) {
 490     OS << " ";
 491     PrintExpr(Node->getThrowExpr());
 492   }
 493   OS << ";\n";
 494 }
 495
 496 void StmtPrinter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *Node) {
 497   Indent() << "@synchronized (";
 498   PrintExpr(Node->getSynchExpr());
 499   OS << ")";
 500   PrintRawCompoundStmt(Node->getSynchBody());
 501   OS << "\n";
 502 }
 503
 504 void StmtPrinter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *Node) {
 505   Indent() << "@autoreleasepool";
 506   PrintRawCompoundStmt(dyn_cast<CompoundStmt>(Node->getSubStmt()));
 507   OS << "\n";
 508 }
 509
 510 void StmtPrinter::PrintRawCXXCatchStmt(CXXCatchStmt *Node) {
 511   OS << "catch (";
 512   if (Decl *ExDecl = Node->getExceptionDecl())
 513     PrintRawDecl(ExDecl);
 514   else
 515     OS << "...";
 516   OS << ") ";
 517   PrintRawCompoundStmt(cast<CompoundStmt>(Node->getHandlerBlock()));
 518 }
 519
 520 void StmtPrinter::VisitCXXCatchStmt(CXXCatchStmt *Node) {
 521   Indent();
 522   PrintRawCXXCatchStmt(Node);
 523   OS << "\n";
 524 }
 525
 526 void StmtPrinter::VisitCXXTryStmt(CXXTryStmt *Node) {
 527   Indent() << "try ";
 528   PrintRawCompoundStmt(Node->getTryBlock());
 529   for (unsigned i = 0, e = Node->getNumHandlers(); i < e; ++i) {
 530     OS << " ";
 531     PrintRawCXXCatchStmt(Node->getHandler(i));
 532   }
 533   OS << "\n";
 534 }
 535
 536 void StmtPrinter::VisitSEHTryStmt(SEHTryStmt *Node) {
 537   Indent() << (Node->getIsCXXTry() ? "try " : "__try ");
 538   PrintRawCompoundStmt(Node->getTryBlock());
 539   SEHExceptStmt *E = Node->getExceptHandler();
 540   SEHFinallyStmt *F = Node->getFinallyHandler();
 541   if(E)
 542     PrintRawSEHExceptHandler(E);
 543   else {
 544     assert(F && "Must have a finally block...");
 545     PrintRawSEHFinallyStmt(F);
 546   }
 547   OS << "\n";
 548 }
 549
 550 void StmtPrinter::PrintRawSEHFinallyStmt(SEHFinallyStmt *Node) {
 551   OS << "__finally ";
 552   PrintRawCompoundStmt(Node->getBlock());
 553   OS << "\n";
 554 }
 555
 556 void StmtPrinter::PrintRawSEHExceptHandler(SEHExceptStmt *Node) {
 557   OS << "__except (";
 558   VisitExpr(Node->getFilterExpr());
 559   OS << ")\n";
 560   PrintRawCompoundStmt(Node->getBlock());
 561   OS << "\n";
 562 }
 563
 564 void StmtPrinter::VisitSEHExceptStmt(SEHExceptStmt *Node) {
 565   Indent();
 566   PrintRawSEHExceptHandler(Node);
 567   OS << "\n";
 568 }
 569
 570 void StmtPrinter::VisitSEHFinallyStmt(SEHFinallyStmt *Node) {
 571   Indent();
 572   PrintRawSEHFinallyStmt(Node);
 573   OS << "\n";
 574 }
 575
 576 void StmtPrinter::VisitSEHLeaveStmt(SEHLeaveStmt *Node) {
 577   Indent() << "__leave;";
 578   if (Policy.IncludeNewlines) OS << "\n";
 579 }
 580
 581 //===----------------------------------------------------------------------===//
 582 //  OpenMP clauses printing methods
 583 //===----------------------------------------------------------------------===//
 584
 585 namespace {
 586 class OMPClausePrinter : public OMPClauseVisitor<OMPClausePrinter> {
 587   raw_ostream &OS;
 588   const PrintingPolicy &Policy;
 589   /// \brief Process clauses with list of variables.
 590   template <typename T>
 591   void VisitOMPClauseList(T *Node, char StartSym);
 592 public:
 593   OMPClausePrinter(raw_ostream &OS, const PrintingPolicy &Policy)
 594     : OS(OS), Policy(Policy) { }
 595 #define OPENMP_CLAUSE(Name, Class)                              \
 596   void Visit##Class(Class *S);
 597 #include "clang/Basic/OpenMPKinds.def"
 598 };
 599
 600 void OMPClausePrinter::VisitOMPIfClause(OMPIfClause *Node) {
 601   OS << "if(";
 602   Node->getCondition()->printPretty(OS, nullptr, Policy, 0);
 603   OS << ")";
 604 }
 605
 606 void OMPClausePrinter::VisitOMPFinalClause(OMPFinalClause *Node) {
 607   OS << "final(";
 608   Node->getCondition()->printPretty(OS, nullptr, Policy, 0);
 609   OS << ")";
 610 }
 611
 612 void OMPClausePrinter::VisitOMPNumThreadsClause(OMPNumThreadsClause *Node) {
 613   OS << "num_threads(";
 614   Node->getNumThreads()->printPretty(OS, nullptr, Policy, 0);
 615   OS << ")";
 616 }
 617
 618 void OMPClausePrinter::VisitOMPSafelenClause(OMPSafelenClause *Node) {
 619   OS << "safelen(";
 620   Node->getSafelen()->printPretty(OS, nullptr, Policy, 0);
 621   OS << ")";
 622 }
 623
 624 void OMPClausePrinter::VisitOMPCollapseClause(OMPCollapseClause *Node) {
 625   OS << "collapse(";
 626   Node->getNumForLoops()->printPretty(OS, nullptr, Policy, 0);
 627   OS << ")";
 628 }
 629
 630 void OMPClausePrinter::VisitOMPDefaultClause(OMPDefaultClause *Node) {
 631   OS << "default("
 632      << getOpenMPSimpleClauseTypeName(OMPC_default, Node->getDefaultKind())
 633      << ")";
 634 }
 635
 636 void OMPClausePrinter::VisitOMPProcBindClause(OMPProcBindClause *Node) {
 637   OS << "proc_bind("
 638      << getOpenMPSimpleClauseTypeName(OMPC_proc_bind, Node->getProcBindKind())
 639      << ")";
 640 }
 641
 642 void OMPClausePrinter::VisitOMPScheduleClause(OMPScheduleClause *Node) {
 643   OS << "schedule("
 644      << getOpenMPSimpleClauseTypeName(OMPC_schedule, Node->getScheduleKind());
 645   if (Node->getChunkSize()) {
 646     OS << ", ";
 647     Node->getChunkSize()->printPretty(OS, nullptr, Policy);
 648   }
 649   OS << ")";
 650 }
 651
 652 void OMPClausePrinter::VisitOMPOrderedClause(OMPOrderedClause *) {
 653   OS << "ordered";
 654 }
 655
 656 void OMPClausePrinter::VisitOMPNowaitClause(OMPNowaitClause *) {
 657   OS << "nowait";
 658 }
 659
 660 void OMPClausePrinter::VisitOMPUntiedClause(OMPUntiedClause *) {
 661   OS << "untied";
 662 }
 663
 664 void OMPClausePrinter::VisitOMPMergeableClause(OMPMergeableClause *) {
 665   OS << "mergeable";
 666 }
 667
 668 void OMPClausePrinter::VisitOMPReadClause(OMPReadClause *) { OS << "read"; }
 669
 670 void OMPClausePrinter::VisitOMPWriteClause(OMPWriteClause *) { OS << "write"; }
 671
 672 void OMPClausePrinter::VisitOMPUpdateClause(OMPUpdateClause *) {
 673   OS << "update";
 674 }
 675
 676 void OMPClausePrinter::VisitOMPCaptureClause(OMPCaptureClause *) {
 677   OS << "capture";
 678 }
 679
 680 void OMPClausePrinter::VisitOMPSeqCstClause(OMPSeqCstClause *) {
 681   OS << "seq_cst";
 682 }
 683
 684 template<typename T>
 685 void OMPClausePrinter::VisitOMPClauseList(T *Node, char StartSym) {
 686   for (typename T::varlist_iterator I = Node->varlist_begin(),
 687                                     E = Node->varlist_end();
 688          I != E; ++I) {
 689     assert(*I && "Expected non-null Stmt");
 690     if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(*I)) {
 691       OS << (I == Node->varlist_begin() ? StartSym : ',');
 692       cast<NamedDecl>(DRE->getDecl())->printQualifiedName(OS);
 693     } else {
 694       OS << (I == Node->varlist_begin() ? StartSym : ',');
 695       (*I)->printPretty(OS, nullptr, Policy, 0);
 696     }
 697   }
 698 }
 699
 700 void OMPClausePrinter::VisitOMPPrivateClause(OMPPrivateClause *Node) {
 701   if (!Node->varlist_empty()) {
 702     OS << "private";
 703     VisitOMPClauseList(Node, '(');
 704     OS << ")";
 705   }
 706 }
 707
 708 void OMPClausePrinter::VisitOMPFirstprivateClause(OMPFirstprivateClause *Node) {
 709   if (!Node->varlist_empty()) {
 710     OS << "firstprivate";
 711     VisitOMPClauseList(Node, '(');
 712     OS << ")";
 713   }
 714 }
 715
 716 void OMPClausePrinter::VisitOMPLastprivateClause(OMPLastprivateClause *Node) {
 717   if (!Node->varlist_empty()) {
 718     OS << "lastprivate";
 719     VisitOMPClauseList(Node, '(');
 720     OS << ")";
 721   }
 722 }
 723
 724 void OMPClausePrinter::VisitOMPSharedClause(OMPSharedClause *Node) {
 725   if (!Node->varlist_empty()) {
 726     OS << "shared";
 727     VisitOMPClauseList(Node, '(');
 728     OS << ")";
 729   }
 730 }
 731
 732 void OMPClausePrinter::VisitOMPReductionClause(OMPReductionClause *Node) {
 733   if (!Node->varlist_empty()) {
 734     OS << "reduction(";
 735     NestedNameSpecifier *QualifierLoc =
 736         Node->getQualifierLoc().getNestedNameSpecifier();
 737     OverloadedOperatorKind OOK =
 738         Node->getNameInfo().getName().getCXXOverloadedOperator();
 739     if (QualifierLoc == nullptr && OOK != OO_None) {
 740       // Print reduction identifier in C format
 741       OS << getOperatorSpelling(OOK);
 742     } else {
 743       // Use C++ format
 744       if (QualifierLoc != nullptr)
 745         QualifierLoc->print(OS, Policy);
 746       OS << Node->getNameInfo();
 747     }
 748     OS << ":";
 749     VisitOMPClauseList(Node, ' ');
 750     OS << ")";
 751   }
 752 }
 753
 754 void OMPClausePrinter::VisitOMPLinearClause(OMPLinearClause *Node) {
 755   if (!Node->varlist_empty()) {
 756     OS << "linear";
 757     VisitOMPClauseList(Node, '(');
 758     if (Node->getStep() != nullptr) {
 759       OS << ": ";
 760       Node->getStep()->printPretty(OS, nullptr, Policy, 0);
 761     }
 762     OS << ")";
 763   }
 764 }
 765
 766 void OMPClausePrinter::VisitOMPAlignedClause(OMPAlignedClause *Node) {
 767   if (!Node->varlist_empty()) {
 768     OS << "aligned";
 769     VisitOMPClauseList(Node, '(');
 770     if (Node->getAlignment() != nullptr) {
 771       OS << ": ";
 772       Node->getAlignment()->printPretty(OS, nullptr, Policy, 0);
 773     }
 774     OS << ")";
 775   }
 776 }
 777
 778 void OMPClausePrinter::VisitOMPCopyinClause(OMPCopyinClause *Node) {
 779   if (!Node->varlist_empty()) {
 780     OS << "copyin";
 781     VisitOMPClauseList(Node, '(');
 782     OS << ")";
 783   }
 784 }
 785
 786 void OMPClausePrinter::VisitOMPCopyprivateClause(OMPCopyprivateClause *Node) {
 787   if (!Node->varlist_empty()) {
 788     OS << "copyprivate";
 789     VisitOMPClauseList(Node, '(');
 790     OS << ")";
 791   }
 792 }
 793
 794 void OMPClausePrinter::VisitOMPFlushClause(OMPFlushClause *Node) {
 795   if (!Node->varlist_empty()) {
 796     VisitOMPClauseList(Node, '(');
 797     OS << ")";
 798   }
 799 }
 800 }
 801
 802 //===----------------------------------------------------------------------===//
 803 //  OpenMP directives printing methods
 804 //===----------------------------------------------------------------------===//
 805
 806 void StmtPrinter::PrintOMPExecutableDirective(OMPExecutableDirective *S) {
 807   OMPClausePrinter Printer(OS, Policy);
 808   ArrayRef<OMPClause *> Clauses = S->clauses();
 809   for (ArrayRef<OMPClause *>::iterator I = Clauses.begin(), E = Clauses.end();
 810        I != E; ++I)
 811     if (*I && !(*I)->isImplicit()) {
 812       Printer.Visit(*I);
 813       OS << ' ';
 814     }
 815   OS << "\n";
 816   if (S->hasAssociatedStmt() && S->getAssociatedStmt()) {
 817     assert(isa<CapturedStmt>(S->getAssociatedStmt()) &&
 818            "Expected captured statement!");
 819     Stmt *CS = cast<CapturedStmt>(S->getAssociatedStmt())->getCapturedStmt();
 820     PrintStmt(CS);
 821   }
 822 }
 823
 824 void StmtPrinter::VisitOMPParallelDirective(OMPParallelDirective *Node) {
 825   Indent() << "#pragma omp parallel ";
 826   PrintOMPExecutableDirective(Node);
 827 }
 828
 829 void StmtPrinter::VisitOMPSimdDirective(OMPSimdDirective *Node) {
 830   Indent() << "#pragma omp simd ";
 831   PrintOMPExecutableDirective(Node);
 832 }
 833
 834 void StmtPrinter::VisitOMPForDirective(OMPForDirective *Node) {
 835   Indent() << "#pragma omp for ";
 836   PrintOMPExecutableDirective(Node);
 837 }
 838
 839 void StmtPrinter::VisitOMPForSimdDirective(OMPForSimdDirective *Node) {
 840   Indent() << "#pragma omp for simd ";
 841   PrintOMPExecutableDirective(Node);
 842 }
 843
 844 void StmtPrinter::VisitOMPSectionsDirective(OMPSectionsDirective *Node) {
 845   Indent() << "#pragma omp sections ";
 846   PrintOMPExecutableDirective(Node);
 847 }
 848
 849 void StmtPrinter::VisitOMPSectionDirective(OMPSectionDirective *Node) {
 850   Indent() << "#pragma omp section";
 851   PrintOMPExecutableDirective(Node);
 852 }
 853
 854 void StmtPrinter::VisitOMPSingleDirective(OMPSingleDirective *Node) {
 855   Indent() << "#pragma omp single ";
 856   PrintOMPExecutableDirective(Node);
 857 }
 858
 859 void StmtPrinter::VisitOMPMasterDirective(OMPMasterDirective *Node) {
 860   Indent() << "#pragma omp master";
 861   PrintOMPExecutableDirective(Node);
 862 }
 863
 864 void StmtPrinter::VisitOMPCriticalDirective(OMPCriticalDirective *Node) {
 865   Indent() << "#pragma omp critical";
 866   if (Node->getDirectiveName().getName()) {
 867     OS << " (";
 868     Node->getDirectiveName().printName(OS);
 869     OS << ")";
 870   }
 871   PrintOMPExecutableDirective(Node);
 872 }
 873
 874 void StmtPrinter::VisitOMPParallelForDirective(OMPParallelForDirective *Node) {
 875   Indent() << "#pragma omp parallel for ";
 876   PrintOMPExecutableDirective(Node);
 877 }
 878
 879 void StmtPrinter::VisitOMPParallelForSimdDirective(
 880     OMPParallelForSimdDirective *Node) {
 881   Indent() << "#pragma omp parallel for simd ";
 882   PrintOMPExecutableDirective(Node);
 883 }
 884
 885 void StmtPrinter::VisitOMPParallelSectionsDirective(
 886     OMPParallelSectionsDirective *Node) {
 887   Indent() << "#pragma omp parallel sections ";
 888   PrintOMPExecutableDirective(Node);
 889 }
 890
 891 void StmtPrinter::VisitOMPTaskDirective(OMPTaskDirective *Node) {
 892   Indent() << "#pragma omp task ";
 893   PrintOMPExecutableDirective(Node);
 894 }
 895
 896 void StmtPrinter::VisitOMPTaskyieldDirective(OMPTaskyieldDirective *Node) {
 897   Indent() << "#pragma omp taskyield";
 898   PrintOMPExecutableDirective(Node);
 899 }
 900
 901 void StmtPrinter::VisitOMPBarrierDirective(OMPBarrierDirective *Node) {
 902   Indent() << "#pragma omp barrier";
 903   PrintOMPExecutableDirective(Node);
 904 }
 905
 906 void StmtPrinter::VisitOMPTaskwaitDirective(OMPTaskwaitDirective *Node) {
 907   Indent() << "#pragma omp taskwait";
 908   PrintOMPExecutableDirective(Node);
 909 }
 910
 911 void StmtPrinter::VisitOMPFlushDirective(OMPFlushDirective *Node) {
 912   Indent() << "#pragma omp flush ";
 913   PrintOMPExecutableDirective(Node);
 914 }
 915
 916 void StmtPrinter::VisitOMPOrderedDirective(OMPOrderedDirective *Node) {
 917   Indent() << "#pragma omp ordered";
 918   PrintOMPExecutableDirective(Node);
 919 }
 920
 921 void StmtPrinter::VisitOMPAtomicDirective(OMPAtomicDirective *Node) {
 922   Indent() << "#pragma omp atomic ";
 923   PrintOMPExecutableDirective(Node);
 924 }
 925
 926 void StmtPrinter::VisitOMPTargetDirective(OMPTargetDirective *Node) {
 927   Indent() << "#pragma omp target ";
 928   PrintOMPExecutableDirective(Node);
 929 }
 930
 931 void StmtPrinter::VisitOMPTeamsDirective(OMPTeamsDirective *Node) {
 932   Indent() << "#pragma omp teams ";
 933   PrintOMPExecutableDirective(Node);
 934 }
 935
 936 //===----------------------------------------------------------------------===//
 937 //  Expr printing methods.
 938 //===----------------------------------------------------------------------===//
 939
 940 void StmtPrinter::VisitDeclRefExpr(DeclRefExpr *Node) {
 941   if (NestedNameSpecifier *Qualifier = Node->getQualifier())
 942     Qualifier->print(OS, Policy);
 943   if (Node->hasTemplateKeyword())
 944     OS << "template ";
 945   OS << Node->getNameInfo();
 946   if (Node->hasExplicitTemplateArgs())
 947     TemplateSpecializationType::PrintTemplateArgumentList(
 948         OS, Node->getTemplateArgs(), Node->getNumTemplateArgs(), Policy);
 949 }
 950
 951 void StmtPrinter::VisitDependentScopeDeclRefExpr(
 952                                            DependentScopeDeclRefExpr *Node) {
 953   if (NestedNameSpecifier *Qualifier = Node->getQualifier())
 954     Qualifier->print(OS, Policy);
 955   if (Node->hasTemplateKeyword())
 956     OS << "template ";
 957   OS << Node->getNameInfo();
 958   if (Node->hasExplicitTemplateArgs())
 959     TemplateSpecializationType::PrintTemplateArgumentList(
 960         OS, Node->getTemplateArgs(), Node->getNumTemplateArgs(), Policy);
 961 }
 962
 963 void StmtPrinter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *Node) {
 964   if (Node->getQualifier())
 965     Node->getQualifier()->print(OS, Policy);
 966   if (Node->hasTemplateKeyword())
 967     OS << "template ";
 968   OS << Node->getNameInfo();
 969   if (Node->hasExplicitTemplateArgs())
 970     TemplateSpecializationType::PrintTemplateArgumentList(
 971         OS, Node->getTemplateArgs(), Node->getNumTemplateArgs(), Policy);
 972 }
 973
 974 void StmtPrinter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *Node) {
 975   if (Node->getBase()) {
 976     PrintExpr(Node->getBase());
 977     OS << (Node->isArrow() ? "->" : ".");
 978   }
 979   OS << *Node->getDecl();
 980 }
 981
 982 void StmtPrinter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *Node) {
 983   if (Node->isSuperReceiver())
 984     OS << "super.";
 985   else if (Node->isObjectReceiver() && Node->getBase()) {
 986     PrintExpr(Node->getBase());
 987     OS << ".";
 988   } else if (Node->isClassReceiver() && Node->getClassReceiver()) {
 989     OS << Node->getClassReceiver()->getName() << ".";
 990   }
 991
 992   if (Node->isImplicitProperty())
 993     Node->getImplicitPropertyGetter()->getSelector().print(OS);
 994   else
 995     OS << Node->getExplicitProperty()->getName();
 996 }
 997
 998 void StmtPrinter::VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr *Node) {
 999
1000   PrintExpr(Node->getBaseExpr());
1001   OS << "[";
1002   PrintExpr(Node->getKeyExpr());
1003   OS << "]";
1004 }
1005
1006 void StmtPrinter::VisitPredefinedExpr(PredefinedExpr *Node) {
1007   OS << PredefinedExpr::getIdentTypeName(Node->getIdentType());
1008 }
1009
1010 void StmtPrinter::VisitCharacterLiteral(CharacterLiteral *Node) {
1011   unsigned value = Node->getValue();
1012
1013   switch (Node->getKind()) {
1014   case CharacterLiteral::Ascii: break; // no prefix.
1015   case CharacterLiteral::Wide:  OS << 'L'; break;
1016   case CharacterLiteral::UTF16: OS << 'u'; break;
1017   case CharacterLiteral::UTF32: OS << 'U'; break;
1018   }
1019
1020   switch (value) {
1021   case '\\':
1022     OS << "'\\\\'";
1023     break;
1024   case '\'':
1025     OS << "'\\''";
1026     break;
1027   case '\a':
1028     // TODO: K&R: the meaning of '\\a' is different in traditional C
1029     OS << "'\\a'";
1030     break;
1031   case '\b':
1032     OS << "'\\b'";
1033     break;
1034   // Nonstandard escape sequence.
1035   /*case '\e':
1036     OS << "'\\e'";
1037     break;*/
1038   case '\f':
1039     OS << "'\\f'";
1040     break;
1041   case '\n':
1042     OS << "'\\n'";
1043     break;
1044   case '\r':
1045     OS << "'\\r'";
1046     break;
1047   case '\t':
1048     OS << "'\\t'";
1049     break;
1050   case '\v':
1051     OS << "'\\v'";
1052     break;
1053   default:
1054     if (value < 256 && isPrintable((unsigned char)value))
1055       OS << "'" << (char)value << "'";
1056     else if (value < 256)
1057       OS << "'\\x" << llvm::format("%02x", value) << "'";
1058     else if (value <= 0xFFFF)
1059       OS << "'\\u" << llvm::format("%04x", value) << "'";
1060     else
1061       OS << "'\\U" << llvm::format("%08x", value) << "'";
1062   }
1063 }
1064
1065 void StmtPrinter::VisitIntegerLiteral(IntegerLiteral *Node) {
1066   bool isSigned = Node->getType()->isSignedIntegerType();
1067   OS << Node->getValue().toString(10, isSigned);
1068
1069   // Emit suffixes.  Integer literals are always a builtin integer type.
1070   switch (Node->getType()->getAs<BuiltinType>()->getKind()) {
1071   default: llvm_unreachable("Unexpected type for integer literal!");
1072   case BuiltinType::SChar:     OS << "i8"; break;
1073   case BuiltinType::UChar:     OS << "Ui8"; break;
1074   case BuiltinType::Short:     OS << "i16"; break;
1075   case BuiltinType::UShort:    OS << "Ui16"; break;
1076   case BuiltinType::Int:       break; // no suffix.
1077   case BuiltinType::UInt:      OS << 'U'; break;
1078   case BuiltinType::Long:      OS << 'L'; break;
1079   case BuiltinType::ULong:     OS << "UL"; break;
1080   case BuiltinType::LongLong:  OS << "LL"; break;
1081   case BuiltinType::ULongLong: OS << "ULL"; break;
1082   case BuiltinType::Int128:    OS << "i128"; break;
1083   case BuiltinType::UInt128:   OS << "Ui128"; break;
1084   }
1085 }
1086
1087 static void PrintFloatingLiteral(raw_ostream &OS, FloatingLiteral *Node,
1088                                  bool PrintSuffix) {
1089   SmallString<16> Str;
1090   Node->getValue().toString(Str);
1091   OS << Str;
1092   if (Str.find_first_not_of("-0123456789") == StringRef::npos)
1093     OS << '.'; // Trailing dot in order to separate from ints.
1094
1095   if (!PrintSuffix)
1096     return;
1097
1098   // Emit suffixes.  Float literals are always a builtin float type.
1099   switch (Node->getType()->getAs<BuiltinType>()->getKind()) {
1100   default: llvm_unreachable("Unexpected type for float literal!");
1101   case BuiltinType::Half:       break; // FIXME: suffix?
1102   case BuiltinType::Double:     break; // no suffix.
1103   case BuiltinType::Float:      OS << 'F'; break;
1104   case BuiltinType::LongDouble: OS << 'L'; break;
1105   }
1106 }
1107
1108 void StmtPrinter::VisitFloatingLiteral(FloatingLiteral *Node) {
1109   PrintFloatingLiteral(OS, Node, /*PrintSuffix=*/true);
1110 }
1111
1112 void StmtPrinter::VisitImaginaryLiteral(ImaginaryLiteral *Node) {
1113   PrintExpr(Node->getSubExpr());
1114   OS << "i";
1115 }
1116
1117 void StmtPrinter::VisitStringLiteral(StringLiteral *Str) {
1118   Str->outputString(OS);
1119 }
1120 void StmtPrinter::VisitParenExpr(ParenExpr *Node) {
1121   OS << "(";
1122   PrintExpr(Node->getSubExpr());
1123   OS << ")";
1124 }
1125 void StmtPrinter::VisitUnaryOperator(UnaryOperator *Node) {
1126   if (!Node->isPostfix()) {
1127     OS << UnaryOperator::getOpcodeStr(Node->getOpcode());
1128
1129     // Print a space if this is an "identifier operator" like __real, or if
1130     // it might be concatenated incorrectly like '+'.
1131     switch (Node->getOpcode()) {
1132     default: break;
1133     case UO_Real:
1134     case UO_Imag:
1135     case UO_Extension:
1136       OS << ' ';
1137       break;
1138     case UO_Plus:
1139     case UO_Minus:
1140       if (isa<UnaryOperator>(Node->getSubExpr()))
1141         OS << ' ';
1142       break;
1143     }
1144   }
1145   PrintExpr(Node->getSubExpr());
1146
1147   if (Node->isPostfix())
1148     OS << UnaryOperator::getOpcodeStr(Node->getOpcode());
1149 }
1150
1151 void StmtPrinter::VisitOffsetOfExpr(OffsetOfExpr *Node) {
1152   OS << "__builtin_offsetof(";
1153   Node->getTypeSourceInfo()->getType().print(OS, Policy);
1154   OS << ", ";
1155   bool PrintedSomething = false;
1156   for (unsigned i = 0, n = Node->getNumComponents(); i < n; ++i) {
1157     OffsetOfExpr::OffsetOfNode ON = Node->getComponent(i);
1158     if (ON.getKind() == OffsetOfExpr::OffsetOfNode::Array) {
1159       // Array node
1160       OS << "[";
1161       PrintExpr(Node->getIndexExpr(ON.getArrayExprIndex()));
1162       OS << "]";
1163       PrintedSomething = true;
1164       continue;
1165     }
1166
1167     // Skip implicit base indirections.
1168     if (ON.getKind() == OffsetOfExpr::OffsetOfNode::Base)
1169       continue;
1170
1171     // Field or identifier node.
1172     IdentifierInfo *Id = ON.getFieldName();
1173     if (!Id)
1174       continue;
1175
1176     if (PrintedSomething)
1177       OS << ".";
1178     else
1179       PrintedSomething = true;
1180     OS << Id->getName();
1181   }
1182   OS << ")";
1183 }
1184
1185 void StmtPrinter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *Node){
1186   switch(Node->getKind()) {
1187   case UETT_SizeOf:
1188     OS << "sizeof";
1189     break;
1190   case UETT_AlignOf:
1191     if (Policy.LangOpts.CPlusPlus)
1192       OS << "alignof";
1193     else if (Policy.LangOpts.C11)
1194       OS << "_Alignof";
1195     else
1196       OS << "__alignof";
1197     break;
1198   case UETT_VecStep:
1199     OS << "vec_step";
1200     break;
1201   }
1202   if (Node->isArgumentType()) {
1203     OS << '(';
1204     Node->getArgumentType().print(OS, Policy);
1205     OS << ')';
1206   } else {
1207     OS << " ";
1208     PrintExpr(Node->getArgumentExpr());
1209   }
1210 }
1211
1212 void StmtPrinter::VisitGenericSelectionExpr(GenericSelectionExpr *Node) {
1213   OS << "_Generic(";
1214   PrintExpr(Node->getControllingExpr());
1215   for (unsigned i = 0; i != Node->getNumAssocs(); ++i) {
1216     OS << ", ";
1217     QualType T = Node->getAssocType(i);
1218     if (T.isNull())
1219       OS << "default";
1220     else
1221       T.print(OS, Policy);
1222     OS << ": ";
1223     PrintExpr(Node->getAssocExpr(i));
1224   }
1225   OS << ")";
1226 }
1227
1228 void StmtPrinter::VisitArraySubscriptExpr(ArraySubscriptExpr *Node) {
1229   PrintExpr(Node->getLHS());
1230   OS << "[";
1231   PrintExpr(Node->getRHS());
1232   OS << "]";
1233 }
1234
1235 void StmtPrinter::PrintCallArgs(CallExpr *Call) {
1236   for (unsigned i = 0, e = Call->getNumArgs(); i != e; ++i) {
1237     if (isa<CXXDefaultArgExpr>(Call->getArg(i))) {
1238       // Don't print any defaulted arguments
1239       break;
1240     }
1241
1242     if (i) OS << ", ";
1243     PrintExpr(Call->getArg(i));
1244   }
1245 }
1246
1247 void StmtPrinter::VisitCallExpr(CallExpr *Call) {
1248   PrintExpr(Call->getCallee());
1249   OS << "(";
1250   PrintCallArgs(Call);
1251   OS << ")";
1252 }
1253 void StmtPrinter::VisitMemberExpr(MemberExpr *Node) {
1254   // FIXME: Suppress printing implicit bases (like "this")
1255   PrintExpr(Node->getBase());
1256
1257   MemberExpr *ParentMember = dyn_cast<MemberExpr>(Node->getBase());
1258   FieldDecl  *ParentDecl   = ParentMember
1259     ? dyn_cast<FieldDecl>(ParentMember->getMemberDecl()) : nullptr;
1260
1261   if (!ParentDecl || !ParentDecl->isAnonymousStructOrUnion())
1262     OS << (Node->isArrow() ? "->" : ".");
1263
1264   if (FieldDecl *FD = dyn_cast<FieldDecl>(Node->getMemberDecl()))
1265     if (FD->isAnonymousStructOrUnion())
1266       return;
1267
1268   if (NestedNameSpecifier *Qualifier = Node->getQualifier())
1269     Qualifier->print(OS, Policy);
1270   if (Node->hasTemplateKeyword())
1271     OS << "template ";
1272   OS << Node->getMemberNameInfo();
1273   if (Node->hasExplicitTemplateArgs())
1274     TemplateSpecializationType::PrintTemplateArgumentList(
1275         OS, Node->getTemplateArgs(), Node->getNumTemplateArgs(), Policy);
1276 }
1277 void StmtPrinter::VisitObjCIsaExpr(ObjCIsaExpr *Node) {
1278   PrintExpr(Node->getBase());
1279   OS << (Node->isArrow() ? "->isa" : ".isa");
1280 }
1281
1282 void StmtPrinter::VisitExtVectorElementExpr(ExtVectorElementExpr *Node) {
1283   PrintExpr(Node->getBase());
1284   OS << ".";
1285   OS << Node->getAccessor().getName();
1286 }
1287 void StmtPrinter::VisitCStyleCastExpr(CStyleCastExpr *Node) {
1288   OS << '(';
1289   Node->getTypeAsWritten().print(OS, Policy);
1290   OS << ')';
1291   PrintExpr(Node->getSubExpr());
1292 }
1293 void StmtPrinter::VisitCompoundLiteralExpr(CompoundLiteralExpr *Node) {
1294   OS << '(';
1295   Node->getType().print(OS, Policy);
1296   OS << ')';
1297   PrintExpr(Node->getInitializer());
1298 }
1299 void StmtPrinter::VisitImplicitCastExpr(ImplicitCastExpr *Node) {
1300   // No need to print anything, simply forward to the subexpression.
1301   PrintExpr(Node->getSubExpr());
1302 }
1303 void StmtPrinter::VisitBinaryOperator(BinaryOperator *Node) {
1304   PrintExpr(Node->getLHS());
1305   OS << " " << BinaryOperator::getOpcodeStr(Node->getOpcode()) << " ";
1306   PrintExpr(Node->getRHS());
1307 }
1308 void StmtPrinter::VisitCompoundAssignOperator(CompoundAssignOperator *Node) {
1309   PrintExpr(Node->getLHS());
1310   OS << " " << BinaryOperator::getOpcodeStr(Node->getOpcode()) << " ";
1311   PrintExpr(Node->getRHS());
1312 }
1313 void StmtPrinter::VisitConditionalOperator(ConditionalOperator *Node) {
1314   PrintExpr(Node->getCond());
1315   OS << " ? ";
1316   PrintExpr(Node->getLHS());
1317   OS << " : ";
1318   PrintExpr(Node->getRHS());
1319 }
1320
1321 // GNU extensions.
1322
1323 void
1324 StmtPrinter::VisitBinaryConditionalOperator(BinaryConditionalOperator *Node) {
1325   PrintExpr(Node->getCommon());
1326   OS << " ?: ";
1327   PrintExpr(Node->getFalseExpr());
1328 }
1329 void StmtPrinter::VisitAddrLabelExpr(AddrLabelExpr *Node) {
1330   OS << "&&" << Node->getLabel()->getName();
1331 }
1332
1333 void StmtPrinter::VisitStmtExpr(StmtExpr *E) {
1334   OS << "(";
1335   PrintRawCompoundStmt(E->getSubStmt());
1336   OS << ")";
1337 }
1338
1339 void StmtPrinter::VisitChooseExpr(ChooseExpr *Node) {
1340   OS << "__builtin_choose_expr(";
1341   PrintExpr(Node->getCond());
1342   OS << ", ";
1343   PrintExpr(Node->getLHS());
1344   OS << ", ";
1345   PrintExpr(Node->getRHS());
1346   OS << ")";
1347 }
1348
1349 void StmtPrinter::VisitGNUNullExpr(GNUNullExpr *) {
1350   OS << "__null";
1351 }
1352
1353 void StmtPrinter::VisitShuffleVectorExpr(ShuffleVectorExpr *Node) {
1354   OS << "__builtin_shufflevector(";
1355   for (unsigned i = 0, e = Node->getNumSubExprs(); i != e; ++i) {
1356     if (i) OS << ", ";
1357     PrintExpr(Node->getExpr(i));
1358   }
1359   OS << ")";
1360 }
1361
1362 void StmtPrinter::VisitConvertVectorExpr(ConvertVectorExpr *Node) {
1363   OS << "__builtin_convertvector(";
1364   PrintExpr(Node->getSrcExpr());
1365   OS << ", ";
1366   Node->getType().print(OS, Policy);
1367   OS << ")";
1368 }
1369
1370 void StmtPrinter::VisitInitListExpr(InitListExpr* Node) {
1371   if (Node->getSyntacticForm()) {
1372     Visit(Node->getSyntacticForm());
1373     return;
1374   }
1375
1376   OS << "{ ";
1377   for (unsigned i = 0, e = Node->getNumInits(); i != e; ++i) {
1378     if (i) OS << ", ";
1379     if (Node->getInit(i))
1380       PrintExpr(Node->getInit(i));
1381     else
1382       OS << "0";
1383   }
1384   OS << " }";
1385 }
1386
1387 void StmtPrinter::VisitParenListExpr(ParenListExpr* Node) {
1388   OS << "( ";
1389   for (unsigned i = 0, e = Node->getNumExprs(); i != e; ++i) {
1390     if (i) OS << ", ";
1391     PrintExpr(Node->getExpr(i));
1392   }
1393   OS << " )";
1394 }
1395
1396 void StmtPrinter::VisitDesignatedInitExpr(DesignatedInitExpr *Node) {
1397   for (DesignatedInitExpr::designators_iterator D = Node->designators_begin(),
1398                       DEnd = Node->designators_end();
1399        D != DEnd; ++D) {
1400     if (D->isFieldDesignator()) {
1401       if (D->getDotLoc().isInvalid()) {
1402         if (IdentifierInfo *II = D->getFieldName())
1403           OS << II->getName() << ":";
1404       } else {
1405         OS << "." << D->getFieldName()->getName();
1406       }
1407     } else {
1408       OS << "[";
1409       if (D->isArrayDesignator()) {
1410         PrintExpr(Node->getArrayIndex(*D));
1411       } else {
1412         PrintExpr(Node->getArrayRangeStart(*D));
1413         OS << " ... ";
1414         PrintExpr(Node->getArrayRangeEnd(*D));
1415       }
1416       OS << "]";
1417     }
1418   }
1419
1420   OS << " = ";
1421   PrintExpr(Node->getInit());
1422 }
1423
1424 void StmtPrinter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *Node) {
1425   if (Policy.LangOpts.CPlusPlus) {
1426     OS << "/*implicit*/";
1427     Node->getType().print(OS, Policy);
1428     OS << "()";
1429   } else {
1430     OS << "/*implicit*/(";
1431     Node->getType().print(OS, Policy);
1432     OS << ')';
1433     if (Node->getType()->isRecordType())
1434       OS << "{}";
1435     else
1436       OS << 0;
1437   }
1438 }
1439
1440 void StmtPrinter::VisitVAArgExpr(VAArgExpr *Node) {
1441   OS << "__builtin_va_arg(";
1442   PrintExpr(Node->getSubExpr());
1443   OS << ", ";
1444   Node->getType().print(OS, Policy);
1445   OS << ")";
1446 }
1447
1448 void StmtPrinter::VisitPseudoObjectExpr(PseudoObjectExpr *Node) {
1449   PrintExpr(Node->getSyntacticForm());
1450 }
1451
1452 void StmtPrinter::VisitAtomicExpr(AtomicExpr *Node) {
1453   const char *Name = nullptr;
1454   switch (Node->getOp()) {
1455 #define BUILTIN(ID, TYPE, ATTRS)
1456 #define ATOMIC_BUILTIN(ID, TYPE, ATTRS) \
1457   case AtomicExpr::AO ## ID: \
1458     Name = #ID "("; \
1459     break;
1460 #include "clang/Basic/Builtins.def"
1461   }
1462   OS << Name;
1463
1464   // AtomicExpr stores its subexpressions in a permuted order.
1465   PrintExpr(Node->getPtr());
1466   if (Node->getOp() != AtomicExpr::AO__c11_atomic_load &&
1467       Node->getOp() != AtomicExpr::AO__atomic_load_n) {
1468     OS << ", ";
1469     PrintExpr(Node->getVal1());
1470   }
1471   if (Node->getOp() == AtomicExpr::AO__atomic_exchange ||
1472       Node->isCmpXChg()) {
1473     OS << ", ";
1474     PrintExpr(Node->getVal2());
1475   }
1476   if (Node->getOp() == AtomicExpr::AO__atomic_compare_exchange ||
1477       Node->getOp() == AtomicExpr::AO__atomic_compare_exchange_n) {
1478     OS << ", ";
1479     PrintExpr(Node->getWeak());
1480   }
1481   if (Node->getOp() != AtomicExpr::AO__c11_atomic_init) {
1482     OS << ", ";
1483     PrintExpr(Node->getOrder());
1484   }
1485   if (Node->isCmpXChg()) {
1486     OS << ", ";
1487     PrintExpr(Node->getOrderFail());
1488   }
1489   OS << ")";
1490 }
1491
1492 // C++
1493 void StmtPrinter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *Node) {
1494   const char *OpStrings[NUM_OVERLOADED_OPERATORS] = {
1495     "",
1496 #define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
1497     Spelling,
1498 #include "clang/Basic/OperatorKinds.def"
1499   };
1500
1501   OverloadedOperatorKind Kind = Node->getOperator();
1502   if (Kind == OO_PlusPlus || Kind == OO_MinusMinus) {
1503     if (Node->getNumArgs() == 1) {
1504       OS << OpStrings[Kind] << ' ';
1505       PrintExpr(Node->getArg(0));
1506     } else {
1507       PrintExpr(Node->getArg(0));
1508       OS << ' ' << OpStrings[Kind];
1509     }
1510   } else if (Kind == OO_Arrow) {
1511     PrintExpr(Node->getArg(0));
1512   } else if (Kind == OO_Call) {
1513     PrintExpr(Node->getArg(0));
1514     OS << '(';
1515     for (unsigned ArgIdx = 1; ArgIdx < Node->getNumArgs(); ++ArgIdx) {
1516       if (ArgIdx > 1)
1517         OS << ", ";
1518       if (!isa<CXXDefaultArgExpr>(Node->getArg(ArgIdx)))
1519         PrintExpr(Node->getArg(ArgIdx));
1520     }
1521     OS << ')';
1522   } else if (Kind == OO_Subscript) {
1523     PrintExpr(Node->getArg(0));
1524     OS << '[';
1525     PrintExpr(Node->getArg(1));
1526     OS << ']';
1527   } else if (Node->getNumArgs() == 1) {
1528     OS << OpStrings[Kind] << ' ';
1529     PrintExpr(Node->getArg(0));
1530   } else if (Node->getNumArgs() == 2) {
1531     PrintExpr(Node->getArg(0));
1532     OS << ' ' << OpStrings[Kind] << ' ';
1533     PrintExpr(Node->getArg(1));
1534   } else {
1535     llvm_unreachable("unknown overloaded operator");
1536   }
1537 }
1538
1539 void StmtPrinter::VisitCXXMemberCallExpr(CXXMemberCallExpr *Node) {
1540   // If we have a conversion operator call only print the argument.
1541   CXXMethodDecl *MD = Node->getMethodDecl();
1542   if (MD && isa<CXXConversionDecl>(MD)) {
1543     PrintExpr(Node->getImplicitObjectArgument());
1544     return;
1545   }
1546   VisitCallExpr(cast<CallExpr>(Node));
1547 }
1548
1549 void StmtPrinter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *Node) {
1550   PrintExpr(Node->getCallee());
1551   OS << "<<<";
1552   PrintCallArgs(Node->getConfig());
1553   OS << ">>>(";
1554   PrintCallArgs(Node);
1555   OS << ")";
1556 }
1557
1558 void StmtPrinter::VisitCXXNamedCastExpr(CXXNamedCastExpr *Node) {
1559   OS << Node->getCastName() << '<';
1560   Node->getTypeAsWritten().print(OS, Policy);
1561   OS << ">(";
1562   PrintExpr(Node->getSubExpr());
1563   OS << ")";
1564 }
1565
1566 void StmtPrinter::VisitCXXStaticCastExpr(CXXStaticCastExpr *Node) {
1567   VisitCXXNamedCastExpr(Node);
1568 }
1569
1570 void StmtPrinter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *Node) {
1571   VisitCXXNamedCastExpr(Node);
1572 }
1573
1574 void StmtPrinter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *Node) {
1575   VisitCXXNamedCastExpr(Node);
1576 }
1577
1578 void StmtPrinter::VisitCXXConstCastExpr(CXXConstCastExpr *Node) {
1579   VisitCXXNamedCastExpr(Node);
1580 }
1581
1582 void StmtPrinter::VisitCXXTypeidExpr(CXXTypeidExpr *Node) {
1583   OS << "typeid(";
1584   if (Node->isTypeOperand()) {
1585     Node->getTypeOperandSourceInfo()->getType().print(OS, Policy);
1586   } else {
1587     PrintExpr(Node->getExprOperand());
1588   }
1589   OS << ")";
1590 }
1591
1592 void StmtPrinter::VisitCXXUuidofExpr(CXXUuidofExpr *Node) {
1593   OS << "__uuidof(";
1594   if (Node->isTypeOperand()) {
1595     Node->getTypeOperandSourceInfo()->getType().print(OS, Policy);
1596   } else {
1597     PrintExpr(Node->getExprOperand());
1598   }
1599   OS << ")";
1600 }
1601
1602 void StmtPrinter::VisitMSPropertyRefExpr(MSPropertyRefExpr *Node) {
1603   PrintExpr(Node->getBaseExpr());
1604   if (Node->isArrow())
1605     OS << "->";
1606   else
1607     OS << ".";
1608   if (NestedNameSpecifier *Qualifier =
1609       Node->getQualifierLoc().getNestedNameSpecifier())
1610     Qualifier->print(OS, Policy);
1611   OS << Node->getPropertyDecl()->getDeclName();
1612 }
1613
1614 void StmtPrinter::VisitUserDefinedLiteral(UserDefinedLiteral *Node) {
1615   switch (Node->getLiteralOperatorKind()) {
1616   case UserDefinedLiteral::LOK_Raw:
1617     OS << cast<StringLiteral>(Node->getArg(0)->IgnoreImpCasts())->getString();
1618     break;
1619   case UserDefinedLiteral::LOK_Template: {
1620     DeclRefExpr *DRE = cast<DeclRefExpr>(Node->getCallee()->IgnoreImpCasts());
1621     const TemplateArgumentList *Args =
1622       cast<FunctionDecl>(DRE->getDecl())->getTemplateSpecializationArgs();
1623     assert(Args);
1624     const TemplateArgument &Pack = Args->get(0);
1625     for (const auto &P : Pack.pack_elements()) {
1626       char C = (char)P.getAsIntegral().getZExtValue();
1627       OS << C;
1628     }
1629     break;
1630   }
1631   case UserDefinedLiteral::LOK_Integer: {
1632     // Print integer literal without suffix.
1633     IntegerLiteral *Int = cast<IntegerLiteral>(Node->getCookedLiteral());
1634     OS << Int->getValue().toString(10, /*isSigned*/false);
1635     break;
1636   }
1637   case UserDefinedLiteral::LOK_Floating: {
1638     // Print floating literal without suffix.
1639     FloatingLiteral *Float = cast<FloatingLiteral>(Node->getCookedLiteral());
1640     PrintFloatingLiteral(OS, Float, /*PrintSuffix=*/false);
1641     break;
1642   }
1643   case UserDefinedLiteral::LOK_String:
1644   case UserDefinedLiteral::LOK_Character:
1645     PrintExpr(Node->getCookedLiteral());
1646     break;
1647   }
1648   OS << Node->getUDSuffix()->getName();
1649 }
1650
1651 void StmtPrinter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *Node) {
1652   OS << (Node->getValue() ? "true" : "false");
1653 }
1654
1655 void StmtPrinter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *Node) {
1656   OS << "nullptr";
1657 }
1658
1659 void StmtPrinter::VisitCXXThisExpr(CXXThisExpr *Node) {
1660   OS << "this";
1661 }
1662
1663 void StmtPrinter::VisitCXXThrowExpr(CXXThrowExpr *Node) {
1664   if (!Node->getSubExpr())
1665     OS << "throw";
1666   else {
1667     OS << "throw ";
1668     PrintExpr(Node->getSubExpr());
1669   }
1670 }
1671
1672 void StmtPrinter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *Node) {
1673   // Nothing to print: we picked up the default argument.
1674 }
1675
1676 void StmtPrinter::VisitCXXDefaultInitExpr(CXXDefaultInitExpr *Node) {
1677   // Nothing to print: we picked up the default initializer.
1678 }
1679
1680 void StmtPrinter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *Node) {
1681   Node->getType().print(OS, Policy);
1682   OS << "(";
1683   PrintExpr(Node->getSubExpr());
1684   OS << ")";
1685 }
1686
1687 void StmtPrinter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *Node) {
1688   PrintExpr(Node->getSubExpr());
1689 }
1690
1691 void StmtPrinter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *Node) {
1692   Node->getType().print(OS, Policy);
1693   OS << "(";
1694   for (CXXTemporaryObjectExpr::arg_iterator Arg = Node->arg_begin(),
1695                                          ArgEnd = Node->arg_end();
1696        Arg != ArgEnd; ++Arg) {
1697     if (Arg->isDefaultArgument())
1698       break;
1699     if (Arg != Node->arg_begin())
1700       OS << ", ";
1701     PrintExpr(*Arg);
1702   }
1703   OS << ")";
1704 }
1705
1706 void StmtPrinter::VisitLambdaExpr(LambdaExpr *Node) {
1707   OS << '[';
1708   bool NeedComma = false;
1709   switch (Node->getCaptureDefault()) {
1710   case LCD_None:
1711     break;
1712
1713   case LCD_ByCopy:
1714     OS << '=';
1715     NeedComma = true;
1716     break;
1717
1718   case LCD_ByRef:
1719     OS << '&';
1720     NeedComma = true;
1721     break;
1722   }
1723   for (LambdaExpr::capture_iterator C = Node->explicit_capture_begin(),
1724                                  CEnd = Node->explicit_capture_end();
1725        C != CEnd;
1726        ++C) {
1727     if (NeedComma)
1728       OS << ", ";
1729     NeedComma = true;
1730
1731     switch (C->getCaptureKind()) {
1732     case LCK_This:
1733       OS << "this";
1734       break;
1735
1736     case LCK_ByRef:
1737       if (Node->getCaptureDefault() != LCD_ByRef || C->isInitCapture())
1738         OS << '&';
1739       OS << C->getCapturedVar()->getName();
1740       break;
1741
1742     case LCK_ByCopy:
1743       OS << C->getCapturedVar()->getName();
1744       break;
1745     case LCK_VLAType:
1746       llvm_unreachable("VLA type in explicit captures.");
1747     }
1748
1749     if (C->isInitCapture())
1750       PrintExpr(C->getCapturedVar()->getInit());
1751   }
1752   OS << ']';
1753
1754   if (Node->hasExplicitParameters()) {
1755     OS << " (";
1756     CXXMethodDecl *Method = Node->getCallOperator();
1757     NeedComma = false;
1758     for (auto P : Method->params()) {
1759       if (NeedComma) {
1760         OS << ", ";
1761       } else {
1762         NeedComma = true;
1763       }
1764       std::string ParamStr = P->getNameAsString();
1765       P->getOriginalType().print(OS, Policy, ParamStr);
1766     }
1767     if (Method->isVariadic()) {
1768       if (NeedComma)
1769         OS << ", ";
1770       OS << "...";
1771     }
1772     OS << ')';
1773
1774     if (Node->isMutable())
1775       OS << " mutable";
1776
1777     const FunctionProtoType *Proto
1778       = Method->getType()->getAs<FunctionProtoType>();
1779     Proto->printExceptionSpecification(OS, Policy);
1780
1781     // FIXME: Attributes
1782
1783     // Print the trailing return type if it was specified in the source.
1784     if (Node->hasExplicitResultType()) {
1785       OS << " -> ";
1786       Proto->getReturnType().print(OS, Policy);
1787     }
1788   }
1789
1790   // Print the body.
1791   CompoundStmt *Body = Node->getBody();
1792   OS << ' ';
1793   PrintStmt(Body);
1794 }
1795
1796 void StmtPrinter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *Node) {
1797   if (TypeSourceInfo *TSInfo = Node->getTypeSourceInfo())
1798     TSInfo->getType().print(OS, Policy);
1799   else
1800     Node->getType().print(OS, Policy);
1801   OS << "()";
1802 }
1803
1804 void StmtPrinter::VisitCXXNewExpr(CXXNewExpr *E) {
1805   if (E->isGlobalNew())
1806     OS << "::";
1807   OS << "new ";
1808   unsigned NumPlace = E->getNumPlacementArgs();
1809   if (NumPlace > 0 && !isa<CXXDefaultArgExpr>(E->getPlacementArg(0))) {
1810     OS << "(";
1811     PrintExpr(E->getPlacementArg(0));
1812     for (unsigned i = 1; i < NumPlace; ++i) {
1813       if (isa<CXXDefaultArgExpr>(E->getPlacementArg(i)))
1814         break;
1815       OS << ", ";
1816       PrintExpr(E->getPlacementArg(i));
1817     }
1818     OS << ") ";
1819   }
1820   if (E->isParenTypeId())
1821     OS << "(";
1822   std::string TypeS;
1823   if (Expr *Size = E->getArraySize()) {
1824     llvm::raw_string_ostream s(TypeS);
1825     s << '[';
1826     Size->printPretty(s, Helper, Policy);
1827     s << ']';
1828   }
1829   E->getAllocatedType().print(OS, Policy, TypeS);
1830   if (E->isParenTypeId())
1831     OS << ")";
1832
1833   CXXNewExpr::InitializationStyle InitStyle = E->getInitializationStyle();
1834   if (InitStyle) {
1835     if (InitStyle == CXXNewExpr::CallInit)
1836       OS << "(";
1837     PrintExpr(E->getInitializer());
1838     if (InitStyle == CXXNewExpr::CallInit)
1839       OS << ")";
1840   }
1841 }
1842
1843 void StmtPrinter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
1844   if (E->isGlobalDelete())
1845     OS << "::";
1846   OS << "delete ";
1847   if (E->isArrayForm())
1848     OS << "[] ";
1849   PrintExpr(E->getArgument());
1850 }
1851
1852 void StmtPrinter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1853   PrintExpr(E->getBase());
1854   if (E->isArrow())
1855     OS << "->";
1856   else
1857     OS << '.';
1858   if (E->getQualifier())
1859     E->getQualifier()->print(OS, Policy);
1860   OS << "~";
1861
1862   if (IdentifierInfo *II = E->getDestroyedTypeIdentifier())
1863     OS << II->getName();
1864   else
1865     E->getDestroyedType().print(OS, Policy);
1866 }
1867
1868 void StmtPrinter::VisitCXXConstructExpr(CXXConstructExpr *E) {
1869   if (E->isListInitialization())
1870     OS << "{ ";
1871
1872   for (unsigned i = 0, e = E->getNumArgs(); i != e; ++i) {
1873     if (isa<CXXDefaultArgExpr>(E->getArg(i))) {
1874       // Don't print any defaulted arguments
1875       break;
1876     }
1877
1878     if (i) OS << ", ";
1879     PrintExpr(E->getArg(i));
1880   }
1881
1882   if (E->isListInitialization())
1883     OS << " }";
1884 }
1885
1886 void StmtPrinter::VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E) {
1887   PrintExpr(E->getSubExpr());
1888 }
1889
1890 void StmtPrinter::VisitExprWithCleanups(ExprWithCleanups *E) {
1891   // Just forward to the subexpression.
1892   PrintExpr(E->getSubExpr());
1893 }
1894
1895 void
1896 StmtPrinter::VisitCXXUnresolvedConstructExpr(
1897                                            CXXUnresolvedConstructExpr *Node) {
1898   Node->getTypeAsWritten().print(OS, Policy);
1899   OS << "(";
1900   for (CXXUnresolvedConstructExpr::arg_iterator Arg = Node->arg_begin(),
1901                                              ArgEnd = Node->arg_end();
1902        Arg != ArgEnd; ++Arg) {
1903     if (Arg != Node->arg_begin())
1904       OS << ", ";
1905     PrintExpr(*Arg);
1906   }
1907   OS << ")";
1908 }
1909
1910 void StmtPrinter::VisitCXXDependentScopeMemberExpr(
1911                                          CXXDependentScopeMemberExpr *Node) {
1912   if (!Node->isImplicitAccess()) {
1913     PrintExpr(Node->getBase());
1914     OS << (Node->isArrow() ? "->" : ".");
1915   }
1916   if (NestedNameSpecifier *Qualifier = Node->getQualifier())
1917     Qualifier->print(OS, Policy);
1918   if (Node->hasTemplateKeyword())
1919     OS << "template ";
1920   OS << Node->getMemberNameInfo();
1921   if (Node->hasExplicitTemplateArgs())
1922     TemplateSpecializationType::PrintTemplateArgumentList(
1923         OS, Node->getTemplateArgs(), Node->getNumTemplateArgs(), Policy);
1924 }
1925
1926 void StmtPrinter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *Node) {
1927   if (!Node->isImplicitAccess()) {
1928     PrintExpr(Node->getBase());
1929     OS << (Node->isArrow() ? "->" : ".");
1930   }
1931   if (NestedNameSpecifier *Qualifier = Node->getQualifier())
1932     Qualifier->print(OS, Policy);
1933   if (Node->hasTemplateKeyword())
1934     OS << "template ";
1935   OS << Node->getMemberNameInfo();
1936   if (Node->hasExplicitTemplateArgs())
1937     TemplateSpecializationType::PrintTemplateArgumentList(
1938         OS, Node->getTemplateArgs(), Node->getNumTemplateArgs(), Policy);
1939 }
1940
1941 static const char *getTypeTraitName(TypeTrait TT) {
1942   switch (TT) {
1943 #define TYPE_TRAIT_1(Spelling, Name, Key) \
1944 case clang::UTT_##Name: return #Spelling;
1945 #define TYPE_TRAIT_2(Spelling, Name, Key) \
1946 case clang::BTT_##Name: return #Spelling;
1947 #define TYPE_TRAIT_N(Spelling, Name, Key) \
1948   case clang::TT_##Name: return #Spelling;
1949 #include "clang/Basic/TokenKinds.def"
1950   }
1951   llvm_unreachable("Type trait not covered by switch");
1952 }
1953
1954 static const char *getTypeTraitName(ArrayTypeTrait ATT) {
1955   switch (ATT) {
1956   case ATT_ArrayRank:        return "__array_rank";
1957   case ATT_ArrayExtent:      return "__array_extent";
1958   }
1959   llvm_unreachable("Array type trait not covered by switch");
1960 }
1961
1962 static const char *getExpressionTraitName(ExpressionTrait ET) {
1963   switch (ET) {
1964   case ET_IsLValueExpr:      return "__is_lvalue_expr";
1965   case ET_IsRValueExpr:      return "__is_rvalue_expr";
1966   }
1967   llvm_unreachable("Expression type trait not covered by switch");
1968 }
1969
1970 void StmtPrinter::VisitTypeTraitExpr(TypeTraitExpr *E) {
1971   OS << getTypeTraitName(E->getTrait()) << "(";
1972   for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I) {
1973     if (I > 0)
1974       OS << ", ";
1975     E->getArg(I)->getType().print(OS, Policy);
1976   }
1977   OS << ")";
1978 }
1979
1980 void StmtPrinter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
1981   OS << getTypeTraitName(E->getTrait()) << '(';
1982   E->getQueriedType().print(OS, Policy);
1983   OS << ')';
1984 }
1985
1986 void StmtPrinter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
1987   OS << getExpressionTraitName(E->getTrait()) << '(';
1988   PrintExpr(E->getQueriedExpression());
1989   OS << ')';
1990 }
1991
1992 void StmtPrinter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
1993   OS << "noexcept(";
1994   PrintExpr(E->getOperand());
1995   OS << ")";
1996 }
1997
1998 void StmtPrinter::VisitPackExpansionExpr(PackExpansionExpr *E) {
1999   PrintExpr(E->getPattern());
2000   OS << "...";
2001 }
2002
2003 void StmtPrinter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
2004   OS << "sizeof...(" << *E->getPack() << ")";
2005 }
2006
2007 void StmtPrinter::VisitSubstNonTypeTemplateParmPackExpr(
2008                                        SubstNonTypeTemplateParmPackExpr *Node) {
2009   OS << *Node->getParameterPack();
2010 }
2011
2012 void StmtPrinter::VisitSubstNonTypeTemplateParmExpr(
2013                                        SubstNonTypeTemplateParmExpr *Node) {
2014   Visit(Node->getReplacement());
2015 }
2016
2017 void StmtPrinter::VisitFunctionParmPackExpr(FunctionParmPackExpr *E) {
2018   OS << *E->getParameterPack();
2019 }
2020
2021 void StmtPrinter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *Node){
2022   PrintExpr(Node->GetTemporaryExpr());
2023 }
2024
2025 void StmtPrinter::VisitCXXFoldExpr(CXXFoldExpr *E) {
2026   OS << "(";
2027   if (E->getLHS()) {
2028     PrintExpr(E->getLHS());
2029     OS << " " << BinaryOperator::getOpcodeStr(E->getOperator()) << " ";
2030   }
2031   OS << "...";
2032   if (E->getRHS()) {
2033     OS << " " << BinaryOperator::getOpcodeStr(E->getOperator()) << " ";
2034     PrintExpr(E->getRHS());
2035   }
2036   OS << ")";
2037 }
2038
2039 // Obj-C
2040
2041 void StmtPrinter::VisitObjCStringLiteral(ObjCStringLiteral *Node) {
2042   OS << "@";
2043   VisitStringLiteral(Node->getString());
2044 }
2045
2046 void StmtPrinter::VisitObjCBoxedExpr(ObjCBoxedExpr *E) {
2047   OS << "@";
2048   Visit(E->getSubExpr());
2049 }
2050
2051 void StmtPrinter::VisitObjCArrayLiteral(ObjCArrayLiteral *E) {
2052   OS << "@[ ";
2053   StmtRange ch = E->children();
2054   if (ch.first != ch.second) {
2055     while (1) {
2056       Visit(*ch.first);
2057       ++ch.first;
2058       if (ch.first == ch.second) break;
2059       OS << ", ";
2060     }
2061   }
2062   OS << " ]";
2063 }
2064
2065 void StmtPrinter::VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) {
2066   OS << "@{ ";
2067   for (unsigned I = 0, N = E->getNumElements(); I != N; ++I) {
2068     if (I > 0)
2069       OS << ", ";
2070
2071     ObjCDictionaryElement Element = E->getKeyValueElement(I);
2072     Visit(Element.Key);
2073     OS << " : ";
2074     Visit(Element.Value);
2075     if (Element.isPackExpansion())
2076       OS << "...";
2077   }
2078   OS << " }";
2079 }
2080
2081 void StmtPrinter::VisitObjCEncodeExpr(ObjCEncodeExpr *Node) {
2082   OS << "@encode(";
2083   Node->getEncodedType().print(OS, Policy);
2084   OS << ')';
2085 }
2086
2087 void StmtPrinter::VisitObjCSelectorExpr(ObjCSelectorExpr *Node) {
2088   OS << "@selector(";
2089   Node->getSelector().print(OS);
2090   OS << ')';
2091 }
2092
2093 void StmtPrinter::VisitObjCProtocolExpr(ObjCProtocolExpr *Node) {
2094   OS << "@protocol(" << *Node->getProtocol() << ')';
2095 }
2096
2097 void StmtPrinter::VisitObjCMessageExpr(ObjCMessageExpr *Mess) {
2098   OS << "[";
2099   switch (Mess->getReceiverKind()) {
2100   case ObjCMessageExpr::Instance:
2101     PrintExpr(Mess->getInstanceReceiver());
2102     break;
2103
2104   case ObjCMessageExpr::Class:
2105     Mess->getClassReceiver().print(OS, Policy);
2106     break;
2107
2108   case ObjCMessageExpr::SuperInstance:
2109   case ObjCMessageExpr::SuperClass:
2110     OS << "Super";
2111     break;
2112   }
2113
2114   OS << ' ';
2115   Selector selector = Mess->getSelector();
2116   if (selector.isUnarySelector()) {
2117     OS << selector.getNameForSlot(0);
2118   } else {
2119     for (unsigned i = 0, e = Mess->getNumArgs(); i != e; ++i) {
2120       if (i < selector.getNumArgs()) {
2121         if (i > 0) OS << ' ';
2122         if (selector.getIdentifierInfoForSlot(i))
2123           OS << selector.getIdentifierInfoForSlot(i)->getName() << ':';
2124         else
2125            OS << ":";
2126       }
2127       else OS << ", "; // Handle variadic methods.
2128
2129       PrintExpr(Mess->getArg(i));
2130     }
2131   }
2132   OS << "]";
2133 }
2134
2135 void StmtPrinter::VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr *Node) {
2136   OS << (Node->getValue() ? "__objc_yes" : "__objc_no");
2137 }
2138
2139 void
2140 StmtPrinter::VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) {
2141   PrintExpr(E->getSubExpr());
2142 }
2143
2144 void
2145 StmtPrinter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) {
2146   OS << '(' << E->getBridgeKindName();
2147   E->getType().print(OS, Policy);
2148   OS << ')';
2149   PrintExpr(E->getSubExpr());
2150 }
2151
2152 void StmtPrinter::VisitBlockExpr(BlockExpr *Node) {
2153   BlockDecl *BD = Node->getBlockDecl();
2154   OS << "^";
2155
2156   const FunctionType *AFT = Node->getFunctionType();
2157
2158   if (isa<FunctionNoProtoType>(AFT)) {
2159     OS << "()";
2160   } else if (!BD->param_empty() || cast<FunctionProtoType>(AFT)->isVariadic()) {
2161     OS << '(';
2162     for (BlockDecl::param_iterator AI = BD->param_begin(),
2163          E = BD->param_end(); AI != E; ++AI) {
2164       if (AI != BD->param_begin()) OS << ", ";
2165       std::string ParamStr = (*AI)->getNameAsString();
2166       (*AI)->getType().print(OS, Policy, ParamStr);
2167     }
2168
2169     const FunctionProtoType *FT = cast<FunctionProtoType>(AFT);
2170     if (FT->isVariadic()) {
2171       if (!BD->param_empty()) OS << ", ";
2172       OS << "...";
2173     }
2174     OS << ')';
2175   }
2176   OS << "{ }";
2177 }
2178
2179 void StmtPrinter::VisitOpaqueValueExpr(OpaqueValueExpr *Node) {
2180   PrintExpr(Node->getSourceExpr());
2181 }
2182
2183 void StmtPrinter::VisitTypoExpr(TypoExpr *Node) {
2184   // TODO: Print something reasonable for a TypoExpr, if necessary.
2185   assert(false && "Cannot print TypoExpr nodes");
2186 }
2187
2188 void StmtPrinter::VisitAsTypeExpr(AsTypeExpr *Node) {
2189   OS << "__builtin_astype(";
2190   PrintExpr(Node->getSrcExpr());
2191   OS << ", ";
2192   Node->getType().print(OS, Policy);
2193   OS << ")";
2194 }
2195
2196 //===----------------------------------------------------------------------===//
2197 // Stmt method implementations
2198 //===----------------------------------------------------------------------===//
2199
2200 void Stmt::dumpPretty(const ASTContext &Context) const {
2201   printPretty(llvm::errs(), nullptr, PrintingPolicy(Context.getLangOpts()));
2202 }
2203
2204 void Stmt::printPretty(raw_ostream &OS,
2205                        PrinterHelper *Helper,
2206                        const PrintingPolicy &Policy,
2207                        unsigned Indentation) const {
2208   StmtPrinter P(OS, Helper, Policy, Indentation);
2209   P.Visit(const_cast<Stmt*>(this));
2210 }
2211
2212 //===----------------------------------------------------------------------===//
2213 // PrinterHelper
2214 //===----------------------------------------------------------------------===//
2215
2216 // Implement virtual destructor.
2217 PrinterHelper::~PrinterHelper() {}