1 //===- StmtIterator.h - Iterators for Statements ----------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file defines the StmtIterator and ConstStmtIterator classes.
12 //===----------------------------------------------------------------------===//
14 #ifndef LLVM_CLANG_AST_STMTITERATOR_H
15 #define LLVM_CLANG_AST_STMTITERATOR_H
26 class VariableArrayType;
28 class StmtIteratorBase {
32 SizeOfTypeVAMode = 0x1,
41 uintptr_t RawVAPtr = 0;
44 StmtIteratorBase(Stmt **s) : stmt(s) {}
45 StmtIteratorBase(const VariableArrayType *t);
46 StmtIteratorBase(Decl **dgi, Decl **dge);
47 StmtIteratorBase() : stmt(nullptr) {}
49 bool inDeclGroup() const {
50 return (RawVAPtr & Flags) == DeclGroupMode;
53 bool inSizeOfTypeVA() const {
54 return (RawVAPtr & Flags) == SizeOfTypeVAMode;
58 return (RawVAPtr & Flags) == StmtMode;
61 const VariableArrayType *getVAPtr() const {
62 return reinterpret_cast<const VariableArrayType*>(RawVAPtr & ~Flags);
65 void setVAPtr(const VariableArrayType *P) {
66 assert(inDeclGroup() || inSizeOfTypeVA());
67 RawVAPtr = reinterpret_cast<uintptr_t>(P) | (RawVAPtr & Flags);
70 void NextDecl(bool ImmediateAdvance = true);
71 bool HandleDecl(Decl* D);
74 Stmt*& GetDeclExpr() const;
77 template <typename DERIVED, typename REFERENCE>
78 class StmtIteratorImpl : public StmtIteratorBase,
79 public std::iterator<std::forward_iterator_tag,
81 REFERENCE, REFERENCE> {
83 StmtIteratorImpl(const StmtIteratorBase& RHS) : StmtIteratorBase(RHS) {}
86 StmtIteratorImpl() = default;
87 StmtIteratorImpl(Stmt **s) : StmtIteratorBase(s) {}
88 StmtIteratorImpl(Decl **dgi, Decl **dge) : StmtIteratorBase(dgi, dge) {}
89 StmtIteratorImpl(const VariableArrayType *t) : StmtIteratorBase(t) {}
91 DERIVED& operator++() {
99 return static_cast<DERIVED&>(*this);
102 DERIVED operator++(int) {
103 DERIVED tmp = static_cast<DERIVED&>(*this);
108 bool operator==(const DERIVED& RHS) const {
109 return stmt == RHS.stmt && DGI == RHS.DGI && RawVAPtr == RHS.RawVAPtr;
112 bool operator!=(const DERIVED& RHS) const {
113 return stmt != RHS.stmt || DGI != RHS.DGI || RawVAPtr != RHS.RawVAPtr;
116 REFERENCE operator*() const {
117 return inStmt() ? *stmt : GetDeclExpr();
120 REFERENCE operator->() const { return operator*(); }
123 struct ConstStmtIterator;
125 struct StmtIterator : public StmtIteratorImpl<StmtIterator, Stmt*&> {
126 explicit StmtIterator() = default;
127 StmtIterator(Stmt** S) : StmtIteratorImpl<StmtIterator, Stmt*&>(S) {}
128 StmtIterator(Decl** dgi, Decl** dge)
129 : StmtIteratorImpl<StmtIterator, Stmt*&>(dgi, dge) {}
130 StmtIterator(const VariableArrayType *t)
131 : StmtIteratorImpl<StmtIterator, Stmt*&>(t) {}
134 StmtIterator(const StmtIteratorBase &RHS)
135 : StmtIteratorImpl<StmtIterator, Stmt *&>(RHS) {}
137 inline friend StmtIterator
138 cast_away_const(const ConstStmtIterator &RHS);
141 struct ConstStmtIterator : public StmtIteratorImpl<ConstStmtIterator,
143 explicit ConstStmtIterator() = default;
144 ConstStmtIterator(const StmtIterator& RHS)
145 : StmtIteratorImpl<ConstStmtIterator, const Stmt*>(RHS) {}
147 ConstStmtIterator(Stmt * const *S)
148 : StmtIteratorImpl<ConstStmtIterator, const Stmt *>(
149 const_cast<Stmt **>(S)) {}
152 inline StmtIterator cast_away_const(const ConstStmtIterator &RHS) {
158 #endif // LLVM_CLANG_AST_STMTITERATOR_H