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