blob: 67b7c1f4408c1441fb8a3d032eb59d5ce7534313 [file] [log] [blame]
Chris Lattner4c6f9522009-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 Gregor39da0b82009-09-09 23:08:42 +000016#include "clang/AST/DeclCXX.h"
Chris Lattner4c6f9522009-04-27 05:14:47 +000017#include "clang/AST/StmtVisitor.h"
18using namespace clang;
19
Chris Lattner4c6f9522009-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 Jahanian09105f52009-08-20 17:02:02 +0000105 unsigned VisitObjCImplicitSetterGetterRefExpr(
106 ObjCImplicitSetterGetterRefExpr *E);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000107 unsigned VisitObjCMessageExpr(ObjCMessageExpr *E);
108 unsigned VisitObjCSuperExpr(ObjCSuperExpr *E);
Steve Narofff242b1b2009-07-24 17:54:45 +0000109 unsigned VisitObjCIsaExpr(ObjCIsaExpr *E);
Mike Stump1eb44332009-09-09 15:08:12 +0000110
Chris Lattner4c6f9522009-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 Kyrtzidisba0a9002009-07-14 03:19:21 +0000117
118 unsigned VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000119 unsigned VisitCXXConstructExpr(CXXConstructExpr *E);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000120 };
121}
122
123unsigned PCHStmtReader::VisitStmt(Stmt *S) {
124 assert(Idx == NumStmtFields && "Incorrect statement field count");
125 return 0;
126}
127
128unsigned PCHStmtReader::VisitNullStmt(NullStmt *S) {
129 VisitStmt(S);
130 S->setSemiLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
131 return 0;
132}
133
134unsigned PCHStmtReader::VisitCompoundStmt(CompoundStmt *S) {
135 VisitStmt(S);
136 unsigned NumStmts = Record[Idx++];
Mike Stump1eb44332009-09-09 15:08:12 +0000137 S->setStmts(*Reader.getContext(),
Jay Foadbeaaccd2009-05-21 09:52:38 +0000138 StmtStack.data() + StmtStack.size() - NumStmts, NumStmts);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000139 S->setLBracLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
140 S->setRBracLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
141 return NumStmts;
142}
143
144unsigned PCHStmtReader::VisitSwitchCase(SwitchCase *S) {
145 VisitStmt(S);
146 Reader.RecordSwitchCaseID(S, Record[Idx++]);
147 return 0;
148}
149
150unsigned PCHStmtReader::VisitCaseStmt(CaseStmt *S) {
151 VisitSwitchCase(S);
152 S->setLHS(cast<Expr>(StmtStack[StmtStack.size() - 3]));
153 S->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
154 S->setSubStmt(StmtStack.back());
155 S->setCaseLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregordbb26db2009-05-15 23:57:33 +0000156 S->setEllipsisLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
157 S->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000158 return 3;
159}
160
161unsigned PCHStmtReader::VisitDefaultStmt(DefaultStmt *S) {
162 VisitSwitchCase(S);
163 S->setSubStmt(StmtStack.back());
164 S->setDefaultLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregordbb26db2009-05-15 23:57:33 +0000165 S->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000166 return 1;
167}
168
169unsigned PCHStmtReader::VisitLabelStmt(LabelStmt *S) {
170 VisitStmt(S);
171 S->setID(Reader.GetIdentifierInfo(Record, Idx));
172 S->setSubStmt(StmtStack.back());
173 S->setIdentLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
174 Reader.RecordLabelStmt(S, Record[Idx++]);
175 return 1;
176}
177
178unsigned PCHStmtReader::VisitIfStmt(IfStmt *S) {
179 VisitStmt(S);
180 S->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
181 S->setThen(StmtStack[StmtStack.size() - 2]);
182 S->setElse(StmtStack[StmtStack.size() - 1]);
183 S->setIfLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregord06f6ca2009-05-15 18:53:42 +0000184 S->setElseLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000185 return 3;
186}
187
188unsigned PCHStmtReader::VisitSwitchStmt(SwitchStmt *S) {
189 VisitStmt(S);
190 S->setCond(cast<Expr>(StmtStack[StmtStack.size() - 2]));
191 S->setBody(StmtStack.back());
192 S->setSwitchLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
193 SwitchCase *PrevSC = 0;
194 for (unsigned N = Record.size(); Idx != N; ++Idx) {
195 SwitchCase *SC = Reader.getSwitchCaseWithID(Record[Idx]);
196 if (PrevSC)
197 PrevSC->setNextSwitchCase(SC);
198 else
199 S->setSwitchCaseList(SC);
Mike Stump1eb44332009-09-09 15:08:12 +0000200
Douglas Gregor43d9d922009-08-08 01:41:12 +0000201 // Retain this SwitchCase, since SwitchStmt::addSwitchCase() would
202 // normally retain it (but we aren't calling addSwitchCase).
203 SC->Retain();
Chris Lattner4c6f9522009-04-27 05:14:47 +0000204 PrevSC = SC;
205 }
206 return 2;
207}
208
209unsigned PCHStmtReader::VisitWhileStmt(WhileStmt *S) {
210 VisitStmt(S);
211 S->setCond(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
212 S->setBody(StmtStack.back());
213 S->setWhileLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
214 return 2;
215}
216
217unsigned PCHStmtReader::VisitDoStmt(DoStmt *S) {
218 VisitStmt(S);
219 S->setCond(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
220 S->setBody(StmtStack.back());
221 S->setDoLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor9f3ca2a2009-05-15 21:56:04 +0000222 S->setWhileLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner98913592009-06-12 23:04:47 +0000223 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000224 return 2;
225}
226
227unsigned PCHStmtReader::VisitForStmt(ForStmt *S) {
228 VisitStmt(S);
229 S->setInit(StmtStack[StmtStack.size() - 4]);
230 S->setCond(cast_or_null<Expr>(StmtStack[StmtStack.size() - 3]));
231 S->setInc(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
232 S->setBody(StmtStack.back());
233 S->setForLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor5831c6a2009-05-15 22:12:32 +0000234 S->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
235 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000236 return 4;
237}
238
239unsigned PCHStmtReader::VisitGotoStmt(GotoStmt *S) {
240 VisitStmt(S);
241 Reader.SetLabelOf(S, Record[Idx++]);
242 S->setGotoLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
243 S->setLabelLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
244 return 0;
245}
246
247unsigned PCHStmtReader::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
248 VisitStmt(S);
249 S->setGotoLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor5f1b9e62009-05-16 00:20:29 +0000250 S->setStarLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000251 S->setTarget(cast_or_null<Expr>(StmtStack.back()));
252 return 1;
253}
254
255unsigned PCHStmtReader::VisitContinueStmt(ContinueStmt *S) {
256 VisitStmt(S);
257 S->setContinueLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
258 return 0;
259}
260
261unsigned PCHStmtReader::VisitBreakStmt(BreakStmt *S) {
262 VisitStmt(S);
263 S->setBreakLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
264 return 0;
265}
266
267unsigned PCHStmtReader::VisitReturnStmt(ReturnStmt *S) {
268 VisitStmt(S);
269 S->setRetValue(cast_or_null<Expr>(StmtStack.back()));
270 S->setReturnLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
271 return 1;
272}
273
274unsigned PCHStmtReader::VisitDeclStmt(DeclStmt *S) {
275 VisitStmt(S);
276 S->setStartLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
277 S->setEndLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
278
279 if (Idx + 1 == Record.size()) {
280 // Single declaration
281 S->setDeclGroup(DeclGroupRef(Reader.GetDecl(Record[Idx++])));
282 } else {
283 llvm::SmallVector<Decl *, 16> Decls;
284 Decls.reserve(Record.size() - Idx);
285 for (unsigned N = Record.size(); Idx != N; ++Idx)
286 Decls.push_back(Reader.GetDecl(Record[Idx]));
Chris Lattnerd1d64a02009-04-27 21:45:14 +0000287 S->setDeclGroup(DeclGroupRef(DeclGroup::Create(*Reader.getContext(),
Douglas Gregor75fdb232009-05-22 22:45:36 +0000288 Decls.data(),
289 Decls.size())));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000290 }
291 return 0;
292}
293
294unsigned PCHStmtReader::VisitAsmStmt(AsmStmt *S) {
295 VisitStmt(S);
296 unsigned NumOutputs = Record[Idx++];
297 unsigned NumInputs = Record[Idx++];
298 unsigned NumClobbers = Record[Idx++];
299 S->setAsmLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
300 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
301 S->setVolatile(Record[Idx++]);
302 S->setSimple(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000303
304 unsigned StackIdx
Chris Lattner4c6f9522009-04-27 05:14:47 +0000305 = StmtStack.size() - (NumOutputs*2 + NumInputs*2 + NumClobbers + 1);
306 S->setAsmString(cast_or_null<StringLiteral>(StmtStack[StackIdx++]));
307
308 // Outputs and inputs
309 llvm::SmallVector<std::string, 16> Names;
310 llvm::SmallVector<StringLiteral*, 16> Constraints;
311 llvm::SmallVector<Stmt*, 16> Exprs;
312 for (unsigned I = 0, N = NumOutputs + NumInputs; I != N; ++I) {
313 Names.push_back(Reader.ReadString(Record, Idx));
314 Constraints.push_back(cast_or_null<StringLiteral>(StmtStack[StackIdx++]));
315 Exprs.push_back(StmtStack[StackIdx++]);
316 }
317 S->setOutputsAndInputs(NumOutputs, NumInputs,
Jay Foadbeaaccd2009-05-21 09:52:38 +0000318 Names.data(), Constraints.data(), Exprs.data());
Chris Lattner4c6f9522009-04-27 05:14:47 +0000319
320 // Constraints
321 llvm::SmallVector<StringLiteral*, 16> Clobbers;
322 for (unsigned I = 0; I != NumClobbers; ++I)
323 Clobbers.push_back(cast_or_null<StringLiteral>(StmtStack[StackIdx++]));
Jay Foadbeaaccd2009-05-21 09:52:38 +0000324 S->setClobbers(Clobbers.data(), NumClobbers);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000325
326 assert(StackIdx == StmtStack.size() && "Error deserializing AsmStmt");
327 return NumOutputs*2 + NumInputs*2 + NumClobbers + 1;
328}
329
330unsigned PCHStmtReader::VisitExpr(Expr *E) {
331 VisitStmt(E);
332 E->setType(Reader.GetType(Record[Idx++]));
333 E->setTypeDependent(Record[Idx++]);
334 E->setValueDependent(Record[Idx++]);
335 assert(Idx == NumExprFields && "Incorrect expression field count");
336 return 0;
337}
338
339unsigned PCHStmtReader::VisitPredefinedExpr(PredefinedExpr *E) {
340 VisitExpr(E);
341 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
342 E->setIdentType((PredefinedExpr::IdentType)Record[Idx++]);
343 return 0;
344}
345
346unsigned PCHStmtReader::VisitDeclRefExpr(DeclRefExpr *E) {
347 VisitExpr(E);
348 E->setDecl(cast<NamedDecl>(Reader.GetDecl(Record[Idx++])));
349 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
350 return 0;
351}
352
353unsigned PCHStmtReader::VisitIntegerLiteral(IntegerLiteral *E) {
354 VisitExpr(E);
355 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
356 E->setValue(Reader.ReadAPInt(Record, Idx));
357 return 0;
358}
359
360unsigned PCHStmtReader::VisitFloatingLiteral(FloatingLiteral *E) {
361 VisitExpr(E);
362 E->setValue(Reader.ReadAPFloat(Record, Idx));
363 E->setExact(Record[Idx++]);
364 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
365 return 0;
366}
367
368unsigned PCHStmtReader::VisitImaginaryLiteral(ImaginaryLiteral *E) {
369 VisitExpr(E);
370 E->setSubExpr(cast<Expr>(StmtStack.back()));
371 return 1;
372}
373
374unsigned PCHStmtReader::VisitStringLiteral(StringLiteral *E) {
375 VisitExpr(E);
376 unsigned Len = Record[Idx++];
Mike Stump1eb44332009-09-09 15:08:12 +0000377 assert(Record[Idx] == E->getNumConcatenated() &&
Chris Lattner4c6f9522009-04-27 05:14:47 +0000378 "Wrong number of concatenated tokens!");
379 ++Idx;
380 E->setWide(Record[Idx++]);
381
Mike Stump1eb44332009-09-09 15:08:12 +0000382 // Read string data
Chris Lattner4c6f9522009-04-27 05:14:47 +0000383 llvm::SmallVector<char, 16> Str(&Record[Idx], &Record[Idx] + Len);
Douglas Gregor75fdb232009-05-22 22:45:36 +0000384 E->setStrData(*Reader.getContext(), Str.data(), Len);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000385 Idx += Len;
386
387 // Read source locations
388 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
389 E->setStrTokenLoc(I, SourceLocation::getFromRawEncoding(Record[Idx++]));
390
391 return 0;
392}
393
394unsigned PCHStmtReader::VisitCharacterLiteral(CharacterLiteral *E) {
395 VisitExpr(E);
396 E->setValue(Record[Idx++]);
397 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
398 E->setWide(Record[Idx++]);
399 return 0;
400}
401
402unsigned PCHStmtReader::VisitParenExpr(ParenExpr *E) {
403 VisitExpr(E);
404 E->setLParen(SourceLocation::getFromRawEncoding(Record[Idx++]));
405 E->setRParen(SourceLocation::getFromRawEncoding(Record[Idx++]));
406 E->setSubExpr(cast<Expr>(StmtStack.back()));
407 return 1;
408}
409
410unsigned PCHStmtReader::VisitUnaryOperator(UnaryOperator *E) {
411 VisitExpr(E);
412 E->setSubExpr(cast<Expr>(StmtStack.back()));
413 E->setOpcode((UnaryOperator::Opcode)Record[Idx++]);
414 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
415 return 1;
416}
417
418unsigned PCHStmtReader::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) {
419 VisitExpr(E);
420 E->setSizeof(Record[Idx++]);
421 if (Record[Idx] == 0) {
422 E->setArgument(cast<Expr>(StmtStack.back()));
423 ++Idx;
424 } else {
425 E->setArgument(Reader.GetType(Record[Idx++]));
426 }
427 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
428 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
429 return E->isArgumentType()? 0 : 1;
430}
431
432unsigned PCHStmtReader::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
433 VisitExpr(E);
434 E->setLHS(cast<Expr>(StmtStack[StmtStack.size() - 2]));
435 E->setRHS(cast<Expr>(StmtStack[StmtStack.size() - 1]));
436 E->setRBracketLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
437 return 2;
438}
439
440unsigned PCHStmtReader::VisitCallExpr(CallExpr *E) {
441 VisitExpr(E);
Chris Lattnerd1d64a02009-04-27 21:45:14 +0000442 E->setNumArgs(*Reader.getContext(), Record[Idx++]);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000443 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
444 E->setCallee(cast<Expr>(StmtStack[StmtStack.size() - E->getNumArgs() - 1]));
445 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
446 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
447 return E->getNumArgs() + 1;
448}
449
450unsigned PCHStmtReader::VisitMemberExpr(MemberExpr *E) {
451 VisitExpr(E);
452 E->setBase(cast<Expr>(StmtStack.back()));
453 E->setMemberDecl(cast<NamedDecl>(Reader.GetDecl(Record[Idx++])));
454 E->setMemberLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
455 E->setArrow(Record[Idx++]);
456 return 1;
457}
458
Steve Narofff242b1b2009-07-24 17:54:45 +0000459unsigned PCHStmtReader::VisitObjCIsaExpr(ObjCIsaExpr *E) {
460 VisitExpr(E);
461 E->setBase(cast<Expr>(StmtStack.back()));
462 E->setIsaMemberLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
463 E->setArrow(Record[Idx++]);
464 return 1;
465}
466
Chris Lattner4c6f9522009-04-27 05:14:47 +0000467unsigned PCHStmtReader::VisitCastExpr(CastExpr *E) {
468 VisitExpr(E);
469 E->setSubExpr(cast<Expr>(StmtStack.back()));
Anders Carlssoncdef2b72009-07-31 00:48:10 +0000470 E->setCastKind((CastExpr::CastKind)Record[Idx++]);
471
Chris Lattner4c6f9522009-04-27 05:14:47 +0000472 return 1;
473}
474
475unsigned PCHStmtReader::VisitBinaryOperator(BinaryOperator *E) {
476 VisitExpr(E);
477 E->setLHS(cast<Expr>(StmtStack.end()[-2]));
478 E->setRHS(cast<Expr>(StmtStack.end()[-1]));
479 E->setOpcode((BinaryOperator::Opcode)Record[Idx++]);
480 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
481 return 2;
482}
483
484unsigned PCHStmtReader::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
485 VisitBinaryOperator(E);
486 E->setComputationLHSType(Reader.GetType(Record[Idx++]));
487 E->setComputationResultType(Reader.GetType(Record[Idx++]));
488 return 2;
489}
490
491unsigned PCHStmtReader::VisitConditionalOperator(ConditionalOperator *E) {
492 VisitExpr(E);
493 E->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
494 E->setLHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
495 E->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 1]));
Douglas Gregor47e1f7c2009-08-26 14:37:04 +0000496 E->setQuestionLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
497 E->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000498 return 3;
499}
500
501unsigned PCHStmtReader::VisitImplicitCastExpr(ImplicitCastExpr *E) {
502 VisitCastExpr(E);
503 E->setLvalueCast(Record[Idx++]);
504 return 1;
505}
506
507unsigned PCHStmtReader::VisitExplicitCastExpr(ExplicitCastExpr *E) {
508 VisitCastExpr(E);
509 E->setTypeAsWritten(Reader.GetType(Record[Idx++]));
510 return 1;
511}
512
513unsigned PCHStmtReader::VisitCStyleCastExpr(CStyleCastExpr *E) {
514 VisitExplicitCastExpr(E);
515 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
516 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
517 return 1;
518}
519
520unsigned PCHStmtReader::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
521 VisitExpr(E);
522 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
523 E->setInitializer(cast<Expr>(StmtStack.back()));
524 E->setFileScope(Record[Idx++]);
525 return 1;
526}
527
528unsigned PCHStmtReader::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
529 VisitExpr(E);
530 E->setBase(cast<Expr>(StmtStack.back()));
531 E->setAccessor(Reader.GetIdentifierInfo(Record, Idx));
532 E->setAccessorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
533 return 1;
534}
535
536unsigned PCHStmtReader::VisitInitListExpr(InitListExpr *E) {
537 VisitExpr(E);
538 unsigned NumInits = Record[Idx++];
539 E->reserveInits(NumInits);
540 for (unsigned I = 0; I != NumInits; ++I)
Mike Stump1eb44332009-09-09 15:08:12 +0000541 E->updateInit(I,
Chris Lattner4c6f9522009-04-27 05:14:47 +0000542 cast<Expr>(StmtStack[StmtStack.size() - NumInits - 1 + I]));
543 E->setSyntacticForm(cast_or_null<InitListExpr>(StmtStack.back()));
544 E->setLBraceLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
545 E->setRBraceLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
546 E->setInitializedFieldInUnion(
547 cast_or_null<FieldDecl>(Reader.GetDecl(Record[Idx++])));
548 E->sawArrayRangeDesignator(Record[Idx++]);
549 return NumInits + 1;
550}
551
552unsigned PCHStmtReader::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
553 typedef DesignatedInitExpr::Designator Designator;
554
555 VisitExpr(E);
556 unsigned NumSubExprs = Record[Idx++];
557 assert(NumSubExprs == E->getNumSubExprs() && "Wrong number of subexprs");
558 for (unsigned I = 0; I != NumSubExprs; ++I)
559 E->setSubExpr(I, cast<Expr>(StmtStack[StmtStack.size() - NumSubExprs + I]));
560 E->setEqualOrColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
561 E->setGNUSyntax(Record[Idx++]);
562
563 llvm::SmallVector<Designator, 4> Designators;
564 while (Idx < Record.size()) {
565 switch ((pch::DesignatorTypes)Record[Idx++]) {
566 case pch::DESIG_FIELD_DECL: {
567 FieldDecl *Field = cast<FieldDecl>(Reader.GetDecl(Record[Idx++]));
Mike Stump1eb44332009-09-09 15:08:12 +0000568 SourceLocation DotLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000569 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000570 SourceLocation FieldLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000571 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000572 Designators.push_back(Designator(Field->getIdentifier(), DotLoc,
Chris Lattner4c6f9522009-04-27 05:14:47 +0000573 FieldLoc));
574 Designators.back().setField(Field);
575 break;
576 }
577
578 case pch::DESIG_FIELD_NAME: {
579 const IdentifierInfo *Name = Reader.GetIdentifierInfo(Record, Idx);
Mike Stump1eb44332009-09-09 15:08:12 +0000580 SourceLocation DotLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000581 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000582 SourceLocation FieldLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000583 = SourceLocation::getFromRawEncoding(Record[Idx++]);
584 Designators.push_back(Designator(Name, DotLoc, FieldLoc));
585 break;
586 }
Mike Stump1eb44332009-09-09 15:08:12 +0000587
Chris Lattner4c6f9522009-04-27 05:14:47 +0000588 case pch::DESIG_ARRAY: {
589 unsigned Index = Record[Idx++];
590 SourceLocation LBracketLoc
591 = SourceLocation::getFromRawEncoding(Record[Idx++]);
592 SourceLocation RBracketLoc
593 = SourceLocation::getFromRawEncoding(Record[Idx++]);
594 Designators.push_back(Designator(Index, LBracketLoc, RBracketLoc));
595 break;
596 }
597
598 case pch::DESIG_ARRAY_RANGE: {
599 unsigned Index = Record[Idx++];
600 SourceLocation LBracketLoc
601 = SourceLocation::getFromRawEncoding(Record[Idx++]);
602 SourceLocation EllipsisLoc
603 = SourceLocation::getFromRawEncoding(Record[Idx++]);
604 SourceLocation RBracketLoc
605 = SourceLocation::getFromRawEncoding(Record[Idx++]);
606 Designators.push_back(Designator(Index, LBracketLoc, EllipsisLoc,
607 RBracketLoc));
608 break;
609 }
610 }
611 }
Douglas Gregor75fdb232009-05-22 22:45:36 +0000612 E->setDesignators(Designators.data(), Designators.size());
Chris Lattner4c6f9522009-04-27 05:14:47 +0000613
614 return NumSubExprs;
615}
616
617unsigned PCHStmtReader::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
618 VisitExpr(E);
619 return 0;
620}
621
622unsigned PCHStmtReader::VisitVAArgExpr(VAArgExpr *E) {
623 VisitExpr(E);
624 E->setSubExpr(cast<Expr>(StmtStack.back()));
625 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
626 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
627 return 1;
628}
629
630unsigned PCHStmtReader::VisitAddrLabelExpr(AddrLabelExpr *E) {
631 VisitExpr(E);
632 E->setAmpAmpLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
633 E->setLabelLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
634 Reader.SetLabelOf(E, Record[Idx++]);
635 return 0;
636}
637
638unsigned PCHStmtReader::VisitStmtExpr(StmtExpr *E) {
639 VisitExpr(E);
640 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
641 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
642 E->setSubStmt(cast_or_null<CompoundStmt>(StmtStack.back()));
643 return 1;
644}
645
646unsigned PCHStmtReader::VisitTypesCompatibleExpr(TypesCompatibleExpr *E) {
647 VisitExpr(E);
648 E->setArgType1(Reader.GetType(Record[Idx++]));
649 E->setArgType2(Reader.GetType(Record[Idx++]));
650 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
651 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
652 return 0;
653}
654
655unsigned PCHStmtReader::VisitChooseExpr(ChooseExpr *E) {
656 VisitExpr(E);
657 E->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
658 E->setLHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
659 E->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 1]));
660 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
661 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
662 return 3;
663}
664
665unsigned PCHStmtReader::VisitGNUNullExpr(GNUNullExpr *E) {
666 VisitExpr(E);
667 E->setTokenLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
668 return 0;
669}
670
671unsigned PCHStmtReader::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
672 VisitExpr(E);
673 unsigned NumExprs = Record[Idx++];
Mike Stump1eb44332009-09-09 15:08:12 +0000674 E->setExprs(*Reader.getContext(),
Nate Begeman888376a2009-08-12 02:28:50 +0000675 (Expr **)&StmtStack[StmtStack.size() - NumExprs], NumExprs);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000676 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
677 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
678 return NumExprs;
679}
680
681unsigned PCHStmtReader::VisitBlockExpr(BlockExpr *E) {
682 VisitExpr(E);
683 E->setBlockDecl(cast_or_null<BlockDecl>(Reader.GetDecl(Record[Idx++])));
684 E->setHasBlockDeclRefExprs(Record[Idx++]);
685 return 0;
686}
687
688unsigned PCHStmtReader::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
689 VisitExpr(E);
690 E->setDecl(cast<ValueDecl>(Reader.GetDecl(Record[Idx++])));
691 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
692 E->setByRef(Record[Idx++]);
Fariborz Jahanian9b0b57c2009-06-20 00:02:26 +0000693 E->setConstQualAdded(Record[Idx++]);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000694 return 0;
695}
696
697//===----------------------------------------------------------------------===//
698// Objective-C Expressions and Statements
699
700unsigned PCHStmtReader::VisitObjCStringLiteral(ObjCStringLiteral *E) {
701 VisitExpr(E);
702 E->setString(cast<StringLiteral>(StmtStack.back()));
703 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
704 return 1;
705}
706
707unsigned PCHStmtReader::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
708 VisitExpr(E);
709 E->setEncodedType(Reader.GetType(Record[Idx++]));
710 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
711 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
712 return 0;
713}
714
715unsigned PCHStmtReader::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
716 VisitExpr(E);
717 E->setSelector(Reader.GetSelector(Record, Idx));
718 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
719 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
720 return 0;
721}
722
723unsigned PCHStmtReader::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
724 VisitExpr(E);
725 E->setProtocol(cast<ObjCProtocolDecl>(Reader.GetDecl(Record[Idx++])));
726 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
727 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
728 return 0;
729}
730
731unsigned PCHStmtReader::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
732 VisitExpr(E);
733 E->setDecl(cast<ObjCIvarDecl>(Reader.GetDecl(Record[Idx++])));
734 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
735 E->setBase(cast<Expr>(StmtStack.back()));
736 E->setIsArrow(Record[Idx++]);
737 E->setIsFreeIvar(Record[Idx++]);
738 return 1;
739}
740
741unsigned PCHStmtReader::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
742 VisitExpr(E);
743 E->setProperty(cast<ObjCPropertyDecl>(Reader.GetDecl(Record[Idx++])));
744 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
745 E->setBase(cast<Expr>(StmtStack.back()));
746 return 1;
747}
748
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000749unsigned PCHStmtReader::VisitObjCImplicitSetterGetterRefExpr(
750 ObjCImplicitSetterGetterRefExpr *E) {
Chris Lattner4c6f9522009-04-27 05:14:47 +0000751 VisitExpr(E);
752 E->setGetterMethod(
753 cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
754 E->setSetterMethod(
755 cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
Fariborz Jahaniand2ae5aa2009-08-18 21:37:33 +0000756 E->setInterfaceDecl(
Chris Lattner4c6f9522009-04-27 05:14:47 +0000757 cast_or_null<ObjCInterfaceDecl>(Reader.GetDecl(Record[Idx++])));
758 E->setBase(cast_or_null<Expr>(StmtStack.back()));
759 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
760 E->setClassLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
761 return 1;
762}
763
764unsigned PCHStmtReader::VisitObjCMessageExpr(ObjCMessageExpr *E) {
765 VisitExpr(E);
766 E->setNumArgs(Record[Idx++]);
767 E->setLeftLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
768 E->setRightLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
769 E->setSelector(Reader.GetSelector(Record, Idx));
770 E->setMethodDecl(cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
Mike Stump1eb44332009-09-09 15:08:12 +0000771
Chris Lattner4c6f9522009-04-27 05:14:47 +0000772 E->setReceiver(
773 cast_or_null<Expr>(StmtStack[StmtStack.size() - E->getNumArgs() - 1]));
774 if (!E->getReceiver()) {
775 ObjCMessageExpr::ClassInfo CI;
776 CI.first = cast_or_null<ObjCInterfaceDecl>(Reader.GetDecl(Record[Idx++]));
777 CI.second = Reader.GetIdentifierInfo(Record, Idx);
778 E->setClassInfo(CI);
779 }
780
781 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
782 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
783 return E->getNumArgs() + 1;
784}
785
786unsigned PCHStmtReader::VisitObjCSuperExpr(ObjCSuperExpr *E) {
787 VisitExpr(E);
788 E->setLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
789 return 0;
790}
791
792unsigned PCHStmtReader::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
793 VisitStmt(S);
794 S->setElement(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 3]));
795 S->setCollection(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
796 S->setBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
797 S->setForLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
798 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
799 return 3;
800}
801
802unsigned PCHStmtReader::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
803 VisitStmt(S);
804 S->setCatchBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
805 S->setNextCatchStmt(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
806 S->setCatchParamDecl(cast_or_null<ParmVarDecl>(Reader.GetDecl(Record[Idx++])));
807 S->setAtCatchLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
808 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
809 return 2;
810}
811
812unsigned PCHStmtReader::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
813 VisitStmt(S);
814 S->setFinallyBody(StmtStack.back());
815 S->setAtFinallyLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
816 return 1;
817}
818
819unsigned PCHStmtReader::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
820 VisitStmt(S);
821 S->setTryBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 3]));
822 S->setCatchStmts(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
823 S->setFinallyStmt(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
824 S->setAtTryLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
825 return 3;
826}
827
828unsigned PCHStmtReader::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
829 VisitStmt(S);
830 S->setSynchExpr(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
831 S->setSynchBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
832 S->setAtSynchronizedLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
833 return 2;
834}
835
836unsigned PCHStmtReader::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
837 VisitStmt(S);
838 S->setThrowExpr(StmtStack.back());
839 S->setThrowLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
840 return 1;
841}
842
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000843//===----------------------------------------------------------------------===//
844// C++ Expressions and Statements
845
846unsigned PCHStmtReader::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
847 unsigned num = VisitCallExpr(E);
848 E->setOperator((OverloadedOperatorKind)Record[Idx++]);
849 return num;
850}
851
Douglas Gregor39da0b82009-09-09 23:08:42 +0000852unsigned PCHStmtReader::VisitCXXConstructExpr(CXXConstructExpr *E) {
853 VisitExpr(E);
854 E->setConstructor(cast<CXXConstructorDecl>(Reader.GetDecl(Record[Idx++])));
855 E->setElidable(Record[Idx++]);
856 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
857 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
858 return E->getNumArgs();
859}
Chris Lattner4c6f9522009-04-27 05:14:47 +0000860
Chris Lattner52e97d12009-04-27 05:41:06 +0000861// Within the bitstream, expressions are stored in Reverse Polish
862// Notation, with each of the subexpressions preceding the
863// expression they are stored in. To evaluate expressions, we
864// continue reading expressions and placing them on the stack, with
865// expressions having operands removing those operands from the
866// stack. Evaluation terminates when we see a STMT_STOP record, and
867// the single remaining expression on the stack is our result.
868Stmt *PCHReader::ReadStmt(llvm::BitstreamCursor &Cursor) {
Chris Lattner4c6f9522009-04-27 05:14:47 +0000869 RecordData Record;
870 unsigned Idx;
871 llvm::SmallVector<Stmt *, 16> StmtStack;
872 PCHStmtReader Reader(*this, Record, Idx, StmtStack);
873 Stmt::EmptyShell Empty;
874
875 while (true) {
Chris Lattner52e97d12009-04-27 05:41:06 +0000876 unsigned Code = Cursor.ReadCode();
Chris Lattner4c6f9522009-04-27 05:14:47 +0000877 if (Code == llvm::bitc::END_BLOCK) {
Chris Lattner52e97d12009-04-27 05:41:06 +0000878 if (Cursor.ReadBlockEnd()) {
Douglas Gregora02b1472009-04-28 21:53:25 +0000879 Error("error at end of block in PCH file");
Chris Lattner4c6f9522009-04-27 05:14:47 +0000880 return 0;
881 }
882 break;
883 }
884
885 if (Code == llvm::bitc::ENTER_SUBBLOCK) {
886 // No known subblocks, always skip them.
Chris Lattner52e97d12009-04-27 05:41:06 +0000887 Cursor.ReadSubBlockID();
888 if (Cursor.SkipBlock()) {
Douglas Gregora02b1472009-04-28 21:53:25 +0000889 Error("malformed block record in PCH file");
Chris Lattner4c6f9522009-04-27 05:14:47 +0000890 return 0;
891 }
892 continue;
893 }
894
895 if (Code == llvm::bitc::DEFINE_ABBREV) {
Chris Lattner52e97d12009-04-27 05:41:06 +0000896 Cursor.ReadAbbrevRecord();
Chris Lattner4c6f9522009-04-27 05:14:47 +0000897 continue;
898 }
899
900 Stmt *S = 0;
901 Idx = 0;
902 Record.clear();
903 bool Finished = false;
Chris Lattner52e97d12009-04-27 05:41:06 +0000904 switch ((pch::StmtCode)Cursor.ReadRecord(Code, Record)) {
Chris Lattner4c6f9522009-04-27 05:14:47 +0000905 case pch::STMT_STOP:
906 Finished = true;
907 break;
908
Mike Stump1eb44332009-09-09 15:08:12 +0000909 case pch::STMT_NULL_PTR:
910 S = 0;
Chris Lattner4c6f9522009-04-27 05:14:47 +0000911 break;
912
913 case pch::STMT_NULL:
914 S = new (Context) NullStmt(Empty);
915 break;
916
917 case pch::STMT_COMPOUND:
918 S = new (Context) CompoundStmt(Empty);
919 break;
920
921 case pch::STMT_CASE:
922 S = new (Context) CaseStmt(Empty);
923 break;
924
925 case pch::STMT_DEFAULT:
926 S = new (Context) DefaultStmt(Empty);
927 break;
928
929 case pch::STMT_LABEL:
930 S = new (Context) LabelStmt(Empty);
931 break;
932
933 case pch::STMT_IF:
934 S = new (Context) IfStmt(Empty);
935 break;
936
937 case pch::STMT_SWITCH:
938 S = new (Context) SwitchStmt(Empty);
939 break;
940
941 case pch::STMT_WHILE:
942 S = new (Context) WhileStmt(Empty);
943 break;
944
945 case pch::STMT_DO:
946 S = new (Context) DoStmt(Empty);
947 break;
Mike Stump1eb44332009-09-09 15:08:12 +0000948
Chris Lattner4c6f9522009-04-27 05:14:47 +0000949 case pch::STMT_FOR:
950 S = new (Context) ForStmt(Empty);
951 break;
952
953 case pch::STMT_GOTO:
954 S = new (Context) GotoStmt(Empty);
955 break;
Mike Stump1eb44332009-09-09 15:08:12 +0000956
Chris Lattner4c6f9522009-04-27 05:14:47 +0000957 case pch::STMT_INDIRECT_GOTO:
958 S = new (Context) IndirectGotoStmt(Empty);
959 break;
960
961 case pch::STMT_CONTINUE:
962 S = new (Context) ContinueStmt(Empty);
963 break;
964
965 case pch::STMT_BREAK:
966 S = new (Context) BreakStmt(Empty);
967 break;
968
969 case pch::STMT_RETURN:
970 S = new (Context) ReturnStmt(Empty);
971 break;
972
973 case pch::STMT_DECL:
974 S = new (Context) DeclStmt(Empty);
975 break;
976
977 case pch::STMT_ASM:
978 S = new (Context) AsmStmt(Empty);
979 break;
980
981 case pch::EXPR_PREDEFINED:
982 S = new (Context) PredefinedExpr(Empty);
983 break;
Mike Stump1eb44332009-09-09 15:08:12 +0000984
985 case pch::EXPR_DECL_REF:
986 S = new (Context) DeclRefExpr(Empty);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000987 break;
Mike Stump1eb44332009-09-09 15:08:12 +0000988
989 case pch::EXPR_INTEGER_LITERAL:
Chris Lattner4c6f9522009-04-27 05:14:47 +0000990 S = new (Context) IntegerLiteral(Empty);
991 break;
Mike Stump1eb44332009-09-09 15:08:12 +0000992
Chris Lattner4c6f9522009-04-27 05:14:47 +0000993 case pch::EXPR_FLOATING_LITERAL:
994 S = new (Context) FloatingLiteral(Empty);
995 break;
Mike Stump1eb44332009-09-09 15:08:12 +0000996
Chris Lattner4c6f9522009-04-27 05:14:47 +0000997 case pch::EXPR_IMAGINARY_LITERAL:
998 S = new (Context) ImaginaryLiteral(Empty);
999 break;
1000
1001 case pch::EXPR_STRING_LITERAL:
Mike Stump1eb44332009-09-09 15:08:12 +00001002 S = StringLiteral::CreateEmpty(*Context,
Chris Lattner4c6f9522009-04-27 05:14:47 +00001003 Record[PCHStmtReader::NumExprFields + 1]);
1004 break;
1005
1006 case pch::EXPR_CHARACTER_LITERAL:
1007 S = new (Context) CharacterLiteral(Empty);
1008 break;
1009
1010 case pch::EXPR_PAREN:
1011 S = new (Context) ParenExpr(Empty);
1012 break;
1013
1014 case pch::EXPR_UNARY_OPERATOR:
1015 S = new (Context) UnaryOperator(Empty);
1016 break;
1017
1018 case pch::EXPR_SIZEOF_ALIGN_OF:
1019 S = new (Context) SizeOfAlignOfExpr(Empty);
1020 break;
1021
1022 case pch::EXPR_ARRAY_SUBSCRIPT:
1023 S = new (Context) ArraySubscriptExpr(Empty);
1024 break;
1025
1026 case pch::EXPR_CALL:
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001027 S = new (Context) CallExpr(*Context, Stmt::CallExprClass, Empty);
Chris Lattner4c6f9522009-04-27 05:14:47 +00001028 break;
1029
1030 case pch::EXPR_MEMBER:
1031 S = new (Context) MemberExpr(Empty);
1032 break;
1033
1034 case pch::EXPR_BINARY_OPERATOR:
1035 S = new (Context) BinaryOperator(Empty);
1036 break;
1037
1038 case pch::EXPR_COMPOUND_ASSIGN_OPERATOR:
1039 S = new (Context) CompoundAssignOperator(Empty);
1040 break;
1041
1042 case pch::EXPR_CONDITIONAL_OPERATOR:
1043 S = new (Context) ConditionalOperator(Empty);
1044 break;
1045
1046 case pch::EXPR_IMPLICIT_CAST:
1047 S = new (Context) ImplicitCastExpr(Empty);
1048 break;
1049
1050 case pch::EXPR_CSTYLE_CAST:
1051 S = new (Context) CStyleCastExpr(Empty);
1052 break;
1053
1054 case pch::EXPR_COMPOUND_LITERAL:
1055 S = new (Context) CompoundLiteralExpr(Empty);
1056 break;
1057
1058 case pch::EXPR_EXT_VECTOR_ELEMENT:
1059 S = new (Context) ExtVectorElementExpr(Empty);
1060 break;
1061
1062 case pch::EXPR_INIT_LIST:
1063 S = new (Context) InitListExpr(Empty);
1064 break;
1065
1066 case pch::EXPR_DESIGNATED_INIT:
Chris Lattnerd1d64a02009-04-27 21:45:14 +00001067 S = DesignatedInitExpr::CreateEmpty(*Context,
Chris Lattner4c6f9522009-04-27 05:14:47 +00001068 Record[PCHStmtReader::NumExprFields] - 1);
Mike Stump1eb44332009-09-09 15:08:12 +00001069
Chris Lattner4c6f9522009-04-27 05:14:47 +00001070 break;
1071
1072 case pch::EXPR_IMPLICIT_VALUE_INIT:
1073 S = new (Context) ImplicitValueInitExpr(Empty);
1074 break;
1075
1076 case pch::EXPR_VA_ARG:
1077 S = new (Context) VAArgExpr(Empty);
1078 break;
1079
1080 case pch::EXPR_ADDR_LABEL:
1081 S = new (Context) AddrLabelExpr(Empty);
1082 break;
1083
1084 case pch::EXPR_STMT:
1085 S = new (Context) StmtExpr(Empty);
1086 break;
1087
1088 case pch::EXPR_TYPES_COMPATIBLE:
1089 S = new (Context) TypesCompatibleExpr(Empty);
1090 break;
1091
1092 case pch::EXPR_CHOOSE:
1093 S = new (Context) ChooseExpr(Empty);
1094 break;
1095
1096 case pch::EXPR_GNU_NULL:
1097 S = new (Context) GNUNullExpr(Empty);
1098 break;
1099
1100 case pch::EXPR_SHUFFLE_VECTOR:
1101 S = new (Context) ShuffleVectorExpr(Empty);
1102 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001103
Chris Lattner4c6f9522009-04-27 05:14:47 +00001104 case pch::EXPR_BLOCK:
1105 S = new (Context) BlockExpr(Empty);
1106 break;
1107
1108 case pch::EXPR_BLOCK_DECL_REF:
1109 S = new (Context) BlockDeclRefExpr(Empty);
1110 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001111
Chris Lattner4c6f9522009-04-27 05:14:47 +00001112 case pch::EXPR_OBJC_STRING_LITERAL:
1113 S = new (Context) ObjCStringLiteral(Empty);
1114 break;
1115 case pch::EXPR_OBJC_ENCODE:
1116 S = new (Context) ObjCEncodeExpr(Empty);
1117 break;
1118 case pch::EXPR_OBJC_SELECTOR_EXPR:
1119 S = new (Context) ObjCSelectorExpr(Empty);
1120 break;
1121 case pch::EXPR_OBJC_PROTOCOL_EXPR:
1122 S = new (Context) ObjCProtocolExpr(Empty);
1123 break;
1124 case pch::EXPR_OBJC_IVAR_REF_EXPR:
1125 S = new (Context) ObjCIvarRefExpr(Empty);
1126 break;
1127 case pch::EXPR_OBJC_PROPERTY_REF_EXPR:
1128 S = new (Context) ObjCPropertyRefExpr(Empty);
1129 break;
1130 case pch::EXPR_OBJC_KVC_REF_EXPR:
Fariborz Jahanian09105f52009-08-20 17:02:02 +00001131 S = new (Context) ObjCImplicitSetterGetterRefExpr(Empty);
Chris Lattner4c6f9522009-04-27 05:14:47 +00001132 break;
1133 case pch::EXPR_OBJC_MESSAGE_EXPR:
1134 S = new (Context) ObjCMessageExpr(Empty);
1135 break;
1136 case pch::EXPR_OBJC_SUPER_EXPR:
1137 S = new (Context) ObjCSuperExpr(Empty);
1138 break;
Steve Narofff242b1b2009-07-24 17:54:45 +00001139 case pch::EXPR_OBJC_ISA:
1140 S = new (Context) ObjCIsaExpr(Empty);
1141 break;
Chris Lattner4c6f9522009-04-27 05:14:47 +00001142 case pch::STMT_OBJC_FOR_COLLECTION:
1143 S = new (Context) ObjCForCollectionStmt(Empty);
1144 break;
1145 case pch::STMT_OBJC_CATCH:
1146 S = new (Context) ObjCAtCatchStmt(Empty);
1147 break;
1148 case pch::STMT_OBJC_FINALLY:
1149 S = new (Context) ObjCAtFinallyStmt(Empty);
1150 break;
1151 case pch::STMT_OBJC_AT_TRY:
1152 S = new (Context) ObjCAtTryStmt(Empty);
1153 break;
1154 case pch::STMT_OBJC_AT_SYNCHRONIZED:
1155 S = new (Context) ObjCAtSynchronizedStmt(Empty);
1156 break;
1157 case pch::STMT_OBJC_AT_THROW:
1158 S = new (Context) ObjCAtThrowStmt(Empty);
1159 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001160
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001161 case pch::EXPR_CXX_OPERATOR_CALL:
1162 S = new (Context) CXXOperatorCallExpr(*Context, Empty);
1163 break;
Douglas Gregor39da0b82009-09-09 23:08:42 +00001164
1165 case pch::EXPR_CXX_CONSTRUCT:
1166 S = new (Context) CXXConstructExpr(Empty, *Context,
1167 Record[PCHStmtReader::NumExprFields + 2]);
1168 break;
Chris Lattner4c6f9522009-04-27 05:14:47 +00001169 }
1170
1171 // We hit a STMT_STOP, so we're done with this expression.
1172 if (Finished)
1173 break;
1174
1175 ++NumStatementsRead;
1176
1177 if (S) {
1178 unsigned NumSubStmts = Reader.Visit(S);
1179 while (NumSubStmts > 0) {
1180 StmtStack.pop_back();
1181 --NumSubStmts;
1182 }
1183 }
1184
1185 assert(Idx == Record.size() && "Invalid deserialization of statement");
1186 StmtStack.push_back(S);
1187 }
1188 assert(StmtStack.size() == 1 && "Extra expressions on stack!");
Chris Lattner4c6f9522009-04-27 05:14:47 +00001189 return StmtStack.back();
1190}