blob: a3f18dea39a3caa617301a8b684a11b9fcc9fd48 [file] [log] [blame]
Chris Lattner63377d52009-04-27 06:20:01 +00001//===--- PCHWriterStmt.cpp - Statement and Expression Serialization -------===//
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// This file implements serialization for Statements and Expressions.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/Frontend/PCHWriter.h"
Douglas Gregor39da0b82009-09-09 23:08:42 +000015#include "clang/AST/DeclCXX.h"
Chris Lattner63377d52009-04-27 06:20:01 +000016#include "clang/AST/DeclObjC.h"
17#include "clang/AST/StmtVisitor.h"
18#include "llvm/Bitcode/BitstreamWriter.h"
19using namespace clang;
20
21//===----------------------------------------------------------------------===//
22// Statement/expression serialization
23//===----------------------------------------------------------------------===//
24
25namespace {
26 class PCHStmtWriter : public StmtVisitor<PCHStmtWriter, void> {
Chris Lattner63377d52009-04-27 06:20:01 +000027 PCHWriter &Writer;
28 PCHWriter::RecordData &Record;
29
30 public:
31 pch::StmtCode Code;
32
33 PCHStmtWriter(PCHWriter &Writer, PCHWriter::RecordData &Record)
34 : Writer(Writer), Record(Record) { }
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +000035
36 void
37 AddExplicitTemplateArgumentList(const ExplicitTemplateArgumentList &Args);
Chris Lattner63377d52009-04-27 06:20:01 +000038
39 void VisitStmt(Stmt *S);
40 void VisitNullStmt(NullStmt *S);
41 void VisitCompoundStmt(CompoundStmt *S);
42 void VisitSwitchCase(SwitchCase *S);
43 void VisitCaseStmt(CaseStmt *S);
44 void VisitDefaultStmt(DefaultStmt *S);
45 void VisitLabelStmt(LabelStmt *S);
46 void VisitIfStmt(IfStmt *S);
47 void VisitSwitchStmt(SwitchStmt *S);
48 void VisitWhileStmt(WhileStmt *S);
49 void VisitDoStmt(DoStmt *S);
50 void VisitForStmt(ForStmt *S);
51 void VisitGotoStmt(GotoStmt *S);
52 void VisitIndirectGotoStmt(IndirectGotoStmt *S);
53 void VisitContinueStmt(ContinueStmt *S);
54 void VisitBreakStmt(BreakStmt *S);
55 void VisitReturnStmt(ReturnStmt *S);
56 void VisitDeclStmt(DeclStmt *S);
57 void VisitAsmStmt(AsmStmt *S);
58 void VisitExpr(Expr *E);
59 void VisitPredefinedExpr(PredefinedExpr *E);
60 void VisitDeclRefExpr(DeclRefExpr *E);
61 void VisitIntegerLiteral(IntegerLiteral *E);
62 void VisitFloatingLiteral(FloatingLiteral *E);
63 void VisitImaginaryLiteral(ImaginaryLiteral *E);
64 void VisitStringLiteral(StringLiteral *E);
65 void VisitCharacterLiteral(CharacterLiteral *E);
66 void VisitParenExpr(ParenExpr *E);
67 void VisitUnaryOperator(UnaryOperator *E);
Douglas Gregor8ecdb652010-04-28 22:16:22 +000068 void VisitOffsetOfExpr(OffsetOfExpr *E);
Chris Lattner63377d52009-04-27 06:20:01 +000069 void VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E);
70 void VisitArraySubscriptExpr(ArraySubscriptExpr *E);
71 void VisitCallExpr(CallExpr *E);
72 void VisitMemberExpr(MemberExpr *E);
73 void VisitCastExpr(CastExpr *E);
74 void VisitBinaryOperator(BinaryOperator *E);
75 void VisitCompoundAssignOperator(CompoundAssignOperator *E);
76 void VisitConditionalOperator(ConditionalOperator *E);
77 void VisitImplicitCastExpr(ImplicitCastExpr *E);
78 void VisitExplicitCastExpr(ExplicitCastExpr *E);
79 void VisitCStyleCastExpr(CStyleCastExpr *E);
80 void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
81 void VisitExtVectorElementExpr(ExtVectorElementExpr *E);
82 void VisitInitListExpr(InitListExpr *E);
83 void VisitDesignatedInitExpr(DesignatedInitExpr *E);
84 void VisitImplicitValueInitExpr(ImplicitValueInitExpr *E);
85 void VisitVAArgExpr(VAArgExpr *E);
86 void VisitAddrLabelExpr(AddrLabelExpr *E);
87 void VisitStmtExpr(StmtExpr *E);
88 void VisitTypesCompatibleExpr(TypesCompatibleExpr *E);
89 void VisitChooseExpr(ChooseExpr *E);
90 void VisitGNUNullExpr(GNUNullExpr *E);
91 void VisitShuffleVectorExpr(ShuffleVectorExpr *E);
92 void VisitBlockExpr(BlockExpr *E);
93 void VisitBlockDeclRefExpr(BlockDeclRefExpr *E);
Mike Stump1eb44332009-09-09 15:08:12 +000094
Chris Lattner63377d52009-04-27 06:20:01 +000095 // Objective-C Expressions
96 void VisitObjCStringLiteral(ObjCStringLiteral *E);
97 void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
98 void VisitObjCSelectorExpr(ObjCSelectorExpr *E);
99 void VisitObjCProtocolExpr(ObjCProtocolExpr *E);
100 void VisitObjCIvarRefExpr(ObjCIvarRefExpr *E);
101 void VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E);
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000102 void VisitObjCImplicitSetterGetterRefExpr(
103 ObjCImplicitSetterGetterRefExpr *E);
Chris Lattner63377d52009-04-27 06:20:01 +0000104 void VisitObjCMessageExpr(ObjCMessageExpr *E);
105 void VisitObjCSuperExpr(ObjCSuperExpr *E);
Steve Narofff242b1b2009-07-24 17:54:45 +0000106 void VisitObjCIsaExpr(ObjCIsaExpr *E);
Mike Stump1eb44332009-09-09 15:08:12 +0000107
108 // Objective-C Statements
Chris Lattner63377d52009-04-27 06:20:01 +0000109 void VisitObjCForCollectionStmt(ObjCForCollectionStmt *);
110 void VisitObjCAtCatchStmt(ObjCAtCatchStmt *);
111 void VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *);
112 void VisitObjCAtTryStmt(ObjCAtTryStmt *);
113 void VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *);
114 void VisitObjCAtThrowStmt(ObjCAtThrowStmt *);
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000115
Mike Stump1eb44332009-09-09 15:08:12 +0000116 // C++ Statements
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000117 void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
Chris Lattner1817bd42010-05-09 05:36:05 +0000118 void VisitCXXMemberCallExpr(CXXMemberCallExpr *E);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000119 void VisitCXXConstructExpr(CXXConstructExpr *E);
Sam Weinigce757a72010-01-16 21:21:01 +0000120 void VisitCXXNamedCastExpr(CXXNamedCastExpr *E);
121 void VisitCXXStaticCastExpr(CXXStaticCastExpr *E);
122 void VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E);
123 void VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E);
124 void VisitCXXConstCastExpr(CXXConstCastExpr *E);
125 void VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E);
Sam Weinigeb7f9612010-02-07 06:32:43 +0000126 void VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E);
127 void VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E);
Chris Lattner14ab24f2010-05-09 06:03:39 +0000128 void VisitCXXTypeidExpr(CXXTypeidExpr *E);
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000129 void VisitCXXThisExpr(CXXThisExpr *E);
130 void VisitCXXThrowExpr(CXXThrowExpr *E);
Chris Lattner030854b2010-05-09 06:40:08 +0000131 void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E);
Chris Lattnerd2598362010-05-10 00:25:06 +0000132 void VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +0000133
Chris Lattner59218632010-05-10 01:22:27 +0000134 void VisitCXXZeroInitValueExpr(CXXZeroInitValueExpr *E);
135 void VisitCXXNewExpr(CXXNewExpr *E);
Argyrios Kyrtzidis95fc98c2010-06-22 17:07:59 +0000136 void VisitCXXDeleteExpr(CXXDeleteExpr *E);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +0000137
Chris Lattnerd2598362010-05-10 00:25:06 +0000138 void VisitCXXExprWithTemporaries(CXXExprWithTemporaries *E);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +0000139 void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
140 void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +0000141
142 void VisitOverloadExpr(OverloadExpr *E);
143 void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E);
Argyrios Kyrtzidisbd65bb52010-06-25 09:03:34 +0000144 void VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E);
Chris Lattner63377d52009-04-27 06:20:01 +0000145 };
146}
147
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +0000148/// \brief Return the number of Exprs contained in the given TemplateArgument.
149static unsigned NumExprsContainedIn(const TemplateArgument &Arg) {
150 switch (Arg.getKind()) {
151 default: break;
152 case TemplateArgument::Expression:
153 return 1;
154 case TemplateArgument::Pack: {
155 unsigned Count = 0;
156 for (TemplateArgument::pack_iterator I=Arg.pack_begin(), E=Arg.pack_end();
157 I != E; ++I)
158 Count += NumExprsContainedIn(*I);
159 return Count;
160 }
161 }
162
163 return 0;
164}
165
166/// \brief Return the number of Exprs contained in the given
167/// ExplicitTemplateArgumentList.
168static unsigned NumExprsContainedIn(const ExplicitTemplateArgumentList &Args) {
169 unsigned Count = 0;
170 for (unsigned i=0; i != Args.NumTemplateArgs; ++i) {
171 const TemplateArgumentLoc &ArgLoc = Args.getTemplateArgs()[i];
172 const TemplateArgument &TemplA = ArgLoc.getArgument();
173 Count += NumExprsContainedIn(TemplA);
174 if (TemplA.getKind() == TemplateArgument::Expression &&
175 TemplA.getAsExpr() != ArgLoc.getLocInfo().getAsExpr())
176 ++Count; // 1 in TemplateArgumentLocInfo.
177 }
178
179 return Count;
180}
181
182void PCHStmtWriter::
183AddExplicitTemplateArgumentList(const ExplicitTemplateArgumentList &Args) {
184 Record.push_back(NumExprsContainedIn(Args));
185 Writer.AddSourceLocation(Args.LAngleLoc, Record);
186 Writer.AddSourceLocation(Args.RAngleLoc, Record);
187 for (unsigned i=0; i != Args.NumTemplateArgs; ++i)
188 Writer.AddTemplateArgumentLoc(Args.getTemplateArgs()[i], Record);
189}
190
Mike Stump1eb44332009-09-09 15:08:12 +0000191void PCHStmtWriter::VisitStmt(Stmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000192}
193
194void PCHStmtWriter::VisitNullStmt(NullStmt *S) {
195 VisitStmt(S);
196 Writer.AddSourceLocation(S->getSemiLoc(), Record);
197 Code = pch::STMT_NULL;
198}
199
200void PCHStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
201 VisitStmt(S);
202 Record.push_back(S->size());
203 for (CompoundStmt::body_iterator CS = S->body_begin(), CSEnd = S->body_end();
204 CS != CSEnd; ++CS)
205 Writer.WriteSubStmt(*CS);
206 Writer.AddSourceLocation(S->getLBracLoc(), Record);
207 Writer.AddSourceLocation(S->getRBracLoc(), Record);
208 Code = pch::STMT_COMPOUND;
209}
210
211void PCHStmtWriter::VisitSwitchCase(SwitchCase *S) {
212 VisitStmt(S);
213 Record.push_back(Writer.RecordSwitchCaseID(S));
214}
215
216void PCHStmtWriter::VisitCaseStmt(CaseStmt *S) {
217 VisitSwitchCase(S);
218 Writer.WriteSubStmt(S->getLHS());
219 Writer.WriteSubStmt(S->getRHS());
220 Writer.WriteSubStmt(S->getSubStmt());
221 Writer.AddSourceLocation(S->getCaseLoc(), Record);
Douglas Gregordbb26db2009-05-15 23:57:33 +0000222 Writer.AddSourceLocation(S->getEllipsisLoc(), Record);
223 Writer.AddSourceLocation(S->getColonLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000224 Code = pch::STMT_CASE;
225}
226
227void PCHStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
228 VisitSwitchCase(S);
229 Writer.WriteSubStmt(S->getSubStmt());
230 Writer.AddSourceLocation(S->getDefaultLoc(), Record);
Douglas Gregordbb26db2009-05-15 23:57:33 +0000231 Writer.AddSourceLocation(S->getColonLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000232 Code = pch::STMT_DEFAULT;
233}
234
235void PCHStmtWriter::VisitLabelStmt(LabelStmt *S) {
236 VisitStmt(S);
237 Writer.AddIdentifierRef(S->getID(), Record);
238 Writer.WriteSubStmt(S->getSubStmt());
239 Writer.AddSourceLocation(S->getIdentLoc(), Record);
240 Record.push_back(Writer.GetLabelID(S));
241 Code = pch::STMT_LABEL;
242}
243
244void PCHStmtWriter::VisitIfStmt(IfStmt *S) {
245 VisitStmt(S);
Douglas Gregor8cfe5a72009-11-23 23:44:04 +0000246 Writer.AddDeclRef(S->getConditionVariable(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000247 Writer.WriteSubStmt(S->getCond());
248 Writer.WriteSubStmt(S->getThen());
249 Writer.WriteSubStmt(S->getElse());
250 Writer.AddSourceLocation(S->getIfLoc(), Record);
Douglas Gregord06f6ca2009-05-15 18:53:42 +0000251 Writer.AddSourceLocation(S->getElseLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000252 Code = pch::STMT_IF;
253}
254
255void PCHStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
256 VisitStmt(S);
Douglas Gregord3d53012009-11-24 17:07:59 +0000257 Writer.AddDeclRef(S->getConditionVariable(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000258 Writer.WriteSubStmt(S->getCond());
259 Writer.WriteSubStmt(S->getBody());
260 Writer.AddSourceLocation(S->getSwitchLoc(), Record);
Mike Stump1eb44332009-09-09 15:08:12 +0000261 for (SwitchCase *SC = S->getSwitchCaseList(); SC;
Chris Lattner63377d52009-04-27 06:20:01 +0000262 SC = SC->getNextSwitchCase())
263 Record.push_back(Writer.getSwitchCaseID(SC));
264 Code = pch::STMT_SWITCH;
265}
266
267void PCHStmtWriter::VisitWhileStmt(WhileStmt *S) {
268 VisitStmt(S);
Douglas Gregor5656e142009-11-24 21:15:44 +0000269 Writer.AddDeclRef(S->getConditionVariable(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000270 Writer.WriteSubStmt(S->getCond());
271 Writer.WriteSubStmt(S->getBody());
272 Writer.AddSourceLocation(S->getWhileLoc(), Record);
273 Code = pch::STMT_WHILE;
274}
275
276void PCHStmtWriter::VisitDoStmt(DoStmt *S) {
277 VisitStmt(S);
278 Writer.WriteSubStmt(S->getCond());
279 Writer.WriteSubStmt(S->getBody());
280 Writer.AddSourceLocation(S->getDoLoc(), Record);
Douglas Gregor9f3ca2a2009-05-15 21:56:04 +0000281 Writer.AddSourceLocation(S->getWhileLoc(), Record);
Chris Lattner98913592009-06-12 23:04:47 +0000282 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000283 Code = pch::STMT_DO;
284}
285
286void PCHStmtWriter::VisitForStmt(ForStmt *S) {
287 VisitStmt(S);
288 Writer.WriteSubStmt(S->getInit());
289 Writer.WriteSubStmt(S->getCond());
Douglas Gregor99e9b4d2009-11-25 00:27:52 +0000290 Writer.AddDeclRef(S->getConditionVariable(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000291 Writer.WriteSubStmt(S->getInc());
292 Writer.WriteSubStmt(S->getBody());
293 Writer.AddSourceLocation(S->getForLoc(), Record);
Douglas Gregor5831c6a2009-05-15 22:12:32 +0000294 Writer.AddSourceLocation(S->getLParenLoc(), Record);
295 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000296 Code = pch::STMT_FOR;
297}
298
299void PCHStmtWriter::VisitGotoStmt(GotoStmt *S) {
300 VisitStmt(S);
301 Record.push_back(Writer.GetLabelID(S->getLabel()));
302 Writer.AddSourceLocation(S->getGotoLoc(), Record);
303 Writer.AddSourceLocation(S->getLabelLoc(), Record);
304 Code = pch::STMT_GOTO;
305}
306
307void PCHStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
308 VisitStmt(S);
309 Writer.AddSourceLocation(S->getGotoLoc(), Record);
Douglas Gregor5f1b9e62009-05-16 00:20:29 +0000310 Writer.AddSourceLocation(S->getStarLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000311 Writer.WriteSubStmt(S->getTarget());
312 Code = pch::STMT_INDIRECT_GOTO;
313}
314
315void PCHStmtWriter::VisitContinueStmt(ContinueStmt *S) {
316 VisitStmt(S);
317 Writer.AddSourceLocation(S->getContinueLoc(), Record);
318 Code = pch::STMT_CONTINUE;
319}
320
321void PCHStmtWriter::VisitBreakStmt(BreakStmt *S) {
322 VisitStmt(S);
323 Writer.AddSourceLocation(S->getBreakLoc(), Record);
324 Code = pch::STMT_BREAK;
325}
326
327void PCHStmtWriter::VisitReturnStmt(ReturnStmt *S) {
328 VisitStmt(S);
329 Writer.WriteSubStmt(S->getRetValue());
330 Writer.AddSourceLocation(S->getReturnLoc(), Record);
Douglas Gregor5077c382010-05-15 06:01:05 +0000331 Writer.AddDeclRef(S->getNRVOCandidate(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000332 Code = pch::STMT_RETURN;
333}
334
335void PCHStmtWriter::VisitDeclStmt(DeclStmt *S) {
336 VisitStmt(S);
337 Writer.AddSourceLocation(S->getStartLoc(), Record);
338 Writer.AddSourceLocation(S->getEndLoc(), Record);
339 DeclGroupRef DG = S->getDeclGroup();
340 for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
341 Writer.AddDeclRef(*D, Record);
342 Code = pch::STMT_DECL;
343}
344
345void PCHStmtWriter::VisitAsmStmt(AsmStmt *S) {
346 VisitStmt(S);
347 Record.push_back(S->getNumOutputs());
348 Record.push_back(S->getNumInputs());
349 Record.push_back(S->getNumClobbers());
350 Writer.AddSourceLocation(S->getAsmLoc(), Record);
351 Writer.AddSourceLocation(S->getRParenLoc(), Record);
352 Record.push_back(S->isVolatile());
353 Record.push_back(S->isSimple());
Mike Stump3b11fd32010-01-04 22:37:17 +0000354 Record.push_back(S->isMSAsm());
Chris Lattner63377d52009-04-27 06:20:01 +0000355 Writer.WriteSubStmt(S->getAsmString());
356
357 // Outputs
Anders Carlssonff93dbd2010-01-30 22:25:16 +0000358 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
359 Writer.AddIdentifierRef(S->getOutputIdentifier(I), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000360 Writer.WriteSubStmt(S->getOutputConstraintLiteral(I));
361 Writer.WriteSubStmt(S->getOutputExpr(I));
362 }
363
364 // Inputs
365 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
Anders Carlssonff93dbd2010-01-30 22:25:16 +0000366 Writer.AddIdentifierRef(S->getInputIdentifier(I), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000367 Writer.WriteSubStmt(S->getInputConstraintLiteral(I));
368 Writer.WriteSubStmt(S->getInputExpr(I));
369 }
370
371 // Clobbers
372 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
373 Writer.WriteSubStmt(S->getClobber(I));
374
375 Code = pch::STMT_ASM;
376}
377
378void PCHStmtWriter::VisitExpr(Expr *E) {
379 VisitStmt(E);
380 Writer.AddTypeRef(E->getType(), Record);
381 Record.push_back(E->isTypeDependent());
382 Record.push_back(E->isValueDependent());
383}
384
385void PCHStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
386 VisitExpr(E);
387 Writer.AddSourceLocation(E->getLocation(), Record);
388 Record.push_back(E->getIdentType()); // FIXME: stable encoding
389 Code = pch::EXPR_PREDEFINED;
390}
391
392void PCHStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
393 VisitExpr(E);
394 Writer.AddDeclRef(E->getDecl(), Record);
395 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregora2813ce2009-10-23 18:54:35 +0000396 // FIXME: write qualifier
397 // FIXME: write explicit template arguments
Chris Lattner63377d52009-04-27 06:20:01 +0000398 Code = pch::EXPR_DECL_REF;
399}
400
401void PCHStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
402 VisitExpr(E);
403 Writer.AddSourceLocation(E->getLocation(), Record);
404 Writer.AddAPInt(E->getValue(), Record);
405 Code = pch::EXPR_INTEGER_LITERAL;
406}
407
408void PCHStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
409 VisitExpr(E);
410 Writer.AddAPFloat(E->getValue(), Record);
411 Record.push_back(E->isExact());
412 Writer.AddSourceLocation(E->getLocation(), Record);
413 Code = pch::EXPR_FLOATING_LITERAL;
414}
415
416void PCHStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
417 VisitExpr(E);
418 Writer.WriteSubStmt(E->getSubExpr());
419 Code = pch::EXPR_IMAGINARY_LITERAL;
420}
421
422void PCHStmtWriter::VisitStringLiteral(StringLiteral *E) {
423 VisitExpr(E);
424 Record.push_back(E->getByteLength());
425 Record.push_back(E->getNumConcatenated());
426 Record.push_back(E->isWide());
427 // FIXME: String data should be stored as a blob at the end of the
428 // StringLiteral. However, we can't do so now because we have no
429 // provision for coping with abbreviations when we're jumping around
430 // the PCH file during deserialization.
Mike Stump1eb44332009-09-09 15:08:12 +0000431 Record.insert(Record.end(),
Chris Lattner63377d52009-04-27 06:20:01 +0000432 E->getStrData(), E->getStrData() + E->getByteLength());
433 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
434 Writer.AddSourceLocation(E->getStrTokenLoc(I), Record);
435 Code = pch::EXPR_STRING_LITERAL;
436}
437
438void PCHStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
439 VisitExpr(E);
440 Record.push_back(E->getValue());
Chris Lattner018d8e02009-08-24 17:39:36 +0000441 Writer.AddSourceLocation(E->getLocation(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000442 Record.push_back(E->isWide());
443 Code = pch::EXPR_CHARACTER_LITERAL;
444}
445
446void PCHStmtWriter::VisitParenExpr(ParenExpr *E) {
447 VisitExpr(E);
448 Writer.AddSourceLocation(E->getLParen(), Record);
449 Writer.AddSourceLocation(E->getRParen(), Record);
450 Writer.WriteSubStmt(E->getSubExpr());
451 Code = pch::EXPR_PAREN;
452}
453
454void PCHStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
455 VisitExpr(E);
456 Writer.WriteSubStmt(E->getSubExpr());
457 Record.push_back(E->getOpcode()); // FIXME: stable encoding
458 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
459 Code = pch::EXPR_UNARY_OPERATOR;
460}
461
Douglas Gregor8ecdb652010-04-28 22:16:22 +0000462void PCHStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
463 VisitExpr(E);
464 Record.push_back(E->getNumComponents());
465 Record.push_back(E->getNumExpressions());
466 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
467 Writer.AddSourceLocation(E->getRParenLoc(), Record);
468 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
469 for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
470 const OffsetOfExpr::OffsetOfNode &ON = E->getComponent(I);
471 Record.push_back(ON.getKind()); // FIXME: Stable encoding
472 Writer.AddSourceLocation(ON.getRange().getBegin(), Record);
473 Writer.AddSourceLocation(ON.getRange().getEnd(), Record);
474 switch (ON.getKind()) {
475 case OffsetOfExpr::OffsetOfNode::Array:
476 Record.push_back(ON.getArrayExprIndex());
477 break;
478
479 case OffsetOfExpr::OffsetOfNode::Field:
480 Writer.AddDeclRef(ON.getField(), Record);
481 break;
482
483 case OffsetOfExpr::OffsetOfNode::Identifier:
484 Writer.AddIdentifierRef(ON.getFieldName(), Record);
485 break;
Douglas Gregorcc8a5d52010-04-29 00:18:15 +0000486
487 case OffsetOfExpr::OffsetOfNode::Base:
488 // FIXME: Implement this!
489 llvm_unreachable("PCH for offsetof(base-specifier) not implemented");
490 break;
Douglas Gregor8ecdb652010-04-28 22:16:22 +0000491 }
492 }
493 for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
494 Writer.WriteSubStmt(E->getIndexExpr(I));
495 Code = pch::EXPR_OFFSETOF;
496}
497
Mike Stump1eb44332009-09-09 15:08:12 +0000498void PCHStmtWriter::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000499 VisitExpr(E);
500 Record.push_back(E->isSizeOf());
501 if (E->isArgumentType())
John McCalla93c9342009-12-07 02:54:59 +0000502 Writer.AddTypeSourceInfo(E->getArgumentTypeInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000503 else {
504 Record.push_back(0);
505 Writer.WriteSubStmt(E->getArgumentExpr());
506 }
507 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
508 Writer.AddSourceLocation(E->getRParenLoc(), Record);
509 Code = pch::EXPR_SIZEOF_ALIGN_OF;
510}
511
512void PCHStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
513 VisitExpr(E);
514 Writer.WriteSubStmt(E->getLHS());
515 Writer.WriteSubStmt(E->getRHS());
516 Writer.AddSourceLocation(E->getRBracketLoc(), Record);
517 Code = pch::EXPR_ARRAY_SUBSCRIPT;
518}
519
520void PCHStmtWriter::VisitCallExpr(CallExpr *E) {
521 VisitExpr(E);
522 Record.push_back(E->getNumArgs());
523 Writer.AddSourceLocation(E->getRParenLoc(), Record);
524 Writer.WriteSubStmt(E->getCallee());
525 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
526 Arg != ArgEnd; ++Arg)
527 Writer.WriteSubStmt(*Arg);
528 Code = pch::EXPR_CALL;
529}
530
531void PCHStmtWriter::VisitMemberExpr(MemberExpr *E) {
532 VisitExpr(E);
533 Writer.WriteSubStmt(E->getBase());
534 Writer.AddDeclRef(E->getMemberDecl(), Record);
535 Writer.AddSourceLocation(E->getMemberLoc(), Record);
536 Record.push_back(E->isArrow());
Douglas Gregor303e6d72009-08-31 23:44:04 +0000537 // FIXME: C++ nested-name-specifier
Douglas Gregorc4bf26f2009-09-01 00:37:14 +0000538 // FIXME: C++ template argument list
Chris Lattner63377d52009-04-27 06:20:01 +0000539 Code = pch::EXPR_MEMBER;
540}
541
Steve Narofff242b1b2009-07-24 17:54:45 +0000542void PCHStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
543 VisitExpr(E);
544 Writer.WriteSubStmt(E->getBase());
545 Writer.AddSourceLocation(E->getIsaMemberLoc(), Record);
546 Record.push_back(E->isArrow());
547 Code = pch::EXPR_OBJC_ISA;
548}
549
Chris Lattner63377d52009-04-27 06:20:01 +0000550void PCHStmtWriter::VisitCastExpr(CastExpr *E) {
551 VisitExpr(E);
552 Writer.WriteSubStmt(E->getSubExpr());
Anders Carlssoncdef2b72009-07-31 00:48:10 +0000553 Record.push_back(E->getCastKind()); // FIXME: stable encoding
Chris Lattner63377d52009-04-27 06:20:01 +0000554}
555
556void PCHStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
557 VisitExpr(E);
558 Writer.WriteSubStmt(E->getLHS());
559 Writer.WriteSubStmt(E->getRHS());
560 Record.push_back(E->getOpcode()); // FIXME: stable encoding
561 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
562 Code = pch::EXPR_BINARY_OPERATOR;
563}
564
565void PCHStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
566 VisitBinaryOperator(E);
567 Writer.AddTypeRef(E->getComputationLHSType(), Record);
568 Writer.AddTypeRef(E->getComputationResultType(), Record);
569 Code = pch::EXPR_COMPOUND_ASSIGN_OPERATOR;
570}
571
572void PCHStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
573 VisitExpr(E);
574 Writer.WriteSubStmt(E->getCond());
575 Writer.WriteSubStmt(E->getLHS());
576 Writer.WriteSubStmt(E->getRHS());
Douglas Gregor47e1f7c2009-08-26 14:37:04 +0000577 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
578 Writer.AddSourceLocation(E->getColonLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000579 Code = pch::EXPR_CONDITIONAL_OPERATOR;
580}
581
582void PCHStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
583 VisitCastExpr(E);
584 Record.push_back(E->isLvalueCast());
585 Code = pch::EXPR_IMPLICIT_CAST;
586}
587
588void PCHStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
589 VisitCastExpr(E);
John McCall9d125032010-01-15 18:39:57 +0000590 Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000591}
592
593void PCHStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
594 VisitExplicitCastExpr(E);
595 Writer.AddSourceLocation(E->getLParenLoc(), Record);
596 Writer.AddSourceLocation(E->getRParenLoc(), Record);
597 Code = pch::EXPR_CSTYLE_CAST;
598}
599
600void PCHStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
601 VisitExpr(E);
602 Writer.AddSourceLocation(E->getLParenLoc(), Record);
John McCall42f56b52010-01-18 19:35:47 +0000603 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000604 Writer.WriteSubStmt(E->getInitializer());
605 Record.push_back(E->isFileScope());
606 Code = pch::EXPR_COMPOUND_LITERAL;
607}
608
609void PCHStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
610 VisitExpr(E);
611 Writer.WriteSubStmt(E->getBase());
612 Writer.AddIdentifierRef(&E->getAccessor(), Record);
613 Writer.AddSourceLocation(E->getAccessorLoc(), Record);
614 Code = pch::EXPR_EXT_VECTOR_ELEMENT;
615}
616
617void PCHStmtWriter::VisitInitListExpr(InitListExpr *E) {
618 VisitExpr(E);
619 Record.push_back(E->getNumInits());
620 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
621 Writer.WriteSubStmt(E->getInit(I));
622 Writer.WriteSubStmt(E->getSyntacticForm());
623 Writer.AddSourceLocation(E->getLBraceLoc(), Record);
624 Writer.AddSourceLocation(E->getRBraceLoc(), Record);
625 Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
626 Record.push_back(E->hadArrayRangeDesignator());
627 Code = pch::EXPR_INIT_LIST;
628}
629
630void PCHStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
631 VisitExpr(E);
632 Record.push_back(E->getNumSubExprs());
633 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
634 Writer.WriteSubStmt(E->getSubExpr(I));
635 Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
636 Record.push_back(E->usesGNUSyntax());
637 for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
638 DEnd = E->designators_end();
639 D != DEnd; ++D) {
640 if (D->isFieldDesignator()) {
641 if (FieldDecl *Field = D->getField()) {
642 Record.push_back(pch::DESIG_FIELD_DECL);
643 Writer.AddDeclRef(Field, Record);
644 } else {
645 Record.push_back(pch::DESIG_FIELD_NAME);
646 Writer.AddIdentifierRef(D->getFieldName(), Record);
647 }
648 Writer.AddSourceLocation(D->getDotLoc(), Record);
649 Writer.AddSourceLocation(D->getFieldLoc(), Record);
650 } else if (D->isArrayDesignator()) {
651 Record.push_back(pch::DESIG_ARRAY);
652 Record.push_back(D->getFirstExprIndex());
653 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
654 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
655 } else {
656 assert(D->isArrayRangeDesignator() && "Unknown designator");
657 Record.push_back(pch::DESIG_ARRAY_RANGE);
658 Record.push_back(D->getFirstExprIndex());
659 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
660 Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
661 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
662 }
663 }
664 Code = pch::EXPR_DESIGNATED_INIT;
665}
666
667void PCHStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
668 VisitExpr(E);
669 Code = pch::EXPR_IMPLICIT_VALUE_INIT;
670}
671
672void PCHStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
673 VisitExpr(E);
674 Writer.WriteSubStmt(E->getSubExpr());
675 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
676 Writer.AddSourceLocation(E->getRParenLoc(), Record);
677 Code = pch::EXPR_VA_ARG;
678}
679
680void PCHStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
681 VisitExpr(E);
682 Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
683 Writer.AddSourceLocation(E->getLabelLoc(), Record);
684 Record.push_back(Writer.GetLabelID(E->getLabel()));
685 Code = pch::EXPR_ADDR_LABEL;
686}
687
688void PCHStmtWriter::VisitStmtExpr(StmtExpr *E) {
689 VisitExpr(E);
690 Writer.WriteSubStmt(E->getSubStmt());
691 Writer.AddSourceLocation(E->getLParenLoc(), Record);
692 Writer.AddSourceLocation(E->getRParenLoc(), Record);
693 Code = pch::EXPR_STMT;
694}
695
696void PCHStmtWriter::VisitTypesCompatibleExpr(TypesCompatibleExpr *E) {
697 VisitExpr(E);
698 Writer.AddTypeRef(E->getArgType1(), Record);
699 Writer.AddTypeRef(E->getArgType2(), Record);
700 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
701 Writer.AddSourceLocation(E->getRParenLoc(), Record);
702 Code = pch::EXPR_TYPES_COMPATIBLE;
703}
704
705void PCHStmtWriter::VisitChooseExpr(ChooseExpr *E) {
706 VisitExpr(E);
707 Writer.WriteSubStmt(E->getCond());
708 Writer.WriteSubStmt(E->getLHS());
709 Writer.WriteSubStmt(E->getRHS());
710 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
711 Writer.AddSourceLocation(E->getRParenLoc(), Record);
712 Code = pch::EXPR_CHOOSE;
713}
714
715void PCHStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
716 VisitExpr(E);
717 Writer.AddSourceLocation(E->getTokenLocation(), Record);
718 Code = pch::EXPR_GNU_NULL;
719}
720
721void PCHStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
722 VisitExpr(E);
723 Record.push_back(E->getNumSubExprs());
724 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
725 Writer.WriteSubStmt(E->getExpr(I));
726 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
727 Writer.AddSourceLocation(E->getRParenLoc(), Record);
728 Code = pch::EXPR_SHUFFLE_VECTOR;
729}
730
731void PCHStmtWriter::VisitBlockExpr(BlockExpr *E) {
732 VisitExpr(E);
733 Writer.AddDeclRef(E->getBlockDecl(), Record);
734 Record.push_back(E->hasBlockDeclRefExprs());
735 Code = pch::EXPR_BLOCK;
736}
737
738void PCHStmtWriter::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
739 VisitExpr(E);
740 Writer.AddDeclRef(E->getDecl(), Record);
741 Writer.AddSourceLocation(E->getLocation(), Record);
742 Record.push_back(E->isByRef());
Fariborz Jahanian9b0b57c2009-06-20 00:02:26 +0000743 Record.push_back(E->isConstQualAdded());
Fariborz Jahanian89f9d3a2010-06-04 19:06:53 +0000744 Writer.WriteSubStmt(E->getCopyConstructorExpr());
Chris Lattner63377d52009-04-27 06:20:01 +0000745 Code = pch::EXPR_BLOCK_DECL_REF;
746}
747
748//===----------------------------------------------------------------------===//
749// Objective-C Expressions and Statements.
750//===----------------------------------------------------------------------===//
751
752void PCHStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
753 VisitExpr(E);
754 Writer.WriteSubStmt(E->getString());
755 Writer.AddSourceLocation(E->getAtLoc(), Record);
756 Code = pch::EXPR_OBJC_STRING_LITERAL;
757}
758
Mike Stump1eb44332009-09-09 15:08:12 +0000759void PCHStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000760 VisitExpr(E);
Douglas Gregor81d34662010-04-20 15:39:42 +0000761 Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000762 Writer.AddSourceLocation(E->getAtLoc(), Record);
763 Writer.AddSourceLocation(E->getRParenLoc(), Record);
764 Code = pch::EXPR_OBJC_ENCODE;
765}
766
767void PCHStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
768 VisitExpr(E);
769 Writer.AddSelectorRef(E->getSelector(), Record);
770 Writer.AddSourceLocation(E->getAtLoc(), Record);
771 Writer.AddSourceLocation(E->getRParenLoc(), Record);
772 Code = pch::EXPR_OBJC_SELECTOR_EXPR;
773}
774
775void PCHStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
776 VisitExpr(E);
777 Writer.AddDeclRef(E->getProtocol(), Record);
778 Writer.AddSourceLocation(E->getAtLoc(), Record);
779 Writer.AddSourceLocation(E->getRParenLoc(), Record);
780 Code = pch::EXPR_OBJC_PROTOCOL_EXPR;
781}
782
783void PCHStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
784 VisitExpr(E);
785 Writer.AddDeclRef(E->getDecl(), Record);
786 Writer.AddSourceLocation(E->getLocation(), Record);
787 Writer.WriteSubStmt(E->getBase());
788 Record.push_back(E->isArrow());
789 Record.push_back(E->isFreeIvar());
790 Code = pch::EXPR_OBJC_IVAR_REF_EXPR;
791}
792
793void PCHStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
794 VisitExpr(E);
795 Writer.AddDeclRef(E->getProperty(), Record);
796 Writer.AddSourceLocation(E->getLocation(), Record);
797 Writer.WriteSubStmt(E->getBase());
798 Code = pch::EXPR_OBJC_PROPERTY_REF_EXPR;
799}
800
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000801void PCHStmtWriter::VisitObjCImplicitSetterGetterRefExpr(
802 ObjCImplicitSetterGetterRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000803 VisitExpr(E);
804 Writer.AddDeclRef(E->getGetterMethod(), Record);
805 Writer.AddDeclRef(E->getSetterMethod(), Record);
Mike Stump1eb44332009-09-09 15:08:12 +0000806
Fariborz Jahaniand2ae5aa2009-08-18 21:37:33 +0000807 // NOTE: InterfaceDecl and Base are mutually exclusive.
808 Writer.AddDeclRef(E->getInterfaceDecl(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000809 Writer.WriteSubStmt(E->getBase());
810 Writer.AddSourceLocation(E->getLocation(), Record);
811 Writer.AddSourceLocation(E->getClassLoc(), Record);
812 Code = pch::EXPR_OBJC_KVC_REF_EXPR;
813}
814
815void PCHStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
816 VisitExpr(E);
817 Record.push_back(E->getNumArgs());
Douglas Gregor04badcf2010-04-21 00:45:42 +0000818 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
819 switch (E->getReceiverKind()) {
820 case ObjCMessageExpr::Instance:
821 Writer.WriteSubStmt(E->getInstanceReceiver());
822 break;
823
824 case ObjCMessageExpr::Class:
825 Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
826 break;
827
828 case ObjCMessageExpr::SuperClass:
829 case ObjCMessageExpr::SuperInstance:
830 Writer.AddTypeRef(E->getSuperType(), Record);
831 Writer.AddSourceLocation(E->getSuperLoc(), Record);
832 break;
833 }
834
835 if (E->getMethodDecl()) {
836 Record.push_back(1);
837 Writer.AddDeclRef(E->getMethodDecl(), Record);
838 } else {
839 Record.push_back(0);
840 Writer.AddSelectorRef(E->getSelector(), Record);
841 }
842
Chris Lattner63377d52009-04-27 06:20:01 +0000843 Writer.AddSourceLocation(E->getLeftLoc(), Record);
844 Writer.AddSourceLocation(E->getRightLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000845
846 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
847 Arg != ArgEnd; ++Arg)
848 Writer.WriteSubStmt(*Arg);
849 Code = pch::EXPR_OBJC_MESSAGE_EXPR;
850}
851
852void PCHStmtWriter::VisitObjCSuperExpr(ObjCSuperExpr *E) {
853 VisitExpr(E);
854 Writer.AddSourceLocation(E->getLoc(), Record);
855 Code = pch::EXPR_OBJC_SUPER_EXPR;
856}
857
858void PCHStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
859 VisitStmt(S);
860 Writer.WriteSubStmt(S->getElement());
861 Writer.WriteSubStmt(S->getCollection());
862 Writer.WriteSubStmt(S->getBody());
863 Writer.AddSourceLocation(S->getForLoc(), Record);
864 Writer.AddSourceLocation(S->getRParenLoc(), Record);
865 Code = pch::STMT_OBJC_FOR_COLLECTION;
866}
867
868void PCHStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
869 Writer.WriteSubStmt(S->getCatchBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000870 Writer.AddDeclRef(S->getCatchParamDecl(), Record);
871 Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
872 Writer.AddSourceLocation(S->getRParenLoc(), Record);
873 Code = pch::STMT_OBJC_CATCH;
874}
875
876void PCHStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
877 Writer.WriteSubStmt(S->getFinallyBody());
878 Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
879 Code = pch::STMT_OBJC_FINALLY;
880}
881
882void PCHStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000883 Record.push_back(S->getNumCatchStmts());
884 Record.push_back(S->getFinallyStmt() != 0);
Chris Lattner63377d52009-04-27 06:20:01 +0000885 Writer.WriteSubStmt(S->getTryBody());
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000886 for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
887 Writer.WriteSubStmt(S->getCatchStmt(I));
888 if (S->getFinallyStmt())
889 Writer.WriteSubStmt(S->getFinallyStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000890 Writer.AddSourceLocation(S->getAtTryLoc(), Record);
891 Code = pch::STMT_OBJC_AT_TRY;
892}
893
894void PCHStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
895 Writer.WriteSubStmt(S->getSynchExpr());
896 Writer.WriteSubStmt(S->getSynchBody());
897 Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
898 Code = pch::STMT_OBJC_AT_SYNCHRONIZED;
899}
900
901void PCHStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
902 Writer.WriteSubStmt(S->getThrowExpr());
903 Writer.AddSourceLocation(S->getThrowLoc(), Record);
904 Code = pch::STMT_OBJC_AT_THROW;
905}
906
907//===----------------------------------------------------------------------===//
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000908// C++ Expressions and Statements.
909//===----------------------------------------------------------------------===//
910
911void PCHStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
912 VisitCallExpr(E);
913 Record.push_back(E->getOperator());
914 Code = pch::EXPR_CXX_OPERATOR_CALL;
915}
916
Chris Lattner1817bd42010-05-09 05:36:05 +0000917void PCHStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
918 VisitCallExpr(E);
919 Code = pch::EXPR_CXX_MEMBER_CALL;
920}
921
Douglas Gregor39da0b82009-09-09 23:08:42 +0000922void PCHStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
923 VisitExpr(E);
Argyrios Kyrtzidis5e1b7c22010-06-24 08:57:09 +0000924 Record.push_back(E->getNumArgs());
925 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
926 Writer.WriteSubStmt(E->getArg(I));
Douglas Gregor39da0b82009-09-09 23:08:42 +0000927 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor99a2e602009-12-16 01:38:02 +0000928 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000929 Record.push_back(E->isElidable());
Douglas Gregor16006c92009-12-16 18:50:27 +0000930 Record.push_back(E->requiresZeroInitialization());
Douglas Gregor3c9034c2010-05-15 00:13:29 +0000931 Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
Douglas Gregor39da0b82009-09-09 23:08:42 +0000932 Code = pch::EXPR_CXX_CONSTRUCT;
933}
934
Sam Weinigce757a72010-01-16 21:21:01 +0000935void PCHStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
936 VisitExplicitCastExpr(E);
937 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
938}
939
940void PCHStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
941 VisitCXXNamedCastExpr(E);
942 Code = pch::EXPR_CXX_STATIC_CAST;
943}
944
945void PCHStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
946 VisitCXXNamedCastExpr(E);
947 Code = pch::EXPR_CXX_DYNAMIC_CAST;
948}
949
950void PCHStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
951 VisitCXXNamedCastExpr(E);
952 Code = pch::EXPR_CXX_REINTERPRET_CAST;
953}
954
955void PCHStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
956 VisitCXXNamedCastExpr(E);
957 Code = pch::EXPR_CXX_CONST_CAST;
958}
959
960void PCHStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
961 VisitExplicitCastExpr(E);
962 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
963 Writer.AddSourceLocation(E->getRParenLoc(), Record);
964 Code = pch::EXPR_CXX_FUNCTIONAL_CAST;
965}
966
Sam Weinigeb7f9612010-02-07 06:32:43 +0000967void PCHStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
968 VisitExpr(E);
969 Record.push_back(E->getValue());
970 Writer.AddSourceLocation(E->getLocation(), Record);
971 Code = pch::EXPR_CXX_BOOL_LITERAL;
972}
973
974void PCHStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
975 VisitExpr(E);
976 Writer.AddSourceLocation(E->getLocation(), Record);
977 Code = pch::EXPR_CXX_NULL_PTR_LITERAL;
978}
979
Chris Lattner14ab24f2010-05-09 06:03:39 +0000980void PCHStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
981 VisitExpr(E);
982 Writer.AddSourceRange(E->getSourceRange(), Record);
983 if (E->isTypeOperand()) {
984 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
985 Code = pch::EXPR_CXX_TYPEID_TYPE;
986 } else {
987 Writer.WriteSubStmt(E->getExprOperand());
988 Code = pch::EXPR_CXX_TYPEID_EXPR;
989 }
990}
991
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000992void PCHStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
993 VisitExpr(E);
994 Writer.AddSourceLocation(E->getLocation(), Record);
995 Record.push_back(E->isImplicit());
996 Code = pch::EXPR_CXX_THIS;
997}
998
999void PCHStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
Chris Lattner030854b2010-05-09 06:40:08 +00001000 VisitExpr(E);
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001001 Writer.AddSourceLocation(E->getThrowLoc(), Record);
1002 Writer.WriteSubStmt(E->getSubExpr());
1003 Code = pch::EXPR_CXX_THROW;
1004}
1005
Chris Lattner030854b2010-05-09 06:40:08 +00001006void PCHStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
1007 VisitExpr(E);
1008 Writer.AddSourceLocation(E->getUsedLocation(), Record);
1009 if (E->isExprStored()) {
1010 Record.push_back(1);
1011 Writer.WriteSubStmt(E->getExpr());
1012 } else {
1013 Record.push_back(0);
1014 }
1015
1016 Code = pch::EXPR_CXX_DEFAULT_ARG;
1017}
1018
Chris Lattnerd2598362010-05-10 00:25:06 +00001019void PCHStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
1020 VisitExpr(E);
1021 Writer.AddCXXTemporary(E->getTemporary(), Record);
1022 Writer.WriteSubStmt(E->getSubExpr());
1023 Code = pch::EXPR_CXX_BIND_TEMPORARY;
1024}
1025
Chris Lattner59218632010-05-10 01:22:27 +00001026void PCHStmtWriter::VisitCXXZeroInitValueExpr(CXXZeroInitValueExpr *E) {
1027 VisitExpr(E);
1028 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1029 Writer.AddSourceLocation(E->getRParenLoc(), Record);
1030 Code = pch::EXPR_CXX_ZERO_INIT_VALUE;
1031}
1032
1033void PCHStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
1034 VisitExpr(E);
1035 Record.push_back(E->isGlobalNew());
1036 Record.push_back(E->isParenTypeId());
1037 Record.push_back(E->hasInitializer());
1038 Record.push_back(E->isArray());
1039 Record.push_back(E->getNumPlacementArgs());
1040 Record.push_back(E->getNumConstructorArgs());
1041 Writer.AddDeclRef(E->getOperatorNew(), Record);
1042 Writer.AddDeclRef(E->getOperatorDelete(), Record);
1043 Writer.AddDeclRef(E->getConstructor(), Record);
1044 Writer.AddSourceLocation(E->getStartLoc(), Record);
1045 Writer.AddSourceLocation(E->getEndLoc(), Record);
1046 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
1047 I != e; ++I)
1048 Writer.WriteSubStmt(*I);
1049
1050 Code = pch::EXPR_CXX_NEW;
1051}
1052
Argyrios Kyrtzidis95fc98c2010-06-22 17:07:59 +00001053void PCHStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
1054 VisitExpr(E);
1055 Record.push_back(E->isGlobalDelete());
1056 Record.push_back(E->isArrayForm());
1057 Writer.AddDeclRef(E->getOperatorDelete(), Record);
1058 Writer.WriteSubStmt(E->getArgument());
1059 Writer.AddSourceLocation(E->getSourceRange().getBegin(), Record);
1060
1061 Code = pch::EXPR_CXX_DELETE;
1062}
Chris Lattner59218632010-05-10 01:22:27 +00001063
Chris Lattnerd2598362010-05-10 00:25:06 +00001064void PCHStmtWriter::VisitCXXExprWithTemporaries(CXXExprWithTemporaries *E) {
1065 VisitExpr(E);
1066 Record.push_back(E->getNumTemporaries());
1067 for (unsigned i = 0, e = E->getNumTemporaries(); i != e; ++i)
1068 Writer.AddCXXTemporary(E->getTemporary(i), Record);
1069
1070 Writer.WriteSubStmt(E->getSubExpr());
1071 Code = pch::EXPR_CXX_EXPR_WITH_TEMPORARIES;
1072}
1073
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001074void
1075PCHStmtWriter::VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E){
1076 VisitExpr(E);
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001077
1078 // Don't emit anything here, NumTemplateArgs must be emitted first.
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001079
1080 if (E->hasExplicitTemplateArgs()) {
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001081 const ExplicitTemplateArgumentList &Args
1082 = *E->getExplicitTemplateArgumentList();
1083 assert(Args.NumTemplateArgs &&
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001084 "Num of template args was zero! PCH reading will mess up!");
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001085 Record.push_back(Args.NumTemplateArgs);
1086 AddExplicitTemplateArgumentList(Args);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001087 } else {
1088 Record.push_back(0);
1089 }
1090
1091 if (!E->isImplicitAccess())
1092 Writer.WriteSubStmt(E->getBase());
1093 else
1094 Writer.WriteSubStmt(0);
1095 Writer.AddTypeRef(E->getBaseType(), Record);
1096 Record.push_back(E->isArrow());
1097 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1098 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1099 Writer.AddSourceRange(E->getQualifierRange(), Record);
1100 Writer.AddDeclRef(E->getFirstQualifierFoundInScope(), Record);
1101 Writer.AddDeclarationName(E->getMember(), Record);
1102 Writer.AddSourceLocation(E->getMemberLoc(), Record);
1103 Code = pch::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
1104}
1105
1106void
1107PCHStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
1108 VisitExpr(E);
1109 Record.push_back(E->arg_size());
1110 for (CXXUnresolvedConstructExpr::arg_iterator
1111 ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
1112 Writer.WriteSubStmt(*ArgI);
1113 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1114 Writer.AddTypeRef(E->getTypeAsWritten(), Record);
1115 Writer.AddSourceLocation(E->getLParenLoc(), Record);
1116 Writer.AddSourceLocation(E->getRParenLoc(), Record);
1117 Code = pch::EXPR_CXX_UNRESOLVED_CONSTRUCT;
1118}
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001119
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001120void PCHStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
1121 VisitExpr(E);
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001122
1123 // Don't emit anything here, NumTemplateArgs must be emitted first.
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001124
1125 if (E->hasExplicitTemplateArgs()) {
1126 const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
1127 assert(Args.NumTemplateArgs &&
1128 "Num of template args was zero! PCH reading will mess up!");
1129 Record.push_back(Args.NumTemplateArgs);
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001130 AddExplicitTemplateArgumentList(Args);
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001131 } else {
1132 Record.push_back(0);
1133 }
1134
1135 Record.push_back(E->getNumDecls());
1136 for (OverloadExpr::decls_iterator
1137 OvI = E->decls_begin(), OvE = E->decls_end(); OvI != OvE; ++OvI) {
1138 Writer.AddDeclRef(OvI.getDecl(), Record);
1139 Record.push_back(OvI.getAccess());
1140 }
1141
1142 Writer.AddDeclarationName(E->getName(), Record);
1143 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1144 Writer.AddSourceRange(E->getQualifierRange(), Record);
1145 Writer.AddSourceLocation(E->getNameLoc(), Record);
1146}
1147
1148void PCHStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
1149 VisitOverloadExpr(E);
1150 Record.push_back(E->isArrow());
1151 Record.push_back(E->hasUnresolvedUsing());
1152 Writer.WriteSubStmt(!E->isImplicitAccess() ? E->getBase() : 0);
1153 Writer.AddTypeRef(E->getBaseType(), Record);
1154 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1155 Code = pch::EXPR_CXX_UNRESOLVED_MEMBER;
1156}
1157
Argyrios Kyrtzidisbd65bb52010-06-25 09:03:34 +00001158void PCHStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
1159 VisitOverloadExpr(E);
1160 Record.push_back(E->requiresADL());
1161 Record.push_back(E->isOverloaded());
1162 Writer.AddDeclRef(E->getNamingClass(), Record);
1163 Code = pch::EXPR_CXX_UNRESOLVED_LOOKUP;
1164}
1165
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001166//===----------------------------------------------------------------------===//
Chris Lattner63377d52009-04-27 06:20:01 +00001167// PCHWriter Implementation
1168//===----------------------------------------------------------------------===//
1169
1170unsigned PCHWriter::RecordSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001171 assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001172 "SwitchCase recorded twice");
1173 unsigned NextID = SwitchCaseIDs.size();
1174 SwitchCaseIDs[S] = NextID;
1175 return NextID;
1176}
1177
1178unsigned PCHWriter::getSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001179 assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001180 "SwitchCase hasn't been seen yet");
1181 return SwitchCaseIDs[S];
1182}
1183
1184/// \brief Retrieve the ID for the given label statement, which may
1185/// or may not have been emitted yet.
1186unsigned PCHWriter::GetLabelID(LabelStmt *S) {
1187 std::map<LabelStmt *, unsigned>::iterator Pos = LabelIDs.find(S);
1188 if (Pos != LabelIDs.end())
1189 return Pos->second;
Mike Stump1eb44332009-09-09 15:08:12 +00001190
Chris Lattner63377d52009-04-27 06:20:01 +00001191 unsigned NextID = LabelIDs.size();
1192 LabelIDs[S] = NextID;
1193 return NextID;
1194}
1195
1196/// \brief Write the given substatement or subexpression to the
1197/// bitstream.
1198void PCHWriter::WriteSubStmt(Stmt *S) {
1199 RecordData Record;
1200 PCHStmtWriter Writer(*this, Record);
1201 ++NumStatements;
Mike Stump1eb44332009-09-09 15:08:12 +00001202
Chris Lattner63377d52009-04-27 06:20:01 +00001203 if (!S) {
1204 Stream.EmitRecord(pch::STMT_NULL_PTR, Record);
1205 return;
1206 }
Mike Stump1eb44332009-09-09 15:08:12 +00001207
Chris Lattner63377d52009-04-27 06:20:01 +00001208 Writer.Code = pch::STMT_NULL_PTR;
1209 Writer.Visit(S);
Chris Lattner59218632010-05-10 01:22:27 +00001210
1211#ifndef NDEBUG
1212 if (Writer.Code == pch::STMT_NULL_PTR) {
Argyrios Kyrtzidis12bf28f2010-06-25 09:03:12 +00001213 SourceManager &SrcMgr
1214 = DeclIDs.begin()->first->getASTContext().getSourceManager();
1215 S->dump(SrcMgr);
Chris Lattner59218632010-05-10 01:22:27 +00001216 assert(0 && "Unhandled sub statement writing PCH file");
1217 }
1218#endif
Mike Stump1eb44332009-09-09 15:08:12 +00001219 Stream.EmitRecord(Writer.Code, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001220}
1221
1222/// \brief Flush all of the statements that have been added to the
1223/// queue via AddStmt().
1224void PCHWriter::FlushStmts() {
1225 RecordData Record;
1226 PCHStmtWriter Writer(*this, Record);
Argyrios Kyrtzidis17cfded2010-06-28 09:31:42 +00001227
1228 EmittingStmts = true;
Mike Stump1eb44332009-09-09 15:08:12 +00001229
Chris Lattner63377d52009-04-27 06:20:01 +00001230 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
1231 ++NumStatements;
1232 Stmt *S = StmtsToEmit[I];
Mike Stump1eb44332009-09-09 15:08:12 +00001233
Chris Lattner63377d52009-04-27 06:20:01 +00001234 if (!S) {
1235 Stream.EmitRecord(pch::STMT_NULL_PTR, Record);
Argyrios Kyrtzidisc4117aa2010-06-25 16:25:02 +00001236 Stream.EmitRecord(pch::STMT_STOP, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001237 continue;
1238 }
Mike Stump1eb44332009-09-09 15:08:12 +00001239
Chris Lattner63377d52009-04-27 06:20:01 +00001240 Writer.Code = pch::STMT_NULL_PTR;
1241 Writer.Visit(S);
Chris Lattnerd2598362010-05-10 00:25:06 +00001242#ifndef NDEBUG
1243 if (Writer.Code == pch::STMT_NULL_PTR) {
1244 S->dump();
1245 assert(0 && "Unhandled expression writing PCH file");
1246 }
1247#endif
Mike Stump1eb44332009-09-09 15:08:12 +00001248 Stream.EmitRecord(Writer.Code, Record);
1249
1250 assert(N == StmtsToEmit.size() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001251 "Substatement writen via AddStmt rather than WriteSubStmt!");
Mike Stump1eb44332009-09-09 15:08:12 +00001252
Chris Lattner63377d52009-04-27 06:20:01 +00001253 // Note that we are at the end of a full expression. Any
1254 // expression records that follow this one are part of a different
1255 // expression.
1256 Record.clear();
1257 Stream.EmitRecord(pch::STMT_STOP, Record);
1258 }
Mike Stump1eb44332009-09-09 15:08:12 +00001259
Argyrios Kyrtzidis17cfded2010-06-28 09:31:42 +00001260 EmittingStmts = false;
1261
Chris Lattner63377d52009-04-27 06:20:01 +00001262 StmtsToEmit.clear();
Chris Lattner63377d52009-04-27 06:20:01 +00001263}