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
39     AddExplicitTemplateArgumentList(const ASTTemplateArgumentListInfo &Args);
40 
41     void VisitStmt(Stmt *S);
42 #define STMT(Type, Base) \
43     void Visit##Type(Type *);
44 #include "clang/AST/StmtNodes.inc"
45   };
46 }
47 
48 void ASTStmtWriter::
49 AddExplicitTemplateArgumentList(const ASTTemplateArgumentListInfo &Args) {
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->hasExplicitTemplateArgs());
268 
269   if (E->hasExplicitTemplateArgs()) {
270     unsigned NumTemplateArgs = E->getNumTemplateArgs();
271     Record.push_back(NumTemplateArgs);
272   }
273 
274   DeclarationName::NameKind nk = (E->getDecl()->getDeclName().getNameKind());
275 
276   if ((!E->hasExplicitTemplateArgs()) && (!E->hasQualifier()) &&
277       (E->getDecl() == E->getFoundDecl()) &&
278       nk == DeclarationName::Identifier) {
279     AbbrevToUse = Writer.getDeclRefExprAbbrev();
280   }
281 
282   if (E->hasQualifier())
283     Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
284 
285   if (E->getDecl() != E->getFoundDecl())
286     Writer.AddDeclRef(E->getFoundDecl(), Record);
287 
288   if (E->hasExplicitTemplateArgs())
289     AddExplicitTemplateArgumentList(E->getExplicitTemplateArgs());
290 
291   Writer.AddDeclRef(E->getDecl(), Record);
292   Writer.AddSourceLocation(E->getLocation(), Record);
293   Writer.AddDeclarationNameLoc(E->DNLoc, E->getDecl()->getDeclName(), Record);
294   Code = serialization::EXPR_DECL_REF;
295 }
296 
297 void ASTStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
298   VisitExpr(E);
299   Writer.AddSourceLocation(E->getLocation(), Record);
300   Writer.AddAPInt(E->getValue(), Record);
301 
302   if (E->getValue().getBitWidth() == 32) {
303     AbbrevToUse = Writer.getIntegerLiteralAbbrev();
304   }
305 
306   Code = serialization::EXPR_INTEGER_LITERAL;
307 }
308 
309 void ASTStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
310   VisitExpr(E);
311   Writer.AddAPFloat(E->getValue(), Record);
312   Record.push_back(E->isExact());
313   Writer.AddSourceLocation(E->getLocation(), Record);
314   Code = serialization::EXPR_FLOATING_LITERAL;
315 }
316 
317 void ASTStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
318   VisitExpr(E);
319   Writer.AddStmt(E->getSubExpr());
320   Code = serialization::EXPR_IMAGINARY_LITERAL;
321 }
322 
323 void ASTStmtWriter::VisitStringLiteral(StringLiteral *E) {
324   VisitExpr(E);
325   Record.push_back(E->getByteLength());
326   Record.push_back(E->getNumConcatenated());
327   Record.push_back(E->getKind());
328   Record.push_back(E->isPascal());
329   // FIXME: String data should be stored as a blob at the end of the
330   // StringLiteral. However, we can't do so now because we have no
331   // provision for coping with abbreviations when we're jumping around
332   // the AST file during deserialization.
333   Record.append(E->getString().begin(), E->getString().end());
334   for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
335     Writer.AddSourceLocation(E->getStrTokenLoc(I), Record);
336   Code = serialization::EXPR_STRING_LITERAL;
337 }
338 
339 void ASTStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
340   VisitExpr(E);
341   Record.push_back(E->getValue());
342   Writer.AddSourceLocation(E->getLocation(), Record);
343   Record.push_back(E->getKind());
344 
345   AbbrevToUse = Writer.getCharacterLiteralAbbrev();
346 
347   Code = serialization::EXPR_CHARACTER_LITERAL;
348 }
349 
350 void ASTStmtWriter::VisitParenExpr(ParenExpr *E) {
351   VisitExpr(E);
352   Writer.AddSourceLocation(E->getLParen(), Record);
353   Writer.AddSourceLocation(E->getRParen(), Record);
354   Writer.AddStmt(E->getSubExpr());
355   Code = serialization::EXPR_PAREN;
356 }
357 
358 void ASTStmtWriter::VisitParenListExpr(ParenListExpr *E) {
359   VisitExpr(E);
360   Record.push_back(E->NumExprs);
361   for (unsigned i=0; i != E->NumExprs; ++i)
362     Writer.AddStmt(E->Exprs[i]);
363   Writer.AddSourceLocation(E->LParenLoc, Record);
364   Writer.AddSourceLocation(E->RParenLoc, Record);
365   Code = serialization::EXPR_PAREN_LIST;
366 }
367 
368 void ASTStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
369   VisitExpr(E);
370   Writer.AddStmt(E->getSubExpr());
371   Record.push_back(E->getOpcode()); // FIXME: stable encoding
372   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
373   Code = serialization::EXPR_UNARY_OPERATOR;
374 }
375 
376 void ASTStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
377   VisitExpr(E);
378   Record.push_back(E->getNumComponents());
379   Record.push_back(E->getNumExpressions());
380   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
381   Writer.AddSourceLocation(E->getRParenLoc(), Record);
382   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
383   for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
384     const OffsetOfExpr::OffsetOfNode &ON = E->getComponent(I);
385     Record.push_back(ON.getKind()); // FIXME: Stable encoding
386     Writer.AddSourceLocation(ON.getSourceRange().getBegin(), Record);
387     Writer.AddSourceLocation(ON.getSourceRange().getEnd(), Record);
388     switch (ON.getKind()) {
389     case OffsetOfExpr::OffsetOfNode::Array:
390       Record.push_back(ON.getArrayExprIndex());
391       break;
392 
393     case OffsetOfExpr::OffsetOfNode::Field:
394       Writer.AddDeclRef(ON.getField(), Record);
395       break;
396 
397     case OffsetOfExpr::OffsetOfNode::Identifier:
398       Writer.AddIdentifierRef(ON.getFieldName(), Record);
399       break;
400 
401     case OffsetOfExpr::OffsetOfNode::Base:
402       Writer.AddCXXBaseSpecifier(*ON.getBase(), Record);
403       break;
404     }
405   }
406   for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
407     Writer.AddStmt(E->getIndexExpr(I));
408   Code = serialization::EXPR_OFFSETOF;
409 }
410 
411 void ASTStmtWriter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E) {
412   VisitExpr(E);
413   Record.push_back(E->getKind());
414   if (E->isArgumentType())
415     Writer.AddTypeSourceInfo(E->getArgumentTypeInfo(), Record);
416   else {
417     Record.push_back(0);
418     Writer.AddStmt(E->getArgumentExpr());
419   }
420   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
421   Writer.AddSourceLocation(E->getRParenLoc(), Record);
422   Code = serialization::EXPR_SIZEOF_ALIGN_OF;
423 }
424 
425 void ASTStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
426   VisitExpr(E);
427   Writer.AddStmt(E->getLHS());
428   Writer.AddStmt(E->getRHS());
429   Writer.AddSourceLocation(E->getRBracketLoc(), Record);
430   Code = serialization::EXPR_ARRAY_SUBSCRIPT;
431 }
432 
433 void ASTStmtWriter::VisitCallExpr(CallExpr *E) {
434   VisitExpr(E);
435   Record.push_back(E->getNumArgs());
436   Writer.AddSourceLocation(E->getRParenLoc(), Record);
437   Writer.AddStmt(E->getCallee());
438   for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
439        Arg != ArgEnd; ++Arg)
440     Writer.AddStmt(*Arg);
441   Code = serialization::EXPR_CALL;
442 }
443 
444 void ASTStmtWriter::VisitMemberExpr(MemberExpr *E) {
445   // Don't call VisitExpr, we'll write everything here.
446 
447   Record.push_back(E->hasQualifier());
448   if (E->hasQualifier())
449     Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
450 
451   Record.push_back(E->hasExplicitTemplateArgs());
452   if (E->hasExplicitTemplateArgs()) {
453     unsigned NumTemplateArgs = E->getNumTemplateArgs();
454     Record.push_back(NumTemplateArgs);
455     Writer.AddSourceLocation(E->getLAngleLoc(), Record);
456     Writer.AddSourceLocation(E->getRAngleLoc(), Record);
457     for (unsigned i=0; i != NumTemplateArgs; ++i)
458       Writer.AddTemplateArgumentLoc(E->getTemplateArgs()[i], Record);
459   }
460 
461   DeclAccessPair FoundDecl = E->getFoundDecl();
462   Writer.AddDeclRef(FoundDecl.getDecl(), Record);
463   Record.push_back(FoundDecl.getAccess());
464 
465   Writer.AddTypeRef(E->getType(), Record);
466   Record.push_back(E->getValueKind());
467   Record.push_back(E->getObjectKind());
468   Writer.AddStmt(E->getBase());
469   Writer.AddDeclRef(E->getMemberDecl(), Record);
470   Writer.AddSourceLocation(E->getMemberLoc(), Record);
471   Record.push_back(E->isArrow());
472   Writer.AddDeclarationNameLoc(E->MemberDNLoc,
473                                E->getMemberDecl()->getDeclName(), Record);
474   Code = serialization::EXPR_MEMBER;
475 }
476 
477 void ASTStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
478   VisitExpr(E);
479   Writer.AddStmt(E->getBase());
480   Writer.AddSourceLocation(E->getIsaMemberLoc(), Record);
481   Record.push_back(E->isArrow());
482   Code = serialization::EXPR_OBJC_ISA;
483 }
484 
485 void ASTStmtWriter::
486 VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) {
487   VisitExpr(E);
488   Writer.AddStmt(E->getSubExpr());
489   Record.push_back(E->shouldCopy());
490   Code = serialization::EXPR_OBJC_INDIRECT_COPY_RESTORE;
491 }
492 
493 void ASTStmtWriter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) {
494   VisitExplicitCastExpr(E);
495   Writer.AddSourceLocation(E->getLParenLoc(), Record);
496   Writer.AddSourceLocation(E->getBridgeKeywordLoc(), Record);
497   Record.push_back(E->getBridgeKind()); // FIXME: Stable encoding
498   Code = serialization::EXPR_OBJC_BRIDGED_CAST;
499 }
500 
501 void ASTStmtWriter::VisitCastExpr(CastExpr *E) {
502   VisitExpr(E);
503   Record.push_back(E->path_size());
504   Writer.AddStmt(E->getSubExpr());
505   Record.push_back(E->getCastKind()); // FIXME: stable encoding
506 
507   for (CastExpr::path_iterator
508          PI = E->path_begin(), PE = E->path_end(); PI != PE; ++PI)
509     Writer.AddCXXBaseSpecifier(**PI, Record);
510 }
511 
512 void ASTStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
513   VisitExpr(E);
514   Writer.AddStmt(E->getLHS());
515   Writer.AddStmt(E->getRHS());
516   Record.push_back(E->getOpcode()); // FIXME: stable encoding
517   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
518   Code = serialization::EXPR_BINARY_OPERATOR;
519 }
520 
521 void ASTStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
522   VisitBinaryOperator(E);
523   Writer.AddTypeRef(E->getComputationLHSType(), Record);
524   Writer.AddTypeRef(E->getComputationResultType(), Record);
525   Code = serialization::EXPR_COMPOUND_ASSIGN_OPERATOR;
526 }
527 
528 void ASTStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
529   VisitExpr(E);
530   Writer.AddStmt(E->getCond());
531   Writer.AddStmt(E->getLHS());
532   Writer.AddStmt(E->getRHS());
533   Writer.AddSourceLocation(E->getQuestionLoc(), Record);
534   Writer.AddSourceLocation(E->getColonLoc(), Record);
535   Code = serialization::EXPR_CONDITIONAL_OPERATOR;
536 }
537 
538 void
539 ASTStmtWriter::VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
540   VisitExpr(E);
541   Writer.AddStmt(E->getOpaqueValue());
542   Writer.AddStmt(E->getCommon());
543   Writer.AddStmt(E->getCond());
544   Writer.AddStmt(E->getTrueExpr());
545   Writer.AddStmt(E->getFalseExpr());
546   Writer.AddSourceLocation(E->getQuestionLoc(), Record);
547   Writer.AddSourceLocation(E->getColonLoc(), Record);
548   Code = serialization::EXPR_BINARY_CONDITIONAL_OPERATOR;
549 }
550 
551 void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
552   VisitCastExpr(E);
553   Code = serialization::EXPR_IMPLICIT_CAST;
554 }
555 
556 void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
557   VisitCastExpr(E);
558   Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
559 }
560 
561 void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
562   VisitExplicitCastExpr(E);
563   Writer.AddSourceLocation(E->getLParenLoc(), Record);
564   Writer.AddSourceLocation(E->getRParenLoc(), Record);
565   Code = serialization::EXPR_CSTYLE_CAST;
566 }
567 
568 void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
569   VisitExpr(E);
570   Writer.AddSourceLocation(E->getLParenLoc(), Record);
571   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
572   Writer.AddStmt(E->getInitializer());
573   Record.push_back(E->isFileScope());
574   Code = serialization::EXPR_COMPOUND_LITERAL;
575 }
576 
577 void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
578   VisitExpr(E);
579   Writer.AddStmt(E->getBase());
580   Writer.AddIdentifierRef(&E->getAccessor(), Record);
581   Writer.AddSourceLocation(E->getAccessorLoc(), Record);
582   Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
583 }
584 
585 void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
586   VisitExpr(E);
587   Writer.AddStmt(E->getSyntacticForm());
588   Writer.AddSourceLocation(E->getLBraceLoc(), Record);
589   Writer.AddSourceLocation(E->getRBraceLoc(), Record);
590   bool isArrayFiller = E->ArrayFillerOrUnionFieldInit.is<Expr*>();
591   Record.push_back(isArrayFiller);
592   if (isArrayFiller)
593     Writer.AddStmt(E->getArrayFiller());
594   else
595     Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
596   Record.push_back(E->hadArrayRangeDesignator());
597   Record.push_back(E->getNumInits());
598   if (isArrayFiller) {
599     // ArrayFiller may have filled "holes" due to designated initializer.
600     // Replace them by 0 to indicate that the filler goes in that place.
601     Expr *filler = E->getArrayFiller();
602     for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
603       Writer.AddStmt(E->getInit(I) != filler ? E->getInit(I) : 0);
604   } else {
605     for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
606       Writer.AddStmt(E->getInit(I));
607   }
608   Code = serialization::EXPR_INIT_LIST;
609 }
610 
611 void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
612   VisitExpr(E);
613   Record.push_back(E->getNumSubExprs());
614   for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
615     Writer.AddStmt(E->getSubExpr(I));
616   Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
617   Record.push_back(E->usesGNUSyntax());
618   for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
619                                              DEnd = E->designators_end();
620        D != DEnd; ++D) {
621     if (D->isFieldDesignator()) {
622       if (FieldDecl *Field = D->getField()) {
623         Record.push_back(serialization::DESIG_FIELD_DECL);
624         Writer.AddDeclRef(Field, Record);
625       } else {
626         Record.push_back(serialization::DESIG_FIELD_NAME);
627         Writer.AddIdentifierRef(D->getFieldName(), Record);
628       }
629       Writer.AddSourceLocation(D->getDotLoc(), Record);
630       Writer.AddSourceLocation(D->getFieldLoc(), Record);
631     } else if (D->isArrayDesignator()) {
632       Record.push_back(serialization::DESIG_ARRAY);
633       Record.push_back(D->getFirstExprIndex());
634       Writer.AddSourceLocation(D->getLBracketLoc(), Record);
635       Writer.AddSourceLocation(D->getRBracketLoc(), Record);
636     } else {
637       assert(D->isArrayRangeDesignator() && "Unknown designator");
638       Record.push_back(serialization::DESIG_ARRAY_RANGE);
639       Record.push_back(D->getFirstExprIndex());
640       Writer.AddSourceLocation(D->getLBracketLoc(), Record);
641       Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
642       Writer.AddSourceLocation(D->getRBracketLoc(), Record);
643     }
644   }
645   Code = serialization::EXPR_DESIGNATED_INIT;
646 }
647 
648 void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
649   VisitExpr(E);
650   Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
651 }
652 
653 void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
654   VisitExpr(E);
655   Writer.AddStmt(E->getSubExpr());
656   Writer.AddTypeSourceInfo(E->getWrittenTypeInfo(), Record);
657   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
658   Writer.AddSourceLocation(E->getRParenLoc(), Record);
659   Code = serialization::EXPR_VA_ARG;
660 }
661 
662 void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
663   VisitExpr(E);
664   Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
665   Writer.AddSourceLocation(E->getLabelLoc(), Record);
666   Writer.AddDeclRef(E->getLabel(), Record);
667   Code = serialization::EXPR_ADDR_LABEL;
668 }
669 
670 void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
671   VisitExpr(E);
672   Writer.AddStmt(E->getSubStmt());
673   Writer.AddSourceLocation(E->getLParenLoc(), Record);
674   Writer.AddSourceLocation(E->getRParenLoc(), Record);
675   Code = serialization::EXPR_STMT;
676 }
677 
678 void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
679   VisitExpr(E);
680   Writer.AddStmt(E->getCond());
681   Writer.AddStmt(E->getLHS());
682   Writer.AddStmt(E->getRHS());
683   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
684   Writer.AddSourceLocation(E->getRParenLoc(), Record);
685   Code = serialization::EXPR_CHOOSE;
686 }
687 
688 void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
689   VisitExpr(E);
690   Writer.AddSourceLocation(E->getTokenLocation(), Record);
691   Code = serialization::EXPR_GNU_NULL;
692 }
693 
694 void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
695   VisitExpr(E);
696   Record.push_back(E->getNumSubExprs());
697   for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
698     Writer.AddStmt(E->getExpr(I));
699   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
700   Writer.AddSourceLocation(E->getRParenLoc(), Record);
701   Code = serialization::EXPR_SHUFFLE_VECTOR;
702 }
703 
704 void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
705   VisitExpr(E);
706   Writer.AddDeclRef(E->getBlockDecl(), Record);
707   Code = serialization::EXPR_BLOCK;
708 }
709 
710 void ASTStmtWriter::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
711   VisitExpr(E);
712   Writer.AddDeclRef(E->getDecl(), Record);
713   Writer.AddSourceLocation(E->getLocation(), Record);
714   Record.push_back(E->isByRef());
715   Record.push_back(E->isConstQualAdded());
716   Code = serialization::EXPR_BLOCK_DECL_REF;
717 }
718 
719 void ASTStmtWriter::VisitGenericSelectionExpr(GenericSelectionExpr *E) {
720   VisitExpr(E);
721   Record.push_back(E->getNumAssocs());
722 
723   Writer.AddStmt(E->getControllingExpr());
724   for (unsigned I = 0, N = E->getNumAssocs(); I != N; ++I) {
725     Writer.AddTypeSourceInfo(E->getAssocTypeSourceInfo(I), Record);
726     Writer.AddStmt(E->getAssocExpr(I));
727   }
728   Record.push_back(E->isResultDependent() ? -1U : E->getResultIndex());
729 
730   Writer.AddSourceLocation(E->getGenericLoc(), Record);
731   Writer.AddSourceLocation(E->getDefaultLoc(), Record);
732   Writer.AddSourceLocation(E->getRParenLoc(), Record);
733   Code = serialization::EXPR_GENERIC_SELECTION;
734 }
735 
736 //===----------------------------------------------------------------------===//
737 // Objective-C Expressions and Statements.
738 //===----------------------------------------------------------------------===//
739 
740 void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
741   VisitExpr(E);
742   Writer.AddStmt(E->getString());
743   Writer.AddSourceLocation(E->getAtLoc(), Record);
744   Code = serialization::EXPR_OBJC_STRING_LITERAL;
745 }
746 
747 void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
748   VisitExpr(E);
749   Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
750   Writer.AddSourceLocation(E->getAtLoc(), Record);
751   Writer.AddSourceLocation(E->getRParenLoc(), Record);
752   Code = serialization::EXPR_OBJC_ENCODE;
753 }
754 
755 void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
756   VisitExpr(E);
757   Writer.AddSelectorRef(E->getSelector(), Record);
758   Writer.AddSourceLocation(E->getAtLoc(), Record);
759   Writer.AddSourceLocation(E->getRParenLoc(), Record);
760   Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
761 }
762 
763 void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
764   VisitExpr(E);
765   Writer.AddDeclRef(E->getProtocol(), Record);
766   Writer.AddSourceLocation(E->getAtLoc(), Record);
767   Writer.AddSourceLocation(E->getRParenLoc(), Record);
768   Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
769 }
770 
771 void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
772   VisitExpr(E);
773   Writer.AddDeclRef(E->getDecl(), Record);
774   Writer.AddSourceLocation(E->getLocation(), Record);
775   Writer.AddStmt(E->getBase());
776   Record.push_back(E->isArrow());
777   Record.push_back(E->isFreeIvar());
778   Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
779 }
780 
781 void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
782   VisitExpr(E);
783   Record.push_back(E->isImplicitProperty());
784   if (E->isImplicitProperty()) {
785     Writer.AddDeclRef(E->getImplicitPropertyGetter(), Record);
786     Writer.AddDeclRef(E->getImplicitPropertySetter(), Record);
787   } else {
788     Writer.AddDeclRef(E->getExplicitProperty(), Record);
789   }
790   Writer.AddSourceLocation(E->getLocation(), Record);
791   Writer.AddSourceLocation(E->getReceiverLocation(), Record);
792   if (E->isObjectReceiver()) {
793     Record.push_back(0);
794     Writer.AddStmt(E->getBase());
795   } else if (E->isSuperReceiver()) {
796     Record.push_back(1);
797     Writer.AddTypeRef(E->getSuperReceiverType(), Record);
798   } else {
799     Record.push_back(2);
800     Writer.AddDeclRef(E->getClassReceiver(), Record);
801   }
802 
803   Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
804 }
805 
806 void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
807   VisitExpr(E);
808   Record.push_back(E->getNumArgs());
809   Record.push_back(E->getNumStoredSelLocs());
810   Record.push_back(E->SelLocsKind);
811   Record.push_back(E->isDelegateInitCall());
812   Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
813   switch (E->getReceiverKind()) {
814   case ObjCMessageExpr::Instance:
815     Writer.AddStmt(E->getInstanceReceiver());
816     break;
817 
818   case ObjCMessageExpr::Class:
819     Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
820     break;
821 
822   case ObjCMessageExpr::SuperClass:
823   case ObjCMessageExpr::SuperInstance:
824     Writer.AddTypeRef(E->getSuperType(), Record);
825     Writer.AddSourceLocation(E->getSuperLoc(), Record);
826     break;
827   }
828 
829   if (E->getMethodDecl()) {
830     Record.push_back(1);
831     Writer.AddDeclRef(E->getMethodDecl(), Record);
832   } else {
833     Record.push_back(0);
834     Writer.AddSelectorRef(E->getSelector(), Record);
835   }
836 
837   Writer.AddSourceLocation(E->getLeftLoc(), Record);
838   Writer.AddSourceLocation(E->getRightLoc(), Record);
839 
840   for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
841        Arg != ArgEnd; ++Arg)
842     Writer.AddStmt(*Arg);
843 
844   SourceLocation *Locs = E->getStoredSelLocs();
845   for (unsigned i = 0, e = E->getNumStoredSelLocs(); i != e; ++i)
846     Writer.AddSourceLocation(Locs[i], Record);
847 
848   Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
849 }
850 
851 void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
852   VisitStmt(S);
853   Writer.AddStmt(S->getElement());
854   Writer.AddStmt(S->getCollection());
855   Writer.AddStmt(S->getBody());
856   Writer.AddSourceLocation(S->getForLoc(), Record);
857   Writer.AddSourceLocation(S->getRParenLoc(), Record);
858   Code = serialization::STMT_OBJC_FOR_COLLECTION;
859 }
860 
861 void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
862   Writer.AddStmt(S->getCatchBody());
863   Writer.AddDeclRef(S->getCatchParamDecl(), Record);
864   Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
865   Writer.AddSourceLocation(S->getRParenLoc(), Record);
866   Code = serialization::STMT_OBJC_CATCH;
867 }
868 
869 void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
870   Writer.AddStmt(S->getFinallyBody());
871   Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
872   Code = serialization::STMT_OBJC_FINALLY;
873 }
874 
875 void ASTStmtWriter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *S) {
876   Writer.AddStmt(S->getSubStmt());
877   Writer.AddSourceLocation(S->getAtLoc(), Record);
878   Code = serialization::STMT_OBJC_AUTORELEASE_POOL;
879 }
880 
881 void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
882   Record.push_back(S->getNumCatchStmts());
883   Record.push_back(S->getFinallyStmt() != 0);
884   Writer.AddStmt(S->getTryBody());
885   for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
886     Writer.AddStmt(S->getCatchStmt(I));
887   if (S->getFinallyStmt())
888     Writer.AddStmt(S->getFinallyStmt());
889   Writer.AddSourceLocation(S->getAtTryLoc(), Record);
890   Code = serialization::STMT_OBJC_AT_TRY;
891 }
892 
893 void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
894   Writer.AddStmt(S->getSynchExpr());
895   Writer.AddStmt(S->getSynchBody());
896   Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
897   Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
898 }
899 
900 void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
901   Writer.AddStmt(S->getThrowExpr());
902   Writer.AddSourceLocation(S->getThrowLoc(), Record);
903   Code = serialization::STMT_OBJC_AT_THROW;
904 }
905 
906 //===----------------------------------------------------------------------===//
907 // C++ Expressions and Statements.
908 //===----------------------------------------------------------------------===//
909 
910 void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
911   VisitStmt(S);
912   Writer.AddSourceLocation(S->getCatchLoc(), Record);
913   Writer.AddDeclRef(S->getExceptionDecl(), Record);
914   Writer.AddStmt(S->getHandlerBlock());
915   Code = serialization::STMT_CXX_CATCH;
916 }
917 
918 void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
919   VisitStmt(S);
920   Record.push_back(S->getNumHandlers());
921   Writer.AddSourceLocation(S->getTryLoc(), Record);
922   Writer.AddStmt(S->getTryBlock());
923   for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
924     Writer.AddStmt(S->getHandler(i));
925   Code = serialization::STMT_CXX_TRY;
926 }
927 
928 void ASTStmtWriter::VisitCXXForRangeStmt(CXXForRangeStmt *S) {
929   VisitStmt(S);
930   Writer.AddSourceLocation(S->getForLoc(), Record);
931   Writer.AddSourceLocation(S->getColonLoc(), Record);
932   Writer.AddSourceLocation(S->getRParenLoc(), Record);
933   Writer.AddStmt(S->getRangeStmt());
934   Writer.AddStmt(S->getBeginEndStmt());
935   Writer.AddStmt(S->getCond());
936   Writer.AddStmt(S->getInc());
937   Writer.AddStmt(S->getLoopVarStmt());
938   Writer.AddStmt(S->getBody());
939   Code = serialization::STMT_CXX_FOR_RANGE;
940 }
941 
942 void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
943   VisitCallExpr(E);
944   Record.push_back(E->getOperator());
945   Code = serialization::EXPR_CXX_OPERATOR_CALL;
946 }
947 
948 void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
949   VisitCallExpr(E);
950   Code = serialization::EXPR_CXX_MEMBER_CALL;
951 }
952 
953 void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
954   VisitExpr(E);
955   Record.push_back(E->getNumArgs());
956   for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
957     Writer.AddStmt(E->getArg(I));
958   Writer.AddDeclRef(E->getConstructor(), Record);
959   Writer.AddSourceLocation(E->getLocation(), Record);
960   Record.push_back(E->isElidable());
961   Record.push_back(E->requiresZeroInitialization());
962   Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
963   Writer.AddSourceRange(E->getParenRange(), Record);
964   Code = serialization::EXPR_CXX_CONSTRUCT;
965 }
966 
967 void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
968   VisitCXXConstructExpr(E);
969   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
970   Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
971 }
972 
973 void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
974   VisitExplicitCastExpr(E);
975   Writer.AddSourceRange(SourceRange(E->getOperatorLoc(), E->getRParenLoc()),
976                         Record);
977 }
978 
979 void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
980   VisitCXXNamedCastExpr(E);
981   Code = serialization::EXPR_CXX_STATIC_CAST;
982 }
983 
984 void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
985   VisitCXXNamedCastExpr(E);
986   Code = serialization::EXPR_CXX_DYNAMIC_CAST;
987 }
988 
989 void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
990   VisitCXXNamedCastExpr(E);
991   Code = serialization::EXPR_CXX_REINTERPRET_CAST;
992 }
993 
994 void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
995   VisitCXXNamedCastExpr(E);
996   Code = serialization::EXPR_CXX_CONST_CAST;
997 }
998 
999 void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
1000   VisitExplicitCastExpr(E);
1001   Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1002   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1003   Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
1004 }
1005 
1006 void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
1007   VisitExpr(E);
1008   Record.push_back(E->getValue());
1009   Writer.AddSourceLocation(E->getLocation(), Record);
1010   Code = serialization::EXPR_CXX_BOOL_LITERAL;
1011 }
1012 
1013 void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
1014   VisitExpr(E);
1015   Writer.AddSourceLocation(E->getLocation(), Record);
1016   Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
1017 }
1018 
1019 void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1020   VisitExpr(E);
1021   Writer.AddSourceRange(E->getSourceRange(), Record);
1022   if (E->isTypeOperand()) {
1023     Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1024     Code = serialization::EXPR_CXX_TYPEID_TYPE;
1025   } else {
1026     Writer.AddStmt(E->getExprOperand());
1027     Code = serialization::EXPR_CXX_TYPEID_EXPR;
1028   }
1029 }
1030 
1031 void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
1032   VisitExpr(E);
1033   Writer.AddSourceLocation(E->getLocation(), Record);
1034   Record.push_back(E->isImplicit());
1035   Code = serialization::EXPR_CXX_THIS;
1036 }
1037 
1038 void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
1039   VisitExpr(E);
1040   Writer.AddSourceLocation(E->getThrowLoc(), Record);
1041   Writer.AddStmt(E->getSubExpr());
1042   Record.push_back(E->isThrownVariableInScope());
1043   Code = serialization::EXPR_CXX_THROW;
1044 }
1045 
1046 void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
1047   VisitExpr(E);
1048 
1049   bool HasOtherExprStored = E->Param.getInt();
1050   // Store these first, the reader reads them before creation.
1051   Record.push_back(HasOtherExprStored);
1052   if (HasOtherExprStored)
1053     Writer.AddStmt(E->getExpr());
1054   Writer.AddDeclRef(E->getParam(), Record);
1055   Writer.AddSourceLocation(E->getUsedLocation(), Record);
1056 
1057   Code = serialization::EXPR_CXX_DEFAULT_ARG;
1058 }
1059 
1060 void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
1061   VisitExpr(E);
1062   Writer.AddCXXTemporary(E->getTemporary(), Record);
1063   Writer.AddStmt(E->getSubExpr());
1064   Code = serialization::EXPR_CXX_BIND_TEMPORARY;
1065 }
1066 
1067 void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
1068   VisitExpr(E);
1069   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
1070   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1071   Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
1072 }
1073 
1074 void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
1075   VisitExpr(E);
1076   Record.push_back(E->isGlobalNew());
1077   Record.push_back(E->hasInitializer());
1078   Record.push_back(E->doesUsualArrayDeleteWantSize());
1079   Record.push_back(E->isArray());
1080   Record.push_back(E->getNumPlacementArgs());
1081   Record.push_back(E->getNumConstructorArgs());
1082   Writer.AddDeclRef(E->getOperatorNew(), Record);
1083   Writer.AddDeclRef(E->getOperatorDelete(), Record);
1084   Writer.AddDeclRef(E->getConstructor(), Record);
1085   Writer.AddTypeSourceInfo(E->getAllocatedTypeSourceInfo(), Record);
1086   Writer.AddSourceRange(E->getTypeIdParens(), Record);
1087   Writer.AddSourceLocation(E->getStartLoc(), Record);
1088   Writer.AddSourceLocation(E->getEndLoc(), Record);
1089   Writer.AddSourceLocation(E->getConstructorLParen(), Record);
1090   Writer.AddSourceLocation(E->getConstructorRParen(), Record);
1091   for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
1092        I != e; ++I)
1093     Writer.AddStmt(*I);
1094 
1095   Code = serialization::EXPR_CXX_NEW;
1096 }
1097 
1098 void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
1099   VisitExpr(E);
1100   Record.push_back(E->isGlobalDelete());
1101   Record.push_back(E->isArrayForm());
1102   Record.push_back(E->isArrayFormAsWritten());
1103   Record.push_back(E->doesUsualArrayDeleteWantSize());
1104   Writer.AddDeclRef(E->getOperatorDelete(), Record);
1105   Writer.AddStmt(E->getArgument());
1106   Writer.AddSourceLocation(E->getSourceRange().getBegin(), Record);
1107 
1108   Code = serialization::EXPR_CXX_DELETE;
1109 }
1110 
1111 void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1112   VisitExpr(E);
1113 
1114   Writer.AddStmt(E->getBase());
1115   Record.push_back(E->isArrow());
1116   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1117   Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1118   Writer.AddTypeSourceInfo(E->getScopeTypeInfo(), Record);
1119   Writer.AddSourceLocation(E->getColonColonLoc(), Record);
1120   Writer.AddSourceLocation(E->getTildeLoc(), Record);
1121 
1122   // PseudoDestructorTypeStorage.
1123   Writer.AddIdentifierRef(E->getDestroyedTypeIdentifier(), Record);
1124   if (E->getDestroyedTypeIdentifier())
1125     Writer.AddSourceLocation(E->getDestroyedTypeLoc(), Record);
1126   else
1127     Writer.AddTypeSourceInfo(E->getDestroyedTypeInfo(), Record);
1128 
1129   Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
1130 }
1131 
1132 void ASTStmtWriter::VisitExprWithCleanups(ExprWithCleanups *E) {
1133   VisitExpr(E);
1134   Record.push_back(E->getNumTemporaries());
1135   for (unsigned i = 0, e = E->getNumTemporaries(); i != e; ++i)
1136     Writer.AddCXXTemporary(E->getTemporary(i), Record);
1137 
1138   Writer.AddStmt(E->getSubExpr());
1139   Code = serialization::EXPR_EXPR_WITH_CLEANUPS;
1140 }
1141 
1142 void
1143 ASTStmtWriter::VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E){
1144   VisitExpr(E);
1145 
1146   // Don't emit anything here, hasExplicitTemplateArgs() must be
1147   // emitted first.
1148 
1149   Record.push_back(E->hasExplicitTemplateArgs());
1150   if (E->hasExplicitTemplateArgs()) {
1151     const ASTTemplateArgumentListInfo &Args = E->getExplicitTemplateArgs();
1152     Record.push_back(Args.NumTemplateArgs);
1153     AddExplicitTemplateArgumentList(Args);
1154   }
1155 
1156   if (!E->isImplicitAccess())
1157     Writer.AddStmt(E->getBase());
1158   else
1159     Writer.AddStmt(0);
1160   Writer.AddTypeRef(E->getBaseType(), Record);
1161   Record.push_back(E->isArrow());
1162   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1163   Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1164   Writer.AddDeclRef(E->getFirstQualifierFoundInScope(), Record);
1165   Writer.AddDeclarationNameInfo(E->MemberNameInfo, Record);
1166   Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
1167 }
1168 
1169 void
1170 ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1171   VisitExpr(E);
1172 
1173   // Don't emit anything here, hasExplicitTemplateArgs() must be
1174   // emitted first.
1175   Record.push_back(E->hasExplicitTemplateArgs());
1176   if (E->hasExplicitTemplateArgs()) {
1177     const ASTTemplateArgumentListInfo &Args = E->getExplicitTemplateArgs();
1178     Record.push_back(Args.NumTemplateArgs);
1179     AddExplicitTemplateArgumentList(Args);
1180   }
1181 
1182   Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1183   Writer.AddDeclarationNameInfo(E->NameInfo, Record);
1184   Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
1185 }
1186 
1187 void
1188 ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
1189   VisitExpr(E);
1190   Record.push_back(E->arg_size());
1191   for (CXXUnresolvedConstructExpr::arg_iterator
1192          ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
1193     Writer.AddStmt(*ArgI);
1194   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
1195   Writer.AddSourceLocation(E->getLParenLoc(), Record);
1196   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1197   Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
1198 }
1199 
1200 void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
1201   VisitExpr(E);
1202 
1203   // Don't emit anything here, hasExplicitTemplateArgs() must be emitted first.
1204   Record.push_back(E->hasExplicitTemplateArgs());
1205   if (E->hasExplicitTemplateArgs()) {
1206     const ASTTemplateArgumentListInfo &Args = E->getExplicitTemplateArgs();
1207     Record.push_back(Args.NumTemplateArgs);
1208     AddExplicitTemplateArgumentList(Args);
1209   }
1210 
1211   Record.push_back(E->getNumDecls());
1212   for (OverloadExpr::decls_iterator
1213          OvI = E->decls_begin(), OvE = E->decls_end(); OvI != OvE; ++OvI) {
1214     Writer.AddDeclRef(OvI.getDecl(), Record);
1215     Record.push_back(OvI.getAccess());
1216   }
1217 
1218   Writer.AddDeclarationNameInfo(E->NameInfo, Record);
1219   Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1220 }
1221 
1222 void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
1223   VisitOverloadExpr(E);
1224   Record.push_back(E->isArrow());
1225   Record.push_back(E->hasUnresolvedUsing());
1226   Writer.AddStmt(!E->isImplicitAccess() ? E->getBase() : 0);
1227   Writer.AddTypeRef(E->getBaseType(), Record);
1228   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1229   Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
1230 }
1231 
1232 void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
1233   VisitOverloadExpr(E);
1234   Record.push_back(E->requiresADL());
1235   if (E->requiresADL())
1236     Record.push_back(E->isStdAssociatedNamespace());
1237   Record.push_back(E->isOverloaded());
1238   Writer.AddDeclRef(E->getNamingClass(), Record);
1239   Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
1240 }
1241 
1242 void ASTStmtWriter::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
1243   VisitExpr(E);
1244   Record.push_back(E->getTrait());
1245   Record.push_back(E->getValue());
1246   Writer.AddSourceRange(E->getSourceRange(), Record);
1247   Writer.AddTypeSourceInfo(E->getQueriedTypeSourceInfo(), Record);
1248   Code = serialization::EXPR_CXX_UNARY_TYPE_TRAIT;
1249 }
1250 
1251 void ASTStmtWriter::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
1252   VisitExpr(E);
1253   Record.push_back(E->getTrait());
1254   Record.push_back(E->getValue());
1255   Writer.AddSourceRange(E->getSourceRange(), Record);
1256   Writer.AddTypeSourceInfo(E->getLhsTypeSourceInfo(), Record);
1257   Writer.AddTypeSourceInfo(E->getRhsTypeSourceInfo(), Record);
1258   Code = serialization::EXPR_BINARY_TYPE_TRAIT;
1259 }
1260 
1261 void ASTStmtWriter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
1262   VisitExpr(E);
1263   Record.push_back(E->getTrait());
1264   Record.push_back(E->getValue());
1265   Writer.AddSourceRange(E->getSourceRange(), Record);
1266   Writer.AddTypeSourceInfo(E->getQueriedTypeSourceInfo(), Record);
1267   Code = serialization::EXPR_ARRAY_TYPE_TRAIT;
1268 }
1269 
1270 void ASTStmtWriter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
1271   VisitExpr(E);
1272   Record.push_back(E->getTrait());
1273   Record.push_back(E->getValue());
1274   Writer.AddSourceRange(E->getSourceRange(), Record);
1275   Writer.AddStmt(E->getQueriedExpression());
1276   Code = serialization::EXPR_CXX_EXPRESSION_TRAIT;
1277 }
1278 
1279 void ASTStmtWriter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
1280   VisitExpr(E);
1281   Record.push_back(E->getValue());
1282   Writer.AddSourceRange(E->getSourceRange(), Record);
1283   Writer.AddStmt(E->getOperand());
1284   Code = serialization::EXPR_CXX_NOEXCEPT;
1285 }
1286 
1287 void ASTStmtWriter::VisitPackExpansionExpr(PackExpansionExpr *E) {
1288   VisitExpr(E);
1289   Writer.AddSourceLocation(E->getEllipsisLoc(), Record);
1290   Record.push_back(E->NumExpansions);
1291   Writer.AddStmt(E->getPattern());
1292   Code = serialization::EXPR_PACK_EXPANSION;
1293 }
1294 
1295 void ASTStmtWriter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
1296   VisitExpr(E);
1297   Writer.AddSourceLocation(E->OperatorLoc, Record);
1298   Writer.AddSourceLocation(E->PackLoc, Record);
1299   Writer.AddSourceLocation(E->RParenLoc, Record);
1300   Record.push_back(E->Length);
1301   Writer.AddDeclRef(E->Pack, Record);
1302   Code = serialization::EXPR_SIZEOF_PACK;
1303 }
1304 
1305 void ASTStmtWriter::VisitSubstNonTypeTemplateParmExpr(
1306                                               SubstNonTypeTemplateParmExpr *E) {
1307   VisitExpr(E);
1308   Writer.AddDeclRef(E->getParameter(), Record);
1309   Writer.AddSourceLocation(E->getNameLoc(), Record);
1310   Writer.AddStmt(E->getReplacement());
1311   Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM;
1312 }
1313 
1314 void ASTStmtWriter::VisitSubstNonTypeTemplateParmPackExpr(
1315                                           SubstNonTypeTemplateParmPackExpr *E) {
1316   VisitExpr(E);
1317   Writer.AddDeclRef(E->getParameterPack(), Record);
1318   Writer.AddTemplateArgument(E->getArgumentPack(), Record);
1319   Writer.AddSourceLocation(E->getParameterPackLocation(), Record);
1320   Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM_PACK;
1321 }
1322 
1323 void ASTStmtWriter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E) {
1324   VisitExpr(E);
1325   Writer.AddStmt(E->Temporary);
1326   Code = serialization::EXPR_MATERIALIZE_TEMPORARY;
1327 }
1328 
1329 void ASTStmtWriter::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
1330   VisitExpr(E);
1331   Record.push_back(Writer.getOpaqueValueID(E));
1332   Writer.AddSourceLocation(E->getLocation(), Record);
1333   Code = serialization::EXPR_OPAQUE_VALUE;
1334 }
1335 
1336 //===----------------------------------------------------------------------===//
1337 // CUDA Expressions and Statements.
1338 //===----------------------------------------------------------------------===//
1339 
1340 void ASTStmtWriter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *E) {
1341   VisitCallExpr(E);
1342   Writer.AddStmt(E->getConfig());
1343   Code = serialization::EXPR_CUDA_KERNEL_CALL;
1344 }
1345 
1346 //===----------------------------------------------------------------------===//
1347 // OpenCL Expressions and Statements.
1348 //===----------------------------------------------------------------------===//
1349 void ASTStmtWriter::VisitAsTypeExpr(AsTypeExpr *E) {
1350   VisitExpr(E);
1351   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
1352   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1353   Writer.AddStmt(E->getSrcExpr());
1354   Code = serialization::EXPR_ASTYPE;
1355 }
1356 
1357 //===----------------------------------------------------------------------===//
1358 // Microsoft Expressions and Statements.
1359 //===----------------------------------------------------------------------===//
1360 void ASTStmtWriter::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
1361   VisitExpr(E);
1362   Writer.AddSourceRange(E->getSourceRange(), Record);
1363   if (E->isTypeOperand()) {
1364     Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1365     Code = serialization::EXPR_CXX_UUIDOF_TYPE;
1366   } else {
1367     Writer.AddStmt(E->getExprOperand());
1368     Code = serialization::EXPR_CXX_UUIDOF_EXPR;
1369   }
1370 }
1371 
1372 void ASTStmtWriter::VisitSEHExceptStmt(SEHExceptStmt *S) {
1373   VisitStmt(S);
1374   Writer.AddSourceLocation(S->getExceptLoc(), Record);
1375   Writer.AddStmt(S->getFilterExpr());
1376   Writer.AddStmt(S->getBlock());
1377   Code = serialization::STMT_SEH_EXCEPT;
1378 }
1379 
1380 void ASTStmtWriter::VisitSEHFinallyStmt(SEHFinallyStmt *S) {
1381   VisitStmt(S);
1382   Writer.AddSourceLocation(S->getFinallyLoc(), Record);
1383   Writer.AddStmt(S->getBlock());
1384   Code = serialization::STMT_SEH_FINALLY;
1385 }
1386 
1387 void ASTStmtWriter::VisitSEHTryStmt(SEHTryStmt *S) {
1388   VisitStmt(S);
1389   Record.push_back(S->getIsCXXTry());
1390   Writer.AddSourceLocation(S->getTryLoc(), Record);
1391   Writer.AddStmt(S->getTryBlock());
1392   Writer.AddStmt(S->getHandler());
1393   Code = serialization::STMT_SEH_TRY;
1394 }
1395 
1396 //===----------------------------------------------------------------------===//
1397 // ASTWriter Implementation
1398 //===----------------------------------------------------------------------===//
1399 
1400 unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
1401   assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
1402          "SwitchCase recorded twice");
1403   unsigned NextID = SwitchCaseIDs.size();
1404   SwitchCaseIDs[S] = NextID;
1405   return NextID;
1406 }
1407 
1408 unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
1409   assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
1410          "SwitchCase hasn't been seen yet");
1411   return SwitchCaseIDs[S];
1412 }
1413 
1414 void ASTWriter::ClearSwitchCaseIDs() {
1415   SwitchCaseIDs.clear();
1416 }
1417 
1418 unsigned ASTWriter::getOpaqueValueID(OpaqueValueExpr *e) {
1419   unsigned &entry = OpaqueValues[e];
1420   if (!entry) entry = OpaqueValues.size();
1421   return entry;
1422 }
1423 
1424 /// \brief Write the given substatement or subexpression to the
1425 /// bitstream.
1426 void ASTWriter::WriteSubStmt(Stmt *S) {
1427   RecordData Record;
1428   ASTStmtWriter Writer(*this, Record);
1429   ++NumStatements;
1430 
1431   if (!S) {
1432     Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
1433     return;
1434   }
1435 
1436   // Redirect ASTWriter::AddStmt to collect sub stmts.
1437   SmallVector<Stmt *, 16> SubStmts;
1438   CollectedStmts = &SubStmts;
1439 
1440   Writer.Code = serialization::STMT_NULL_PTR;
1441   Writer.AbbrevToUse = 0;
1442   Writer.Visit(S);
1443 
1444 #ifndef NDEBUG
1445   if (Writer.Code == serialization::STMT_NULL_PTR) {
1446     SourceManager &SrcMgr
1447       = DeclIDs.begin()->first->getASTContext().getSourceManager();
1448     S->dump(SrcMgr);
1449     llvm_unreachable("Unhandled sub statement writing AST file");
1450   }
1451 #endif
1452 
1453   // Revert ASTWriter::AddStmt.
1454   CollectedStmts = &StmtsToEmit;
1455 
1456   // Write the sub stmts in reverse order, last to first. When reading them back
1457   // we will read them in correct order by "pop"ing them from the Stmts stack.
1458   // This simplifies reading and allows to store a variable number of sub stmts
1459   // without knowing it in advance.
1460   while (!SubStmts.empty())
1461     WriteSubStmt(SubStmts.pop_back_val());
1462 
1463   Stream.EmitRecord(Writer.Code, Record, Writer.AbbrevToUse);
1464 }
1465 
1466 /// \brief Flush all of the statements that have been added to the
1467 /// queue via AddStmt().
1468 void ASTWriter::FlushStmts() {
1469   RecordData Record;
1470 
1471   for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
1472     WriteSubStmt(StmtsToEmit[I]);
1473 
1474     assert(N == StmtsToEmit.size() &&
1475            "Substatement written via AddStmt rather than WriteSubStmt!");
1476 
1477     // Note that we are at the end of a full expression. Any
1478     // expression records that follow this one are part of a different
1479     // expression.
1480     Stream.EmitRecord(serialization::STMT_STOP, Record);
1481   }
1482 
1483   StmtsToEmit.clear();
1484 }
1485