blob: a9ee43527cec3ba5a12a7fffd306a4d28ffb9526 [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) { }
35
36 void VisitStmt(Stmt *S);
37 void VisitNullStmt(NullStmt *S);
38 void VisitCompoundStmt(CompoundStmt *S);
39 void VisitSwitchCase(SwitchCase *S);
40 void VisitCaseStmt(CaseStmt *S);
41 void VisitDefaultStmt(DefaultStmt *S);
42 void VisitLabelStmt(LabelStmt *S);
43 void VisitIfStmt(IfStmt *S);
44 void VisitSwitchStmt(SwitchStmt *S);
45 void VisitWhileStmt(WhileStmt *S);
46 void VisitDoStmt(DoStmt *S);
47 void VisitForStmt(ForStmt *S);
48 void VisitGotoStmt(GotoStmt *S);
49 void VisitIndirectGotoStmt(IndirectGotoStmt *S);
50 void VisitContinueStmt(ContinueStmt *S);
51 void VisitBreakStmt(BreakStmt *S);
52 void VisitReturnStmt(ReturnStmt *S);
53 void VisitDeclStmt(DeclStmt *S);
54 void VisitAsmStmt(AsmStmt *S);
55 void VisitExpr(Expr *E);
56 void VisitPredefinedExpr(PredefinedExpr *E);
57 void VisitDeclRefExpr(DeclRefExpr *E);
58 void VisitIntegerLiteral(IntegerLiteral *E);
59 void VisitFloatingLiteral(FloatingLiteral *E);
60 void VisitImaginaryLiteral(ImaginaryLiteral *E);
61 void VisitStringLiteral(StringLiteral *E);
62 void VisitCharacterLiteral(CharacterLiteral *E);
63 void VisitParenExpr(ParenExpr *E);
64 void VisitUnaryOperator(UnaryOperator *E);
Douglas Gregor8ecdb652010-04-28 22:16:22 +000065 void VisitOffsetOfExpr(OffsetOfExpr *E);
Chris Lattner63377d52009-04-27 06:20:01 +000066 void VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E);
67 void VisitArraySubscriptExpr(ArraySubscriptExpr *E);
68 void VisitCallExpr(CallExpr *E);
69 void VisitMemberExpr(MemberExpr *E);
70 void VisitCastExpr(CastExpr *E);
71 void VisitBinaryOperator(BinaryOperator *E);
72 void VisitCompoundAssignOperator(CompoundAssignOperator *E);
73 void VisitConditionalOperator(ConditionalOperator *E);
74 void VisitImplicitCastExpr(ImplicitCastExpr *E);
75 void VisitExplicitCastExpr(ExplicitCastExpr *E);
76 void VisitCStyleCastExpr(CStyleCastExpr *E);
77 void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
78 void VisitExtVectorElementExpr(ExtVectorElementExpr *E);
79 void VisitInitListExpr(InitListExpr *E);
80 void VisitDesignatedInitExpr(DesignatedInitExpr *E);
81 void VisitImplicitValueInitExpr(ImplicitValueInitExpr *E);
82 void VisitVAArgExpr(VAArgExpr *E);
83 void VisitAddrLabelExpr(AddrLabelExpr *E);
84 void VisitStmtExpr(StmtExpr *E);
85 void VisitTypesCompatibleExpr(TypesCompatibleExpr *E);
86 void VisitChooseExpr(ChooseExpr *E);
87 void VisitGNUNullExpr(GNUNullExpr *E);
88 void VisitShuffleVectorExpr(ShuffleVectorExpr *E);
89 void VisitBlockExpr(BlockExpr *E);
90 void VisitBlockDeclRefExpr(BlockDeclRefExpr *E);
Mike Stump1eb44332009-09-09 15:08:12 +000091
Chris Lattner63377d52009-04-27 06:20:01 +000092 // Objective-C Expressions
93 void VisitObjCStringLiteral(ObjCStringLiteral *E);
94 void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
95 void VisitObjCSelectorExpr(ObjCSelectorExpr *E);
96 void VisitObjCProtocolExpr(ObjCProtocolExpr *E);
97 void VisitObjCIvarRefExpr(ObjCIvarRefExpr *E);
98 void VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E);
Fariborz Jahanian09105f52009-08-20 17:02:02 +000099 void VisitObjCImplicitSetterGetterRefExpr(
100 ObjCImplicitSetterGetterRefExpr *E);
Chris Lattner63377d52009-04-27 06:20:01 +0000101 void VisitObjCMessageExpr(ObjCMessageExpr *E);
102 void VisitObjCSuperExpr(ObjCSuperExpr *E);
Steve Narofff242b1b2009-07-24 17:54:45 +0000103 void VisitObjCIsaExpr(ObjCIsaExpr *E);
Mike Stump1eb44332009-09-09 15:08:12 +0000104
105 // Objective-C Statements
Chris Lattner63377d52009-04-27 06:20:01 +0000106 void VisitObjCForCollectionStmt(ObjCForCollectionStmt *);
107 void VisitObjCAtCatchStmt(ObjCAtCatchStmt *);
108 void VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *);
109 void VisitObjCAtTryStmt(ObjCAtTryStmt *);
110 void VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *);
111 void VisitObjCAtThrowStmt(ObjCAtThrowStmt *);
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000112
Mike Stump1eb44332009-09-09 15:08:12 +0000113 // C++ Statements
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000114 void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
Chris Lattner1817bd42010-05-09 05:36:05 +0000115 void VisitCXXMemberCallExpr(CXXMemberCallExpr *E);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000116 void VisitCXXConstructExpr(CXXConstructExpr *E);
Sam Weinigce757a72010-01-16 21:21:01 +0000117 void VisitCXXNamedCastExpr(CXXNamedCastExpr *E);
118 void VisitCXXStaticCastExpr(CXXStaticCastExpr *E);
119 void VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E);
120 void VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E);
121 void VisitCXXConstCastExpr(CXXConstCastExpr *E);
122 void VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E);
Sam Weinigeb7f9612010-02-07 06:32:43 +0000123 void VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E);
124 void VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E);
Chris Lattner14ab24f2010-05-09 06:03:39 +0000125 void VisitCXXTypeidExpr(CXXTypeidExpr *E);
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000126 void VisitCXXThisExpr(CXXThisExpr *E);
127 void VisitCXXThrowExpr(CXXThrowExpr *E);
Chris Lattner030854b2010-05-09 06:40:08 +0000128 void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E);
Chris Lattnerd2598362010-05-10 00:25:06 +0000129 void VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E);
130
Chris Lattner59218632010-05-10 01:22:27 +0000131 void VisitCXXZeroInitValueExpr(CXXZeroInitValueExpr *E);
132 void VisitCXXNewExpr(CXXNewExpr *E);
133
Chris Lattnerd2598362010-05-10 00:25:06 +0000134 void VisitCXXExprWithTemporaries(CXXExprWithTemporaries *E);
Chris Lattner63377d52009-04-27 06:20:01 +0000135 };
136}
137
Mike Stump1eb44332009-09-09 15:08:12 +0000138void PCHStmtWriter::VisitStmt(Stmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000139}
140
141void PCHStmtWriter::VisitNullStmt(NullStmt *S) {
142 VisitStmt(S);
143 Writer.AddSourceLocation(S->getSemiLoc(), Record);
144 Code = pch::STMT_NULL;
145}
146
147void PCHStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
148 VisitStmt(S);
149 Record.push_back(S->size());
150 for (CompoundStmt::body_iterator CS = S->body_begin(), CSEnd = S->body_end();
151 CS != CSEnd; ++CS)
152 Writer.WriteSubStmt(*CS);
153 Writer.AddSourceLocation(S->getLBracLoc(), Record);
154 Writer.AddSourceLocation(S->getRBracLoc(), Record);
155 Code = pch::STMT_COMPOUND;
156}
157
158void PCHStmtWriter::VisitSwitchCase(SwitchCase *S) {
159 VisitStmt(S);
160 Record.push_back(Writer.RecordSwitchCaseID(S));
161}
162
163void PCHStmtWriter::VisitCaseStmt(CaseStmt *S) {
164 VisitSwitchCase(S);
165 Writer.WriteSubStmt(S->getLHS());
166 Writer.WriteSubStmt(S->getRHS());
167 Writer.WriteSubStmt(S->getSubStmt());
168 Writer.AddSourceLocation(S->getCaseLoc(), Record);
Douglas Gregordbb26db2009-05-15 23:57:33 +0000169 Writer.AddSourceLocation(S->getEllipsisLoc(), Record);
170 Writer.AddSourceLocation(S->getColonLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000171 Code = pch::STMT_CASE;
172}
173
174void PCHStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
175 VisitSwitchCase(S);
176 Writer.WriteSubStmt(S->getSubStmt());
177 Writer.AddSourceLocation(S->getDefaultLoc(), Record);
Douglas Gregordbb26db2009-05-15 23:57:33 +0000178 Writer.AddSourceLocation(S->getColonLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000179 Code = pch::STMT_DEFAULT;
180}
181
182void PCHStmtWriter::VisitLabelStmt(LabelStmt *S) {
183 VisitStmt(S);
184 Writer.AddIdentifierRef(S->getID(), Record);
185 Writer.WriteSubStmt(S->getSubStmt());
186 Writer.AddSourceLocation(S->getIdentLoc(), Record);
187 Record.push_back(Writer.GetLabelID(S));
188 Code = pch::STMT_LABEL;
189}
190
191void PCHStmtWriter::VisitIfStmt(IfStmt *S) {
192 VisitStmt(S);
Douglas Gregor8cfe5a72009-11-23 23:44:04 +0000193 Writer.AddDeclRef(S->getConditionVariable(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000194 Writer.WriteSubStmt(S->getCond());
195 Writer.WriteSubStmt(S->getThen());
196 Writer.WriteSubStmt(S->getElse());
197 Writer.AddSourceLocation(S->getIfLoc(), Record);
Douglas Gregord06f6ca2009-05-15 18:53:42 +0000198 Writer.AddSourceLocation(S->getElseLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000199 Code = pch::STMT_IF;
200}
201
202void PCHStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
203 VisitStmt(S);
Douglas Gregord3d53012009-11-24 17:07:59 +0000204 Writer.AddDeclRef(S->getConditionVariable(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000205 Writer.WriteSubStmt(S->getCond());
206 Writer.WriteSubStmt(S->getBody());
207 Writer.AddSourceLocation(S->getSwitchLoc(), Record);
Mike Stump1eb44332009-09-09 15:08:12 +0000208 for (SwitchCase *SC = S->getSwitchCaseList(); SC;
Chris Lattner63377d52009-04-27 06:20:01 +0000209 SC = SC->getNextSwitchCase())
210 Record.push_back(Writer.getSwitchCaseID(SC));
211 Code = pch::STMT_SWITCH;
212}
213
214void PCHStmtWriter::VisitWhileStmt(WhileStmt *S) {
215 VisitStmt(S);
Douglas Gregor5656e142009-11-24 21:15:44 +0000216 Writer.AddDeclRef(S->getConditionVariable(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000217 Writer.WriteSubStmt(S->getCond());
218 Writer.WriteSubStmt(S->getBody());
219 Writer.AddSourceLocation(S->getWhileLoc(), Record);
220 Code = pch::STMT_WHILE;
221}
222
223void PCHStmtWriter::VisitDoStmt(DoStmt *S) {
224 VisitStmt(S);
225 Writer.WriteSubStmt(S->getCond());
226 Writer.WriteSubStmt(S->getBody());
227 Writer.AddSourceLocation(S->getDoLoc(), Record);
Douglas Gregor9f3ca2a2009-05-15 21:56:04 +0000228 Writer.AddSourceLocation(S->getWhileLoc(), Record);
Chris Lattner98913592009-06-12 23:04:47 +0000229 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000230 Code = pch::STMT_DO;
231}
232
233void PCHStmtWriter::VisitForStmt(ForStmt *S) {
234 VisitStmt(S);
235 Writer.WriteSubStmt(S->getInit());
236 Writer.WriteSubStmt(S->getCond());
Douglas Gregor99e9b4d2009-11-25 00:27:52 +0000237 Writer.AddDeclRef(S->getConditionVariable(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000238 Writer.WriteSubStmt(S->getInc());
239 Writer.WriteSubStmt(S->getBody());
240 Writer.AddSourceLocation(S->getForLoc(), Record);
Douglas Gregor5831c6a2009-05-15 22:12:32 +0000241 Writer.AddSourceLocation(S->getLParenLoc(), Record);
242 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000243 Code = pch::STMT_FOR;
244}
245
246void PCHStmtWriter::VisitGotoStmt(GotoStmt *S) {
247 VisitStmt(S);
248 Record.push_back(Writer.GetLabelID(S->getLabel()));
249 Writer.AddSourceLocation(S->getGotoLoc(), Record);
250 Writer.AddSourceLocation(S->getLabelLoc(), Record);
251 Code = pch::STMT_GOTO;
252}
253
254void PCHStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
255 VisitStmt(S);
256 Writer.AddSourceLocation(S->getGotoLoc(), Record);
Douglas Gregor5f1b9e62009-05-16 00:20:29 +0000257 Writer.AddSourceLocation(S->getStarLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000258 Writer.WriteSubStmt(S->getTarget());
259 Code = pch::STMT_INDIRECT_GOTO;
260}
261
262void PCHStmtWriter::VisitContinueStmt(ContinueStmt *S) {
263 VisitStmt(S);
264 Writer.AddSourceLocation(S->getContinueLoc(), Record);
265 Code = pch::STMT_CONTINUE;
266}
267
268void PCHStmtWriter::VisitBreakStmt(BreakStmt *S) {
269 VisitStmt(S);
270 Writer.AddSourceLocation(S->getBreakLoc(), Record);
271 Code = pch::STMT_BREAK;
272}
273
274void PCHStmtWriter::VisitReturnStmt(ReturnStmt *S) {
275 VisitStmt(S);
276 Writer.WriteSubStmt(S->getRetValue());
277 Writer.AddSourceLocation(S->getReturnLoc(), Record);
Douglas Gregor5077c382010-05-15 06:01:05 +0000278 Writer.AddDeclRef(S->getNRVOCandidate(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000279 Code = pch::STMT_RETURN;
280}
281
282void PCHStmtWriter::VisitDeclStmt(DeclStmt *S) {
283 VisitStmt(S);
284 Writer.AddSourceLocation(S->getStartLoc(), Record);
285 Writer.AddSourceLocation(S->getEndLoc(), Record);
286 DeclGroupRef DG = S->getDeclGroup();
287 for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
288 Writer.AddDeclRef(*D, Record);
289 Code = pch::STMT_DECL;
290}
291
292void PCHStmtWriter::VisitAsmStmt(AsmStmt *S) {
293 VisitStmt(S);
294 Record.push_back(S->getNumOutputs());
295 Record.push_back(S->getNumInputs());
296 Record.push_back(S->getNumClobbers());
297 Writer.AddSourceLocation(S->getAsmLoc(), Record);
298 Writer.AddSourceLocation(S->getRParenLoc(), Record);
299 Record.push_back(S->isVolatile());
300 Record.push_back(S->isSimple());
Mike Stump3b11fd32010-01-04 22:37:17 +0000301 Record.push_back(S->isMSAsm());
Chris Lattner63377d52009-04-27 06:20:01 +0000302 Writer.WriteSubStmt(S->getAsmString());
303
304 // Outputs
Anders Carlssonff93dbd2010-01-30 22:25:16 +0000305 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
306 Writer.AddIdentifierRef(S->getOutputIdentifier(I), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000307 Writer.WriteSubStmt(S->getOutputConstraintLiteral(I));
308 Writer.WriteSubStmt(S->getOutputExpr(I));
309 }
310
311 // Inputs
312 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
Anders Carlssonff93dbd2010-01-30 22:25:16 +0000313 Writer.AddIdentifierRef(S->getInputIdentifier(I), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000314 Writer.WriteSubStmt(S->getInputConstraintLiteral(I));
315 Writer.WriteSubStmt(S->getInputExpr(I));
316 }
317
318 // Clobbers
319 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
320 Writer.WriteSubStmt(S->getClobber(I));
321
322 Code = pch::STMT_ASM;
323}
324
325void PCHStmtWriter::VisitExpr(Expr *E) {
326 VisitStmt(E);
327 Writer.AddTypeRef(E->getType(), Record);
328 Record.push_back(E->isTypeDependent());
329 Record.push_back(E->isValueDependent());
330}
331
332void PCHStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
333 VisitExpr(E);
334 Writer.AddSourceLocation(E->getLocation(), Record);
335 Record.push_back(E->getIdentType()); // FIXME: stable encoding
336 Code = pch::EXPR_PREDEFINED;
337}
338
339void PCHStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
340 VisitExpr(E);
341 Writer.AddDeclRef(E->getDecl(), Record);
342 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregora2813ce2009-10-23 18:54:35 +0000343 // FIXME: write qualifier
344 // FIXME: write explicit template arguments
Chris Lattner63377d52009-04-27 06:20:01 +0000345 Code = pch::EXPR_DECL_REF;
346}
347
348void PCHStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
349 VisitExpr(E);
350 Writer.AddSourceLocation(E->getLocation(), Record);
351 Writer.AddAPInt(E->getValue(), Record);
352 Code = pch::EXPR_INTEGER_LITERAL;
353}
354
355void PCHStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
356 VisitExpr(E);
357 Writer.AddAPFloat(E->getValue(), Record);
358 Record.push_back(E->isExact());
359 Writer.AddSourceLocation(E->getLocation(), Record);
360 Code = pch::EXPR_FLOATING_LITERAL;
361}
362
363void PCHStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
364 VisitExpr(E);
365 Writer.WriteSubStmt(E->getSubExpr());
366 Code = pch::EXPR_IMAGINARY_LITERAL;
367}
368
369void PCHStmtWriter::VisitStringLiteral(StringLiteral *E) {
370 VisitExpr(E);
371 Record.push_back(E->getByteLength());
372 Record.push_back(E->getNumConcatenated());
373 Record.push_back(E->isWide());
374 // FIXME: String data should be stored as a blob at the end of the
375 // StringLiteral. However, we can't do so now because we have no
376 // provision for coping with abbreviations when we're jumping around
377 // the PCH file during deserialization.
Mike Stump1eb44332009-09-09 15:08:12 +0000378 Record.insert(Record.end(),
Chris Lattner63377d52009-04-27 06:20:01 +0000379 E->getStrData(), E->getStrData() + E->getByteLength());
380 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
381 Writer.AddSourceLocation(E->getStrTokenLoc(I), Record);
382 Code = pch::EXPR_STRING_LITERAL;
383}
384
385void PCHStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
386 VisitExpr(E);
387 Record.push_back(E->getValue());
Chris Lattner018d8e02009-08-24 17:39:36 +0000388 Writer.AddSourceLocation(E->getLocation(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000389 Record.push_back(E->isWide());
390 Code = pch::EXPR_CHARACTER_LITERAL;
391}
392
393void PCHStmtWriter::VisitParenExpr(ParenExpr *E) {
394 VisitExpr(E);
395 Writer.AddSourceLocation(E->getLParen(), Record);
396 Writer.AddSourceLocation(E->getRParen(), Record);
397 Writer.WriteSubStmt(E->getSubExpr());
398 Code = pch::EXPR_PAREN;
399}
400
401void PCHStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
402 VisitExpr(E);
403 Writer.WriteSubStmt(E->getSubExpr());
404 Record.push_back(E->getOpcode()); // FIXME: stable encoding
405 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
406 Code = pch::EXPR_UNARY_OPERATOR;
407}
408
Douglas Gregor8ecdb652010-04-28 22:16:22 +0000409void PCHStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
410 VisitExpr(E);
411 Record.push_back(E->getNumComponents());
412 Record.push_back(E->getNumExpressions());
413 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
414 Writer.AddSourceLocation(E->getRParenLoc(), Record);
415 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
416 for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
417 const OffsetOfExpr::OffsetOfNode &ON = E->getComponent(I);
418 Record.push_back(ON.getKind()); // FIXME: Stable encoding
419 Writer.AddSourceLocation(ON.getRange().getBegin(), Record);
420 Writer.AddSourceLocation(ON.getRange().getEnd(), Record);
421 switch (ON.getKind()) {
422 case OffsetOfExpr::OffsetOfNode::Array:
423 Record.push_back(ON.getArrayExprIndex());
424 break;
425
426 case OffsetOfExpr::OffsetOfNode::Field:
427 Writer.AddDeclRef(ON.getField(), Record);
428 break;
429
430 case OffsetOfExpr::OffsetOfNode::Identifier:
431 Writer.AddIdentifierRef(ON.getFieldName(), Record);
432 break;
Douglas Gregorcc8a5d52010-04-29 00:18:15 +0000433
434 case OffsetOfExpr::OffsetOfNode::Base:
435 // FIXME: Implement this!
436 llvm_unreachable("PCH for offsetof(base-specifier) not implemented");
437 break;
Douglas Gregor8ecdb652010-04-28 22:16:22 +0000438 }
439 }
440 for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
441 Writer.WriteSubStmt(E->getIndexExpr(I));
442 Code = pch::EXPR_OFFSETOF;
443}
444
Mike Stump1eb44332009-09-09 15:08:12 +0000445void PCHStmtWriter::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000446 VisitExpr(E);
447 Record.push_back(E->isSizeOf());
448 if (E->isArgumentType())
John McCalla93c9342009-12-07 02:54:59 +0000449 Writer.AddTypeSourceInfo(E->getArgumentTypeInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000450 else {
451 Record.push_back(0);
452 Writer.WriteSubStmt(E->getArgumentExpr());
453 }
454 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
455 Writer.AddSourceLocation(E->getRParenLoc(), Record);
456 Code = pch::EXPR_SIZEOF_ALIGN_OF;
457}
458
459void PCHStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
460 VisitExpr(E);
461 Writer.WriteSubStmt(E->getLHS());
462 Writer.WriteSubStmt(E->getRHS());
463 Writer.AddSourceLocation(E->getRBracketLoc(), Record);
464 Code = pch::EXPR_ARRAY_SUBSCRIPT;
465}
466
467void PCHStmtWriter::VisitCallExpr(CallExpr *E) {
468 VisitExpr(E);
469 Record.push_back(E->getNumArgs());
470 Writer.AddSourceLocation(E->getRParenLoc(), Record);
471 Writer.WriteSubStmt(E->getCallee());
472 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
473 Arg != ArgEnd; ++Arg)
474 Writer.WriteSubStmt(*Arg);
475 Code = pch::EXPR_CALL;
476}
477
478void PCHStmtWriter::VisitMemberExpr(MemberExpr *E) {
479 VisitExpr(E);
480 Writer.WriteSubStmt(E->getBase());
481 Writer.AddDeclRef(E->getMemberDecl(), Record);
482 Writer.AddSourceLocation(E->getMemberLoc(), Record);
483 Record.push_back(E->isArrow());
Douglas Gregor303e6d72009-08-31 23:44:04 +0000484 // FIXME: C++ nested-name-specifier
Douglas Gregorc4bf26f2009-09-01 00:37:14 +0000485 // FIXME: C++ template argument list
Chris Lattner63377d52009-04-27 06:20:01 +0000486 Code = pch::EXPR_MEMBER;
487}
488
Steve Narofff242b1b2009-07-24 17:54:45 +0000489void PCHStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
490 VisitExpr(E);
491 Writer.WriteSubStmt(E->getBase());
492 Writer.AddSourceLocation(E->getIsaMemberLoc(), Record);
493 Record.push_back(E->isArrow());
494 Code = pch::EXPR_OBJC_ISA;
495}
496
Chris Lattner63377d52009-04-27 06:20:01 +0000497void PCHStmtWriter::VisitCastExpr(CastExpr *E) {
498 VisitExpr(E);
499 Writer.WriteSubStmt(E->getSubExpr());
Anders Carlssoncdef2b72009-07-31 00:48:10 +0000500 Record.push_back(E->getCastKind()); // FIXME: stable encoding
Chris Lattner63377d52009-04-27 06:20:01 +0000501}
502
503void PCHStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
504 VisitExpr(E);
505 Writer.WriteSubStmt(E->getLHS());
506 Writer.WriteSubStmt(E->getRHS());
507 Record.push_back(E->getOpcode()); // FIXME: stable encoding
508 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
509 Code = pch::EXPR_BINARY_OPERATOR;
510}
511
512void PCHStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
513 VisitBinaryOperator(E);
514 Writer.AddTypeRef(E->getComputationLHSType(), Record);
515 Writer.AddTypeRef(E->getComputationResultType(), Record);
516 Code = pch::EXPR_COMPOUND_ASSIGN_OPERATOR;
517}
518
519void PCHStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
520 VisitExpr(E);
521 Writer.WriteSubStmt(E->getCond());
522 Writer.WriteSubStmt(E->getLHS());
523 Writer.WriteSubStmt(E->getRHS());
Douglas Gregor47e1f7c2009-08-26 14:37:04 +0000524 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
525 Writer.AddSourceLocation(E->getColonLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000526 Code = pch::EXPR_CONDITIONAL_OPERATOR;
527}
528
529void PCHStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
530 VisitCastExpr(E);
531 Record.push_back(E->isLvalueCast());
532 Code = pch::EXPR_IMPLICIT_CAST;
533}
534
535void PCHStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
536 VisitCastExpr(E);
John McCall9d125032010-01-15 18:39:57 +0000537 Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000538}
539
540void PCHStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
541 VisitExplicitCastExpr(E);
542 Writer.AddSourceLocation(E->getLParenLoc(), Record);
543 Writer.AddSourceLocation(E->getRParenLoc(), Record);
544 Code = pch::EXPR_CSTYLE_CAST;
545}
546
547void PCHStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
548 VisitExpr(E);
549 Writer.AddSourceLocation(E->getLParenLoc(), Record);
John McCall42f56b52010-01-18 19:35:47 +0000550 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000551 Writer.WriteSubStmt(E->getInitializer());
552 Record.push_back(E->isFileScope());
553 Code = pch::EXPR_COMPOUND_LITERAL;
554}
555
556void PCHStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
557 VisitExpr(E);
558 Writer.WriteSubStmt(E->getBase());
559 Writer.AddIdentifierRef(&E->getAccessor(), Record);
560 Writer.AddSourceLocation(E->getAccessorLoc(), Record);
561 Code = pch::EXPR_EXT_VECTOR_ELEMENT;
562}
563
564void PCHStmtWriter::VisitInitListExpr(InitListExpr *E) {
565 VisitExpr(E);
566 Record.push_back(E->getNumInits());
567 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
568 Writer.WriteSubStmt(E->getInit(I));
569 Writer.WriteSubStmt(E->getSyntacticForm());
570 Writer.AddSourceLocation(E->getLBraceLoc(), Record);
571 Writer.AddSourceLocation(E->getRBraceLoc(), Record);
572 Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
573 Record.push_back(E->hadArrayRangeDesignator());
574 Code = pch::EXPR_INIT_LIST;
575}
576
577void PCHStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
578 VisitExpr(E);
579 Record.push_back(E->getNumSubExprs());
580 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
581 Writer.WriteSubStmt(E->getSubExpr(I));
582 Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
583 Record.push_back(E->usesGNUSyntax());
584 for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
585 DEnd = E->designators_end();
586 D != DEnd; ++D) {
587 if (D->isFieldDesignator()) {
588 if (FieldDecl *Field = D->getField()) {
589 Record.push_back(pch::DESIG_FIELD_DECL);
590 Writer.AddDeclRef(Field, Record);
591 } else {
592 Record.push_back(pch::DESIG_FIELD_NAME);
593 Writer.AddIdentifierRef(D->getFieldName(), Record);
594 }
595 Writer.AddSourceLocation(D->getDotLoc(), Record);
596 Writer.AddSourceLocation(D->getFieldLoc(), Record);
597 } else if (D->isArrayDesignator()) {
598 Record.push_back(pch::DESIG_ARRAY);
599 Record.push_back(D->getFirstExprIndex());
600 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
601 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
602 } else {
603 assert(D->isArrayRangeDesignator() && "Unknown designator");
604 Record.push_back(pch::DESIG_ARRAY_RANGE);
605 Record.push_back(D->getFirstExprIndex());
606 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
607 Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
608 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
609 }
610 }
611 Code = pch::EXPR_DESIGNATED_INIT;
612}
613
614void PCHStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
615 VisitExpr(E);
616 Code = pch::EXPR_IMPLICIT_VALUE_INIT;
617}
618
619void PCHStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
620 VisitExpr(E);
621 Writer.WriteSubStmt(E->getSubExpr());
622 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
623 Writer.AddSourceLocation(E->getRParenLoc(), Record);
624 Code = pch::EXPR_VA_ARG;
625}
626
627void PCHStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
628 VisitExpr(E);
629 Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
630 Writer.AddSourceLocation(E->getLabelLoc(), Record);
631 Record.push_back(Writer.GetLabelID(E->getLabel()));
632 Code = pch::EXPR_ADDR_LABEL;
633}
634
635void PCHStmtWriter::VisitStmtExpr(StmtExpr *E) {
636 VisitExpr(E);
637 Writer.WriteSubStmt(E->getSubStmt());
638 Writer.AddSourceLocation(E->getLParenLoc(), Record);
639 Writer.AddSourceLocation(E->getRParenLoc(), Record);
640 Code = pch::EXPR_STMT;
641}
642
643void PCHStmtWriter::VisitTypesCompatibleExpr(TypesCompatibleExpr *E) {
644 VisitExpr(E);
645 Writer.AddTypeRef(E->getArgType1(), Record);
646 Writer.AddTypeRef(E->getArgType2(), Record);
647 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
648 Writer.AddSourceLocation(E->getRParenLoc(), Record);
649 Code = pch::EXPR_TYPES_COMPATIBLE;
650}
651
652void PCHStmtWriter::VisitChooseExpr(ChooseExpr *E) {
653 VisitExpr(E);
654 Writer.WriteSubStmt(E->getCond());
655 Writer.WriteSubStmt(E->getLHS());
656 Writer.WriteSubStmt(E->getRHS());
657 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
658 Writer.AddSourceLocation(E->getRParenLoc(), Record);
659 Code = pch::EXPR_CHOOSE;
660}
661
662void PCHStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
663 VisitExpr(E);
664 Writer.AddSourceLocation(E->getTokenLocation(), Record);
665 Code = pch::EXPR_GNU_NULL;
666}
667
668void PCHStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
669 VisitExpr(E);
670 Record.push_back(E->getNumSubExprs());
671 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
672 Writer.WriteSubStmt(E->getExpr(I));
673 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
674 Writer.AddSourceLocation(E->getRParenLoc(), Record);
675 Code = pch::EXPR_SHUFFLE_VECTOR;
676}
677
678void PCHStmtWriter::VisitBlockExpr(BlockExpr *E) {
679 VisitExpr(E);
680 Writer.AddDeclRef(E->getBlockDecl(), Record);
681 Record.push_back(E->hasBlockDeclRefExprs());
682 Code = pch::EXPR_BLOCK;
683}
684
685void PCHStmtWriter::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
686 VisitExpr(E);
687 Writer.AddDeclRef(E->getDecl(), Record);
688 Writer.AddSourceLocation(E->getLocation(), Record);
689 Record.push_back(E->isByRef());
Fariborz Jahanian9b0b57c2009-06-20 00:02:26 +0000690 Record.push_back(E->isConstQualAdded());
Chris Lattner63377d52009-04-27 06:20:01 +0000691 Code = pch::EXPR_BLOCK_DECL_REF;
692}
693
694//===----------------------------------------------------------------------===//
695// Objective-C Expressions and Statements.
696//===----------------------------------------------------------------------===//
697
698void PCHStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
699 VisitExpr(E);
700 Writer.WriteSubStmt(E->getString());
701 Writer.AddSourceLocation(E->getAtLoc(), Record);
702 Code = pch::EXPR_OBJC_STRING_LITERAL;
703}
704
Mike Stump1eb44332009-09-09 15:08:12 +0000705void PCHStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000706 VisitExpr(E);
Douglas Gregor81d34662010-04-20 15:39:42 +0000707 Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000708 Writer.AddSourceLocation(E->getAtLoc(), Record);
709 Writer.AddSourceLocation(E->getRParenLoc(), Record);
710 Code = pch::EXPR_OBJC_ENCODE;
711}
712
713void PCHStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
714 VisitExpr(E);
715 Writer.AddSelectorRef(E->getSelector(), Record);
716 Writer.AddSourceLocation(E->getAtLoc(), Record);
717 Writer.AddSourceLocation(E->getRParenLoc(), Record);
718 Code = pch::EXPR_OBJC_SELECTOR_EXPR;
719}
720
721void PCHStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
722 VisitExpr(E);
723 Writer.AddDeclRef(E->getProtocol(), Record);
724 Writer.AddSourceLocation(E->getAtLoc(), Record);
725 Writer.AddSourceLocation(E->getRParenLoc(), Record);
726 Code = pch::EXPR_OBJC_PROTOCOL_EXPR;
727}
728
729void PCHStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
730 VisitExpr(E);
731 Writer.AddDeclRef(E->getDecl(), Record);
732 Writer.AddSourceLocation(E->getLocation(), Record);
733 Writer.WriteSubStmt(E->getBase());
734 Record.push_back(E->isArrow());
735 Record.push_back(E->isFreeIvar());
736 Code = pch::EXPR_OBJC_IVAR_REF_EXPR;
737}
738
739void PCHStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
740 VisitExpr(E);
741 Writer.AddDeclRef(E->getProperty(), Record);
742 Writer.AddSourceLocation(E->getLocation(), Record);
743 Writer.WriteSubStmt(E->getBase());
744 Code = pch::EXPR_OBJC_PROPERTY_REF_EXPR;
745}
746
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000747void PCHStmtWriter::VisitObjCImplicitSetterGetterRefExpr(
748 ObjCImplicitSetterGetterRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000749 VisitExpr(E);
750 Writer.AddDeclRef(E->getGetterMethod(), Record);
751 Writer.AddDeclRef(E->getSetterMethod(), Record);
Mike Stump1eb44332009-09-09 15:08:12 +0000752
Fariborz Jahaniand2ae5aa2009-08-18 21:37:33 +0000753 // NOTE: InterfaceDecl and Base are mutually exclusive.
754 Writer.AddDeclRef(E->getInterfaceDecl(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000755 Writer.WriteSubStmt(E->getBase());
756 Writer.AddSourceLocation(E->getLocation(), Record);
757 Writer.AddSourceLocation(E->getClassLoc(), Record);
758 Code = pch::EXPR_OBJC_KVC_REF_EXPR;
759}
760
761void PCHStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
762 VisitExpr(E);
763 Record.push_back(E->getNumArgs());
Douglas Gregor04badcf2010-04-21 00:45:42 +0000764 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
765 switch (E->getReceiverKind()) {
766 case ObjCMessageExpr::Instance:
767 Writer.WriteSubStmt(E->getInstanceReceiver());
768 break;
769
770 case ObjCMessageExpr::Class:
771 Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
772 break;
773
774 case ObjCMessageExpr::SuperClass:
775 case ObjCMessageExpr::SuperInstance:
776 Writer.AddTypeRef(E->getSuperType(), Record);
777 Writer.AddSourceLocation(E->getSuperLoc(), Record);
778 break;
779 }
780
781 if (E->getMethodDecl()) {
782 Record.push_back(1);
783 Writer.AddDeclRef(E->getMethodDecl(), Record);
784 } else {
785 Record.push_back(0);
786 Writer.AddSelectorRef(E->getSelector(), Record);
787 }
788
Chris Lattner63377d52009-04-27 06:20:01 +0000789 Writer.AddSourceLocation(E->getLeftLoc(), Record);
790 Writer.AddSourceLocation(E->getRightLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000791
792 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
793 Arg != ArgEnd; ++Arg)
794 Writer.WriteSubStmt(*Arg);
795 Code = pch::EXPR_OBJC_MESSAGE_EXPR;
796}
797
798void PCHStmtWriter::VisitObjCSuperExpr(ObjCSuperExpr *E) {
799 VisitExpr(E);
800 Writer.AddSourceLocation(E->getLoc(), Record);
801 Code = pch::EXPR_OBJC_SUPER_EXPR;
802}
803
804void PCHStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
805 VisitStmt(S);
806 Writer.WriteSubStmt(S->getElement());
807 Writer.WriteSubStmt(S->getCollection());
808 Writer.WriteSubStmt(S->getBody());
809 Writer.AddSourceLocation(S->getForLoc(), Record);
810 Writer.AddSourceLocation(S->getRParenLoc(), Record);
811 Code = pch::STMT_OBJC_FOR_COLLECTION;
812}
813
814void PCHStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
815 Writer.WriteSubStmt(S->getCatchBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000816 Writer.AddDeclRef(S->getCatchParamDecl(), Record);
817 Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
818 Writer.AddSourceLocation(S->getRParenLoc(), Record);
819 Code = pch::STMT_OBJC_CATCH;
820}
821
822void PCHStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
823 Writer.WriteSubStmt(S->getFinallyBody());
824 Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
825 Code = pch::STMT_OBJC_FINALLY;
826}
827
828void PCHStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000829 Record.push_back(S->getNumCatchStmts());
830 Record.push_back(S->getFinallyStmt() != 0);
Chris Lattner63377d52009-04-27 06:20:01 +0000831 Writer.WriteSubStmt(S->getTryBody());
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000832 for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
833 Writer.WriteSubStmt(S->getCatchStmt(I));
834 if (S->getFinallyStmt())
835 Writer.WriteSubStmt(S->getFinallyStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000836 Writer.AddSourceLocation(S->getAtTryLoc(), Record);
837 Code = pch::STMT_OBJC_AT_TRY;
838}
839
840void PCHStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
841 Writer.WriteSubStmt(S->getSynchExpr());
842 Writer.WriteSubStmt(S->getSynchBody());
843 Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
844 Code = pch::STMT_OBJC_AT_SYNCHRONIZED;
845}
846
847void PCHStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
848 Writer.WriteSubStmt(S->getThrowExpr());
849 Writer.AddSourceLocation(S->getThrowLoc(), Record);
850 Code = pch::STMT_OBJC_AT_THROW;
851}
852
853//===----------------------------------------------------------------------===//
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000854// C++ Expressions and Statements.
855//===----------------------------------------------------------------------===//
856
857void PCHStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
858 VisitCallExpr(E);
859 Record.push_back(E->getOperator());
860 Code = pch::EXPR_CXX_OPERATOR_CALL;
861}
862
Chris Lattner1817bd42010-05-09 05:36:05 +0000863void PCHStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
864 VisitCallExpr(E);
865 Code = pch::EXPR_CXX_MEMBER_CALL;
866}
867
Douglas Gregor39da0b82009-09-09 23:08:42 +0000868void PCHStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
869 VisitExpr(E);
870 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor99a2e602009-12-16 01:38:02 +0000871 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000872 Record.push_back(E->isElidable());
Douglas Gregor16006c92009-12-16 18:50:27 +0000873 Record.push_back(E->requiresZeroInitialization());
Douglas Gregor39da0b82009-09-09 23:08:42 +0000874 Record.push_back(E->getNumArgs());
875 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
876 Writer.WriteSubStmt(E->getArg(I));
Douglas Gregor3c9034c2010-05-15 00:13:29 +0000877 Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
Douglas Gregor39da0b82009-09-09 23:08:42 +0000878 Code = pch::EXPR_CXX_CONSTRUCT;
879}
880
Sam Weinigce757a72010-01-16 21:21:01 +0000881void PCHStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
882 VisitExplicitCastExpr(E);
883 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
884}
885
886void PCHStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
887 VisitCXXNamedCastExpr(E);
888 Code = pch::EXPR_CXX_STATIC_CAST;
889}
890
891void PCHStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
892 VisitCXXNamedCastExpr(E);
893 Code = pch::EXPR_CXX_DYNAMIC_CAST;
894}
895
896void PCHStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
897 VisitCXXNamedCastExpr(E);
898 Code = pch::EXPR_CXX_REINTERPRET_CAST;
899}
900
901void PCHStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
902 VisitCXXNamedCastExpr(E);
903 Code = pch::EXPR_CXX_CONST_CAST;
904}
905
906void PCHStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
907 VisitExplicitCastExpr(E);
908 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
909 Writer.AddSourceLocation(E->getRParenLoc(), Record);
910 Code = pch::EXPR_CXX_FUNCTIONAL_CAST;
911}
912
Sam Weinigeb7f9612010-02-07 06:32:43 +0000913void PCHStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
914 VisitExpr(E);
915 Record.push_back(E->getValue());
916 Writer.AddSourceLocation(E->getLocation(), Record);
917 Code = pch::EXPR_CXX_BOOL_LITERAL;
918}
919
920void PCHStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
921 VisitExpr(E);
922 Writer.AddSourceLocation(E->getLocation(), Record);
923 Code = pch::EXPR_CXX_NULL_PTR_LITERAL;
924}
925
Chris Lattner14ab24f2010-05-09 06:03:39 +0000926void PCHStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
927 VisitExpr(E);
928 Writer.AddSourceRange(E->getSourceRange(), Record);
929 if (E->isTypeOperand()) {
930 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
931 Code = pch::EXPR_CXX_TYPEID_TYPE;
932 } else {
933 Writer.WriteSubStmt(E->getExprOperand());
934 Code = pch::EXPR_CXX_TYPEID_EXPR;
935 }
936}
937
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000938void PCHStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
939 VisitExpr(E);
940 Writer.AddSourceLocation(E->getLocation(), Record);
941 Record.push_back(E->isImplicit());
942 Code = pch::EXPR_CXX_THIS;
943}
944
945void PCHStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
Chris Lattner030854b2010-05-09 06:40:08 +0000946 VisitExpr(E);
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000947 Writer.AddSourceLocation(E->getThrowLoc(), Record);
948 Writer.WriteSubStmt(E->getSubExpr());
949 Code = pch::EXPR_CXX_THROW;
950}
951
Chris Lattner030854b2010-05-09 06:40:08 +0000952void PCHStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
953 VisitExpr(E);
954 Writer.AddSourceLocation(E->getUsedLocation(), Record);
955 if (E->isExprStored()) {
956 Record.push_back(1);
957 Writer.WriteSubStmt(E->getExpr());
958 } else {
959 Record.push_back(0);
960 }
961
962 Code = pch::EXPR_CXX_DEFAULT_ARG;
963}
964
Chris Lattnerd2598362010-05-10 00:25:06 +0000965void PCHStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
966 VisitExpr(E);
967 Writer.AddCXXTemporary(E->getTemporary(), Record);
968 Writer.WriteSubStmt(E->getSubExpr());
969 Code = pch::EXPR_CXX_BIND_TEMPORARY;
970}
971
Chris Lattner59218632010-05-10 01:22:27 +0000972void PCHStmtWriter::VisitCXXZeroInitValueExpr(CXXZeroInitValueExpr *E) {
973 VisitExpr(E);
974 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
975 Writer.AddSourceLocation(E->getRParenLoc(), Record);
976 Code = pch::EXPR_CXX_ZERO_INIT_VALUE;
977}
978
979void PCHStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
980 VisitExpr(E);
981 Record.push_back(E->isGlobalNew());
982 Record.push_back(E->isParenTypeId());
983 Record.push_back(E->hasInitializer());
984 Record.push_back(E->isArray());
985 Record.push_back(E->getNumPlacementArgs());
986 Record.push_back(E->getNumConstructorArgs());
987 Writer.AddDeclRef(E->getOperatorNew(), Record);
988 Writer.AddDeclRef(E->getOperatorDelete(), Record);
989 Writer.AddDeclRef(E->getConstructor(), Record);
990 Writer.AddSourceLocation(E->getStartLoc(), Record);
991 Writer.AddSourceLocation(E->getEndLoc(), Record);
992 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
993 I != e; ++I)
994 Writer.WriteSubStmt(*I);
995
996 Code = pch::EXPR_CXX_NEW;
997}
998
999
Chris Lattnerd2598362010-05-10 00:25:06 +00001000void PCHStmtWriter::VisitCXXExprWithTemporaries(CXXExprWithTemporaries *E) {
1001 VisitExpr(E);
1002 Record.push_back(E->getNumTemporaries());
1003 for (unsigned i = 0, e = E->getNumTemporaries(); i != e; ++i)
1004 Writer.AddCXXTemporary(E->getTemporary(i), Record);
1005
1006 Writer.WriteSubStmt(E->getSubExpr());
1007 Code = pch::EXPR_CXX_EXPR_WITH_TEMPORARIES;
1008}
1009
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001010
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001011//===----------------------------------------------------------------------===//
Chris Lattner63377d52009-04-27 06:20:01 +00001012// PCHWriter Implementation
1013//===----------------------------------------------------------------------===//
1014
1015unsigned PCHWriter::RecordSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001016 assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001017 "SwitchCase recorded twice");
1018 unsigned NextID = SwitchCaseIDs.size();
1019 SwitchCaseIDs[S] = NextID;
1020 return NextID;
1021}
1022
1023unsigned PCHWriter::getSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001024 assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001025 "SwitchCase hasn't been seen yet");
1026 return SwitchCaseIDs[S];
1027}
1028
1029/// \brief Retrieve the ID for the given label statement, which may
1030/// or may not have been emitted yet.
1031unsigned PCHWriter::GetLabelID(LabelStmt *S) {
1032 std::map<LabelStmt *, unsigned>::iterator Pos = LabelIDs.find(S);
1033 if (Pos != LabelIDs.end())
1034 return Pos->second;
Mike Stump1eb44332009-09-09 15:08:12 +00001035
Chris Lattner63377d52009-04-27 06:20:01 +00001036 unsigned NextID = LabelIDs.size();
1037 LabelIDs[S] = NextID;
1038 return NextID;
1039}
1040
1041/// \brief Write the given substatement or subexpression to the
1042/// bitstream.
1043void PCHWriter::WriteSubStmt(Stmt *S) {
1044 RecordData Record;
1045 PCHStmtWriter Writer(*this, Record);
1046 ++NumStatements;
Mike Stump1eb44332009-09-09 15:08:12 +00001047
Chris Lattner63377d52009-04-27 06:20:01 +00001048 if (!S) {
1049 Stream.EmitRecord(pch::STMT_NULL_PTR, Record);
1050 return;
1051 }
Mike Stump1eb44332009-09-09 15:08:12 +00001052
Chris Lattner63377d52009-04-27 06:20:01 +00001053 Writer.Code = pch::STMT_NULL_PTR;
1054 Writer.Visit(S);
Chris Lattner59218632010-05-10 01:22:27 +00001055
1056#ifndef NDEBUG
1057 if (Writer.Code == pch::STMT_NULL_PTR) {
1058 S->dump();
1059 assert(0 && "Unhandled sub statement writing PCH file");
1060 }
1061#endif
Mike Stump1eb44332009-09-09 15:08:12 +00001062 Stream.EmitRecord(Writer.Code, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001063}
1064
1065/// \brief Flush all of the statements that have been added to the
1066/// queue via AddStmt().
1067void PCHWriter::FlushStmts() {
1068 RecordData Record;
1069 PCHStmtWriter Writer(*this, Record);
Mike Stump1eb44332009-09-09 15:08:12 +00001070
Chris Lattner63377d52009-04-27 06:20:01 +00001071 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
1072 ++NumStatements;
1073 Stmt *S = StmtsToEmit[I];
Mike Stump1eb44332009-09-09 15:08:12 +00001074
Chris Lattner63377d52009-04-27 06:20:01 +00001075 if (!S) {
1076 Stream.EmitRecord(pch::STMT_NULL_PTR, Record);
1077 continue;
1078 }
Mike Stump1eb44332009-09-09 15:08:12 +00001079
Chris Lattner63377d52009-04-27 06:20:01 +00001080 Writer.Code = pch::STMT_NULL_PTR;
1081 Writer.Visit(S);
Chris Lattnerd2598362010-05-10 00:25:06 +00001082#ifndef NDEBUG
1083 if (Writer.Code == pch::STMT_NULL_PTR) {
1084 S->dump();
1085 assert(0 && "Unhandled expression writing PCH file");
1086 }
1087#endif
Mike Stump1eb44332009-09-09 15:08:12 +00001088 Stream.EmitRecord(Writer.Code, Record);
1089
1090 assert(N == StmtsToEmit.size() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001091 "Substatement writen via AddStmt rather than WriteSubStmt!");
Mike Stump1eb44332009-09-09 15:08:12 +00001092
Chris Lattner63377d52009-04-27 06:20:01 +00001093 // Note that we are at the end of a full expression. Any
1094 // expression records that follow this one are part of a different
1095 // expression.
1096 Record.clear();
1097 Stream.EmitRecord(pch::STMT_STOP, Record);
1098 }
Mike Stump1eb44332009-09-09 15:08:12 +00001099
Chris Lattner63377d52009-04-27 06:20:01 +00001100 StmtsToEmit.clear();
Chris Lattner63377d52009-04-27 06:20:01 +00001101}