blob: 21c9cbf17cd7e449845f4180c56219f75105594f [file] [log] [blame]
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001//===--- PCHReaderStmt.cpp - Stmt/Expr Deserialization ----------*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// Statement/expression deserialization. This implements the
11// PCHReader::ReadStmt method.
12//
13//===----------------------------------------------------------------------===//
14
15#include "clang/Frontend/PCHReader.h"
Douglas Gregor5d3507d2009-09-09 23:08:42 +000016#include "clang/AST/DeclCXX.h"
Chris Lattner92ba5ff2009-04-27 05:14:47 +000017#include "clang/AST/StmtVisitor.h"
18using namespace clang;
19
Chris Lattner92ba5ff2009-04-27 05:14:47 +000020namespace {
21 class PCHStmtReader : public StmtVisitor<PCHStmtReader, unsigned> {
22 PCHReader &Reader;
23 const PCHReader::RecordData &Record;
24 unsigned &Idx;
25 llvm::SmallVectorImpl<Stmt *> &StmtStack;
26
27 public:
28 PCHStmtReader(PCHReader &Reader, const PCHReader::RecordData &Record,
29 unsigned &Idx, llvm::SmallVectorImpl<Stmt *> &StmtStack)
30 : Reader(Reader), Record(Record), Idx(Idx), StmtStack(StmtStack) { }
31
32 /// \brief The number of record fields required for the Stmt class
33 /// itself.
34 static const unsigned NumStmtFields = 0;
35
36 /// \brief The number of record fields required for the Expr class
37 /// itself.
38 static const unsigned NumExprFields = NumStmtFields + 3;
39
40 // Each of the Visit* functions reads in part of the expression
41 // from the given record and the current expression stack, then
42 // return the total number of operands that it read from the
43 // expression stack.
44
45 unsigned VisitStmt(Stmt *S);
46 unsigned VisitNullStmt(NullStmt *S);
47 unsigned VisitCompoundStmt(CompoundStmt *S);
48 unsigned VisitSwitchCase(SwitchCase *S);
49 unsigned VisitCaseStmt(CaseStmt *S);
50 unsigned VisitDefaultStmt(DefaultStmt *S);
51 unsigned VisitLabelStmt(LabelStmt *S);
52 unsigned VisitIfStmt(IfStmt *S);
53 unsigned VisitSwitchStmt(SwitchStmt *S);
54 unsigned VisitWhileStmt(WhileStmt *S);
55 unsigned VisitDoStmt(DoStmt *S);
56 unsigned VisitForStmt(ForStmt *S);
57 unsigned VisitGotoStmt(GotoStmt *S);
58 unsigned VisitIndirectGotoStmt(IndirectGotoStmt *S);
59 unsigned VisitContinueStmt(ContinueStmt *S);
60 unsigned VisitBreakStmt(BreakStmt *S);
61 unsigned VisitReturnStmt(ReturnStmt *S);
62 unsigned VisitDeclStmt(DeclStmt *S);
63 unsigned VisitAsmStmt(AsmStmt *S);
64 unsigned VisitExpr(Expr *E);
65 unsigned VisitPredefinedExpr(PredefinedExpr *E);
66 unsigned VisitDeclRefExpr(DeclRefExpr *E);
67 unsigned VisitIntegerLiteral(IntegerLiteral *E);
68 unsigned VisitFloatingLiteral(FloatingLiteral *E);
69 unsigned VisitImaginaryLiteral(ImaginaryLiteral *E);
70 unsigned VisitStringLiteral(StringLiteral *E);
71 unsigned VisitCharacterLiteral(CharacterLiteral *E);
72 unsigned VisitParenExpr(ParenExpr *E);
73 unsigned VisitUnaryOperator(UnaryOperator *E);
74 unsigned VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E);
75 unsigned VisitArraySubscriptExpr(ArraySubscriptExpr *E);
76 unsigned VisitCallExpr(CallExpr *E);
77 unsigned VisitMemberExpr(MemberExpr *E);
78 unsigned VisitCastExpr(CastExpr *E);
79 unsigned VisitBinaryOperator(BinaryOperator *E);
80 unsigned VisitCompoundAssignOperator(CompoundAssignOperator *E);
81 unsigned VisitConditionalOperator(ConditionalOperator *E);
82 unsigned VisitImplicitCastExpr(ImplicitCastExpr *E);
83 unsigned VisitExplicitCastExpr(ExplicitCastExpr *E);
84 unsigned VisitCStyleCastExpr(CStyleCastExpr *E);
85 unsigned VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
86 unsigned VisitExtVectorElementExpr(ExtVectorElementExpr *E);
87 unsigned VisitInitListExpr(InitListExpr *E);
88 unsigned VisitDesignatedInitExpr(DesignatedInitExpr *E);
89 unsigned VisitImplicitValueInitExpr(ImplicitValueInitExpr *E);
90 unsigned VisitVAArgExpr(VAArgExpr *E);
91 unsigned VisitAddrLabelExpr(AddrLabelExpr *E);
92 unsigned VisitStmtExpr(StmtExpr *E);
93 unsigned VisitTypesCompatibleExpr(TypesCompatibleExpr *E);
94 unsigned VisitChooseExpr(ChooseExpr *E);
95 unsigned VisitGNUNullExpr(GNUNullExpr *E);
96 unsigned VisitShuffleVectorExpr(ShuffleVectorExpr *E);
97 unsigned VisitBlockExpr(BlockExpr *E);
98 unsigned VisitBlockDeclRefExpr(BlockDeclRefExpr *E);
99 unsigned VisitObjCStringLiteral(ObjCStringLiteral *E);
100 unsigned VisitObjCEncodeExpr(ObjCEncodeExpr *E);
101 unsigned VisitObjCSelectorExpr(ObjCSelectorExpr *E);
102 unsigned VisitObjCProtocolExpr(ObjCProtocolExpr *E);
103 unsigned VisitObjCIvarRefExpr(ObjCIvarRefExpr *E);
104 unsigned VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E);
Fariborz Jahanian9a846652009-08-20 17:02:02 +0000105 unsigned VisitObjCImplicitSetterGetterRefExpr(
106 ObjCImplicitSetterGetterRefExpr *E);
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000107 unsigned VisitObjCMessageExpr(ObjCMessageExpr *E);
108 unsigned VisitObjCSuperExpr(ObjCSuperExpr *E);
Steve Naroffe87026a2009-07-24 17:54:45 +0000109 unsigned VisitObjCIsaExpr(ObjCIsaExpr *E);
Mike Stump11289f42009-09-09 15:08:12 +0000110
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000111 unsigned VisitObjCForCollectionStmt(ObjCForCollectionStmt *);
112 unsigned VisitObjCAtCatchStmt(ObjCAtCatchStmt *);
113 unsigned VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *);
114 unsigned VisitObjCAtTryStmt(ObjCAtTryStmt *);
115 unsigned VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *);
116 unsigned VisitObjCAtThrowStmt(ObjCAtThrowStmt *);
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +0000117
118 unsigned VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
Douglas Gregor5d3507d2009-09-09 23:08:42 +0000119 unsigned VisitCXXConstructExpr(CXXConstructExpr *E);
Sam Weinigd01101e2010-01-16 21:21:01 +0000120 unsigned VisitCXXNamedCastExpr(CXXNamedCastExpr *E);
121 unsigned VisitCXXStaticCastExpr(CXXStaticCastExpr *E);
122 unsigned VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E);
123 unsigned VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E);
124 unsigned VisitCXXConstCastExpr(CXXConstCastExpr *E);
125 unsigned VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E);
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000126 };
127}
128
129unsigned PCHStmtReader::VisitStmt(Stmt *S) {
130 assert(Idx == NumStmtFields && "Incorrect statement field count");
131 return 0;
132}
133
134unsigned PCHStmtReader::VisitNullStmt(NullStmt *S) {
135 VisitStmt(S);
136 S->setSemiLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
137 return 0;
138}
139
140unsigned PCHStmtReader::VisitCompoundStmt(CompoundStmt *S) {
141 VisitStmt(S);
142 unsigned NumStmts = Record[Idx++];
Mike Stump11289f42009-09-09 15:08:12 +0000143 S->setStmts(*Reader.getContext(),
Jay Foad7d0479f2009-05-21 09:52:38 +0000144 StmtStack.data() + StmtStack.size() - NumStmts, NumStmts);
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000145 S->setLBracLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
146 S->setRBracLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
147 return NumStmts;
148}
149
150unsigned PCHStmtReader::VisitSwitchCase(SwitchCase *S) {
151 VisitStmt(S);
152 Reader.RecordSwitchCaseID(S, Record[Idx++]);
153 return 0;
154}
155
156unsigned PCHStmtReader::VisitCaseStmt(CaseStmt *S) {
157 VisitSwitchCase(S);
158 S->setLHS(cast<Expr>(StmtStack[StmtStack.size() - 3]));
159 S->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
160 S->setSubStmt(StmtStack.back());
161 S->setCaseLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor2a2d00f2009-05-15 23:57:33 +0000162 S->setEllipsisLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
163 S->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000164 return 3;
165}
166
167unsigned PCHStmtReader::VisitDefaultStmt(DefaultStmt *S) {
168 VisitSwitchCase(S);
169 S->setSubStmt(StmtStack.back());
170 S->setDefaultLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor2a2d00f2009-05-15 23:57:33 +0000171 S->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000172 return 1;
173}
174
175unsigned PCHStmtReader::VisitLabelStmt(LabelStmt *S) {
176 VisitStmt(S);
177 S->setID(Reader.GetIdentifierInfo(Record, Idx));
178 S->setSubStmt(StmtStack.back());
179 S->setIdentLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
180 Reader.RecordLabelStmt(S, Record[Idx++]);
181 return 1;
182}
183
184unsigned PCHStmtReader::VisitIfStmt(IfStmt *S) {
185 VisitStmt(S);
Douglas Gregor633caca2009-11-23 23:44:04 +0000186 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000187 S->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
188 S->setThen(StmtStack[StmtStack.size() - 2]);
189 S->setElse(StmtStack[StmtStack.size() - 1]);
190 S->setIfLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor9d73cab2009-05-15 18:53:42 +0000191 S->setElseLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000192 return 3;
193}
194
195unsigned PCHStmtReader::VisitSwitchStmt(SwitchStmt *S) {
196 VisitStmt(S);
Douglas Gregordcf19622009-11-24 17:07:59 +0000197 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000198 S->setCond(cast<Expr>(StmtStack[StmtStack.size() - 2]));
199 S->setBody(StmtStack.back());
200 S->setSwitchLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
201 SwitchCase *PrevSC = 0;
202 for (unsigned N = Record.size(); Idx != N; ++Idx) {
203 SwitchCase *SC = Reader.getSwitchCaseWithID(Record[Idx]);
204 if (PrevSC)
205 PrevSC->setNextSwitchCase(SC);
206 else
207 S->setSwitchCaseList(SC);
Mike Stump11289f42009-09-09 15:08:12 +0000208
Douglas Gregor2c742022009-08-08 01:41:12 +0000209 // Retain this SwitchCase, since SwitchStmt::addSwitchCase() would
210 // normally retain it (but we aren't calling addSwitchCase).
211 SC->Retain();
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000212 PrevSC = SC;
213 }
214 return 2;
215}
216
217unsigned PCHStmtReader::VisitWhileStmt(WhileStmt *S) {
218 VisitStmt(S);
Douglas Gregor680f8612009-11-24 21:15:44 +0000219 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000220 S->setCond(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
221 S->setBody(StmtStack.back());
222 S->setWhileLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
223 return 2;
224}
225
226unsigned PCHStmtReader::VisitDoStmt(DoStmt *S) {
227 VisitStmt(S);
228 S->setCond(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
229 S->setBody(StmtStack.back());
230 S->setDoLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor3daa82d2009-05-15 21:56:04 +0000231 S->setWhileLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner815b70e2009-06-12 23:04:47 +0000232 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000233 return 2;
234}
235
236unsigned PCHStmtReader::VisitForStmt(ForStmt *S) {
237 VisitStmt(S);
238 S->setInit(StmtStack[StmtStack.size() - 4]);
239 S->setCond(cast_or_null<Expr>(StmtStack[StmtStack.size() - 3]));
Douglas Gregor7bab5ff2009-11-25 00:27:52 +0000240 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000241 S->setInc(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
242 S->setBody(StmtStack.back());
243 S->setForLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor5d138682009-05-15 22:12:32 +0000244 S->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
245 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000246 return 4;
247}
248
249unsigned PCHStmtReader::VisitGotoStmt(GotoStmt *S) {
250 VisitStmt(S);
251 Reader.SetLabelOf(S, Record[Idx++]);
252 S->setGotoLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
253 S->setLabelLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
254 return 0;
255}
256
257unsigned PCHStmtReader::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
258 VisitStmt(S);
259 S->setGotoLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor30776d42009-05-16 00:20:29 +0000260 S->setStarLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000261 S->setTarget(cast_or_null<Expr>(StmtStack.back()));
262 return 1;
263}
264
265unsigned PCHStmtReader::VisitContinueStmt(ContinueStmt *S) {
266 VisitStmt(S);
267 S->setContinueLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
268 return 0;
269}
270
271unsigned PCHStmtReader::VisitBreakStmt(BreakStmt *S) {
272 VisitStmt(S);
273 S->setBreakLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
274 return 0;
275}
276
277unsigned PCHStmtReader::VisitReturnStmt(ReturnStmt *S) {
278 VisitStmt(S);
279 S->setRetValue(cast_or_null<Expr>(StmtStack.back()));
280 S->setReturnLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
281 return 1;
282}
283
284unsigned PCHStmtReader::VisitDeclStmt(DeclStmt *S) {
285 VisitStmt(S);
286 S->setStartLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
287 S->setEndLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
288
289 if (Idx + 1 == Record.size()) {
290 // Single declaration
291 S->setDeclGroup(DeclGroupRef(Reader.GetDecl(Record[Idx++])));
292 } else {
293 llvm::SmallVector<Decl *, 16> Decls;
294 Decls.reserve(Record.size() - Idx);
295 for (unsigned N = Record.size(); Idx != N; ++Idx)
296 Decls.push_back(Reader.GetDecl(Record[Idx]));
Chris Lattner8575daa2009-04-27 21:45:14 +0000297 S->setDeclGroup(DeclGroupRef(DeclGroup::Create(*Reader.getContext(),
Douglas Gregor038c3382009-05-22 22:45:36 +0000298 Decls.data(),
299 Decls.size())));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000300 }
301 return 0;
302}
303
304unsigned PCHStmtReader::VisitAsmStmt(AsmStmt *S) {
305 VisitStmt(S);
306 unsigned NumOutputs = Record[Idx++];
307 unsigned NumInputs = Record[Idx++];
308 unsigned NumClobbers = Record[Idx++];
309 S->setAsmLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
310 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
311 S->setVolatile(Record[Idx++]);
312 S->setSimple(Record[Idx++]);
Mike Stump90be58a2010-01-04 22:37:17 +0000313 S->setMSAsm(Record[Idx++]);
Mike Stump11289f42009-09-09 15:08:12 +0000314
315 unsigned StackIdx
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000316 = StmtStack.size() - (NumOutputs*2 + NumInputs*2 + NumClobbers + 1);
317 S->setAsmString(cast_or_null<StringLiteral>(StmtStack[StackIdx++]));
318
319 // Outputs and inputs
320 llvm::SmallVector<std::string, 16> Names;
321 llvm::SmallVector<StringLiteral*, 16> Constraints;
322 llvm::SmallVector<Stmt*, 16> Exprs;
323 for (unsigned I = 0, N = NumOutputs + NumInputs; I != N; ++I) {
324 Names.push_back(Reader.ReadString(Record, Idx));
325 Constraints.push_back(cast_or_null<StringLiteral>(StmtStack[StackIdx++]));
326 Exprs.push_back(StmtStack[StackIdx++]);
327 }
328 S->setOutputsAndInputs(NumOutputs, NumInputs,
Jay Foad7d0479f2009-05-21 09:52:38 +0000329 Names.data(), Constraints.data(), Exprs.data());
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000330
331 // Constraints
332 llvm::SmallVector<StringLiteral*, 16> Clobbers;
333 for (unsigned I = 0; I != NumClobbers; ++I)
334 Clobbers.push_back(cast_or_null<StringLiteral>(StmtStack[StackIdx++]));
Jay Foad7d0479f2009-05-21 09:52:38 +0000335 S->setClobbers(Clobbers.data(), NumClobbers);
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000336
337 assert(StackIdx == StmtStack.size() && "Error deserializing AsmStmt");
338 return NumOutputs*2 + NumInputs*2 + NumClobbers + 1;
339}
340
341unsigned PCHStmtReader::VisitExpr(Expr *E) {
342 VisitStmt(E);
343 E->setType(Reader.GetType(Record[Idx++]));
344 E->setTypeDependent(Record[Idx++]);
345 E->setValueDependent(Record[Idx++]);
346 assert(Idx == NumExprFields && "Incorrect expression field count");
347 return 0;
348}
349
350unsigned PCHStmtReader::VisitPredefinedExpr(PredefinedExpr *E) {
351 VisitExpr(E);
352 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
353 E->setIdentType((PredefinedExpr::IdentType)Record[Idx++]);
354 return 0;
355}
356
357unsigned PCHStmtReader::VisitDeclRefExpr(DeclRefExpr *E) {
358 VisitExpr(E);
John McCallce546572009-12-08 09:08:17 +0000359 E->setDecl(cast<ValueDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000360 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor4bd90e52009-10-23 18:54:35 +0000361 // FIXME: read qualifier
362 // FIXME: read explicit template arguments
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000363 return 0;
364}
365
366unsigned PCHStmtReader::VisitIntegerLiteral(IntegerLiteral *E) {
367 VisitExpr(E);
368 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
369 E->setValue(Reader.ReadAPInt(Record, Idx));
370 return 0;
371}
372
373unsigned PCHStmtReader::VisitFloatingLiteral(FloatingLiteral *E) {
374 VisitExpr(E);
375 E->setValue(Reader.ReadAPFloat(Record, Idx));
376 E->setExact(Record[Idx++]);
377 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
378 return 0;
379}
380
381unsigned PCHStmtReader::VisitImaginaryLiteral(ImaginaryLiteral *E) {
382 VisitExpr(E);
383 E->setSubExpr(cast<Expr>(StmtStack.back()));
384 return 1;
385}
386
387unsigned PCHStmtReader::VisitStringLiteral(StringLiteral *E) {
388 VisitExpr(E);
389 unsigned Len = Record[Idx++];
Mike Stump11289f42009-09-09 15:08:12 +0000390 assert(Record[Idx] == E->getNumConcatenated() &&
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000391 "Wrong number of concatenated tokens!");
392 ++Idx;
393 E->setWide(Record[Idx++]);
394
Mike Stump11289f42009-09-09 15:08:12 +0000395 // Read string data
Daniel Dunbar36217882009-09-22 03:27:33 +0000396 llvm::SmallString<16> Str(&Record[Idx], &Record[Idx] + Len);
397 E->setString(*Reader.getContext(), Str.str());
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000398 Idx += Len;
399
400 // Read source locations
401 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
402 E->setStrTokenLoc(I, SourceLocation::getFromRawEncoding(Record[Idx++]));
403
404 return 0;
405}
406
407unsigned PCHStmtReader::VisitCharacterLiteral(CharacterLiteral *E) {
408 VisitExpr(E);
409 E->setValue(Record[Idx++]);
410 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
411 E->setWide(Record[Idx++]);
412 return 0;
413}
414
415unsigned PCHStmtReader::VisitParenExpr(ParenExpr *E) {
416 VisitExpr(E);
417 E->setLParen(SourceLocation::getFromRawEncoding(Record[Idx++]));
418 E->setRParen(SourceLocation::getFromRawEncoding(Record[Idx++]));
419 E->setSubExpr(cast<Expr>(StmtStack.back()));
420 return 1;
421}
422
423unsigned PCHStmtReader::VisitUnaryOperator(UnaryOperator *E) {
424 VisitExpr(E);
425 E->setSubExpr(cast<Expr>(StmtStack.back()));
426 E->setOpcode((UnaryOperator::Opcode)Record[Idx++]);
427 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
428 return 1;
429}
430
431unsigned PCHStmtReader::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) {
432 VisitExpr(E);
433 E->setSizeof(Record[Idx++]);
434 if (Record[Idx] == 0) {
435 E->setArgument(cast<Expr>(StmtStack.back()));
436 ++Idx;
437 } else {
John McCallbcd03502009-12-07 02:54:59 +0000438 E->setArgument(Reader.GetTypeSourceInfo(Record, Idx));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000439 }
440 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
441 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
442 return E->isArgumentType()? 0 : 1;
443}
444
445unsigned PCHStmtReader::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
446 VisitExpr(E);
447 E->setLHS(cast<Expr>(StmtStack[StmtStack.size() - 2]));
448 E->setRHS(cast<Expr>(StmtStack[StmtStack.size() - 1]));
449 E->setRBracketLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
450 return 2;
451}
452
453unsigned PCHStmtReader::VisitCallExpr(CallExpr *E) {
454 VisitExpr(E);
Chris Lattner8575daa2009-04-27 21:45:14 +0000455 E->setNumArgs(*Reader.getContext(), Record[Idx++]);
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000456 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
457 E->setCallee(cast<Expr>(StmtStack[StmtStack.size() - E->getNumArgs() - 1]));
458 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
459 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
460 return E->getNumArgs() + 1;
461}
462
463unsigned PCHStmtReader::VisitMemberExpr(MemberExpr *E) {
464 VisitExpr(E);
465 E->setBase(cast<Expr>(StmtStack.back()));
Eli Friedman3de20c52009-12-04 06:46:54 +0000466 E->setMemberDecl(cast<ValueDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000467 E->setMemberLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
468 E->setArrow(Record[Idx++]);
469 return 1;
470}
471
Steve Naroffe87026a2009-07-24 17:54:45 +0000472unsigned PCHStmtReader::VisitObjCIsaExpr(ObjCIsaExpr *E) {
473 VisitExpr(E);
474 E->setBase(cast<Expr>(StmtStack.back()));
475 E->setIsaMemberLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
476 E->setArrow(Record[Idx++]);
477 return 1;
478}
479
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000480unsigned PCHStmtReader::VisitCastExpr(CastExpr *E) {
481 VisitExpr(E);
482 E->setSubExpr(cast<Expr>(StmtStack.back()));
Anders Carlssona2615922009-07-31 00:48:10 +0000483 E->setCastKind((CastExpr::CastKind)Record[Idx++]);
484
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000485 return 1;
486}
487
488unsigned PCHStmtReader::VisitBinaryOperator(BinaryOperator *E) {
489 VisitExpr(E);
490 E->setLHS(cast<Expr>(StmtStack.end()[-2]));
491 E->setRHS(cast<Expr>(StmtStack.end()[-1]));
492 E->setOpcode((BinaryOperator::Opcode)Record[Idx++]);
493 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
494 return 2;
495}
496
497unsigned PCHStmtReader::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
498 VisitBinaryOperator(E);
499 E->setComputationLHSType(Reader.GetType(Record[Idx++]));
500 E->setComputationResultType(Reader.GetType(Record[Idx++]));
501 return 2;
502}
503
504unsigned PCHStmtReader::VisitConditionalOperator(ConditionalOperator *E) {
505 VisitExpr(E);
506 E->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
507 E->setLHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
508 E->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 1]));
Douglas Gregor7e112b02009-08-26 14:37:04 +0000509 E->setQuestionLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
510 E->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000511 return 3;
512}
513
514unsigned PCHStmtReader::VisitImplicitCastExpr(ImplicitCastExpr *E) {
515 VisitCastExpr(E);
516 E->setLvalueCast(Record[Idx++]);
517 return 1;
518}
519
520unsigned PCHStmtReader::VisitExplicitCastExpr(ExplicitCastExpr *E) {
521 VisitCastExpr(E);
John McCall97513962010-01-15 18:39:57 +0000522 E->setTypeInfoAsWritten(Reader.GetTypeSourceInfo(Record, Idx));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000523 return 1;
524}
525
526unsigned PCHStmtReader::VisitCStyleCastExpr(CStyleCastExpr *E) {
527 VisitExplicitCastExpr(E);
528 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
529 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
530 return 1;
531}
532
533unsigned PCHStmtReader::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
534 VisitExpr(E);
535 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
John McCalle15bbff2010-01-18 19:35:47 +0000536 E->setTypeSourceInfo(Reader.GetTypeSourceInfo(Record, Idx));
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000537 E->setInitializer(cast<Expr>(StmtStack.back()));
538 E->setFileScope(Record[Idx++]);
539 return 1;
540}
541
542unsigned PCHStmtReader::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
543 VisitExpr(E);
544 E->setBase(cast<Expr>(StmtStack.back()));
545 E->setAccessor(Reader.GetIdentifierInfo(Record, Idx));
546 E->setAccessorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
547 return 1;
548}
549
550unsigned PCHStmtReader::VisitInitListExpr(InitListExpr *E) {
551 VisitExpr(E);
552 unsigned NumInits = Record[Idx++];
553 E->reserveInits(NumInits);
554 for (unsigned I = 0; I != NumInits; ++I)
Mike Stump11289f42009-09-09 15:08:12 +0000555 E->updateInit(I,
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000556 cast<Expr>(StmtStack[StmtStack.size() - NumInits - 1 + I]));
557 E->setSyntacticForm(cast_or_null<InitListExpr>(StmtStack.back()));
558 E->setLBraceLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
559 E->setRBraceLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
560 E->setInitializedFieldInUnion(
561 cast_or_null<FieldDecl>(Reader.GetDecl(Record[Idx++])));
562 E->sawArrayRangeDesignator(Record[Idx++]);
563 return NumInits + 1;
564}
565
566unsigned PCHStmtReader::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
567 typedef DesignatedInitExpr::Designator Designator;
568
569 VisitExpr(E);
570 unsigned NumSubExprs = Record[Idx++];
571 assert(NumSubExprs == E->getNumSubExprs() && "Wrong number of subexprs");
572 for (unsigned I = 0; I != NumSubExprs; ++I)
573 E->setSubExpr(I, cast<Expr>(StmtStack[StmtStack.size() - NumSubExprs + I]));
574 E->setEqualOrColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
575 E->setGNUSyntax(Record[Idx++]);
576
577 llvm::SmallVector<Designator, 4> Designators;
578 while (Idx < Record.size()) {
579 switch ((pch::DesignatorTypes)Record[Idx++]) {
580 case pch::DESIG_FIELD_DECL: {
581 FieldDecl *Field = cast<FieldDecl>(Reader.GetDecl(Record[Idx++]));
Mike Stump11289f42009-09-09 15:08:12 +0000582 SourceLocation DotLoc
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000583 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump11289f42009-09-09 15:08:12 +0000584 SourceLocation FieldLoc
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000585 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump11289f42009-09-09 15:08:12 +0000586 Designators.push_back(Designator(Field->getIdentifier(), DotLoc,
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000587 FieldLoc));
588 Designators.back().setField(Field);
589 break;
590 }
591
592 case pch::DESIG_FIELD_NAME: {
593 const IdentifierInfo *Name = Reader.GetIdentifierInfo(Record, Idx);
Mike Stump11289f42009-09-09 15:08:12 +0000594 SourceLocation DotLoc
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000595 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump11289f42009-09-09 15:08:12 +0000596 SourceLocation FieldLoc
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000597 = SourceLocation::getFromRawEncoding(Record[Idx++]);
598 Designators.push_back(Designator(Name, DotLoc, FieldLoc));
599 break;
600 }
Mike Stump11289f42009-09-09 15:08:12 +0000601
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000602 case pch::DESIG_ARRAY: {
603 unsigned Index = Record[Idx++];
604 SourceLocation LBracketLoc
605 = SourceLocation::getFromRawEncoding(Record[Idx++]);
606 SourceLocation RBracketLoc
607 = SourceLocation::getFromRawEncoding(Record[Idx++]);
608 Designators.push_back(Designator(Index, LBracketLoc, RBracketLoc));
609 break;
610 }
611
612 case pch::DESIG_ARRAY_RANGE: {
613 unsigned Index = Record[Idx++];
614 SourceLocation LBracketLoc
615 = SourceLocation::getFromRawEncoding(Record[Idx++]);
616 SourceLocation EllipsisLoc
617 = SourceLocation::getFromRawEncoding(Record[Idx++]);
618 SourceLocation RBracketLoc
619 = SourceLocation::getFromRawEncoding(Record[Idx++]);
620 Designators.push_back(Designator(Index, LBracketLoc, EllipsisLoc,
621 RBracketLoc));
622 break;
623 }
624 }
625 }
Douglas Gregor03e8bdc2010-01-06 23:17:19 +0000626 E->setDesignators(*Reader.getContext(),
627 Designators.data(), Designators.size());
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000628
629 return NumSubExprs;
630}
631
632unsigned PCHStmtReader::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
633 VisitExpr(E);
634 return 0;
635}
636
637unsigned PCHStmtReader::VisitVAArgExpr(VAArgExpr *E) {
638 VisitExpr(E);
639 E->setSubExpr(cast<Expr>(StmtStack.back()));
640 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
641 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
642 return 1;
643}
644
645unsigned PCHStmtReader::VisitAddrLabelExpr(AddrLabelExpr *E) {
646 VisitExpr(E);
647 E->setAmpAmpLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
648 E->setLabelLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
649 Reader.SetLabelOf(E, Record[Idx++]);
650 return 0;
651}
652
653unsigned PCHStmtReader::VisitStmtExpr(StmtExpr *E) {
654 VisitExpr(E);
655 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
656 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
657 E->setSubStmt(cast_or_null<CompoundStmt>(StmtStack.back()));
658 return 1;
659}
660
661unsigned PCHStmtReader::VisitTypesCompatibleExpr(TypesCompatibleExpr *E) {
662 VisitExpr(E);
663 E->setArgType1(Reader.GetType(Record[Idx++]));
664 E->setArgType2(Reader.GetType(Record[Idx++]));
665 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
666 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
667 return 0;
668}
669
670unsigned PCHStmtReader::VisitChooseExpr(ChooseExpr *E) {
671 VisitExpr(E);
672 E->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
673 E->setLHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
674 E->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 1]));
675 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
676 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
677 return 3;
678}
679
680unsigned PCHStmtReader::VisitGNUNullExpr(GNUNullExpr *E) {
681 VisitExpr(E);
682 E->setTokenLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
683 return 0;
684}
685
686unsigned PCHStmtReader::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
687 VisitExpr(E);
688 unsigned NumExprs = Record[Idx++];
Mike Stump11289f42009-09-09 15:08:12 +0000689 E->setExprs(*Reader.getContext(),
Nate Begeman48745922009-08-12 02:28:50 +0000690 (Expr **)&StmtStack[StmtStack.size() - NumExprs], NumExprs);
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000691 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
692 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
693 return NumExprs;
694}
695
696unsigned PCHStmtReader::VisitBlockExpr(BlockExpr *E) {
697 VisitExpr(E);
698 E->setBlockDecl(cast_or_null<BlockDecl>(Reader.GetDecl(Record[Idx++])));
699 E->setHasBlockDeclRefExprs(Record[Idx++]);
700 return 0;
701}
702
703unsigned PCHStmtReader::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
704 VisitExpr(E);
705 E->setDecl(cast<ValueDecl>(Reader.GetDecl(Record[Idx++])));
706 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
707 E->setByRef(Record[Idx++]);
Fariborz Jahaniane8918d52009-06-20 00:02:26 +0000708 E->setConstQualAdded(Record[Idx++]);
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000709 return 0;
710}
711
712//===----------------------------------------------------------------------===//
713// Objective-C Expressions and Statements
714
715unsigned PCHStmtReader::VisitObjCStringLiteral(ObjCStringLiteral *E) {
716 VisitExpr(E);
717 E->setString(cast<StringLiteral>(StmtStack.back()));
718 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
719 return 1;
720}
721
722unsigned PCHStmtReader::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
723 VisitExpr(E);
724 E->setEncodedType(Reader.GetType(Record[Idx++]));
725 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
726 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
727 return 0;
728}
729
730unsigned PCHStmtReader::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
731 VisitExpr(E);
732 E->setSelector(Reader.GetSelector(Record, Idx));
733 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
734 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
735 return 0;
736}
737
738unsigned PCHStmtReader::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
739 VisitExpr(E);
740 E->setProtocol(cast<ObjCProtocolDecl>(Reader.GetDecl(Record[Idx++])));
741 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
742 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
743 return 0;
744}
745
746unsigned PCHStmtReader::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
747 VisitExpr(E);
748 E->setDecl(cast<ObjCIvarDecl>(Reader.GetDecl(Record[Idx++])));
749 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
750 E->setBase(cast<Expr>(StmtStack.back()));
751 E->setIsArrow(Record[Idx++]);
752 E->setIsFreeIvar(Record[Idx++]);
753 return 1;
754}
755
756unsigned PCHStmtReader::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
757 VisitExpr(E);
758 E->setProperty(cast<ObjCPropertyDecl>(Reader.GetDecl(Record[Idx++])));
759 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
760 E->setBase(cast<Expr>(StmtStack.back()));
761 return 1;
762}
763
Fariborz Jahanian9a846652009-08-20 17:02:02 +0000764unsigned PCHStmtReader::VisitObjCImplicitSetterGetterRefExpr(
765 ObjCImplicitSetterGetterRefExpr *E) {
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000766 VisitExpr(E);
767 E->setGetterMethod(
768 cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
769 E->setSetterMethod(
770 cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
Fariborz Jahanian19380c42009-08-18 21:37:33 +0000771 E->setInterfaceDecl(
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000772 cast_or_null<ObjCInterfaceDecl>(Reader.GetDecl(Record[Idx++])));
773 E->setBase(cast_or_null<Expr>(StmtStack.back()));
774 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
775 E->setClassLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
776 return 1;
777}
778
779unsigned PCHStmtReader::VisitObjCMessageExpr(ObjCMessageExpr *E) {
780 VisitExpr(E);
781 E->setNumArgs(Record[Idx++]);
782 E->setLeftLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
783 E->setRightLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
784 E->setSelector(Reader.GetSelector(Record, Idx));
785 E->setMethodDecl(cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
Mike Stump11289f42009-09-09 15:08:12 +0000786
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000787 E->setReceiver(
788 cast_or_null<Expr>(StmtStack[StmtStack.size() - E->getNumArgs() - 1]));
789 if (!E->getReceiver()) {
790 ObjCMessageExpr::ClassInfo CI;
791 CI.first = cast_or_null<ObjCInterfaceDecl>(Reader.GetDecl(Record[Idx++]));
792 CI.second = Reader.GetIdentifierInfo(Record, Idx);
793 E->setClassInfo(CI);
794 }
795
796 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
797 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
798 return E->getNumArgs() + 1;
799}
800
801unsigned PCHStmtReader::VisitObjCSuperExpr(ObjCSuperExpr *E) {
802 VisitExpr(E);
803 E->setLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
804 return 0;
805}
806
807unsigned PCHStmtReader::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
808 VisitStmt(S);
809 S->setElement(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 3]));
810 S->setCollection(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
811 S->setBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
812 S->setForLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
813 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
814 return 3;
815}
816
817unsigned PCHStmtReader::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
818 VisitStmt(S);
819 S->setCatchBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
820 S->setNextCatchStmt(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
821 S->setCatchParamDecl(cast_or_null<ParmVarDecl>(Reader.GetDecl(Record[Idx++])));
822 S->setAtCatchLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
823 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
824 return 2;
825}
826
827unsigned PCHStmtReader::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
828 VisitStmt(S);
829 S->setFinallyBody(StmtStack.back());
830 S->setAtFinallyLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
831 return 1;
832}
833
834unsigned PCHStmtReader::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
835 VisitStmt(S);
836 S->setTryBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 3]));
837 S->setCatchStmts(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
838 S->setFinallyStmt(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
839 S->setAtTryLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
840 return 3;
841}
842
843unsigned PCHStmtReader::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
844 VisitStmt(S);
845 S->setSynchExpr(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
846 S->setSynchBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
847 S->setAtSynchronizedLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
848 return 2;
849}
850
851unsigned PCHStmtReader::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
852 VisitStmt(S);
853 S->setThrowExpr(StmtStack.back());
854 S->setThrowLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
855 return 1;
856}
857
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +0000858//===----------------------------------------------------------------------===//
859// C++ Expressions and Statements
860
861unsigned PCHStmtReader::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
862 unsigned num = VisitCallExpr(E);
863 E->setOperator((OverloadedOperatorKind)Record[Idx++]);
864 return num;
865}
866
Douglas Gregor5d3507d2009-09-09 23:08:42 +0000867unsigned PCHStmtReader::VisitCXXConstructExpr(CXXConstructExpr *E) {
868 VisitExpr(E);
869 E->setConstructor(cast<CXXConstructorDecl>(Reader.GetDecl(Record[Idx++])));
Douglas Gregor85dabae2009-12-16 01:38:02 +0000870 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor5d3507d2009-09-09 23:08:42 +0000871 E->setElidable(Record[Idx++]);
Douglas Gregor4f4b1862009-12-16 18:50:27 +0000872 E->setRequiresZeroInitialization(Record[Idx++]);
Douglas Gregor5d3507d2009-09-09 23:08:42 +0000873 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
874 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
875 return E->getNumArgs();
876}
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000877
Sam Weinigd01101e2010-01-16 21:21:01 +0000878unsigned PCHStmtReader::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
879 unsigned num = VisitExplicitCastExpr(E);
880 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
881 return num;
882}
883
884unsigned PCHStmtReader::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
885 return VisitCXXNamedCastExpr(E);
886}
887
888unsigned PCHStmtReader::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
889 return VisitCXXNamedCastExpr(E);
890}
891
892unsigned PCHStmtReader::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
893 return VisitCXXNamedCastExpr(E);
894}
895
896unsigned PCHStmtReader::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
897 return VisitCXXNamedCastExpr(E);
898}
899
900unsigned PCHStmtReader::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
901 unsigned num = VisitExplicitCastExpr(E);
902 E->setTypeBeginLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
903 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
904 return num;
905}
906
Chris Lattnerf4262532009-04-27 05:41:06 +0000907// Within the bitstream, expressions are stored in Reverse Polish
908// Notation, with each of the subexpressions preceding the
909// expression they are stored in. To evaluate expressions, we
910// continue reading expressions and placing them on the stack, with
911// expressions having operands removing those operands from the
912// stack. Evaluation terminates when we see a STMT_STOP record, and
913// the single remaining expression on the stack is our result.
914Stmt *PCHReader::ReadStmt(llvm::BitstreamCursor &Cursor) {
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000915 RecordData Record;
916 unsigned Idx;
917 llvm::SmallVector<Stmt *, 16> StmtStack;
918 PCHStmtReader Reader(*this, Record, Idx, StmtStack);
919 Stmt::EmptyShell Empty;
920
921 while (true) {
Chris Lattnerf4262532009-04-27 05:41:06 +0000922 unsigned Code = Cursor.ReadCode();
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000923 if (Code == llvm::bitc::END_BLOCK) {
Chris Lattnerf4262532009-04-27 05:41:06 +0000924 if (Cursor.ReadBlockEnd()) {
Douglas Gregor6f00bf82009-04-28 21:53:25 +0000925 Error("error at end of block in PCH file");
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000926 return 0;
927 }
928 break;
929 }
930
931 if (Code == llvm::bitc::ENTER_SUBBLOCK) {
932 // No known subblocks, always skip them.
Chris Lattnerf4262532009-04-27 05:41:06 +0000933 Cursor.ReadSubBlockID();
934 if (Cursor.SkipBlock()) {
Douglas Gregor6f00bf82009-04-28 21:53:25 +0000935 Error("malformed block record in PCH file");
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000936 return 0;
937 }
938 continue;
939 }
940
941 if (Code == llvm::bitc::DEFINE_ABBREV) {
Chris Lattnerf4262532009-04-27 05:41:06 +0000942 Cursor.ReadAbbrevRecord();
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000943 continue;
944 }
945
946 Stmt *S = 0;
947 Idx = 0;
948 Record.clear();
949 bool Finished = false;
Chris Lattnerf4262532009-04-27 05:41:06 +0000950 switch ((pch::StmtCode)Cursor.ReadRecord(Code, Record)) {
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000951 case pch::STMT_STOP:
952 Finished = true;
953 break;
954
Mike Stump11289f42009-09-09 15:08:12 +0000955 case pch::STMT_NULL_PTR:
956 S = 0;
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000957 break;
958
959 case pch::STMT_NULL:
960 S = new (Context) NullStmt(Empty);
961 break;
962
963 case pch::STMT_COMPOUND:
964 S = new (Context) CompoundStmt(Empty);
965 break;
966
967 case pch::STMT_CASE:
968 S = new (Context) CaseStmt(Empty);
969 break;
970
971 case pch::STMT_DEFAULT:
972 S = new (Context) DefaultStmt(Empty);
973 break;
974
975 case pch::STMT_LABEL:
976 S = new (Context) LabelStmt(Empty);
977 break;
978
979 case pch::STMT_IF:
980 S = new (Context) IfStmt(Empty);
981 break;
982
983 case pch::STMT_SWITCH:
984 S = new (Context) SwitchStmt(Empty);
985 break;
986
987 case pch::STMT_WHILE:
988 S = new (Context) WhileStmt(Empty);
989 break;
990
991 case pch::STMT_DO:
992 S = new (Context) DoStmt(Empty);
993 break;
Mike Stump11289f42009-09-09 15:08:12 +0000994
Chris Lattner92ba5ff2009-04-27 05:14:47 +0000995 case pch::STMT_FOR:
996 S = new (Context) ForStmt(Empty);
997 break;
998
999 case pch::STMT_GOTO:
1000 S = new (Context) GotoStmt(Empty);
1001 break;
Mike Stump11289f42009-09-09 15:08:12 +00001002
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001003 case pch::STMT_INDIRECT_GOTO:
1004 S = new (Context) IndirectGotoStmt(Empty);
1005 break;
1006
1007 case pch::STMT_CONTINUE:
1008 S = new (Context) ContinueStmt(Empty);
1009 break;
1010
1011 case pch::STMT_BREAK:
1012 S = new (Context) BreakStmt(Empty);
1013 break;
1014
1015 case pch::STMT_RETURN:
1016 S = new (Context) ReturnStmt(Empty);
1017 break;
1018
1019 case pch::STMT_DECL:
1020 S = new (Context) DeclStmt(Empty);
1021 break;
1022
1023 case pch::STMT_ASM:
1024 S = new (Context) AsmStmt(Empty);
1025 break;
1026
1027 case pch::EXPR_PREDEFINED:
1028 S = new (Context) PredefinedExpr(Empty);
1029 break;
Mike Stump11289f42009-09-09 15:08:12 +00001030
1031 case pch::EXPR_DECL_REF:
1032 S = new (Context) DeclRefExpr(Empty);
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001033 break;
Mike Stump11289f42009-09-09 15:08:12 +00001034
1035 case pch::EXPR_INTEGER_LITERAL:
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001036 S = new (Context) IntegerLiteral(Empty);
1037 break;
Mike Stump11289f42009-09-09 15:08:12 +00001038
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001039 case pch::EXPR_FLOATING_LITERAL:
1040 S = new (Context) FloatingLiteral(Empty);
1041 break;
Mike Stump11289f42009-09-09 15:08:12 +00001042
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001043 case pch::EXPR_IMAGINARY_LITERAL:
1044 S = new (Context) ImaginaryLiteral(Empty);
1045 break;
1046
1047 case pch::EXPR_STRING_LITERAL:
Mike Stump11289f42009-09-09 15:08:12 +00001048 S = StringLiteral::CreateEmpty(*Context,
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001049 Record[PCHStmtReader::NumExprFields + 1]);
1050 break;
1051
1052 case pch::EXPR_CHARACTER_LITERAL:
1053 S = new (Context) CharacterLiteral(Empty);
1054 break;
1055
1056 case pch::EXPR_PAREN:
1057 S = new (Context) ParenExpr(Empty);
1058 break;
1059
1060 case pch::EXPR_UNARY_OPERATOR:
1061 S = new (Context) UnaryOperator(Empty);
1062 break;
1063
1064 case pch::EXPR_SIZEOF_ALIGN_OF:
1065 S = new (Context) SizeOfAlignOfExpr(Empty);
1066 break;
1067
1068 case pch::EXPR_ARRAY_SUBSCRIPT:
1069 S = new (Context) ArraySubscriptExpr(Empty);
1070 break;
1071
1072 case pch::EXPR_CALL:
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +00001073 S = new (Context) CallExpr(*Context, Stmt::CallExprClass, Empty);
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001074 break;
1075
1076 case pch::EXPR_MEMBER:
1077 S = new (Context) MemberExpr(Empty);
1078 break;
1079
1080 case pch::EXPR_BINARY_OPERATOR:
1081 S = new (Context) BinaryOperator(Empty);
1082 break;
1083
1084 case pch::EXPR_COMPOUND_ASSIGN_OPERATOR:
1085 S = new (Context) CompoundAssignOperator(Empty);
1086 break;
1087
1088 case pch::EXPR_CONDITIONAL_OPERATOR:
1089 S = new (Context) ConditionalOperator(Empty);
1090 break;
1091
1092 case pch::EXPR_IMPLICIT_CAST:
1093 S = new (Context) ImplicitCastExpr(Empty);
1094 break;
1095
1096 case pch::EXPR_CSTYLE_CAST:
1097 S = new (Context) CStyleCastExpr(Empty);
1098 break;
1099
1100 case pch::EXPR_COMPOUND_LITERAL:
1101 S = new (Context) CompoundLiteralExpr(Empty);
1102 break;
1103
1104 case pch::EXPR_EXT_VECTOR_ELEMENT:
1105 S = new (Context) ExtVectorElementExpr(Empty);
1106 break;
1107
1108 case pch::EXPR_INIT_LIST:
1109 S = new (Context) InitListExpr(Empty);
1110 break;
1111
1112 case pch::EXPR_DESIGNATED_INIT:
Chris Lattner8575daa2009-04-27 21:45:14 +00001113 S = DesignatedInitExpr::CreateEmpty(*Context,
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001114 Record[PCHStmtReader::NumExprFields] - 1);
Mike Stump11289f42009-09-09 15:08:12 +00001115
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001116 break;
1117
1118 case pch::EXPR_IMPLICIT_VALUE_INIT:
1119 S = new (Context) ImplicitValueInitExpr(Empty);
1120 break;
1121
1122 case pch::EXPR_VA_ARG:
1123 S = new (Context) VAArgExpr(Empty);
1124 break;
1125
1126 case pch::EXPR_ADDR_LABEL:
1127 S = new (Context) AddrLabelExpr(Empty);
1128 break;
1129
1130 case pch::EXPR_STMT:
1131 S = new (Context) StmtExpr(Empty);
1132 break;
1133
1134 case pch::EXPR_TYPES_COMPATIBLE:
1135 S = new (Context) TypesCompatibleExpr(Empty);
1136 break;
1137
1138 case pch::EXPR_CHOOSE:
1139 S = new (Context) ChooseExpr(Empty);
1140 break;
1141
1142 case pch::EXPR_GNU_NULL:
1143 S = new (Context) GNUNullExpr(Empty);
1144 break;
1145
1146 case pch::EXPR_SHUFFLE_VECTOR:
1147 S = new (Context) ShuffleVectorExpr(Empty);
1148 break;
Mike Stump11289f42009-09-09 15:08:12 +00001149
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001150 case pch::EXPR_BLOCK:
1151 S = new (Context) BlockExpr(Empty);
1152 break;
1153
1154 case pch::EXPR_BLOCK_DECL_REF:
1155 S = new (Context) BlockDeclRefExpr(Empty);
1156 break;
Mike Stump11289f42009-09-09 15:08:12 +00001157
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001158 case pch::EXPR_OBJC_STRING_LITERAL:
1159 S = new (Context) ObjCStringLiteral(Empty);
1160 break;
1161 case pch::EXPR_OBJC_ENCODE:
1162 S = new (Context) ObjCEncodeExpr(Empty);
1163 break;
1164 case pch::EXPR_OBJC_SELECTOR_EXPR:
1165 S = new (Context) ObjCSelectorExpr(Empty);
1166 break;
1167 case pch::EXPR_OBJC_PROTOCOL_EXPR:
1168 S = new (Context) ObjCProtocolExpr(Empty);
1169 break;
1170 case pch::EXPR_OBJC_IVAR_REF_EXPR:
1171 S = new (Context) ObjCIvarRefExpr(Empty);
1172 break;
1173 case pch::EXPR_OBJC_PROPERTY_REF_EXPR:
1174 S = new (Context) ObjCPropertyRefExpr(Empty);
1175 break;
1176 case pch::EXPR_OBJC_KVC_REF_EXPR:
Fariborz Jahanian9a846652009-08-20 17:02:02 +00001177 S = new (Context) ObjCImplicitSetterGetterRefExpr(Empty);
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001178 break;
1179 case pch::EXPR_OBJC_MESSAGE_EXPR:
1180 S = new (Context) ObjCMessageExpr(Empty);
1181 break;
1182 case pch::EXPR_OBJC_SUPER_EXPR:
1183 S = new (Context) ObjCSuperExpr(Empty);
1184 break;
Steve Naroffe87026a2009-07-24 17:54:45 +00001185 case pch::EXPR_OBJC_ISA:
1186 S = new (Context) ObjCIsaExpr(Empty);
1187 break;
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001188 case pch::STMT_OBJC_FOR_COLLECTION:
1189 S = new (Context) ObjCForCollectionStmt(Empty);
1190 break;
1191 case pch::STMT_OBJC_CATCH:
1192 S = new (Context) ObjCAtCatchStmt(Empty);
1193 break;
1194 case pch::STMT_OBJC_FINALLY:
1195 S = new (Context) ObjCAtFinallyStmt(Empty);
1196 break;
1197 case pch::STMT_OBJC_AT_TRY:
1198 S = new (Context) ObjCAtTryStmt(Empty);
1199 break;
1200 case pch::STMT_OBJC_AT_SYNCHRONIZED:
1201 S = new (Context) ObjCAtSynchronizedStmt(Empty);
1202 break;
1203 case pch::STMT_OBJC_AT_THROW:
1204 S = new (Context) ObjCAtThrowStmt(Empty);
1205 break;
Mike Stump11289f42009-09-09 15:08:12 +00001206
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +00001207 case pch::EXPR_CXX_OPERATOR_CALL:
1208 S = new (Context) CXXOperatorCallExpr(*Context, Empty);
1209 break;
Douglas Gregor5d3507d2009-09-09 23:08:42 +00001210
1211 case pch::EXPR_CXX_CONSTRUCT:
1212 S = new (Context) CXXConstructExpr(Empty, *Context,
1213 Record[PCHStmtReader::NumExprFields + 2]);
1214 break;
Sam Weinigd01101e2010-01-16 21:21:01 +00001215
1216 case pch::EXPR_CXX_STATIC_CAST:
1217 S = new (Context) CXXStaticCastExpr(Empty);
1218 break;
1219
1220 case pch::EXPR_CXX_DYNAMIC_CAST:
1221 S = new (Context) CXXDynamicCastExpr(Empty);
1222 break;
1223
1224 case pch::EXPR_CXX_REINTERPRET_CAST:
1225 S = new (Context) CXXReinterpretCastExpr(Empty);
1226 break;
1227
1228 case pch::EXPR_CXX_CONST_CAST:
1229 S = new (Context) CXXConstCastExpr(Empty);
1230 break;
1231
1232 case pch::EXPR_CXX_FUNCTIONAL_CAST:
1233 S = new (Context) CXXFunctionalCastExpr(Empty);
1234 break;
1235
1236
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001237 }
1238
1239 // We hit a STMT_STOP, so we're done with this expression.
1240 if (Finished)
1241 break;
1242
1243 ++NumStatementsRead;
1244
1245 if (S) {
1246 unsigned NumSubStmts = Reader.Visit(S);
1247 while (NumSubStmts > 0) {
1248 StmtStack.pop_back();
1249 --NumSubStmts;
1250 }
1251 }
1252
1253 assert(Idx == Record.size() && "Invalid deserialization of statement");
1254 StmtStack.push_back(S);
1255 }
1256 assert(StmtStack.size() == 1 && "Extra expressions on stack!");
Chris Lattner92ba5ff2009-04-27 05:14:47 +00001257 return StmtStack.back();
1258}