blob: b3501d0984456d95d2a28f5b740dffe126ba1291 [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());
Fariborz Jahanian89f9d3a2010-06-04 19:06:53 +0000691 Writer.WriteSubStmt(E->getCopyConstructorExpr());
Chris Lattner63377d52009-04-27 06:20:01 +0000692 Code = pch::EXPR_BLOCK_DECL_REF;
693}
694
695//===----------------------------------------------------------------------===//
696// Objective-C Expressions and Statements.
697//===----------------------------------------------------------------------===//
698
699void PCHStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
700 VisitExpr(E);
701 Writer.WriteSubStmt(E->getString());
702 Writer.AddSourceLocation(E->getAtLoc(), Record);
703 Code = pch::EXPR_OBJC_STRING_LITERAL;
704}
705
Mike Stump1eb44332009-09-09 15:08:12 +0000706void PCHStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000707 VisitExpr(E);
Douglas Gregor81d34662010-04-20 15:39:42 +0000708 Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000709 Writer.AddSourceLocation(E->getAtLoc(), Record);
710 Writer.AddSourceLocation(E->getRParenLoc(), Record);
711 Code = pch::EXPR_OBJC_ENCODE;
712}
713
714void PCHStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
715 VisitExpr(E);
716 Writer.AddSelectorRef(E->getSelector(), Record);
717 Writer.AddSourceLocation(E->getAtLoc(), Record);
718 Writer.AddSourceLocation(E->getRParenLoc(), Record);
719 Code = pch::EXPR_OBJC_SELECTOR_EXPR;
720}
721
722void PCHStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
723 VisitExpr(E);
724 Writer.AddDeclRef(E->getProtocol(), Record);
725 Writer.AddSourceLocation(E->getAtLoc(), Record);
726 Writer.AddSourceLocation(E->getRParenLoc(), Record);
727 Code = pch::EXPR_OBJC_PROTOCOL_EXPR;
728}
729
730void PCHStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
731 VisitExpr(E);
732 Writer.AddDeclRef(E->getDecl(), Record);
733 Writer.AddSourceLocation(E->getLocation(), Record);
734 Writer.WriteSubStmt(E->getBase());
735 Record.push_back(E->isArrow());
736 Record.push_back(E->isFreeIvar());
737 Code = pch::EXPR_OBJC_IVAR_REF_EXPR;
738}
739
740void PCHStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
741 VisitExpr(E);
742 Writer.AddDeclRef(E->getProperty(), Record);
743 Writer.AddSourceLocation(E->getLocation(), Record);
744 Writer.WriteSubStmt(E->getBase());
745 Code = pch::EXPR_OBJC_PROPERTY_REF_EXPR;
746}
747
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000748void PCHStmtWriter::VisitObjCImplicitSetterGetterRefExpr(
749 ObjCImplicitSetterGetterRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000750 VisitExpr(E);
751 Writer.AddDeclRef(E->getGetterMethod(), Record);
752 Writer.AddDeclRef(E->getSetterMethod(), Record);
Mike Stump1eb44332009-09-09 15:08:12 +0000753
Fariborz Jahaniand2ae5aa2009-08-18 21:37:33 +0000754 // NOTE: InterfaceDecl and Base are mutually exclusive.
755 Writer.AddDeclRef(E->getInterfaceDecl(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000756 Writer.WriteSubStmt(E->getBase());
757 Writer.AddSourceLocation(E->getLocation(), Record);
758 Writer.AddSourceLocation(E->getClassLoc(), Record);
759 Code = pch::EXPR_OBJC_KVC_REF_EXPR;
760}
761
762void PCHStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
763 VisitExpr(E);
764 Record.push_back(E->getNumArgs());
Douglas Gregor04badcf2010-04-21 00:45:42 +0000765 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
766 switch (E->getReceiverKind()) {
767 case ObjCMessageExpr::Instance:
768 Writer.WriteSubStmt(E->getInstanceReceiver());
769 break;
770
771 case ObjCMessageExpr::Class:
772 Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
773 break;
774
775 case ObjCMessageExpr::SuperClass:
776 case ObjCMessageExpr::SuperInstance:
777 Writer.AddTypeRef(E->getSuperType(), Record);
778 Writer.AddSourceLocation(E->getSuperLoc(), Record);
779 break;
780 }
781
782 if (E->getMethodDecl()) {
783 Record.push_back(1);
784 Writer.AddDeclRef(E->getMethodDecl(), Record);
785 } else {
786 Record.push_back(0);
787 Writer.AddSelectorRef(E->getSelector(), Record);
788 }
789
Chris Lattner63377d52009-04-27 06:20:01 +0000790 Writer.AddSourceLocation(E->getLeftLoc(), Record);
791 Writer.AddSourceLocation(E->getRightLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000792
793 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
794 Arg != ArgEnd; ++Arg)
795 Writer.WriteSubStmt(*Arg);
796 Code = pch::EXPR_OBJC_MESSAGE_EXPR;
797}
798
799void PCHStmtWriter::VisitObjCSuperExpr(ObjCSuperExpr *E) {
800 VisitExpr(E);
801 Writer.AddSourceLocation(E->getLoc(), Record);
802 Code = pch::EXPR_OBJC_SUPER_EXPR;
803}
804
805void PCHStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
806 VisitStmt(S);
807 Writer.WriteSubStmt(S->getElement());
808 Writer.WriteSubStmt(S->getCollection());
809 Writer.WriteSubStmt(S->getBody());
810 Writer.AddSourceLocation(S->getForLoc(), Record);
811 Writer.AddSourceLocation(S->getRParenLoc(), Record);
812 Code = pch::STMT_OBJC_FOR_COLLECTION;
813}
814
815void PCHStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
816 Writer.WriteSubStmt(S->getCatchBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000817 Writer.AddDeclRef(S->getCatchParamDecl(), Record);
818 Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
819 Writer.AddSourceLocation(S->getRParenLoc(), Record);
820 Code = pch::STMT_OBJC_CATCH;
821}
822
823void PCHStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
824 Writer.WriteSubStmt(S->getFinallyBody());
825 Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
826 Code = pch::STMT_OBJC_FINALLY;
827}
828
829void PCHStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000830 Record.push_back(S->getNumCatchStmts());
831 Record.push_back(S->getFinallyStmt() != 0);
Chris Lattner63377d52009-04-27 06:20:01 +0000832 Writer.WriteSubStmt(S->getTryBody());
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000833 for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
834 Writer.WriteSubStmt(S->getCatchStmt(I));
835 if (S->getFinallyStmt())
836 Writer.WriteSubStmt(S->getFinallyStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000837 Writer.AddSourceLocation(S->getAtTryLoc(), Record);
838 Code = pch::STMT_OBJC_AT_TRY;
839}
840
841void PCHStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
842 Writer.WriteSubStmt(S->getSynchExpr());
843 Writer.WriteSubStmt(S->getSynchBody());
844 Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
845 Code = pch::STMT_OBJC_AT_SYNCHRONIZED;
846}
847
848void PCHStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
849 Writer.WriteSubStmt(S->getThrowExpr());
850 Writer.AddSourceLocation(S->getThrowLoc(), Record);
851 Code = pch::STMT_OBJC_AT_THROW;
852}
853
854//===----------------------------------------------------------------------===//
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000855// C++ Expressions and Statements.
856//===----------------------------------------------------------------------===//
857
858void PCHStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
859 VisitCallExpr(E);
860 Record.push_back(E->getOperator());
861 Code = pch::EXPR_CXX_OPERATOR_CALL;
862}
863
Chris Lattner1817bd42010-05-09 05:36:05 +0000864void PCHStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
865 VisitCallExpr(E);
866 Code = pch::EXPR_CXX_MEMBER_CALL;
867}
868
Douglas Gregor39da0b82009-09-09 23:08:42 +0000869void PCHStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
870 VisitExpr(E);
871 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor99a2e602009-12-16 01:38:02 +0000872 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000873 Record.push_back(E->isElidable());
Douglas Gregor16006c92009-12-16 18:50:27 +0000874 Record.push_back(E->requiresZeroInitialization());
Douglas Gregor39da0b82009-09-09 23:08:42 +0000875 Record.push_back(E->getNumArgs());
876 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
877 Writer.WriteSubStmt(E->getArg(I));
Douglas Gregor3c9034c2010-05-15 00:13:29 +0000878 Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
Douglas Gregor39da0b82009-09-09 23:08:42 +0000879 Code = pch::EXPR_CXX_CONSTRUCT;
880}
881
Sam Weinigce757a72010-01-16 21:21:01 +0000882void PCHStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
883 VisitExplicitCastExpr(E);
884 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
885}
886
887void PCHStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
888 VisitCXXNamedCastExpr(E);
889 Code = pch::EXPR_CXX_STATIC_CAST;
890}
891
892void PCHStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
893 VisitCXXNamedCastExpr(E);
894 Code = pch::EXPR_CXX_DYNAMIC_CAST;
895}
896
897void PCHStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
898 VisitCXXNamedCastExpr(E);
899 Code = pch::EXPR_CXX_REINTERPRET_CAST;
900}
901
902void PCHStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
903 VisitCXXNamedCastExpr(E);
904 Code = pch::EXPR_CXX_CONST_CAST;
905}
906
907void PCHStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
908 VisitExplicitCastExpr(E);
909 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
910 Writer.AddSourceLocation(E->getRParenLoc(), Record);
911 Code = pch::EXPR_CXX_FUNCTIONAL_CAST;
912}
913
Sam Weinigeb7f9612010-02-07 06:32:43 +0000914void PCHStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
915 VisitExpr(E);
916 Record.push_back(E->getValue());
917 Writer.AddSourceLocation(E->getLocation(), Record);
918 Code = pch::EXPR_CXX_BOOL_LITERAL;
919}
920
921void PCHStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
922 VisitExpr(E);
923 Writer.AddSourceLocation(E->getLocation(), Record);
924 Code = pch::EXPR_CXX_NULL_PTR_LITERAL;
925}
926
Chris Lattner14ab24f2010-05-09 06:03:39 +0000927void PCHStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
928 VisitExpr(E);
929 Writer.AddSourceRange(E->getSourceRange(), Record);
930 if (E->isTypeOperand()) {
931 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
932 Code = pch::EXPR_CXX_TYPEID_TYPE;
933 } else {
934 Writer.WriteSubStmt(E->getExprOperand());
935 Code = pch::EXPR_CXX_TYPEID_EXPR;
936 }
937}
938
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000939void PCHStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
940 VisitExpr(E);
941 Writer.AddSourceLocation(E->getLocation(), Record);
942 Record.push_back(E->isImplicit());
943 Code = pch::EXPR_CXX_THIS;
944}
945
946void PCHStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
Chris Lattner030854b2010-05-09 06:40:08 +0000947 VisitExpr(E);
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000948 Writer.AddSourceLocation(E->getThrowLoc(), Record);
949 Writer.WriteSubStmt(E->getSubExpr());
950 Code = pch::EXPR_CXX_THROW;
951}
952
Chris Lattner030854b2010-05-09 06:40:08 +0000953void PCHStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
954 VisitExpr(E);
955 Writer.AddSourceLocation(E->getUsedLocation(), Record);
956 if (E->isExprStored()) {
957 Record.push_back(1);
958 Writer.WriteSubStmt(E->getExpr());
959 } else {
960 Record.push_back(0);
961 }
962
963 Code = pch::EXPR_CXX_DEFAULT_ARG;
964}
965
Chris Lattnerd2598362010-05-10 00:25:06 +0000966void PCHStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
967 VisitExpr(E);
968 Writer.AddCXXTemporary(E->getTemporary(), Record);
969 Writer.WriteSubStmt(E->getSubExpr());
970 Code = pch::EXPR_CXX_BIND_TEMPORARY;
971}
972
Chris Lattner59218632010-05-10 01:22:27 +0000973void PCHStmtWriter::VisitCXXZeroInitValueExpr(CXXZeroInitValueExpr *E) {
974 VisitExpr(E);
975 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
976 Writer.AddSourceLocation(E->getRParenLoc(), Record);
977 Code = pch::EXPR_CXX_ZERO_INIT_VALUE;
978}
979
980void PCHStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
981 VisitExpr(E);
982 Record.push_back(E->isGlobalNew());
983 Record.push_back(E->isParenTypeId());
984 Record.push_back(E->hasInitializer());
985 Record.push_back(E->isArray());
986 Record.push_back(E->getNumPlacementArgs());
987 Record.push_back(E->getNumConstructorArgs());
988 Writer.AddDeclRef(E->getOperatorNew(), Record);
989 Writer.AddDeclRef(E->getOperatorDelete(), Record);
990 Writer.AddDeclRef(E->getConstructor(), Record);
991 Writer.AddSourceLocation(E->getStartLoc(), Record);
992 Writer.AddSourceLocation(E->getEndLoc(), Record);
993 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
994 I != e; ++I)
995 Writer.WriteSubStmt(*I);
996
997 Code = pch::EXPR_CXX_NEW;
998}
999
1000
Chris Lattnerd2598362010-05-10 00:25:06 +00001001void PCHStmtWriter::VisitCXXExprWithTemporaries(CXXExprWithTemporaries *E) {
1002 VisitExpr(E);
1003 Record.push_back(E->getNumTemporaries());
1004 for (unsigned i = 0, e = E->getNumTemporaries(); i != e; ++i)
1005 Writer.AddCXXTemporary(E->getTemporary(i), Record);
1006
1007 Writer.WriteSubStmt(E->getSubExpr());
1008 Code = pch::EXPR_CXX_EXPR_WITH_TEMPORARIES;
1009}
1010
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001011
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001012//===----------------------------------------------------------------------===//
Chris Lattner63377d52009-04-27 06:20:01 +00001013// PCHWriter Implementation
1014//===----------------------------------------------------------------------===//
1015
1016unsigned PCHWriter::RecordSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001017 assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001018 "SwitchCase recorded twice");
1019 unsigned NextID = SwitchCaseIDs.size();
1020 SwitchCaseIDs[S] = NextID;
1021 return NextID;
1022}
1023
1024unsigned PCHWriter::getSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001025 assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001026 "SwitchCase hasn't been seen yet");
1027 return SwitchCaseIDs[S];
1028}
1029
1030/// \brief Retrieve the ID for the given label statement, which may
1031/// or may not have been emitted yet.
1032unsigned PCHWriter::GetLabelID(LabelStmt *S) {
1033 std::map<LabelStmt *, unsigned>::iterator Pos = LabelIDs.find(S);
1034 if (Pos != LabelIDs.end())
1035 return Pos->second;
Mike Stump1eb44332009-09-09 15:08:12 +00001036
Chris Lattner63377d52009-04-27 06:20:01 +00001037 unsigned NextID = LabelIDs.size();
1038 LabelIDs[S] = NextID;
1039 return NextID;
1040}
1041
1042/// \brief Write the given substatement or subexpression to the
1043/// bitstream.
1044void PCHWriter::WriteSubStmt(Stmt *S) {
1045 RecordData Record;
1046 PCHStmtWriter Writer(*this, Record);
1047 ++NumStatements;
Mike Stump1eb44332009-09-09 15:08:12 +00001048
Chris Lattner63377d52009-04-27 06:20:01 +00001049 if (!S) {
1050 Stream.EmitRecord(pch::STMT_NULL_PTR, Record);
1051 return;
1052 }
Mike Stump1eb44332009-09-09 15:08:12 +00001053
Chris Lattner63377d52009-04-27 06:20:01 +00001054 Writer.Code = pch::STMT_NULL_PTR;
1055 Writer.Visit(S);
Chris Lattner59218632010-05-10 01:22:27 +00001056
1057#ifndef NDEBUG
1058 if (Writer.Code == pch::STMT_NULL_PTR) {
1059 S->dump();
1060 assert(0 && "Unhandled sub statement writing PCH file");
1061 }
1062#endif
Mike Stump1eb44332009-09-09 15:08:12 +00001063 Stream.EmitRecord(Writer.Code, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001064}
1065
1066/// \brief Flush all of the statements that have been added to the
1067/// queue via AddStmt().
1068void PCHWriter::FlushStmts() {
1069 RecordData Record;
1070 PCHStmtWriter Writer(*this, Record);
Mike Stump1eb44332009-09-09 15:08:12 +00001071
Chris Lattner63377d52009-04-27 06:20:01 +00001072 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
1073 ++NumStatements;
1074 Stmt *S = StmtsToEmit[I];
Mike Stump1eb44332009-09-09 15:08:12 +00001075
Chris Lattner63377d52009-04-27 06:20:01 +00001076 if (!S) {
1077 Stream.EmitRecord(pch::STMT_NULL_PTR, Record);
1078 continue;
1079 }
Mike Stump1eb44332009-09-09 15:08:12 +00001080
Chris Lattner63377d52009-04-27 06:20:01 +00001081 Writer.Code = pch::STMT_NULL_PTR;
1082 Writer.Visit(S);
Chris Lattnerd2598362010-05-10 00:25:06 +00001083#ifndef NDEBUG
1084 if (Writer.Code == pch::STMT_NULL_PTR) {
1085 S->dump();
1086 assert(0 && "Unhandled expression writing PCH file");
1087 }
1088#endif
Mike Stump1eb44332009-09-09 15:08:12 +00001089 Stream.EmitRecord(Writer.Code, Record);
1090
1091 assert(N == StmtsToEmit.size() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001092 "Substatement writen via AddStmt rather than WriteSubStmt!");
Mike Stump1eb44332009-09-09 15:08:12 +00001093
Chris Lattner63377d52009-04-27 06:20:01 +00001094 // Note that we are at the end of a full expression. Any
1095 // expression records that follow this one are part of a different
1096 // expression.
1097 Record.clear();
1098 Stream.EmitRecord(pch::STMT_STOP, Record);
1099 }
Mike Stump1eb44332009-09-09 15:08:12 +00001100
Chris Lattner63377d52009-04-27 06:20:01 +00001101 StmtsToEmit.clear();
Chris Lattner63377d52009-04-27 06:20:01 +00001102}