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