blob: 65e9f824999364ce59f3f83a27f6a693e5abc63d [file] [log] [blame]
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00001//===--- StmtPrinter.cpp - Printing implementation for Stmt ASTs ----------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner5b12ab82007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00007//
8//===----------------------------------------------------------------------===//
9//
Chris Lattnercbe4f772007-08-08 22:51:59 +000010// This file implements the Stmt::dumpPretty/Stmt::printPretty methods, which
11// pretty print the AST back out to C code.
Chris Lattnera3bcb7a2006-11-04 07:16:25 +000012//
13//===----------------------------------------------------------------------===//
14
Benjamin Kramer4ab984e2012-07-04 20:19:54 +000015#include "clang/AST/ASTContext.h"
Benjamin Kramerea70eb32012-12-01 15:09:41 +000016#include "clang/AST/Attr.h"
Douglas Gregorc7acfdf2009-01-06 05:10:23 +000017#include "clang/AST/DeclCXX.h"
Ted Kremenek5c84c012007-10-17 18:36:42 +000018#include "clang/AST/DeclObjC.h"
Alexey Bataev90c228f2016-02-08 09:29:13 +000019#include "clang/AST/DeclOpenMP.h"
Douglas Gregorcdbc5392011-01-15 01:15:58 +000020#include "clang/AST/DeclTemplate.h"
Douglas Gregor882211c2010-04-28 22:16:22 +000021#include "clang/AST/Expr.h"
Douglas Gregor2b88c112010-09-08 00:15:04 +000022#include "clang/AST/ExprCXX.h"
Alexey Bataev1a3320e2015-08-25 14:24:04 +000023#include "clang/AST/ExprOpenMP.h"
Benjamin Kramerea70eb32012-12-01 15:09:41 +000024#include "clang/AST/PrettyPrinter.h"
25#include "clang/AST/StmtVisitor.h"
Jordan Rose00d1b592013-02-08 22:30:27 +000026#include "clang/Basic/CharInfo.h"
Benjamin Kramer49038022012-02-04 13:45:25 +000027#include "llvm/ADT/SmallString.h"
Jordan Rose00d1b592013-02-08 22:30:27 +000028#include "llvm/Support/Format.h"
Chris Lattnera3bcb7a2006-11-04 07:16:25 +000029using namespace clang;
30
31//===----------------------------------------------------------------------===//
32// StmtPrinter Visitor
33//===----------------------------------------------------------------------===//
34
35namespace {
Benjamin Kramer26222b62009-11-28 19:03:38 +000036 class StmtPrinter : public StmtVisitor<StmtPrinter> {
Chris Lattner0e62c1c2011-07-23 10:55:15 +000037 raw_ostream &OS;
Chris Lattnera3bcb7a2006-11-04 07:16:25 +000038 unsigned IndentLevel;
Ted Kremenek04f3cee2007-08-31 21:30:12 +000039 clang::PrinterHelper* Helper;
Douglas Gregor7de59662009-05-29 20:38:28 +000040 PrintingPolicy Policy;
41
Chris Lattnera3bcb7a2006-11-04 07:16:25 +000042 public:
Richard Smith235341b2012-08-16 03:56:14 +000043 StmtPrinter(raw_ostream &os, PrinterHelper* helper,
Chris Lattnerc61089a2009-06-30 01:26:17 +000044 const PrintingPolicy &Policy,
Douglas Gregor7de59662009-05-29 20:38:28 +000045 unsigned Indentation = 0)
Richard Smith235341b2012-08-16 03:56:14 +000046 : OS(os), IndentLevel(Indentation), Helper(helper), Policy(Policy) {}
Mike Stump11289f42009-09-09 15:08:12 +000047
Douglas Gregor7de59662009-05-29 20:38:28 +000048 void PrintStmt(Stmt *S) {
49 PrintStmt(S, Policy.Indentation);
50 }
51
52 void PrintStmt(Stmt *S, int SubIndent) {
Chris Lattnerb9eb5a12007-05-20 22:52:15 +000053 IndentLevel += SubIndent;
Chris Lattnera076fde2007-05-31 18:21:33 +000054 if (S && isa<Expr>(S)) {
Chris Lattner882f7882006-11-04 18:52:07 +000055 // If this is an expr used in a stmt context, indent and newline it.
56 Indent();
Chris Lattner62249a62007-08-21 04:04:25 +000057 Visit(S);
Chris Lattnerfc068c12007-05-30 17:57:36 +000058 OS << ";\n";
Chris Lattner882f7882006-11-04 18:52:07 +000059 } else if (S) {
Chris Lattner62249a62007-08-21 04:04:25 +000060 Visit(S);
Chris Lattner882f7882006-11-04 18:52:07 +000061 } else {
Chris Lattner2f6ac262007-05-28 01:47:47 +000062 Indent() << "<<<NULL STATEMENT>>>\n";
Chris Lattner882f7882006-11-04 18:52:07 +000063 }
Chris Lattnerb9eb5a12007-05-20 22:52:15 +000064 IndentLevel -= SubIndent;
Chris Lattner882f7882006-11-04 18:52:07 +000065 }
Eli Friedman15ea8802009-05-30 00:19:54 +000066
Chris Lattner073926e2007-05-20 23:04:55 +000067 void PrintRawCompoundStmt(CompoundStmt *S);
Chris Lattnerfdc195a2007-06-05 20:52:47 +000068 void PrintRawDecl(Decl *D);
Eli Friedmanf86e5072012-10-16 23:45:15 +000069 void PrintRawDeclStmt(const DeclStmt *S);
Chris Lattnerc0a38dd2007-06-11 22:26:23 +000070 void PrintRawIfStmt(IfStmt *If);
Sebastian Redl9b244a82008-12-22 21:35:02 +000071 void PrintRawCXXCatchStmt(CXXCatchStmt *Catch);
Peter Collingbourne4b279a02011-02-08 21:17:54 +000072 void PrintCallArgs(CallExpr *E);
John Wiegley1c0675e2011-04-28 01:08:34 +000073 void PrintRawSEHExceptHandler(SEHExceptStmt *S);
74 void PrintRawSEHFinallyStmt(SEHFinallyStmt *S);
Alexey Bataev1b59ab52014-02-27 08:29:12 +000075 void PrintOMPExecutableDirective(OMPExecutableDirective *S);
Mike Stump11289f42009-09-09 15:08:12 +000076
Chris Lattner882f7882006-11-04 18:52:07 +000077 void PrintExpr(Expr *E) {
78 if (E)
Chris Lattner62249a62007-08-21 04:04:25 +000079 Visit(E);
Chris Lattnera3bcb7a2006-11-04 07:16:25 +000080 else
Chris Lattner882f7882006-11-04 18:52:07 +000081 OS << "<null expr>";
Chris Lattnera3bcb7a2006-11-04 07:16:25 +000082 }
Mike Stump11289f42009-09-09 15:08:12 +000083
Chris Lattner0e62c1c2011-07-23 10:55:15 +000084 raw_ostream &Indent(int Delta = 0) {
Douglas Gregor7de59662009-05-29 20:38:28 +000085 for (int i = 0, e = IndentLevel+Delta; i < e; ++i)
86 OS << " ";
Chris Lattnera3bcb7a2006-11-04 07:16:25 +000087 return OS;
88 }
Mike Stump11289f42009-09-09 15:08:12 +000089
Mike Stump11289f42009-09-09 15:08:12 +000090 void Visit(Stmt* S) {
Ted Kremenek04f3cee2007-08-31 21:30:12 +000091 if (Helper && Helper->handledStmt(S,OS))
92 return;
93 else StmtVisitor<StmtPrinter>::Visit(S);
94 }
Alexey Bataev1b59ab52014-02-27 08:29:12 +000095
Chandler Carruthb7967b92010-10-23 08:21:37 +000096 void VisitStmt(Stmt *Node) LLVM_ATTRIBUTE_UNUSED {
Argyrios Kyrtzidis30805532010-08-15 01:15:33 +000097 Indent() << "<<unknown stmt type>>\n";
98 }
Chandler Carruthb7967b92010-10-23 08:21:37 +000099 void VisitExpr(Expr *Node) LLVM_ATTRIBUTE_UNUSED {
Argyrios Kyrtzidis30805532010-08-15 01:15:33 +0000100 OS << "<<unknown expr type>>";
101 }
102 void VisitCXXNamedCastExpr(CXXNamedCastExpr *Node);
Mike Stump11289f42009-09-09 15:08:12 +0000103
Argyrios Kyrtzidis30805532010-08-15 01:15:33 +0000104#define ABSTRACT_STMT(CLASS)
Douglas Gregorbe35ce92008-11-14 12:46:07 +0000105#define STMT(CLASS, PARENT) \
Chris Lattner62249a62007-08-21 04:04:25 +0000106 void Visit##CLASS(CLASS *Node);
Alexis Hunt656bb312010-05-05 15:24:00 +0000107#include "clang/AST/StmtNodes.inc"
Chris Lattner71e23ce2006-11-04 20:18:38 +0000108 };
Alexander Kornienkoab9db512015-06-22 23:07:51 +0000109}
Chris Lattnera3bcb7a2006-11-04 07:16:25 +0000110
Chris Lattner71e23ce2006-11-04 20:18:38 +0000111//===----------------------------------------------------------------------===//
112// Stmt printing methods.
113//===----------------------------------------------------------------------===//
114
Chris Lattner073926e2007-05-20 23:04:55 +0000115/// PrintRawCompoundStmt - Print a compound stmt without indenting the {, and
116/// with no newline after the }.
117void StmtPrinter::PrintRawCompoundStmt(CompoundStmt *Node) {
118 OS << "{\n";
Aaron Ballmanc7e4e212014-03-17 14:19:37 +0000119 for (auto *I : Node->body())
120 PrintStmt(I);
Mike Stump11289f42009-09-09 15:08:12 +0000121
Chris Lattner073926e2007-05-20 23:04:55 +0000122 Indent() << "}";
123}
124
Chris Lattnerfdc195a2007-06-05 20:52:47 +0000125void StmtPrinter::PrintRawDecl(Decl *D) {
Argyrios Kyrtzidis8a803cc2009-06-30 02:35:04 +0000126 D->print(OS, Policy, IndentLevel);
Mike Stump74a76472009-02-10 20:16:46 +0000127}
128
Eli Friedmanf86e5072012-10-16 23:45:15 +0000129void StmtPrinter::PrintRawDeclStmt(const DeclStmt *S) {
Aaron Ballman535bbcc2014-03-14 17:01:24 +0000130 SmallVector<Decl*, 2> Decls(S->decls());
Argyrios Kyrtzidis8a803cc2009-06-30 02:35:04 +0000131 Decl::printGroup(Decls.data(), Decls.size(), OS, Policy, IndentLevel);
Ted Kremenek15e6b402008-10-06 18:39:36 +0000132}
Chris Lattnerfdc195a2007-06-05 20:52:47 +0000133
134void StmtPrinter::VisitNullStmt(NullStmt *Node) {
135 Indent() << ";\n";
136}
137
138void StmtPrinter::VisitDeclStmt(DeclStmt *Node) {
Eli Friedman15ea8802009-05-30 00:19:54 +0000139 Indent();
140 PrintRawDeclStmt(Node);
141 OS << ";\n";
Steve Naroff2a8ad182007-05-29 22:59:26 +0000142}
143
Chris Lattner073926e2007-05-20 23:04:55 +0000144void StmtPrinter::VisitCompoundStmt(CompoundStmt *Node) {
145 Indent();
146 PrintRawCompoundStmt(Node);
Chris Lattnerdf3cafb2007-05-31 05:08:56 +0000147 OS << "\n";
Chris Lattnera3bcb7a2006-11-04 07:16:25 +0000148}
149
Chris Lattner6c0ff132006-11-05 00:19:50 +0000150void StmtPrinter::VisitCaseStmt(CaseStmt *Node) {
Chris Lattnerb9eb5a12007-05-20 22:52:15 +0000151 Indent(-1) << "case ";
Chris Lattner6c0ff132006-11-05 00:19:50 +0000152 PrintExpr(Node->getLHS());
153 if (Node->getRHS()) {
154 OS << " ... ";
155 PrintExpr(Node->getRHS());
156 }
157 OS << ":\n";
Mike Stump11289f42009-09-09 15:08:12 +0000158
Chris Lattnerb9eb5a12007-05-20 22:52:15 +0000159 PrintStmt(Node->getSubStmt(), 0);
Chris Lattner6c0ff132006-11-05 00:19:50 +0000160}
161
162void StmtPrinter::VisitDefaultStmt(DefaultStmt *Node) {
Chris Lattnerb9eb5a12007-05-20 22:52:15 +0000163 Indent(-1) << "default:\n";
164 PrintStmt(Node->getSubStmt(), 0);
Chris Lattner6c0ff132006-11-05 00:19:50 +0000165}
166
167void StmtPrinter::VisitLabelStmt(LabelStmt *Node) {
Chris Lattnereefa10e2007-05-28 06:56:27 +0000168 Indent(-1) << Node->getName() << ":\n";
Chris Lattnerb9eb5a12007-05-20 22:52:15 +0000169 PrintStmt(Node->getSubStmt(), 0);
Chris Lattner6c0ff132006-11-05 00:19:50 +0000170}
171
Richard Smithc202b282012-04-14 00:33:13 +0000172void StmtPrinter::VisitAttributedStmt(AttributedStmt *Node) {
Aaron Ballmanb06b15a2014-06-06 12:40:24 +0000173 for (const auto *Attr : Node->getAttrs()) {
Tyler Nowickie8b07ed2014-06-13 17:57:25 +0000174 Attr->printPretty(OS, Policy);
Aaron Ballmanb06b15a2014-06-06 12:40:24 +0000175 }
176
Richard Smithc202b282012-04-14 00:33:13 +0000177 PrintStmt(Node->getSubStmt(), 0);
178}
179
Chris Lattnerc0a38dd2007-06-11 22:26:23 +0000180void StmtPrinter::PrintRawIfStmt(IfStmt *If) {
Sebastian Redl7b7cec62009-02-07 20:05:48 +0000181 OS << "if (";
Eli Friedmanf86e5072012-10-16 23:45:15 +0000182 if (const DeclStmt *DS = If->getConditionVariableDeclStmt())
183 PrintRawDeclStmt(DS);
184 else
185 PrintExpr(If->getCond());
Sebastian Redl7b7cec62009-02-07 20:05:48 +0000186 OS << ')';
Mike Stump11289f42009-09-09 15:08:12 +0000187
Chris Lattner073926e2007-05-20 23:04:55 +0000188 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(If->getThen())) {
189 OS << ' ';
190 PrintRawCompoundStmt(CS);
191 OS << (If->getElse() ? ' ' : '\n');
192 } else {
193 OS << '\n';
194 PrintStmt(If->getThen());
195 if (If->getElse()) Indent();
196 }
Mike Stump11289f42009-09-09 15:08:12 +0000197
Chris Lattner073926e2007-05-20 23:04:55 +0000198 if (Stmt *Else = If->getElse()) {
199 OS << "else";
Mike Stump11289f42009-09-09 15:08:12 +0000200
Chris Lattner073926e2007-05-20 23:04:55 +0000201 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Else)) {
202 OS << ' ';
203 PrintRawCompoundStmt(CS);
204 OS << '\n';
Chris Lattnerc0a38dd2007-06-11 22:26:23 +0000205 } else if (IfStmt *ElseIf = dyn_cast<IfStmt>(Else)) {
206 OS << ' ';
207 PrintRawIfStmt(ElseIf);
Chris Lattner073926e2007-05-20 23:04:55 +0000208 } else {
209 OS << '\n';
210 PrintStmt(If->getElse());
211 }
Chris Lattner882f7882006-11-04 18:52:07 +0000212 }
Chris Lattner85ed8732006-11-04 20:40:44 +0000213}
214
Chris Lattnerc0a38dd2007-06-11 22:26:23 +0000215void StmtPrinter::VisitIfStmt(IfStmt *If) {
216 Indent();
217 PrintRawIfStmt(If);
218}
219
Chris Lattnerf2174b62006-11-04 20:59:27 +0000220void StmtPrinter::VisitSwitchStmt(SwitchStmt *Node) {
221 Indent() << "switch (";
Eli Friedmanf86e5072012-10-16 23:45:15 +0000222 if (const DeclStmt *DS = Node->getConditionVariableDeclStmt())
223 PrintRawDeclStmt(DS);
224 else
225 PrintExpr(Node->getCond());
Chris Lattner073926e2007-05-20 23:04:55 +0000226 OS << ")";
Mike Stump11289f42009-09-09 15:08:12 +0000227
Chris Lattner073926e2007-05-20 23:04:55 +0000228 // Pretty print compoundstmt bodies (very common).
229 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
230 OS << " ";
231 PrintRawCompoundStmt(CS);
232 OS << "\n";
233 } else {
234 OS << "\n";
235 PrintStmt(Node->getBody());
236 }
Chris Lattnerf2174b62006-11-04 20:59:27 +0000237}
238
Chris Lattner85ed8732006-11-04 20:40:44 +0000239void StmtPrinter::VisitWhileStmt(WhileStmt *Node) {
240 Indent() << "while (";
Eli Friedmanf86e5072012-10-16 23:45:15 +0000241 if (const DeclStmt *DS = Node->getConditionVariableDeclStmt())
242 PrintRawDeclStmt(DS);
243 else
244 PrintExpr(Node->getCond());
Chris Lattner85ed8732006-11-04 20:40:44 +0000245 OS << ")\n";
246 PrintStmt(Node->getBody());
247}
248
249void StmtPrinter::VisitDoStmt(DoStmt *Node) {
Chris Lattner5a4e9d22007-09-15 21:49:37 +0000250 Indent() << "do ";
251 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
252 PrintRawCompoundStmt(CS);
253 OS << " ";
254 } else {
255 OS << "\n";
256 PrintStmt(Node->getBody());
257 Indent();
258 }
Mike Stump11289f42009-09-09 15:08:12 +0000259
Eli Friedman8f1d33e2009-05-17 01:05:34 +0000260 OS << "while (";
Chris Lattner85ed8732006-11-04 20:40:44 +0000261 PrintExpr(Node->getCond());
Eli Friedman8f1d33e2009-05-17 01:05:34 +0000262 OS << ");\n";
Chris Lattnera3bcb7a2006-11-04 07:16:25 +0000263}
264
Chris Lattner71e23ce2006-11-04 20:18:38 +0000265void StmtPrinter::VisitForStmt(ForStmt *Node) {
266 Indent() << "for (";
Chris Lattnerfdc195a2007-06-05 20:52:47 +0000267 if (Node->getInit()) {
268 if (DeclStmt *DS = dyn_cast<DeclStmt>(Node->getInit()))
Ted Kremenek15e6b402008-10-06 18:39:36 +0000269 PrintRawDeclStmt(DS);
Chris Lattnerfdc195a2007-06-05 20:52:47 +0000270 else
271 PrintExpr(cast<Expr>(Node->getInit()));
272 }
Chris Lattner5a4e9d22007-09-15 21:49:37 +0000273 OS << ";";
274 if (Node->getCond()) {
275 OS << " ";
Chris Lattnerfdc195a2007-06-05 20:52:47 +0000276 PrintExpr(Node->getCond());
Chris Lattner5a4e9d22007-09-15 21:49:37 +0000277 }
278 OS << ";";
279 if (Node->getInc()) {
280 OS << " ";
Chris Lattnerfdc195a2007-06-05 20:52:47 +0000281 PrintExpr(Node->getInc());
Chris Lattner5a4e9d22007-09-15 21:49:37 +0000282 }
283 OS << ") ";
Mike Stump11289f42009-09-09 15:08:12 +0000284
Chris Lattner5a4e9d22007-09-15 21:49:37 +0000285 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
286 PrintRawCompoundStmt(CS);
287 OS << "\n";
288 } else {
289 OS << "\n";
290 PrintStmt(Node->getBody());
291 }
Chris Lattner71e23ce2006-11-04 20:18:38 +0000292}
293
Ted Kremenek1b0ea822008-01-07 19:49:32 +0000294void StmtPrinter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *Node) {
Fariborz Jahanian83615522008-01-02 22:54:34 +0000295 Indent() << "for (";
296 if (DeclStmt *DS = dyn_cast<DeclStmt>(Node->getElement()))
Ted Kremenek15e6b402008-10-06 18:39:36 +0000297 PrintRawDeclStmt(DS);
Fariborz Jahanian83615522008-01-02 22:54:34 +0000298 else
299 PrintExpr(cast<Expr>(Node->getElement()));
300 OS << " in ";
301 PrintExpr(Node->getCollection());
302 OS << ") ";
Mike Stump11289f42009-09-09 15:08:12 +0000303
Fariborz Jahanian83615522008-01-02 22:54:34 +0000304 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(Node->getBody())) {
305 PrintRawCompoundStmt(CS);
306 OS << "\n";
307 } else {
308 OS << "\n";
309 PrintStmt(Node->getBody());
310 }
311}
312
Richard Smith02e85f32011-04-14 22:09:26 +0000313void StmtPrinter::VisitCXXForRangeStmt(CXXForRangeStmt *Node) {
314 Indent() << "for (";
315 PrintingPolicy SubPolicy(Policy);
316 SubPolicy.SuppressInitializers = true;
317 Node->getLoopVariable()->print(OS, SubPolicy, IndentLevel);
318 OS << " : ";
319 PrintExpr(Node->getRangeInit());
320 OS << ") {\n";
321 PrintStmt(Node->getBody());
Ted Kremenek88446602013-12-11 23:44:02 +0000322 Indent() << "}";
323 if (Policy.IncludeNewlines) OS << "\n";
Richard Smith02e85f32011-04-14 22:09:26 +0000324}
325
Douglas Gregordeb4a2be2011-10-25 01:33:02 +0000326void StmtPrinter::VisitMSDependentExistsStmt(MSDependentExistsStmt *Node) {
327 Indent();
328 if (Node->isIfExists())
329 OS << "__if_exists (";
330 else
331 OS << "__if_not_exists (";
332
333 if (NestedNameSpecifier *Qualifier
334 = Node->getQualifierLoc().getNestedNameSpecifier())
335 Qualifier->print(OS, Policy);
336
337 OS << Node->getNameInfo() << ") ";
338
339 PrintRawCompoundStmt(Node->getSubStmt());
340}
341
Chris Lattner16976d32006-11-05 01:46:01 +0000342void StmtPrinter::VisitGotoStmt(GotoStmt *Node) {
Ted Kremenek88446602013-12-11 23:44:02 +0000343 Indent() << "goto " << Node->getLabel()->getName() << ";";
344 if (Policy.IncludeNewlines) OS << "\n";
Chris Lattner16976d32006-11-05 01:46:01 +0000345}
346
347void StmtPrinter::VisitIndirectGotoStmt(IndirectGotoStmt *Node) {
Chris Lattner36ad1232006-11-05 01:51:06 +0000348 Indent() << "goto *";
Chris Lattner16976d32006-11-05 01:46:01 +0000349 PrintExpr(Node->getTarget());
Ted Kremenek88446602013-12-11 23:44:02 +0000350 OS << ";";
351 if (Policy.IncludeNewlines) OS << "\n";
Chris Lattner16976d32006-11-05 01:46:01 +0000352}
353
354void StmtPrinter::VisitContinueStmt(ContinueStmt *Node) {
Ted Kremenek88446602013-12-11 23:44:02 +0000355 Indent() << "continue;";
356 if (Policy.IncludeNewlines) OS << "\n";
Chris Lattner16976d32006-11-05 01:46:01 +0000357}
358
359void StmtPrinter::VisitBreakStmt(BreakStmt *Node) {
Ted Kremenek88446602013-12-11 23:44:02 +0000360 Indent() << "break;";
361 if (Policy.IncludeNewlines) OS << "\n";
Chris Lattner16976d32006-11-05 01:46:01 +0000362}
363
364
Chris Lattner882f7882006-11-04 18:52:07 +0000365void StmtPrinter::VisitReturnStmt(ReturnStmt *Node) {
366 Indent() << "return";
367 if (Node->getRetValue()) {
368 OS << " ";
369 PrintExpr(Node->getRetValue());
Chris Lattnera3bcb7a2006-11-04 07:16:25 +0000370 }
Ted Kremenek88446602013-12-11 23:44:02 +0000371 OS << ";";
372 if (Policy.IncludeNewlines) OS << "\n";
Chris Lattnera3bcb7a2006-11-04 07:16:25 +0000373}
374
Chris Lattner73c56c02007-10-29 04:04:16 +0000375
Chad Rosierde70e0e2012-08-25 00:11:56 +0000376void StmtPrinter::VisitGCCAsmStmt(GCCAsmStmt *Node) {
Anders Carlsson660bdd12007-11-23 23:12:25 +0000377 Indent() << "asm ";
Mike Stump11289f42009-09-09 15:08:12 +0000378
Anders Carlsson660bdd12007-11-23 23:12:25 +0000379 if (Node->isVolatile())
380 OS << "volatile ";
Mike Stump11289f42009-09-09 15:08:12 +0000381
Anders Carlsson660bdd12007-11-23 23:12:25 +0000382 OS << "(";
Anders Carlsson81a5a692007-11-20 19:21:03 +0000383 VisitStringLiteral(Node->getAsmString());
Mike Stump11289f42009-09-09 15:08:12 +0000384
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000385 // Outputs
386 if (Node->getNumOutputs() != 0 || Node->getNumInputs() != 0 ||
387 Node->getNumClobbers() != 0)
388 OS << " : ";
Mike Stump11289f42009-09-09 15:08:12 +0000389
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000390 for (unsigned i = 0, e = Node->getNumOutputs(); i != e; ++i) {
391 if (i != 0)
392 OS << ", ";
Mike Stump11289f42009-09-09 15:08:12 +0000393
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000394 if (!Node->getOutputName(i).empty()) {
395 OS << '[';
396 OS << Node->getOutputName(i);
397 OS << "] ";
398 }
Mike Stump11289f42009-09-09 15:08:12 +0000399
Chris Lattner72bbf172009-03-10 04:59:06 +0000400 VisitStringLiteral(Node->getOutputConstraintLiteral(i));
Jonathan Roelofs78f9e032015-06-09 14:13:31 +0000401 OS << " (";
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000402 Visit(Node->getOutputExpr(i));
Jonathan Roelofs78f9e032015-06-09 14:13:31 +0000403 OS << ")";
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000404 }
Mike Stump11289f42009-09-09 15:08:12 +0000405
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000406 // Inputs
407 if (Node->getNumInputs() != 0 || Node->getNumClobbers() != 0)
408 OS << " : ";
Mike Stump11289f42009-09-09 15:08:12 +0000409
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000410 for (unsigned i = 0, e = Node->getNumInputs(); i != e; ++i) {
411 if (i != 0)
412 OS << ", ";
Mike Stump11289f42009-09-09 15:08:12 +0000413
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000414 if (!Node->getInputName(i).empty()) {
415 OS << '[';
416 OS << Node->getInputName(i);
417 OS << "] ";
418 }
Mike Stump11289f42009-09-09 15:08:12 +0000419
Chris Lattner72bbf172009-03-10 04:59:06 +0000420 VisitStringLiteral(Node->getInputConstraintLiteral(i));
Jonathan Roelofs78f9e032015-06-09 14:13:31 +0000421 OS << " (";
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000422 Visit(Node->getInputExpr(i));
Jonathan Roelofs78f9e032015-06-09 14:13:31 +0000423 OS << ")";
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000424 }
Mike Stump11289f42009-09-09 15:08:12 +0000425
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000426 // Clobbers
427 if (Node->getNumClobbers() != 0)
428 OS << " : ";
Mike Stump11289f42009-09-09 15:08:12 +0000429
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000430 for (unsigned i = 0, e = Node->getNumClobbers(); i != e; ++i) {
431 if (i != 0)
432 OS << ", ";
Mike Stump11289f42009-09-09 15:08:12 +0000433
Chad Rosierd9fb09a2012-08-27 23:28:41 +0000434 VisitStringLiteral(Node->getClobberStringLiteral(i));
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000435 }
Mike Stump11289f42009-09-09 15:08:12 +0000436
Ted Kremenek88446602013-12-11 23:44:02 +0000437 OS << ");";
438 if (Policy.IncludeNewlines) OS << "\n";
Chris Lattner73c56c02007-10-29 04:04:16 +0000439}
440
Chad Rosier32503022012-06-11 20:47:18 +0000441void StmtPrinter::VisitMSAsmStmt(MSAsmStmt *Node) {
442 // FIXME: Implement MS style inline asm statement printer.
Chad Rosierb6f46c12012-08-15 16:53:30 +0000443 Indent() << "__asm ";
444 if (Node->hasBraces())
445 OS << "{\n";
John McCallf413f5e2013-05-03 00:10:13 +0000446 OS << Node->getAsmString() << "\n";
Chad Rosierb6f46c12012-08-15 16:53:30 +0000447 if (Node->hasBraces())
448 Indent() << "}\n";
Chad Rosier32503022012-06-11 20:47:18 +0000449}
450
Tareq A. Siraj24110cc2013-04-16 18:53:08 +0000451void StmtPrinter::VisitCapturedStmt(CapturedStmt *Node) {
Tareq A. Siraj6dfa25a2013-04-16 19:37:38 +0000452 PrintStmt(Node->getCapturedDecl()->getBody());
Tareq A. Siraj24110cc2013-04-16 18:53:08 +0000453}
454
Ted Kremenek1b0ea822008-01-07 19:49:32 +0000455void StmtPrinter::VisitObjCAtTryStmt(ObjCAtTryStmt *Node) {
Fariborz Jahanian88157952007-11-02 18:16:07 +0000456 Indent() << "@try";
457 if (CompoundStmt *TS = dyn_cast<CompoundStmt>(Node->getTryBody())) {
458 PrintRawCompoundStmt(TS);
459 OS << "\n";
460 }
Mike Stump11289f42009-09-09 15:08:12 +0000461
Douglas Gregor96c79492010-04-23 22:50:49 +0000462 for (unsigned I = 0, N = Node->getNumCatchStmts(); I != N; ++I) {
463 ObjCAtCatchStmt *catchStmt = Node->getCatchStmt(I);
Fariborz Jahanian88157952007-11-02 18:16:07 +0000464 Indent() << "@catch(";
Steve Naroff371b8fb2009-03-03 19:52:17 +0000465 if (catchStmt->getCatchParamDecl()) {
466 if (Decl *DS = catchStmt->getCatchParamDecl())
467 PrintRawDecl(DS);
Fariborz Jahanian88157952007-11-02 18:16:07 +0000468 }
469 OS << ")";
Mike Stump11289f42009-09-09 15:08:12 +0000470 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(catchStmt->getCatchBody())) {
471 PrintRawCompoundStmt(CS);
472 OS << "\n";
473 }
Fariborz Jahanian88157952007-11-02 18:16:07 +0000474 }
Mike Stump11289f42009-09-09 15:08:12 +0000475
476 if (ObjCAtFinallyStmt *FS = static_cast<ObjCAtFinallyStmt *>(
477 Node->getFinallyStmt())) {
Fariborz Jahaniandefbf9a2007-11-07 00:46:42 +0000478 Indent() << "@finally";
479 PrintRawCompoundStmt(dyn_cast<CompoundStmt>(FS->getFinallyBody()));
Fariborz Jahanian88157952007-11-02 18:16:07 +0000480 OS << "\n";
Mike Stump11289f42009-09-09 15:08:12 +0000481 }
Fariborz Jahanian65590b22007-11-01 21:12:44 +0000482}
483
Ted Kremenek1b0ea822008-01-07 19:49:32 +0000484void StmtPrinter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *Node) {
Fariborz Jahanian65590b22007-11-01 21:12:44 +0000485}
486
Ted Kremenek1b0ea822008-01-07 19:49:32 +0000487void StmtPrinter::VisitObjCAtCatchStmt (ObjCAtCatchStmt *Node) {
Fariborz Jahanian65590b22007-11-01 21:12:44 +0000488 Indent() << "@catch (...) { /* todo */ } \n";
489}
490
Fariborz Jahanian049fa582008-01-30 17:38:29 +0000491void StmtPrinter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *Node) {
Fariborz Jahanianadfbbc32007-11-07 02:00:49 +0000492 Indent() << "@throw";
493 if (Node->getThrowExpr()) {
494 OS << " ";
495 PrintExpr(Node->getThrowExpr());
496 }
497 OS << ";\n";
498}
499
Erik Pilkington29099de2016-07-16 00:35:23 +0000500void StmtPrinter::VisitObjCAvailabilityCheckExpr(
501 ObjCAvailabilityCheckExpr *Node) {
502 OS << "@available(...)";
503}
504
Fariborz Jahanian049fa582008-01-30 17:38:29 +0000505void StmtPrinter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *Node) {
Fariborz Jahanianf89ca382008-01-29 18:21:32 +0000506 Indent() << "@synchronized (";
507 PrintExpr(Node->getSynchExpr());
508 OS << ")";
Fariborz Jahanian049fa582008-01-30 17:38:29 +0000509 PrintRawCompoundStmt(Node->getSynchBody());
510 OS << "\n";
Fariborz Jahanianf89ca382008-01-29 18:21:32 +0000511}
512
John McCall31168b02011-06-15 23:02:42 +0000513void StmtPrinter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *Node) {
514 Indent() << "@autoreleasepool";
515 PrintRawCompoundStmt(dyn_cast<CompoundStmt>(Node->getSubStmt()));
516 OS << "\n";
517}
518
Sebastian Redl9b244a82008-12-22 21:35:02 +0000519void StmtPrinter::PrintRawCXXCatchStmt(CXXCatchStmt *Node) {
520 OS << "catch (";
Sebastian Redl54c04d42008-12-22 19:15:10 +0000521 if (Decl *ExDecl = Node->getExceptionDecl())
522 PrintRawDecl(ExDecl);
523 else
524 OS << "...";
525 OS << ") ";
526 PrintRawCompoundStmt(cast<CompoundStmt>(Node->getHandlerBlock()));
Sebastian Redl9b244a82008-12-22 21:35:02 +0000527}
528
529void StmtPrinter::VisitCXXCatchStmt(CXXCatchStmt *Node) {
530 Indent();
531 PrintRawCXXCatchStmt(Node);
532 OS << "\n";
533}
534
535void StmtPrinter::VisitCXXTryStmt(CXXTryStmt *Node) {
536 Indent() << "try ";
537 PrintRawCompoundStmt(Node->getTryBlock());
Mike Stump11289f42009-09-09 15:08:12 +0000538 for (unsigned i = 0, e = Node->getNumHandlers(); i < e; ++i) {
Sebastian Redl9b244a82008-12-22 21:35:02 +0000539 OS << " ";
540 PrintRawCXXCatchStmt(Node->getHandler(i));
541 }
Sebastian Redl54c04d42008-12-22 19:15:10 +0000542 OS << "\n";
543}
544
John Wiegley1c0675e2011-04-28 01:08:34 +0000545void StmtPrinter::VisitSEHTryStmt(SEHTryStmt *Node) {
546 Indent() << (Node->getIsCXXTry() ? "try " : "__try ");
547 PrintRawCompoundStmt(Node->getTryBlock());
548 SEHExceptStmt *E = Node->getExceptHandler();
549 SEHFinallyStmt *F = Node->getFinallyHandler();
550 if(E)
551 PrintRawSEHExceptHandler(E);
552 else {
553 assert(F && "Must have a finally block...");
554 PrintRawSEHFinallyStmt(F);
555 }
556 OS << "\n";
557}
558
559void StmtPrinter::PrintRawSEHFinallyStmt(SEHFinallyStmt *Node) {
560 OS << "__finally ";
561 PrintRawCompoundStmt(Node->getBlock());
562 OS << "\n";
563}
564
565void StmtPrinter::PrintRawSEHExceptHandler(SEHExceptStmt *Node) {
566 OS << "__except (";
567 VisitExpr(Node->getFilterExpr());
Joao Matos566359c2012-09-04 17:49:35 +0000568 OS << ")\n";
John Wiegley1c0675e2011-04-28 01:08:34 +0000569 PrintRawCompoundStmt(Node->getBlock());
570 OS << "\n";
571}
572
573void StmtPrinter::VisitSEHExceptStmt(SEHExceptStmt *Node) {
574 Indent();
575 PrintRawSEHExceptHandler(Node);
576 OS << "\n";
577}
578
579void StmtPrinter::VisitSEHFinallyStmt(SEHFinallyStmt *Node) {
580 Indent();
581 PrintRawSEHFinallyStmt(Node);
582 OS << "\n";
583}
584
Nico Weber9b982072014-07-07 00:12:30 +0000585void StmtPrinter::VisitSEHLeaveStmt(SEHLeaveStmt *Node) {
586 Indent() << "__leave;";
587 if (Policy.IncludeNewlines) OS << "\n";
588}
589
Chris Lattner71e23ce2006-11-04 20:18:38 +0000590//===----------------------------------------------------------------------===//
Alexey Bataev5ec3eb12013-07-19 03:13:43 +0000591// OpenMP clauses printing methods
592//===----------------------------------------------------------------------===//
593
594namespace {
595class OMPClausePrinter : public OMPClauseVisitor<OMPClausePrinter> {
596 raw_ostream &OS;
Alexey Bataevaadd52e2014-02-13 05:29:23 +0000597 const PrintingPolicy &Policy;
Alexey Bataev756c1962013-09-24 03:17:45 +0000598 /// \brief Process clauses with list of variables.
599 template <typename T>
600 void VisitOMPClauseList(T *Node, char StartSym);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +0000601public:
Alexey Bataevaadd52e2014-02-13 05:29:23 +0000602 OMPClausePrinter(raw_ostream &OS, const PrintingPolicy &Policy)
603 : OS(OS), Policy(Policy) { }
Alexey Bataev5ec3eb12013-07-19 03:13:43 +0000604#define OPENMP_CLAUSE(Name, Class) \
605 void Visit##Class(Class *S);
606#include "clang/Basic/OpenMPKinds.def"
607};
608
Alexey Bataevaadd52e2014-02-13 05:29:23 +0000609void OMPClausePrinter::VisitOMPIfClause(OMPIfClause *Node) {
610 OS << "if(";
Alexey Bataev6b8046a2015-09-03 07:23:48 +0000611 if (Node->getNameModifier() != OMPD_unknown)
612 OS << getOpenMPDirectiveName(Node->getNameModifier()) << ": ";
Craig Topper36250ad2014-05-12 05:36:57 +0000613 Node->getCondition()->printPretty(OS, nullptr, Policy, 0);
Alexey Bataevaadd52e2014-02-13 05:29:23 +0000614 OS << ")";
615}
616
Alexey Bataev3778b602014-07-17 07:32:53 +0000617void OMPClausePrinter::VisitOMPFinalClause(OMPFinalClause *Node) {
618 OS << "final(";
619 Node->getCondition()->printPretty(OS, nullptr, Policy, 0);
620 OS << ")";
621}
622
Alexey Bataev568a8332014-03-06 06:15:19 +0000623void OMPClausePrinter::VisitOMPNumThreadsClause(OMPNumThreadsClause *Node) {
624 OS << "num_threads(";
Craig Topper36250ad2014-05-12 05:36:57 +0000625 Node->getNumThreads()->printPretty(OS, nullptr, Policy, 0);
Alexey Bataev568a8332014-03-06 06:15:19 +0000626 OS << ")";
627}
628
Alexey Bataev62c87d22014-03-21 04:51:18 +0000629void OMPClausePrinter::VisitOMPSafelenClause(OMPSafelenClause *Node) {
630 OS << "safelen(";
Craig Topper36250ad2014-05-12 05:36:57 +0000631 Node->getSafelen()->printPretty(OS, nullptr, Policy, 0);
Alexey Bataev62c87d22014-03-21 04:51:18 +0000632 OS << ")";
633}
634
Alexey Bataev66b15b52015-08-21 11:14:16 +0000635void OMPClausePrinter::VisitOMPSimdlenClause(OMPSimdlenClause *Node) {
636 OS << "simdlen(";
637 Node->getSimdlen()->printPretty(OS, nullptr, Policy, 0);
638 OS << ")";
639}
640
Alexander Musman8bd31e62014-05-27 15:12:19 +0000641void OMPClausePrinter::VisitOMPCollapseClause(OMPCollapseClause *Node) {
642 OS << "collapse(";
643 Node->getNumForLoops()->printPretty(OS, nullptr, Policy, 0);
644 OS << ")";
645}
646
Alexey Bataev5ec3eb12013-07-19 03:13:43 +0000647void OMPClausePrinter::VisitOMPDefaultClause(OMPDefaultClause *Node) {
648 OS << "default("
649 << getOpenMPSimpleClauseTypeName(OMPC_default, Node->getDefaultKind())
650 << ")";
651}
652
Alexey Bataevbcbadb62014-05-06 06:04:14 +0000653void OMPClausePrinter::VisitOMPProcBindClause(OMPProcBindClause *Node) {
654 OS << "proc_bind("
655 << getOpenMPSimpleClauseTypeName(OMPC_proc_bind, Node->getProcBindKind())
656 << ")";
657}
658
Alexey Bataev56dafe82014-06-20 07:16:17 +0000659void OMPClausePrinter::VisitOMPScheduleClause(OMPScheduleClause *Node) {
Alexey Bataev6402bca2015-12-28 07:25:51 +0000660 OS << "schedule(";
661 if (Node->getFirstScheduleModifier() != OMPC_SCHEDULE_MODIFIER_unknown) {
662 OS << getOpenMPSimpleClauseTypeName(OMPC_schedule,
663 Node->getFirstScheduleModifier());
664 if (Node->getSecondScheduleModifier() != OMPC_SCHEDULE_MODIFIER_unknown) {
665 OS << ", ";
666 OS << getOpenMPSimpleClauseTypeName(OMPC_schedule,
667 Node->getSecondScheduleModifier());
668 }
669 OS << ": ";
670 }
671 OS << getOpenMPSimpleClauseTypeName(OMPC_schedule, Node->getScheduleKind());
Alexey Bataev3392d762016-02-16 11:18:12 +0000672 if (auto *E = Node->getChunkSize()) {
Alexey Bataev56dafe82014-06-20 07:16:17 +0000673 OS << ", ";
Alexey Bataev60da77e2016-02-29 05:54:20 +0000674 E->printPretty(OS, nullptr, Policy);
Alexey Bataev56dafe82014-06-20 07:16:17 +0000675 }
676 OS << ")";
677}
678
Alexey Bataev10e775f2015-07-30 11:36:16 +0000679void OMPClausePrinter::VisitOMPOrderedClause(OMPOrderedClause *Node) {
Alexey Bataev142e1fc2014-06-20 09:44:06 +0000680 OS << "ordered";
Alexey Bataev10e775f2015-07-30 11:36:16 +0000681 if (auto *Num = Node->getNumForLoops()) {
682 OS << "(";
683 Num->printPretty(OS, nullptr, Policy, 0);
684 OS << ")";
685 }
Alexey Bataev142e1fc2014-06-20 09:44:06 +0000686}
687
Alexey Bataev236070f2014-06-20 11:19:47 +0000688void OMPClausePrinter::VisitOMPNowaitClause(OMPNowaitClause *) {
689 OS << "nowait";
690}
691
Alexey Bataev7aea99a2014-07-17 12:19:31 +0000692void OMPClausePrinter::VisitOMPUntiedClause(OMPUntiedClause *) {
693 OS << "untied";
694}
695
Alexey Bataevb825de12015-12-07 10:51:44 +0000696void OMPClausePrinter::VisitOMPNogroupClause(OMPNogroupClause *) {
697 OS << "nogroup";
698}
699
Alexey Bataev74ba3a52014-07-17 12:47:03 +0000700void OMPClausePrinter::VisitOMPMergeableClause(OMPMergeableClause *) {
701 OS << "mergeable";
702}
703
Alexey Bataevf98b00c2014-07-23 02:27:21 +0000704void OMPClausePrinter::VisitOMPReadClause(OMPReadClause *) { OS << "read"; }
705
Alexey Bataevdea47612014-07-23 07:46:59 +0000706void OMPClausePrinter::VisitOMPWriteClause(OMPWriteClause *) { OS << "write"; }
707
Alexey Bataev67a4f222014-07-23 10:25:33 +0000708void OMPClausePrinter::VisitOMPUpdateClause(OMPUpdateClause *) {
709 OS << "update";
710}
711
Alexey Bataev459dec02014-07-24 06:46:57 +0000712void OMPClausePrinter::VisitOMPCaptureClause(OMPCaptureClause *) {
713 OS << "capture";
714}
715
Alexey Bataev82bad8b2014-07-24 08:55:34 +0000716void OMPClausePrinter::VisitOMPSeqCstClause(OMPSeqCstClause *) {
717 OS << "seq_cst";
718}
719
Alexey Bataev346265e2015-09-25 10:37:12 +0000720void OMPClausePrinter::VisitOMPThreadsClause(OMPThreadsClause *) {
721 OS << "threads";
722}
723
Alexey Bataevd14d1e62015-09-28 06:39:35 +0000724void OMPClausePrinter::VisitOMPSIMDClause(OMPSIMDClause *) { OS << "simd"; }
725
Michael Wonge710d542015-08-07 16:16:36 +0000726void OMPClausePrinter::VisitOMPDeviceClause(OMPDeviceClause *Node) {
727 OS << "device(";
728 Node->getDevice()->printPretty(OS, nullptr, Policy, 0);
729 OS << ")";
730}
731
Kelvin Li099bb8c2015-11-24 20:50:12 +0000732void OMPClausePrinter::VisitOMPNumTeamsClause(OMPNumTeamsClause *Node) {
733 OS << "num_teams(";
734 Node->getNumTeams()->printPretty(OS, nullptr, Policy, 0);
735 OS << ")";
736}
737
Kelvin Lia15fb1a2015-11-27 18:47:36 +0000738void OMPClausePrinter::VisitOMPThreadLimitClause(OMPThreadLimitClause *Node) {
739 OS << "thread_limit(";
740 Node->getThreadLimit()->printPretty(OS, nullptr, Policy, 0);
741 OS << ")";
742}
743
Alexey Bataeva0569352015-12-01 10:17:31 +0000744void OMPClausePrinter::VisitOMPPriorityClause(OMPPriorityClause *Node) {
745 OS << "priority(";
746 Node->getPriority()->printPretty(OS, nullptr, Policy, 0);
747 OS << ")";
748}
749
Alexey Bataev1fd4aed2015-12-07 12:52:51 +0000750void OMPClausePrinter::VisitOMPGrainsizeClause(OMPGrainsizeClause *Node) {
751 OS << "grainsize(";
752 Node->getGrainsize()->printPretty(OS, nullptr, Policy, 0);
753 OS << ")";
754}
755
Alexey Bataev382967a2015-12-08 12:06:20 +0000756void OMPClausePrinter::VisitOMPNumTasksClause(OMPNumTasksClause *Node) {
757 OS << "num_tasks(";
758 Node->getNumTasks()->printPretty(OS, nullptr, Policy, 0);
759 OS << ")";
760}
761
Alexey Bataev28c75412015-12-15 08:19:24 +0000762void OMPClausePrinter::VisitOMPHintClause(OMPHintClause *Node) {
763 OS << "hint(";
764 Node->getHint()->printPretty(OS, nullptr, Policy, 0);
765 OS << ")";
766}
767
Alexey Bataev756c1962013-09-24 03:17:45 +0000768template<typename T>
769void OMPClausePrinter::VisitOMPClauseList(T *Node, char StartSym) {
770 for (typename T::varlist_iterator I = Node->varlist_begin(),
771 E = Node->varlist_end();
Alexey Bataev90c228f2016-02-08 09:29:13 +0000772 I != E; ++I) {
Richard Trieuddd01ce2014-06-09 22:53:25 +0000773 assert(*I && "Expected non-null Stmt");
Alexey Bataev90c228f2016-02-08 09:29:13 +0000774 OS << (I == Node->varlist_begin() ? StartSym : ',');
Alexey Bataevd48bcd82014-03-31 03:36:38 +0000775 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(*I)) {
Alexey Bataev60da77e2016-02-29 05:54:20 +0000776 if (isa<OMPCapturedExprDecl>(DRE->getDecl()))
777 DRE->printPretty(OS, nullptr, Policy, 0);
Alexey Bataev90c228f2016-02-08 09:29:13 +0000778 else
779 DRE->getDecl()->printQualifiedName(OS);
780 } else
Craig Topper36250ad2014-05-12 05:36:57 +0000781 (*I)->printPretty(OS, nullptr, Policy, 0);
Alexey Bataevd48bcd82014-03-31 03:36:38 +0000782 }
Alexey Bataev756c1962013-09-24 03:17:45 +0000783}
Alexey Bataev5ec3eb12013-07-19 03:13:43 +0000784
785void OMPClausePrinter::VisitOMPPrivateClause(OMPPrivateClause *Node) {
786 if (!Node->varlist_empty()) {
787 OS << "private";
Alexey Bataev756c1962013-09-24 03:17:45 +0000788 VisitOMPClauseList(Node, '(');
Alexey Bataev5ec3eb12013-07-19 03:13:43 +0000789 OS << ")";
790 }
791}
792
Alexey Bataevd5af8e42013-10-01 05:32:34 +0000793void OMPClausePrinter::VisitOMPFirstprivateClause(OMPFirstprivateClause *Node) {
794 if (!Node->varlist_empty()) {
795 OS << "firstprivate";
796 VisitOMPClauseList(Node, '(');
797 OS << ")";
798 }
799}
800
Alexander Musman1bb328c2014-06-04 13:06:39 +0000801void OMPClausePrinter::VisitOMPLastprivateClause(OMPLastprivateClause *Node) {
802 if (!Node->varlist_empty()) {
803 OS << "lastprivate";
804 VisitOMPClauseList(Node, '(');
805 OS << ")";
806 }
807}
808
Alexey Bataev758e55e2013-09-06 18:03:48 +0000809void OMPClausePrinter::VisitOMPSharedClause(OMPSharedClause *Node) {
810 if (!Node->varlist_empty()) {
811 OS << "shared";
Alexey Bataev756c1962013-09-24 03:17:45 +0000812 VisitOMPClauseList(Node, '(');
Alexey Bataev758e55e2013-09-06 18:03:48 +0000813 OS << ")";
814 }
815}
816
Alexey Bataevc5e02582014-06-16 07:08:35 +0000817void OMPClausePrinter::VisitOMPReductionClause(OMPReductionClause *Node) {
818 if (!Node->varlist_empty()) {
819 OS << "reduction(";
820 NestedNameSpecifier *QualifierLoc =
821 Node->getQualifierLoc().getNestedNameSpecifier();
822 OverloadedOperatorKind OOK =
823 Node->getNameInfo().getName().getCXXOverloadedOperator();
824 if (QualifierLoc == nullptr && OOK != OO_None) {
825 // Print reduction identifier in C format
826 OS << getOperatorSpelling(OOK);
827 } else {
828 // Use C++ format
829 if (QualifierLoc != nullptr)
830 QualifierLoc->print(OS, Policy);
831 OS << Node->getNameInfo();
832 }
833 OS << ":";
834 VisitOMPClauseList(Node, ' ');
835 OS << ")";
836 }
837}
838
Alexander Musman8dba6642014-04-22 13:09:42 +0000839void OMPClausePrinter::VisitOMPLinearClause(OMPLinearClause *Node) {
840 if (!Node->varlist_empty()) {
841 OS << "linear";
Alexey Bataev182227b2015-08-20 10:54:39 +0000842 if (Node->getModifierLoc().isValid()) {
843 OS << '('
844 << getOpenMPSimpleClauseTypeName(OMPC_linear, Node->getModifier());
845 }
Alexander Musman8dba6642014-04-22 13:09:42 +0000846 VisitOMPClauseList(Node, '(');
Alexey Bataev182227b2015-08-20 10:54:39 +0000847 if (Node->getModifierLoc().isValid())
848 OS << ')';
Craig Topper36250ad2014-05-12 05:36:57 +0000849 if (Node->getStep() != nullptr) {
Alexander Musman8dba6642014-04-22 13:09:42 +0000850 OS << ": ";
Craig Topper36250ad2014-05-12 05:36:57 +0000851 Node->getStep()->printPretty(OS, nullptr, Policy, 0);
Alexander Musman8dba6642014-04-22 13:09:42 +0000852 }
853 OS << ")";
854 }
855}
856
Alexander Musmanf0d76e72014-05-29 14:36:25 +0000857void OMPClausePrinter::VisitOMPAlignedClause(OMPAlignedClause *Node) {
858 if (!Node->varlist_empty()) {
859 OS << "aligned";
860 VisitOMPClauseList(Node, '(');
861 if (Node->getAlignment() != nullptr) {
862 OS << ": ";
863 Node->getAlignment()->printPretty(OS, nullptr, Policy, 0);
864 }
865 OS << ")";
866 }
867}
868
Alexey Bataevd48bcd82014-03-31 03:36:38 +0000869void OMPClausePrinter::VisitOMPCopyinClause(OMPCopyinClause *Node) {
870 if (!Node->varlist_empty()) {
871 OS << "copyin";
872 VisitOMPClauseList(Node, '(');
873 OS << ")";
874 }
875}
876
Alexey Bataevbae9a792014-06-27 10:37:06 +0000877void OMPClausePrinter::VisitOMPCopyprivateClause(OMPCopyprivateClause *Node) {
878 if (!Node->varlist_empty()) {
879 OS << "copyprivate";
880 VisitOMPClauseList(Node, '(');
881 OS << ")";
882 }
883}
884
Alexey Bataev6125da92014-07-21 11:26:11 +0000885void OMPClausePrinter::VisitOMPFlushClause(OMPFlushClause *Node) {
886 if (!Node->varlist_empty()) {
887 VisitOMPClauseList(Node, '(');
888 OS << ")";
889 }
890}
Alexey Bataev1c2cfbc2015-06-23 14:25:19 +0000891
892void OMPClausePrinter::VisitOMPDependClause(OMPDependClause *Node) {
Alexey Bataeveb482352015-12-18 05:05:56 +0000893 OS << "depend(";
894 OS << getOpenMPSimpleClauseTypeName(Node->getClauseKind(),
895 Node->getDependencyKind());
Alexey Bataev1c2cfbc2015-06-23 14:25:19 +0000896 if (!Node->varlist_empty()) {
Alexey Bataeveb482352015-12-18 05:05:56 +0000897 OS << " :";
Alexey Bataev1c2cfbc2015-06-23 14:25:19 +0000898 VisitOMPClauseList(Node, ' ');
Alexey Bataev1c2cfbc2015-06-23 14:25:19 +0000899 }
Alexey Bataeveb482352015-12-18 05:05:56 +0000900 OS << ")";
Alexey Bataev1c2cfbc2015-06-23 14:25:19 +0000901}
Kelvin Li0bff7af2015-11-23 05:32:03 +0000902
903void OMPClausePrinter::VisitOMPMapClause(OMPMapClause *Node) {
904 if (!Node->varlist_empty()) {
905 OS << "map(";
906 if (Node->getMapType() != OMPC_MAP_unknown) {
907 if (Node->getMapTypeModifier() != OMPC_MAP_unknown) {
908 OS << getOpenMPSimpleClauseTypeName(OMPC_map,
909 Node->getMapTypeModifier());
910 OS << ',';
911 }
912 OS << getOpenMPSimpleClauseTypeName(OMPC_map, Node->getMapType());
913 OS << ':';
914 }
915 VisitOMPClauseList(Node, ' ');
916 OS << ")";
917 }
918}
Carlo Bertollib4adf552016-01-15 18:50:31 +0000919
Samuel Antao661c0902016-05-26 17:39:58 +0000920void OMPClausePrinter::VisitOMPToClause(OMPToClause *Node) {
921 if (!Node->varlist_empty()) {
922 OS << "to";
923 VisitOMPClauseList(Node, '(');
924 OS << ")";
925 }
926}
927
Samuel Antaoec172c62016-05-26 17:49:04 +0000928void OMPClausePrinter::VisitOMPFromClause(OMPFromClause *Node) {
929 if (!Node->varlist_empty()) {
930 OS << "from";
931 VisitOMPClauseList(Node, '(');
932 OS << ")";
933 }
934}
935
Carlo Bertollib4adf552016-01-15 18:50:31 +0000936void OMPClausePrinter::VisitOMPDistScheduleClause(OMPDistScheduleClause *Node) {
937 OS << "dist_schedule(" << getOpenMPSimpleClauseTypeName(
938 OMPC_dist_schedule, Node->getDistScheduleKind());
Alexey Bataev3392d762016-02-16 11:18:12 +0000939 if (auto *E = Node->getChunkSize()) {
Carlo Bertollib4adf552016-01-15 18:50:31 +0000940 OS << ", ";
Alexey Bataev60da77e2016-02-29 05:54:20 +0000941 E->printPretty(OS, nullptr, Policy);
Carlo Bertollib4adf552016-01-15 18:50:31 +0000942 }
943 OS << ")";
944}
Arpith Chacko Jacob3cf89042016-01-26 16:37:23 +0000945
946void OMPClausePrinter::VisitOMPDefaultmapClause(OMPDefaultmapClause *Node) {
947 OS << "defaultmap(";
948 OS << getOpenMPSimpleClauseTypeName(OMPC_defaultmap,
949 Node->getDefaultmapModifier());
950 OS << ": ";
951 OS << getOpenMPSimpleClauseTypeName(OMPC_defaultmap,
952 Node->getDefaultmapKind());
953 OS << ")";
954}
Carlo Bertolli2404b172016-07-13 15:37:16 +0000955
956void OMPClausePrinter::VisitOMPUseDevicePtrClause(OMPUseDevicePtrClause *Node) {
957 if (!Node->varlist_empty()) {
958 OS << "use_device_ptr";
959 VisitOMPClauseList(Node, '(');
960 OS << ")";
961 }
962}
Carlo Bertolli70594e92016-07-13 17:16:49 +0000963
964void OMPClausePrinter::VisitOMPIsDevicePtrClause(OMPIsDevicePtrClause *Node) {
965 if (!Node->varlist_empty()) {
966 OS << "is_device_ptr";
967 VisitOMPClauseList(Node, '(');
968 OS << ")";
969 }
970}
Alexander Kornienkoab9db512015-06-22 23:07:51 +0000971}
Alexey Bataev5ec3eb12013-07-19 03:13:43 +0000972
973//===----------------------------------------------------------------------===//
974// OpenMP directives printing methods
975//===----------------------------------------------------------------------===//
976
Alexey Bataev1b59ab52014-02-27 08:29:12 +0000977void StmtPrinter::PrintOMPExecutableDirective(OMPExecutableDirective *S) {
Alexey Bataevaadd52e2014-02-13 05:29:23 +0000978 OMPClausePrinter Printer(OS, Policy);
Alexey Bataev1b59ab52014-02-27 08:29:12 +0000979 ArrayRef<OMPClause *> Clauses = S->clauses();
Alexey Bataev5ec3eb12013-07-19 03:13:43 +0000980 for (ArrayRef<OMPClause *>::iterator I = Clauses.begin(), E = Clauses.end();
981 I != E; ++I)
982 if (*I && !(*I)->isImplicit()) {
983 Printer.Visit(*I);
984 OS << ' ';
985 }
986 OS << "\n";
Alexey Bataev68446b72014-07-18 07:47:19 +0000987 if (S->hasAssociatedStmt() && S->getAssociatedStmt()) {
Alexey Bataev1b59ab52014-02-27 08:29:12 +0000988 assert(isa<CapturedStmt>(S->getAssociatedStmt()) &&
Alexey Bataev5ec3eb12013-07-19 03:13:43 +0000989 "Expected captured statement!");
Alexey Bataev1b59ab52014-02-27 08:29:12 +0000990 Stmt *CS = cast<CapturedStmt>(S->getAssociatedStmt())->getCapturedStmt();
Alexey Bataev5ec3eb12013-07-19 03:13:43 +0000991 PrintStmt(CS);
992 }
993}
Alexey Bataev1b59ab52014-02-27 08:29:12 +0000994
995void StmtPrinter::VisitOMPParallelDirective(OMPParallelDirective *Node) {
996 Indent() << "#pragma omp parallel ";
997 PrintOMPExecutableDirective(Node);
998}
999
1000void StmtPrinter::VisitOMPSimdDirective(OMPSimdDirective *Node) {
1001 Indent() << "#pragma omp simd ";
1002 PrintOMPExecutableDirective(Node);
1003}
1004
Alexey Bataevf29276e2014-06-18 04:14:57 +00001005void StmtPrinter::VisitOMPForDirective(OMPForDirective *Node) {
1006 Indent() << "#pragma omp for ";
1007 PrintOMPExecutableDirective(Node);
1008}
1009
Alexander Musmanf82886e2014-09-18 05:12:34 +00001010void StmtPrinter::VisitOMPForSimdDirective(OMPForSimdDirective *Node) {
1011 Indent() << "#pragma omp for simd ";
1012 PrintOMPExecutableDirective(Node);
1013}
1014
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00001015void StmtPrinter::VisitOMPSectionsDirective(OMPSectionsDirective *Node) {
1016 Indent() << "#pragma omp sections ";
1017 PrintOMPExecutableDirective(Node);
1018}
1019
Alexey Bataev1e0498a2014-06-26 08:21:58 +00001020void StmtPrinter::VisitOMPSectionDirective(OMPSectionDirective *Node) {
1021 Indent() << "#pragma omp section";
1022 PrintOMPExecutableDirective(Node);
1023}
1024
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00001025void StmtPrinter::VisitOMPSingleDirective(OMPSingleDirective *Node) {
1026 Indent() << "#pragma omp single ";
1027 PrintOMPExecutableDirective(Node);
1028}
1029
Alexander Musman80c22892014-07-17 08:54:58 +00001030void StmtPrinter::VisitOMPMasterDirective(OMPMasterDirective *Node) {
1031 Indent() << "#pragma omp master";
1032 PrintOMPExecutableDirective(Node);
1033}
1034
Alexander Musmand9ed09f2014-07-21 09:42:05 +00001035void StmtPrinter::VisitOMPCriticalDirective(OMPCriticalDirective *Node) {
1036 Indent() << "#pragma omp critical";
1037 if (Node->getDirectiveName().getName()) {
1038 OS << " (";
1039 Node->getDirectiveName().printName(OS);
1040 OS << ")";
1041 }
Alexey Bataev28c75412015-12-15 08:19:24 +00001042 OS << " ";
Alexander Musmand9ed09f2014-07-21 09:42:05 +00001043 PrintOMPExecutableDirective(Node);
1044}
1045
Alexey Bataev4acb8592014-07-07 13:01:15 +00001046void StmtPrinter::VisitOMPParallelForDirective(OMPParallelForDirective *Node) {
1047 Indent() << "#pragma omp parallel for ";
1048 PrintOMPExecutableDirective(Node);
1049}
1050
Alexander Musmane4e893b2014-09-23 09:33:00 +00001051void StmtPrinter::VisitOMPParallelForSimdDirective(
1052 OMPParallelForSimdDirective *Node) {
1053 Indent() << "#pragma omp parallel for simd ";
1054 PrintOMPExecutableDirective(Node);
1055}
1056
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00001057void StmtPrinter::VisitOMPParallelSectionsDirective(
1058 OMPParallelSectionsDirective *Node) {
1059 Indent() << "#pragma omp parallel sections ";
1060 PrintOMPExecutableDirective(Node);
1061}
1062
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00001063void StmtPrinter::VisitOMPTaskDirective(OMPTaskDirective *Node) {
1064 Indent() << "#pragma omp task ";
1065 PrintOMPExecutableDirective(Node);
1066}
1067
Alexey Bataev68446b72014-07-18 07:47:19 +00001068void StmtPrinter::VisitOMPTaskyieldDirective(OMPTaskyieldDirective *Node) {
1069 Indent() << "#pragma omp taskyield";
1070 PrintOMPExecutableDirective(Node);
1071}
1072
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00001073void StmtPrinter::VisitOMPBarrierDirective(OMPBarrierDirective *Node) {
1074 Indent() << "#pragma omp barrier";
1075 PrintOMPExecutableDirective(Node);
1076}
1077
Alexey Bataev2df347a2014-07-18 10:17:07 +00001078void StmtPrinter::VisitOMPTaskwaitDirective(OMPTaskwaitDirective *Node) {
1079 Indent() << "#pragma omp taskwait";
1080 PrintOMPExecutableDirective(Node);
1081}
1082
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00001083void StmtPrinter::VisitOMPTaskgroupDirective(OMPTaskgroupDirective *Node) {
1084 Indent() << "#pragma omp taskgroup";
1085 PrintOMPExecutableDirective(Node);
1086}
1087
Alexey Bataev6125da92014-07-21 11:26:11 +00001088void StmtPrinter::VisitOMPFlushDirective(OMPFlushDirective *Node) {
1089 Indent() << "#pragma omp flush ";
1090 PrintOMPExecutableDirective(Node);
1091}
1092
Alexey Bataev9fb6e642014-07-22 06:45:04 +00001093void StmtPrinter::VisitOMPOrderedDirective(OMPOrderedDirective *Node) {
Alexey Bataev346265e2015-09-25 10:37:12 +00001094 Indent() << "#pragma omp ordered ";
Alexey Bataev9fb6e642014-07-22 06:45:04 +00001095 PrintOMPExecutableDirective(Node);
1096}
1097
Alexey Bataev0162e452014-07-22 10:10:35 +00001098void StmtPrinter::VisitOMPAtomicDirective(OMPAtomicDirective *Node) {
1099 Indent() << "#pragma omp atomic ";
1100 PrintOMPExecutableDirective(Node);
1101}
1102
Alexey Bataev0bd520b2014-09-19 08:19:49 +00001103void StmtPrinter::VisitOMPTargetDirective(OMPTargetDirective *Node) {
1104 Indent() << "#pragma omp target ";
1105 PrintOMPExecutableDirective(Node);
1106}
1107
Michael Wong65f367f2015-07-21 13:44:28 +00001108void StmtPrinter::VisitOMPTargetDataDirective(OMPTargetDataDirective *Node) {
1109 Indent() << "#pragma omp target data ";
1110 PrintOMPExecutableDirective(Node);
1111}
1112
Samuel Antaodf67fc42016-01-19 19:15:56 +00001113void StmtPrinter::VisitOMPTargetEnterDataDirective(
1114 OMPTargetEnterDataDirective *Node) {
1115 Indent() << "#pragma omp target enter data ";
1116 PrintOMPExecutableDirective(Node);
1117}
1118
Samuel Antao72590762016-01-19 20:04:50 +00001119void StmtPrinter::VisitOMPTargetExitDataDirective(
1120 OMPTargetExitDataDirective *Node) {
1121 Indent() << "#pragma omp target exit data ";
1122 PrintOMPExecutableDirective(Node);
1123}
1124
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00001125void StmtPrinter::VisitOMPTargetParallelDirective(
1126 OMPTargetParallelDirective *Node) {
1127 Indent() << "#pragma omp target parallel ";
1128 PrintOMPExecutableDirective(Node);
1129}
1130
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00001131void StmtPrinter::VisitOMPTargetParallelForDirective(
1132 OMPTargetParallelForDirective *Node) {
1133 Indent() << "#pragma omp target parallel for ";
1134 PrintOMPExecutableDirective(Node);
1135}
1136
Alexey Bataev13314bf2014-10-09 04:18:56 +00001137void StmtPrinter::VisitOMPTeamsDirective(OMPTeamsDirective *Node) {
1138 Indent() << "#pragma omp teams ";
1139 PrintOMPExecutableDirective(Node);
1140}
1141
Alexey Bataev6d4ed052015-07-01 06:57:41 +00001142void StmtPrinter::VisitOMPCancellationPointDirective(
1143 OMPCancellationPointDirective *Node) {
1144 Indent() << "#pragma omp cancellation point "
1145 << getOpenMPDirectiveName(Node->getCancelRegion());
1146 PrintOMPExecutableDirective(Node);
1147}
Alexey Bataev80909872015-07-02 11:25:17 +00001148
1149void StmtPrinter::VisitOMPCancelDirective(OMPCancelDirective *Node) {
1150 Indent() << "#pragma omp cancel "
Alexey Bataev87933c72015-09-18 08:07:34 +00001151 << getOpenMPDirectiveName(Node->getCancelRegion()) << " ";
Alexey Bataev80909872015-07-02 11:25:17 +00001152 PrintOMPExecutableDirective(Node);
1153}
Alexey Bataev49f6e782015-12-01 04:18:41 +00001154
1155void StmtPrinter::VisitOMPTaskLoopDirective(OMPTaskLoopDirective *Node) {
1156 Indent() << "#pragma omp taskloop ";
1157 PrintOMPExecutableDirective(Node);
1158}
1159
Alexey Bataev0a6ed842015-12-03 09:40:15 +00001160void StmtPrinter::VisitOMPTaskLoopSimdDirective(
1161 OMPTaskLoopSimdDirective *Node) {
1162 Indent() << "#pragma omp taskloop simd ";
1163 PrintOMPExecutableDirective(Node);
1164}
1165
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00001166void StmtPrinter::VisitOMPDistributeDirective(OMPDistributeDirective *Node) {
1167 Indent() << "#pragma omp distribute ";
1168 PrintOMPExecutableDirective(Node);
1169}
1170
Samuel Antao686c70c2016-05-26 17:30:50 +00001171void StmtPrinter::VisitOMPTargetUpdateDirective(
1172 OMPTargetUpdateDirective *Node) {
1173 Indent() << "#pragma omp target update ";
1174 PrintOMPExecutableDirective(Node);
1175}
1176
Carlo Bertolli9925f152016-06-27 14:55:37 +00001177void StmtPrinter::VisitOMPDistributeParallelForDirective(
1178 OMPDistributeParallelForDirective *Node) {
1179 Indent() << "#pragma omp distribute parallel for ";
1180 PrintOMPExecutableDirective(Node);
1181}
1182
Kelvin Li4a39add2016-07-05 05:00:15 +00001183void StmtPrinter::VisitOMPDistributeParallelForSimdDirective(
1184 OMPDistributeParallelForSimdDirective *Node) {
1185 Indent() << "#pragma omp distribute parallel for simd ";
1186 PrintOMPExecutableDirective(Node);
1187}
1188
Kelvin Li787f3fc2016-07-06 04:45:38 +00001189void StmtPrinter::VisitOMPDistributeSimdDirective(
1190 OMPDistributeSimdDirective *Node) {
1191 Indent() << "#pragma omp distribute simd ";
1192 PrintOMPExecutableDirective(Node);
1193}
1194
Kelvin Lia579b912016-07-14 02:54:56 +00001195void StmtPrinter::VisitOMPTargetParallelForSimdDirective(
1196 OMPTargetParallelForSimdDirective *Node) {
1197 Indent() << "#pragma omp target parallel for simd ";
1198 PrintOMPExecutableDirective(Node);
1199}
1200
Kelvin Li986330c2016-07-20 22:57:10 +00001201void StmtPrinter::VisitOMPTargetSimdDirective(OMPTargetSimdDirective *Node) {
1202 Indent() << "#pragma omp target simd ";
1203 PrintOMPExecutableDirective(Node);
1204}
1205
Kelvin Li02532872016-08-05 14:37:37 +00001206void StmtPrinter::VisitOMPTeamsDistributeDirective(
1207 OMPTeamsDistributeDirective *Node) {
1208 Indent() << "#pragma omp teams distribute ";
1209 PrintOMPExecutableDirective(Node);
1210}
1211
Kelvin Li4e325f72016-10-25 12:50:55 +00001212void StmtPrinter::VisitOMPTeamsDistributeSimdDirective(
1213 OMPTeamsDistributeSimdDirective *Node) {
1214 Indent() << "#pragma omp teams distribute simd ";
1215 PrintOMPExecutableDirective(Node);
1216}
1217
Kelvin Li579e41c2016-11-30 23:51:03 +00001218void StmtPrinter::VisitOMPTeamsDistributeParallelForSimdDirective(
1219 OMPTeamsDistributeParallelForSimdDirective *Node) {
1220 Indent() << "#pragma omp teams distribute parallel for simd ";
1221 PrintOMPExecutableDirective(Node);
1222}
1223
Kelvin Li7ade93f2016-12-09 03:24:30 +00001224void StmtPrinter::VisitOMPTeamsDistributeParallelForDirective(
1225 OMPTeamsDistributeParallelForDirective *Node) {
1226 Indent() << "#pragma omp teams distribute parallel for ";
1227 PrintOMPExecutableDirective(Node);
1228}
1229
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001230//===----------------------------------------------------------------------===//
Chris Lattner71e23ce2006-11-04 20:18:38 +00001231// Expr printing methods.
1232//===----------------------------------------------------------------------===//
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00001233
Chris Lattner882f7882006-11-04 18:52:07 +00001234void StmtPrinter::VisitDeclRefExpr(DeclRefExpr *Node) {
Alexey Bataev60da77e2016-02-29 05:54:20 +00001235 if (auto *OCED = dyn_cast<OMPCapturedExprDecl>(Node->getDecl())) {
1236 OCED->getInit()->IgnoreImpCasts()->printPretty(OS, nullptr, Policy);
1237 return;
1238 }
Douglas Gregor4bd90e52009-10-23 18:54:35 +00001239 if (NestedNameSpecifier *Qualifier = Node->getQualifier())
1240 Qualifier->print(OS, Policy);
Abramo Bagnara7945c982012-01-27 09:46:47 +00001241 if (Node->hasTemplateKeyword())
1242 OS << "template ";
Abramo Bagnarad6d2f182010-08-11 22:01:17 +00001243 OS << Node->getNameInfo();
John McCallb3774b52010-08-19 23:49:38 +00001244 if (Node->hasExplicitTemplateArgs())
Benjamin Kramer9170e912013-02-22 15:46:01 +00001245 TemplateSpecializationType::PrintTemplateArgumentList(
David Majnemer6fbeee32016-07-07 04:43:07 +00001246 OS, Node->template_arguments(), Policy);
Douglas Gregorc7acfdf2009-01-06 05:10:23 +00001247}
1248
John McCall8cd78132009-11-19 22:55:06 +00001249void StmtPrinter::VisitDependentScopeDeclRefExpr(
1250 DependentScopeDeclRefExpr *Node) {
Axel Naumann20b27862011-01-24 15:44:00 +00001251 if (NestedNameSpecifier *Qualifier = Node->getQualifier())
1252 Qualifier->print(OS, Policy);
Abramo Bagnara7945c982012-01-27 09:46:47 +00001253 if (Node->hasTemplateKeyword())
1254 OS << "template ";
Abramo Bagnarad6d2f182010-08-11 22:01:17 +00001255 OS << Node->getNameInfo();
John McCalle66edc12009-11-24 19:00:30 +00001256 if (Node->hasExplicitTemplateArgs())
Benjamin Kramer9170e912013-02-22 15:46:01 +00001257 TemplateSpecializationType::PrintTemplateArgumentList(
David Majnemer6fbeee32016-07-07 04:43:07 +00001258 OS, Node->template_arguments(), Policy);
Douglas Gregor90a1a652009-03-19 17:26:29 +00001259}
1260
John McCalld14a8642009-11-21 08:51:07 +00001261void StmtPrinter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *Node) {
Douglas Gregora727cb92009-06-30 22:34:41 +00001262 if (Node->getQualifier())
1263 Node->getQualifier()->print(OS, Policy);
Abramo Bagnara7945c982012-01-27 09:46:47 +00001264 if (Node->hasTemplateKeyword())
1265 OS << "template ";
Abramo Bagnarad6d2f182010-08-11 22:01:17 +00001266 OS << Node->getNameInfo();
John McCalle66edc12009-11-24 19:00:30 +00001267 if (Node->hasExplicitTemplateArgs())
Benjamin Kramer9170e912013-02-22 15:46:01 +00001268 TemplateSpecializationType::PrintTemplateArgumentList(
David Majnemer6fbeee32016-07-07 04:43:07 +00001269 OS, Node->template_arguments(), Policy);
Douglas Gregora727cb92009-06-30 22:34:41 +00001270}
1271
Steve Naroffe46504b2007-11-12 14:29:37 +00001272void StmtPrinter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *Node) {
Fariborz Jahanian21f54ee2007-11-12 22:29:28 +00001273 if (Node->getBase()) {
1274 PrintExpr(Node->getBase());
1275 OS << (Node->isArrow() ? "->" : ".");
1276 }
Benjamin Kramerb89514a2011-10-14 18:45:37 +00001277 OS << *Node->getDecl();
Steve Naroffe46504b2007-11-12 14:29:37 +00001278}
1279
Steve Naroffec944032008-05-30 00:40:33 +00001280void StmtPrinter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *Node) {
Fariborz Jahanian681c0752010-10-14 16:04:05 +00001281 if (Node->isSuperReceiver())
1282 OS << "super.";
Richard Trieu6d92e502014-02-06 23:26:23 +00001283 else if (Node->isObjectReceiver() && Node->getBase()) {
Steve Naroffec944032008-05-30 00:40:33 +00001284 PrintExpr(Node->getBase());
1285 OS << ".";
Richard Trieu6d92e502014-02-06 23:26:23 +00001286 } else if (Node->isClassReceiver() && Node->getClassReceiver()) {
1287 OS << Node->getClassReceiver()->getName() << ".";
Steve Naroffec944032008-05-30 00:40:33 +00001288 }
Fariborz Jahanian681c0752010-10-14 16:04:05 +00001289
John McCallb7bd14f2010-12-02 01:19:52 +00001290 if (Node->isImplicitProperty())
Aaron Ballmanb190f972014-01-03 17:59:55 +00001291 Node->getImplicitPropertyGetter()->getSelector().print(OS);
John McCallb7bd14f2010-12-02 01:19:52 +00001292 else
1293 OS << Node->getExplicitProperty()->getName();
Fariborz Jahanian8a1810f2008-11-22 18:39:36 +00001294}
1295
Ted Kremeneke65b0862012-03-06 20:05:56 +00001296void StmtPrinter::VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr *Node) {
1297
1298 PrintExpr(Node->getBaseExpr());
1299 OS << "[";
1300 PrintExpr(Node->getKeyExpr());
1301 OS << "]";
1302}
1303
Chris Lattner6307f192008-08-10 01:53:14 +00001304void StmtPrinter::VisitPredefinedExpr(PredefinedExpr *Node) {
Alexey Bataevec474782014-10-09 08:45:04 +00001305 OS << PredefinedExpr::getIdentTypeName(Node->getIdentType());
Anders Carlsson625bfc82007-07-21 05:21:51 +00001306}
1307
Steve Naroffae4143e2007-04-26 20:39:23 +00001308void StmtPrinter::VisitCharacterLiteral(CharacterLiteral *Node) {
Chris Lattner6e9d9b32007-07-13 05:18:11 +00001309 unsigned value = Node->getValue();
Douglas Gregorfb65e592011-07-27 05:40:30 +00001310
1311 switch (Node->getKind()) {
1312 case CharacterLiteral::Ascii: break; // no prefix.
1313 case CharacterLiteral::Wide: OS << 'L'; break;
Aaron Ballman9a17c852016-01-07 20:59:26 +00001314 case CharacterLiteral::UTF8: OS << "u8"; break;
Douglas Gregorfb65e592011-07-27 05:40:30 +00001315 case CharacterLiteral::UTF16: OS << 'u'; break;
1316 case CharacterLiteral::UTF32: OS << 'U'; break;
1317 }
1318
Chris Lattner666115c2007-07-13 23:58:20 +00001319 switch (value) {
1320 case '\\':
1321 OS << "'\\\\'";
1322 break;
1323 case '\'':
1324 OS << "'\\''";
1325 break;
1326 case '\a':
1327 // TODO: K&R: the meaning of '\\a' is different in traditional C
1328 OS << "'\\a'";
1329 break;
1330 case '\b':
1331 OS << "'\\b'";
1332 break;
1333 // Nonstandard escape sequence.
1334 /*case '\e':
1335 OS << "'\\e'";
1336 break;*/
1337 case '\f':
1338 OS << "'\\f'";
1339 break;
1340 case '\n':
1341 OS << "'\\n'";
1342 break;
1343 case '\r':
1344 OS << "'\\r'";
1345 break;
1346 case '\t':
1347 OS << "'\\t'";
1348 break;
1349 case '\v':
1350 OS << "'\\v'";
1351 break;
1352 default:
Steven Watanabee43ae192016-02-13 02:31:28 +00001353 // A character literal might be sign-extended, which
1354 // would result in an invalid \U escape sequence.
1355 // FIXME: multicharacter literals such as '\xFF\xFF\xFF\xFF'
1356 // are not correctly handled.
1357 if ((value & ~0xFFu) == ~0xFFu && Node->getKind() == CharacterLiteral::Ascii)
1358 value &= 0xFFu;
Jordan Rose00d1b592013-02-08 22:30:27 +00001359 if (value < 256 && isPrintable((unsigned char)value))
Chris Lattner666115c2007-07-13 23:58:20 +00001360 OS << "'" << (char)value << "'";
Jordan Rose00d1b592013-02-08 22:30:27 +00001361 else if (value < 256)
1362 OS << "'\\x" << llvm::format("%02x", value) << "'";
1363 else if (value <= 0xFFFF)
1364 OS << "'\\u" << llvm::format("%04x", value) << "'";
1365 else
1366 OS << "'\\U" << llvm::format("%08x", value) << "'";
Chris Lattner6e9d9b32007-07-13 05:18:11 +00001367 }
Steve Naroffae4143e2007-04-26 20:39:23 +00001368}
1369
Steve Naroffdf7855b2007-02-21 23:46:25 +00001370void StmtPrinter::VisitIntegerLiteral(IntegerLiteral *Node) {
Chris Lattner06430412007-05-21 05:45:03 +00001371 bool isSigned = Node->getType()->isSignedIntegerType();
1372 OS << Node->getValue().toString(10, isSigned);
Mike Stump11289f42009-09-09 15:08:12 +00001373
Chris Lattner06430412007-05-21 05:45:03 +00001374 // Emit suffixes. Integer literals are always a builtin integer type.
John McCall9dd450b2009-09-21 23:43:11 +00001375 switch (Node->getType()->getAs<BuiltinType>()->getKind()) {
David Blaikie83d382b2011-09-23 05:06:16 +00001376 default: llvm_unreachable("Unexpected type for integer literal!");
David Majnemerbe09e8e2015-03-06 18:04:22 +00001377 case BuiltinType::Char_S:
1378 case BuiltinType::Char_U: OS << "i8"; break;
David Majnemer65a407c2014-06-21 18:46:07 +00001379 case BuiltinType::UChar: OS << "Ui8"; break;
1380 case BuiltinType::Short: OS << "i16"; break;
1381 case BuiltinType::UShort: OS << "Ui16"; break;
Chris Lattner06430412007-05-21 05:45:03 +00001382 case BuiltinType::Int: break; // no suffix.
1383 case BuiltinType::UInt: OS << 'U'; break;
1384 case BuiltinType::Long: OS << 'L'; break;
1385 case BuiltinType::ULong: OS << "UL"; break;
1386 case BuiltinType::LongLong: OS << "LL"; break;
1387 case BuiltinType::ULongLong: OS << "ULL"; break;
1388 }
Chris Lattner882f7882006-11-04 18:52:07 +00001389}
Benjamin Kramer8a526762012-09-20 14:07:17 +00001390
1391static void PrintFloatingLiteral(raw_ostream &OS, FloatingLiteral *Node,
1392 bool PrintSuffix) {
Dylan Noblesmith2c1dd272012-02-05 02:13:05 +00001393 SmallString<16> Str;
Eli Friedman53392062011-10-05 20:32:03 +00001394 Node->getValue().toString(Str);
1395 OS << Str;
Benjamin Kramer8a526762012-09-20 14:07:17 +00001396 if (Str.find_first_not_of("-0123456789") == StringRef::npos)
1397 OS << '.'; // Trailing dot in order to separate from ints.
1398
1399 if (!PrintSuffix)
1400 return;
1401
1402 // Emit suffixes. Float literals are always a builtin float type.
1403 switch (Node->getType()->getAs<BuiltinType>()->getKind()) {
1404 default: llvm_unreachable("Unexpected type for float literal!");
1405 case BuiltinType::Half: break; // FIXME: suffix?
1406 case BuiltinType::Double: break; // no suffix.
1407 case BuiltinType::Float: OS << 'F'; break;
1408 case BuiltinType::LongDouble: OS << 'L'; break;
Nemanja Ivanovicbb1ea2d2016-05-09 08:52:33 +00001409 case BuiltinType::Float128: OS << 'Q'; break;
Benjamin Kramer8a526762012-09-20 14:07:17 +00001410 }
1411}
1412
1413void StmtPrinter::VisitFloatingLiteral(FloatingLiteral *Node) {
1414 PrintFloatingLiteral(OS, Node, /*PrintSuffix=*/true);
Chris Lattner882f7882006-11-04 18:52:07 +00001415}
Chris Lattner1c20a172007-08-26 03:42:43 +00001416
1417void StmtPrinter::VisitImaginaryLiteral(ImaginaryLiteral *Node) {
1418 PrintExpr(Node->getSubExpr());
1419 OS << "i";
1420}
1421
Steve Naroffdf7855b2007-02-21 23:46:25 +00001422void StmtPrinter::VisitStringLiteral(StringLiteral *Str) {
Richard Trieudc355912012-06-13 20:25:24 +00001423 Str->outputString(OS);
Chris Lattner882f7882006-11-04 18:52:07 +00001424}
1425void StmtPrinter::VisitParenExpr(ParenExpr *Node) {
1426 OS << "(";
1427 PrintExpr(Node->getSubExpr());
Chris Lattnera076fde2007-05-31 18:21:33 +00001428 OS << ")";
Chris Lattner882f7882006-11-04 18:52:07 +00001429}
1430void StmtPrinter::VisitUnaryOperator(UnaryOperator *Node) {
Chris Lattnere5b60442007-08-23 21:46:40 +00001431 if (!Node->isPostfix()) {
Chris Lattner15768702006-11-05 23:54:51 +00001432 OS << UnaryOperator::getOpcodeStr(Node->getOpcode());
Mike Stump11289f42009-09-09 15:08:12 +00001433
Eli Friedman1cf25362009-06-14 22:39:26 +00001434 // Print a space if this is an "identifier operator" like __real, or if
1435 // it might be concatenated incorrectly like '+'.
Chris Lattnere5b60442007-08-23 21:46:40 +00001436 switch (Node->getOpcode()) {
1437 default: break;
John McCalle3027922010-08-25 11:45:40 +00001438 case UO_Real:
1439 case UO_Imag:
1440 case UO_Extension:
Chris Lattnere5b60442007-08-23 21:46:40 +00001441 OS << ' ';
1442 break;
John McCalle3027922010-08-25 11:45:40 +00001443 case UO_Plus:
1444 case UO_Minus:
Eli Friedman1cf25362009-06-14 22:39:26 +00001445 if (isa<UnaryOperator>(Node->getSubExpr()))
1446 OS << ' ';
1447 break;
Chris Lattnere5b60442007-08-23 21:46:40 +00001448 }
1449 }
Chris Lattner882f7882006-11-04 18:52:07 +00001450 PrintExpr(Node->getSubExpr());
Mike Stump11289f42009-09-09 15:08:12 +00001451
Chris Lattner15768702006-11-05 23:54:51 +00001452 if (Node->isPostfix())
1453 OS << UnaryOperator::getOpcodeStr(Node->getOpcode());
Chris Lattner882f7882006-11-04 18:52:07 +00001454}
Chris Lattner98dbf0a2007-08-30 17:59:59 +00001455
Douglas Gregor882211c2010-04-28 22:16:22 +00001456void StmtPrinter::VisitOffsetOfExpr(OffsetOfExpr *Node) {
1457 OS << "__builtin_offsetof(";
Benjamin Kramerc5720e92013-02-22 14:19:01 +00001458 Node->getTypeSourceInfo()->getType().print(OS, Policy);
1459 OS << ", ";
Douglas Gregor882211c2010-04-28 22:16:22 +00001460 bool PrintedSomething = false;
1461 for (unsigned i = 0, n = Node->getNumComponents(); i < n; ++i) {
James Y Knight7281c352015-12-29 22:31:18 +00001462 OffsetOfNode ON = Node->getComponent(i);
1463 if (ON.getKind() == OffsetOfNode::Array) {
Douglas Gregor882211c2010-04-28 22:16:22 +00001464 // Array node
1465 OS << "[";
1466 PrintExpr(Node->getIndexExpr(ON.getArrayExprIndex()));
1467 OS << "]";
1468 PrintedSomething = true;
1469 continue;
1470 }
Douglas Gregord1702062010-04-29 00:18:15 +00001471
1472 // Skip implicit base indirections.
James Y Knight7281c352015-12-29 22:31:18 +00001473 if (ON.getKind() == OffsetOfNode::Base)
Douglas Gregord1702062010-04-29 00:18:15 +00001474 continue;
1475
Douglas Gregor882211c2010-04-28 22:16:22 +00001476 // Field or identifier node.
1477 IdentifierInfo *Id = ON.getFieldName();
1478 if (!Id)
1479 continue;
Alexis Hunta8136cc2010-05-05 15:23:54 +00001480
Douglas Gregor882211c2010-04-28 22:16:22 +00001481 if (PrintedSomething)
1482 OS << ".";
1483 else
1484 PrintedSomething = true;
Alexis Hunta8136cc2010-05-05 15:23:54 +00001485 OS << Id->getName();
Douglas Gregor882211c2010-04-28 22:16:22 +00001486 }
1487 OS << ")";
1488}
1489
Peter Collingbournee190dee2011-03-11 19:24:49 +00001490void StmtPrinter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *Node){
1491 switch(Node->getKind()) {
1492 case UETT_SizeOf:
1493 OS << "sizeof";
1494 break;
1495 case UETT_AlignOf:
Richard Smith301bc212016-05-19 01:39:10 +00001496 if (Policy.Alignof)
Jordan Rose58d54722012-06-30 21:33:57 +00001497 OS << "alignof";
Richard Smith301bc212016-05-19 01:39:10 +00001498 else if (Policy.UnderscoreAlignof)
Jordan Rose58d54722012-06-30 21:33:57 +00001499 OS << "_Alignof";
1500 else
1501 OS << "__alignof";
Peter Collingbournee190dee2011-03-11 19:24:49 +00001502 break;
1503 case UETT_VecStep:
1504 OS << "vec_step";
1505 break;
Alexey Bataev00396512015-07-02 03:40:19 +00001506 case UETT_OpenMPRequiredSimdAlign:
1507 OS << "__builtin_omp_required_simd_align";
1508 break;
Peter Collingbournee190dee2011-03-11 19:24:49 +00001509 }
Benjamin Kramerc5720e92013-02-22 14:19:01 +00001510 if (Node->isArgumentType()) {
1511 OS << '(';
1512 Node->getArgumentType().print(OS, Policy);
1513 OS << ')';
1514 } else {
Sebastian Redl6f282892008-11-11 17:56:53 +00001515 OS << " ";
1516 PrintExpr(Node->getArgumentExpr());
1517 }
Chris Lattner882f7882006-11-04 18:52:07 +00001518}
Peter Collingbourne91147592011-04-15 00:35:48 +00001519
1520void StmtPrinter::VisitGenericSelectionExpr(GenericSelectionExpr *Node) {
1521 OS << "_Generic(";
1522 PrintExpr(Node->getControllingExpr());
1523 for (unsigned i = 0; i != Node->getNumAssocs(); ++i) {
1524 OS << ", ";
1525 QualType T = Node->getAssocType(i);
1526 if (T.isNull())
1527 OS << "default";
1528 else
Benjamin Kramerc5720e92013-02-22 14:19:01 +00001529 T.print(OS, Policy);
Peter Collingbourne91147592011-04-15 00:35:48 +00001530 OS << ": ";
1531 PrintExpr(Node->getAssocExpr(i));
1532 }
1533 OS << ")";
1534}
1535
Chris Lattner882f7882006-11-04 18:52:07 +00001536void StmtPrinter::VisitArraySubscriptExpr(ArraySubscriptExpr *Node) {
Ted Kremenekc81614d2007-08-20 16:18:38 +00001537 PrintExpr(Node->getLHS());
Chris Lattner882f7882006-11-04 18:52:07 +00001538 OS << "[";
Ted Kremenekc81614d2007-08-20 16:18:38 +00001539 PrintExpr(Node->getRHS());
Chris Lattner882f7882006-11-04 18:52:07 +00001540 OS << "]";
1541}
1542
Alexey Bataev1a3320e2015-08-25 14:24:04 +00001543void StmtPrinter::VisitOMPArraySectionExpr(OMPArraySectionExpr *Node) {
1544 PrintExpr(Node->getBase());
1545 OS << "[";
1546 if (Node->getLowerBound())
1547 PrintExpr(Node->getLowerBound());
Alexey Bataeved5fb672015-09-11 04:54:28 +00001548 if (Node->getColonLoc().isValid()) {
1549 OS << ":";
1550 if (Node->getLength())
1551 PrintExpr(Node->getLength());
1552 }
Alexey Bataev1a3320e2015-08-25 14:24:04 +00001553 OS << "]";
1554}
1555
Peter Collingbourne4b279a02011-02-08 21:17:54 +00001556void StmtPrinter::PrintCallArgs(CallExpr *Call) {
Chris Lattner882f7882006-11-04 18:52:07 +00001557 for (unsigned i = 0, e = Call->getNumArgs(); i != e; ++i) {
Chris Lattneraa9c7ae2008-04-08 04:40:51 +00001558 if (isa<CXXDefaultArgExpr>(Call->getArg(i))) {
1559 // Don't print any defaulted arguments
1560 break;
1561 }
1562
Chris Lattner882f7882006-11-04 18:52:07 +00001563 if (i) OS << ", ";
1564 PrintExpr(Call->getArg(i));
1565 }
Peter Collingbourne4b279a02011-02-08 21:17:54 +00001566}
1567
1568void StmtPrinter::VisitCallExpr(CallExpr *Call) {
1569 PrintExpr(Call->getCallee());
1570 OS << "(";
1571 PrintCallArgs(Call);
Chris Lattner882f7882006-11-04 18:52:07 +00001572 OS << ")";
1573}
1574void StmtPrinter::VisitMemberExpr(MemberExpr *Node) {
Douglas Gregore4b414c2009-01-08 22:45:41 +00001575 // FIXME: Suppress printing implicit bases (like "this")
1576 PrintExpr(Node->getBase());
David Blaikie8fab8e52012-11-12 19:12:12 +00001577
1578 MemberExpr *ParentMember = dyn_cast<MemberExpr>(Node->getBase());
David Blaikie6bffd6f2012-11-12 19:32:32 +00001579 FieldDecl *ParentDecl = ParentMember
Craig Topper36250ad2014-05-12 05:36:57 +00001580 ? dyn_cast<FieldDecl>(ParentMember->getMemberDecl()) : nullptr;
David Blaikie8fab8e52012-11-12 19:12:12 +00001581
David Blaikie6bffd6f2012-11-12 19:32:32 +00001582 if (!ParentDecl || !ParentDecl->isAnonymousStructOrUnion())
David Blaikie8fab8e52012-11-12 19:12:12 +00001583 OS << (Node->isArrow() ? "->" : ".");
David Blaikie8fab8e52012-11-12 19:12:12 +00001584
Fariborz Jahanian2990c022010-01-11 21:17:32 +00001585 if (FieldDecl *FD = dyn_cast<FieldDecl>(Node->getMemberDecl()))
1586 if (FD->isAnonymousStructOrUnion())
1587 return;
David Blaikie8fab8e52012-11-12 19:12:12 +00001588
Douglas Gregorf405d7e2009-08-31 23:41:50 +00001589 if (NestedNameSpecifier *Qualifier = Node->getQualifier())
1590 Qualifier->print(OS, Policy);
Abramo Bagnara7945c982012-01-27 09:46:47 +00001591 if (Node->hasTemplateKeyword())
1592 OS << "template ";
Abramo Bagnarad6d2f182010-08-11 22:01:17 +00001593 OS << Node->getMemberNameInfo();
John McCallb3774b52010-08-19 23:49:38 +00001594 if (Node->hasExplicitTemplateArgs())
Benjamin Kramer9170e912013-02-22 15:46:01 +00001595 TemplateSpecializationType::PrintTemplateArgumentList(
David Majnemer6fbeee32016-07-07 04:43:07 +00001596 OS, Node->template_arguments(), Policy);
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00001597}
Steve Naroffe87026a2009-07-24 17:54:45 +00001598void StmtPrinter::VisitObjCIsaExpr(ObjCIsaExpr *Node) {
1599 PrintExpr(Node->getBase());
1600 OS << (Node->isArrow() ? "->isa" : ".isa");
1601}
1602
Nate Begemance4d7fc2008-04-18 23:10:10 +00001603void StmtPrinter::VisitExtVectorElementExpr(ExtVectorElementExpr *Node) {
Steve Narofff7a5da12007-07-28 23:10:27 +00001604 PrintExpr(Node->getBase());
1605 OS << ".";
1606 OS << Node->getAccessor().getName();
1607}
Douglas Gregorf19b2312008-10-28 15:36:24 +00001608void StmtPrinter::VisitCStyleCastExpr(CStyleCastExpr *Node) {
Benjamin Kramerc5720e92013-02-22 14:19:01 +00001609 OS << '(';
1610 Node->getTypeAsWritten().print(OS, Policy);
1611 OS << ')';
Chris Lattner882f7882006-11-04 18:52:07 +00001612 PrintExpr(Node->getSubExpr());
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00001613}
Steve Naroff57eb2c52007-07-19 21:32:11 +00001614void StmtPrinter::VisitCompoundLiteralExpr(CompoundLiteralExpr *Node) {
Benjamin Kramerc5720e92013-02-22 14:19:01 +00001615 OS << '(';
1616 Node->getType().print(OS, Policy);
1617 OS << ')';
Steve Naroff57eb2c52007-07-19 21:32:11 +00001618 PrintExpr(Node->getInitializer());
1619}
Steve Naroff7a5af782007-07-13 16:58:59 +00001620void StmtPrinter::VisitImplicitCastExpr(ImplicitCastExpr *Node) {
Alp Toker028ed912013-12-06 17:56:43 +00001621 // No need to print anything, simply forward to the subexpression.
Steve Naroffb8ea4fb2007-07-13 23:32:42 +00001622 PrintExpr(Node->getSubExpr());
Steve Naroff7a5af782007-07-13 16:58:59 +00001623}
Chris Lattner882f7882006-11-04 18:52:07 +00001624void StmtPrinter::VisitBinaryOperator(BinaryOperator *Node) {
1625 PrintExpr(Node->getLHS());
1626 OS << " " << BinaryOperator::getOpcodeStr(Node->getOpcode()) << " ";
1627 PrintExpr(Node->getRHS());
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00001628}
Chris Lattner86928112007-08-25 02:00:02 +00001629void StmtPrinter::VisitCompoundAssignOperator(CompoundAssignOperator *Node) {
1630 PrintExpr(Node->getLHS());
1631 OS << " " << BinaryOperator::getOpcodeStr(Node->getOpcode()) << " ";
1632 PrintExpr(Node->getRHS());
1633}
Chris Lattner882f7882006-11-04 18:52:07 +00001634void StmtPrinter::VisitConditionalOperator(ConditionalOperator *Node) {
1635 PrintExpr(Node->getCond());
John McCallc07a0c72011-02-17 10:25:35 +00001636 OS << " ? ";
1637 PrintExpr(Node->getLHS());
1638 OS << " : ";
Chris Lattner882f7882006-11-04 18:52:07 +00001639 PrintExpr(Node->getRHS());
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00001640}
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00001641
Chris Lattnereefa10e2007-05-28 06:56:27 +00001642// GNU extensions.
1643
John McCallc07a0c72011-02-17 10:25:35 +00001644void
1645StmtPrinter::VisitBinaryConditionalOperator(BinaryConditionalOperator *Node) {
1646 PrintExpr(Node->getCommon());
1647 OS << " ?: ";
1648 PrintExpr(Node->getFalseExpr());
1649}
Chris Lattnerd268a7a2007-08-03 17:31:20 +00001650void StmtPrinter::VisitAddrLabelExpr(AddrLabelExpr *Node) {
Chris Lattnereefa10e2007-05-28 06:56:27 +00001651 OS << "&&" << Node->getLabel()->getName();
Chris Lattnereefa10e2007-05-28 06:56:27 +00001652}
1653
Chris Lattner366727f2007-07-24 16:58:17 +00001654void StmtPrinter::VisitStmtExpr(StmtExpr *E) {
1655 OS << "(";
1656 PrintRawCompoundStmt(E->getSubStmt());
1657 OS << ")";
1658}
1659
Steve Naroff9efdabc2007-08-03 21:21:27 +00001660void StmtPrinter::VisitChooseExpr(ChooseExpr *Node) {
1661 OS << "__builtin_choose_expr(";
1662 PrintExpr(Node->getCond());
Chris Lattner81a96882007-08-04 00:20:15 +00001663 OS << ", ";
Steve Naroff9efdabc2007-08-03 21:21:27 +00001664 PrintExpr(Node->getLHS());
Chris Lattner81a96882007-08-04 00:20:15 +00001665 OS << ", ";
Steve Naroff9efdabc2007-08-03 21:21:27 +00001666 PrintExpr(Node->getRHS());
1667 OS << ")";
1668}
Chris Lattner366727f2007-07-24 16:58:17 +00001669
Douglas Gregor3be4b122008-11-29 04:51:27 +00001670void StmtPrinter::VisitGNUNullExpr(GNUNullExpr *) {
1671 OS << "__null";
1672}
1673
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001674void StmtPrinter::VisitShuffleVectorExpr(ShuffleVectorExpr *Node) {
1675 OS << "__builtin_shufflevector(";
1676 for (unsigned i = 0, e = Node->getNumSubExprs(); i != e; ++i) {
1677 if (i) OS << ", ";
1678 PrintExpr(Node->getExpr(i));
1679 }
1680 OS << ")";
1681}
1682
Hal Finkelc4d7c822013-09-18 03:29:45 +00001683void StmtPrinter::VisitConvertVectorExpr(ConvertVectorExpr *Node) {
1684 OS << "__builtin_convertvector(";
1685 PrintExpr(Node->getSrcExpr());
1686 OS << ", ";
1687 Node->getType().print(OS, Policy);
1688 OS << ")";
1689}
1690
Anders Carlsson4692db02007-08-31 04:56:16 +00001691void StmtPrinter::VisitInitListExpr(InitListExpr* Node) {
Douglas Gregor36098ff2009-05-30 00:56:08 +00001692 if (Node->getSyntacticForm()) {
1693 Visit(Node->getSyntacticForm());
1694 return;
1695 }
1696
Richard Smithd0e102f2015-01-30 02:04:26 +00001697 OS << "{";
Anders Carlsson4692db02007-08-31 04:56:16 +00001698 for (unsigned i = 0, e = Node->getNumInits(); i != e; ++i) {
1699 if (i) OS << ", ";
Douglas Gregor347f7ea2009-01-28 21:54:33 +00001700 if (Node->getInit(i))
1701 PrintExpr(Node->getInit(i));
1702 else
Richard Smithd0e102f2015-01-30 02:04:26 +00001703 OS << "{}";
Anders Carlsson4692db02007-08-31 04:56:16 +00001704 }
Richard Smithd0e102f2015-01-30 02:04:26 +00001705 OS << "}";
Anders Carlsson4692db02007-08-31 04:56:16 +00001706}
1707
Richard Smith410306b2016-12-12 02:53:20 +00001708void StmtPrinter::VisitArrayInitLoopExpr(ArrayInitLoopExpr *Node) {
1709 // There's no way to express this expression in any of our supported
1710 // languages, so just emit something terse and (hopefully) clear.
1711 OS << "{";
1712 PrintExpr(Node->getSubExpr());
1713 OS << "}";
1714}
1715
1716void StmtPrinter::VisitArrayInitIndexExpr(ArrayInitIndexExpr *Node) {
1717 OS << "*";
1718}
1719
Nate Begeman5ec4b312009-08-10 23:49:36 +00001720void StmtPrinter::VisitParenListExpr(ParenListExpr* Node) {
Richard Smithd0e102f2015-01-30 02:04:26 +00001721 OS << "(";
Nate Begeman5ec4b312009-08-10 23:49:36 +00001722 for (unsigned i = 0, e = Node->getNumExprs(); i != e; ++i) {
1723 if (i) OS << ", ";
1724 PrintExpr(Node->getExpr(i));
1725 }
Richard Smithd0e102f2015-01-30 02:04:26 +00001726 OS << ")";
Nate Begeman5ec4b312009-08-10 23:49:36 +00001727}
1728
Douglas Gregore4a0bb72009-01-22 00:58:24 +00001729void StmtPrinter::VisitDesignatedInitExpr(DesignatedInitExpr *Node) {
Justin Bognercb337032015-05-28 22:19:36 +00001730 bool NeedsEquals = true;
David Majnemerf7e36092016-06-23 00:15:04 +00001731 for (const DesignatedInitExpr::Designator &D : Node->designators()) {
1732 if (D.isFieldDesignator()) {
1733 if (D.getDotLoc().isInvalid()) {
1734 if (IdentifierInfo *II = D.getFieldName()) {
Serge Pavloveb57aa62014-06-20 17:08:28 +00001735 OS << II->getName() << ":";
Justin Bognercb337032015-05-28 22:19:36 +00001736 NeedsEquals = false;
1737 }
Serge Pavloveb57aa62014-06-20 17:08:28 +00001738 } else {
David Majnemerf7e36092016-06-23 00:15:04 +00001739 OS << "." << D.getFieldName()->getName();
Serge Pavloveb57aa62014-06-20 17:08:28 +00001740 }
Douglas Gregor347f7ea2009-01-28 21:54:33 +00001741 } else {
1742 OS << "[";
David Majnemerf7e36092016-06-23 00:15:04 +00001743 if (D.isArrayDesignator()) {
1744 PrintExpr(Node->getArrayIndex(D));
Douglas Gregor347f7ea2009-01-28 21:54:33 +00001745 } else {
David Majnemerf7e36092016-06-23 00:15:04 +00001746 PrintExpr(Node->getArrayRangeStart(D));
Douglas Gregor347f7ea2009-01-28 21:54:33 +00001747 OS << " ... ";
David Majnemerf7e36092016-06-23 00:15:04 +00001748 PrintExpr(Node->getArrayRangeEnd(D));
Douglas Gregor347f7ea2009-01-28 21:54:33 +00001749 }
1750 OS << "]";
1751 }
1752 }
1753
Justin Bognercb337032015-05-28 22:19:36 +00001754 if (NeedsEquals)
1755 OS << " = ";
1756 else
1757 OS << " ";
Douglas Gregor347f7ea2009-01-28 21:54:33 +00001758 PrintExpr(Node->getInit());
Douglas Gregore4a0bb72009-01-22 00:58:24 +00001759}
1760
Yunzhong Gaocb779302015-06-10 00:27:52 +00001761void StmtPrinter::VisitDesignatedInitUpdateExpr(
1762 DesignatedInitUpdateExpr *Node) {
1763 OS << "{";
1764 OS << "/*base*/";
1765 PrintExpr(Node->getBase());
1766 OS << ", ";
1767
1768 OS << "/*updater*/";
1769 PrintExpr(Node->getUpdater());
1770 OS << "}";
1771}
1772
1773void StmtPrinter::VisitNoInitExpr(NoInitExpr *Node) {
1774 OS << "/*no init*/";
1775}
1776
Douglas Gregor0202cb42009-01-29 17:44:32 +00001777void StmtPrinter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *Node) {
Richard Smith301bc212016-05-19 01:39:10 +00001778 if (Node->getType()->getAsCXXRecordDecl()) {
Benjamin Kramerc5720e92013-02-22 14:19:01 +00001779 OS << "/*implicit*/";
1780 Node->getType().print(OS, Policy);
1781 OS << "()";
1782 } else {
1783 OS << "/*implicit*/(";
1784 Node->getType().print(OS, Policy);
1785 OS << ')';
Douglas Gregor278f52e2009-05-30 00:08:05 +00001786 if (Node->getType()->isRecordType())
1787 OS << "{}";
1788 else
1789 OS << 0;
1790 }
Douglas Gregor0202cb42009-01-29 17:44:32 +00001791}
1792
Anders Carlsson7e13ab82007-10-15 20:28:48 +00001793void StmtPrinter::VisitVAArgExpr(VAArgExpr *Node) {
Eli Friedman79635842009-05-30 04:20:30 +00001794 OS << "__builtin_va_arg(";
Anders Carlsson7e13ab82007-10-15 20:28:48 +00001795 PrintExpr(Node->getSubExpr());
1796 OS << ", ";
Benjamin Kramerc5720e92013-02-22 14:19:01 +00001797 Node->getType().print(OS, Policy);
Anders Carlsson7e13ab82007-10-15 20:28:48 +00001798 OS << ")";
1799}
1800
John McCallfe96e0b2011-11-06 09:01:30 +00001801void StmtPrinter::VisitPseudoObjectExpr(PseudoObjectExpr *Node) {
1802 PrintExpr(Node->getSyntacticForm());
1803}
1804
Eli Friedmandf14b3a2011-10-11 02:20:01 +00001805void StmtPrinter::VisitAtomicExpr(AtomicExpr *Node) {
Craig Topper36250ad2014-05-12 05:36:57 +00001806 const char *Name = nullptr;
Eli Friedmandf14b3a2011-10-11 02:20:01 +00001807 switch (Node->getOp()) {
Richard Smithfeea8832012-04-12 05:08:17 +00001808#define BUILTIN(ID, TYPE, ATTRS)
1809#define ATOMIC_BUILTIN(ID, TYPE, ATTRS) \
1810 case AtomicExpr::AO ## ID: \
1811 Name = #ID "("; \
1812 break;
1813#include "clang/Basic/Builtins.def"
Eli Friedmandf14b3a2011-10-11 02:20:01 +00001814 }
1815 OS << Name;
Richard Smithfeea8832012-04-12 05:08:17 +00001816
1817 // AtomicExpr stores its subexpressions in a permuted order.
Eli Friedmandf14b3a2011-10-11 02:20:01 +00001818 PrintExpr(Node->getPtr());
Richard Smithfeea8832012-04-12 05:08:17 +00001819 if (Node->getOp() != AtomicExpr::AO__c11_atomic_load &&
1820 Node->getOp() != AtomicExpr::AO__atomic_load_n) {
Eli Friedmandf14b3a2011-10-11 02:20:01 +00001821 OS << ", ";
Richard Smithdf6bee82013-05-01 19:02:43 +00001822 PrintExpr(Node->getVal1());
Eli Friedmandf14b3a2011-10-11 02:20:01 +00001823 }
Richard Smithfeea8832012-04-12 05:08:17 +00001824 if (Node->getOp() == AtomicExpr::AO__atomic_exchange ||
1825 Node->isCmpXChg()) {
Eli Friedmandf14b3a2011-10-11 02:20:01 +00001826 OS << ", ";
Richard Smithdf6bee82013-05-01 19:02:43 +00001827 PrintExpr(Node->getVal2());
Eli Friedmandf14b3a2011-10-11 02:20:01 +00001828 }
Richard Smithfeea8832012-04-12 05:08:17 +00001829 if (Node->getOp() == AtomicExpr::AO__atomic_compare_exchange ||
1830 Node->getOp() == AtomicExpr::AO__atomic_compare_exchange_n) {
Richard Smithfeea8832012-04-12 05:08:17 +00001831 OS << ", ";
Richard Smithdf6bee82013-05-01 19:02:43 +00001832 PrintExpr(Node->getWeak());
Richard Smithfeea8832012-04-12 05:08:17 +00001833 }
Richard Smithdf6bee82013-05-01 19:02:43 +00001834 if (Node->getOp() != AtomicExpr::AO__c11_atomic_init) {
1835 OS << ", ";
David Chisnallfa35df62012-01-16 17:27:18 +00001836 PrintExpr(Node->getOrder());
Richard Smithdf6bee82013-05-01 19:02:43 +00001837 }
Eli Friedmandf14b3a2011-10-11 02:20:01 +00001838 if (Node->isCmpXChg()) {
1839 OS << ", ";
1840 PrintExpr(Node->getOrderFail());
1841 }
1842 OS << ")";
1843}
1844
Chris Lattnereefa10e2007-05-28 06:56:27 +00001845// C++
Douglas Gregor993603d2008-11-14 16:09:21 +00001846void StmtPrinter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *Node) {
1847 const char *OpStrings[NUM_OVERLOADED_OPERATORS] = {
1848 "",
1849#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
1850 Spelling,
1851#include "clang/Basic/OperatorKinds.def"
1852 };
1853
1854 OverloadedOperatorKind Kind = Node->getOperator();
1855 if (Kind == OO_PlusPlus || Kind == OO_MinusMinus) {
1856 if (Node->getNumArgs() == 1) {
1857 OS << OpStrings[Kind] << ' ';
1858 PrintExpr(Node->getArg(0));
1859 } else {
1860 PrintExpr(Node->getArg(0));
1861 OS << ' ' << OpStrings[Kind];
1862 }
Eli Friedman8e236422012-10-12 22:45:14 +00001863 } else if (Kind == OO_Arrow) {
1864 PrintExpr(Node->getArg(0));
Douglas Gregor993603d2008-11-14 16:09:21 +00001865 } else if (Kind == OO_Call) {
1866 PrintExpr(Node->getArg(0));
1867 OS << '(';
1868 for (unsigned ArgIdx = 1; ArgIdx < Node->getNumArgs(); ++ArgIdx) {
1869 if (ArgIdx > 1)
1870 OS << ", ";
1871 if (!isa<CXXDefaultArgExpr>(Node->getArg(ArgIdx)))
1872 PrintExpr(Node->getArg(ArgIdx));
1873 }
1874 OS << ')';
1875 } else if (Kind == OO_Subscript) {
1876 PrintExpr(Node->getArg(0));
1877 OS << '[';
1878 PrintExpr(Node->getArg(1));
1879 OS << ']';
1880 } else if (Node->getNumArgs() == 1) {
1881 OS << OpStrings[Kind] << ' ';
1882 PrintExpr(Node->getArg(0));
1883 } else if (Node->getNumArgs() == 2) {
1884 PrintExpr(Node->getArg(0));
1885 OS << ' ' << OpStrings[Kind] << ' ';
1886 PrintExpr(Node->getArg(1));
1887 } else {
David Blaikie83d382b2011-09-23 05:06:16 +00001888 llvm_unreachable("unknown overloaded operator");
Douglas Gregor993603d2008-11-14 16:09:21 +00001889 }
1890}
Chris Lattnereefa10e2007-05-28 06:56:27 +00001891
Douglas Gregor97fd6e22008-12-22 05:46:06 +00001892void StmtPrinter::VisitCXXMemberCallExpr(CXXMemberCallExpr *Node) {
Benjamin Kramer00e8a192014-02-25 18:03:55 +00001893 // If we have a conversion operator call only print the argument.
1894 CXXMethodDecl *MD = Node->getMethodDecl();
1895 if (MD && isa<CXXConversionDecl>(MD)) {
1896 PrintExpr(Node->getImplicitObjectArgument());
1897 return;
1898 }
Douglas Gregor97fd6e22008-12-22 05:46:06 +00001899 VisitCallExpr(cast<CallExpr>(Node));
1900}
1901
Peter Collingbourne41f85462011-02-09 21:07:24 +00001902void StmtPrinter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *Node) {
1903 PrintExpr(Node->getCallee());
1904 OS << "<<<";
1905 PrintCallArgs(Node->getConfig());
1906 OS << ">>>(";
1907 PrintCallArgs(Node);
1908 OS << ")";
1909}
1910
Douglas Gregore200adc2008-10-27 19:41:14 +00001911void StmtPrinter::VisitCXXNamedCastExpr(CXXNamedCastExpr *Node) {
1912 OS << Node->getCastName() << '<';
Benjamin Kramerc5720e92013-02-22 14:19:01 +00001913 Node->getTypeAsWritten().print(OS, Policy);
1914 OS << ">(";
Chris Lattnereefa10e2007-05-28 06:56:27 +00001915 PrintExpr(Node->getSubExpr());
1916 OS << ")";
1917}
1918
Douglas Gregore200adc2008-10-27 19:41:14 +00001919void StmtPrinter::VisitCXXStaticCastExpr(CXXStaticCastExpr *Node) {
1920 VisitCXXNamedCastExpr(Node);
1921}
1922
1923void StmtPrinter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *Node) {
1924 VisitCXXNamedCastExpr(Node);
1925}
1926
1927void StmtPrinter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *Node) {
1928 VisitCXXNamedCastExpr(Node);
1929}
1930
1931void StmtPrinter::VisitCXXConstCastExpr(CXXConstCastExpr *Node) {
1932 VisitCXXNamedCastExpr(Node);
1933}
1934
Sebastian Redlc4704762008-11-11 11:37:55 +00001935void StmtPrinter::VisitCXXTypeidExpr(CXXTypeidExpr *Node) {
1936 OS << "typeid(";
1937 if (Node->isTypeOperand()) {
David Majnemer143c55e2013-09-27 07:04:31 +00001938 Node->getTypeOperandSourceInfo()->getType().print(OS, Policy);
Sebastian Redlc4704762008-11-11 11:37:55 +00001939 } else {
1940 PrintExpr(Node->getExprOperand());
1941 }
1942 OS << ")";
1943}
1944
Francois Pichet9f4f2072010-09-08 12:20:18 +00001945void StmtPrinter::VisitCXXUuidofExpr(CXXUuidofExpr *Node) {
1946 OS << "__uuidof(";
1947 if (Node->isTypeOperand()) {
David Majnemer143c55e2013-09-27 07:04:31 +00001948 Node->getTypeOperandSourceInfo()->getType().print(OS, Policy);
Francois Pichet9f4f2072010-09-08 12:20:18 +00001949 } else {
1950 PrintExpr(Node->getExprOperand());
1951 }
1952 OS << ")";
1953}
1954
John McCall5e77d762013-04-16 07:28:30 +00001955void StmtPrinter::VisitMSPropertyRefExpr(MSPropertyRefExpr *Node) {
1956 PrintExpr(Node->getBaseExpr());
1957 if (Node->isArrow())
1958 OS << "->";
1959 else
1960 OS << ".";
1961 if (NestedNameSpecifier *Qualifier =
1962 Node->getQualifierLoc().getNestedNameSpecifier())
1963 Qualifier->print(OS, Policy);
1964 OS << Node->getPropertyDecl()->getDeclName();
1965}
1966
Alexey Bataevf7630272015-11-25 12:01:00 +00001967void StmtPrinter::VisitMSPropertySubscriptExpr(MSPropertySubscriptExpr *Node) {
1968 PrintExpr(Node->getBase());
1969 OS << "[";
1970 PrintExpr(Node->getIdx());
1971 OS << "]";
1972}
1973
Richard Smithc67fdd42012-03-07 08:35:16 +00001974void StmtPrinter::VisitUserDefinedLiteral(UserDefinedLiteral *Node) {
1975 switch (Node->getLiteralOperatorKind()) {
1976 case UserDefinedLiteral::LOK_Raw:
Richard Smith29e95952012-03-09 10:10:02 +00001977 OS << cast<StringLiteral>(Node->getArg(0)->IgnoreImpCasts())->getString();
Richard Smithc67fdd42012-03-07 08:35:16 +00001978 break;
1979 case UserDefinedLiteral::LOK_Template: {
Richard Smith29e95952012-03-09 10:10:02 +00001980 DeclRefExpr *DRE = cast<DeclRefExpr>(Node->getCallee()->IgnoreImpCasts());
1981 const TemplateArgumentList *Args =
1982 cast<FunctionDecl>(DRE->getDecl())->getTemplateSpecializationArgs();
1983 assert(Args);
David Majnemer314b72f2015-04-05 05:32:54 +00001984
1985 if (Args->size() != 1) {
Richard Smithe87aeb32015-10-08 00:17:59 +00001986 OS << "operator\"\"" << Node->getUDSuffix()->getName();
David Majnemer314b72f2015-04-05 05:32:54 +00001987 TemplateSpecializationType::PrintTemplateArgumentList(
David Majnemer6fbeee32016-07-07 04:43:07 +00001988 OS, Args->asArray(), Policy);
David Majnemer314b72f2015-04-05 05:32:54 +00001989 OS << "()";
1990 return;
1991 }
1992
Richard Smith29e95952012-03-09 10:10:02 +00001993 const TemplateArgument &Pack = Args->get(0);
Aaron Ballman2a89e852014-07-15 21:32:31 +00001994 for (const auto &P : Pack.pack_elements()) {
1995 char C = (char)P.getAsIntegral().getZExtValue();
Richard Smithc67fdd42012-03-07 08:35:16 +00001996 OS << C;
1997 }
1998 break;
1999 }
Richard Smith75025ba2012-03-08 09:02:38 +00002000 case UserDefinedLiteral::LOK_Integer: {
2001 // Print integer literal without suffix.
2002 IntegerLiteral *Int = cast<IntegerLiteral>(Node->getCookedLiteral());
2003 OS << Int->getValue().toString(10, /*isSigned*/false);
2004 break;
2005 }
Benjamin Kramer8a526762012-09-20 14:07:17 +00002006 case UserDefinedLiteral::LOK_Floating: {
2007 // Print floating literal without suffix.
2008 FloatingLiteral *Float = cast<FloatingLiteral>(Node->getCookedLiteral());
2009 PrintFloatingLiteral(OS, Float, /*PrintSuffix=*/false);
2010 break;
2011 }
Richard Smithc67fdd42012-03-07 08:35:16 +00002012 case UserDefinedLiteral::LOK_String:
2013 case UserDefinedLiteral::LOK_Character:
2014 PrintExpr(Node->getCookedLiteral());
2015 break;
2016 }
2017 OS << Node->getUDSuffix()->getName();
2018}
2019
Chris Lattnereefa10e2007-05-28 06:56:27 +00002020void StmtPrinter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *Node) {
2021 OS << (Node->getValue() ? "true" : "false");
2022}
2023
Sebastian Redl576fd422009-05-10 18:38:11 +00002024void StmtPrinter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *Node) {
2025 OS << "nullptr";
2026}
2027
Douglas Gregor97a9c812008-11-04 14:32:21 +00002028void StmtPrinter::VisitCXXThisExpr(CXXThisExpr *Node) {
2029 OS << "this";
2030}
2031
Chris Lattnerb7e656b2008-02-26 00:51:44 +00002032void StmtPrinter::VisitCXXThrowExpr(CXXThrowExpr *Node) {
Craig Topper36250ad2014-05-12 05:36:57 +00002033 if (!Node->getSubExpr())
Chris Lattnerb7e656b2008-02-26 00:51:44 +00002034 OS << "throw";
2035 else {
2036 OS << "throw ";
2037 PrintExpr(Node->getSubExpr());
2038 }
2039}
2040
Chris Lattneraa9c7ae2008-04-08 04:40:51 +00002041void StmtPrinter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *Node) {
Richard Smith852c9db2013-04-20 22:23:05 +00002042 // Nothing to print: we picked up the default argument.
2043}
2044
2045void StmtPrinter::VisitCXXDefaultInitExpr(CXXDefaultInitExpr *Node) {
2046 // Nothing to print: we picked up the default initializer.
Chris Lattneraa9c7ae2008-04-08 04:40:51 +00002047}
2048
Argyrios Kyrtzidis857fcc22008-08-22 15:38:55 +00002049void StmtPrinter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *Node) {
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002050 Node->getType().print(OS, Policy);
Richard Smith1ae689c2015-01-28 22:06:01 +00002051 // If there are no parens, this is list-initialization, and the braces are
2052 // part of the syntax of the inner construct.
2053 if (Node->getLParenLoc().isValid())
2054 OS << "(";
Argyrios Kyrtzidis857fcc22008-08-22 15:38:55 +00002055 PrintExpr(Node->getSubExpr());
Richard Smith1ae689c2015-01-28 22:06:01 +00002056 if (Node->getLParenLoc().isValid())
2057 OS << ")";
Argyrios Kyrtzidis857fcc22008-08-22 15:38:55 +00002058}
2059
Anders Carlsson993a4b32009-05-30 20:03:25 +00002060void StmtPrinter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *Node) {
2061 PrintExpr(Node->getSubExpr());
2062}
2063
Douglas Gregordd04d332009-01-16 18:33:17 +00002064void StmtPrinter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *Node) {
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002065 Node->getType().print(OS, Policy);
Richard Smithed83ebd2015-02-05 07:02:11 +00002066 if (Node->isStdInitListInitialization())
2067 /* Nothing to do; braces are part of creating the std::initializer_list. */;
2068 else if (Node->isListInitialization())
Richard Smith1ae689c2015-01-28 22:06:01 +00002069 OS << "{";
2070 else
2071 OS << "(";
Douglas Gregordd04d332009-01-16 18:33:17 +00002072 for (CXXTemporaryObjectExpr::arg_iterator Arg = Node->arg_begin(),
Mike Stump11289f42009-09-09 15:08:12 +00002073 ArgEnd = Node->arg_end();
Douglas Gregordd04d332009-01-16 18:33:17 +00002074 Arg != ArgEnd; ++Arg) {
Benjamin Kramerf48ee442015-07-18 14:35:53 +00002075 if ((*Arg)->isDefaultArgument())
Rafael Espindola6f6f3c42013-05-24 16:11:44 +00002076 break;
Douglas Gregordd04d332009-01-16 18:33:17 +00002077 if (Arg != Node->arg_begin())
2078 OS << ", ";
2079 PrintExpr(*Arg);
2080 }
Richard Smithed83ebd2015-02-05 07:02:11 +00002081 if (Node->isStdInitListInitialization())
2082 /* See above. */;
2083 else if (Node->isListInitialization())
Richard Smith1ae689c2015-01-28 22:06:01 +00002084 OS << "}";
2085 else
2086 OS << ")";
Douglas Gregordd04d332009-01-16 18:33:17 +00002087}
2088
Douglas Gregore31e6062012-02-07 10:09:13 +00002089void StmtPrinter::VisitLambdaExpr(LambdaExpr *Node) {
2090 OS << '[';
2091 bool NeedComma = false;
2092 switch (Node->getCaptureDefault()) {
2093 case LCD_None:
2094 break;
2095
2096 case LCD_ByCopy:
2097 OS << '=';
2098 NeedComma = true;
2099 break;
2100
2101 case LCD_ByRef:
2102 OS << '&';
2103 NeedComma = true;
2104 break;
2105 }
2106 for (LambdaExpr::capture_iterator C = Node->explicit_capture_begin(),
2107 CEnd = Node->explicit_capture_end();
2108 C != CEnd;
2109 ++C) {
2110 if (NeedComma)
2111 OS << ", ";
2112 NeedComma = true;
2113
2114 switch (C->getCaptureKind()) {
2115 case LCK_This:
2116 OS << "this";
2117 break;
Faisal Validc6b5962016-03-21 09:25:37 +00002118 case LCK_StarThis:
2119 OS << "*this";
2120 break;
Douglas Gregore31e6062012-02-07 10:09:13 +00002121 case LCK_ByRef:
James Dennettdd2ffea22015-05-07 18:48:18 +00002122 if (Node->getCaptureDefault() != LCD_ByRef || Node->isInitCapture(C))
Douglas Gregore31e6062012-02-07 10:09:13 +00002123 OS << '&';
2124 OS << C->getCapturedVar()->getName();
2125 break;
2126
2127 case LCK_ByCopy:
Douglas Gregore31e6062012-02-07 10:09:13 +00002128 OS << C->getCapturedVar()->getName();
2129 break;
Alexey Bataev39c81e22014-08-28 04:28:19 +00002130 case LCK_VLAType:
2131 llvm_unreachable("VLA type in explicit captures.");
Douglas Gregore31e6062012-02-07 10:09:13 +00002132 }
Richard Smithbb13c9a2013-09-28 04:02:39 +00002133
James Dennettdd2ffea22015-05-07 18:48:18 +00002134 if (Node->isInitCapture(C))
Richard Smithbb13c9a2013-09-28 04:02:39 +00002135 PrintExpr(C->getCapturedVar()->getInit());
Douglas Gregore31e6062012-02-07 10:09:13 +00002136 }
2137 OS << ']';
2138
2139 if (Node->hasExplicitParameters()) {
2140 OS << " (";
2141 CXXMethodDecl *Method = Node->getCallOperator();
2142 NeedComma = false;
David Majnemer59f77922016-06-24 04:05:48 +00002143 for (auto P : Method->parameters()) {
Douglas Gregore31e6062012-02-07 10:09:13 +00002144 if (NeedComma) {
2145 OS << ", ";
2146 } else {
2147 NeedComma = true;
2148 }
Aaron Ballman43b68be2014-03-07 17:50:17 +00002149 std::string ParamStr = P->getNameAsString();
2150 P->getOriginalType().print(OS, Policy, ParamStr);
Douglas Gregore31e6062012-02-07 10:09:13 +00002151 }
2152 if (Method->isVariadic()) {
2153 if (NeedComma)
2154 OS << ", ";
2155 OS << "...";
2156 }
2157 OS << ')';
2158
2159 if (Node->isMutable())
2160 OS << " mutable";
2161
2162 const FunctionProtoType *Proto
2163 = Method->getType()->getAs<FunctionProtoType>();
Benjamin Kramer9170e912013-02-22 15:46:01 +00002164 Proto->printExceptionSpecification(OS, Policy);
Douglas Gregore31e6062012-02-07 10:09:13 +00002165
Bill Wendling44426052012-12-20 19:22:21 +00002166 // FIXME: Attributes
Douglas Gregore31e6062012-02-07 10:09:13 +00002167
Douglas Gregor0c46b2b2012-02-13 22:00:16 +00002168 // Print the trailing return type if it was specified in the source.
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002169 if (Node->hasExplicitResultType()) {
2170 OS << " -> ";
Alp Toker314cc812014-01-25 16:55:45 +00002171 Proto->getReturnType().print(OS, Policy);
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002172 }
Douglas Gregore31e6062012-02-07 10:09:13 +00002173 }
2174
2175 // Print the body.
2176 CompoundStmt *Body = Node->getBody();
2177 OS << ' ';
2178 PrintStmt(Body);
2179}
2180
Douglas Gregor747eb782010-07-08 06:14:04 +00002181void StmtPrinter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *Node) {
Douglas Gregor2b88c112010-09-08 00:15:04 +00002182 if (TypeSourceInfo *TSInfo = Node->getTypeSourceInfo())
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002183 TSInfo->getType().print(OS, Policy);
Douglas Gregor2b88c112010-09-08 00:15:04 +00002184 else
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002185 Node->getType().print(OS, Policy);
2186 OS << "()";
Argyrios Kyrtzidis857fcc22008-08-22 15:38:55 +00002187}
2188
Sebastian Redlbd150f42008-11-21 19:14:01 +00002189void StmtPrinter::VisitCXXNewExpr(CXXNewExpr *E) {
2190 if (E->isGlobalNew())
2191 OS << "::";
2192 OS << "new ";
2193 unsigned NumPlace = E->getNumPlacementArgs();
Eli Friedmana6fdfaa2012-10-18 20:54:37 +00002194 if (NumPlace > 0 && !isa<CXXDefaultArgExpr>(E->getPlacementArg(0))) {
Sebastian Redlbd150f42008-11-21 19:14:01 +00002195 OS << "(";
2196 PrintExpr(E->getPlacementArg(0));
2197 for (unsigned i = 1; i < NumPlace; ++i) {
Eli Friedmana6fdfaa2012-10-18 20:54:37 +00002198 if (isa<CXXDefaultArgExpr>(E->getPlacementArg(i)))
2199 break;
Sebastian Redlbd150f42008-11-21 19:14:01 +00002200 OS << ", ";
2201 PrintExpr(E->getPlacementArg(i));
2202 }
2203 OS << ") ";
2204 }
2205 if (E->isParenTypeId())
2206 OS << "(";
Sebastian Redld6d55ee2008-12-02 22:08:59 +00002207 std::string TypeS;
2208 if (Expr *Size = E->getArraySize()) {
2209 llvm::raw_string_ostream s(TypeS);
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002210 s << '[';
Richard Smith235341b2012-08-16 03:56:14 +00002211 Size->printPretty(s, Helper, Policy);
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002212 s << ']';
Sebastian Redld6d55ee2008-12-02 22:08:59 +00002213 }
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002214 E->getAllocatedType().print(OS, Policy, TypeS);
Sebastian Redlbd150f42008-11-21 19:14:01 +00002215 if (E->isParenTypeId())
2216 OS << ")";
2217
Sebastian Redl6047f072012-02-16 12:22:20 +00002218 CXXNewExpr::InitializationStyle InitStyle = E->getInitializationStyle();
2219 if (InitStyle) {
2220 if (InitStyle == CXXNewExpr::CallInit)
2221 OS << "(";
2222 PrintExpr(E->getInitializer());
2223 if (InitStyle == CXXNewExpr::CallInit)
2224 OS << ")";
Sebastian Redlbd150f42008-11-21 19:14:01 +00002225 }
2226}
2227
2228void StmtPrinter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
2229 if (E->isGlobalDelete())
2230 OS << "::";
2231 OS << "delete ";
2232 if (E->isArrayForm())
2233 OS << "[] ";
2234 PrintExpr(E->getArgument());
2235}
2236
Douglas Gregorad8a3362009-09-04 17:36:40 +00002237void StmtPrinter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
2238 PrintExpr(E->getBase());
2239 if (E->isArrow())
2240 OS << "->";
2241 else
2242 OS << '.';
2243 if (E->getQualifier())
2244 E->getQualifier()->print(OS, Policy);
Eli Friedman9cc8ac52012-10-23 20:26:57 +00002245 OS << "~";
Mike Stump11289f42009-09-09 15:08:12 +00002246
Douglas Gregor678f90d2010-02-25 01:56:36 +00002247 if (IdentifierInfo *II = E->getDestroyedTypeIdentifier())
2248 OS << II->getName();
2249 else
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002250 E->getDestroyedType().print(OS, Policy);
Douglas Gregorad8a3362009-09-04 17:36:40 +00002251}
2252
Anders Carlsson0781ce72009-04-23 02:32:43 +00002253void StmtPrinter::VisitCXXConstructExpr(CXXConstructExpr *E) {
Richard Smithed83ebd2015-02-05 07:02:11 +00002254 if (E->isListInitialization() && !E->isStdInitListInitialization())
Richard Smithd0e102f2015-01-30 02:04:26 +00002255 OS << "{";
Richard Smithd59b8322012-12-19 01:39:02 +00002256
Douglas Gregorbaba85d2011-01-24 17:25:03 +00002257 for (unsigned i = 0, e = E->getNumArgs(); i != e; ++i) {
2258 if (isa<CXXDefaultArgExpr>(E->getArg(i))) {
2259 // Don't print any defaulted arguments
2260 break;
2261 }
2262
2263 if (i) OS << ", ";
2264 PrintExpr(E->getArg(i));
Fariborz Jahaniand0bbf662010-01-13 21:41:11 +00002265 }
Richard Smithd59b8322012-12-19 01:39:02 +00002266
Richard Smithed83ebd2015-02-05 07:02:11 +00002267 if (E->isListInitialization() && !E->isStdInitListInitialization())
Richard Smithd0e102f2015-01-30 02:04:26 +00002268 OS << "}";
Anders Carlsson0781ce72009-04-23 02:32:43 +00002269}
2270
Richard Smith5179eb72016-06-28 19:03:57 +00002271void StmtPrinter::VisitCXXInheritedCtorInitExpr(CXXInheritedCtorInitExpr *E) {
2272 // Parens are printed by the surrounding context.
2273 OS << "<forwarded>";
2274}
2275
Richard Smithcc1b96d2013-06-12 22:31:48 +00002276void StmtPrinter::VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E) {
2277 PrintExpr(E->getSubExpr());
2278}
2279
John McCall5d413782010-12-06 08:20:24 +00002280void StmtPrinter::VisitExprWithCleanups(ExprWithCleanups *E) {
Alp Toker028ed912013-12-06 17:56:43 +00002281 // Just forward to the subexpression.
Anders Carlssondefc6442009-04-24 22:47:04 +00002282 PrintExpr(E->getSubExpr());
2283}
2284
Mike Stump11289f42009-09-09 15:08:12 +00002285void
Douglas Gregorce934142009-05-20 18:46:25 +00002286StmtPrinter::VisitCXXUnresolvedConstructExpr(
2287 CXXUnresolvedConstructExpr *Node) {
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002288 Node->getTypeAsWritten().print(OS, Policy);
Douglas Gregorce934142009-05-20 18:46:25 +00002289 OS << "(";
2290 for (CXXUnresolvedConstructExpr::arg_iterator Arg = Node->arg_begin(),
Mike Stump11289f42009-09-09 15:08:12 +00002291 ArgEnd = Node->arg_end();
Douglas Gregorce934142009-05-20 18:46:25 +00002292 Arg != ArgEnd; ++Arg) {
2293 if (Arg != Node->arg_begin())
2294 OS << ", ";
2295 PrintExpr(*Arg);
2296 }
2297 OS << ")";
2298}
2299
John McCall8cd78132009-11-19 22:55:06 +00002300void StmtPrinter::VisitCXXDependentScopeMemberExpr(
2301 CXXDependentScopeMemberExpr *Node) {
John McCall2d74de92009-12-01 22:10:20 +00002302 if (!Node->isImplicitAccess()) {
2303 PrintExpr(Node->getBase());
2304 OS << (Node->isArrow() ? "->" : ".");
2305 }
Douglas Gregorc26e0f62009-09-03 16:14:30 +00002306 if (NestedNameSpecifier *Qualifier = Node->getQualifier())
2307 Qualifier->print(OS, Policy);
Abramo Bagnara7945c982012-01-27 09:46:47 +00002308 if (Node->hasTemplateKeyword())
Douglas Gregor308047d2009-09-09 00:23:06 +00002309 OS << "template ";
Abramo Bagnarad6d2f182010-08-11 22:01:17 +00002310 OS << Node->getMemberNameInfo();
Benjamin Kramer9170e912013-02-22 15:46:01 +00002311 if (Node->hasExplicitTemplateArgs())
2312 TemplateSpecializationType::PrintTemplateArgumentList(
David Majnemer6fbeee32016-07-07 04:43:07 +00002313 OS, Node->template_arguments(), Policy);
Douglas Gregora8db9542009-05-22 21:13:27 +00002314}
2315
John McCall10eae182009-11-30 22:42:35 +00002316void StmtPrinter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *Node) {
John McCall2d74de92009-12-01 22:10:20 +00002317 if (!Node->isImplicitAccess()) {
2318 PrintExpr(Node->getBase());
2319 OS << (Node->isArrow() ? "->" : ".");
2320 }
John McCall10eae182009-11-30 22:42:35 +00002321 if (NestedNameSpecifier *Qualifier = Node->getQualifier())
2322 Qualifier->print(OS, Policy);
Abramo Bagnara7945c982012-01-27 09:46:47 +00002323 if (Node->hasTemplateKeyword())
2324 OS << "template ";
Abramo Bagnarad6d2f182010-08-11 22:01:17 +00002325 OS << Node->getMemberNameInfo();
Benjamin Kramer9170e912013-02-22 15:46:01 +00002326 if (Node->hasExplicitTemplateArgs())
2327 TemplateSpecializationType::PrintTemplateArgumentList(
David Majnemer6fbeee32016-07-07 04:43:07 +00002328 OS, Node->template_arguments(), Policy);
John McCall10eae182009-11-30 22:42:35 +00002329}
2330
Douglas Gregor29c42f22012-02-24 07:38:34 +00002331static const char *getTypeTraitName(TypeTrait TT) {
2332 switch (TT) {
Alp Toker95e7ff22014-01-01 05:57:51 +00002333#define TYPE_TRAIT_1(Spelling, Name, Key) \
2334case clang::UTT_##Name: return #Spelling;
Alp Tokercbb90342013-12-13 20:49:58 +00002335#define TYPE_TRAIT_2(Spelling, Name, Key) \
2336case clang::BTT_##Name: return #Spelling;
Alp Toker40f9b1c2013-12-12 21:23:03 +00002337#define TYPE_TRAIT_N(Spelling, Name, Key) \
2338 case clang::TT_##Name: return #Spelling;
2339#include "clang/Basic/TokenKinds.def"
Douglas Gregor29c42f22012-02-24 07:38:34 +00002340 }
2341 llvm_unreachable("Type trait not covered by switch");
2342}
2343
John Wiegley6242b6a2011-04-28 00:16:57 +00002344static const char *getTypeTraitName(ArrayTypeTrait ATT) {
2345 switch (ATT) {
2346 case ATT_ArrayRank: return "__array_rank";
2347 case ATT_ArrayExtent: return "__array_extent";
2348 }
Chandler Carruthe46eaf32011-05-01 07:23:23 +00002349 llvm_unreachable("Array type trait not covered by switch");
John Wiegley6242b6a2011-04-28 00:16:57 +00002350}
2351
John Wiegleyf9f65842011-04-25 06:54:41 +00002352static const char *getExpressionTraitName(ExpressionTrait ET) {
2353 switch (ET) {
John Wiegleyf9f65842011-04-25 06:54:41 +00002354 case ET_IsLValueExpr: return "__is_lvalue_expr";
2355 case ET_IsRValueExpr: return "__is_rvalue_expr";
2356 }
Chandler Carruthe46eaf32011-05-01 07:23:23 +00002357 llvm_unreachable("Expression type trait not covered by switch");
John Wiegleyf9f65842011-04-25 06:54:41 +00002358}
2359
Douglas Gregor29c42f22012-02-24 07:38:34 +00002360void StmtPrinter::VisitTypeTraitExpr(TypeTraitExpr *E) {
2361 OS << getTypeTraitName(E->getTrait()) << "(";
2362 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I) {
2363 if (I > 0)
2364 OS << ", ";
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002365 E->getArg(I)->getType().print(OS, Policy);
Douglas Gregor29c42f22012-02-24 07:38:34 +00002366 }
2367 OS << ")";
2368}
2369
John Wiegley6242b6a2011-04-28 00:16:57 +00002370void StmtPrinter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002371 OS << getTypeTraitName(E->getTrait()) << '(';
2372 E->getQueriedType().print(OS, Policy);
2373 OS << ')';
John Wiegley6242b6a2011-04-28 00:16:57 +00002374}
2375
John Wiegleyf9f65842011-04-25 06:54:41 +00002376void StmtPrinter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002377 OS << getExpressionTraitName(E->getTrait()) << '(';
2378 PrintExpr(E->getQueriedExpression());
2379 OS << ')';
John Wiegleyf9f65842011-04-25 06:54:41 +00002380}
2381
Sebastian Redl4202c0f2010-09-10 20:55:43 +00002382void StmtPrinter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
2383 OS << "noexcept(";
2384 PrintExpr(E->getOperand());
2385 OS << ")";
2386}
2387
Douglas Gregor820ba7b2011-01-04 17:33:58 +00002388void StmtPrinter::VisitPackExpansionExpr(PackExpansionExpr *E) {
Douglas Gregore8e9dd62011-01-03 17:17:50 +00002389 PrintExpr(E->getPattern());
2390 OS << "...";
2391}
2392
Douglas Gregor820ba7b2011-01-04 17:33:58 +00002393void StmtPrinter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
Benjamin Kramerdb0fc512012-02-07 11:57:57 +00002394 OS << "sizeof...(" << *E->getPack() << ")";
Douglas Gregor820ba7b2011-01-04 17:33:58 +00002395}
2396
Douglas Gregorcdbc5392011-01-15 01:15:58 +00002397void StmtPrinter::VisitSubstNonTypeTemplateParmPackExpr(
2398 SubstNonTypeTemplateParmPackExpr *Node) {
Benjamin Kramerdb0fc512012-02-07 11:57:57 +00002399 OS << *Node->getParameterPack();
Douglas Gregorcdbc5392011-01-15 01:15:58 +00002400}
2401
John McCall7c454bb2011-07-15 05:09:51 +00002402void StmtPrinter::VisitSubstNonTypeTemplateParmExpr(
2403 SubstNonTypeTemplateParmExpr *Node) {
2404 Visit(Node->getReplacement());
2405}
2406
Richard Smithb15fe3a2012-09-12 00:56:43 +00002407void StmtPrinter::VisitFunctionParmPackExpr(FunctionParmPackExpr *E) {
2408 OS << *E->getParameterPack();
2409}
2410
Douglas Gregorfe314812011-06-21 17:03:29 +00002411void StmtPrinter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *Node){
2412 PrintExpr(Node->GetTemporaryExpr());
2413}
2414
Richard Smith0f0af192014-11-08 05:07:16 +00002415void StmtPrinter::VisitCXXFoldExpr(CXXFoldExpr *E) {
2416 OS << "(";
2417 if (E->getLHS()) {
2418 PrintExpr(E->getLHS());
2419 OS << " " << BinaryOperator::getOpcodeStr(E->getOperator()) << " ";
2420 }
2421 OS << "...";
2422 if (E->getRHS()) {
2423 OS << " " << BinaryOperator::getOpcodeStr(E->getOperator()) << " ";
2424 PrintExpr(E->getRHS());
2425 }
2426 OS << ")";
2427}
2428
Richard Smith9f690bd2015-10-27 06:02:45 +00002429// C++ Coroutines TS
2430
2431void StmtPrinter::VisitCoroutineBodyStmt(CoroutineBodyStmt *S) {
2432 Visit(S->getBody());
2433}
2434
2435void StmtPrinter::VisitCoreturnStmt(CoreturnStmt *S) {
2436 OS << "co_return";
2437 if (S->getOperand()) {
2438 OS << " ";
2439 Visit(S->getOperand());
2440 }
2441 OS << ";";
2442}
2443
2444void StmtPrinter::VisitCoawaitExpr(CoawaitExpr *S) {
2445 OS << "co_await ";
2446 PrintExpr(S->getOperand());
2447}
2448
2449void StmtPrinter::VisitCoyieldExpr(CoyieldExpr *S) {
2450 OS << "co_yield ";
2451 PrintExpr(S->getOperand());
2452}
2453
Mike Stump11289f42009-09-09 15:08:12 +00002454// Obj-C
Anders Carlsson76f4a902007-08-21 17:43:55 +00002455
2456void StmtPrinter::VisitObjCStringLiteral(ObjCStringLiteral *Node) {
2457 OS << "@";
2458 VisitStringLiteral(Node->getString());
2459}
Chris Lattnereefa10e2007-05-28 06:56:27 +00002460
Patrick Beard0caa3942012-04-19 00:25:12 +00002461void StmtPrinter::VisitObjCBoxedExpr(ObjCBoxedExpr *E) {
Ted Kremeneke65b0862012-03-06 20:05:56 +00002462 OS << "@";
Patrick Beard0caa3942012-04-19 00:25:12 +00002463 Visit(E->getSubExpr());
Ted Kremeneke65b0862012-03-06 20:05:56 +00002464}
2465
2466void StmtPrinter::VisitObjCArrayLiteral(ObjCArrayLiteral *E) {
2467 OS << "@[ ";
Benjamin Kramer5733e352015-07-18 17:09:36 +00002468 ObjCArrayLiteral::child_range Ch = E->children();
2469 for (auto I = Ch.begin(), E = Ch.end(); I != E; ++I) {
2470 if (I != Ch.begin())
Ted Kremeneke65b0862012-03-06 20:05:56 +00002471 OS << ", ";
Benjamin Kramer5733e352015-07-18 17:09:36 +00002472 Visit(*I);
Ted Kremeneke65b0862012-03-06 20:05:56 +00002473 }
2474 OS << " ]";
2475}
2476
2477void StmtPrinter::VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) {
2478 OS << "@{ ";
2479 for (unsigned I = 0, N = E->getNumElements(); I != N; ++I) {
2480 if (I > 0)
2481 OS << ", ";
2482
2483 ObjCDictionaryElement Element = E->getKeyValueElement(I);
2484 Visit(Element.Key);
2485 OS << " : ";
2486 Visit(Element.Value);
2487 if (Element.isPackExpansion())
2488 OS << "...";
2489 }
2490 OS << " }";
2491}
2492
Anders Carlssonc5a81eb2007-08-22 15:14:15 +00002493void StmtPrinter::VisitObjCEncodeExpr(ObjCEncodeExpr *Node) {
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002494 OS << "@encode(";
2495 Node->getEncodedType().print(OS, Policy);
2496 OS << ')';
Anders Carlssonc5a81eb2007-08-22 15:14:15 +00002497}
2498
Fariborz Jahanian4bef4622007-10-16 20:40:23 +00002499void StmtPrinter::VisitObjCSelectorExpr(ObjCSelectorExpr *Node) {
Aaron Ballmanb190f972014-01-03 17:59:55 +00002500 OS << "@selector(";
2501 Node->getSelector().print(OS);
2502 OS << ')';
Fariborz Jahanian4bef4622007-10-16 20:40:23 +00002503}
2504
Fariborz Jahaniana32aaef2007-10-17 16:58:11 +00002505void StmtPrinter::VisitObjCProtocolExpr(ObjCProtocolExpr *Node) {
Benjamin Kramerb89514a2011-10-14 18:45:37 +00002506 OS << "@protocol(" << *Node->getProtocol() << ')';
Fariborz Jahaniana32aaef2007-10-17 16:58:11 +00002507}
2508
Steve Naroffd54978b2007-09-18 23:55:05 +00002509void StmtPrinter::VisitObjCMessageExpr(ObjCMessageExpr *Mess) {
2510 OS << "[";
Douglas Gregor9a129192010-04-21 00:45:42 +00002511 switch (Mess->getReceiverKind()) {
2512 case ObjCMessageExpr::Instance:
2513 PrintExpr(Mess->getInstanceReceiver());
2514 break;
2515
2516 case ObjCMessageExpr::Class:
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002517 Mess->getClassReceiver().print(OS, Policy);
Douglas Gregor9a129192010-04-21 00:45:42 +00002518 break;
2519
2520 case ObjCMessageExpr::SuperInstance:
2521 case ObjCMessageExpr::SuperClass:
2522 OS << "Super";
2523 break;
2524 }
2525
Ted Kremeneka06e7122008-05-02 17:32:38 +00002526 OS << ' ';
Ted Kremenekb65a67d2008-04-16 04:30:16 +00002527 Selector selector = Mess->getSelector();
Steve Naroffc6814ea2007-10-02 20:01:56 +00002528 if (selector.isUnarySelector()) {
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002529 OS << selector.getNameForSlot(0);
Steve Naroffc6814ea2007-10-02 20:01:56 +00002530 } else {
2531 for (unsigned i = 0, e = Mess->getNumArgs(); i != e; ++i) {
Ted Kremeneka06e7122008-05-02 17:32:38 +00002532 if (i < selector.getNumArgs()) {
2533 if (i > 0) OS << ' ';
2534 if (selector.getIdentifierInfoForSlot(i))
Chris Lattner1cbaacc2008-11-24 04:00:27 +00002535 OS << selector.getIdentifierInfoForSlot(i)->getName() << ':';
Ted Kremeneka06e7122008-05-02 17:32:38 +00002536 else
2537 OS << ":";
2538 }
2539 else OS << ", "; // Handle variadic methods.
Mike Stump11289f42009-09-09 15:08:12 +00002540
Steve Naroffc6814ea2007-10-02 20:01:56 +00002541 PrintExpr(Mess->getArg(i));
2542 }
Steve Naroffd54978b2007-09-18 23:55:05 +00002543 }
2544 OS << "]";
2545}
2546
Ted Kremeneke65b0862012-03-06 20:05:56 +00002547void StmtPrinter::VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr *Node) {
2548 OS << (Node->getValue() ? "__objc_yes" : "__objc_no");
2549}
2550
John McCall31168b02011-06-15 23:02:42 +00002551void
2552StmtPrinter::VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) {
2553 PrintExpr(E->getSubExpr());
2554}
2555
2556void
2557StmtPrinter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) {
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002558 OS << '(' << E->getBridgeKindName();
2559 E->getType().print(OS, Policy);
2560 OS << ')';
John McCall31168b02011-06-15 23:02:42 +00002561 PrintExpr(E->getSubExpr());
2562}
Douglas Gregor8ea1f532008-11-04 14:56:14 +00002563
Steve Naroffc540d662008-09-03 18:15:37 +00002564void StmtPrinter::VisitBlockExpr(BlockExpr *Node) {
Steve Naroff415d3d52008-10-08 17:01:13 +00002565 BlockDecl *BD = Node->getBlockDecl();
Steve Naroffc540d662008-09-03 18:15:37 +00002566 OS << "^";
Mike Stump11289f42009-09-09 15:08:12 +00002567
Steve Naroffc540d662008-09-03 18:15:37 +00002568 const FunctionType *AFT = Node->getFunctionType();
Mike Stump11289f42009-09-09 15:08:12 +00002569
Douglas Gregordeaad8c2009-02-26 23:50:07 +00002570 if (isa<FunctionNoProtoType>(AFT)) {
Steve Naroffc540d662008-09-03 18:15:37 +00002571 OS << "()";
Douglas Gregordeaad8c2009-02-26 23:50:07 +00002572 } else if (!BD->param_empty() || cast<FunctionProtoType>(AFT)->isVariadic()) {
Steve Naroffc540d662008-09-03 18:15:37 +00002573 OS << '(';
Steve Naroff415d3d52008-10-08 17:01:13 +00002574 for (BlockDecl::param_iterator AI = BD->param_begin(),
2575 E = BD->param_end(); AI != E; ++AI) {
2576 if (AI != BD->param_begin()) OS << ", ";
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002577 std::string ParamStr = (*AI)->getNameAsString();
2578 (*AI)->getType().print(OS, Policy, ParamStr);
Steve Naroffc540d662008-09-03 18:15:37 +00002579 }
Mike Stump11289f42009-09-09 15:08:12 +00002580
Douglas Gregordeaad8c2009-02-26 23:50:07 +00002581 const FunctionProtoType *FT = cast<FunctionProtoType>(AFT);
Steve Naroffc540d662008-09-03 18:15:37 +00002582 if (FT->isVariadic()) {
Steve Naroff415d3d52008-10-08 17:01:13 +00002583 if (!BD->param_empty()) OS << ", ";
Steve Naroffc540d662008-09-03 18:15:37 +00002584 OS << "...";
2585 }
2586 OS << ')';
2587 }
Fariborz Jahanian1ace8cb2012-12-04 21:15:23 +00002588 OS << "{ }";
Steve Naroffc540d662008-09-03 18:15:37 +00002589}
2590
Ted Kremenekf551f322011-11-30 22:08:08 +00002591void StmtPrinter::VisitOpaqueValueExpr(OpaqueValueExpr *Node) {
2592 PrintExpr(Node->getSourceExpr());
2593}
John McCall8d69a212010-11-15 23:31:06 +00002594
Kaelyn Takatae1f49d52014-10-27 18:07:20 +00002595void StmtPrinter::VisitTypoExpr(TypoExpr *Node) {
2596 // TODO: Print something reasonable for a TypoExpr, if necessary.
Davide Italiano04839a52016-01-30 08:03:54 +00002597 llvm_unreachable("Cannot print TypoExpr nodes");
Kaelyn Takatae1f49d52014-10-27 18:07:20 +00002598}
2599
Tanya Lattner55808c12011-06-04 00:47:47 +00002600void StmtPrinter::VisitAsTypeExpr(AsTypeExpr *Node) {
2601 OS << "__builtin_astype(";
2602 PrintExpr(Node->getSrcExpr());
Benjamin Kramerc5720e92013-02-22 14:19:01 +00002603 OS << ", ";
2604 Node->getType().print(OS, Policy);
Tanya Lattner55808c12011-06-04 00:47:47 +00002605 OS << ")";
2606}
2607
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00002608//===----------------------------------------------------------------------===//
2609// Stmt method implementations
2610//===----------------------------------------------------------------------===//
2611
Craig Topperc571c812013-08-22 06:02:26 +00002612void Stmt::dumpPretty(const ASTContext &Context) const {
Craig Topper36250ad2014-05-12 05:36:57 +00002613 printPretty(llvm::errs(), nullptr, PrintingPolicy(Context.getLangOpts()));
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00002614}
2615
Richard Smith235341b2012-08-16 03:56:14 +00002616void Stmt::printPretty(raw_ostream &OS,
2617 PrinterHelper *Helper,
Douglas Gregor7de59662009-05-29 20:38:28 +00002618 const PrintingPolicy &Policy,
2619 unsigned Indentation) const {
Richard Smith235341b2012-08-16 03:56:14 +00002620 StmtPrinter P(OS, Helper, Policy, Indentation);
Chris Lattner62249a62007-08-21 04:04:25 +00002621 P.Visit(const_cast<Stmt*>(this));
Chris Lattnera3bcb7a2006-11-04 07:16:25 +00002622}
Ted Kremenek04f3cee2007-08-31 21:30:12 +00002623
2624//===----------------------------------------------------------------------===//
2625// PrinterHelper
2626//===----------------------------------------------------------------------===//
2627
2628// Implement virtual destructor.
Angel Garcia Gomez637d1e62015-10-20 13:23:58 +00002629PrinterHelper::~PrinterHelper() {}