1 //===--- StmtPrinter.cpp - Printing implementation for Stmt ASTs ----------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file implements the Stmt::dumpPretty/Stmt::printPretty methods, which
11 // pretty print the AST back out to C code.
13 //===----------------------------------------------------------------------===//
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;
31 //===----------------------------------------------------------------------===//
32 // StmtPrinter Visitor
33 //===----------------------------------------------------------------------===//
36 class StmtPrinter : public StmtVisitor<StmtPrinter> {
39 clang::PrinterHelper* Helper;
40 PrintingPolicy Policy;
43 StmtPrinter(raw_ostream &os, PrinterHelper* helper,
44 const PrintingPolicy &Policy,
45 unsigned Indentation = 0)
46 : OS(os), IndentLevel(Indentation), Helper(helper), Policy(Policy) {}
48 void PrintStmt(Stmt *S) {
49 PrintStmt(S, Policy.Indentation);
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.
62 Indent() << "<<<NULL STATEMENT>>>\n";
64 IndentLevel -= SubIndent;
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);
77 void PrintExpr(Expr *E) {
84 raw_ostream &Indent(int Delta = 0) {
85 for (int i = 0, e = IndentLevel+Delta; i < e; ++i)
91 if (Helper && Helper->handledStmt(S,OS))
93 else StmtVisitor<StmtPrinter>::Visit(S);
96 void VisitStmt(Stmt *Node) LLVM_ATTRIBUTE_UNUSED {
97 Indent() << "<<unknown stmt type>>\n";
99 void VisitExpr(Expr *Node) LLVM_ATTRIBUTE_UNUSED {
100 OS << "<<unknown expr type>>";
102 void VisitCXXNamedCastExpr(CXXNamedCastExpr *Node);
104 #define ABSTRACT_STMT(CLASS)
105 #define STMT(CLASS, PARENT) \
106 void Visit##CLASS(CLASS *Node);
107 #include "clang/AST/StmtNodes.inc"
111 //===----------------------------------------------------------------------===//
112 // Stmt printing methods.
113 //===----------------------------------------------------------------------===//
115 /// PrintRawCompoundStmt - Print a compound stmt without indenting the {, and
116 /// with no newline after the }.
117 void StmtPrinter::PrintRawCompoundStmt(CompoundStmt *Node) {
119 for (auto *I : Node->body())
125 void StmtPrinter::PrintRawDecl(Decl *D) {
126 D->print(OS, Policy, IndentLevel);
129 void StmtPrinter::PrintRawDeclStmt(const DeclStmt *S) {
130 SmallVector<Decl*, 2> Decls(S->decls());
131 Decl::printGroup(Decls.data(), Decls.size(), OS, Policy, IndentLevel);
134 void StmtPrinter::VisitNullStmt(NullStmt *Node) {
138 void StmtPrinter::VisitDeclStmt(DeclStmt *Node) {
140 PrintRawDeclStmt(Node);
144 void StmtPrinter::VisitCompoundStmt(CompoundStmt *Node) {
146 PrintRawCompoundStmt(Node);
150 void StmtPrinter::VisitCaseStmt(CaseStmt *Node) {
151 Indent(-1) << "case ";
152 PrintExpr(Node->getLHS());
153 if (Node->getRHS()) {
155 PrintExpr(Node->getRHS());
159 PrintStmt(Node->getSubStmt(), 0);
162 void StmtPrinter::VisitDefaultStmt(DefaultStmt *Node) {
163 Indent(-1) << "default:\n";
164 PrintStmt(Node->getSubStmt(), 0);
167 void StmtPrinter::VisitLabelStmt(LabelStmt *Node) {
168 Indent(-1) << Node->getName() << ":\n";
169 PrintStmt(Node->getSubStmt(), 0);
172 void StmtPrinter::VisitAttributedStmt(AttributedStmt *Node) {
173 for (const auto *Attr : Node->getAttrs()) {
174 Attr->printPretty(OS, Policy);
177 PrintStmt(Node->getSubStmt(), 0);
180 void StmtPrinter::PrintRawIfStmt(IfStmt *If) {
182 if (const DeclStmt *DS = If->getConditionVariableDeclStmt())
183 PrintRawDeclStmt(DS);
185 PrintExpr(If->getCond());
188 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(If->getThen())) {
190 PrintRawCompoundStmt(CS);
191 OS << (If->getElse() ? ' ' : '\n');
194 PrintStmt(If->getThen());
195 if (If->getElse()) Indent();
198 if (Stmt *Else = If->getElse()) {
201 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Else)) {
203 PrintRawCompoundStmt(CS);
205 } else if (IfStmt *ElseIf = dyn_cast<IfStmt>(Else)) {
207 PrintRawIfStmt(ElseIf);
210 PrintStmt(If->getElse());
215 void StmtPrinter::VisitIfStmt(IfStmt *If) {
220 void StmtPrinter::VisitSwitchStmt(SwitchStmt *Node) {
221 Indent() << "switch (";
222 if (const DeclStmt *DS = Node->getConditionVariableDeclStmt())
223 PrintRawDeclStmt(DS);
225 PrintExpr(Node->getCond());
228 // Pretty print compoundstmt bodies (very common).
229 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
231 PrintRawCompoundStmt(CS);
235 PrintStmt(Node->getBody());
239 void StmtPrinter::VisitWhileStmt(WhileStmt *Node) {
240 Indent() << "while (";
241 if (const DeclStmt *DS = Node->getConditionVariableDeclStmt())
242 PrintRawDeclStmt(DS);
244 PrintExpr(Node->getCond());
246 PrintStmt(Node->getBody());
249 void StmtPrinter::VisitDoStmt(DoStmt *Node) {
251 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
252 PrintRawCompoundStmt(CS);
256 PrintStmt(Node->getBody());
261 PrintExpr(Node->getCond());
265 void StmtPrinter::VisitForStmt(ForStmt *Node) {
267 if (Node->getInit()) {
268 if (DeclStmt *DS = dyn_cast<DeclStmt>(Node->getInit()))
269 PrintRawDeclStmt(DS);
271 PrintExpr(cast<Expr>(Node->getInit()));
274 if (Node->getCond()) {
276 PrintExpr(Node->getCond());
279 if (Node->getInc()) {
281 PrintExpr(Node->getInc());
285 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
286 PrintRawCompoundStmt(CS);
290 PrintStmt(Node->getBody());
294 void StmtPrinter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *Node) {
296 if (DeclStmt *DS = dyn_cast<DeclStmt>(Node->getElement()))
297 PrintRawDeclStmt(DS);
299 PrintExpr(cast<Expr>(Node->getElement()));
301 PrintExpr(Node->getCollection());
304 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
305 PrintRawCompoundStmt(CS);
309 PrintStmt(Node->getBody());
313 void StmtPrinter::VisitCXXForRangeStmt(CXXForRangeStmt *Node) {
315 PrintingPolicy SubPolicy(Policy);
316 SubPolicy.SuppressInitializers = true;
317 Node->getLoopVariable()->print(OS, SubPolicy, IndentLevel);
319 PrintExpr(Node->getRangeInit());
321 PrintStmt(Node->getBody());
323 if (Policy.IncludeNewlines) OS << "\n";
326 void StmtPrinter::VisitMSDependentExistsStmt(MSDependentExistsStmt *Node) {
328 if (Node->isIfExists())
329 OS << "__if_exists (";
331 OS << "__if_not_exists (";
333 if (NestedNameSpecifier *Qualifier
334 = Node->getQualifierLoc().getNestedNameSpecifier())
335 Qualifier->print(OS, Policy);
337 OS << Node->getNameInfo() << ") ";
339 PrintRawCompoundStmt(Node->getSubStmt());
342 void StmtPrinter::VisitGotoStmt(GotoStmt *Node) {
343 Indent() << "goto " << Node->getLabel()->getName() << ";";
344 if (Policy.IncludeNewlines) OS << "\n";
347 void StmtPrinter::VisitIndirectGotoStmt(IndirectGotoStmt *Node) {
348 Indent() << "goto *";
349 PrintExpr(Node->getTarget());
351 if (Policy.IncludeNewlines) OS << "\n";
354 void StmtPrinter::VisitContinueStmt(ContinueStmt *Node) {
355 Indent() << "continue;";
356 if (Policy.IncludeNewlines) OS << "\n";
359 void StmtPrinter::VisitBreakStmt(BreakStmt *Node) {
360 Indent() << "break;";
361 if (Policy.IncludeNewlines) OS << "\n";
365 void StmtPrinter::VisitReturnStmt(ReturnStmt *Node) {
366 Indent() << "return";
367 if (Node->getRetValue()) {
369 PrintExpr(Node->getRetValue());
372 if (Policy.IncludeNewlines) OS << "\n";
376 void StmtPrinter::VisitGCCAsmStmt(GCCAsmStmt *Node) {
379 if (Node->isVolatile())
383 VisitStringLiteral(Node->getAsmString());
386 if (Node->getNumOutputs() != 0 || Node->getNumInputs() != 0 ||
387 Node->getNumClobbers() != 0)
390 for (unsigned i = 0, e = Node->getNumOutputs(); i != e; ++i) {
394 if (!Node->getOutputName(i).empty()) {
396 OS << Node->getOutputName(i);
400 VisitStringLiteral(Node->getOutputConstraintLiteral(i));
402 Visit(Node->getOutputExpr(i));
407 if (Node->getNumInputs() != 0 || Node->getNumClobbers() != 0)
410 for (unsigned i = 0, e = Node->getNumInputs(); i != e; ++i) {
414 if (!Node->getInputName(i).empty()) {
416 OS << Node->getInputName(i);
420 VisitStringLiteral(Node->getInputConstraintLiteral(i));
422 Visit(Node->getInputExpr(i));
427 if (Node->getNumClobbers() != 0)
430 for (unsigned i = 0, e = Node->getNumClobbers(); i != e; ++i) {
434 VisitStringLiteral(Node->getClobberStringLiteral(i));
438 if (Policy.IncludeNewlines) OS << "\n";
441 void StmtPrinter::VisitMSAsmStmt(MSAsmStmt *Node) {
442 // FIXME: Implement MS style inline asm statement printer.
443 Indent() << "__asm ";
444 if (Node->hasBraces())
446 OS << Node->getAsmString() << "\n";
447 if (Node->hasBraces())
451 void StmtPrinter::VisitCapturedStmt(CapturedStmt *Node) {
452 PrintStmt(Node->getCapturedDecl()->getBody());
455 void StmtPrinter::VisitObjCAtTryStmt(ObjCAtTryStmt *Node) {
457 if (CompoundStmt *TS = dyn_cast<CompoundStmt>(Node->getTryBody())) {
458 PrintRawCompoundStmt(TS);
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())
470 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(catchStmt->getCatchBody())) {
471 PrintRawCompoundStmt(CS);
476 if (ObjCAtFinallyStmt *FS = static_cast<ObjCAtFinallyStmt *>(
477 Node->getFinallyStmt())) {
478 Indent() << "@finally";
479 PrintRawCompoundStmt(dyn_cast<CompoundStmt>(FS->getFinallyBody()));
484 void StmtPrinter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *Node) {
487 void StmtPrinter::VisitObjCAtCatchStmt (ObjCAtCatchStmt *Node) {
488 Indent() << "@catch (...) { /* todo */ } \n";
491 void StmtPrinter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *Node) {
492 Indent() << "@throw";
493 if (Node->getThrowExpr()) {
495 PrintExpr(Node->getThrowExpr());
500 void StmtPrinter::VisitObjCAvailabilityCheckExpr(
501 ObjCAvailabilityCheckExpr *Node) {
502 OS << "@available(...)";
505 void StmtPrinter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *Node) {
506 Indent() << "@synchronized (";
507 PrintExpr(Node->getSynchExpr());
509 PrintRawCompoundStmt(Node->getSynchBody());
513 void StmtPrinter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *Node) {
514 Indent() << "@autoreleasepool";
515 PrintRawCompoundStmt(dyn_cast<CompoundStmt>(Node->getSubStmt()));
519 void StmtPrinter::PrintRawCXXCatchStmt(CXXCatchStmt *Node) {
521 if (Decl *ExDecl = Node->getExceptionDecl())
522 PrintRawDecl(ExDecl);
526 PrintRawCompoundStmt(cast<CompoundStmt>(Node->getHandlerBlock()));
529 void StmtPrinter::VisitCXXCatchStmt(CXXCatchStmt *Node) {
531 PrintRawCXXCatchStmt(Node);
535 void StmtPrinter::VisitCXXTryStmt(CXXTryStmt *Node) {
537 PrintRawCompoundStmt(Node->getTryBlock());
538 for (unsigned i = 0, e = Node->getNumHandlers(); i < e; ++i) {
540 PrintRawCXXCatchStmt(Node->getHandler(i));
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();
551 PrintRawSEHExceptHandler(E);
553 assert(F && "Must have a finally block...");
554 PrintRawSEHFinallyStmt(F);
559 void StmtPrinter::PrintRawSEHFinallyStmt(SEHFinallyStmt *Node) {
561 PrintRawCompoundStmt(Node->getBlock());
565 void StmtPrinter::PrintRawSEHExceptHandler(SEHExceptStmt *Node) {
567 VisitExpr(Node->getFilterExpr());
569 PrintRawCompoundStmt(Node->getBlock());
573 void StmtPrinter::VisitSEHExceptStmt(SEHExceptStmt *Node) {
575 PrintRawSEHExceptHandler(Node);
579 void StmtPrinter::VisitSEHFinallyStmt(SEHFinallyStmt *Node) {
581 PrintRawSEHFinallyStmt(Node);
585 void StmtPrinter::VisitSEHLeaveStmt(SEHLeaveStmt *Node) {
586 Indent() << "__leave;";
587 if (Policy.IncludeNewlines) OS << "\n";
590 //===----------------------------------------------------------------------===//
591 // OpenMP clauses printing methods
592 //===----------------------------------------------------------------------===//
595 class OMPClausePrinter : public OMPClauseVisitor<OMPClausePrinter> {
597 const PrintingPolicy &Policy;
598 /// \brief Process clauses with list of variables.
599 template <typename T>
600 void VisitOMPClauseList(T *Node, char StartSym);
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"
609 void OMPClausePrinter::VisitOMPIfClause(OMPIfClause *Node) {
611 if (Node->getNameModifier() != OMPD_unknown)
612 OS << getOpenMPDirectiveName(Node->getNameModifier()) << ": ";
613 Node->getCondition()->printPretty(OS, nullptr, Policy, 0);
617 void OMPClausePrinter::VisitOMPFinalClause(OMPFinalClause *Node) {
619 Node->getCondition()->printPretty(OS, nullptr, Policy, 0);
623 void OMPClausePrinter::VisitOMPNumThreadsClause(OMPNumThreadsClause *Node) {
624 OS << "num_threads(";
625 Node->getNumThreads()->printPretty(OS, nullptr, Policy, 0);
629 void OMPClausePrinter::VisitOMPSafelenClause(OMPSafelenClause *Node) {
631 Node->getSafelen()->printPretty(OS, nullptr, Policy, 0);
635 void OMPClausePrinter::VisitOMPSimdlenClause(OMPSimdlenClause *Node) {
637 Node->getSimdlen()->printPretty(OS, nullptr, Policy, 0);
641 void OMPClausePrinter::VisitOMPCollapseClause(OMPCollapseClause *Node) {
643 Node->getNumForLoops()->printPretty(OS, nullptr, Policy, 0);
647 void OMPClausePrinter::VisitOMPDefaultClause(OMPDefaultClause *Node) {
649 << getOpenMPSimpleClauseTypeName(OMPC_default, Node->getDefaultKind())
653 void OMPClausePrinter::VisitOMPProcBindClause(OMPProcBindClause *Node) {
655 << getOpenMPSimpleClauseTypeName(OMPC_proc_bind, Node->getProcBindKind())
659 void OMPClausePrinter::VisitOMPScheduleClause(OMPScheduleClause *Node) {
661 if (Node->getFirstScheduleModifier() != OMPC_SCHEDULE_MODIFIER_unknown) {
662 OS << getOpenMPSimpleClauseTypeName(OMPC_schedule,
663 Node->getFirstScheduleModifier());
664 if (Node->getSecondScheduleModifier() != OMPC_SCHEDULE_MODIFIER_unknown) {
666 OS << getOpenMPSimpleClauseTypeName(OMPC_schedule,
667 Node->getSecondScheduleModifier());
671 OS << getOpenMPSimpleClauseTypeName(OMPC_schedule, Node->getScheduleKind());
672 if (auto *E = Node->getChunkSize()) {
674 E->printPretty(OS, nullptr, Policy);
679 void OMPClausePrinter::VisitOMPOrderedClause(OMPOrderedClause *Node) {
681 if (auto *Num = Node->getNumForLoops()) {
683 Num->printPretty(OS, nullptr, Policy, 0);
688 void OMPClausePrinter::VisitOMPNowaitClause(OMPNowaitClause *) {
692 void OMPClausePrinter::VisitOMPUntiedClause(OMPUntiedClause *) {
696 void OMPClausePrinter::VisitOMPNogroupClause(OMPNogroupClause *) {
700 void OMPClausePrinter::VisitOMPMergeableClause(OMPMergeableClause *) {
704 void OMPClausePrinter::VisitOMPReadClause(OMPReadClause *) { OS << "read"; }
706 void OMPClausePrinter::VisitOMPWriteClause(OMPWriteClause *) { OS << "write"; }
708 void OMPClausePrinter::VisitOMPUpdateClause(OMPUpdateClause *) {
712 void OMPClausePrinter::VisitOMPCaptureClause(OMPCaptureClause *) {
716 void OMPClausePrinter::VisitOMPSeqCstClause(OMPSeqCstClause *) {
720 void OMPClausePrinter::VisitOMPThreadsClause(OMPThreadsClause *) {
724 void OMPClausePrinter::VisitOMPSIMDClause(OMPSIMDClause *) { OS << "simd"; }
726 void OMPClausePrinter::VisitOMPDeviceClause(OMPDeviceClause *Node) {
728 Node->getDevice()->printPretty(OS, nullptr, Policy, 0);
732 void OMPClausePrinter::VisitOMPNumTeamsClause(OMPNumTeamsClause *Node) {
734 Node->getNumTeams()->printPretty(OS, nullptr, Policy, 0);
738 void OMPClausePrinter::VisitOMPThreadLimitClause(OMPThreadLimitClause *Node) {
739 OS << "thread_limit(";
740 Node->getThreadLimit()->printPretty(OS, nullptr, Policy, 0);
744 void OMPClausePrinter::VisitOMPPriorityClause(OMPPriorityClause *Node) {
746 Node->getPriority()->printPretty(OS, nullptr, Policy, 0);
750 void OMPClausePrinter::VisitOMPGrainsizeClause(OMPGrainsizeClause *Node) {
752 Node->getGrainsize()->printPretty(OS, nullptr, Policy, 0);
756 void OMPClausePrinter::VisitOMPNumTasksClause(OMPNumTasksClause *Node) {
758 Node->getNumTasks()->printPretty(OS, nullptr, Policy, 0);
762 void OMPClausePrinter::VisitOMPHintClause(OMPHintClause *Node) {
764 Node->getHint()->printPretty(OS, nullptr, Policy, 0);
769 void OMPClausePrinter::VisitOMPClauseList(T *Node, char StartSym) {
770 for (typename T::varlist_iterator I = Node->varlist_begin(),
771 E = Node->varlist_end();
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);
779 DRE->getDecl()->printQualifiedName(OS);
781 (*I)->printPretty(OS, nullptr, Policy, 0);
785 void OMPClausePrinter::VisitOMPPrivateClause(OMPPrivateClause *Node) {
786 if (!Node->varlist_empty()) {
788 VisitOMPClauseList(Node, '(');
793 void OMPClausePrinter::VisitOMPFirstprivateClause(OMPFirstprivateClause *Node) {
794 if (!Node->varlist_empty()) {
795 OS << "firstprivate";
796 VisitOMPClauseList(Node, '(');
801 void OMPClausePrinter::VisitOMPLastprivateClause(OMPLastprivateClause *Node) {
802 if (!Node->varlist_empty()) {
804 VisitOMPClauseList(Node, '(');
809 void OMPClausePrinter::VisitOMPSharedClause(OMPSharedClause *Node) {
810 if (!Node->varlist_empty()) {
812 VisitOMPClauseList(Node, '(');
817 void OMPClausePrinter::VisitOMPReductionClause(OMPReductionClause *Node) {
818 if (!Node->varlist_empty()) {
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);
829 if (QualifierLoc != nullptr)
830 QualifierLoc->print(OS, Policy);
831 OS << Node->getNameInfo();
834 VisitOMPClauseList(Node, ' ');
839 void OMPClausePrinter::VisitOMPLinearClause(OMPLinearClause *Node) {
840 if (!Node->varlist_empty()) {
842 if (Node->getModifierLoc().isValid()) {
844 << getOpenMPSimpleClauseTypeName(OMPC_linear, Node->getModifier());
846 VisitOMPClauseList(Node, '(');
847 if (Node->getModifierLoc().isValid())
849 if (Node->getStep() != nullptr) {
851 Node->getStep()->printPretty(OS, nullptr, Policy, 0);
857 void OMPClausePrinter::VisitOMPAlignedClause(OMPAlignedClause *Node) {
858 if (!Node->varlist_empty()) {
860 VisitOMPClauseList(Node, '(');
861 if (Node->getAlignment() != nullptr) {
863 Node->getAlignment()->printPretty(OS, nullptr, Policy, 0);
869 void OMPClausePrinter::VisitOMPCopyinClause(OMPCopyinClause *Node) {
870 if (!Node->varlist_empty()) {
872 VisitOMPClauseList(Node, '(');
877 void OMPClausePrinter::VisitOMPCopyprivateClause(OMPCopyprivateClause *Node) {
878 if (!Node->varlist_empty()) {
880 VisitOMPClauseList(Node, '(');
885 void OMPClausePrinter::VisitOMPFlushClause(OMPFlushClause *Node) {
886 if (!Node->varlist_empty()) {
887 VisitOMPClauseList(Node, '(');
892 void OMPClausePrinter::VisitOMPDependClause(OMPDependClause *Node) {
894 OS << getOpenMPSimpleClauseTypeName(Node->getClauseKind(),
895 Node->getDependencyKind());
896 if (!Node->varlist_empty()) {
898 VisitOMPClauseList(Node, ' ');
903 void OMPClausePrinter::VisitOMPMapClause(OMPMapClause *Node) {
904 if (!Node->varlist_empty()) {
906 if (Node->getMapType() != OMPC_MAP_unknown) {
907 if (Node->getMapTypeModifier() != OMPC_MAP_unknown) {
908 OS << getOpenMPSimpleClauseTypeName(OMPC_map,
909 Node->getMapTypeModifier());
912 OS << getOpenMPSimpleClauseTypeName(OMPC_map, Node->getMapType());
915 VisitOMPClauseList(Node, ' ');
920 void OMPClausePrinter::VisitOMPToClause(OMPToClause *Node) {
921 if (!Node->varlist_empty()) {
923 VisitOMPClauseList(Node, '(');
928 void OMPClausePrinter::VisitOMPFromClause(OMPFromClause *Node) {
929 if (!Node->varlist_empty()) {
931 VisitOMPClauseList(Node, '(');
936 void OMPClausePrinter::VisitOMPDistScheduleClause(OMPDistScheduleClause *Node) {
937 OS << "dist_schedule(" << getOpenMPSimpleClauseTypeName(
938 OMPC_dist_schedule, Node->getDistScheduleKind());
939 if (auto *E = Node->getChunkSize()) {
941 E->printPretty(OS, nullptr, Policy);
946 void OMPClausePrinter::VisitOMPDefaultmapClause(OMPDefaultmapClause *Node) {
948 OS << getOpenMPSimpleClauseTypeName(OMPC_defaultmap,
949 Node->getDefaultmapModifier());
951 OS << getOpenMPSimpleClauseTypeName(OMPC_defaultmap,
952 Node->getDefaultmapKind());
956 void OMPClausePrinter::VisitOMPUseDevicePtrClause(OMPUseDevicePtrClause *Node) {
957 if (!Node->varlist_empty()) {
958 OS << "use_device_ptr";
959 VisitOMPClauseList(Node, '(');
964 void OMPClausePrinter::VisitOMPIsDevicePtrClause(OMPIsDevicePtrClause *Node) {
965 if (!Node->varlist_empty()) {
966 OS << "is_device_ptr";
967 VisitOMPClauseList(Node, '(');
973 //===----------------------------------------------------------------------===//
974 // OpenMP directives printing methods
975 //===----------------------------------------------------------------------===//
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();
982 if (*I && !(*I)->isImplicit()) {
987 if (S->hasAssociatedStmt() && S->getAssociatedStmt()) {
988 assert(isa<CapturedStmt>(S->getAssociatedStmt()) &&
989 "Expected captured statement!");
990 Stmt *CS = cast<CapturedStmt>(S->getAssociatedStmt())->getCapturedStmt();
995 void StmtPrinter::VisitOMPParallelDirective(OMPParallelDirective *Node) {
996 Indent() << "#pragma omp parallel ";
997 PrintOMPExecutableDirective(Node);
1000 void StmtPrinter::VisitOMPSimdDirective(OMPSimdDirective *Node) {
1001 Indent() << "#pragma omp simd ";
1002 PrintOMPExecutableDirective(Node);
1005 void StmtPrinter::VisitOMPForDirective(OMPForDirective *Node) {
1006 Indent() << "#pragma omp for ";
1007 PrintOMPExecutableDirective(Node);
1010 void StmtPrinter::VisitOMPForSimdDirective(OMPForSimdDirective *Node) {
1011 Indent() << "#pragma omp for simd ";
1012 PrintOMPExecutableDirective(Node);
1015 void StmtPrinter::VisitOMPSectionsDirective(OMPSectionsDirective *Node) {
1016 Indent() << "#pragma omp sections ";
1017 PrintOMPExecutableDirective(Node);
1020 void StmtPrinter::VisitOMPSectionDirective(OMPSectionDirective *Node) {
1021 Indent() << "#pragma omp section";
1022 PrintOMPExecutableDirective(Node);
1025 void StmtPrinter::VisitOMPSingleDirective(OMPSingleDirective *Node) {
1026 Indent() << "#pragma omp single ";
1027 PrintOMPExecutableDirective(Node);
1030 void StmtPrinter::VisitOMPMasterDirective(OMPMasterDirective *Node) {
1031 Indent() << "#pragma omp master";
1032 PrintOMPExecutableDirective(Node);
1035 void StmtPrinter::VisitOMPCriticalDirective(OMPCriticalDirective *Node) {
1036 Indent() << "#pragma omp critical";
1037 if (Node->getDirectiveName().getName()) {
1039 Node->getDirectiveName().printName(OS);
1043 PrintOMPExecutableDirective(Node);
1046 void StmtPrinter::VisitOMPParallelForDirective(OMPParallelForDirective *Node) {
1047 Indent() << "#pragma omp parallel for ";
1048 PrintOMPExecutableDirective(Node);
1051 void StmtPrinter::VisitOMPParallelForSimdDirective(
1052 OMPParallelForSimdDirective *Node) {
1053 Indent() << "#pragma omp parallel for simd ";
1054 PrintOMPExecutableDirective(Node);
1057 void StmtPrinter::VisitOMPParallelSectionsDirective(
1058 OMPParallelSectionsDirective *Node) {
1059 Indent() << "#pragma omp parallel sections ";
1060 PrintOMPExecutableDirective(Node);
1063 void StmtPrinter::VisitOMPTaskDirective(OMPTaskDirective *Node) {
1064 Indent() << "#pragma omp task ";
1065 PrintOMPExecutableDirective(Node);
1068 void StmtPrinter::VisitOMPTaskyieldDirective(OMPTaskyieldDirective *Node) {
1069 Indent() << "#pragma omp taskyield";
1070 PrintOMPExecutableDirective(Node);
1073 void StmtPrinter::VisitOMPBarrierDirective(OMPBarrierDirective *Node) {
1074 Indent() << "#pragma omp barrier";
1075 PrintOMPExecutableDirective(Node);
1078 void StmtPrinter::VisitOMPTaskwaitDirective(OMPTaskwaitDirective *Node) {
1079 Indent() << "#pragma omp taskwait";
1080 PrintOMPExecutableDirective(Node);
1083 void StmtPrinter::VisitOMPTaskgroupDirective(OMPTaskgroupDirective *Node) {
1084 Indent() << "#pragma omp taskgroup";
1085 PrintOMPExecutableDirective(Node);
1088 void StmtPrinter::VisitOMPFlushDirective(OMPFlushDirective *Node) {
1089 Indent() << "#pragma omp flush ";
1090 PrintOMPExecutableDirective(Node);
1093 void StmtPrinter::VisitOMPOrderedDirective(OMPOrderedDirective *Node) {
1094 Indent() << "#pragma omp ordered ";
1095 PrintOMPExecutableDirective(Node);
1098 void StmtPrinter::VisitOMPAtomicDirective(OMPAtomicDirective *Node) {
1099 Indent() << "#pragma omp atomic ";
1100 PrintOMPExecutableDirective(Node);
1103 void StmtPrinter::VisitOMPTargetDirective(OMPTargetDirective *Node) {
1104 Indent() << "#pragma omp target ";
1105 PrintOMPExecutableDirective(Node);
1108 void StmtPrinter::VisitOMPTargetDataDirective(OMPTargetDataDirective *Node) {
1109 Indent() << "#pragma omp target data ";
1110 PrintOMPExecutableDirective(Node);
1113 void StmtPrinter::VisitOMPTargetEnterDataDirective(
1114 OMPTargetEnterDataDirective *Node) {
1115 Indent() << "#pragma omp target enter data ";
1116 PrintOMPExecutableDirective(Node);
1119 void StmtPrinter::VisitOMPTargetExitDataDirective(
1120 OMPTargetExitDataDirective *Node) {
1121 Indent() << "#pragma omp target exit data ";
1122 PrintOMPExecutableDirective(Node);
1125 void StmtPrinter::VisitOMPTargetParallelDirective(
1126 OMPTargetParallelDirective *Node) {
1127 Indent() << "#pragma omp target parallel ";
1128 PrintOMPExecutableDirective(Node);
1131 void StmtPrinter::VisitOMPTargetParallelForDirective(
1132 OMPTargetParallelForDirective *Node) {
1133 Indent() << "#pragma omp target parallel for ";
1134 PrintOMPExecutableDirective(Node);
1137 void StmtPrinter::VisitOMPTeamsDirective(OMPTeamsDirective *Node) {
1138 Indent() << "#pragma omp teams ";
1139 PrintOMPExecutableDirective(Node);
1142 void StmtPrinter::VisitOMPCancellationPointDirective(
1143 OMPCancellationPointDirective *Node) {
1144 Indent() << "#pragma omp cancellation point "
1145 << getOpenMPDirectiveName(Node->getCancelRegion());
1146 PrintOMPExecutableDirective(Node);
1149 void StmtPrinter::VisitOMPCancelDirective(OMPCancelDirective *Node) {
1150 Indent() << "#pragma omp cancel "
1151 << getOpenMPDirectiveName(Node->getCancelRegion()) << " ";
1152 PrintOMPExecutableDirective(Node);
1155 void StmtPrinter::VisitOMPTaskLoopDirective(OMPTaskLoopDirective *Node) {
1156 Indent() << "#pragma omp taskloop ";
1157 PrintOMPExecutableDirective(Node);
1160 void StmtPrinter::VisitOMPTaskLoopSimdDirective(
1161 OMPTaskLoopSimdDirective *Node) {
1162 Indent() << "#pragma omp taskloop simd ";
1163 PrintOMPExecutableDirective(Node);
1166 void StmtPrinter::VisitOMPDistributeDirective(OMPDistributeDirective *Node) {
1167 Indent() << "#pragma omp distribute ";
1168 PrintOMPExecutableDirective(Node);
1171 void StmtPrinter::VisitOMPTargetUpdateDirective(
1172 OMPTargetUpdateDirective *Node) {
1173 Indent() << "#pragma omp target update ";
1174 PrintOMPExecutableDirective(Node);
1177 void StmtPrinter::VisitOMPDistributeParallelForDirective(
1178 OMPDistributeParallelForDirective *Node) {
1179 Indent() << "#pragma omp distribute parallel for ";
1180 PrintOMPExecutableDirective(Node);
1183 void StmtPrinter::VisitOMPDistributeParallelForSimdDirective(
1184 OMPDistributeParallelForSimdDirective *Node) {
1185 Indent() << "#pragma omp distribute parallel for simd ";
1186 PrintOMPExecutableDirective(Node);
1189 void StmtPrinter::VisitOMPDistributeSimdDirective(
1190 OMPDistributeSimdDirective *Node) {
1191 Indent() << "#pragma omp distribute simd ";
1192 PrintOMPExecutableDirective(Node);
1195 void StmtPrinter::VisitOMPTargetParallelForSimdDirective(
1196 OMPTargetParallelForSimdDirective *Node) {
1197 Indent() << "#pragma omp target parallel for simd ";
1198 PrintOMPExecutableDirective(Node);
1201 void StmtPrinter::VisitOMPTargetSimdDirective(OMPTargetSimdDirective *Node) {
1202 Indent() << "#pragma omp target simd ";
1203 PrintOMPExecutableDirective(Node);
1206 void StmtPrinter::VisitOMPTeamsDistributeDirective(
1207 OMPTeamsDistributeDirective *Node) {
1208 Indent() << "#pragma omp teams distribute ";
1209 PrintOMPExecutableDirective(Node);
1212 void StmtPrinter::VisitOMPTeamsDistributeSimdDirective(
1213 OMPTeamsDistributeSimdDirective *Node) {
1214 Indent() << "#pragma omp teams distribute simd ";
1215 PrintOMPExecutableDirective(Node);
1218 void StmtPrinter::VisitOMPTeamsDistributeParallelForSimdDirective(
1219 OMPTeamsDistributeParallelForSimdDirective *Node) {
1220 Indent() << "#pragma omp teams distribute parallel for simd ";
1221 PrintOMPExecutableDirective(Node);
1224 void StmtPrinter::VisitOMPTeamsDistributeParallelForDirective(
1225 OMPTeamsDistributeParallelForDirective *Node) {
1226 Indent() << "#pragma omp teams distribute parallel for ";
1227 PrintOMPExecutableDirective(Node);
1230 void StmtPrinter::VisitOMPTargetTeamsDirective(OMPTargetTeamsDirective *Node) {
1231 Indent() << "#pragma omp target teams ";
1232 PrintOMPExecutableDirective(Node);
1235 void StmtPrinter::VisitOMPTargetTeamsDistributeDirective(
1236 OMPTargetTeamsDistributeDirective *Node) {
1237 Indent() << "#pragma omp target teams distribute ";
1238 PrintOMPExecutableDirective(Node);
1241 void StmtPrinter::VisitOMPTargetTeamsDistributeParallelForDirective(
1242 OMPTargetTeamsDistributeParallelForDirective *Node) {
1243 Indent() << "#pragma omp target teams distribute parallel for ";
1244 PrintOMPExecutableDirective(Node);
1247 void StmtPrinter::VisitOMPTargetTeamsDistributeParallelForSimdDirective(
1248 OMPTargetTeamsDistributeParallelForSimdDirective *Node) {
1249 Indent() << "#pragma omp target teams distribute parallel for simd ";
1250 PrintOMPExecutableDirective(Node);
1253 //===----------------------------------------------------------------------===//
1254 // Expr printing methods.
1255 //===----------------------------------------------------------------------===//
1257 void StmtPrinter::VisitDeclRefExpr(DeclRefExpr *Node) {
1258 if (auto *OCED = dyn_cast<OMPCapturedExprDecl>(Node->getDecl())) {
1259 OCED->getInit()->IgnoreImpCasts()->printPretty(OS, nullptr, Policy);
1262 if (NestedNameSpecifier *Qualifier = Node->getQualifier())
1263 Qualifier->print(OS, Policy);
1264 if (Node->hasTemplateKeyword())
1266 OS << Node->getNameInfo();
1267 if (Node->hasExplicitTemplateArgs())
1268 TemplateSpecializationType::PrintTemplateArgumentList(
1269 OS, Node->template_arguments(), Policy);
1272 void StmtPrinter::VisitDependentScopeDeclRefExpr(
1273 DependentScopeDeclRefExpr *Node) {
1274 if (NestedNameSpecifier *Qualifier = Node->getQualifier())
1275 Qualifier->print(OS, Policy);
1276 if (Node->hasTemplateKeyword())
1278 OS << Node->getNameInfo();
1279 if (Node->hasExplicitTemplateArgs())
1280 TemplateSpecializationType::PrintTemplateArgumentList(
1281 OS, Node->template_arguments(), Policy);
1284 void StmtPrinter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *Node) {
1285 if (Node->getQualifier())
1286 Node->getQualifier()->print(OS, Policy);
1287 if (Node->hasTemplateKeyword())
1289 OS << Node->getNameInfo();
1290 if (Node->hasExplicitTemplateArgs())
1291 TemplateSpecializationType::PrintTemplateArgumentList(
1292 OS, Node->template_arguments(), Policy);
1295 void StmtPrinter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *Node) {
1296 if (Node->getBase()) {
1297 PrintExpr(Node->getBase());
1298 OS << (Node->isArrow() ? "->" : ".");
1300 OS << *Node->getDecl();
1303 void StmtPrinter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *Node) {
1304 if (Node->isSuperReceiver())
1306 else if (Node->isObjectReceiver() && Node->getBase()) {
1307 PrintExpr(Node->getBase());
1309 } else if (Node->isClassReceiver() && Node->getClassReceiver()) {
1310 OS << Node->getClassReceiver()->getName() << ".";
1313 if (Node->isImplicitProperty())
1314 Node->getImplicitPropertyGetter()->getSelector().print(OS);
1316 OS << Node->getExplicitProperty()->getName();
1319 void StmtPrinter::VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr *Node) {
1321 PrintExpr(Node->getBaseExpr());
1323 PrintExpr(Node->getKeyExpr());
1327 void StmtPrinter::VisitPredefinedExpr(PredefinedExpr *Node) {
1328 OS << PredefinedExpr::getIdentTypeName(Node->getIdentType());
1331 void StmtPrinter::VisitCharacterLiteral(CharacterLiteral *Node) {
1332 unsigned value = Node->getValue();
1334 switch (Node->getKind()) {
1335 case CharacterLiteral::Ascii: break; // no prefix.
1336 case CharacterLiteral::Wide: OS << 'L'; break;
1337 case CharacterLiteral::UTF8: OS << "u8"; break;
1338 case CharacterLiteral::UTF16: OS << 'u'; break;
1339 case CharacterLiteral::UTF32: OS << 'U'; break;
1350 // TODO: K&R: the meaning of '\\a' is different in traditional C
1356 // Nonstandard escape sequence.
1376 // A character literal might be sign-extended, which
1377 // would result in an invalid \U escape sequence.
1378 // FIXME: multicharacter literals such as '\xFF\xFF\xFF\xFF'
1379 // are not correctly handled.
1380 if ((value & ~0xFFu) == ~0xFFu && Node->getKind() == CharacterLiteral::Ascii)
1382 if (value < 256 && isPrintable((unsigned char)value))
1383 OS << "'" << (char)value << "'";
1384 else if (value < 256)
1385 OS << "'\\x" << llvm::format("%02x", value) << "'";
1386 else if (value <= 0xFFFF)
1387 OS << "'\\u" << llvm::format("%04x", value) << "'";
1389 OS << "'\\U" << llvm::format("%08x", value) << "'";
1393 void StmtPrinter::VisitIntegerLiteral(IntegerLiteral *Node) {
1394 bool isSigned = Node->getType()->isSignedIntegerType();
1395 OS << Node->getValue().toString(10, isSigned);
1397 // Emit suffixes. Integer literals are always a builtin integer type.
1398 switch (Node->getType()->getAs<BuiltinType>()->getKind()) {
1399 default: llvm_unreachable("Unexpected type for integer literal!");
1400 case BuiltinType::Char_S:
1401 case BuiltinType::Char_U: OS << "i8"; break;
1402 case BuiltinType::UChar: OS << "Ui8"; break;
1403 case BuiltinType::Short: OS << "i16"; break;
1404 case BuiltinType::UShort: OS << "Ui16"; break;
1405 case BuiltinType::Int: break; // no suffix.
1406 case BuiltinType::UInt: OS << 'U'; break;
1407 case BuiltinType::Long: OS << 'L'; break;
1408 case BuiltinType::ULong: OS << "UL"; break;
1409 case BuiltinType::LongLong: OS << "LL"; break;
1410 case BuiltinType::ULongLong: OS << "ULL"; break;
1414 static void PrintFloatingLiteral(raw_ostream &OS, FloatingLiteral *Node,
1416 SmallString<16> Str;
1417 Node->getValue().toString(Str);
1419 if (Str.find_first_not_of("-0123456789") == StringRef::npos)
1420 OS << '.'; // Trailing dot in order to separate from ints.
1425 // Emit suffixes. Float literals are always a builtin float type.
1426 switch (Node->getType()->getAs<BuiltinType>()->getKind()) {
1427 default: llvm_unreachable("Unexpected type for float literal!");
1428 case BuiltinType::Half: break; // FIXME: suffix?
1429 case BuiltinType::Double: break; // no suffix.
1430 case BuiltinType::Float: OS << 'F'; break;
1431 case BuiltinType::LongDouble: OS << 'L'; break;
1432 case BuiltinType::Float128: OS << 'Q'; break;
1436 void StmtPrinter::VisitFloatingLiteral(FloatingLiteral *Node) {
1437 PrintFloatingLiteral(OS, Node, /*PrintSuffix=*/true);
1440 void StmtPrinter::VisitImaginaryLiteral(ImaginaryLiteral *Node) {
1441 PrintExpr(Node->getSubExpr());
1445 void StmtPrinter::VisitStringLiteral(StringLiteral *Str) {
1446 Str->outputString(OS);
1448 void StmtPrinter::VisitParenExpr(ParenExpr *Node) {
1450 PrintExpr(Node->getSubExpr());
1453 void StmtPrinter::VisitUnaryOperator(UnaryOperator *Node) {
1454 if (!Node->isPostfix()) {
1455 OS << UnaryOperator::getOpcodeStr(Node->getOpcode());
1457 // Print a space if this is an "identifier operator" like __real, or if
1458 // it might be concatenated incorrectly like '+'.
1459 switch (Node->getOpcode()) {
1468 if (isa<UnaryOperator>(Node->getSubExpr()))
1473 PrintExpr(Node->getSubExpr());
1475 if (Node->isPostfix())
1476 OS << UnaryOperator::getOpcodeStr(Node->getOpcode());
1479 void StmtPrinter::VisitOffsetOfExpr(OffsetOfExpr *Node) {
1480 OS << "__builtin_offsetof(";
1481 Node->getTypeSourceInfo()->getType().print(OS, Policy);
1483 bool PrintedSomething = false;
1484 for (unsigned i = 0, n = Node->getNumComponents(); i < n; ++i) {
1485 OffsetOfNode ON = Node->getComponent(i);
1486 if (ON.getKind() == OffsetOfNode::Array) {
1489 PrintExpr(Node->getIndexExpr(ON.getArrayExprIndex()));
1491 PrintedSomething = true;
1495 // Skip implicit base indirections.
1496 if (ON.getKind() == OffsetOfNode::Base)
1499 // Field or identifier node.
1500 IdentifierInfo *Id = ON.getFieldName();
1504 if (PrintedSomething)
1507 PrintedSomething = true;
1508 OS << Id->getName();
1513 void StmtPrinter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *Node){
1514 switch(Node->getKind()) {
1521 else if (Policy.UnderscoreAlignof)
1529 case UETT_OpenMPRequiredSimdAlign:
1530 OS << "__builtin_omp_required_simd_align";
1533 if (Node->isArgumentType()) {
1535 Node->getArgumentType().print(OS, Policy);
1539 PrintExpr(Node->getArgumentExpr());
1543 void StmtPrinter::VisitGenericSelectionExpr(GenericSelectionExpr *Node) {
1545 PrintExpr(Node->getControllingExpr());
1546 for (unsigned i = 0; i != Node->getNumAssocs(); ++i) {
1548 QualType T = Node->getAssocType(i);
1552 T.print(OS, Policy);
1554 PrintExpr(Node->getAssocExpr(i));
1559 void StmtPrinter::VisitArraySubscriptExpr(ArraySubscriptExpr *Node) {
1560 PrintExpr(Node->getLHS());
1562 PrintExpr(Node->getRHS());
1566 void StmtPrinter::VisitOMPArraySectionExpr(OMPArraySectionExpr *Node) {
1567 PrintExpr(Node->getBase());
1569 if (Node->getLowerBound())
1570 PrintExpr(Node->getLowerBound());
1571 if (Node->getColonLoc().isValid()) {
1573 if (Node->getLength())
1574 PrintExpr(Node->getLength());
1579 void StmtPrinter::PrintCallArgs(CallExpr *Call) {
1580 for (unsigned i = 0, e = Call->getNumArgs(); i != e; ++i) {
1581 if (isa<CXXDefaultArgExpr>(Call->getArg(i))) {
1582 // Don't print any defaulted arguments
1587 PrintExpr(Call->getArg(i));
1591 void StmtPrinter::VisitCallExpr(CallExpr *Call) {
1592 PrintExpr(Call->getCallee());
1594 PrintCallArgs(Call);
1597 void StmtPrinter::VisitMemberExpr(MemberExpr *Node) {
1598 // FIXME: Suppress printing implicit bases (like "this")
1599 PrintExpr(Node->getBase());
1601 MemberExpr *ParentMember = dyn_cast<MemberExpr>(Node->getBase());
1602 FieldDecl *ParentDecl = ParentMember
1603 ? dyn_cast<FieldDecl>(ParentMember->getMemberDecl()) : nullptr;
1605 if (!ParentDecl || !ParentDecl->isAnonymousStructOrUnion())
1606 OS << (Node->isArrow() ? "->" : ".");
1608 if (FieldDecl *FD = dyn_cast<FieldDecl>(Node->getMemberDecl()))
1609 if (FD->isAnonymousStructOrUnion())
1612 if (NestedNameSpecifier *Qualifier = Node->getQualifier())
1613 Qualifier->print(OS, Policy);
1614 if (Node->hasTemplateKeyword())
1616 OS << Node->getMemberNameInfo();
1617 if (Node->hasExplicitTemplateArgs())
1618 TemplateSpecializationType::PrintTemplateArgumentList(
1619 OS, Node->template_arguments(), Policy);
1621 void StmtPrinter::VisitObjCIsaExpr(ObjCIsaExpr *Node) {
1622 PrintExpr(Node->getBase());
1623 OS << (Node->isArrow() ? "->isa" : ".isa");
1626 void StmtPrinter::VisitExtVectorElementExpr(ExtVectorElementExpr *Node) {
1627 PrintExpr(Node->getBase());
1629 OS << Node->getAccessor().getName();
1631 void StmtPrinter::VisitCStyleCastExpr(CStyleCastExpr *Node) {
1633 Node->getTypeAsWritten().print(OS, Policy);
1635 PrintExpr(Node->getSubExpr());
1637 void StmtPrinter::VisitCompoundLiteralExpr(CompoundLiteralExpr *Node) {
1639 Node->getType().print(OS, Policy);
1641 PrintExpr(Node->getInitializer());
1643 void StmtPrinter::VisitImplicitCastExpr(ImplicitCastExpr *Node) {
1644 // No need to print anything, simply forward to the subexpression.
1645 PrintExpr(Node->getSubExpr());
1647 void StmtPrinter::VisitBinaryOperator(BinaryOperator *Node) {
1648 PrintExpr(Node->getLHS());
1649 OS << " " << BinaryOperator::getOpcodeStr(Node->getOpcode()) << " ";
1650 PrintExpr(Node->getRHS());
1652 void StmtPrinter::VisitCompoundAssignOperator(CompoundAssignOperator *Node) {
1653 PrintExpr(Node->getLHS());
1654 OS << " " << BinaryOperator::getOpcodeStr(Node->getOpcode()) << " ";
1655 PrintExpr(Node->getRHS());
1657 void StmtPrinter::VisitConditionalOperator(ConditionalOperator *Node) {
1658 PrintExpr(Node->getCond());
1660 PrintExpr(Node->getLHS());
1662 PrintExpr(Node->getRHS());
1668 StmtPrinter::VisitBinaryConditionalOperator(BinaryConditionalOperator *Node) {
1669 PrintExpr(Node->getCommon());
1671 PrintExpr(Node->getFalseExpr());
1673 void StmtPrinter::VisitAddrLabelExpr(AddrLabelExpr *Node) {
1674 OS << "&&" << Node->getLabel()->getName();
1677 void StmtPrinter::VisitStmtExpr(StmtExpr *E) {
1679 PrintRawCompoundStmt(E->getSubStmt());
1683 void StmtPrinter::VisitChooseExpr(ChooseExpr *Node) {
1684 OS << "__builtin_choose_expr(";
1685 PrintExpr(Node->getCond());
1687 PrintExpr(Node->getLHS());
1689 PrintExpr(Node->getRHS());
1693 void StmtPrinter::VisitGNUNullExpr(GNUNullExpr *) {
1697 void StmtPrinter::VisitShuffleVectorExpr(ShuffleVectorExpr *Node) {
1698 OS << "__builtin_shufflevector(";
1699 for (unsigned i = 0, e = Node->getNumSubExprs(); i != e; ++i) {
1701 PrintExpr(Node->getExpr(i));
1706 void StmtPrinter::VisitConvertVectorExpr(ConvertVectorExpr *Node) {
1707 OS << "__builtin_convertvector(";
1708 PrintExpr(Node->getSrcExpr());
1710 Node->getType().print(OS, Policy);
1714 void StmtPrinter::VisitInitListExpr(InitListExpr* Node) {
1715 if (Node->getSyntacticForm()) {
1716 Visit(Node->getSyntacticForm());
1721 for (unsigned i = 0, e = Node->getNumInits(); i != e; ++i) {
1723 if (Node->getInit(i))
1724 PrintExpr(Node->getInit(i));
1731 void StmtPrinter::VisitArrayInitLoopExpr(ArrayInitLoopExpr *Node) {
1732 // There's no way to express this expression in any of our supported
1733 // languages, so just emit something terse and (hopefully) clear.
1735 PrintExpr(Node->getSubExpr());
1739 void StmtPrinter::VisitArrayInitIndexExpr(ArrayInitIndexExpr *Node) {
1743 void StmtPrinter::VisitParenListExpr(ParenListExpr* Node) {
1745 for (unsigned i = 0, e = Node->getNumExprs(); i != e; ++i) {
1747 PrintExpr(Node->getExpr(i));
1752 void StmtPrinter::VisitDesignatedInitExpr(DesignatedInitExpr *Node) {
1753 bool NeedsEquals = true;
1754 for (const DesignatedInitExpr::Designator &D : Node->designators()) {
1755 if (D.isFieldDesignator()) {
1756 if (D.getDotLoc().isInvalid()) {
1757 if (IdentifierInfo *II = D.getFieldName()) {
1758 OS << II->getName() << ":";
1759 NeedsEquals = false;
1762 OS << "." << D.getFieldName()->getName();
1766 if (D.isArrayDesignator()) {
1767 PrintExpr(Node->getArrayIndex(D));
1769 PrintExpr(Node->getArrayRangeStart(D));
1771 PrintExpr(Node->getArrayRangeEnd(D));
1781 PrintExpr(Node->getInit());
1784 void StmtPrinter::VisitDesignatedInitUpdateExpr(
1785 DesignatedInitUpdateExpr *Node) {
1788 PrintExpr(Node->getBase());
1791 OS << "/*updater*/";
1792 PrintExpr(Node->getUpdater());
1796 void StmtPrinter::VisitNoInitExpr(NoInitExpr *Node) {
1797 OS << "/*no init*/";
1800 void StmtPrinter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *Node) {
1801 if (Node->getType()->getAsCXXRecordDecl()) {
1802 OS << "/*implicit*/";
1803 Node->getType().print(OS, Policy);
1806 OS << "/*implicit*/(";
1807 Node->getType().print(OS, Policy);
1809 if (Node->getType()->isRecordType())
1816 void StmtPrinter::VisitVAArgExpr(VAArgExpr *Node) {
1817 OS << "__builtin_va_arg(";
1818 PrintExpr(Node->getSubExpr());
1820 Node->getType().print(OS, Policy);
1824 void StmtPrinter::VisitPseudoObjectExpr(PseudoObjectExpr *Node) {
1825 PrintExpr(Node->getSyntacticForm());
1828 void StmtPrinter::VisitAtomicExpr(AtomicExpr *Node) {
1829 const char *Name = nullptr;
1830 switch (Node->getOp()) {
1831 #define BUILTIN(ID, TYPE, ATTRS)
1832 #define ATOMIC_BUILTIN(ID, TYPE, ATTRS) \
1833 case AtomicExpr::AO ## ID: \
1836 #include "clang/Basic/Builtins.def"
1840 // AtomicExpr stores its subexpressions in a permuted order.
1841 PrintExpr(Node->getPtr());
1842 if (Node->getOp() != AtomicExpr::AO__c11_atomic_load &&
1843 Node->getOp() != AtomicExpr::AO__atomic_load_n) {
1845 PrintExpr(Node->getVal1());
1847 if (Node->getOp() == AtomicExpr::AO__atomic_exchange ||
1848 Node->isCmpXChg()) {
1850 PrintExpr(Node->getVal2());
1852 if (Node->getOp() == AtomicExpr::AO__atomic_compare_exchange ||
1853 Node->getOp() == AtomicExpr::AO__atomic_compare_exchange_n) {
1855 PrintExpr(Node->getWeak());
1857 if (Node->getOp() != AtomicExpr::AO__c11_atomic_init) {
1859 PrintExpr(Node->getOrder());
1861 if (Node->isCmpXChg()) {
1863 PrintExpr(Node->getOrderFail());
1869 void StmtPrinter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *Node) {
1870 const char *OpStrings[NUM_OVERLOADED_OPERATORS] = {
1872 #define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
1874 #include "clang/Basic/OperatorKinds.def"
1877 OverloadedOperatorKind Kind = Node->getOperator();
1878 if (Kind == OO_PlusPlus || Kind == OO_MinusMinus) {
1879 if (Node->getNumArgs() == 1) {
1880 OS << OpStrings[Kind] << ' ';
1881 PrintExpr(Node->getArg(0));
1883 PrintExpr(Node->getArg(0));
1884 OS << ' ' << OpStrings[Kind];
1886 } else if (Kind == OO_Arrow) {
1887 PrintExpr(Node->getArg(0));
1888 } else if (Kind == OO_Call) {
1889 PrintExpr(Node->getArg(0));
1891 for (unsigned ArgIdx = 1; ArgIdx < Node->getNumArgs(); ++ArgIdx) {
1894 if (!isa<CXXDefaultArgExpr>(Node->getArg(ArgIdx)))
1895 PrintExpr(Node->getArg(ArgIdx));
1898 } else if (Kind == OO_Subscript) {
1899 PrintExpr(Node->getArg(0));
1901 PrintExpr(Node->getArg(1));
1903 } else if (Node->getNumArgs() == 1) {
1904 OS << OpStrings[Kind] << ' ';
1905 PrintExpr(Node->getArg(0));
1906 } else if (Node->getNumArgs() == 2) {
1907 PrintExpr(Node->getArg(0));
1908 OS << ' ' << OpStrings[Kind] << ' ';
1909 PrintExpr(Node->getArg(1));
1911 llvm_unreachable("unknown overloaded operator");
1915 void StmtPrinter::VisitCXXMemberCallExpr(CXXMemberCallExpr *Node) {
1916 // If we have a conversion operator call only print the argument.
1917 CXXMethodDecl *MD = Node->getMethodDecl();
1918 if (MD && isa<CXXConversionDecl>(MD)) {
1919 PrintExpr(Node->getImplicitObjectArgument());
1922 VisitCallExpr(cast<CallExpr>(Node));
1925 void StmtPrinter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *Node) {
1926 PrintExpr(Node->getCallee());
1928 PrintCallArgs(Node->getConfig());
1930 PrintCallArgs(Node);
1934 void StmtPrinter::VisitCXXNamedCastExpr(CXXNamedCastExpr *Node) {
1935 OS << Node->getCastName() << '<';
1936 Node->getTypeAsWritten().print(OS, Policy);
1938 PrintExpr(Node->getSubExpr());
1942 void StmtPrinter::VisitCXXStaticCastExpr(CXXStaticCastExpr *Node) {
1943 VisitCXXNamedCastExpr(Node);
1946 void StmtPrinter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *Node) {
1947 VisitCXXNamedCastExpr(Node);
1950 void StmtPrinter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *Node) {
1951 VisitCXXNamedCastExpr(Node);
1954 void StmtPrinter::VisitCXXConstCastExpr(CXXConstCastExpr *Node) {
1955 VisitCXXNamedCastExpr(Node);
1958 void StmtPrinter::VisitCXXTypeidExpr(CXXTypeidExpr *Node) {
1960 if (Node->isTypeOperand()) {
1961 Node->getTypeOperandSourceInfo()->getType().print(OS, Policy);
1963 PrintExpr(Node->getExprOperand());
1968 void StmtPrinter::VisitCXXUuidofExpr(CXXUuidofExpr *Node) {
1970 if (Node->isTypeOperand()) {
1971 Node->getTypeOperandSourceInfo()->getType().print(OS, Policy);
1973 PrintExpr(Node->getExprOperand());
1978 void StmtPrinter::VisitMSPropertyRefExpr(MSPropertyRefExpr *Node) {
1979 PrintExpr(Node->getBaseExpr());
1980 if (Node->isArrow())
1984 if (NestedNameSpecifier *Qualifier =
1985 Node->getQualifierLoc().getNestedNameSpecifier())
1986 Qualifier->print(OS, Policy);
1987 OS << Node->getPropertyDecl()->getDeclName();
1990 void StmtPrinter::VisitMSPropertySubscriptExpr(MSPropertySubscriptExpr *Node) {
1991 PrintExpr(Node->getBase());
1993 PrintExpr(Node->getIdx());
1997 void StmtPrinter::VisitUserDefinedLiteral(UserDefinedLiteral *Node) {
1998 switch (Node->getLiteralOperatorKind()) {
1999 case UserDefinedLiteral::LOK_Raw:
2000 OS << cast<StringLiteral>(Node->getArg(0)->IgnoreImpCasts())->getString();
2002 case UserDefinedLiteral::LOK_Template: {
2003 DeclRefExpr *DRE = cast<DeclRefExpr>(Node->getCallee()->IgnoreImpCasts());
2004 const TemplateArgumentList *Args =
2005 cast<FunctionDecl>(DRE->getDecl())->getTemplateSpecializationArgs();
2008 if (Args->size() != 1) {
2009 OS << "operator\"\"" << Node->getUDSuffix()->getName();
2010 TemplateSpecializationType::PrintTemplateArgumentList(
2011 OS, Args->asArray(), Policy);
2016 const TemplateArgument &Pack = Args->get(0);
2017 for (const auto &P : Pack.pack_elements()) {
2018 char C = (char)P.getAsIntegral().getZExtValue();
2023 case UserDefinedLiteral::LOK_Integer: {
2024 // Print integer literal without suffix.
2025 IntegerLiteral *Int = cast<IntegerLiteral>(Node->getCookedLiteral());
2026 OS << Int->getValue().toString(10, /*isSigned*/false);
2029 case UserDefinedLiteral::LOK_Floating: {
2030 // Print floating literal without suffix.
2031 FloatingLiteral *Float = cast<FloatingLiteral>(Node->getCookedLiteral());
2032 PrintFloatingLiteral(OS, Float, /*PrintSuffix=*/false);
2035 case UserDefinedLiteral::LOK_String:
2036 case UserDefinedLiteral::LOK_Character:
2037 PrintExpr(Node->getCookedLiteral());
2040 OS << Node->getUDSuffix()->getName();
2043 void StmtPrinter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *Node) {
2044 OS << (Node->getValue() ? "true" : "false");
2047 void StmtPrinter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *Node) {
2051 void StmtPrinter::VisitCXXThisExpr(CXXThisExpr *Node) {
2055 void StmtPrinter::VisitCXXThrowExpr(CXXThrowExpr *Node) {
2056 if (!Node->getSubExpr())
2060 PrintExpr(Node->getSubExpr());
2064 void StmtPrinter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *Node) {
2065 // Nothing to print: we picked up the default argument.
2068 void StmtPrinter::VisitCXXDefaultInitExpr(CXXDefaultInitExpr *Node) {
2069 // Nothing to print: we picked up the default initializer.
2072 void StmtPrinter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *Node) {
2073 Node->getType().print(OS, Policy);
2074 // If there are no parens, this is list-initialization, and the braces are
2075 // part of the syntax of the inner construct.
2076 if (Node->getLParenLoc().isValid())
2078 PrintExpr(Node->getSubExpr());
2079 if (Node->getLParenLoc().isValid())
2083 void StmtPrinter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *Node) {
2084 PrintExpr(Node->getSubExpr());
2087 void StmtPrinter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *Node) {
2088 Node->getType().print(OS, Policy);
2089 if (Node->isStdInitListInitialization())
2090 /* Nothing to do; braces are part of creating the std::initializer_list. */;
2091 else if (Node->isListInitialization())
2095 for (CXXTemporaryObjectExpr::arg_iterator Arg = Node->arg_begin(),
2096 ArgEnd = Node->arg_end();
2097 Arg != ArgEnd; ++Arg) {
2098 if ((*Arg)->isDefaultArgument())
2100 if (Arg != Node->arg_begin())
2104 if (Node->isStdInitListInitialization())
2106 else if (Node->isListInitialization())
2112 void StmtPrinter::VisitLambdaExpr(LambdaExpr *Node) {
2114 bool NeedComma = false;
2115 switch (Node->getCaptureDefault()) {
2129 for (LambdaExpr::capture_iterator C = Node->explicit_capture_begin(),
2130 CEnd = Node->explicit_capture_end();
2137 switch (C->getCaptureKind()) {
2145 if (Node->getCaptureDefault() != LCD_ByRef || Node->isInitCapture(C))
2147 OS << C->getCapturedVar()->getName();
2151 OS << C->getCapturedVar()->getName();
2154 llvm_unreachable("VLA type in explicit captures.");
2157 if (Node->isInitCapture(C))
2158 PrintExpr(C->getCapturedVar()->getInit());
2162 if (Node->hasExplicitParameters()) {
2164 CXXMethodDecl *Method = Node->getCallOperator();
2166 for (auto P : Method->parameters()) {
2172 std::string ParamStr = P->getNameAsString();
2173 P->getOriginalType().print(OS, Policy, ParamStr);
2175 if (Method->isVariadic()) {
2182 if (Node->isMutable())
2185 const FunctionProtoType *Proto
2186 = Method->getType()->getAs<FunctionProtoType>();
2187 Proto->printExceptionSpecification(OS, Policy);
2189 // FIXME: Attributes
2191 // Print the trailing return type if it was specified in the source.
2192 if (Node->hasExplicitResultType()) {
2194 Proto->getReturnType().print(OS, Policy);
2199 CompoundStmt *Body = Node->getBody();
2204 void StmtPrinter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *Node) {
2205 if (TypeSourceInfo *TSInfo = Node->getTypeSourceInfo())
2206 TSInfo->getType().print(OS, Policy);
2208 Node->getType().print(OS, Policy);
2212 void StmtPrinter::VisitCXXNewExpr(CXXNewExpr *E) {
2213 if (E->isGlobalNew())
2216 unsigned NumPlace = E->getNumPlacementArgs();
2217 if (NumPlace > 0 && !isa<CXXDefaultArgExpr>(E->getPlacementArg(0))) {
2219 PrintExpr(E->getPlacementArg(0));
2220 for (unsigned i = 1; i < NumPlace; ++i) {
2221 if (isa<CXXDefaultArgExpr>(E->getPlacementArg(i)))
2224 PrintExpr(E->getPlacementArg(i));
2228 if (E->isParenTypeId())
2231 if (Expr *Size = E->getArraySize()) {
2232 llvm::raw_string_ostream s(TypeS);
2234 Size->printPretty(s, Helper, Policy);
2237 E->getAllocatedType().print(OS, Policy, TypeS);
2238 if (E->isParenTypeId())
2241 CXXNewExpr::InitializationStyle InitStyle = E->getInitializationStyle();
2243 if (InitStyle == CXXNewExpr::CallInit)
2245 PrintExpr(E->getInitializer());
2246 if (InitStyle == CXXNewExpr::CallInit)
2251 void StmtPrinter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
2252 if (E->isGlobalDelete())
2255 if (E->isArrayForm())
2257 PrintExpr(E->getArgument());
2260 void StmtPrinter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
2261 PrintExpr(E->getBase());
2266 if (E->getQualifier())
2267 E->getQualifier()->print(OS, Policy);
2270 if (IdentifierInfo *II = E->getDestroyedTypeIdentifier())
2271 OS << II->getName();
2273 E->getDestroyedType().print(OS, Policy);
2276 void StmtPrinter::VisitCXXConstructExpr(CXXConstructExpr *E) {
2277 if (E->isListInitialization() && !E->isStdInitListInitialization())
2280 for (unsigned i = 0, e = E->getNumArgs(); i != e; ++i) {
2281 if (isa<CXXDefaultArgExpr>(E->getArg(i))) {
2282 // Don't print any defaulted arguments
2287 PrintExpr(E->getArg(i));
2290 if (E->isListInitialization() && !E->isStdInitListInitialization())
2294 void StmtPrinter::VisitCXXInheritedCtorInitExpr(CXXInheritedCtorInitExpr *E) {
2295 // Parens are printed by the surrounding context.
2296 OS << "<forwarded>";
2299 void StmtPrinter::VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E) {
2300 PrintExpr(E->getSubExpr());
2303 void StmtPrinter::VisitExprWithCleanups(ExprWithCleanups *E) {
2304 // Just forward to the subexpression.
2305 PrintExpr(E->getSubExpr());
2309 StmtPrinter::VisitCXXUnresolvedConstructExpr(
2310 CXXUnresolvedConstructExpr *Node) {
2311 Node->getTypeAsWritten().print(OS, Policy);
2313 for (CXXUnresolvedConstructExpr::arg_iterator Arg = Node->arg_begin(),
2314 ArgEnd = Node->arg_end();
2315 Arg != ArgEnd; ++Arg) {
2316 if (Arg != Node->arg_begin())
2323 void StmtPrinter::VisitCXXDependentScopeMemberExpr(
2324 CXXDependentScopeMemberExpr *Node) {
2325 if (!Node->isImplicitAccess()) {
2326 PrintExpr(Node->getBase());
2327 OS << (Node->isArrow() ? "->" : ".");
2329 if (NestedNameSpecifier *Qualifier = Node->getQualifier())
2330 Qualifier->print(OS, Policy);
2331 if (Node->hasTemplateKeyword())
2333 OS << Node->getMemberNameInfo();
2334 if (Node->hasExplicitTemplateArgs())
2335 TemplateSpecializationType::PrintTemplateArgumentList(
2336 OS, Node->template_arguments(), Policy);
2339 void StmtPrinter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *Node) {
2340 if (!Node->isImplicitAccess()) {
2341 PrintExpr(Node->getBase());
2342 OS << (Node->isArrow() ? "->" : ".");
2344 if (NestedNameSpecifier *Qualifier = Node->getQualifier())
2345 Qualifier->print(OS, Policy);
2346 if (Node->hasTemplateKeyword())
2348 OS << Node->getMemberNameInfo();
2349 if (Node->hasExplicitTemplateArgs())
2350 TemplateSpecializationType::PrintTemplateArgumentList(
2351 OS, Node->template_arguments(), Policy);
2354 static const char *getTypeTraitName(TypeTrait TT) {
2356 #define TYPE_TRAIT_1(Spelling, Name, Key) \
2357 case clang::UTT_##Name: return #Spelling;
2358 #define TYPE_TRAIT_2(Spelling, Name, Key) \
2359 case clang::BTT_##Name: return #Spelling;
2360 #define TYPE_TRAIT_N(Spelling, Name, Key) \
2361 case clang::TT_##Name: return #Spelling;
2362 #include "clang/Basic/TokenKinds.def"
2364 llvm_unreachable("Type trait not covered by switch");
2367 static const char *getTypeTraitName(ArrayTypeTrait ATT) {
2369 case ATT_ArrayRank: return "__array_rank";
2370 case ATT_ArrayExtent: return "__array_extent";
2372 llvm_unreachable("Array type trait not covered by switch");
2375 static const char *getExpressionTraitName(ExpressionTrait ET) {
2377 case ET_IsLValueExpr: return "__is_lvalue_expr";
2378 case ET_IsRValueExpr: return "__is_rvalue_expr";
2380 llvm_unreachable("Expression type trait not covered by switch");
2383 void StmtPrinter::VisitTypeTraitExpr(TypeTraitExpr *E) {
2384 OS << getTypeTraitName(E->getTrait()) << "(";
2385 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I) {
2388 E->getArg(I)->getType().print(OS, Policy);
2393 void StmtPrinter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
2394 OS << getTypeTraitName(E->getTrait()) << '(';
2395 E->getQueriedType().print(OS, Policy);
2399 void StmtPrinter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
2400 OS << getExpressionTraitName(E->getTrait()) << '(';
2401 PrintExpr(E->getQueriedExpression());
2405 void StmtPrinter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
2407 PrintExpr(E->getOperand());
2411 void StmtPrinter::VisitPackExpansionExpr(PackExpansionExpr *E) {
2412 PrintExpr(E->getPattern());
2416 void StmtPrinter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
2417 OS << "sizeof...(" << *E->getPack() << ")";
2420 void StmtPrinter::VisitSubstNonTypeTemplateParmPackExpr(
2421 SubstNonTypeTemplateParmPackExpr *Node) {
2422 OS << *Node->getParameterPack();
2425 void StmtPrinter::VisitSubstNonTypeTemplateParmExpr(
2426 SubstNonTypeTemplateParmExpr *Node) {
2427 Visit(Node->getReplacement());
2430 void StmtPrinter::VisitFunctionParmPackExpr(FunctionParmPackExpr *E) {
2431 OS << *E->getParameterPack();
2434 void StmtPrinter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *Node){
2435 PrintExpr(Node->GetTemporaryExpr());
2438 void StmtPrinter::VisitCXXFoldExpr(CXXFoldExpr *E) {
2441 PrintExpr(E->getLHS());
2442 OS << " " << BinaryOperator::getOpcodeStr(E->getOperator()) << " ";
2446 OS << " " << BinaryOperator::getOpcodeStr(E->getOperator()) << " ";
2447 PrintExpr(E->getRHS());
2452 // C++ Coroutines TS
2454 void StmtPrinter::VisitCoroutineBodyStmt(CoroutineBodyStmt *S) {
2455 Visit(S->getBody());
2458 void StmtPrinter::VisitCoreturnStmt(CoreturnStmt *S) {
2460 if (S->getOperand()) {
2462 Visit(S->getOperand());
2467 void StmtPrinter::VisitCoawaitExpr(CoawaitExpr *S) {
2469 PrintExpr(S->getOperand());
2472 void StmtPrinter::VisitCoyieldExpr(CoyieldExpr *S) {
2474 PrintExpr(S->getOperand());
2479 void StmtPrinter::VisitObjCStringLiteral(ObjCStringLiteral *Node) {
2481 VisitStringLiteral(Node->getString());
2484 void StmtPrinter::VisitObjCBoxedExpr(ObjCBoxedExpr *E) {
2486 Visit(E->getSubExpr());
2489 void StmtPrinter::VisitObjCArrayLiteral(ObjCArrayLiteral *E) {
2491 ObjCArrayLiteral::child_range Ch = E->children();
2492 for (auto I = Ch.begin(), E = Ch.end(); I != E; ++I) {
2493 if (I != Ch.begin())
2500 void StmtPrinter::VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) {
2502 for (unsigned I = 0, N = E->getNumElements(); I != N; ++I) {
2506 ObjCDictionaryElement Element = E->getKeyValueElement(I);
2509 Visit(Element.Value);
2510 if (Element.isPackExpansion())
2516 void StmtPrinter::VisitObjCEncodeExpr(ObjCEncodeExpr *Node) {
2518 Node->getEncodedType().print(OS, Policy);
2522 void StmtPrinter::VisitObjCSelectorExpr(ObjCSelectorExpr *Node) {
2524 Node->getSelector().print(OS);
2528 void StmtPrinter::VisitObjCProtocolExpr(ObjCProtocolExpr *Node) {
2529 OS << "@protocol(" << *Node->getProtocol() << ')';
2532 void StmtPrinter::VisitObjCMessageExpr(ObjCMessageExpr *Mess) {
2534 switch (Mess->getReceiverKind()) {
2535 case ObjCMessageExpr::Instance:
2536 PrintExpr(Mess->getInstanceReceiver());
2539 case ObjCMessageExpr::Class:
2540 Mess->getClassReceiver().print(OS, Policy);
2543 case ObjCMessageExpr::SuperInstance:
2544 case ObjCMessageExpr::SuperClass:
2550 Selector selector = Mess->getSelector();
2551 if (selector.isUnarySelector()) {
2552 OS << selector.getNameForSlot(0);
2554 for (unsigned i = 0, e = Mess->getNumArgs(); i != e; ++i) {
2555 if (i < selector.getNumArgs()) {
2556 if (i > 0) OS << ' ';
2557 if (selector.getIdentifierInfoForSlot(i))
2558 OS << selector.getIdentifierInfoForSlot(i)->getName() << ':';
2562 else OS << ", "; // Handle variadic methods.
2564 PrintExpr(Mess->getArg(i));
2570 void StmtPrinter::VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr *Node) {
2571 OS << (Node->getValue() ? "__objc_yes" : "__objc_no");
2575 StmtPrinter::VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) {
2576 PrintExpr(E->getSubExpr());
2580 StmtPrinter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) {
2581 OS << '(' << E->getBridgeKindName();
2582 E->getType().print(OS, Policy);
2584 PrintExpr(E->getSubExpr());
2587 void StmtPrinter::VisitBlockExpr(BlockExpr *Node) {
2588 BlockDecl *BD = Node->getBlockDecl();
2591 const FunctionType *AFT = Node->getFunctionType();
2593 if (isa<FunctionNoProtoType>(AFT)) {
2595 } else if (!BD->param_empty() || cast<FunctionProtoType>(AFT)->isVariadic()) {
2597 for (BlockDecl::param_iterator AI = BD->param_begin(),
2598 E = BD->param_end(); AI != E; ++AI) {
2599 if (AI != BD->param_begin()) OS << ", ";
2600 std::string ParamStr = (*AI)->getNameAsString();
2601 (*AI)->getType().print(OS, Policy, ParamStr);
2604 const FunctionProtoType *FT = cast<FunctionProtoType>(AFT);
2605 if (FT->isVariadic()) {
2606 if (!BD->param_empty()) OS << ", ";
2614 void StmtPrinter::VisitOpaqueValueExpr(OpaqueValueExpr *Node) {
2615 PrintExpr(Node->getSourceExpr());
2618 void StmtPrinter::VisitTypoExpr(TypoExpr *Node) {
2619 // TODO: Print something reasonable for a TypoExpr, if necessary.
2620 llvm_unreachable("Cannot print TypoExpr nodes");
2623 void StmtPrinter::VisitAsTypeExpr(AsTypeExpr *Node) {
2624 OS << "__builtin_astype(";
2625 PrintExpr(Node->getSrcExpr());
2627 Node->getType().print(OS, Policy);
2631 //===----------------------------------------------------------------------===//
2632 // Stmt method implementations
2633 //===----------------------------------------------------------------------===//
2635 void Stmt::dumpPretty(const ASTContext &Context) const {
2636 printPretty(llvm::errs(), nullptr, PrintingPolicy(Context.getLangOpts()));
2639 void Stmt::printPretty(raw_ostream &OS,
2640 PrinterHelper *Helper,
2641 const PrintingPolicy &Policy,
2642 unsigned Indentation) const {
2643 StmtPrinter P(OS, Helper, Policy, Indentation);
2644 P.Visit(const_cast<Stmt*>(this));
2647 //===----------------------------------------------------------------------===//
2649 //===----------------------------------------------------------------------===//
2651 // Implement virtual destructor.
2652 PrinterHelper::~PrinterHelper() {}