xref: /llvm-project-15.0.7/clang/lib/AST/Stmt.cpp (revision ef99e4d8)
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 
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 
52 void *Stmt::operator new(size_t bytes, const ASTContext& C,
53                          unsigned alignment) {
54   return ::operator new(bytes, C, alignment);
55 }
56 
57 const char *Stmt::getStmtClassName() const {
58   return getStmtInfoTableEntry((StmtClass) StmtBits.sClass).Name;
59 }
60 
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 
86 void Stmt::addStmtClass(StmtClass s) {
87   ++getStmtInfoTableEntry(s).Counter;
88 }
89 
90 bool Stmt::StatisticsEnabled = false;
91 void Stmt::EnableStatistics() {
92   StatisticsEnabled = true;
93 }
94 
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 Skip no-op (attributed, compound) container stmts and skip captured
108 /// stmt at the top, if \a IgnoreCaptured is true.
109 Stmt *Stmt::IgnoreContainers(bool IgnoreCaptured) {
110   Stmt *S = this;
111   if (IgnoreCaptured)
112     if (auto CapS = dyn_cast_or_null<CapturedStmt>(S))
113       S = CapS->getCapturedStmt();
114   while (true) {
115     if (auto AS = dyn_cast_or_null<AttributedStmt>(S))
116       S = AS->getSubStmt();
117     else if (auto CS = dyn_cast_or_null<CompoundStmt>(S)) {
118       if (CS->size() != 1)
119         break;
120       S = CS->body_back();
121     } else
122       break;
123   }
124   return S;
125 }
126 
127 /// \brief Strip off all label-like statements.
128 ///
129 /// This will strip off label statements, case statements, attributed
130 /// statements and default statements recursively.
131 const Stmt *Stmt::stripLabelLikeStatements() const {
132   const Stmt *S = this;
133   while (true) {
134     if (const LabelStmt *LS = dyn_cast<LabelStmt>(S))
135       S = LS->getSubStmt();
136     else if (const SwitchCase *SC = dyn_cast<SwitchCase>(S))
137       S = SC->getSubStmt();
138     else if (const AttributedStmt *AS = dyn_cast<AttributedStmt>(S))
139       S = AS->getSubStmt();
140     else
141       return S;
142   }
143 }
144 
145 namespace {
146   struct good {};
147   struct bad {};
148 
149   // These silly little functions have to be static inline to suppress
150   // unused warnings, and they have to be defined to suppress other
151   // warnings.
152   static inline good is_good(good) { return good(); }
153 
154   typedef Stmt::child_range children_t();
155   template <class T> good implements_children(children_t T::*) {
156     return good();
157   }
158   LLVM_ATTRIBUTE_UNUSED
159   static inline bad implements_children(children_t Stmt::*) {
160     return bad();
161   }
162 
163   typedef SourceLocation getLocStart_t() const;
164   template <class T> good implements_getLocStart(getLocStart_t T::*) {
165     return good();
166   }
167   LLVM_ATTRIBUTE_UNUSED
168   static inline bad implements_getLocStart(getLocStart_t Stmt::*) {
169     return bad();
170   }
171 
172   typedef SourceLocation getLocEnd_t() const;
173   template <class T> good implements_getLocEnd(getLocEnd_t T::*) {
174     return good();
175   }
176   LLVM_ATTRIBUTE_UNUSED
177   static inline bad implements_getLocEnd(getLocEnd_t Stmt::*) {
178     return bad();
179   }
180 
181 #define ASSERT_IMPLEMENTS_children(type) \
182   (void) is_good(implements_children(&type::children))
183 #define ASSERT_IMPLEMENTS_getLocStart(type) \
184   (void) is_good(implements_getLocStart(&type::getLocStart))
185 #define ASSERT_IMPLEMENTS_getLocEnd(type) \
186   (void) is_good(implements_getLocEnd(&type::getLocEnd))
187 }
188 
189 /// Check whether the various Stmt classes implement their member
190 /// functions.
191 LLVM_ATTRIBUTE_UNUSED
192 static inline void check_implementations() {
193 #define ABSTRACT_STMT(type)
194 #define STMT(type, base) \
195   ASSERT_IMPLEMENTS_children(type); \
196   ASSERT_IMPLEMENTS_getLocStart(type); \
197   ASSERT_IMPLEMENTS_getLocEnd(type);
198 #include "clang/AST/StmtNodes.inc"
199 }
200 
201 Stmt::child_range Stmt::children() {
202   switch (getStmtClass()) {
203   case Stmt::NoStmtClass: llvm_unreachable("statement without class");
204 #define ABSTRACT_STMT(type)
205 #define STMT(type, base) \
206   case Stmt::type##Class: \
207     return static_cast<type*>(this)->children();
208 #include "clang/AST/StmtNodes.inc"
209   }
210   llvm_unreachable("unknown statement kind!");
211 }
212 
213 // Amusing macro metaprogramming hack: check whether a class provides
214 // a more specific implementation of getSourceRange.
215 //
216 // See also Expr.cpp:getExprLoc().
217 namespace {
218   /// This implementation is used when a class provides a custom
219   /// implementation of getSourceRange.
220   template <class S, class T>
221   SourceRange getSourceRangeImpl(const Stmt *stmt,
222                                  SourceRange (T::*v)() const) {
223     return static_cast<const S*>(stmt)->getSourceRange();
224   }
225 
226   /// This implementation is used when a class doesn't provide a custom
227   /// implementation of getSourceRange.  Overload resolution should pick it over
228   /// the implementation above because it's more specialized according to
229   /// function template partial ordering.
230   template <class S>
231   SourceRange getSourceRangeImpl(const Stmt *stmt,
232                                  SourceRange (Stmt::*v)() const) {
233     return SourceRange(static_cast<const S*>(stmt)->getLocStart(),
234                        static_cast<const S*>(stmt)->getLocEnd());
235   }
236 }
237 
238 SourceRange Stmt::getSourceRange() const {
239   switch (getStmtClass()) {
240   case Stmt::NoStmtClass: llvm_unreachable("statement without class");
241 #define ABSTRACT_STMT(type)
242 #define STMT(type, base) \
243   case Stmt::type##Class: \
244     return getSourceRangeImpl<type>(this, &type::getSourceRange);
245 #include "clang/AST/StmtNodes.inc"
246   }
247   llvm_unreachable("unknown statement kind!");
248 }
249 
250 SourceLocation Stmt::getLocStart() const {
251 //  llvm::errs() << "getLocStart() for " << getStmtClassName() << "\n";
252   switch (getStmtClass()) {
253   case Stmt::NoStmtClass: llvm_unreachable("statement without class");
254 #define ABSTRACT_STMT(type)
255 #define STMT(type, base) \
256   case Stmt::type##Class: \
257     return static_cast<const type*>(this)->getLocStart();
258 #include "clang/AST/StmtNodes.inc"
259   }
260   llvm_unreachable("unknown statement kind");
261 }
262 
263 SourceLocation Stmt::getLocEnd() const {
264   switch (getStmtClass()) {
265   case Stmt::NoStmtClass: llvm_unreachable("statement without class");
266 #define ABSTRACT_STMT(type)
267 #define STMT(type, base) \
268   case Stmt::type##Class: \
269     return static_cast<const type*>(this)->getLocEnd();
270 #include "clang/AST/StmtNodes.inc"
271   }
272   llvm_unreachable("unknown statement kind");
273 }
274 
275 CompoundStmt::CompoundStmt(const ASTContext &C, ArrayRef<Stmt*> Stmts,
276                            SourceLocation LB, SourceLocation RB)
277   : Stmt(CompoundStmtClass), LBracLoc(LB), RBracLoc(RB) {
278   CompoundStmtBits.NumStmts = Stmts.size();
279   assert(CompoundStmtBits.NumStmts == Stmts.size() &&
280          "NumStmts doesn't fit in bits of CompoundStmtBits.NumStmts!");
281 
282   if (Stmts.size() == 0) {
283     Body = nullptr;
284     return;
285   }
286 
287   Body = new (C) Stmt*[Stmts.size()];
288   std::copy(Stmts.begin(), Stmts.end(), Body);
289 }
290 
291 void CompoundStmt::setStmts(const ASTContext &C, Stmt **Stmts,
292                             unsigned NumStmts) {
293   if (this->Body)
294     C.Deallocate(Body);
295   this->CompoundStmtBits.NumStmts = NumStmts;
296 
297   Body = new (C) Stmt*[NumStmts];
298   memcpy(Body, Stmts, sizeof(Stmt *) * NumStmts);
299 }
300 
301 const char *LabelStmt::getName() const {
302   return getDecl()->getIdentifier()->getNameStart();
303 }
304 
305 AttributedStmt *AttributedStmt::Create(const ASTContext &C, SourceLocation Loc,
306                                        ArrayRef<const Attr*> Attrs,
307                                        Stmt *SubStmt) {
308   assert(!Attrs.empty() && "Attrs should not be empty");
309   void *Mem = C.Allocate(sizeof(AttributedStmt) + sizeof(Attr *) * Attrs.size(),
310                          llvm::alignOf<AttributedStmt>());
311   return new (Mem) AttributedStmt(Loc, Attrs, SubStmt);
312 }
313 
314 AttributedStmt *AttributedStmt::CreateEmpty(const ASTContext &C,
315                                             unsigned NumAttrs) {
316   assert(NumAttrs > 0 && "NumAttrs should be greater than zero");
317   void *Mem = C.Allocate(sizeof(AttributedStmt) + sizeof(Attr *) * NumAttrs,
318                          llvm::alignOf<AttributedStmt>());
319   return new (Mem) AttributedStmt(EmptyShell(), NumAttrs);
320 }
321 
322 std::string AsmStmt::generateAsmString(const ASTContext &C) const {
323   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
324     return gccAsmStmt->generateAsmString(C);
325   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
326     return msAsmStmt->generateAsmString(C);
327   llvm_unreachable("unknown asm statement kind!");
328 }
329 
330 StringRef AsmStmt::getOutputConstraint(unsigned i) const {
331   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
332     return gccAsmStmt->getOutputConstraint(i);
333   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
334     return msAsmStmt->getOutputConstraint(i);
335   llvm_unreachable("unknown asm statement kind!");
336 }
337 
338 const Expr *AsmStmt::getOutputExpr(unsigned i) const {
339   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
340     return gccAsmStmt->getOutputExpr(i);
341   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
342     return msAsmStmt->getOutputExpr(i);
343   llvm_unreachable("unknown asm statement kind!");
344 }
345 
346 StringRef AsmStmt::getInputConstraint(unsigned i) const {
347   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
348     return gccAsmStmt->getInputConstraint(i);
349   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
350     return msAsmStmt->getInputConstraint(i);
351   llvm_unreachable("unknown asm statement kind!");
352 }
353 
354 const Expr *AsmStmt::getInputExpr(unsigned i) const {
355   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
356     return gccAsmStmt->getInputExpr(i);
357   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
358     return msAsmStmt->getInputExpr(i);
359   llvm_unreachable("unknown asm statement kind!");
360 }
361 
362 StringRef AsmStmt::getClobber(unsigned i) const {
363   if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
364     return gccAsmStmt->getClobber(i);
365   if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
366     return msAsmStmt->getClobber(i);
367   llvm_unreachable("unknown asm statement kind!");
368 }
369 
370 /// getNumPlusOperands - Return the number of output operands that have a "+"
371 /// constraint.
372 unsigned AsmStmt::getNumPlusOperands() const {
373   unsigned Res = 0;
374   for (unsigned i = 0, e = getNumOutputs(); i != e; ++i)
375     if (isOutputPlusConstraint(i))
376       ++Res;
377   return Res;
378 }
379 
380 char GCCAsmStmt::AsmStringPiece::getModifier() const {
381   assert(isOperand() && "Only Operands can have modifiers.");
382   return isLetter(Str[0]) ? Str[0] : '\0';
383 }
384 
385 StringRef GCCAsmStmt::getClobber(unsigned i) const {
386   return getClobberStringLiteral(i)->getString();
387 }
388 
389 Expr *GCCAsmStmt::getOutputExpr(unsigned i) {
390   return cast<Expr>(Exprs[i]);
391 }
392 
393 /// getOutputConstraint - Return the constraint string for the specified
394 /// output operand.  All output constraints are known to be non-empty (either
395 /// '=' or '+').
396 StringRef GCCAsmStmt::getOutputConstraint(unsigned i) const {
397   return getOutputConstraintLiteral(i)->getString();
398 }
399 
400 Expr *GCCAsmStmt::getInputExpr(unsigned i) {
401   return cast<Expr>(Exprs[i + NumOutputs]);
402 }
403 void GCCAsmStmt::setInputExpr(unsigned i, Expr *E) {
404   Exprs[i + NumOutputs] = E;
405 }
406 
407 /// getInputConstraint - Return the specified input constraint.  Unlike output
408 /// constraints, these can be empty.
409 StringRef GCCAsmStmt::getInputConstraint(unsigned i) const {
410   return getInputConstraintLiteral(i)->getString();
411 }
412 
413 void GCCAsmStmt::setOutputsAndInputsAndClobbers(const ASTContext &C,
414                                                 IdentifierInfo **Names,
415                                                 StringLiteral **Constraints,
416                                                 Stmt **Exprs,
417                                                 unsigned NumOutputs,
418                                                 unsigned NumInputs,
419                                                 StringLiteral **Clobbers,
420                                                 unsigned NumClobbers) {
421   this->NumOutputs = NumOutputs;
422   this->NumInputs = NumInputs;
423   this->NumClobbers = NumClobbers;
424 
425   unsigned NumExprs = NumOutputs + NumInputs;
426 
427   C.Deallocate(this->Names);
428   this->Names = new (C) IdentifierInfo*[NumExprs];
429   std::copy(Names, Names + NumExprs, this->Names);
430 
431   C.Deallocate(this->Exprs);
432   this->Exprs = new (C) Stmt*[NumExprs];
433   std::copy(Exprs, Exprs + NumExprs, this->Exprs);
434 
435   C.Deallocate(this->Constraints);
436   this->Constraints = new (C) StringLiteral*[NumExprs];
437   std::copy(Constraints, Constraints + NumExprs, this->Constraints);
438 
439   C.Deallocate(this->Clobbers);
440   this->Clobbers = new (C) StringLiteral*[NumClobbers];
441   std::copy(Clobbers, Clobbers + NumClobbers, this->Clobbers);
442 }
443 
444 /// getNamedOperand - Given a symbolic operand reference like %[foo],
445 /// translate this into a numeric value needed to reference the same operand.
446 /// This returns -1 if the operand name is invalid.
447 int GCCAsmStmt::getNamedOperand(StringRef SymbolicName) const {
448   unsigned NumPlusOperands = 0;
449 
450   // Check if this is an output operand.
451   for (unsigned i = 0, e = getNumOutputs(); i != e; ++i) {
452     if (getOutputName(i) == SymbolicName)
453       return i;
454   }
455 
456   for (unsigned i = 0, e = getNumInputs(); i != e; ++i)
457     if (getInputName(i) == SymbolicName)
458       return getNumOutputs() + NumPlusOperands + i;
459 
460   // Not found.
461   return -1;
462 }
463 
464 /// AnalyzeAsmString - Analyze the asm string of the current asm, decomposing
465 /// it into pieces.  If the asm string is erroneous, emit errors and return
466 /// true, otherwise return false.
467 unsigned GCCAsmStmt::AnalyzeAsmString(SmallVectorImpl<AsmStringPiece>&Pieces,
468                                 const ASTContext &C, unsigned &DiagOffs) const {
469   StringRef Str = getAsmString()->getString();
470   const char *StrStart = Str.begin();
471   const char *StrEnd = Str.end();
472   const char *CurPtr = StrStart;
473 
474   // "Simple" inline asms have no constraints or operands, just convert the asm
475   // string to escape $'s.
476   if (isSimple()) {
477     std::string Result;
478     for (; CurPtr != StrEnd; ++CurPtr) {
479       switch (*CurPtr) {
480       case '$':
481         Result += "$$";
482         break;
483       default:
484         Result += *CurPtr;
485         break;
486       }
487     }
488     Pieces.push_back(AsmStringPiece(Result));
489     return 0;
490   }
491 
492   // CurStringPiece - The current string that we are building up as we scan the
493   // asm string.
494   std::string CurStringPiece;
495 
496   bool HasVariants = !C.getTargetInfo().hasNoAsmVariants();
497 
498   while (1) {
499     // Done with the string?
500     if (CurPtr == StrEnd) {
501       if (!CurStringPiece.empty())
502         Pieces.push_back(AsmStringPiece(CurStringPiece));
503       return 0;
504     }
505 
506     char CurChar = *CurPtr++;
507     switch (CurChar) {
508     case '$': CurStringPiece += "$$"; continue;
509     case '{': CurStringPiece += (HasVariants ? "$(" : "{"); continue;
510     case '|': CurStringPiece += (HasVariants ? "$|" : "|"); continue;
511     case '}': CurStringPiece += (HasVariants ? "$)" : "}"); continue;
512     case '%':
513       break;
514     default:
515       CurStringPiece += CurChar;
516       continue;
517     }
518 
519     // Escaped "%" character in asm string.
520     if (CurPtr == StrEnd) {
521       // % at end of string is invalid (no escape).
522       DiagOffs = CurPtr-StrStart-1;
523       return diag::err_asm_invalid_escape;
524     }
525 
526     char EscapedChar = *CurPtr++;
527     if (EscapedChar == '%') {  // %% -> %
528       // Escaped percentage sign.
529       CurStringPiece += '%';
530       continue;
531     }
532 
533     if (EscapedChar == '=') {  // %= -> Generate an unique ID.
534       CurStringPiece += "${:uid}";
535       continue;
536     }
537 
538     // Otherwise, we have an operand.  If we have accumulated a string so far,
539     // add it to the Pieces list.
540     if (!CurStringPiece.empty()) {
541       Pieces.push_back(AsmStringPiece(CurStringPiece));
542       CurStringPiece.clear();
543     }
544 
545     // Handle operands that have asmSymbolicName (e.g., %x[foo]) and those that
546     // don't (e.g., %x4). 'x' following the '%' is the constraint modifier.
547 
548     const char *Begin = CurPtr - 1; // Points to the character following '%'.
549     const char *Percent = Begin - 1; // Points to '%'.
550 
551     if (isLetter(EscapedChar)) {
552       if (CurPtr == StrEnd) { // Premature end.
553         DiagOffs = CurPtr-StrStart-1;
554         return diag::err_asm_invalid_escape;
555       }
556       EscapedChar = *CurPtr++;
557     }
558 
559     const TargetInfo &TI = C.getTargetInfo();
560     const SourceManager &SM = C.getSourceManager();
561     const LangOptions &LO = C.getLangOpts();
562 
563     // Handle operands that don't have asmSymbolicName (e.g., %x4).
564     if (isDigit(EscapedChar)) {
565       // %n - Assembler operand n
566       unsigned N = 0;
567 
568       --CurPtr;
569       while (CurPtr != StrEnd && isDigit(*CurPtr))
570         N = N*10 + ((*CurPtr++)-'0');
571 
572       unsigned NumOperands =
573         getNumOutputs() + getNumPlusOperands() + getNumInputs();
574       if (N >= NumOperands) {
575         DiagOffs = CurPtr-StrStart-1;
576         return diag::err_asm_invalid_operand_number;
577       }
578 
579       // Str contains "x4" (Operand without the leading %).
580       std::string Str(Begin, CurPtr - Begin);
581 
582       // (BeginLoc, EndLoc) represents the range of the operand we are currently
583       // processing. Unlike Str, the range includes the leading '%'.
584       SourceLocation BeginLoc =
585           getAsmString()->getLocationOfByte(Percent - StrStart, SM, LO, TI);
586       SourceLocation EndLoc =
587           getAsmString()->getLocationOfByte(CurPtr - StrStart, SM, LO, TI);
588 
589       Pieces.push_back(AsmStringPiece(N, Str, BeginLoc, EndLoc));
590       continue;
591     }
592 
593     // Handle operands that have asmSymbolicName (e.g., %x[foo]).
594     if (EscapedChar == '[') {
595       DiagOffs = CurPtr-StrStart-1;
596 
597       // Find the ']'.
598       const char *NameEnd = (const char*)memchr(CurPtr, ']', StrEnd-CurPtr);
599       if (NameEnd == nullptr)
600         return diag::err_asm_unterminated_symbolic_operand_name;
601       if (NameEnd == CurPtr)
602         return diag::err_asm_empty_symbolic_operand_name;
603 
604       StringRef SymbolicName(CurPtr, NameEnd - CurPtr);
605 
606       int N = getNamedOperand(SymbolicName);
607       if (N == -1) {
608         // Verify that an operand with that name exists.
609         DiagOffs = CurPtr-StrStart;
610         return diag::err_asm_unknown_symbolic_operand_name;
611       }
612 
613       // Str contains "x[foo]" (Operand without the leading %).
614       std::string Str(Begin, NameEnd + 1 - Begin);
615 
616       // (BeginLoc, EndLoc) represents the range of the operand we are currently
617       // processing. Unlike Str, the range includes the leading '%'.
618       SourceLocation BeginLoc =
619           getAsmString()->getLocationOfByte(Percent - StrStart, SM, LO, TI);
620       SourceLocation EndLoc =
621           getAsmString()->getLocationOfByte(NameEnd + 1 - StrStart, SM, LO, TI);
622 
623       Pieces.push_back(AsmStringPiece(N, Str, BeginLoc, EndLoc));
624 
625       CurPtr = NameEnd+1;
626       continue;
627     }
628 
629     DiagOffs = CurPtr-StrStart-1;
630     return diag::err_asm_invalid_escape;
631   }
632 }
633 
634 /// Assemble final IR asm string (GCC-style).
635 std::string GCCAsmStmt::generateAsmString(const ASTContext &C) const {
636   // Analyze the asm string to decompose it into its pieces.  We know that Sema
637   // has already done this, so it is guaranteed to be successful.
638   SmallVector<GCCAsmStmt::AsmStringPiece, 4> Pieces;
639   unsigned DiagOffs;
640   AnalyzeAsmString(Pieces, C, DiagOffs);
641 
642   std::string AsmString;
643   for (unsigned i = 0, e = Pieces.size(); i != e; ++i) {
644     if (Pieces[i].isString())
645       AsmString += Pieces[i].getString();
646     else if (Pieces[i].getModifier() == '\0')
647       AsmString += '$' + llvm::utostr(Pieces[i].getOperandNo());
648     else
649       AsmString += "${" + llvm::utostr(Pieces[i].getOperandNo()) + ':' +
650                    Pieces[i].getModifier() + '}';
651   }
652   return AsmString;
653 }
654 
655 /// Assemble final IR asm string (MS-style).
656 std::string MSAsmStmt::generateAsmString(const ASTContext &C) const {
657   // FIXME: This needs to be translated into the IR string representation.
658   return AsmStr;
659 }
660 
661 Expr *MSAsmStmt::getOutputExpr(unsigned i) {
662   return cast<Expr>(Exprs[i]);
663 }
664 
665 Expr *MSAsmStmt::getInputExpr(unsigned i) {
666   return cast<Expr>(Exprs[i + NumOutputs]);
667 }
668 void MSAsmStmt::setInputExpr(unsigned i, Expr *E) {
669   Exprs[i + NumOutputs] = E;
670 }
671 
672 QualType CXXCatchStmt::getCaughtType() const {
673   if (ExceptionDecl)
674     return ExceptionDecl->getType();
675   return QualType();
676 }
677 
678 //===----------------------------------------------------------------------===//
679 // Constructors
680 //===----------------------------------------------------------------------===//
681 
682 GCCAsmStmt::GCCAsmStmt(const ASTContext &C, SourceLocation asmloc,
683                        bool issimple, bool isvolatile, unsigned numoutputs,
684                        unsigned numinputs, IdentifierInfo **names,
685                        StringLiteral **constraints, Expr **exprs,
686                        StringLiteral *asmstr, unsigned numclobbers,
687                        StringLiteral **clobbers, SourceLocation rparenloc)
688   : AsmStmt(GCCAsmStmtClass, asmloc, issimple, isvolatile, numoutputs,
689             numinputs, numclobbers), RParenLoc(rparenloc), AsmStr(asmstr) {
690 
691   unsigned NumExprs = NumOutputs + NumInputs;
692 
693   Names = new (C) IdentifierInfo*[NumExprs];
694   std::copy(names, names + NumExprs, Names);
695 
696   Exprs = new (C) Stmt*[NumExprs];
697   std::copy(exprs, exprs + NumExprs, Exprs);
698 
699   Constraints = new (C) StringLiteral*[NumExprs];
700   std::copy(constraints, constraints + NumExprs, Constraints);
701 
702   Clobbers = new (C) StringLiteral*[NumClobbers];
703   std::copy(clobbers, clobbers + NumClobbers, Clobbers);
704 }
705 
706 MSAsmStmt::MSAsmStmt(const ASTContext &C, SourceLocation asmloc,
707                      SourceLocation lbraceloc, bool issimple, bool isvolatile,
708                      ArrayRef<Token> asmtoks, unsigned numoutputs,
709                      unsigned numinputs,
710                      ArrayRef<StringRef> constraints, ArrayRef<Expr*> exprs,
711                      StringRef asmstr, ArrayRef<StringRef> clobbers,
712                      SourceLocation endloc)
713   : AsmStmt(MSAsmStmtClass, asmloc, issimple, isvolatile, numoutputs,
714             numinputs, clobbers.size()), LBraceLoc(lbraceloc),
715             EndLoc(endloc), NumAsmToks(asmtoks.size()) {
716 
717   initialize(C, asmstr, asmtoks, constraints, exprs, clobbers);
718 }
719 
720 static StringRef copyIntoContext(const ASTContext &C, StringRef str) {
721   size_t size = str.size();
722   char *buffer = new (C) char[size];
723   memcpy(buffer, str.data(), size);
724   return StringRef(buffer, size);
725 }
726 
727 void MSAsmStmt::initialize(const ASTContext &C, StringRef asmstr,
728                            ArrayRef<Token> asmtoks,
729                            ArrayRef<StringRef> constraints,
730                            ArrayRef<Expr*> exprs,
731                            ArrayRef<StringRef> clobbers) {
732   assert(NumAsmToks == asmtoks.size());
733   assert(NumClobbers == clobbers.size());
734 
735   unsigned NumExprs = exprs.size();
736   assert(NumExprs == NumOutputs + NumInputs);
737   assert(NumExprs == constraints.size());
738 
739   AsmStr = copyIntoContext(C, asmstr);
740 
741   Exprs = new (C) Stmt*[NumExprs];
742   for (unsigned i = 0, e = NumExprs; i != e; ++i)
743     Exprs[i] = exprs[i];
744 
745   AsmToks = new (C) Token[NumAsmToks];
746   for (unsigned i = 0, e = NumAsmToks; i != e; ++i)
747     AsmToks[i] = asmtoks[i];
748 
749   Constraints = new (C) StringRef[NumExprs];
750   for (unsigned i = 0, e = NumExprs; i != e; ++i) {
751     Constraints[i] = copyIntoContext(C, constraints[i]);
752   }
753 
754   Clobbers = new (C) StringRef[NumClobbers];
755   for (unsigned i = 0, e = NumClobbers; i != e; ++i) {
756     // FIXME: Avoid the allocation/copy if at all possible.
757     Clobbers[i] = copyIntoContext(C, clobbers[i]);
758   }
759 }
760 
761 ObjCForCollectionStmt::ObjCForCollectionStmt(Stmt *Elem, Expr *Collect,
762                                              Stmt *Body,  SourceLocation FCL,
763                                              SourceLocation RPL)
764 : Stmt(ObjCForCollectionStmtClass) {
765   SubExprs[ELEM] = Elem;
766   SubExprs[COLLECTION] = Collect;
767   SubExprs[BODY] = Body;
768   ForLoc = FCL;
769   RParenLoc = RPL;
770 }
771 
772 ObjCAtTryStmt::ObjCAtTryStmt(SourceLocation atTryLoc, Stmt *atTryStmt,
773                              Stmt **CatchStmts, unsigned NumCatchStmts,
774                              Stmt *atFinallyStmt)
775   : Stmt(ObjCAtTryStmtClass), AtTryLoc(atTryLoc),
776     NumCatchStmts(NumCatchStmts), HasFinally(atFinallyStmt != nullptr) {
777   Stmt **Stmts = getStmts();
778   Stmts[0] = atTryStmt;
779   for (unsigned I = 0; I != NumCatchStmts; ++I)
780     Stmts[I + 1] = CatchStmts[I];
781 
782   if (HasFinally)
783     Stmts[NumCatchStmts + 1] = atFinallyStmt;
784 }
785 
786 ObjCAtTryStmt *ObjCAtTryStmt::Create(const ASTContext &Context,
787                                      SourceLocation atTryLoc,
788                                      Stmt *atTryStmt,
789                                      Stmt **CatchStmts,
790                                      unsigned NumCatchStmts,
791                                      Stmt *atFinallyStmt) {
792   unsigned Size = sizeof(ObjCAtTryStmt) +
793     (1 + NumCatchStmts + (atFinallyStmt != nullptr)) * sizeof(Stmt *);
794   void *Mem = Context.Allocate(Size, llvm::alignOf<ObjCAtTryStmt>());
795   return new (Mem) ObjCAtTryStmt(atTryLoc, atTryStmt, CatchStmts, NumCatchStmts,
796                                  atFinallyStmt);
797 }
798 
799 ObjCAtTryStmt *ObjCAtTryStmt::CreateEmpty(const ASTContext &Context,
800                                           unsigned NumCatchStmts,
801                                           bool HasFinally) {
802   unsigned Size = sizeof(ObjCAtTryStmt) +
803     (1 + NumCatchStmts + HasFinally) * sizeof(Stmt *);
804   void *Mem = Context.Allocate(Size, llvm::alignOf<ObjCAtTryStmt>());
805   return new (Mem) ObjCAtTryStmt(EmptyShell(), NumCatchStmts, HasFinally);
806 }
807 
808 SourceLocation ObjCAtTryStmt::getLocEnd() const {
809   if (HasFinally)
810     return getFinallyStmt()->getLocEnd();
811   if (NumCatchStmts)
812     return getCatchStmt(NumCatchStmts - 1)->getLocEnd();
813   return getTryBody()->getLocEnd();
814 }
815 
816 CXXTryStmt *CXXTryStmt::Create(const ASTContext &C, SourceLocation tryLoc,
817                                Stmt *tryBlock, ArrayRef<Stmt*> handlers) {
818   std::size_t Size = sizeof(CXXTryStmt);
819   Size += ((handlers.size() + 1) * sizeof(Stmt));
820 
821   void *Mem = C.Allocate(Size, llvm::alignOf<CXXTryStmt>());
822   return new (Mem) CXXTryStmt(tryLoc, tryBlock, handlers);
823 }
824 
825 CXXTryStmt *CXXTryStmt::Create(const ASTContext &C, EmptyShell Empty,
826                                unsigned numHandlers) {
827   std::size_t Size = sizeof(CXXTryStmt);
828   Size += ((numHandlers + 1) * sizeof(Stmt));
829 
830   void *Mem = C.Allocate(Size, llvm::alignOf<CXXTryStmt>());
831   return new (Mem) CXXTryStmt(Empty, numHandlers);
832 }
833 
834 CXXTryStmt::CXXTryStmt(SourceLocation tryLoc, Stmt *tryBlock,
835                        ArrayRef<Stmt*> handlers)
836   : Stmt(CXXTryStmtClass), TryLoc(tryLoc), NumHandlers(handlers.size()) {
837   Stmt **Stmts = reinterpret_cast<Stmt **>(this + 1);
838   Stmts[0] = tryBlock;
839   std::copy(handlers.begin(), handlers.end(), Stmts + 1);
840 }
841 
842 CXXForRangeStmt::CXXForRangeStmt(DeclStmt *Range, DeclStmt *BeginEndStmt,
843                                  Expr *Cond, Expr *Inc, DeclStmt *LoopVar,
844                                  Stmt *Body, SourceLocation FL,
845                                  SourceLocation CL, SourceLocation RPL)
846   : Stmt(CXXForRangeStmtClass), ForLoc(FL), ColonLoc(CL), RParenLoc(RPL) {
847   SubExprs[RANGE] = Range;
848   SubExprs[BEGINEND] = BeginEndStmt;
849   SubExprs[COND] = Cond;
850   SubExprs[INC] = Inc;
851   SubExprs[LOOPVAR] = LoopVar;
852   SubExprs[BODY] = Body;
853 }
854 
855 Expr *CXXForRangeStmt::getRangeInit() {
856   DeclStmt *RangeStmt = getRangeStmt();
857   VarDecl *RangeDecl = dyn_cast_or_null<VarDecl>(RangeStmt->getSingleDecl());
858   assert(RangeDecl && "for-range should have a single var decl");
859   return RangeDecl->getInit();
860 }
861 
862 const Expr *CXXForRangeStmt::getRangeInit() const {
863   return const_cast<CXXForRangeStmt*>(this)->getRangeInit();
864 }
865 
866 VarDecl *CXXForRangeStmt::getLoopVariable() {
867   Decl *LV = cast<DeclStmt>(getLoopVarStmt())->getSingleDecl();
868   assert(LV && "No loop variable in CXXForRangeStmt");
869   return cast<VarDecl>(LV);
870 }
871 
872 const VarDecl *CXXForRangeStmt::getLoopVariable() const {
873   return const_cast<CXXForRangeStmt*>(this)->getLoopVariable();
874 }
875 
876 IfStmt::IfStmt(const ASTContext &C, SourceLocation IL, VarDecl *var, Expr *cond,
877                Stmt *then, SourceLocation EL, Stmt *elsev)
878   : Stmt(IfStmtClass), IfLoc(IL), ElseLoc(EL)
879 {
880   setConditionVariable(C, var);
881   SubExprs[COND] = cond;
882   SubExprs[THEN] = then;
883   SubExprs[ELSE] = elsev;
884 }
885 
886 VarDecl *IfStmt::getConditionVariable() const {
887   if (!SubExprs[VAR])
888     return nullptr;
889 
890   DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
891   return cast<VarDecl>(DS->getSingleDecl());
892 }
893 
894 void IfStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
895   if (!V) {
896     SubExprs[VAR] = nullptr;
897     return;
898   }
899 
900   SourceRange VarRange = V->getSourceRange();
901   SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
902                                    VarRange.getEnd());
903 }
904 
905 ForStmt::ForStmt(const ASTContext &C, Stmt *Init, Expr *Cond, VarDecl *condVar,
906                  Expr *Inc, Stmt *Body, SourceLocation FL, SourceLocation LP,
907                  SourceLocation RP)
908   : Stmt(ForStmtClass), ForLoc(FL), LParenLoc(LP), RParenLoc(RP)
909 {
910   SubExprs[INIT] = Init;
911   setConditionVariable(C, condVar);
912   SubExprs[COND] = Cond;
913   SubExprs[INC] = Inc;
914   SubExprs[BODY] = Body;
915 }
916 
917 VarDecl *ForStmt::getConditionVariable() const {
918   if (!SubExprs[CONDVAR])
919     return nullptr;
920 
921   DeclStmt *DS = cast<DeclStmt>(SubExprs[CONDVAR]);
922   return cast<VarDecl>(DS->getSingleDecl());
923 }
924 
925 void ForStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
926   if (!V) {
927     SubExprs[CONDVAR] = nullptr;
928     return;
929   }
930 
931   SourceRange VarRange = V->getSourceRange();
932   SubExprs[CONDVAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
933                                        VarRange.getEnd());
934 }
935 
936 SwitchStmt::SwitchStmt(const ASTContext &C, VarDecl *Var, Expr *cond)
937   : Stmt(SwitchStmtClass), FirstCase(nullptr), AllEnumCasesCovered(0)
938 {
939   setConditionVariable(C, Var);
940   SubExprs[COND] = cond;
941   SubExprs[BODY] = nullptr;
942 }
943 
944 VarDecl *SwitchStmt::getConditionVariable() const {
945   if (!SubExprs[VAR])
946     return nullptr;
947 
948   DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
949   return cast<VarDecl>(DS->getSingleDecl());
950 }
951 
952 void SwitchStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
953   if (!V) {
954     SubExprs[VAR] = nullptr;
955     return;
956   }
957 
958   SourceRange VarRange = V->getSourceRange();
959   SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
960                                    VarRange.getEnd());
961 }
962 
963 Stmt *SwitchCase::getSubStmt() {
964   if (isa<CaseStmt>(this))
965     return cast<CaseStmt>(this)->getSubStmt();
966   return cast<DefaultStmt>(this)->getSubStmt();
967 }
968 
969 WhileStmt::WhileStmt(const ASTContext &C, VarDecl *Var, Expr *cond, Stmt *body,
970                      SourceLocation WL)
971   : Stmt(WhileStmtClass) {
972   setConditionVariable(C, Var);
973   SubExprs[COND] = cond;
974   SubExprs[BODY] = body;
975   WhileLoc = WL;
976 }
977 
978 VarDecl *WhileStmt::getConditionVariable() const {
979   if (!SubExprs[VAR])
980     return nullptr;
981 
982   DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
983   return cast<VarDecl>(DS->getSingleDecl());
984 }
985 
986 void WhileStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
987   if (!V) {
988     SubExprs[VAR] = nullptr;
989     return;
990   }
991 
992   SourceRange VarRange = V->getSourceRange();
993   SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
994                                    VarRange.getEnd());
995 }
996 
997 // IndirectGotoStmt
998 LabelDecl *IndirectGotoStmt::getConstantTarget() {
999   if (AddrLabelExpr *E =
1000         dyn_cast<AddrLabelExpr>(getTarget()->IgnoreParenImpCasts()))
1001     return E->getLabel();
1002   return nullptr;
1003 }
1004 
1005 // ReturnStmt
1006 const Expr* ReturnStmt::getRetValue() const {
1007   return cast_or_null<Expr>(RetExpr);
1008 }
1009 Expr* ReturnStmt::getRetValue() {
1010   return cast_or_null<Expr>(RetExpr);
1011 }
1012 
1013 SEHTryStmt::SEHTryStmt(bool IsCXXTry,
1014                        SourceLocation TryLoc,
1015                        Stmt *TryBlock,
1016                        Stmt *Handler)
1017   : Stmt(SEHTryStmtClass),
1018     IsCXXTry(IsCXXTry),
1019     TryLoc(TryLoc)
1020 {
1021   Children[TRY]     = TryBlock;
1022   Children[HANDLER] = Handler;
1023 }
1024 
1025 SEHTryStmt* SEHTryStmt::Create(const ASTContext &C, bool IsCXXTry,
1026                                SourceLocation TryLoc, Stmt *TryBlock,
1027                                Stmt *Handler) {
1028   return new(C) SEHTryStmt(IsCXXTry,TryLoc,TryBlock,Handler);
1029 }
1030 
1031 SEHExceptStmt* SEHTryStmt::getExceptHandler() const {
1032   return dyn_cast<SEHExceptStmt>(getHandler());
1033 }
1034 
1035 SEHFinallyStmt* SEHTryStmt::getFinallyHandler() const {
1036   return dyn_cast<SEHFinallyStmt>(getHandler());
1037 }
1038 
1039 SEHExceptStmt::SEHExceptStmt(SourceLocation Loc,
1040                              Expr *FilterExpr,
1041                              Stmt *Block)
1042   : Stmt(SEHExceptStmtClass),
1043     Loc(Loc)
1044 {
1045   Children[FILTER_EXPR] = FilterExpr;
1046   Children[BLOCK]       = Block;
1047 }
1048 
1049 SEHExceptStmt* SEHExceptStmt::Create(const ASTContext &C, SourceLocation Loc,
1050                                      Expr *FilterExpr, Stmt *Block) {
1051   return new(C) SEHExceptStmt(Loc,FilterExpr,Block);
1052 }
1053 
1054 SEHFinallyStmt::SEHFinallyStmt(SourceLocation Loc,
1055                                Stmt *Block)
1056   : Stmt(SEHFinallyStmtClass),
1057     Loc(Loc),
1058     Block(Block)
1059 {}
1060 
1061 SEHFinallyStmt* SEHFinallyStmt::Create(const ASTContext &C, SourceLocation Loc,
1062                                        Stmt *Block) {
1063   return new(C)SEHFinallyStmt(Loc,Block);
1064 }
1065 
1066 CapturedStmt::Capture *CapturedStmt::getStoredCaptures() const {
1067   unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (NumCaptures + 1);
1068 
1069   // Offset of the first Capture object.
1070   unsigned FirstCaptureOffset =
1071     llvm::RoundUpToAlignment(Size, llvm::alignOf<Capture>());
1072 
1073   return reinterpret_cast<Capture *>(
1074       reinterpret_cast<char *>(const_cast<CapturedStmt *>(this))
1075       + FirstCaptureOffset);
1076 }
1077 
1078 CapturedStmt::CapturedStmt(Stmt *S, CapturedRegionKind Kind,
1079                            ArrayRef<Capture> Captures,
1080                            ArrayRef<Expr *> CaptureInits,
1081                            CapturedDecl *CD,
1082                            RecordDecl *RD)
1083   : Stmt(CapturedStmtClass), NumCaptures(Captures.size()),
1084     CapDeclAndKind(CD, Kind), TheRecordDecl(RD) {
1085   assert( S && "null captured statement");
1086   assert(CD && "null captured declaration for captured statement");
1087   assert(RD && "null record declaration for captured statement");
1088 
1089   // Copy initialization expressions.
1090   Stmt **Stored = getStoredStmts();
1091   for (unsigned I = 0, N = NumCaptures; I != N; ++I)
1092     *Stored++ = CaptureInits[I];
1093 
1094   // Copy the statement being captured.
1095   *Stored = S;
1096 
1097   // Copy all Capture objects.
1098   Capture *Buffer = getStoredCaptures();
1099   std::copy(Captures.begin(), Captures.end(), Buffer);
1100 }
1101 
1102 CapturedStmt::CapturedStmt(EmptyShell Empty, unsigned NumCaptures)
1103   : Stmt(CapturedStmtClass, Empty), NumCaptures(NumCaptures),
1104     CapDeclAndKind(nullptr, CR_Default), TheRecordDecl(nullptr) {
1105   getStoredStmts()[NumCaptures] = nullptr;
1106 }
1107 
1108 CapturedStmt *CapturedStmt::Create(const ASTContext &Context, Stmt *S,
1109                                    CapturedRegionKind Kind,
1110                                    ArrayRef<Capture> Captures,
1111                                    ArrayRef<Expr *> CaptureInits,
1112                                    CapturedDecl *CD,
1113                                    RecordDecl *RD) {
1114   // The layout is
1115   //
1116   // -----------------------------------------------------------
1117   // | CapturedStmt, Init, ..., Init, S, Capture, ..., Capture |
1118   // ----------------^-------------------^----------------------
1119   //                 getStoredStmts()    getStoredCaptures()
1120   //
1121   // where S is the statement being captured.
1122   //
1123   assert(CaptureInits.size() == Captures.size() && "wrong number of arguments");
1124 
1125   unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (Captures.size() + 1);
1126   if (!Captures.empty()) {
1127     // Realign for the following Capture array.
1128     Size = llvm::RoundUpToAlignment(Size, llvm::alignOf<Capture>());
1129     Size += sizeof(Capture) * Captures.size();
1130   }
1131 
1132   void *Mem = Context.Allocate(Size);
1133   return new (Mem) CapturedStmt(S, Kind, Captures, CaptureInits, CD, RD);
1134 }
1135 
1136 CapturedStmt *CapturedStmt::CreateDeserialized(const ASTContext &Context,
1137                                                unsigned NumCaptures) {
1138   unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (NumCaptures + 1);
1139   if (NumCaptures > 0) {
1140     // Realign for the following Capture array.
1141     Size = llvm::RoundUpToAlignment(Size, llvm::alignOf<Capture>());
1142     Size += sizeof(Capture) * NumCaptures;
1143   }
1144 
1145   void *Mem = Context.Allocate(Size);
1146   return new (Mem) CapturedStmt(EmptyShell(), NumCaptures);
1147 }
1148 
1149 Stmt::child_range CapturedStmt::children() {
1150   // Children are captured field initilizers.
1151   return child_range(getStoredStmts(), getStoredStmts() + NumCaptures);
1152 }
1153 
1154 bool CapturedStmt::capturesVariable(const VarDecl *Var) const {
1155   for (const auto &I : captures()) {
1156     if (!I.capturesVariable())
1157       continue;
1158 
1159     // This does not handle variable redeclarations. This should be
1160     // extended to capture variables with redeclarations, for example
1161     // a thread-private variable in OpenMP.
1162     if (I.getCapturedVar() == Var)
1163       return true;
1164   }
1165 
1166   return false;
1167 }
1168 
1169 StmtRange OMPClause::children() {
1170   switch(getClauseKind()) {
1171   default : break;
1172 #define OPENMP_CLAUSE(Name, Class)                                       \
1173   case OMPC_ ## Name : return static_cast<Class *>(this)->children();
1174 #include "clang/Basic/OpenMPKinds.def"
1175   }
1176   llvm_unreachable("unknown OMPClause");
1177 }
1178 
1179 OMPPrivateClause *OMPPrivateClause::Create(const ASTContext &C,
1180                                            SourceLocation StartLoc,
1181                                            SourceLocation LParenLoc,
1182                                            SourceLocation EndLoc,
1183                                            ArrayRef<Expr *> VL) {
1184   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPPrivateClause),
1185                                                   llvm::alignOf<Expr *>()) +
1186                          sizeof(Expr *) * VL.size());
1187   OMPPrivateClause *Clause = new (Mem) OMPPrivateClause(StartLoc, LParenLoc,
1188                                                         EndLoc, VL.size());
1189   Clause->setVarRefs(VL);
1190   return Clause;
1191 }
1192 
1193 OMPPrivateClause *OMPPrivateClause::CreateEmpty(const ASTContext &C,
1194                                                 unsigned N) {
1195   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPPrivateClause),
1196                                                   llvm::alignOf<Expr *>()) +
1197                          sizeof(Expr *) * N);
1198   return new (Mem) OMPPrivateClause(N);
1199 }
1200 
1201 OMPFirstprivateClause *OMPFirstprivateClause::Create(const ASTContext &C,
1202                                                      SourceLocation StartLoc,
1203                                                      SourceLocation LParenLoc,
1204                                                      SourceLocation EndLoc,
1205                                                      ArrayRef<Expr *> VL) {
1206   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPFirstprivateClause),
1207                                                   llvm::alignOf<Expr *>()) +
1208                          sizeof(Expr *) * VL.size());
1209   OMPFirstprivateClause *Clause = new (Mem) OMPFirstprivateClause(StartLoc,
1210                                                                   LParenLoc,
1211                                                                   EndLoc,
1212                                                                   VL.size());
1213   Clause->setVarRefs(VL);
1214   return Clause;
1215 }
1216 
1217 OMPFirstprivateClause *OMPFirstprivateClause::CreateEmpty(const ASTContext &C,
1218                                                           unsigned N) {
1219   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPFirstprivateClause),
1220                                                   llvm::alignOf<Expr *>()) +
1221                          sizeof(Expr *) * N);
1222   return new (Mem) OMPFirstprivateClause(N);
1223 }
1224 
1225 OMPLastprivateClause *OMPLastprivateClause::Create(const ASTContext &C,
1226                                                    SourceLocation StartLoc,
1227                                                    SourceLocation LParenLoc,
1228                                                    SourceLocation EndLoc,
1229                                                    ArrayRef<Expr *> VL) {
1230   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLastprivateClause),
1231                                                   llvm::alignOf<Expr *>()) +
1232                          sizeof(Expr *) * VL.size());
1233   OMPLastprivateClause *Clause =
1234       new (Mem) OMPLastprivateClause(StartLoc, LParenLoc, EndLoc, VL.size());
1235   Clause->setVarRefs(VL);
1236   return Clause;
1237 }
1238 
1239 OMPLastprivateClause *OMPLastprivateClause::CreateEmpty(const ASTContext &C,
1240                                                         unsigned N) {
1241   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLastprivateClause),
1242                                                   llvm::alignOf<Expr *>()) +
1243                          sizeof(Expr *) * N);
1244   return new (Mem) OMPLastprivateClause(N);
1245 }
1246 
1247 OMPSharedClause *OMPSharedClause::Create(const ASTContext &C,
1248                                          SourceLocation StartLoc,
1249                                          SourceLocation LParenLoc,
1250                                          SourceLocation EndLoc,
1251                                          ArrayRef<Expr *> VL) {
1252   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPSharedClause),
1253                                                   llvm::alignOf<Expr *>()) +
1254                          sizeof(Expr *) * VL.size());
1255   OMPSharedClause *Clause = new (Mem) OMPSharedClause(StartLoc, LParenLoc,
1256                                                       EndLoc, VL.size());
1257   Clause->setVarRefs(VL);
1258   return Clause;
1259 }
1260 
1261 OMPSharedClause *OMPSharedClause::CreateEmpty(const ASTContext &C,
1262                                               unsigned N) {
1263   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPSharedClause),
1264                                                   llvm::alignOf<Expr *>()) +
1265                          sizeof(Expr *) * N);
1266   return new (Mem) OMPSharedClause(N);
1267 }
1268 
1269 OMPLinearClause *OMPLinearClause::Create(const ASTContext &C,
1270                                          SourceLocation StartLoc,
1271                                          SourceLocation LParenLoc,
1272                                          SourceLocation ColonLoc,
1273                                          SourceLocation EndLoc,
1274                                          ArrayRef<Expr *> VL, Expr *Step) {
1275   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLinearClause),
1276                                                   llvm::alignOf<Expr *>()) +
1277                          sizeof(Expr *) * (VL.size() + 1));
1278   OMPLinearClause *Clause = new (Mem)
1279       OMPLinearClause(StartLoc, LParenLoc, ColonLoc, EndLoc, VL.size());
1280   Clause->setVarRefs(VL);
1281   Clause->setStep(Step);
1282   return Clause;
1283 }
1284 
1285 OMPLinearClause *OMPLinearClause::CreateEmpty(const ASTContext &C,
1286                                               unsigned NumVars) {
1287   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPLinearClause),
1288                                                   llvm::alignOf<Expr *>()) +
1289                          sizeof(Expr *) * (NumVars + 1));
1290   return new (Mem) OMPLinearClause(NumVars);
1291 }
1292 
1293 OMPAlignedClause *
1294 OMPAlignedClause::Create(const ASTContext &C, SourceLocation StartLoc,
1295                          SourceLocation LParenLoc, SourceLocation ColonLoc,
1296                          SourceLocation EndLoc, ArrayRef<Expr *> VL, Expr *A) {
1297   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPAlignedClause),
1298                                                   llvm::alignOf<Expr *>()) +
1299                          sizeof(Expr *) * (VL.size() + 1));
1300   OMPAlignedClause *Clause = new (Mem)
1301       OMPAlignedClause(StartLoc, LParenLoc, ColonLoc, EndLoc, VL.size());
1302   Clause->setVarRefs(VL);
1303   Clause->setAlignment(A);
1304   return Clause;
1305 }
1306 
1307 OMPAlignedClause *OMPAlignedClause::CreateEmpty(const ASTContext &C,
1308                                                 unsigned NumVars) {
1309   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPAlignedClause),
1310                                                   llvm::alignOf<Expr *>()) +
1311                          sizeof(Expr *) * (NumVars + 1));
1312   return new (Mem) OMPAlignedClause(NumVars);
1313 }
1314 
1315 OMPCopyinClause *OMPCopyinClause::Create(const ASTContext &C,
1316                                          SourceLocation StartLoc,
1317                                          SourceLocation LParenLoc,
1318                                          SourceLocation EndLoc,
1319                                          ArrayRef<Expr *> VL) {
1320   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyinClause),
1321                                                   llvm::alignOf<Expr *>()) +
1322                          sizeof(Expr *) * VL.size());
1323   OMPCopyinClause *Clause = new (Mem) OMPCopyinClause(StartLoc, LParenLoc,
1324                                                       EndLoc, VL.size());
1325   Clause->setVarRefs(VL);
1326   return Clause;
1327 }
1328 
1329 OMPCopyinClause *OMPCopyinClause::CreateEmpty(const ASTContext &C,
1330                                               unsigned N) {
1331   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyinClause),
1332                                                   llvm::alignOf<Expr *>()) +
1333                          sizeof(Expr *) * N);
1334   return new (Mem) OMPCopyinClause(N);
1335 }
1336 
1337 OMPCopyprivateClause *OMPCopyprivateClause::Create(const ASTContext &C,
1338                                                    SourceLocation StartLoc,
1339                                                    SourceLocation LParenLoc,
1340                                                    SourceLocation EndLoc,
1341                                                    ArrayRef<Expr *> VL) {
1342   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyprivateClause),
1343                                                   llvm::alignOf<Expr *>()) +
1344                          sizeof(Expr *) * VL.size());
1345   OMPCopyprivateClause *Clause =
1346       new (Mem) OMPCopyprivateClause(StartLoc, LParenLoc, EndLoc, VL.size());
1347   Clause->setVarRefs(VL);
1348   return Clause;
1349 }
1350 
1351 OMPCopyprivateClause *OMPCopyprivateClause::CreateEmpty(const ASTContext &C,
1352                                                         unsigned N) {
1353   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPCopyprivateClause),
1354                                                   llvm::alignOf<Expr *>()) +
1355                          sizeof(Expr *) * N);
1356   return new (Mem) OMPCopyprivateClause(N);
1357 }
1358 
1359 void OMPExecutableDirective::setClauses(ArrayRef<OMPClause *> Clauses) {
1360   assert(Clauses.size() == getNumClauses() &&
1361          "Number of clauses is not the same as the preallocated buffer");
1362   std::copy(Clauses.begin(), Clauses.end(), getClauses().begin());
1363 }
1364 
1365 void OMPLoopDirective::setCounters(ArrayRef<Expr *> A) {
1366   assert(A.size() == getCollapsedNumber() &&
1367          "Number of loop counters is not the same as the collapsed number");
1368   std::copy(A.begin(), A.end(), getCounters().begin());
1369 }
1370 
1371 void OMPLoopDirective::setUpdates(ArrayRef<Expr *> A) {
1372   assert(A.size() == getCollapsedNumber() &&
1373          "Number of counter updates is not the same as the collapsed number");
1374   std::copy(A.begin(), A.end(), getUpdates().begin());
1375 }
1376 
1377 void OMPLoopDirective::setFinals(ArrayRef<Expr *> A) {
1378   assert(A.size() == getCollapsedNumber() &&
1379          "Number of counter finals is not the same as the collapsed number");
1380   std::copy(A.begin(), A.end(), getFinals().begin());
1381 }
1382 
1383 OMPReductionClause *OMPReductionClause::Create(
1384     const ASTContext &C, SourceLocation StartLoc, SourceLocation LParenLoc,
1385     SourceLocation EndLoc, SourceLocation ColonLoc, ArrayRef<Expr *> VL,
1386     NestedNameSpecifierLoc QualifierLoc, const DeclarationNameInfo &NameInfo) {
1387   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPReductionClause),
1388                                                   llvm::alignOf<Expr *>()) +
1389                          sizeof(Expr *) * VL.size());
1390   OMPReductionClause *Clause = new (Mem) OMPReductionClause(
1391       StartLoc, LParenLoc, EndLoc, ColonLoc, VL.size(), QualifierLoc, NameInfo);
1392   Clause->setVarRefs(VL);
1393   return Clause;
1394 }
1395 
1396 OMPReductionClause *OMPReductionClause::CreateEmpty(const ASTContext &C,
1397                                                     unsigned N) {
1398   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPReductionClause),
1399                                                   llvm::alignOf<Expr *>()) +
1400                          sizeof(Expr *) * N);
1401   return new (Mem) OMPReductionClause(N);
1402 }
1403 
1404 OMPFlushClause *OMPFlushClause::Create(const ASTContext &C,
1405                                        SourceLocation StartLoc,
1406                                        SourceLocation LParenLoc,
1407                                        SourceLocation EndLoc,
1408                                        ArrayRef<Expr *> VL) {
1409   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPFlushClause),
1410                                                   llvm::alignOf<Expr *>()) +
1411                          sizeof(Expr *) * VL.size());
1412   OMPFlushClause *Clause =
1413       new (Mem) OMPFlushClause(StartLoc, LParenLoc, EndLoc, VL.size());
1414   Clause->setVarRefs(VL);
1415   return Clause;
1416 }
1417 
1418 OMPFlushClause *OMPFlushClause::CreateEmpty(const ASTContext &C, unsigned N) {
1419   void *Mem = C.Allocate(llvm::RoundUpToAlignment(sizeof(OMPFlushClause),
1420                                                   llvm::alignOf<Expr *>()) +
1421                          sizeof(Expr *) * N);
1422   return new (Mem) OMPFlushClause(N);
1423 }
1424 
1425 OMPParallelDirective *OMPParallelDirective::Create(
1426                                               const ASTContext &C,
1427                                               SourceLocation StartLoc,
1428                                               SourceLocation EndLoc,
1429                                               ArrayRef<OMPClause *> Clauses,
1430                                               Stmt *AssociatedStmt) {
1431   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelDirective),
1432                                            llvm::alignOf<OMPClause *>());
1433   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1434                          sizeof(Stmt *));
1435   OMPParallelDirective *Dir = new (Mem) OMPParallelDirective(StartLoc, EndLoc,
1436                                                              Clauses.size());
1437   Dir->setClauses(Clauses);
1438   Dir->setAssociatedStmt(AssociatedStmt);
1439   return Dir;
1440 }
1441 
1442 OMPParallelDirective *OMPParallelDirective::CreateEmpty(const ASTContext &C,
1443                                                         unsigned NumClauses,
1444                                                         EmptyShell) {
1445   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelDirective),
1446                                            llvm::alignOf<OMPClause *>());
1447   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1448                          sizeof(Stmt *));
1449   return new (Mem) OMPParallelDirective(NumClauses);
1450 }
1451 
1452 OMPSimdDirective *
1453 OMPSimdDirective::Create(const ASTContext &C, SourceLocation StartLoc,
1454                          SourceLocation EndLoc, unsigned CollapsedNum,
1455                          ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
1456                          Expr *IV, Expr *LastIteration, Expr *CalcLastIteration,
1457                          Expr *PreCond, Expr *Cond, Expr *SeparatedCond,
1458                          Expr *Init, Expr *Inc, ArrayRef<Expr *> Counters,
1459                          ArrayRef<Expr *> Updates, ArrayRef<Expr *> Finals) {
1460   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSimdDirective),
1461                                            llvm::alignOf<OMPClause *>());
1462   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1463                          sizeof(Stmt *) * numLoopChildren(CollapsedNum));
1464   OMPSimdDirective *Dir = new (Mem)
1465       OMPSimdDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
1466   Dir->setClauses(Clauses);
1467   Dir->setAssociatedStmt(AssociatedStmt);
1468   Dir->setIterationVariable(IV);
1469   Dir->setLastIteration(LastIteration);
1470   Dir->setCalcLastIteration(CalcLastIteration);
1471   Dir->setPreCond(PreCond);
1472   Dir->setCond(Cond, SeparatedCond);
1473   Dir->setInit(Init);
1474   Dir->setInc(Inc);
1475   Dir->setCounters(Counters);
1476   Dir->setUpdates(Updates);
1477   Dir->setFinals(Finals);
1478   return Dir;
1479 }
1480 
1481 OMPSimdDirective *OMPSimdDirective::CreateEmpty(const ASTContext &C,
1482                                                 unsigned NumClauses,
1483                                                 unsigned CollapsedNum,
1484                                                 EmptyShell) {
1485   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSimdDirective),
1486                                            llvm::alignOf<OMPClause *>());
1487   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1488                          sizeof(Stmt *) * numLoopChildren(CollapsedNum));
1489   return new (Mem) OMPSimdDirective(CollapsedNum, NumClauses);
1490 }
1491 
1492 OMPForDirective *
1493 OMPForDirective::Create(const ASTContext &C, SourceLocation StartLoc,
1494                         SourceLocation EndLoc, unsigned CollapsedNum,
1495                         ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
1496                         Expr *IV, Expr *LastIteration, Expr *CalcLastIteration,
1497                         Expr *PreCond, Expr *Cond, Expr *SeparatedCond,
1498                         Expr *Init, Expr *Inc, ArrayRef<Expr *> Counters,
1499                         ArrayRef<Expr *> Updates, ArrayRef<Expr *> Finals) {
1500   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPForDirective),
1501                                            llvm::alignOf<OMPClause *>());
1502   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1503                          sizeof(Stmt *) * numLoopChildren(CollapsedNum));
1504   OMPForDirective *Dir =
1505       new (Mem) OMPForDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
1506   Dir->setClauses(Clauses);
1507   Dir->setAssociatedStmt(AssociatedStmt);
1508   Dir->setIterationVariable(IV);
1509   Dir->setLastIteration(LastIteration);
1510   Dir->setCalcLastIteration(CalcLastIteration);
1511   Dir->setPreCond(PreCond);
1512   Dir->setCond(Cond, SeparatedCond);
1513   Dir->setInit(Init);
1514   Dir->setInc(Inc);
1515   Dir->setCounters(Counters);
1516   Dir->setUpdates(Updates);
1517   Dir->setFinals(Finals);
1518   return Dir;
1519 }
1520 
1521 OMPForDirective *OMPForDirective::CreateEmpty(const ASTContext &C,
1522                                               unsigned NumClauses,
1523                                               unsigned CollapsedNum,
1524                                               EmptyShell) {
1525   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPForDirective),
1526                                            llvm::alignOf<OMPClause *>());
1527   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1528                          sizeof(Stmt *) * numLoopChildren(CollapsedNum));
1529   return new (Mem) OMPForDirective(CollapsedNum, NumClauses);
1530 }
1531 
1532 OMPForSimdDirective *OMPForSimdDirective::Create(
1533     const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
1534     unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
1535     Expr *IV, Expr *LastIteration, Expr *CalcLastIteration, Expr *PreCond,
1536     Expr *Cond, Expr *SeparatedCond, Expr *Init, Expr *Inc,
1537     ArrayRef<Expr *> Counters, ArrayRef<Expr *> Updates,
1538     ArrayRef<Expr *> Finals) {
1539   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPForSimdDirective),
1540                                            llvm::alignOf<OMPClause *>());
1541   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1542                          sizeof(Stmt *) * numLoopChildren(CollapsedNum));
1543   OMPForSimdDirective *Dir = new (Mem)
1544       OMPForSimdDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
1545   Dir->setClauses(Clauses);
1546   Dir->setAssociatedStmt(AssociatedStmt);
1547   Dir->setIterationVariable(IV);
1548   Dir->setLastIteration(LastIteration);
1549   Dir->setCalcLastIteration(CalcLastIteration);
1550   Dir->setPreCond(PreCond);
1551   Dir->setCond(Cond, SeparatedCond);
1552   Dir->setInit(Init);
1553   Dir->setInc(Inc);
1554   Dir->setCounters(Counters);
1555   Dir->setUpdates(Updates);
1556   Dir->setFinals(Finals);
1557   return Dir;
1558 }
1559 
1560 OMPForSimdDirective *OMPForSimdDirective::CreateEmpty(const ASTContext &C,
1561                                                       unsigned NumClauses,
1562                                                       unsigned CollapsedNum,
1563                                                       EmptyShell) {
1564   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPForSimdDirective),
1565                                            llvm::alignOf<OMPClause *>());
1566   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1567                          sizeof(Stmt *) * numLoopChildren(CollapsedNum));
1568   return new (Mem) OMPForSimdDirective(CollapsedNum, NumClauses);
1569 }
1570 
1571 OMPSectionsDirective *OMPSectionsDirective::Create(
1572     const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
1573     ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
1574   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionsDirective),
1575                                            llvm::alignOf<OMPClause *>());
1576   void *Mem =
1577       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1578   OMPSectionsDirective *Dir =
1579       new (Mem) OMPSectionsDirective(StartLoc, EndLoc, Clauses.size());
1580   Dir->setClauses(Clauses);
1581   Dir->setAssociatedStmt(AssociatedStmt);
1582   return Dir;
1583 }
1584 
1585 OMPSectionsDirective *OMPSectionsDirective::CreateEmpty(const ASTContext &C,
1586                                                         unsigned NumClauses,
1587                                                         EmptyShell) {
1588   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionsDirective),
1589                                            llvm::alignOf<OMPClause *>());
1590   void *Mem =
1591       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1592   return new (Mem) OMPSectionsDirective(NumClauses);
1593 }
1594 
1595 OMPSectionDirective *OMPSectionDirective::Create(const ASTContext &C,
1596                                                  SourceLocation StartLoc,
1597                                                  SourceLocation EndLoc,
1598                                                  Stmt *AssociatedStmt) {
1599   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionsDirective),
1600                                            llvm::alignOf<Stmt *>());
1601   void *Mem = C.Allocate(Size + sizeof(Stmt *));
1602   OMPSectionDirective *Dir = new (Mem) OMPSectionDirective(StartLoc, EndLoc);
1603   Dir->setAssociatedStmt(AssociatedStmt);
1604   return Dir;
1605 }
1606 
1607 OMPSectionDirective *OMPSectionDirective::CreateEmpty(const ASTContext &C,
1608                                                       EmptyShell) {
1609   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSectionDirective),
1610                                            llvm::alignOf<Stmt *>());
1611   void *Mem = C.Allocate(Size + sizeof(Stmt *));
1612   return new (Mem) OMPSectionDirective();
1613 }
1614 
1615 OMPSingleDirective *OMPSingleDirective::Create(const ASTContext &C,
1616                                                SourceLocation StartLoc,
1617                                                SourceLocation EndLoc,
1618                                                ArrayRef<OMPClause *> Clauses,
1619                                                Stmt *AssociatedStmt) {
1620   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSingleDirective),
1621                                            llvm::alignOf<OMPClause *>());
1622   void *Mem =
1623       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1624   OMPSingleDirective *Dir =
1625       new (Mem) OMPSingleDirective(StartLoc, EndLoc, Clauses.size());
1626   Dir->setClauses(Clauses);
1627   Dir->setAssociatedStmt(AssociatedStmt);
1628   return Dir;
1629 }
1630 
1631 OMPSingleDirective *OMPSingleDirective::CreateEmpty(const ASTContext &C,
1632                                                     unsigned NumClauses,
1633                                                     EmptyShell) {
1634   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPSingleDirective),
1635                                            llvm::alignOf<OMPClause *>());
1636   void *Mem =
1637       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1638   return new (Mem) OMPSingleDirective(NumClauses);
1639 }
1640 
1641 OMPMasterDirective *OMPMasterDirective::Create(const ASTContext &C,
1642                                                SourceLocation StartLoc,
1643                                                SourceLocation EndLoc,
1644                                                Stmt *AssociatedStmt) {
1645   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPMasterDirective),
1646                                            llvm::alignOf<Stmt *>());
1647   void *Mem = C.Allocate(Size + sizeof(Stmt *));
1648   OMPMasterDirective *Dir = new (Mem) OMPMasterDirective(StartLoc, EndLoc);
1649   Dir->setAssociatedStmt(AssociatedStmt);
1650   return Dir;
1651 }
1652 
1653 OMPMasterDirective *OMPMasterDirective::CreateEmpty(const ASTContext &C,
1654                                                     EmptyShell) {
1655   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPMasterDirective),
1656                                            llvm::alignOf<Stmt *>());
1657   void *Mem = C.Allocate(Size + sizeof(Stmt *));
1658   return new (Mem) OMPMasterDirective();
1659 }
1660 
1661 OMPCriticalDirective *OMPCriticalDirective::Create(
1662     const ASTContext &C, const DeclarationNameInfo &Name,
1663     SourceLocation StartLoc, SourceLocation EndLoc, Stmt *AssociatedStmt) {
1664   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPCriticalDirective),
1665                                            llvm::alignOf<Stmt *>());
1666   void *Mem = C.Allocate(Size + sizeof(Stmt *));
1667   OMPCriticalDirective *Dir =
1668       new (Mem) OMPCriticalDirective(Name, StartLoc, EndLoc);
1669   Dir->setAssociatedStmt(AssociatedStmt);
1670   return Dir;
1671 }
1672 
1673 OMPCriticalDirective *OMPCriticalDirective::CreateEmpty(const ASTContext &C,
1674                                                         EmptyShell) {
1675   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPCriticalDirective),
1676                                            llvm::alignOf<Stmt *>());
1677   void *Mem = C.Allocate(Size + sizeof(Stmt *));
1678   return new (Mem) OMPCriticalDirective();
1679 }
1680 
1681 OMPParallelForDirective *OMPParallelForDirective::Create(
1682     const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
1683     unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
1684     Expr *IV, Expr *LastIteration, Expr *CalcLastIteration, Expr *PreCond,
1685     Expr *Cond, Expr *SeparatedCond, Expr *Init, Expr *Inc,
1686     ArrayRef<Expr *> Counters, ArrayRef<Expr *> Updates,
1687     ArrayRef<Expr *> Finals) {
1688   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelForDirective),
1689                                            llvm::alignOf<OMPClause *>());
1690   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1691                          sizeof(Stmt *) * numLoopChildren(CollapsedNum));
1692   OMPParallelForDirective *Dir = new (Mem)
1693       OMPParallelForDirective(StartLoc, EndLoc, CollapsedNum, Clauses.size());
1694   Dir->setClauses(Clauses);
1695   Dir->setAssociatedStmt(AssociatedStmt);
1696   Dir->setIterationVariable(IV);
1697   Dir->setLastIteration(LastIteration);
1698   Dir->setCalcLastIteration(CalcLastIteration);
1699   Dir->setPreCond(PreCond);
1700   Dir->setCond(Cond, SeparatedCond);
1701   Dir->setInit(Init);
1702   Dir->setInc(Inc);
1703   Dir->setCounters(Counters);
1704   Dir->setUpdates(Updates);
1705   Dir->setFinals(Finals);
1706   return Dir;
1707 }
1708 
1709 OMPParallelForDirective *
1710 OMPParallelForDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
1711                                      unsigned CollapsedNum, EmptyShell) {
1712   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelForDirective),
1713                                            llvm::alignOf<OMPClause *>());
1714   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1715                          sizeof(Stmt *) * numLoopChildren(CollapsedNum));
1716   return new (Mem) OMPParallelForDirective(CollapsedNum, NumClauses);
1717 }
1718 
1719 OMPParallelForSimdDirective *OMPParallelForSimdDirective::Create(
1720     const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
1721     unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
1722     Expr *IV, Expr *LastIteration, Expr *CalcLastIteration, Expr *PreCond,
1723     Expr *Cond, Expr *SeparatedCond, Expr *Init, Expr *Inc,
1724     ArrayRef<Expr *> Counters, ArrayRef<Expr *> Updates,
1725     ArrayRef<Expr *> Finals) {
1726   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelForSimdDirective),
1727                                            llvm::alignOf<OMPClause *>());
1728   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() +
1729                          sizeof(Stmt *) * numLoopChildren(CollapsedNum));
1730   OMPParallelForSimdDirective *Dir = new (Mem) OMPParallelForSimdDirective(
1731       StartLoc, EndLoc, CollapsedNum, Clauses.size());
1732   Dir->setClauses(Clauses);
1733   Dir->setAssociatedStmt(AssociatedStmt);
1734   Dir->setIterationVariable(IV);
1735   Dir->setLastIteration(LastIteration);
1736   Dir->setCalcLastIteration(CalcLastIteration);
1737   Dir->setPreCond(PreCond);
1738   Dir->setCond(Cond, SeparatedCond);
1739   Dir->setInit(Init);
1740   Dir->setInc(Inc);
1741   Dir->setCounters(Counters);
1742   Dir->setUpdates(Updates);
1743   Dir->setFinals(Finals);
1744   return Dir;
1745 }
1746 
1747 OMPParallelForSimdDirective *
1748 OMPParallelForSimdDirective::CreateEmpty(const ASTContext &C,
1749                                          unsigned NumClauses,
1750                                          unsigned CollapsedNum, EmptyShell) {
1751   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelForSimdDirective),
1752                                            llvm::alignOf<OMPClause *>());
1753   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses +
1754                          sizeof(Stmt *) * numLoopChildren(CollapsedNum));
1755   return new (Mem) OMPParallelForSimdDirective(CollapsedNum, NumClauses);
1756 }
1757 
1758 OMPParallelSectionsDirective *OMPParallelSectionsDirective::Create(
1759     const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
1760     ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
1761   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelSectionsDirective),
1762                                            llvm::alignOf<OMPClause *>());
1763   void *Mem =
1764       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1765   OMPParallelSectionsDirective *Dir =
1766       new (Mem) OMPParallelSectionsDirective(StartLoc, EndLoc, Clauses.size());
1767   Dir->setClauses(Clauses);
1768   Dir->setAssociatedStmt(AssociatedStmt);
1769   return Dir;
1770 }
1771 
1772 OMPParallelSectionsDirective *
1773 OMPParallelSectionsDirective::CreateEmpty(const ASTContext &C,
1774                                           unsigned NumClauses, EmptyShell) {
1775   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPParallelSectionsDirective),
1776                                            llvm::alignOf<OMPClause *>());
1777   void *Mem =
1778       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1779   return new (Mem) OMPParallelSectionsDirective(NumClauses);
1780 }
1781 
1782 OMPTaskDirective *OMPTaskDirective::Create(const ASTContext &C,
1783                                            SourceLocation StartLoc,
1784                                            SourceLocation EndLoc,
1785                                            ArrayRef<OMPClause *> Clauses,
1786                                            Stmt *AssociatedStmt) {
1787   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPTaskDirective),
1788                                            llvm::alignOf<OMPClause *>());
1789   void *Mem =
1790       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1791   OMPTaskDirective *Dir =
1792       new (Mem) OMPTaskDirective(StartLoc, EndLoc, Clauses.size());
1793   Dir->setClauses(Clauses);
1794   Dir->setAssociatedStmt(AssociatedStmt);
1795   return Dir;
1796 }
1797 
1798 OMPTaskDirective *OMPTaskDirective::CreateEmpty(const ASTContext &C,
1799                                                 unsigned NumClauses,
1800                                                 EmptyShell) {
1801   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPTaskDirective),
1802                                            llvm::alignOf<OMPClause *>());
1803   void *Mem =
1804       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1805   return new (Mem) OMPTaskDirective(NumClauses);
1806 }
1807 
1808 OMPTaskyieldDirective *OMPTaskyieldDirective::Create(const ASTContext &C,
1809                                                      SourceLocation StartLoc,
1810                                                      SourceLocation EndLoc) {
1811   void *Mem = C.Allocate(sizeof(OMPTaskyieldDirective));
1812   OMPTaskyieldDirective *Dir =
1813       new (Mem) OMPTaskyieldDirective(StartLoc, EndLoc);
1814   return Dir;
1815 }
1816 
1817 OMPTaskyieldDirective *OMPTaskyieldDirective::CreateEmpty(const ASTContext &C,
1818                                                           EmptyShell) {
1819   void *Mem = C.Allocate(sizeof(OMPTaskyieldDirective));
1820   return new (Mem) OMPTaskyieldDirective();
1821 }
1822 
1823 OMPBarrierDirective *OMPBarrierDirective::Create(const ASTContext &C,
1824                                                  SourceLocation StartLoc,
1825                                                  SourceLocation EndLoc) {
1826   void *Mem = C.Allocate(sizeof(OMPBarrierDirective));
1827   OMPBarrierDirective *Dir = new (Mem) OMPBarrierDirective(StartLoc, EndLoc);
1828   return Dir;
1829 }
1830 
1831 OMPBarrierDirective *OMPBarrierDirective::CreateEmpty(const ASTContext &C,
1832                                                       EmptyShell) {
1833   void *Mem = C.Allocate(sizeof(OMPBarrierDirective));
1834   return new (Mem) OMPBarrierDirective();
1835 }
1836 
1837 OMPTaskwaitDirective *OMPTaskwaitDirective::Create(const ASTContext &C,
1838                                                    SourceLocation StartLoc,
1839                                                    SourceLocation EndLoc) {
1840   void *Mem = C.Allocate(sizeof(OMPTaskwaitDirective));
1841   OMPTaskwaitDirective *Dir = new (Mem) OMPTaskwaitDirective(StartLoc, EndLoc);
1842   return Dir;
1843 }
1844 
1845 OMPTaskwaitDirective *OMPTaskwaitDirective::CreateEmpty(const ASTContext &C,
1846                                                         EmptyShell) {
1847   void *Mem = C.Allocate(sizeof(OMPTaskwaitDirective));
1848   return new (Mem) OMPTaskwaitDirective();
1849 }
1850 
1851 OMPFlushDirective *OMPFlushDirective::Create(const ASTContext &C,
1852                                              SourceLocation StartLoc,
1853                                              SourceLocation EndLoc,
1854                                              ArrayRef<OMPClause *> Clauses) {
1855   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPFlushDirective),
1856                                            llvm::alignOf<OMPClause *>());
1857   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * Clauses.size());
1858   OMPFlushDirective *Dir =
1859       new (Mem) OMPFlushDirective(StartLoc, EndLoc, Clauses.size());
1860   Dir->setClauses(Clauses);
1861   return Dir;
1862 }
1863 
1864 OMPFlushDirective *OMPFlushDirective::CreateEmpty(const ASTContext &C,
1865                                                   unsigned NumClauses,
1866                                                   EmptyShell) {
1867   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPFlushDirective),
1868                                            llvm::alignOf<OMPClause *>());
1869   void *Mem = C.Allocate(Size + sizeof(OMPClause *) * NumClauses);
1870   return new (Mem) OMPFlushDirective(NumClauses);
1871 }
1872 
1873 OMPOrderedDirective *OMPOrderedDirective::Create(const ASTContext &C,
1874                                                  SourceLocation StartLoc,
1875                                                  SourceLocation EndLoc,
1876                                                  Stmt *AssociatedStmt) {
1877   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPOrderedDirective),
1878                                            llvm::alignOf<Stmt *>());
1879   void *Mem = C.Allocate(Size + sizeof(Stmt *));
1880   OMPOrderedDirective *Dir = new (Mem) OMPOrderedDirective(StartLoc, EndLoc);
1881   Dir->setAssociatedStmt(AssociatedStmt);
1882   return Dir;
1883 }
1884 
1885 OMPOrderedDirective *OMPOrderedDirective::CreateEmpty(const ASTContext &C,
1886                                                       EmptyShell) {
1887   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPOrderedDirective),
1888                                            llvm::alignOf<Stmt *>());
1889   void *Mem = C.Allocate(Size + sizeof(Stmt *));
1890   return new (Mem) OMPOrderedDirective();
1891 }
1892 
1893 OMPAtomicDirective *OMPAtomicDirective::Create(const ASTContext &C,
1894                                                SourceLocation StartLoc,
1895                                                SourceLocation EndLoc,
1896                                                ArrayRef<OMPClause *> Clauses,
1897                                                Stmt *AssociatedStmt) {
1898   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPAtomicDirective),
1899                                            llvm::alignOf<OMPClause *>());
1900   void *Mem =
1901       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1902   OMPAtomicDirective *Dir =
1903       new (Mem) OMPAtomicDirective(StartLoc, EndLoc, Clauses.size());
1904   Dir->setClauses(Clauses);
1905   Dir->setAssociatedStmt(AssociatedStmt);
1906   return Dir;
1907 }
1908 
1909 OMPAtomicDirective *OMPAtomicDirective::CreateEmpty(const ASTContext &C,
1910                                                     unsigned NumClauses,
1911                                                     EmptyShell) {
1912   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPAtomicDirective),
1913                                            llvm::alignOf<OMPClause *>());
1914   void *Mem =
1915       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1916   return new (Mem) OMPAtomicDirective(NumClauses);
1917 }
1918 
1919 OMPTargetDirective *OMPTargetDirective::Create(const ASTContext &C,
1920                                                SourceLocation StartLoc,
1921                                                SourceLocation EndLoc,
1922                                                ArrayRef<OMPClause *> Clauses,
1923                                                Stmt *AssociatedStmt) {
1924   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPTargetDirective),
1925                                            llvm::alignOf<OMPClause *>());
1926   void *Mem =
1927       C.Allocate(Size + sizeof(OMPClause *) * Clauses.size() + sizeof(Stmt *));
1928   OMPTargetDirective *Dir =
1929       new (Mem) OMPTargetDirective(StartLoc, EndLoc, Clauses.size());
1930   Dir->setClauses(Clauses);
1931   Dir->setAssociatedStmt(AssociatedStmt);
1932   return Dir;
1933 }
1934 
1935 OMPTargetDirective *OMPTargetDirective::CreateEmpty(const ASTContext &C,
1936                                                     unsigned NumClauses,
1937                                                     EmptyShell) {
1938   unsigned Size = llvm::RoundUpToAlignment(sizeof(OMPTargetDirective),
1939                                            llvm::alignOf<OMPClause *>());
1940   void *Mem =
1941       C.Allocate(Size + sizeof(OMPClause *) * NumClauses + sizeof(Stmt *));
1942   return new (Mem) OMPTargetDirective(NumClauses);
1943 }
1944 
1945