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