Stmt.h revision 4ba2a17694148e16eaa8d3917f657ffcd3667be4
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"
235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "llvm/ADT/SmallVector.h"
248189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek#include "clang/AST/ASTContext.h"
256e340496341a4704be0ede9c1ff4f8eacea7ee2cChris Lattner#include <string>
267573098b83e780d1c5bea13b384b610d8f155676Steve Naroffusing llvm::dyn_cast_or_null;
277573098b83e780d1c5bea13b384b610d8f155676Steve Naroff
2841ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregornamespace llvm {
2941ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  class FoldingSetNodeID;
3041ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor}
3141ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor
325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencernamespace clang {
33e2563ca02a519c2ad6d64dfed87d6e86c5d3c072Sam Bishop  class ASTContext;
345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  class Expr;
355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  class Decl;
367ba138abd329e591a8f6d5001f60dd7082f71b3bSteve Naroff  class ParmVarDecl;
374b07b2968f87f3cd5a3d8c76145f1cbfd718d42dSebastian Redl  class QualType;
385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  class IdentifierInfo;
390c727a35718556866a978f64ac549d9798735f08Chris Lattner  class SourceManager;
406a0ef4b83c91a6d6d5acb4ed5577c4659fe022a3Anders Carlsson  class StringLiteral;
416c36be5b383875b490684bcf439d6d427298c1afChris Lattner  class SwitchStmt;
421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
43ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  //===----------------------------------------------------------------------===//
44ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  // ExprIterator - Iterators for iterating over Stmt* arrays that contain
45ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  //  only Expr*.  This is needed because AST nodes use Stmt* arrays to store
46ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  //  references to children (to be compatible with StmtIterator).
47ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  //===----------------------------------------------------------------------===//
481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
49ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  class Stmt;
50ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  class Expr;
511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
52ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  class ExprIterator {
53ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    Stmt** I;
54ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  public:
55ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ExprIterator(Stmt** i) : I(i) {}
561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    ExprIterator() : I(0) {}
57ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ExprIterator& operator++() { ++I; return *this; }
58ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ExprIterator operator-(size_t i) { return I-i; }
59ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ExprIterator operator+(size_t i) { return I+i; }
60ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    Expr* operator[](size_t idx);
61ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    // FIXME: Verify that this will correctly return a signed distance.
62ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    signed operator-(const ExprIterator& R) const { return I - R.I; }
63ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    Expr* operator*() const;
64ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    Expr* operator->() const;
65ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    bool operator==(const ExprIterator& R) const { return I == R.I; }
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  };
701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
71ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  class ConstExprIterator {
721705fe9ec0efb65f77a46e669e48302923204fe8Benjamin Kramer    const Stmt * const *I;
73ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  public:
741705fe9ec0efb65f77a46e669e48302923204fe8Benjamin Kramer    ConstExprIterator(const Stmt * const *i) : I(i) {}
751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    ConstExprIterator() : I(0) {}
76ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    ConstExprIterator& operator++() { ++I; return *this; }
771705fe9ec0efb65f77a46e669e48302923204fe8Benjamin Kramer    ConstExprIterator operator+(size_t i) const { return I+i; }
781705fe9ec0efb65f77a46e669e48302923204fe8Benjamin Kramer    ConstExprIterator operator-(size_t i) const { return I-i; }
79ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    const Expr * operator[](size_t idx) const;
80ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    signed operator-(const ConstExprIterator& R) const { return I - R.I; }
81ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    const Expr * operator*() const;
82ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    const Expr * operator->() const;
83ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    bool operator==(const ConstExprIterator& R) const { return I == R.I; }
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; }
871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  };
881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
89ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek//===----------------------------------------------------------------------===//
90ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek// AST classes for statements.
91ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek//===----------------------------------------------------------------------===//
921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// Stmt - This represents one statement.
945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass Stmt {
965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  enum StmtClass {
98f2cad8633e46cce12fc3d77c0bd451ffd7264bafDouglas Gregor    NoStmtClass = 0,
99f2cad8633e46cce12fc3d77c0bd451ffd7264bafDouglas Gregor#define STMT(CLASS, PARENT) CLASS##Class,
1007381d5cfbd599fa2b9e215011ad7cbd449de231aSean Hunt#define STMT_RANGE(BASE, FIRST, LAST) \
1019a55591af3e5506b95a9718e15380129fbfc5ebcSean Hunt        first##BASE##Constant=FIRST##Class, last##BASE##Constant=LAST##Class,
1027381d5cfbd599fa2b9e215011ad7cbd449de231aSean Hunt#define LAST_STMT_RANGE(BASE, FIRST, LAST) \
1039a55591af3e5506b95a9718e15380129fbfc5ebcSean Hunt        first##BASE##Constant=FIRST##Class, last##BASE##Constant=LAST##Class
1047381d5cfbd599fa2b9e215011ad7cbd449de231aSean Hunt#define ABSTRACT_STMT(STMT)
1054bfe1968410ea8ffe3b4f629addd7c4bcf484765Sean Hunt#include "clang/AST/StmtNodes.inc"
1068e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  };
1071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1088189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  // Make vanilla 'new' and 'delete' illegal for Stmts.
1098189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenekprotected:
1108189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  void* operator new(size_t bytes) throw() {
1118189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    assert(0 && "Stmts cannot be allocated with regular 'new'.");
1128189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return 0;
1138189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
1148189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  void operator delete(void* data) throw() {
1158189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    assert(0 && "Stmts cannot be released with regular 'delete'.");
1168189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
1171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1188e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  class StmtBitfields {
1198e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    friend class Stmt;
1208e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
1218e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    /// \brief The statement class.
1228e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned sClass : 8;
1238e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  };
1248e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  enum { NumStmtBits = 8 };
1258e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
1268e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  class CompoundStmtBitfields {
1278e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    friend class CompoundStmt;
1288e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned : NumStmtBits;
1298e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
1308e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned NumStmts : 32 - NumStmtBits;
1318e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  };
1328e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
1338e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  class LabelStmtBitfields {
1348e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    friend class LabelStmt;
1358e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned : NumStmtBits;
1368e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
1378e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned Used : 1;
1388e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned HasUnusedAttr : 1;
1398e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  };
1408e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
1418e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  class ExprBitfields {
1428e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    friend class Expr;
1438e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    friend class DeclRefExpr; // computeDependence
1448e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    friend class InitListExpr; // ctor
1458e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    friend class DesignatedInitExpr; // ctor
146bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class ASTStmtReader; // deserialization
147bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class CXXNewExpr; // ctor
148bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class DependentScopeDeclRefExpr; // ctor
149bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class CXXConstructExpr; // ctor
150bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class CallExpr; // ctor
151bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class OffsetOfExpr; // ctor
152bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class ObjCMessageExpr; // ctor
153bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class ShuffleVectorExpr; // ctor
154bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class ParenListExpr; // ctor
155bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class CXXUnresolvedConstructExpr; // ctor
156bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class CXXDependentScopeMemberExpr; // ctor
157bebbe0d9b7568ce43a464286bee49429489ef483Douglas Gregor    friend class OverloadExpr; // ctor
1588e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned : NumStmtBits;
1598e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
1608e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned ValueKind : 2;
161f89e55ab1bfb3ea997f8b02997c611a02254eb2dJohn McCall    unsigned ObjectKind : 2;
1628e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned TypeDependent : 1;
1638e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned ValueDependent : 1;
164d0937224f383c7cc72c947119380f9713a070c73Douglas Gregor    unsigned ContainsUnexpandedParameterPack : 1;
1658e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  };
166d0937224f383c7cc72c947119380f9713a070c73Douglas Gregor  enum { NumExprBits = 15 };
1678e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
1688e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  class CastExprBitfields {
1698e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    friend class CastExpr;
1708e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    unsigned : NumExprBits;
1718e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
1722bb5d00fcf71a7b4d478d478be778fff0494aff6John McCall    unsigned Kind : 6;
1732bb5d00fcf71a7b4d478d478be778fff0494aff6John McCall    unsigned BasePathSize : 32 - 6 - NumExprBits;
1748e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  };
1758e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
1768e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  union {
1778e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    StmtBitfields StmtBits;
1788e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    CompoundStmtBitfields CompoundStmtBits;
1798e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    LabelStmtBitfields LabelStmtBits;
1808e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    ExprBitfields ExprBits;
1818e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    CastExprBitfields CastExprBits;
1828e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  };
1838e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
184d0937224f383c7cc72c947119380f9713a070c73Douglas Gregor  friend class ASTStmtReader;
185d0937224f383c7cc72c947119380f9713a070c73Douglas Gregor
1868189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenekpublic:
1878189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  // Only allow allocation of Stmts using the allocator in ASTContext
1881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // or by doing a placement new.
1898189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  void* operator new(size_t bytes, ASTContext& C,
190f4e689b8528770001f4792f1f4ebdfb09d859e3dDouglas Gregor                     unsigned alignment = 8) throw() {
1918189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return ::operator new(bytes, C, alignment);
1928189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
1931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1948189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  void* operator new(size_t bytes, ASTContext* C,
195f4e689b8528770001f4792f1f4ebdfb09d859e3dDouglas Gregor                     unsigned alignment = 8) throw() {
1968189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return ::operator new(bytes, *C, alignment);
1978189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
1981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1998189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  void* operator new(size_t bytes, void* mem) throw() {
2008189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return mem;
2018189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
202e2dedf8f61b8f306f704781456b482eb61871e8eDouglas Gregor
203fb523e16dd1f860ff02a3ae03e5e3e25327a5860Chris Lattner  void operator delete(void*, ASTContext&, unsigned) throw() { }
204fb523e16dd1f860ff02a3ae03e5e3e25327a5860Chris Lattner  void operator delete(void*, ASTContext*, unsigned) throw() { }
205e2dedf8f61b8f306f704781456b482eb61871e8eDouglas Gregor  void operator delete(void*, std::size_t) throw() { }
206fb523e16dd1f860ff02a3ae03e5e3e25327a5860Chris Lattner  void operator delete(void*, void*) throw() { }
207e2dedf8f61b8f306f704781456b482eb61871e8eDouglas Gregor
208025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregorpublic:
209025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief A placeholder type used to construct an empty shell of a
210025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// type, that will be filled in later (e.g., by some
211025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// de-serialization).
212025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  struct EmptyShell { };
213025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
2149c1863ef36a74e8203f00289d19856ad956f48b9Ted Kremenekprotected:
215025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Construct an empty statement.
2168e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  explicit Stmt(StmtClass SC, EmptyShell) {
2178e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    StmtBits.sClass = SC;
218025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor    if (Stmt::CollectingStats()) Stmt::addStmtClass(SC);
219025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  }
220025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
2215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
2228e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  Stmt(StmtClass SC) {
2238e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    StmtBits.sClass = SC;
2245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (Stmt::CollectingStats()) Stmt::addStmtClass(SC);
2255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
2265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  virtual ~Stmt() {}
2271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
228c302113179a1c2b1254224ea9b6f5316ceeb375cSean Hunt  StmtClass getStmtClass() const {
2298e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    return static_cast<StmtClass>(StmtBits.sClass);
230b2f81cf7f8445e0c65c0428f4fbb0442566916b8Douglas Gregor  }
2315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  const char *getStmtClassName() const;
2321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
233b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  /// SourceLocation tokens are not useful in isolation - they are low level
234b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  /// value objects created/interpreted by SourceManager. We assume AST
235b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  /// clients will have a pointer to the respective SourceManager.
236b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  virtual SourceRange getSourceRange() const = 0;
237311ff02fae0392bee6abe7723cdf5a69b2899a47Chris Lattner  SourceLocation getLocStart() const { return getSourceRange().getBegin(); }
238311ff02fae0392bee6abe7723cdf5a69b2899a47Chris Lattner  SourceLocation getLocEnd() const { return getSourceRange().getEnd(); }
2395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
2405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // global temp stats (until we have a per-module visitor)
2415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static void addStmtClass(const StmtClass s);
2422024f4d4b0d57616f79ea742fa782d633d414462Kovarththanan Rajaratnam  static bool CollectingStats(bool Enable = false);
2435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static void PrintStats();
2446000dace22f110d8768476989313e9d981d690d0Chris Lattner
2456000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// dump - This does a local dump of the specified AST fragment.  It dumps the
2466000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// specified node and a few nodes underneath it, but not the whole subtree.
2476000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// This is useful in a debugger.
2485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void dump() const;
249e300c870f08d08badf2ebcb53ded49f304af37fcChris Lattner  void dump(SourceManager &SM) const;
2509668033ee4c25efd019e6c7e6dd96aa2e6364a46Argyrios Kyrtzidis  void dump(llvm::raw_ostream &OS, SourceManager &SM) const;
2516000dace22f110d8768476989313e9d981d690d0Chris Lattner
2526000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// dumpAll - This does a dump of the specified AST fragment and all subtrees.
2536000dace22f110d8768476989313e9d981d690d0Chris Lattner  void dumpAll() const;
254e300c870f08d08badf2ebcb53ded49f304af37fcChris Lattner  void dumpAll(SourceManager &SM) const;
2556000dace22f110d8768476989313e9d981d690d0Chris Lattner
2566000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// dumpPretty/printPretty - These two methods do a "pretty print" of the AST
2576000dace22f110d8768476989313e9d981d690d0Chris Lattner  /// back to its original source language syntax.
25848d14a222276fad5279e994d1a062f36ae6fcbceEli Friedman  void dumpPretty(ASTContext& Context) const;
2591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  void printPretty(llvm::raw_ostream &OS, PrinterHelper *Helper,
260e4f2142d00fa5fdb580c4e2413da91882d955381Chris Lattner                   const PrintingPolicy &Policy,
26148d14a222276fad5279e994d1a062f36ae6fcbceEli Friedman                   unsigned Indentation = 0) const {
26248d14a222276fad5279e994d1a062f36ae6fcbceEli Friedman    printPretty(OS, *(ASTContext*)0, Helper, Policy, Indentation);
26348d14a222276fad5279e994d1a062f36ae6fcbceEli Friedman  }
264e4f2142d00fa5fdb580c4e2413da91882d955381Chris Lattner  void printPretty(llvm::raw_ostream &OS, ASTContext &Context,
2651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                   PrinterHelper *Helper,
266e4f2142d00fa5fdb580c4e2413da91882d955381Chris Lattner                   const PrintingPolicy &Policy,
267d249e1d1f1498b81314459ceda19d6ff25c278adDouglas Gregor                   unsigned Indentation = 0) const;
2681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
269d2a4a1af1088fca80e2dc76eb3369db0fbbfdefdTed Kremenek  /// viewAST - Visualize an AST rooted at this Stmt* using GraphViz.  Only
270d2a4a1af1088fca80e2dc76eb3369db0fbbfdefdTed Kremenek  ///   works on systems with GraphViz (Mac OS X) or dot+gv installed.
271d2a4a1af1088fca80e2dc76eb3369db0fbbfdefdTed Kremenek  void viewAST() const;
2721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Implement isa<T> support.
2741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *) { return true; }
2751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
276d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  /// hasImplicitControlFlow - Some statements (e.g. short circuited operations)
277d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  ///  contain implicit control-flow in the order their subexpressions
278d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  ///  are evaluated.  This predicate returns true if this statement has
279d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  ///  such implicit control-flow.  Such statements are also specially handled
280d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  ///  within CFGs.
281d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek  bool hasImplicitControlFlow() const;
282d48ade633d96b94cb435d73e2c935ea457152decTed Kremenek
28377ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  /// Child Iterators: All subclasses must implement child_begin and child_end
28477ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  ///  to permit easy iteration over the substatements/subexpessions of an
28577ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  ///  AST node.  This permits easy iteration over all nodes in the AST.
2869caf8b1ca6beb254f420dada3c0e94d5ef027f58Ted Kremenek  typedef StmtIterator       child_iterator;
2879caf8b1ca6beb254f420dada3c0e94d5ef027f58Ted Kremenek  typedef ConstStmtIterator  const_child_iterator;
2881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2898297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin() = 0;
2908297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end()   = 0;
2911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
29277ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  const_child_iterator child_begin() const {
2939caf8b1ca6beb254f420dada3c0e94d5ef027f58Ted Kremenek    return const_child_iterator(const_cast<Stmt*>(this)->child_begin());
29477ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  }
2951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
29677ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  const_child_iterator child_end() const {
2979caf8b1ca6beb254f420dada3c0e94d5ef027f58Ted Kremenek    return const_child_iterator(const_cast<Stmt*>(this)->child_end());
29877ed8e4edf6ed78c53fb20ec3210aff2a59c9d87Ted Kremenek  }
29941ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor
30041ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// \brief Produce a unique representation of the given statement.
30141ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  ///
30241ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// \brief ID once the profiling operation is complete, will contain
30341ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// the unique representation of the given statement.
30441ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  ///
30541ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// \brief Context the AST context in which the statement resides
30641ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  ///
30741ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// \brief Canonical whether the profile should be based on the canonical
30841ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// representation of this statement (e.g., where non-type template
3091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  /// parameters are identified by index/level rather than their
31041ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// declaration pointers) or the exact representation of the statement as
31141ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor  /// written in the source.
3124ba2a17694148e16eaa8d3917f657ffcd3667be4Jay Foad  void Profile(llvm::FoldingSetNodeID &ID, const ASTContext &Context,
31341ef0c3472a3d09c29bc1792f3d26842f2b8a695Douglas Gregor               bool Canonical);
3145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
3155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
3165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// DeclStmt - Adaptor class for mixing declarations with statements and
3175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// expressions. For example, CompoundStmt mixes statements, expressions
3181eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// and declarations (variables, types). Another example is ForStmt, where
3195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// the first statement can be an expression or a declaration.
3205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
3215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass DeclStmt : public Stmt {
3229653db7bd0e3665b955a0445859285f2e1e7dacdDouglas Gregor  DeclGroupRef DG;
32381c018d9482e7cc2addadc6202dcf162a01faefdChris Lattner  SourceLocation StartLoc, EndLoc;
3241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
3261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  DeclStmt(DeclGroupRef dg, SourceLocation startLoc,
3278ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek           SourceLocation endLoc) : Stmt(DeclStmtClass), DG(dg),
3288ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek                                    StartLoc(startLoc), EndLoc(endLoc) {}
3291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
33084f2170062014d268951902164bed0d8bdea0e82Douglas Gregor  /// \brief Build an empty declaration statement.
33184f2170062014d268951902164bed0d8bdea0e82Douglas Gregor  explicit DeclStmt(EmptyShell Empty) : Stmt(DeclStmtClass, Empty) { }
33284f2170062014d268951902164bed0d8bdea0e82Douglas Gregor
3337e24e82a70a2c681f4291a3397bcd1e1005f251aChris Lattner  /// isSingleDecl - This method returns true if this DeclStmt refers
334fe95deaf66e4fbd82d44b5f6afa8162fa69cb85cChris Lattner  /// to a single Decl.
3357e24e82a70a2c681f4291a3397bcd1e1005f251aChris Lattner  bool isSingleDecl() const {
336fe95deaf66e4fbd82d44b5f6afa8162fa69cb85cChris Lattner    return DG.isSingleDecl();
3378ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  }
3381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3397e24e82a70a2c681f4291a3397bcd1e1005f251aChris Lattner  const Decl *getSingleDecl() const { return DG.getSingleDecl(); }
3401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Decl *getSingleDecl() { return DG.getSingleDecl(); }
3411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
342e66a8cf9117e5fb95a05ff76ec06615e63dd5adeChris Lattner  const DeclGroupRef getDeclGroup() const { return DG; }
343e66a8cf9117e5fb95a05ff76ec06615e63dd5adeChris Lattner  DeclGroupRef getDeclGroup() { return DG; }
34484f2170062014d268951902164bed0d8bdea0e82Douglas Gregor  void setDeclGroup(DeclGroupRef DGR) { DG = DGR; }
345b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
34681c018d9482e7cc2addadc6202dcf162a01faefdChris Lattner  SourceLocation getStartLoc() const { return StartLoc; }
34784f2170062014d268951902164bed0d8bdea0e82Douglas Gregor  void setStartLoc(SourceLocation L) { StartLoc = L; }
34881c018d9482e7cc2addadc6202dcf162a01faefdChris Lattner  SourceLocation getEndLoc() const { return EndLoc; }
34984f2170062014d268951902164bed0d8bdea0e82Douglas Gregor  void setEndLoc(SourceLocation L) { EndLoc = L; }
35084f2170062014d268951902164bed0d8bdea0e82Douglas Gregor
3518ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  SourceRange getSourceRange() const {
35281c018d9482e7cc2addadc6202dcf162a01faefdChris Lattner    return SourceRange(StartLoc, EndLoc);
35381c018d9482e7cc2addadc6202dcf162a01faefdChris Lattner  }
3541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
3561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == DeclStmtClass;
3575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
3585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const DeclStmt *) { return true; }
3591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
36014f8b4ff660bcaa763974b8d0fae81857c594495Ted Kremenek  // Iterators over subexpressions.
3618297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
3628297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
3631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3640632dd6fe068011af5710c0d6a745724021ff620Chris Lattner  typedef DeclGroupRef::iterator decl_iterator;
3650632dd6fe068011af5710c0d6a745724021ff620Chris Lattner  typedef DeclGroupRef::const_iterator const_decl_iterator;
3661eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3678ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  decl_iterator decl_begin() { return DG.begin(); }
3688ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  decl_iterator decl_end() { return DG.end(); }
3698ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  const_decl_iterator decl_begin() const { return DG.begin(); }
3708ffb159441e923322bef6b5dee1aaf24c738d75eTed Kremenek  const_decl_iterator decl_end() const { return DG.end(); }
3715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
3725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
3735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// NullStmt - This is the null statement ";": C99 6.8.3p3.
3745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
3755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass NullStmt : public Stmt {
3765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  SourceLocation SemiLoc;
37744aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis
37844aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis  /// \brief Whether the null statement was preceded by an empty macro, e.g:
37944aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis  /// @code
38044aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis  ///   #define CALL(x)
38144aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis  ///   CALL(0);
38244aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis  /// @endcode
38344aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis  bool LeadingEmptyMacro;
3845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
38544aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis  NullStmt(SourceLocation L, bool LeadingEmptyMacro = false)
38644aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis    : Stmt(NullStmtClass), SemiLoc(L), LeadingEmptyMacro(LeadingEmptyMacro) {}
3875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
388025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build an empty null statement.
389025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit NullStmt(EmptyShell Empty) : Stmt(NullStmtClass, Empty) { }
390025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
3915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  SourceLocation getSemiLoc() const { return SemiLoc; }
392025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setSemiLoc(SourceLocation L) { SemiLoc = L; }
393b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
39444aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis  bool hasLeadingEmptyMacro() const { return LeadingEmptyMacro; }
39544aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis
396507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  virtual SourceRange getSourceRange() const { return SourceRange(SemiLoc); }
3971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
3991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == NullStmtClass;
4005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
4015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const NullStmt *) { return true; }
4021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4038297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
4048297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
4058297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
40644aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis
40744aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis  friend class ASTStmtReader;
40844aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis  friend class ASTStmtWriter;
4095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
4105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
4115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// CompoundStmt - This represents a group of statements like { stmt stmt }.
4125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
4135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass CompoundStmt : public Stmt {
4148189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  Stmt** Body;
415b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation LBracLoc, RBracLoc;
4165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
4178e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  CompoundStmt(ASTContext& C, Stmt **StmtStart, unsigned NumStmts,
4188e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall               SourceLocation LB, SourceLocation RB)
4198e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  : Stmt(CompoundStmtClass), LBracLoc(LB), RBracLoc(RB) {
4208e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    CompoundStmtBits.NumStmts = NumStmts;
4218e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall
422026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner    if (NumStmts == 0) {
423026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner      Body = 0;
424026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner      return;
425026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner    }
4261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
427026dc96ac6ece60da5e1b98e2a71bd0ff0939fd8Chris Lattner    Body = new (C) Stmt*[NumStmts];
4288e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    memcpy(Body, StmtStart, NumStmts * sizeof(*Body));
4291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  }
430025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
431025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  // \brief Build an empty compound statement.
432025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit CompoundStmt(EmptyShell Empty)
4338e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    : Stmt(CompoundStmtClass, Empty), Body(0) {
4348e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    CompoundStmtBits.NumStmts = 0;
4358e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  }
436025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
437025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setStmts(ASTContext &C, Stmt **Stmts, unsigned NumStmts);
4381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4398e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  bool body_empty() const { return CompoundStmtBits.NumStmts == 0; }
4408e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  unsigned size() const { return CompoundStmtBits.NumStmts; }
441025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
4428189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  typedef Stmt** body_iterator;
4438189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  body_iterator body_begin() { return Body; }
4448e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  body_iterator body_end() { return Body + size(); }
4458e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  Stmt *body_back() { return !body_empty() ? Body[size()-1] : 0; }
446e946fc833d8592aa2890bfd9839f1ad839b3d284Fariborz Jahanian
4478e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  void setLastStmt(Stmt *S) {
4488e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    assert(!body_empty() && "setLastStmt");
4498e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    Body[size()-1] = S;
4508e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  }
4515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
4528189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  typedef Stmt* const * const_body_iterator;
4538189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  const_body_iterator body_begin() const { return Body; }
4548e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  const_body_iterator body_end() const { return Body + size(); }
4558e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  const Stmt *body_back() const { return !body_empty() ? Body[size()-1] : 0; }
4568189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek
4578189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  typedef std::reverse_iterator<body_iterator> reverse_body_iterator;
4588189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  reverse_body_iterator body_rbegin() {
4598189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return reverse_body_iterator(body_end());
4608189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
4618189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  reverse_body_iterator body_rend() {
4628189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return reverse_body_iterator(body_begin());
4638189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
4644ce854736dd196e2304f554ebeac8b43c89cf9e2Ted Kremenek
4658189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  typedef std::reverse_iterator<const_body_iterator>
4668189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek          const_reverse_body_iterator;
4674ce854736dd196e2304f554ebeac8b43c89cf9e2Ted Kremenek
4688189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  const_reverse_body_iterator body_rbegin() const {
4698189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return const_reverse_body_iterator(body_end());
4708189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
4711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4728189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  const_reverse_body_iterator body_rend() const {
4738189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek    return const_reverse_body_iterator(body_begin());
4748189cde56b4f6f938cd65f53c932fe1860d0204cTed Kremenek  }
4751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
4771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(LBracLoc, RBracLoc);
478b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
4791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
480cc326204dd97771c336b9aab3b9963ea30d69c29Ted Kremenek  SourceLocation getLBracLoc() const { return LBracLoc; }
481025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setLBracLoc(SourceLocation L) { LBracLoc = L; }
482cc326204dd97771c336b9aab3b9963ea30d69c29Ted Kremenek  SourceLocation getRBracLoc() const { return RBracLoc; }
483025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setRBracLoc(SourceLocation L) { RBracLoc = L; }
4841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
4861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == CompoundStmtClass;
4875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
4885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const CompoundStmt *) { return true; }
4891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4908297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
4918297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
4928297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
4935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
4945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
495c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson// SwitchCase is the base class for CaseStmt and DefaultStmt,
496c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlssonclass SwitchCase : public Stmt {
497103fc81f12aa635aa0a573c94b1aceb496b4e587Ted Kremenekprotected:
498c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  // A pointer to the following CaseStmt or DefaultStmt class,
499c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  // used by SwitchStmt.
500c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  SwitchCase *NextSwitchCase;
501103fc81f12aa635aa0a573c94b1aceb496b4e587Ted Kremenek
502d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  SwitchCase(StmtClass SC) : Stmt(SC), NextSwitchCase(0) {}
5031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
504c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlssonpublic:
505c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  const SwitchCase *getNextSwitchCase() const { return NextSwitchCase; }
506c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson
507c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  SwitchCase *getNextSwitchCase() { return NextSwitchCase; }
508c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson
509c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  void setNextSwitchCase(SwitchCase *SC) { NextSwitchCase = SC; }
510d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek
511d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Stmt *getSubStmt() { return v_getSubStmt(); }
512b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
513b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  virtual SourceRange getSourceRange() const { return SourceRange(); }
5141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
5161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == CaseStmtClass ||
517c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson    T->getStmtClass() == DefaultStmtClass;
518c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  }
519c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  static bool classof(const SwitchCase *) { return true; }
5203fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattnerprotected:
5211eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual Stmt* v_getSubStmt() = 0;
522c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson};
523c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson
524c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlssonclass CaseStmt : public SwitchCase {
525d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  enum { SUBSTMT, LHS, RHS, END_EXPR };
5261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Stmt* SubExprs[END_EXPR];  // The expression for the RHS is Non-null for
527d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek                             // GNU "case 1 ... 4" extension
528b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation CaseLoc;
529dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation EllipsisLoc;
530dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation ColonLoc;
531dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor
5323fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner  virtual Stmt* v_getSubStmt() { return getSubStmt(); }
5335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
534dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  CaseStmt(Expr *lhs, Expr *rhs, SourceLocation caseLoc,
5351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump           SourceLocation ellipsisLoc, SourceLocation colonLoc)
536d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek    : SwitchCase(CaseStmtClass) {
53724e1e707b4c362f18e371e2bbf054a8345b57bfaChris Lattner    SubExprs[SUBSTMT] = 0;
538d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek    SubExprs[LHS] = reinterpret_cast<Stmt*>(lhs);
539d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek    SubExprs[RHS] = reinterpret_cast<Stmt*>(rhs);
540b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff    CaseLoc = caseLoc;
541dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor    EllipsisLoc = ellipsisLoc;
542dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor    ColonLoc = colonLoc;
543d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  }
544025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
545025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build an empty switch case statement.
546025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit CaseStmt(EmptyShell Empty) : SwitchCase(CaseStmtClass) { }
547025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
548764a7ce5217f9569e100a3445f47496ee82daf86Chris Lattner  SourceLocation getCaseLoc() const { return CaseLoc; }
549025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setCaseLoc(SourceLocation L) { CaseLoc = L; }
550dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation getEllipsisLoc() const { return EllipsisLoc; }
551dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  void setEllipsisLoc(SourceLocation L) { EllipsisLoc = L; }
552dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation getColonLoc() const { return ColonLoc; }
553dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  void setColonLoc(SourceLocation L) { ColonLoc = L; }
554025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
555d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Expr *getLHS() { return reinterpret_cast<Expr*>(SubExprs[LHS]); }
556d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Expr *getRHS() { return reinterpret_cast<Expr*>(SubExprs[RHS]); }
557d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Stmt *getSubStmt() { return SubExprs[SUBSTMT]; }
558025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
5591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const Expr *getLHS() const {
5601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return reinterpret_cast<const Expr*>(SubExprs[LHS]);
56151b09f2c528c8460b5465c676173324e44176d62Devang Patel  }
5621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const Expr *getRHS() const {
5631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return reinterpret_cast<const Expr*>(SubExprs[RHS]);
56451b09f2c528c8460b5465c676173324e44176d62Devang Patel  }
56551b09f2c528c8460b5465c676173324e44176d62Devang Patel  const Stmt *getSubStmt() const { return SubExprs[SUBSTMT]; }
5665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
56720dabe8b8ff7ed6d2d0158fee43755a4bc7642c3Chris Lattner  void setSubStmt(Stmt *S) { SubExprs[SUBSTMT] = S; }
56820dabe8b8ff7ed6d2d0158fee43755a4bc7642c3Chris Lattner  void setLHS(Expr *Val) { SubExprs[LHS] = reinterpret_cast<Stmt*>(Val); }
56920dabe8b8ff7ed6d2d0158fee43755a4bc7642c3Chris Lattner  void setRHS(Expr *Val) { SubExprs[RHS] = reinterpret_cast<Stmt*>(Val); }
5701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5723fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner  virtual SourceRange getSourceRange() const {
5733fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner    // Handle deeply nested case statements with iteration instead of recursion.
5743fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner    const CaseStmt *CS = this;
57591ee0140ecb60b5c1402edc9e577257636c4ca60Chris Lattner    while (const CaseStmt *CS2 = dyn_cast<CaseStmt>(CS->getSubStmt()))
5763fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner      CS = CS2;
5771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(CaseLoc, CS->getSubStmt()->getLocEnd());
579b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
5801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
5811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == CaseStmtClass;
5825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
5835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const CaseStmt *) { return true; }
5841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
585d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  // Iterators
586d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  virtual child_iterator child_begin();
587d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  virtual child_iterator child_end();
5885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
5895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
590c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlssonclass DefaultStmt : public SwitchCase {
591d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Stmt* SubStmt;
5926c36be5b383875b490684bcf439d6d427298c1afChris Lattner  SourceLocation DefaultLoc;
593dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation ColonLoc;
5943fb94a4918bd427fdb12df997dd87fd1017f0388Chris Lattner  virtual Stmt* v_getSubStmt() { return getSubStmt(); }
5955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
5961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  DefaultStmt(SourceLocation DL, SourceLocation CL, Stmt *substmt) :
597dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor    SwitchCase(DefaultStmtClass), SubStmt(substmt), DefaultLoc(DL),
598dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor    ColonLoc(CL) {}
599025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
600025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build an empty default statement.
601025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit DefaultStmt(EmptyShell) : SwitchCase(DefaultStmtClass) { }
602025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
603d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  Stmt *getSubStmt() { return SubStmt; }
60451b09f2c528c8460b5465c676173324e44176d62Devang Patel  const Stmt *getSubStmt() const { return SubStmt; }
605025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setSubStmt(Stmt *S) { SubStmt = S; }
606025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
6076c36be5b383875b490684bcf439d6d427298c1afChris Lattner  SourceLocation getDefaultLoc() const { return DefaultLoc; }
608025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setDefaultLoc(SourceLocation L) { DefaultLoc = L; }
609dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  SourceLocation getColonLoc() const { return ColonLoc; }
610dbb26db1d426fb6caaaf1b4fa47b46d1947c12c9Douglas Gregor  void setColonLoc(SourceLocation L) { ColonLoc = L; }
6115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
6131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(DefaultLoc, SubStmt->getLocEnd());
614b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
6151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
6161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == DefaultStmtClass;
6175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
6185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const DefaultStmt *) { return true; }
6191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
620d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  // Iterators
621d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  virtual child_iterator child_begin();
622d97bb6c1384cb773ba5cdbd198008dec127cebadTed Kremenek  virtual child_iterator child_end();
6235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
6245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass LabelStmt : public Stmt {
6265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  IdentifierInfo *Label;
6275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  Stmt *SubStmt;
628b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation IdentLoc;
6295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
6301a18600b85aaa691122983dd8dcf4225cfc9ef68Argyrios Kyrtzidis  LabelStmt(SourceLocation IL, IdentifierInfo *label, Stmt *substmt,
6311a18600b85aaa691122983dd8dcf4225cfc9ef68Argyrios Kyrtzidis            bool hasUnusedAttr = false)
6328e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    : Stmt(LabelStmtClass), Label(label), SubStmt(substmt), IdentLoc(IL) {
6338e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    LabelStmtBits.Used = false;
6348e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    LabelStmtBits.HasUnusedAttr = hasUnusedAttr;
6358e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  }
6361de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor
6371de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  // \brief Build an empty label statement.
6381de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  explicit LabelStmt(EmptyShell Empty) : Stmt(LabelStmtClass, Empty) { }
6391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  SourceLocation getIdentLoc() const { return IdentLoc; }
6415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  IdentifierInfo *getID() const { return Label; }
6421de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  void setID(IdentifierInfo *II) { Label = II; }
6435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  const char *getName() const;
6445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  Stmt *getSubStmt() { return SubStmt; }
6455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  const Stmt *getSubStmt() const { return SubStmt; }
6465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void setIdentLoc(SourceLocation L) { IdentLoc = L; }
6475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void setSubStmt(Stmt *SS) { SubStmt = SS; }
648b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
649355a9fe26a6dec89680ddf713dd5bc7a671b298aArgyrios Kyrtzidis  /// \brief Whether this label was used.
6501a18600b85aaa691122983dd8dcf4225cfc9ef68Argyrios Kyrtzidis  bool isUsed(bool CheckUnusedAttr = true) const {
6518e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall    return LabelStmtBits.Used ||
6528e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall           (CheckUnusedAttr && LabelStmtBits.HasUnusedAttr);
6531a18600b85aaa691122983dd8dcf4225cfc9ef68Argyrios Kyrtzidis  }
6548e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  void setUsed(bool U = true) { LabelStmtBits.Used = U; }
655355a9fe26a6dec89680ddf713dd5bc7a671b298aArgyrios Kyrtzidis
6568e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  bool HasUnusedAttribute() const { return LabelStmtBits.HasUnusedAttr; }
6578e6285af719adc6f86d6faa235d22a08eb68ee3aJohn McCall  void setUnusedAttribute(bool U) { LabelStmtBits.HasUnusedAttr = U; }
6581a18600b85aaa691122983dd8dcf4225cfc9ef68Argyrios Kyrtzidis
6591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
6601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(IdentLoc, SubStmt->getLocEnd());
6611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  }
6621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
6631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == LabelStmtClass;
6645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
6655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const LabelStmt *) { return true; }
6661eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6678297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
6688297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
6698297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
6705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
6715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// IfStmt - This represents an if/then/else.
6745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
6755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass IfStmt : public Stmt {
67643dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  enum { VAR, COND, THEN, ELSE, END_EXPR };
6778297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt* SubExprs[END_EXPR];
6788cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor
679b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation IfLoc;
680d06f6ca61062f85926eb9d409eb3d4f8afcf93c7Douglas Gregor  SourceLocation ElseLoc;
68144aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis
6825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
68343dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  IfStmt(ASTContext &C, SourceLocation IL, VarDecl *var, Expr *cond,
68444aa1f397855f130e88e62ffc1029f7f83bb5d2eArgyrios Kyrtzidis         Stmt *then, SourceLocation EL = SourceLocation(), Stmt *elsev = 0);
68543dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor
686025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build an empty if/then/else statement
687025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit IfStmt(EmptyShell Empty) : Stmt(IfStmtClass, Empty) { }
688025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
6898cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// \brief Retrieve the variable declared in this "if" statement, if any.
6908cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  ///
6918cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// In the following example, "x" is the condition variable.
6928cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// \code
6938cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// if (int x = foo()) {
6948cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  ///   printf("x is %d", x);
6958cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// }
6968cfe5a784133d90bf329fd20801824a6f71bb8caDouglas Gregor  /// \endcode
69743dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  VarDecl *getConditionVariable() const;
69843dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  void setConditionVariable(ASTContext &C, VarDecl *V);
699c302113179a1c2b1254224ea9b6f5316ceeb375cSean Hunt
7008297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getCond() const { return reinterpret_cast<Expr*>(SubExprs[COND]);}
701025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setCond(Expr *E) { SubExprs[COND] = reinterpret_cast<Stmt *>(E); }
7028297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getThen() const { return SubExprs[THEN]; }
7031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  void setThen(Stmt *S) { SubExprs[THEN] = S; }
7048297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getElse() const { return SubExprs[ELSE]; }
705025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setElse(Stmt *S) { SubExprs[ELSE] = S; }
7065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7078297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getCond() { return reinterpret_cast<Expr*>(SubExprs[COND]); }
7088297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getThen() { return SubExprs[THEN]; }
7098297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getElse() { return SubExprs[ELSE]; }
710b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
711025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  SourceLocation getIfLoc() const { return IfLoc; }
712025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setIfLoc(SourceLocation L) { IfLoc = L; }
713d06f6ca61062f85926eb9d409eb3d4f8afcf93c7Douglas Gregor  SourceLocation getElseLoc() const { return ElseLoc; }
714d06f6ca61062f85926eb9d409eb3d4f8afcf93c7Douglas Gregor  void setElseLoc(SourceLocation L) { ElseLoc = L; }
715025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
7161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
717b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff    if (SubExprs[ELSE])
718b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff      return SourceRange(IfLoc, SubExprs[ELSE]->getLocEnd());
719b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff    else
720b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff      return SourceRange(IfLoc, SubExprs[THEN]->getLocEnd());
721b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
7221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
7241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == IfStmtClass;
7255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
7265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const IfStmt *) { return true; }
7271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
72835628d1f17c817f8c240208db7ba490e3109981bTed Kremenek  // Iterators over subexpressions.  The iterators will include iterating
72935628d1f17c817f8c240208db7ba490e3109981bTed Kremenek  // over the initialization expression referenced by the condition variable.
7308297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
7318297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
7325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
7335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// SwitchStmt - This represents a 'switch' stmt.
7355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
7365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass SwitchStmt : public Stmt {
73743dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  enum { VAR, COND, BODY, END_EXPR };
7381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Stmt* SubExprs[END_EXPR];
739c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  // This points to a linked list of case and default statements.
740c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  SwitchCase *FirstCase;
7419dcbfa450d751bd68fc4af8b75da381d4f6984b9Steve Naroff  SourceLocation SwitchLoc;
7421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
743559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  /// If the SwitchStmt is a switch on an enum value, this records whether
744559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  /// all the enum values were covered by CaseStmts.  This value is meant to
745559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  /// be a hint for possible clients.
746559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  unsigned AllEnumCasesCovered : 1;
747559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek
7485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
74943dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  SwitchStmt(ASTContext &C, VarDecl *Var, Expr *cond);
7501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
751025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build a empty switch statement.
752025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit SwitchStmt(EmptyShell Empty) : Stmt(SwitchStmtClass, Empty) { }
753025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
754d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// \brief Retrieve the variable declared in this "switch" statement, if any.
755d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  ///
756d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// In the following example, "x" is the condition variable.
757d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// \code
758d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// switch (int x = foo()) {
759d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  ///   case 0: break;
760d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  ///   // ...
761d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// }
762d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor  /// \endcode
76343dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  VarDecl *getConditionVariable() const;
76443dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  void setConditionVariable(ASTContext &C, VarDecl *V);
765d3d5301c44138b92bf01286183f5bf310cdd37cfDouglas Gregor
7668297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getCond() const { return reinterpret_cast<Expr*>(SubExprs[COND]);}
7678297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getBody() const { return SubExprs[BODY]; }
768c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  const SwitchCase *getSwitchCaseList() const { return FirstCase; }
769c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson
7708297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getCond() { return reinterpret_cast<Expr*>(SubExprs[COND]);}
771025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setCond(Expr *E) { SubExprs[COND] = reinterpret_cast<Stmt *>(E); }
7728297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getBody() { return SubExprs[BODY]; }
773025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setBody(Stmt *S) { SubExprs[BODY] = S; }
774c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  SwitchCase *getSwitchCaseList() { return FirstCase; }
7751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
77643d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// \brief Set the case list for this switch statement.
77743d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  ///
77843d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// The caller is responsible for incrementing the retain counts on
77943d9d9243329b1b75d1a6efdad9f16d6fb386b8eDouglas Gregor  /// all of the SwitchCase statements in this list.
780025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setSwitchCaseList(SwitchCase *SC) { FirstCase = SC; }
781025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
782025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  SourceLocation getSwitchLoc() const { return SwitchLoc; }
783025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setSwitchLoc(SourceLocation L) { SwitchLoc = L; }
784c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson
7851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  void setBody(Stmt *S, SourceLocation SL) {
7861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    SubExprs[BODY] = S;
7879dcbfa450d751bd68fc4af8b75da381d4f6984b9Steve Naroff    SwitchLoc = SL;
7881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  }
789c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  void addSwitchCase(SwitchCase *SC) {
790c2ee10d79f70036af652a395ac1f8273f3d04e12Douglas Gregor    assert(!SC->getNextSwitchCase() && "case/default already added to a switch");
791c2ee10d79f70036af652a395ac1f8273f3d04e12Douglas Gregor    SC->setNextSwitchCase(FirstCase);
792c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson    FirstCase = SC;
793c1fcb7762673be706b0a40477d5e93411e918f93Anders Carlsson  }
794559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek
795559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  /// Set a flag in the SwitchStmt indicating that if the 'switch (X)' is a
796559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  /// switch over an enum value then all cases have been explicitly covered.
797559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  void setAllEnumCasesCovered() {
798559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek    AllEnumCasesCovered = 1;
799559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  }
800559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek
801559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  /// Returns true if the SwitchStmt is a switch of an enum value and all cases
802559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  /// have been explicitly covered.
803559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  bool isAllEnumCasesCovered() const {
804559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek    return (bool) AllEnumCasesCovered;
805559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek  }
806559fb554602bedb57dbbf3cc14ac8a38264b4547Ted Kremenek
8071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
8081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(SwitchLoc, SubExprs[BODY]->getLocEnd());
8099dcbfa450d751bd68fc4af8b75da381d4f6984b9Steve Naroff  }
8101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
8111eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == SwitchStmtClass;
8125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
8135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const SwitchStmt *) { return true; }
8141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8158297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
8168297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
8178297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
8185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
8195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
8205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
8215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// WhileStmt - This represents a 'while' stmt.
8225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
8235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass WhileStmt : public Stmt {
82443dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  enum { VAR, COND, BODY, END_EXPR };
8258297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt* SubExprs[END_EXPR];
826b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation WhileLoc;
8275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
82843dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  WhileStmt(ASTContext &C, VarDecl *Var, Expr *cond, Stmt *body,
82943dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor            SourceLocation WL);
8301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
831d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  /// \brief Build an empty while statement.
832d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  explicit WhileStmt(EmptyShell Empty) : Stmt(WhileStmtClass, Empty) { }
833d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor
8345656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// \brief Retrieve the variable declared in this "while" statement, if any.
8355656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  ///
8365656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// In the following example, "x" is the condition variable.
8375656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// \code
8385656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// while (int x = random()) {
8395656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  ///   // ...
8405656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// }
8415656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor  /// \endcode
84243dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  VarDecl *getConditionVariable() const;
84343dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  void setConditionVariable(ASTContext &C, VarDecl *V);
8445656e14d91405417182171a705ed3e3d2d6d7aa3Douglas Gregor
8458297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getCond() { return reinterpret_cast<Expr*>(SubExprs[COND]); }
8468297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getCond() const { return reinterpret_cast<Expr*>(SubExprs[COND]);}
847d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  void setCond(Expr *E) { SubExprs[COND] = reinterpret_cast<Stmt*>(E); }
8488297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getBody() { return SubExprs[BODY]; }
8498297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getBody() const { return SubExprs[BODY]; }
850d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  void setBody(Stmt *S) { SubExprs[BODY] = S; }
851d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor
852d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  SourceLocation getWhileLoc() const { return WhileLoc; }
853d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  void setWhileLoc(SourceLocation L) { WhileLoc = L; }
854b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
8551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
8561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(WhileLoc, SubExprs[BODY]->getLocEnd());
857b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
8581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
8591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == WhileStmtClass;
8605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
8615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const WhileStmt *) { return true; }
8621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8638297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
8648297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
8658297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
8665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
8675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
8685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// DoStmt - This represents a 'do/while' stmt.
8695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
8705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass DoStmt : public Stmt {
871c7b5ed6da7410849b51ba9a9ea04d2cc7b720f48Douglas Gregor  enum { BODY, COND, END_EXPR };
8728297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt* SubExprs[END_EXPR];
873b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation DoLoc;
8749f3ca2a7747bd47f14d7693f333103fac29a24d2Douglas Gregor  SourceLocation WhileLoc;
875989135901c750af61ef012b6b0a0368be415bc46Chris Lattner  SourceLocation RParenLoc;  // Location of final ')' in do stmt condition.
8769f3ca2a7747bd47f14d7693f333103fac29a24d2Douglas Gregor
8775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
878989135901c750af61ef012b6b0a0368be415bc46Chris Lattner  DoStmt(Stmt *body, Expr *cond, SourceLocation DL, SourceLocation WL,
879989135901c750af61ef012b6b0a0368be415bc46Chris Lattner         SourceLocation RP)
880989135901c750af61ef012b6b0a0368be415bc46Chris Lattner    : Stmt(DoStmtClass), DoLoc(DL), WhileLoc(WL), RParenLoc(RP) {
8818297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek    SubExprs[COND] = reinterpret_cast<Stmt*>(cond);
8828297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek    SubExprs[BODY] = body;
8831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  }
88467d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor
88567d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  /// \brief Build an empty do-while statement.
88667d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  explicit DoStmt(EmptyShell Empty) : Stmt(DoStmtClass, Empty) { }
8871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8888297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getCond() { return reinterpret_cast<Expr*>(SubExprs[COND]); }
8898297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getCond() const { return reinterpret_cast<Expr*>(SubExprs[COND]);}
89067d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setCond(Expr *E) { SubExprs[COND] = reinterpret_cast<Stmt*>(E); }
8918297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getBody() { return SubExprs[BODY]; }
8921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const Stmt *getBody() const { return SubExprs[BODY]; }
89367d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setBody(Stmt *S) { SubExprs[BODY] = S; }
89467d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor
89567d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  SourceLocation getDoLoc() const { return DoLoc; }
89667d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setDoLoc(SourceLocation L) { DoLoc = L; }
8979f3ca2a7747bd47f14d7693f333103fac29a24d2Douglas Gregor  SourceLocation getWhileLoc() const { return WhileLoc; }
8989f3ca2a7747bd47f14d7693f333103fac29a24d2Douglas Gregor  void setWhileLoc(SourceLocation L) { WhileLoc = L; }
899b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
900989135901c750af61ef012b6b0a0368be415bc46Chris Lattner  SourceLocation getRParenLoc() const { return RParenLoc; }
901989135901c750af61ef012b6b0a0368be415bc46Chris Lattner  void setRParenLoc(SourceLocation L) { RParenLoc = L; }
902989135901c750af61ef012b6b0a0368be415bc46Chris Lattner
9031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
9041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(DoLoc, RParenLoc);
905b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
9061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
9071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == DoStmtClass;
9085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
9095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const DoStmt *) { return true; }
9108297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek
9118297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
9128297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
9138297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
9145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
9155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
9165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
9175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// ForStmt - This represents a 'for (init;cond;inc)' stmt.  Note that any of
9185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// the init/cond/inc parts of the ForStmt will be null if they were not
9195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// specified in the source.
9205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
9215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass ForStmt : public Stmt {
92243dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  enum { INIT, CONDVAR, COND, INC, BODY, END_EXPR };
9238297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt* SubExprs[END_EXPR]; // SubExprs[INIT] is an expression or declstmt.
924b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  SourceLocation ForLoc;
9255831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor  SourceLocation LParenLoc, RParenLoc;
9265831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor
9275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
92843dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  ForStmt(ASTContext &C, Stmt *Init, Expr *Cond, VarDecl *condVar, Expr *Inc,
92943dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor          Stmt *Body, SourceLocation FL, SourceLocation LP, SourceLocation RP);
9301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
93167d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  /// \brief Build an empty for statement.
93267d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  explicit ForStmt(EmptyShell Empty) : Stmt(ForStmtClass, Empty) { }
93367d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor
9348297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getInit() { return SubExprs[INIT]; }
935c302113179a1c2b1254224ea9b6f5316ceeb375cSean Hunt
93699e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// \brief Retrieve the variable declared in this "for" statement, if any.
93799e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  ///
93899e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// In the following example, "y" is the condition variable.
93999e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// \code
94099e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// for (int x = random(); int y = mangle(x); ++x) {
94199e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  ///   // ...
94299e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// }
94399e9b4d172f6877e6ba5ebe75bb8238721f5e01cDouglas Gregor  /// \endcode
94443dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  VarDecl *getConditionVariable() const;
94543dec6bbde2d0a16c35978983761c8b7030c8e18Douglas Gregor  void setConditionVariable(ASTContext &C, VarDecl *V);
946c302113179a1c2b1254224ea9b6f5316ceeb375cSean Hunt
9478297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getCond() { return reinterpret_cast<Expr*>(SubExprs[COND]); }
9488297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Expr *getInc()  { return reinterpret_cast<Expr*>(SubExprs[INC]); }
9498297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  Stmt *getBody() { return SubExprs[BODY]; }
9508297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek
9518297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getInit() const { return SubExprs[INIT]; }
9528297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getCond() const { return reinterpret_cast<Expr*>(SubExprs[COND]);}
9538297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Expr *getInc()  const { return reinterpret_cast<Expr*>(SubExprs[INC]); }
9548297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  const Stmt *getBody() const { return SubExprs[BODY]; }
955b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
95667d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setInit(Stmt *S) { SubExprs[INIT] = S; }
95767d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setCond(Expr *E) { SubExprs[COND] = reinterpret_cast<Stmt*>(E); }
95867d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setInc(Expr *E) { SubExprs[INC] = reinterpret_cast<Stmt*>(E); }
95967d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setBody(Stmt *S) { SubExprs[BODY] = S; }
96067d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor
96167d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  SourceLocation getForLoc() const { return ForLoc; }
96267d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor  void setForLoc(SourceLocation L) { ForLoc = L; }
9635831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor  SourceLocation getLParenLoc() const { return LParenLoc; }
9645831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor  void setLParenLoc(SourceLocation L) { LParenLoc = L; }
9655831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor  SourceLocation getRParenLoc() const { return RParenLoc; }
9665831c6a1efc47e6a19d82fe3dd25b5b8fef6979dDouglas Gregor  void setRParenLoc(SourceLocation L) { RParenLoc = L; }
96767d8249924ef38a5375ff9c92cd21c1854d6ababDouglas Gregor
9681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
9691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(ForLoc, SubExprs[BODY]->getLocEnd());
970b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff  }
9711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
9721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == ForStmtClass;
9735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
9745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const ForStmt *) { return true; }
9751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9768297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
9778297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
9788297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
9795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
9801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// GotoStmt - This represents a direct goto.
9825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
9835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass GotoStmt : public Stmt {
9845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  LabelStmt *Label;
985507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  SourceLocation GotoLoc;
98661f62165220e75694fe333179c78815e2e48d71fTed Kremenek  SourceLocation LabelLoc;
9875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
9881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  GotoStmt(LabelStmt *label, SourceLocation GL, SourceLocation LL)
98961f62165220e75694fe333179c78815e2e48d71fTed Kremenek    : Stmt(GotoStmtClass), Label(label), GotoLoc(GL), LabelLoc(LL) {}
9901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9911de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  /// \brief Build an empty goto statement.
9921de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  explicit GotoStmt(EmptyShell Empty) : Stmt(GotoStmtClass, Empty) { }
9931de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor
9945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  LabelStmt *getLabel() const { return Label; }
9951de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  void setLabel(LabelStmt *S) { Label = S; }
9961de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor
9971de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  SourceLocation getGotoLoc() const { return GotoLoc; }
9981de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  void setGotoLoc(SourceLocation L) { GotoLoc = L; }
9991de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  SourceLocation getLabelLoc() const { return LabelLoc; }
10001de05feeeafe5b215fe7617594a7076a5192a6e2Douglas Gregor  void setLabelLoc(SourceLocation L) { LabelLoc = L; }
1001b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
10021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
10031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(GotoLoc, LabelLoc);
1004507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  }
10051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
10061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == GotoStmtClass;
10075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
10085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const GotoStmt *) { return true; }
10091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10108297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
10118297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
10128297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
10135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
10145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// IndirectGotoStmt - This represents an indirect goto.
10165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
10175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass IndirectGotoStmt : public Stmt {
1018ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner  SourceLocation GotoLoc;
10195f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor  SourceLocation StarLoc;
10201060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek  Stmt *Target;
10215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
10221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  IndirectGotoStmt(SourceLocation gotoLoc, SourceLocation starLoc,
10235f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor                   Expr *target)
10245f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor    : Stmt(IndirectGotoStmtClass), GotoLoc(gotoLoc), StarLoc(starLoc),
10255f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor      Target((Stmt*)target) {}
10267d5c2f241c74e5f8d9ec492e8f9f268e5e9ae41fDouglas Gregor
10277d5c2f241c74e5f8d9ec492e8f9f268e5e9ae41fDouglas Gregor  /// \brief Build an empty indirect goto statement.
10281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  explicit IndirectGotoStmt(EmptyShell Empty)
10297d5c2f241c74e5f8d9ec492e8f9f268e5e9ae41fDouglas Gregor    : Stmt(IndirectGotoStmtClass, Empty) { }
10301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1031ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner  void setGotoLoc(SourceLocation L) { GotoLoc = L; }
1032ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner  SourceLocation getGotoLoc() const { return GotoLoc; }
10335f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor  void setStarLoc(SourceLocation L) { StarLoc = L; }
10345f1b9e689fa5c101512aef99225f2afea1673449Douglas Gregor  SourceLocation getStarLoc() const { return StarLoc; }
10351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
103695c225de9fa3d79f70ef5008c0279580a7d9dcadJohn McCall  Expr *getTarget() { return reinterpret_cast<Expr*>(Target); }
103795c225de9fa3d79f70ef5008c0279580a7d9dcadJohn McCall  const Expr *getTarget() const {return reinterpret_cast<const Expr*>(Target);}
10387d5c2f241c74e5f8d9ec492e8f9f268e5e9ae41fDouglas Gregor  void setTarget(Expr *E) { Target = reinterpret_cast<Stmt*>(E); }
1039b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
104095c225de9fa3d79f70ef5008c0279580a7d9dcadJohn McCall  /// getConstantTarget - Returns the fixed target of this indirect
104195c225de9fa3d79f70ef5008c0279580a7d9dcadJohn McCall  /// goto, if one exists.
104295c225de9fa3d79f70ef5008c0279580a7d9dcadJohn McCall  LabelStmt *getConstantTarget();
104395c225de9fa3d79f70ef5008c0279580a7d9dcadJohn McCall  const LabelStmt *getConstantTarget() const {
104495c225de9fa3d79f70ef5008c0279580a7d9dcadJohn McCall    return const_cast<IndirectGotoStmt*>(this)->getConstantTarget();
104595c225de9fa3d79f70ef5008c0279580a7d9dcadJohn McCall  }
104695c225de9fa3d79f70ef5008c0279580a7d9dcadJohn McCall
1047ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner  virtual SourceRange getSourceRange() const {
1048ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner    return SourceRange(GotoLoc, Target->getLocEnd());
1049ad56d684259f706b7c0ae5ad9c23adb4f2926817Chris Lattner  }
10501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
10521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == IndirectGotoStmtClass;
10535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
10545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const IndirectGotoStmt *) { return true; }
10551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10568297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
10578297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
10588297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
10595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
10605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// ContinueStmt - This represents a continue.
10635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
10645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass ContinueStmt : public Stmt {
1065507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  SourceLocation ContinueLoc;
10665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
1067507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  ContinueStmt(SourceLocation CL) : Stmt(ContinueStmtClass), ContinueLoc(CL) {}
10681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1069d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  /// \brief Build an empty continue statement.
1070d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  explicit ContinueStmt(EmptyShell Empty) : Stmt(ContinueStmtClass, Empty) { }
1071d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor
1072d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  SourceLocation getContinueLoc() const { return ContinueLoc; }
1073d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor  void setContinueLoc(SourceLocation L) { ContinueLoc = L; }
1074d921cf976b4769af8d06d6763a2547dadf7940abDouglas Gregor
10751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual SourceRange getSourceRange() const {
10761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return SourceRange(ContinueLoc);
1077507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  }
1078861ce3178c70cfb0fa50baf685e1ad363538eaa9Douglas Gregor
10791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
10801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == ContinueStmtClass;
10815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
10825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const ContinueStmt *) { return true; }
10831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10848297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
10858297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
10868297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
10875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
10885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// BreakStmt - This represents a break.
10905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
10915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass BreakStmt : public Stmt {
1092507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  SourceLocation BreakLoc;
10935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
1094507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  BreakStmt(SourceLocation BL) : Stmt(BreakStmtClass), BreakLoc(BL) {}
10951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1096025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  /// \brief Build an empty break statement.
1097025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  explicit BreakStmt(EmptyShell Empty) : Stmt(BreakStmtClass, Empty) { }
1098025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
1099025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  SourceLocation getBreakLoc() const { return BreakLoc; }
1100025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor  void setBreakLoc(SourceLocation L) { BreakLoc = L; }
1101025452fa0eda63e150cfaeebe64f0a19c96b3a06Douglas Gregor
1102507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  virtual SourceRange getSourceRange() const { return SourceRange(BreakLoc); }
1103b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
11041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
11051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == BreakStmtClass;
11065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
11075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const BreakStmt *) { return true; }
11081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11098297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
11108297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
11118297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
11125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
11135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
11145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1115c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner/// ReturnStmt - This represents a return, optionally of an expression:
1116c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner///   return;
1117c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner///   return 4;
1118c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner///
1119c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner/// Note that GCC allows return with no argument in a function declared to
1120c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner/// return a value, and it allows returning a value in functions declared to
1121c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner/// return void.  We explicitly model this in the AST, which means you can't
1122c8edf6bd5f76bf50411d037c83d36847edfa68f0Chris Lattner/// depend on the return type of the function and the presence of an argument.
11235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
11245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass ReturnStmt : public Stmt {
11251060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek  Stmt *RetExpr;
1126507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  SourceLocation RetLoc;
11275077c3876beeaed32280af88244e8050078619a8Douglas Gregor  const VarDecl *NRVOCandidate;
11285077c3876beeaed32280af88244e8050078619a8Douglas Gregor
11295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
11305077c3876beeaed32280af88244e8050078619a8Douglas Gregor  ReturnStmt(SourceLocation RL)
113125a0fe2a99aaf2d1bf5bdfdae7ab11e2a5e7622fAbramo Bagnara    : Stmt(ReturnStmtClass), RetExpr(0), RetLoc(RL), NRVOCandidate(0) { }
11325077c3876beeaed32280af88244e8050078619a8Douglas Gregor
11335077c3876beeaed32280af88244e8050078619a8Douglas Gregor  ReturnStmt(SourceLocation RL, Expr *E, const VarDecl *NRVOCandidate)
11345077c3876beeaed32280af88244e8050078619a8Douglas Gregor    : Stmt(ReturnStmtClass), RetExpr((Stmt*) E), RetLoc(RL),
11355077c3876beeaed32280af88244e8050078619a8Douglas Gregor      NRVOCandidate(NRVOCandidate) {}
11360de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor
11370de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor  /// \brief Build an empty return expression.
11380de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor  explicit ReturnStmt(EmptyShell Empty) : Stmt(ReturnStmtClass, Empty) { }
11390de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor
11401060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek  const Expr *getRetValue() const;
11411060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek  Expr *getRetValue();
11420de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor  void setRetValue(Expr *E) { RetExpr = reinterpret_cast<Stmt*>(E); }
11430de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor
11440de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor  SourceLocation getReturnLoc() const { return RetLoc; }
11450de9d8857b715c2f45c987651f4ce06d73330d93Douglas Gregor  void setReturnLoc(SourceLocation L) { RetLoc = L; }
1146b5a69586f1b8855ee4c1f0bb7a8f0ff4fe32ce09Steve Naroff
11475077c3876beeaed32280af88244e8050078619a8Douglas Gregor  /// \brief Retrieve the variable that might be used for the named return
11485077c3876beeaed32280af88244e8050078619a8Douglas Gregor  /// value optimization.
11495077c3876beeaed32280af88244e8050078619a8Douglas Gregor  ///
11505077c3876beeaed32280af88244e8050078619a8Douglas Gregor  /// The optimization itself can only be performed if the variable is
11515077c3876beeaed32280af88244e8050078619a8Douglas Gregor  /// also marked as an NRVO object.
11525077c3876beeaed32280af88244e8050078619a8Douglas Gregor  const VarDecl *getNRVOCandidate() const { return NRVOCandidate; }
11535077c3876beeaed32280af88244e8050078619a8Douglas Gregor  void setNRVOCandidate(const VarDecl *Var) { NRVOCandidate = Var; }
11545077c3876beeaed32280af88244e8050078619a8Douglas Gregor
1155507f2d5811bd7da1a4d9d2f4960f32177dfab9deSteve Naroff  virtual SourceRange getSourceRange() const;
11561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static bool classof(const Stmt *T) {
11581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return T->getStmtClass() == ReturnStmtClass;
11595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
11605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  static bool classof(const ReturnStmt *) { return true; }
11611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11628297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  // Iterators
11638297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_begin();
11648297777fbe19c4d39e8a70c55346474868055fa1Ted Kremenek  virtual child_iterator child_end();
11655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
11665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1167fe795956194141c91ae555985c9b930595bff43fChris Lattner/// AsmStmt - This represents a GNU inline-assembly statement extension.
1168fe795956194141c91ae555985c9b930595bff43fChris Lattner///
1169fe795956194141c91ae555985c9b930595bff43fChris Lattnerclass AsmStmt : public Stmt {
1170fe795956194141c91ae555985c9b930595bff43fChris Lattner  SourceLocation AsmLoc, RParenLoc;
11716a0ef4b83c91a6d6d5acb4ed5577c4659fe022a3Anders Carlsson  StringLiteral *AsmStr;
1172b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson
1173dfab34a696d1dba8622248c31aaf605906cb6109Anders Carlsson  bool IsSimple;
117439c47b56f45437bbc49c9568b7308a400234a730Anders Carlsson  bool IsVolatile;
11753b11fd3b52e7b88233c602407c151d07cb093e75Mike Stump  bool MSAsm;
11761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1177b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  unsigned NumOutputs;
1178b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  unsigned NumInputs;
1179966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson  unsigned NumClobbers;
11801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1181966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson  // FIXME: If we wanted to, we could allocate all of these in one big array.
1182966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson  IdentifierInfo **Names;
1183966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson  StringLiteral **Constraints;
1184966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson  Stmt **Exprs;
1185966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson  StringLiteral **Clobbers;
1186c302113179a1c2b1254224ea9b6f5316ceeb375cSean Hunt
1187fe795956194141c91ae555985c9b930595bff43fChris Lattnerpublic:
1188c302113179a1c2b1254224ea9b6f5316ceeb375cSean Hunt  AsmStmt(ASTContext &C, SourceLocation asmloc, bool issimple, bool isvolatile,
1189966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson          bool msasm, unsigned numoutputs, unsigned numinputs,
1190ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson          IdentifierInfo **names, StringLiteral **constraints,
1191db6ed1786bf460e1143f67d14bf2d71ad9856f81Chris Lattner          Expr **exprs, StringLiteral *asmstr, unsigned numclobbers,
1192db6ed1786bf460e1143f67d14bf2d71ad9856f81Chris Lattner          StringLiteral **clobbers, SourceLocation rparenloc);
1193b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson
1194cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  /// \brief Build an empty inline-assembly statement.
1195c302113179a1c2b1254224ea9b6f5316ceeb375cSean Hunt  explicit AsmStmt(EmptyShell Empty) : Stmt(AsmStmtClass, Empty),
1196966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson    Names(0), Constraints(0), Exprs(0), Clobbers(0) { }
1197cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
1198cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  SourceLocation getAsmLoc() const { return AsmLoc; }
1199cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  void setAsmLoc(SourceLocation L) { AsmLoc = L; }
1200cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  SourceLocation getRParenLoc() const { return RParenLoc; }
1201cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  void setRParenLoc(SourceLocation L) { RParenLoc = L; }
1202cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
120339c47b56f45437bbc49c9568b7308a400234a730Anders Carlsson  bool isVolatile() const { return IsVolatile; }
1204cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  void setVolatile(bool V) { IsVolatile = V; }
1205dfab34a696d1dba8622248c31aaf605906cb6109Anders Carlsson  bool isSimple() const { return IsSimple; }
12066ffe643322949dd776285a6df60d3578f3918be4Chris Lattner  void setSimple(bool V) { IsSimple = V; }
12073b11fd3b52e7b88233c602407c151d07cb093e75Mike Stump  bool isMSAsm() const { return MSAsm; }
12083b11fd3b52e7b88233c602407c151d07cb093e75Mike Stump  void setMSAsm(bool V) { MSAsm = V; }
1209dfab34a696d1dba8622248c31aaf605906cb6109Anders Carlsson
1210458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  //===--- Asm String Analysis ---===//
1211458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner
1212458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  const StringLiteral *getAsmString() const { return AsmStr; }
1213458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  StringLiteral *getAsmString() { return AsmStr; }
1214cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor  void setAsmString(StringLiteral *E) { AsmStr = E; }
1215cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
1216458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// AsmStringPiece - this is part of a decomposed asm string specification
1217458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// (for use with the AnalyzeAsmString function below).  An asm string is
1218458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// considered to be a concatenation of these parts.
1219458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  class AsmStringPiece {
1220458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  public:
1221458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    enum Kind {
1222458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      String,  // String in .ll asm string form, "$" -> "$$" and "%%" -> "%".
1223458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      Operand  // Operand reference, with optional modifier %c4.
1224458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    };
1225458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  private:
1226458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    Kind MyKind;
1227458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    std::string Str;
1228458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    unsigned OperandNo;
1229458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  public:
1230458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    AsmStringPiece(const std::string &S) : MyKind(String), Str(S) {}
1231458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    AsmStringPiece(unsigned OpNo, char Modifier)
1232458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      : MyKind(Operand), Str(), OperandNo(OpNo) {
1233458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      Str += Modifier;
1234458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    }
12351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1236458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    bool isString() const { return MyKind == String; }
1237458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    bool isOperand() const { return MyKind == Operand; }
12381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1239458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    const std::string &getString() const {
1240458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      assert(isString());
1241458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      return Str;
1242458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    }
1243458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner
1244458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    unsigned getOperandNo() const {
1245458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      assert(isOperand());
1246458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      return OperandNo;
1247458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    }
12481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1249458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    /// getModifier - Get the modifier for this operand, if present.  This
1250458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    /// returns '\0' if there was no modifier.
1251458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    char getModifier() const {
1252458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      assert(isOperand());
1253458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner      return Str[0];
1254458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner    }
1255458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  };
12561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1257458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// AnalyzeAsmString - Analyze the asm string of the current asm, decomposing
1258458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// it into pieces.  If the asm string is erroneous, emit errors and return
1259458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// true, otherwise return false.  This handles canonicalization and
1260458cd9c8a79b25b87dcea43c9d97a4c59f194799Chris Lattner  /// translation of strings from GCC syntax to LLVM IR syntax, and handles
12611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  //// flattening of named references like %[foo] to Operand AsmStringPiece's.
1262fb5058ef67c054296c88db18ab1b3717845cb71dChris Lattner  unsigned AnalyzeAsmString(llvm::SmallVectorImpl<AsmStringPiece> &Pieces,
1263fb5058ef67c054296c88db18ab1b3717845cb71dChris Lattner                            ASTContext &C, unsigned &DiagOffs) const;
12641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1266b327793860737d1c103a73aeda8057dd628a101dChris Lattner  //===--- Output operands ---===//
1267b327793860737d1c103a73aeda8057dd628a101dChris Lattner
1268b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  unsigned getNumOutputs() const { return NumOutputs; }
1269ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek
1270ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson  IdentifierInfo *getOutputIdentifier(unsigned i) const {
1271ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    return Names[i];
1272ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
1273ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek
1274ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson  llvm::StringRef getOutputName(unsigned i) const {
1275ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson    if (IdentifierInfo *II = getOutputIdentifier(i))
1276ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson      return II->getName();
1277c302113179a1c2b1254224ea9b6f5316ceeb375cSean Hunt
1278ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson    return llvm::StringRef();
1279ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson  }
1280ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson
1281b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// getOutputConstraint - Return the constraint string for the specified
1282b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// output operand.  All output constraints are known to be non-empty (either
1283b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// '=' or '+').
1284acb6bcb442936f3c14126fdd473307c17647cb84Anders Carlsson  llvm::StringRef getOutputConstraint(unsigned i) const;
12851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1286b327793860737d1c103a73aeda8057dd628a101dChris Lattner  const StringLiteral *getOutputConstraintLiteral(unsigned i) const {
1287ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    return Constraints[i];
1288ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
1289b327793860737d1c103a73aeda8057dd628a101dChris Lattner  StringLiteral *getOutputConstraintLiteral(unsigned i) {
1290b327793860737d1c103a73aeda8057dd628a101dChris Lattner    return Constraints[i];
1291b327793860737d1c103a73aeda8057dd628a101dChris Lattner  }
12921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1293ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  Expr *getOutputExpr(unsigned i);
12941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1295b327793860737d1c103a73aeda8057dd628a101dChris Lattner  const Expr *getOutputExpr(unsigned i) const {
1296b327793860737d1c103a73aeda8057dd628a101dChris Lattner    return const_cast<AsmStmt*>(this)->getOutputExpr(i);
1297b327793860737d1c103a73aeda8057dd628a101dChris Lattner  }
12981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1299b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// isOutputPlusConstraint - Return true if the specified output constraint
1300b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// is a "+" constraint (which is both an input and an output) or false if it
1301b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// is an "=" constraint (just an output).
1302b327793860737d1c103a73aeda8057dd628a101dChris Lattner  bool isOutputPlusConstraint(unsigned i) const {
1303b327793860737d1c103a73aeda8057dd628a101dChris Lattner    return getOutputConstraint(i)[0] == '+';
1304b327793860737d1c103a73aeda8057dd628a101dChris Lattner  }
13051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
130685759278332404e96d4bb89d0e976e46158cd026Chris Lattner  /// getNumPlusOperands - Return the number of output operands that have a "+"
130785759278332404e96d4bb89d0e976e46158cd026Chris Lattner  /// constraint.
130885759278332404e96d4bb89d0e976e46158cd026Chris Lattner  unsigned getNumPlusOperands() const;
13091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1310b327793860737d1c103a73aeda8057dd628a101dChris Lattner  //===--- Input operands ---===//
13111eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
13121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  unsigned getNumInputs() const { return NumInputs; }
13131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1314ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson  IdentifierInfo *getInputIdentifier(unsigned i) const {
1315ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    return Names[i + NumOutputs];
1316ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
13171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1318ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson  llvm::StringRef getInputName(unsigned i) const {
1319ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson    if (IdentifierInfo *II = getInputIdentifier(i))
1320ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson      return II->getName();
1321ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson
1322ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson    return llvm::StringRef();
1323ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson  }
1324ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson
1325b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// getInputConstraint - Return the specified input constraint.  Unlike output
1326b327793860737d1c103a73aeda8057dd628a101dChris Lattner  /// constraints, these can be empty.
1327acb6bcb442936f3c14126fdd473307c17647cb84Anders Carlsson  llvm::StringRef getInputConstraint(unsigned i) const;
13281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1329b327793860737d1c103a73aeda8057dd628a101dChris Lattner  const StringLiteral *getInputConstraintLiteral(unsigned i) const {
1330ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    return Constraints[i + NumOutputs];
1331ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
1332b327793860737d1c103a73aeda8057dd628a101dChris Lattner  StringLiteral *getInputConstraintLiteral(unsigned i) {
1333ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek    return Constraints[i + NumOutputs];
1334ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
13351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1336ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  Expr *getInputExpr(unsigned i);
13371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1338b327793860737d1c103a73aeda8057dd628a101dChris Lattner  const Expr *getInputExpr(unsigned i) const {
1339b327793860737d1c103a73aeda8057dd628a101dChris Lattner    return const_cast<AsmStmt*>(this)->getInputExpr(i);
1340b327793860737d1c103a73aeda8057dd628a101dChris Lattner  }
1341cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
1342acb6bcb442936f3c14126fdd473307c17647cb84Anders Carlsson  void setOutputsAndInputsAndClobbers(ASTContext &C,
1343ff93dbd887e40588ed55d135037bb9287488b285Anders Carlsson                                      IdentifierInfo **Names,
1344fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      StringLiteral **Constraints,
1345fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      Stmt **Exprs,
1346fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      unsigned NumOutputs,
1347c302113179a1c2b1254224ea9b6f5316ceeb375cSean Hunt                                      unsigned NumInputs,
1348fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      StringLiteral **Clobbers,
1349fdba9c069023f686e2608affde02c82131ee1cf8Anders Carlsson                                      unsigned NumClobbers);
1350cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
135110ca96ae9aed6906c3302403ef1a146a8d4c6b74Chris Lattner  //===--- Other ---===//
13521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
135310ca96ae9aed6906c3302403ef1a146a8d4c6b74Chris Lattner  /// getNamedOperand - Given a symbolic operand reference like %[foo],
135410ca96ae9aed6906c3302403ef1a146a8d4c6b74Chris Lattner  /// translate this into a numeric value needed to reference the same operand.
135510ca96ae9aed6906c3302403ef1a146a8d4c6b74Chris Lattner  /// This returns -1 if the operand name is invalid.
1356acb6bcb442936f3c14126fdd473307c17647cb84Anders Carlsson  int getNamedOperand(llvm::StringRef SymbolicName) const;
135710ca96ae9aed6906c3302403ef1a146a8d4c6b74Chris Lattner
1358966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson  unsigned getNumClobbers() const { return NumClobbers; }
1359b235fc2cf37621c7fc6511bb2b8788c95f9fb9fcAnders Carlsson  StringLiteral *getClobber(unsigned i) { return Clobbers[i]; }
1360d04c6e23f2e10eeb9936778d67f4a1c4a14cc4f6Anders Carlsson  const StringLiteral *getClobber(unsigned i) const { return Clobbers[i]; }
1361cd7d5a9dc558178ed7a66032f888781b3c592e4fDouglas Gregor
1362fe795956194141c91ae555985c9b930595bff43fChris Lattner  virtual SourceRange getSourceRange() const {
1363fe795956194141c91ae555985c9b930595bff43fChris Lattner    return SourceRange(AsmLoc, RParenLoc);
1364fe795956194141c91ae555985c9b930595bff43fChris Lattner  }
13651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1366fe795956194141c91ae555985c9b930595bff43fChris Lattner  static bool classof(const Stmt *T) {return T->getStmtClass() == AsmStmtClass;}
1367fe795956194141c91ae555985c9b930595bff43fChris Lattner  static bool classof(const AsmStmt *) { return true; }
13681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1369ec2f7dccb1f30ae137f74e764e44c2332b0a2ec0Ted Kremenek  // Input expr iterators.
13701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1371ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  typedef ExprIterator inputs_iterator;
1372ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  typedef ConstExprIterator const_inputs_iterator;
13731eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1374ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  inputs_iterator begin_inputs() {
1375966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson    return &Exprs[0] + NumOutputs;
1376ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
13771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1378ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  inputs_iterator end_inputs() {
1379966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson    return &Exprs[0] + NumOutputs + NumInputs;
1380ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
13811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1382ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  const_inputs_iterator begin_inputs() const {
1383966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson    return &Exprs[0] + NumOutputs;
1384ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  }
13851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1386ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  const_inputs_iterator end_inputs() const {
1387966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson    return &Exprs[0] + NumOutputs + NumInputs;
1388d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  }
13891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1390ec2f7dccb1f30ae137f74e764e44c2332b0a2ec0Ted Kremenek  // Output expr iterators.
13911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1392ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  typedef ExprIterator outputs_iterator;
1393ce2fc3a343ea6098a96d587071cee7299f11957aTed Kremenek  typedef ConstExprIterator const_outputs_iterator;
13941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1395d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  outputs_iterator begin_outputs() {
1396966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson    return &Exprs[0];
1397d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  }
1398d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  outputs_iterator end_outputs() {
1399966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson    return &Exprs[0] + NumOutputs;
1400d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  }
14011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1402d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  const_outputs_iterator begin_outputs() const {
1403966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson    return &Exprs[0];
1404d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  }
1405d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  const_outputs_iterator end_outputs() const {
1406966146e89141804ff6492739a2a6e6592ca671c7Anders Carlsson    return &Exprs[0] + NumOutputs;
1407d59a5bd0ab7821a643c540df8c84c70fef76a1c2Ted Kremenek  }
14081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
14091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Child iterators
14101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1411fe795956194141c91ae555985c9b930595bff43fChris Lattner  virtual child_iterator child_begin();
1412fe795956194141c91ae555985c9b930595bff43fChris Lattner  virtual child_iterator child_end();
1413fe795956194141c91ae555985c9b930595bff43fChris Lattner};
141478492dae00fb85e0da0f966df4745edafdafb66cTed Kremenek
14155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}  // end namespace clang
14165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
14175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#endif
1418