blob: d253654f7c50d4707b4ae55676bd0a9d4d5821dd [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);
Sam Weinigeb7f9612010-02-07 06:32:43 +0000126 unsigned VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E);
127 unsigned VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000128 };
129}
130
131unsigned PCHStmtReader::VisitStmt(Stmt *S) {
132 assert(Idx == NumStmtFields && "Incorrect statement field count");
133 return 0;
134}
135
136unsigned PCHStmtReader::VisitNullStmt(NullStmt *S) {
137 VisitStmt(S);
138 S->setSemiLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
139 return 0;
140}
141
142unsigned PCHStmtReader::VisitCompoundStmt(CompoundStmt *S) {
143 VisitStmt(S);
144 unsigned NumStmts = Record[Idx++];
Mike Stump1eb44332009-09-09 15:08:12 +0000145 S->setStmts(*Reader.getContext(),
Jay Foadbeaaccd2009-05-21 09:52:38 +0000146 StmtStack.data() + StmtStack.size() - NumStmts, NumStmts);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000147 S->setLBracLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
148 S->setRBracLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
149 return NumStmts;
150}
151
152unsigned PCHStmtReader::VisitSwitchCase(SwitchCase *S) {
153 VisitStmt(S);
154 Reader.RecordSwitchCaseID(S, Record[Idx++]);
155 return 0;
156}
157
158unsigned PCHStmtReader::VisitCaseStmt(CaseStmt *S) {
159 VisitSwitchCase(S);
160 S->setLHS(cast<Expr>(StmtStack[StmtStack.size() - 3]));
161 S->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
162 S->setSubStmt(StmtStack.back());
163 S->setCaseLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregordbb26db2009-05-15 23:57:33 +0000164 S->setEllipsisLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
165 S->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000166 return 3;
167}
168
169unsigned PCHStmtReader::VisitDefaultStmt(DefaultStmt *S) {
170 VisitSwitchCase(S);
171 S->setSubStmt(StmtStack.back());
172 S->setDefaultLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregordbb26db2009-05-15 23:57:33 +0000173 S->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000174 return 1;
175}
176
177unsigned PCHStmtReader::VisitLabelStmt(LabelStmt *S) {
178 VisitStmt(S);
179 S->setID(Reader.GetIdentifierInfo(Record, Idx));
180 S->setSubStmt(StmtStack.back());
181 S->setIdentLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
182 Reader.RecordLabelStmt(S, Record[Idx++]);
183 return 1;
184}
185
186unsigned PCHStmtReader::VisitIfStmt(IfStmt *S) {
187 VisitStmt(S);
Douglas Gregor8cfe5a72009-11-23 23:44:04 +0000188 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000189 S->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
190 S->setThen(StmtStack[StmtStack.size() - 2]);
191 S->setElse(StmtStack[StmtStack.size() - 1]);
192 S->setIfLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregord06f6ca2009-05-15 18:53:42 +0000193 S->setElseLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000194 return 3;
195}
196
197unsigned PCHStmtReader::VisitSwitchStmt(SwitchStmt *S) {
198 VisitStmt(S);
Douglas Gregord3d53012009-11-24 17:07:59 +0000199 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000200 S->setCond(cast<Expr>(StmtStack[StmtStack.size() - 2]));
201 S->setBody(StmtStack.back());
202 S->setSwitchLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
203 SwitchCase *PrevSC = 0;
204 for (unsigned N = Record.size(); Idx != N; ++Idx) {
205 SwitchCase *SC = Reader.getSwitchCaseWithID(Record[Idx]);
206 if (PrevSC)
207 PrevSC->setNextSwitchCase(SC);
208 else
209 S->setSwitchCaseList(SC);
Mike Stump1eb44332009-09-09 15:08:12 +0000210
Douglas Gregor43d9d922009-08-08 01:41:12 +0000211 // Retain this SwitchCase, since SwitchStmt::addSwitchCase() would
212 // normally retain it (but we aren't calling addSwitchCase).
213 SC->Retain();
Chris Lattner4c6f9522009-04-27 05:14:47 +0000214 PrevSC = SC;
215 }
216 return 2;
217}
218
219unsigned PCHStmtReader::VisitWhileStmt(WhileStmt *S) {
220 VisitStmt(S);
Douglas Gregor5656e142009-11-24 21:15:44 +0000221 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000222 S->setCond(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
223 S->setBody(StmtStack.back());
224 S->setWhileLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
225 return 2;
226}
227
228unsigned PCHStmtReader::VisitDoStmt(DoStmt *S) {
229 VisitStmt(S);
230 S->setCond(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
231 S->setBody(StmtStack.back());
232 S->setDoLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor9f3ca2a2009-05-15 21:56:04 +0000233 S->setWhileLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner98913592009-06-12 23:04:47 +0000234 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000235 return 2;
236}
237
238unsigned PCHStmtReader::VisitForStmt(ForStmt *S) {
239 VisitStmt(S);
240 S->setInit(StmtStack[StmtStack.size() - 4]);
241 S->setCond(cast_or_null<Expr>(StmtStack[StmtStack.size() - 3]));
Douglas Gregor99e9b4d2009-11-25 00:27:52 +0000242 S->setConditionVariable(cast_or_null<VarDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000243 S->setInc(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
244 S->setBody(StmtStack.back());
245 S->setForLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor5831c6a2009-05-15 22:12:32 +0000246 S->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
247 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000248 return 4;
249}
250
251unsigned PCHStmtReader::VisitGotoStmt(GotoStmt *S) {
252 VisitStmt(S);
253 Reader.SetLabelOf(S, Record[Idx++]);
254 S->setGotoLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
255 S->setLabelLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
256 return 0;
257}
258
259unsigned PCHStmtReader::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
260 VisitStmt(S);
261 S->setGotoLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor5f1b9e62009-05-16 00:20:29 +0000262 S->setStarLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000263 S->setTarget(cast_or_null<Expr>(StmtStack.back()));
264 return 1;
265}
266
267unsigned PCHStmtReader::VisitContinueStmt(ContinueStmt *S) {
268 VisitStmt(S);
269 S->setContinueLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
270 return 0;
271}
272
273unsigned PCHStmtReader::VisitBreakStmt(BreakStmt *S) {
274 VisitStmt(S);
275 S->setBreakLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
276 return 0;
277}
278
279unsigned PCHStmtReader::VisitReturnStmt(ReturnStmt *S) {
280 VisitStmt(S);
281 S->setRetValue(cast_or_null<Expr>(StmtStack.back()));
282 S->setReturnLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
283 return 1;
284}
285
286unsigned PCHStmtReader::VisitDeclStmt(DeclStmt *S) {
287 VisitStmt(S);
288 S->setStartLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
289 S->setEndLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
290
291 if (Idx + 1 == Record.size()) {
292 // Single declaration
293 S->setDeclGroup(DeclGroupRef(Reader.GetDecl(Record[Idx++])));
294 } else {
295 llvm::SmallVector<Decl *, 16> Decls;
296 Decls.reserve(Record.size() - Idx);
297 for (unsigned N = Record.size(); Idx != N; ++Idx)
298 Decls.push_back(Reader.GetDecl(Record[Idx]));
Chris Lattnerd1d64a02009-04-27 21:45:14 +0000299 S->setDeclGroup(DeclGroupRef(DeclGroup::Create(*Reader.getContext(),
Douglas Gregor75fdb232009-05-22 22:45:36 +0000300 Decls.data(),
301 Decls.size())));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000302 }
303 return 0;
304}
305
306unsigned PCHStmtReader::VisitAsmStmt(AsmStmt *S) {
307 VisitStmt(S);
308 unsigned NumOutputs = Record[Idx++];
309 unsigned NumInputs = Record[Idx++];
310 unsigned NumClobbers = Record[Idx++];
311 S->setAsmLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
312 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
313 S->setVolatile(Record[Idx++]);
314 S->setSimple(Record[Idx++]);
Mike Stump3b11fd32010-01-04 22:37:17 +0000315 S->setMSAsm(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000316
317 unsigned StackIdx
Chris Lattner4c6f9522009-04-27 05:14:47 +0000318 = StmtStack.size() - (NumOutputs*2 + NumInputs*2 + NumClobbers + 1);
319 S->setAsmString(cast_or_null<StringLiteral>(StmtStack[StackIdx++]));
320
321 // Outputs and inputs
Anders Carlssonff93dbd2010-01-30 22:25:16 +0000322 llvm::SmallVector<IdentifierInfo *, 16> Names;
Chris Lattner4c6f9522009-04-27 05:14:47 +0000323 llvm::SmallVector<StringLiteral*, 16> Constraints;
324 llvm::SmallVector<Stmt*, 16> Exprs;
325 for (unsigned I = 0, N = NumOutputs + NumInputs; I != N; ++I) {
Anders Carlssonff93dbd2010-01-30 22:25:16 +0000326 Names.push_back(Reader.GetIdentifierInfo(Record, Idx));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000327 Constraints.push_back(cast_or_null<StringLiteral>(StmtStack[StackIdx++]));
328 Exprs.push_back(StmtStack[StackIdx++]);
329 }
Chris Lattner4c6f9522009-04-27 05:14:47 +0000330
331 // Constraints
332 llvm::SmallVector<StringLiteral*, 16> Clobbers;
333 for (unsigned I = 0; I != NumClobbers; ++I)
334 Clobbers.push_back(cast_or_null<StringLiteral>(StmtStack[StackIdx++]));
Anders Carlssonfdba9c02010-01-30 19:34:25 +0000335
Anders Carlssonacb6bcb2010-01-30 20:38:10 +0000336 S->setOutputsAndInputsAndClobbers(*Reader.getContext(),
337 Names.data(), Constraints.data(),
Anders Carlssonfdba9c02010-01-30 19:34:25 +0000338 Exprs.data(), NumOutputs, NumInputs,
339 Clobbers.data(), NumClobbers);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000340
341 assert(StackIdx == StmtStack.size() && "Error deserializing AsmStmt");
342 return NumOutputs*2 + NumInputs*2 + NumClobbers + 1;
343}
344
345unsigned PCHStmtReader::VisitExpr(Expr *E) {
346 VisitStmt(E);
347 E->setType(Reader.GetType(Record[Idx++]));
348 E->setTypeDependent(Record[Idx++]);
349 E->setValueDependent(Record[Idx++]);
350 assert(Idx == NumExprFields && "Incorrect expression field count");
351 return 0;
352}
353
354unsigned PCHStmtReader::VisitPredefinedExpr(PredefinedExpr *E) {
355 VisitExpr(E);
356 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
357 E->setIdentType((PredefinedExpr::IdentType)Record[Idx++]);
358 return 0;
359}
360
361unsigned PCHStmtReader::VisitDeclRefExpr(DeclRefExpr *E) {
362 VisitExpr(E);
John McCalldbd872f2009-12-08 09:08:17 +0000363 E->setDecl(cast<ValueDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000364 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregora2813ce2009-10-23 18:54:35 +0000365 // FIXME: read qualifier
366 // FIXME: read explicit template arguments
Chris Lattner4c6f9522009-04-27 05:14:47 +0000367 return 0;
368}
369
370unsigned PCHStmtReader::VisitIntegerLiteral(IntegerLiteral *E) {
371 VisitExpr(E);
372 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
373 E->setValue(Reader.ReadAPInt(Record, Idx));
374 return 0;
375}
376
377unsigned PCHStmtReader::VisitFloatingLiteral(FloatingLiteral *E) {
378 VisitExpr(E);
379 E->setValue(Reader.ReadAPFloat(Record, Idx));
380 E->setExact(Record[Idx++]);
381 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
382 return 0;
383}
384
385unsigned PCHStmtReader::VisitImaginaryLiteral(ImaginaryLiteral *E) {
386 VisitExpr(E);
387 E->setSubExpr(cast<Expr>(StmtStack.back()));
388 return 1;
389}
390
391unsigned PCHStmtReader::VisitStringLiteral(StringLiteral *E) {
392 VisitExpr(E);
393 unsigned Len = Record[Idx++];
Mike Stump1eb44332009-09-09 15:08:12 +0000394 assert(Record[Idx] == E->getNumConcatenated() &&
Chris Lattner4c6f9522009-04-27 05:14:47 +0000395 "Wrong number of concatenated tokens!");
396 ++Idx;
397 E->setWide(Record[Idx++]);
398
Mike Stump1eb44332009-09-09 15:08:12 +0000399 // Read string data
Daniel Dunbarb6480232009-09-22 03:27:33 +0000400 llvm::SmallString<16> Str(&Record[Idx], &Record[Idx] + Len);
401 E->setString(*Reader.getContext(), Str.str());
Chris Lattner4c6f9522009-04-27 05:14:47 +0000402 Idx += Len;
403
404 // Read source locations
405 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
406 E->setStrTokenLoc(I, SourceLocation::getFromRawEncoding(Record[Idx++]));
407
408 return 0;
409}
410
411unsigned PCHStmtReader::VisitCharacterLiteral(CharacterLiteral *E) {
412 VisitExpr(E);
413 E->setValue(Record[Idx++]);
414 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
415 E->setWide(Record[Idx++]);
416 return 0;
417}
418
419unsigned PCHStmtReader::VisitParenExpr(ParenExpr *E) {
420 VisitExpr(E);
421 E->setLParen(SourceLocation::getFromRawEncoding(Record[Idx++]));
422 E->setRParen(SourceLocation::getFromRawEncoding(Record[Idx++]));
423 E->setSubExpr(cast<Expr>(StmtStack.back()));
424 return 1;
425}
426
427unsigned PCHStmtReader::VisitUnaryOperator(UnaryOperator *E) {
428 VisitExpr(E);
429 E->setSubExpr(cast<Expr>(StmtStack.back()));
430 E->setOpcode((UnaryOperator::Opcode)Record[Idx++]);
431 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
432 return 1;
433}
434
435unsigned PCHStmtReader::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) {
436 VisitExpr(E);
437 E->setSizeof(Record[Idx++]);
438 if (Record[Idx] == 0) {
439 E->setArgument(cast<Expr>(StmtStack.back()));
440 ++Idx;
441 } else {
John McCalla93c9342009-12-07 02:54:59 +0000442 E->setArgument(Reader.GetTypeSourceInfo(Record, Idx));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000443 }
444 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
445 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
446 return E->isArgumentType()? 0 : 1;
447}
448
449unsigned PCHStmtReader::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
450 VisitExpr(E);
451 E->setLHS(cast<Expr>(StmtStack[StmtStack.size() - 2]));
452 E->setRHS(cast<Expr>(StmtStack[StmtStack.size() - 1]));
453 E->setRBracketLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
454 return 2;
455}
456
457unsigned PCHStmtReader::VisitCallExpr(CallExpr *E) {
458 VisitExpr(E);
Chris Lattnerd1d64a02009-04-27 21:45:14 +0000459 E->setNumArgs(*Reader.getContext(), Record[Idx++]);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000460 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
461 E->setCallee(cast<Expr>(StmtStack[StmtStack.size() - E->getNumArgs() - 1]));
462 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
463 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
464 return E->getNumArgs() + 1;
465}
466
467unsigned PCHStmtReader::VisitMemberExpr(MemberExpr *E) {
468 VisitExpr(E);
469 E->setBase(cast<Expr>(StmtStack.back()));
Eli Friedmandd27a2d2009-12-04 06:46:54 +0000470 E->setMemberDecl(cast<ValueDecl>(Reader.GetDecl(Record[Idx++])));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000471 E->setMemberLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
472 E->setArrow(Record[Idx++]);
473 return 1;
474}
475
Steve Narofff242b1b2009-07-24 17:54:45 +0000476unsigned PCHStmtReader::VisitObjCIsaExpr(ObjCIsaExpr *E) {
477 VisitExpr(E);
478 E->setBase(cast<Expr>(StmtStack.back()));
479 E->setIsaMemberLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
480 E->setArrow(Record[Idx++]);
481 return 1;
482}
483
Chris Lattner4c6f9522009-04-27 05:14:47 +0000484unsigned PCHStmtReader::VisitCastExpr(CastExpr *E) {
485 VisitExpr(E);
486 E->setSubExpr(cast<Expr>(StmtStack.back()));
Anders Carlssoncdef2b72009-07-31 00:48:10 +0000487 E->setCastKind((CastExpr::CastKind)Record[Idx++]);
488
Chris Lattner4c6f9522009-04-27 05:14:47 +0000489 return 1;
490}
491
492unsigned PCHStmtReader::VisitBinaryOperator(BinaryOperator *E) {
493 VisitExpr(E);
494 E->setLHS(cast<Expr>(StmtStack.end()[-2]));
495 E->setRHS(cast<Expr>(StmtStack.end()[-1]));
496 E->setOpcode((BinaryOperator::Opcode)Record[Idx++]);
497 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
498 return 2;
499}
500
501unsigned PCHStmtReader::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
502 VisitBinaryOperator(E);
503 E->setComputationLHSType(Reader.GetType(Record[Idx++]));
504 E->setComputationResultType(Reader.GetType(Record[Idx++]));
505 return 2;
506}
507
508unsigned PCHStmtReader::VisitConditionalOperator(ConditionalOperator *E) {
509 VisitExpr(E);
510 E->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
511 E->setLHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
512 E->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 1]));
Douglas Gregor47e1f7c2009-08-26 14:37:04 +0000513 E->setQuestionLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
514 E->setColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000515 return 3;
516}
517
518unsigned PCHStmtReader::VisitImplicitCastExpr(ImplicitCastExpr *E) {
519 VisitCastExpr(E);
520 E->setLvalueCast(Record[Idx++]);
521 return 1;
522}
523
524unsigned PCHStmtReader::VisitExplicitCastExpr(ExplicitCastExpr *E) {
525 VisitCastExpr(E);
John McCall9d125032010-01-15 18:39:57 +0000526 E->setTypeInfoAsWritten(Reader.GetTypeSourceInfo(Record, Idx));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000527 return 1;
528}
529
530unsigned PCHStmtReader::VisitCStyleCastExpr(CStyleCastExpr *E) {
531 VisitExplicitCastExpr(E);
532 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
533 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
534 return 1;
535}
536
537unsigned PCHStmtReader::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
538 VisitExpr(E);
539 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
John McCall42f56b52010-01-18 19:35:47 +0000540 E->setTypeSourceInfo(Reader.GetTypeSourceInfo(Record, Idx));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000541 E->setInitializer(cast<Expr>(StmtStack.back()));
542 E->setFileScope(Record[Idx++]);
543 return 1;
544}
545
546unsigned PCHStmtReader::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
547 VisitExpr(E);
548 E->setBase(cast<Expr>(StmtStack.back()));
549 E->setAccessor(Reader.GetIdentifierInfo(Record, Idx));
550 E->setAccessorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
551 return 1;
552}
553
554unsigned PCHStmtReader::VisitInitListExpr(InitListExpr *E) {
555 VisitExpr(E);
556 unsigned NumInits = Record[Idx++];
Ted Kremenek709210f2010-04-13 23:39:13 +0000557 E->reserveInits(*Reader.getContext(), NumInits);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000558 for (unsigned I = 0; I != NumInits; ++I)
Ted Kremenek709210f2010-04-13 23:39:13 +0000559 E->updateInit(*Reader.getContext(), I,
Chris Lattner4c6f9522009-04-27 05:14:47 +0000560 cast<Expr>(StmtStack[StmtStack.size() - NumInits - 1 + I]));
561 E->setSyntacticForm(cast_or_null<InitListExpr>(StmtStack.back()));
562 E->setLBraceLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
563 E->setRBraceLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
564 E->setInitializedFieldInUnion(
565 cast_or_null<FieldDecl>(Reader.GetDecl(Record[Idx++])));
566 E->sawArrayRangeDesignator(Record[Idx++]);
567 return NumInits + 1;
568}
569
570unsigned PCHStmtReader::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
571 typedef DesignatedInitExpr::Designator Designator;
572
573 VisitExpr(E);
574 unsigned NumSubExprs = Record[Idx++];
575 assert(NumSubExprs == E->getNumSubExprs() && "Wrong number of subexprs");
576 for (unsigned I = 0; I != NumSubExprs; ++I)
577 E->setSubExpr(I, cast<Expr>(StmtStack[StmtStack.size() - NumSubExprs + I]));
578 E->setEqualOrColonLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
579 E->setGNUSyntax(Record[Idx++]);
580
581 llvm::SmallVector<Designator, 4> Designators;
582 while (Idx < Record.size()) {
583 switch ((pch::DesignatorTypes)Record[Idx++]) {
584 case pch::DESIG_FIELD_DECL: {
585 FieldDecl *Field = cast<FieldDecl>(Reader.GetDecl(Record[Idx++]));
Mike Stump1eb44332009-09-09 15:08:12 +0000586 SourceLocation DotLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000587 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000588 SourceLocation FieldLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000589 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000590 Designators.push_back(Designator(Field->getIdentifier(), DotLoc,
Chris Lattner4c6f9522009-04-27 05:14:47 +0000591 FieldLoc));
592 Designators.back().setField(Field);
593 break;
594 }
595
596 case pch::DESIG_FIELD_NAME: {
597 const IdentifierInfo *Name = Reader.GetIdentifierInfo(Record, Idx);
Mike Stump1eb44332009-09-09 15:08:12 +0000598 SourceLocation DotLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000599 = SourceLocation::getFromRawEncoding(Record[Idx++]);
Mike Stump1eb44332009-09-09 15:08:12 +0000600 SourceLocation FieldLoc
Chris Lattner4c6f9522009-04-27 05:14:47 +0000601 = SourceLocation::getFromRawEncoding(Record[Idx++]);
602 Designators.push_back(Designator(Name, DotLoc, FieldLoc));
603 break;
604 }
Mike Stump1eb44332009-09-09 15:08:12 +0000605
Chris Lattner4c6f9522009-04-27 05:14:47 +0000606 case pch::DESIG_ARRAY: {
607 unsigned Index = Record[Idx++];
608 SourceLocation LBracketLoc
609 = SourceLocation::getFromRawEncoding(Record[Idx++]);
610 SourceLocation RBracketLoc
611 = SourceLocation::getFromRawEncoding(Record[Idx++]);
612 Designators.push_back(Designator(Index, LBracketLoc, RBracketLoc));
613 break;
614 }
615
616 case pch::DESIG_ARRAY_RANGE: {
617 unsigned Index = Record[Idx++];
618 SourceLocation LBracketLoc
619 = SourceLocation::getFromRawEncoding(Record[Idx++]);
620 SourceLocation EllipsisLoc
621 = SourceLocation::getFromRawEncoding(Record[Idx++]);
622 SourceLocation RBracketLoc
623 = SourceLocation::getFromRawEncoding(Record[Idx++]);
624 Designators.push_back(Designator(Index, LBracketLoc, EllipsisLoc,
625 RBracketLoc));
626 break;
627 }
628 }
629 }
Douglas Gregor319d57f2010-01-06 23:17:19 +0000630 E->setDesignators(*Reader.getContext(),
631 Designators.data(), Designators.size());
Chris Lattner4c6f9522009-04-27 05:14:47 +0000632
633 return NumSubExprs;
634}
635
636unsigned PCHStmtReader::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
637 VisitExpr(E);
638 return 0;
639}
640
641unsigned PCHStmtReader::VisitVAArgExpr(VAArgExpr *E) {
642 VisitExpr(E);
643 E->setSubExpr(cast<Expr>(StmtStack.back()));
644 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
645 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
646 return 1;
647}
648
649unsigned PCHStmtReader::VisitAddrLabelExpr(AddrLabelExpr *E) {
650 VisitExpr(E);
651 E->setAmpAmpLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
652 E->setLabelLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
653 Reader.SetLabelOf(E, Record[Idx++]);
654 return 0;
655}
656
657unsigned PCHStmtReader::VisitStmtExpr(StmtExpr *E) {
658 VisitExpr(E);
659 E->setLParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
660 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
661 E->setSubStmt(cast_or_null<CompoundStmt>(StmtStack.back()));
662 return 1;
663}
664
665unsigned PCHStmtReader::VisitTypesCompatibleExpr(TypesCompatibleExpr *E) {
666 VisitExpr(E);
667 E->setArgType1(Reader.GetType(Record[Idx++]));
668 E->setArgType2(Reader.GetType(Record[Idx++]));
669 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
670 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
671 return 0;
672}
673
674unsigned PCHStmtReader::VisitChooseExpr(ChooseExpr *E) {
675 VisitExpr(E);
676 E->setCond(cast<Expr>(StmtStack[StmtStack.size() - 3]));
677 E->setLHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
678 E->setRHS(cast_or_null<Expr>(StmtStack[StmtStack.size() - 1]));
679 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
680 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
681 return 3;
682}
683
684unsigned PCHStmtReader::VisitGNUNullExpr(GNUNullExpr *E) {
685 VisitExpr(E);
686 E->setTokenLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
687 return 0;
688}
689
690unsigned PCHStmtReader::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
691 VisitExpr(E);
692 unsigned NumExprs = Record[Idx++];
Mike Stump1eb44332009-09-09 15:08:12 +0000693 E->setExprs(*Reader.getContext(),
Nate Begeman888376a2009-08-12 02:28:50 +0000694 (Expr **)&StmtStack[StmtStack.size() - NumExprs], NumExprs);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000695 E->setBuiltinLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
696 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
697 return NumExprs;
698}
699
700unsigned PCHStmtReader::VisitBlockExpr(BlockExpr *E) {
701 VisitExpr(E);
702 E->setBlockDecl(cast_or_null<BlockDecl>(Reader.GetDecl(Record[Idx++])));
703 E->setHasBlockDeclRefExprs(Record[Idx++]);
704 return 0;
705}
706
707unsigned PCHStmtReader::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
708 VisitExpr(E);
709 E->setDecl(cast<ValueDecl>(Reader.GetDecl(Record[Idx++])));
710 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
711 E->setByRef(Record[Idx++]);
Fariborz Jahanian9b0b57c2009-06-20 00:02:26 +0000712 E->setConstQualAdded(Record[Idx++]);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000713 return 0;
714}
715
716//===----------------------------------------------------------------------===//
717// Objective-C Expressions and Statements
718
719unsigned PCHStmtReader::VisitObjCStringLiteral(ObjCStringLiteral *E) {
720 VisitExpr(E);
721 E->setString(cast<StringLiteral>(StmtStack.back()));
722 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
723 return 1;
724}
725
726unsigned PCHStmtReader::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
727 VisitExpr(E);
Douglas Gregor81d34662010-04-20 15:39:42 +0000728 E->setEncodedTypeSourceInfo(Reader.GetTypeSourceInfo(Record, Idx));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000729 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
730 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
731 return 0;
732}
733
734unsigned PCHStmtReader::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
735 VisitExpr(E);
736 E->setSelector(Reader.GetSelector(Record, Idx));
737 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
738 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
739 return 0;
740}
741
742unsigned PCHStmtReader::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
743 VisitExpr(E);
744 E->setProtocol(cast<ObjCProtocolDecl>(Reader.GetDecl(Record[Idx++])));
745 E->setAtLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
746 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
747 return 0;
748}
749
750unsigned PCHStmtReader::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
751 VisitExpr(E);
752 E->setDecl(cast<ObjCIvarDecl>(Reader.GetDecl(Record[Idx++])));
753 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
754 E->setBase(cast<Expr>(StmtStack.back()));
755 E->setIsArrow(Record[Idx++]);
756 E->setIsFreeIvar(Record[Idx++]);
757 return 1;
758}
759
760unsigned PCHStmtReader::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
761 VisitExpr(E);
762 E->setProperty(cast<ObjCPropertyDecl>(Reader.GetDecl(Record[Idx++])));
763 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
764 E->setBase(cast<Expr>(StmtStack.back()));
765 return 1;
766}
767
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000768unsigned PCHStmtReader::VisitObjCImplicitSetterGetterRefExpr(
769 ObjCImplicitSetterGetterRefExpr *E) {
Chris Lattner4c6f9522009-04-27 05:14:47 +0000770 VisitExpr(E);
771 E->setGetterMethod(
772 cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
773 E->setSetterMethod(
774 cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
Fariborz Jahaniand2ae5aa2009-08-18 21:37:33 +0000775 E->setInterfaceDecl(
Chris Lattner4c6f9522009-04-27 05:14:47 +0000776 cast_or_null<ObjCInterfaceDecl>(Reader.GetDecl(Record[Idx++])));
777 E->setBase(cast_or_null<Expr>(StmtStack.back()));
778 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
779 E->setClassLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
780 return 1;
781}
782
783unsigned PCHStmtReader::VisitObjCMessageExpr(ObjCMessageExpr *E) {
784 VisitExpr(E);
Douglas Gregor04badcf2010-04-21 00:45:42 +0000785 assert(Record[Idx] == E->getNumArgs());
786 ++Idx;
787 ObjCMessageExpr::ReceiverKind Kind
788 = static_cast<ObjCMessageExpr::ReceiverKind>(Record[Idx++]);
789 switch (Kind) {
790 case ObjCMessageExpr::Instance:
791 E->setInstanceReceiver(
792 cast_or_null<Expr>(StmtStack[StmtStack.size() - E->getNumArgs() - 1]));
793 break;
794
795 case ObjCMessageExpr::Class:
796 E->setClassReceiver(Reader.GetTypeSourceInfo(Record, Idx));
797 break;
798
799 case ObjCMessageExpr::SuperClass:
800 case ObjCMessageExpr::SuperInstance: {
801 QualType T = Reader.GetType(Record[Idx++]);
802 SourceLocation SuperLoc = SourceLocation::getFromRawEncoding(Record[Idx++]);
803 E->setSuper(SuperLoc, T, Kind == ObjCMessageExpr::SuperInstance);
804 break;
805 }
806 }
807
808 assert(Kind == E->getReceiverKind());
809
810 if (Record[Idx++])
811 E->setMethodDecl(cast_or_null<ObjCMethodDecl>(Reader.GetDecl(Record[Idx++])));
812 else
813 E->setSelector(Reader.GetSelector(Record, Idx));
814
Chris Lattner4c6f9522009-04-27 05:14:47 +0000815 E->setLeftLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
816 E->setRightLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
Chris Lattner4c6f9522009-04-27 05:14:47 +0000817
818 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
819 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
Douglas Gregor04badcf2010-04-21 00:45:42 +0000820 return E->getNumArgs() + (Kind == ObjCMessageExpr::Instance);
Chris Lattner4c6f9522009-04-27 05:14:47 +0000821}
822
823unsigned PCHStmtReader::VisitObjCSuperExpr(ObjCSuperExpr *E) {
824 VisitExpr(E);
825 E->setLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
826 return 0;
827}
828
829unsigned PCHStmtReader::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
830 VisitStmt(S);
831 S->setElement(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 3]));
832 S->setCollection(cast_or_null<Expr>(StmtStack[StmtStack.size() - 2]));
833 S->setBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
834 S->setForLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
835 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
836 return 3;
837}
838
839unsigned PCHStmtReader::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
840 VisitStmt(S);
841 S->setCatchBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
842 S->setNextCatchStmt(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
843 S->setCatchParamDecl(cast_or_null<ParmVarDecl>(Reader.GetDecl(Record[Idx++])));
844 S->setAtCatchLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
845 S->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
846 return 2;
847}
848
849unsigned PCHStmtReader::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
850 VisitStmt(S);
851 S->setFinallyBody(StmtStack.back());
852 S->setAtFinallyLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
853 return 1;
854}
855
856unsigned PCHStmtReader::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
857 VisitStmt(S);
858 S->setTryBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 3]));
859 S->setCatchStmts(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
860 S->setFinallyStmt(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
861 S->setAtTryLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
862 return 3;
863}
864
865unsigned PCHStmtReader::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
866 VisitStmt(S);
867 S->setSynchExpr(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 2]));
868 S->setSynchBody(cast_or_null<Stmt>(StmtStack[StmtStack.size() - 1]));
869 S->setAtSynchronizedLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
870 return 2;
871}
872
873unsigned PCHStmtReader::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
874 VisitStmt(S);
875 S->setThrowExpr(StmtStack.back());
876 S->setThrowLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
877 return 1;
878}
879
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000880//===----------------------------------------------------------------------===//
881// C++ Expressions and Statements
882
883unsigned PCHStmtReader::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
884 unsigned num = VisitCallExpr(E);
885 E->setOperator((OverloadedOperatorKind)Record[Idx++]);
886 return num;
887}
888
Douglas Gregor39da0b82009-09-09 23:08:42 +0000889unsigned PCHStmtReader::VisitCXXConstructExpr(CXXConstructExpr *E) {
890 VisitExpr(E);
891 E->setConstructor(cast<CXXConstructorDecl>(Reader.GetDecl(Record[Idx++])));
Douglas Gregor99a2e602009-12-16 01:38:02 +0000892 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
Douglas Gregor39da0b82009-09-09 23:08:42 +0000893 E->setElidable(Record[Idx++]);
Douglas Gregor16006c92009-12-16 18:50:27 +0000894 E->setRequiresZeroInitialization(Record[Idx++]);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000895 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
896 E->setArg(I, cast<Expr>(StmtStack[StmtStack.size() - N + I]));
897 return E->getNumArgs();
898}
Chris Lattner4c6f9522009-04-27 05:14:47 +0000899
Sam Weinigce757a72010-01-16 21:21:01 +0000900unsigned PCHStmtReader::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
901 unsigned num = VisitExplicitCastExpr(E);
902 E->setOperatorLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
903 return num;
904}
905
906unsigned PCHStmtReader::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
907 return VisitCXXNamedCastExpr(E);
908}
909
910unsigned PCHStmtReader::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
911 return VisitCXXNamedCastExpr(E);
912}
913
914unsigned PCHStmtReader::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
915 return VisitCXXNamedCastExpr(E);
916}
917
918unsigned PCHStmtReader::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
919 return VisitCXXNamedCastExpr(E);
920}
921
922unsigned PCHStmtReader::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
923 unsigned num = VisitExplicitCastExpr(E);
924 E->setTypeBeginLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
925 E->setRParenLoc(SourceLocation::getFromRawEncoding(Record[Idx++]));
926 return num;
927}
928
Sam Weinigeb7f9612010-02-07 06:32:43 +0000929unsigned PCHStmtReader::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
930 VisitExpr(E);
931 E->setValue(Record[Idx++]);
932 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
933 return 0;
934}
935
936unsigned PCHStmtReader::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
937 VisitExpr(E);
938 E->setLocation(SourceLocation::getFromRawEncoding(Record[Idx++]));
939 return 0;
940}
941
Chris Lattner52e97d12009-04-27 05:41:06 +0000942// Within the bitstream, expressions are stored in Reverse Polish
943// Notation, with each of the subexpressions preceding the
944// expression they are stored in. To evaluate expressions, we
945// continue reading expressions and placing them on the stack, with
946// expressions having operands removing those operands from the
947// stack. Evaluation terminates when we see a STMT_STOP record, and
948// the single remaining expression on the stack is our result.
949Stmt *PCHReader::ReadStmt(llvm::BitstreamCursor &Cursor) {
Chris Lattner4c6f9522009-04-27 05:14:47 +0000950 RecordData Record;
951 unsigned Idx;
952 llvm::SmallVector<Stmt *, 16> StmtStack;
953 PCHStmtReader Reader(*this, Record, Idx, StmtStack);
954 Stmt::EmptyShell Empty;
955
956 while (true) {
Chris Lattner52e97d12009-04-27 05:41:06 +0000957 unsigned Code = Cursor.ReadCode();
Chris Lattner4c6f9522009-04-27 05:14:47 +0000958 if (Code == llvm::bitc::END_BLOCK) {
Chris Lattner52e97d12009-04-27 05:41:06 +0000959 if (Cursor.ReadBlockEnd()) {
Douglas Gregora02b1472009-04-28 21:53:25 +0000960 Error("error at end of block in PCH file");
Chris Lattner4c6f9522009-04-27 05:14:47 +0000961 return 0;
962 }
963 break;
964 }
965
966 if (Code == llvm::bitc::ENTER_SUBBLOCK) {
967 // No known subblocks, always skip them.
Chris Lattner52e97d12009-04-27 05:41:06 +0000968 Cursor.ReadSubBlockID();
969 if (Cursor.SkipBlock()) {
Douglas Gregora02b1472009-04-28 21:53:25 +0000970 Error("malformed block record in PCH file");
Chris Lattner4c6f9522009-04-27 05:14:47 +0000971 return 0;
972 }
973 continue;
974 }
975
976 if (Code == llvm::bitc::DEFINE_ABBREV) {
Chris Lattner52e97d12009-04-27 05:41:06 +0000977 Cursor.ReadAbbrevRecord();
Chris Lattner4c6f9522009-04-27 05:14:47 +0000978 continue;
979 }
980
981 Stmt *S = 0;
982 Idx = 0;
983 Record.clear();
984 bool Finished = false;
Chris Lattner52e97d12009-04-27 05:41:06 +0000985 switch ((pch::StmtCode)Cursor.ReadRecord(Code, Record)) {
Chris Lattner4c6f9522009-04-27 05:14:47 +0000986 case pch::STMT_STOP:
987 Finished = true;
988 break;
989
Mike Stump1eb44332009-09-09 15:08:12 +0000990 case pch::STMT_NULL_PTR:
991 S = 0;
Chris Lattner4c6f9522009-04-27 05:14:47 +0000992 break;
993
994 case pch::STMT_NULL:
995 S = new (Context) NullStmt(Empty);
996 break;
997
998 case pch::STMT_COMPOUND:
999 S = new (Context) CompoundStmt(Empty);
1000 break;
1001
1002 case pch::STMT_CASE:
1003 S = new (Context) CaseStmt(Empty);
1004 break;
1005
1006 case pch::STMT_DEFAULT:
1007 S = new (Context) DefaultStmt(Empty);
1008 break;
1009
1010 case pch::STMT_LABEL:
1011 S = new (Context) LabelStmt(Empty);
1012 break;
1013
1014 case pch::STMT_IF:
1015 S = new (Context) IfStmt(Empty);
1016 break;
1017
1018 case pch::STMT_SWITCH:
1019 S = new (Context) SwitchStmt(Empty);
1020 break;
1021
1022 case pch::STMT_WHILE:
1023 S = new (Context) WhileStmt(Empty);
1024 break;
1025
1026 case pch::STMT_DO:
1027 S = new (Context) DoStmt(Empty);
1028 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001029
Chris Lattner4c6f9522009-04-27 05:14:47 +00001030 case pch::STMT_FOR:
1031 S = new (Context) ForStmt(Empty);
1032 break;
1033
1034 case pch::STMT_GOTO:
1035 S = new (Context) GotoStmt(Empty);
1036 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001037
Chris Lattner4c6f9522009-04-27 05:14:47 +00001038 case pch::STMT_INDIRECT_GOTO:
1039 S = new (Context) IndirectGotoStmt(Empty);
1040 break;
1041
1042 case pch::STMT_CONTINUE:
1043 S = new (Context) ContinueStmt(Empty);
1044 break;
1045
1046 case pch::STMT_BREAK:
1047 S = new (Context) BreakStmt(Empty);
1048 break;
1049
1050 case pch::STMT_RETURN:
1051 S = new (Context) ReturnStmt(Empty);
1052 break;
1053
1054 case pch::STMT_DECL:
1055 S = new (Context) DeclStmt(Empty);
1056 break;
1057
1058 case pch::STMT_ASM:
1059 S = new (Context) AsmStmt(Empty);
1060 break;
1061
1062 case pch::EXPR_PREDEFINED:
1063 S = new (Context) PredefinedExpr(Empty);
1064 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001065
1066 case pch::EXPR_DECL_REF:
1067 S = new (Context) DeclRefExpr(Empty);
Chris Lattner4c6f9522009-04-27 05:14:47 +00001068 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001069
1070 case pch::EXPR_INTEGER_LITERAL:
Chris Lattner4c6f9522009-04-27 05:14:47 +00001071 S = new (Context) IntegerLiteral(Empty);
1072 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001073
Chris Lattner4c6f9522009-04-27 05:14:47 +00001074 case pch::EXPR_FLOATING_LITERAL:
1075 S = new (Context) FloatingLiteral(Empty);
1076 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001077
Chris Lattner4c6f9522009-04-27 05:14:47 +00001078 case pch::EXPR_IMAGINARY_LITERAL:
1079 S = new (Context) ImaginaryLiteral(Empty);
1080 break;
1081
1082 case pch::EXPR_STRING_LITERAL:
Mike Stump1eb44332009-09-09 15:08:12 +00001083 S = StringLiteral::CreateEmpty(*Context,
Chris Lattner4c6f9522009-04-27 05:14:47 +00001084 Record[PCHStmtReader::NumExprFields + 1]);
1085 break;
1086
1087 case pch::EXPR_CHARACTER_LITERAL:
1088 S = new (Context) CharacterLiteral(Empty);
1089 break;
1090
1091 case pch::EXPR_PAREN:
1092 S = new (Context) ParenExpr(Empty);
1093 break;
1094
1095 case pch::EXPR_UNARY_OPERATOR:
1096 S = new (Context) UnaryOperator(Empty);
1097 break;
1098
1099 case pch::EXPR_SIZEOF_ALIGN_OF:
1100 S = new (Context) SizeOfAlignOfExpr(Empty);
1101 break;
1102
1103 case pch::EXPR_ARRAY_SUBSCRIPT:
1104 S = new (Context) ArraySubscriptExpr(Empty);
1105 break;
1106
1107 case pch::EXPR_CALL:
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001108 S = new (Context) CallExpr(*Context, Stmt::CallExprClass, Empty);
Chris Lattner4c6f9522009-04-27 05:14:47 +00001109 break;
1110
1111 case pch::EXPR_MEMBER:
1112 S = new (Context) MemberExpr(Empty);
1113 break;
1114
1115 case pch::EXPR_BINARY_OPERATOR:
1116 S = new (Context) BinaryOperator(Empty);
1117 break;
1118
1119 case pch::EXPR_COMPOUND_ASSIGN_OPERATOR:
1120 S = new (Context) CompoundAssignOperator(Empty);
1121 break;
1122
1123 case pch::EXPR_CONDITIONAL_OPERATOR:
1124 S = new (Context) ConditionalOperator(Empty);
1125 break;
1126
1127 case pch::EXPR_IMPLICIT_CAST:
1128 S = new (Context) ImplicitCastExpr(Empty);
1129 break;
1130
1131 case pch::EXPR_CSTYLE_CAST:
1132 S = new (Context) CStyleCastExpr(Empty);
1133 break;
1134
1135 case pch::EXPR_COMPOUND_LITERAL:
1136 S = new (Context) CompoundLiteralExpr(Empty);
1137 break;
1138
1139 case pch::EXPR_EXT_VECTOR_ELEMENT:
1140 S = new (Context) ExtVectorElementExpr(Empty);
1141 break;
1142
1143 case pch::EXPR_INIT_LIST:
Ted Kremenek709210f2010-04-13 23:39:13 +00001144 S = new (Context) InitListExpr(*getContext(), Empty);
Chris Lattner4c6f9522009-04-27 05:14:47 +00001145 break;
1146
1147 case pch::EXPR_DESIGNATED_INIT:
Chris Lattnerd1d64a02009-04-27 21:45:14 +00001148 S = DesignatedInitExpr::CreateEmpty(*Context,
Chris Lattner4c6f9522009-04-27 05:14:47 +00001149 Record[PCHStmtReader::NumExprFields] - 1);
Mike Stump1eb44332009-09-09 15:08:12 +00001150
Chris Lattner4c6f9522009-04-27 05:14:47 +00001151 break;
1152
1153 case pch::EXPR_IMPLICIT_VALUE_INIT:
1154 S = new (Context) ImplicitValueInitExpr(Empty);
1155 break;
1156
1157 case pch::EXPR_VA_ARG:
1158 S = new (Context) VAArgExpr(Empty);
1159 break;
1160
1161 case pch::EXPR_ADDR_LABEL:
1162 S = new (Context) AddrLabelExpr(Empty);
1163 break;
1164
1165 case pch::EXPR_STMT:
1166 S = new (Context) StmtExpr(Empty);
1167 break;
1168
1169 case pch::EXPR_TYPES_COMPATIBLE:
1170 S = new (Context) TypesCompatibleExpr(Empty);
1171 break;
1172
1173 case pch::EXPR_CHOOSE:
1174 S = new (Context) ChooseExpr(Empty);
1175 break;
1176
1177 case pch::EXPR_GNU_NULL:
1178 S = new (Context) GNUNullExpr(Empty);
1179 break;
1180
1181 case pch::EXPR_SHUFFLE_VECTOR:
1182 S = new (Context) ShuffleVectorExpr(Empty);
1183 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001184
Chris Lattner4c6f9522009-04-27 05:14:47 +00001185 case pch::EXPR_BLOCK:
1186 S = new (Context) BlockExpr(Empty);
1187 break;
1188
1189 case pch::EXPR_BLOCK_DECL_REF:
1190 S = new (Context) BlockDeclRefExpr(Empty);
1191 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001192
Chris Lattner4c6f9522009-04-27 05:14:47 +00001193 case pch::EXPR_OBJC_STRING_LITERAL:
1194 S = new (Context) ObjCStringLiteral(Empty);
1195 break;
1196 case pch::EXPR_OBJC_ENCODE:
1197 S = new (Context) ObjCEncodeExpr(Empty);
1198 break;
1199 case pch::EXPR_OBJC_SELECTOR_EXPR:
1200 S = new (Context) ObjCSelectorExpr(Empty);
1201 break;
1202 case pch::EXPR_OBJC_PROTOCOL_EXPR:
1203 S = new (Context) ObjCProtocolExpr(Empty);
1204 break;
1205 case pch::EXPR_OBJC_IVAR_REF_EXPR:
1206 S = new (Context) ObjCIvarRefExpr(Empty);
1207 break;
1208 case pch::EXPR_OBJC_PROPERTY_REF_EXPR:
1209 S = new (Context) ObjCPropertyRefExpr(Empty);
1210 break;
1211 case pch::EXPR_OBJC_KVC_REF_EXPR:
Fariborz Jahanian09105f52009-08-20 17:02:02 +00001212 S = new (Context) ObjCImplicitSetterGetterRefExpr(Empty);
Chris Lattner4c6f9522009-04-27 05:14:47 +00001213 break;
1214 case pch::EXPR_OBJC_MESSAGE_EXPR:
Douglas Gregor04badcf2010-04-21 00:45:42 +00001215 S = ObjCMessageExpr::CreateEmpty(*Context,
1216 Record[PCHStmtReader::NumExprFields]);
Chris Lattner4c6f9522009-04-27 05:14:47 +00001217 break;
1218 case pch::EXPR_OBJC_SUPER_EXPR:
1219 S = new (Context) ObjCSuperExpr(Empty);
1220 break;
Steve Narofff242b1b2009-07-24 17:54:45 +00001221 case pch::EXPR_OBJC_ISA:
1222 S = new (Context) ObjCIsaExpr(Empty);
1223 break;
Chris Lattner4c6f9522009-04-27 05:14:47 +00001224 case pch::STMT_OBJC_FOR_COLLECTION:
1225 S = new (Context) ObjCForCollectionStmt(Empty);
1226 break;
1227 case pch::STMT_OBJC_CATCH:
1228 S = new (Context) ObjCAtCatchStmt(Empty);
1229 break;
1230 case pch::STMT_OBJC_FINALLY:
1231 S = new (Context) ObjCAtFinallyStmt(Empty);
1232 break;
1233 case pch::STMT_OBJC_AT_TRY:
1234 S = new (Context) ObjCAtTryStmt(Empty);
1235 break;
1236 case pch::STMT_OBJC_AT_SYNCHRONIZED:
1237 S = new (Context) ObjCAtSynchronizedStmt(Empty);
1238 break;
1239 case pch::STMT_OBJC_AT_THROW:
1240 S = new (Context) ObjCAtThrowStmt(Empty);
1241 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001242
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001243 case pch::EXPR_CXX_OPERATOR_CALL:
1244 S = new (Context) CXXOperatorCallExpr(*Context, Empty);
1245 break;
Douglas Gregor39da0b82009-09-09 23:08:42 +00001246
1247 case pch::EXPR_CXX_CONSTRUCT:
1248 S = new (Context) CXXConstructExpr(Empty, *Context,
1249 Record[PCHStmtReader::NumExprFields + 2]);
1250 break;
Sam Weinigce757a72010-01-16 21:21:01 +00001251
1252 case pch::EXPR_CXX_STATIC_CAST:
1253 S = new (Context) CXXStaticCastExpr(Empty);
1254 break;
1255
1256 case pch::EXPR_CXX_DYNAMIC_CAST:
1257 S = new (Context) CXXDynamicCastExpr(Empty);
1258 break;
1259
1260 case pch::EXPR_CXX_REINTERPRET_CAST:
1261 S = new (Context) CXXReinterpretCastExpr(Empty);
1262 break;
1263
1264 case pch::EXPR_CXX_CONST_CAST:
1265 S = new (Context) CXXConstCastExpr(Empty);
1266 break;
1267
1268 case pch::EXPR_CXX_FUNCTIONAL_CAST:
1269 S = new (Context) CXXFunctionalCastExpr(Empty);
1270 break;
1271
Sam Weinigeb7f9612010-02-07 06:32:43 +00001272 case pch::EXPR_CXX_BOOL_LITERAL:
1273 S = new (Context) CXXBoolLiteralExpr(Empty);
1274 break;
Sam Weinigce757a72010-01-16 21:21:01 +00001275
Sam Weinigeb7f9612010-02-07 06:32:43 +00001276 case pch::EXPR_CXX_NULL_PTR_LITERAL:
1277 S = new (Context) CXXNullPtrLiteralExpr(Empty);
1278 break;
Chris Lattner4c6f9522009-04-27 05:14:47 +00001279 }
1280
1281 // We hit a STMT_STOP, so we're done with this expression.
1282 if (Finished)
1283 break;
1284
1285 ++NumStatementsRead;
1286
1287 if (S) {
1288 unsigned NumSubStmts = Reader.Visit(S);
1289 while (NumSubStmts > 0) {
1290 StmtStack.pop_back();
1291 --NumSubStmts;
1292 }
1293 }
1294
1295 assert(Idx == Record.size() && "Invalid deserialization of statement");
1296 StmtStack.push_back(S);
1297 }
1298 assert(StmtStack.size() == 1 && "Extra expressions on stack!");
Chris Lattner4c6f9522009-04-27 05:14:47 +00001299 return StmtStack.back();
1300}