blob: 8d4f6133efeb32ba3bdd1103f0260c89ecab96db [file] [log] [blame]
Reid Spencer5f016e22007-07-11 17:01:13 +00001//===--- Stmt.cpp - Statement AST Node Implementation ---------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner0bc735f2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Reid Spencer5f016e22007-07-11 17:01:13 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the Stmt class and statement subclasses.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/AST/Stmt.h"
15#include "clang/AST/ExprCXX.h"
16#include "clang/AST/StmtVisitor.h"
Chris Lattnerc7229c32007-10-07 08:58:51 +000017#include "clang/Basic/IdentifierTable.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000018using namespace clang;
19
Reid Spencer5f016e22007-07-11 17:01:13 +000020static struct StmtClassNameTable {
Chris Lattner63381352007-08-25 01:42:24 +000021 const char *Name;
22 unsigned Counter;
23 unsigned Size;
Chris Lattner1f683e92007-08-25 01:55:00 +000024} StmtClassInfo[Stmt::lastExprConstant+1];
Chris Lattner63381352007-08-25 01:42:24 +000025
26static StmtClassNameTable &getStmtInfoTableEntry(Stmt::StmtClass E) {
27 static bool Initialized = false;
28 if (Initialized)
29 return StmtClassInfo[E];
30
31 // Intialize the table on the first use.
32 Initialized = true;
33#define STMT(N, CLASS, PARENT) \
34 StmtClassInfo[N].Name = #CLASS; \
35 StmtClassInfo[N].Size = sizeof(CLASS);
Reid Spencer5f016e22007-07-11 17:01:13 +000036#include "clang/AST/StmtNodes.def"
Reid Spencer5f016e22007-07-11 17:01:13 +000037
Chris Lattner63381352007-08-25 01:42:24 +000038 return StmtClassInfo[E];
39}
40
Reid Spencer5f016e22007-07-11 17:01:13 +000041const char *Stmt::getStmtClassName() const {
Chris Lattner63381352007-08-25 01:42:24 +000042 return getStmtInfoTableEntry(sClass).Name;
Reid Spencer5f016e22007-07-11 17:01:13 +000043}
44
45void Stmt::PrintStats() {
Chris Lattner63381352007-08-25 01:42:24 +000046 // Ensure the table is primed.
47 getStmtInfoTableEntry(Stmt::NullStmtClass);
48
Reid Spencer5f016e22007-07-11 17:01:13 +000049 unsigned sum = 0;
50 fprintf(stderr, "*** Stmt/Expr Stats:\n");
Chris Lattner1f683e92007-08-25 01:55:00 +000051 for (int i = 0; i != Stmt::lastExprConstant+1; i++) {
Chris Lattner63381352007-08-25 01:42:24 +000052 if (StmtClassInfo[i].Name == 0) continue;
53 sum += StmtClassInfo[i].Counter;
Reid Spencer5f016e22007-07-11 17:01:13 +000054 }
55 fprintf(stderr, " %d stmts/exprs total.\n", sum);
56 sum = 0;
Chris Lattner1f683e92007-08-25 01:55:00 +000057 for (int i = 0; i != Stmt::lastExprConstant+1; i++) {
Chris Lattner63381352007-08-25 01:42:24 +000058 if (StmtClassInfo[i].Name == 0) continue;
Reid Spencer5f016e22007-07-11 17:01:13 +000059 fprintf(stderr, " %d %s, %d each (%d bytes)\n",
Chris Lattner63381352007-08-25 01:42:24 +000060 StmtClassInfo[i].Counter, StmtClassInfo[i].Name,
61 StmtClassInfo[i].Size,
62 StmtClassInfo[i].Counter*StmtClassInfo[i].Size);
63 sum += StmtClassInfo[i].Counter*StmtClassInfo[i].Size;
Reid Spencer5f016e22007-07-11 17:01:13 +000064 }
65 fprintf(stderr, "Total bytes = %d\n", sum);
66}
67
68void Stmt::addStmtClass(StmtClass s) {
Chris Lattner63381352007-08-25 01:42:24 +000069 ++getStmtInfoTableEntry(s).Counter;
Reid Spencer5f016e22007-07-11 17:01:13 +000070}
71
72static bool StatSwitch = false;
73
74bool Stmt::CollectingStats(bool enable) {
75 if (enable) StatSwitch = true;
Chris Lattner63381352007-08-25 01:42:24 +000076 return StatSwitch;
Reid Spencer5f016e22007-07-11 17:01:13 +000077}
78
79
Reid Spencer5f016e22007-07-11 17:01:13 +000080const char *LabelStmt::getName() const {
81 return getID()->getName();
82}
83
Steve Naroff507f2d52007-08-31 23:49:30 +000084// This is defined here to avoid polluting Stmt.h with importing Expr.h
85SourceRange ReturnStmt::getSourceRange() const {
86 if (RetExpr)
87 return SourceRange(RetLoc, RetExpr->getLocEnd());
88 else
89 return SourceRange(RetLoc);
90}
91
Ted Kremenekd48ade62007-10-01 16:34:52 +000092bool Stmt::hasImplicitControlFlow() const {
93 switch (sClass) {
94 default:
95 return false;
96
97 case CallExprClass:
98 case ConditionalOperatorClass:
99 case ChooseExprClass:
100 case StmtExprClass:
101 case DeclStmtClass:
102 return true;
103
104 case Stmt::BinaryOperatorClass: {
105 const BinaryOperator* B = cast<BinaryOperator>(this);
106 if (B->isLogicalOp() || B->getOpcode() == BinaryOperator::Comma)
107 return true;
108 else
109 return false;
110 }
111 }
112}
113
Anders Carlssonb235fc22007-11-22 01:36:19 +0000114AsmStmt::AsmStmt(SourceLocation asmloc,
Anders Carlsson39c47b52007-11-23 23:12:25 +0000115 bool isvolatile,
Anders Carlssonb235fc22007-11-22 01:36:19 +0000116 unsigned numoutputs,
117 unsigned numinputs,
118 std::string *names,
119 StringLiteral **constraints,
120 Expr **exprs,
121 StringLiteral *asmstr,
122 unsigned numclobbers,
123 StringLiteral **clobbers,
124 SourceLocation rparenloc)
125 : Stmt(AsmStmtClass), AsmLoc(asmloc), RParenLoc(rparenloc), AsmStr(asmstr)
Anders Carlsson39c47b52007-11-23 23:12:25 +0000126 , IsVolatile(isvolatile), NumOutputs(numoutputs), NumInputs(numinputs)
Anders Carlssonb235fc22007-11-22 01:36:19 +0000127{
128 for (unsigned i = 0, e = numinputs + numoutputs; i != e; i++) {
129 Names.push_back(names[i]);
130 Exprs.push_back(exprs[i]);
131 Constraints.push_back(constraints[i]);
132 }
133
134 for (unsigned i = 0; i != numclobbers; i++)
135 Clobbers.push_back(clobbers[i]);
136}
137
Ted Kremenek82977772007-08-24 21:09:09 +0000138//===----------------------------------------------------------------------===//
139// Child Iterators for iterating over subexpressions/substatements
140//===----------------------------------------------------------------------===//
141
142// DeclStmt
Ted Kremenek9ac59282007-10-18 23:28:49 +0000143Stmt::child_iterator DeclStmt::child_begin() { return getDecl(); }
144Stmt::child_iterator DeclStmt::child_end() { return child_iterator(); }
Ted Kremenek82977772007-08-24 21:09:09 +0000145
146// NullStmt
Ted Kremenek9ac59282007-10-18 23:28:49 +0000147Stmt::child_iterator NullStmt::child_begin() { return child_iterator(); }
148Stmt::child_iterator NullStmt::child_end() { return child_iterator(); }
Ted Kremenek82977772007-08-24 21:09:09 +0000149
150// CompoundStmt
151Stmt::child_iterator CompoundStmt::child_begin() { return &Body[0]; }
152Stmt::child_iterator CompoundStmt::child_end() { return &Body[0]+Body.size(); }
153
Ted Kremenekd97bb6c2007-08-30 16:50:46 +0000154// CaseStmt
155Stmt::child_iterator CaseStmt::child_begin() { return &SubExprs[0]; }
156Stmt::child_iterator CaseStmt::child_end() { return &SubExprs[END_EXPR]; }
157
158// DefaultStmt
159Stmt::child_iterator DefaultStmt::child_begin() { return &SubStmt; }
160Stmt::child_iterator DefaultStmt::child_end() { return &SubStmt+1; }
Ted Kremenek82977772007-08-24 21:09:09 +0000161
162// LabelStmt
163Stmt::child_iterator LabelStmt::child_begin() { return &SubStmt; }
Chris Lattnerb3938792007-08-30 00:53:54 +0000164Stmt::child_iterator LabelStmt::child_end() { return &SubStmt+1; }
Ted Kremenek82977772007-08-24 21:09:09 +0000165
166// IfStmt
167Stmt::child_iterator IfStmt::child_begin() { return &SubExprs[0]; }
168Stmt::child_iterator IfStmt::child_end() { return &SubExprs[0]+END_EXPR; }
169
170// SwitchStmt
171Stmt::child_iterator SwitchStmt::child_begin() { return &SubExprs[0]; }
172Stmt::child_iterator SwitchStmt::child_end() { return &SubExprs[0]+END_EXPR; }
173
174// WhileStmt
175Stmt::child_iterator WhileStmt::child_begin() { return &SubExprs[0]; }
176Stmt::child_iterator WhileStmt::child_end() { return &SubExprs[0]+END_EXPR; }
177
178// DoStmt
179Stmt::child_iterator DoStmt::child_begin() { return &SubExprs[0]; }
180Stmt::child_iterator DoStmt::child_end() { return &SubExprs[0]+END_EXPR; }
181
182// ForStmt
183Stmt::child_iterator ForStmt::child_begin() { return &SubExprs[0]; }
184Stmt::child_iterator ForStmt::child_end() { return &SubExprs[0]+END_EXPR; }
185
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000186// ObjCForCollectionStmt
187Stmt::child_iterator ObjCForCollectionStmt::child_begin() {
Fariborz Jahanian0196cab2008-01-02 22:54:34 +0000188 return &SubExprs[0];
189}
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000190Stmt::child_iterator ObjCForCollectionStmt::child_end() {
Fariborz Jahanian0196cab2008-01-02 22:54:34 +0000191 return &SubExprs[0]+END_EXPR;
192}
193
Ted Kremenek82977772007-08-24 21:09:09 +0000194// GotoStmt
Ted Kremenek9ac59282007-10-18 23:28:49 +0000195Stmt::child_iterator GotoStmt::child_begin() { return child_iterator(); }
196Stmt::child_iterator GotoStmt::child_end() { return child_iterator(); }
Ted Kremenek82977772007-08-24 21:09:09 +0000197
198// IndirectGotoStmt
199Stmt::child_iterator IndirectGotoStmt::child_begin() {
200 return reinterpret_cast<Stmt**>(&Target);
201}
202
Ted Kremenek9caf8b12007-10-18 00:24:38 +0000203Stmt::child_iterator IndirectGotoStmt::child_end() { return ++child_begin(); }
Ted Kremenek82977772007-08-24 21:09:09 +0000204
205// ContinueStmt
Ted Kremenek9ac59282007-10-18 23:28:49 +0000206Stmt::child_iterator ContinueStmt::child_begin() { return child_iterator(); }
207Stmt::child_iterator ContinueStmt::child_end() { return child_iterator(); }
Ted Kremenek82977772007-08-24 21:09:09 +0000208
209// BreakStmt
Ted Kremenek9ac59282007-10-18 23:28:49 +0000210Stmt::child_iterator BreakStmt::child_begin() { return child_iterator(); }
211Stmt::child_iterator BreakStmt::child_end() { return child_iterator(); }
Ted Kremenek82977772007-08-24 21:09:09 +0000212
213// ReturnStmt
Ted Kremenek2298f912007-08-27 20:58:16 +0000214Stmt::child_iterator ReturnStmt::child_begin() {
215 if (RetExpr) return reinterpret_cast<Stmt**>(&RetExpr);
Ted Kremenek9ac59282007-10-18 23:28:49 +0000216 else return child_iterator();
Ted Kremenek82977772007-08-24 21:09:09 +0000217}
218
Ted Kremenek2298f912007-08-27 20:58:16 +0000219Stmt::child_iterator ReturnStmt::child_end() {
220 if (RetExpr) return reinterpret_cast<Stmt**>(&RetExpr)+1;
Ted Kremenek9ac59282007-10-18 23:28:49 +0000221 else return child_iterator();
Ted Kremenek2298f912007-08-27 20:58:16 +0000222}
Ted Kremenek82977772007-08-24 21:09:09 +0000223
Chris Lattnerfe795952007-10-29 04:04:16 +0000224// AsmStmt
225Stmt::child_iterator AsmStmt::child_begin() { return child_iterator(); }
226Stmt::child_iterator AsmStmt::child_end() { return child_iterator(); }
227
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000228// ObjCAtCatchStmt
229Stmt::child_iterator ObjCAtCatchStmt::child_begin() { return &SubExprs[0]; }
230Stmt::child_iterator ObjCAtCatchStmt::child_end() {
Fariborz Jahanian3b1191d2007-11-01 23:59:59 +0000231 return &SubExprs[0]+END_EXPR;
232}
Fariborz Jahanianb210bd02007-11-01 21:12:44 +0000233
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000234// ObjCAtFinallyStmt
235Stmt::child_iterator ObjCAtFinallyStmt::child_begin() { return &AtFinallyStmt; }
236Stmt::child_iterator ObjCAtFinallyStmt::child_end() { return &AtFinallyStmt+1; }
Fariborz Jahanianb210bd02007-11-01 21:12:44 +0000237
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000238// ObjCAtTryStmt
239Stmt::child_iterator ObjCAtTryStmt::child_begin() { return &SubStmts[0]; }
240Stmt::child_iterator ObjCAtTryStmt::child_end() {
Fariborz Jahanian89079ea2007-11-07 17:43:16 +0000241 return &SubStmts[0]+END_EXPR;
Fariborz Jahanian3b1191d2007-11-01 23:59:59 +0000242}
Fariborz Jahanianb210bd02007-11-01 21:12:44 +0000243
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000244// ObjCAtThrowStmt
245Stmt::child_iterator ObjCAtThrowStmt::child_begin() {
Fariborz Jahanian39f8f152007-11-07 02:00:49 +0000246 return &Throw;
247}
248
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000249Stmt::child_iterator ObjCAtThrowStmt::child_end() {
Fariborz Jahanian39f8f152007-11-07 02:00:49 +0000250 return &Throw+1;
251}