1 //===--- ASTWriterStmt.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/Serialization/ASTWriter.h"
15 #include "clang/AST/DeclCXX.h"
16 #include "clang/AST/DeclObjC.h"
17 #include "clang/AST/StmtVisitor.h"
18 #include "llvm/Bitcode/BitstreamWriter.h"
19 using namespace clang;
20 
21 //===----------------------------------------------------------------------===//
22 // Statement/expression serialization
23 //===----------------------------------------------------------------------===//
24 
25 namespace clang {
26   class ASTStmtWriter : public StmtVisitor<ASTStmtWriter, void> {
27     ASTWriter &Writer;
28     ASTWriter::RecordData &Record;
29 
30   public:
31     serialization::StmtCode Code;
32 
33     ASTStmtWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
34       : Writer(Writer), Record(Record) { }
35 
36     void
37     AddExplicitTemplateArgumentList(const ExplicitTemplateArgumentList &Args);
38 
39     void VisitStmt(Stmt *S);
40     void VisitNullStmt(NullStmt *S);
41     void VisitCompoundStmt(CompoundStmt *S);
42     void VisitSwitchCase(SwitchCase *S);
43     void VisitCaseStmt(CaseStmt *S);
44     void VisitDefaultStmt(DefaultStmt *S);
45     void VisitLabelStmt(LabelStmt *S);
46     void VisitIfStmt(IfStmt *S);
47     void VisitSwitchStmt(SwitchStmt *S);
48     void VisitWhileStmt(WhileStmt *S);
49     void VisitDoStmt(DoStmt *S);
50     void VisitForStmt(ForStmt *S);
51     void VisitGotoStmt(GotoStmt *S);
52     void VisitIndirectGotoStmt(IndirectGotoStmt *S);
53     void VisitContinueStmt(ContinueStmt *S);
54     void VisitBreakStmt(BreakStmt *S);
55     void VisitReturnStmt(ReturnStmt *S);
56     void VisitDeclStmt(DeclStmt *S);
57     void VisitAsmStmt(AsmStmt *S);
58     void VisitExpr(Expr *E);
59     void VisitPredefinedExpr(PredefinedExpr *E);
60     void VisitDeclRefExpr(DeclRefExpr *E);
61     void VisitIntegerLiteral(IntegerLiteral *E);
62     void VisitFloatingLiteral(FloatingLiteral *E);
63     void VisitImaginaryLiteral(ImaginaryLiteral *E);
64     void VisitStringLiteral(StringLiteral *E);
65     void VisitCharacterLiteral(CharacterLiteral *E);
66     void VisitParenExpr(ParenExpr *E);
67     void VisitParenListExpr(ParenListExpr *E);
68     void VisitUnaryOperator(UnaryOperator *E);
69     void VisitOffsetOfExpr(OffsetOfExpr *E);
70     void VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E);
71     void VisitArraySubscriptExpr(ArraySubscriptExpr *E);
72     void VisitCallExpr(CallExpr *E);
73     void VisitMemberExpr(MemberExpr *E);
74     void VisitCastExpr(CastExpr *E);
75     void VisitBinaryOperator(BinaryOperator *E);
76     void VisitCompoundAssignOperator(CompoundAssignOperator *E);
77     void VisitConditionalOperator(ConditionalOperator *E);
78     void VisitImplicitCastExpr(ImplicitCastExpr *E);
79     void VisitExplicitCastExpr(ExplicitCastExpr *E);
80     void VisitCStyleCastExpr(CStyleCastExpr *E);
81     void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
82     void VisitExtVectorElementExpr(ExtVectorElementExpr *E);
83     void VisitInitListExpr(InitListExpr *E);
84     void VisitDesignatedInitExpr(DesignatedInitExpr *E);
85     void VisitImplicitValueInitExpr(ImplicitValueInitExpr *E);
86     void VisitVAArgExpr(VAArgExpr *E);
87     void VisitAddrLabelExpr(AddrLabelExpr *E);
88     void VisitStmtExpr(StmtExpr *E);
89     void VisitChooseExpr(ChooseExpr *E);
90     void VisitGNUNullExpr(GNUNullExpr *E);
91     void VisitShuffleVectorExpr(ShuffleVectorExpr *E);
92     void VisitBlockExpr(BlockExpr *E);
93     void VisitBlockDeclRefExpr(BlockDeclRefExpr *E);
94 
95     // Objective-C Expressions
96     void VisitObjCStringLiteral(ObjCStringLiteral *E);
97     void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
98     void VisitObjCSelectorExpr(ObjCSelectorExpr *E);
99     void VisitObjCProtocolExpr(ObjCProtocolExpr *E);
100     void VisitObjCIvarRefExpr(ObjCIvarRefExpr *E);
101     void VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E);
102     void VisitObjCMessageExpr(ObjCMessageExpr *E);
103     void VisitObjCIsaExpr(ObjCIsaExpr *E);
104 
105     // Objective-C Statements
106     void VisitObjCForCollectionStmt(ObjCForCollectionStmt *);
107     void VisitObjCAtCatchStmt(ObjCAtCatchStmt *);
108     void VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *);
109     void VisitObjCAtTryStmt(ObjCAtTryStmt *);
110     void VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *);
111     void VisitObjCAtThrowStmt(ObjCAtThrowStmt *);
112 
113     // C++ Statements
114     void VisitCXXCatchStmt(CXXCatchStmt *S);
115     void VisitCXXTryStmt(CXXTryStmt *S);
116 
117     void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
118     void VisitCXXMemberCallExpr(CXXMemberCallExpr *E);
119     void VisitCXXConstructExpr(CXXConstructExpr *E);
120     void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
121     void VisitCXXNamedCastExpr(CXXNamedCastExpr *E);
122     void VisitCXXStaticCastExpr(CXXStaticCastExpr *E);
123     void VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E);
124     void VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E);
125     void VisitCXXConstCastExpr(CXXConstCastExpr *E);
126     void VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E);
127     void VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E);
128     void VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E);
129     void VisitCXXTypeidExpr(CXXTypeidExpr *E);
130     void VisitCXXUuidofExpr(CXXUuidofExpr *E);
131     void VisitCXXThisExpr(CXXThisExpr *E);
132     void VisitCXXThrowExpr(CXXThrowExpr *E);
133     void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E);
134     void VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E);
135 
136     void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
137     void VisitCXXNewExpr(CXXNewExpr *E);
138     void VisitCXXDeleteExpr(CXXDeleteExpr *E);
139     void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
140 
141     void VisitExprWithCleanups(ExprWithCleanups *E);
142     void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
143     void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
144     void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
145 
146     void VisitOverloadExpr(OverloadExpr *E);
147     void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E);
148     void VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E);
149 
150     void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
151     void VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E);
152     void VisitCXXNoexceptExpr(CXXNoexceptExpr *E);
153     void VisitPackExpansionExpr(PackExpansionExpr *E);
154     void VisitSizeOfPackExpr(SizeOfPackExpr *E);
155     void VisitOpaqueValueExpr(OpaqueValueExpr *E);
156   };
157 }
158 
159 void ASTStmtWriter::
160 AddExplicitTemplateArgumentList(const ExplicitTemplateArgumentList &Args) {
161   Writer.AddSourceLocation(Args.LAngleLoc, Record);
162   Writer.AddSourceLocation(Args.RAngleLoc, Record);
163   for (unsigned i=0; i != Args.NumTemplateArgs; ++i)
164     Writer.AddTemplateArgumentLoc(Args.getTemplateArgs()[i], Record);
165 }
166 
167 void ASTStmtWriter::VisitStmt(Stmt *S) {
168 }
169 
170 void ASTStmtWriter::VisitNullStmt(NullStmt *S) {
171   VisitStmt(S);
172   Writer.AddSourceLocation(S->getSemiLoc(), Record);
173   Record.push_back(S->LeadingEmptyMacro);
174   Code = serialization::STMT_NULL;
175 }
176 
177 void ASTStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
178   VisitStmt(S);
179   Record.push_back(S->size());
180   for (CompoundStmt::body_iterator CS = S->body_begin(), CSEnd = S->body_end();
181        CS != CSEnd; ++CS)
182     Writer.AddStmt(*CS);
183   Writer.AddSourceLocation(S->getLBracLoc(), Record);
184   Writer.AddSourceLocation(S->getRBracLoc(), Record);
185   Code = serialization::STMT_COMPOUND;
186 }
187 
188 void ASTStmtWriter::VisitSwitchCase(SwitchCase *S) {
189   VisitStmt(S);
190   Record.push_back(Writer.getSwitchCaseID(S));
191 }
192 
193 void ASTStmtWriter::VisitCaseStmt(CaseStmt *S) {
194   VisitSwitchCase(S);
195   Writer.AddStmt(S->getLHS());
196   Writer.AddStmt(S->getRHS());
197   Writer.AddStmt(S->getSubStmt());
198   Writer.AddSourceLocation(S->getCaseLoc(), Record);
199   Writer.AddSourceLocation(S->getEllipsisLoc(), Record);
200   Writer.AddSourceLocation(S->getColonLoc(), Record);
201   Code = serialization::STMT_CASE;
202 }
203 
204 void ASTStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
205   VisitSwitchCase(S);
206   Writer.AddStmt(S->getSubStmt());
207   Writer.AddSourceLocation(S->getDefaultLoc(), Record);
208   Writer.AddSourceLocation(S->getColonLoc(), Record);
209   Code = serialization::STMT_DEFAULT;
210 }
211 
212 void ASTStmtWriter::VisitLabelStmt(LabelStmt *S) {
213   VisitStmt(S);
214   Writer.AddIdentifierRef(S->getID(), Record);
215   Writer.AddStmt(S->getSubStmt());
216   Writer.AddSourceLocation(S->getIdentLoc(), Record);
217   Record.push_back(S->isUsed());
218   Record.push_back(S->HasUnusedAttribute());
219   Record.push_back(Writer.GetLabelID(S));
220   Code = serialization::STMT_LABEL;
221 }
222 
223 void ASTStmtWriter::VisitIfStmt(IfStmt *S) {
224   VisitStmt(S);
225   Writer.AddDeclRef(S->getConditionVariable(), Record);
226   Writer.AddStmt(S->getCond());
227   Writer.AddStmt(S->getThen());
228   Writer.AddStmt(S->getElse());
229   Writer.AddSourceLocation(S->getIfLoc(), Record);
230   Writer.AddSourceLocation(S->getElseLoc(), Record);
231   Code = serialization::STMT_IF;
232 }
233 
234 void ASTStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
235   VisitStmt(S);
236   Writer.AddDeclRef(S->getConditionVariable(), Record);
237   Writer.AddStmt(S->getCond());
238   Writer.AddStmt(S->getBody());
239   Writer.AddSourceLocation(S->getSwitchLoc(), Record);
240   Record.push_back(S->isAllEnumCasesCovered());
241   for (SwitchCase *SC = S->getSwitchCaseList(); SC;
242        SC = SC->getNextSwitchCase())
243     Record.push_back(Writer.RecordSwitchCaseID(SC));
244   Code = serialization::STMT_SWITCH;
245 }
246 
247 void ASTStmtWriter::VisitWhileStmt(WhileStmt *S) {
248   VisitStmt(S);
249   Writer.AddDeclRef(S->getConditionVariable(), Record);
250   Writer.AddStmt(S->getCond());
251   Writer.AddStmt(S->getBody());
252   Writer.AddSourceLocation(S->getWhileLoc(), Record);
253   Code = serialization::STMT_WHILE;
254 }
255 
256 void ASTStmtWriter::VisitDoStmt(DoStmt *S) {
257   VisitStmt(S);
258   Writer.AddStmt(S->getCond());
259   Writer.AddStmt(S->getBody());
260   Writer.AddSourceLocation(S->getDoLoc(), Record);
261   Writer.AddSourceLocation(S->getWhileLoc(), Record);
262   Writer.AddSourceLocation(S->getRParenLoc(), Record);
263   Code = serialization::STMT_DO;
264 }
265 
266 void ASTStmtWriter::VisitForStmt(ForStmt *S) {
267   VisitStmt(S);
268   Writer.AddStmt(S->getInit());
269   Writer.AddStmt(S->getCond());
270   Writer.AddDeclRef(S->getConditionVariable(), Record);
271   Writer.AddStmt(S->getInc());
272   Writer.AddStmt(S->getBody());
273   Writer.AddSourceLocation(S->getForLoc(), Record);
274   Writer.AddSourceLocation(S->getLParenLoc(), Record);
275   Writer.AddSourceLocation(S->getRParenLoc(), Record);
276   Code = serialization::STMT_FOR;
277 }
278 
279 void ASTStmtWriter::VisitGotoStmt(GotoStmt *S) {
280   VisitStmt(S);
281   Record.push_back(Writer.GetLabelID(S->getLabel()));
282   Writer.AddSourceLocation(S->getGotoLoc(), Record);
283   Writer.AddSourceLocation(S->getLabelLoc(), Record);
284   Code = serialization::STMT_GOTO;
285 }
286 
287 void ASTStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
288   VisitStmt(S);
289   Writer.AddSourceLocation(S->getGotoLoc(), Record);
290   Writer.AddSourceLocation(S->getStarLoc(), Record);
291   Writer.AddStmt(S->getTarget());
292   Code = serialization::STMT_INDIRECT_GOTO;
293 }
294 
295 void ASTStmtWriter::VisitContinueStmt(ContinueStmt *S) {
296   VisitStmt(S);
297   Writer.AddSourceLocation(S->getContinueLoc(), Record);
298   Code = serialization::STMT_CONTINUE;
299 }
300 
301 void ASTStmtWriter::VisitBreakStmt(BreakStmt *S) {
302   VisitStmt(S);
303   Writer.AddSourceLocation(S->getBreakLoc(), Record);
304   Code = serialization::STMT_BREAK;
305 }
306 
307 void ASTStmtWriter::VisitReturnStmt(ReturnStmt *S) {
308   VisitStmt(S);
309   Writer.AddStmt(S->getRetValue());
310   Writer.AddSourceLocation(S->getReturnLoc(), Record);
311   Writer.AddDeclRef(S->getNRVOCandidate(), Record);
312   Code = serialization::STMT_RETURN;
313 }
314 
315 void ASTStmtWriter::VisitDeclStmt(DeclStmt *S) {
316   VisitStmt(S);
317   Writer.AddSourceLocation(S->getStartLoc(), Record);
318   Writer.AddSourceLocation(S->getEndLoc(), Record);
319   DeclGroupRef DG = S->getDeclGroup();
320   for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
321     Writer.AddDeclRef(*D, Record);
322   Code = serialization::STMT_DECL;
323 }
324 
325 void ASTStmtWriter::VisitAsmStmt(AsmStmt *S) {
326   VisitStmt(S);
327   Record.push_back(S->getNumOutputs());
328   Record.push_back(S->getNumInputs());
329   Record.push_back(S->getNumClobbers());
330   Writer.AddSourceLocation(S->getAsmLoc(), Record);
331   Writer.AddSourceLocation(S->getRParenLoc(), Record);
332   Record.push_back(S->isVolatile());
333   Record.push_back(S->isSimple());
334   Record.push_back(S->isMSAsm());
335   Writer.AddStmt(S->getAsmString());
336 
337   // Outputs
338   for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
339     Writer.AddIdentifierRef(S->getOutputIdentifier(I), Record);
340     Writer.AddStmt(S->getOutputConstraintLiteral(I));
341     Writer.AddStmt(S->getOutputExpr(I));
342   }
343 
344   // Inputs
345   for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
346     Writer.AddIdentifierRef(S->getInputIdentifier(I), Record);
347     Writer.AddStmt(S->getInputConstraintLiteral(I));
348     Writer.AddStmt(S->getInputExpr(I));
349   }
350 
351   // Clobbers
352   for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
353     Writer.AddStmt(S->getClobber(I));
354 
355   Code = serialization::STMT_ASM;
356 }
357 
358 void ASTStmtWriter::VisitExpr(Expr *E) {
359   VisitStmt(E);
360   Writer.AddTypeRef(E->getType(), Record);
361   Record.push_back(E->isTypeDependent());
362   Record.push_back(E->isValueDependent());
363   Record.push_back(E->containsUnexpandedParameterPack());
364   Record.push_back(E->getValueKind());
365   Record.push_back(E->getObjectKind());
366 }
367 
368 void ASTStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
369   VisitExpr(E);
370   Writer.AddSourceLocation(E->getLocation(), Record);
371   Record.push_back(E->getIdentType()); // FIXME: stable encoding
372   Code = serialization::EXPR_PREDEFINED;
373 }
374 
375 void ASTStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
376   VisitExpr(E);
377 
378   Record.push_back(E->hasQualifier());
379   unsigned NumTemplateArgs = E->getNumTemplateArgs();
380   assert((NumTemplateArgs != 0) == E->hasExplicitTemplateArgs() &&
381          "Template args list with no args ?");
382   Record.push_back(NumTemplateArgs);
383 
384   if (E->hasQualifier()) {
385     Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
386     Writer.AddSourceRange(E->getQualifierRange(), Record);
387   }
388 
389   if (NumTemplateArgs)
390     AddExplicitTemplateArgumentList(E->getExplicitTemplateArgs());
391 
392   Writer.AddDeclRef(E->getDecl(), Record);
393   Writer.AddSourceLocation(E->getLocation(), Record);
394   Writer.AddDeclarationNameLoc(E->DNLoc, E->getDecl()->getDeclName(), Record);
395   Code = serialization::EXPR_DECL_REF;
396 }
397 
398 void ASTStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
399   VisitExpr(E);
400   Writer.AddSourceLocation(E->getLocation(), Record);
401   Writer.AddAPInt(E->getValue(), Record);
402   Code = serialization::EXPR_INTEGER_LITERAL;
403 }
404 
405 void ASTStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
406   VisitExpr(E);
407   Writer.AddAPFloat(E->getValue(), Record);
408   Record.push_back(E->isExact());
409   Writer.AddSourceLocation(E->getLocation(), Record);
410   Code = serialization::EXPR_FLOATING_LITERAL;
411 }
412 
413 void ASTStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
414   VisitExpr(E);
415   Writer.AddStmt(E->getSubExpr());
416   Code = serialization::EXPR_IMAGINARY_LITERAL;
417 }
418 
419 void ASTStmtWriter::VisitStringLiteral(StringLiteral *E) {
420   VisitExpr(E);
421   Record.push_back(E->getByteLength());
422   Record.push_back(E->getNumConcatenated());
423   Record.push_back(E->isWide());
424   // FIXME: String data should be stored as a blob at the end of the
425   // StringLiteral. However, we can't do so now because we have no
426   // provision for coping with abbreviations when we're jumping around
427   // the AST file during deserialization.
428   Record.append(E->getString().begin(), E->getString().end());
429   for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
430     Writer.AddSourceLocation(E->getStrTokenLoc(I), Record);
431   Code = serialization::EXPR_STRING_LITERAL;
432 }
433 
434 void ASTStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
435   VisitExpr(E);
436   Record.push_back(E->getValue());
437   Writer.AddSourceLocation(E->getLocation(), Record);
438   Record.push_back(E->isWide());
439   Code = serialization::EXPR_CHARACTER_LITERAL;
440 }
441 
442 void ASTStmtWriter::VisitParenExpr(ParenExpr *E) {
443   VisitExpr(E);
444   Writer.AddSourceLocation(E->getLParen(), Record);
445   Writer.AddSourceLocation(E->getRParen(), Record);
446   Writer.AddStmt(E->getSubExpr());
447   Code = serialization::EXPR_PAREN;
448 }
449 
450 void ASTStmtWriter::VisitParenListExpr(ParenListExpr *E) {
451   VisitExpr(E);
452   Record.push_back(E->NumExprs);
453   for (unsigned i=0; i != E->NumExprs; ++i)
454     Writer.AddStmt(E->Exprs[i]);
455   Writer.AddSourceLocation(E->LParenLoc, Record);
456   Writer.AddSourceLocation(E->RParenLoc, Record);
457   Code = serialization::EXPR_PAREN_LIST;
458 }
459 
460 void ASTStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
461   VisitExpr(E);
462   Writer.AddStmt(E->getSubExpr());
463   Record.push_back(E->getOpcode()); // FIXME: stable encoding
464   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
465   Code = serialization::EXPR_UNARY_OPERATOR;
466 }
467 
468 void ASTStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
469   VisitExpr(E);
470   Record.push_back(E->getNumComponents());
471   Record.push_back(E->getNumExpressions());
472   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
473   Writer.AddSourceLocation(E->getRParenLoc(), Record);
474   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
475   for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
476     const OffsetOfExpr::OffsetOfNode &ON = E->getComponent(I);
477     Record.push_back(ON.getKind()); // FIXME: Stable encoding
478     Writer.AddSourceLocation(ON.getRange().getBegin(), Record);
479     Writer.AddSourceLocation(ON.getRange().getEnd(), Record);
480     switch (ON.getKind()) {
481     case OffsetOfExpr::OffsetOfNode::Array:
482       Record.push_back(ON.getArrayExprIndex());
483       break;
484 
485     case OffsetOfExpr::OffsetOfNode::Field:
486       Writer.AddDeclRef(ON.getField(), Record);
487       break;
488 
489     case OffsetOfExpr::OffsetOfNode::Identifier:
490       Writer.AddIdentifierRef(ON.getFieldName(), Record);
491       break;
492 
493     case OffsetOfExpr::OffsetOfNode::Base:
494       Writer.AddCXXBaseSpecifier(*ON.getBase(), Record);
495       break;
496     }
497   }
498   for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
499     Writer.AddStmt(E->getIndexExpr(I));
500   Code = serialization::EXPR_OFFSETOF;
501 }
502 
503 void ASTStmtWriter::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) {
504   VisitExpr(E);
505   Record.push_back(E->isSizeOf());
506   if (E->isArgumentType())
507     Writer.AddTypeSourceInfo(E->getArgumentTypeInfo(), Record);
508   else {
509     Record.push_back(0);
510     Writer.AddStmt(E->getArgumentExpr());
511   }
512   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
513   Writer.AddSourceLocation(E->getRParenLoc(), Record);
514   Code = serialization::EXPR_SIZEOF_ALIGN_OF;
515 }
516 
517 void ASTStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
518   VisitExpr(E);
519   Writer.AddStmt(E->getLHS());
520   Writer.AddStmt(E->getRHS());
521   Writer.AddSourceLocation(E->getRBracketLoc(), Record);
522   Code = serialization::EXPR_ARRAY_SUBSCRIPT;
523 }
524 
525 void ASTStmtWriter::VisitCallExpr(CallExpr *E) {
526   VisitExpr(E);
527   Record.push_back(E->getNumArgs());
528   Writer.AddSourceLocation(E->getRParenLoc(), Record);
529   Writer.AddStmt(E->getCallee());
530   for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
531        Arg != ArgEnd; ++Arg)
532     Writer.AddStmt(*Arg);
533   Code = serialization::EXPR_CALL;
534 }
535 
536 void ASTStmtWriter::VisitMemberExpr(MemberExpr *E) {
537   // Don't call VisitExpr, we'll write everything here.
538 
539   Record.push_back(E->hasQualifier());
540   if (E->hasQualifier()) {
541     Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
542     Writer.AddSourceRange(E->getQualifierRange(), Record);
543   }
544 
545   unsigned NumTemplateArgs = E->getNumTemplateArgs();
546   assert((NumTemplateArgs != 0) == E->hasExplicitTemplateArgs() &&
547          "Template args list with no args ?");
548   Record.push_back(NumTemplateArgs);
549   if (NumTemplateArgs) {
550     Writer.AddSourceLocation(E->getLAngleLoc(), Record);
551     Writer.AddSourceLocation(E->getRAngleLoc(), Record);
552     for (unsigned i=0; i != NumTemplateArgs; ++i)
553       Writer.AddTemplateArgumentLoc(E->getTemplateArgs()[i], Record);
554   }
555 
556   DeclAccessPair FoundDecl = E->getFoundDecl();
557   Writer.AddDeclRef(FoundDecl.getDecl(), Record);
558   Record.push_back(FoundDecl.getAccess());
559 
560   Writer.AddTypeRef(E->getType(), Record);
561   Record.push_back(E->getValueKind());
562   Record.push_back(E->getObjectKind());
563   Writer.AddStmt(E->getBase());
564   Writer.AddDeclRef(E->getMemberDecl(), Record);
565   Writer.AddSourceLocation(E->getMemberLoc(), Record);
566   Record.push_back(E->isArrow());
567   Writer.AddDeclarationNameLoc(E->MemberDNLoc,
568                                E->getMemberDecl()->getDeclName(), Record);
569   Code = serialization::EXPR_MEMBER;
570 }
571 
572 void ASTStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
573   VisitExpr(E);
574   Writer.AddStmt(E->getBase());
575   Writer.AddSourceLocation(E->getIsaMemberLoc(), Record);
576   Record.push_back(E->isArrow());
577   Code = serialization::EXPR_OBJC_ISA;
578 }
579 
580 void ASTStmtWriter::VisitCastExpr(CastExpr *E) {
581   VisitExpr(E);
582   Record.push_back(E->path_size());
583   Writer.AddStmt(E->getSubExpr());
584   Record.push_back(E->getCastKind()); // FIXME: stable encoding
585 
586   for (CastExpr::path_iterator
587          PI = E->path_begin(), PE = E->path_end(); PI != PE; ++PI)
588     Writer.AddCXXBaseSpecifier(**PI, Record);
589 }
590 
591 void ASTStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
592   VisitExpr(E);
593   Writer.AddStmt(E->getLHS());
594   Writer.AddStmt(E->getRHS());
595   Record.push_back(E->getOpcode()); // FIXME: stable encoding
596   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
597   Code = serialization::EXPR_BINARY_OPERATOR;
598 }
599 
600 void ASTStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
601   VisitBinaryOperator(E);
602   Writer.AddTypeRef(E->getComputationLHSType(), Record);
603   Writer.AddTypeRef(E->getComputationResultType(), Record);
604   Code = serialization::EXPR_COMPOUND_ASSIGN_OPERATOR;
605 }
606 
607 void ASTStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
608   VisitExpr(E);
609   Writer.AddStmt(E->getCond());
610   Writer.AddStmt(E->getLHS());
611   Writer.AddStmt(E->getRHS());
612   Writer.AddStmt(E->getSAVE());
613   Writer.AddSourceLocation(E->getQuestionLoc(), Record);
614   Writer.AddSourceLocation(E->getColonLoc(), Record);
615   Code = serialization::EXPR_CONDITIONAL_OPERATOR;
616 }
617 
618 void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
619   VisitCastExpr(E);
620   Code = serialization::EXPR_IMPLICIT_CAST;
621 }
622 
623 void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
624   VisitCastExpr(E);
625   Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
626 }
627 
628 void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
629   VisitExplicitCastExpr(E);
630   Writer.AddSourceLocation(E->getLParenLoc(), Record);
631   Writer.AddSourceLocation(E->getRParenLoc(), Record);
632   Code = serialization::EXPR_CSTYLE_CAST;
633 }
634 
635 void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
636   VisitExpr(E);
637   Writer.AddSourceLocation(E->getLParenLoc(), Record);
638   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
639   Writer.AddStmt(E->getInitializer());
640   Record.push_back(E->isFileScope());
641   Code = serialization::EXPR_COMPOUND_LITERAL;
642 }
643 
644 void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
645   VisitExpr(E);
646   Writer.AddStmt(E->getBase());
647   Writer.AddIdentifierRef(&E->getAccessor(), Record);
648   Writer.AddSourceLocation(E->getAccessorLoc(), Record);
649   Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
650 }
651 
652 void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
653   VisitExpr(E);
654   Record.push_back(E->getNumInits());
655   for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
656     Writer.AddStmt(E->getInit(I));
657   Writer.AddStmt(E->getSyntacticForm());
658   Writer.AddSourceLocation(E->getLBraceLoc(), Record);
659   Writer.AddSourceLocation(E->getRBraceLoc(), Record);
660   Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
661   Record.push_back(E->hadArrayRangeDesignator());
662   Code = serialization::EXPR_INIT_LIST;
663 }
664 
665 void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
666   VisitExpr(E);
667   Record.push_back(E->getNumSubExprs());
668   for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
669     Writer.AddStmt(E->getSubExpr(I));
670   Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
671   Record.push_back(E->usesGNUSyntax());
672   for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
673                                              DEnd = E->designators_end();
674        D != DEnd; ++D) {
675     if (D->isFieldDesignator()) {
676       if (FieldDecl *Field = D->getField()) {
677         Record.push_back(serialization::DESIG_FIELD_DECL);
678         Writer.AddDeclRef(Field, Record);
679       } else {
680         Record.push_back(serialization::DESIG_FIELD_NAME);
681         Writer.AddIdentifierRef(D->getFieldName(), Record);
682       }
683       Writer.AddSourceLocation(D->getDotLoc(), Record);
684       Writer.AddSourceLocation(D->getFieldLoc(), Record);
685     } else if (D->isArrayDesignator()) {
686       Record.push_back(serialization::DESIG_ARRAY);
687       Record.push_back(D->getFirstExprIndex());
688       Writer.AddSourceLocation(D->getLBracketLoc(), Record);
689       Writer.AddSourceLocation(D->getRBracketLoc(), Record);
690     } else {
691       assert(D->isArrayRangeDesignator() && "Unknown designator");
692       Record.push_back(serialization::DESIG_ARRAY_RANGE);
693       Record.push_back(D->getFirstExprIndex());
694       Writer.AddSourceLocation(D->getLBracketLoc(), Record);
695       Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
696       Writer.AddSourceLocation(D->getRBracketLoc(), Record);
697     }
698   }
699   Code = serialization::EXPR_DESIGNATED_INIT;
700 }
701 
702 void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
703   VisitExpr(E);
704   Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
705 }
706 
707 void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
708   VisitExpr(E);
709   Writer.AddStmt(E->getSubExpr());
710   Writer.AddTypeSourceInfo(E->getWrittenTypeInfo(), Record);
711   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
712   Writer.AddSourceLocation(E->getRParenLoc(), Record);
713   Code = serialization::EXPR_VA_ARG;
714 }
715 
716 void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
717   VisitExpr(E);
718   Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
719   Writer.AddSourceLocation(E->getLabelLoc(), Record);
720   Record.push_back(Writer.GetLabelID(E->getLabel()));
721   Code = serialization::EXPR_ADDR_LABEL;
722 }
723 
724 void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
725   VisitExpr(E);
726   Writer.AddStmt(E->getSubStmt());
727   Writer.AddSourceLocation(E->getLParenLoc(), Record);
728   Writer.AddSourceLocation(E->getRParenLoc(), Record);
729   Code = serialization::EXPR_STMT;
730 }
731 
732 void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
733   VisitExpr(E);
734   Writer.AddStmt(E->getCond());
735   Writer.AddStmt(E->getLHS());
736   Writer.AddStmt(E->getRHS());
737   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
738   Writer.AddSourceLocation(E->getRParenLoc(), Record);
739   Code = serialization::EXPR_CHOOSE;
740 }
741 
742 void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
743   VisitExpr(E);
744   Writer.AddSourceLocation(E->getTokenLocation(), Record);
745   Code = serialization::EXPR_GNU_NULL;
746 }
747 
748 void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
749   VisitExpr(E);
750   Record.push_back(E->getNumSubExprs());
751   for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
752     Writer.AddStmt(E->getExpr(I));
753   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
754   Writer.AddSourceLocation(E->getRParenLoc(), Record);
755   Code = serialization::EXPR_SHUFFLE_VECTOR;
756 }
757 
758 void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
759   VisitExpr(E);
760   Writer.AddDeclRef(E->getBlockDecl(), Record);
761   Record.push_back(E->hasBlockDeclRefExprs());
762   Code = serialization::EXPR_BLOCK;
763 }
764 
765 void ASTStmtWriter::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
766   VisitExpr(E);
767   Writer.AddDeclRef(E->getDecl(), Record);
768   Writer.AddSourceLocation(E->getLocation(), Record);
769   Record.push_back(E->isByRef());
770   Record.push_back(E->isConstQualAdded());
771   Writer.AddStmt(E->getCopyConstructorExpr());
772   Code = serialization::EXPR_BLOCK_DECL_REF;
773 }
774 
775 //===----------------------------------------------------------------------===//
776 // Objective-C Expressions and Statements.
777 //===----------------------------------------------------------------------===//
778 
779 void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
780   VisitExpr(E);
781   Writer.AddStmt(E->getString());
782   Writer.AddSourceLocation(E->getAtLoc(), Record);
783   Code = serialization::EXPR_OBJC_STRING_LITERAL;
784 }
785 
786 void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
787   VisitExpr(E);
788   Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
789   Writer.AddSourceLocation(E->getAtLoc(), Record);
790   Writer.AddSourceLocation(E->getRParenLoc(), Record);
791   Code = serialization::EXPR_OBJC_ENCODE;
792 }
793 
794 void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
795   VisitExpr(E);
796   Writer.AddSelectorRef(E->getSelector(), Record);
797   Writer.AddSourceLocation(E->getAtLoc(), Record);
798   Writer.AddSourceLocation(E->getRParenLoc(), Record);
799   Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
800 }
801 
802 void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
803   VisitExpr(E);
804   Writer.AddDeclRef(E->getProtocol(), Record);
805   Writer.AddSourceLocation(E->getAtLoc(), Record);
806   Writer.AddSourceLocation(E->getRParenLoc(), Record);
807   Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
808 }
809 
810 void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
811   VisitExpr(E);
812   Writer.AddDeclRef(E->getDecl(), Record);
813   Writer.AddSourceLocation(E->getLocation(), Record);
814   Writer.AddStmt(E->getBase());
815   Record.push_back(E->isArrow());
816   Record.push_back(E->isFreeIvar());
817   Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
818 }
819 
820 void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
821   VisitExpr(E);
822   Record.push_back(E->isImplicitProperty());
823   if (E->isImplicitProperty()) {
824     Writer.AddDeclRef(E->getImplicitPropertyGetter(), Record);
825     Writer.AddDeclRef(E->getImplicitPropertySetter(), Record);
826   } else {
827     Writer.AddDeclRef(E->getExplicitProperty(), Record);
828   }
829   Writer.AddSourceLocation(E->getLocation(), Record);
830   Writer.AddSourceLocation(E->getReceiverLocation(), Record);
831   if (E->isObjectReceiver()) {
832     Record.push_back(0);
833     Writer.AddStmt(E->getBase());
834   } else if (E->isSuperReceiver()) {
835     Record.push_back(1);
836     Writer.AddTypeRef(E->getSuperReceiverType(), Record);
837   } else {
838     Record.push_back(2);
839     Writer.AddDeclRef(E->getClassReceiver(), Record);
840   }
841 
842   Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
843 }
844 
845 void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
846   VisitExpr(E);
847   Record.push_back(E->getNumArgs());
848   Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
849   switch (E->getReceiverKind()) {
850   case ObjCMessageExpr::Instance:
851     Writer.AddStmt(E->getInstanceReceiver());
852     break;
853 
854   case ObjCMessageExpr::Class:
855     Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
856     break;
857 
858   case ObjCMessageExpr::SuperClass:
859   case ObjCMessageExpr::SuperInstance:
860     Writer.AddTypeRef(E->getSuperType(), Record);
861     Writer.AddSourceLocation(E->getSuperLoc(), Record);
862     break;
863   }
864 
865   if (E->getMethodDecl()) {
866     Record.push_back(1);
867     Writer.AddDeclRef(E->getMethodDecl(), Record);
868   } else {
869     Record.push_back(0);
870     Writer.AddSelectorRef(E->getSelector(), Record);
871   }
872 
873   Writer.AddSourceLocation(E->getLeftLoc(), Record);
874   Writer.AddSourceLocation(E->getRightLoc(), Record);
875   Writer.AddSourceLocation(E->getSelectorLoc(), Record);
876 
877   for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
878        Arg != ArgEnd; ++Arg)
879     Writer.AddStmt(*Arg);
880   Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
881 }
882 
883 void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
884   VisitStmt(S);
885   Writer.AddStmt(S->getElement());
886   Writer.AddStmt(S->getCollection());
887   Writer.AddStmt(S->getBody());
888   Writer.AddSourceLocation(S->getForLoc(), Record);
889   Writer.AddSourceLocation(S->getRParenLoc(), Record);
890   Code = serialization::STMT_OBJC_FOR_COLLECTION;
891 }
892 
893 void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
894   Writer.AddStmt(S->getCatchBody());
895   Writer.AddDeclRef(S->getCatchParamDecl(), Record);
896   Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
897   Writer.AddSourceLocation(S->getRParenLoc(), Record);
898   Code = serialization::STMT_OBJC_CATCH;
899 }
900 
901 void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
902   Writer.AddStmt(S->getFinallyBody());
903   Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
904   Code = serialization::STMT_OBJC_FINALLY;
905 }
906 
907 void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
908   Record.push_back(S->getNumCatchStmts());
909   Record.push_back(S->getFinallyStmt() != 0);
910   Writer.AddStmt(S->getTryBody());
911   for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
912     Writer.AddStmt(S->getCatchStmt(I));
913   if (S->getFinallyStmt())
914     Writer.AddStmt(S->getFinallyStmt());
915   Writer.AddSourceLocation(S->getAtTryLoc(), Record);
916   Code = serialization::STMT_OBJC_AT_TRY;
917 }
918 
919 void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
920   Writer.AddStmt(S->getSynchExpr());
921   Writer.AddStmt(S->getSynchBody());
922   Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
923   Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
924 }
925 
926 void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
927   Writer.AddStmt(S->getThrowExpr());
928   Writer.AddSourceLocation(S->getThrowLoc(), Record);
929   Code = serialization::STMT_OBJC_AT_THROW;
930 }
931 
932 //===----------------------------------------------------------------------===//
933 // C++ Expressions and Statements.
934 //===----------------------------------------------------------------------===//
935 
936 void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
937   VisitStmt(S);
938   Writer.AddSourceLocation(S->getCatchLoc(), Record);
939   Writer.AddDeclRef(S->getExceptionDecl(), Record);
940   Writer.AddStmt(S->getHandlerBlock());
941   Code = serialization::STMT_CXX_CATCH;
942 }
943 
944 void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
945   VisitStmt(S);
946   Record.push_back(S->getNumHandlers());
947   Writer.AddSourceLocation(S->getTryLoc(), Record);
948   Writer.AddStmt(S->getTryBlock());
949   for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
950     Writer.AddStmt(S->getHandler(i));
951   Code = serialization::STMT_CXX_TRY;
952 }
953 
954 void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
955   VisitCallExpr(E);
956   Record.push_back(E->getOperator());
957   Code = serialization::EXPR_CXX_OPERATOR_CALL;
958 }
959 
960 void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
961   VisitCallExpr(E);
962   Code = serialization::EXPR_CXX_MEMBER_CALL;
963 }
964 
965 void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
966   VisitExpr(E);
967   Record.push_back(E->getNumArgs());
968   for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
969     Writer.AddStmt(E->getArg(I));
970   Writer.AddDeclRef(E->getConstructor(), Record);
971   Writer.AddSourceLocation(E->getLocation(), Record);
972   Record.push_back(E->isElidable());
973   Record.push_back(E->requiresZeroInitialization());
974   Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
975   Writer.AddSourceRange(E->getParenRange(), Record);
976   Code = serialization::EXPR_CXX_CONSTRUCT;
977 }
978 
979 void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
980   VisitCXXConstructExpr(E);
981   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
982   Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
983 }
984 
985 void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
986   VisitExplicitCastExpr(E);
987   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
988 }
989 
990 void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
991   VisitCXXNamedCastExpr(E);
992   Code = serialization::EXPR_CXX_STATIC_CAST;
993 }
994 
995 void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
996   VisitCXXNamedCastExpr(E);
997   Code = serialization::EXPR_CXX_DYNAMIC_CAST;
998 }
999 
1000 void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
1001   VisitCXXNamedCastExpr(E);
1002   Code = serialization::EXPR_CXX_REINTERPRET_CAST;
1003 }
1004 
1005 void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
1006   VisitCXXNamedCastExpr(E);
1007   Code = serialization::EXPR_CXX_CONST_CAST;
1008 }
1009 
1010 void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
1011   VisitExplicitCastExpr(E);
1012   Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1013   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1014   Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
1015 }
1016 
1017 void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
1018   VisitExpr(E);
1019   Record.push_back(E->getValue());
1020   Writer.AddSourceLocation(E->getLocation(), Record);
1021   Code = serialization::EXPR_CXX_BOOL_LITERAL;
1022 }
1023 
1024 void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
1025   VisitExpr(E);
1026   Writer.AddSourceLocation(E->getLocation(), Record);
1027   Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
1028 }
1029 
1030 void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1031   VisitExpr(E);
1032   Writer.AddSourceRange(E->getSourceRange(), Record);
1033   if (E->isTypeOperand()) {
1034     Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1035     Code = serialization::EXPR_CXX_TYPEID_TYPE;
1036   } else {
1037     Writer.AddStmt(E->getExprOperand());
1038     Code = serialization::EXPR_CXX_TYPEID_EXPR;
1039   }
1040 }
1041 
1042 void ASTStmtWriter::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
1043   VisitExpr(E);
1044   Writer.AddSourceRange(E->getSourceRange(), Record);
1045   if (E->isTypeOperand()) {
1046     Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1047     Code = serialization::EXPR_CXX_UUIDOF_TYPE;
1048   } else {
1049     Writer.AddStmt(E->getExprOperand());
1050     Code = serialization::EXPR_CXX_UUIDOF_EXPR;
1051   }
1052 }
1053 
1054 void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
1055   VisitExpr(E);
1056   Writer.AddSourceLocation(E->getLocation(), Record);
1057   Record.push_back(E->isImplicit());
1058   Code = serialization::EXPR_CXX_THIS;
1059 }
1060 
1061 void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
1062   VisitExpr(E);
1063   Writer.AddSourceLocation(E->getThrowLoc(), Record);
1064   Writer.AddStmt(E->getSubExpr());
1065   Code = serialization::EXPR_CXX_THROW;
1066 }
1067 
1068 void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
1069   VisitExpr(E);
1070 
1071   bool HasOtherExprStored = E->Param.getInt();
1072   // Store these first, the reader reads them before creation.
1073   Record.push_back(HasOtherExprStored);
1074   if (HasOtherExprStored)
1075     Writer.AddStmt(E->getExpr());
1076   Writer.AddDeclRef(E->getParam(), Record);
1077   Writer.AddSourceLocation(E->getUsedLocation(), Record);
1078 
1079   Code = serialization::EXPR_CXX_DEFAULT_ARG;
1080 }
1081 
1082 void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
1083   VisitExpr(E);
1084   Writer.AddCXXTemporary(E->getTemporary(), Record);
1085   Writer.AddStmt(E->getSubExpr());
1086   Code = serialization::EXPR_CXX_BIND_TEMPORARY;
1087 }
1088 
1089 void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
1090   VisitExpr(E);
1091   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
1092   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1093   Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
1094 }
1095 
1096 void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
1097   VisitExpr(E);
1098   Record.push_back(E->isGlobalNew());
1099   Record.push_back(E->hasInitializer());
1100   Record.push_back(E->isArray());
1101   Record.push_back(E->getNumPlacementArgs());
1102   Record.push_back(E->getNumConstructorArgs());
1103   Writer.AddDeclRef(E->getOperatorNew(), Record);
1104   Writer.AddDeclRef(E->getOperatorDelete(), Record);
1105   Writer.AddDeclRef(E->getConstructor(), Record);
1106   Writer.AddTypeSourceInfo(E->getAllocatedTypeSourceInfo(), Record);
1107   Writer.AddSourceRange(E->getTypeIdParens(), Record);
1108   Writer.AddSourceLocation(E->getStartLoc(), Record);
1109   Writer.AddSourceLocation(E->getEndLoc(), Record);
1110   Writer.AddSourceLocation(E->getConstructorLParen(), Record);
1111   Writer.AddSourceLocation(E->getConstructorRParen(), Record);
1112   for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
1113        I != e; ++I)
1114     Writer.AddStmt(*I);
1115 
1116   Code = serialization::EXPR_CXX_NEW;
1117 }
1118 
1119 void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
1120   VisitExpr(E);
1121   Record.push_back(E->isGlobalDelete());
1122   Record.push_back(E->isArrayForm());
1123   Record.push_back(E->isArrayFormAsWritten());
1124   Writer.AddDeclRef(E->getOperatorDelete(), Record);
1125   Writer.AddStmt(E->getArgument());
1126   Writer.AddSourceLocation(E->getSourceRange().getBegin(), Record);
1127 
1128   Code = serialization::EXPR_CXX_DELETE;
1129 }
1130 
1131 void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1132   VisitExpr(E);
1133 
1134   Writer.AddStmt(E->getBase());
1135   Record.push_back(E->isArrow());
1136   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1137   Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1138   Writer.AddSourceRange(E->getQualifierRange(), Record);
1139   Writer.AddTypeSourceInfo(E->getScopeTypeInfo(), Record);
1140   Writer.AddSourceLocation(E->getColonColonLoc(), Record);
1141   Writer.AddSourceLocation(E->getTildeLoc(), Record);
1142 
1143   // PseudoDestructorTypeStorage.
1144   Writer.AddIdentifierRef(E->getDestroyedTypeIdentifier(), Record);
1145   if (E->getDestroyedTypeIdentifier())
1146     Writer.AddSourceLocation(E->getDestroyedTypeLoc(), Record);
1147   else
1148     Writer.AddTypeSourceInfo(E->getDestroyedTypeInfo(), Record);
1149 
1150   Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
1151 }
1152 
1153 void ASTStmtWriter::VisitExprWithCleanups(ExprWithCleanups *E) {
1154   VisitExpr(E);
1155   Record.push_back(E->getNumTemporaries());
1156   for (unsigned i = 0, e = E->getNumTemporaries(); i != e; ++i)
1157     Writer.AddCXXTemporary(E->getTemporary(i), Record);
1158 
1159   Writer.AddStmt(E->getSubExpr());
1160   Code = serialization::EXPR_EXPR_WITH_CLEANUPS;
1161 }
1162 
1163 void
1164 ASTStmtWriter::VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E){
1165   VisitExpr(E);
1166 
1167   // Don't emit anything here, NumTemplateArgs must be emitted first.
1168 
1169   if (E->hasExplicitTemplateArgs()) {
1170     const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
1171     assert(Args.NumTemplateArgs &&
1172            "Num of template args was zero! AST reading will mess up!");
1173     Record.push_back(Args.NumTemplateArgs);
1174     AddExplicitTemplateArgumentList(Args);
1175   } else {
1176     Record.push_back(0);
1177   }
1178 
1179   if (!E->isImplicitAccess())
1180     Writer.AddStmt(E->getBase());
1181   else
1182     Writer.AddStmt(0);
1183   Writer.AddTypeRef(E->getBaseType(), Record);
1184   Record.push_back(E->isArrow());
1185   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1186   Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1187   Writer.AddSourceRange(E->getQualifierRange(), Record);
1188   Writer.AddDeclRef(E->getFirstQualifierFoundInScope(), Record);
1189   Writer.AddDeclarationNameInfo(E->MemberNameInfo, Record);
1190   Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
1191 }
1192 
1193 void
1194 ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1195   VisitExpr(E);
1196 
1197   // Don't emit anything here, NumTemplateArgs must be emitted first.
1198 
1199   if (E->hasExplicitTemplateArgs()) {
1200     const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
1201     assert(Args.NumTemplateArgs &&
1202            "Num of template args was zero! AST reading will mess up!");
1203     Record.push_back(Args.NumTemplateArgs);
1204     AddExplicitTemplateArgumentList(Args);
1205   } else {
1206     Record.push_back(0);
1207   }
1208 
1209   Writer.AddDeclarationNameInfo(E->NameInfo, Record);
1210   Writer.AddSourceRange(E->getQualifierRange(), Record);
1211   Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1212   Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
1213 }
1214 
1215 void
1216 ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
1217   VisitExpr(E);
1218   Record.push_back(E->arg_size());
1219   for (CXXUnresolvedConstructExpr::arg_iterator
1220          ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
1221     Writer.AddStmt(*ArgI);
1222   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
1223   Writer.AddSourceLocation(E->getLParenLoc(), Record);
1224   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1225   Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
1226 }
1227 
1228 void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
1229   VisitExpr(E);
1230 
1231   // Don't emit anything here, NumTemplateArgs must be emitted first.
1232 
1233   if (E->hasExplicitTemplateArgs()) {
1234     const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
1235     assert(Args.NumTemplateArgs &&
1236            "Num of template args was zero! AST reading will mess up!");
1237     Record.push_back(Args.NumTemplateArgs);
1238     AddExplicitTemplateArgumentList(Args);
1239   } else {
1240     Record.push_back(0);
1241   }
1242 
1243   Record.push_back(E->getNumDecls());
1244   for (OverloadExpr::decls_iterator
1245          OvI = E->decls_begin(), OvE = E->decls_end(); OvI != OvE; ++OvI) {
1246     Writer.AddDeclRef(OvI.getDecl(), Record);
1247     Record.push_back(OvI.getAccess());
1248   }
1249 
1250   Writer.AddDeclarationNameInfo(E->NameInfo, Record);
1251   Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1252   Writer.AddSourceRange(E->getQualifierRange(), Record);
1253 }
1254 
1255 void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
1256   VisitOverloadExpr(E);
1257   Record.push_back(E->isArrow());
1258   Record.push_back(E->hasUnresolvedUsing());
1259   Writer.AddStmt(!E->isImplicitAccess() ? E->getBase() : 0);
1260   Writer.AddTypeRef(E->getBaseType(), Record);
1261   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1262   Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
1263 }
1264 
1265 void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
1266   VisitOverloadExpr(E);
1267   Record.push_back(E->requiresADL());
1268   Record.push_back(E->isOverloaded());
1269   Writer.AddDeclRef(E->getNamingClass(), Record);
1270   Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
1271 }
1272 
1273 void ASTStmtWriter::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
1274   VisitExpr(E);
1275   Record.push_back(E->getTrait());
1276   Record.push_back(E->getValue());
1277   Writer.AddSourceRange(E->getSourceRange(), Record);
1278   Writer.AddTypeSourceInfo(E->getQueriedTypeSourceInfo(), Record);
1279   Code = serialization::EXPR_CXX_UNARY_TYPE_TRAIT;
1280 }
1281 
1282 void ASTStmtWriter::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
1283   VisitExpr(E);
1284   Record.push_back(E->getTrait());
1285   Record.push_back(E->getValue());
1286   Writer.AddSourceRange(E->getSourceRange(), Record);
1287   Writer.AddTypeSourceInfo(E->getLhsTypeSourceInfo(), Record);
1288   Writer.AddTypeSourceInfo(E->getRhsTypeSourceInfo(), Record);
1289   Code = serialization::EXPR_BINARY_TYPE_TRAIT;
1290 }
1291 
1292 void ASTStmtWriter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
1293   VisitExpr(E);
1294   Record.push_back(E->getValue());
1295   Writer.AddSourceRange(E->getSourceRange(), Record);
1296   Writer.AddStmt(E->getOperand());
1297   Code = serialization::EXPR_CXX_NOEXCEPT;
1298 }
1299 
1300 void ASTStmtWriter::VisitPackExpansionExpr(PackExpansionExpr *E) {
1301   VisitExpr(E);
1302   Writer.AddSourceLocation(E->getEllipsisLoc(), Record);
1303   Writer.AddStmt(E->getPattern());
1304   Code = serialization::EXPR_PACK_EXPANSION;
1305 }
1306 
1307 void ASTStmtWriter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
1308   VisitExpr(E);
1309   Writer.AddSourceLocation(E->OperatorLoc, Record);
1310   Writer.AddSourceLocation(E->PackLoc, Record);
1311   Writer.AddSourceLocation(E->RParenLoc, Record);
1312   Record.push_back(E->Length);
1313   Writer.AddDeclRef(E->Pack, Record);
1314   Code = serialization::EXPR_SIZEOF_PACK;
1315 }
1316 
1317 void ASTStmtWriter::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
1318   VisitExpr(E);
1319   Code = serialization::EXPR_OPAQUE_VALUE;
1320 }
1321 
1322 //===----------------------------------------------------------------------===//
1323 // ASTWriter Implementation
1324 //===----------------------------------------------------------------------===//
1325 
1326 unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
1327   assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
1328          "SwitchCase recorded twice");
1329   unsigned NextID = SwitchCaseIDs.size();
1330   SwitchCaseIDs[S] = NextID;
1331   return NextID;
1332 }
1333 
1334 unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
1335   assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
1336          "SwitchCase hasn't been seen yet");
1337   return SwitchCaseIDs[S];
1338 }
1339 
1340 void ASTWriter::ClearSwitchCaseIDs() {
1341   SwitchCaseIDs.clear();
1342 }
1343 
1344 /// \brief Retrieve the ID for the given label statement, which may
1345 /// or may not have been emitted yet.
1346 unsigned ASTWriter::GetLabelID(LabelStmt *S) {
1347   std::map<LabelStmt *, unsigned>::iterator Pos = LabelIDs.find(S);
1348   if (Pos != LabelIDs.end())
1349     return Pos->second;
1350 
1351   unsigned NextID = LabelIDs.size();
1352   LabelIDs[S] = NextID;
1353   return NextID;
1354 }
1355 
1356 /// \brief Write the given substatement or subexpression to the
1357 /// bitstream.
1358 void ASTWriter::WriteSubStmt(Stmt *S) {
1359   RecordData Record;
1360   ASTStmtWriter Writer(*this, Record);
1361   ++NumStatements;
1362 
1363   if (!S) {
1364     Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
1365     return;
1366   }
1367 
1368   // Redirect ASTWriter::AddStmt to collect sub stmts.
1369   llvm::SmallVector<Stmt *, 16> SubStmts;
1370   CollectedStmts = &SubStmts;
1371 
1372   Writer.Code = serialization::STMT_NULL_PTR;
1373   Writer.Visit(S);
1374 
1375 #ifndef NDEBUG
1376   if (Writer.Code == serialization::STMT_NULL_PTR) {
1377     SourceManager &SrcMgr
1378       = DeclIDs.begin()->first->getASTContext().getSourceManager();
1379     S->dump(SrcMgr);
1380     assert(0 && "Unhandled sub statement writing AST file");
1381   }
1382 #endif
1383 
1384   // Revert ASTWriter::AddStmt.
1385   CollectedStmts = &StmtsToEmit;
1386 
1387   // Write the sub stmts in reverse order, last to first. When reading them back
1388   // we will read them in correct order by "pop"ing them from the Stmts stack.
1389   // This simplifies reading and allows to store a variable number of sub stmts
1390   // without knowing it in advance.
1391   while (!SubStmts.empty())
1392     WriteSubStmt(SubStmts.pop_back_val());
1393 
1394   Stream.EmitRecord(Writer.Code, Record);
1395 }
1396 
1397 /// \brief Flush all of the statements that have been added to the
1398 /// queue via AddStmt().
1399 void ASTWriter::FlushStmts() {
1400   RecordData Record;
1401 
1402   for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
1403     WriteSubStmt(StmtsToEmit[I]);
1404 
1405     assert(N == StmtsToEmit.size() &&
1406            "Substatement writen via AddStmt rather than WriteSubStmt!");
1407 
1408     // Note that we are at the end of a full expression. Any
1409     // expression records that follow this one are part of a different
1410     // expression.
1411     Stream.EmitRecord(serialization::STMT_STOP, Record);
1412   }
1413 
1414   StmtsToEmit.clear();
1415 }
1416