123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775 |
- //===- StmtPrinter.cpp - Printing implementation for Stmt ASTs ------------===//
- //
- // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
- // See https://llvm.org/LICENSE.txt for license information.
- // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
- //
- //===----------------------------------------------------------------------===//
- //
- // 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/JsonSupport.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/StringExtras.h"
- #include "llvm/ADT/StringRef.h"
- #include "llvm/Support/Casting.h"
- #include "llvm/Support/Compiler.h"
- #include "llvm/Support/ErrorHandling.h"
- #include "llvm/Support/raw_ostream.h"
- #include <cassert>
- #include <optional>
- #include <string>
- using namespace clang;
- //===----------------------------------------------------------------------===//
- // StmtPrinter Visitor
- //===----------------------------------------------------------------------===//
- namespace {
- class StmtPrinter : public StmtVisitor<StmtPrinter> {
- raw_ostream &OS;
- unsigned IndentLevel;
- PrinterHelper* Helper;
- PrintingPolicy Policy;
- std::string NL;
- const ASTContext *Context;
- public:
- StmtPrinter(raw_ostream &os, PrinterHelper *helper,
- const PrintingPolicy &Policy, unsigned Indentation = 0,
- StringRef NL = "\n", const ASTContext *Context = nullptr)
- : OS(os), IndentLevel(Indentation), Helper(helper), Policy(Policy),
- NL(NL), 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 << ";" << NL;
- } else if (S) {
- Visit(S);
- } else {
- Indent() << "<<<NULL STATEMENT>>>" << NL;
- }
- IndentLevel -= SubIndent;
- }
- void PrintInitStmt(Stmt *S, unsigned PrefixWidth) {
- // FIXME: Cope better with odd prefix widths.
- IndentLevel += (PrefixWidth + 1) / 2;
- if (auto *DS = dyn_cast<DeclStmt>(S))
- PrintRawDeclStmt(DS);
- else
- PrintExpr(cast<Expr>(S));
- OS << "; ";
- IndentLevel -= (PrefixWidth + 1) / 2;
- }
- void PrintControlledStmt(Stmt *S) {
- if (auto *CS = dyn_cast<CompoundStmt>(S)) {
- OS << " ";
- PrintRawCompoundStmt(CS);
- OS << NL;
- } else {
- OS << NL;
- PrintStmt(S);
- }
- }
- 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 PrintFPPragmas(CompoundStmt *S);
- 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>>" << NL;
- }
- 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 << "{" << NL;
- PrintFPPragmas(Node);
- for (auto *I : Node->body())
- PrintStmt(I);
- Indent() << "}";
- }
- void StmtPrinter::PrintFPPragmas(CompoundStmt *S) {
- if (!S->hasStoredFPFeatures())
- return;
- FPOptionsOverride FPO = S->getStoredFPFeatures();
- bool FEnvAccess = false;
- if (FPO.hasAllowFEnvAccessOverride()) {
- FEnvAccess = FPO.getAllowFEnvAccessOverride();
- Indent() << "#pragma STDC FENV_ACCESS " << (FEnvAccess ? "ON" : "OFF")
- << NL;
- }
- if (FPO.hasSpecifiedExceptionModeOverride()) {
- LangOptions::FPExceptionModeKind EM =
- FPO.getSpecifiedExceptionModeOverride();
- if (!FEnvAccess || EM != LangOptions::FPE_Strict) {
- Indent() << "#pragma clang fp exceptions(";
- switch (FPO.getSpecifiedExceptionModeOverride()) {
- default:
- break;
- case LangOptions::FPE_Ignore:
- OS << "ignore";
- break;
- case LangOptions::FPE_MayTrap:
- OS << "maytrap";
- break;
- case LangOptions::FPE_Strict:
- OS << "strict";
- break;
- }
- OS << ")\n";
- }
- }
- if (FPO.hasConstRoundingModeOverride()) {
- LangOptions::RoundingMode RM = FPO.getConstRoundingModeOverride();
- Indent() << "#pragma STDC FENV_ROUND ";
- switch (RM) {
- case llvm::RoundingMode::TowardZero:
- OS << "FE_TOWARDZERO";
- break;
- case llvm::RoundingMode::NearestTiesToEven:
- OS << "FE_TONEAREST";
- break;
- case llvm::RoundingMode::TowardPositive:
- OS << "FE_UPWARD";
- break;
- case llvm::RoundingMode::TowardNegative:
- OS << "FE_DOWNWARD";
- break;
- case llvm::RoundingMode::NearestTiesToAway:
- OS << "FE_TONEARESTFROMZERO";
- break;
- case llvm::RoundingMode::Dynamic:
- OS << "FE_DYNAMIC";
- break;
- default:
- llvm_unreachable("Invalid rounding mode");
- }
- OS << NL;
- }
- }
- 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() << ";" << NL;
- }
- void StmtPrinter::VisitDeclStmt(DeclStmt *Node) {
- Indent();
- PrintRawDeclStmt(Node);
- OS << ";" << NL;
- }
- void StmtPrinter::VisitCompoundStmt(CompoundStmt *Node) {
- Indent();
- PrintRawCompoundStmt(Node);
- OS << "" << NL;
- }
- void StmtPrinter::VisitCaseStmt(CaseStmt *Node) {
- Indent(-1) << "case ";
- PrintExpr(Node->getLHS());
- if (Node->getRHS()) {
- OS << " ... ";
- PrintExpr(Node->getRHS());
- }
- OS << ":" << NL;
- PrintStmt(Node->getSubStmt(), 0);
- }
- void StmtPrinter::VisitDefaultStmt(DefaultStmt *Node) {
- Indent(-1) << "default:" << NL;
- PrintStmt(Node->getSubStmt(), 0);
- }
- void StmtPrinter::VisitLabelStmt(LabelStmt *Node) {
- Indent(-1) << Node->getName() << ":" << NL;
- 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) {
- if (If->isConsteval()) {
- OS << "if ";
- if (If->isNegatedConsteval())
- OS << "!";
- OS << "consteval";
- OS << NL;
- PrintStmt(If->getThen());
- if (Stmt *Else = If->getElse()) {
- Indent();
- OS << "else";
- PrintStmt(Else);
- OS << NL;
- }
- return;
- }
- OS << "if (";
- if (If->getInit())
- PrintInitStmt(If->getInit(), 4);
- 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() ? " " : NL);
- } else {
- OS << NL;
- 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 << NL;
- } else if (auto *ElseIf = dyn_cast<IfStmt>(Else)) {
- OS << ' ';
- PrintRawIfStmt(ElseIf);
- } else {
- OS << NL;
- PrintStmt(If->getElse());
- }
- }
- }
- void StmtPrinter::VisitIfStmt(IfStmt *If) {
- Indent();
- PrintRawIfStmt(If);
- }
- void StmtPrinter::VisitSwitchStmt(SwitchStmt *Node) {
- Indent() << "switch (";
- if (Node->getInit())
- PrintInitStmt(Node->getInit(), 8);
- if (const DeclStmt *DS = Node->getConditionVariableDeclStmt())
- PrintRawDeclStmt(DS);
- else
- PrintExpr(Node->getCond());
- OS << ")";
- PrintControlledStmt(Node->getBody());
- }
- void StmtPrinter::VisitWhileStmt(WhileStmt *Node) {
- Indent() << "while (";
- if (const DeclStmt *DS = Node->getConditionVariableDeclStmt())
- PrintRawDeclStmt(DS);
- else
- PrintExpr(Node->getCond());
- OS << ")" << NL;
- PrintStmt(Node->getBody());
- }
- void StmtPrinter::VisitDoStmt(DoStmt *Node) {
- Indent() << "do ";
- if (auto *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
- PrintRawCompoundStmt(CS);
- OS << " ";
- } else {
- OS << NL;
- PrintStmt(Node->getBody());
- Indent();
- }
- OS << "while (";
- PrintExpr(Node->getCond());
- OS << ");" << NL;
- }
- void StmtPrinter::VisitForStmt(ForStmt *Node) {
- Indent() << "for (";
- if (Node->getInit())
- PrintInitStmt(Node->getInit(), 5);
- else
- OS << (Node->getCond() ? "; " : ";");
- if (Node->getCond())
- PrintExpr(Node->getCond());
- OS << ";";
- if (Node->getInc()) {
- OS << " ";
- PrintExpr(Node->getInc());
- }
- OS << ")";
- PrintControlledStmt(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 << ")";
- PrintControlledStmt(Node->getBody());
- }
- void StmtPrinter::VisitCXXForRangeStmt(CXXForRangeStmt *Node) {
- Indent() << "for (";
- if (Node->getInit())
- PrintInitStmt(Node->getInit(), 5);
- PrintingPolicy SubPolicy(Policy);
- SubPolicy.SuppressInitializers = true;
- Node->getLoopVariable()->print(OS, SubPolicy, IndentLevel);
- OS << " : ";
- PrintExpr(Node->getRangeInit());
- OS << ")";
- PrintControlledStmt(Node->getBody());
- }
- 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 << NL;
- }
- void StmtPrinter::VisitIndirectGotoStmt(IndirectGotoStmt *Node) {
- Indent() << "goto *";
- PrintExpr(Node->getTarget());
- OS << ";";
- if (Policy.IncludeNewlines) OS << NL;
- }
- void StmtPrinter::VisitContinueStmt(ContinueStmt *Node) {
- Indent() << "continue;";
- if (Policy.IncludeNewlines) OS << NL;
- }
- void StmtPrinter::VisitBreakStmt(BreakStmt *Node) {
- Indent() << "break;";
- if (Policy.IncludeNewlines) OS << NL;
- }
- void StmtPrinter::VisitReturnStmt(ReturnStmt *Node) {
- Indent() << "return";
- if (Node->getRetValue()) {
- OS << " ";
- PrintExpr(Node->getRetValue());
- }
- OS << ";";
- if (Policy.IncludeNewlines) OS << NL;
- }
- void StmtPrinter::VisitGCCAsmStmt(GCCAsmStmt *Node) {
- Indent() << "asm ";
- if (Node->isVolatile())
- OS << "volatile ";
- if (Node->isAsmGoto())
- OS << "goto ";
- OS << "(";
- VisitStringLiteral(Node->getAsmString());
- // Outputs
- if (Node->getNumOutputs() != 0 || Node->getNumInputs() != 0 ||
- Node->getNumClobbers() != 0 || Node->getNumLabels() != 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 ||
- Node->getNumLabels() != 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 || Node->getNumLabels())
- OS << " : ";
- for (unsigned i = 0, e = Node->getNumClobbers(); i != e; ++i) {
- if (i != 0)
- OS << ", ";
- VisitStringLiteral(Node->getClobberStringLiteral(i));
- }
- // Labels
- if (Node->getNumLabels() != 0)
- OS << " : ";
- for (unsigned i = 0, e = Node->getNumLabels(); i != e; ++i) {
- if (i != 0)
- OS << ", ";
- OS << Node->getLabelName(i);
- }
- OS << ");";
- if (Policy.IncludeNewlines) OS << NL;
- }
- void StmtPrinter::VisitMSAsmStmt(MSAsmStmt *Node) {
- // FIXME: Implement MS style inline asm statement printer.
- Indent() << "__asm ";
- if (Node->hasBraces())
- OS << "{" << NL;
- OS << Node->getAsmString() << NL;
- if (Node->hasBraces())
- Indent() << "}" << NL;
- }
- 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 << NL;
- }
- for (ObjCAtCatchStmt *catchStmt : Node->catch_stmts()) {
- Indent() << "@catch(";
- if (Decl *DS = catchStmt->getCatchParamDecl())
- PrintRawDecl(DS);
- OS << ")";
- if (auto *CS = dyn_cast<CompoundStmt>(catchStmt->getCatchBody())) {
- PrintRawCompoundStmt(CS);
- OS << NL;
- }
- }
- if (auto *FS = static_cast<ObjCAtFinallyStmt *>(Node->getFinallyStmt())) {
- Indent() << "@finally";
- PrintRawCompoundStmt(dyn_cast<CompoundStmt>(FS->getFinallyBody()));
- OS << NL;
- }
- }
- void StmtPrinter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *Node) {
- }
- void StmtPrinter::VisitObjCAtCatchStmt (ObjCAtCatchStmt *Node) {
- Indent() << "@catch (...) { /* todo */ } " << NL;
- }
- void StmtPrinter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *Node) {
- Indent() << "@throw";
- if (Node->getThrowExpr()) {
- OS << " ";
- PrintExpr(Node->getThrowExpr());
- }
- OS << ";" << NL;
- }
- void StmtPrinter::VisitObjCAvailabilityCheckExpr(
- ObjCAvailabilityCheckExpr *Node) {
- OS << "@available(...)";
- }
- void StmtPrinter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *Node) {
- Indent() << "@synchronized (";
- PrintExpr(Node->getSynchExpr());
- OS << ")";
- PrintRawCompoundStmt(Node->getSynchBody());
- OS << NL;
- }
- void StmtPrinter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *Node) {
- Indent() << "@autoreleasepool";
- PrintRawCompoundStmt(dyn_cast<CompoundStmt>(Node->getSubStmt()));
- OS << NL;
- }
- 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 << NL;
- }
- 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 << NL;
- }
- 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 << NL;
- }
- void StmtPrinter::PrintRawSEHFinallyStmt(SEHFinallyStmt *Node) {
- OS << "__finally ";
- PrintRawCompoundStmt(Node->getBlock());
- OS << NL;
- }
- void StmtPrinter::PrintRawSEHExceptHandler(SEHExceptStmt *Node) {
- OS << "__except (";
- VisitExpr(Node->getFilterExpr());
- OS << ")" << NL;
- PrintRawCompoundStmt(Node->getBlock());
- OS << NL;
- }
- void StmtPrinter::VisitSEHExceptStmt(SEHExceptStmt *Node) {
- Indent();
- PrintRawSEHExceptHandler(Node);
- OS << NL;
- }
- void StmtPrinter::VisitSEHFinallyStmt(SEHFinallyStmt *Node) {
- Indent();
- PrintRawSEHFinallyStmt(Node);
- OS << NL;
- }
- void StmtPrinter::VisitSEHLeaveStmt(SEHLeaveStmt *Node) {
- Indent() << "__leave;";
- if (Policy.IncludeNewlines) OS << NL;
- }
- //===----------------------------------------------------------------------===//
- // OpenMP directives printing methods
- //===----------------------------------------------------------------------===//
- void StmtPrinter::VisitOMPCanonicalLoop(OMPCanonicalLoop *Node) {
- PrintStmt(Node->getLoopStmt());
- }
- 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 << NL;
- if (!ForceNoStmt && S->hasAssociatedStmt())
- PrintStmt(S->getRawStmt());
- }
- void StmtPrinter::VisitOMPMetaDirective(OMPMetaDirective *Node) {
- Indent() << "#pragma omp metadirective";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPParallelDirective(OMPParallelDirective *Node) {
- Indent() << "#pragma omp parallel";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPSimdDirective(OMPSimdDirective *Node) {
- Indent() << "#pragma omp simd";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPTileDirective(OMPTileDirective *Node) {
- Indent() << "#pragma omp tile";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPUnrollDirective(OMPUnrollDirective *Node) {
- Indent() << "#pragma omp unroll";
- 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, Policy);
- 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::VisitOMPParallelMasterDirective(
- OMPParallelMasterDirective *Node) {
- Indent() << "#pragma omp parallel master";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPParallelMaskedDirective(
- OMPParallelMaskedDirective *Node) {
- Indent() << "#pragma omp parallel masked";
- 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::VisitOMPErrorDirective(OMPErrorDirective *Node) {
- Indent() << "#pragma omp error";
- 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::VisitOMPDepobjDirective(OMPDepobjDirective *Node) {
- Indent() << "#pragma omp depobj";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPScanDirective(OMPScanDirective *Node) {
- Indent() << "#pragma omp scan";
- 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::VisitOMPMasterTaskLoopDirective(
- OMPMasterTaskLoopDirective *Node) {
- Indent() << "#pragma omp master taskloop";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPMaskedTaskLoopDirective(
- OMPMaskedTaskLoopDirective *Node) {
- Indent() << "#pragma omp masked taskloop";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPMasterTaskLoopSimdDirective(
- OMPMasterTaskLoopSimdDirective *Node) {
- Indent() << "#pragma omp master taskloop simd";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPMaskedTaskLoopSimdDirective(
- OMPMaskedTaskLoopSimdDirective *Node) {
- Indent() << "#pragma omp masked taskloop simd";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPParallelMasterTaskLoopDirective(
- OMPParallelMasterTaskLoopDirective *Node) {
- Indent() << "#pragma omp parallel master taskloop";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPParallelMaskedTaskLoopDirective(
- OMPParallelMaskedTaskLoopDirective *Node) {
- Indent() << "#pragma omp parallel masked taskloop";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPParallelMasterTaskLoopSimdDirective(
- OMPParallelMasterTaskLoopSimdDirective *Node) {
- Indent() << "#pragma omp parallel master taskloop simd";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPParallelMaskedTaskLoopSimdDirective(
- OMPParallelMaskedTaskLoopSimdDirective *Node) {
- Indent() << "#pragma omp parallel masked 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);
- }
- void StmtPrinter::VisitOMPInteropDirective(OMPInteropDirective *Node) {
- Indent() << "#pragma omp interop";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPDispatchDirective(OMPDispatchDirective *Node) {
- Indent() << "#pragma omp dispatch";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPMaskedDirective(OMPMaskedDirective *Node) {
- Indent() << "#pragma omp masked";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPGenericLoopDirective(OMPGenericLoopDirective *Node) {
- Indent() << "#pragma omp loop";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPTeamsGenericLoopDirective(
- OMPTeamsGenericLoopDirective *Node) {
- Indent() << "#pragma omp teams loop";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPTargetTeamsGenericLoopDirective(
- OMPTargetTeamsGenericLoopDirective *Node) {
- Indent() << "#pragma omp target teams loop";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPParallelGenericLoopDirective(
- OMPParallelGenericLoopDirective *Node) {
- Indent() << "#pragma omp parallel loop";
- PrintOMPExecutableDirective(Node);
- }
- void StmtPrinter::VisitOMPTargetParallelGenericLoopDirective(
- OMPTargetParallelGenericLoopDirective *Node) {
- Indent() << "#pragma omp target parallel loop";
- PrintOMPExecutableDirective(Node);
- }
- //===----------------------------------------------------------------------===//
- // Expr printing methods.
- //===----------------------------------------------------------------------===//
- void StmtPrinter::VisitSourceLocExpr(SourceLocExpr *Node) {
- OS << Node->getBuiltinStr() << "()";
- }
- void StmtPrinter::VisitConstantExpr(ConstantExpr *Node) {
- PrintExpr(Node->getSubExpr());
- }
- void StmtPrinter::VisitDeclRefExpr(DeclRefExpr *Node) {
- if (const auto *OCED = dyn_cast<OMPCapturedExprDecl>(Node->getDecl())) {
- OCED->getInit()->IgnoreImpCasts()->printPretty(OS, nullptr, Policy);
- return;
- }
- if (const auto *TPOD = dyn_cast<TemplateParamObjectDecl>(Node->getDecl())) {
- TPOD->printAsExpr(OS, Policy);
- return;
- }
- if (NestedNameSpecifier *Qualifier = Node->getQualifier())
- Qualifier->print(OS, Policy);
- if (Node->hasTemplateKeyword())
- OS << "template ";
- if (Policy.CleanUglifiedParameters &&
- isa<ParmVarDecl, NonTypeTemplateParmDecl>(Node->getDecl()) &&
- Node->getDecl()->getIdentifier())
- OS << Node->getDecl()->getIdentifier()->deuglifiedName();
- else
- Node->getNameInfo().printName(OS, Policy);
- if (Node->hasExplicitTemplateArgs()) {
- const TemplateParameterList *TPL = nullptr;
- if (!Node->hadMultipleCandidates())
- if (auto *TD = dyn_cast<TemplateDecl>(Node->getDecl()))
- TPL = TD->getTemplateParameters();
- printTemplateArgumentList(OS, Node->template_arguments(), Policy, TPL);
- }
- }
- 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->getBeginLoc().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::VisitSYCLUniqueStableNameExpr(
- SYCLUniqueStableNameExpr *Node) {
- OS << "__builtin_sycl_unique_stable_name(";
- Node->getTypeSourceInfo()->getType().print(OS, Policy);
- OS << ")";
- }
- void StmtPrinter::VisitPredefinedExpr(PredefinedExpr *Node) {
- OS << PredefinedExpr::getIdentKindName(Node->getIdentKind());
- }
- void StmtPrinter::VisitCharacterLiteral(CharacterLiteral *Node) {
- CharacterLiteral::print(Node->getValue(), Node->getKind(), OS);
- }
- /// 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 << toString(Node->getValue(), 10, isSigned);
- if (isa<BitIntType>(Node->getType())) {
- OS << (isSigned ? "wb" : "uwb");
- return;
- }
- // Emit suffixes. Integer literals are always a builtin integer type.
- switch (Node->getType()->castAs<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::SChar: OS << "i8"; 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;
- case BuiltinType::Int128:
- break; // no suffix.
- case BuiltinType::UInt128:
- break; // no suffix.
- case BuiltinType::WChar_S:
- case BuiltinType::WChar_U:
- break; // no suffix
- }
- }
- void StmtPrinter::VisitFixedPointLiteral(FixedPointLiteral *Node) {
- if (Policy.ConstantsAsWritten && printExprAsWritten(OS, Node, Context))
- return;
- OS << Node->getValueAsString(/*Radix=*/10);
- switch (Node->getType()->castAs<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()->castAs<BuiltinType>()->getKind()) {
- default: llvm_unreachable("Unexpected type for float literal!");
- case BuiltinType::Half: break; // FIXME: suffix?
- case BuiltinType::Ibm128: break; // FIXME: No suffix for ibm128 literal
- 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) {
- const char *Spelling = getTraitSpelling(Node->getKind());
- if (Node->getKind() == UETT_AlignOf) {
- if (Policy.Alignof)
- Spelling = "alignof";
- else if (Policy.UnderscoreAlignof)
- Spelling = "_Alignof";
- else
- Spelling = "__alignof";
- }
- OS << Spelling;
- 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 (const GenericSelectionExpr::Association Assoc : Node->associations()) {
- OS << ", ";
- QualType T = Assoc.getType();
- if (T.isNull())
- OS << "default";
- else
- T.print(OS, Policy);
- OS << ": ";
- PrintExpr(Assoc.getAssociationExpr());
- }
- OS << ")";
- }
- void StmtPrinter::VisitArraySubscriptExpr(ArraySubscriptExpr *Node) {
- PrintExpr(Node->getLHS());
- OS << "[";
- PrintExpr(Node->getRHS());
- OS << "]";
- }
- void StmtPrinter::VisitMatrixSubscriptExpr(MatrixSubscriptExpr *Node) {
- PrintExpr(Node->getBase());
- OS << "[";
- PrintExpr(Node->getRowIdx());
- OS << "]";
- OS << "[";
- PrintExpr(Node->getColumnIdx());
- OS << "]";
- }
- void StmtPrinter::VisitOMPArraySectionExpr(OMPArraySectionExpr *Node) {
- PrintExpr(Node->getBase());
- OS << "[";
- if (Node->getLowerBound())
- PrintExpr(Node->getLowerBound());
- if (Node->getColonLocFirst().isValid()) {
- OS << ":";
- if (Node->getLength())
- PrintExpr(Node->getLength());
- }
- if (Node->getColonLocSecond().isValid()) {
- OS << ":";
- if (Node->getStride())
- PrintExpr(Node->getStride());
- }
- OS << "]";
- }
- void StmtPrinter::VisitOMPArrayShapingExpr(OMPArrayShapingExpr *Node) {
- OS << "(";
- for (Expr *E : Node->getDimensions()) {
- OS << "[";
- PrintExpr(E);
- OS << "]";
- }
- OS << ")";
- PrintExpr(Node->getBase());
- }
- void StmtPrinter::VisitOMPIteratorExpr(OMPIteratorExpr *Node) {
- OS << "iterator(";
- for (unsigned I = 0, E = Node->numOfIterators(); I < E; ++I) {
- auto *VD = cast<ValueDecl>(Node->getIteratorDecl(I));
- VD->getType().print(OS, Policy);
- const OMPIteratorExpr::IteratorRange Range = Node->getIteratorRange(I);
- OS << " " << VD->getName() << " = ";
- PrintExpr(Range.Begin);
- OS << ":";
- PrintExpr(Range.End);
- if (Range.Step) {
- OS << ":";
- PrintExpr(Range.Step);
- }
- if (I < E - 1)
- OS << ", ";
- }
- 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();
- const TemplateParameterList *TPL = nullptr;
- if (auto *FD = dyn_cast<FunctionDecl>(Node->getMemberDecl())) {
- if (!Node->hadMultipleCandidates())
- if (auto *FTD = FD->getPrimaryTemplate())
- TPL = FTD->getTemplateParameters();
- } else if (auto *VTSD =
- dyn_cast<VarTemplateSpecializationDecl>(Node->getMemberDecl()))
- TPL = VTSD->getSpecializedTemplate()->getTemplateParameters();
- if (Node->hasExplicitTemplateArgs())
- printTemplateArgumentList(OS, Node->template_arguments(), Policy, TPL);
- }
- 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 &&
- Node->getOp() != AtomicExpr::AO__hip_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) {
- OverloadedOperatorKind Kind = Node->getOperator();
- if (Kind == OO_PlusPlus || Kind == OO_MinusMinus) {
- if (Node->getNumArgs() == 1) {
- OS << getOperatorSpelling(Kind) << ' ';
- PrintExpr(Node->getArg(0));
- } else {
- PrintExpr(Node->getArg(0));
- OS << ' ' << getOperatorSpelling(Kind);
- }
- } else if (Kind == OO_Arrow) {
- PrintExpr(Node->getArg(0));
- } else if (Kind == OO_Call || Kind == OO_Subscript) {
- PrintExpr(Node->getArg(0));
- OS << (Kind == OO_Call ? '(' : '[');
- for (unsigned ArgIdx = 1; ArgIdx < Node->getNumArgs(); ++ArgIdx) {
- if (ArgIdx > 1)
- OS << ", ";
- if (!isa<CXXDefaultArgExpr>(Node->getArg(ArgIdx)))
- PrintExpr(Node->getArg(ArgIdx));
- }
- OS << (Kind == OO_Call ? ')' : ']');
- } else if (Node->getNumArgs() == 1) {
- OS << getOperatorSpelling(Kind) << ' ';
- PrintExpr(Node->getArg(0));
- } else if (Node->getNumArgs() == 2) {
- PrintExpr(Node->getArg(0));
- OS << ' ' << getOperatorSpelling(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::VisitCXXRewrittenBinaryOperator(
- CXXRewrittenBinaryOperator *Node) {
- CXXRewrittenBinaryOperator::DecomposedForm Decomposed =
- Node->getDecomposedForm();
- PrintExpr(const_cast<Expr*>(Decomposed.LHS));
- OS << ' ' << BinaryOperator::getOpcodeStr(Decomposed.Opcode) << ' ';
- PrintExpr(const_cast<Expr*>(Decomposed.RHS));
- }
- 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::VisitBuiltinBitCastExpr(BuiltinBitCastExpr *Node) {
- OS << "__builtin_bit_cast(";
- Node->getTypeInfoAsWritten()->getType().print(OS, Policy);
- OS << ", ";
- PrintExpr(Node->getSubExpr());
- OS << ")";
- }
- void StmtPrinter::VisitCXXAddrspaceCastExpr(CXXAddrspaceCastExpr *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 || Args->get(0).getKind() != TemplateArgument::Pack) {
- const TemplateParameterList *TPL = nullptr;
- if (!DRE->hadMultipleCandidates())
- if (const auto *TD = dyn_cast<TemplateDecl>(DRE->getDecl()))
- TPL = TD->getTemplateParameters();
- OS << "operator\"\"" << Node->getUDSuffix()->getName();
- printTemplateArgumentList(OS, Args->asArray(), Policy, TPL);
- 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 << toString(Int->getValue(), 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) {
- auto TargetType = Node->getType();
- auto *Auto = TargetType->getContainedDeducedType();
- bool Bare = Auto && Auto->isDeduced();
- // Parenthesize deduced casts.
- if (Bare)
- OS << '(';
- TargetType.print(OS, Policy);
- if (Bare)
- OS << ')';
- // No extra braces surrounding the inner construct.
- if (!Node->isListInitialization())
- OS << '(';
- PrintExpr(Node->getSubExpr());
- if (!Node->isListInitialization())
- 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 (C->isPackExpansion())
- OS << "...";
- if (Node->isInitCapture(C)) {
- // Init captures are always VarDecl.
- auto *D = cast<VarDecl>(C->getCapturedVar());
- llvm::StringRef Pre;
- llvm::StringRef Post;
- if (D->getInitStyle() == VarDecl::CallInit &&
- !isa<ParenListExpr>(D->getInit())) {
- Pre = "(";
- Post = ")";
- } else if (D->getInitStyle() == VarDecl::CInit) {
- Pre = " = ";
- }
- OS << Pre;
- PrintExpr(D->getInit());
- OS << Post;
- }
- }
- OS << ']';
- if (!Node->getExplicitTemplateParameters().empty()) {
- Node->getTemplateParameterList()->print(
- OS, Node->getLambdaClass()->getASTContext(),
- /*OmitTemplateKW*/true);
- }
- 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 =
- (Policy.CleanUglifiedParameters && P->getIdentifier())
- ? P->getIdentifier()->deuglifiedName().str()
- : 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()->castAs<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.
- OS << ' ';
- if (Policy.TerseOutput)
- OS << "{}";
- else
- PrintRawCompoundStmt(Node->getCompoundStmtBody());
- }
- 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 (E->isArray()) {
- llvm::raw_string_ostream s(TypeS);
- s << '[';
- if (std::optional<Expr *> Size = E->getArraySize())
- (*Size)->printPretty(s, Helper, Policy);
- s << ']';
- }
- E->getAllocatedType().print(OS, Policy, TypeS);
- if (E->isParenTypeId())
- OS << ")";
- CXXNewExpr::InitializationStyle InitStyle = E->getInitializationStyle();
- if (InitStyle != CXXNewExpr::NoInit) {
- bool Bare = InitStyle == CXXNewExpr::CallInit &&
- !isa<ParenListExpr>(E->getInitializer());
- if (Bare)
- OS << "(";
- PrintExpr(E->getInitializer());
- if (Bare)
- 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);
- if (!Node->isListInitialization())
- OS << '(';
- for (auto Arg = Node->arg_begin(), ArgEnd = Node->arg_end(); Arg != ArgEnd;
- ++Arg) {
- if (Arg != Node->arg_begin())
- OS << ", ";
- PrintExpr(*Arg);
- }
- if (!Node->isListInitialization())
- 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);
- }
- void StmtPrinter::VisitTypeTraitExpr(TypeTraitExpr *E) {
- OS << getTraitSpelling(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 << getTraitSpelling(E->getTrait()) << '(';
- E->getQueriedType().print(OS, Policy);
- OS << ')';
- }
- void StmtPrinter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
- OS << getTraitSpelling(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->getSubExpr());
- }
- 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 << ")";
- }
- void StmtPrinter::VisitCXXParenListInitExpr(CXXParenListInitExpr *Node) {
- OS << "(";
- llvm::interleaveComma(Node->getInitExprs(), OS,
- [&](Expr *E) { PrintExpr(E); });
- OS << ")";
- }
- void StmtPrinter::VisitConceptSpecializationExpr(ConceptSpecializationExpr *E) {
- NestedNameSpecifierLoc NNS = E->getNestedNameSpecifierLoc();
- if (NNS)
- NNS.getNestedNameSpecifier()->print(OS, Policy);
- if (E->getTemplateKWLoc().isValid())
- OS << "template ";
- OS << E->getFoundDecl()->getName();
- printTemplateArgumentList(OS, E->getTemplateArgsAsWritten()->arguments(),
- Policy,
- E->getNamedConcept()->getTemplateParameters());
- }
- void StmtPrinter::VisitRequiresExpr(RequiresExpr *E) {
- OS << "requires ";
- auto LocalParameters = E->getLocalParameters();
- if (!LocalParameters.empty()) {
- OS << "(";
- for (ParmVarDecl *LocalParam : LocalParameters) {
- PrintRawDecl(LocalParam);
- if (LocalParam != LocalParameters.back())
- OS << ", ";
- }
- OS << ") ";
- }
- OS << "{ ";
- auto Requirements = E->getRequirements();
- for (concepts::Requirement *Req : Requirements) {
- if (auto *TypeReq = dyn_cast<concepts::TypeRequirement>(Req)) {
- if (TypeReq->isSubstitutionFailure())
- OS << "<<error-type>>";
- else
- TypeReq->getType()->getType().print(OS, Policy);
- } else if (auto *ExprReq = dyn_cast<concepts::ExprRequirement>(Req)) {
- if (ExprReq->isCompound())
- OS << "{ ";
- if (ExprReq->isExprSubstitutionFailure())
- OS << "<<error-expression>>";
- else
- PrintExpr(ExprReq->getExpr());
- if (ExprReq->isCompound()) {
- OS << " }";
- if (ExprReq->getNoexceptLoc().isValid())
- OS << " noexcept";
- const auto &RetReq = ExprReq->getReturnTypeRequirement();
- if (!RetReq.isEmpty()) {
- OS << " -> ";
- if (RetReq.isSubstitutionFailure())
- OS << "<<error-type>>";
- else if (RetReq.isTypeConstraint())
- RetReq.getTypeConstraint()->print(OS, Policy);
- }
- }
- } else {
- auto *NestedReq = cast<concepts::NestedRequirement>(Req);
- OS << "requires ";
- if (NestedReq->hasInvalidConstraint())
- OS << "<<error-expression>>";
- else
- PrintExpr(NestedReq->getConstraintExpr());
- }
- OS << "; ";
- }
- 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::VisitRecoveryExpr(RecoveryExpr *Node) {
- OS << "<recovery-expr>(";
- const char *Sep = "";
- for (Expr *E : Node->subExpressions()) {
- OS << Sep;
- PrintExpr(E);
- Sep = ", ";
- }
- OS << ')';
- }
- 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 &Out, PrinterHelper *Helper,
- const PrintingPolicy &Policy, unsigned Indentation,
- StringRef NL, const ASTContext *Context) const {
- StmtPrinter P(Out, Helper, Policy, Indentation, NL, Context);
- P.Visit(const_cast<Stmt *>(this));
- }
- void Stmt::printPrettyControlled(raw_ostream &Out, PrinterHelper *Helper,
- const PrintingPolicy &Policy,
- unsigned Indentation, StringRef NL,
- const ASTContext *Context) const {
- StmtPrinter P(Out, Helper, Policy, Indentation, NL, Context);
- P.PrintControlledStmt(const_cast<Stmt *>(this));
- }
- void Stmt::printJson(raw_ostream &Out, PrinterHelper *Helper,
- const PrintingPolicy &Policy, bool AddQuotes) const {
- std::string Buf;
- llvm::raw_string_ostream TempOut(Buf);
- printPretty(TempOut, Helper, Policy);
- Out << JsonFormat(TempOut.str(), AddQuotes);
- }
- //===----------------------------------------------------------------------===//
- // PrinterHelper
- //===----------------------------------------------------------------------===//
- // Implement virtual destructor.
- PrinterHelper::~PrinterHelper() = default;
|