• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===--- Stmt.cpp - Statement AST Node Implementation ---------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements the Stmt class and statement subclasses.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/AST/ASTContext.h"
15 #include "clang/AST/ASTDiagnostic.h"
16 #include "clang/AST/ExprCXX.h"
17 #include "clang/AST/ExprObjC.h"
18 #include "clang/AST/Stmt.h"
19 #include "clang/AST/StmtCXX.h"
20 #include "clang/AST/StmtObjC.h"
21 #include "clang/AST/StmtOpenMP.h"
22 #include "clang/AST/Type.h"
23 #include "clang/Basic/CharInfo.h"
24 #include "clang/Basic/TargetInfo.h"
25 #include "clang/Lex/Token.h"
26 #include "llvm/ADT/StringExtras.h"
27 #include "llvm/Support/raw_ostream.h"
28 using namespace clang;
29 
30 static struct StmtClassNameTable {
31   const char *Name;
32   unsigned Counter;
33   unsigned Size;
34 } StmtClassInfo[Stmt::lastStmtConstant+1];
35 
getStmtInfoTableEntry(Stmt::StmtClass E)36 static 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;
43 #define ABSTRACT_STMT(STMT)
44 #define STMT(CLASS, PARENT) \
45   StmtClassInfo[(unsigned)Stmt::CLASS##Class].Name = #CLASS;    \
46   StmtClassInfo[(unsigned)Stmt::CLASS##Class].Size = sizeof(CLASS);
47 #include "clang/AST/StmtNodes.inc"
48 
49   return StmtClassInfo[E];
50 }
51 
operator new(size_t bytes,const ASTContext & C,unsigned alignment)52 void *Stmt::operator new(size_t bytes, const ASTContext& C,
53                          unsigned alignment) {
54   return ::operator new(bytes, C, alignment);
55 }
56 
getStmtClassName() const57 const char *Stmt::getStmtClassName() const {
58   return getStmtInfoTableEntry((StmtClass) StmtBits.sClass).Name;
59 }
60 
PrintStats()61 void Stmt::PrintStats() {
62   // Ensure the table is primed.
63   getStmtInfoTableEntry(Stmt::NullStmtClass);
64 
65   unsigned sum = 0;
66   llvm::errs() << "\n*** Stmt/Expr Stats:\n";
67   for (int i = 0; i != Stmt::lastStmtConstant+1; i++) {
68     if (StmtClassInfo[i].Name == nullptr) continue;
69     sum += StmtClassInfo[i].Counter;
70   }
71   llvm::errs() << "  " << sum << " stmts/exprs total.\n";
72   sum = 0;
73   for (int i = 0; i != Stmt::lastStmtConstant+1; i++) {
74     if (StmtClassInfo[i].Name == nullptr) continue;
75     if (StmtClassInfo[i].Counter == 0) continue;
76     llvm::errs() << "    " << StmtClassInfo[i].Counter << " "
77                  << StmtClassInfo[i].Name << ", " << StmtClassInfo[i].Size
78                  << " each (" << StmtClassInfo[i].Counter*StmtClassInfo[i].Size
79                  << " bytes)\n";
80     sum += StmtClassInfo[i].Counter*StmtClassInfo[i].Size;
81   }
82 
83   llvm::errs() << "Total bytes = " << sum << "\n";
84 }
85 
addStmtClass(StmtClass s)86 void Stmt::addStmtClass(StmtClass s) {
87   ++getStmtInfoTableEntry(s).Counter;
88 }
89 
90 bool Stmt::StatisticsEnabled = false;
EnableStatistics()91 void Stmt::EnableStatistics() {
92   StatisticsEnabled = true;
93 }
94 
IgnoreImplicit()95 Stmt *Stmt::IgnoreImplicit() {
96   Stmt *s = this;
97 
98   if (ExprWithCleanups *ewc = dyn_cast<ExprWithCleanups>(s))
99     s = ewc->getSubExpr();
100 
101   while (ImplicitCastExpr *ice = dyn_cast<ImplicitCastExpr>(s))
102     s = ice->getSubExpr();
103 
104   return s;
105 }
106 
107 /// \brief Strip off all label-like statements.
108 ///
109 /// This will strip off label statements, case statements, attributed
110 /// statements and default statements recursively.
stripLabelLikeStatements() const111 const Stmt *Stmt::stripLabelLikeStatements() const {
112   const Stmt *S = this;
113   while (true) {
114     if (const LabelStmt *LS = dyn_cast<LabelStmt>(S))
115       S = LS->getSubStmt();
116     else if (const SwitchCase *SC = dyn_cast<SwitchCase>(S))
117       S = SC->getSubStmt();
118     else if (const AttributedStmt *AS = dyn_cast<AttributedStmt>(S))
119       S = AS->getSubStmt();
120     else
121       return S;
122   }
123 }
124 
125 namespace {
126   struct good {};
127   struct bad {};
128 
129   // These silly little functions have to be static inline to suppress
130   // unused warnings, and they have to be defined to suppress other
131   // warnings.
is_good(good)132   static inline good is_good(good) { return good(); }
133 
134   typedef Stmt::child_range children_t();
implements_children(children_t T::*)135   template <class T> good implements_children(children_t T::*) {
136     return good();
137   }
138   LLVM_ATTRIBUTE_UNUSED
implements_children(children_t Stmt::*)139   static inline bad implements_children(children_t Stmt::*) {
140     return bad();
141   }
142 
143   typedef SourceLocation getLocStart_t() const;
implements_getLocStart(getLocStart_t T::*)144   template <class T> good implements_getLocStart(getLocStart_t T::*) {
145     return good();
146   }
147   LLVM_ATTRIBUTE_UNUSED
implements_getLocStart(getLocStart_t Stmt::*)148   static inline bad implements_getLocStart(getLocStart_t Stmt::*) {
149     return bad();
150   }
151 
152   typedef SourceLocation getLocEnd_t() const;
implements_getLocEnd(getLocEnd_t T::*)153   template <class T> good implements_getLocEnd(getLocEnd_t T::*) {
154     return good();
155   }
156   LLVM_ATTRIBUTE_UNUSED
implements_getLocEnd(getLocEnd_t Stmt::*)157   static inline bad implements_getLocEnd(getLocEnd_t Stmt::*) {
158     return bad();
159   }
160 
161 #define ASSERT_IMPLEMENTS_children(type) \
162   (void) is_good(implements_children(&type::children))
163 #define ASSERT_IMPLEMENTS_getLocStart(type) \
164   (void) is_good(implements_getLocStart(&type::getLocStart))
165 #define ASSERT_IMPLEMENTS_getLocEnd(type) \
166   (void) is_good(implements_getLocEnd(&type::getLocEnd))
167 }
168 
169 /// Check whether the various Stmt classes implement their member
170 /// functions.
171 LLVM_ATTRIBUTE_UNUSED
check_implementations()172 static inline void check_implementations() {
173 #define ABSTRACT_STMT(type)
174 #define STMT(type, base) \
175   ASSERT_IMPLEMENTS_children(type); \
176   ASSERT_IMPLEMENTS_getLocStart(type); \
177   ASSERT_IMPLEMENTS_getLocEnd(type);
178 #include "clang/AST/StmtNodes.inc"
179 }
180 
children()181 Stmt::child_range Stmt::children() {
182   switch (getStmtClass()) {
183   case Stmt::NoStmtClass: llvm_unreachable("statement without class");
184 #define ABSTRACT_STMT(type)
185 #define STMT(type, base) \
186   case Stmt::type##Class: \
187     return static_cast<type*>(this)->children();
188 #include "clang/AST/StmtNodes.inc"
189   }
190   llvm_unreachable("unknown statement kind!");
191 }
192 
193 // Amusing macro metaprogramming hack: check whether a class provides
194 // a more specific implementation of getSourceRange.
195 //
196 // See also Expr.cpp:getExprLoc().
197 namespace {
198   /// This implementation is used when a class provides a custom
199   /// implementation of getSourceRange.
200   template <class S, class T>
getSourceRangeImpl(const Stmt * stmt,SourceRange (T::* v)()const)201   SourceRange getSourceRangeImpl(const Stmt *stmt,
202                                  SourceRange (T::*v)() const) {
203     return static_cast<const S*>(stmt)->getSourceRange();
204   }
205 
206   /// This implementation is used when a class doesn't provide a custom
207   /// implementation of getSourceRange.  Overload resolution should pick it over
208   /// the implementation above because it's more specialized according to
209   /// function template partial ordering.
210   template <class S>
getSourceRangeImpl(const Stmt * stmt,SourceRange (Stmt::* v)()const)211   SourceRange getSourceRangeImpl(const Stmt *stmt,
212                                  SourceRange (Stmt::*v)() const) {
213     return SourceRange(static_cast<const S*>(stmt)->getLocStart(),
214                        static_cast<const S*>(stmt)->getLocEnd());
215   }
216 }
217 
getSourceRange() const218 SourceRange Stmt::getSourceRange() const {
219   switch (getStmtClass()) {
220   case Stmt::NoStmtClass: llvm_unreachable("statement without class");
221 #define ABSTRACT_STMT(type)
222 #define STMT(type, base) \
223   case Stmt::type##Class: \
224     return getSourceRangeImpl<type>(this, &type::getSourceRange);
225 #include "clang/AST/StmtNodes.inc"
226   }
227   llvm_unreachable("unknown statement kind!");
228 }
229 
getLocStart() const230 SourceLocation Stmt::getLocStart() const {
231 //  llvm::errs() << "getLocStart() for " << getStmtClassName() << "\n";
232   switch (getStmtClass()) {
233   case Stmt::NoStmtClass: llvm_unreachable("statement without class");
234 #define ABSTRACT_STMT(type)
235 #define STMT(type, base) \
236   case Stmt::type##Class: \
237     return static_cast<const type*>(this)->getLocStart();
238 #include "clang/AST/StmtNodes.inc"
239   }
240   llvm_unreachable("unknown statement kind");
241 }
242 
getLocEnd() const243 SourceLocation Stmt::getLocEnd() const {
244   switch (getStmtClass()) {
245   case Stmt::NoStmtClass: llvm_unreachable("statement without class");
246 #define ABSTRACT_STMT(type)
247 #define STMT(type, base) \
248   case Stmt::type##Class: \
249     return static_cast<const type*>(this)->getLocEnd();
250 #include "clang/AST/StmtNodes.inc"
251   }
252   llvm_unreachable("unknown statement kind");
253 }
254 
CompoundStmt(const ASTContext & C,ArrayRef<Stmt * > Stmts,SourceLocation LB,SourceLocation RB)255 CompoundStmt::CompoundStmt(const ASTContext &C, ArrayRef<Stmt*> Stmts,
256                            SourceLocation LB, SourceLocation RB)
257   : Stmt(CompoundStmtClass), LBracLoc(LB), RBracLoc(RB) {
258   CompoundStmtBits.NumStmts = Stmts.size();
259   assert(CompoundStmtBits.NumStmts == Stmts.size() &&
260          "NumStmts doesn't fit in bits of CompoundStmtBits.NumStmts!");
261 
262   if (Stmts.size() == 0) {
263     Body = nullptr;
264     return;
265   }
266 
267   Body = new (C) Stmt*[Stmts.size()];
268   std::copy(Stmts.begin(), Stmts.end(), Body);
269 }
270 
setStmts(const ASTContext & C,Stmt ** Stmts,unsigned NumStmts)271 void CompoundStmt::setStmts(const ASTContext &C, Stmt **Stmts,
272                             unsigned NumStmts) {
273   if (this->Body)
274     C.Deallocate(Body);
275   this->CompoundStmtBits.NumStmts = NumStmts;
276 
277   Body = new (C) Stmt*[NumStmts];
278   memcpy(Body, Stmts, sizeof(Stmt *) * NumStmts);
279 }
280 
getName() const281 const char *LabelStmt::getName() const {
282   return getDecl()->getIdentifier()->getNameStart();
283 }
284 
Create(const ASTContext & C,SourceLocation Loc,ArrayRef<const Attr * > Attrs,Stmt * SubStmt)285 AttributedStmt *AttributedStmt::Create(const ASTContext &C, SourceLocation Loc,
286                                        ArrayRef<const Attr*> Attrs,
287                                        Stmt *SubStmt) {
288   assert(!Attrs.empty() && "Attrs should not be empty");
289   void *Mem = C.Allocate(sizeof(AttributedStmt) + sizeof(Attr *) * Attrs.size(),
290                          llvm::alignOf<AttributedStmt>());
291   return new (Mem) AttributedStmt(Loc, Attrs, SubStmt);
292 }
293 
CreateEmpty(const ASTContext & C,unsigned NumAttrs)294 AttributedStmt *AttributedStmt::CreateEmpty(const ASTContext &C,
295                                             unsigned NumAttrs) {
296   assert(NumAttrs > 0 && "NumAttrs should be greater than zero");
297   void *Mem = C.Allocate(sizeof(AttributedStmt) + sizeof(Attr *) * NumAttrs,
298                          llvm::alignOf<AttributedStmt>());
299   return new (Mem) AttributedStmt(EmptyShell(), NumAttrs);
300 }
301 
generateAsmString(const ASTContext & C) const302 std::string AsmStmt::generateAsmString(const ASTContext &C) const {
303   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
304     return gccAsmStmt->generateAsmString(C);
305   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
306     return msAsmStmt->generateAsmString(C);
307   llvm_unreachable("unknown asm statement kind!");
308 }
309 
getOutputConstraint(unsigned i) const310 StringRef AsmStmt::getOutputConstraint(unsigned i) const {
311   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
312     return gccAsmStmt->getOutputConstraint(i);
313   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
314     return msAsmStmt->getOutputConstraint(i);
315   llvm_unreachable("unknown asm statement kind!");
316 }
317 
getOutputExpr(unsigned i) const318 const Expr *AsmStmt::getOutputExpr(unsigned i) const {
319   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
320     return gccAsmStmt->getOutputExpr(i);
321   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
322     return msAsmStmt->getOutputExpr(i);
323   llvm_unreachable("unknown asm statement kind!");
324 }
325 
getInputConstraint(unsigned i) const326 StringRef AsmStmt::getInputConstraint(unsigned i) const {
327   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
328     return gccAsmStmt->getInputConstraint(i);
329   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
330     return msAsmStmt->getInputConstraint(i);
331   llvm_unreachable("unknown asm statement kind!");
332 }
333 
getInputExpr(unsigned i) const334 const Expr *AsmStmt::getInputExpr(unsigned i) const {
335   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
336     return gccAsmStmt->getInputExpr(i);
337   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
338     return msAsmStmt->getInputExpr(i);
339   llvm_unreachable("unknown asm statement kind!");
340 }
341 
getClobber(unsigned i) const342 StringRef AsmStmt::getClobber(unsigned i) const {
343   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
344     return gccAsmStmt->getClobber(i);
345   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
346     return msAsmStmt->getClobber(i);
347   llvm_unreachable("unknown asm statement kind!");
348 }
349 
350 /// getNumPlusOperands - Return the number of output operands that have a "+"
351 /// constraint.
getNumPlusOperands() const352 unsigned AsmStmt::getNumPlusOperands() const {
353   unsigned Res = 0;
354   for (unsigned i = 0, e = getNumOutputs(); i != e; ++i)
355     if (isOutputPlusConstraint(i))
356       ++Res;
357   return Res;
358 }
359 
getClobber(unsigned i) const360 StringRef GCCAsmStmt::getClobber(unsigned i) const {
361   return getClobberStringLiteral(i)->getString();
362 }
363 
getOutputExpr(unsigned i)364 Expr *GCCAsmStmt::getOutputExpr(unsigned i) {
365   return cast<Expr>(Exprs[i]);
366 }
367 
368 /// getOutputConstraint - Return the constraint string for the specified
369 /// output operand.  All output constraints are known to be non-empty (either
370 /// '=' or '+').
getOutputConstraint(unsigned i) const371 StringRef GCCAsmStmt::getOutputConstraint(unsigned i) const {
372   return getOutputConstraintLiteral(i)->getString();
373 }
374 
getInputExpr(unsigned i)375 Expr *GCCAsmStmt::getInputExpr(unsigned i) {
376   return cast<Expr>(Exprs[i + NumOutputs]);
377 }
setInputExpr(unsigned i,Expr * E)378 void GCCAsmStmt::setInputExpr(unsigned i, Expr *E) {
379   Exprs[i + NumOutputs] = E;
380 }
381 
382 /// getInputConstraint - Return the specified input constraint.  Unlike output
383 /// constraints, these can be empty.
getInputConstraint(unsigned i) const384 StringRef GCCAsmStmt::getInputConstraint(unsigned i) const {
385   return getInputConstraintLiteral(i)->getString();
386 }
387 
setOutputsAndInputsAndClobbers(const ASTContext & C,IdentifierInfo ** Names,StringLiteral ** Constraints,Stmt ** Exprs,unsigned NumOutputs,unsigned NumInputs,StringLiteral ** Clobbers,unsigned NumClobbers)388 void GCCAsmStmt::setOutputsAndInputsAndClobbers(const ASTContext &C,
389                                                 IdentifierInfo **Names,
390                                                 StringLiteral **Constraints,
391                                                 Stmt **Exprs,
392                                                 unsigned NumOutputs,
393                                                 unsigned NumInputs,
394                                                 StringLiteral **Clobbers,
395                                                 unsigned NumClobbers) {
396   this->NumOutputs = NumOutputs;
397   this->NumInputs = NumInputs;
398   this->NumClobbers = NumClobbers;
399 
400   unsigned NumExprs = NumOutputs + NumInputs;
401 
402   C.Deallocate(this->Names);
403   this->Names = new (C) IdentifierInfo*[NumExprs];
404   std::copy(Names, Names + NumExprs, this->Names);
405 
406   C.Deallocate(this->Exprs);
407   this->Exprs = new (C) Stmt*[NumExprs];
408   std::copy(Exprs, Exprs + NumExprs, this->Exprs);
409 
410   C.Deallocate(this->Constraints);
411   this->Constraints = new (C) StringLiteral*[NumExprs];
412   std::copy(Constraints, Constraints + NumExprs, this->Constraints);
413 
414   C.Deallocate(this->Clobbers);
415   this->Clobbers = new (C) StringLiteral*[NumClobbers];
416   std::copy(Clobbers, Clobbers + NumClobbers, this->Clobbers);
417 }
418 
419 /// getNamedOperand - Given a symbolic operand reference like %[foo],
420 /// translate this into a numeric value needed to reference the same operand.
421 /// This returns -1 if the operand name is invalid.
getNamedOperand(StringRef SymbolicName) const422 int GCCAsmStmt::getNamedOperand(StringRef SymbolicName) const {
423   unsigned NumPlusOperands = 0;
424 
425   // Check if this is an output operand.
426   for (unsigned i = 0, e = getNumOutputs(); i != e; ++i) {
427     if (getOutputName(i) == SymbolicName)
428       return i;
429   }
430 
431   for (unsigned i = 0, e = getNumInputs(); i != e; ++i)
432     if (getInputName(i) == SymbolicName)
433       return getNumOutputs() + NumPlusOperands + i;
434 
435   // Not found.
436   return -1;
437 }
438 
439 /// AnalyzeAsmString - Analyze the asm string of the current asm, decomposing
440 /// it into pieces.  If the asm string is erroneous, emit errors and return
441 /// true, otherwise return false.
AnalyzeAsmString(SmallVectorImpl<AsmStringPiece> & Pieces,const ASTContext & C,unsigned & DiagOffs) const442 unsigned GCCAsmStmt::AnalyzeAsmString(SmallVectorImpl<AsmStringPiece>&Pieces,
443                                 const ASTContext &C, unsigned &DiagOffs) const {
444   StringRef Str = getAsmString()->getString();
445   const char *StrStart = Str.begin();
446   const char *StrEnd = Str.end();
447   const char *CurPtr = StrStart;
448 
449   // "Simple" inline asms have no constraints or operands, just convert the asm
450   // string to escape $'s.
451   if (isSimple()) {
452     std::string Result;
453     for (; CurPtr != StrEnd; ++CurPtr) {
454       switch (*CurPtr) {
455       case '$':
456         Result += "$$";
457         break;
458       default:
459         Result += *CurPtr;
460         break;
461       }
462     }
463     Pieces.push_back(AsmStringPiece(Result));
464     return 0;
465   }
466 
467   // CurStringPiece - The current string that we are building up as we scan the
468   // asm string.
469   std::string CurStringPiece;
470 
471   bool HasVariants = !C.getTargetInfo().hasNoAsmVariants();
472 
473   while (1) {
474     // Done with the string?
475     if (CurPtr == StrEnd) {
476       if (!CurStringPiece.empty())
477         Pieces.push_back(AsmStringPiece(CurStringPiece));
478       return 0;
479     }
480 
481     char CurChar = *CurPtr++;
482     switch (CurChar) {
483     case '$': CurStringPiece += "$$"; continue;
484     case '{': CurStringPiece += (HasVariants ? "$(" : "{"); continue;
485     case '|': CurStringPiece += (HasVariants ? "$|" : "|"); continue;
486     case '}': CurStringPiece += (HasVariants ? "$)" : "}"); continue;
487     case '%':
488       break;
489     default:
490       CurStringPiece += CurChar;
491       continue;
492     }
493 
494     // Escaped "%" character in asm string.
495     if (CurPtr == StrEnd) {
496       // % at end of string is invalid (no escape).
497       DiagOffs = CurPtr-StrStart-1;
498       return diag::err_asm_invalid_escape;
499     }
500 
501     char EscapedChar = *CurPtr++;
502     if (EscapedChar == '%') {  // %% -> %
503       // Escaped percentage sign.
504       CurStringPiece += '%';
505       continue;
506     }
507 
508     if (EscapedChar == '=') {  // %= -> Generate an unique ID.
509       CurStringPiece += "${:uid}";
510       continue;
511     }
512 
513     // Otherwise, we have an operand.  If we have accumulated a string so far,
514     // add it to the Pieces list.
515     if (!CurStringPiece.empty()) {
516       Pieces.push_back(AsmStringPiece(CurStringPiece));
517       CurStringPiece.clear();
518     }
519 
520     // Handle %x4 and %x[foo] by capturing x as the modifier character.
521     char Modifier = '\0';
522     if (isLetter(EscapedChar)) {
523       if (CurPtr == StrEnd) { // Premature end.
524         DiagOffs = CurPtr-StrStart-1;
525         return diag::err_asm_invalid_escape;
526       }
527       Modifier = EscapedChar;
528       EscapedChar = *CurPtr++;
529     }
530 
531     if (isDigit(EscapedChar)) {
532       // %n - Assembler operand n
533       unsigned N = 0;
534 
535       --CurPtr;
536       while (CurPtr != StrEnd && isDigit(*CurPtr))
537         N = N*10 + ((*CurPtr++)-'0');
538 
539       unsigned NumOperands =
540         getNumOutputs() + getNumPlusOperands() + getNumInputs();
541       if (N >= NumOperands) {
542         DiagOffs = CurPtr-StrStart-1;
543         return diag::err_asm_invalid_operand_number;
544       }
545 
546       Pieces.push_back(AsmStringPiece(N, Modifier));
547       continue;
548     }
549 
550     // Handle %[foo], a symbolic operand reference.
551     if (EscapedChar == '[') {
552       DiagOffs = CurPtr-StrStart-1;
553 
554       // Find the ']'.
555       const char *NameEnd = (const char*)memchr(CurPtr, ']', StrEnd-CurPtr);
556       if (NameEnd == nullptr)
557         return diag::err_asm_unterminated_symbolic_operand_name;
558       if (NameEnd == CurPtr)
559         return diag::err_asm_empty_symbolic_operand_name;
560 
561       StringRef SymbolicName(CurPtr, NameEnd - CurPtr);
562 
563       int N = getNamedOperand(SymbolicName);
564       if (N == -1) {
565         // Verify that an operand with that name exists.
566         DiagOffs = CurPtr-StrStart;
567         return diag::err_asm_unknown_symbolic_operand_name;
568       }
569       Pieces.push_back(AsmStringPiece(N, Modifier));
570 
571       CurPtr = NameEnd+1;
572       continue;
573     }
574 
575     DiagOffs = CurPtr-StrStart-1;
576     return diag::err_asm_invalid_escape;
577   }
578 }
579 
580 /// Assemble final IR asm string (GCC-style).
generateAsmString(const ASTContext & C) const581 std::string GCCAsmStmt::generateAsmString(const ASTContext &C) const {
582   // Analyze the asm string to decompose it into its pieces.  We know that Sema
583   // has already done this, so it is guaranteed to be successful.
584   SmallVector<GCCAsmStmt::AsmStringPiece, 4> Pieces;
585   unsigned DiagOffs;
586   AnalyzeAsmString(Pieces, C, DiagOffs);
587 
588   std::string AsmString;
589   for (unsigned i = 0, e = Pieces.size(); i != e; ++i) {
590     if (Pieces[i].isString())
591       AsmString += Pieces[i].getString();
592     else if (Pieces[i].getModifier() == '\0')
593       AsmString += '$' + llvm::utostr(Pieces[i].getOperandNo());
594     else
595       AsmString += "${" + llvm::utostr(Pieces[i].getOperandNo()) + ':' +
596                    Pieces[i].getModifier() + '}';
597   }
598   return AsmString;
599 }
600 
601 /// Assemble final IR asm string (MS-style).
generateAsmString(const ASTContext & C) const602 std::string MSAsmStmt::generateAsmString(const ASTContext &C) const {
603   // FIXME: This needs to be translated into the IR string representation.
604   return AsmStr;
605 }
606 
getOutputExpr(unsigned i)607 Expr *MSAsmStmt::getOutputExpr(unsigned i) {
608   return cast<Expr>(Exprs[i]);
609 }
610 
getInputExpr(unsigned i)611 Expr *MSAsmStmt::getInputExpr(unsigned i) {
612   return cast<Expr>(Exprs[i + NumOutputs]);
613 }
setInputExpr(unsigned i,Expr * E)614 void MSAsmStmt::setInputExpr(unsigned i, Expr *E) {
615   Exprs[i + NumOutputs] = E;
616 }
617 
getCaughtType() const618 QualType CXXCatchStmt::getCaughtType() const {
619   if (ExceptionDecl)
620     return ExceptionDecl->getType();
621   return QualType();
622 }
623 
624 //===----------------------------------------------------------------------===//
625 // Constructors
626 //===----------------------------------------------------------------------===//
627 
GCCAsmStmt(const ASTContext & C,SourceLocation asmloc,bool issimple,bool isvolatile,unsigned numoutputs,unsigned numinputs,IdentifierInfo ** names,StringLiteral ** constraints,Expr ** exprs,StringLiteral * asmstr,unsigned numclobbers,StringLiteral ** clobbers,SourceLocation rparenloc)628 GCCAsmStmt::GCCAsmStmt(const ASTContext &C, SourceLocation asmloc,
629                        bool issimple, bool isvolatile, unsigned numoutputs,
630                        unsigned numinputs, IdentifierInfo **names,
631                        StringLiteral **constraints, Expr **exprs,
632                        StringLiteral *asmstr, unsigned numclobbers,
633                        StringLiteral **clobbers, SourceLocation rparenloc)
634   : AsmStmt(GCCAsmStmtClass, asmloc, issimple, isvolatile, numoutputs,
635             numinputs, numclobbers), RParenLoc(rparenloc), AsmStr(asmstr) {
636 
637   unsigned NumExprs = NumOutputs + NumInputs;
638 
639   Names = new (C) IdentifierInfo*[NumExprs];
640   std::copy(names, names + NumExprs, Names);
641 
642   Exprs = new (C) Stmt*[NumExprs];
643   std::copy(exprs, exprs + NumExprs, Exprs);
644 
645   Constraints = new (C) StringLiteral*[NumExprs];
646   std::copy(constraints, constraints + NumExprs, Constraints);
647 
648   Clobbers = new (C) StringLiteral*[NumClobbers];
649   std::copy(clobbers, clobbers + NumClobbers, Clobbers);
650 }
651 
MSAsmStmt(const ASTContext & C,SourceLocation asmloc,SourceLocation lbraceloc,bool issimple,bool isvolatile,ArrayRef<Token> asmtoks,unsigned numoutputs,unsigned numinputs,ArrayRef<StringRef> constraints,ArrayRef<Expr * > exprs,StringRef asmstr,ArrayRef<StringRef> clobbers,SourceLocation endloc)652 MSAsmStmt::MSAsmStmt(const ASTContext &C, SourceLocation asmloc,
653                      SourceLocation lbraceloc, bool issimple, bool isvolatile,
654                      ArrayRef<Token> asmtoks, unsigned numoutputs,
655                      unsigned numinputs,
656                      ArrayRef<StringRef> constraints, ArrayRef<Expr*> exprs,
657                      StringRef asmstr, ArrayRef<StringRef> clobbers,
658                      SourceLocation endloc)
659   : AsmStmt(MSAsmStmtClass, asmloc, issimple, isvolatile, numoutputs,
660             numinputs, clobbers.size()), LBraceLoc(lbraceloc),
661             EndLoc(endloc), NumAsmToks(asmtoks.size()) {
662 
663   initialize(C, asmstr, asmtoks, constraints, exprs, clobbers);
664 }
665 
copyIntoContext(const ASTContext & C,StringRef str)666 static StringRef copyIntoContext(const ASTContext &C, StringRef str) {
667   size_t size = str.size();
668   char *buffer = new (C) char[size];
669   memcpy(buffer, str.data(), size);
670   return StringRef(buffer, size);
671 }
672 
initialize(const ASTContext & C,StringRef asmstr,ArrayRef<Token> asmtoks,ArrayRef<StringRef> constraints,ArrayRef<Expr * > exprs,ArrayRef<StringRef> clobbers)673 void MSAsmStmt::initialize(const ASTContext &C, StringRef asmstr,
674                            ArrayRef<Token> asmtoks,
675                            ArrayRef<StringRef> constraints,
676                            ArrayRef<Expr*> exprs,
677                            ArrayRef<StringRef> clobbers) {
678   assert(NumAsmToks == asmtoks.size());
679   assert(NumClobbers == clobbers.size());
680 
681   unsigned NumExprs = exprs.size();
682   assert(NumExprs == NumOutputs + NumInputs);
683   assert(NumExprs == constraints.size());
684 
685   AsmStr = copyIntoContext(C, asmstr);
686 
687   Exprs = new (C) Stmt*[NumExprs];
688   for (unsigned i = 0, e = NumExprs; i != e; ++i)
689     Exprs[i] = exprs[i];
690 
691   AsmToks = new (C) Token[NumAsmToks];
692   for (unsigned i = 0, e = NumAsmToks; i != e; ++i)
693     AsmToks[i] = asmtoks[i];
694 
695   Constraints = new (C) StringRef[NumExprs];
696   for (unsigned i = 0, e = NumExprs; i != e; ++i) {
697     Constraints[i] = copyIntoContext(C, constraints[i]);
698   }
699 
700   Clobbers = new (C) StringRef[NumClobbers];
701   for (unsigned i = 0, e = NumClobbers; i != e; ++i) {
702     // FIXME: Avoid the allocation/copy if at all possible.
703     Clobbers[i] = copyIntoContext(C, clobbers[i]);
704   }
705 }
706 
ObjCForCollectionStmt(Stmt * Elem,Expr * Collect,Stmt * Body,SourceLocation FCL,SourceLocation RPL)707 ObjCForCollectionStmt::ObjCForCollectionStmt(Stmt *Elem, Expr *Collect,
708                                              Stmt *Body,  SourceLocation FCL,
709                                              SourceLocation RPL)
710 : Stmt(ObjCForCollectionStmtClass) {
711   SubExprs[ELEM] = Elem;
712   SubExprs[COLLECTION] = Collect;
713   SubExprs[BODY] = Body;
714   ForLoc = FCL;
715   RParenLoc = RPL;
716 }
717 
ObjCAtTryStmt(SourceLocation atTryLoc,Stmt * atTryStmt,Stmt ** CatchStmts,unsigned NumCatchStmts,Stmt * atFinallyStmt)718 ObjCAtTryStmt::ObjCAtTryStmt(SourceLocation atTryLoc, Stmt *atTryStmt,
719                              Stmt **CatchStmts, unsigned NumCatchStmts,
720                              Stmt *atFinallyStmt)
721   : Stmt(ObjCAtTryStmtClass), AtTryLoc(atTryLoc),
722     NumCatchStmts(NumCatchStmts), HasFinally(atFinallyStmt != nullptr) {
723   Stmt **Stmts = getStmts();
724   Stmts[0] = atTryStmt;
725   for (unsigned I = 0; I != NumCatchStmts; ++I)
726     Stmts[I + 1] = CatchStmts[I];
727 
728   if (HasFinally)
729     Stmts[NumCatchStmts + 1] = atFinallyStmt;
730 }
731 
Create(const ASTContext & Context,SourceLocation atTryLoc,Stmt * atTryStmt,Stmt ** CatchStmts,unsigned NumCatchStmts,Stmt * atFinallyStmt)732 ObjCAtTryStmt *ObjCAtTryStmt::Create(const ASTContext &Context,
733                                      SourceLocation atTryLoc,
734                                      Stmt *atTryStmt,
735                                      Stmt **CatchStmts,
736                                      unsigned NumCatchStmts,
737                                      Stmt *atFinallyStmt) {
738   unsigned Size = sizeof(ObjCAtTryStmt) +
739     (1 + NumCatchStmts + (atFinallyStmt != nullptr)) * sizeof(Stmt *);
740   void *Mem = Context.Allocate(Size, llvm::alignOf<ObjCAtTryStmt>());
741   return new (Mem) ObjCAtTryStmt(atTryLoc, atTryStmt, CatchStmts, NumCatchStmts,
742                                  atFinallyStmt);
743 }
744 
CreateEmpty(const ASTContext & Context,unsigned NumCatchStmts,bool HasFinally)745 ObjCAtTryStmt *ObjCAtTryStmt::CreateEmpty(const ASTContext &Context,
746                                           unsigned NumCatchStmts,
747                                           bool HasFinally) {
748   unsigned Size = sizeof(ObjCAtTryStmt) +
749     (1 + NumCatchStmts + HasFinally) * sizeof(Stmt *);
750   void *Mem = Context.Allocate(Size, llvm::alignOf<ObjCAtTryStmt>());
751   return new (Mem) ObjCAtTryStmt(EmptyShell(), NumCatchStmts, HasFinally);
752 }
753 
getLocEnd() const754 SourceLocation ObjCAtTryStmt::getLocEnd() const {
755   if (HasFinally)
756     return getFinallyStmt()->getLocEnd();
757   if (NumCatchStmts)
758     return getCatchStmt(NumCatchStmts - 1)->getLocEnd();
759   return getTryBody()->getLocEnd();
760 }
761 
Create(const ASTContext & C,SourceLocation tryLoc,Stmt * tryBlock,ArrayRef<Stmt * > handlers)762 CXXTryStmt *CXXTryStmt::Create(const ASTContext &C, SourceLocation tryLoc,
763                                Stmt *tryBlock, ArrayRef<Stmt*> handlers) {
764   std::size_t Size = sizeof(CXXTryStmt);
765   Size += ((handlers.size() + 1) * sizeof(Stmt));
766 
767   void *Mem = C.Allocate(Size, llvm::alignOf<CXXTryStmt>());
768   return new (Mem) CXXTryStmt(tryLoc, tryBlock, handlers);
769 }
770 
Create(const ASTContext & C,EmptyShell Empty,unsigned numHandlers)771 CXXTryStmt *CXXTryStmt::Create(const ASTContext &C, EmptyShell Empty,
772                                unsigned numHandlers) {
773   std::size_t Size = sizeof(CXXTryStmt);
774   Size += ((numHandlers + 1) * sizeof(Stmt));
775 
776   void *Mem = C.Allocate(Size, llvm::alignOf<CXXTryStmt>());
777   return new (Mem) CXXTryStmt(Empty, numHandlers);
778 }
779 
CXXTryStmt(SourceLocation tryLoc,Stmt * tryBlock,ArrayRef<Stmt * > handlers)780 CXXTryStmt::CXXTryStmt(SourceLocation tryLoc, Stmt *tryBlock,
781                        ArrayRef<Stmt*> handlers)
782   : Stmt(CXXTryStmtClass), TryLoc(tryLoc), NumHandlers(handlers.size()) {
783   Stmt **Stmts = reinterpret_cast<Stmt **>(this + 1);
784   Stmts[0] = tryBlock;
785   std::copy(handlers.begin(), handlers.end(), Stmts + 1);
786 }
787 
CXXForRangeStmt(DeclStmt * Range,DeclStmt * BeginEndStmt,Expr * Cond,Expr * Inc,DeclStmt * LoopVar,Stmt * Body,SourceLocation FL,SourceLocation CL,SourceLocation RPL)788 CXXForRangeStmt::CXXForRangeStmt(DeclStmt *Range, DeclStmt *BeginEndStmt,
789                                  Expr *Cond, Expr *Inc, DeclStmt *LoopVar,
790                                  Stmt *Body, SourceLocation FL,
791                                  SourceLocation CL, SourceLocation RPL)
792   : Stmt(CXXForRangeStmtClass), ForLoc(FL), ColonLoc(CL), RParenLoc(RPL) {
793   SubExprs[RANGE] = Range;
794   SubExprs[BEGINEND] = BeginEndStmt;
795   SubExprs[COND] = Cond;
796   SubExprs[INC] = Inc;
797   SubExprs[LOOPVAR] = LoopVar;
798   SubExprs[BODY] = Body;
799 }
800 
getRangeInit()801 Expr *CXXForRangeStmt::getRangeInit() {
802   DeclStmt *RangeStmt = getRangeStmt();
803   VarDecl *RangeDecl = dyn_cast_or_null<VarDecl>(RangeStmt->getSingleDecl());
804   assert(RangeDecl && "for-range should have a single var decl");
805   return RangeDecl->getInit();
806 }
807 
getRangeInit() const808 const Expr *CXXForRangeStmt::getRangeInit() const {
809   return const_cast<CXXForRangeStmt*>(this)->getRangeInit();
810 }
811 
getLoopVariable()812 VarDecl *CXXForRangeStmt::getLoopVariable() {
813   Decl *LV = cast<DeclStmt>(getLoopVarStmt())->getSingleDecl();
814   assert(LV && "No loop variable in CXXForRangeStmt");
815   return cast<VarDecl>(LV);
816 }
817 
getLoopVariable() const818 const VarDecl *CXXForRangeStmt::getLoopVariable() const {
819   return const_cast<CXXForRangeStmt*>(this)->getLoopVariable();
820 }
821 
IfStmt(const ASTContext & C,SourceLocation IL,VarDecl * var,Expr * cond,Stmt * then,SourceLocation EL,Stmt * elsev)822 IfStmt::IfStmt(const ASTContext &C, SourceLocation IL, VarDecl *var, Expr *cond,
823                Stmt *then, SourceLocation EL, Stmt *elsev)
824   : Stmt(IfStmtClass), IfLoc(IL), ElseLoc(EL)
825 {
826   setConditionVariable(C, var);
827   SubExprs[COND] = cond;
828   SubExprs[THEN] = then;
829   SubExprs[ELSE] = elsev;
830 }
831 
getConditionVariable() const832 VarDecl *IfStmt::getConditionVariable() const {
833   if (!SubExprs[VAR])
834     return nullptr;
835 
836   DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
837   return cast<VarDecl>(DS->getSingleDecl());
838 }
839 
setConditionVariable(const ASTContext & C,VarDecl * V)840 void IfStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
841   if (!V) {
842     SubExprs[VAR] = nullptr;
843     return;
844   }
845 
846   SourceRange VarRange = V->getSourceRange();
847   SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
848                                    VarRange.getEnd());
849 }
850 
ForStmt(const ASTContext & C,Stmt * Init,Expr * Cond,VarDecl * condVar,Expr * Inc,Stmt * Body,SourceLocation FL,SourceLocation LP,SourceLocation RP)851 ForStmt::ForStmt(const ASTContext &C, Stmt *Init, Expr *Cond, VarDecl *condVar,
852                  Expr *Inc, Stmt *Body, SourceLocation FL, SourceLocation LP,
853                  SourceLocation RP)
854   : Stmt(ForStmtClass), ForLoc(FL), LParenLoc(LP), RParenLoc(RP)
855 {
856   SubExprs[INIT] = Init;
857   setConditionVariable(C, condVar);
858   SubExprs[COND] = Cond;
859   SubExprs[INC] = Inc;
860   SubExprs[BODY] = Body;
861 }
862 
getConditionVariable() const863 VarDecl *ForStmt::getConditionVariable() const {
864   if (!SubExprs[CONDVAR])
865     return nullptr;
866 
867   DeclStmt *DS = cast<DeclStmt>(SubExprs[CONDVAR]);
868   return cast<VarDecl>(DS->getSingleDecl());
869 }
870 
setConditionVariable(const ASTContext & C,VarDecl * V)871 void ForStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
872   if (!V) {
873     SubExprs[CONDVAR] = nullptr;
874     return;
875   }
876 
877   SourceRange VarRange = V->getSourceRange();
878   SubExprs[CONDVAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
879                                        VarRange.getEnd());
880 }
881 
SwitchStmt(const ASTContext & C,VarDecl * Var,Expr * cond)882 SwitchStmt::SwitchStmt(const ASTContext &C, VarDecl *Var, Expr *cond)
883   : Stmt(SwitchStmtClass), FirstCase(nullptr), AllEnumCasesCovered(0)
884 {
885   setConditionVariable(C, Var);
886   SubExprs[COND] = cond;
887   SubExprs[BODY] = nullptr;
888 }
889 
getConditionVariable() const890 VarDecl *SwitchStmt::getConditionVariable() const {
891   if (!SubExprs[VAR])
892     return nullptr;
893 
894   DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
895   return cast<VarDecl>(DS->getSingleDecl());
896 }
897 
setConditionVariable(const ASTContext & C,VarDecl * V)898 void SwitchStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
899   if (!V) {
900     SubExprs[VAR] = nullptr;
901     return;
902   }
903 
904   SourceRange VarRange = V->getSourceRange();
905   SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
906                                    VarRange.getEnd());
907 }
908 
getSubStmt()909 Stmt *SwitchCase::getSubStmt() {
910   if (isa<CaseStmt>(this))
911     return cast<CaseStmt>(this)->getSubStmt();
912   return cast<DefaultStmt>(this)->getSubStmt();
913 }
914 
WhileStmt(const ASTContext & C,VarDecl * Var,Expr * cond,Stmt * body,SourceLocation WL)915 WhileStmt::WhileStmt(const ASTContext &C, VarDecl *Var, Expr *cond, Stmt *body,
916                      SourceLocation WL)
917   : Stmt(WhileStmtClass) {
918   setConditionVariable(C, Var);
919   SubExprs[COND] = cond;
920   SubExprs[BODY] = body;
921   WhileLoc = WL;
922 }
923 
getConditionVariable() const924 VarDecl *WhileStmt::getConditionVariable() const {
925   if (!SubExprs[VAR])
926     return nullptr;
927 
928   DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
929   return cast<VarDecl>(DS->getSingleDecl());
930 }
931 
setConditionVariable(const ASTContext & C,VarDecl * V)932 void WhileStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
933   if (!V) {
934     SubExprs[VAR] = nullptr;
935     return;
936   }
937 
938   SourceRange VarRange = V->getSourceRange();
939   SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
940                                    VarRange.getEnd());
941 }
942 
943 // IndirectGotoStmt
getConstantTarget()944 LabelDecl *IndirectGotoStmt::getConstantTarget() {
945   if (AddrLabelExpr *E =
946         dyn_cast<AddrLabelExpr>(getTarget()->IgnoreParenImpCasts()))
947     return E->getLabel();
948   return nullptr;
949 }
950 
951 // ReturnStmt
getRetValue() const952 const Expr* ReturnStmt::getRetValue() const {
953   return cast_or_null<Expr>(RetExpr);
954 }
getRetValue()955 Expr* ReturnStmt::getRetValue() {
956   return cast_or_null<Expr>(RetExpr);
957 }
958 
SEHTryStmt(bool IsCXXTry,SourceLocation TryLoc,Stmt * TryBlock,Stmt * Handler)959 SEHTryStmt::SEHTryStmt(bool IsCXXTry,
960                        SourceLocation TryLoc,
961                        Stmt *TryBlock,
962                        Stmt *Handler)
963   : Stmt(SEHTryStmtClass),
964     IsCXXTry(IsCXXTry),
965     TryLoc(TryLoc)
966 {
967   Children[TRY]     = TryBlock;
968   Children[HANDLER] = Handler;
969 }
970 
Create(const ASTContext & C,bool IsCXXTry,SourceLocation TryLoc,Stmt * TryBlock,Stmt * Handler)971 SEHTryStmt* SEHTryStmt::Create(const ASTContext &C, bool IsCXXTry,
972                                SourceLocation TryLoc, Stmt *TryBlock,
973                                Stmt *Handler) {
974   return new(C) SEHTryStmt(IsCXXTry,TryLoc,TryBlock,Handler);
975 }
976 
getExceptHandler() const977 SEHExceptStmt* SEHTryStmt::getExceptHandler() const {
978   return dyn_cast<SEHExceptStmt>(getHandler());
979 }
980 
getFinallyHandler() const981 SEHFinallyStmt* SEHTryStmt::getFinallyHandler() const {
982   return dyn_cast<SEHFinallyStmt>(getHandler());
983 }
984 
SEHExceptStmt(SourceLocation Loc,Expr * FilterExpr,Stmt * Block)985 SEHExceptStmt::SEHExceptStmt(SourceLocation Loc,
986                              Expr *FilterExpr,
987                              Stmt *Block)
988   : Stmt(SEHExceptStmtClass),
989     Loc(Loc)
990 {
991   Children[FILTER_EXPR] = FilterExpr;
992   Children[BLOCK]       = Block;
993 }
994 
Create(const ASTContext & C,SourceLocation Loc,Expr * FilterExpr,Stmt * Block)995 SEHExceptStmt* SEHExceptStmt::Create(const ASTContext &C, SourceLocation Loc,
996                                      Expr *FilterExpr, Stmt *Block) {
997   return new(C) SEHExceptStmt(Loc,FilterExpr,Block);
998 }
999 
SEHFinallyStmt(SourceLocation Loc,Stmt * Block)1000 SEHFinallyStmt::SEHFinallyStmt(SourceLocation Loc,
1001                                Stmt *Block)
1002   : Stmt(SEHFinallyStmtClass),
1003     Loc(Loc),
1004     Block(Block)
1005 {}
1006 
Create(const ASTContext & C,SourceLocation Loc,Stmt * Block)1007 SEHFinallyStmt* SEHFinallyStmt::Create(const ASTContext &C, SourceLocation Loc,
1008                                        Stmt *Block) {
1009   return new(C)SEHFinallyStmt(Loc,Block);
1010 }
1011 
getStoredCaptures() const1012 CapturedStmt::Capture *CapturedStmt::getStoredCaptures() const {
1013   unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (NumCaptures + 1);
1014 
1015   // Offset of the first Capture object.
1016   unsigned FirstCaptureOffset =
1017     llvm::RoundUpToAlignment(Size, llvm::alignOf<Capture>());
1018 
1019   return reinterpret_cast<Capture *>(
1020       reinterpret_cast<char *>(const_cast<CapturedStmt *>(this))
1021       + FirstCaptureOffset);
1022 }
1023 
CapturedStmt(Stmt * S,CapturedRegionKind Kind,ArrayRef<Capture> Captures,ArrayRef<Expr * > CaptureInits,CapturedDecl * CD,RecordDecl * RD)1024 CapturedStmt::CapturedStmt(Stmt *S, CapturedRegionKind Kind,
1025                            ArrayRef<Capture> Captures,
1026                            ArrayRef<Expr *> CaptureInits,
1027                            CapturedDecl *CD,
1028                            RecordDecl *RD)
1029   : Stmt(CapturedStmtClass), NumCaptures(Captures.size()),
1030     CapDeclAndKind(CD, Kind), TheRecordDecl(RD) {
1031   assert( S && "null captured statement");
1032   assert(CD && "null captured declaration for captured statement");
1033   assert(RD && "null record declaration for captured statement");
1034 
1035   // Copy initialization expressions.
1036   Stmt **Stored = getStoredStmts();
1037   for (unsigned I = 0, N = NumCaptures; I != N; ++I)
1038     *Stored++ = CaptureInits[I];
1039 
1040   // Copy the statement being captured.
1041   *Stored = S;
1042 
1043   // Copy all Capture objects.
1044   Capture *Buffer = getStoredCaptures();
1045   std::copy(Captures.begin(), Captures.end(), Buffer);
1046 }
1047 
CapturedStmt(EmptyShell Empty,unsigned NumCaptures)1048 CapturedStmt::CapturedStmt(EmptyShell Empty, unsigned NumCaptures)
1049   : Stmt(CapturedStmtClass, Empty), NumCaptures(NumCaptures),
1050     CapDeclAndKind(nullptr, CR_Default), TheRecordDecl(nullptr) {
1051   getStoredStmts()[NumCaptures] = nullptr;
1052 }
1053 
Create(const ASTContext & Context,Stmt * S,CapturedRegionKind Kind,ArrayRef<Capture> Captures,ArrayRef<Expr * > CaptureInits,CapturedDecl * CD,RecordDecl * RD)1054 CapturedStmt *CapturedStmt::Create(const ASTContext &Context, Stmt *S,
1055                                    CapturedRegionKind Kind,
1056                                    ArrayRef<Capture> Captures,
1057                                    ArrayRef<Expr *> CaptureInits,
1058                                    CapturedDecl *CD,
1059                                    RecordDecl *RD) {
1060   // The layout is
1061   //
1062   // -----------------------------------------------------------
1063   // | CapturedStmt, Init, ..., Init, S, Capture, ..., Capture |
1064   // ----------------^-------------------^----------------------
1065   //                 getStoredStmts()    getStoredCaptures()
1066   //
1067   // where S is the statement being captured.
1068   //
1069   assert(CaptureInits.size() == Captures.size() && "wrong number of arguments");
1070 
1071   unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (Captures.size() + 1);
1072   if (!Captures.empty()) {
1073     // Realign for the following Capture array.
1074     Size = llvm::RoundUpToAlignment(Size, llvm::alignOf<Capture>());
1075     Size += sizeof(Capture) * Captures.size();
1076   }
1077 
1078   void *Mem = Context.Allocate(Size);
1079   return new (Mem) CapturedStmt(S, Kind, Captures, CaptureInits, CD, RD);
1080 }
1081 
CreateDeserialized(const ASTContext & Context,unsigned NumCaptures)1082 CapturedStmt *CapturedStmt::CreateDeserialized(const ASTContext &Context,
1083                                                unsigned NumCaptures) {
1084   unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (NumCaptures + 1);
1085   if (NumCaptures > 0) {
1086     // Realign for the following Capture array.
1087     Size = llvm::RoundUpToAlignment(Size, llvm::alignOf<Capture>());
1088     Size += sizeof(Capture) * NumCaptures;
1089   }
1090 
1091   void *Mem = Context.Allocate(Size);
1092   return new (Mem) CapturedStmt(EmptyShell(), NumCaptures);
1093 }
1094 
children()1095 Stmt::child_range CapturedStmt::children() {
1096   // Children are captured field initilizers.
1097   return child_range(getStoredStmts(), getStoredStmts() + NumCaptures);
1098 }
1099 
capturesVariable(const VarDecl * Var) const1100 bool CapturedStmt::capturesVariable(const VarDecl *Var) const {
1101   for (const auto &I : captures()) {
1102     if (!I.capturesVariable())
1103       continue;
1104 
1105     // This does not handle variable redeclarations. This should be
1106     // extended to capture variables with redeclarations, for example
1107     // a thread-private variable in OpenMP.
1108     if (I.getCapturedVar() == Var)
1109       return true;
1110   }
1111 
1112   return false;
1113 }
1114 
children()1115 StmtRange OMPClause::children() {
1116   switch(getClauseKind()) {
1117   default : break;
1118 #define OPENMP_CLAUSE(Name, Class)                                       \
1119   case OMPC_ ## Name : return static_cast<Class *>(this)->children();
1120 #include "clang/Basic/OpenMPKinds.def"
1121   }
1122   llvm_unreachable("unknown OMPClause");
1123 }
1124 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc,ArrayRef<Expr * > VL)1125 OMPPrivateClause *OMPPrivateClause::Create(const ASTContext &C,
1126                                            SourceLocation StartLoc,
1127                                            SourceLocation LParenLoc,
1128                                            SourceLocation EndLoc,
1129                                            ArrayRef<Expr *> VL) {
1130   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPPrivateClause),
1131                                                   llvm::alignOf<Expr *>()) +
1132                          sizeof(Expr *) * VL.size());
1133   OMPPrivateClause *Clause = new (Mem) OMPPrivateClause(StartLoc, LParenLoc,
1134                                                         EndLoc, VL.size());
1135   Clause->setVarRefs(VL);
1136   return Clause;
1137 }
1138 
CreateEmpty(const ASTContext & C,unsigned N)1139 OMPPrivateClause *OMPPrivateClause::CreateEmpty(const ASTContext &C,
1140                                                 unsigned N) {
1141   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPPrivateClause),
1142                                                   llvm::alignOf<Expr *>()) +
1143                          sizeof(Expr *) * N);
1144   return new (Mem) OMPPrivateClause(N);
1145 }
1146 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc,ArrayRef<Expr * > VL)1147 OMPFirstprivateClause *OMPFirstprivateClause::Create(const ASTContext &C,
1148                                                      SourceLocation StartLoc,
1149                                                      SourceLocation LParenLoc,
1150                                                      SourceLocation EndLoc,
1151                                                      ArrayRef<Expr *> VL) {
1152   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPFirstprivateClause),
1153                                                   llvm::alignOf<Expr *>()) +
1154                          sizeof(Expr *) * VL.size());
1155   OMPFirstprivateClause *Clause = new (Mem) OMPFirstprivateClause(StartLoc,
1156                                                                   LParenLoc,
1157                                                                   EndLoc,
1158                                                                   VL.size());
1159   Clause->setVarRefs(VL);
1160   return Clause;
1161 }
1162 
CreateEmpty(const ASTContext & C,unsigned N)1163 OMPFirstprivateClause *OMPFirstprivateClause::CreateEmpty(const ASTContext &C,
1164                                                           unsigned N) {
1165   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPFirstprivateClause),
1166                                                   llvm::alignOf<Expr *>()) +
1167                          sizeof(Expr *) * N);
1168   return new (Mem) OMPFirstprivateClause(N);
1169 }
1170 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc,ArrayRef<Expr * > VL)1171 OMPLastprivateClause *OMPLastprivateClause::Create(const ASTContext &C,
1172                                                    SourceLocation StartLoc,
1173                                                    SourceLocation LParenLoc,
1174                                                    SourceLocation EndLoc,
1175                                                    ArrayRef<Expr *> VL) {
1176   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLastprivateClause),
1177                                                   llvm::alignOf<Expr *>()) +
1178                          sizeof(Expr *) * VL.size());
1179   OMPLastprivateClause *Clause =
1180       new (Mem) OMPLastprivateClause(StartLoc, LParenLoc, EndLoc, VL.size());
1181   Clause->setVarRefs(VL);
1182   return Clause;
1183 }
1184 
CreateEmpty(const ASTContext & C,unsigned N)1185 OMPLastprivateClause *OMPLastprivateClause::CreateEmpty(const ASTContext &C,
1186                                                         unsigned N) {
1187   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLastprivateClause),
1188                                                   llvm::alignOf<Expr *>()) +
1189                          sizeof(Expr *) * N);
1190   return new (Mem) OMPLastprivateClause(N);
1191 }
1192 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc,ArrayRef<Expr * > VL)1193 OMPSharedClause *OMPSharedClause::Create(const ASTContext &C,
1194                                          SourceLocation StartLoc,
1195                                          SourceLocation LParenLoc,
1196                                          SourceLocation EndLoc,
1197                                          ArrayRef<Expr *> VL) {
1198   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPSharedClause),
1199                                                   llvm::alignOf<Expr *>()) +
1200                          sizeof(Expr *) * VL.size());
1201   OMPSharedClause *Clause = new (Mem) OMPSharedClause(StartLoc, LParenLoc,
1202                                                       EndLoc, VL.size());
1203   Clause->setVarRefs(VL);
1204   return Clause;
1205 }
1206 
CreateEmpty(const ASTContext & C,unsigned N)1207 OMPSharedClause *OMPSharedClause::CreateEmpty(const ASTContext &C,
1208                                               unsigned N) {
1209   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPSharedClause),
1210                                                   llvm::alignOf<Expr *>()) +
1211                          sizeof(Expr *) * N);
1212   return new (Mem) OMPSharedClause(N);
1213 }
1214 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation ColonLoc,SourceLocation EndLoc,ArrayRef<Expr * > VL,Expr * Step)1215 OMPLinearClause *OMPLinearClause::Create(const ASTContext &C,
1216                                          SourceLocation StartLoc,
1217                                          SourceLocation LParenLoc,
1218                                          SourceLocation ColonLoc,
1219                                          SourceLocation EndLoc,
1220                                          ArrayRef<Expr *> VL, Expr *Step) {
1221   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLinearClause),
1222                                                   llvm::alignOf<Expr *>()) +
1223                          sizeof(Expr *) * (VL.size() + 1));
1224   OMPLinearClause *Clause = new (Mem)
1225       OMPLinearClause(StartLoc, LParenLoc, ColonLoc, EndLoc, VL.size());
1226   Clause->setVarRefs(VL);
1227   Clause->setStep(Step);
1228   return Clause;
1229 }
1230 
CreateEmpty(const ASTContext & C,unsigned NumVars)1231 OMPLinearClause *OMPLinearClause::CreateEmpty(const ASTContext &C,
1232                                               unsigned NumVars) {
1233   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLinearClause),
1234                                                   llvm::alignOf<Expr *>()) +
1235                          sizeof(Expr *) * (NumVars + 1));
1236   return new (Mem) OMPLinearClause(NumVars);
1237 }
1238 
1239 OMPAlignedClause *
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation ColonLoc,SourceLocation EndLoc,ArrayRef<Expr * > VL,Expr * A)1240 OMPAlignedClause::Create(const ASTContext &C, SourceLocation StartLoc,
1241                          SourceLocation LParenLoc, SourceLocation ColonLoc,
1242                          SourceLocation EndLoc, ArrayRef<Expr *> VL, Expr *A) {
1243   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPAlignedClause),
1244                                                   llvm::alignOf<Expr *>()) +
1245                          sizeof(Expr *) * (VL.size() + 1));
1246   OMPAlignedClause *Clause = new (Mem)
1247       OMPAlignedClause(StartLoc, LParenLoc, ColonLoc, EndLoc, VL.size());
1248   Clause->setVarRefs(VL);
1249   Clause->setAlignment(A);
1250   return Clause;
1251 }
1252 
CreateEmpty(const ASTContext & C,unsigned NumVars)1253 OMPAlignedClause *OMPAlignedClause::CreateEmpty(const ASTContext &C,
1254                                                 unsigned NumVars) {
1255   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPAlignedClause),
1256                                                   llvm::alignOf<Expr *>()) +
1257                          sizeof(Expr *) * (NumVars + 1));
1258   return new (Mem) OMPAlignedClause(NumVars);
1259 }
1260 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc,ArrayRef<Expr * > VL)1261 OMPCopyinClause *OMPCopyinClause::Create(const ASTContext &C,
1262                                          SourceLocation StartLoc,
1263                                          SourceLocation LParenLoc,
1264                                          SourceLocation EndLoc,
1265                                          ArrayRef<Expr *> VL) {
1266   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyinClause),
1267                                                   llvm::alignOf<Expr *>()) +
1268                          sizeof(Expr *) * VL.size());
1269   OMPCopyinClause *Clause = new (Mem) OMPCopyinClause(StartLoc, LParenLoc,
1270                                                       EndLoc, VL.size());
1271   Clause->setVarRefs(VL);
1272   return Clause;
1273 }
1274 
CreateEmpty(const ASTContext & C,unsigned N)1275 OMPCopyinClause *OMPCopyinClause::CreateEmpty(const ASTContext &C,
1276                                               unsigned N) {
1277   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyinClause),
1278                                                   llvm::alignOf<Expr *>()) +
1279                          sizeof(Expr *) * N);
1280   return new (Mem) OMPCopyinClause(N);
1281 }
1282 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc,ArrayRef<Expr * > VL)1283 OMPCopyprivateClause *OMPCopyprivateClause::Create(const ASTContext &C,
1284                                                    SourceLocation StartLoc,
1285                                                    SourceLocation LParenLoc,
1286                                                    SourceLocation EndLoc,
1287                                                    ArrayRef<Expr *> VL) {
1288   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyprivateClause),
1289                                                   llvm::alignOf<Expr *>()) +
1290                          sizeof(Expr *) * VL.size());
1291   OMPCopyprivateClause *Clause =
1292       new (Mem) OMPCopyprivateClause(StartLoc, LParenLoc, EndLoc, VL.size());
1293   Clause->setVarRefs(VL);
1294   return Clause;
1295 }
1296 
CreateEmpty(const ASTContext & C,unsigned N)1297 OMPCopyprivateClause *OMPCopyprivateClause::CreateEmpty(const ASTContext &C,
1298                                                         unsigned N) {
1299   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyprivateClause),
1300                                                   llvm::alignOf<Expr *>()) +
1301                          sizeof(Expr *) * N);
1302   return new (Mem) OMPCopyprivateClause(N);
1303 }
1304 
setClauses(ArrayRef<OMPClause * > Clauses)1305 void OMPExecutableDirective::setClauses(ArrayRef<OMPClause *> Clauses) {
1306   assert(Clauses.size() == getNumClauses() &&
1307          "Number of clauses is not the same as the preallocated buffer");
1308   std::copy(Clauses.begin(), Clauses.end(), getClauses().begin());
1309 }
1310 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc,SourceLocation ColonLoc,ArrayRef<Expr * > VL,NestedNameSpecifierLoc QualifierLoc,const DeclarationNameInfo & NameInfo)1311 OMPReductionClause *OMPReductionClause::Create(
1312     const ASTContext &C, SourceLocation StartLoc, SourceLocation LParenLoc,
1313     SourceLocation EndLoc, SourceLocation ColonLoc, ArrayRef<Expr *> VL,
1314     NestedNameSpecifierLoc QualifierLoc, const DeclarationNameInfo &NameInfo) {
1315   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPReductionClause),
1316                                                   llvm::alignOf<Expr *>()) +
1317                          sizeof(Expr *) * VL.size());
1318   OMPReductionClause *Clause = new (Mem) OMPReductionClause(
1319       StartLoc, LParenLoc, EndLoc, ColonLoc, VL.size(), QualifierLoc, NameInfo);
1320   Clause->setVarRefs(VL);
1321   return Clause;
1322 }
1323 
CreateEmpty(const ASTContext & C,unsigned N)1324 OMPReductionClause *OMPReductionClause::CreateEmpty(const ASTContext &C,
1325                                                     unsigned N) {
1326   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPReductionClause),
1327                                                   llvm::alignOf<Expr *>()) +
1328                          sizeof(Expr *) * N);
1329   return new (Mem) OMPReductionClause(N);
1330 }
1331 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation EndLoc,ArrayRef<OMPClause * > Clauses,Stmt * AssociatedStmt)1332 OMPParallelDirective *OMPParallelDirective::Create(
1333                                               const ASTContext &C,
1334                                               SourceLocation StartLoc,
1335                                               SourceLocation EndLoc,
1336                                               ArrayRef<OMPClause *> Clauses,
1337                                               Stmt *AssociatedStmt) {
1338   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelDirective),
1339                                            llvm::alignOf<OMPClause *>());
1340   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1341                          sizeof(Stmt *));
1342   OMPParallelDirective *Dir = new (Mem) OMPParallelDirective(StartLoc, EndLoc,
1343                                                              Clauses.size());
1344   Dir->setClauses(Clauses);
1345   Dir->setAssociatedStmt(AssociatedStmt);
1346   return Dir;
1347 }
1348 
CreateEmpty(const ASTContext & C,unsigned NumClauses,EmptyShell)1349 OMPParallelDirective *OMPParallelDirective::CreateEmpty(const ASTContext &C,
1350                                                         unsigned NumClauses,
1351                                                         EmptyShell) {
1352   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelDirective),
1353                                            llvm::alignOf<OMPClause *>());
1354   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1355                          sizeof(Stmt *));
1356   return new (Mem) OMPParallelDirective(NumClauses);
1357 }
1358 
1359 OMPSimdDirective *
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation EndLoc,unsigned CollapsedNum,ArrayRef<OMPClause * > Clauses,Stmt * AssociatedStmt)1360 OMPSimdDirective::Create(const ASTContext &C, SourceLocation StartLoc,
1361                          SourceLocation EndLoc, unsigned CollapsedNum,
1362                          ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
1363   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSimdDirective),
1364                                            llvm::alignOf<OMPClause *>());
1365   void *Mem =
1366       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1367   OMPSimdDirective *Dir = new (Mem)
1368       OMPSimdDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
1369   Dir->setClauses(Clauses);
1370   Dir->setAssociatedStmt(AssociatedStmt);
1371   return Dir;
1372 }
1373 
CreateEmpty(const ASTContext & C,unsigned NumClauses,unsigned CollapsedNum,EmptyShell)1374 OMPSimdDirective *OMPSimdDirective::CreateEmpty(const ASTContext &C,
1375                                                 unsigned NumClauses,
1376                                                 unsigned CollapsedNum,
1377                                                 EmptyShell) {
1378   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSimdDirective),
1379                                            llvm::alignOf<OMPClause *>());
1380   void *Mem =
1381       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1382   return new (Mem) OMPSimdDirective(CollapsedNum, NumClauses);
1383 }
1384 
1385 OMPForDirective *
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation EndLoc,unsigned CollapsedNum,ArrayRef<OMPClause * > Clauses,Stmt * AssociatedStmt)1386 OMPForDirective::Create(const ASTContext &C, SourceLocation StartLoc,
1387                         SourceLocation EndLoc, unsigned CollapsedNum,
1388                         ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
1389   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPForDirective),
1390                                            llvm::alignOf<OMPClause *>());
1391   void *Mem =
1392       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1393   OMPForDirective *Dir =
1394       new (Mem) OMPForDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
1395   Dir->setClauses(Clauses);
1396   Dir->setAssociatedStmt(AssociatedStmt);
1397   return Dir;
1398 }
1399 
CreateEmpty(const ASTContext & C,unsigned NumClauses,unsigned CollapsedNum,EmptyShell)1400 OMPForDirective *OMPForDirective::CreateEmpty(const ASTContext &C,
1401                                               unsigned NumClauses,
1402                                               unsigned CollapsedNum,
1403                                               EmptyShell) {
1404   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPForDirective),
1405                                            llvm::alignOf<OMPClause *>());
1406   void *Mem =
1407       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1408   return new (Mem) OMPForDirective(CollapsedNum, NumClauses);
1409 }
1410 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation EndLoc,ArrayRef<OMPClause * > Clauses,Stmt * AssociatedStmt)1411 OMPSectionsDirective *OMPSectionsDirective::Create(
1412     const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
1413     ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
1414   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionsDirective),
1415                                            llvm::alignOf<OMPClause *>());
1416   void *Mem =
1417       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1418   OMPSectionsDirective *Dir =
1419       new (Mem) OMPSectionsDirective(StartLoc, EndLoc, Clauses.size());
1420   Dir->setClauses(Clauses);
1421   Dir->setAssociatedStmt(AssociatedStmt);
1422   return Dir;
1423 }
1424 
CreateEmpty(const ASTContext & C,unsigned NumClauses,EmptyShell)1425 OMPSectionsDirective *OMPSectionsDirective::CreateEmpty(const ASTContext &C,
1426                                                         unsigned NumClauses,
1427                                                         EmptyShell) {
1428   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionsDirective),
1429                                            llvm::alignOf<OMPClause *>());
1430   void *Mem =
1431       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1432   return new (Mem) OMPSectionsDirective(NumClauses);
1433 }
1434 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation EndLoc,Stmt * AssociatedStmt)1435 OMPSectionDirective *OMPSectionDirective::Create(const ASTContext &C,
1436                                                  SourceLocation StartLoc,
1437                                                  SourceLocation EndLoc,
1438                                                  Stmt *AssociatedStmt) {
1439   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionsDirective),
1440                                            llvm::alignOf<Stmt *>());
1441   void *Mem = C.Allocate(Size + sizeof(Stmt *));
1442   OMPSectionDirective *Dir = new (Mem) OMPSectionDirective(StartLoc, EndLoc);
1443   Dir->setAssociatedStmt(AssociatedStmt);
1444   return Dir;
1445 }
1446 
CreateEmpty(const ASTContext & C,EmptyShell)1447 OMPSectionDirective *OMPSectionDirective::CreateEmpty(const ASTContext &C,
1448                                                       EmptyShell) {
1449   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionDirective),
1450                                            llvm::alignOf<Stmt *>());
1451   void *Mem = C.Allocate(Size + sizeof(Stmt *));
1452   return new (Mem) OMPSectionDirective();
1453 }
1454 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation EndLoc,ArrayRef<OMPClause * > Clauses,Stmt * AssociatedStmt)1455 OMPSingleDirective *OMPSingleDirective::Create(const ASTContext &C,
1456                                                SourceLocation StartLoc,
1457                                                SourceLocation EndLoc,
1458                                                ArrayRef<OMPClause *> Clauses,
1459                                                Stmt *AssociatedStmt) {
1460   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSingleDirective),
1461                                            llvm::alignOf<OMPClause *>());
1462   void *Mem =
1463       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1464   OMPSingleDirective *Dir =
1465       new (Mem) OMPSingleDirective(StartLoc, EndLoc, Clauses.size());
1466   Dir->setClauses(Clauses);
1467   Dir->setAssociatedStmt(AssociatedStmt);
1468   return Dir;
1469 }
1470 
CreateEmpty(const ASTContext & C,unsigned NumClauses,EmptyShell)1471 OMPSingleDirective *OMPSingleDirective::CreateEmpty(const ASTContext &C,
1472                                                     unsigned NumClauses,
1473                                                     EmptyShell) {
1474   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSingleDirective),
1475                                            llvm::alignOf<OMPClause *>());
1476   void *Mem =
1477       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1478   return new (Mem) OMPSingleDirective(NumClauses);
1479 }
1480 
1481 OMPParallelForDirective *
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation EndLoc,unsigned CollapsedNum,ArrayRef<OMPClause * > Clauses,Stmt * AssociatedStmt)1482 OMPParallelForDirective::Create(const ASTContext &C, SourceLocation StartLoc,
1483                                 SourceLocation EndLoc, unsigned CollapsedNum,
1484                                 ArrayRef<OMPClause *> Clauses,
1485                                 Stmt *AssociatedStmt) {
1486   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelForDirective),
1487                                            llvm::alignOf<OMPClause *>());
1488   void *Mem =
1489       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1490   OMPParallelForDirective *Dir = new (Mem)
1491       OMPParallelForDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
1492   Dir->setClauses(Clauses);
1493   Dir->setAssociatedStmt(AssociatedStmt);
1494   return Dir;
1495 }
1496 
1497 OMPParallelForDirective *
CreateEmpty(const ASTContext & C,unsigned NumClauses,unsigned CollapsedNum,EmptyShell)1498 OMPParallelForDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
1499                                      unsigned CollapsedNum, EmptyShell) {
1500   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelForDirective),
1501                                            llvm::alignOf<OMPClause *>());
1502   void *Mem =
1503       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1504   return new (Mem) OMPParallelForDirective(CollapsedNum, NumClauses);
1505 }
1506 
Create(const ASTContext & C,SourceLocation StartLoc,SourceLocation EndLoc,ArrayRef<OMPClause * > Clauses,Stmt * AssociatedStmt)1507 OMPParallelSectionsDirective *OMPParallelSectionsDirective::Create(
1508     const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
1509     ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
1510   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelSectionsDirective),
1511                                            llvm::alignOf<OMPClause *>());
1512   void *Mem =
1513       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1514   OMPParallelSectionsDirective *Dir =
1515       new (Mem) OMPParallelSectionsDirective(StartLoc, EndLoc, Clauses.size());
1516   Dir->setClauses(Clauses);
1517   Dir->setAssociatedStmt(AssociatedStmt);
1518   return Dir;
1519 }
1520 
1521 OMPParallelSectionsDirective *
CreateEmpty(const ASTContext & C,unsigned NumClauses,EmptyShell)1522 OMPParallelSectionsDirective::CreateEmpty(const ASTContext &C,
1523                                           unsigned NumClauses, EmptyShell) {
1524   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelSectionsDirective),
1525                                            llvm::alignOf<OMPClause *>());
1526   void *Mem =
1527       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1528   return new (Mem) OMPParallelSectionsDirective(NumClauses);
1529 }
1530 
1531