| //===- StmtPrinter.cpp - Printing implementation for Stmt ASTs ------------===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // This file implements the Stmt::dumpPretty/Stmt::printPretty methods, which |
| // pretty print the AST back out to C code. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "clang/AST/ASTContext.h" |
| #include "clang/AST/Attr.h" |
| #include "clang/AST/Decl.h" |
| #include "clang/AST/DeclBase.h" |
| #include "clang/AST/DeclCXX.h" |
| #include "clang/AST/DeclObjC.h" |
| #include "clang/AST/DeclOpenMP.h" |
| #include "clang/AST/DeclTemplate.h" |
| #include "clang/AST/Expr.h" |
| #include "clang/AST/ExprCXX.h" |
| #include "clang/AST/ExprObjC.h" |
| #include "clang/AST/ExprOpenMP.h" |
| #include "clang/AST/NestedNameSpecifier.h" |
| #include "clang/AST/OpenMPClause.h" |
| #include "clang/AST/PrettyPrinter.h" |
| #include "clang/AST/Stmt.h" |
| #include "clang/AST/StmtCXX.h" |
| #include "clang/AST/StmtObjC.h" |
| #include "clang/AST/StmtOpenMP.h" |
| #include "clang/AST/StmtVisitor.h" |
| #include "clang/AST/TemplateBase.h" |
| #include "clang/AST/Type.h" |
| #include "clang/Basic/CharInfo.h" |
| #include "clang/Basic/ExpressionTraits.h" |
| #include "clang/Basic/IdentifierTable.h" |
| #include "clang/Basic/LLVM.h" |
| #include "clang/Basic/Lambda.h" |
| #include "clang/Basic/OpenMPKinds.h" |
| #include "clang/Basic/OperatorKinds.h" |
| #include "clang/Basic/SourceLocation.h" |
| #include "clang/Basic/TypeTraits.h" |
| #include "clang/Lex/Lexer.h" |
| #include "llvm/ADT/ArrayRef.h" |
| #include "llvm/ADT/SmallString.h" |
| #include "llvm/ADT/SmallVector.h" |
| #include "llvm/ADT/StringRef.h" |
| #include "llvm/Support/Casting.h" |
| #include "llvm/Support/Compiler.h" |
| #include "llvm/Support/ErrorHandling.h" |
| #include "llvm/Support/Format.h" |
| #include "llvm/Support/raw_ostream.h" |
| #include <cassert> |
| #include <string> |
| |
| using namespace clang; |
| |
| //===----------------------------------------------------------------------===// |
| // StmtPrinter Visitor |
| //===----------------------------------------------------------------------===// |
| |
| namespace { |
| |
| class StmtPrinter : public StmtVisitor<StmtPrinter> { |
| raw_ostream &OS; |
| unsigned IndentLevel; |
| PrinterHelper* Helper; |
| PrintingPolicy Policy; |
| const ASTContext *Context; |
| |
| public: |
| StmtPrinter(raw_ostream &os, PrinterHelper *helper, |
| const PrintingPolicy &Policy, unsigned Indentation = 0, |
| const ASTContext *Context = nullptr) |
| : OS(os), IndentLevel(Indentation), Helper(helper), Policy(Policy), |
| Context(Context) {} |
| |
| void PrintStmt(Stmt *S) { |
| PrintStmt(S, Policy.Indentation); |
| } |
| |
| void PrintStmt(Stmt *S, int SubIndent) { |
| IndentLevel += SubIndent; |
| if (S && isa<Expr>(S)) { |
| // If this is an expr used in a stmt context, indent and newline it. |
| Indent(); |
| Visit(S); |
| OS << ";\n"; |
| } else if (S) { |
| Visit(S); |
| } else { |
| Indent() << "<<<NULL STATEMENT>>>\n"; |
| } |
| IndentLevel -= SubIndent; |
| } |
| |
| void PrintRawCompoundStmt(CompoundStmt *S); |
| void PrintRawDecl(Decl *D); |
| void PrintRawDeclStmt(const DeclStmt *S); |
| void PrintRawIfStmt(IfStmt *If); |
| void PrintRawCXXCatchStmt(CXXCatchStmt *Catch); |
| void PrintCallArgs(CallExpr *E); |
| void PrintRawSEHExceptHandler(SEHExceptStmt *S); |
| void PrintRawSEHFinallyStmt(SEHFinallyStmt *S); |
| void PrintOMPExecutableDirective(OMPExecutableDirective *S, |
| bool ForceNoStmt = false); |
| |
| void PrintExpr(Expr *E) { |
| if (E) |
| Visit(E); |
| else |
| OS << "<null expr>"; |
| } |
| |
| raw_ostream &Indent(int Delta = 0) { |
| for (int i = 0, e = IndentLevel+Delta; i < e; ++i) |
| OS << " "; |
| return OS; |
| } |
| |
| void Visit(Stmt* S) { |
| if (Helper && Helper->handledStmt(S,OS)) |
| return; |
| else StmtVisitor<StmtPrinter>::Visit(S); |
| } |
| |
| void VisitStmt(Stmt *Node) LLVM_ATTRIBUTE_UNUSED { |
| Indent() << "<<unknown stmt type>>\n"; |
| } |
| |
| void VisitExpr(Expr *Node) LLVM_ATTRIBUTE_UNUSED { |
| OS << "<<unknown expr type>>"; |
| } |
| |
| void VisitCXXNamedCastExpr(CXXNamedCastExpr *Node); |
| |
| #define ABSTRACT_STMT(CLASS) |
| #define STMT(CLASS, PARENT) \ |
| void Visit##CLASS(CLASS *Node); |
| #include "clang/AST/StmtNodes.inc" |
| }; |
| |
| } // namespace |
| |
| //===----------------------------------------------------------------------===// |
| // Stmt printing methods. |
| //===----------------------------------------------------------------------===// |
| |
| /// PrintRawCompoundStmt - Print a compound stmt without indenting the {, and |
| /// with no newline after the }. |
| void StmtPrinter::PrintRawCompoundStmt(CompoundStmt *Node) { |
| OS << "{\n"; |
| for (auto *I : Node->body()) |
| PrintStmt(I); |
| |
| Indent() << "}"; |
| } |
| |
| void StmtPrinter::PrintRawDecl(Decl *D) { |
| D->print(OS, Policy, IndentLevel); |
| } |
| |
| void StmtPrinter::PrintRawDeclStmt(const DeclStmt *S) { |
| SmallVector<Decl *, 2> Decls(S->decls()); |
| Decl::printGroup(Decls.data(), Decls.size(), OS, Policy, IndentLevel); |
| } |
| |
| void StmtPrinter::VisitNullStmt(NullStmt *Node) { |
| Indent() << ";\n"; |
| } |
| |
| void StmtPrinter::VisitDeclStmt(DeclStmt *Node) { |
| Indent(); |
| PrintRawDeclStmt(Node); |
| OS << ";\n"; |
| } |
| |
| void StmtPrinter::VisitCompoundStmt(CompoundStmt *Node) { |
| Indent(); |
| PrintRawCompoundStmt(Node); |
| OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitCaseStmt(CaseStmt *Node) { |
| Indent(-1) << "case "; |
| PrintExpr(Node->getLHS()); |
| if (Node->getRHS()) { |
| OS << " ... "; |
| PrintExpr(Node->getRHS()); |
| } |
| OS << ":\n"; |
| |
| PrintStmt(Node->getSubStmt(), 0); |
| } |
| |
| void StmtPrinter::VisitDefaultStmt(DefaultStmt *Node) { |
| Indent(-1) << "default:\n"; |
| PrintStmt(Node->getSubStmt(), 0); |
| } |
| |
| void StmtPrinter::VisitLabelStmt(LabelStmt *Node) { |
| Indent(-1) << Node->getName() << ":\n"; |
| PrintStmt(Node->getSubStmt(), 0); |
| } |
| |
| void StmtPrinter::VisitAttributedStmt(AttributedStmt *Node) { |
| for (const auto *Attr : Node->getAttrs()) { |
| Attr->printPretty(OS, Policy); |
| } |
| |
| PrintStmt(Node->getSubStmt(), 0); |
| } |
| |
| void StmtPrinter::PrintRawIfStmt(IfStmt *If) { |
| OS << "if ("; |
| if (const DeclStmt *DS = If->getConditionVariableDeclStmt()) |
| PrintRawDeclStmt(DS); |
| else |
| PrintExpr(If->getCond()); |
| OS << ')'; |
| |
| if (auto *CS = dyn_cast<CompoundStmt>(If->getThen())) { |
| OS << ' '; |
| PrintRawCompoundStmt(CS); |
| OS << (If->getElse() ? ' ' : '\n'); |
| } else { |
| OS << '\n'; |
| PrintStmt(If->getThen()); |
| if (If->getElse()) Indent(); |
| } |
| |
| if (Stmt *Else = If->getElse()) { |
| OS << "else"; |
| |
| if (auto *CS = dyn_cast<CompoundStmt>(Else)) { |
| OS << ' '; |
| PrintRawCompoundStmt(CS); |
| OS << '\n'; |
| } else if (auto *ElseIf = dyn_cast<IfStmt>(Else)) { |
| OS << ' '; |
| PrintRawIfStmt(ElseIf); |
| } else { |
| OS << '\n'; |
| PrintStmt(If->getElse()); |
| } |
| } |
| } |
| |
| void StmtPrinter::VisitIfStmt(IfStmt *If) { |
| Indent(); |
| PrintRawIfStmt(If); |
| } |
| |
| void StmtPrinter::VisitSwitchStmt(SwitchStmt *Node) { |
| Indent() << "switch ("; |
| if (const DeclStmt *DS = Node->getConditionVariableDeclStmt()) |
| PrintRawDeclStmt(DS); |
| else |
| PrintExpr(Node->getCond()); |
| OS << ")"; |
| |
| // Pretty print compoundstmt bodies (very common). |
| if (auto *CS = dyn_cast<CompoundStmt>(Node->getBody())) { |
| OS << " "; |
| PrintRawCompoundStmt(CS); |
| OS << "\n"; |
| } else { |
| OS << "\n"; |
| PrintStmt(Node->getBody()); |
| } |
| } |
| |
| void StmtPrinter::VisitWhileStmt(WhileStmt *Node) { |
| Indent() << "while ("; |
| if (const DeclStmt *DS = Node->getConditionVariableDeclStmt()) |
| PrintRawDeclStmt(DS); |
| else |
| PrintExpr(Node->getCond()); |
| OS << ")\n"; |
| PrintStmt(Node->getBody()); |
| } |
| |
| void StmtPrinter::VisitDoStmt(DoStmt *Node) { |
| Indent() << "do "; |
| if (auto *CS = dyn_cast<CompoundStmt>(Node->getBody())) { |
| PrintRawCompoundStmt(CS); |
| OS << " "; |
| } else { |
| OS << "\n"; |
| PrintStmt(Node->getBody()); |
| Indent(); |
| } |
| |
| OS << "while ("; |
| PrintExpr(Node->getCond()); |
| OS << ");\n"; |
| } |
| |
| void StmtPrinter::VisitForStmt(ForStmt *Node) { |
| Indent() << "for ("; |
| if (Node->getInit()) { |
| if (auto *DS = dyn_cast<DeclStmt>(Node->getInit())) |
| PrintRawDeclStmt(DS); |
| else |
| PrintExpr(cast<Expr>(Node->getInit())); |
| } |
| OS << ";"; |
| if (Node->getCond()) { |
| OS << " "; |
| PrintExpr(Node->getCond()); |
| } |
| OS << ";"; |
| if (Node->getInc()) { |
| OS << " "; |
| PrintExpr(Node->getInc()); |
| } |
| OS << ") "; |
| |
| if (auto *CS = dyn_cast<CompoundStmt>(Node->getBody())) { |
| PrintRawCompoundStmt(CS); |
| OS << "\n"; |
| } else { |
| OS << "\n"; |
| PrintStmt(Node->getBody()); |
| } |
| } |
| |
| void StmtPrinter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *Node) { |
| Indent() << "for ("; |
| if (auto *DS = dyn_cast<DeclStmt>(Node->getElement())) |
| PrintRawDeclStmt(DS); |
| else |
| PrintExpr(cast<Expr>(Node->getElement())); |
| OS << " in "; |
| PrintExpr(Node->getCollection()); |
| OS << ") "; |
| |
| if (auto *CS = dyn_cast<CompoundStmt>(Node->getBody())) { |
| PrintRawCompoundStmt(CS); |
| OS << "\n"; |
| } else { |
| OS << "\n"; |
| PrintStmt(Node->getBody()); |
| } |
| } |
| |
| void StmtPrinter::VisitCXXForRangeStmt(CXXForRangeStmt *Node) { |
| Indent() << "for ("; |
| PrintingPolicy SubPolicy(Policy); |
| SubPolicy.SuppressInitializers = true; |
| Node->getLoopVariable()->print(OS, SubPolicy, IndentLevel); |
| OS << " : "; |
| PrintExpr(Node->getRangeInit()); |
| OS << ") {\n"; |
| PrintStmt(Node->getBody()); |
| Indent() << "}"; |
| if (Policy.IncludeNewlines) OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitMSDependentExistsStmt(MSDependentExistsStmt *Node) { |
| Indent(); |
| if (Node->isIfExists()) |
| OS << "__if_exists ("; |
| else |
| OS << "__if_not_exists ("; |
| |
| if (NestedNameSpecifier *Qualifier |
| = Node->getQualifierLoc().getNestedNameSpecifier()) |
| Qualifier->print(OS, Policy); |
| |
| OS << Node->getNameInfo() << ") "; |
| |
| PrintRawCompoundStmt(Node->getSubStmt()); |
| } |
| |
| void StmtPrinter::VisitGotoStmt(GotoStmt *Node) { |
| Indent() << "goto " << Node->getLabel()->getName() << ";"; |
| if (Policy.IncludeNewlines) OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitIndirectGotoStmt(IndirectGotoStmt *Node) { |
| Indent() << "goto *"; |
| PrintExpr(Node->getTarget()); |
| OS << ";"; |
| if (Policy.IncludeNewlines) OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitContinueStmt(ContinueStmt *Node) { |
| Indent() << "continue;"; |
| if (Policy.IncludeNewlines) OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitBreakStmt(BreakStmt *Node) { |
| Indent() << "break;"; |
| if (Policy.IncludeNewlines) OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitReturnStmt(ReturnStmt *Node) { |
| Indent() << "return"; |
| if (Node->getRetValue()) { |
| OS << " "; |
| PrintExpr(Node->getRetValue()); |
| } |
| OS << ";"; |
| if (Policy.IncludeNewlines) OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitGCCAsmStmt(GCCAsmStmt *Node) { |
| Indent() << "asm "; |
| |
| if (Node->isVolatile()) |
| OS << "volatile "; |
| |
| OS << "("; |
| VisitStringLiteral(Node->getAsmString()); |
| |
| // Outputs |
| if (Node->getNumOutputs() != 0 || Node->getNumInputs() != 0 || |
| Node->getNumClobbers() != 0) |
| OS << " : "; |
| |
| for (unsigned i = 0, e = Node->getNumOutputs(); i != e; ++i) { |
| if (i != 0) |
| OS << ", "; |
| |
| if (!Node->getOutputName(i).empty()) { |
| OS << '['; |
| OS << Node->getOutputName(i); |
| OS << "] "; |
| } |
| |
| VisitStringLiteral(Node->getOutputConstraintLiteral(i)); |
| OS << " ("; |
| Visit(Node->getOutputExpr(i)); |
| OS << ")"; |
| } |
| |
| // Inputs |
| if (Node->getNumInputs() != 0 || Node->getNumClobbers() != 0) |
| OS << " : "; |
| |
| for (unsigned i = 0, e = Node->getNumInputs(); i != e; ++i) { |
| if (i != 0) |
| OS << ", "; |
| |
| if (!Node->getInputName(i).empty()) { |
| OS << '['; |
| OS << Node->getInputName(i); |
| OS << "] "; |
| } |
| |
| VisitStringLiteral(Node->getInputConstraintLiteral(i)); |
| OS << " ("; |
| Visit(Node->getInputExpr(i)); |
| OS << ")"; |
| } |
| |
| // Clobbers |
| if (Node->getNumClobbers() != 0) |
| OS << " : "; |
| |
| for (unsigned i = 0, e = Node->getNumClobbers(); i != e; ++i) { |
| if (i != 0) |
| OS << ", "; |
| |
| VisitStringLiteral(Node->getClobberStringLiteral(i)); |
| } |
| |
| OS << ");"; |
| if (Policy.IncludeNewlines) OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitMSAsmStmt(MSAsmStmt *Node) { |
| // FIXME: Implement MS style inline asm statement printer. |
| Indent() << "__asm "; |
| if (Node->hasBraces()) |
| OS << "{\n"; |
| OS << Node->getAsmString() << "\n"; |
| if (Node->hasBraces()) |
| Indent() << "}\n"; |
| } |
| |
| void StmtPrinter::VisitCapturedStmt(CapturedStmt *Node) { |
| PrintStmt(Node->getCapturedDecl()->getBody()); |
| } |
| |
| void StmtPrinter::VisitObjCAtTryStmt(ObjCAtTryStmt *Node) { |
| Indent() << "@try"; |
| if (auto *TS = dyn_cast<CompoundStmt>(Node->getTryBody())) { |
| PrintRawCompoundStmt(TS); |
| OS << "\n"; |
| } |
| |
| for (unsigned I = 0, N = Node->getNumCatchStmts(); I != N; ++I) { |
| ObjCAtCatchStmt *catchStmt = Node->getCatchStmt(I); |
| Indent() << "@catch("; |
| if (catchStmt->getCatchParamDecl()) { |
| if (Decl *DS = catchStmt->getCatchParamDecl()) |
| PrintRawDecl(DS); |
| } |
| OS << ")"; |
| if (auto *CS = dyn_cast<CompoundStmt>(catchStmt->getCatchBody())) { |
| PrintRawCompoundStmt(CS); |
| OS << "\n"; |
| } |
| } |
| |
| if (auto *FS = static_cast<ObjCAtFinallyStmt *>(Node->getFinallyStmt())) { |
| Indent() << "@finally"; |
| PrintRawCompoundStmt(dyn_cast<CompoundStmt>(FS->getFinallyBody())); |
| OS << "\n"; |
| } |
| } |
| |
| void StmtPrinter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *Node) { |
| } |
| |
| void StmtPrinter::VisitObjCAtCatchStmt (ObjCAtCatchStmt *Node) { |
| Indent() << "@catch (...) { /* todo */ } \n"; |
| } |
| |
| void StmtPrinter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *Node) { |
| Indent() << "@throw"; |
| if (Node->getThrowExpr()) { |
| OS << " "; |
| PrintExpr(Node->getThrowExpr()); |
| } |
| OS << ";\n"; |
| } |
| |
| void StmtPrinter::VisitObjCAvailabilityCheckExpr( |
| ObjCAvailabilityCheckExpr *Node) { |
| OS << "@available(...)"; |
| } |
| |
| void StmtPrinter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *Node) { |
| Indent() << "@synchronized ("; |
| PrintExpr(Node->getSynchExpr()); |
| OS << ")"; |
| PrintRawCompoundStmt(Node->getSynchBody()); |
| OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *Node) { |
| Indent() << "@autoreleasepool"; |
| PrintRawCompoundStmt(dyn_cast<CompoundStmt>(Node->getSubStmt())); |
| OS << "\n"; |
| } |
| |
| void StmtPrinter::PrintRawCXXCatchStmt(CXXCatchStmt *Node) { |
| OS << "catch ("; |
| if (Decl *ExDecl = Node->getExceptionDecl()) |
| PrintRawDecl(ExDecl); |
| else |
| OS << "..."; |
| OS << ") "; |
| PrintRawCompoundStmt(cast<CompoundStmt>(Node->getHandlerBlock())); |
| } |
| |
| void StmtPrinter::VisitCXXCatchStmt(CXXCatchStmt *Node) { |
| Indent(); |
| PrintRawCXXCatchStmt(Node); |
| OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitCXXTryStmt(CXXTryStmt *Node) { |
| Indent() << "try "; |
| PrintRawCompoundStmt(Node->getTryBlock()); |
| for (unsigned i = 0, e = Node->getNumHandlers(); i < e; ++i) { |
| OS << " "; |
| PrintRawCXXCatchStmt(Node->getHandler(i)); |
| } |
| OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitSEHTryStmt(SEHTryStmt *Node) { |
| Indent() << (Node->getIsCXXTry() ? "try " : "__try "); |
| PrintRawCompoundStmt(Node->getTryBlock()); |
| SEHExceptStmt *E = Node->getExceptHandler(); |
| SEHFinallyStmt *F = Node->getFinallyHandler(); |
| if(E) |
| PrintRawSEHExceptHandler(E); |
| else { |
| assert(F && "Must have a finally block..."); |
| PrintRawSEHFinallyStmt(F); |
| } |
| OS << "\n"; |
| } |
| |
| void StmtPrinter::PrintRawSEHFinallyStmt(SEHFinallyStmt *Node) { |
| OS << "__finally "; |
| PrintRawCompoundStmt(Node->getBlock()); |
| OS << "\n"; |
| } |
| |
| void StmtPrinter::PrintRawSEHExceptHandler(SEHExceptStmt *Node) { |
| OS << "__except ("; |
| VisitExpr(Node->getFilterExpr()); |
| OS << ")\n"; |
| PrintRawCompoundStmt(Node->getBlock()); |
| OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitSEHExceptStmt(SEHExceptStmt *Node) { |
| Indent(); |
| PrintRawSEHExceptHandler(Node); |
| OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitSEHFinallyStmt(SEHFinallyStmt *Node) { |
| Indent(); |
| PrintRawSEHFinallyStmt(Node); |
| OS << "\n"; |
| } |
| |
| void StmtPrinter::VisitSEHLeaveStmt(SEHLeaveStmt *Node) { |
| Indent() << "__leave;"; |
| if (Policy.IncludeNewlines) OS << "\n"; |
| } |
| |
| //===----------------------------------------------------------------------===// |
| // OpenMP clauses printing methods |
| //===----------------------------------------------------------------------===// |
| |
| namespace { |
| |
| class OMPClausePrinter : public OMPClauseVisitor<OMPClausePrinter> { |
| raw_ostream &OS; |
| const PrintingPolicy &Policy; |
| |
| /// Process clauses with list of variables. |
| template <typename T> |
| void VisitOMPClauseList(T *Node, char StartSym); |
| |
| public: |
| OMPClausePrinter(raw_ostream &OS, const PrintingPolicy &Policy) |
| : OS(OS), Policy(Policy) {} |
| |
| #define OPENMP_CLAUSE(Name, Class) \ |
| void Visit##Class(Class *S); |
| #include "clang/Basic/OpenMPKinds.def" |
| }; |
| |
| } // namespace |
| |
| void OMPClausePrinter::VisitOMPIfClause(OMPIfClause *Node) { |
| OS << "if("; |
| if (Node->getNameModifier() != OMPD_unknown) |
| OS << getOpenMPDirectiveName(Node->getNameModifier()) << ": "; |
| Node->getCondition()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPFinalClause(OMPFinalClause *Node) { |
| OS << "final("; |
| Node->getCondition()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPNumThreadsClause(OMPNumThreadsClause *Node) { |
| OS << "num_threads("; |
| Node->getNumThreads()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPSafelenClause(OMPSafelenClause *Node) { |
| OS << "safelen("; |
| Node->getSafelen()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPSimdlenClause(OMPSimdlenClause *Node) { |
| OS << "simdlen("; |
| Node->getSimdlen()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPCollapseClause(OMPCollapseClause *Node) { |
| OS << "collapse("; |
| Node->getNumForLoops()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPDefaultClause(OMPDefaultClause *Node) { |
| OS << "default(" |
| << getOpenMPSimpleClauseTypeName(OMPC_default, Node->getDefaultKind()) |
| << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPProcBindClause(OMPProcBindClause *Node) { |
| OS << "proc_bind(" |
| << getOpenMPSimpleClauseTypeName(OMPC_proc_bind, Node->getProcBindKind()) |
| << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPScheduleClause(OMPScheduleClause *Node) { |
| OS << "schedule("; |
| if (Node->getFirstScheduleModifier() != OMPC_SCHEDULE_MODIFIER_unknown) { |
| OS << getOpenMPSimpleClauseTypeName(OMPC_schedule, |
| Node->getFirstScheduleModifier()); |
| if (Node->getSecondScheduleModifier() != OMPC_SCHEDULE_MODIFIER_unknown) { |
| OS << ", "; |
| OS << getOpenMPSimpleClauseTypeName(OMPC_schedule, |
| Node->getSecondScheduleModifier()); |
| } |
| OS << ": "; |
| } |
| OS << getOpenMPSimpleClauseTypeName(OMPC_schedule, Node->getScheduleKind()); |
| if (auto *E = Node->getChunkSize()) { |
| OS << ", "; |
| E->printPretty(OS, nullptr, Policy); |
| } |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPOrderedClause(OMPOrderedClause *Node) { |
| OS << "ordered"; |
| if (auto *Num = Node->getNumForLoops()) { |
| OS << "("; |
| Num->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPNowaitClause(OMPNowaitClause *) { |
| OS << "nowait"; |
| } |
| |
| void OMPClausePrinter::VisitOMPUntiedClause(OMPUntiedClause *) { |
| OS << "untied"; |
| } |
| |
| void OMPClausePrinter::VisitOMPNogroupClause(OMPNogroupClause *) { |
| OS << "nogroup"; |
| } |
| |
| void OMPClausePrinter::VisitOMPMergeableClause(OMPMergeableClause *) { |
| OS << "mergeable"; |
| } |
| |
| void OMPClausePrinter::VisitOMPReadClause(OMPReadClause *) { OS << "read"; } |
| |
| void OMPClausePrinter::VisitOMPWriteClause(OMPWriteClause *) { OS << "write"; } |
| |
| void OMPClausePrinter::VisitOMPUpdateClause(OMPUpdateClause *) { |
| OS << "update"; |
| } |
| |
| void OMPClausePrinter::VisitOMPCaptureClause(OMPCaptureClause *) { |
| OS << "capture"; |
| } |
| |
| void OMPClausePrinter::VisitOMPSeqCstClause(OMPSeqCstClause *) { |
| OS << "seq_cst"; |
| } |
| |
| void OMPClausePrinter::VisitOMPThreadsClause(OMPThreadsClause *) { |
| OS << "threads"; |
| } |
| |
| void OMPClausePrinter::VisitOMPSIMDClause(OMPSIMDClause *) { OS << "simd"; } |
| |
| void OMPClausePrinter::VisitOMPDeviceClause(OMPDeviceClause *Node) { |
| OS << "device("; |
| Node->getDevice()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPNumTeamsClause(OMPNumTeamsClause *Node) { |
| OS << "num_teams("; |
| Node->getNumTeams()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPThreadLimitClause(OMPThreadLimitClause *Node) { |
| OS << "thread_limit("; |
| Node->getThreadLimit()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPPriorityClause(OMPPriorityClause *Node) { |
| OS << "priority("; |
| Node->getPriority()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPGrainsizeClause(OMPGrainsizeClause *Node) { |
| OS << "grainsize("; |
| Node->getGrainsize()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPNumTasksClause(OMPNumTasksClause *Node) { |
| OS << "num_tasks("; |
| Node->getNumTasks()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPHintClause(OMPHintClause *Node) { |
| OS << "hint("; |
| Node->getHint()->printPretty(OS, nullptr, Policy, 0); |
| OS << ")"; |
| } |
| |
| template<typename T> |
| void OMPClausePrinter::VisitOMPClauseList(T *Node, char StartSym) { |
| for (typename T::varlist_iterator I = Node->varlist_begin(), |
| E = Node->varlist_end(); |
| I != E; ++I) { |
| assert(*I && "Expected non-null Stmt"); |
| OS << (I == Node->varlist_begin() ? StartSym : ','); |
| if (auto *DRE = dyn_cast<DeclRefExpr>(*I)) { |
| if (isa<OMPCapturedExprDecl>(DRE->getDecl())) |
| DRE->printPretty(OS, nullptr, Policy, 0); |
| else |
| DRE->getDecl()->printQualifiedName(OS); |
| } else |
| (*I)->printPretty(OS, nullptr, Policy, 0); |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPPrivateClause(OMPPrivateClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "private"; |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPFirstprivateClause(OMPFirstprivateClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "firstprivate"; |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPLastprivateClause(OMPLastprivateClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "lastprivate"; |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPSharedClause(OMPSharedClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "shared"; |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPReductionClause(OMPReductionClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "reduction("; |
| NestedNameSpecifier *QualifierLoc = |
| Node->getQualifierLoc().getNestedNameSpecifier(); |
| OverloadedOperatorKind OOK = |
| Node->getNameInfo().getName().getCXXOverloadedOperator(); |
| if (QualifierLoc == nullptr && OOK != OO_None) { |
| // Print reduction identifier in C format |
| OS << getOperatorSpelling(OOK); |
| } else { |
| // Use C++ format |
| if (QualifierLoc != nullptr) |
| QualifierLoc->print(OS, Policy); |
| OS << Node->getNameInfo(); |
| } |
| OS << ":"; |
| VisitOMPClauseList(Node, ' '); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPTaskReductionClause( |
| OMPTaskReductionClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "task_reduction("; |
| NestedNameSpecifier *QualifierLoc = |
| Node->getQualifierLoc().getNestedNameSpecifier(); |
| OverloadedOperatorKind OOK = |
| Node->getNameInfo().getName().getCXXOverloadedOperator(); |
| if (QualifierLoc == nullptr && OOK != OO_None) { |
| // Print reduction identifier in C format |
| OS << getOperatorSpelling(OOK); |
| } else { |
| // Use C++ format |
| if (QualifierLoc != nullptr) |
| QualifierLoc->print(OS, Policy); |
| OS << Node->getNameInfo(); |
| } |
| OS << ":"; |
| VisitOMPClauseList(Node, ' '); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPInReductionClause(OMPInReductionClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "in_reduction("; |
| NestedNameSpecifier *QualifierLoc = |
| Node->getQualifierLoc().getNestedNameSpecifier(); |
| OverloadedOperatorKind OOK = |
| Node->getNameInfo().getName().getCXXOverloadedOperator(); |
| if (QualifierLoc == nullptr && OOK != OO_None) { |
| // Print reduction identifier in C format |
| OS << getOperatorSpelling(OOK); |
| } else { |
| // Use C++ format |
| if (QualifierLoc != nullptr) |
| QualifierLoc->print(OS, Policy); |
| OS << Node->getNameInfo(); |
| } |
| OS << ":"; |
| VisitOMPClauseList(Node, ' '); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPLinearClause(OMPLinearClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "linear"; |
| if (Node->getModifierLoc().isValid()) { |
| OS << '(' |
| << getOpenMPSimpleClauseTypeName(OMPC_linear, Node->getModifier()); |
| } |
| VisitOMPClauseList(Node, '('); |
| if (Node->getModifierLoc().isValid()) |
| OS << ')'; |
| if (Node->getStep() != nullptr) { |
| OS << ": "; |
| Node->getStep()->printPretty(OS, nullptr, Policy, 0); |
| } |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPAlignedClause(OMPAlignedClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "aligned"; |
| VisitOMPClauseList(Node, '('); |
| if (Node->getAlignment() != nullptr) { |
| OS << ": "; |
| Node->getAlignment()->printPretty(OS, nullptr, Policy, 0); |
| } |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPCopyinClause(OMPCopyinClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "copyin"; |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPCopyprivateClause(OMPCopyprivateClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "copyprivate"; |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPFlushClause(OMPFlushClause *Node) { |
| if (!Node->varlist_empty()) { |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPDependClause(OMPDependClause *Node) { |
| OS << "depend("; |
| OS << getOpenMPSimpleClauseTypeName(Node->getClauseKind(), |
| Node->getDependencyKind()); |
| if (!Node->varlist_empty()) { |
| OS << " :"; |
| VisitOMPClauseList(Node, ' '); |
| } |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPMapClause(OMPMapClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "map("; |
| if (Node->getMapType() != OMPC_MAP_unknown) { |
| if (Node->getMapTypeModifier() != OMPC_MAP_unknown) { |
| OS << getOpenMPSimpleClauseTypeName(OMPC_map, |
| Node->getMapTypeModifier()); |
| OS << ','; |
| } |
| OS << getOpenMPSimpleClauseTypeName(OMPC_map, Node->getMapType()); |
| OS << ':'; |
| } |
| VisitOMPClauseList(Node, ' '); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPToClause(OMPToClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "to"; |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPFromClause(OMPFromClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "from"; |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPDistScheduleClause(OMPDistScheduleClause *Node) { |
| OS << "dist_schedule(" << getOpenMPSimpleClauseTypeName( |
| OMPC_dist_schedule, Node->getDistScheduleKind()); |
| if (auto *E = Node->getChunkSize()) { |
| OS << ", "; |
| E->printPretty(OS, nullptr, Policy); |
| } |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPDefaultmapClause(OMPDefaultmapClause *Node) { |
| OS << "defaultmap("; |
| OS << getOpenMPSimpleClauseTypeName(OMPC_defaultmap, |
| Node->getDefaultmapModifier()); |
| OS << ": "; |
| OS << getOpenMPSimpleClauseTypeName(OMPC_defaultmap, |
| Node->getDefaultmapKind()); |
| OS << ")"; |
| } |
| |
| void OMPClausePrinter::VisitOMPUseDevicePtrClause(OMPUseDevicePtrClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "use_device_ptr"; |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| void OMPClausePrinter::VisitOMPIsDevicePtrClause(OMPIsDevicePtrClause *Node) { |
| if (!Node->varlist_empty()) { |
| OS << "is_device_ptr"; |
| VisitOMPClauseList(Node, '('); |
| OS << ")"; |
| } |
| } |
| |
| //===----------------------------------------------------------------------===// |
| // OpenMP directives printing methods |
| //===----------------------------------------------------------------------===// |
| |
| void StmtPrinter::PrintOMPExecutableDirective(OMPExecutableDirective *S, |
| bool ForceNoStmt) { |
| OMPClausePrinter Printer(OS, Policy); |
| ArrayRef<OMPClause *> Clauses = S->clauses(); |
| for (auto *Clause : Clauses) |
| if (Clause && !Clause->isImplicit()) { |
| OS << ' '; |
| Printer.Visit(Clause); |
| } |
| OS << "\n"; |
| if (!ForceNoStmt && S->hasAssociatedStmt()) |
| PrintStmt(S->getInnermostCapturedStmt()->getCapturedStmt()); |
| } |
| |
| void StmtPrinter::VisitOMPParallelDirective(OMPParallelDirective *Node) { |
| Indent() << "#pragma omp parallel"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPSimdDirective(OMPSimdDirective *Node) { |
| Indent() << "#pragma omp simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPForDirective(OMPForDirective *Node) { |
| Indent() << "#pragma omp for"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPForSimdDirective(OMPForSimdDirective *Node) { |
| Indent() << "#pragma omp for simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPSectionsDirective(OMPSectionsDirective *Node) { |
| Indent() << "#pragma omp sections"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPSectionDirective(OMPSectionDirective *Node) { |
| Indent() << "#pragma omp section"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPSingleDirective(OMPSingleDirective *Node) { |
| Indent() << "#pragma omp single"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPMasterDirective(OMPMasterDirective *Node) { |
| Indent() << "#pragma omp master"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPCriticalDirective(OMPCriticalDirective *Node) { |
| Indent() << "#pragma omp critical"; |
| if (Node->getDirectiveName().getName()) { |
| OS << " ("; |
| Node->getDirectiveName().printName(OS); |
| OS << ")"; |
| } |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPParallelForDirective(OMPParallelForDirective *Node) { |
| Indent() << "#pragma omp parallel for"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPParallelForSimdDirective( |
| OMPParallelForSimdDirective *Node) { |
| Indent() << "#pragma omp parallel for simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPParallelSectionsDirective( |
| OMPParallelSectionsDirective *Node) { |
| Indent() << "#pragma omp parallel sections"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTaskDirective(OMPTaskDirective *Node) { |
| Indent() << "#pragma omp task"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTaskyieldDirective(OMPTaskyieldDirective *Node) { |
| Indent() << "#pragma omp taskyield"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPBarrierDirective(OMPBarrierDirective *Node) { |
| Indent() << "#pragma omp barrier"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTaskwaitDirective(OMPTaskwaitDirective *Node) { |
| Indent() << "#pragma omp taskwait"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTaskgroupDirective(OMPTaskgroupDirective *Node) { |
| Indent() << "#pragma omp taskgroup"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPFlushDirective(OMPFlushDirective *Node) { |
| Indent() << "#pragma omp flush"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPOrderedDirective(OMPOrderedDirective *Node) { |
| Indent() << "#pragma omp ordered"; |
| PrintOMPExecutableDirective(Node, Node->hasClausesOfKind<OMPDependClause>()); |
| } |
| |
| void StmtPrinter::VisitOMPAtomicDirective(OMPAtomicDirective *Node) { |
| Indent() << "#pragma omp atomic"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetDirective(OMPTargetDirective *Node) { |
| Indent() << "#pragma omp target"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetDataDirective(OMPTargetDataDirective *Node) { |
| Indent() << "#pragma omp target data"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetEnterDataDirective( |
| OMPTargetEnterDataDirective *Node) { |
| Indent() << "#pragma omp target enter data"; |
| PrintOMPExecutableDirective(Node, /*ForceNoStmt=*/true); |
| } |
| |
| void StmtPrinter::VisitOMPTargetExitDataDirective( |
| OMPTargetExitDataDirective *Node) { |
| Indent() << "#pragma omp target exit data"; |
| PrintOMPExecutableDirective(Node, /*ForceNoStmt=*/true); |
| } |
| |
| void StmtPrinter::VisitOMPTargetParallelDirective( |
| OMPTargetParallelDirective *Node) { |
| Indent() << "#pragma omp target parallel"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetParallelForDirective( |
| OMPTargetParallelForDirective *Node) { |
| Indent() << "#pragma omp target parallel for"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTeamsDirective(OMPTeamsDirective *Node) { |
| Indent() << "#pragma omp teams"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPCancellationPointDirective( |
| OMPCancellationPointDirective *Node) { |
| Indent() << "#pragma omp cancellation point " |
| << getOpenMPDirectiveName(Node->getCancelRegion()); |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPCancelDirective(OMPCancelDirective *Node) { |
| Indent() << "#pragma omp cancel " |
| << getOpenMPDirectiveName(Node->getCancelRegion()); |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTaskLoopDirective(OMPTaskLoopDirective *Node) { |
| Indent() << "#pragma omp taskloop"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTaskLoopSimdDirective( |
| OMPTaskLoopSimdDirective *Node) { |
| Indent() << "#pragma omp taskloop simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPDistributeDirective(OMPDistributeDirective *Node) { |
| Indent() << "#pragma omp distribute"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetUpdateDirective( |
| OMPTargetUpdateDirective *Node) { |
| Indent() << "#pragma omp target update"; |
| PrintOMPExecutableDirective(Node, /*ForceNoStmt=*/true); |
| } |
| |
| void StmtPrinter::VisitOMPDistributeParallelForDirective( |
| OMPDistributeParallelForDirective *Node) { |
| Indent() << "#pragma omp distribute parallel for"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPDistributeParallelForSimdDirective( |
| OMPDistributeParallelForSimdDirective *Node) { |
| Indent() << "#pragma omp distribute parallel for simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPDistributeSimdDirective( |
| OMPDistributeSimdDirective *Node) { |
| Indent() << "#pragma omp distribute simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetParallelForSimdDirective( |
| OMPTargetParallelForSimdDirective *Node) { |
| Indent() << "#pragma omp target parallel for simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetSimdDirective(OMPTargetSimdDirective *Node) { |
| Indent() << "#pragma omp target simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTeamsDistributeDirective( |
| OMPTeamsDistributeDirective *Node) { |
| Indent() << "#pragma omp teams distribute"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTeamsDistributeSimdDirective( |
| OMPTeamsDistributeSimdDirective *Node) { |
| Indent() << "#pragma omp teams distribute simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTeamsDistributeParallelForSimdDirective( |
| OMPTeamsDistributeParallelForSimdDirective *Node) { |
| Indent() << "#pragma omp teams distribute parallel for simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTeamsDistributeParallelForDirective( |
| OMPTeamsDistributeParallelForDirective *Node) { |
| Indent() << "#pragma omp teams distribute parallel for"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetTeamsDirective(OMPTargetTeamsDirective *Node) { |
| Indent() << "#pragma omp target teams"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetTeamsDistributeDirective( |
| OMPTargetTeamsDistributeDirective *Node) { |
| Indent() << "#pragma omp target teams distribute"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetTeamsDistributeParallelForDirective( |
| OMPTargetTeamsDistributeParallelForDirective *Node) { |
| Indent() << "#pragma omp target teams distribute parallel for"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetTeamsDistributeParallelForSimdDirective( |
| OMPTargetTeamsDistributeParallelForSimdDirective *Node) { |
| Indent() << "#pragma omp target teams distribute parallel for simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| void StmtPrinter::VisitOMPTargetTeamsDistributeSimdDirective( |
| OMPTargetTeamsDistributeSimdDirective *Node) { |
| Indent() << "#pragma omp target teams distribute simd"; |
| PrintOMPExecutableDirective(Node); |
| } |
| |
| //===----------------------------------------------------------------------===// |
| // Expr printing methods. |
| //===----------------------------------------------------------------------===// |
| |
| void StmtPrinter::VisitDeclRefExpr(DeclRefExpr *Node) { |
| if (const auto *OCED = dyn_cast<OMPCapturedExprDecl>(Node->getDecl())) { |
| OCED->getInit()->IgnoreImpCasts()->printPretty(OS, nullptr, Policy); |
| return; |
| } |
| if (NestedNameSpecifier *Qualifier = Node->getQualifier()) |
| Qualifier->print(OS, Policy); |
| if (Node->hasTemplateKeyword()) |
| OS << "template "; |
| OS << Node->getNameInfo(); |
| if (Node->hasExplicitTemplateArgs()) |
| printTemplateArgumentList(OS, Node->template_arguments(), Policy); |
| } |
| |
| void StmtPrinter::VisitDependentScopeDeclRefExpr( |
| DependentScopeDeclRefExpr *Node) { |
| if (NestedNameSpecifier *Qualifier = Node->getQualifier()) |
| Qualifier->print(OS, Policy); |
| if (Node->hasTemplateKeyword()) |
| OS << "template "; |
| OS << Node->getNameInfo(); |
| if (Node->hasExplicitTemplateArgs()) |
| printTemplateArgumentList(OS, Node->template_arguments(), Policy); |
| } |
| |
| void StmtPrinter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *Node) { |
| if (Node->getQualifier()) |
| Node->getQualifier()->print(OS, Policy); |
| if (Node->hasTemplateKeyword()) |
| OS << "template "; |
| OS << Node->getNameInfo(); |
| if (Node->hasExplicitTemplateArgs()) |
| printTemplateArgumentList(OS, Node->template_arguments(), Policy); |
| } |
| |
| static bool isImplicitSelf(const Expr *E) { |
| if (const auto *DRE = dyn_cast<DeclRefExpr>(E)) { |
| if (const auto *PD = dyn_cast<ImplicitParamDecl>(DRE->getDecl())) { |
| if (PD->getParameterKind() == ImplicitParamDecl::ObjCSelf && |
| DRE->getLocStart().isInvalid()) |
| return true; |
| } |
| } |
| return false; |
| } |
| |
| void StmtPrinter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *Node) { |
| if (Node->getBase()) { |
| if (!Policy.SuppressImplicitBase || |
| !isImplicitSelf(Node->getBase()->IgnoreImpCasts())) { |
| PrintExpr(Node->getBase()); |
| OS << (Node->isArrow() ? "->" : "."); |
| } |
| } |
| OS << *Node->getDecl(); |
| } |
| |
| void StmtPrinter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *Node) { |
| if (Node->isSuperReceiver()) |
| OS << "super."; |
| else if (Node->isObjectReceiver() && Node->getBase()) { |
| PrintExpr(Node->getBase()); |
| OS << "."; |
| } else if (Node->isClassReceiver() && Node->getClassReceiver()) { |
| OS << Node->getClassReceiver()->getName() << "."; |
| } |
| |
| if (Node->isImplicitProperty()) { |
| if (const auto *Getter = Node->getImplicitPropertyGetter()) |
| Getter->getSelector().print(OS); |
| else |
| OS << SelectorTable::getPropertyNameFromSetterSelector( |
| Node->getImplicitPropertySetter()->getSelector()); |
| } else |
| OS << Node->getExplicitProperty()->getName(); |
| } |
| |
| void StmtPrinter::VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr *Node) { |
| PrintExpr(Node->getBaseExpr()); |
| OS << "["; |
| PrintExpr(Node->getKeyExpr()); |
| OS << "]"; |
| } |
| |
| void StmtPrinter::VisitPredefinedExpr(PredefinedExpr *Node) { |
| OS << PredefinedExpr::getIdentTypeName(Node->getIdentType()); |
| } |
| |
| void StmtPrinter::VisitCharacterLiteral(CharacterLiteral *Node) { |
| unsigned value = Node->getValue(); |
| |
| switch (Node->getKind()) { |
| case CharacterLiteral::Ascii: break; // no prefix. |
| case CharacterLiteral::Wide: OS << 'L'; break; |
| case CharacterLiteral::UTF8: OS << "u8"; break; |
| case CharacterLiteral::UTF16: OS << 'u'; break; |
| case CharacterLiteral::UTF32: OS << 'U'; break; |
| } |
| |
| switch (value) { |
| case '\\': |
| OS << "'\\\\'"; |
| break; |
| case '\'': |
| OS << "'\\''"; |
| break; |
| case '\a': |
| // TODO: K&R: the meaning of '\\a' is different in traditional C |
| OS << "'\\a'"; |
| break; |
| case '\b': |
| OS << "'\\b'"; |
| break; |
| // Nonstandard escape sequence. |
| /*case '\e': |
| OS << "'\\e'"; |
| break;*/ |
| case '\f': |
| OS << "'\\f'"; |
| break; |
| case '\n': |
| OS << "'\\n'"; |
| break; |
| case '\r': |
| OS << "'\\r'"; |
| break; |
| case '\t': |
| OS << "'\\t'"; |
| break; |
| case '\v': |
| OS << "'\\v'"; |
| break; |
| default: |
| // A character literal might be sign-extended, which |
| // would result in an invalid \U escape sequence. |
| // FIXME: multicharacter literals such as '\xFF\xFF\xFF\xFF' |
| // are not correctly handled. |
| if ((value & ~0xFFu) == ~0xFFu && Node->getKind() == CharacterLiteral::Ascii) |
| value &= 0xFFu; |
| if (value < 256 && isPrintable((unsigned char)value)) |
| OS << "'" << (char)value << "'"; |
| else if (value < 256) |
| OS << "'\\x" << llvm::format("%02x", value) << "'"; |
| else if (value <= 0xFFFF) |
| OS << "'\\u" << llvm::format("%04x", value) << "'"; |
| else |
| OS << "'\\U" << llvm::format("%08x", value) << "'"; |
| } |
| } |
| |
| /// Prints the given expression using the original source text. Returns true on |
| /// success, false otherwise. |
| static bool printExprAsWritten(raw_ostream &OS, Expr *E, |
| const ASTContext *Context) { |
| if (!Context) |
| return false; |
| bool Invalid = false; |
| StringRef Source = Lexer::getSourceText( |
| CharSourceRange::getTokenRange(E->getSourceRange()), |
| Context->getSourceManager(), Context->getLangOpts(), &Invalid); |
| if (!Invalid) { |
| OS << Source; |
| return true; |
| } |
| return false; |
| } |
| |
| void StmtPrinter::VisitIntegerLiteral(IntegerLiteral *Node) { |
| if (Policy.ConstantsAsWritten && printExprAsWritten(OS, Node, Context)) |
| return; |
| bool isSigned = Node->getType()->isSignedIntegerType(); |
| OS << Node->getValue().toString(10, isSigned); |
| |
| // Emit suffixes. Integer literals are always a builtin integer type. |
| switch (Node->getType()->getAs<BuiltinType>()->getKind()) { |
| default: llvm_unreachable("Unexpected type for integer literal!"); |
| case BuiltinType::Char_S: |
| case BuiltinType::Char_U: OS << "i8"; break; |
| case BuiltinType::UChar: OS << "Ui8"; break; |
| case BuiltinType::Short: OS << "i16"; break; |
| case BuiltinType::UShort: OS << "Ui16"; break; |
| case BuiltinType::Int: break; // no suffix. |
| case BuiltinType::UInt: OS << 'U'; break; |
| case BuiltinType::Long: OS << 'L'; break; |
| case BuiltinType::ULong: OS << "UL"; break; |
| case BuiltinType::LongLong: OS << "LL"; break; |
| case BuiltinType::ULongLong: OS << "ULL"; break; |
| } |
| } |
| |
| void StmtPrinter::VisitFixedPointLiteral(FixedPointLiteral *Node) { |
| if (Policy.ConstantsAsWritten && printExprAsWritten(OS, Node, Context)) |
| return; |
| OS << Node->getValueAsString(/*Radix=*/10); |
| |
| switch (Node->getType()->getAs<BuiltinType>()->getKind()) { |
| default: llvm_unreachable("Unexpected type for fixed point literal!"); |
| case BuiltinType::ShortFract: OS << "hr"; break; |
| case BuiltinType::ShortAccum: OS << "hk"; break; |
| case BuiltinType::UShortFract: OS << "uhr"; break; |
| case BuiltinType::UShortAccum: OS << "uhk"; break; |
| case BuiltinType::Fract: OS << "r"; break; |
| case BuiltinType::Accum: OS << "k"; break; |
| case BuiltinType::UFract: OS << "ur"; break; |
| case BuiltinType::UAccum: OS << "uk"; break; |
| case BuiltinType::LongFract: OS << "lr"; break; |
| case BuiltinType::LongAccum: OS << "lk"; break; |
| case BuiltinType::ULongFract: OS << "ulr"; break; |
| case BuiltinType::ULongAccum: OS << "ulk"; break; |
| } |
| } |
| |
| static void PrintFloatingLiteral(raw_ostream &OS, FloatingLiteral *Node, |
| bool PrintSuffix) { |
| SmallString<16> Str; |
| Node->getValue().toString(Str); |
| OS << Str; |
| if (Str.find_first_not_of("-0123456789") == StringRef::npos) |
| OS << '.'; // Trailing dot in order to separate from ints. |
| |
| if (!PrintSuffix) |
| return; |
| |
| // Emit suffixes. Float literals are always a builtin float type. |
| switch (Node->getType()->getAs<BuiltinType>()->getKind()) { |
| default: llvm_unreachable("Unexpected type for float literal!"); |
| case BuiltinType::Half: break; // FIXME: suffix? |
| case BuiltinType::Double: break; // no suffix. |
| case BuiltinType::Float16: OS << "F16"; break; |
| case BuiltinType::Float: OS << 'F'; break; |
| case BuiltinType::LongDouble: OS << 'L'; break; |
| case BuiltinType::Float128: OS << 'Q'; break; |
| } |
| } |
| |
| void StmtPrinter::VisitFloatingLiteral(FloatingLiteral *Node) { |
| if (Policy.ConstantsAsWritten && printExprAsWritten(OS, Node, Context)) |
| return; |
| PrintFloatingLiteral(OS, Node, /*PrintSuffix=*/true); |
| } |
| |
| void StmtPrinter::VisitImaginaryLiteral(ImaginaryLiteral *Node) { |
| PrintExpr(Node->getSubExpr()); |
| OS << "i"; |
| } |
| |
| void StmtPrinter::VisitStringLiteral(StringLiteral *Str) { |
| Str->outputString(OS); |
| } |
| |
| void StmtPrinter::VisitParenExpr(ParenExpr *Node) { |
| OS << "("; |
| PrintExpr(Node->getSubExpr()); |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitUnaryOperator(UnaryOperator *Node) { |
| if (!Node->isPostfix()) { |
| OS << UnaryOperator::getOpcodeStr(Node->getOpcode()); |
| |
| // Print a space if this is an "identifier operator" like __real, or if |
| // it might be concatenated incorrectly like '+'. |
| switch (Node->getOpcode()) { |
| default: break; |
| case UO_Real: |
| case UO_Imag: |
| case UO_Extension: |
| OS << ' '; |
| break; |
| case UO_Plus: |
| case UO_Minus: |
| if (isa<UnaryOperator>(Node->getSubExpr())) |
| OS << ' '; |
| break; |
| } |
| } |
| PrintExpr(Node->getSubExpr()); |
| |
| if (Node->isPostfix()) |
| OS << UnaryOperator::getOpcodeStr(Node->getOpcode()); |
| } |
| |
| void StmtPrinter::VisitOffsetOfExpr(OffsetOfExpr *Node) { |
| OS << "__builtin_offsetof("; |
| Node->getTypeSourceInfo()->getType().print(OS, Policy); |
| OS << ", "; |
| bool PrintedSomething = false; |
| for (unsigned i = 0, n = Node->getNumComponents(); i < n; ++i) { |
| OffsetOfNode ON = Node->getComponent(i); |
| if (ON.getKind() == OffsetOfNode::Array) { |
| // Array node |
| OS << "["; |
| PrintExpr(Node->getIndexExpr(ON.getArrayExprIndex())); |
| OS << "]"; |
| PrintedSomething = true; |
| continue; |
| } |
| |
| // Skip implicit base indirections. |
| if (ON.getKind() == OffsetOfNode::Base) |
| continue; |
| |
| // Field or identifier node. |
| IdentifierInfo *Id = ON.getFieldName(); |
| if (!Id) |
| continue; |
| |
| if (PrintedSomething) |
| OS << "."; |
| else |
| PrintedSomething = true; |
| OS << Id->getName(); |
| } |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *Node){ |
| switch(Node->getKind()) { |
| case UETT_SizeOf: |
| OS << "sizeof"; |
| break; |
| case UETT_AlignOf: |
| if (Policy.Alignof) |
| OS << "alignof"; |
| else if (Policy.UnderscoreAlignof) |
| OS << "_Alignof"; |
| else |
| OS << "__alignof"; |
| break; |
| case UETT_VecStep: |
| OS << "vec_step"; |
| break; |
| case UETT_OpenMPRequiredSimdAlign: |
| OS << "__builtin_omp_required_simd_align"; |
| break; |
| } |
| if (Node->isArgumentType()) { |
| OS << '('; |
| Node->getArgumentType().print(OS, Policy); |
| OS << ')'; |
| } else { |
| OS << " "; |
| PrintExpr(Node->getArgumentExpr()); |
| } |
| } |
| |
| void StmtPrinter::VisitGenericSelectionExpr(GenericSelectionExpr *Node) { |
| OS << "_Generic("; |
| PrintExpr(Node->getControllingExpr()); |
| for (unsigned i = 0; i != Node->getNumAssocs(); ++i) { |
| OS << ", "; |
| QualType T = Node->getAssocType(i); |
| if (T.isNull()) |
| OS << "default"; |
| else |
| T.print(OS, Policy); |
| OS << ": "; |
| PrintExpr(Node->getAssocExpr(i)); |
| } |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitArraySubscriptExpr(ArraySubscriptExpr *Node) { |
| PrintExpr(Node->getLHS()); |
| OS << "["; |
| PrintExpr(Node->getRHS()); |
| OS << "]"; |
| } |
| |
| void StmtPrinter::VisitOMPArraySectionExpr(OMPArraySectionExpr *Node) { |
| PrintExpr(Node->getBase()); |
| OS << "["; |
| if (Node->getLowerBound()) |
| PrintExpr(Node->getLowerBound()); |
| if (Node->getColonLoc().isValid()) { |
| OS << ":"; |
| if (Node->getLength()) |
| PrintExpr(Node->getLength()); |
| } |
| OS << "]"; |
| } |
| |
| void StmtPrinter::PrintCallArgs(CallExpr *Call) { |
| for (unsigned i = 0, e = Call->getNumArgs(); i != e; ++i) { |
| if (isa<CXXDefaultArgExpr>(Call->getArg(i))) { |
| // Don't print any defaulted arguments |
| break; |
| } |
| |
| if (i) OS << ", "; |
| PrintExpr(Call->getArg(i)); |
| } |
| } |
| |
| void StmtPrinter::VisitCallExpr(CallExpr *Call) { |
| PrintExpr(Call->getCallee()); |
| OS << "("; |
| PrintCallArgs(Call); |
| OS << ")"; |
| } |
| |
| static bool isImplicitThis(const Expr *E) { |
| if (const auto *TE = dyn_cast<CXXThisExpr>(E)) |
| return TE->isImplicit(); |
| return false; |
| } |
| |
| void StmtPrinter::VisitMemberExpr(MemberExpr *Node) { |
| if (!Policy.SuppressImplicitBase || !isImplicitThis(Node->getBase())) { |
| PrintExpr(Node->getBase()); |
| |
| auto *ParentMember = dyn_cast<MemberExpr>(Node->getBase()); |
| FieldDecl *ParentDecl = |
| ParentMember ? dyn_cast<FieldDecl>(ParentMember->getMemberDecl()) |
| : nullptr; |
| |
| if (!ParentDecl || !ParentDecl->isAnonymousStructOrUnion()) |
| OS << (Node->isArrow() ? "->" : "."); |
| } |
| |
| if (auto *FD = dyn_cast<FieldDecl>(Node->getMemberDecl())) |
| if (FD->isAnonymousStructOrUnion()) |
| return; |
| |
| if (NestedNameSpecifier *Qualifier = Node->getQualifier()) |
| Qualifier->print(OS, Policy); |
| if (Node->hasTemplateKeyword()) |
| OS << "template "; |
| OS << Node->getMemberNameInfo(); |
| if (Node->hasExplicitTemplateArgs()) |
| printTemplateArgumentList(OS, Node->template_arguments(), Policy); |
| } |
| |
| void StmtPrinter::VisitObjCIsaExpr(ObjCIsaExpr *Node) { |
| PrintExpr(Node->getBase()); |
| OS << (Node->isArrow() ? "->isa" : ".isa"); |
| } |
| |
| void StmtPrinter::VisitExtVectorElementExpr(ExtVectorElementExpr *Node) { |
| PrintExpr(Node->getBase()); |
| OS << "."; |
| OS << Node->getAccessor().getName(); |
| } |
| |
| void StmtPrinter::VisitCStyleCastExpr(CStyleCastExpr *Node) { |
| OS << '('; |
| Node->getTypeAsWritten().print(OS, Policy); |
| OS << ')'; |
| PrintExpr(Node->getSubExpr()); |
| } |
| |
| void StmtPrinter::VisitCompoundLiteralExpr(CompoundLiteralExpr *Node) { |
| OS << '('; |
| Node->getType().print(OS, Policy); |
| OS << ')'; |
| PrintExpr(Node->getInitializer()); |
| } |
| |
| void StmtPrinter::VisitImplicitCastExpr(ImplicitCastExpr *Node) { |
| // No need to print anything, simply forward to the subexpression. |
| PrintExpr(Node->getSubExpr()); |
| } |
| |
| void StmtPrinter::VisitBinaryOperator(BinaryOperator *Node) { |
| PrintExpr(Node->getLHS()); |
| OS << " " << BinaryOperator::getOpcodeStr(Node->getOpcode()) << " "; |
| PrintExpr(Node->getRHS()); |
| } |
| |
| void StmtPrinter::VisitCompoundAssignOperator(CompoundAssignOperator *Node) { |
| PrintExpr(Node->getLHS()); |
| OS << " " << BinaryOperator::getOpcodeStr(Node->getOpcode()) << " "; |
| PrintExpr(Node->getRHS()); |
| } |
| |
| void StmtPrinter::VisitConditionalOperator(ConditionalOperator *Node) { |
| PrintExpr(Node->getCond()); |
| OS << " ? "; |
| PrintExpr(Node->getLHS()); |
| OS << " : "; |
| PrintExpr(Node->getRHS()); |
| } |
| |
| // GNU extensions. |
| |
| void |
| StmtPrinter::VisitBinaryConditionalOperator(BinaryConditionalOperator *Node) { |
| PrintExpr(Node->getCommon()); |
| OS << " ?: "; |
| PrintExpr(Node->getFalseExpr()); |
| } |
| |
| void StmtPrinter::VisitAddrLabelExpr(AddrLabelExpr *Node) { |
| OS << "&&" << Node->getLabel()->getName(); |
| } |
| |
| void StmtPrinter::VisitStmtExpr(StmtExpr *E) { |
| OS << "("; |
| PrintRawCompoundStmt(E->getSubStmt()); |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitChooseExpr(ChooseExpr *Node) { |
| OS << "__builtin_choose_expr("; |
| PrintExpr(Node->getCond()); |
| OS << ", "; |
| PrintExpr(Node->getLHS()); |
| OS << ", "; |
| PrintExpr(Node->getRHS()); |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitGNUNullExpr(GNUNullExpr *) { |
| OS << "__null"; |
| } |
| |
| void StmtPrinter::VisitShuffleVectorExpr(ShuffleVectorExpr *Node) { |
| OS << "__builtin_shufflevector("; |
| for (unsigned i = 0, e = Node->getNumSubExprs(); i != e; ++i) { |
| if (i) OS << ", "; |
| PrintExpr(Node->getExpr(i)); |
| } |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitConvertVectorExpr(ConvertVectorExpr *Node) { |
| OS << "__builtin_convertvector("; |
| PrintExpr(Node->getSrcExpr()); |
| OS << ", "; |
| Node->getType().print(OS, Policy); |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitInitListExpr(InitListExpr* Node) { |
| if (Node->getSyntacticForm()) { |
| Visit(Node->getSyntacticForm()); |
| return; |
| } |
| |
| OS << "{"; |
| for (unsigned i = 0, e = Node->getNumInits(); i != e; ++i) { |
| if (i) OS << ", "; |
| if (Node->getInit(i)) |
| PrintExpr(Node->getInit(i)); |
| else |
| OS << "{}"; |
| } |
| OS << "}"; |
| } |
| |
| void StmtPrinter::VisitArrayInitLoopExpr(ArrayInitLoopExpr *Node) { |
| // There's no way to express this expression in any of our supported |
| // languages, so just emit something terse and (hopefully) clear. |
| OS << "{"; |
| PrintExpr(Node->getSubExpr()); |
| OS << "}"; |
| } |
| |
| void StmtPrinter::VisitArrayInitIndexExpr(ArrayInitIndexExpr *Node) { |
| OS << "*"; |
| } |
| |
| void StmtPrinter::VisitParenListExpr(ParenListExpr* Node) { |
| OS << "("; |
| for (unsigned i = 0, e = Node->getNumExprs(); i != e; ++i) { |
| if (i) OS << ", "; |
| PrintExpr(Node->getExpr(i)); |
| } |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitDesignatedInitExpr(DesignatedInitExpr *Node) { |
| bool NeedsEquals = true; |
| for (const DesignatedInitExpr::Designator &D : Node->designators()) { |
| if (D.isFieldDesignator()) { |
| if (D.getDotLoc().isInvalid()) { |
| if (IdentifierInfo *II = D.getFieldName()) { |
| OS << II->getName() << ":"; |
| NeedsEquals = false; |
| } |
| } else { |
| OS << "." << D.getFieldName()->getName(); |
| } |
| } else { |
| OS << "["; |
| if (D.isArrayDesignator()) { |
| PrintExpr(Node->getArrayIndex(D)); |
| } else { |
| PrintExpr(Node->getArrayRangeStart(D)); |
| OS << " ... "; |
| PrintExpr(Node->getArrayRangeEnd(D)); |
| } |
| OS << "]"; |
| } |
| } |
| |
| if (NeedsEquals) |
| OS << " = "; |
| else |
| OS << " "; |
| PrintExpr(Node->getInit()); |
| } |
| |
| void StmtPrinter::VisitDesignatedInitUpdateExpr( |
| DesignatedInitUpdateExpr *Node) { |
| OS << "{"; |
| OS << "/*base*/"; |
| PrintExpr(Node->getBase()); |
| OS << ", "; |
| |
| OS << "/*updater*/"; |
| PrintExpr(Node->getUpdater()); |
| OS << "}"; |
| } |
| |
| void StmtPrinter::VisitNoInitExpr(NoInitExpr *Node) { |
| OS << "/*no init*/"; |
| } |
| |
| void StmtPrinter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *Node) { |
| if (Node->getType()->getAsCXXRecordDecl()) { |
| OS << "/*implicit*/"; |
| Node->getType().print(OS, Policy); |
| OS << "()"; |
| } else { |
| OS << "/*implicit*/("; |
| Node->getType().print(OS, Policy); |
| OS << ')'; |
| if (Node->getType()->isRecordType()) |
| OS << "{}"; |
| else |
| OS << 0; |
| } |
| } |
| |
| void StmtPrinter::VisitVAArgExpr(VAArgExpr *Node) { |
| OS << "__builtin_va_arg("; |
| PrintExpr(Node->getSubExpr()); |
| OS << ", "; |
| Node->getType().print(OS, Policy); |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitPseudoObjectExpr(PseudoObjectExpr *Node) { |
| PrintExpr(Node->getSyntacticForm()); |
| } |
| |
| void StmtPrinter::VisitAtomicExpr(AtomicExpr *Node) { |
| const char *Name = nullptr; |
| switch (Node->getOp()) { |
| #define BUILTIN(ID, TYPE, ATTRS) |
| #define ATOMIC_BUILTIN(ID, TYPE, ATTRS) \ |
| case AtomicExpr::AO ## ID: \ |
| Name = #ID "("; \ |
| break; |
| #include "clang/Basic/Builtins.def" |
| } |
| OS << Name; |
| |
| // AtomicExpr stores its subexpressions in a permuted order. |
| PrintExpr(Node->getPtr()); |
| if (Node->getOp() != AtomicExpr::AO__c11_atomic_load && |
| Node->getOp() != AtomicExpr::AO__atomic_load_n && |
| Node->getOp() != AtomicExpr::AO__opencl_atomic_load) { |
| OS << ", "; |
| PrintExpr(Node->getVal1()); |
| } |
| if (Node->getOp() == AtomicExpr::AO__atomic_exchange || |
| Node->isCmpXChg()) { |
| OS << ", "; |
| PrintExpr(Node->getVal2()); |
| } |
| if (Node->getOp() == AtomicExpr::AO__atomic_compare_exchange || |
| Node->getOp() == AtomicExpr::AO__atomic_compare_exchange_n) { |
| OS << ", "; |
| PrintExpr(Node->getWeak()); |
| } |
| if (Node->getOp() != AtomicExpr::AO__c11_atomic_init && |
| Node->getOp() != AtomicExpr::AO__opencl_atomic_init) { |
| OS << ", "; |
| PrintExpr(Node->getOrder()); |
| } |
| if (Node->isCmpXChg()) { |
| OS << ", "; |
| PrintExpr(Node->getOrderFail()); |
| } |
| OS << ")"; |
| } |
| |
| // C++ |
| void StmtPrinter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *Node) { |
| const char *OpStrings[NUM_OVERLOADED_OPERATORS] = { |
| "", |
| #define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \ |
| Spelling, |
| #include "clang/Basic/OperatorKinds.def" |
| }; |
| |
| OverloadedOperatorKind Kind = Node->getOperator(); |
| if (Kind == OO_PlusPlus || Kind == OO_MinusMinus) { |
| if (Node->getNumArgs() == 1) { |
| OS << OpStrings[Kind] << ' '; |
| PrintExpr(Node->getArg(0)); |
| } else { |
| PrintExpr(Node->getArg(0)); |
| OS << ' ' << OpStrings[Kind]; |
| } |
| } else if (Kind == OO_Arrow) { |
| PrintExpr(Node->getArg(0)); |
| } else if (Kind == OO_Call) { |
| PrintExpr(Node->getArg(0)); |
| OS << '('; |
| for (unsigned ArgIdx = 1; ArgIdx < Node->getNumArgs(); ++ArgIdx) { |
| if (ArgIdx > 1) |
| OS << ", "; |
| if (!isa<CXXDefaultArgExpr>(Node->getArg(ArgIdx))) |
| PrintExpr(Node->getArg(ArgIdx)); |
| } |
| OS << ')'; |
| } else if (Kind == OO_Subscript) { |
| PrintExpr(Node->getArg(0)); |
| OS << '['; |
| PrintExpr(Node->getArg(1)); |
| OS << ']'; |
| } else if (Node->getNumArgs() == 1) { |
| OS << OpStrings[Kind] << ' '; |
| PrintExpr(Node->getArg(0)); |
| } else if (Node->getNumArgs() == 2) { |
| PrintExpr(Node->getArg(0)); |
| OS << ' ' << OpStrings[Kind] << ' '; |
| PrintExpr(Node->getArg(1)); |
| } else { |
| llvm_unreachable("unknown overloaded operator"); |
| } |
| } |
| |
| void StmtPrinter::VisitCXXMemberCallExpr(CXXMemberCallExpr *Node) { |
| // If we have a conversion operator call only print the argument. |
| CXXMethodDecl *MD = Node->getMethodDecl(); |
| if (MD && isa<CXXConversionDecl>(MD)) { |
| PrintExpr(Node->getImplicitObjectArgument()); |
| return; |
| } |
| VisitCallExpr(cast<CallExpr>(Node)); |
| } |
| |
| void StmtPrinter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *Node) { |
| PrintExpr(Node->getCallee()); |
| OS << "<<<"; |
| PrintCallArgs(Node->getConfig()); |
| OS << ">>>("; |
| PrintCallArgs(Node); |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitCXXNamedCastExpr(CXXNamedCastExpr *Node) { |
| OS << Node->getCastName() << '<'; |
| Node->getTypeAsWritten().print(OS, Policy); |
| OS << ">("; |
| PrintExpr(Node->getSubExpr()); |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitCXXStaticCastExpr(CXXStaticCastExpr *Node) { |
| VisitCXXNamedCastExpr(Node); |
| } |
| |
| void StmtPrinter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *Node) { |
| VisitCXXNamedCastExpr(Node); |
| } |
| |
| void StmtPrinter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *Node) { |
| VisitCXXNamedCastExpr(Node); |
| } |
| |
| void StmtPrinter::VisitCXXConstCastExpr(CXXConstCastExpr *Node) { |
| VisitCXXNamedCastExpr(Node); |
| } |
| |
| void StmtPrinter::VisitCXXTypeidExpr(CXXTypeidExpr *Node) { |
| OS << "typeid("; |
| if (Node->isTypeOperand()) { |
| Node->getTypeOperandSourceInfo()->getType().print(OS, Policy); |
| } else { |
| PrintExpr(Node->getExprOperand()); |
| } |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitCXXUuidofExpr(CXXUuidofExpr *Node) { |
| OS << "__uuidof("; |
| if (Node->isTypeOperand()) { |
| Node->getTypeOperandSourceInfo()->getType().print(OS, Policy); |
| } else { |
| PrintExpr(Node->getExprOperand()); |
| } |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitMSPropertyRefExpr(MSPropertyRefExpr *Node) { |
| PrintExpr(Node->getBaseExpr()); |
| if (Node->isArrow()) |
| OS << "->"; |
| else |
| OS << "."; |
| if (NestedNameSpecifier *Qualifier = |
| Node->getQualifierLoc().getNestedNameSpecifier()) |
| Qualifier->print(OS, Policy); |
| OS << Node->getPropertyDecl()->getDeclName(); |
| } |
| |
| void StmtPrinter::VisitMSPropertySubscriptExpr(MSPropertySubscriptExpr *Node) { |
| PrintExpr(Node->getBase()); |
| OS << "["; |
| PrintExpr(Node->getIdx()); |
| OS << "]"; |
| } |
| |
| void StmtPrinter::VisitUserDefinedLiteral(UserDefinedLiteral *Node) { |
| switch (Node->getLiteralOperatorKind()) { |
| case UserDefinedLiteral::LOK_Raw: |
| OS << cast<StringLiteral>(Node->getArg(0)->IgnoreImpCasts())->getString(); |
| break; |
| case UserDefinedLiteral::LOK_Template: { |
| const auto *DRE = cast<DeclRefExpr>(Node->getCallee()->IgnoreImpCasts()); |
| const TemplateArgumentList *Args = |
| cast<FunctionDecl>(DRE->getDecl())->getTemplateSpecializationArgs(); |
| assert(Args); |
| |
| if (Args->size() != 1) { |
| OS << "operator\"\"" << Node->getUDSuffix()->getName(); |
| printTemplateArgumentList(OS, Args->asArray(), Policy); |
| OS << "()"; |
| return; |
| } |
| |
| const TemplateArgument &Pack = Args->get(0); |
| for (const auto &P : Pack.pack_elements()) { |
| char C = (char)P.getAsIntegral().getZExtValue(); |
| OS << C; |
| } |
| break; |
| } |
| case UserDefinedLiteral::LOK_Integer: { |
| // Print integer literal without suffix. |
| const auto *Int = cast<IntegerLiteral>(Node->getCookedLiteral()); |
| OS << Int->getValue().toString(10, /*isSigned*/false); |
| break; |
| } |
| case UserDefinedLiteral::LOK_Floating: { |
| // Print floating literal without suffix. |
| auto *Float = cast<FloatingLiteral>(Node->getCookedLiteral()); |
| PrintFloatingLiteral(OS, Float, /*PrintSuffix=*/false); |
| break; |
| } |
| case UserDefinedLiteral::LOK_String: |
| case UserDefinedLiteral::LOK_Character: |
| PrintExpr(Node->getCookedLiteral()); |
| break; |
| } |
| OS << Node->getUDSuffix()->getName(); |
| } |
| |
| void StmtPrinter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *Node) { |
| OS << (Node->getValue() ? "true" : "false"); |
| } |
| |
| void StmtPrinter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *Node) { |
| OS << "nullptr"; |
| } |
| |
| void StmtPrinter::VisitCXXThisExpr(CXXThisExpr *Node) { |
| OS << "this"; |
| } |
| |
| void StmtPrinter::VisitCXXThrowExpr(CXXThrowExpr *Node) { |
| if (!Node->getSubExpr()) |
| OS << "throw"; |
| else { |
| OS << "throw "; |
| PrintExpr(Node->getSubExpr()); |
| } |
| } |
| |
| void StmtPrinter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *Node) { |
| // Nothing to print: we picked up the default argument. |
| } |
| |
| void StmtPrinter::VisitCXXDefaultInitExpr(CXXDefaultInitExpr *Node) { |
| // Nothing to print: we picked up the default initializer. |
| } |
| |
| void StmtPrinter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *Node) { |
| Node->getType().print(OS, Policy); |
| // If there are no parens, this is list-initialization, and the braces are |
| // part of the syntax of the inner construct. |
| if (Node->getLParenLoc().isValid()) |
| OS << "("; |
| PrintExpr(Node->getSubExpr()); |
| if (Node->getLParenLoc().isValid()) |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *Node) { |
| PrintExpr(Node->getSubExpr()); |
| } |
| |
| void StmtPrinter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *Node) { |
| Node->getType().print(OS, Policy); |
| if (Node->isStdInitListInitialization()) |
| /* Nothing to do; braces are part of creating the std::initializer_list. */; |
| else if (Node->isListInitialization()) |
| OS << "{"; |
| else |
| OS << "("; |
| for (CXXTemporaryObjectExpr::arg_iterator Arg = Node->arg_begin(), |
| ArgEnd = Node->arg_end(); |
| Arg != ArgEnd; ++Arg) { |
| if ((*Arg)->isDefaultArgument()) |
| break; |
| if (Arg != Node->arg_begin()) |
| OS << ", "; |
| PrintExpr(*Arg); |
| } |
| if (Node->isStdInitListInitialization()) |
| /* See above. */; |
| else if (Node->isListInitialization()) |
| OS << "}"; |
| else |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitLambdaExpr(LambdaExpr *Node) { |
| OS << '['; |
| bool NeedComma = false; |
| switch (Node->getCaptureDefault()) { |
| case LCD_None: |
| break; |
| |
| case LCD_ByCopy: |
| OS << '='; |
| NeedComma = true; |
| break; |
| |
| case LCD_ByRef: |
| OS << '&'; |
| NeedComma = true; |
| break; |
| } |
| for (LambdaExpr::capture_iterator C = Node->explicit_capture_begin(), |
| CEnd = Node->explicit_capture_end(); |
| C != CEnd; |
| ++C) { |
| if (C->capturesVLAType()) |
| continue; |
| |
| if (NeedComma) |
| OS << ", "; |
| NeedComma = true; |
| |
| switch (C->getCaptureKind()) { |
| case LCK_This: |
| OS << "this"; |
| break; |
| |
| case LCK_StarThis: |
| OS << "*this"; |
| break; |
| |
| case LCK_ByRef: |
| if (Node->getCaptureDefault() != LCD_ByRef || Node->isInitCapture(C)) |
| OS << '&'; |
| OS << C->getCapturedVar()->getName(); |
| break; |
| |
| case LCK_ByCopy: |
| OS << C->getCapturedVar()->getName(); |
| break; |
| |
| case LCK_VLAType: |
| llvm_unreachable("VLA type in explicit captures."); |
| } |
| |
| if (Node->isInitCapture(C)) |
| PrintExpr(C->getCapturedVar()->getInit()); |
| } |
| OS << ']'; |
| |
| if (Node->hasExplicitParameters()) { |
| OS << " ("; |
| CXXMethodDecl *Method = Node->getCallOperator(); |
| NeedComma = false; |
| for (const auto *P : Method->parameters()) { |
| if (NeedComma) { |
| OS << ", "; |
| } else { |
| NeedComma = true; |
| } |
| std::string ParamStr = P->getNameAsString(); |
| P->getOriginalType().print(OS, Policy, ParamStr); |
| } |
| if (Method->isVariadic()) { |
| if (NeedComma) |
| OS << ", "; |
| OS << "..."; |
| } |
| OS << ')'; |
| |
| if (Node->isMutable()) |
| OS << " mutable"; |
| |
| auto *Proto = Method->getType()->getAs<FunctionProtoType>(); |
| Proto->printExceptionSpecification(OS, Policy); |
| |
| // FIXME: Attributes |
| |
| // Print the trailing return type if it was specified in the source. |
| if (Node->hasExplicitResultType()) { |
| OS << " -> "; |
| Proto->getReturnType().print(OS, Policy); |
| } |
| } |
| |
| // Print the body. |
| CompoundStmt *Body = Node->getBody(); |
| OS << ' '; |
| PrintStmt(Body); |
| } |
| |
| void StmtPrinter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *Node) { |
| if (TypeSourceInfo *TSInfo = Node->getTypeSourceInfo()) |
| TSInfo->getType().print(OS, Policy); |
| else |
| Node->getType().print(OS, Policy); |
| OS << "()"; |
| } |
| |
| void StmtPrinter::VisitCXXNewExpr(CXXNewExpr *E) { |
| if (E->isGlobalNew()) |
| OS << "::"; |
| OS << "new "; |
| unsigned NumPlace = E->getNumPlacementArgs(); |
| if (NumPlace > 0 && !isa<CXXDefaultArgExpr>(E->getPlacementArg(0))) { |
| OS << "("; |
| PrintExpr(E->getPlacementArg(0)); |
| for (unsigned i = 1; i < NumPlace; ++i) { |
| if (isa<CXXDefaultArgExpr>(E->getPlacementArg(i))) |
| break; |
| OS << ", "; |
| PrintExpr(E->getPlacementArg(i)); |
| } |
| OS << ") "; |
| } |
| if (E->isParenTypeId()) |
| OS << "("; |
| std::string TypeS; |
| if (Expr *Size = E->getArraySize()) { |
| llvm::raw_string_ostream s(TypeS); |
| s << '['; |
| Size->printPretty(s, Helper, Policy); |
| s << ']'; |
| } |
| E->getAllocatedType().print(OS, Policy, TypeS); |
| if (E->isParenTypeId()) |
| OS << ")"; |
| |
| CXXNewExpr::InitializationStyle InitStyle = E->getInitializationStyle(); |
| if (InitStyle) { |
| if (InitStyle == CXXNewExpr::CallInit) |
| OS << "("; |
| PrintExpr(E->getInitializer()); |
| if (InitStyle == CXXNewExpr::CallInit) |
| OS << ")"; |
| } |
| } |
| |
| void StmtPrinter::VisitCXXDeleteExpr(CXXDeleteExpr *E) { |
| if (E->isGlobalDelete()) |
| OS << "::"; |
| OS << "delete "; |
| if (E->isArrayForm()) |
| OS << "[] "; |
| PrintExpr(E->getArgument()); |
| } |
| |
| void StmtPrinter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) { |
| PrintExpr(E->getBase()); |
| if (E->isArrow()) |
| OS << "->"; |
| else |
| OS << '.'; |
| if (E->getQualifier()) |
| E->getQualifier()->print(OS, Policy); |
| OS << "~"; |
| |
| if (IdentifierInfo *II = E->getDestroyedTypeIdentifier()) |
| OS << II->getName(); |
| else |
| E->getDestroyedType().print(OS, Policy); |
| } |
| |
| void StmtPrinter::VisitCXXConstructExpr(CXXConstructExpr *E) { |
| if (E->isListInitialization() && !E->isStdInitListInitialization()) |
| OS << "{"; |
| |
| for (unsigned i = 0, e = E->getNumArgs(); i != e; ++i) { |
| if (isa<CXXDefaultArgExpr>(E->getArg(i))) { |
| // Don't print any defaulted arguments |
| break; |
| } |
| |
| if (i) OS << ", "; |
| PrintExpr(E->getArg(i)); |
| } |
| |
| if (E->isListInitialization() && !E->isStdInitListInitialization()) |
| OS << "}"; |
| } |
| |
| void StmtPrinter::VisitCXXInheritedCtorInitExpr(CXXInheritedCtorInitExpr *E) { |
| // Parens are printed by the surrounding context. |
| OS << "<forwarded>"; |
| } |
| |
| void StmtPrinter::VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E) { |
| PrintExpr(E->getSubExpr()); |
| } |
| |
| void StmtPrinter::VisitExprWithCleanups(ExprWithCleanups *E) { |
| // Just forward to the subexpression. |
| PrintExpr(E->getSubExpr()); |
| } |
| |
| void |
| StmtPrinter::VisitCXXUnresolvedConstructExpr( |
| CXXUnresolvedConstructExpr *Node) { |
| Node->getTypeAsWritten().print(OS, Policy); |
| OS << "("; |
| for (CXXUnresolvedConstructExpr::arg_iterator Arg = Node->arg_begin(), |
| ArgEnd = Node->arg_end(); |
| Arg != ArgEnd; ++Arg) { |
| if (Arg != Node->arg_begin()) |
| OS << ", "; |
| PrintExpr(*Arg); |
| } |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitCXXDependentScopeMemberExpr( |
| CXXDependentScopeMemberExpr *Node) { |
| if (!Node->isImplicitAccess()) { |
| PrintExpr(Node->getBase()); |
| OS << (Node->isArrow() ? "->" : "."); |
| } |
| if (NestedNameSpecifier *Qualifier = Node->getQualifier()) |
| Qualifier->print(OS, Policy); |
| if (Node->hasTemplateKeyword()) |
| OS << "template "; |
| OS << Node->getMemberNameInfo(); |
| if (Node->hasExplicitTemplateArgs()) |
| printTemplateArgumentList(OS, Node->template_arguments(), Policy); |
| } |
| |
| void StmtPrinter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *Node) { |
| if (!Node->isImplicitAccess()) { |
| PrintExpr(Node->getBase()); |
| OS << (Node->isArrow() ? "->" : "."); |
| } |
| if (NestedNameSpecifier *Qualifier = Node->getQualifier()) |
| Qualifier->print(OS, Policy); |
| if (Node->hasTemplateKeyword()) |
| OS << "template "; |
| OS << Node->getMemberNameInfo(); |
| if (Node->hasExplicitTemplateArgs()) |
| printTemplateArgumentList(OS, Node->template_arguments(), Policy); |
| } |
| |
| static const char *getTypeTraitName(TypeTrait TT) { |
| switch (TT) { |
| #define TYPE_TRAIT_1(Spelling, Name, Key) \ |
| case clang::UTT_##Name: return #Spelling; |
| #define TYPE_TRAIT_2(Spelling, Name, Key) \ |
| case clang::BTT_##Name: return #Spelling; |
| #define TYPE_TRAIT_N(Spelling, Name, Key) \ |
| case clang::TT_##Name: return #Spelling; |
| #include "clang/Basic/TokenKinds.def" |
| } |
| llvm_unreachable("Type trait not covered by switch"); |
| } |
| |
| static const char *getTypeTraitName(ArrayTypeTrait ATT) { |
| switch (ATT) { |
| case ATT_ArrayRank: return "__array_rank"; |
| case ATT_ArrayExtent: return "__array_extent"; |
| } |
| llvm_unreachable("Array type trait not covered by switch"); |
| } |
| |
| static const char *getExpressionTraitName(ExpressionTrait ET) { |
| switch (ET) { |
| case ET_IsLValueExpr: return "__is_lvalue_expr"; |
| case ET_IsRValueExpr: return "__is_rvalue_expr"; |
| } |
| llvm_unreachable("Expression type trait not covered by switch"); |
| } |
| |
| void StmtPrinter::VisitTypeTraitExpr(TypeTraitExpr *E) { |
| OS << getTypeTraitName(E->getTrait()) << "("; |
| for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I) { |
| if (I > 0) |
| OS << ", "; |
| E->getArg(I)->getType().print(OS, Policy); |
| } |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) { |
| OS << getTypeTraitName(E->getTrait()) << '('; |
| E->getQueriedType().print(OS, Policy); |
| OS << ')'; |
| } |
| |
| void StmtPrinter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) { |
| OS << getExpressionTraitName(E->getTrait()) << '('; |
| PrintExpr(E->getQueriedExpression()); |
| OS << ')'; |
| } |
| |
| void StmtPrinter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) { |
| OS << "noexcept("; |
| PrintExpr(E->getOperand()); |
| OS << ")"; |
| } |
| |
| void StmtPrinter::VisitPackExpansionExpr(PackExpansionExpr *E) { |
| PrintExpr(E->getPattern()); |
| OS << "..."; |
| } |
| |
| void StmtPrinter::VisitSizeOfPackExpr(SizeOfPackExpr *E) { |
| OS << "sizeof...(" << *E->getPack() << ")"; |
| } |
| |
| void StmtPrinter::VisitSubstNonTypeTemplateParmPackExpr( |
| SubstNonTypeTemplateParmPackExpr *Node) { |
| OS << *Node->getParameterPack(); |
| } |
| |
| void StmtPrinter::VisitSubstNonTypeTemplateParmExpr( |
| SubstNonTypeTemplateParmExpr *Node) { |
| Visit(Node->getReplacement()); |
| } |
| |
| void StmtPrinter::VisitFunctionParmPackExpr(FunctionParmPackExpr *E) { |
| OS << *E->getParameterPack(); |
| } |
| |
| void StmtPrinter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *Node){ |
| PrintExpr(Node->GetTemporaryExpr()); |
| } |
| |
| void StmtPrinter::VisitCXXFoldExpr(CXXFoldExpr *E) { |
| OS << "("; |
| if (E->getLHS()) { |
| PrintExpr(E->getLHS()); |
| OS << " " << BinaryOperator::getOpcodeStr(E->getOperator()) << " "; |
| } |
| OS << "..."; |
| if (E->getRHS()) { |
| OS << " " << BinaryOperator::getOpcodeStr(E->getOperator()) << " "; |
| PrintExpr(E->getRHS()); |
| } |
| OS << ")"; |
| } |
| |
| // C++ Coroutines TS |
| |
| void StmtPrinter::VisitCoroutineBodyStmt(CoroutineBodyStmt *S) { |
| Visit(S->getBody()); |
| } |
| |
| void StmtPrinter::VisitCoreturnStmt(CoreturnStmt *S) { |
| OS << "co_return"; |
| if (S->getOperand()) { |
| OS << " "; |
| Visit(S->getOperand()); |
| } |
| OS << ";"; |
| } |
| |
| void StmtPrinter::VisitCoawaitExpr(CoawaitExpr *S) { |
| OS << "co_await "; |
| PrintExpr(S->getOperand()); |
| } |
| |
| void StmtPrinter::VisitDependentCoawaitExpr(DependentCoawaitExpr *S) { |
| OS << "co_await "; |
| PrintExpr(S->getOperand()); |
| } |
| |
| void StmtPrinter::VisitCoyieldExpr(CoyieldExpr *S) { |
| OS << "co_yield "; |
| PrintExpr(S->getOperand()); |
| } |
| |
| // Obj-C |
| |
| void StmtPrinter::VisitObjCStringLiteral(ObjCStringLiteral *Node) { |
| OS << "@"; |
| VisitStringLiteral(Node->getString()); |
| } |
| |
| void StmtPrinter::VisitObjCBoxedExpr(ObjCBoxedExpr *E) { |
| OS << "@"; |
| Visit(E->getSubExpr()); |
| } |
| |
| void StmtPrinter::VisitObjCArrayLiteral(ObjCArrayLiteral *E) { |
| OS << "@[ "; |
| ObjCArrayLiteral::child_range Ch = E->children(); |
| for (auto I = Ch.begin(), E = Ch.end(); I != E; ++I) { |
| if (I != Ch.begin()) |
| OS << ", "; |
| Visit(*I); |
| } |
| OS << " ]"; |
| } |
| |
| void StmtPrinter::VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) { |
| OS << "@{ "; |
| for (unsigned I = 0, N = E->getNumElements(); I != N; ++I) { |
| if (I > 0) |
| OS << ", "; |
| |
| ObjCDictionaryElement Element = E->getKeyValueElement(I); |
| Visit(Element.Key); |
| OS << " : "; |
| Visit(Element.Value); |
| if (Element.isPackExpansion()) |
| OS << "..."; |
| } |
| OS << " }"; |
| } |
| |
| void StmtPrinter::VisitObjCEncodeExpr(ObjCEncodeExpr *Node) { |
| OS << "@encode("; |
| Node->getEncodedType().print(OS, Policy); |
| OS << ')'; |
| } |
| |
| void StmtPrinter::VisitObjCSelectorExpr(ObjCSelectorExpr *Node) { |
| OS << "@selector("; |
| Node->getSelector().print(OS); |
| OS << ')'; |
| } |
| |
| void StmtPrinter::VisitObjCProtocolExpr(ObjCProtocolExpr *Node) { |
| OS << "@protocol(" << *Node->getProtocol() << ')'; |
| } |
| |
| void StmtPrinter::VisitObjCMessageExpr(ObjCMessageExpr *Mess) { |
| OS << "["; |
| switch (Mess->getReceiverKind()) { |
| case ObjCMessageExpr::Instance: |
| PrintExpr(Mess->getInstanceReceiver()); |
| break; |
| |
| case ObjCMessageExpr::Class: |
| Mess->getClassReceiver().print(OS, Policy); |
| break; |
| |
| case ObjCMessageExpr::SuperInstance: |
| case ObjCMessageExpr::SuperClass: |
| OS << "Super"; |
| break; |
| } |
| |
| OS << ' '; |
| Selector selector = Mess->getSelector(); |
| if (selector.isUnarySelector()) { |
| OS << selector.getNameForSlot(0); |
| } else { |
| for (unsigned i = 0, e = Mess->getNumArgs(); i != e; ++i) { |
| if (i < selector.getNumArgs()) { |
| if (i > 0) OS << ' '; |
| if (selector.getIdentifierInfoForSlot(i)) |
| OS << selector.getIdentifierInfoForSlot(i)->getName() << ':'; |
| else |
| OS << ":"; |
| } |
| else OS << ", "; // Handle variadic methods. |
| |
| PrintExpr(Mess->getArg(i)); |
| } |
| } |
| OS << "]"; |
| } |
| |
| void StmtPrinter::VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr *Node) { |
| OS << (Node->getValue() ? "__objc_yes" : "__objc_no"); |
| } |
| |
| void |
| StmtPrinter::VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) { |
| PrintExpr(E->getSubExpr()); |
| } |
| |
| void |
| StmtPrinter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) { |
| OS << '(' << E->getBridgeKindName(); |
| E->getType().print(OS, Policy); |
| OS << ')'; |
| PrintExpr(E->getSubExpr()); |
| } |
| |
| void StmtPrinter::VisitBlockExpr(BlockExpr *Node) { |
| BlockDecl *BD = Node->getBlockDecl(); |
| OS << "^"; |
| |
| const FunctionType *AFT = Node->getFunctionType(); |
| |
| if (isa<FunctionNoProtoType>(AFT)) { |
| OS << "()"; |
| } else if (!BD->param_empty() || cast<FunctionProtoType>(AFT)->isVariadic()) { |
| OS << '('; |
| for (BlockDecl::param_iterator AI = BD->param_begin(), |
| E = BD->param_end(); AI != E; ++AI) { |
| if (AI != BD->param_begin()) OS << ", "; |
| std::string ParamStr = (*AI)->getNameAsString(); |
| (*AI)->getType().print(OS, Policy, ParamStr); |
| } |
| |
| const auto *FT = cast<FunctionProtoType>(AFT); |
| if (FT->isVariadic()) { |
| if (!BD->param_empty()) OS << ", "; |
| OS << "..."; |
| } |
| OS << ')'; |
| } |
| OS << "{ }"; |
| } |
| |
| void StmtPrinter::VisitOpaqueValueExpr(OpaqueValueExpr *Node) { |
| PrintExpr(Node->getSourceExpr()); |
| } |
| |
| void StmtPrinter::VisitTypoExpr(TypoExpr *Node) { |
| // TODO: Print something reasonable for a TypoExpr, if necessary. |
| llvm_unreachable("Cannot print TypoExpr nodes"); |
| } |
| |
| void StmtPrinter::VisitAsTypeExpr(AsTypeExpr *Node) { |
| OS << "__builtin_astype("; |
| PrintExpr(Node->getSrcExpr()); |
| OS << ", "; |
| Node->getType().print(OS, Policy); |
| OS << ")"; |
| } |
| |
| //===----------------------------------------------------------------------===// |
| // Stmt method implementations |
| //===----------------------------------------------------------------------===// |
| |
| void Stmt::dumpPretty(const ASTContext &Context) const { |
| printPretty(llvm::errs(), nullptr, PrintingPolicy(Context.getLangOpts())); |
| } |
| |
| void Stmt::printPretty(raw_ostream &OS, PrinterHelper *Helper, |
| const PrintingPolicy &Policy, unsigned Indentation, |
| const ASTContext *Context) const { |
| StmtPrinter P(OS, Helper, Policy, Indentation, Context); |
| P.Visit(const_cast<Stmt*>(this)); |
| } |
| |
| //===----------------------------------------------------------------------===// |
| // PrinterHelper |
| //===----------------------------------------------------------------------===// |
| |
| // Implement virtual destructor. |
| PrinterHelper::~PrinterHelper() = default; |