blob: 4262dc39ea59418d76dd93d8877aff18a8d73ddd [file] [log] [blame]
Chris Lattnerf42cce72006-10-25 04:09:21 +00001//===--- Stmt.cpp - Statement AST Node Implementation ---------------------===//
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 Lattnerf42cce72006-10-25 04:09:21 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the Stmt class and statement subclasses.
11//
12//===----------------------------------------------------------------------===//
13
Benjamin Kramer444a1302012-12-01 17:12:56 +000014#include "clang/AST/ASTContext.h"
15#include "clang/AST/ASTDiagnostic.h"
Chris Lattner29375652006-12-04 18:06:35 +000016#include "clang/AST/ExprCXX.h"
Steve Naroff021ca182008-05-29 21:12:08 +000017#include "clang/AST/ExprObjC.h"
Benjamin Kramer444a1302012-12-01 17:12:56 +000018#include "clang/AST/Stmt.h"
Chris Lattnerf0b64d72009-04-26 01:32:48 +000019#include "clang/AST/StmtCXX.h"
20#include "clang/AST/StmtObjC.h"
Alexey Bataev5ec3eb12013-07-19 03:13:43 +000021#include "clang/AST/StmtOpenMP.h"
Sebastian Redl54c04d42008-12-22 19:15:10 +000022#include "clang/AST/Type.h"
Jordan Rosea7d03842013-02-08 22:30:41 +000023#include "clang/Basic/CharInfo.h"
Chris Lattner1a8f3942010-04-23 16:29:58 +000024#include "clang/Basic/TargetInfo.h"
Benjamin Kramer444a1302012-12-01 17:12:56 +000025#include "clang/Lex/Token.h"
Chad Rosier14836ba2012-08-24 17:05:45 +000026#include "llvm/ADT/StringExtras.h"
Chandler Carruthbfb154a2011-07-04 06:13:27 +000027#include "llvm/Support/raw_ostream.h"
Chris Lattnerf42cce72006-10-25 04:09:21 +000028using namespace clang;
29
Steve Narofff84d11f2007-05-23 21:48:04 +000030static struct StmtClassNameTable {
Chris Lattner4d15a0d2007-08-25 01:42:24 +000031 const char *Name;
32 unsigned Counter;
33 unsigned Size;
Alexis Hunt656bb312010-05-05 15:24:00 +000034} StmtClassInfo[Stmt::lastStmtConstant+1];
Chris Lattner4d15a0d2007-08-25 01:42:24 +000035
36static StmtClassNameTable &getStmtInfoTableEntry(Stmt::StmtClass E) {
37 static bool Initialized = false;
38 if (Initialized)
39 return StmtClassInfo[E];
40
41 // Intialize the table on the first use.
42 Initialized = true;
Alexis Huntabb2ac82010-05-18 06:22:21 +000043#define ABSTRACT_STMT(STMT)
Douglas Gregorbe35ce92008-11-14 12:46:07 +000044#define STMT(CLASS, PARENT) \
45 StmtClassInfo[(unsigned)Stmt::CLASS##Class].Name = #CLASS; \
46 StmtClassInfo[(unsigned)Stmt::CLASS##Class].Size = sizeof(CLASS);
Alexis Hunt656bb312010-05-05 15:24:00 +000047#include "clang/AST/StmtNodes.inc"
Nico Weberde565e32008-08-05 23:15:29 +000048
Chris Lattner4d15a0d2007-08-25 01:42:24 +000049 return StmtClassInfo[E];
50}
51
Craig Topper37932912013-08-18 10:09:15 +000052void *Stmt::operator new(size_t bytes, const ASTContext& C,
Craig Topper5a050012013-08-18 17:45:38 +000053 unsigned alignment) {
Benjamin Kramerea70eb32012-12-01 15:09:41 +000054 return ::operator new(bytes, C, alignment);
55}
56
Steve Narofff1e53692007-03-23 22:27:02 +000057const char *Stmt::getStmtClassName() const {
John McCall925b16622010-10-26 08:39:16 +000058 return getStmtInfoTableEntry((StmtClass) StmtBits.sClass).Name;
Steve Narofff1e53692007-03-23 22:27:02 +000059}
Steve Narofff84d11f2007-05-23 21:48:04 +000060
61void Stmt::PrintStats() {
Chris Lattner4d15a0d2007-08-25 01:42:24 +000062 // Ensure the table is primed.
63 getStmtInfoTableEntry(Stmt::NullStmtClass);
Nico Weberde565e32008-08-05 23:15:29 +000064
Steve Narofff84d11f2007-05-23 21:48:04 +000065 unsigned sum = 0;
Chandler Carruthbfb154a2011-07-04 06:13:27 +000066 llvm::errs() << "\n*** Stmt/Expr Stats:\n";
Alexis Hunt656bb312010-05-05 15:24:00 +000067 for (int i = 0; i != Stmt::lastStmtConstant+1; i++) {
Craig Topper36250ad2014-05-12 05:36:57 +000068 if (StmtClassInfo[i].Name == nullptr) continue;
Chris Lattner4d15a0d2007-08-25 01:42:24 +000069 sum += StmtClassInfo[i].Counter;
Steve Narofff84d11f2007-05-23 21:48:04 +000070 }
Chandler Carruthbfb154a2011-07-04 06:13:27 +000071 llvm::errs() << " " << sum << " stmts/exprs total.\n";
Steve Narofff84d11f2007-05-23 21:48:04 +000072 sum = 0;
Alexis Hunt656bb312010-05-05 15:24:00 +000073 for (int i = 0; i != Stmt::lastStmtConstant+1; i++) {
Craig Topper36250ad2014-05-12 05:36:57 +000074 if (StmtClassInfo[i].Name == nullptr) continue;
Douglas Gregora30d0462009-05-26 14:40:08 +000075 if (StmtClassInfo[i].Counter == 0) continue;
Chandler Carruthbfb154a2011-07-04 06:13:27 +000076 llvm::errs() << " " << StmtClassInfo[i].Counter << " "
77 << StmtClassInfo[i].Name << ", " << StmtClassInfo[i].Size
78 << " each (" << StmtClassInfo[i].Counter*StmtClassInfo[i].Size
79 << " bytes)\n";
Chris Lattner4d15a0d2007-08-25 01:42:24 +000080 sum += StmtClassInfo[i].Counter*StmtClassInfo[i].Size;
Steve Narofff84d11f2007-05-23 21:48:04 +000081 }
Chandler Carruthbfb154a2011-07-04 06:13:27 +000082
83 llvm::errs() << "Total bytes = " << sum << "\n";
Steve Narofff84d11f2007-05-23 21:48:04 +000084}
85
86void Stmt::addStmtClass(StmtClass s) {
Chris Lattner4d15a0d2007-08-25 01:42:24 +000087 ++getStmtInfoTableEntry(s).Counter;
Steve Narofff84d11f2007-05-23 21:48:04 +000088}
89
Daniel Dunbar62905572012-03-05 21:42:49 +000090bool Stmt::StatisticsEnabled = false;
91void Stmt::EnableStatistics() {
92 StatisticsEnabled = true;
Steve Narofff84d11f2007-05-23 21:48:04 +000093}
94
John McCall4db5c3c2011-07-07 06:58:02 +000095Stmt *Stmt::IgnoreImplicit() {
96 Stmt *s = this;
97
Richard Smith520449d2015-02-05 06:15:50 +000098 if (auto *ewc = dyn_cast<ExprWithCleanups>(s))
John McCall4db5c3c2011-07-07 06:58:02 +000099 s = ewc->getSubExpr();
100
Richard Smith520449d2015-02-05 06:15:50 +0000101 if (auto *mte = dyn_cast<MaterializeTemporaryExpr>(s))
102 s = mte->GetTemporaryExpr();
103
104 if (auto *bte = dyn_cast<CXXBindTemporaryExpr>(s))
105 s = bte->getSubExpr();
106
107 while (auto *ice = dyn_cast<ImplicitCastExpr>(s))
John McCall4db5c3c2011-07-07 06:58:02 +0000108 s = ice->getSubExpr();
109
110 return s;
111}
112
Alexander Musmana5f070a2014-10-01 06:03:56 +0000113/// \brief Skip no-op (attributed, compound) container stmts and skip captured
114/// stmt at the top, if \a IgnoreCaptured is true.
115Stmt *Stmt::IgnoreContainers(bool IgnoreCaptured) {
116 Stmt *S = this;
117 if (IgnoreCaptured)
118 if (auto CapS = dyn_cast_or_null<CapturedStmt>(S))
119 S = CapS->getCapturedStmt();
120 while (true) {
121 if (auto AS = dyn_cast_or_null<AttributedStmt>(S))
122 S = AS->getSubStmt();
123 else if (auto CS = dyn_cast_or_null<CompoundStmt>(S)) {
124 if (CS->size() != 1)
125 break;
126 S = CS->body_back();
127 } else
128 break;
129 }
130 return S;
131}
132
Chandler Carrutha626d642011-09-10 00:02:34 +0000133/// \brief Strip off all label-like statements.
134///
Richard Smithc202b282012-04-14 00:33:13 +0000135/// This will strip off label statements, case statements, attributed
136/// statements and default statements recursively.
Chandler Carrutha626d642011-09-10 00:02:34 +0000137const Stmt *Stmt::stripLabelLikeStatements() const {
138 const Stmt *S = this;
139 while (true) {
140 if (const LabelStmt *LS = dyn_cast<LabelStmt>(S))
141 S = LS->getSubStmt();
142 else if (const SwitchCase *SC = dyn_cast<SwitchCase>(S))
143 S = SC->getSubStmt();
Richard Smithc202b282012-04-14 00:33:13 +0000144 else if (const AttributedStmt *AS = dyn_cast<AttributedStmt>(S))
145 S = AS->getSubStmt();
Chandler Carrutha626d642011-09-10 00:02:34 +0000146 else
147 return S;
148 }
149}
150
John McCallbd066782011-02-09 08:16:59 +0000151namespace {
152 struct good {};
153 struct bad {};
John McCallf75152f2011-02-09 08:31:17 +0000154
155 // These silly little functions have to be static inline to suppress
156 // unused warnings, and they have to be defined to suppress other
157 // warnings.
Nick Lewyckybae992f2011-02-09 08:42:57 +0000158 static inline good is_good(good) { return good(); }
John McCallbd066782011-02-09 08:16:59 +0000159
160 typedef Stmt::child_range children_t();
Nick Lewyckybae992f2011-02-09 08:42:57 +0000161 template <class T> good implements_children(children_t T::*) {
162 return good();
163 }
Eli Friedmandc41d792013-09-10 22:57:15 +0000164 LLVM_ATTRIBUTE_UNUSED
Nick Lewyckybae992f2011-02-09 08:42:57 +0000165 static inline bad implements_children(children_t Stmt::*) {
166 return bad();
167 }
John McCallbd066782011-02-09 08:16:59 +0000168
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000169 typedef SourceLocation getLocStart_t() const;
170 template <class T> good implements_getLocStart(getLocStart_t T::*) {
Nick Lewyckybae992f2011-02-09 08:42:57 +0000171 return good();
172 }
Eli Friedmandc41d792013-09-10 22:57:15 +0000173 LLVM_ATTRIBUTE_UNUSED
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000174 static inline bad implements_getLocStart(getLocStart_t Stmt::*) {
175 return bad();
176 }
177
178 typedef SourceLocation getLocEnd_t() const;
179 template <class T> good implements_getLocEnd(getLocEnd_t T::*) {
180 return good();
181 }
Eli Friedmandc41d792013-09-10 22:57:15 +0000182 LLVM_ATTRIBUTE_UNUSED
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000183 static inline bad implements_getLocEnd(getLocEnd_t Stmt::*) {
Nick Lewyckybae992f2011-02-09 08:42:57 +0000184 return bad();
185 }
John McCallbd066782011-02-09 08:16:59 +0000186
187#define ASSERT_IMPLEMENTS_children(type) \
Eli Friedmandc41d792013-09-10 22:57:15 +0000188 (void) is_good(implements_children(&type::children))
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000189#define ASSERT_IMPLEMENTS_getLocStart(type) \
Eli Friedmandc41d792013-09-10 22:57:15 +0000190 (void) is_good(implements_getLocStart(&type::getLocStart))
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000191#define ASSERT_IMPLEMENTS_getLocEnd(type) \
Eli Friedmandc41d792013-09-10 22:57:15 +0000192 (void) is_good(implements_getLocEnd(&type::getLocEnd))
John McCallbd066782011-02-09 08:16:59 +0000193}
194
195/// Check whether the various Stmt classes implement their member
196/// functions.
Eli Friedmandc41d792013-09-10 22:57:15 +0000197LLVM_ATTRIBUTE_UNUSED
John McCallbd066782011-02-09 08:16:59 +0000198static inline void check_implementations() {
199#define ABSTRACT_STMT(type)
200#define STMT(type, base) \
201 ASSERT_IMPLEMENTS_children(type); \
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000202 ASSERT_IMPLEMENTS_getLocStart(type); \
203 ASSERT_IMPLEMENTS_getLocEnd(type);
John McCallbd066782011-02-09 08:16:59 +0000204#include "clang/AST/StmtNodes.inc"
205}
206
207Stmt::child_range Stmt::children() {
208 switch (getStmtClass()) {
209 case Stmt::NoStmtClass: llvm_unreachable("statement without class");
210#define ABSTRACT_STMT(type)
211#define STMT(type, base) \
212 case Stmt::type##Class: \
213 return static_cast<type*>(this)->children();
214#include "clang/AST/StmtNodes.inc"
215 }
216 llvm_unreachable("unknown statement kind!");
John McCallbd066782011-02-09 08:16:59 +0000217}
218
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000219// Amusing macro metaprogramming hack: check whether a class provides
220// a more specific implementation of getSourceRange.
221//
222// See also Expr.cpp:getExprLoc().
223namespace {
224 /// This implementation is used when a class provides a custom
225 /// implementation of getSourceRange.
226 template <class S, class T>
227 SourceRange getSourceRangeImpl(const Stmt *stmt,
228 SourceRange (T::*v)() const) {
229 return static_cast<const S*>(stmt)->getSourceRange();
230 }
231
232 /// This implementation is used when a class doesn't provide a custom
233 /// implementation of getSourceRange. Overload resolution should pick it over
234 /// the implementation above because it's more specialized according to
235 /// function template partial ordering.
236 template <class S>
237 SourceRange getSourceRangeImpl(const Stmt *stmt,
238 SourceRange (Stmt::*v)() const) {
239 return SourceRange(static_cast<const S*>(stmt)->getLocStart(),
240 static_cast<const S*>(stmt)->getLocEnd());
241 }
242}
243
John McCallbd066782011-02-09 08:16:59 +0000244SourceRange Stmt::getSourceRange() const {
245 switch (getStmtClass()) {
246 case Stmt::NoStmtClass: llvm_unreachable("statement without class");
247#define ABSTRACT_STMT(type)
248#define STMT(type, base) \
249 case Stmt::type##Class: \
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000250 return getSourceRangeImpl<type>(this, &type::getSourceRange);
John McCallbd066782011-02-09 08:16:59 +0000251#include "clang/AST/StmtNodes.inc"
252 }
253 llvm_unreachable("unknown statement kind!");
John McCallbd066782011-02-09 08:16:59 +0000254}
255
Daniel Dunbarb0ab5e92012-03-09 15:39:19 +0000256SourceLocation Stmt::getLocStart() const {
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000257// llvm::errs() << "getLocStart() for " << getStmtClassName() << "\n";
Daniel Dunbarb0ab5e92012-03-09 15:39:19 +0000258 switch (getStmtClass()) {
259 case Stmt::NoStmtClass: llvm_unreachable("statement without class");
260#define ABSTRACT_STMT(type)
261#define STMT(type, base) \
262 case Stmt::type##Class: \
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000263 return static_cast<const type*>(this)->getLocStart();
Daniel Dunbarb0ab5e92012-03-09 15:39:19 +0000264#include "clang/AST/StmtNodes.inc"
265 }
266 llvm_unreachable("unknown statement kind");
267}
268
269SourceLocation Stmt::getLocEnd() const {
270 switch (getStmtClass()) {
271 case Stmt::NoStmtClass: llvm_unreachable("statement without class");
272#define ABSTRACT_STMT(type)
273#define STMT(type, base) \
274 case Stmt::type##Class: \
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000275 return static_cast<const type*>(this)->getLocEnd();
Daniel Dunbarb0ab5e92012-03-09 15:39:19 +0000276#include "clang/AST/StmtNodes.inc"
277 }
278 llvm_unreachable("unknown statement kind");
279}
280
Craig Toppere6960e22013-08-22 05:28:54 +0000281CompoundStmt::CompoundStmt(const ASTContext &C, ArrayRef<Stmt*> Stmts,
Benjamin Kramere2a929d2012-07-04 17:03:41 +0000282 SourceLocation LB, SourceLocation RB)
Aaron Ballmance6c67e2014-10-22 21:06:18 +0000283 : Stmt(CompoundStmtClass), LBraceLoc(LB), RBraceLoc(RB) {
Nico Webera2a0eb92012-12-29 20:03:39 +0000284 CompoundStmtBits.NumStmts = Stmts.size();
285 assert(CompoundStmtBits.NumStmts == Stmts.size() &&
Benjamin Kramere2a929d2012-07-04 17:03:41 +0000286 "NumStmts doesn't fit in bits of CompoundStmtBits.NumStmts!");
287
Nico Webera2a0eb92012-12-29 20:03:39 +0000288 if (Stmts.size() == 0) {
Craig Topper36250ad2014-05-12 05:36:57 +0000289 Body = nullptr;
Benjamin Kramere2a929d2012-07-04 17:03:41 +0000290 return;
291 }
292
Nico Webera2a0eb92012-12-29 20:03:39 +0000293 Body = new (C) Stmt*[Stmts.size()];
294 std::copy(Stmts.begin(), Stmts.end(), Body);
Benjamin Kramere2a929d2012-07-04 17:03:41 +0000295}
296
Craig Topperc571c812013-08-22 06:02:26 +0000297void CompoundStmt::setStmts(const ASTContext &C, Stmt **Stmts,
298 unsigned NumStmts) {
Douglas Gregora9af1d12009-04-17 00:04:06 +0000299 if (this->Body)
300 C.Deallocate(Body);
John McCall925b16622010-10-26 08:39:16 +0000301 this->CompoundStmtBits.NumStmts = NumStmts;
Douglas Gregora9af1d12009-04-17 00:04:06 +0000302
303 Body = new (C) Stmt*[NumStmts];
304 memcpy(Body, Stmts, sizeof(Stmt *) * NumStmts);
305}
Steve Narofff84d11f2007-05-23 21:48:04 +0000306
Chris Lattnereefa10e2007-05-28 06:56:27 +0000307const char *LabelStmt::getName() const {
Chris Lattnerc8e630e2011-02-17 07:39:24 +0000308 return getDecl()->getIdentifier()->getNameStart();
Chris Lattnereefa10e2007-05-28 06:56:27 +0000309}
310
Craig Toppere6960e22013-08-22 05:28:54 +0000311AttributedStmt *AttributedStmt::Create(const ASTContext &C, SourceLocation Loc,
Alexander Kornienko20f6fc62012-07-09 10:04:07 +0000312 ArrayRef<const Attr*> Attrs,
313 Stmt *SubStmt) {
Aaron Ballmanf3d9b092014-05-13 14:55:01 +0000314 assert(!Attrs.empty() && "Attrs should not be empty");
315 void *Mem = C.Allocate(sizeof(AttributedStmt) + sizeof(Attr *) * Attrs.size(),
Alexander Kornienko20f6fc62012-07-09 10:04:07 +0000316 llvm::alignOf<AttributedStmt>());
317 return new (Mem) AttributedStmt(Loc, Attrs, SubStmt);
318}
319
Craig Toppere6960e22013-08-22 05:28:54 +0000320AttributedStmt *AttributedStmt::CreateEmpty(const ASTContext &C,
321 unsigned NumAttrs) {
Alexander Kornienko20f6fc62012-07-09 10:04:07 +0000322 assert(NumAttrs > 0 && "NumAttrs should be greater than zero");
Aaron Ballmanf3d9b092014-05-13 14:55:01 +0000323 void *Mem = C.Allocate(sizeof(AttributedStmt) + sizeof(Attr *) * NumAttrs,
Alexander Kornienko20f6fc62012-07-09 10:04:07 +0000324 llvm::alignOf<AttributedStmt>());
325 return new (Mem) AttributedStmt(EmptyShell(), NumAttrs);
326}
327
Craig Topperc571c812013-08-22 06:02:26 +0000328std::string AsmStmt::generateAsmString(const ASTContext &C) const {
Chad Rosierf70b7e22012-08-28 18:21:14 +0000329 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
330 return gccAsmStmt->generateAsmString(C);
331 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
332 return msAsmStmt->generateAsmString(C);
Chad Rosierbbbe9ab2012-08-28 17:43:23 +0000333 llvm_unreachable("unknown asm statement kind!");
334}
335
336StringRef AsmStmt::getOutputConstraint(unsigned i) const {
Chad Rosierf70b7e22012-08-28 18:21:14 +0000337 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
338 return gccAsmStmt->getOutputConstraint(i);
339 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
340 return msAsmStmt->getOutputConstraint(i);
Chad Rosierbbbe9ab2012-08-28 17:43:23 +0000341 llvm_unreachable("unknown asm statement kind!");
342}
343
344const Expr *AsmStmt::getOutputExpr(unsigned i) const {
Chad Rosierf70b7e22012-08-28 18:21:14 +0000345 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
346 return gccAsmStmt->getOutputExpr(i);
347 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
348 return msAsmStmt->getOutputExpr(i);
Chad Rosierbbbe9ab2012-08-28 17:43:23 +0000349 llvm_unreachable("unknown asm statement kind!");
350}
351
352StringRef AsmStmt::getInputConstraint(unsigned i) const {
Chad Rosierf70b7e22012-08-28 18:21:14 +0000353 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
354 return gccAsmStmt->getInputConstraint(i);
355 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
356 return msAsmStmt->getInputConstraint(i);
Chad Rosierbbbe9ab2012-08-28 17:43:23 +0000357 llvm_unreachable("unknown asm statement kind!");
358}
359
360const Expr *AsmStmt::getInputExpr(unsigned i) const {
Chad Rosierf70b7e22012-08-28 18:21:14 +0000361 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
362 return gccAsmStmt->getInputExpr(i);
363 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
364 return msAsmStmt->getInputExpr(i);
Chad Rosierbbbe9ab2012-08-28 17:43:23 +0000365 llvm_unreachable("unknown asm statement kind!");
366}
367
368StringRef AsmStmt::getClobber(unsigned i) const {
Chad Rosierf70b7e22012-08-28 18:21:14 +0000369 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
370 return gccAsmStmt->getClobber(i);
371 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
372 return msAsmStmt->getClobber(i);
Chad Rosierbbbe9ab2012-08-28 17:43:23 +0000373 llvm_unreachable("unknown asm statement kind!");
374}
375
Chad Rosiera1b5c8c2012-08-28 00:24:05 +0000376/// getNumPlusOperands - Return the number of output operands that have a "+"
377/// constraint.
378unsigned AsmStmt::getNumPlusOperands() const {
379 unsigned Res = 0;
380 for (unsigned i = 0, e = getNumOutputs(); i != e; ++i)
381 if (isOutputPlusConstraint(i))
382 ++Res;
383 return Res;
384}
385
Akira Hatanaka987f1862014-08-22 06:05:21 +0000386char GCCAsmStmt::AsmStringPiece::getModifier() const {
387 assert(isOperand() && "Only Operands can have modifiers.");
388 return isLetter(Str[0]) ? Str[0] : '\0';
389}
390
Chad Rosier6100ae12012-08-27 23:47:56 +0000391StringRef GCCAsmStmt::getClobber(unsigned i) const {
392 return getClobberStringLiteral(i)->getString();
393}
394
Chad Rosierde70e0e2012-08-25 00:11:56 +0000395Expr *GCCAsmStmt::getOutputExpr(unsigned i) {
Ted Kremenek5778acf2008-10-27 18:40:21 +0000396 return cast<Expr>(Exprs[i]);
397}
Chris Lattner72bbf172009-03-10 04:59:06 +0000398
399/// getOutputConstraint - Return the constraint string for the specified
400/// output operand. All output constraints are known to be non-empty (either
401/// '=' or '+').
Chad Rosierde70e0e2012-08-25 00:11:56 +0000402StringRef GCCAsmStmt::getOutputConstraint(unsigned i) const {
Anders Carlsson66de0812010-01-30 20:38:10 +0000403 return getOutputConstraintLiteral(i)->getString();
Ted Kremenek5778acf2008-10-27 18:40:21 +0000404}
Chris Lattner72bbf172009-03-10 04:59:06 +0000405
Chad Rosierde70e0e2012-08-25 00:11:56 +0000406Expr *GCCAsmStmt::getInputExpr(unsigned i) {
Ted Kremenek5778acf2008-10-27 18:40:21 +0000407 return cast<Expr>(Exprs[i + NumOutputs]);
408}
Chad Rosierde70e0e2012-08-25 00:11:56 +0000409void GCCAsmStmt::setInputExpr(unsigned i, Expr *E) {
Chris Lattner93ede022011-02-21 22:09:29 +0000410 Exprs[i + NumOutputs] = E;
411}
412
Chris Lattner72bbf172009-03-10 04:59:06 +0000413/// getInputConstraint - Return the specified input constraint. Unlike output
414/// constraints, these can be empty.
Chad Rosierde70e0e2012-08-25 00:11:56 +0000415StringRef GCCAsmStmt::getInputConstraint(unsigned i) const {
Anders Carlsson66de0812010-01-30 20:38:10 +0000416 return getInputConstraintLiteral(i)->getString();
Ted Kremenek5778acf2008-10-27 18:40:21 +0000417}
418
Craig Topperc571c812013-08-22 06:02:26 +0000419void GCCAsmStmt::setOutputsAndInputsAndClobbers(const ASTContext &C,
420 IdentifierInfo **Names,
421 StringLiteral **Constraints,
422 Stmt **Exprs,
423 unsigned NumOutputs,
424 unsigned NumInputs,
425 StringLiteral **Clobbers,
426 unsigned NumClobbers) {
Douglas Gregorf994f062009-04-17 20:57:14 +0000427 this->NumOutputs = NumOutputs;
428 this->NumInputs = NumInputs;
Anders Carlsson98323d22010-01-30 23:19:41 +0000429 this->NumClobbers = NumClobbers;
430
431 unsigned NumExprs = NumOutputs + NumInputs;
Chad Rosier42032fa2012-08-07 23:12:23 +0000432
Anders Carlsson98323d22010-01-30 23:19:41 +0000433 C.Deallocate(this->Names);
434 this->Names = new (C) IdentifierInfo*[NumExprs];
435 std::copy(Names, Names + NumExprs, this->Names);
Chad Rosier42032fa2012-08-07 23:12:23 +0000436
Anders Carlsson98323d22010-01-30 23:19:41 +0000437 C.Deallocate(this->Exprs);
438 this->Exprs = new (C) Stmt*[NumExprs];
439 std::copy(Exprs, Exprs + NumExprs, this->Exprs);
Chad Rosier42032fa2012-08-07 23:12:23 +0000440
Anders Carlsson98323d22010-01-30 23:19:41 +0000441 C.Deallocate(this->Constraints);
442 this->Constraints = new (C) StringLiteral*[NumExprs];
443 std::copy(Constraints, Constraints + NumExprs, this->Constraints);
Chad Rosier42032fa2012-08-07 23:12:23 +0000444
Anders Carlsson98323d22010-01-30 23:19:41 +0000445 C.Deallocate(this->Clobbers);
446 this->Clobbers = new (C) StringLiteral*[NumClobbers];
447 std::copy(Clobbers, Clobbers + NumClobbers, this->Clobbers);
Douglas Gregorf994f062009-04-17 20:57:14 +0000448}
449
Chris Lattnerd7d5fdf2009-03-10 06:33:24 +0000450/// getNamedOperand - Given a symbolic operand reference like %[foo],
451/// translate this into a numeric value needed to reference the same operand.
452/// This returns -1 if the operand name is invalid.
Chad Rosierde70e0e2012-08-25 00:11:56 +0000453int GCCAsmStmt::getNamedOperand(StringRef SymbolicName) const {
Chris Lattnerd7d5fdf2009-03-10 06:33:24 +0000454 unsigned NumPlusOperands = 0;
Mike Stump11289f42009-09-09 15:08:12 +0000455
Chris Lattnerd7d5fdf2009-03-10 06:33:24 +0000456 // Check if this is an output operand.
457 for (unsigned i = 0, e = getNumOutputs(); i != e; ++i) {
458 if (getOutputName(i) == SymbolicName)
459 return i;
Chris Lattnerd7d5fdf2009-03-10 06:33:24 +0000460 }
Mike Stump11289f42009-09-09 15:08:12 +0000461
Chris Lattnerd7d5fdf2009-03-10 06:33:24 +0000462 for (unsigned i = 0, e = getNumInputs(); i != e; ++i)
463 if (getInputName(i) == SymbolicName)
464 return getNumOutputs() + NumPlusOperands + i;
465
466 // Not found.
467 return -1;
468}
469
Chris Lattner35b58362009-03-10 23:21:44 +0000470/// AnalyzeAsmString - Analyze the asm string of the current asm, decomposing
471/// it into pieces. If the asm string is erroneous, emit errors and return
472/// true, otherwise return false.
Chad Rosierde70e0e2012-08-25 00:11:56 +0000473unsigned GCCAsmStmt::AnalyzeAsmString(SmallVectorImpl<AsmStringPiece>&Pieces,
Craig Topperc571c812013-08-22 06:02:26 +0000474 const ASTContext &C, unsigned &DiagOffs) const {
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000475 StringRef Str = getAsmString()->getString();
Benjamin Kramer35b077e2010-08-17 12:54:38 +0000476 const char *StrStart = Str.begin();
477 const char *StrEnd = Str.end();
Chris Lattnera41b8472009-03-10 23:51:40 +0000478 const char *CurPtr = StrStart;
Mike Stump11289f42009-09-09 15:08:12 +0000479
Chris Lattner35b58362009-03-10 23:21:44 +0000480 // "Simple" inline asms have no constraints or operands, just convert the asm
481 // string to escape $'s.
482 if (isSimple()) {
483 std::string Result;
Chris Lattnera41b8472009-03-10 23:51:40 +0000484 for (; CurPtr != StrEnd; ++CurPtr) {
485 switch (*CurPtr) {
Chris Lattner35b58362009-03-10 23:21:44 +0000486 case '$':
487 Result += "$$";
488 break;
489 default:
Chris Lattnera41b8472009-03-10 23:51:40 +0000490 Result += *CurPtr;
Chris Lattner35b58362009-03-10 23:21:44 +0000491 break;
492 }
493 }
494 Pieces.push_back(AsmStringPiece(Result));
Chris Lattnera41b8472009-03-10 23:51:40 +0000495 return 0;
Chris Lattner35b58362009-03-10 23:21:44 +0000496 }
497
498 // CurStringPiece - The current string that we are building up as we scan the
499 // asm string.
500 std::string CurStringPiece;
Mike Stump11289f42009-09-09 15:08:12 +0000501
Douglas Gregore8bbc122011-09-02 00:18:52 +0000502 bool HasVariants = !C.getTargetInfo().hasNoAsmVariants();
Chad Rosier42032fa2012-08-07 23:12:23 +0000503
Chris Lattner35b58362009-03-10 23:21:44 +0000504 while (1) {
505 // Done with the string?
Chris Lattnera41b8472009-03-10 23:51:40 +0000506 if (CurPtr == StrEnd) {
Chris Lattner35b58362009-03-10 23:21:44 +0000507 if (!CurStringPiece.empty())
508 Pieces.push_back(AsmStringPiece(CurStringPiece));
Chris Lattnera41b8472009-03-10 23:51:40 +0000509 return 0;
Chris Lattner35b58362009-03-10 23:21:44 +0000510 }
Mike Stump11289f42009-09-09 15:08:12 +0000511
Chris Lattnera41b8472009-03-10 23:51:40 +0000512 char CurChar = *CurPtr++;
Chris Lattnerda081a82010-04-05 18:44:00 +0000513 switch (CurChar) {
514 case '$': CurStringPiece += "$$"; continue;
Chris Lattner1a8f3942010-04-23 16:29:58 +0000515 case '{': CurStringPiece += (HasVariants ? "$(" : "{"); continue;
516 case '|': CurStringPiece += (HasVariants ? "$|" : "|"); continue;
517 case '}': CurStringPiece += (HasVariants ? "$)" : "}"); continue;
Chris Lattnerda081a82010-04-05 18:44:00 +0000518 case '%':
519 break;
520 default:
Chris Lattner35b58362009-03-10 23:21:44 +0000521 CurStringPiece += CurChar;
522 continue;
523 }
Chad Rosier42032fa2012-08-07 23:12:23 +0000524
Chris Lattner35b58362009-03-10 23:21:44 +0000525 // Escaped "%" character in asm string.
Chris Lattner3fa25c62009-03-11 00:06:36 +0000526 if (CurPtr == StrEnd) {
527 // % at end of string is invalid (no escape).
528 DiagOffs = CurPtr-StrStart-1;
529 return diag::err_asm_invalid_escape;
530 }
Mike Stump11289f42009-09-09 15:08:12 +0000531
Chris Lattnera41b8472009-03-10 23:51:40 +0000532 char EscapedChar = *CurPtr++;
Chris Lattner35b58362009-03-10 23:21:44 +0000533 if (EscapedChar == '%') { // %% -> %
534 // Escaped percentage sign.
535 CurStringPiece += '%';
536 continue;
537 }
Mike Stump11289f42009-09-09 15:08:12 +0000538
Chris Lattner35b58362009-03-10 23:21:44 +0000539 if (EscapedChar == '=') { // %= -> Generate an unique ID.
540 CurStringPiece += "${:uid}";
541 continue;
542 }
Mike Stump11289f42009-09-09 15:08:12 +0000543
Chris Lattner35b58362009-03-10 23:21:44 +0000544 // Otherwise, we have an operand. If we have accumulated a string so far,
545 // add it to the Pieces list.
546 if (!CurStringPiece.empty()) {
547 Pieces.push_back(AsmStringPiece(CurStringPiece));
548 CurStringPiece.clear();
549 }
Mike Stump11289f42009-09-09 15:08:12 +0000550
Akira Hatanaka987f1862014-08-22 06:05:21 +0000551 // Handle operands that have asmSymbolicName (e.g., %x[foo]) and those that
552 // don't (e.g., %x4). 'x' following the '%' is the constraint modifier.
553
554 const char *Begin = CurPtr - 1; // Points to the character following '%'.
555 const char *Percent = Begin - 1; // Points to '%'.
556
Jordan Rosea7d03842013-02-08 22:30:41 +0000557 if (isLetter(EscapedChar)) {
Benjamin Kramere87c38b2011-07-05 11:13:37 +0000558 if (CurPtr == StrEnd) { // Premature end.
559 DiagOffs = CurPtr-StrStart-1;
560 return diag::err_asm_invalid_escape;
561 }
Chris Lattnera41b8472009-03-10 23:51:40 +0000562 EscapedChar = *CurPtr++;
Chris Lattner35b58362009-03-10 23:21:44 +0000563 }
Mike Stump11289f42009-09-09 15:08:12 +0000564
Akira Hatanaka987f1862014-08-22 06:05:21 +0000565 const TargetInfo &TI = C.getTargetInfo();
566 const SourceManager &SM = C.getSourceManager();
567 const LangOptions &LO = C.getLangOpts();
568
569 // Handle operands that don't have asmSymbolicName (e.g., %x4).
Jordan Rosea7d03842013-02-08 22:30:41 +0000570 if (isDigit(EscapedChar)) {
Chris Lattner35b58362009-03-10 23:21:44 +0000571 // %n - Assembler operand n
Chris Lattner99d892b2009-03-11 22:52:17 +0000572 unsigned N = 0;
Mike Stump11289f42009-09-09 15:08:12 +0000573
Chris Lattner99d892b2009-03-11 22:52:17 +0000574 --CurPtr;
Jordan Rosea7d03842013-02-08 22:30:41 +0000575 while (CurPtr != StrEnd && isDigit(*CurPtr))
Chris Lattner84f3afa2009-03-11 23:09:16 +0000576 N = N*10 + ((*CurPtr++)-'0');
Mike Stump11289f42009-09-09 15:08:12 +0000577
Chris Lattner14311922009-03-11 00:23:13 +0000578 unsigned NumOperands =
579 getNumOutputs() + getNumPlusOperands() + getNumInputs();
580 if (N >= NumOperands) {
581 DiagOffs = CurPtr-StrStart-1;
582 return diag::err_asm_invalid_operand_number;
583 }
584
Akira Hatanaka987f1862014-08-22 06:05:21 +0000585 // Str contains "x4" (Operand without the leading %).
586 std::string Str(Begin, CurPtr - Begin);
587
588 // (BeginLoc, EndLoc) represents the range of the operand we are currently
589 // processing. Unlike Str, the range includes the leading '%'.
590 SourceLocation BeginLoc =
591 getAsmString()->getLocationOfByte(Percent - StrStart, SM, LO, TI);
592 SourceLocation EndLoc =
593 getAsmString()->getLocationOfByte(CurPtr - StrStart, SM, LO, TI);
594
595 Pieces.push_back(AsmStringPiece(N, Str, BeginLoc, EndLoc));
Chris Lattner35b58362009-03-10 23:21:44 +0000596 continue;
597 }
Mike Stump11289f42009-09-09 15:08:12 +0000598
Akira Hatanaka987f1862014-08-22 06:05:21 +0000599 // Handle operands that have asmSymbolicName (e.g., %x[foo]).
Chris Lattner35b58362009-03-10 23:21:44 +0000600 if (EscapedChar == '[') {
Chris Lattner3fa25c62009-03-11 00:06:36 +0000601 DiagOffs = CurPtr-StrStart-1;
Mike Stump11289f42009-09-09 15:08:12 +0000602
Chris Lattner3fa25c62009-03-11 00:06:36 +0000603 // Find the ']'.
Chris Lattnera41b8472009-03-10 23:51:40 +0000604 const char *NameEnd = (const char*)memchr(CurPtr, ']', StrEnd-CurPtr);
Craig Topper36250ad2014-05-12 05:36:57 +0000605 if (NameEnd == nullptr)
Chris Lattner3fa25c62009-03-11 00:06:36 +0000606 return diag::err_asm_unterminated_symbolic_operand_name;
607 if (NameEnd == CurPtr)
608 return diag::err_asm_empty_symbolic_operand_name;
Mike Stump11289f42009-09-09 15:08:12 +0000609
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000610 StringRef SymbolicName(CurPtr, NameEnd - CurPtr);
Mike Stump11289f42009-09-09 15:08:12 +0000611
Chris Lattner35b58362009-03-10 23:21:44 +0000612 int N = getNamedOperand(SymbolicName);
Chris Lattner3fa25c62009-03-11 00:06:36 +0000613 if (N == -1) {
614 // Verify that an operand with that name exists.
615 DiagOffs = CurPtr-StrStart;
616 return diag::err_asm_unknown_symbolic_operand_name;
617 }
Akira Hatanaka987f1862014-08-22 06:05:21 +0000618
619 // Str contains "x[foo]" (Operand without the leading %).
620 std::string Str(Begin, NameEnd + 1 - Begin);
621
622 // (BeginLoc, EndLoc) represents the range of the operand we are currently
623 // processing. Unlike Str, the range includes the leading '%'.
624 SourceLocation BeginLoc =
625 getAsmString()->getLocationOfByte(Percent - StrStart, SM, LO, TI);
626 SourceLocation EndLoc =
627 getAsmString()->getLocationOfByte(NameEnd + 1 - StrStart, SM, LO, TI);
628
629 Pieces.push_back(AsmStringPiece(N, Str, BeginLoc, EndLoc));
Mike Stump11289f42009-09-09 15:08:12 +0000630
Chris Lattner3fa25c62009-03-11 00:06:36 +0000631 CurPtr = NameEnd+1;
Chris Lattner35b58362009-03-10 23:21:44 +0000632 continue;
633 }
Mike Stump11289f42009-09-09 15:08:12 +0000634
Chris Lattner0cdaa2e2009-03-10 23:57:07 +0000635 DiagOffs = CurPtr-StrStart-1;
Chris Lattnera41b8472009-03-10 23:51:40 +0000636 return diag::err_asm_invalid_escape;
Chris Lattner35b58362009-03-10 23:21:44 +0000637 }
638}
Chad Rosier3b0c2602012-08-27 20:23:31 +0000639
640/// Assemble final IR asm string (GCC-style).
Craig Topperc571c812013-08-22 06:02:26 +0000641std::string GCCAsmStmt::generateAsmString(const ASTContext &C) const {
Chad Rosier14836ba2012-08-24 17:05:45 +0000642 // Analyze the asm string to decompose it into its pieces. We know that Sema
643 // has already done this, so it is guaranteed to be successful.
Chad Rosierde70e0e2012-08-25 00:11:56 +0000644 SmallVector<GCCAsmStmt::AsmStringPiece, 4> Pieces;
Chad Rosier14836ba2012-08-24 17:05:45 +0000645 unsigned DiagOffs;
646 AnalyzeAsmString(Pieces, C, DiagOffs);
647
648 std::string AsmString;
649 for (unsigned i = 0, e = Pieces.size(); i != e; ++i) {
650 if (Pieces[i].isString())
651 AsmString += Pieces[i].getString();
652 else if (Pieces[i].getModifier() == '\0')
653 AsmString += '$' + llvm::utostr(Pieces[i].getOperandNo());
654 else
655 AsmString += "${" + llvm::utostr(Pieces[i].getOperandNo()) + ':' +
656 Pieces[i].getModifier() + '}';
657 }
658 return AsmString;
659}
Chris Lattner35b58362009-03-10 23:21:44 +0000660
Chad Rosier3b0c2602012-08-27 20:23:31 +0000661/// Assemble final IR asm string (MS-style).
Craig Topperc571c812013-08-22 06:02:26 +0000662std::string MSAsmStmt::generateAsmString(const ASTContext &C) const {
Chad Rosier3b0c2602012-08-27 20:23:31 +0000663 // FIXME: This needs to be translated into the IR string representation.
Chad Rosier0bca4692012-08-28 20:33:49 +0000664 return AsmStr;
Chad Rosier3b0c2602012-08-27 20:23:31 +0000665}
666
Chad Rosierfe31e622012-08-24 00:07:09 +0000667Expr *MSAsmStmt::getOutputExpr(unsigned i) {
668 return cast<Expr>(Exprs[i]);
669}
670
671Expr *MSAsmStmt::getInputExpr(unsigned i) {
672 return cast<Expr>(Exprs[i + NumOutputs]);
673}
674void MSAsmStmt::setInputExpr(unsigned i, Expr *E) {
675 Exprs[i + NumOutputs] = E;
676}
677
Sam Weinigebcea982010-02-03 02:09:59 +0000678QualType CXXCatchStmt::getCaughtType() const {
679 if (ExceptionDecl)
680 return ExceptionDecl->getType();
681 return QualType();
682}
683
Chris Lattner86f5e132008-01-30 05:01:46 +0000684//===----------------------------------------------------------------------===//
685// Constructors
686//===----------------------------------------------------------------------===//
687
Craig Toppere6960e22013-08-22 05:28:54 +0000688GCCAsmStmt::GCCAsmStmt(const ASTContext &C, SourceLocation asmloc,
689 bool issimple, bool isvolatile, unsigned numoutputs,
690 unsigned numinputs, IdentifierInfo **names,
691 StringLiteral **constraints, Expr **exprs,
692 StringLiteral *asmstr, unsigned numclobbers,
693 StringLiteral **clobbers, SourceLocation rparenloc)
Chad Rosierfe3352e2012-08-27 19:38:01 +0000694 : AsmStmt(GCCAsmStmtClass, asmloc, issimple, isvolatile, numoutputs,
695 numinputs, numclobbers), RParenLoc(rparenloc), AsmStr(asmstr) {
Nico Weberde565e32008-08-05 23:15:29 +0000696
Chad Rosier42032fa2012-08-07 23:12:23 +0000697 unsigned NumExprs = NumOutputs + NumInputs;
698
Anders Carlsson98323d22010-01-30 23:19:41 +0000699 Names = new (C) IdentifierInfo*[NumExprs];
700 std::copy(names, names + NumExprs, Names);
701
702 Exprs = new (C) Stmt*[NumExprs];
703 std::copy(exprs, exprs + NumExprs, Exprs);
704
705 Constraints = new (C) StringLiteral*[NumExprs];
706 std::copy(constraints, constraints + NumExprs, Constraints);
707
708 Clobbers = new (C) StringLiteral*[NumClobbers];
709 std::copy(clobbers, clobbers + NumClobbers, Clobbers);
Anders Carlsson94ea8aa2007-11-22 01:36:19 +0000710}
711
Craig Toppere6960e22013-08-22 05:28:54 +0000712MSAsmStmt::MSAsmStmt(const ASTContext &C, SourceLocation asmloc,
Chad Rosier7dbef3e2012-08-16 00:06:53 +0000713 SourceLocation lbraceloc, bool issimple, bool isvolatile,
Chad Rosierf8037a12012-10-16 21:55:39 +0000714 ArrayRef<Token> asmtoks, unsigned numoutputs,
John McCallf413f5e2013-05-03 00:10:13 +0000715 unsigned numinputs,
Chad Rosierf8037a12012-10-16 21:55:39 +0000716 ArrayRef<StringRef> constraints, ArrayRef<Expr*> exprs,
717 StringRef asmstr, ArrayRef<StringRef> clobbers,
718 SourceLocation endloc)
719 : AsmStmt(MSAsmStmtClass, asmloc, issimple, isvolatile, numoutputs,
720 numinputs, clobbers.size()), LBraceLoc(lbraceloc),
John McCallf413f5e2013-05-03 00:10:13 +0000721 EndLoc(endloc), NumAsmToks(asmtoks.size()) {
Chad Rosier7dbef3e2012-08-16 00:06:53 +0000722
John McCallf413f5e2013-05-03 00:10:13 +0000723 initialize(C, asmstr, asmtoks, constraints, exprs, clobbers);
724}
Chad Rosier7dbef3e2012-08-16 00:06:53 +0000725
Craig Toppere6960e22013-08-22 05:28:54 +0000726static StringRef copyIntoContext(const ASTContext &C, StringRef str) {
John McCallf413f5e2013-05-03 00:10:13 +0000727 size_t size = str.size();
728 char *buffer = new (C) char[size];
729 memcpy(buffer, str.data(), size);
730 return StringRef(buffer, size);
731}
732
Craig Toppere6960e22013-08-22 05:28:54 +0000733void MSAsmStmt::initialize(const ASTContext &C, StringRef asmstr,
John McCallf413f5e2013-05-03 00:10:13 +0000734 ArrayRef<Token> asmtoks,
735 ArrayRef<StringRef> constraints,
736 ArrayRef<Expr*> exprs,
737 ArrayRef<StringRef> clobbers) {
738 assert(NumAsmToks == asmtoks.size());
739 assert(NumClobbers == clobbers.size());
740
741 unsigned NumExprs = exprs.size();
742 assert(NumExprs == NumOutputs + NumInputs);
743 assert(NumExprs == constraints.size());
744
745 AsmStr = copyIntoContext(C, asmstr);
Chad Rosier99fc3812012-08-07 00:29:06 +0000746
Chad Rosierfe31e622012-08-24 00:07:09 +0000747 Exprs = new (C) Stmt*[NumExprs];
Chad Rosierf8037a12012-10-16 21:55:39 +0000748 for (unsigned i = 0, e = NumExprs; i != e; ++i)
749 Exprs[i] = exprs[i];
Chad Rosierfe31e622012-08-24 00:07:09 +0000750
Chad Rosier99fc3812012-08-07 00:29:06 +0000751 AsmToks = new (C) Token[NumAsmToks];
752 for (unsigned i = 0, e = NumAsmToks; i != e; ++i)
753 AsmToks[i] = asmtoks[i];
Chad Rosier3ed0bd92012-08-08 19:48:07 +0000754
Chad Rosier3dd7bf22012-08-28 20:28:20 +0000755 Constraints = new (C) StringRef[NumExprs];
756 for (unsigned i = 0, e = NumExprs; i != e; ++i) {
John McCallf413f5e2013-05-03 00:10:13 +0000757 Constraints[i] = copyIntoContext(C, constraints[i]);
Chad Rosier3dd7bf22012-08-28 20:28:20 +0000758 }
759
Chad Rosierbaf53f92012-08-10 21:36:25 +0000760 Clobbers = new (C) StringRef[NumClobbers];
Chad Rosierd6ef7042012-08-10 21:06:19 +0000761 for (unsigned i = 0, e = NumClobbers; i != e; ++i) {
762 // FIXME: Avoid the allocation/copy if at all possible.
John McCallf413f5e2013-05-03 00:10:13 +0000763 Clobbers[i] = copyIntoContext(C, clobbers[i]);
Chad Rosierd6ef7042012-08-10 21:06:19 +0000764 }
Chad Rosier32503022012-06-11 20:47:18 +0000765}
766
Chris Lattner86f5e132008-01-30 05:01:46 +0000767ObjCForCollectionStmt::ObjCForCollectionStmt(Stmt *Elem, Expr *Collect,
768 Stmt *Body, SourceLocation FCL,
Nico Weberde565e32008-08-05 23:15:29 +0000769 SourceLocation RPL)
Chris Lattner86f5e132008-01-30 05:01:46 +0000770: Stmt(ObjCForCollectionStmtClass) {
771 SubExprs[ELEM] = Elem;
Pavel Labath515f4db2013-09-03 14:41:16 +0000772 SubExprs[COLLECTION] = Collect;
Chris Lattner86f5e132008-01-30 05:01:46 +0000773 SubExprs[BODY] = Body;
774 ForLoc = FCL;
775 RParenLoc = RPL;
776}
777
Douglas Gregor96c79492010-04-23 22:50:49 +0000778ObjCAtTryStmt::ObjCAtTryStmt(SourceLocation atTryLoc, Stmt *atTryStmt,
779 Stmt **CatchStmts, unsigned NumCatchStmts,
780 Stmt *atFinallyStmt)
781 : Stmt(ObjCAtTryStmtClass), AtTryLoc(atTryLoc),
Craig Topper36250ad2014-05-12 05:36:57 +0000782 NumCatchStmts(NumCatchStmts), HasFinally(atFinallyStmt != nullptr) {
Douglas Gregor96c79492010-04-23 22:50:49 +0000783 Stmt **Stmts = getStmts();
784 Stmts[0] = atTryStmt;
785 for (unsigned I = 0; I != NumCatchStmts; ++I)
786 Stmts[I + 1] = CatchStmts[I];
Chad Rosier42032fa2012-08-07 23:12:23 +0000787
Douglas Gregor96c79492010-04-23 22:50:49 +0000788 if (HasFinally)
789 Stmts[NumCatchStmts + 1] = atFinallyStmt;
790}
Chris Lattner86f5e132008-01-30 05:01:46 +0000791
Craig Toppere6960e22013-08-22 05:28:54 +0000792ObjCAtTryStmt *ObjCAtTryStmt::Create(const ASTContext &Context,
Chad Rosier42032fa2012-08-07 23:12:23 +0000793 SourceLocation atTryLoc,
Douglas Gregor96c79492010-04-23 22:50:49 +0000794 Stmt *atTryStmt,
Chad Rosier42032fa2012-08-07 23:12:23 +0000795 Stmt **CatchStmts,
Douglas Gregor96c79492010-04-23 22:50:49 +0000796 unsigned NumCatchStmts,
797 Stmt *atFinallyStmt) {
Chad Rosier42032fa2012-08-07 23:12:23 +0000798 unsigned Size = sizeof(ObjCAtTryStmt) +
Craig Topper36250ad2014-05-12 05:36:57 +0000799 (1 + NumCatchStmts + (atFinallyStmt != nullptr)) * sizeof(Stmt *);
Chris Lattner5c0b4052010-10-30 05:14:06 +0000800 void *Mem = Context.Allocate(Size, llvm::alignOf<ObjCAtTryStmt>());
Douglas Gregor96c79492010-04-23 22:50:49 +0000801 return new (Mem) ObjCAtTryStmt(atTryLoc, atTryStmt, CatchStmts, NumCatchStmts,
802 atFinallyStmt);
803}
Ted Kremeneka4965842008-02-01 21:28:59 +0000804
Craig Toppere6960e22013-08-22 05:28:54 +0000805ObjCAtTryStmt *ObjCAtTryStmt::CreateEmpty(const ASTContext &Context,
806 unsigned NumCatchStmts,
807 bool HasFinally) {
Chad Rosier42032fa2012-08-07 23:12:23 +0000808 unsigned Size = sizeof(ObjCAtTryStmt) +
Douglas Gregor96c79492010-04-23 22:50:49 +0000809 (1 + NumCatchStmts + HasFinally) * sizeof(Stmt *);
Chris Lattner5c0b4052010-10-30 05:14:06 +0000810 void *Mem = Context.Allocate(Size, llvm::alignOf<ObjCAtTryStmt>());
Chad Rosier42032fa2012-08-07 23:12:23 +0000811 return new (Mem) ObjCAtTryStmt(EmptyShell(), NumCatchStmts, HasFinally);
Douglas Gregor96c79492010-04-23 22:50:49 +0000812}
Nico Weberde565e32008-08-05 23:15:29 +0000813
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000814SourceLocation ObjCAtTryStmt::getLocEnd() const {
Douglas Gregor96c79492010-04-23 22:50:49 +0000815 if (HasFinally)
Erik Verbruggen11a2ecc2012-12-25 14:51:39 +0000816 return getFinallyStmt()->getLocEnd();
817 if (NumCatchStmts)
818 return getCatchStmt(NumCatchStmts - 1)->getLocEnd();
819 return getTryBody()->getLocEnd();
Chris Lattner86f5e132008-01-30 05:01:46 +0000820}
821
Craig Toppere6960e22013-08-22 05:28:54 +0000822CXXTryStmt *CXXTryStmt::Create(const ASTContext &C, SourceLocation tryLoc,
Nico Weber9dff3782012-12-29 20:13:03 +0000823 Stmt *tryBlock, ArrayRef<Stmt*> handlers) {
Sam Weiniga16b0dd2010-02-03 03:56:39 +0000824 std::size_t Size = sizeof(CXXTryStmt);
Nico Weber9dff3782012-12-29 20:13:03 +0000825 Size += ((handlers.size() + 1) * sizeof(Stmt));
Sam Weiniga16b0dd2010-02-03 03:56:39 +0000826
Chris Lattner5c0b4052010-10-30 05:14:06 +0000827 void *Mem = C.Allocate(Size, llvm::alignOf<CXXTryStmt>());
Nico Weber9dff3782012-12-29 20:13:03 +0000828 return new (Mem) CXXTryStmt(tryLoc, tryBlock, handlers);
Sam Weiniga16b0dd2010-02-03 03:56:39 +0000829}
830
Craig Toppere6960e22013-08-22 05:28:54 +0000831CXXTryStmt *CXXTryStmt::Create(const ASTContext &C, EmptyShell Empty,
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000832 unsigned numHandlers) {
833 std::size_t Size = sizeof(CXXTryStmt);
834 Size += ((numHandlers + 1) * sizeof(Stmt));
835
Chris Lattner5c0b4052010-10-30 05:14:06 +0000836 void *Mem = C.Allocate(Size, llvm::alignOf<CXXTryStmt>());
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000837 return new (Mem) CXXTryStmt(Empty, numHandlers);
838}
839
Sam Weiniga16b0dd2010-02-03 03:56:39 +0000840CXXTryStmt::CXXTryStmt(SourceLocation tryLoc, Stmt *tryBlock,
Nico Weber9dff3782012-12-29 20:13:03 +0000841 ArrayRef<Stmt*> handlers)
842 : Stmt(CXXTryStmtClass), TryLoc(tryLoc), NumHandlers(handlers.size()) {
Sam Weiniga16b0dd2010-02-03 03:56:39 +0000843 Stmt **Stmts = reinterpret_cast<Stmt **>(this + 1);
Sam Weinigebcea982010-02-03 02:09:59 +0000844 Stmts[0] = tryBlock;
Nico Weber9dff3782012-12-29 20:13:03 +0000845 std::copy(handlers.begin(), handlers.end(), Stmts + 1);
Sam Weinigebcea982010-02-03 02:09:59 +0000846}
847
Richard Smith02e85f32011-04-14 22:09:26 +0000848CXXForRangeStmt::CXXForRangeStmt(DeclStmt *Range, DeclStmt *BeginEndStmt,
849 Expr *Cond, Expr *Inc, DeclStmt *LoopVar,
850 Stmt *Body, SourceLocation FL,
851 SourceLocation CL, SourceLocation RPL)
852 : Stmt(CXXForRangeStmtClass), ForLoc(FL), ColonLoc(CL), RParenLoc(RPL) {
853 SubExprs[RANGE] = Range;
854 SubExprs[BEGINEND] = BeginEndStmt;
Pavel Labath515f4db2013-09-03 14:41:16 +0000855 SubExprs[COND] = Cond;
856 SubExprs[INC] = Inc;
Richard Smith02e85f32011-04-14 22:09:26 +0000857 SubExprs[LOOPVAR] = LoopVar;
858 SubExprs[BODY] = Body;
859}
860
861Expr *CXXForRangeStmt::getRangeInit() {
862 DeclStmt *RangeStmt = getRangeStmt();
863 VarDecl *RangeDecl = dyn_cast_or_null<VarDecl>(RangeStmt->getSingleDecl());
Richard Trieu86738692014-02-27 23:59:14 +0000864 assert(RangeDecl && "for-range should have a single var decl");
Richard Smith02e85f32011-04-14 22:09:26 +0000865 return RangeDecl->getInit();
866}
867
868const Expr *CXXForRangeStmt::getRangeInit() const {
869 return const_cast<CXXForRangeStmt*>(this)->getRangeInit();
870}
871
872VarDecl *CXXForRangeStmt::getLoopVariable() {
873 Decl *LV = cast<DeclStmt>(getLoopVarStmt())->getSingleDecl();
874 assert(LV && "No loop variable in CXXForRangeStmt");
875 return cast<VarDecl>(LV);
876}
877
878const VarDecl *CXXForRangeStmt::getLoopVariable() const {
879 return const_cast<CXXForRangeStmt*>(this)->getLoopVariable();
880}
881
Craig Toppere6960e22013-08-22 05:28:54 +0000882IfStmt::IfStmt(const ASTContext &C, SourceLocation IL, VarDecl *var, Expr *cond,
Argyrios Kyrtzidisde2bdf62010-11-20 02:04:01 +0000883 Stmt *then, SourceLocation EL, Stmt *elsev)
884 : Stmt(IfStmtClass), IfLoc(IL), ElseLoc(EL)
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000885{
886 setConditionVariable(C, var);
Pavel Labath515f4db2013-09-03 14:41:16 +0000887 SubExprs[COND] = cond;
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000888 SubExprs[THEN] = then;
Chad Rosier42032fa2012-08-07 23:12:23 +0000889 SubExprs[ELSE] = elsev;
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000890}
891
892VarDecl *IfStmt::getConditionVariable() const {
893 if (!SubExprs[VAR])
Craig Topper36250ad2014-05-12 05:36:57 +0000894 return nullptr;
Chad Rosier42032fa2012-08-07 23:12:23 +0000895
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000896 DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
897 return cast<VarDecl>(DS->getSingleDecl());
898}
899
Craig Toppere6960e22013-08-22 05:28:54 +0000900void IfStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000901 if (!V) {
Craig Topper36250ad2014-05-12 05:36:57 +0000902 SubExprs[VAR] = nullptr;
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000903 return;
904 }
Chad Rosier42032fa2012-08-07 23:12:23 +0000905
Daniel Dunbar62ee6412012-03-09 18:35:03 +0000906 SourceRange VarRange = V->getSourceRange();
907 SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
908 VarRange.getEnd());
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000909}
910
Craig Toppere6960e22013-08-22 05:28:54 +0000911ForStmt::ForStmt(const ASTContext &C, Stmt *Init, Expr *Cond, VarDecl *condVar,
Chad Rosier42032fa2012-08-07 23:12:23 +0000912 Expr *Inc, Stmt *Body, SourceLocation FL, SourceLocation LP,
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000913 SourceLocation RP)
Chad Rosier42032fa2012-08-07 23:12:23 +0000914 : Stmt(ForStmtClass), ForLoc(FL), LParenLoc(LP), RParenLoc(RP)
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000915{
916 SubExprs[INIT] = Init;
917 setConditionVariable(C, condVar);
Pavel Labath515f4db2013-09-03 14:41:16 +0000918 SubExprs[COND] = Cond;
919 SubExprs[INC] = Inc;
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000920 SubExprs[BODY] = Body;
921}
922
923VarDecl *ForStmt::getConditionVariable() const {
924 if (!SubExprs[CONDVAR])
Craig Topper36250ad2014-05-12 05:36:57 +0000925 return nullptr;
Chad Rosier42032fa2012-08-07 23:12:23 +0000926
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000927 DeclStmt *DS = cast<DeclStmt>(SubExprs[CONDVAR]);
928 return cast<VarDecl>(DS->getSingleDecl());
929}
930
Craig Toppere6960e22013-08-22 05:28:54 +0000931void ForStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000932 if (!V) {
Craig Topper36250ad2014-05-12 05:36:57 +0000933 SubExprs[CONDVAR] = nullptr;
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000934 return;
935 }
Chad Rosier42032fa2012-08-07 23:12:23 +0000936
Daniel Dunbar62ee6412012-03-09 18:35:03 +0000937 SourceRange VarRange = V->getSourceRange();
938 SubExprs[CONDVAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
939 VarRange.getEnd());
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000940}
941
Craig Toppere6960e22013-08-22 05:28:54 +0000942SwitchStmt::SwitchStmt(const ASTContext &C, VarDecl *Var, Expr *cond)
Craig Topper36250ad2014-05-12 05:36:57 +0000943 : Stmt(SwitchStmtClass), FirstCase(nullptr), AllEnumCasesCovered(0)
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000944{
945 setConditionVariable(C, Var);
Pavel Labath515f4db2013-09-03 14:41:16 +0000946 SubExprs[COND] = cond;
Craig Topper36250ad2014-05-12 05:36:57 +0000947 SubExprs[BODY] = nullptr;
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000948}
949
950VarDecl *SwitchStmt::getConditionVariable() const {
951 if (!SubExprs[VAR])
Craig Topper36250ad2014-05-12 05:36:57 +0000952 return nullptr;
Chad Rosier42032fa2012-08-07 23:12:23 +0000953
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000954 DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
955 return cast<VarDecl>(DS->getSingleDecl());
956}
957
Craig Toppere6960e22013-08-22 05:28:54 +0000958void SwitchStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000959 if (!V) {
Craig Topper36250ad2014-05-12 05:36:57 +0000960 SubExprs[VAR] = nullptr;
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000961 return;
962 }
Chad Rosier42032fa2012-08-07 23:12:23 +0000963
Daniel Dunbar62ee6412012-03-09 18:35:03 +0000964 SourceRange VarRange = V->getSourceRange();
965 SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
966 VarRange.getEnd());
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000967}
968
John McCallbd066782011-02-09 08:16:59 +0000969Stmt *SwitchCase::getSubStmt() {
Chris Lattner80d51eb2011-02-28 00:18:06 +0000970 if (isa<CaseStmt>(this))
971 return cast<CaseStmt>(this)->getSubStmt();
John McCallbd066782011-02-09 08:16:59 +0000972 return cast<DefaultStmt>(this)->getSubStmt();
973}
974
Craig Toppere6960e22013-08-22 05:28:54 +0000975WhileStmt::WhileStmt(const ASTContext &C, VarDecl *Var, Expr *cond, Stmt *body,
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000976 SourceLocation WL)
Chris Lattner80d51eb2011-02-28 00:18:06 +0000977 : Stmt(WhileStmtClass) {
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000978 setConditionVariable(C, Var);
Pavel Labath515f4db2013-09-03 14:41:16 +0000979 SubExprs[COND] = cond;
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000980 SubExprs[BODY] = body;
981 WhileLoc = WL;
982}
983
984VarDecl *WhileStmt::getConditionVariable() const {
985 if (!SubExprs[VAR])
Craig Topper36250ad2014-05-12 05:36:57 +0000986 return nullptr;
Chad Rosier42032fa2012-08-07 23:12:23 +0000987
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000988 DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
989 return cast<VarDecl>(DS->getSingleDecl());
990}
991
Craig Toppere6960e22013-08-22 05:28:54 +0000992void WhileStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000993 if (!V) {
Craig Topper36250ad2014-05-12 05:36:57 +0000994 SubExprs[VAR] = nullptr;
Douglas Gregor27b98ea2010-06-21 23:44:13 +0000995 return;
996 }
Daniel Dunbar62ee6412012-03-09 18:35:03 +0000997
998 SourceRange VarRange = V->getSourceRange();
999 SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
1000 VarRange.getEnd());
Douglas Gregor27b98ea2010-06-21 23:44:13 +00001001}
1002
Ted Kremenek066dd932007-08-24 21:09:09 +00001003// IndirectGotoStmt
Chris Lattnerc8e630e2011-02-17 07:39:24 +00001004LabelDecl *IndirectGotoStmt::getConstantTarget() {
John McCall9de91602010-10-28 08:53:48 +00001005 if (AddrLabelExpr *E =
1006 dyn_cast<AddrLabelExpr>(getTarget()->IgnoreParenImpCasts()))
1007 return E->getLabel();
Craig Topper36250ad2014-05-12 05:36:57 +00001008 return nullptr;
John McCall9de91602010-10-28 08:53:48 +00001009}
Ted Kremenek066dd932007-08-24 21:09:09 +00001010
Ted Kremenek066dd932007-08-24 21:09:09 +00001011// ReturnStmt
Ted Kremenekc6501db2008-06-17 03:11:08 +00001012const Expr* ReturnStmt::getRetValue() const {
1013 return cast_or_null<Expr>(RetExpr);
1014}
1015Expr* ReturnStmt::getRetValue() {
1016 return cast_or_null<Expr>(RetExpr);
Ted Kremenek066dd932007-08-24 21:09:09 +00001017}
John Wiegley1c0675e2011-04-28 01:08:34 +00001018
Warren Huntf6be4cb2014-07-25 20:52:51 +00001019SEHTryStmt::SEHTryStmt(bool IsCXXTry,
1020 SourceLocation TryLoc,
1021 Stmt *TryBlock,
1022 Stmt *Handler)
1023 : Stmt(SEHTryStmtClass),
1024 IsCXXTry(IsCXXTry),
1025 TryLoc(TryLoc)
1026{
1027 Children[TRY] = TryBlock;
John Wiegley1c0675e2011-04-28 01:08:34 +00001028 Children[HANDLER] = Handler;
1029}
1030
Warren Huntf6be4cb2014-07-25 20:52:51 +00001031SEHTryStmt* SEHTryStmt::Create(const ASTContext &C, bool IsCXXTry,
Craig Toppere6960e22013-08-22 05:28:54 +00001032 SourceLocation TryLoc, Stmt *TryBlock,
Warren Huntf6be4cb2014-07-25 20:52:51 +00001033 Stmt *Handler) {
1034 return new(C) SEHTryStmt(IsCXXTry,TryLoc,TryBlock,Handler);
John Wiegley1c0675e2011-04-28 01:08:34 +00001035}
1036
1037SEHExceptStmt* SEHTryStmt::getExceptHandler() const {
1038 return dyn_cast<SEHExceptStmt>(getHandler());
1039}
1040
1041SEHFinallyStmt* SEHTryStmt::getFinallyHandler() const {
1042 return dyn_cast<SEHFinallyStmt>(getHandler());
1043}
1044
1045SEHExceptStmt::SEHExceptStmt(SourceLocation Loc,
1046 Expr *FilterExpr,
1047 Stmt *Block)
1048 : Stmt(SEHExceptStmtClass),
1049 Loc(Loc)
1050{
Pavel Labath515f4db2013-09-03 14:41:16 +00001051 Children[FILTER_EXPR] = FilterExpr;
John Wiegley1c0675e2011-04-28 01:08:34 +00001052 Children[BLOCK] = Block;
1053}
1054
Craig Toppere6960e22013-08-22 05:28:54 +00001055SEHExceptStmt* SEHExceptStmt::Create(const ASTContext &C, SourceLocation Loc,
1056 Expr *FilterExpr, Stmt *Block) {
John Wiegley1c0675e2011-04-28 01:08:34 +00001057 return new(C) SEHExceptStmt(Loc,FilterExpr,Block);
1058}
1059
1060SEHFinallyStmt::SEHFinallyStmt(SourceLocation Loc,
1061 Stmt *Block)
1062 : Stmt(SEHFinallyStmtClass),
1063 Loc(Loc),
1064 Block(Block)
1065{}
1066
Craig Toppere6960e22013-08-22 05:28:54 +00001067SEHFinallyStmt* SEHFinallyStmt::Create(const ASTContext &C, SourceLocation Loc,
John Wiegley1c0675e2011-04-28 01:08:34 +00001068 Stmt *Block) {
1069 return new(C)SEHFinallyStmt(Loc,Block);
1070}
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001071
1072CapturedStmt::Capture *CapturedStmt::getStoredCaptures() const {
1073 unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (NumCaptures + 1);
1074
1075 // Offset of the first Capture object.
1076 unsigned FirstCaptureOffset =
1077 llvm::RoundUpToAlignment(Size, llvm::alignOf<Capture>());
1078
1079 return reinterpret_cast<Capture *>(
1080 reinterpret_cast<char *>(const_cast<CapturedStmt *>(this))
1081 + FirstCaptureOffset);
1082}
1083
Wei Pan17fbf6e2013-05-04 03:59:06 +00001084CapturedStmt::CapturedStmt(Stmt *S, CapturedRegionKind Kind,
1085 ArrayRef<Capture> Captures,
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001086 ArrayRef<Expr *> CaptureInits,
Tareq A. Siraj6dfa25a2013-04-16 19:37:38 +00001087 CapturedDecl *CD,
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001088 RecordDecl *RD)
1089 : Stmt(CapturedStmtClass), NumCaptures(Captures.size()),
Wei Pan17fbf6e2013-05-04 03:59:06 +00001090 CapDeclAndKind(CD, Kind), TheRecordDecl(RD) {
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001091 assert( S && "null captured statement");
Tareq A. Siraj6dfa25a2013-04-16 19:37:38 +00001092 assert(CD && "null captured declaration for captured statement");
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001093 assert(RD && "null record declaration for captured statement");
1094
1095 // Copy initialization expressions.
1096 Stmt **Stored = getStoredStmts();
1097 for (unsigned I = 0, N = NumCaptures; I != N; ++I)
1098 *Stored++ = CaptureInits[I];
1099
1100 // Copy the statement being captured.
1101 *Stored = S;
1102
1103 // Copy all Capture objects.
1104 Capture *Buffer = getStoredCaptures();
1105 std::copy(Captures.begin(), Captures.end(), Buffer);
1106}
1107
1108CapturedStmt::CapturedStmt(EmptyShell Empty, unsigned NumCaptures)
1109 : Stmt(CapturedStmtClass, Empty), NumCaptures(NumCaptures),
Craig Topper36250ad2014-05-12 05:36:57 +00001110 CapDeclAndKind(nullptr, CR_Default), TheRecordDecl(nullptr) {
1111 getStoredStmts()[NumCaptures] = nullptr;
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001112}
1113
Craig Toppere6960e22013-08-22 05:28:54 +00001114CapturedStmt *CapturedStmt::Create(const ASTContext &Context, Stmt *S,
Wei Pan17fbf6e2013-05-04 03:59:06 +00001115 CapturedRegionKind Kind,
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001116 ArrayRef<Capture> Captures,
1117 ArrayRef<Expr *> CaptureInits,
Tareq A. Siraj6dfa25a2013-04-16 19:37:38 +00001118 CapturedDecl *CD,
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001119 RecordDecl *RD) {
1120 // The layout is
1121 //
1122 // -----------------------------------------------------------
1123 // | CapturedStmt, Init, ..., Init, S, Capture, ..., Capture |
1124 // ----------------^-------------------^----------------------
1125 // getStoredStmts() getStoredCaptures()
1126 //
1127 // where S is the statement being captured.
1128 //
1129 assert(CaptureInits.size() == Captures.size() && "wrong number of arguments");
1130
1131 unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (Captures.size() + 1);
1132 if (!Captures.empty()) {
1133 // Realign for the following Capture array.
1134 Size = llvm::RoundUpToAlignment(Size, llvm::alignOf<Capture>());
1135 Size += sizeof(Capture) * Captures.size();
1136 }
1137
1138 void *Mem = Context.Allocate(Size);
Wei Pan17fbf6e2013-05-04 03:59:06 +00001139 return new (Mem) CapturedStmt(S, Kind, Captures, CaptureInits, CD, RD);
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001140}
1141
Craig Toppere6960e22013-08-22 05:28:54 +00001142CapturedStmt *CapturedStmt::CreateDeserialized(const ASTContext &Context,
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001143 unsigned NumCaptures) {
1144 unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (NumCaptures + 1);
1145 if (NumCaptures > 0) {
1146 // Realign for the following Capture array.
1147 Size = llvm::RoundUpToAlignment(Size, llvm::alignOf<Capture>());
1148 Size += sizeof(Capture) * NumCaptures;
1149 }
1150
1151 void *Mem = Context.Allocate(Size);
1152 return new (Mem) CapturedStmt(EmptyShell(), NumCaptures);
1153}
1154
1155Stmt::child_range CapturedStmt::children() {
Tareq A. Siraj6dfa25a2013-04-16 19:37:38 +00001156 // Children are captured field initilizers.
1157 return child_range(getStoredStmts(), getStoredStmts() + NumCaptures);
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001158}
1159
1160bool CapturedStmt::capturesVariable(const VarDecl *Var) const {
Aaron Ballmanc656303a2014-03-14 18:08:33 +00001161 for (const auto &I : captures()) {
1162 if (!I.capturesVariable())
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001163 continue;
1164
1165 // This does not handle variable redeclarations. This should be
1166 // extended to capture variables with redeclarations, for example
1167 // a thread-private variable in OpenMP.
Aaron Ballmanc656303a2014-03-14 18:08:33 +00001168 if (I.getCapturedVar() == Var)
Tareq A. Siraj24110cc2013-04-16 18:53:08 +00001169 return true;
1170 }
1171
1172 return false;
1173}
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001174
Alexey Bataev758e55e2013-09-06 18:03:48 +00001175StmtRange OMPClause::children() {
1176 switch(getClauseKind()) {
1177 default : break;
1178#define OPENMP_CLAUSE(Name, Class) \
1179 case OMPC_ ## Name : return static_cast<Class *>(this)->children();
1180#include "clang/Basic/OpenMPKinds.def"
1181 }
1182 llvm_unreachable("unknown OMPClause");
1183}
1184
Alexey Bataev03b340a2014-10-21 03:16:40 +00001185void OMPPrivateClause::setPrivateCopies(ArrayRef<Expr *> VL) {
1186 assert(VL.size() == varlist_size() &&
1187 "Number of private copies is not the same as the preallocated buffer");
1188 std::copy(VL.begin(), VL.end(), varlist_end());
1189}
1190
1191OMPPrivateClause *
1192OMPPrivateClause::Create(const ASTContext &C, SourceLocation StartLoc,
1193 SourceLocation LParenLoc, SourceLocation EndLoc,
1194 ArrayRef<Expr *> VL, ArrayRef<Expr *> PrivateVL) {
1195 // Allocate space for private variables and initializer expressions.
Alexey Bataev13193812014-02-25 11:25:38 +00001196 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPPrivateClause),
1197 llvm::alignOf<Expr *>()) +
Alexey Bataev03b340a2014-10-21 03:16:40 +00001198 2 * sizeof(Expr *) * VL.size());
1199 OMPPrivateClause *Clause =
1200 new (Mem) OMPPrivateClause(StartLoc, LParenLoc, EndLoc, VL.size());
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001201 Clause->setVarRefs(VL);
Alexey Bataev03b340a2014-10-21 03:16:40 +00001202 Clause->setPrivateCopies(PrivateVL);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001203 return Clause;
1204}
1205
Craig Toppere6960e22013-08-22 05:28:54 +00001206OMPPrivateClause *OMPPrivateClause::CreateEmpty(const ASTContext &C,
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001207 unsigned N) {
Alexey Bataev13193812014-02-25 11:25:38 +00001208 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPPrivateClause),
1209 llvm::alignOf<Expr *>()) +
Alexey Bataev03b340a2014-10-21 03:16:40 +00001210 2 * sizeof(Expr *) * N);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001211 return new (Mem) OMPPrivateClause(N);
1212}
1213
Alexey Bataev4a5bb772014-10-08 14:01:46 +00001214void OMPFirstprivateClause::setPrivateCopies(ArrayRef<Expr *> VL) {
1215 assert(VL.size() == varlist_size() &&
1216 "Number of private copies is not the same as the preallocated buffer");
1217 std::copy(VL.begin(), VL.end(), varlist_end());
1218}
1219
1220void OMPFirstprivateClause::setInits(ArrayRef<Expr *> VL) {
1221 assert(VL.size() == varlist_size() &&
1222 "Number of inits is not the same as the preallocated buffer");
1223 std::copy(VL.begin(), VL.end(), getPrivateCopies().end());
1224}
1225
1226OMPFirstprivateClause *
1227OMPFirstprivateClause::Create(const ASTContext &C, SourceLocation StartLoc,
1228 SourceLocation LParenLoc, SourceLocation EndLoc,
1229 ArrayRef<Expr *> VL, ArrayRef<Expr *> PrivateVL,
1230 ArrayRef<Expr *> InitVL) {
Alexey Bataev13193812014-02-25 11:25:38 +00001231 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPFirstprivateClause),
1232 llvm::alignOf<Expr *>()) +
Alexey Bataev4a5bb772014-10-08 14:01:46 +00001233 3 * sizeof(Expr *) * VL.size());
1234 OMPFirstprivateClause *Clause =
1235 new (Mem) OMPFirstprivateClause(StartLoc, LParenLoc, EndLoc, VL.size());
Alexey Bataevd5af8e42013-10-01 05:32:34 +00001236 Clause->setVarRefs(VL);
Alexey Bataev4a5bb772014-10-08 14:01:46 +00001237 Clause->setPrivateCopies(PrivateVL);
1238 Clause->setInits(InitVL);
Alexey Bataevd5af8e42013-10-01 05:32:34 +00001239 return Clause;
1240}
1241
1242OMPFirstprivateClause *OMPFirstprivateClause::CreateEmpty(const ASTContext &C,
1243 unsigned N) {
Alexey Bataev13193812014-02-25 11:25:38 +00001244 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPFirstprivateClause),
1245 llvm::alignOf<Expr *>()) +
Alexey Bataev4a5bb772014-10-08 14:01:46 +00001246 3 * sizeof(Expr *) * N);
Alexey Bataevd5af8e42013-10-01 05:32:34 +00001247 return new (Mem) OMPFirstprivateClause(N);
1248}
1249
Alexander Musman1bb328c2014-06-04 13:06:39 +00001250OMPLastprivateClause *OMPLastprivateClause::Create(const ASTContext &C,
1251 SourceLocation StartLoc,
1252 SourceLocation LParenLoc,
1253 SourceLocation EndLoc,
1254 ArrayRef<Expr *> VL) {
1255 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLastprivateClause),
1256 llvm::alignOf<Expr *>()) +
1257 sizeof(Expr *) * VL.size());
1258 OMPLastprivateClause *Clause =
1259 new (Mem) OMPLastprivateClause(StartLoc, LParenLoc, EndLoc, VL.size());
1260 Clause->setVarRefs(VL);
1261 return Clause;
1262}
1263
1264OMPLastprivateClause *OMPLastprivateClause::CreateEmpty(const ASTContext &C,
1265 unsigned N) {
1266 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLastprivateClause),
1267 llvm::alignOf<Expr *>()) +
1268 sizeof(Expr *) * N);
1269 return new (Mem) OMPLastprivateClause(N);
1270}
1271
Alexey Bataev758e55e2013-09-06 18:03:48 +00001272OMPSharedClause *OMPSharedClause::Create(const ASTContext &C,
1273 SourceLocation StartLoc,
1274 SourceLocation LParenLoc,
1275 SourceLocation EndLoc,
1276 ArrayRef<Expr *> VL) {
Alexey Bataev13193812014-02-25 11:25:38 +00001277 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPSharedClause),
1278 llvm::alignOf<Expr *>()) +
1279 sizeof(Expr *) * VL.size());
Alexey Bataev758e55e2013-09-06 18:03:48 +00001280 OMPSharedClause *Clause = new (Mem) OMPSharedClause(StartLoc, LParenLoc,
1281 EndLoc, VL.size());
1282 Clause->setVarRefs(VL);
1283 return Clause;
1284}
1285
1286OMPSharedClause *OMPSharedClause::CreateEmpty(const ASTContext &C,
1287 unsigned N) {
Alexey Bataev13193812014-02-25 11:25:38 +00001288 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPSharedClause),
1289 llvm::alignOf<Expr *>()) +
1290 sizeof(Expr *) * N);
Alexey Bataev758e55e2013-09-06 18:03:48 +00001291 return new (Mem) OMPSharedClause(N);
1292}
1293
Alexander Musman3276a272015-03-21 10:12:56 +00001294void OMPLinearClause::setInits(ArrayRef<Expr *> IL) {
1295 assert(IL.size() == varlist_size() &&
1296 "Number of inits is not the same as the preallocated buffer");
1297 std::copy(IL.begin(), IL.end(), varlist_end());
1298}
1299
1300void OMPLinearClause::setUpdates(ArrayRef<Expr *> UL) {
1301 assert(UL.size() == varlist_size() &&
1302 "Number of updates is not the same as the preallocated buffer");
1303 std::copy(UL.begin(), UL.end(), getInits().end());
1304}
1305
1306void OMPLinearClause::setFinals(ArrayRef<Expr *> FL) {
1307 assert(FL.size() == varlist_size() &&
1308 "Number of final updates is not the same as the preallocated buffer");
1309 std::copy(FL.begin(), FL.end(), getUpdates().end());
1310}
1311
1312OMPLinearClause *
1313OMPLinearClause::Create(const ASTContext &C, SourceLocation StartLoc,
1314 SourceLocation LParenLoc, SourceLocation ColonLoc,
1315 SourceLocation EndLoc, ArrayRef<Expr *> VL,
1316 ArrayRef<Expr *> IL, Expr *Step, Expr *CalcStep) {
1317 // Allocate space for 4 lists (Vars, Inits, Updates, Finals) and 2 expressions
1318 // (Step and CalcStep).
Alexander Musman8dba6642014-04-22 13:09:42 +00001319 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLinearClause),
1320 llvm::alignOf<Expr *>()) +
Alexander Musman3276a272015-03-21 10:12:56 +00001321 (4 * VL.size() + 2) * sizeof(Expr *));
Alexander Musman8dba6642014-04-22 13:09:42 +00001322 OMPLinearClause *Clause = new (Mem)
1323 OMPLinearClause(StartLoc, LParenLoc, ColonLoc, EndLoc, VL.size());
1324 Clause->setVarRefs(VL);
Alexander Musman3276a272015-03-21 10:12:56 +00001325 Clause->setInits(IL);
1326 // Fill update and final expressions with zeroes, they are provided later,
1327 // after the directive construction.
1328 std::fill(Clause->getInits().end(), Clause->getInits().end() + VL.size(),
1329 nullptr);
1330 std::fill(Clause->getUpdates().end(), Clause->getUpdates().end() + VL.size(),
1331 nullptr);
Alexander Musman8dba6642014-04-22 13:09:42 +00001332 Clause->setStep(Step);
Alexander Musman3276a272015-03-21 10:12:56 +00001333 Clause->setCalcStep(CalcStep);
Alexander Musman8dba6642014-04-22 13:09:42 +00001334 return Clause;
1335}
1336
1337OMPLinearClause *OMPLinearClause::CreateEmpty(const ASTContext &C,
1338 unsigned NumVars) {
Alexander Musman3276a272015-03-21 10:12:56 +00001339 // Allocate space for 4 lists (Vars, Inits, Updates, Finals) and 2 expressions
1340 // (Step and CalcStep).
Alexander Musman8dba6642014-04-22 13:09:42 +00001341 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLinearClause),
1342 llvm::alignOf<Expr *>()) +
Alexander Musman3276a272015-03-21 10:12:56 +00001343 (4 * NumVars + 2) * sizeof(Expr *));
Alexander Musman8dba6642014-04-22 13:09:42 +00001344 return new (Mem) OMPLinearClause(NumVars);
1345}
1346
Alexander Musmanf0d76e72014-05-29 14:36:25 +00001347OMPAlignedClause *
1348OMPAlignedClause::Create(const ASTContext &C, SourceLocation StartLoc,
1349 SourceLocation LParenLoc, SourceLocation ColonLoc,
1350 SourceLocation EndLoc, ArrayRef<Expr *> VL, Expr *A) {
1351 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPAlignedClause),
1352 llvm::alignOf<Expr *>()) +
1353 sizeof(Expr *) * (VL.size() + 1));
1354 OMPAlignedClause *Clause = new (Mem)
1355 OMPAlignedClause(StartLoc, LParenLoc, ColonLoc, EndLoc, VL.size());
1356 Clause->setVarRefs(VL);
1357 Clause->setAlignment(A);
1358 return Clause;
1359}
1360
1361OMPAlignedClause *OMPAlignedClause::CreateEmpty(const ASTContext &C,
1362 unsigned NumVars) {
1363 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPAlignedClause),
1364 llvm::alignOf<Expr *>()) +
1365 sizeof(Expr *) * (NumVars + 1));
1366 return new (Mem) OMPAlignedClause(NumVars);
1367}
1368
Alexey Bataevd48bcd82014-03-31 03:36:38 +00001369OMPCopyinClause *OMPCopyinClause::Create(const ASTContext &C,
1370 SourceLocation StartLoc,
1371 SourceLocation LParenLoc,
1372 SourceLocation EndLoc,
1373 ArrayRef<Expr *> VL) {
1374 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyinClause),
1375 llvm::alignOf<Expr *>()) +
1376 sizeof(Expr *) * VL.size());
1377 OMPCopyinClause *Clause = new (Mem) OMPCopyinClause(StartLoc, LParenLoc,
1378 EndLoc, VL.size());
1379 Clause->setVarRefs(VL);
1380 return Clause;
1381}
1382
1383OMPCopyinClause *OMPCopyinClause::CreateEmpty(const ASTContext &C,
1384 unsigned N) {
1385 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyinClause),
1386 llvm::alignOf<Expr *>()) +
1387 sizeof(Expr *) * N);
1388 return new (Mem) OMPCopyinClause(N);
1389}
1390
Alexey Bataeva63048e2015-03-23 06:18:07 +00001391void OMPCopyprivateClause::setSourceExprs(ArrayRef<Expr *> SrcExprs) {
1392 assert(SrcExprs.size() == varlist_size() && "Number of source expressions is "
1393 "not the same as the "
1394 "preallocated buffer");
1395 std::copy(SrcExprs.begin(), SrcExprs.end(), varlist_end());
1396}
1397
1398void OMPCopyprivateClause::setDestinationExprs(ArrayRef<Expr *> DstExprs) {
1399 assert(DstExprs.size() == varlist_size() && "Number of destination "
1400 "expressions is not the same as "
1401 "the preallocated buffer");
1402 std::copy(DstExprs.begin(), DstExprs.end(), getSourceExprs().end());
1403}
1404
1405void OMPCopyprivateClause::setAssignmentOps(ArrayRef<Expr *> AssignmentOps) {
1406 assert(AssignmentOps.size() == varlist_size() &&
1407 "Number of assignment expressions is not the same as the preallocated "
1408 "buffer");
1409 std::copy(AssignmentOps.begin(), AssignmentOps.end(),
1410 getDestinationExprs().end());
1411}
1412
1413OMPCopyprivateClause *OMPCopyprivateClause::Create(
1414 const ASTContext &C, SourceLocation StartLoc, SourceLocation LParenLoc,
1415 SourceLocation EndLoc, ArrayRef<Expr *> VL, ArrayRef<Expr *> SrcExprs,
1416 ArrayRef<Expr *> DstExprs, ArrayRef<Expr *> AssignmentOps) {
Alexey Bataevbae9a792014-06-27 10:37:06 +00001417 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyprivateClause),
1418 llvm::alignOf<Expr *>()) +
Alexey Bataeva63048e2015-03-23 06:18:07 +00001419 4 * sizeof(Expr *) * VL.size());
Alexey Bataevbae9a792014-06-27 10:37:06 +00001420 OMPCopyprivateClause *Clause =
1421 new (Mem) OMPCopyprivateClause(StartLoc, LParenLoc, EndLoc, VL.size());
1422 Clause->setVarRefs(VL);
Alexey Bataeva63048e2015-03-23 06:18:07 +00001423 Clause->setSourceExprs(SrcExprs);
1424 Clause->setDestinationExprs(DstExprs);
1425 Clause->setAssignmentOps(AssignmentOps);
Alexey Bataevbae9a792014-06-27 10:37:06 +00001426 return Clause;
1427}
1428
1429OMPCopyprivateClause *OMPCopyprivateClause::CreateEmpty(const ASTContext &C,
1430 unsigned N) {
1431 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyprivateClause),
1432 llvm::alignOf<Expr *>()) +
Alexey Bataeva63048e2015-03-23 06:18:07 +00001433 4 * sizeof(Expr *) * N);
Alexey Bataevbae9a792014-06-27 10:37:06 +00001434 return new (Mem) OMPCopyprivateClause(N);
1435}
1436
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001437void OMPExecutableDirective::setClauses(ArrayRef<OMPClause *> Clauses) {
Alexander Musmanfd2b7592014-03-27 15:14:18 +00001438 assert(Clauses.size() == getNumClauses() &&
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001439 "Number of clauses is not the same as the preallocated buffer");
Alexander Musmanfd2b7592014-03-27 15:14:18 +00001440 std::copy(Clauses.begin(), Clauses.end(), getClauses().begin());
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001441}
1442
Alexander Musmana5f070a2014-10-01 06:03:56 +00001443void OMPLoopDirective::setCounters(ArrayRef<Expr *> A) {
1444 assert(A.size() == getCollapsedNumber() &&
1445 "Number of loop counters is not the same as the collapsed number");
1446 std::copy(A.begin(), A.end(), getCounters().begin());
1447}
1448
1449void OMPLoopDirective::setUpdates(ArrayRef<Expr *> A) {
1450 assert(A.size() == getCollapsedNumber() &&
1451 "Number of counter updates is not the same as the collapsed number");
1452 std::copy(A.begin(), A.end(), getUpdates().begin());
1453}
1454
1455void OMPLoopDirective::setFinals(ArrayRef<Expr *> A) {
1456 assert(A.size() == getCollapsedNumber() &&
1457 "Number of counter finals is not the same as the collapsed number");
1458 std::copy(A.begin(), A.end(), getFinals().begin());
1459}
1460
Alexey Bataevc5e02582014-06-16 07:08:35 +00001461OMPReductionClause *OMPReductionClause::Create(
1462 const ASTContext &C, SourceLocation StartLoc, SourceLocation LParenLoc,
1463 SourceLocation EndLoc, SourceLocation ColonLoc, ArrayRef<Expr *> VL,
1464 NestedNameSpecifierLoc QualifierLoc, const DeclarationNameInfo &NameInfo) {
1465 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPReductionClause),
1466 llvm::alignOf<Expr *>()) +
1467 sizeof(Expr *) * VL.size());
1468 OMPReductionClause *Clause = new (Mem) OMPReductionClause(
1469 StartLoc, LParenLoc, EndLoc, ColonLoc, VL.size(), QualifierLoc, NameInfo);
1470 Clause->setVarRefs(VL);
1471 return Clause;
1472}
1473
1474OMPReductionClause *OMPReductionClause::CreateEmpty(const ASTContext &C,
1475 unsigned N) {
1476 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPReductionClause),
1477 llvm::alignOf<Expr *>()) +
1478 sizeof(Expr *) * N);
1479 return new (Mem) OMPReductionClause(N);
1480}
1481
Alexey Bataev6125da92014-07-21 11:26:11 +00001482OMPFlushClause *OMPFlushClause::Create(const ASTContext &C,
1483 SourceLocation StartLoc,
1484 SourceLocation LParenLoc,
1485 SourceLocation EndLoc,
1486 ArrayRef<Expr *> VL) {
1487 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPFlushClause),
1488 llvm::alignOf<Expr *>()) +
1489 sizeof(Expr *) * VL.size());
1490 OMPFlushClause *Clause =
1491 new (Mem) OMPFlushClause(StartLoc, LParenLoc, EndLoc, VL.size());
1492 Clause->setVarRefs(VL);
1493 return Clause;
1494}
1495
1496OMPFlushClause *OMPFlushClause::CreateEmpty(const ASTContext &C, unsigned N) {
1497 void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPFlushClause),
1498 llvm::alignOf<Expr *>()) +
1499 sizeof(Expr *) * N);
1500 return new (Mem) OMPFlushClause(N);
1501}
1502
Alexey Bataevd74d0602014-10-13 06:02:40 +00001503const OMPClause *
1504OMPExecutableDirective::getSingleClause(OpenMPClauseKind K) const {
1505 auto ClauseFilter =
1506 [=](const OMPClause *C) -> bool { return C->getClauseKind() == K; };
1507 OMPExecutableDirective::filtered_clause_iterator<decltype(ClauseFilter)> I(
1508 clauses(), ClauseFilter);
1509
1510 if (I) {
1511 auto *Clause = *I;
1512 assert(!++I && "There are at least 2 clauses of the specified kind");
1513 return Clause;
1514 }
1515 return nullptr;
1516}
1517
Craig Toppercee61332013-08-21 04:01:01 +00001518OMPParallelDirective *OMPParallelDirective::Create(
Craig Toppere6960e22013-08-22 05:28:54 +00001519 const ASTContext &C,
Craig Toppercee61332013-08-21 04:01:01 +00001520 SourceLocation StartLoc,
1521 SourceLocation EndLoc,
1522 ArrayRef<OMPClause *> Clauses,
1523 Stmt *AssociatedStmt) {
Alexey Bataev13193812014-02-25 11:25:38 +00001524 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelDirective),
1525 llvm::alignOf<OMPClause *>());
1526 void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1527 sizeof(Stmt *));
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001528 OMPParallelDirective *Dir = new (Mem) OMPParallelDirective(StartLoc, EndLoc,
1529 Clauses.size());
1530 Dir->setClauses(Clauses);
1531 Dir->setAssociatedStmt(AssociatedStmt);
1532 return Dir;
1533}
1534
Craig Toppere6960e22013-08-22 05:28:54 +00001535OMPParallelDirective *OMPParallelDirective::CreateEmpty(const ASTContext &C,
Alexey Bataev1b59ab52014-02-27 08:29:12 +00001536 unsigned NumClauses,
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001537 EmptyShell) {
Alexey Bataev13193812014-02-25 11:25:38 +00001538 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelDirective),
1539 llvm::alignOf<OMPClause *>());
Alexey Bataev1b59ab52014-02-27 08:29:12 +00001540 void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1541 sizeof(Stmt *));
1542 return new (Mem) OMPParallelDirective(NumClauses);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001543}
Alexey Bataev1b59ab52014-02-27 08:29:12 +00001544
Alexey Bataevabfc0692014-06-25 06:52:00 +00001545OMPSimdDirective *
1546OMPSimdDirective::Create(const ASTContext &C, SourceLocation StartLoc,
1547 SourceLocation EndLoc, unsigned CollapsedNum,
Alexander Musmana5f070a2014-10-01 06:03:56 +00001548 ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
Alexander Musmanc6388682014-12-15 07:07:06 +00001549 const HelperExprs &Exprs) {
Alexey Bataev1b59ab52014-02-27 08:29:12 +00001550 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSimdDirective),
1551 llvm::alignOf<OMPClause *>());
Alexander Musmanc6388682014-12-15 07:07:06 +00001552 void *Mem =
1553 C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1554 sizeof(Stmt *) * numLoopChildren(CollapsedNum, OMPD_simd));
Alexey Bataevabfc0692014-06-25 06:52:00 +00001555 OMPSimdDirective *Dir = new (Mem)
1556 OMPSimdDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
Alexey Bataev1b59ab52014-02-27 08:29:12 +00001557 Dir->setClauses(Clauses);
1558 Dir->setAssociatedStmt(AssociatedStmt);
Alexander Musmanc6388682014-12-15 07:07:06 +00001559 Dir->setIterationVariable(Exprs.IterationVarRef);
1560 Dir->setLastIteration(Exprs.LastIteration);
1561 Dir->setCalcLastIteration(Exprs.CalcLastIteration);
1562 Dir->setPreCond(Exprs.PreCond);
1563 Dir->setCond(Exprs.Cond, Exprs.SeparatedCond);
1564 Dir->setInit(Exprs.Init);
1565 Dir->setInc(Exprs.Inc);
1566 Dir->setCounters(Exprs.Counters);
1567 Dir->setUpdates(Exprs.Updates);
1568 Dir->setFinals(Exprs.Finals);
Alexey Bataev1b59ab52014-02-27 08:29:12 +00001569 return Dir;
1570}
1571
1572OMPSimdDirective *OMPSimdDirective::CreateEmpty(const ASTContext &C,
1573 unsigned NumClauses,
1574 unsigned CollapsedNum,
1575 EmptyShell) {
1576 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSimdDirective),
1577 llvm::alignOf<OMPClause *>());
Alexander Musmanc6388682014-12-15 07:07:06 +00001578 void *Mem =
1579 C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1580 sizeof(Stmt *) * numLoopChildren(CollapsedNum, OMPD_simd));
Alexey Bataev1b59ab52014-02-27 08:29:12 +00001581 return new (Mem) OMPSimdDirective(CollapsedNum, NumClauses);
1582}
1583
Alexey Bataevabfc0692014-06-25 06:52:00 +00001584OMPForDirective *
1585OMPForDirective::Create(const ASTContext &C, SourceLocation StartLoc,
1586 SourceLocation EndLoc, unsigned CollapsedNum,
Alexander Musmana5f070a2014-10-01 06:03:56 +00001587 ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
Alexander Musmanc6388682014-12-15 07:07:06 +00001588 const HelperExprs &Exprs) {
Alexey Bataevf29276e2014-06-18 04:14:57 +00001589 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPForDirective),
1590 llvm::alignOf<OMPClause *>());
Alexander Musmanc6388682014-12-15 07:07:06 +00001591 void *Mem =
1592 C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1593 sizeof(Stmt *) * numLoopChildren(CollapsedNum, OMPD_for));
Alexey Bataevf29276e2014-06-18 04:14:57 +00001594 OMPForDirective *Dir =
Alexey Bataevabfc0692014-06-25 06:52:00 +00001595 new (Mem) OMPForDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
Alexey Bataevf29276e2014-06-18 04:14:57 +00001596 Dir->setClauses(Clauses);
1597 Dir->setAssociatedStmt(AssociatedStmt);
Alexander Musmanc6388682014-12-15 07:07:06 +00001598 Dir->setIterationVariable(Exprs.IterationVarRef);
1599 Dir->setLastIteration(Exprs.LastIteration);
1600 Dir->setCalcLastIteration(Exprs.CalcLastIteration);
1601 Dir->setPreCond(Exprs.PreCond);
1602 Dir->setCond(Exprs.Cond, Exprs.SeparatedCond);
1603 Dir->setInit(Exprs.Init);
1604 Dir->setInc(Exprs.Inc);
1605 Dir->setIsLastIterVariable(Exprs.IL);
1606 Dir->setLowerBoundVariable(Exprs.LB);
1607 Dir->setUpperBoundVariable(Exprs.UB);
1608 Dir->setStrideVariable(Exprs.ST);
1609 Dir->setEnsureUpperBound(Exprs.EUB);
1610 Dir->setNextLowerBound(Exprs.NLB);
1611 Dir->setNextUpperBound(Exprs.NUB);
1612 Dir->setCounters(Exprs.Counters);
1613 Dir->setUpdates(Exprs.Updates);
1614 Dir->setFinals(Exprs.Finals);
Alexey Bataevf29276e2014-06-18 04:14:57 +00001615 return Dir;
1616}
1617
1618OMPForDirective *OMPForDirective::CreateEmpty(const ASTContext &C,
1619 unsigned NumClauses,
1620 unsigned CollapsedNum,
1621 EmptyShell) {
1622 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPForDirective),
1623 llvm::alignOf<OMPClause *>());
Alexander Musmanc6388682014-12-15 07:07:06 +00001624 void *Mem =
1625 C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1626 sizeof(Stmt *) * numLoopChildren(CollapsedNum, OMPD_for));
Alexey Bataevf29276e2014-06-18 04:14:57 +00001627 return new (Mem) OMPForDirective(CollapsedNum, NumClauses);
1628}
1629
Alexander Musmanc6388682014-12-15 07:07:06 +00001630OMPForSimdDirective *
1631OMPForSimdDirective::Create(const ASTContext &C, SourceLocation StartLoc,
1632 SourceLocation EndLoc, unsigned CollapsedNum,
1633 ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
1634 const HelperExprs &Exprs) {
Alexander Musmanf82886e2014-09-18 05:12:34 +00001635 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPForSimdDirective),
1636 llvm::alignOf<OMPClause *>());
Alexander Musmanc6388682014-12-15 07:07:06 +00001637 void *Mem =
1638 C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1639 sizeof(Stmt *) * numLoopChildren(CollapsedNum, OMPD_for_simd));
Alexander Musmanf82886e2014-09-18 05:12:34 +00001640 OMPForSimdDirective *Dir = new (Mem)
1641 OMPForSimdDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
1642 Dir->setClauses(Clauses);
1643 Dir->setAssociatedStmt(AssociatedStmt);
Alexander Musmanc6388682014-12-15 07:07:06 +00001644 Dir->setIterationVariable(Exprs.IterationVarRef);
1645 Dir->setLastIteration(Exprs.LastIteration);
1646 Dir->setCalcLastIteration(Exprs.CalcLastIteration);
1647 Dir->setPreCond(Exprs.PreCond);
1648 Dir->setCond(Exprs.Cond, Exprs.SeparatedCond);
1649 Dir->setInit(Exprs.Init);
1650 Dir->setInc(Exprs.Inc);
1651 Dir->setIsLastIterVariable(Exprs.IL);
1652 Dir->setLowerBoundVariable(Exprs.LB);
1653 Dir->setUpperBoundVariable(Exprs.UB);
1654 Dir->setStrideVariable(Exprs.ST);
1655 Dir->setEnsureUpperBound(Exprs.EUB);
1656 Dir->setNextLowerBound(Exprs.NLB);
1657 Dir->setNextUpperBound(Exprs.NUB);
1658 Dir->setCounters(Exprs.Counters);
1659 Dir->setUpdates(Exprs.Updates);
1660 Dir->setFinals(Exprs.Finals);
Alexander Musmanf82886e2014-09-18 05:12:34 +00001661 return Dir;
1662}
1663
1664OMPForSimdDirective *OMPForSimdDirective::CreateEmpty(const ASTContext &C,
1665 unsigned NumClauses,
1666 unsigned CollapsedNum,
1667 EmptyShell) {
1668 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPForSimdDirective),
1669 llvm::alignOf<OMPClause *>());
Alexander Musmanc6388682014-12-15 07:07:06 +00001670 void *Mem =
1671 C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1672 sizeof(Stmt *) * numLoopChildren(CollapsedNum, OMPD_for_simd));
Alexander Musmanf82886e2014-09-18 05:12:34 +00001673 return new (Mem) OMPForSimdDirective(CollapsedNum, NumClauses);
1674}
1675
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00001676OMPSectionsDirective *OMPSectionsDirective::Create(
1677 const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
1678 ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
1679 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionsDirective),
1680 llvm::alignOf<OMPClause *>());
1681 void *Mem =
1682 C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1683 OMPSectionsDirective *Dir =
1684 new (Mem) OMPSectionsDirective(StartLoc, EndLoc, Clauses.size());
1685 Dir->setClauses(Clauses);
1686 Dir->setAssociatedStmt(AssociatedStmt);
1687 return Dir;
1688}
1689
1690OMPSectionsDirective *OMPSectionsDirective::CreateEmpty(const ASTContext &C,
1691 unsigned NumClauses,
1692 EmptyShell) {
1693 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionsDirective),
1694 llvm::alignOf<OMPClause *>());
1695 void *Mem =
1696 C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1697 return new (Mem) OMPSectionsDirective(NumClauses);
1698}
1699
Alexey Bataev1e0498a2014-06-26 08:21:58 +00001700OMPSectionDirective *OMPSectionDirective::Create(const ASTContext &C,
1701 SourceLocation StartLoc,
1702 SourceLocation EndLoc,
1703 Stmt *AssociatedStmt) {
1704 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionsDirective),
1705 llvm::alignOf<Stmt *>());
1706 void *Mem = C.Allocate(Size + sizeof(Stmt *));
1707 OMPSectionDirective *Dir = new (Mem) OMPSectionDirective(StartLoc, EndLoc);
1708 Dir->setAssociatedStmt(AssociatedStmt);
1709 return Dir;
1710}
1711
1712OMPSectionDirective *OMPSectionDirective::CreateEmpty(const ASTContext &C,
1713 EmptyShell) {
1714 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionDirective),
1715 llvm::alignOf<Stmt *>());
1716 void *Mem = C.Allocate(Size + sizeof(Stmt *));
1717 return new (Mem) OMPSectionDirective();
1718}
1719
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00001720OMPSingleDirective *OMPSingleDirective::Create(const ASTContext &C,
1721 SourceLocation StartLoc,
1722 SourceLocation EndLoc,
1723 ArrayRef<OMPClause *> Clauses,
1724 Stmt *AssociatedStmt) {
1725 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSingleDirective),
1726 llvm::alignOf<OMPClause *>());
1727 void *Mem =
1728 C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1729 OMPSingleDirective *Dir =
1730 new (Mem) OMPSingleDirective(StartLoc, EndLoc, Clauses.size());
1731 Dir->setClauses(Clauses);
1732 Dir->setAssociatedStmt(AssociatedStmt);
1733 return Dir;
1734}
1735
1736OMPSingleDirective *OMPSingleDirective::CreateEmpty(const ASTContext &C,
1737 unsigned NumClauses,
1738 EmptyShell) {
1739 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSingleDirective),
1740 llvm::alignOf<OMPClause *>());
1741 void *Mem =
1742 C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1743 return new (Mem) OMPSingleDirective(NumClauses);
1744}
1745
Alexander Musman80c22892014-07-17 08:54:58 +00001746OMPMasterDirective *OMPMasterDirective::Create(const ASTContext &C,
1747 SourceLocation StartLoc,
1748 SourceLocation EndLoc,
1749 Stmt *AssociatedStmt) {
1750 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPMasterDirective),
1751 llvm::alignOf<Stmt *>());
1752 void *Mem = C.Allocate(Size + sizeof(Stmt *));
1753 OMPMasterDirective *Dir = new (Mem) OMPMasterDirective(StartLoc, EndLoc);
1754 Dir->setAssociatedStmt(AssociatedStmt);
1755 return Dir;
1756}
1757
1758OMPMasterDirective *OMPMasterDirective::CreateEmpty(const ASTContext &C,
1759 EmptyShell) {
1760 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPMasterDirective),
1761 llvm::alignOf<Stmt *>());
1762 void *Mem = C.Allocate(Size + sizeof(Stmt *));
1763 return new (Mem) OMPMasterDirective();
1764}
1765
Alexander Musmand9ed09f2014-07-21 09:42:05 +00001766OMPCriticalDirective *OMPCriticalDirective::Create(
1767 const ASTContext &C, const DeclarationNameInfo &Name,
1768 SourceLocation StartLoc, SourceLocation EndLoc, Stmt *AssociatedStmt) {
1769 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPCriticalDirective),
1770 llvm::alignOf<Stmt *>());
1771 void *Mem = C.Allocate(Size + sizeof(Stmt *));
1772 OMPCriticalDirective *Dir =
1773 new (Mem) OMPCriticalDirective(Name, StartLoc, EndLoc);
1774 Dir->setAssociatedStmt(AssociatedStmt);
1775 return Dir;
1776}
1777
1778OMPCriticalDirective *OMPCriticalDirective::CreateEmpty(const ASTContext &C,
1779 EmptyShell) {
1780 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPCriticalDirective),
1781 llvm::alignOf<Stmt *>());
1782 void *Mem = C.Allocate(Size + sizeof(Stmt *));
1783 return new (Mem) OMPCriticalDirective();
1784}
1785
Alexander Musmana5f070a2014-10-01 06:03:56 +00001786OMPParallelForDirective *OMPParallelForDirective::Create(
1787 const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
1788 unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
Alexander Musmanc6388682014-12-15 07:07:06 +00001789 const HelperExprs &Exprs) {
Alexey Bataev4acb8592014-07-07 13:01:15 +00001790 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelForDirective),
1791 llvm::alignOf<OMPClause *>());
Alexander Musmana5f070a2014-10-01 06:03:56 +00001792 void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
Alexander Musmanc6388682014-12-15 07:07:06 +00001793 sizeof(Stmt *) *
1794 numLoopChildren(CollapsedNum, OMPD_parallel_for));
Alexey Bataev4acb8592014-07-07 13:01:15 +00001795 OMPParallelForDirective *Dir = new (Mem)
1796 OMPParallelForDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
1797 Dir->setClauses(Clauses);
1798 Dir->setAssociatedStmt(AssociatedStmt);
Alexander Musmanc6388682014-12-15 07:07:06 +00001799 Dir->setIterationVariable(Exprs.IterationVarRef);
1800 Dir->setLastIteration(Exprs.LastIteration);
1801 Dir->setCalcLastIteration(Exprs.CalcLastIteration);
1802 Dir->setPreCond(Exprs.PreCond);
1803 Dir->setCond(Exprs.Cond, Exprs.SeparatedCond);
1804 Dir->setInit(Exprs.Init);
1805 Dir->setInc(Exprs.Inc);
1806 Dir->setIsLastIterVariable(Exprs.IL);
1807 Dir->setLowerBoundVariable(Exprs.LB);
1808 Dir->setUpperBoundVariable(Exprs.UB);
1809 Dir->setStrideVariable(Exprs.ST);
1810 Dir->setEnsureUpperBound(Exprs.EUB);
1811 Dir->setNextLowerBound(Exprs.NLB);
1812 Dir->setNextUpperBound(Exprs.NUB);
1813 Dir->setCounters(Exprs.Counters);
1814 Dir->setUpdates(Exprs.Updates);
1815 Dir->setFinals(Exprs.Finals);
Alexey Bataev4acb8592014-07-07 13:01:15 +00001816 return Dir;
1817}
1818
1819OMPParallelForDirective *
1820OMPParallelForDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
1821 unsigned CollapsedNum, EmptyShell) {
1822 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelForDirective),
1823 llvm::alignOf<OMPClause *>());
Alexander Musmana5f070a2014-10-01 06:03:56 +00001824 void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
Alexander Musmanc6388682014-12-15 07:07:06 +00001825 sizeof(Stmt *) *
1826 numLoopChildren(CollapsedNum, OMPD_parallel_for));
Alexey Bataev4acb8592014-07-07 13:01:15 +00001827 return new (Mem) OMPParallelForDirective(CollapsedNum, NumClauses);
1828}
1829
Alexander Musmane4e893b2014-09-23 09:33:00 +00001830OMPParallelForSimdDirective *OMPParallelForSimdDirective::Create(
1831 const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
Alexander Musmana5f070a2014-10-01 06:03:56 +00001832 unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
Alexander Musmanc6388682014-12-15 07:07:06 +00001833 const HelperExprs &Exprs) {
Alexander Musmane4e893b2014-09-23 09:33:00 +00001834 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelForSimdDirective),
1835 llvm::alignOf<OMPClause *>());
Alexander Musmanc6388682014-12-15 07:07:06 +00001836 void *Mem = C.Allocate(
1837 Size + sizeof(OMPClause *) * Clauses.size() +
1838 sizeof(Stmt *) * numLoopChildren(CollapsedNum, OMPD_parallel_for_simd));
Alexander Musmane4e893b2014-09-23 09:33:00 +00001839 OMPParallelForSimdDirective *Dir = new (Mem) OMPParallelForSimdDirective(
1840 StartLoc, EndLoc, CollapsedNum, Clauses.size());
1841 Dir->setClauses(Clauses);
1842 Dir->setAssociatedStmt(AssociatedStmt);
Alexander Musmanc6388682014-12-15 07:07:06 +00001843 Dir->setIterationVariable(Exprs.IterationVarRef);
1844 Dir->setLastIteration(Exprs.LastIteration);
1845 Dir->setCalcLastIteration(Exprs.CalcLastIteration);
1846 Dir->setPreCond(Exprs.PreCond);
1847 Dir->setCond(Exprs.Cond, Exprs.SeparatedCond);
1848 Dir->setInit(Exprs.Init);
1849 Dir->setInc(Exprs.Inc);
1850 Dir->setIsLastIterVariable(Exprs.IL);
1851 Dir->setLowerBoundVariable(Exprs.LB);
1852 Dir->setUpperBoundVariable(Exprs.UB);
1853 Dir->setStrideVariable(Exprs.ST);
1854 Dir->setEnsureUpperBound(Exprs.EUB);
1855 Dir->setNextLowerBound(Exprs.NLB);
1856 Dir->setNextUpperBound(Exprs.NUB);
1857 Dir->setCounters(Exprs.Counters);
1858 Dir->setUpdates(Exprs.Updates);
1859 Dir->setFinals(Exprs.Finals);
Alexander Musmane4e893b2014-09-23 09:33:00 +00001860 return Dir;
1861}
1862
1863OMPParallelForSimdDirective *
1864OMPParallelForSimdDirective::CreateEmpty(const ASTContext &C,
1865 unsigned NumClauses,
1866 unsigned CollapsedNum, EmptyShell) {
1867 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelForSimdDirective),
1868 llvm::alignOf<OMPClause *>());
Alexander Musmanc6388682014-12-15 07:07:06 +00001869 void *Mem = C.Allocate(
1870 Size + sizeof(OMPClause *) * NumClauses +
1871 sizeof(Stmt *) * numLoopChildren(CollapsedNum, OMPD_parallel_for_simd));
Alexander Musmane4e893b2014-09-23 09:33:00 +00001872 return new (Mem) OMPParallelForSimdDirective(CollapsedNum, NumClauses);
1873}
1874
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00001875OMPParallelSectionsDirective *OMPParallelSectionsDirective::Create(
1876 const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
1877 ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
1878 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelSectionsDirective),
1879 llvm::alignOf<OMPClause *>());
1880 void *Mem =
1881 C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1882 OMPParallelSectionsDirective *Dir =
1883 new (Mem) OMPParallelSectionsDirective(StartLoc, EndLoc, Clauses.size());
1884 Dir->setClauses(Clauses);
1885 Dir->setAssociatedStmt(AssociatedStmt);
1886 return Dir;
1887}
1888
1889OMPParallelSectionsDirective *
1890OMPParallelSectionsDirective::CreateEmpty(const ASTContext &C,
1891 unsigned NumClauses, EmptyShell) {
1892 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelSectionsDirective),
1893 llvm::alignOf<OMPClause *>());
1894 void *Mem =
1895 C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1896 return new (Mem) OMPParallelSectionsDirective(NumClauses);
1897}
1898
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00001899OMPTaskDirective *OMPTaskDirective::Create(const ASTContext &C,
1900 SourceLocation StartLoc,
1901 SourceLocation EndLoc,
1902 ArrayRef<OMPClause *> Clauses,
1903 Stmt *AssociatedStmt) {
1904 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPTaskDirective),
1905 llvm::alignOf<OMPClause *>());
1906 void *Mem =
1907 C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1908 OMPTaskDirective *Dir =
1909 new (Mem) OMPTaskDirective(StartLoc, EndLoc, Clauses.size());
1910 Dir->setClauses(Clauses);
1911 Dir->setAssociatedStmt(AssociatedStmt);
1912 return Dir;
1913}
1914
1915OMPTaskDirective *OMPTaskDirective::CreateEmpty(const ASTContext &C,
1916 unsigned NumClauses,
1917 EmptyShell) {
1918 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPTaskDirective),
1919 llvm::alignOf<OMPClause *>());
1920 void *Mem =
1921 C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1922 return new (Mem) OMPTaskDirective(NumClauses);
1923}
1924
Alexey Bataev68446b72014-07-18 07:47:19 +00001925OMPTaskyieldDirective *OMPTaskyieldDirective::Create(const ASTContext &C,
1926 SourceLocation StartLoc,
1927 SourceLocation EndLoc) {
1928 void *Mem = C.Allocate(sizeof(OMPTaskyieldDirective));
1929 OMPTaskyieldDirective *Dir =
1930 new (Mem) OMPTaskyieldDirective(StartLoc, EndLoc);
1931 return Dir;
1932}
1933
1934OMPTaskyieldDirective *OMPTaskyieldDirective::CreateEmpty(const ASTContext &C,
1935 EmptyShell) {
1936 void *Mem = C.Allocate(sizeof(OMPTaskyieldDirective));
1937 return new (Mem) OMPTaskyieldDirective();
1938}
1939
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00001940OMPBarrierDirective *OMPBarrierDirective::Create(const ASTContext &C,
1941 SourceLocation StartLoc,
1942 SourceLocation EndLoc) {
1943 void *Mem = C.Allocate(sizeof(OMPBarrierDirective));
1944 OMPBarrierDirective *Dir = new (Mem) OMPBarrierDirective(StartLoc, EndLoc);
1945 return Dir;
1946}
1947
1948OMPBarrierDirective *OMPBarrierDirective::CreateEmpty(const ASTContext &C,
1949 EmptyShell) {
1950 void *Mem = C.Allocate(sizeof(OMPBarrierDirective));
1951 return new (Mem) OMPBarrierDirective();
1952}
1953
Alexey Bataev2df347a2014-07-18 10:17:07 +00001954OMPTaskwaitDirective *OMPTaskwaitDirective::Create(const ASTContext &C,
1955 SourceLocation StartLoc,
1956 SourceLocation EndLoc) {
1957 void *Mem = C.Allocate(sizeof(OMPTaskwaitDirective));
1958 OMPTaskwaitDirective *Dir = new (Mem) OMPTaskwaitDirective(StartLoc, EndLoc);
1959 return Dir;
1960}
1961
1962OMPTaskwaitDirective *OMPTaskwaitDirective::CreateEmpty(const ASTContext &C,
1963 EmptyShell) {
1964 void *Mem = C.Allocate(sizeof(OMPTaskwaitDirective));
1965 return new (Mem) OMPTaskwaitDirective();
1966}
1967
Alexey Bataev6125da92014-07-21 11:26:11 +00001968OMPFlushDirective *OMPFlushDirective::Create(const ASTContext &C,
1969 SourceLocation StartLoc,
1970 SourceLocation EndLoc,
1971 ArrayRef<OMPClause *> Clauses) {
1972 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPFlushDirective),
1973 llvm::alignOf<OMPClause *>());
1974 void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size());
1975 OMPFlushDirective *Dir =
1976 new (Mem) OMPFlushDirective(StartLoc, EndLoc, Clauses.size());
1977 Dir->setClauses(Clauses);
1978 return Dir;
1979}
1980
1981OMPFlushDirective *OMPFlushDirective::CreateEmpty(const ASTContext &C,
1982 unsigned NumClauses,
1983 EmptyShell) {
1984 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPFlushDirective),
1985 llvm::alignOf<OMPClause *>());
1986 void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses);
1987 return new (Mem) OMPFlushDirective(NumClauses);
1988}
1989
Alexey Bataev9fb6e642014-07-22 06:45:04 +00001990OMPOrderedDirective *OMPOrderedDirective::Create(const ASTContext &C,
1991 SourceLocation StartLoc,
1992 SourceLocation EndLoc,
1993 Stmt *AssociatedStmt) {
1994 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPOrderedDirective),
1995 llvm::alignOf<Stmt *>());
1996 void *Mem = C.Allocate(Size + sizeof(Stmt *));
1997 OMPOrderedDirective *Dir = new (Mem) OMPOrderedDirective(StartLoc, EndLoc);
1998 Dir->setAssociatedStmt(AssociatedStmt);
1999 return Dir;
2000}
2001
2002OMPOrderedDirective *OMPOrderedDirective::CreateEmpty(const ASTContext &C,
2003 EmptyShell) {
2004 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPOrderedDirective),
2005 llvm::alignOf<Stmt *>());
2006 void *Mem = C.Allocate(Size + sizeof(Stmt *));
2007 return new (Mem) OMPOrderedDirective();
2008}
2009
Alexey Bataev62cec442014-11-18 10:14:22 +00002010OMPAtomicDirective *
2011OMPAtomicDirective::Create(const ASTContext &C, SourceLocation StartLoc,
2012 SourceLocation EndLoc, ArrayRef<OMPClause *> Clauses,
Alexey Bataevb4505a72015-03-30 05:20:59 +00002013 Stmt *AssociatedStmt, Expr *X, Expr *V, Expr *E,
2014 Expr *UE, bool IsXLHSInRHSPart) {
Alexey Bataev0162e452014-07-22 10:10:35 +00002015 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPAtomicDirective),
2016 llvm::alignOf<OMPClause *>());
Alexey Bataev62cec442014-11-18 10:14:22 +00002017 void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
Alexey Bataev1d160b12015-03-13 12:27:31 +00002018 5 * sizeof(Stmt *));
Alexey Bataev0162e452014-07-22 10:10:35 +00002019 OMPAtomicDirective *Dir =
2020 new (Mem) OMPAtomicDirective(StartLoc, EndLoc, Clauses.size());
2021 Dir->setClauses(Clauses);
2022 Dir->setAssociatedStmt(AssociatedStmt);
Alexey Bataev62cec442014-11-18 10:14:22 +00002023 Dir->setX(X);
2024 Dir->setV(V);
2025 Dir->setExpr(E);
Alexey Bataevb4505a72015-03-30 05:20:59 +00002026 Dir->setUpdateExpr(UE);
2027 Dir->IsXLHSInRHSPart = IsXLHSInRHSPart;
Alexey Bataev0162e452014-07-22 10:10:35 +00002028 return Dir;
2029}
2030
2031OMPAtomicDirective *OMPAtomicDirective::CreateEmpty(const ASTContext &C,
2032 unsigned NumClauses,
2033 EmptyShell) {
2034 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPAtomicDirective),
2035 llvm::alignOf<OMPClause *>());
2036 void *Mem =
Alexey Bataev1d160b12015-03-13 12:27:31 +00002037 C.Allocate(Size + sizeof(OMPClause *) * NumClauses + 5 * sizeof(Stmt *));
Alexey Bataev0162e452014-07-22 10:10:35 +00002038 return new (Mem) OMPAtomicDirective(NumClauses);
2039}
2040
Alexey Bataev0bd520b2014-09-19 08:19:49 +00002041OMPTargetDirective *OMPTargetDirective::Create(const ASTContext &C,
2042 SourceLocation StartLoc,
2043 SourceLocation EndLoc,
2044 ArrayRef<OMPClause *> Clauses,
2045 Stmt *AssociatedStmt) {
2046 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPTargetDirective),
2047 llvm::alignOf<OMPClause *>());
2048 void *Mem =
2049 C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
2050 OMPTargetDirective *Dir =
2051 new (Mem) OMPTargetDirective(StartLoc, EndLoc, Clauses.size());
2052 Dir->setClauses(Clauses);
2053 Dir->setAssociatedStmt(AssociatedStmt);
2054 return Dir;
2055}
2056
2057OMPTargetDirective *OMPTargetDirective::CreateEmpty(const ASTContext &C,
2058 unsigned NumClauses,
2059 EmptyShell) {
2060 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPTargetDirective),
2061 llvm::alignOf<OMPClause *>());
2062 void *Mem =
2063 C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
2064 return new (Mem) OMPTargetDirective(NumClauses);
2065}
2066
Alexey Bataev13314bf2014-10-09 04:18:56 +00002067OMPTeamsDirective *OMPTeamsDirective::Create(const ASTContext &C,
2068 SourceLocation StartLoc,
2069 SourceLocation EndLoc,
2070 ArrayRef<OMPClause *> Clauses,
2071 Stmt *AssociatedStmt) {
2072 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPTeamsDirective),
2073 llvm::alignOf<OMPClause *>());
2074 void *Mem =
2075 C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
2076 OMPTeamsDirective *Dir =
2077 new (Mem) OMPTeamsDirective(StartLoc, EndLoc, Clauses.size());
2078 Dir->setClauses(Clauses);
2079 Dir->setAssociatedStmt(AssociatedStmt);
2080 return Dir;
2081}
2082
2083OMPTeamsDirective *OMPTeamsDirective::CreateEmpty(const ASTContext &C,
2084 unsigned NumClauses,
2085 EmptyShell) {
2086 unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPTeamsDirective),
2087 llvm::alignOf<OMPClause *>());
2088 void *Mem =
2089 C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
2090 return new (Mem) OMPTeamsDirective(NumClauses);
2091}
2092