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