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