blob: 9a361d499c4c6d0f79ce2e40bf7c07dee2687ff0 [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);
278 Code = pch::STMT_RETURN;
279}
280
281void PCHStmtWriter::VisitDeclStmt(DeclStmt *S) {
282 VisitStmt(S);
283 Writer.AddSourceLocation(S->getStartLoc(), Record);
284 Writer.AddSourceLocation(S->getEndLoc(), Record);
285 DeclGroupRef DG = S->getDeclGroup();
286 for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
287 Writer.AddDeclRef(*D, Record);
288 Code = pch::STMT_DECL;
289}
290
291void PCHStmtWriter::VisitAsmStmt(AsmStmt *S) {
292 VisitStmt(S);
293 Record.push_back(S->getNumOutputs());
294 Record.push_back(S->getNumInputs());
295 Record.push_back(S->getNumClobbers());
296 Writer.AddSourceLocation(S->getAsmLoc(), Record);
297 Writer.AddSourceLocation(S->getRParenLoc(), Record);
298 Record.push_back(S->isVolatile());
299 Record.push_back(S->isSimple());
Mike Stump3b11fd32010-01-04 22:37:17 +0000300 Record.push_back(S->isMSAsm());
Chris Lattner63377d52009-04-27 06:20:01 +0000301 Writer.WriteSubStmt(S->getAsmString());
302
303 // Outputs
Anders Carlssonff93dbd2010-01-30 22:25:16 +0000304 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
305 Writer.AddIdentifierRef(S->getOutputIdentifier(I), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000306 Writer.WriteSubStmt(S->getOutputConstraintLiteral(I));
307 Writer.WriteSubStmt(S->getOutputExpr(I));
308 }
309
310 // Inputs
311 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
Anders Carlssonff93dbd2010-01-30 22:25:16 +0000312 Writer.AddIdentifierRef(S->getInputIdentifier(I), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000313 Writer.WriteSubStmt(S->getInputConstraintLiteral(I));
314 Writer.WriteSubStmt(S->getInputExpr(I));
315 }
316
317 // Clobbers
318 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
319 Writer.WriteSubStmt(S->getClobber(I));
320
321 Code = pch::STMT_ASM;
322}
323
324void PCHStmtWriter::VisitExpr(Expr *E) {
325 VisitStmt(E);
326 Writer.AddTypeRef(E->getType(), Record);
327 Record.push_back(E->isTypeDependent());
328 Record.push_back(E->isValueDependent());
329}
330
331void PCHStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
332 VisitExpr(E);
333 Writer.AddSourceLocation(E->getLocation(), Record);
334 Record.push_back(E->getIdentType()); // FIXME: stable encoding
335 Code = pch::EXPR_PREDEFINED;
336}
337
338void PCHStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
339 VisitExpr(E);
340 Writer.AddDeclRef(E->getDecl(), Record);
341 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregora2813ce2009-10-23 18:54:35 +0000342 // FIXME: write qualifier
343 // FIXME: write explicit template arguments
Chris Lattner63377d52009-04-27 06:20:01 +0000344 Code = pch::EXPR_DECL_REF;
345}
346
347void PCHStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
348 VisitExpr(E);
349 Writer.AddSourceLocation(E->getLocation(), Record);
350 Writer.AddAPInt(E->getValue(), Record);
351 Code = pch::EXPR_INTEGER_LITERAL;
352}
353
354void PCHStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
355 VisitExpr(E);
356 Writer.AddAPFloat(E->getValue(), Record);
357 Record.push_back(E->isExact());
358 Writer.AddSourceLocation(E->getLocation(), Record);
359 Code = pch::EXPR_FLOATING_LITERAL;
360}
361
362void PCHStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
363 VisitExpr(E);
364 Writer.WriteSubStmt(E->getSubExpr());
365 Code = pch::EXPR_IMAGINARY_LITERAL;
366}
367
368void PCHStmtWriter::VisitStringLiteral(StringLiteral *E) {
369 VisitExpr(E);
370 Record.push_back(E->getByteLength());
371 Record.push_back(E->getNumConcatenated());
372 Record.push_back(E->isWide());
373 // FIXME: String data should be stored as a blob at the end of the
374 // StringLiteral. However, we can't do so now because we have no
375 // provision for coping with abbreviations when we're jumping around
376 // the PCH file during deserialization.
Mike Stump1eb44332009-09-09 15:08:12 +0000377 Record.insert(Record.end(),
Chris Lattner63377d52009-04-27 06:20:01 +0000378 E->getStrData(), E->getStrData() + E->getByteLength());
379 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
380 Writer.AddSourceLocation(E->getStrTokenLoc(I), Record);
381 Code = pch::EXPR_STRING_LITERAL;
382}
383
384void PCHStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
385 VisitExpr(E);
386 Record.push_back(E->getValue());
Chris Lattner018d8e02009-08-24 17:39:36 +0000387 Writer.AddSourceLocation(E->getLocation(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000388 Record.push_back(E->isWide());
389 Code = pch::EXPR_CHARACTER_LITERAL;
390}
391
392void PCHStmtWriter::VisitParenExpr(ParenExpr *E) {
393 VisitExpr(E);
394 Writer.AddSourceLocation(E->getLParen(), Record);
395 Writer.AddSourceLocation(E->getRParen(), Record);
396 Writer.WriteSubStmt(E->getSubExpr());
397 Code = pch::EXPR_PAREN;
398}
399
400void PCHStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
401 VisitExpr(E);
402 Writer.WriteSubStmt(E->getSubExpr());
403 Record.push_back(E->getOpcode()); // FIXME: stable encoding
404 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
405 Code = pch::EXPR_UNARY_OPERATOR;
406}
407
Douglas Gregor8ecdb652010-04-28 22:16:22 +0000408void PCHStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
409 VisitExpr(E);
410 Record.push_back(E->getNumComponents());
411 Record.push_back(E->getNumExpressions());
412 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
413 Writer.AddSourceLocation(E->getRParenLoc(), Record);
414 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
415 for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
416 const OffsetOfExpr::OffsetOfNode &ON = E->getComponent(I);
417 Record.push_back(ON.getKind()); // FIXME: Stable encoding
418 Writer.AddSourceLocation(ON.getRange().getBegin(), Record);
419 Writer.AddSourceLocation(ON.getRange().getEnd(), Record);
420 switch (ON.getKind()) {
421 case OffsetOfExpr::OffsetOfNode::Array:
422 Record.push_back(ON.getArrayExprIndex());
423 break;
424
425 case OffsetOfExpr::OffsetOfNode::Field:
426 Writer.AddDeclRef(ON.getField(), Record);
427 break;
428
429 case OffsetOfExpr::OffsetOfNode::Identifier:
430 Writer.AddIdentifierRef(ON.getFieldName(), Record);
431 break;
Douglas Gregorcc8a5d52010-04-29 00:18:15 +0000432
433 case OffsetOfExpr::OffsetOfNode::Base:
434 // FIXME: Implement this!
435 llvm_unreachable("PCH for offsetof(base-specifier) not implemented");
436 break;
Douglas Gregor8ecdb652010-04-28 22:16:22 +0000437 }
438 }
439 for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
440 Writer.WriteSubStmt(E->getIndexExpr(I));
441 Code = pch::EXPR_OFFSETOF;
442}
443
Mike Stump1eb44332009-09-09 15:08:12 +0000444void PCHStmtWriter::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000445 VisitExpr(E);
446 Record.push_back(E->isSizeOf());
447 if (E->isArgumentType())
John McCalla93c9342009-12-07 02:54:59 +0000448 Writer.AddTypeSourceInfo(E->getArgumentTypeInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000449 else {
450 Record.push_back(0);
451 Writer.WriteSubStmt(E->getArgumentExpr());
452 }
453 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
454 Writer.AddSourceLocation(E->getRParenLoc(), Record);
455 Code = pch::EXPR_SIZEOF_ALIGN_OF;
456}
457
458void PCHStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
459 VisitExpr(E);
460 Writer.WriteSubStmt(E->getLHS());
461 Writer.WriteSubStmt(E->getRHS());
462 Writer.AddSourceLocation(E->getRBracketLoc(), Record);
463 Code = pch::EXPR_ARRAY_SUBSCRIPT;
464}
465
466void PCHStmtWriter::VisitCallExpr(CallExpr *E) {
467 VisitExpr(E);
468 Record.push_back(E->getNumArgs());
469 Writer.AddSourceLocation(E->getRParenLoc(), Record);
470 Writer.WriteSubStmt(E->getCallee());
471 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
472 Arg != ArgEnd; ++Arg)
473 Writer.WriteSubStmt(*Arg);
474 Code = pch::EXPR_CALL;
475}
476
477void PCHStmtWriter::VisitMemberExpr(MemberExpr *E) {
478 VisitExpr(E);
479 Writer.WriteSubStmt(E->getBase());
480 Writer.AddDeclRef(E->getMemberDecl(), Record);
481 Writer.AddSourceLocation(E->getMemberLoc(), Record);
482 Record.push_back(E->isArrow());
Douglas Gregor303e6d72009-08-31 23:44:04 +0000483 // FIXME: C++ nested-name-specifier
Douglas Gregorc4bf26f2009-09-01 00:37:14 +0000484 // FIXME: C++ template argument list
Chris Lattner63377d52009-04-27 06:20:01 +0000485 Code = pch::EXPR_MEMBER;
486}
487
Steve Narofff242b1b2009-07-24 17:54:45 +0000488void PCHStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
489 VisitExpr(E);
490 Writer.WriteSubStmt(E->getBase());
491 Writer.AddSourceLocation(E->getIsaMemberLoc(), Record);
492 Record.push_back(E->isArrow());
493 Code = pch::EXPR_OBJC_ISA;
494}
495
Chris Lattner63377d52009-04-27 06:20:01 +0000496void PCHStmtWriter::VisitCastExpr(CastExpr *E) {
497 VisitExpr(E);
498 Writer.WriteSubStmt(E->getSubExpr());
Anders Carlssoncdef2b72009-07-31 00:48:10 +0000499 Record.push_back(E->getCastKind()); // FIXME: stable encoding
Chris Lattner63377d52009-04-27 06:20:01 +0000500}
501
502void PCHStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
503 VisitExpr(E);
504 Writer.WriteSubStmt(E->getLHS());
505 Writer.WriteSubStmt(E->getRHS());
506 Record.push_back(E->getOpcode()); // FIXME: stable encoding
507 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
508 Code = pch::EXPR_BINARY_OPERATOR;
509}
510
511void PCHStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
512 VisitBinaryOperator(E);
513 Writer.AddTypeRef(E->getComputationLHSType(), Record);
514 Writer.AddTypeRef(E->getComputationResultType(), Record);
515 Code = pch::EXPR_COMPOUND_ASSIGN_OPERATOR;
516}
517
518void PCHStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
519 VisitExpr(E);
520 Writer.WriteSubStmt(E->getCond());
521 Writer.WriteSubStmt(E->getLHS());
522 Writer.WriteSubStmt(E->getRHS());
Douglas Gregor47e1f7c2009-08-26 14:37:04 +0000523 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
524 Writer.AddSourceLocation(E->getColonLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000525 Code = pch::EXPR_CONDITIONAL_OPERATOR;
526}
527
528void PCHStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
529 VisitCastExpr(E);
530 Record.push_back(E->isLvalueCast());
531 Code = pch::EXPR_IMPLICIT_CAST;
532}
533
534void PCHStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
535 VisitCastExpr(E);
John McCall9d125032010-01-15 18:39:57 +0000536 Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000537}
538
539void PCHStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
540 VisitExplicitCastExpr(E);
541 Writer.AddSourceLocation(E->getLParenLoc(), Record);
542 Writer.AddSourceLocation(E->getRParenLoc(), Record);
543 Code = pch::EXPR_CSTYLE_CAST;
544}
545
546void PCHStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
547 VisitExpr(E);
548 Writer.AddSourceLocation(E->getLParenLoc(), Record);
John McCall42f56b52010-01-18 19:35:47 +0000549 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000550 Writer.WriteSubStmt(E->getInitializer());
551 Record.push_back(E->isFileScope());
552 Code = pch::EXPR_COMPOUND_LITERAL;
553}
554
555void PCHStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
556 VisitExpr(E);
557 Writer.WriteSubStmt(E->getBase());
558 Writer.AddIdentifierRef(&E->getAccessor(), Record);
559 Writer.AddSourceLocation(E->getAccessorLoc(), Record);
560 Code = pch::EXPR_EXT_VECTOR_ELEMENT;
561}
562
563void PCHStmtWriter::VisitInitListExpr(InitListExpr *E) {
564 VisitExpr(E);
565 Record.push_back(E->getNumInits());
566 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
567 Writer.WriteSubStmt(E->getInit(I));
568 Writer.WriteSubStmt(E->getSyntacticForm());
569 Writer.AddSourceLocation(E->getLBraceLoc(), Record);
570 Writer.AddSourceLocation(E->getRBraceLoc(), Record);
571 Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
572 Record.push_back(E->hadArrayRangeDesignator());
573 Code = pch::EXPR_INIT_LIST;
574}
575
576void PCHStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
577 VisitExpr(E);
578 Record.push_back(E->getNumSubExprs());
579 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
580 Writer.WriteSubStmt(E->getSubExpr(I));
581 Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
582 Record.push_back(E->usesGNUSyntax());
583 for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
584 DEnd = E->designators_end();
585 D != DEnd; ++D) {
586 if (D->isFieldDesignator()) {
587 if (FieldDecl *Field = D->getField()) {
588 Record.push_back(pch::DESIG_FIELD_DECL);
589 Writer.AddDeclRef(Field, Record);
590 } else {
591 Record.push_back(pch::DESIG_FIELD_NAME);
592 Writer.AddIdentifierRef(D->getFieldName(), Record);
593 }
594 Writer.AddSourceLocation(D->getDotLoc(), Record);
595 Writer.AddSourceLocation(D->getFieldLoc(), Record);
596 } else if (D->isArrayDesignator()) {
597 Record.push_back(pch::DESIG_ARRAY);
598 Record.push_back(D->getFirstExprIndex());
599 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
600 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
601 } else {
602 assert(D->isArrayRangeDesignator() && "Unknown designator");
603 Record.push_back(pch::DESIG_ARRAY_RANGE);
604 Record.push_back(D->getFirstExprIndex());
605 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
606 Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
607 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
608 }
609 }
610 Code = pch::EXPR_DESIGNATED_INIT;
611}
612
613void PCHStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
614 VisitExpr(E);
615 Code = pch::EXPR_IMPLICIT_VALUE_INIT;
616}
617
618void PCHStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
619 VisitExpr(E);
620 Writer.WriteSubStmt(E->getSubExpr());
621 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
622 Writer.AddSourceLocation(E->getRParenLoc(), Record);
623 Code = pch::EXPR_VA_ARG;
624}
625
626void PCHStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
627 VisitExpr(E);
628 Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
629 Writer.AddSourceLocation(E->getLabelLoc(), Record);
630 Record.push_back(Writer.GetLabelID(E->getLabel()));
631 Code = pch::EXPR_ADDR_LABEL;
632}
633
634void PCHStmtWriter::VisitStmtExpr(StmtExpr *E) {
635 VisitExpr(E);
636 Writer.WriteSubStmt(E->getSubStmt());
637 Writer.AddSourceLocation(E->getLParenLoc(), Record);
638 Writer.AddSourceLocation(E->getRParenLoc(), Record);
639 Code = pch::EXPR_STMT;
640}
641
642void PCHStmtWriter::VisitTypesCompatibleExpr(TypesCompatibleExpr *E) {
643 VisitExpr(E);
644 Writer.AddTypeRef(E->getArgType1(), Record);
645 Writer.AddTypeRef(E->getArgType2(), Record);
646 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
647 Writer.AddSourceLocation(E->getRParenLoc(), Record);
648 Code = pch::EXPR_TYPES_COMPATIBLE;
649}
650
651void PCHStmtWriter::VisitChooseExpr(ChooseExpr *E) {
652 VisitExpr(E);
653 Writer.WriteSubStmt(E->getCond());
654 Writer.WriteSubStmt(E->getLHS());
655 Writer.WriteSubStmt(E->getRHS());
656 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
657 Writer.AddSourceLocation(E->getRParenLoc(), Record);
658 Code = pch::EXPR_CHOOSE;
659}
660
661void PCHStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
662 VisitExpr(E);
663 Writer.AddSourceLocation(E->getTokenLocation(), Record);
664 Code = pch::EXPR_GNU_NULL;
665}
666
667void PCHStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
668 VisitExpr(E);
669 Record.push_back(E->getNumSubExprs());
670 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
671 Writer.WriteSubStmt(E->getExpr(I));
672 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
673 Writer.AddSourceLocation(E->getRParenLoc(), Record);
674 Code = pch::EXPR_SHUFFLE_VECTOR;
675}
676
677void PCHStmtWriter::VisitBlockExpr(BlockExpr *E) {
678 VisitExpr(E);
679 Writer.AddDeclRef(E->getBlockDecl(), Record);
680 Record.push_back(E->hasBlockDeclRefExprs());
681 Code = pch::EXPR_BLOCK;
682}
683
684void PCHStmtWriter::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
685 VisitExpr(E);
686 Writer.AddDeclRef(E->getDecl(), Record);
687 Writer.AddSourceLocation(E->getLocation(), Record);
688 Record.push_back(E->isByRef());
Fariborz Jahanian9b0b57c2009-06-20 00:02:26 +0000689 Record.push_back(E->isConstQualAdded());
Chris Lattner63377d52009-04-27 06:20:01 +0000690 Code = pch::EXPR_BLOCK_DECL_REF;
691}
692
693//===----------------------------------------------------------------------===//
694// Objective-C Expressions and Statements.
695//===----------------------------------------------------------------------===//
696
697void PCHStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
698 VisitExpr(E);
699 Writer.WriteSubStmt(E->getString());
700 Writer.AddSourceLocation(E->getAtLoc(), Record);
701 Code = pch::EXPR_OBJC_STRING_LITERAL;
702}
703
Mike Stump1eb44332009-09-09 15:08:12 +0000704void PCHStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000705 VisitExpr(E);
Douglas Gregor81d34662010-04-20 15:39:42 +0000706 Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000707 Writer.AddSourceLocation(E->getAtLoc(), Record);
708 Writer.AddSourceLocation(E->getRParenLoc(), Record);
709 Code = pch::EXPR_OBJC_ENCODE;
710}
711
712void PCHStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
713 VisitExpr(E);
714 Writer.AddSelectorRef(E->getSelector(), Record);
715 Writer.AddSourceLocation(E->getAtLoc(), Record);
716 Writer.AddSourceLocation(E->getRParenLoc(), Record);
717 Code = pch::EXPR_OBJC_SELECTOR_EXPR;
718}
719
720void PCHStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
721 VisitExpr(E);
722 Writer.AddDeclRef(E->getProtocol(), Record);
723 Writer.AddSourceLocation(E->getAtLoc(), Record);
724 Writer.AddSourceLocation(E->getRParenLoc(), Record);
725 Code = pch::EXPR_OBJC_PROTOCOL_EXPR;
726}
727
728void PCHStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
729 VisitExpr(E);
730 Writer.AddDeclRef(E->getDecl(), Record);
731 Writer.AddSourceLocation(E->getLocation(), Record);
732 Writer.WriteSubStmt(E->getBase());
733 Record.push_back(E->isArrow());
734 Record.push_back(E->isFreeIvar());
735 Code = pch::EXPR_OBJC_IVAR_REF_EXPR;
736}
737
738void PCHStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
739 VisitExpr(E);
740 Writer.AddDeclRef(E->getProperty(), Record);
741 Writer.AddSourceLocation(E->getLocation(), Record);
742 Writer.WriteSubStmt(E->getBase());
743 Code = pch::EXPR_OBJC_PROPERTY_REF_EXPR;
744}
745
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000746void PCHStmtWriter::VisitObjCImplicitSetterGetterRefExpr(
747 ObjCImplicitSetterGetterRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000748 VisitExpr(E);
749 Writer.AddDeclRef(E->getGetterMethod(), Record);
750 Writer.AddDeclRef(E->getSetterMethod(), Record);
Mike Stump1eb44332009-09-09 15:08:12 +0000751
Fariborz Jahaniand2ae5aa2009-08-18 21:37:33 +0000752 // NOTE: InterfaceDecl and Base are mutually exclusive.
753 Writer.AddDeclRef(E->getInterfaceDecl(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000754 Writer.WriteSubStmt(E->getBase());
755 Writer.AddSourceLocation(E->getLocation(), Record);
756 Writer.AddSourceLocation(E->getClassLoc(), Record);
757 Code = pch::EXPR_OBJC_KVC_REF_EXPR;
758}
759
760void PCHStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
761 VisitExpr(E);
762 Record.push_back(E->getNumArgs());
Douglas Gregor04badcf2010-04-21 00:45:42 +0000763 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
764 switch (E->getReceiverKind()) {
765 case ObjCMessageExpr::Instance:
766 Writer.WriteSubStmt(E->getInstanceReceiver());
767 break;
768
769 case ObjCMessageExpr::Class:
770 Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
771 break;
772
773 case ObjCMessageExpr::SuperClass:
774 case ObjCMessageExpr::SuperInstance:
775 Writer.AddTypeRef(E->getSuperType(), Record);
776 Writer.AddSourceLocation(E->getSuperLoc(), Record);
777 break;
778 }
779
780 if (E->getMethodDecl()) {
781 Record.push_back(1);
782 Writer.AddDeclRef(E->getMethodDecl(), Record);
783 } else {
784 Record.push_back(0);
785 Writer.AddSelectorRef(E->getSelector(), Record);
786 }
787
Chris Lattner63377d52009-04-27 06:20:01 +0000788 Writer.AddSourceLocation(E->getLeftLoc(), Record);
789 Writer.AddSourceLocation(E->getRightLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000790
791 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
792 Arg != ArgEnd; ++Arg)
793 Writer.WriteSubStmt(*Arg);
794 Code = pch::EXPR_OBJC_MESSAGE_EXPR;
795}
796
797void PCHStmtWriter::VisitObjCSuperExpr(ObjCSuperExpr *E) {
798 VisitExpr(E);
799 Writer.AddSourceLocation(E->getLoc(), Record);
800 Code = pch::EXPR_OBJC_SUPER_EXPR;
801}
802
803void PCHStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
804 VisitStmt(S);
805 Writer.WriteSubStmt(S->getElement());
806 Writer.WriteSubStmt(S->getCollection());
807 Writer.WriteSubStmt(S->getBody());
808 Writer.AddSourceLocation(S->getForLoc(), Record);
809 Writer.AddSourceLocation(S->getRParenLoc(), Record);
810 Code = pch::STMT_OBJC_FOR_COLLECTION;
811}
812
813void PCHStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
814 Writer.WriteSubStmt(S->getCatchBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000815 Writer.AddDeclRef(S->getCatchParamDecl(), Record);
816 Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
817 Writer.AddSourceLocation(S->getRParenLoc(), Record);
818 Code = pch::STMT_OBJC_CATCH;
819}
820
821void PCHStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
822 Writer.WriteSubStmt(S->getFinallyBody());
823 Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
824 Code = pch::STMT_OBJC_FINALLY;
825}
826
827void PCHStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000828 Record.push_back(S->getNumCatchStmts());
829 Record.push_back(S->getFinallyStmt() != 0);
Chris Lattner63377d52009-04-27 06:20:01 +0000830 Writer.WriteSubStmt(S->getTryBody());
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000831 for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
832 Writer.WriteSubStmt(S->getCatchStmt(I));
833 if (S->getFinallyStmt())
834 Writer.WriteSubStmt(S->getFinallyStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000835 Writer.AddSourceLocation(S->getAtTryLoc(), Record);
836 Code = pch::STMT_OBJC_AT_TRY;
837}
838
839void PCHStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
840 Writer.WriteSubStmt(S->getSynchExpr());
841 Writer.WriteSubStmt(S->getSynchBody());
842 Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
843 Code = pch::STMT_OBJC_AT_SYNCHRONIZED;
844}
845
846void PCHStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
847 Writer.WriteSubStmt(S->getThrowExpr());
848 Writer.AddSourceLocation(S->getThrowLoc(), Record);
849 Code = pch::STMT_OBJC_AT_THROW;
850}
851
852//===----------------------------------------------------------------------===//
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000853// C++ Expressions and Statements.
854//===----------------------------------------------------------------------===//
855
856void PCHStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
857 VisitCallExpr(E);
858 Record.push_back(E->getOperator());
859 Code = pch::EXPR_CXX_OPERATOR_CALL;
860}
861
Chris Lattner1817bd42010-05-09 05:36:05 +0000862void PCHStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
863 VisitCallExpr(E);
864 Code = pch::EXPR_CXX_MEMBER_CALL;
865}
866
Douglas Gregor39da0b82009-09-09 23:08:42 +0000867void PCHStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
868 VisitExpr(E);
869 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor99a2e602009-12-16 01:38:02 +0000870 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000871 Record.push_back(E->isElidable());
Douglas Gregor16006c92009-12-16 18:50:27 +0000872 Record.push_back(E->requiresZeroInitialization());
Douglas Gregor39da0b82009-09-09 23:08:42 +0000873 Record.push_back(E->getNumArgs());
874 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
875 Writer.WriteSubStmt(E->getArg(I));
876 Code = pch::EXPR_CXX_CONSTRUCT;
877}
878
Sam Weinigce757a72010-01-16 21:21:01 +0000879void PCHStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
880 VisitExplicitCastExpr(E);
881 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
882}
883
884void PCHStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
885 VisitCXXNamedCastExpr(E);
886 Code = pch::EXPR_CXX_STATIC_CAST;
887}
888
889void PCHStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
890 VisitCXXNamedCastExpr(E);
891 Code = pch::EXPR_CXX_DYNAMIC_CAST;
892}
893
894void PCHStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
895 VisitCXXNamedCastExpr(E);
896 Code = pch::EXPR_CXX_REINTERPRET_CAST;
897}
898
899void PCHStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
900 VisitCXXNamedCastExpr(E);
901 Code = pch::EXPR_CXX_CONST_CAST;
902}
903
904void PCHStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
905 VisitExplicitCastExpr(E);
906 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
907 Writer.AddSourceLocation(E->getRParenLoc(), Record);
908 Code = pch::EXPR_CXX_FUNCTIONAL_CAST;
909}
910
Sam Weinigeb7f9612010-02-07 06:32:43 +0000911void PCHStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
912 VisitExpr(E);
913 Record.push_back(E->getValue());
914 Writer.AddSourceLocation(E->getLocation(), Record);
915 Code = pch::EXPR_CXX_BOOL_LITERAL;
916}
917
918void PCHStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
919 VisitExpr(E);
920 Writer.AddSourceLocation(E->getLocation(), Record);
921 Code = pch::EXPR_CXX_NULL_PTR_LITERAL;
922}
923
Chris Lattner14ab24f2010-05-09 06:03:39 +0000924void PCHStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
925 VisitExpr(E);
926 Writer.AddSourceRange(E->getSourceRange(), Record);
927 if (E->isTypeOperand()) {
928 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
929 Code = pch::EXPR_CXX_TYPEID_TYPE;
930 } else {
931 Writer.WriteSubStmt(E->getExprOperand());
932 Code = pch::EXPR_CXX_TYPEID_EXPR;
933 }
934}
935
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000936void PCHStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
937 VisitExpr(E);
938 Writer.AddSourceLocation(E->getLocation(), Record);
939 Record.push_back(E->isImplicit());
940 Code = pch::EXPR_CXX_THIS;
941}
942
943void PCHStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
Chris Lattner030854b2010-05-09 06:40:08 +0000944 VisitExpr(E);
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000945 Writer.AddSourceLocation(E->getThrowLoc(), Record);
946 Writer.WriteSubStmt(E->getSubExpr());
947 Code = pch::EXPR_CXX_THROW;
948}
949
Chris Lattner030854b2010-05-09 06:40:08 +0000950void PCHStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
951 VisitExpr(E);
952 Writer.AddSourceLocation(E->getUsedLocation(), Record);
953 if (E->isExprStored()) {
954 Record.push_back(1);
955 Writer.WriteSubStmt(E->getExpr());
956 } else {
957 Record.push_back(0);
958 }
959
960 Code = pch::EXPR_CXX_DEFAULT_ARG;
961}
962
Chris Lattnerd2598362010-05-10 00:25:06 +0000963void PCHStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
964 VisitExpr(E);
965 Writer.AddCXXTemporary(E->getTemporary(), Record);
966 Writer.WriteSubStmt(E->getSubExpr());
967 Code = pch::EXPR_CXX_BIND_TEMPORARY;
968}
969
Chris Lattner59218632010-05-10 01:22:27 +0000970void PCHStmtWriter::VisitCXXZeroInitValueExpr(CXXZeroInitValueExpr *E) {
971 VisitExpr(E);
972 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
973 Writer.AddSourceLocation(E->getRParenLoc(), Record);
974 Code = pch::EXPR_CXX_ZERO_INIT_VALUE;
975}
976
977void PCHStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
978 VisitExpr(E);
979 Record.push_back(E->isGlobalNew());
980 Record.push_back(E->isParenTypeId());
981 Record.push_back(E->hasInitializer());
982 Record.push_back(E->isArray());
983 Record.push_back(E->getNumPlacementArgs());
984 Record.push_back(E->getNumConstructorArgs());
985 Writer.AddDeclRef(E->getOperatorNew(), Record);
986 Writer.AddDeclRef(E->getOperatorDelete(), Record);
987 Writer.AddDeclRef(E->getConstructor(), Record);
988 Writer.AddSourceLocation(E->getStartLoc(), Record);
989 Writer.AddSourceLocation(E->getEndLoc(), Record);
990 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
991 I != e; ++I)
992 Writer.WriteSubStmt(*I);
993
994 Code = pch::EXPR_CXX_NEW;
995}
996
997
Chris Lattnerd2598362010-05-10 00:25:06 +0000998void PCHStmtWriter::VisitCXXExprWithTemporaries(CXXExprWithTemporaries *E) {
999 VisitExpr(E);
1000 Record.push_back(E->getNumTemporaries());
1001 for (unsigned i = 0, e = E->getNumTemporaries(); i != e; ++i)
1002 Writer.AddCXXTemporary(E->getTemporary(i), Record);
1003
1004 Writer.WriteSubStmt(E->getSubExpr());
1005 Code = pch::EXPR_CXX_EXPR_WITH_TEMPORARIES;
1006}
1007
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001008
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001009//===----------------------------------------------------------------------===//
Chris Lattner63377d52009-04-27 06:20:01 +00001010// PCHWriter Implementation
1011//===----------------------------------------------------------------------===//
1012
1013unsigned PCHWriter::RecordSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001014 assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001015 "SwitchCase recorded twice");
1016 unsigned NextID = SwitchCaseIDs.size();
1017 SwitchCaseIDs[S] = NextID;
1018 return NextID;
1019}
1020
1021unsigned PCHWriter::getSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001022 assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001023 "SwitchCase hasn't been seen yet");
1024 return SwitchCaseIDs[S];
1025}
1026
1027/// \brief Retrieve the ID for the given label statement, which may
1028/// or may not have been emitted yet.
1029unsigned PCHWriter::GetLabelID(LabelStmt *S) {
1030 std::map<LabelStmt *, unsigned>::iterator Pos = LabelIDs.find(S);
1031 if (Pos != LabelIDs.end())
1032 return Pos->second;
Mike Stump1eb44332009-09-09 15:08:12 +00001033
Chris Lattner63377d52009-04-27 06:20:01 +00001034 unsigned NextID = LabelIDs.size();
1035 LabelIDs[S] = NextID;
1036 return NextID;
1037}
1038
1039/// \brief Write the given substatement or subexpression to the
1040/// bitstream.
1041void PCHWriter::WriteSubStmt(Stmt *S) {
1042 RecordData Record;
1043 PCHStmtWriter Writer(*this, Record);
1044 ++NumStatements;
Mike Stump1eb44332009-09-09 15:08:12 +00001045
Chris Lattner63377d52009-04-27 06:20:01 +00001046 if (!S) {
1047 Stream.EmitRecord(pch::STMT_NULL_PTR, Record);
1048 return;
1049 }
Mike Stump1eb44332009-09-09 15:08:12 +00001050
Chris Lattner63377d52009-04-27 06:20:01 +00001051 Writer.Code = pch::STMT_NULL_PTR;
1052 Writer.Visit(S);
Chris Lattner59218632010-05-10 01:22:27 +00001053
1054#ifndef NDEBUG
1055 if (Writer.Code == pch::STMT_NULL_PTR) {
1056 S->dump();
1057 assert(0 && "Unhandled sub statement writing PCH file");
1058 }
1059#endif
Mike Stump1eb44332009-09-09 15:08:12 +00001060 Stream.EmitRecord(Writer.Code, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001061}
1062
1063/// \brief Flush all of the statements that have been added to the
1064/// queue via AddStmt().
1065void PCHWriter::FlushStmts() {
1066 RecordData Record;
1067 PCHStmtWriter Writer(*this, Record);
Mike Stump1eb44332009-09-09 15:08:12 +00001068
Chris Lattner63377d52009-04-27 06:20:01 +00001069 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
1070 ++NumStatements;
1071 Stmt *S = StmtsToEmit[I];
Mike Stump1eb44332009-09-09 15:08:12 +00001072
Chris Lattner63377d52009-04-27 06:20:01 +00001073 if (!S) {
1074 Stream.EmitRecord(pch::STMT_NULL_PTR, Record);
1075 continue;
1076 }
Mike Stump1eb44332009-09-09 15:08:12 +00001077
Chris Lattner63377d52009-04-27 06:20:01 +00001078 Writer.Code = pch::STMT_NULL_PTR;
1079 Writer.Visit(S);
Chris Lattnerd2598362010-05-10 00:25:06 +00001080#ifndef NDEBUG
1081 if (Writer.Code == pch::STMT_NULL_PTR) {
1082 S->dump();
1083 assert(0 && "Unhandled expression writing PCH file");
1084 }
1085#endif
Mike Stump1eb44332009-09-09 15:08:12 +00001086 Stream.EmitRecord(Writer.Code, Record);
1087
1088 assert(N == StmtsToEmit.size() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001089 "Substatement writen via AddStmt rather than WriteSubStmt!");
Mike Stump1eb44332009-09-09 15:08:12 +00001090
Chris Lattner63377d52009-04-27 06:20:01 +00001091 // Note that we are at the end of a full expression. Any
1092 // expression records that follow this one are part of a different
1093 // expression.
1094 Record.clear();
1095 Stream.EmitRecord(pch::STMT_STOP, Record);
1096 }
Mike Stump1eb44332009-09-09 15:08:12 +00001097
Chris Lattner63377d52009-04-27 06:20:01 +00001098 StmtsToEmit.clear();
Chris Lattner63377d52009-04-27 06:20:01 +00001099}