blob: 4c80c87f19622a1d6fc3829fd61b71f05cda0b14 [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);
Sam Weinigce757a72010-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 Lattner4c6f9522009-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 Stump1eb44332009-09-09 15:08:12 +0000143 S->setStmts(*Reader.getContext(),
Jay Foadbeaaccd2009-05-21 09:52:38 +0000144 StmtStack.data() + StmtStack.size() - NumStmts, NumStmts);
Chris Lattner4c6f9522009-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 Gregordbb26db2009-05-15 23:57:33 +0000162 S->setEllipsisLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
163 S->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-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 Gregordbb26db2009-05-15 23:57:33 +0000171 S->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-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 Gregor8cfe5a72009-11-23 23:44:04 +0000186 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-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 Gregord06f6ca2009-05-15 18:53:42 +0000191 S->setElseLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000192 return 3;
193}
194
195unsigned PCHStmtReader::VisitSwitchStmt(SwitchStmt *S) {
196 VisitStmt(S);
Douglas Gregord3d53012009-11-24 17:07:59 +0000197 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-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 Stump1eb44332009-09-09 15:08:12 +0000208
Douglas Gregor43d9d922009-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 Lattner4c6f9522009-04-27 05:14:47 +0000212 PrevSC = SC;
213 }
214 return 2;
215}
216
217unsigned PCHStmtReader::VisitWhileStmt(WhileStmt *S) {
218 VisitStmt(S);
Douglas Gregor5656e142009-11-24 21:15:44 +0000219 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-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 Gregor9f3ca2a2009-05-15 21:56:04 +0000231 S->setWhileLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner98913592009-06-12 23:04:47 +0000232 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-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 Gregor99e9b4d2009-11-25 00:27:52 +0000240 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-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 Gregor5831c6a2009-05-15 22:12:32 +0000244 S->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
245 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-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 Gregor5f1b9e62009-05-16 00:20:29 +0000260 S->setStarLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-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 Lattnerd1d64a02009-04-27 21:45:14 +0000297 S->setDeclGroup(DeclGroupRef(DeclGroup::Create(*Reader.getContext(),
Douglas Gregor75fdb232009-05-22 22:45:36 +0000298 Decls.data(),
299 Decls.size())));
Chris Lattner4c6f9522009-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 Stump3b11fd32010-01-04 22:37:17 +0000313 S->setMSAsm(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000314
315 unsigned StackIdx
Chris Lattner4c6f9522009-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 }
Chris Lattner4c6f9522009-04-27 05:14:47 +0000328
329 // Constraints
330 llvm::SmallVector<StringLiteral*, 16> Clobbers;
331 for (unsigned I = 0; I != NumClobbers; ++I)
332 Clobbers.push_back(cast_or_null<StringLiteral>(StmtStack[StackIdx++]));
Anders Carlssonfdba9c02010-01-30 19:34:25 +0000333
Anders Carlssonacb6bcb2010-01-30 20:38:10 +0000334 S->setOutputsAndInputsAndClobbers(*Reader.getContext(),
335 Names.data(), Constraints.data(),
Anders Carlssonfdba9c02010-01-30 19:34:25 +0000336 Exprs.data(), NumOutputs, NumInputs,
337 Clobbers.data(), NumClobbers);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000338
339 assert(StackIdx == StmtStack.size() && "Error deserializing AsmStmt");
340 return NumOutputs*2 + NumInputs*2 + NumClobbers + 1;
341}
342
343unsigned PCHStmtReader::VisitExpr(Expr *E) {
344 VisitStmt(E);
345 E->setType(Reader.GetType(Record[Idx++]));
346 E->setTypeDependent(Record[Idx++]);
347 E->setValueDependent(Record[Idx++]);
348 assert(Idx == NumExprFields && "Incorrect expression field count");
349 return 0;
350}
351
352unsigned PCHStmtReader::VisitPredefinedExpr(PredefinedExpr *E) {
353 VisitExpr(E);
354 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
355 E->setIdentType((PredefinedExpr::IdentType)Record[Idx++]);
356 return 0;
357}
358
359unsigned PCHStmtReader::VisitDeclRefExpr(DeclRefExpr *E) {
360 VisitExpr(E);
John McCalldbd872f2009-12-08 09:08:17 +0000361 E->setDecl(cast<ValueDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000362 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregora2813ce2009-10-23 18:54:35 +0000363 // FIXME: read qualifier
364 // FIXME: read explicit template arguments
Chris Lattner4c6f9522009-04-27 05:14:47 +0000365 return 0;
366}
367
368unsigned PCHStmtReader::VisitIntegerLiteral(IntegerLiteral *E) {
369 VisitExpr(E);
370 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
371 E->setValue(Reader.ReadAPInt(Record, Idx));
372 return 0;
373}
374
375unsigned PCHStmtReader::VisitFloatingLiteral(FloatingLiteral *E) {
376 VisitExpr(E);
377 E->setValue(Reader.ReadAPFloat(Record, Idx));
378 E->setExact(Record[Idx++]);
379 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
380 return 0;
381}
382
383unsigned PCHStmtReader::VisitImaginaryLiteral(ImaginaryLiteral *E) {
384 VisitExpr(E);
385 E->setSubExpr(cast<Expr>(StmtStack.back()));
386 return 1;
387}
388
389unsigned PCHStmtReader::VisitStringLiteral(StringLiteral *E) {
390 VisitExpr(E);
391 unsigned Len = Record[Idx++];
Mike Stump1eb44332009-09-09 15:08:12 +0000392 assert(Record[Idx] == E->getNumConcatenated() &&
Chris Lattner4c6f9522009-04-27 05:14:47 +0000393 "Wrong number of concatenated tokens!");
394 ++Idx;
395 E->setWide(Record[Idx++]);
396
Mike Stump1eb44332009-09-09 15:08:12 +0000397 // Read string data
Daniel Dunbarb6480232009-09-22 03:27:33 +0000398 llvm::SmallString<16> Str(&Record[Idx], &Record[Idx] + Len);
399 E->setString(*Reader.getContext(), Str.str());
Chris Lattner4c6f9522009-04-27 05:14:47 +0000400 Idx += Len;
401
402 // Read source locations
403 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
404 E->setStrTokenLoc(I, SourceLocation::getFromRawEncoding(Record[Idx++]));
405
406 return 0;
407}
408
409unsigned PCHStmtReader::VisitCharacterLiteral(CharacterLiteral *E) {
410 VisitExpr(E);
411 E->setValue(Record[Idx++]);
412 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
413 E->setWide(Record[Idx++]);
414 return 0;
415}
416
417unsigned PCHStmtReader::VisitParenExpr(ParenExpr *E) {
418 VisitExpr(E);
419 E->setLParen(SourceLocation::getFromRawEncoding(Record[Idx++]));
420 E->setRParen(SourceLocation::getFromRawEncoding(Record[Idx++]));
421 E->setSubExpr(cast<Expr>(StmtStack.back()));
422 return 1;
423}
424
425unsigned PCHStmtReader::VisitUnaryOperator(UnaryOperator *E) {
426 VisitExpr(E);
427 E->setSubExpr(cast<Expr>(StmtStack.back()));
428 E->setOpcode((UnaryOperator::Opcode)Record[Idx++]);
429 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
430 return 1;
431}
432
433unsigned PCHStmtReader::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) {
434 VisitExpr(E);
435 E->setSizeof(Record[Idx++]);
436 if (Record[Idx] == 0) {
437 E->setArgument(cast<Expr>(StmtStack.back()));
438 ++Idx;
439 } else {
John McCalla93c9342009-12-07 02:54:59 +0000440 E->setArgument(Reader.GetTypeSourceInfo(Record, Idx));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000441 }
442 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
443 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
444 return E->isArgumentType()? 0 : 1;
445}
446
447unsigned PCHStmtReader::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
448 VisitExpr(E);
449 E->setLHS(cast<Expr>(StmtStack[StmtStack.size() - 2]));
450 E->setRHS(cast<Expr>(StmtStack[StmtStack.size() - 1]));
451 E->setRBracketLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
452 return 2;
453}
454
455unsigned PCHStmtReader::VisitCallExpr(CallExpr *E) {
456 VisitExpr(E);
Chris Lattnerd1d64a02009-04-27 21:45:14 +0000457 E->setNumArgs(*Reader.getContext(), Record[Idx++]);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000458 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
459 E->setCallee(cast<Expr>(StmtStack[StmtStack.size() - E->getNumArgs() - 1]));
460 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
461 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
462 return E->getNumArgs() + 1;
463}
464
465unsigned PCHStmtReader::VisitMemberExpr(MemberExpr *E) {
466 VisitExpr(E);
467 E->setBase(cast<Expr>(StmtStack.back()));
Eli Friedmandd27a2d2009-12-04 06:46:54 +0000468 E->setMemberDecl(cast<ValueDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000469 E->setMemberLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
470 E->setArrow(Record[Idx++]);
471 return 1;
472}
473
Steve Narofff242b1b2009-07-24 17:54:45 +0000474unsigned PCHStmtReader::VisitObjCIsaExpr(ObjCIsaExpr *E) {
475 VisitExpr(E);
476 E->setBase(cast<Expr>(StmtStack.back()));
477 E->setIsaMemberLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
478 E->setArrow(Record[Idx++]);
479 return 1;
480}
481
Chris Lattner4c6f9522009-04-27 05:14:47 +0000482unsigned PCHStmtReader::VisitCastExpr(CastExpr *E) {
483 VisitExpr(E);
484 E->setSubExpr(cast<Expr>(StmtStack.back()));
Anders Carlssoncdef2b72009-07-31 00:48:10 +0000485 E->setCastKind((CastExpr::CastKind)Record[Idx++]);
486
Chris Lattner4c6f9522009-04-27 05:14:47 +0000487 return 1;
488}
489
490unsigned PCHStmtReader::VisitBinaryOperator(BinaryOperator *E) {
491 VisitExpr(E);
492 E->setLHS(cast<Expr>(StmtStack.end()[-2]));
493 E->setRHS(cast<Expr>(StmtStack.end()[-1]));
494 E->setOpcode((BinaryOperator::Opcode)Record[Idx++]);
495 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
496 return 2;
497}
498
499unsigned PCHStmtReader::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
500 VisitBinaryOperator(E);
501 E->setComputationLHSType(Reader.GetType(Record[Idx++]));
502 E->setComputationResultType(Reader.GetType(Record[Idx++]));
503 return 2;
504}
505
506unsigned PCHStmtReader::VisitConditionalOperator(ConditionalOperator *E) {
507 VisitExpr(E);
508 E->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
509 E->setLHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
510 E->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 1]));
Douglas Gregor47e1f7c2009-08-26 14:37:04 +0000511 E->setQuestionLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
512 E->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000513 return 3;
514}
515
516unsigned PCHStmtReader::VisitImplicitCastExpr(ImplicitCastExpr *E) {
517 VisitCastExpr(E);
518 E->setLvalueCast(Record[Idx++]);
519 return 1;
520}
521
522unsigned PCHStmtReader::VisitExplicitCastExpr(ExplicitCastExpr *E) {
523 VisitCastExpr(E);
John McCall9d125032010-01-15 18:39:57 +0000524 E->setTypeInfoAsWritten(Reader.GetTypeSourceInfo(Record, Idx));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000525 return 1;
526}
527
528unsigned PCHStmtReader::VisitCStyleCastExpr(CStyleCastExpr *E) {
529 VisitExplicitCastExpr(E);
530 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
531 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
532 return 1;
533}
534
535unsigned PCHStmtReader::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
536 VisitExpr(E);
537 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
John McCall42f56b52010-01-18 19:35:47 +0000538 E->setTypeSourceInfo(Reader.GetTypeSourceInfo(Record, Idx));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000539 E->setInitializer(cast<Expr>(StmtStack.back()));
540 E->setFileScope(Record[Idx++]);
541 return 1;
542}
543
544unsigned PCHStmtReader::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
545 VisitExpr(E);
546 E->setBase(cast<Expr>(StmtStack.back()));
547 E->setAccessor(Reader.GetIdentifierInfo(Record, Idx));
548 E->setAccessorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
549 return 1;
550}
551
552unsigned PCHStmtReader::VisitInitListExpr(InitListExpr *E) {
553 VisitExpr(E);
554 unsigned NumInits = Record[Idx++];
555 E->reserveInits(NumInits);
556 for (unsigned I = 0; I != NumInits; ++I)
Mike Stump1eb44332009-09-09 15:08:12 +0000557 E->updateInit(I,
Chris Lattner4c6f9522009-04-27 05:14:47 +0000558 cast<Expr>(StmtStack[StmtStack.size() - NumInits - 1 + I]));
559 E->setSyntacticForm(cast_or_null<InitListExpr>(StmtStack.back()));
560 E->setLBraceLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
561 E->setRBraceLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
562 E->setInitializedFieldInUnion(
563 cast_or_null<FieldDecl>(Reader.GetDecl(Record[Idx++])));
564 E->sawArrayRangeDesignator(Record[Idx++]);
565 return NumInits + 1;
566}
567
568unsigned PCHStmtReader::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
569 typedef DesignatedInitExpr::Designator Designator;
570
571 VisitExpr(E);
572 unsigned NumSubExprs = Record[Idx++];
573 assert(NumSubExprs == E->getNumSubExprs() && "Wrong number of subexprs");
574 for (unsigned I = 0; I != NumSubExprs; ++I)
575 E->setSubExpr(I, cast<Expr>(StmtStack[StmtStack.size() - NumSubExprs + I]));
576 E->setEqualOrColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
577 E->setGNUSyntax(Record[Idx++]);
578
579 llvm::SmallVector<Designator, 4> Designators;
580 while (Idx < Record.size()) {
581 switch ((pch::DesignatorTypes)Record[Idx++]) {
582 case pch::DESIG_FIELD_DECL: {
583 FieldDecl *Field = cast<FieldDecl>(Reader.GetDecl(Record[Idx++]));
Mike Stump1eb44332009-09-09 15:08:12 +0000584 SourceLocation DotLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000585 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000586 SourceLocation FieldLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000587 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000588 Designators.push_back(Designator(Field->getIdentifier(), DotLoc,
Chris Lattner4c6f9522009-04-27 05:14:47 +0000589 FieldLoc));
590 Designators.back().setField(Field);
591 break;
592 }
593
594 case pch::DESIG_FIELD_NAME: {
595 const IdentifierInfo *Name = Reader.GetIdentifierInfo(Record, Idx);
Mike Stump1eb44332009-09-09 15:08:12 +0000596 SourceLocation DotLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000597 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000598 SourceLocation FieldLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000599 = SourceLocation::getFromRawEncoding(Record[Idx++]);
600 Designators.push_back(Designator(Name, DotLoc, FieldLoc));
601 break;
602 }
Mike Stump1eb44332009-09-09 15:08:12 +0000603
Chris Lattner4c6f9522009-04-27 05:14:47 +0000604 case pch::DESIG_ARRAY: {
605 unsigned Index = Record[Idx++];
606 SourceLocation LBracketLoc
607 = SourceLocation::getFromRawEncoding(Record[Idx++]);
608 SourceLocation RBracketLoc
609 = SourceLocation::getFromRawEncoding(Record[Idx++]);
610 Designators.push_back(Designator(Index, LBracketLoc, RBracketLoc));
611 break;
612 }
613
614 case pch::DESIG_ARRAY_RANGE: {
615 unsigned Index = Record[Idx++];
616 SourceLocation LBracketLoc
617 = SourceLocation::getFromRawEncoding(Record[Idx++]);
618 SourceLocation EllipsisLoc
619 = SourceLocation::getFromRawEncoding(Record[Idx++]);
620 SourceLocation RBracketLoc
621 = SourceLocation::getFromRawEncoding(Record[Idx++]);
622 Designators.push_back(Designator(Index, LBracketLoc, EllipsisLoc,
623 RBracketLoc));
624 break;
625 }
626 }
627 }
Douglas Gregor319d57f2010-01-06 23:17:19 +0000628 E->setDesignators(*Reader.getContext(),
629 Designators.data(), Designators.size());
Chris Lattner4c6f9522009-04-27 05:14:47 +0000630
631 return NumSubExprs;
632}
633
634unsigned PCHStmtReader::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
635 VisitExpr(E);
636 return 0;
637}
638
639unsigned PCHStmtReader::VisitVAArgExpr(VAArgExpr *E) {
640 VisitExpr(E);
641 E->setSubExpr(cast<Expr>(StmtStack.back()));
642 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
643 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
644 return 1;
645}
646
647unsigned PCHStmtReader::VisitAddrLabelExpr(AddrLabelExpr *E) {
648 VisitExpr(E);
649 E->setAmpAmpLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
650 E->setLabelLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
651 Reader.SetLabelOf(E, Record[Idx++]);
652 return 0;
653}
654
655unsigned PCHStmtReader::VisitStmtExpr(StmtExpr *E) {
656 VisitExpr(E);
657 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
658 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
659 E->setSubStmt(cast_or_null<CompoundStmt>(StmtStack.back()));
660 return 1;
661}
662
663unsigned PCHStmtReader::VisitTypesCompatibleExpr(TypesCompatibleExpr *E) {
664 VisitExpr(E);
665 E->setArgType1(Reader.GetType(Record[Idx++]));
666 E->setArgType2(Reader.GetType(Record[Idx++]));
667 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
668 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
669 return 0;
670}
671
672unsigned PCHStmtReader::VisitChooseExpr(ChooseExpr *E) {
673 VisitExpr(E);
674 E->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
675 E->setLHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
676 E->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 1]));
677 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
678 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
679 return 3;
680}
681
682unsigned PCHStmtReader::VisitGNUNullExpr(GNUNullExpr *E) {
683 VisitExpr(E);
684 E->setTokenLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
685 return 0;
686}
687
688unsigned PCHStmtReader::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
689 VisitExpr(E);
690 unsigned NumExprs = Record[Idx++];
Mike Stump1eb44332009-09-09 15:08:12 +0000691 E->setExprs(*Reader.getContext(),
Nate Begeman888376a2009-08-12 02:28:50 +0000692 (Expr **)&StmtStack[StmtStack.size() - NumExprs], NumExprs);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000693 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
694 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
695 return NumExprs;
696}
697
698unsigned PCHStmtReader::VisitBlockExpr(BlockExpr *E) {
699 VisitExpr(E);
700 E->setBlockDecl(cast_or_null<BlockDecl>(Reader.GetDecl(Record[Idx++])));
701 E->setHasBlockDeclRefExprs(Record[Idx++]);
702 return 0;
703}
704
705unsigned PCHStmtReader::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
706 VisitExpr(E);
707 E->setDecl(cast<ValueDecl>(Reader.GetDecl(Record[Idx++])));
708 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
709 E->setByRef(Record[Idx++]);
Fariborz Jahanian9b0b57c2009-06-20 00:02:26 +0000710 E->setConstQualAdded(Record[Idx++]);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000711 return 0;
712}
713
714//===----------------------------------------------------------------------===//
715// Objective-C Expressions and Statements
716
717unsigned PCHStmtReader::VisitObjCStringLiteral(ObjCStringLiteral *E) {
718 VisitExpr(E);
719 E->setString(cast<StringLiteral>(StmtStack.back()));
720 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
721 return 1;
722}
723
724unsigned PCHStmtReader::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
725 VisitExpr(E);
726 E->setEncodedType(Reader.GetType(Record[Idx++]));
727 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
728 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
729 return 0;
730}
731
732unsigned PCHStmtReader::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
733 VisitExpr(E);
734 E->setSelector(Reader.GetSelector(Record, Idx));
735 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
736 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
737 return 0;
738}
739
740unsigned PCHStmtReader::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
741 VisitExpr(E);
742 E->setProtocol(cast<ObjCProtocolDecl>(Reader.GetDecl(Record[Idx++])));
743 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
744 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
745 return 0;
746}
747
748unsigned PCHStmtReader::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
749 VisitExpr(E);
750 E->setDecl(cast<ObjCIvarDecl>(Reader.GetDecl(Record[Idx++])));
751 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
752 E->setBase(cast<Expr>(StmtStack.back()));
753 E->setIsArrow(Record[Idx++]);
754 E->setIsFreeIvar(Record[Idx++]);
755 return 1;
756}
757
758unsigned PCHStmtReader::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
759 VisitExpr(E);
760 E->setProperty(cast<ObjCPropertyDecl>(Reader.GetDecl(Record[Idx++])));
761 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
762 E->setBase(cast<Expr>(StmtStack.back()));
763 return 1;
764}
765
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000766unsigned PCHStmtReader::VisitObjCImplicitSetterGetterRefExpr(
767 ObjCImplicitSetterGetterRefExpr *E) {
Chris Lattner4c6f9522009-04-27 05:14:47 +0000768 VisitExpr(E);
769 E->setGetterMethod(
770 cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
771 E->setSetterMethod(
772 cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
Fariborz Jahaniand2ae5aa2009-08-18 21:37:33 +0000773 E->setInterfaceDecl(
Chris Lattner4c6f9522009-04-27 05:14:47 +0000774 cast_or_null<ObjCInterfaceDecl>(Reader.GetDecl(Record[Idx++])));
775 E->setBase(cast_or_null<Expr>(StmtStack.back()));
776 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
777 E->setClassLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
778 return 1;
779}
780
781unsigned PCHStmtReader::VisitObjCMessageExpr(ObjCMessageExpr *E) {
782 VisitExpr(E);
783 E->setNumArgs(Record[Idx++]);
784 E->setLeftLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
785 E->setRightLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
786 E->setSelector(Reader.GetSelector(Record, Idx));
787 E->setMethodDecl(cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
Mike Stump1eb44332009-09-09 15:08:12 +0000788
Chris Lattner4c6f9522009-04-27 05:14:47 +0000789 E->setReceiver(
790 cast_or_null<Expr>(StmtStack[StmtStack.size() - E->getNumArgs() - 1]));
791 if (!E->getReceiver()) {
792 ObjCMessageExpr::ClassInfo CI;
793 CI.first = cast_or_null<ObjCInterfaceDecl>(Reader.GetDecl(Record[Idx++]));
794 CI.second = Reader.GetIdentifierInfo(Record, Idx);
795 E->setClassInfo(CI);
796 }
797
798 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
799 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
800 return E->getNumArgs() + 1;
801}
802
803unsigned PCHStmtReader::VisitObjCSuperExpr(ObjCSuperExpr *E) {
804 VisitExpr(E);
805 E->setLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
806 return 0;
807}
808
809unsigned PCHStmtReader::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
810 VisitStmt(S);
811 S->setElement(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 3]));
812 S->setCollection(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
813 S->setBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
814 S->setForLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
815 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
816 return 3;
817}
818
819unsigned PCHStmtReader::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
820 VisitStmt(S);
821 S->setCatchBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
822 S->setNextCatchStmt(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
823 S->setCatchParamDecl(cast_or_null<ParmVarDecl>(Reader.GetDecl(Record[Idx++])));
824 S->setAtCatchLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
825 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
826 return 2;
827}
828
829unsigned PCHStmtReader::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
830 VisitStmt(S);
831 S->setFinallyBody(StmtStack.back());
832 S->setAtFinallyLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
833 return 1;
834}
835
836unsigned PCHStmtReader::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
837 VisitStmt(S);
838 S->setTryBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 3]));
839 S->setCatchStmts(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
840 S->setFinallyStmt(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
841 S->setAtTryLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
842 return 3;
843}
844
845unsigned PCHStmtReader::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
846 VisitStmt(S);
847 S->setSynchExpr(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
848 S->setSynchBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
849 S->setAtSynchronizedLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
850 return 2;
851}
852
853unsigned PCHStmtReader::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
854 VisitStmt(S);
855 S->setThrowExpr(StmtStack.back());
856 S->setThrowLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
857 return 1;
858}
859
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000860//===----------------------------------------------------------------------===//
861// C++ Expressions and Statements
862
863unsigned PCHStmtReader::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
864 unsigned num = VisitCallExpr(E);
865 E->setOperator((OverloadedOperatorKind)Record[Idx++]);
866 return num;
867}
868
Douglas Gregor39da0b82009-09-09 23:08:42 +0000869unsigned PCHStmtReader::VisitCXXConstructExpr(CXXConstructExpr *E) {
870 VisitExpr(E);
871 E->setConstructor(cast<CXXConstructorDecl>(Reader.GetDecl(Record[Idx++])));
Douglas Gregor99a2e602009-12-16 01:38:02 +0000872 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor39da0b82009-09-09 23:08:42 +0000873 E->setElidable(Record[Idx++]);
Douglas Gregor16006c92009-12-16 18:50:27 +0000874 E->setRequiresZeroInitialization(Record[Idx++]);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000875 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
876 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
877 return E->getNumArgs();
878}
Chris Lattner4c6f9522009-04-27 05:14:47 +0000879
Sam Weinigce757a72010-01-16 21:21:01 +0000880unsigned PCHStmtReader::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
881 unsigned num = VisitExplicitCastExpr(E);
882 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
883 return num;
884}
885
886unsigned PCHStmtReader::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
887 return VisitCXXNamedCastExpr(E);
888}
889
890unsigned PCHStmtReader::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
891 return VisitCXXNamedCastExpr(E);
892}
893
894unsigned PCHStmtReader::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
895 return VisitCXXNamedCastExpr(E);
896}
897
898unsigned PCHStmtReader::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
899 return VisitCXXNamedCastExpr(E);
900}
901
902unsigned PCHStmtReader::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
903 unsigned num = VisitExplicitCastExpr(E);
904 E->setTypeBeginLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
905 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
906 return num;
907}
908
Chris Lattner52e97d12009-04-27 05:41:06 +0000909// Within the bitstream, expressions are stored in Reverse Polish
910// Notation, with each of the subexpressions preceding the
911// expression they are stored in. To evaluate expressions, we
912// continue reading expressions and placing them on the stack, with
913// expressions having operands removing those operands from the
914// stack. Evaluation terminates when we see a STMT_STOP record, and
915// the single remaining expression on the stack is our result.
916Stmt *PCHReader::ReadStmt(llvm::BitstreamCursor &Cursor) {
Chris Lattner4c6f9522009-04-27 05:14:47 +0000917 RecordData Record;
918 unsigned Idx;
919 llvm::SmallVector<Stmt *, 16> StmtStack;
920 PCHStmtReader Reader(*this, Record, Idx, StmtStack);
921 Stmt::EmptyShell Empty;
922
923 while (true) {
Chris Lattner52e97d12009-04-27 05:41:06 +0000924 unsigned Code = Cursor.ReadCode();
Chris Lattner4c6f9522009-04-27 05:14:47 +0000925 if (Code == llvm::bitc::END_BLOCK) {
Chris Lattner52e97d12009-04-27 05:41:06 +0000926 if (Cursor.ReadBlockEnd()) {
Douglas Gregora02b1472009-04-28 21:53:25 +0000927 Error("error at end of block in PCH file");
Chris Lattner4c6f9522009-04-27 05:14:47 +0000928 return 0;
929 }
930 break;
931 }
932
933 if (Code == llvm::bitc::ENTER_SUBBLOCK) {
934 // No known subblocks, always skip them.
Chris Lattner52e97d12009-04-27 05:41:06 +0000935 Cursor.ReadSubBlockID();
936 if (Cursor.SkipBlock()) {
Douglas Gregora02b1472009-04-28 21:53:25 +0000937 Error("malformed block record in PCH file");
Chris Lattner4c6f9522009-04-27 05:14:47 +0000938 return 0;
939 }
940 continue;
941 }
942
943 if (Code == llvm::bitc::DEFINE_ABBREV) {
Chris Lattner52e97d12009-04-27 05:41:06 +0000944 Cursor.ReadAbbrevRecord();
Chris Lattner4c6f9522009-04-27 05:14:47 +0000945 continue;
946 }
947
948 Stmt *S = 0;
949 Idx = 0;
950 Record.clear();
951 bool Finished = false;
Chris Lattner52e97d12009-04-27 05:41:06 +0000952 switch ((pch::StmtCode)Cursor.ReadRecord(Code, Record)) {
Chris Lattner4c6f9522009-04-27 05:14:47 +0000953 case pch::STMT_STOP:
954 Finished = true;
955 break;
956
Mike Stump1eb44332009-09-09 15:08:12 +0000957 case pch::STMT_NULL_PTR:
958 S = 0;
Chris Lattner4c6f9522009-04-27 05:14:47 +0000959 break;
960
961 case pch::STMT_NULL:
962 S = new (Context) NullStmt(Empty);
963 break;
964
965 case pch::STMT_COMPOUND:
966 S = new (Context) CompoundStmt(Empty);
967 break;
968
969 case pch::STMT_CASE:
970 S = new (Context) CaseStmt(Empty);
971 break;
972
973 case pch::STMT_DEFAULT:
974 S = new (Context) DefaultStmt(Empty);
975 break;
976
977 case pch::STMT_LABEL:
978 S = new (Context) LabelStmt(Empty);
979 break;
980
981 case pch::STMT_IF:
982 S = new (Context) IfStmt(Empty);
983 break;
984
985 case pch::STMT_SWITCH:
986 S = new (Context) SwitchStmt(Empty);
987 break;
988
989 case pch::STMT_WHILE:
990 S = new (Context) WhileStmt(Empty);
991 break;
992
993 case pch::STMT_DO:
994 S = new (Context) DoStmt(Empty);
995 break;
Mike Stump1eb44332009-09-09 15:08:12 +0000996
Chris Lattner4c6f9522009-04-27 05:14:47 +0000997 case pch::STMT_FOR:
998 S = new (Context) ForStmt(Empty);
999 break;
1000
1001 case pch::STMT_GOTO:
1002 S = new (Context) GotoStmt(Empty);
1003 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001004
Chris Lattner4c6f9522009-04-27 05:14:47 +00001005 case pch::STMT_INDIRECT_GOTO:
1006 S = new (Context) IndirectGotoStmt(Empty);
1007 break;
1008
1009 case pch::STMT_CONTINUE:
1010 S = new (Context) ContinueStmt(Empty);
1011 break;
1012
1013 case pch::STMT_BREAK:
1014 S = new (Context) BreakStmt(Empty);
1015 break;
1016
1017 case pch::STMT_RETURN:
1018 S = new (Context) ReturnStmt(Empty);
1019 break;
1020
1021 case pch::STMT_DECL:
1022 S = new (Context) DeclStmt(Empty);
1023 break;
1024
1025 case pch::STMT_ASM:
1026 S = new (Context) AsmStmt(Empty);
1027 break;
1028
1029 case pch::EXPR_PREDEFINED:
1030 S = new (Context) PredefinedExpr(Empty);
1031 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001032
1033 case pch::EXPR_DECL_REF:
1034 S = new (Context) DeclRefExpr(Empty);
Chris Lattner4c6f9522009-04-27 05:14:47 +00001035 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001036
1037 case pch::EXPR_INTEGER_LITERAL:
Chris Lattner4c6f9522009-04-27 05:14:47 +00001038 S = new (Context) IntegerLiteral(Empty);
1039 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001040
Chris Lattner4c6f9522009-04-27 05:14:47 +00001041 case pch::EXPR_FLOATING_LITERAL:
1042 S = new (Context) FloatingLiteral(Empty);
1043 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001044
Chris Lattner4c6f9522009-04-27 05:14:47 +00001045 case pch::EXPR_IMAGINARY_LITERAL:
1046 S = new (Context) ImaginaryLiteral(Empty);
1047 break;
1048
1049 case pch::EXPR_STRING_LITERAL:
Mike Stump1eb44332009-09-09 15:08:12 +00001050 S = StringLiteral::CreateEmpty(*Context,
Chris Lattner4c6f9522009-04-27 05:14:47 +00001051 Record[PCHStmtReader::NumExprFields + 1]);
1052 break;
1053
1054 case pch::EXPR_CHARACTER_LITERAL:
1055 S = new (Context) CharacterLiteral(Empty);
1056 break;
1057
1058 case pch::EXPR_PAREN:
1059 S = new (Context) ParenExpr(Empty);
1060 break;
1061
1062 case pch::EXPR_UNARY_OPERATOR:
1063 S = new (Context) UnaryOperator(Empty);
1064 break;
1065
1066 case pch::EXPR_SIZEOF_ALIGN_OF:
1067 S = new (Context) SizeOfAlignOfExpr(Empty);
1068 break;
1069
1070 case pch::EXPR_ARRAY_SUBSCRIPT:
1071 S = new (Context) ArraySubscriptExpr(Empty);
1072 break;
1073
1074 case pch::EXPR_CALL:
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001075 S = new (Context) CallExpr(*Context, Stmt::CallExprClass, Empty);
Chris Lattner4c6f9522009-04-27 05:14:47 +00001076 break;
1077
1078 case pch::EXPR_MEMBER:
1079 S = new (Context) MemberExpr(Empty);
1080 break;
1081
1082 case pch::EXPR_BINARY_OPERATOR:
1083 S = new (Context) BinaryOperator(Empty);
1084 break;
1085
1086 case pch::EXPR_COMPOUND_ASSIGN_OPERATOR:
1087 S = new (Context) CompoundAssignOperator(Empty);
1088 break;
1089
1090 case pch::EXPR_CONDITIONAL_OPERATOR:
1091 S = new (Context) ConditionalOperator(Empty);
1092 break;
1093
1094 case pch::EXPR_IMPLICIT_CAST:
1095 S = new (Context) ImplicitCastExpr(Empty);
1096 break;
1097
1098 case pch::EXPR_CSTYLE_CAST:
1099 S = new (Context) CStyleCastExpr(Empty);
1100 break;
1101
1102 case pch::EXPR_COMPOUND_LITERAL:
1103 S = new (Context) CompoundLiteralExpr(Empty);
1104 break;
1105
1106 case pch::EXPR_EXT_VECTOR_ELEMENT:
1107 S = new (Context) ExtVectorElementExpr(Empty);
1108 break;
1109
1110 case pch::EXPR_INIT_LIST:
1111 S = new (Context) InitListExpr(Empty);
1112 break;
1113
1114 case pch::EXPR_DESIGNATED_INIT:
Chris Lattnerd1d64a02009-04-27 21:45:14 +00001115 S = DesignatedInitExpr::CreateEmpty(*Context,
Chris Lattner4c6f9522009-04-27 05:14:47 +00001116 Record[PCHStmtReader::NumExprFields] - 1);
Mike Stump1eb44332009-09-09 15:08:12 +00001117
Chris Lattner4c6f9522009-04-27 05:14:47 +00001118 break;
1119
1120 case pch::EXPR_IMPLICIT_VALUE_INIT:
1121 S = new (Context) ImplicitValueInitExpr(Empty);
1122 break;
1123
1124 case pch::EXPR_VA_ARG:
1125 S = new (Context) VAArgExpr(Empty);
1126 break;
1127
1128 case pch::EXPR_ADDR_LABEL:
1129 S = new (Context) AddrLabelExpr(Empty);
1130 break;
1131
1132 case pch::EXPR_STMT:
1133 S = new (Context) StmtExpr(Empty);
1134 break;
1135
1136 case pch::EXPR_TYPES_COMPATIBLE:
1137 S = new (Context) TypesCompatibleExpr(Empty);
1138 break;
1139
1140 case pch::EXPR_CHOOSE:
1141 S = new (Context) ChooseExpr(Empty);
1142 break;
1143
1144 case pch::EXPR_GNU_NULL:
1145 S = new (Context) GNUNullExpr(Empty);
1146 break;
1147
1148 case pch::EXPR_SHUFFLE_VECTOR:
1149 S = new (Context) ShuffleVectorExpr(Empty);
1150 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001151
Chris Lattner4c6f9522009-04-27 05:14:47 +00001152 case pch::EXPR_BLOCK:
1153 S = new (Context) BlockExpr(Empty);
1154 break;
1155
1156 case pch::EXPR_BLOCK_DECL_REF:
1157 S = new (Context) BlockDeclRefExpr(Empty);
1158 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001159
Chris Lattner4c6f9522009-04-27 05:14:47 +00001160 case pch::EXPR_OBJC_STRING_LITERAL:
1161 S = new (Context) ObjCStringLiteral(Empty);
1162 break;
1163 case pch::EXPR_OBJC_ENCODE:
1164 S = new (Context) ObjCEncodeExpr(Empty);
1165 break;
1166 case pch::EXPR_OBJC_SELECTOR_EXPR:
1167 S = new (Context) ObjCSelectorExpr(Empty);
1168 break;
1169 case pch::EXPR_OBJC_PROTOCOL_EXPR:
1170 S = new (Context) ObjCProtocolExpr(Empty);
1171 break;
1172 case pch::EXPR_OBJC_IVAR_REF_EXPR:
1173 S = new (Context) ObjCIvarRefExpr(Empty);
1174 break;
1175 case pch::EXPR_OBJC_PROPERTY_REF_EXPR:
1176 S = new (Context) ObjCPropertyRefExpr(Empty);
1177 break;
1178 case pch::EXPR_OBJC_KVC_REF_EXPR:
Fariborz Jahanian09105f52009-08-20 17:02:02 +00001179 S = new (Context) ObjCImplicitSetterGetterRefExpr(Empty);
Chris Lattner4c6f9522009-04-27 05:14:47 +00001180 break;
1181 case pch::EXPR_OBJC_MESSAGE_EXPR:
1182 S = new (Context) ObjCMessageExpr(Empty);
1183 break;
1184 case pch::EXPR_OBJC_SUPER_EXPR:
1185 S = new (Context) ObjCSuperExpr(Empty);
1186 break;
Steve Narofff242b1b2009-07-24 17:54:45 +00001187 case pch::EXPR_OBJC_ISA:
1188 S = new (Context) ObjCIsaExpr(Empty);
1189 break;
Chris Lattner4c6f9522009-04-27 05:14:47 +00001190 case pch::STMT_OBJC_FOR_COLLECTION:
1191 S = new (Context) ObjCForCollectionStmt(Empty);
1192 break;
1193 case pch::STMT_OBJC_CATCH:
1194 S = new (Context) ObjCAtCatchStmt(Empty);
1195 break;
1196 case pch::STMT_OBJC_FINALLY:
1197 S = new (Context) ObjCAtFinallyStmt(Empty);
1198 break;
1199 case pch::STMT_OBJC_AT_TRY:
1200 S = new (Context) ObjCAtTryStmt(Empty);
1201 break;
1202 case pch::STMT_OBJC_AT_SYNCHRONIZED:
1203 S = new (Context) ObjCAtSynchronizedStmt(Empty);
1204 break;
1205 case pch::STMT_OBJC_AT_THROW:
1206 S = new (Context) ObjCAtThrowStmt(Empty);
1207 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001208
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001209 case pch::EXPR_CXX_OPERATOR_CALL:
1210 S = new (Context) CXXOperatorCallExpr(*Context, Empty);
1211 break;
Douglas Gregor39da0b82009-09-09 23:08:42 +00001212
1213 case pch::EXPR_CXX_CONSTRUCT:
1214 S = new (Context) CXXConstructExpr(Empty, *Context,
1215 Record[PCHStmtReader::NumExprFields + 2]);
1216 break;
Sam Weinigce757a72010-01-16 21:21:01 +00001217
1218 case pch::EXPR_CXX_STATIC_CAST:
1219 S = new (Context) CXXStaticCastExpr(Empty);
1220 break;
1221
1222 case pch::EXPR_CXX_DYNAMIC_CAST:
1223 S = new (Context) CXXDynamicCastExpr(Empty);
1224 break;
1225
1226 case pch::EXPR_CXX_REINTERPRET_CAST:
1227 S = new (Context) CXXReinterpretCastExpr(Empty);
1228 break;
1229
1230 case pch::EXPR_CXX_CONST_CAST:
1231 S = new (Context) CXXConstCastExpr(Empty);
1232 break;
1233
1234 case pch::EXPR_CXX_FUNCTIONAL_CAST:
1235 S = new (Context) CXXFunctionalCastExpr(Empty);
1236 break;
1237
1238
Chris Lattner4c6f9522009-04-27 05:14:47 +00001239 }
1240
1241 // We hit a STMT_STOP, so we're done with this expression.
1242 if (Finished)
1243 break;
1244
1245 ++NumStatementsRead;
1246
1247 if (S) {
1248 unsigned NumSubStmts = Reader.Visit(S);
1249 while (NumSubStmts > 0) {
1250 StmtStack.pop_back();
1251 --NumSubStmts;
1252 }
1253 }
1254
1255 assert(Idx == Record.size() && "Invalid deserialization of statement");
1256 StmtStack.push_back(S);
1257 }
1258 assert(StmtStack.size() == 1 && "Extra expressions on stack!");
Chris Lattner4c6f9522009-04-27 05:14:47 +00001259 return StmtStack.back();
1260}