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