Stmt.h revision acb6bcb442936f3c14126fdd473307c17647cb84
15f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===--- Stmt.h - Classes for representing statements -----------*- C++ -*-===//
25f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
35f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//                     The LLVM Compiler Infrastructure
45f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
50bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// This file is distributed under the University of Illinois Open Source
60bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// License. See LICENSE.TXT for details.
75f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
85f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===//
95f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//  This file defines the Stmt interface and subclasses.
115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===//
135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#ifndef LLVM_CLANG_AST_STMT_H
155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#define LLVM_CLANG_AST_STMT_H
165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
177573098b83e780d1c5bea13b384b610d8f155676Steve Naroff#include "llvm/Support/Casting.h"
18a95d3750441ac8ad03e36af8e6e74039c9a3109dTed Kremenek#include "llvm/Support/raw_ostream.h"
195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Basic/SourceLocation.h"
20d249e1d1f1498b81314459ceda19d6ff25c278adDouglas Gregor#include "clang/AST/PrettyPrinter.h"
219caf8b1ca6beb254f420dada3c0e94d5ef027f58Ted Kremenek#include "clang/AST/StmtIterator.h"
228ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek#include "clang/AST/DeclGroup.h"
235ee56e95c3905d2e7bc403631b03865cdbdd8a42Anders Carlsson#include "clang/AST/FullExpr.h"
245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "llvm/ADT/SmallVector.h"
258189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek#include "clang/AST/ASTContext.h"
266e340496341a4704be0ede9c1ff4f8eacea7ee2cChris Lattner#include <string>
277573098b83e780d1c5bea13b384b610d8f155676Steve Naroffusing llvm::dyn_cast_or_null;
287573098b83e780d1c5bea13b384b610d8f155676Steve Naroff
2941ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregornamespace llvm {
3041ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  class FoldingSetNodeID;
3141ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor}
3241ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor
335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencernamespace clang {
34e2563ca02a519c2ad6d64dfed87d6e86c5d3c072Sam Bishop  class ASTContext;
355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  class Expr;
365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  class Decl;
377ba138abd329e591a8f6d5001f60dd7082f71b3bSteve Naroff  class ParmVarDecl;
384b07b2968f87f3cd5a3d8c76145f1cbfd718d42dSebastian Redl  class QualType;
395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  class IdentifierInfo;
400c727a35718556866a978f64ac549d9798735f08Chris Lattner  class SourceManager;
416a0ef4b83c91a6d6d5acb4ed5577c4659fe022a3Anders Carlsson  class StringLiteral;
426c36be5b383875b490684bcf439d6d427298c1afChris Lattner  class SwitchStmt;
431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
44ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  //===----------------------------------------------------------------------===//
45ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  // ExprIterator - Iterators for iterating over Stmt* arrays that contain
46ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  //  only Expr*.  This is needed because AST nodes use Stmt* arrays to store
47ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  //  references to children (to be compatible with StmtIterator).
48ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  //===----------------------------------------------------------------------===//
491eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
50ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  class Stmt;
51ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  class Expr;
521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
53ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  class ExprIterator {
54ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    Stmt** I;
55ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  public:
56ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ExprIterator(Stmt** i) : I(i) {}
571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    ExprIterator() : I(0) {}
58ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ExprIterator& operator++() { ++I; return *this; }
59ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ExprIterator operator-(size_t i) { return I-i; }
60ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ExprIterator operator+(size_t i) { return I+i; }
61ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    Expr* operator[](size_t idx);
62ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    // FIXME: Verify that this will correctly return a signed distance.
63ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    signed operator-(const ExprIterator& R) const { return I - R.I; }
64ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    Expr* operator*() const;
65ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    Expr* operator->() const;
66ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    bool operator==(const ExprIterator& R) const { return I == R.I; }
67ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    bool operator!=(const ExprIterator& R) const { return I != R.I; }
68ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    bool operator>(const ExprIterator& R) const { return I > R.I; }
69ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    bool operator>=(const ExprIterator& R) const { return I >= R.I; }
70ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  };
711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
72ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  class ConstExprIterator {
73ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    Stmt* const * I;
74ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  public:
75ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ConstExprIterator(Stmt* const* i) : I(i) {}
761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    ConstExprIterator() : I(0) {}
77ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ConstExprIterator& operator++() { ++I; return *this; }
78ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ConstExprIterator operator+(size_t i) { return I+i; }
79ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ConstExprIterator operator-(size_t i) { return I-i; }
80ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    const Expr * operator[](size_t idx) const;
81ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    signed operator-(const ConstExprIterator& R) const { return I - R.I; }
82ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    const Expr * operator*() const;
83ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    const Expr * operator->() const;
84ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    bool operator==(const ConstExprIterator& R) const { return I == R.I; }
85ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    bool operator!=(const ConstExprIterator& R) const { return I != R.I; }
86ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    bool operator>(const ConstExprIterator& R) const { return I > R.I; }
87ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    bool operator>=(const ConstExprIterator& R) const { return I >= R.I; }
881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  };
891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
90ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek//===----------------------------------------------------------------------===//
91ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek// AST classes for statements.
92ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek//===----------------------------------------------------------------------===//
931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// Stmt - This represents one statement.
955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass Stmt {
975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  enum StmtClass {
99f2cad8633e46cce12fc3d77c0bd451ffd7264bafDouglas Gregor    NoStmtClass = 0,
100f2cad8633e46cce12fc3d77c0bd451ffd7264bafDouglas Gregor#define STMT(CLASS, PARENT) CLASS##Class,
101f2cad8633e46cce12fc3d77c0bd451ffd7264bafDouglas Gregor#define FIRST_STMT(CLASS) firstStmtConstant = CLASS##Class,
102f2cad8633e46cce12fc3d77c0bd451ffd7264bafDouglas Gregor#define LAST_STMT(CLASS) lastStmtConstant = CLASS##Class,
103f2cad8633e46cce12fc3d77c0bd451ffd7264bafDouglas Gregor#define FIRST_EXPR(CLASS) firstExprConstant = CLASS##Class,
104f2cad8633e46cce12fc3d77c0bd451ffd7264bafDouglas Gregor#define LAST_EXPR(CLASS) lastExprConstant = CLASS##Class
1055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/AST/StmtNodes.def"
1065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
1075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerprivate:
10843d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// \brief The statement class.
10943d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  const unsigned sClass : 8;
1101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11143d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// \brief The reference count for this statement.
11243d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  unsigned RefCount : 24;
1138189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek
1148189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  // Make vanilla 'new' and 'delete' illegal for Stmts.
1158189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenekprotected:
1168189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  void* operator new(size_t bytes) throw() {
1178189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    assert(0 && "Stmts cannot be allocated with regular 'new'.");
1188189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return 0;
1198189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
1208189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  void operator delete(void* data) throw() {
1218189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    assert(0 && "Stmts cannot be released with regular 'delete'.");
1228189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
1231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1248189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenekpublic:
1258189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  // Only allow allocation of Stmts using the allocator in ASTContext
1261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // or by doing a placement new.
1278189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  void* operator new(size_t bytes, ASTContext& C,
1288189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek                     unsigned alignment = 16) throw() {
1298189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return ::operator new(bytes, C, alignment);
1308189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
1311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1328189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  void* operator new(size_t bytes, ASTContext* C,
1338189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek                     unsigned alignment = 16) throw() {
1348189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return ::operator new(bytes, *C, alignment);
1358189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
1361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1378189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  void* operator new(size_t bytes, void* mem) throw() {
1388189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return mem;
1398189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
140e2dedf8f61b8f306f704781456b482eb61871e8eDouglas Gregor
141fb523e16dd1f860ff02a3ae03e5e3e25327a5860Chris Lattner  void operator delete(void*, ASTContext&, unsigned) throw() { }
142fb523e16dd1f860ff02a3ae03e5e3e25327a5860Chris Lattner  void operator delete(void*, ASTContext*, unsigned) throw() { }
143e2dedf8f61b8f306f704781456b482eb61871e8eDouglas Gregor  void operator delete(void*, std::size_t) throw() { }
144fb523e16dd1f860ff02a3ae03e5e3e25327a5860Chris Lattner  void operator delete(void*, void*) throw() { }
145e2dedf8f61b8f306f704781456b482eb61871e8eDouglas Gregor
146025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregorpublic:
147025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief A placeholder type used to construct an empty shell of a
148025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// type, that will be filled in later (e.g., by some
149025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// de-serialization).
150025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  struct EmptyShell { };
151025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
1529c1863ef36a74e8203f00289d19856ad956f48b9Ted Kremenekprotected:
1539c1863ef36a74e8203f00289d19856ad956f48b9Ted Kremenek  /// DestroyChildren - Invoked by destructors of subclasses of Stmt to
1549c1863ef36a74e8203f00289d19856ad956f48b9Ted Kremenek  ///  recursively release child AST nodes.
15527f8a28bee33bb0e857cfe1a61c281bbc234b338Ted Kremenek  void DestroyChildren(ASTContext& Ctx);
1561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
157025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Construct an empty statement.
1581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  explicit Stmt(StmtClass SC, EmptyShell) : sClass(SC), RefCount(1) {
159025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor    if (Stmt::CollectingStats()) Stmt::addStmtClass(SC);
160025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  }
161025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
16242602bb40aefcc2751d4078ba88aacf4d965c9bdDouglas Gregor  /// \brief Virtual method that performs the actual destruction of
16342602bb40aefcc2751d4078ba88aacf4d965c9bdDouglas Gregor  /// this statement.
16442602bb40aefcc2751d4078ba88aacf4d965c9bdDouglas Gregor  ///
16542602bb40aefcc2751d4078ba88aacf4d965c9bdDouglas Gregor  /// Subclasses should override this method (not Destroy()) to
16642602bb40aefcc2751d4078ba88aacf4d965c9bdDouglas Gregor  /// provide class-specific destruction.
16742602bb40aefcc2751d4078ba88aacf4d965c9bdDouglas Gregor  virtual void DoDestroy(ASTContext &Ctx);
1681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
1701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Stmt(StmtClass SC) : sClass(SC), RefCount(1) {
1715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (Stmt::CollectingStats()) Stmt::addStmtClass(SC);
1725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  virtual ~Stmt() {}
1741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
175129e2df52ed7e0434b3f1cf1867fd6a5cb083ff6John McCall#ifndef NDEBUG
176129e2df52ed7e0434b3f1cf1867fd6a5cb083ff6John McCall  /// \brief True if this statement's refcount is in a valid state.
177129e2df52ed7e0434b3f1cf1867fd6a5cb083ff6John McCall  /// Should be used only in assertions.
178129e2df52ed7e0434b3f1cf1867fd6a5cb083ff6John McCall  bool isRetained() const {
179129e2df52ed7e0434b3f1cf1867fd6a5cb083ff6John McCall    return (RefCount >= 1);
180129e2df52ed7e0434b3f1cf1867fd6a5cb083ff6John McCall  }
181129e2df52ed7e0434b3f1cf1867fd6a5cb083ff6John McCall#endif
182129e2df52ed7e0434b3f1cf1867fd6a5cb083ff6John McCall
18342602bb40aefcc2751d4078ba88aacf4d965c9bdDouglas Gregor  /// \brief Destroy the current statement and its children.
1841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  void Destroy(ASTContext &Ctx) {
185cf5c19b9b7145e7fc092fa1309085deded25b0f1Fariborz Jahanian    assert(RefCount >= 1);
18643d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor    if (--RefCount == 0)
1871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      DoDestroy(Ctx);
18843d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  }
1895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
19043d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// \brief Increases the reference count for this statement.
19143d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  ///
19243d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// Invoke the Retain() operation when this statement or expression
19343d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// is being shared by another owner.
19443d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  Stmt *Retain() {
1958b55ce79391dcce51c52f8aff923f4607e6cd977Douglas Gregor    assert(RefCount >= 1);
19643d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor    ++RefCount;
19743d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor    return this;
19843d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  }
1991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
200b2f81cf7f8445e0c65c0428f4fbb0442566916b8Douglas Gregor  StmtClass getStmtClass() const {
201b2f81cf7f8445e0c65c0428f4fbb0442566916b8Douglas Gregor    assert(RefCount >= 1 && "Referencing already-destroyed statement!");
202b2f81cf7f8445e0c65c0428f4fbb0442566916b8Douglas Gregor    return (StmtClass)sClass;
203b2f81cf7f8445e0c65c0428f4fbb0442566916b8Douglas Gregor  }
2045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  const char *getStmtClassName() const;
2051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
206b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  /// SourceLocation tokens are not useful in isolation - they are low level
207b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  /// value objects created/interpreted by SourceManager. We assume AST
208b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  /// clients will have a pointer to the respective SourceManager.
209b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  virtual SourceRange getSourceRange() const = 0;
210311ff02fae0392bee6abe7723cdf5a69b2899a47Chris Lattner  SourceLocation getLocStart() const { return getSourceRange().getBegin(); }
211311ff02fae0392bee6abe7723cdf5a69b2899a47Chris Lattner  SourceLocation getLocEnd() const { return getSourceRange().getEnd(); }
2125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
2135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // global temp stats (until we have a per-module visitor)
2145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static void addStmtClass(const StmtClass s);
2152024f4d4b0d57616f79ea742fa782d633d414462Kovarththanan Rajaratnam  static bool CollectingStats(bool Enable = false);
2165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static void PrintStats();
2176000dace22f110d8768476989313e9d981d690d0Chris Lattner
2186000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// dump - This does a local dump of the specified AST fragment.  It dumps the
2196000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// specified node and a few nodes underneath it, but not the whole subtree.
2206000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// This is useful in a debugger.
2215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void dump() const;
222e300c870f08d08badf2ebcb53ded49f304af37fcChris Lattner  void dump(SourceManager &SM) const;
2236000dace22f110d8768476989313e9d981d690d0Chris Lattner
2246000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// dumpAll - This does a dump of the specified AST fragment and all subtrees.
2256000dace22f110d8768476989313e9d981d690d0Chris Lattner  void dumpAll() const;
226e300c870f08d08badf2ebcb53ded49f304af37fcChris Lattner  void dumpAll(SourceManager &SM) const;
2276000dace22f110d8768476989313e9d981d690d0Chris Lattner
2286000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// dumpPretty/printPretty - These two methods do a "pretty print" of the AST
2296000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// back to its original source language syntax.
23048d14a222276fad5279e994d1a062f36ae6fcbceEli Friedman  void dumpPretty(ASTContext& Context) const;
2311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  void printPretty(llvm::raw_ostream &OS, PrinterHelper *Helper,
232e4f2142d00fa5fdb580c4e2413da91882d955381Chris Lattner                   const PrintingPolicy &Policy,
23348d14a222276fad5279e994d1a062f36ae6fcbceEli Friedman                   unsigned Indentation = 0) const {
23448d14a222276fad5279e994d1a062f36ae6fcbceEli Friedman    printPretty(OS, *(ASTContext*)0, Helper, Policy, Indentation);
23548d14a222276fad5279e994d1a062f36ae6fcbceEli Friedman  }
236e4f2142d00fa5fdb580c4e2413da91882d955381Chris Lattner  void printPretty(llvm::raw_ostream &OS, ASTContext &Context,
2371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                   PrinterHelper *Helper,
238e4f2142d00fa5fdb580c4e2413da91882d955381Chris Lattner                   const PrintingPolicy &Policy,
239d249e1d1f1498b81314459ceda19d6ff25c278adDouglas Gregor                   unsigned Indentation = 0) const;
2401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
241d2a4a1af1088fca80e2dc76eb3369db0fbbfdefdTed Kremenek  /// viewAST - Visualize an AST rooted at this Stmt* using GraphViz.  Only
242d2a4a1af1088fca80e2dc76eb3369db0fbbfdefdTed Kremenek  ///   works on systems with GraphViz (Mac OS X) or dot+gv installed.
243d2a4a1af1088fca80e2dc76eb3369db0fbbfdefdTed Kremenek  void viewAST() const;
2441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Implement isa<T> support.
2461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *) { return true; }
2471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
248d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  /// hasImplicitControlFlow - Some statements (e.g. short circuited operations)
249d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  ///  contain implicit control-flow in the order their subexpressions
250d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  ///  are evaluated.  This predicate returns true if this statement has
251d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  ///  such implicit control-flow.  Such statements are also specially handled
252d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  ///  within CFGs.
253d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  bool hasImplicitControlFlow() const;
254d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek
25577ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  /// Child Iterators: All subclasses must implement child_begin and child_end
25677ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  ///  to permit easy iteration over the substatements/subexpessions of an
25777ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  ///  AST node.  This permits easy iteration over all nodes in the AST.
2589caf8b1ca6beb254f420dada3c0e94d5ef027f58Ted Kremenek  typedef StmtIterator       child_iterator;
2599caf8b1ca6beb254f420dada3c0e94d5ef027f58Ted Kremenek  typedef ConstStmtIterator  const_child_iterator;
2601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2618297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin() = 0;
2628297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end()   = 0;
2631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
26477ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  const_child_iterator child_begin() const {
2659caf8b1ca6beb254f420dada3c0e94d5ef027f58Ted Kremenek    return const_child_iterator(const_cast<Stmt*>(this)->child_begin());
26677ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  }
2671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
26877ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  const_child_iterator child_end() const {
2699caf8b1ca6beb254f420dada3c0e94d5ef027f58Ted Kremenek    return const_child_iterator(const_cast<Stmt*>(this)->child_end());
27077ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  }
27141ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor
27241ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// \brief Produce a unique representation of the given statement.
27341ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  ///
27441ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// \brief ID once the profiling operation is complete, will contain
27541ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// the unique representation of the given statement.
27641ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  ///
27741ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// \brief Context the AST context in which the statement resides
27841ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  ///
27941ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// \brief Canonical whether the profile should be based on the canonical
28041ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// representation of this statement (e.g., where non-type template
2811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  /// parameters are identified by index/level rather than their
28241ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// declaration pointers) or the exact representation of the statement as
28341ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// written in the source.
28441ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  void Profile(llvm::FoldingSetNodeID &ID, ASTContext &Context,
28541ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor               bool Canonical);
2865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
2875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
2885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// DeclStmt - Adaptor class for mixing declarations with statements and
2895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// expressions. For example, CompoundStmt mixes statements, expressions
2901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// and declarations (variables, types). Another example is ForStmt, where
2915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// the first statement can be an expression or a declaration.
2925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
2935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass DeclStmt : public Stmt {
2949653db7bd0e3665b955a0445859285f2e1e7dacdDouglas Gregor  DeclGroupRef DG;
29581c018d9482e7cc2addadc6202dcf162a01faefdChris Lattner  SourceLocation StartLoc, EndLoc;
2961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
297e7809d49413febf078d0503753987fe9f6061a68Ted Kremenekprotected:
298e7809d49413febf078d0503753987fe9f6061a68Ted Kremenek  virtual void DoDestroy(ASTContext &Ctx);
299e7809d49413febf078d0503753987fe9f6061a68Ted Kremenek
3005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
3011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  DeclStmt(DeclGroupRef dg, SourceLocation startLoc,
3028ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek           SourceLocation endLoc) : Stmt(DeclStmtClass), DG(dg),
3038ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek                                    StartLoc(startLoc), EndLoc(endLoc) {}
3041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
30584f2170062014d268951902164bed0d8bdea0e82Douglas Gregor  /// \brief Build an empty declaration statement.
30684f2170062014d268951902164bed0d8bdea0e82Douglas Gregor  explicit DeclStmt(EmptyShell Empty) : Stmt(DeclStmtClass, Empty) { }
30784f2170062014d268951902164bed0d8bdea0e82Douglas Gregor
3087e24e82a70a2c681f4291a3397bcd1e1005f251aChris Lattner  /// isSingleDecl - This method returns true if this DeclStmt refers
309fe95deaf66e4fbd82d44b5f6afa8162fa69cb85cChris Lattner  /// to a single Decl.
3107e24e82a70a2c681f4291a3397bcd1e1005f251aChris Lattner  bool isSingleDecl() const {
311fe95deaf66e4fbd82d44b5f6afa8162fa69cb85cChris Lattner    return DG.isSingleDecl();
3128ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  }
3131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3147e24e82a70a2c681f4291a3397bcd1e1005f251aChris Lattner  const Decl *getSingleDecl() const { return DG.getSingleDecl(); }
3151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Decl *getSingleDecl() { return DG.getSingleDecl(); }
3161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
317e66a8cf9117e5fb95a05ff76ec06615e63dd5adeChris Lattner  const DeclGroupRef getDeclGroup() const { return DG; }
318e66a8cf9117e5fb95a05ff76ec06615e63dd5adeChris Lattner  DeclGroupRef getDeclGroup() { return DG; }
31984f2170062014d268951902164bed0d8bdea0e82Douglas Gregor  void setDeclGroup(DeclGroupRef DGR) { DG = DGR; }
320b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
32181c018d9482e7cc2addadc6202dcf162a01faefdChris Lattner  SourceLocation getStartLoc() const { return StartLoc; }
32284f2170062014d268951902164bed0d8bdea0e82Douglas Gregor  void setStartLoc(SourceLocation L) { StartLoc = L; }
32381c018d9482e7cc2addadc6202dcf162a01faefdChris Lattner  SourceLocation getEndLoc() const { return EndLoc; }
32484f2170062014d268951902164bed0d8bdea0e82Douglas Gregor  void setEndLoc(SourceLocation L) { EndLoc = L; }
32584f2170062014d268951902164bed0d8bdea0e82Douglas Gregor
3268ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  SourceRange getSourceRange() const {
32781c018d9482e7cc2addadc6202dcf162a01faefdChris Lattner    return SourceRange(StartLoc, EndLoc);
32881c018d9482e7cc2addadc6202dcf162a01faefdChris Lattner  }
3291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
3311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == DeclStmtClass;
3325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
3335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const DeclStmt *) { return true; }
3341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
33514f8b4ff660bcaa763974b8d0fae81857c594495Ted Kremenek  // Iterators over subexpressions.
3368297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
3378297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
3381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3390632dd6fe068011af5710c0d6a745724021ff620Chris Lattner  typedef DeclGroupRef::iterator decl_iterator;
3400632dd6fe068011af5710c0d6a745724021ff620Chris Lattner  typedef DeclGroupRef::const_iterator const_decl_iterator;
3411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3428ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  decl_iterator decl_begin() { return DG.begin(); }
3438ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  decl_iterator decl_end() { return DG.end(); }
3448ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  const_decl_iterator decl_begin() const { return DG.begin(); }
3458ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  const_decl_iterator decl_end() const { return DG.end(); }
3465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
3475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
3485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// NullStmt - This is the null statement ";": C99 6.8.3p3.
3495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
3505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass NullStmt : public Stmt {
3515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  SourceLocation SemiLoc;
3525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
3535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  NullStmt(SourceLocation L) : Stmt(NullStmtClass), SemiLoc(L) {}
3545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
355025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build an empty null statement.
356025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit NullStmt(EmptyShell Empty) : Stmt(NullStmtClass, Empty) { }
357025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
3585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  SourceLocation getSemiLoc() const { return SemiLoc; }
359025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setSemiLoc(SourceLocation L) { SemiLoc = L; }
360b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
361507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  virtual SourceRange getSourceRange() const { return SourceRange(SemiLoc); }
3621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
3641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == NullStmtClass;
3655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
3665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const NullStmt *) { return true; }
3671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3688297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
3698297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
3708297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
3715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
3725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
3735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// CompoundStmt - This represents a group of statements like { stmt stmt }.
3745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
3755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass CompoundStmt : public Stmt {
3768189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  Stmt** Body;
3778189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  unsigned NumStmts;
378b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation LBracLoc, RBracLoc;
3795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
3801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  CompoundStmt(ASTContext& C, Stmt **StmtStart, unsigned numStmts,
3818189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek                             SourceLocation LB, SourceLocation RB)
3828189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  : Stmt(CompoundStmtClass), NumStmts(numStmts), LBracLoc(LB), RBracLoc(RB) {
383026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner    if (NumStmts == 0) {
384026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner      Body = 0;
385026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner      return;
386026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner    }
3871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
388026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner    Body = new (C) Stmt*[NumStmts];
389026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner    memcpy(Body, StmtStart, numStmts * sizeof(*Body));
3901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  }
391025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
392025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  // \brief Build an empty compound statement.
393025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit CompoundStmt(EmptyShell Empty)
394025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor    : Stmt(CompoundStmtClass, Empty), Body(0), NumStmts(0) { }
395025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
396025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setStmts(ASTContext &C, Stmt **Stmts, unsigned NumStmts);
3971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3988189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  bool body_empty() const { return NumStmts == 0; }
399025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  unsigned size() const { return NumStmts; }
400025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
4018189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  typedef Stmt** body_iterator;
4028189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  body_iterator body_begin() { return Body; }
4038189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  body_iterator body_end() { return Body + NumStmts; }
4048189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  Stmt *body_back() { return NumStmts ? Body[NumStmts-1] : 0; }
4055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
4068189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  typedef Stmt* const * const_body_iterator;
4078189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  const_body_iterator body_begin() const { return Body; }
4088189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  const_body_iterator body_end() const { return Body + NumStmts; }
4098189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  const Stmt *body_back() const { return NumStmts ? Body[NumStmts-1] : 0; }
4108189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek
4118189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  typedef std::reverse_iterator<body_iterator> reverse_body_iterator;
4128189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  reverse_body_iterator body_rbegin() {
4138189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return reverse_body_iterator(body_end());
4148189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
4158189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  reverse_body_iterator body_rend() {
4168189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return reverse_body_iterator(body_begin());
4178189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
4184ce854736dd196e2304f554ebeac8b43c89cf9e2Ted Kremenek
4198189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  typedef std::reverse_iterator<const_body_iterator>
4208189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek          const_reverse_body_iterator;
4214ce854736dd196e2304f554ebeac8b43c89cf9e2Ted Kremenek
4228189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  const_reverse_body_iterator body_rbegin() const {
4238189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return const_reverse_body_iterator(body_end());
4248189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
4251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4268189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  const_reverse_body_iterator body_rend() const {
4278189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return const_reverse_body_iterator(body_begin());
4288189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
4291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
4311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(LBracLoc, RBracLoc);
432b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
4331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
434cc326204dd97771c336b9aab3b9963ea30d69c29Ted Kremenek  SourceLocation getLBracLoc() const { return LBracLoc; }
435025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setLBracLoc(SourceLocation L) { LBracLoc = L; }
436cc326204dd97771c336b9aab3b9963ea30d69c29Ted Kremenek  SourceLocation getRBracLoc() const { return RBracLoc; }
437025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setRBracLoc(SourceLocation L) { RBracLoc = L; }
4381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
4401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == CompoundStmtClass;
4415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
4425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const CompoundStmt *) { return true; }
4431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4448297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
4458297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
4468297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
4475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
4485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
449c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson// SwitchCase is the base class for CaseStmt and DefaultStmt,
450c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlssonclass SwitchCase : public Stmt {
451103fc81f12aa635aa0a573c94b1aceb496b4e587Ted Kremenekprotected:
452c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  // A pointer to the following CaseStmt or DefaultStmt class,
453c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  // used by SwitchStmt.
454c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  SwitchCase *NextSwitchCase;
455103fc81f12aa635aa0a573c94b1aceb496b4e587Ted Kremenek
456d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  SwitchCase(StmtClass SC) : Stmt(SC), NextSwitchCase(0) {}
4571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
458c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlssonpublic:
459c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  const SwitchCase *getNextSwitchCase() const { return NextSwitchCase; }
460c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson
461c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  SwitchCase *getNextSwitchCase() { return NextSwitchCase; }
462c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson
463c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  void setNextSwitchCase(SwitchCase *SC) { NextSwitchCase = SC; }
464d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek
465d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Stmt *getSubStmt() { return v_getSubStmt(); }
466b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
467b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  virtual SourceRange getSourceRange() const { return SourceRange(); }
4681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
4701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == CaseStmtClass ||
471c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson    T->getStmtClass() == DefaultStmtClass;
472c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  }
473c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  static bool classof(const SwitchCase *) { return true; }
4743fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattnerprotected:
4751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual Stmt* v_getSubStmt() = 0;
476c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson};
477c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson
478c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlssonclass CaseStmt : public SwitchCase {
479d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  enum { SUBSTMT, LHS, RHS, END_EXPR };
4801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Stmt* SubExprs[END_EXPR];  // The expression for the RHS is Non-null for
481d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek                             // GNU "case 1 ... 4" extension
482b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation CaseLoc;
483dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation EllipsisLoc;
484dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation ColonLoc;
485dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor
4863fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner  virtual Stmt* v_getSubStmt() { return getSubStmt(); }
4875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
488dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  CaseStmt(Expr *lhs, Expr *rhs, SourceLocation caseLoc,
4891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump           SourceLocation ellipsisLoc, SourceLocation colonLoc)
490d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek    : SwitchCase(CaseStmtClass) {
49124e1e707b4c362f18e371e2bbf054a8345b57bfaChris Lattner    SubExprs[SUBSTMT] = 0;
492d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek    SubExprs[LHS] = reinterpret_cast<Stmt*>(lhs);
493d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek    SubExprs[RHS] = reinterpret_cast<Stmt*>(rhs);
494b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff    CaseLoc = caseLoc;
495dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor    EllipsisLoc = ellipsisLoc;
496dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor    ColonLoc = colonLoc;
497d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  }
498025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
499025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build an empty switch case statement.
500025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit CaseStmt(EmptyShell Empty) : SwitchCase(CaseStmtClass) { }
501025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
502764a7ce5217f9569e100a3445f47496ee82daf86Chris Lattner  SourceLocation getCaseLoc() const { return CaseLoc; }
503025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setCaseLoc(SourceLocation L) { CaseLoc = L; }
504dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation getEllipsisLoc() const { return EllipsisLoc; }
505dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  void setEllipsisLoc(SourceLocation L) { EllipsisLoc = L; }
506dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation getColonLoc() const { return ColonLoc; }
507dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  void setColonLoc(SourceLocation L) { ColonLoc = L; }
508025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
509d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Expr *getLHS() { return reinterpret_cast<Expr*>(SubExprs[LHS]); }
510d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Expr *getRHS() { return reinterpret_cast<Expr*>(SubExprs[RHS]); }
511d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Stmt *getSubStmt() { return SubExprs[SUBSTMT]; }
512025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
5131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const Expr *getLHS() const {
5141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return reinterpret_cast<const Expr*>(SubExprs[LHS]);
51551b09f2c528c8460b5465c676173324e44176d62Devang Patel  }
5161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const Expr *getRHS() const {
5171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return reinterpret_cast<const Expr*>(SubExprs[RHS]);
51851b09f2c528c8460b5465c676173324e44176d62Devang Patel  }
51951b09f2c528c8460b5465c676173324e44176d62Devang Patel  const Stmt *getSubStmt() const { return SubExprs[SUBSTMT]; }
5205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
52120dabe8b8ff7ed6d2d0158fee43755a4bc7642c3Chris Lattner  void setSubStmt(Stmt *S) { SubExprs[SUBSTMT] = S; }
52220dabe8b8ff7ed6d2d0158fee43755a4bc7642c3Chris Lattner  void setLHS(Expr *Val) { SubExprs[LHS] = reinterpret_cast<Stmt*>(Val); }
52320dabe8b8ff7ed6d2d0158fee43755a4bc7642c3Chris Lattner  void setRHS(Expr *Val) { SubExprs[RHS] = reinterpret_cast<Stmt*>(Val); }
5241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5263fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner  virtual SourceRange getSourceRange() const {
5273fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner    // Handle deeply nested case statements with iteration instead of recursion.
5283fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner    const CaseStmt *CS = this;
52991ee0140ecb60b5c1402edc9e577257636c4ca60Chris Lattner    while (const CaseStmt *CS2 = dyn_cast<CaseStmt>(CS->getSubStmt()))
5303fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner      CS = CS2;
5311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(CaseLoc, CS->getSubStmt()->getLocEnd());
533b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
5341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
5351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == CaseStmtClass;
5365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
5375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const CaseStmt *) { return true; }
5381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
539d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  // Iterators
540d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  virtual child_iterator child_begin();
541d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  virtual child_iterator child_end();
5425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
5435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
544c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlssonclass DefaultStmt : public SwitchCase {
545d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Stmt* SubStmt;
5466c36be5b383875b490684bcf439d6d427298c1afChris Lattner  SourceLocation DefaultLoc;
547dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation ColonLoc;
5483fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner  virtual Stmt* v_getSubStmt() { return getSubStmt(); }
5495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
5501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  DefaultStmt(SourceLocation DL, SourceLocation CL, Stmt *substmt) :
551dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor    SwitchCase(DefaultStmtClass), SubStmt(substmt), DefaultLoc(DL),
552dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor    ColonLoc(CL) {}
553025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
554025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build an empty default statement.
555025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit DefaultStmt(EmptyShell) : SwitchCase(DefaultStmtClass) { }
556025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
557d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Stmt *getSubStmt() { return SubStmt; }
55851b09f2c528c8460b5465c676173324e44176d62Devang Patel  const Stmt *getSubStmt() const { return SubStmt; }
559025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setSubStmt(Stmt *S) { SubStmt = S; }
560025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
5616c36be5b383875b490684bcf439d6d427298c1afChris Lattner  SourceLocation getDefaultLoc() const { return DefaultLoc; }
562025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setDefaultLoc(SourceLocation L) { DefaultLoc = L; }
563dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation getColonLoc() const { return ColonLoc; }
564dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  void setColonLoc(SourceLocation L) { ColonLoc = L; }
5655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
5661eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
5671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(DefaultLoc, SubStmt->getLocEnd());
568b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
5691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
5701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == DefaultStmtClass;
5715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
5725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const DefaultStmt *) { return true; }
5731eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
574d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  // Iterators
575d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  virtual child_iterator child_begin();
576d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  virtual child_iterator child_end();
5775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
5785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
5795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass LabelStmt : public Stmt {
5805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  IdentifierInfo *Label;
5815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  Stmt *SubStmt;
582b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation IdentLoc;
5835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
5841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  LabelStmt(SourceLocation IL, IdentifierInfo *label, Stmt *substmt)
5851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    : Stmt(LabelStmtClass), Label(label),
586b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff      SubStmt(substmt), IdentLoc(IL) {}
5871de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor
5881de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  // \brief Build an empty label statement.
5891de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  explicit LabelStmt(EmptyShell Empty) : Stmt(LabelStmtClass, Empty) { }
5901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  SourceLocation getIdentLoc() const { return IdentLoc; }
5925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  IdentifierInfo *getID() const { return Label; }
5931de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  void setID(IdentifierInfo *II) { Label = II; }
5945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  const char *getName() const;
5955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  Stmt *getSubStmt() { return SubStmt; }
5965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  const Stmt *getSubStmt() const { return SubStmt; }
5975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void setIdentLoc(SourceLocation L) { IdentLoc = L; }
5985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void setSubStmt(Stmt *SS) { SubStmt = SS; }
599b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
6001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
6011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(IdentLoc, SubStmt->getLocEnd());
6021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  }
6031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
6041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == LabelStmtClass;
6055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
6065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const LabelStmt *) { return true; }
6071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6088297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
6098297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
6108297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
6115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
6125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// IfStmt - This represents an if/then/else.
6155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
6165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass IfStmt : public Stmt {
6178297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  enum { COND, THEN, ELSE, END_EXPR };
6188297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt* SubExprs[END_EXPR];
6198cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor
6208cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// \brief If non-NULL, the declaration in the "if" statement.
6218cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  VarDecl *Var;
6228cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor
623b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation IfLoc;
624d06f6ca61062f85926eb9d409eb3d4f8afcf93c7Douglas Gregor  SourceLocation ElseLoc;
6258cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor
6265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
62735628d1f17c817f8c240208db7ba490e3109981bTed Kremenek  IfStmt(SourceLocation IL, VarDecl *var, Expr *cond, Stmt *then,
6281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump         SourceLocation EL = SourceLocation(), Stmt *elsev = 0)
62935628d1f17c817f8c240208db7ba490e3109981bTed Kremenek    : Stmt(IfStmtClass), Var(var), IfLoc(IL), ElseLoc(EL)  {
6308297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek    SubExprs[COND] = reinterpret_cast<Stmt*>(cond);
6318297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek    SubExprs[THEN] = then;
6328297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek    SubExprs[ELSE] = elsev;
6338297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  }
6341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
635025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build an empty if/then/else statement
636025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit IfStmt(EmptyShell Empty) : Stmt(IfStmtClass, Empty) { }
637025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
6388cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// \brief Retrieve the variable declared in this "if" statement, if any.
6398cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  ///
6408cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// In the following example, "x" is the condition variable.
6418cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// \code
6428cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// if (int x = foo()) {
6438cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  ///   printf("x is %d", x);
6448cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// }
6458cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// \endcode
6468cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  VarDecl *getConditionVariable() const { return Var; }
6478cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  void setConditionVariable(VarDecl *V) { Var = V; }
6488cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor
6498297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getCond() const { return reinterpret_cast<Expr*>(SubExprs[COND]);}
650025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setCond(Expr *E) { SubExprs[COND] = reinterpret_cast<Stmt *>(E); }
6518297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getThen() const { return SubExprs[THEN]; }
6521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  void setThen(Stmt *S) { SubExprs[THEN] = S; }
6538297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getElse() const { return SubExprs[ELSE]; }
654025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setElse(Stmt *S) { SubExprs[ELSE] = S; }
6555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6568297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getCond() { return reinterpret_cast<Expr*>(SubExprs[COND]); }
6578297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getThen() { return SubExprs[THEN]; }
6588297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getElse() { return SubExprs[ELSE]; }
659b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
660025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  SourceLocation getIfLoc() const { return IfLoc; }
661025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setIfLoc(SourceLocation L) { IfLoc = L; }
662d06f6ca61062f85926eb9d409eb3d4f8afcf93c7Douglas Gregor  SourceLocation getElseLoc() const { return ElseLoc; }
663d06f6ca61062f85926eb9d409eb3d4f8afcf93c7Douglas Gregor  void setElseLoc(SourceLocation L) { ElseLoc = L; }
664025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
6651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
666b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff    if (SubExprs[ELSE])
667b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff      return SourceRange(IfLoc, SubExprs[ELSE]->getLocEnd());
668b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff    else
669b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff      return SourceRange(IfLoc, SubExprs[THEN]->getLocEnd());
670b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
6711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
6731eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == IfStmtClass;
6745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
6755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const IfStmt *) { return true; }
6761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
67735628d1f17c817f8c240208db7ba490e3109981bTed Kremenek  // Iterators over subexpressions.  The iterators will include iterating
67835628d1f17c817f8c240208db7ba490e3109981bTed Kremenek  // over the initialization expression referenced by the condition variable.
6798297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
6808297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
68135628d1f17c817f8c240208db7ba490e3109981bTed Kremenek
68235628d1f17c817f8c240208db7ba490e3109981bTed Kremenekprotected:
68335628d1f17c817f8c240208db7ba490e3109981bTed Kremenek  virtual void DoDestroy(ASTContext &Ctx);
6845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
6855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// SwitchStmt - This represents a 'switch' stmt.
6875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
6885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass SwitchStmt : public Stmt {
6898297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  enum { COND, BODY, END_EXPR };
6901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Stmt* SubExprs[END_EXPR];
691d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  VarDecl *Var;
692c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  // This points to a linked list of case and default statements.
693c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  SwitchCase *FirstCase;
6949dcbfa450d751bd68fc4af8b75da381d4f6984b9Steve Naroff  SourceLocation SwitchLoc;
6951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
69643d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregorprotected:
69743d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  virtual void DoDestroy(ASTContext &Ctx);
6981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
700d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  SwitchStmt(VarDecl *Var, Expr *cond)
701d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor    : Stmt(SwitchStmtClass), Var(Var), FirstCase(0)
702d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  {
703d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor    SubExprs[COND] = reinterpret_cast<Stmt*>(cond);
704d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor    SubExprs[BODY] = NULL;
705d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  }
7061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
707025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build a empty switch statement.
708025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit SwitchStmt(EmptyShell Empty) : Stmt(SwitchStmtClass, Empty) { }
709025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
710d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// \brief Retrieve the variable declared in this "switch" statement, if any.
711d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  ///
712d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// In the following example, "x" is the condition variable.
713d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// \code
714d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// switch (int x = foo()) {
715d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  ///   case 0: break;
716d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  ///   // ...
717d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// }
718d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// \endcode
719d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  VarDecl *getConditionVariable() const { return Var; }
720d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  void setConditionVariable(VarDecl *V) { Var = V; }
721d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor
7228297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getCond() const { return reinterpret_cast<Expr*>(SubExprs[COND]);}
7238297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getBody() const { return SubExprs[BODY]; }
724c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  const SwitchCase *getSwitchCaseList() const { return FirstCase; }
725c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson
7268297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getCond() { return reinterpret_cast<Expr*>(SubExprs[COND]);}
727025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setCond(Expr *E) { SubExprs[COND] = reinterpret_cast<Stmt *>(E); }
7288297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getBody() { return SubExprs[BODY]; }
729025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setBody(Stmt *S) { SubExprs[BODY] = S; }
730c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  SwitchCase *getSwitchCaseList() { return FirstCase; }
7311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
73243d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// \brief Set the case list for this switch statement.
73343d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  ///
73443d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// The caller is responsible for incrementing the retain counts on
73543d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// all of the SwitchCase statements in this list.
736025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setSwitchCaseList(SwitchCase *SC) { FirstCase = SC; }
737025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
738025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  SourceLocation getSwitchLoc() const { return SwitchLoc; }
739025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setSwitchLoc(SourceLocation L) { SwitchLoc = L; }
740c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson
7411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  void setBody(Stmt *S, SourceLocation SL) {
7421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    SubExprs[BODY] = S;
7439dcbfa450d751bd68fc4af8b75da381d4f6984b9Steve Naroff    SwitchLoc = SL;
7441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  }
745c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  void addSwitchCase(SwitchCase *SC) {
746c2ee10d79f70036af652a395ac1f8273f3d04e12Douglas Gregor    assert(!SC->getNextSwitchCase() && "case/default already added to a switch");
74743d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor    SC->Retain();
748c2ee10d79f70036af652a395ac1f8273f3d04e12Douglas Gregor    SC->setNextSwitchCase(FirstCase);
749c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson    FirstCase = SC;
750c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  }
7511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
7521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(SwitchLoc, SubExprs[BODY]->getLocEnd());
7539dcbfa450d751bd68fc4af8b75da381d4f6984b9Steve Naroff  }
7541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
7551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == SwitchStmtClass;
7565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
7575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const SwitchStmt *) { return true; }
7581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7598297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
7608297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
7618297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
7625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
7635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// WhileStmt - This represents a 'while' stmt.
7665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
7675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass WhileStmt : public Stmt {
7688297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  enum { COND, BODY, END_EXPR };
7695656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  VarDecl *Var;
7708297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt* SubExprs[END_EXPR];
771b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation WhileLoc;
7725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
7735656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  WhileStmt(VarDecl *Var, Expr *cond, Stmt *body, SourceLocation WL)
7745656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor    : Stmt(WhileStmtClass), Var(Var)
7755656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  {
7768297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek    SubExprs[COND] = reinterpret_cast<Stmt*>(cond);
7778297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek    SubExprs[BODY] = body;
778b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff    WhileLoc = WL;
7798297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  }
7801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
781d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  /// \brief Build an empty while statement.
782d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  explicit WhileStmt(EmptyShell Empty) : Stmt(WhileStmtClass, Empty) { }
783d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor
7845656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// \brief Retrieve the variable declared in this "while" statement, if any.
7855656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  ///
7865656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// In the following example, "x" is the condition variable.
7875656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// \code
7885656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// while (int x = random()) {
7895656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  ///   // ...
7905656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// }
7915656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// \endcode
7925656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  VarDecl *getConditionVariable() const { return Var; }
7935656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  void setConditionVariable(VarDecl *V) { Var = V; }
7945656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor
7958297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getCond() { return reinterpret_cast<Expr*>(SubExprs[COND]); }
7968297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getCond() const { return reinterpret_cast<Expr*>(SubExprs[COND]);}
797d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  void setCond(Expr *E) { SubExprs[COND] = reinterpret_cast<Stmt*>(E); }
7988297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getBody() { return SubExprs[BODY]; }
7998297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getBody() const { return SubExprs[BODY]; }
800d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  void setBody(Stmt *S) { SubExprs[BODY] = S; }
801d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor
802d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  SourceLocation getWhileLoc() const { return WhileLoc; }
803d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  void setWhileLoc(SourceLocation L) { WhileLoc = L; }
804b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
8051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
8061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(WhileLoc, SubExprs[BODY]->getLocEnd());
807b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
8081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
8091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == WhileStmtClass;
8105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
8115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const WhileStmt *) { return true; }
8121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8138297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
8148297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
8158297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
8167d02b8c3be58676a03046eef9d056063e55ada3eTed Kremenek
8177d02b8c3be58676a03046eef9d056063e55ada3eTed Kremenekprotected:
8187d02b8c3be58676a03046eef9d056063e55ada3eTed Kremenek  virtual void DoDestroy(ASTContext &Ctx);
8195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
8205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
8215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// DoStmt - This represents a 'do/while' stmt.
8225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
8235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass DoStmt : public Stmt {
8248297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  enum { COND, BODY, END_EXPR };
8258297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt* SubExprs[END_EXPR];
826b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation DoLoc;
8279f3ca2a7747bd47f14d7693f333103fac29a24d2Douglas Gregor  SourceLocation WhileLoc;
828989135901c750af61ef012b6b0a0368be415bc46Chris Lattner  SourceLocation RParenLoc;  // Location of final ')' in do stmt condition.
8299f3ca2a7747bd47f14d7693f333103fac29a24d2Douglas Gregor
8305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
831989135901c750af61ef012b6b0a0368be415bc46Chris Lattner  DoStmt(Stmt *body, Expr *cond, SourceLocation DL, SourceLocation WL,
832989135901c750af61ef012b6b0a0368be415bc46Chris Lattner         SourceLocation RP)
833989135901c750af61ef012b6b0a0368be415bc46Chris Lattner    : Stmt(DoStmtClass), DoLoc(DL), WhileLoc(WL), RParenLoc(RP) {
8348297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek    SubExprs[COND] = reinterpret_cast<Stmt*>(cond);
8358297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek    SubExprs[BODY] = body;
8361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  }
83767d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor
83867d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  /// \brief Build an empty do-while statement.
83967d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  explicit DoStmt(EmptyShell Empty) : Stmt(DoStmtClass, Empty) { }
8401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8418297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getCond() { return reinterpret_cast<Expr*>(SubExprs[COND]); }
8428297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getCond() const { return reinterpret_cast<Expr*>(SubExprs[COND]);}
84367d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setCond(Expr *E) { SubExprs[COND] = reinterpret_cast<Stmt*>(E); }
8448297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getBody() { return SubExprs[BODY]; }
8451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const Stmt *getBody() const { return SubExprs[BODY]; }
84667d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setBody(Stmt *S) { SubExprs[BODY] = S; }
84767d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor
84867d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  SourceLocation getDoLoc() const { return DoLoc; }
84967d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setDoLoc(SourceLocation L) { DoLoc = L; }
8509f3ca2a7747bd47f14d7693f333103fac29a24d2Douglas Gregor  SourceLocation getWhileLoc() const { return WhileLoc; }
8519f3ca2a7747bd47f14d7693f333103fac29a24d2Douglas Gregor  void setWhileLoc(SourceLocation L) { WhileLoc = L; }
852b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
853989135901c750af61ef012b6b0a0368be415bc46Chris Lattner  SourceLocation getRParenLoc() const { return RParenLoc; }
854989135901c750af61ef012b6b0a0368be415bc46Chris Lattner  void setRParenLoc(SourceLocation L) { RParenLoc = L; }
855989135901c750af61ef012b6b0a0368be415bc46Chris Lattner
8561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
8571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(DoLoc, RParenLoc);
858b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
8591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
8601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == DoStmtClass;
8615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
8625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const DoStmt *) { return true; }
8638297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek
8648297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
8658297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
8668297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
8675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
8685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
8695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
8705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// ForStmt - This represents a 'for (init;cond;inc)' stmt.  Note that any of
8715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// the init/cond/inc parts of the ForStmt will be null if they were not
8725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// specified in the source.
8735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
8745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass ForStmt : public Stmt {
8758297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  enum { INIT, COND, INC, BODY, END_EXPR };
8768297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt* SubExprs[END_EXPR]; // SubExprs[INIT] is an expression or declstmt.
87799e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  VarDecl *CondVar;
878b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation ForLoc;
8795831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor  SourceLocation LParenLoc, RParenLoc;
8805831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor
8815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
8826281213d87e71b76f64c4ca62fbe7a97f18ae0a0Ted Kremenek  ForStmt(Stmt *Init, Expr *Cond, VarDecl *condVar, Expr *Inc, Stmt *Body,
88399e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor          SourceLocation FL, SourceLocation LP, SourceLocation RP)
8846281213d87e71b76f64c4ca62fbe7a97f18ae0a0Ted Kremenek    : Stmt(ForStmtClass), CondVar(condVar), ForLoc(FL), LParenLoc(LP),
88599e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor      RParenLoc(RP)
88699e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  {
887af458c9f658335bdea482395ff4a25c75935e129Chris Lattner    SubExprs[INIT] = Init;
888af458c9f658335bdea482395ff4a25c75935e129Chris Lattner    SubExprs[COND] = reinterpret_cast<Stmt*>(Cond);
889af458c9f658335bdea482395ff4a25c75935e129Chris Lattner    SubExprs[INC] = reinterpret_cast<Stmt*>(Inc);
890af458c9f658335bdea482395ff4a25c75935e129Chris Lattner    SubExprs[BODY] = Body;
891af458c9f658335bdea482395ff4a25c75935e129Chris Lattner  }
8921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
89367d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  /// \brief Build an empty for statement.
89467d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  explicit ForStmt(EmptyShell Empty) : Stmt(ForStmtClass, Empty) { }
89567d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor
8968297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getInit() { return SubExprs[INIT]; }
89799e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor
89899e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// \brief Retrieve the variable declared in this "for" statement, if any.
89999e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  ///
90099e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// In the following example, "y" is the condition variable.
90199e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// \code
90299e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// for (int x = random(); int y = mangle(x); ++x) {
90399e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  ///   // ...
90499e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// }
90599e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// \endcode
90699e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  VarDecl *getConditionVariable() const { return CondVar; }
90799e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  void setConditionVariable(VarDecl *V) { CondVar = V; }
90899e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor
9098297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getCond() { return reinterpret_cast<Expr*>(SubExprs[COND]); }
9108297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getInc()  { return reinterpret_cast<Expr*>(SubExprs[INC]); }
9118297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getBody() { return SubExprs[BODY]; }
9128297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek
9138297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getInit() const { return SubExprs[INIT]; }
9148297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getCond() const { return reinterpret_cast<Expr*>(SubExprs[COND]);}
9158297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getInc()  const { return reinterpret_cast<Expr*>(SubExprs[INC]); }
9168297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getBody() const { return SubExprs[BODY]; }
917b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
91867d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setInit(Stmt *S) { SubExprs[INIT] = S; }
91967d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setCond(Expr *E) { SubExprs[COND] = reinterpret_cast<Stmt*>(E); }
92067d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setInc(Expr *E) { SubExprs[INC] = reinterpret_cast<Stmt*>(E); }
92167d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setBody(Stmt *S) { SubExprs[BODY] = S; }
92267d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor
92367d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  SourceLocation getForLoc() const { return ForLoc; }
92467d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setForLoc(SourceLocation L) { ForLoc = L; }
9255831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor  SourceLocation getLParenLoc() const { return LParenLoc; }
9265831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor  void setLParenLoc(SourceLocation L) { LParenLoc = L; }
9275831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor  SourceLocation getRParenLoc() const { return RParenLoc; }
9285831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor  void setRParenLoc(SourceLocation L) { RParenLoc = L; }
92967d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor
9301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
9311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(ForLoc, SubExprs[BODY]->getLocEnd());
932b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
9331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
9341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == ForStmtClass;
9355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
9365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const ForStmt *) { return true; }
9371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9388297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
9398297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
9408297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
941f0d975f25b1d71612fcbd58ce8abbbc4532c6bb5Ted Kremenek
942f0d975f25b1d71612fcbd58ce8abbbc4532c6bb5Ted Kremenekprotected:
943f0d975f25b1d71612fcbd58ce8abbbc4532c6bb5Ted Kremenek  virtual void DoDestroy(ASTContext &Ctx);
9445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
9451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// GotoStmt - This represents a direct goto.
9475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
9485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass GotoStmt : public Stmt {
9495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  LabelStmt *Label;
950507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  SourceLocation GotoLoc;
95161f62165220e75694fe333179c78815e2e48d71fTed Kremenek  SourceLocation LabelLoc;
9525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
9531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  GotoStmt(LabelStmt *label, SourceLocation GL, SourceLocation LL)
95461f62165220e75694fe333179c78815e2e48d71fTed Kremenek    : Stmt(GotoStmtClass), Label(label), GotoLoc(GL), LabelLoc(LL) {}
9551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9561de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  /// \brief Build an empty goto statement.
9571de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  explicit GotoStmt(EmptyShell Empty) : Stmt(GotoStmtClass, Empty) { }
9581de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor
9595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  LabelStmt *getLabel() const { return Label; }
9601de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  void setLabel(LabelStmt *S) { Label = S; }
9611de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor
9621de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  SourceLocation getGotoLoc() const { return GotoLoc; }
9631de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  void setGotoLoc(SourceLocation L) { GotoLoc = L; }
9641de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  SourceLocation getLabelLoc() const { return LabelLoc; }
9651de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  void setLabelLoc(SourceLocation L) { LabelLoc = L; }
966b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
9671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
9681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(GotoLoc, LabelLoc);
969507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  }
9701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
9711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == GotoStmtClass;
9725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
9735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const GotoStmt *) { return true; }
9741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9758297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
9768297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
9778297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
9785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
9795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
9805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// IndirectGotoStmt - This represents an indirect goto.
9815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
9825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass IndirectGotoStmt : public Stmt {
983ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner  SourceLocation GotoLoc;
9845f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor  SourceLocation StarLoc;
9851060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek  Stmt *Target;
9865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
9871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  IndirectGotoStmt(SourceLocation gotoLoc, SourceLocation starLoc,
9885f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor                   Expr *target)
9895f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor    : Stmt(IndirectGotoStmtClass), GotoLoc(gotoLoc), StarLoc(starLoc),
9905f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor      Target((Stmt*)target) {}
9917d5c2f241c74e5f8d9ec492e8f9f268e5e9ae41fDouglas Gregor
9927d5c2f241c74e5f8d9ec492e8f9f268e5e9ae41fDouglas Gregor  /// \brief Build an empty indirect goto statement.
9931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  explicit IndirectGotoStmt(EmptyShell Empty)
9947d5c2f241c74e5f8d9ec492e8f9f268e5e9ae41fDouglas Gregor    : Stmt(IndirectGotoStmtClass, Empty) { }
9951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
996ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner  void setGotoLoc(SourceLocation L) { GotoLoc = L; }
997ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner  SourceLocation getGotoLoc() const { return GotoLoc; }
9985f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor  void setStarLoc(SourceLocation L) { StarLoc = L; }
9995f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor  SourceLocation getStarLoc() const { return StarLoc; }
10001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10011060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek  Expr *getTarget();
10021060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek  const Expr *getTarget() const;
10037d5c2f241c74e5f8d9ec492e8f9f268e5e9ae41fDouglas Gregor  void setTarget(Expr *E) { Target = reinterpret_cast<Stmt*>(E); }
1004b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
1005ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner  virtual SourceRange getSourceRange() const {
1006ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner    return SourceRange(GotoLoc, Target->getLocEnd());
1007ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner  }
10081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
10101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == IndirectGotoStmtClass;
10115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
10125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const IndirectGotoStmt *) { return true; }
10131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10148297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
10158297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
10168297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
10175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
10185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// ContinueStmt - This represents a continue.
10215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
10225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass ContinueStmt : public Stmt {
1023507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  SourceLocation ContinueLoc;
10245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
1025507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  ContinueStmt(SourceLocation CL) : Stmt(ContinueStmtClass), ContinueLoc(CL) {}
10261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1027d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  /// \brief Build an empty continue statement.
1028d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  explicit ContinueStmt(EmptyShell Empty) : Stmt(ContinueStmtClass, Empty) { }
1029d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor
1030d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  SourceLocation getContinueLoc() const { return ContinueLoc; }
1031d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  void setContinueLoc(SourceLocation L) { ContinueLoc = L; }
1032d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor
10331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
10341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(ContinueLoc);
1035507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  }
1036861ce3178c70cfb0fa50baf685e1ad363538eaa9Douglas Gregor
10371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
10381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == ContinueStmtClass;
10395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
10405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const ContinueStmt *) { return true; }
10411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10428297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
10438297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
10448297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
10455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
10465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// BreakStmt - This represents a break.
10485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
10495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass BreakStmt : public Stmt {
1050507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  SourceLocation BreakLoc;
10515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
1052507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  BreakStmt(SourceLocation BL) : Stmt(BreakStmtClass), BreakLoc(BL) {}
10531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1054025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build an empty break statement.
1055025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit BreakStmt(EmptyShell Empty) : Stmt(BreakStmtClass, Empty) { }
1056025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
1057025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  SourceLocation getBreakLoc() const { return BreakLoc; }
1058025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setBreakLoc(SourceLocation L) { BreakLoc = L; }
1059025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
1060507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  virtual SourceRange getSourceRange() const { return SourceRange(BreakLoc); }
1061b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
10621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
10631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == BreakStmtClass;
10645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
10655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const BreakStmt *) { return true; }
10661eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10678297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
10688297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
10698297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
10705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
10715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1073c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner/// ReturnStmt - This represents a return, optionally of an expression:
1074c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner///   return;
1075c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner///   return 4;
1076c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner///
1077c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner/// Note that GCC allows return with no argument in a function declared to
1078c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner/// return a value, and it allows returning a value in functions declared to
1079c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner/// return void.  We explicitly model this in the AST, which means you can't
1080c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner/// depend on the return type of the function and the presence of an argument.
10815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
10825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass ReturnStmt : public Stmt {
10831060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek  Stmt *RetExpr;
1084507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  SourceLocation RetLoc;
10855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
10861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  ReturnStmt(SourceLocation RL, Expr *E = 0) : Stmt(ReturnStmtClass),
10871060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek    RetExpr((Stmt*) E), RetLoc(RL) {}
10880de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor
10890de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor  /// \brief Build an empty return expression.
10900de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor  explicit ReturnStmt(EmptyShell Empty) : Stmt(ReturnStmtClass, Empty) { }
10910de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor
10921060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek  const Expr *getRetValue() const;
10931060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek  Expr *getRetValue();
10940de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor  void setRetValue(Expr *E) { RetExpr = reinterpret_cast<Stmt*>(E); }
10950de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor
10960de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor  SourceLocation getReturnLoc() const { return RetLoc; }
10970de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor  void setReturnLoc(SourceLocation L) { RetLoc = L; }
1098b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
1099507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  virtual SourceRange getSourceRange() const;
11001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
11021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == ReturnStmtClass;
11035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
11045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const ReturnStmt *) { return true; }
11051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11068297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
11078297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
11088297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
11095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
11105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1111fe795956194141c91ae555985c9b930595bff43fChris Lattner/// AsmStmt - This represents a GNU inline-assembly statement extension.
1112fe795956194141c91ae555985c9b930595bff43fChris Lattner///
1113fe795956194141c91ae555985c9b930595bff43fChris Lattnerclass AsmStmt : public Stmt {
1114fe795956194141c91ae555985c9b930595bff43fChris Lattner  SourceLocation AsmLoc, RParenLoc;
11156a0ef4b83c91a6d6d5acb4ed5577c4659fe022a3Anders Carlsson  StringLiteral *AsmStr;
1116b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson
1117dfab34a696d1dba8622248c31aaf605906cb6109Anders Carlsson  bool IsSimple;
111839c47b56f45437bbc49c9568b7308a400234a730Anders Carlsson  bool IsVolatile;
11193b11fd3b52e7b88233c602407c151d07cb093e75Mike Stump  bool MSAsm;
11201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1121b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  unsigned NumOutputs;
1122b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  unsigned NumInputs;
11231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1124b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  llvm::SmallVector<std::string, 4> Names;
1125b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  llvm::SmallVector<StringLiteral*, 4> Constraints;
1126ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  llvm::SmallVector<Stmt*, 4> Exprs;
1127b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson
1128b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  llvm::SmallVector<StringLiteral*, 4> Clobbers;
1129fe795956194141c91ae555985c9b930595bff43fChris Lattnerpublic:
11303b11fd3b52e7b88233c602407c151d07cb093e75Mike Stump  AsmStmt(SourceLocation asmloc, bool issimple, bool isvolatile, bool msasm,
11311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump          unsigned numoutputs, unsigned numinputs,
1132703e39486689d6660e75f6b6de0068db031a51c7Anders Carlsson          const std::string *names, StringLiteral **constraints,
1133db6ed1786bf460e1143f67d14bf2d71ad9856f81Chris Lattner          Expr **exprs, StringLiteral *asmstr, unsigned numclobbers,
1134db6ed1786bf460e1143f67d14bf2d71ad9856f81Chris Lattner          StringLiteral **clobbers, SourceLocation rparenloc);
1135b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson
1136cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  /// \brief Build an empty inline-assembly statement.
1137cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  explicit AsmStmt(EmptyShell Empty) : Stmt(AsmStmtClass, Empty) { }
1138cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
1139cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  SourceLocation getAsmLoc() const { return AsmLoc; }
1140cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  void setAsmLoc(SourceLocation L) { AsmLoc = L; }
1141cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  SourceLocation getRParenLoc() const { return RParenLoc; }
1142cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  void setRParenLoc(SourceLocation L) { RParenLoc = L; }
1143cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
114439c47b56f45437bbc49c9568b7308a400234a730Anders Carlsson  bool isVolatile() const { return IsVolatile; }
1145cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  void setVolatile(bool V) { IsVolatile = V; }
1146dfab34a696d1dba8622248c31aaf605906cb6109Anders Carlsson  bool isSimple() const { return IsSimple; }
11476ffe643322949dd776285a6df60d3578f3918be4Chris Lattner  void setSimple(bool V) { IsSimple = V; }
11483b11fd3b52e7b88233c602407c151d07cb093e75Mike Stump  bool isMSAsm() const { return MSAsm; }
11493b11fd3b52e7b88233c602407c151d07cb093e75Mike Stump  void setMSAsm(bool V) { MSAsm = V; }
1150dfab34a696d1dba8622248c31aaf605906cb6109Anders Carlsson
1151458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  //===--- Asm String Analysis ---===//
1152458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner
1153458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  const StringLiteral *getAsmString() const { return AsmStr; }
1154458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  StringLiteral *getAsmString() { return AsmStr; }
1155cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  void setAsmString(StringLiteral *E) { AsmStr = E; }
1156cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
1157458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// AsmStringPiece - this is part of a decomposed asm string specification
1158458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// (for use with the AnalyzeAsmString function below).  An asm string is
1159458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// considered to be a concatenation of these parts.
1160458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  class AsmStringPiece {
1161458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  public:
1162458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    enum Kind {
1163458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      String,  // String in .ll asm string form, "$" -> "$$" and "%%" -> "%".
1164458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      Operand  // Operand reference, with optional modifier %c4.
1165458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    };
1166458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  private:
1167458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    Kind MyKind;
1168458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    std::string Str;
1169458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    unsigned OperandNo;
1170458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  public:
1171458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    AsmStringPiece(const std::string &S) : MyKind(String), Str(S) {}
1172458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    AsmStringPiece(unsigned OpNo, char Modifier)
1173458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      : MyKind(Operand), Str(), OperandNo(OpNo) {
1174458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      Str += Modifier;
1175458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    }
11761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1177458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    bool isString() const { return MyKind == String; }
1178458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    bool isOperand() const { return MyKind == Operand; }
11791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1180458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    const std::string &getString() const {
1181458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      assert(isString());
1182458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      return Str;
1183458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    }
1184458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner
1185458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    unsigned getOperandNo() const {
1186458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      assert(isOperand());
1187458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      return OperandNo;
1188458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    }
11891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1190458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    /// getModifier - Get the modifier for this operand, if present.  This
1191458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    /// returns '\0' if there was no modifier.
1192458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    char getModifier() const {
1193458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      assert(isOperand());
1194458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      return Str[0];
1195458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    }
1196458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  };
11971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1198458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// AnalyzeAsmString - Analyze the asm string of the current asm, decomposing
1199458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// it into pieces.  If the asm string is erroneous, emit errors and return
1200458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// true, otherwise return false.  This handles canonicalization and
1201458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// translation of strings from GCC syntax to LLVM IR syntax, and handles
12021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  //// flattening of named references like %[foo] to Operand AsmStringPiece's.
1203fb5058ef67c054296c88db18ab1b3717845cb71dChris Lattner  unsigned AnalyzeAsmString(llvm::SmallVectorImpl<AsmStringPiece> &Pieces,
1204fb5058ef67c054296c88db18ab1b3717845cb71dChris Lattner                            ASTContext &C, unsigned &DiagOffs) const;
12051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1207b327793860737d1c103a73aeda8057dd628a101dChris Lattner  //===--- Output operands ---===//
1208b327793860737d1c103a73aeda8057dd628a101dChris Lattner
1209b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  unsigned getNumOutputs() const { return NumOutputs; }
1210ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek
1211ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  const std::string &getOutputName(unsigned i) const {
1212ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    return Names[i];
1213ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
1214ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek
1215b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// getOutputConstraint - Return the constraint string for the specified
1216b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// output operand.  All output constraints are known to be non-empty (either
1217b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// '=' or '+').
1218acb6bcb442936f3c14126fdd473307c17647cb84Anders Carlsson  llvm::StringRef getOutputConstraint(unsigned i) const;
12191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1220b327793860737d1c103a73aeda8057dd628a101dChris Lattner  const StringLiteral *getOutputConstraintLiteral(unsigned i) const {
1221ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    return Constraints[i];
1222ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
1223b327793860737d1c103a73aeda8057dd628a101dChris Lattner  StringLiteral *getOutputConstraintLiteral(unsigned i) {
1224b327793860737d1c103a73aeda8057dd628a101dChris Lattner    return Constraints[i];
1225b327793860737d1c103a73aeda8057dd628a101dChris Lattner  }
12261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1228ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  Expr *getOutputExpr(unsigned i);
12291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1230b327793860737d1c103a73aeda8057dd628a101dChris Lattner  const Expr *getOutputExpr(unsigned i) const {
1231b327793860737d1c103a73aeda8057dd628a101dChris Lattner    return const_cast<AsmStmt*>(this)->getOutputExpr(i);
1232b327793860737d1c103a73aeda8057dd628a101dChris Lattner  }
12331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1234b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// isOutputPlusConstraint - Return true if the specified output constraint
1235b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// is a "+" constraint (which is both an input and an output) or false if it
1236b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// is an "=" constraint (just an output).
1237b327793860737d1c103a73aeda8057dd628a101dChris Lattner  bool isOutputPlusConstraint(unsigned i) const {
1238b327793860737d1c103a73aeda8057dd628a101dChris Lattner    return getOutputConstraint(i)[0] == '+';
1239b327793860737d1c103a73aeda8057dd628a101dChris Lattner  }
12401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
124185759278332404e96d4bb89d0e976e46158cd026Chris Lattner  /// getNumPlusOperands - Return the number of output operands that have a "+"
124285759278332404e96d4bb89d0e976e46158cd026Chris Lattner  /// constraint.
124385759278332404e96d4bb89d0e976e46158cd026Chris Lattner  unsigned getNumPlusOperands() const;
12441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1245b327793860737d1c103a73aeda8057dd628a101dChris Lattner  //===--- Input operands ---===//
12461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  unsigned getNumInputs() const { return NumInputs; }
12481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1249ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  const std::string &getInputName(unsigned i) const {
1250ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    return Names[i + NumOutputs];
1251ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
12521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1253b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// getInputConstraint - Return the specified input constraint.  Unlike output
1254b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// constraints, these can be empty.
1255acb6bcb442936f3c14126fdd473307c17647cb84Anders Carlsson  llvm::StringRef getInputConstraint(unsigned i) const;
12561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1257b327793860737d1c103a73aeda8057dd628a101dChris Lattner  const StringLiteral *getInputConstraintLiteral(unsigned i) const {
1258ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    return Constraints[i + NumOutputs];
1259ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
1260b327793860737d1c103a73aeda8057dd628a101dChris Lattner  StringLiteral *getInputConstraintLiteral(unsigned i) {
1261ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    return Constraints[i + NumOutputs];
1262ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
12631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1264ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  Expr *getInputExpr(unsigned i);
12651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1266b327793860737d1c103a73aeda8057dd628a101dChris Lattner  const Expr *getInputExpr(unsigned i) const {
1267b327793860737d1c103a73aeda8057dd628a101dChris Lattner    return const_cast<AsmStmt*>(this)->getInputExpr(i);
1268b327793860737d1c103a73aeda8057dd628a101dChris Lattner  }
1269cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
1270acb6bcb442936f3c14126fdd473307c17647cb84Anders Carlsson  void setOutputsAndInputsAndClobbers(ASTContext &C,
1271acb6bcb442936f3c14126fdd473307c17647cb84Anders Carlsson                                      const std::string *Names,
1272fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      StringLiteral **Constraints,
1273fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      Stmt **Exprs,
1274fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      unsigned NumOutputs,
1275fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      unsigned NumInputs,
1276fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      StringLiteral **Clobbers,
1277fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      unsigned NumClobbers);
1278cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
127910ca96ae9aed6906c3302403ef1a146a8d4c6b74Chris Lattner  //===--- Other ---===//
12801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
128110ca96ae9aed6906c3302403ef1a146a8d4c6b74Chris Lattner  /// getNamedOperand - Given a symbolic operand reference like %[foo],
128210ca96ae9aed6906c3302403ef1a146a8d4c6b74Chris Lattner  /// translate this into a numeric value needed to reference the same operand.
128310ca96ae9aed6906c3302403ef1a146a8d4c6b74Chris Lattner  /// This returns -1 if the operand name is invalid.
1284acb6bcb442936f3c14126fdd473307c17647cb84Anders Carlsson  int getNamedOperand(llvm::StringRef SymbolicName) const;
128510ca96ae9aed6906c3302403ef1a146a8d4c6b74Chris Lattner
1286b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  unsigned getNumClobbers() const { return Clobbers.size(); }
1287b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  StringLiteral *getClobber(unsigned i) { return Clobbers[i]; }
1288d04c6e23f2e10eeb9936778d67f4a1c4a14cc4f6Anders Carlsson  const StringLiteral *getClobber(unsigned i) const { return Clobbers[i]; }
1289cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
1290fe795956194141c91ae555985c9b930595bff43fChris Lattner  virtual SourceRange getSourceRange() const {
1291fe795956194141c91ae555985c9b930595bff43fChris Lattner    return SourceRange(AsmLoc, RParenLoc);
1292fe795956194141c91ae555985c9b930595bff43fChris Lattner  }
12931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1294fe795956194141c91ae555985c9b930595bff43fChris Lattner  static bool classof(const Stmt *T) {return T->getStmtClass() == AsmStmtClass;}
1295fe795956194141c91ae555985c9b930595bff43fChris Lattner  static bool classof(const AsmStmt *) { return true; }
12961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1297ec2f7dccb1f30ae137f74e764e44c2332b0a2ec0Ted Kremenek  // Input expr iterators.
12981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1299ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  typedef ExprIterator inputs_iterator;
1300ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  typedef ConstExprIterator const_inputs_iterator;
13011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1302ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  inputs_iterator begin_inputs() {
130315dffd863b4d8f6f7b04b5c80566ea3531a14f41Ted Kremenek    return Exprs.data() + NumOutputs;
1304ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
13051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1306ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  inputs_iterator end_inputs() {
130715dffd863b4d8f6f7b04b5c80566ea3531a14f41Ted Kremenek    return Exprs.data() + NumOutputs + NumInputs;
1308ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
13091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1310ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  const_inputs_iterator begin_inputs() const {
131115dffd863b4d8f6f7b04b5c80566ea3531a14f41Ted Kremenek    return Exprs.data() + NumOutputs;
1312ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
13131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1314ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  const_inputs_iterator end_inputs() const {
131515dffd863b4d8f6f7b04b5c80566ea3531a14f41Ted Kremenek    return Exprs.data() + NumOutputs + NumInputs;
1316d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  }
13171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1318ec2f7dccb1f30ae137f74e764e44c2332b0a2ec0Ted Kremenek  // Output expr iterators.
13191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1320ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  typedef ExprIterator outputs_iterator;
1321ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  typedef ConstExprIterator const_outputs_iterator;
13221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1323d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  outputs_iterator begin_outputs() {
132415dffd863b4d8f6f7b04b5c80566ea3531a14f41Ted Kremenek    return Exprs.data();
1325d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  }
1326d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  outputs_iterator end_outputs() {
132715dffd863b4d8f6f7b04b5c80566ea3531a14f41Ted Kremenek    return Exprs.data() + NumOutputs;
1328d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  }
13291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1330d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  const_outputs_iterator begin_outputs() const {
133115dffd863b4d8f6f7b04b5c80566ea3531a14f41Ted Kremenek    return Exprs.data();
1332d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  }
1333d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  const_outputs_iterator end_outputs() const {
133415dffd863b4d8f6f7b04b5c80566ea3531a14f41Ted Kremenek    return Exprs.data() + NumOutputs;
1335d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  }
13361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
13371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Child iterators
13381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1339fe795956194141c91ae555985c9b930595bff43fChris Lattner  virtual child_iterator child_begin();
1340fe795956194141c91ae555985c9b930595bff43fChris Lattner  virtual child_iterator child_end();
1341fe795956194141c91ae555985c9b930595bff43fChris Lattner};
134278492dae00fb85e0da0f966df4745edafdafb66cTed Kremenek
13435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}  // end namespace clang
13445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
13455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#endif
1346