]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/llvm/tools/clang/lib/Serialization/ASTWriterStmt.cpp
Merge clang trunk r238337 from ^/vendor/clang/dist, resolve conflicts,
[FreeBSD/FreeBSD.git] / contrib / llvm / tools / clang / lib / Serialization / ASTWriterStmt.cpp
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 /// \file
11 /// \brief Implements serialization for Statements and Expressions.
12 ///
13 //===----------------------------------------------------------------------===//
14
15 #include "clang/Serialization/ASTWriter.h"
16 #include "clang/AST/ASTContext.h"
17 #include "clang/AST/DeclCXX.h"
18 #include "clang/AST/DeclObjC.h"
19 #include "clang/AST/DeclTemplate.h"
20 #include "clang/AST/StmtVisitor.h"
21 #include "clang/Lex/Token.h"
22 #include "llvm/Bitcode/BitstreamWriter.h"
23 using namespace clang;
24
25 //===----------------------------------------------------------------------===//
26 // Statement/expression serialization
27 //===----------------------------------------------------------------------===//
28
29 namespace clang {
30
31   class ASTStmtWriter : public StmtVisitor<ASTStmtWriter, void> {
32     friend class OMPClauseWriter;
33     ASTWriter &Writer;
34     ASTWriter::RecordData &Record;
35
36   public:
37     serialization::StmtCode Code;
38     unsigned AbbrevToUse;
39
40     ASTStmtWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
41       : Writer(Writer), Record(Record) { }
42
43     void AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo &Args);
44
45     void VisitStmt(Stmt *S);
46 #define STMT(Type, Base) \
47     void Visit##Type(Type *);
48 #include "clang/AST/StmtNodes.inc"
49   };
50 }
51
52 void ASTStmtWriter::
53 AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo &Args) {
54   Writer.AddSourceLocation(Args.getTemplateKeywordLoc(), Record);
55   Writer.AddSourceLocation(Args.LAngleLoc, Record);
56   Writer.AddSourceLocation(Args.RAngleLoc, Record);
57   for (unsigned i=0; i != Args.NumTemplateArgs; ++i)
58     Writer.AddTemplateArgumentLoc(Args.getTemplateArgs()[i], Record);
59 }
60
61 void ASTStmtWriter::VisitStmt(Stmt *S) {
62 }
63
64 void ASTStmtWriter::VisitNullStmt(NullStmt *S) {
65   VisitStmt(S);
66   Writer.AddSourceLocation(S->getSemiLoc(), Record);
67   Record.push_back(S->HasLeadingEmptyMacro);
68   Code = serialization::STMT_NULL;
69 }
70
71 void ASTStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
72   VisitStmt(S);
73   Record.push_back(S->size());
74   for (auto *CS : S->body())
75     Writer.AddStmt(CS);
76   Writer.AddSourceLocation(S->getLBracLoc(), Record);
77   Writer.AddSourceLocation(S->getRBracLoc(), Record);
78   Code = serialization::STMT_COMPOUND;
79 }
80
81 void ASTStmtWriter::VisitSwitchCase(SwitchCase *S) {
82   VisitStmt(S);
83   Record.push_back(Writer.getSwitchCaseID(S));
84   Writer.AddSourceLocation(S->getKeywordLoc(), Record);
85   Writer.AddSourceLocation(S->getColonLoc(), Record);
86 }
87
88 void ASTStmtWriter::VisitCaseStmt(CaseStmt *S) {
89   VisitSwitchCase(S);
90   Writer.AddStmt(S->getLHS());
91   Writer.AddStmt(S->getRHS());
92   Writer.AddStmt(S->getSubStmt());
93   Writer.AddSourceLocation(S->getEllipsisLoc(), Record);
94   Code = serialization::STMT_CASE;
95 }
96
97 void ASTStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
98   VisitSwitchCase(S);
99   Writer.AddStmt(S->getSubStmt());
100   Code = serialization::STMT_DEFAULT;
101 }
102
103 void ASTStmtWriter::VisitLabelStmt(LabelStmt *S) {
104   VisitStmt(S);
105   Writer.AddDeclRef(S->getDecl(), Record);
106   Writer.AddStmt(S->getSubStmt());
107   Writer.AddSourceLocation(S->getIdentLoc(), Record);
108   Code = serialization::STMT_LABEL;
109 }
110
111 void ASTStmtWriter::VisitAttributedStmt(AttributedStmt *S) {
112   VisitStmt(S);
113   Record.push_back(S->getAttrs().size());
114   Writer.WriteAttributes(S->getAttrs(), Record);
115   Writer.AddStmt(S->getSubStmt());
116   Writer.AddSourceLocation(S->getAttrLoc(), Record);
117   Code = serialization::STMT_ATTRIBUTED;
118 }
119
120 void ASTStmtWriter::VisitIfStmt(IfStmt *S) {
121   VisitStmt(S);
122   Writer.AddDeclRef(S->getConditionVariable(), Record);
123   Writer.AddStmt(S->getCond());
124   Writer.AddStmt(S->getThen());
125   Writer.AddStmt(S->getElse());
126   Writer.AddSourceLocation(S->getIfLoc(), Record);
127   Writer.AddSourceLocation(S->getElseLoc(), Record);
128   Code = serialization::STMT_IF;
129 }
130
131 void ASTStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
132   VisitStmt(S);
133   Writer.AddDeclRef(S->getConditionVariable(), Record);
134   Writer.AddStmt(S->getCond());
135   Writer.AddStmt(S->getBody());
136   Writer.AddSourceLocation(S->getSwitchLoc(), Record);
137   Record.push_back(S->isAllEnumCasesCovered());
138   for (SwitchCase *SC = S->getSwitchCaseList(); SC;
139        SC = SC->getNextSwitchCase())
140     Record.push_back(Writer.RecordSwitchCaseID(SC));
141   Code = serialization::STMT_SWITCH;
142 }
143
144 void ASTStmtWriter::VisitWhileStmt(WhileStmt *S) {
145   VisitStmt(S);
146   Writer.AddDeclRef(S->getConditionVariable(), Record);
147   Writer.AddStmt(S->getCond());
148   Writer.AddStmt(S->getBody());
149   Writer.AddSourceLocation(S->getWhileLoc(), Record);
150   Code = serialization::STMT_WHILE;
151 }
152
153 void ASTStmtWriter::VisitDoStmt(DoStmt *S) {
154   VisitStmt(S);
155   Writer.AddStmt(S->getCond());
156   Writer.AddStmt(S->getBody());
157   Writer.AddSourceLocation(S->getDoLoc(), Record);
158   Writer.AddSourceLocation(S->getWhileLoc(), Record);
159   Writer.AddSourceLocation(S->getRParenLoc(), Record);
160   Code = serialization::STMT_DO;
161 }
162
163 void ASTStmtWriter::VisitForStmt(ForStmt *S) {
164   VisitStmt(S);
165   Writer.AddStmt(S->getInit());
166   Writer.AddStmt(S->getCond());
167   Writer.AddDeclRef(S->getConditionVariable(), Record);
168   Writer.AddStmt(S->getInc());
169   Writer.AddStmt(S->getBody());
170   Writer.AddSourceLocation(S->getForLoc(), Record);
171   Writer.AddSourceLocation(S->getLParenLoc(), Record);
172   Writer.AddSourceLocation(S->getRParenLoc(), Record);
173   Code = serialization::STMT_FOR;
174 }
175
176 void ASTStmtWriter::VisitGotoStmt(GotoStmt *S) {
177   VisitStmt(S);
178   Writer.AddDeclRef(S->getLabel(), Record);
179   Writer.AddSourceLocation(S->getGotoLoc(), Record);
180   Writer.AddSourceLocation(S->getLabelLoc(), Record);
181   Code = serialization::STMT_GOTO;
182 }
183
184 void ASTStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
185   VisitStmt(S);
186   Writer.AddSourceLocation(S->getGotoLoc(), Record);
187   Writer.AddSourceLocation(S->getStarLoc(), Record);
188   Writer.AddStmt(S->getTarget());
189   Code = serialization::STMT_INDIRECT_GOTO;
190 }
191
192 void ASTStmtWriter::VisitContinueStmt(ContinueStmt *S) {
193   VisitStmt(S);
194   Writer.AddSourceLocation(S->getContinueLoc(), Record);
195   Code = serialization::STMT_CONTINUE;
196 }
197
198 void ASTStmtWriter::VisitBreakStmt(BreakStmt *S) {
199   VisitStmt(S);
200   Writer.AddSourceLocation(S->getBreakLoc(), Record);
201   Code = serialization::STMT_BREAK;
202 }
203
204 void ASTStmtWriter::VisitReturnStmt(ReturnStmt *S) {
205   VisitStmt(S);
206   Writer.AddStmt(S->getRetValue());
207   Writer.AddSourceLocation(S->getReturnLoc(), Record);
208   Writer.AddDeclRef(S->getNRVOCandidate(), Record);
209   Code = serialization::STMT_RETURN;
210 }
211
212 void ASTStmtWriter::VisitDeclStmt(DeclStmt *S) {
213   VisitStmt(S);
214   Writer.AddSourceLocation(S->getStartLoc(), Record);
215   Writer.AddSourceLocation(S->getEndLoc(), Record);
216   DeclGroupRef DG = S->getDeclGroup();
217   for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
218     Writer.AddDeclRef(*D, Record);
219   Code = serialization::STMT_DECL;
220 }
221
222 void ASTStmtWriter::VisitAsmStmt(AsmStmt *S) {
223   VisitStmt(S);
224   Record.push_back(S->getNumOutputs());
225   Record.push_back(S->getNumInputs());
226   Record.push_back(S->getNumClobbers());
227   Writer.AddSourceLocation(S->getAsmLoc(), Record);
228   Record.push_back(S->isVolatile());
229   Record.push_back(S->isSimple());
230 }
231
232 void ASTStmtWriter::VisitGCCAsmStmt(GCCAsmStmt *S) {
233   VisitAsmStmt(S);
234   Writer.AddSourceLocation(S->getRParenLoc(), Record);
235   Writer.AddStmt(S->getAsmString());
236
237   // Outputs
238   for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {      
239     Writer.AddIdentifierRef(S->getOutputIdentifier(I), Record);
240     Writer.AddStmt(S->getOutputConstraintLiteral(I));
241     Writer.AddStmt(S->getOutputExpr(I));
242   }
243
244   // Inputs
245   for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
246     Writer.AddIdentifierRef(S->getInputIdentifier(I), Record);
247     Writer.AddStmt(S->getInputConstraintLiteral(I));
248     Writer.AddStmt(S->getInputExpr(I));
249   }
250
251   // Clobbers
252   for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
253     Writer.AddStmt(S->getClobberStringLiteral(I));
254
255   Code = serialization::STMT_GCCASM;
256 }
257
258 void ASTStmtWriter::VisitMSAsmStmt(MSAsmStmt *S) {
259   VisitAsmStmt(S);
260   Writer.AddSourceLocation(S->getLBraceLoc(), Record);
261   Writer.AddSourceLocation(S->getEndLoc(), Record);
262   Record.push_back(S->getNumAsmToks());
263   Writer.AddString(S->getAsmString(), Record);
264
265   // Tokens
266   for (unsigned I = 0, N = S->getNumAsmToks(); I != N; ++I) {
267     Writer.AddToken(S->getAsmToks()[I], Record);
268   }
269
270   // Clobbers
271   for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I) {
272     Writer.AddString(S->getClobber(I), Record);
273   }
274
275   // Outputs
276   for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {      
277     Writer.AddStmt(S->getOutputExpr(I));
278     Writer.AddString(S->getOutputConstraint(I), Record);
279   }
280
281   // Inputs
282   for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
283     Writer.AddStmt(S->getInputExpr(I));
284     Writer.AddString(S->getInputConstraint(I), Record);
285   }
286
287   Code = serialization::STMT_MSASM;
288 }
289
290 void ASTStmtWriter::VisitCapturedStmt(CapturedStmt *S) {
291   VisitStmt(S);
292   // NumCaptures
293   Record.push_back(std::distance(S->capture_begin(), S->capture_end()));
294
295   // CapturedDecl and captured region kind
296   Writer.AddDeclRef(S->getCapturedDecl(), Record);
297   Record.push_back(S->getCapturedRegionKind());
298
299   Writer.AddDeclRef(S->getCapturedRecordDecl(), Record);
300
301   // Capture inits
302   for (auto *I : S->capture_inits())
303     Writer.AddStmt(I);
304
305   // Body
306   Writer.AddStmt(S->getCapturedStmt());
307
308   // Captures
309   for (const auto &I : S->captures()) {
310     if (I.capturesThis() || I.capturesVariableArrayType())
311       Writer.AddDeclRef(nullptr, Record);
312     else
313       Writer.AddDeclRef(I.getCapturedVar(), Record);
314     Record.push_back(I.getCaptureKind());
315     Writer.AddSourceLocation(I.getLocation(), Record);
316   }
317
318   Code = serialization::STMT_CAPTURED;
319 }
320
321 void ASTStmtWriter::VisitExpr(Expr *E) {
322   VisitStmt(E);
323   Writer.AddTypeRef(E->getType(), Record);
324   Record.push_back(E->isTypeDependent());
325   Record.push_back(E->isValueDependent());
326   Record.push_back(E->isInstantiationDependent());
327   Record.push_back(E->containsUnexpandedParameterPack());
328   Record.push_back(E->getValueKind());
329   Record.push_back(E->getObjectKind());
330 }
331
332 void ASTStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
333   VisitExpr(E);
334   Writer.AddSourceLocation(E->getLocation(), Record);
335   Record.push_back(E->getIdentType()); // FIXME: stable encoding
336   Writer.AddStmt(E->getFunctionName());
337   Code = serialization::EXPR_PREDEFINED;
338 }
339
340 void ASTStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
341   VisitExpr(E);
342
343   Record.push_back(E->hasQualifier());
344   Record.push_back(E->getDecl() != E->getFoundDecl());
345   Record.push_back(E->hasTemplateKWAndArgsInfo());
346   Record.push_back(E->hadMultipleCandidates());
347   Record.push_back(E->refersToEnclosingVariableOrCapture());
348
349   if (E->hasTemplateKWAndArgsInfo()) {
350     unsigned NumTemplateArgs = E->getNumTemplateArgs();
351     Record.push_back(NumTemplateArgs);
352   }
353
354   DeclarationName::NameKind nk = (E->getDecl()->getDeclName().getNameKind());
355
356   if ((!E->hasTemplateKWAndArgsInfo()) && (!E->hasQualifier()) &&
357       (E->getDecl() == E->getFoundDecl()) &&
358       nk == DeclarationName::Identifier) {
359     AbbrevToUse = Writer.getDeclRefExprAbbrev();
360   }
361
362   if (E->hasQualifier())
363     Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
364
365   if (E->getDecl() != E->getFoundDecl())
366     Writer.AddDeclRef(E->getFoundDecl(), Record);
367
368   if (E->hasTemplateKWAndArgsInfo())
369     AddTemplateKWAndArgsInfo(*E->getTemplateKWAndArgsInfo());
370
371   Writer.AddDeclRef(E->getDecl(), Record);
372   Writer.AddSourceLocation(E->getLocation(), Record);
373   Writer.AddDeclarationNameLoc(E->DNLoc, E->getDecl()->getDeclName(), Record);
374   Code = serialization::EXPR_DECL_REF;
375 }
376
377 void ASTStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
378   VisitExpr(E);
379   Writer.AddSourceLocation(E->getLocation(), Record);
380   Writer.AddAPInt(E->getValue(), Record);
381
382   if (E->getValue().getBitWidth() == 32) {
383     AbbrevToUse = Writer.getIntegerLiteralAbbrev();
384   }
385
386   Code = serialization::EXPR_INTEGER_LITERAL;
387 }
388
389 void ASTStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
390   VisitExpr(E);
391   Record.push_back(E->getRawSemantics());
392   Record.push_back(E->isExact());
393   Writer.AddAPFloat(E->getValue(), Record);
394   Writer.AddSourceLocation(E->getLocation(), Record);
395   Code = serialization::EXPR_FLOATING_LITERAL;
396 }
397
398 void ASTStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
399   VisitExpr(E);
400   Writer.AddStmt(E->getSubExpr());
401   Code = serialization::EXPR_IMAGINARY_LITERAL;
402 }
403
404 void ASTStmtWriter::VisitStringLiteral(StringLiteral *E) {
405   VisitExpr(E);
406   Record.push_back(E->getByteLength());
407   Record.push_back(E->getNumConcatenated());
408   Record.push_back(E->getKind());
409   Record.push_back(E->isPascal());
410   // FIXME: String data should be stored as a blob at the end of the
411   // StringLiteral. However, we can't do so now because we have no
412   // provision for coping with abbreviations when we're jumping around
413   // the AST file during deserialization.
414   Record.append(E->getBytes().begin(), E->getBytes().end());
415   for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
416     Writer.AddSourceLocation(E->getStrTokenLoc(I), Record);
417   Code = serialization::EXPR_STRING_LITERAL;
418 }
419
420 void ASTStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
421   VisitExpr(E);
422   Record.push_back(E->getValue());
423   Writer.AddSourceLocation(E->getLocation(), Record);
424   Record.push_back(E->getKind());
425
426   AbbrevToUse = Writer.getCharacterLiteralAbbrev();
427
428   Code = serialization::EXPR_CHARACTER_LITERAL;
429 }
430
431 void ASTStmtWriter::VisitParenExpr(ParenExpr *E) {
432   VisitExpr(E);
433   Writer.AddSourceLocation(E->getLParen(), Record);
434   Writer.AddSourceLocation(E->getRParen(), Record);
435   Writer.AddStmt(E->getSubExpr());
436   Code = serialization::EXPR_PAREN;
437 }
438
439 void ASTStmtWriter::VisitParenListExpr(ParenListExpr *E) {
440   VisitExpr(E);
441   Record.push_back(E->NumExprs);
442   for (unsigned i=0; i != E->NumExprs; ++i)
443     Writer.AddStmt(E->Exprs[i]);
444   Writer.AddSourceLocation(E->LParenLoc, Record);
445   Writer.AddSourceLocation(E->RParenLoc, Record);
446   Code = serialization::EXPR_PAREN_LIST;
447 }
448
449 void ASTStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
450   VisitExpr(E);
451   Writer.AddStmt(E->getSubExpr());
452   Record.push_back(E->getOpcode()); // FIXME: stable encoding
453   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
454   Code = serialization::EXPR_UNARY_OPERATOR;
455 }
456
457 void ASTStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
458   VisitExpr(E);
459   Record.push_back(E->getNumComponents());
460   Record.push_back(E->getNumExpressions());
461   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
462   Writer.AddSourceLocation(E->getRParenLoc(), Record);
463   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
464   for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
465     const OffsetOfExpr::OffsetOfNode &ON = E->getComponent(I);
466     Record.push_back(ON.getKind()); // FIXME: Stable encoding
467     Writer.AddSourceLocation(ON.getSourceRange().getBegin(), Record);
468     Writer.AddSourceLocation(ON.getSourceRange().getEnd(), Record);
469     switch (ON.getKind()) {
470     case OffsetOfExpr::OffsetOfNode::Array:
471       Record.push_back(ON.getArrayExprIndex());
472       break;
473         
474     case OffsetOfExpr::OffsetOfNode::Field:
475       Writer.AddDeclRef(ON.getField(), Record);
476       break;
477         
478     case OffsetOfExpr::OffsetOfNode::Identifier:
479       Writer.AddIdentifierRef(ON.getFieldName(), Record);
480       break;
481         
482     case OffsetOfExpr::OffsetOfNode::Base:
483       Writer.AddCXXBaseSpecifier(*ON.getBase(), Record);
484       break;
485     }
486   }
487   for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
488     Writer.AddStmt(E->getIndexExpr(I));
489   Code = serialization::EXPR_OFFSETOF;
490 }
491
492 void ASTStmtWriter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E) {
493   VisitExpr(E);
494   Record.push_back(E->getKind());
495   if (E->isArgumentType())
496     Writer.AddTypeSourceInfo(E->getArgumentTypeInfo(), Record);
497   else {
498     Record.push_back(0);
499     Writer.AddStmt(E->getArgumentExpr());
500   }
501   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
502   Writer.AddSourceLocation(E->getRParenLoc(), Record);
503   Code = serialization::EXPR_SIZEOF_ALIGN_OF;
504 }
505
506 void ASTStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
507   VisitExpr(E);
508   Writer.AddStmt(E->getLHS());
509   Writer.AddStmt(E->getRHS());
510   Writer.AddSourceLocation(E->getRBracketLoc(), Record);
511   Code = serialization::EXPR_ARRAY_SUBSCRIPT;
512 }
513
514 void ASTStmtWriter::VisitCallExpr(CallExpr *E) {
515   VisitExpr(E);
516   Record.push_back(E->getNumArgs());
517   Writer.AddSourceLocation(E->getRParenLoc(), Record);
518   Writer.AddStmt(E->getCallee());
519   for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
520        Arg != ArgEnd; ++Arg)
521     Writer.AddStmt(*Arg);
522   Code = serialization::EXPR_CALL;
523 }
524
525 void ASTStmtWriter::VisitMemberExpr(MemberExpr *E) {
526   // Don't call VisitExpr, we'll write everything here.
527
528   Record.push_back(E->hasQualifier());
529   if (E->hasQualifier())
530     Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
531
532   Record.push_back(E->HasTemplateKWAndArgsInfo);
533   if (E->HasTemplateKWAndArgsInfo) {
534     Writer.AddSourceLocation(E->getTemplateKeywordLoc(), Record);
535     unsigned NumTemplateArgs = E->getNumTemplateArgs();
536     Record.push_back(NumTemplateArgs);
537     Writer.AddSourceLocation(E->getLAngleLoc(), Record);
538     Writer.AddSourceLocation(E->getRAngleLoc(), Record);
539     for (unsigned i=0; i != NumTemplateArgs; ++i)
540       Writer.AddTemplateArgumentLoc(E->getTemplateArgs()[i], Record);
541   }
542
543   Record.push_back(E->hadMultipleCandidates());
544
545   DeclAccessPair FoundDecl = E->getFoundDecl();
546   Writer.AddDeclRef(FoundDecl.getDecl(), Record);
547   Record.push_back(FoundDecl.getAccess());
548
549   Writer.AddTypeRef(E->getType(), Record);
550   Record.push_back(E->getValueKind());
551   Record.push_back(E->getObjectKind());
552   Writer.AddStmt(E->getBase());
553   Writer.AddDeclRef(E->getMemberDecl(), Record);
554   Writer.AddSourceLocation(E->getMemberLoc(), Record);
555   Record.push_back(E->isArrow());
556   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
557   Writer.AddDeclarationNameLoc(E->MemberDNLoc,
558                                E->getMemberDecl()->getDeclName(), Record);
559   Code = serialization::EXPR_MEMBER;
560 }
561
562 void ASTStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
563   VisitExpr(E);
564   Writer.AddStmt(E->getBase());
565   Writer.AddSourceLocation(E->getIsaMemberLoc(), Record);
566   Writer.AddSourceLocation(E->getOpLoc(), Record);
567   Record.push_back(E->isArrow());
568   Code = serialization::EXPR_OBJC_ISA;
569 }
570
571 void ASTStmtWriter::
572 VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) {
573   VisitExpr(E);
574   Writer.AddStmt(E->getSubExpr());
575   Record.push_back(E->shouldCopy());
576   Code = serialization::EXPR_OBJC_INDIRECT_COPY_RESTORE;
577 }
578
579 void ASTStmtWriter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) {
580   VisitExplicitCastExpr(E);
581   Writer.AddSourceLocation(E->getLParenLoc(), Record);
582   Writer.AddSourceLocation(E->getBridgeKeywordLoc(), Record);
583   Record.push_back(E->getBridgeKind()); // FIXME: Stable encoding
584   Code = serialization::EXPR_OBJC_BRIDGED_CAST;
585 }
586
587 void ASTStmtWriter::VisitCastExpr(CastExpr *E) {
588   VisitExpr(E);
589   Record.push_back(E->path_size());
590   Writer.AddStmt(E->getSubExpr());
591   Record.push_back(E->getCastKind()); // FIXME: stable encoding
592
593   for (CastExpr::path_iterator
594          PI = E->path_begin(), PE = E->path_end(); PI != PE; ++PI)
595     Writer.AddCXXBaseSpecifier(**PI, Record);
596 }
597
598 void ASTStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
599   VisitExpr(E);
600   Writer.AddStmt(E->getLHS());
601   Writer.AddStmt(E->getRHS());
602   Record.push_back(E->getOpcode()); // FIXME: stable encoding
603   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
604   Record.push_back(E->isFPContractable());
605   Code = serialization::EXPR_BINARY_OPERATOR;
606 }
607
608 void ASTStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
609   VisitBinaryOperator(E);
610   Writer.AddTypeRef(E->getComputationLHSType(), Record);
611   Writer.AddTypeRef(E->getComputationResultType(), Record);
612   Code = serialization::EXPR_COMPOUND_ASSIGN_OPERATOR;
613 }
614
615 void ASTStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
616   VisitExpr(E);
617   Writer.AddStmt(E->getCond());
618   Writer.AddStmt(E->getLHS());
619   Writer.AddStmt(E->getRHS());
620   Writer.AddSourceLocation(E->getQuestionLoc(), Record);
621   Writer.AddSourceLocation(E->getColonLoc(), Record);
622   Code = serialization::EXPR_CONDITIONAL_OPERATOR;
623 }
624
625 void
626 ASTStmtWriter::VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
627   VisitExpr(E);
628   Writer.AddStmt(E->getOpaqueValue());
629   Writer.AddStmt(E->getCommon());
630   Writer.AddStmt(E->getCond());
631   Writer.AddStmt(E->getTrueExpr());
632   Writer.AddStmt(E->getFalseExpr());
633   Writer.AddSourceLocation(E->getQuestionLoc(), Record);
634   Writer.AddSourceLocation(E->getColonLoc(), Record);
635   Code = serialization::EXPR_BINARY_CONDITIONAL_OPERATOR;
636 }
637
638 void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
639   VisitCastExpr(E);
640
641   if (E->path_size() == 0)
642     AbbrevToUse = Writer.getExprImplicitCastAbbrev();
643
644   Code = serialization::EXPR_IMPLICIT_CAST;
645 }
646
647 void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
648   VisitCastExpr(E);
649   Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
650 }
651
652 void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
653   VisitExplicitCastExpr(E);
654   Writer.AddSourceLocation(E->getLParenLoc(), Record);
655   Writer.AddSourceLocation(E->getRParenLoc(), Record);
656   Code = serialization::EXPR_CSTYLE_CAST;
657 }
658
659 void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
660   VisitExpr(E);
661   Writer.AddSourceLocation(E->getLParenLoc(), Record);
662   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
663   Writer.AddStmt(E->getInitializer());
664   Record.push_back(E->isFileScope());
665   Code = serialization::EXPR_COMPOUND_LITERAL;
666 }
667
668 void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
669   VisitExpr(E);
670   Writer.AddStmt(E->getBase());
671   Writer.AddIdentifierRef(&E->getAccessor(), Record);
672   Writer.AddSourceLocation(E->getAccessorLoc(), Record);
673   Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
674 }
675
676 void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
677   VisitExpr(E);
678   // NOTE: only add the (possibly null) syntactic form.
679   // No need to serialize the isSemanticForm flag and the semantic form.
680   Writer.AddStmt(E->getSyntacticForm());
681   Writer.AddSourceLocation(E->getLBraceLoc(), Record);
682   Writer.AddSourceLocation(E->getRBraceLoc(), Record);
683   bool isArrayFiller = E->ArrayFillerOrUnionFieldInit.is<Expr*>();
684   Record.push_back(isArrayFiller);
685   if (isArrayFiller)
686     Writer.AddStmt(E->getArrayFiller());
687   else
688     Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
689   Record.push_back(E->hadArrayRangeDesignator());
690   Record.push_back(E->getNumInits());
691   if (isArrayFiller) {
692     // ArrayFiller may have filled "holes" due to designated initializer.
693     // Replace them by 0 to indicate that the filler goes in that place.
694     Expr *filler = E->getArrayFiller();
695     for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
696       Writer.AddStmt(E->getInit(I) != filler ? E->getInit(I) : nullptr);
697   } else {
698     for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
699       Writer.AddStmt(E->getInit(I));
700   }
701   Code = serialization::EXPR_INIT_LIST;
702 }
703
704 void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
705   VisitExpr(E);
706   Record.push_back(E->getNumSubExprs());
707   for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
708     Writer.AddStmt(E->getSubExpr(I));
709   Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
710   Record.push_back(E->usesGNUSyntax());
711   for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
712                                              DEnd = E->designators_end();
713        D != DEnd; ++D) {
714     if (D->isFieldDesignator()) {
715       if (FieldDecl *Field = D->getField()) {
716         Record.push_back(serialization::DESIG_FIELD_DECL);
717         Writer.AddDeclRef(Field, Record);
718       } else {
719         Record.push_back(serialization::DESIG_FIELD_NAME);
720         Writer.AddIdentifierRef(D->getFieldName(), Record);
721       }
722       Writer.AddSourceLocation(D->getDotLoc(), Record);
723       Writer.AddSourceLocation(D->getFieldLoc(), Record);
724     } else if (D->isArrayDesignator()) {
725       Record.push_back(serialization::DESIG_ARRAY);
726       Record.push_back(D->getFirstExprIndex());
727       Writer.AddSourceLocation(D->getLBracketLoc(), Record);
728       Writer.AddSourceLocation(D->getRBracketLoc(), Record);
729     } else {
730       assert(D->isArrayRangeDesignator() && "Unknown designator");
731       Record.push_back(serialization::DESIG_ARRAY_RANGE);
732       Record.push_back(D->getFirstExprIndex());
733       Writer.AddSourceLocation(D->getLBracketLoc(), Record);
734       Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
735       Writer.AddSourceLocation(D->getRBracketLoc(), Record);
736     }
737   }
738   Code = serialization::EXPR_DESIGNATED_INIT;
739 }
740
741 void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
742   VisitExpr(E);
743   Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
744 }
745
746 void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
747   VisitExpr(E);
748   Writer.AddStmt(E->getSubExpr());
749   Writer.AddTypeSourceInfo(E->getWrittenTypeInfo(), Record);
750   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
751   Writer.AddSourceLocation(E->getRParenLoc(), Record);
752   Code = serialization::EXPR_VA_ARG;
753 }
754
755 void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
756   VisitExpr(E);
757   Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
758   Writer.AddSourceLocation(E->getLabelLoc(), Record);
759   Writer.AddDeclRef(E->getLabel(), Record);
760   Code = serialization::EXPR_ADDR_LABEL;
761 }
762
763 void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
764   VisitExpr(E);
765   Writer.AddStmt(E->getSubStmt());
766   Writer.AddSourceLocation(E->getLParenLoc(), Record);
767   Writer.AddSourceLocation(E->getRParenLoc(), Record);
768   Code = serialization::EXPR_STMT;
769 }
770
771 void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
772   VisitExpr(E);
773   Writer.AddStmt(E->getCond());
774   Writer.AddStmt(E->getLHS());
775   Writer.AddStmt(E->getRHS());
776   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
777   Writer.AddSourceLocation(E->getRParenLoc(), Record);
778   Record.push_back(E->isConditionDependent() ? false : E->isConditionTrue());
779   Code = serialization::EXPR_CHOOSE;
780 }
781
782 void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
783   VisitExpr(E);
784   Writer.AddSourceLocation(E->getTokenLocation(), Record);
785   Code = serialization::EXPR_GNU_NULL;
786 }
787
788 void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
789   VisitExpr(E);
790   Record.push_back(E->getNumSubExprs());
791   for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
792     Writer.AddStmt(E->getExpr(I));
793   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
794   Writer.AddSourceLocation(E->getRParenLoc(), Record);
795   Code = serialization::EXPR_SHUFFLE_VECTOR;
796 }
797
798 void ASTStmtWriter::VisitConvertVectorExpr(ConvertVectorExpr *E) {
799   VisitExpr(E);
800   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
801   Writer.AddSourceLocation(E->getRParenLoc(), Record);
802   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
803   Writer.AddStmt(E->getSrcExpr());
804   Code = serialization::EXPR_CONVERT_VECTOR;
805 }
806
807 void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
808   VisitExpr(E);
809   Writer.AddDeclRef(E->getBlockDecl(), Record);
810   Code = serialization::EXPR_BLOCK;
811 }
812
813 void ASTStmtWriter::VisitGenericSelectionExpr(GenericSelectionExpr *E) {
814   VisitExpr(E);
815   Record.push_back(E->getNumAssocs());
816
817   Writer.AddStmt(E->getControllingExpr());
818   for (unsigned I = 0, N = E->getNumAssocs(); I != N; ++I) {
819     Writer.AddTypeSourceInfo(E->getAssocTypeSourceInfo(I), Record);
820     Writer.AddStmt(E->getAssocExpr(I));
821   }
822   Record.push_back(E->isResultDependent() ? -1U : E->getResultIndex());
823
824   Writer.AddSourceLocation(E->getGenericLoc(), Record);
825   Writer.AddSourceLocation(E->getDefaultLoc(), Record);
826   Writer.AddSourceLocation(E->getRParenLoc(), Record);
827   Code = serialization::EXPR_GENERIC_SELECTION;
828 }
829
830 void ASTStmtWriter::VisitPseudoObjectExpr(PseudoObjectExpr *E) {
831   VisitExpr(E);
832   Record.push_back(E->getNumSemanticExprs());
833
834   // Push the result index.  Currently, this needs to exactly match
835   // the encoding used internally for ResultIndex.
836   unsigned result = E->getResultExprIndex();
837   result = (result == PseudoObjectExpr::NoResult ? 0 : result + 1);
838   Record.push_back(result);
839
840   Writer.AddStmt(E->getSyntacticForm());
841   for (PseudoObjectExpr::semantics_iterator
842          i = E->semantics_begin(), e = E->semantics_end(); i != e; ++i) {
843     Writer.AddStmt(*i);
844   }
845   Code = serialization::EXPR_PSEUDO_OBJECT;
846 }
847
848 void ASTStmtWriter::VisitAtomicExpr(AtomicExpr *E) {
849   VisitExpr(E);
850   Record.push_back(E->getOp());
851   for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
852     Writer.AddStmt(E->getSubExprs()[I]);
853   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
854   Writer.AddSourceLocation(E->getRParenLoc(), Record);
855   Code = serialization::EXPR_ATOMIC;
856 }
857
858 //===----------------------------------------------------------------------===//
859 // Objective-C Expressions and Statements.
860 //===----------------------------------------------------------------------===//
861
862 void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
863   VisitExpr(E);
864   Writer.AddStmt(E->getString());
865   Writer.AddSourceLocation(E->getAtLoc(), Record);
866   Code = serialization::EXPR_OBJC_STRING_LITERAL;
867 }
868
869 void ASTStmtWriter::VisitObjCBoxedExpr(ObjCBoxedExpr *E) {
870   VisitExpr(E);
871   Writer.AddStmt(E->getSubExpr());
872   Writer.AddDeclRef(E->getBoxingMethod(), Record);
873   Writer.AddSourceRange(E->getSourceRange(), Record);
874   Code = serialization::EXPR_OBJC_BOXED_EXPRESSION;
875 }
876
877 void ASTStmtWriter::VisitObjCArrayLiteral(ObjCArrayLiteral *E) {
878   VisitExpr(E);
879   Record.push_back(E->getNumElements());
880   for (unsigned i = 0; i < E->getNumElements(); i++)
881     Writer.AddStmt(E->getElement(i));
882   Writer.AddDeclRef(E->getArrayWithObjectsMethod(), Record);
883   Writer.AddSourceRange(E->getSourceRange(), Record);
884   Code = serialization::EXPR_OBJC_ARRAY_LITERAL;
885 }
886
887 void ASTStmtWriter::VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) {
888   VisitExpr(E);
889   Record.push_back(E->getNumElements());
890   Record.push_back(E->HasPackExpansions);
891   for (unsigned i = 0; i < E->getNumElements(); i++) {
892     ObjCDictionaryElement Element = E->getKeyValueElement(i);
893     Writer.AddStmt(Element.Key);
894     Writer.AddStmt(Element.Value);
895     if (E->HasPackExpansions) {
896       Writer.AddSourceLocation(Element.EllipsisLoc, Record);
897       unsigned NumExpansions = 0;
898       if (Element.NumExpansions)
899         NumExpansions = *Element.NumExpansions + 1;
900       Record.push_back(NumExpansions);
901     }
902   }
903     
904   Writer.AddDeclRef(E->getDictWithObjectsMethod(), Record);
905   Writer.AddSourceRange(E->getSourceRange(), Record);
906   Code = serialization::EXPR_OBJC_DICTIONARY_LITERAL;
907 }
908
909 void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
910   VisitExpr(E);
911   Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
912   Writer.AddSourceLocation(E->getAtLoc(), Record);
913   Writer.AddSourceLocation(E->getRParenLoc(), Record);
914   Code = serialization::EXPR_OBJC_ENCODE;
915 }
916
917 void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
918   VisitExpr(E);
919   Writer.AddSelectorRef(E->getSelector(), Record);
920   Writer.AddSourceLocation(E->getAtLoc(), Record);
921   Writer.AddSourceLocation(E->getRParenLoc(), Record);
922   Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
923 }
924
925 void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
926   VisitExpr(E);
927   Writer.AddDeclRef(E->getProtocol(), Record);
928   Writer.AddSourceLocation(E->getAtLoc(), Record);
929   Writer.AddSourceLocation(E->ProtoLoc, Record);
930   Writer.AddSourceLocation(E->getRParenLoc(), Record);
931   Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
932 }
933
934 void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
935   VisitExpr(E);
936   Writer.AddDeclRef(E->getDecl(), Record);
937   Writer.AddSourceLocation(E->getLocation(), Record);
938   Writer.AddSourceLocation(E->getOpLoc(), Record);
939   Writer.AddStmt(E->getBase());
940   Record.push_back(E->isArrow());
941   Record.push_back(E->isFreeIvar());
942   Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
943 }
944
945 void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
946   VisitExpr(E);
947   Record.push_back(E->SetterAndMethodRefFlags.getInt());
948   Record.push_back(E->isImplicitProperty());
949   if (E->isImplicitProperty()) {
950     Writer.AddDeclRef(E->getImplicitPropertyGetter(), Record);
951     Writer.AddDeclRef(E->getImplicitPropertySetter(), Record);
952   } else {
953     Writer.AddDeclRef(E->getExplicitProperty(), Record);
954   }
955   Writer.AddSourceLocation(E->getLocation(), Record);
956   Writer.AddSourceLocation(E->getReceiverLocation(), Record);
957   if (E->isObjectReceiver()) {
958     Record.push_back(0);
959     Writer.AddStmt(E->getBase());
960   } else if (E->isSuperReceiver()) {
961     Record.push_back(1);
962     Writer.AddTypeRef(E->getSuperReceiverType(), Record);
963   } else {
964     Record.push_back(2);
965     Writer.AddDeclRef(E->getClassReceiver(), Record);
966   }
967   
968   Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
969 }
970
971 void ASTStmtWriter::VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr *E) {
972   VisitExpr(E);
973   Writer.AddSourceLocation(E->getRBracket(), Record);
974   Writer.AddStmt(E->getBaseExpr());
975   Writer.AddStmt(E->getKeyExpr());
976   Writer.AddDeclRef(E->getAtIndexMethodDecl(), Record);
977   Writer.AddDeclRef(E->setAtIndexMethodDecl(), Record);
978   
979   Code = serialization::EXPR_OBJC_SUBSCRIPT_REF_EXPR;
980 }
981
982 void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
983   VisitExpr(E);
984   Record.push_back(E->getNumArgs());
985   Record.push_back(E->getNumStoredSelLocs());
986   Record.push_back(E->SelLocsKind);
987   Record.push_back(E->isDelegateInitCall());
988   Record.push_back(E->IsImplicit);
989   Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
990   switch (E->getReceiverKind()) {
991   case ObjCMessageExpr::Instance:
992     Writer.AddStmt(E->getInstanceReceiver());
993     break;
994
995   case ObjCMessageExpr::Class:
996     Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
997     break;
998
999   case ObjCMessageExpr::SuperClass:
1000   case ObjCMessageExpr::SuperInstance:
1001     Writer.AddTypeRef(E->getSuperType(), Record);
1002     Writer.AddSourceLocation(E->getSuperLoc(), Record);
1003     break;
1004   }
1005
1006   if (E->getMethodDecl()) {
1007     Record.push_back(1);
1008     Writer.AddDeclRef(E->getMethodDecl(), Record);
1009   } else {
1010     Record.push_back(0);
1011     Writer.AddSelectorRef(E->getSelector(), Record);    
1012   }
1013     
1014   Writer.AddSourceLocation(E->getLeftLoc(), Record);
1015   Writer.AddSourceLocation(E->getRightLoc(), Record);
1016
1017   for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
1018        Arg != ArgEnd; ++Arg)
1019     Writer.AddStmt(*Arg);
1020
1021   SourceLocation *Locs = E->getStoredSelLocs();
1022   for (unsigned i = 0, e = E->getNumStoredSelLocs(); i != e; ++i)
1023     Writer.AddSourceLocation(Locs[i], Record);
1024
1025   Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
1026 }
1027
1028 void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
1029   VisitStmt(S);
1030   Writer.AddStmt(S->getElement());
1031   Writer.AddStmt(S->getCollection());
1032   Writer.AddStmt(S->getBody());
1033   Writer.AddSourceLocation(S->getForLoc(), Record);
1034   Writer.AddSourceLocation(S->getRParenLoc(), Record);
1035   Code = serialization::STMT_OBJC_FOR_COLLECTION;
1036 }
1037
1038 void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
1039   Writer.AddStmt(S->getCatchBody());
1040   Writer.AddDeclRef(S->getCatchParamDecl(), Record);
1041   Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
1042   Writer.AddSourceLocation(S->getRParenLoc(), Record);
1043   Code = serialization::STMT_OBJC_CATCH;
1044 }
1045
1046 void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
1047   Writer.AddStmt(S->getFinallyBody());
1048   Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
1049   Code = serialization::STMT_OBJC_FINALLY;
1050 }
1051
1052 void ASTStmtWriter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *S) {
1053   Writer.AddStmt(S->getSubStmt());
1054   Writer.AddSourceLocation(S->getAtLoc(), Record);
1055   Code = serialization::STMT_OBJC_AUTORELEASE_POOL;
1056 }
1057
1058 void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
1059   Record.push_back(S->getNumCatchStmts());
1060   Record.push_back(S->getFinallyStmt() != nullptr);
1061   Writer.AddStmt(S->getTryBody());
1062   for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
1063     Writer.AddStmt(S->getCatchStmt(I));
1064   if (S->getFinallyStmt())
1065     Writer.AddStmt(S->getFinallyStmt());
1066   Writer.AddSourceLocation(S->getAtTryLoc(), Record);
1067   Code = serialization::STMT_OBJC_AT_TRY;
1068 }
1069
1070 void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
1071   Writer.AddStmt(S->getSynchExpr());
1072   Writer.AddStmt(S->getSynchBody());
1073   Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
1074   Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
1075 }
1076
1077 void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
1078   Writer.AddStmt(S->getThrowExpr());
1079   Writer.AddSourceLocation(S->getThrowLoc(), Record);
1080   Code = serialization::STMT_OBJC_AT_THROW;
1081 }
1082
1083 void ASTStmtWriter::VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr *E) {
1084   VisitExpr(E);
1085   Record.push_back(E->getValue());
1086   Writer.AddSourceLocation(E->getLocation(), Record);
1087   Code = serialization::EXPR_OBJC_BOOL_LITERAL;
1088 }
1089
1090 //===----------------------------------------------------------------------===//
1091 // C++ Expressions and Statements.
1092 //===----------------------------------------------------------------------===//
1093
1094 void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
1095   VisitStmt(S);
1096   Writer.AddSourceLocation(S->getCatchLoc(), Record);
1097   Writer.AddDeclRef(S->getExceptionDecl(), Record);
1098   Writer.AddStmt(S->getHandlerBlock());
1099   Code = serialization::STMT_CXX_CATCH;
1100 }
1101
1102 void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
1103   VisitStmt(S);
1104   Record.push_back(S->getNumHandlers());
1105   Writer.AddSourceLocation(S->getTryLoc(), Record);
1106   Writer.AddStmt(S->getTryBlock());
1107   for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
1108     Writer.AddStmt(S->getHandler(i));
1109   Code = serialization::STMT_CXX_TRY;
1110 }
1111
1112 void ASTStmtWriter::VisitCXXForRangeStmt(CXXForRangeStmt *S) {
1113   VisitStmt(S);
1114   Writer.AddSourceLocation(S->getForLoc(), Record);
1115   Writer.AddSourceLocation(S->getColonLoc(), Record);
1116   Writer.AddSourceLocation(S->getRParenLoc(), Record);
1117   Writer.AddStmt(S->getRangeStmt());
1118   Writer.AddStmt(S->getBeginEndStmt());
1119   Writer.AddStmt(S->getCond());
1120   Writer.AddStmt(S->getInc());
1121   Writer.AddStmt(S->getLoopVarStmt());
1122   Writer.AddStmt(S->getBody());
1123   Code = serialization::STMT_CXX_FOR_RANGE;
1124 }
1125
1126 void ASTStmtWriter::VisitMSDependentExistsStmt(MSDependentExistsStmt *S) {
1127   VisitStmt(S);
1128   Writer.AddSourceLocation(S->getKeywordLoc(), Record);
1129   Record.push_back(S->isIfExists());
1130   Writer.AddNestedNameSpecifierLoc(S->getQualifierLoc(), Record);
1131   Writer.AddDeclarationNameInfo(S->getNameInfo(), Record);
1132   Writer.AddStmt(S->getSubStmt());
1133   Code = serialization::STMT_MS_DEPENDENT_EXISTS;
1134 }
1135
1136 void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
1137   VisitCallExpr(E);
1138   Record.push_back(E->getOperator());
1139   Writer.AddSourceRange(E->Range, Record);
1140   Record.push_back(E->isFPContractable());
1141   Code = serialization::EXPR_CXX_OPERATOR_CALL;
1142 }
1143
1144 void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
1145   VisitCallExpr(E);
1146   Code = serialization::EXPR_CXX_MEMBER_CALL;
1147 }
1148
1149 void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
1150   VisitExpr(E);
1151   Record.push_back(E->getNumArgs());
1152   for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
1153     Writer.AddStmt(E->getArg(I));
1154   Writer.AddDeclRef(E->getConstructor(), Record);
1155   Writer.AddSourceLocation(E->getLocation(), Record);
1156   Record.push_back(E->isElidable());
1157   Record.push_back(E->hadMultipleCandidates());
1158   Record.push_back(E->isListInitialization());
1159   Record.push_back(E->isStdInitListInitialization());
1160   Record.push_back(E->requiresZeroInitialization());
1161   Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
1162   Writer.AddSourceRange(E->getParenOrBraceRange(), Record);
1163   Code = serialization::EXPR_CXX_CONSTRUCT;
1164 }
1165
1166 void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
1167   VisitCXXConstructExpr(E);
1168   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
1169   Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
1170 }
1171
1172 void ASTStmtWriter::VisitLambdaExpr(LambdaExpr *E) {
1173   VisitExpr(E);
1174   Record.push_back(E->NumCaptures);
1175   unsigned NumArrayIndexVars = 0;
1176   if (E->HasArrayIndexVars)
1177     NumArrayIndexVars = E->getArrayIndexStarts()[E->NumCaptures];
1178   Record.push_back(NumArrayIndexVars);
1179   Writer.AddSourceRange(E->IntroducerRange, Record);
1180   Record.push_back(E->CaptureDefault); // FIXME: stable encoding
1181   Writer.AddSourceLocation(E->CaptureDefaultLoc, Record);
1182   Record.push_back(E->ExplicitParams);
1183   Record.push_back(E->ExplicitResultType);
1184   Writer.AddSourceLocation(E->ClosingBrace, Record);
1185   
1186   // Add capture initializers.
1187   for (LambdaExpr::capture_init_iterator C = E->capture_init_begin(),
1188                                       CEnd = E->capture_init_end();
1189        C != CEnd; ++C) {
1190     Writer.AddStmt(*C);
1191   }
1192   
1193   // Add array index variables, if any.
1194   if (NumArrayIndexVars) {
1195     Record.append(E->getArrayIndexStarts(), 
1196                   E->getArrayIndexStarts() + E->NumCaptures + 1);
1197     VarDecl **ArrayIndexVars = E->getArrayIndexVars();
1198     for (unsigned I = 0; I != NumArrayIndexVars; ++I)
1199       Writer.AddDeclRef(ArrayIndexVars[I], Record);
1200   }
1201   
1202   Code = serialization::EXPR_LAMBDA;
1203 }
1204
1205 void ASTStmtWriter::VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E) {
1206   VisitExpr(E);
1207   Writer.AddStmt(E->getSubExpr());
1208   Code = serialization::EXPR_CXX_STD_INITIALIZER_LIST;
1209 }
1210
1211 void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
1212   VisitExplicitCastExpr(E);
1213   Writer.AddSourceRange(SourceRange(E->getOperatorLoc(), E->getRParenLoc()),
1214                         Record);
1215   Writer.AddSourceRange(E->getAngleBrackets(), Record);
1216 }
1217
1218 void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
1219   VisitCXXNamedCastExpr(E);
1220   Code = serialization::EXPR_CXX_STATIC_CAST;
1221 }
1222
1223 void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
1224   VisitCXXNamedCastExpr(E);
1225   Code = serialization::EXPR_CXX_DYNAMIC_CAST;
1226 }
1227
1228 void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
1229   VisitCXXNamedCastExpr(E);
1230   Code = serialization::EXPR_CXX_REINTERPRET_CAST;
1231 }
1232
1233 void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
1234   VisitCXXNamedCastExpr(E);
1235   Code = serialization::EXPR_CXX_CONST_CAST;
1236 }
1237
1238 void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
1239   VisitExplicitCastExpr(E);
1240   Writer.AddSourceLocation(E->getLParenLoc(), Record);
1241   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1242   Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
1243 }
1244
1245 void ASTStmtWriter::VisitUserDefinedLiteral(UserDefinedLiteral *E) {
1246   VisitCallExpr(E);
1247   Writer.AddSourceLocation(E->UDSuffixLoc, Record);
1248   Code = serialization::EXPR_USER_DEFINED_LITERAL;
1249 }
1250
1251 void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
1252   VisitExpr(E);
1253   Record.push_back(E->getValue());
1254   Writer.AddSourceLocation(E->getLocation(), Record);
1255   Code = serialization::EXPR_CXX_BOOL_LITERAL;
1256 }
1257
1258 void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
1259   VisitExpr(E);
1260   Writer.AddSourceLocation(E->getLocation(), Record);
1261   Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
1262 }
1263
1264 void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1265   VisitExpr(E);
1266   Writer.AddSourceRange(E->getSourceRange(), Record);
1267   if (E->isTypeOperand()) {
1268     Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1269     Code = serialization::EXPR_CXX_TYPEID_TYPE;
1270   } else {
1271     Writer.AddStmt(E->getExprOperand());
1272     Code = serialization::EXPR_CXX_TYPEID_EXPR;
1273   }
1274 }
1275
1276 void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
1277   VisitExpr(E);
1278   Writer.AddSourceLocation(E->getLocation(), Record);
1279   Record.push_back(E->isImplicit());
1280   Code = serialization::EXPR_CXX_THIS;
1281 }
1282
1283 void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
1284   VisitExpr(E);
1285   Writer.AddSourceLocation(E->getThrowLoc(), Record);
1286   Writer.AddStmt(E->getSubExpr());
1287   Record.push_back(E->isThrownVariableInScope());
1288   Code = serialization::EXPR_CXX_THROW;
1289 }
1290
1291 void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
1292   VisitExpr(E);
1293
1294   bool HasOtherExprStored = E->Param.getInt();
1295   // Store these first, the reader reads them before creation.
1296   Record.push_back(HasOtherExprStored);
1297   if (HasOtherExprStored)
1298     Writer.AddStmt(E->getExpr());
1299   Writer.AddDeclRef(E->getParam(), Record);
1300   Writer.AddSourceLocation(E->getUsedLocation(), Record);
1301
1302   Code = serialization::EXPR_CXX_DEFAULT_ARG;
1303 }
1304
1305 void ASTStmtWriter::VisitCXXDefaultInitExpr(CXXDefaultInitExpr *E) {
1306   VisitExpr(E);
1307   Writer.AddDeclRef(E->getField(), Record);
1308   Writer.AddSourceLocation(E->getExprLoc(), Record);
1309   Code = serialization::EXPR_CXX_DEFAULT_INIT;
1310 }
1311
1312 void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
1313   VisitExpr(E);
1314   Writer.AddCXXTemporary(E->getTemporary(), Record);
1315   Writer.AddStmt(E->getSubExpr());
1316   Code = serialization::EXPR_CXX_BIND_TEMPORARY;
1317 }
1318
1319 void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
1320   VisitExpr(E);
1321   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
1322   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1323   Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
1324 }
1325
1326 void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
1327   VisitExpr(E);
1328   Record.push_back(E->isGlobalNew());
1329   Record.push_back(E->isArray());
1330   Record.push_back(E->doesUsualArrayDeleteWantSize());
1331   Record.push_back(E->getNumPlacementArgs());
1332   Record.push_back(E->StoredInitializationStyle);
1333   Writer.AddDeclRef(E->getOperatorNew(), Record);
1334   Writer.AddDeclRef(E->getOperatorDelete(), Record);
1335   Writer.AddTypeSourceInfo(E->getAllocatedTypeSourceInfo(), Record);
1336   Writer.AddSourceRange(E->getTypeIdParens(), Record);
1337   Writer.AddSourceRange(E->getSourceRange(), Record);
1338   Writer.AddSourceRange(E->getDirectInitRange(), Record);
1339   for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
1340        I != e; ++I)
1341     Writer.AddStmt(*I);
1342
1343   Code = serialization::EXPR_CXX_NEW;
1344 }
1345
1346 void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
1347   VisitExpr(E);
1348   Record.push_back(E->isGlobalDelete());
1349   Record.push_back(E->isArrayForm());
1350   Record.push_back(E->isArrayFormAsWritten());
1351   Record.push_back(E->doesUsualArrayDeleteWantSize());
1352   Writer.AddDeclRef(E->getOperatorDelete(), Record);
1353   Writer.AddStmt(E->getArgument());
1354   Writer.AddSourceLocation(E->getSourceRange().getBegin(), Record);
1355   
1356   Code = serialization::EXPR_CXX_DELETE;
1357 }
1358
1359 void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1360   VisitExpr(E);
1361
1362   Writer.AddStmt(E->getBase());
1363   Record.push_back(E->isArrow());
1364   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1365   Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1366   Writer.AddTypeSourceInfo(E->getScopeTypeInfo(), Record);
1367   Writer.AddSourceLocation(E->getColonColonLoc(), Record);
1368   Writer.AddSourceLocation(E->getTildeLoc(), Record);
1369
1370   // PseudoDestructorTypeStorage.
1371   Writer.AddIdentifierRef(E->getDestroyedTypeIdentifier(), Record);
1372   if (E->getDestroyedTypeIdentifier())
1373     Writer.AddSourceLocation(E->getDestroyedTypeLoc(), Record);
1374   else
1375     Writer.AddTypeSourceInfo(E->getDestroyedTypeInfo(), Record);
1376
1377   Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
1378 }
1379
1380 void ASTStmtWriter::VisitExprWithCleanups(ExprWithCleanups *E) {
1381   VisitExpr(E);
1382   Record.push_back(E->getNumObjects());
1383   for (unsigned i = 0, e = E->getNumObjects(); i != e; ++i)
1384     Writer.AddDeclRef(E->getObject(i), Record);
1385   
1386   Writer.AddStmt(E->getSubExpr());
1387   Code = serialization::EXPR_EXPR_WITH_CLEANUPS;
1388 }
1389
1390 void
1391 ASTStmtWriter::VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E){
1392   VisitExpr(E);
1393
1394   // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1395   // emitted first.
1396
1397   Record.push_back(E->HasTemplateKWAndArgsInfo);
1398   if (E->HasTemplateKWAndArgsInfo) {
1399     const ASTTemplateKWAndArgsInfo &Args = *E->getTemplateKWAndArgsInfo();
1400     Record.push_back(Args.NumTemplateArgs);
1401     AddTemplateKWAndArgsInfo(Args);
1402   }
1403
1404   if (!E->isImplicitAccess())
1405     Writer.AddStmt(E->getBase());
1406   else
1407     Writer.AddStmt(nullptr);
1408   Writer.AddTypeRef(E->getBaseType(), Record);
1409   Record.push_back(E->isArrow());
1410   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1411   Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1412   Writer.AddDeclRef(E->getFirstQualifierFoundInScope(), Record);
1413   Writer.AddDeclarationNameInfo(E->MemberNameInfo, Record);
1414   Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
1415 }
1416
1417 void
1418 ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1419   VisitExpr(E);
1420
1421   // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1422   // emitted first.
1423
1424   Record.push_back(E->HasTemplateKWAndArgsInfo);
1425   if (E->HasTemplateKWAndArgsInfo) {
1426     const ASTTemplateKWAndArgsInfo &Args = *E->getTemplateKWAndArgsInfo();
1427     Record.push_back(Args.NumTemplateArgs);
1428     AddTemplateKWAndArgsInfo(Args);
1429   }
1430
1431   Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1432   Writer.AddDeclarationNameInfo(E->NameInfo, Record);
1433   Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
1434 }
1435
1436 void
1437 ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
1438   VisitExpr(E);
1439   Record.push_back(E->arg_size());
1440   for (CXXUnresolvedConstructExpr::arg_iterator
1441          ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
1442     Writer.AddStmt(*ArgI);
1443   Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
1444   Writer.AddSourceLocation(E->getLParenLoc(), Record);
1445   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1446   Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
1447 }
1448
1449 void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
1450   VisitExpr(E);
1451
1452   // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1453   // emitted first.
1454
1455   Record.push_back(E->HasTemplateKWAndArgsInfo);
1456   if (E->HasTemplateKWAndArgsInfo) {
1457     const ASTTemplateKWAndArgsInfo &Args = *E->getTemplateKWAndArgsInfo();
1458     Record.push_back(Args.NumTemplateArgs);
1459     AddTemplateKWAndArgsInfo(Args);
1460   }
1461
1462   Record.push_back(E->getNumDecls());
1463   for (OverloadExpr::decls_iterator
1464          OvI = E->decls_begin(), OvE = E->decls_end(); OvI != OvE; ++OvI) {
1465     Writer.AddDeclRef(OvI.getDecl(), Record);
1466     Record.push_back(OvI.getAccess());
1467   }
1468
1469   Writer.AddDeclarationNameInfo(E->NameInfo, Record);
1470   Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1471 }
1472
1473 void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
1474   VisitOverloadExpr(E);
1475   Record.push_back(E->isArrow());
1476   Record.push_back(E->hasUnresolvedUsing());
1477   Writer.AddStmt(!E->isImplicitAccess() ? E->getBase() : nullptr);
1478   Writer.AddTypeRef(E->getBaseType(), Record);
1479   Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1480   Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
1481 }
1482
1483 void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
1484   VisitOverloadExpr(E);
1485   Record.push_back(E->requiresADL());
1486   Record.push_back(E->isOverloaded());
1487   Writer.AddDeclRef(E->getNamingClass(), Record);
1488   Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
1489 }
1490
1491 void ASTStmtWriter::VisitTypeTraitExpr(TypeTraitExpr *E) {
1492   VisitExpr(E);
1493   Record.push_back(E->TypeTraitExprBits.NumArgs);
1494   Record.push_back(E->TypeTraitExprBits.Kind); // FIXME: Stable encoding
1495   Record.push_back(E->TypeTraitExprBits.Value);
1496   Writer.AddSourceRange(E->getSourceRange(), Record);
1497   for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
1498     Writer.AddTypeSourceInfo(E->getArg(I), Record);
1499   Code = serialization::EXPR_TYPE_TRAIT;
1500 }
1501
1502 void ASTStmtWriter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
1503   VisitExpr(E);
1504   Record.push_back(E->getTrait());
1505   Record.push_back(E->getValue());
1506   Writer.AddSourceRange(E->getSourceRange(), Record);
1507   Writer.AddTypeSourceInfo(E->getQueriedTypeSourceInfo(), Record);
1508   Code = serialization::EXPR_ARRAY_TYPE_TRAIT;
1509 }
1510
1511 void ASTStmtWriter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
1512   VisitExpr(E);
1513   Record.push_back(E->getTrait());
1514   Record.push_back(E->getValue());
1515   Writer.AddSourceRange(E->getSourceRange(), Record);
1516   Writer.AddStmt(E->getQueriedExpression());
1517   Code = serialization::EXPR_CXX_EXPRESSION_TRAIT;
1518 }
1519
1520 void ASTStmtWriter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
1521   VisitExpr(E);
1522   Record.push_back(E->getValue());
1523   Writer.AddSourceRange(E->getSourceRange(), Record);
1524   Writer.AddStmt(E->getOperand());
1525   Code = serialization::EXPR_CXX_NOEXCEPT;
1526 }
1527
1528 void ASTStmtWriter::VisitPackExpansionExpr(PackExpansionExpr *E) {
1529   VisitExpr(E);
1530   Writer.AddSourceLocation(E->getEllipsisLoc(), Record);
1531   Record.push_back(E->NumExpansions);
1532   Writer.AddStmt(E->getPattern());
1533   Code = serialization::EXPR_PACK_EXPANSION;
1534 }
1535
1536 void ASTStmtWriter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
1537   VisitExpr(E);
1538   Writer.AddSourceLocation(E->OperatorLoc, Record);
1539   Writer.AddSourceLocation(E->PackLoc, Record);
1540   Writer.AddSourceLocation(E->RParenLoc, Record);
1541   Record.push_back(E->Length);
1542   Writer.AddDeclRef(E->Pack, Record);
1543   Code = serialization::EXPR_SIZEOF_PACK;
1544 }
1545
1546 void ASTStmtWriter::VisitSubstNonTypeTemplateParmExpr(
1547                                               SubstNonTypeTemplateParmExpr *E) {
1548   VisitExpr(E);
1549   Writer.AddDeclRef(E->getParameter(), Record);
1550   Writer.AddSourceLocation(E->getNameLoc(), Record);
1551   Writer.AddStmt(E->getReplacement());
1552   Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM;
1553 }
1554
1555 void ASTStmtWriter::VisitSubstNonTypeTemplateParmPackExpr(
1556                                           SubstNonTypeTemplateParmPackExpr *E) {
1557   VisitExpr(E);
1558   Writer.AddDeclRef(E->getParameterPack(), Record);
1559   Writer.AddTemplateArgument(E->getArgumentPack(), Record);
1560   Writer.AddSourceLocation(E->getParameterPackLocation(), Record);
1561   Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM_PACK;
1562 }
1563
1564 void ASTStmtWriter::VisitFunctionParmPackExpr(FunctionParmPackExpr *E) {
1565   VisitExpr(E);
1566   Record.push_back(E->getNumExpansions());
1567   Writer.AddDeclRef(E->getParameterPack(), Record);
1568   Writer.AddSourceLocation(E->getParameterPackLocation(), Record);
1569   for (FunctionParmPackExpr::iterator I = E->begin(), End = E->end();
1570        I != End; ++I)
1571     Writer.AddDeclRef(*I, Record);
1572   Code = serialization::EXPR_FUNCTION_PARM_PACK;
1573 }
1574
1575 void ASTStmtWriter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E) {
1576   VisitExpr(E);
1577   Writer.AddStmt(E->getTemporary());
1578   Writer.AddDeclRef(E->getExtendingDecl(), Record);
1579   Record.push_back(E->getManglingNumber());
1580   Code = serialization::EXPR_MATERIALIZE_TEMPORARY;
1581 }
1582
1583 void ASTStmtWriter::VisitCXXFoldExpr(CXXFoldExpr *E) {
1584   VisitExpr(E);
1585   Writer.AddSourceLocation(E->LParenLoc, Record);
1586   Writer.AddSourceLocation(E->EllipsisLoc, Record);
1587   Writer.AddSourceLocation(E->RParenLoc, Record);
1588   Writer.AddStmt(E->SubExprs[0]);
1589   Writer.AddStmt(E->SubExprs[1]);
1590   Record.push_back(E->Opcode);
1591   Code = serialization::EXPR_CXX_FOLD;
1592 }
1593
1594 void ASTStmtWriter::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
1595   VisitExpr(E);
1596   Writer.AddStmt(E->getSourceExpr());
1597   Writer.AddSourceLocation(E->getLocation(), Record);
1598   Code = serialization::EXPR_OPAQUE_VALUE;
1599 }
1600
1601 void ASTStmtWriter::VisitTypoExpr(TypoExpr *E) {
1602   VisitExpr(E);
1603   // TODO: Figure out sane writer behavior for a TypoExpr, if necessary
1604   assert(false && "Cannot write TypoExpr nodes");
1605 }
1606
1607 //===----------------------------------------------------------------------===//
1608 // CUDA Expressions and Statements.
1609 //===----------------------------------------------------------------------===//
1610
1611 void ASTStmtWriter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *E) {
1612   VisitCallExpr(E);
1613   Writer.AddStmt(E->getConfig());
1614   Code = serialization::EXPR_CUDA_KERNEL_CALL;
1615 }
1616
1617 //===----------------------------------------------------------------------===//
1618 // OpenCL Expressions and Statements.
1619 //===----------------------------------------------------------------------===//
1620 void ASTStmtWriter::VisitAsTypeExpr(AsTypeExpr *E) {
1621   VisitExpr(E);
1622   Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
1623   Writer.AddSourceLocation(E->getRParenLoc(), Record);
1624   Writer.AddStmt(E->getSrcExpr());
1625   Code = serialization::EXPR_ASTYPE;
1626 }
1627
1628 //===----------------------------------------------------------------------===//
1629 // Microsoft Expressions and Statements.
1630 //===----------------------------------------------------------------------===//
1631 void ASTStmtWriter::VisitMSPropertyRefExpr(MSPropertyRefExpr *E) {
1632   VisitExpr(E);
1633   Record.push_back(E->isArrow());
1634   Writer.AddStmt(E->getBaseExpr());
1635   Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1636   Writer.AddSourceLocation(E->getMemberLoc(), Record);
1637   Writer.AddDeclRef(E->getPropertyDecl(), Record);
1638   Code = serialization::EXPR_CXX_PROPERTY_REF_EXPR;
1639 }
1640
1641 void ASTStmtWriter::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
1642   VisitExpr(E);
1643   Writer.AddSourceRange(E->getSourceRange(), Record);
1644   if (E->isTypeOperand()) {
1645     Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1646     Code = serialization::EXPR_CXX_UUIDOF_TYPE;
1647   } else {
1648     Writer.AddStmt(E->getExprOperand());
1649     Code = serialization::EXPR_CXX_UUIDOF_EXPR;
1650   }
1651 }
1652
1653 void ASTStmtWriter::VisitSEHExceptStmt(SEHExceptStmt *S) {
1654   VisitStmt(S);
1655   Writer.AddSourceLocation(S->getExceptLoc(), Record);
1656   Writer.AddStmt(S->getFilterExpr());
1657   Writer.AddStmt(S->getBlock());
1658   Code = serialization::STMT_SEH_EXCEPT;
1659 }
1660
1661 void ASTStmtWriter::VisitSEHFinallyStmt(SEHFinallyStmt *S) {
1662   VisitStmt(S);
1663   Writer.AddSourceLocation(S->getFinallyLoc(), Record);
1664   Writer.AddStmt(S->getBlock());
1665   Code = serialization::STMT_SEH_FINALLY;
1666 }
1667
1668 void ASTStmtWriter::VisitSEHTryStmt(SEHTryStmt *S) {
1669   VisitStmt(S);
1670   Record.push_back(S->getIsCXXTry());
1671   Writer.AddSourceLocation(S->getTryLoc(), Record);
1672   Writer.AddStmt(S->getTryBlock());
1673   Writer.AddStmt(S->getHandler());
1674   Code = serialization::STMT_SEH_TRY;
1675 }
1676
1677 void ASTStmtWriter::VisitSEHLeaveStmt(SEHLeaveStmt *S) {
1678   VisitStmt(S);
1679   Writer.AddSourceLocation(S->getLeaveLoc(), Record);
1680   Code = serialization::STMT_SEH_LEAVE;
1681 }
1682
1683 //===----------------------------------------------------------------------===//
1684 // OpenMP Clauses.
1685 //===----------------------------------------------------------------------===//
1686
1687 namespace clang {
1688 class OMPClauseWriter : public OMPClauseVisitor<OMPClauseWriter> {
1689   ASTStmtWriter *Writer;
1690   ASTWriter::RecordData &Record;
1691 public:
1692   OMPClauseWriter(ASTStmtWriter *W, ASTWriter::RecordData &Record)
1693     : Writer(W), Record(Record) { }
1694 #define OPENMP_CLAUSE(Name, Class)    \
1695   void Visit##Class(Class *S);
1696 #include "clang/Basic/OpenMPKinds.def"
1697   void writeClause(OMPClause *C);
1698 };
1699 }
1700
1701 void OMPClauseWriter::writeClause(OMPClause *C) {
1702   Record.push_back(C->getClauseKind());
1703   Visit(C);
1704   Writer->Writer.AddSourceLocation(C->getLocStart(), Record);
1705   Writer->Writer.AddSourceLocation(C->getLocEnd(), Record);
1706 }
1707
1708 void OMPClauseWriter::VisitOMPIfClause(OMPIfClause *C) {
1709   Writer->Writer.AddStmt(C->getCondition());
1710   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1711 }
1712
1713 void OMPClauseWriter::VisitOMPFinalClause(OMPFinalClause *C) {
1714   Writer->Writer.AddStmt(C->getCondition());
1715   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1716 }
1717
1718 void OMPClauseWriter::VisitOMPNumThreadsClause(OMPNumThreadsClause *C) {
1719   Writer->Writer.AddStmt(C->getNumThreads());
1720   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1721 }
1722
1723 void OMPClauseWriter::VisitOMPSafelenClause(OMPSafelenClause *C) {
1724   Writer->Writer.AddStmt(C->getSafelen());
1725   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1726 }
1727
1728 void OMPClauseWriter::VisitOMPCollapseClause(OMPCollapseClause *C) {
1729   Writer->Writer.AddStmt(C->getNumForLoops());
1730   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1731 }
1732
1733 void OMPClauseWriter::VisitOMPDefaultClause(OMPDefaultClause *C) {
1734   Record.push_back(C->getDefaultKind());
1735   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1736   Writer->Writer.AddSourceLocation(C->getDefaultKindKwLoc(), Record);
1737 }
1738
1739 void OMPClauseWriter::VisitOMPProcBindClause(OMPProcBindClause *C) {
1740   Record.push_back(C->getProcBindKind());
1741   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1742   Writer->Writer.AddSourceLocation(C->getProcBindKindKwLoc(), Record);
1743 }
1744
1745 void OMPClauseWriter::VisitOMPScheduleClause(OMPScheduleClause *C) {
1746   Record.push_back(C->getScheduleKind());
1747   Writer->Writer.AddStmt(C->getChunkSize());
1748   Writer->Writer.AddStmt(C->getHelperChunkSize());
1749   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1750   Writer->Writer.AddSourceLocation(C->getScheduleKindLoc(), Record);
1751   Writer->Writer.AddSourceLocation(C->getCommaLoc(), Record);
1752 }
1753
1754 void OMPClauseWriter::VisitOMPOrderedClause(OMPOrderedClause *) {}
1755
1756 void OMPClauseWriter::VisitOMPNowaitClause(OMPNowaitClause *) {}
1757
1758 void OMPClauseWriter::VisitOMPUntiedClause(OMPUntiedClause *) {}
1759
1760 void OMPClauseWriter::VisitOMPMergeableClause(OMPMergeableClause *) {}
1761
1762 void OMPClauseWriter::VisitOMPReadClause(OMPReadClause *) {}
1763
1764 void OMPClauseWriter::VisitOMPWriteClause(OMPWriteClause *) {}
1765
1766 void OMPClauseWriter::VisitOMPUpdateClause(OMPUpdateClause *) {}
1767
1768 void OMPClauseWriter::VisitOMPCaptureClause(OMPCaptureClause *) {}
1769
1770 void OMPClauseWriter::VisitOMPSeqCstClause(OMPSeqCstClause *) {}
1771
1772 void OMPClauseWriter::VisitOMPPrivateClause(OMPPrivateClause *C) {
1773   Record.push_back(C->varlist_size());
1774   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1775   for (auto *VE : C->varlists()) {
1776     Writer->Writer.AddStmt(VE);
1777   }
1778   for (auto *VE : C->private_copies()) {
1779     Writer->Writer.AddStmt(VE);
1780   }
1781 }
1782
1783 void OMPClauseWriter::VisitOMPFirstprivateClause(OMPFirstprivateClause *C) {
1784   Record.push_back(C->varlist_size());
1785   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1786   for (auto *VE : C->varlists()) {
1787     Writer->Writer.AddStmt(VE);
1788   }
1789   for (auto *VE : C->private_copies()) {
1790     Writer->Writer.AddStmt(VE);
1791   }
1792   for (auto *VE : C->inits()) {
1793     Writer->Writer.AddStmt(VE);
1794   }
1795 }
1796
1797 void OMPClauseWriter::VisitOMPLastprivateClause(OMPLastprivateClause *C) {
1798   Record.push_back(C->varlist_size());
1799   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1800   for (auto *VE : C->varlists())
1801     Writer->Writer.AddStmt(VE);
1802   for (auto *E : C->private_copies())
1803     Writer->Writer.AddStmt(E);
1804   for (auto *E : C->source_exprs())
1805     Writer->Writer.AddStmt(E);
1806   for (auto *E : C->destination_exprs())
1807     Writer->Writer.AddStmt(E);
1808   for (auto *E : C->assignment_ops())
1809     Writer->Writer.AddStmt(E);
1810 }
1811
1812 void OMPClauseWriter::VisitOMPSharedClause(OMPSharedClause *C) {
1813   Record.push_back(C->varlist_size());
1814   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1815   for (auto *VE : C->varlists())
1816     Writer->Writer.AddStmt(VE);
1817 }
1818
1819 void OMPClauseWriter::VisitOMPReductionClause(OMPReductionClause *C) {
1820   Record.push_back(C->varlist_size());
1821   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1822   Writer->Writer.AddSourceLocation(C->getColonLoc(), Record);
1823   Writer->Writer.AddNestedNameSpecifierLoc(C->getQualifierLoc(), Record);
1824   Writer->Writer.AddDeclarationNameInfo(C->getNameInfo(), Record);
1825   for (auto *VE : C->varlists())
1826     Writer->Writer.AddStmt(VE);
1827   for (auto *E : C->lhs_exprs())
1828     Writer->Writer.AddStmt(E);
1829   for (auto *E : C->rhs_exprs())
1830     Writer->Writer.AddStmt(E);
1831   for (auto *E : C->reduction_ops())
1832     Writer->Writer.AddStmt(E);
1833 }
1834
1835 void OMPClauseWriter::VisitOMPLinearClause(OMPLinearClause *C) {
1836   Record.push_back(C->varlist_size());
1837   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1838   Writer->Writer.AddSourceLocation(C->getColonLoc(), Record);
1839   for (auto *VE : C->varlists()) {
1840     Writer->Writer.AddStmt(VE);
1841   }
1842   for (auto *VE : C->inits()) {
1843     Writer->Writer.AddStmt(VE);
1844   }
1845   for (auto *VE : C->updates()) {
1846     Writer->Writer.AddStmt(VE);
1847   }
1848   for (auto *VE : C->finals()) {
1849     Writer->Writer.AddStmt(VE);
1850   }
1851   Writer->Writer.AddStmt(C->getStep());
1852   Writer->Writer.AddStmt(C->getCalcStep());
1853 }
1854
1855 void OMPClauseWriter::VisitOMPAlignedClause(OMPAlignedClause *C) {
1856   Record.push_back(C->varlist_size());
1857   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1858   Writer->Writer.AddSourceLocation(C->getColonLoc(), Record);
1859   for (auto *VE : C->varlists())
1860     Writer->Writer.AddStmt(VE);
1861   Writer->Writer.AddStmt(C->getAlignment());
1862 }
1863
1864 void OMPClauseWriter::VisitOMPCopyinClause(OMPCopyinClause *C) {
1865   Record.push_back(C->varlist_size());
1866   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1867   for (auto *VE : C->varlists())
1868     Writer->Writer.AddStmt(VE);
1869   for (auto *E : C->source_exprs())
1870     Writer->Writer.AddStmt(E);
1871   for (auto *E : C->destination_exprs())
1872     Writer->Writer.AddStmt(E);
1873   for (auto *E : C->assignment_ops())
1874     Writer->Writer.AddStmt(E);
1875 }
1876
1877 void OMPClauseWriter::VisitOMPCopyprivateClause(OMPCopyprivateClause *C) {
1878   Record.push_back(C->varlist_size());
1879   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1880   for (auto *VE : C->varlists())
1881     Writer->Writer.AddStmt(VE);
1882   for (auto *E : C->source_exprs())
1883     Writer->Writer.AddStmt(E);
1884   for (auto *E : C->destination_exprs())
1885     Writer->Writer.AddStmt(E);
1886   for (auto *E : C->assignment_ops())
1887     Writer->Writer.AddStmt(E);
1888 }
1889
1890 void OMPClauseWriter::VisitOMPFlushClause(OMPFlushClause *C) {
1891   Record.push_back(C->varlist_size());
1892   Writer->Writer.AddSourceLocation(C->getLParenLoc(), Record);
1893   for (auto *VE : C->varlists())
1894     Writer->Writer.AddStmt(VE);
1895 }
1896
1897 //===----------------------------------------------------------------------===//
1898 // OpenMP Directives.
1899 //===----------------------------------------------------------------------===//
1900 void ASTStmtWriter::VisitOMPExecutableDirective(OMPExecutableDirective *E) {
1901   Writer.AddSourceLocation(E->getLocStart(), Record);
1902   Writer.AddSourceLocation(E->getLocEnd(), Record);
1903   OMPClauseWriter ClauseWriter(this, Record);
1904   for (unsigned i = 0; i < E->getNumClauses(); ++i) {
1905     ClauseWriter.writeClause(E->getClause(i));
1906   }
1907   if (E->hasAssociatedStmt())
1908     Writer.AddStmt(E->getAssociatedStmt());
1909 }
1910
1911 void ASTStmtWriter::VisitOMPLoopDirective(OMPLoopDirective *D) {
1912   VisitStmt(D);
1913   Record.push_back(D->getNumClauses());
1914   Record.push_back(D->getCollapsedNumber());
1915   VisitOMPExecutableDirective(D);
1916   Writer.AddStmt(D->getIterationVariable());
1917   Writer.AddStmt(D->getLastIteration());
1918   Writer.AddStmt(D->getCalcLastIteration());
1919   Writer.AddStmt(D->getPreCond());
1920   Writer.AddStmt(D->getCond(/* SeparateIter */ false));
1921   Writer.AddStmt(D->getCond(/* SeparateIter */ true));
1922   Writer.AddStmt(D->getInit());
1923   Writer.AddStmt(D->getInc());
1924   if (isOpenMPWorksharingDirective(D->getDirectiveKind())) {
1925     Writer.AddStmt(D->getIsLastIterVariable());
1926     Writer.AddStmt(D->getLowerBoundVariable());
1927     Writer.AddStmt(D->getUpperBoundVariable());
1928     Writer.AddStmt(D->getStrideVariable());
1929     Writer.AddStmt(D->getEnsureUpperBound());
1930     Writer.AddStmt(D->getNextLowerBound());
1931     Writer.AddStmt(D->getNextUpperBound());
1932   }
1933   for (auto I : D->counters()) {
1934     Writer.AddStmt(I);
1935   }
1936   for (auto I : D->updates()) {
1937     Writer.AddStmt(I);
1938   }
1939   for (auto I : D->finals()) {
1940     Writer.AddStmt(I);
1941   }
1942 }
1943
1944 void ASTStmtWriter::VisitOMPParallelDirective(OMPParallelDirective *D) {
1945   VisitStmt(D);
1946   Record.push_back(D->getNumClauses());
1947   VisitOMPExecutableDirective(D);
1948   Code = serialization::STMT_OMP_PARALLEL_DIRECTIVE;
1949 }
1950
1951 void ASTStmtWriter::VisitOMPSimdDirective(OMPSimdDirective *D) {
1952   VisitOMPLoopDirective(D);
1953   Code = serialization::STMT_OMP_SIMD_DIRECTIVE;
1954 }
1955
1956 void ASTStmtWriter::VisitOMPForDirective(OMPForDirective *D) {
1957   VisitOMPLoopDirective(D);
1958   Code = serialization::STMT_OMP_FOR_DIRECTIVE;
1959 }
1960
1961 void ASTStmtWriter::VisitOMPForSimdDirective(OMPForSimdDirective *D) {
1962   VisitOMPLoopDirective(D);
1963   Code = serialization::STMT_OMP_FOR_SIMD_DIRECTIVE;
1964 }
1965
1966 void ASTStmtWriter::VisitOMPSectionsDirective(OMPSectionsDirective *D) {
1967   VisitStmt(D);
1968   Record.push_back(D->getNumClauses());
1969   VisitOMPExecutableDirective(D);
1970   Code = serialization::STMT_OMP_SECTIONS_DIRECTIVE;
1971 }
1972
1973 void ASTStmtWriter::VisitOMPSectionDirective(OMPSectionDirective *D) {
1974   VisitStmt(D);
1975   VisitOMPExecutableDirective(D);
1976   Code = serialization::STMT_OMP_SECTION_DIRECTIVE;
1977 }
1978
1979 void ASTStmtWriter::VisitOMPSingleDirective(OMPSingleDirective *D) {
1980   VisitStmt(D);
1981   Record.push_back(D->getNumClauses());
1982   VisitOMPExecutableDirective(D);
1983   Code = serialization::STMT_OMP_SINGLE_DIRECTIVE;
1984 }
1985
1986 void ASTStmtWriter::VisitOMPMasterDirective(OMPMasterDirective *D) {
1987   VisitStmt(D);
1988   VisitOMPExecutableDirective(D);
1989   Code = serialization::STMT_OMP_MASTER_DIRECTIVE;
1990 }
1991
1992 void ASTStmtWriter::VisitOMPCriticalDirective(OMPCriticalDirective *D) {
1993   VisitStmt(D);
1994   VisitOMPExecutableDirective(D);
1995   Writer.AddDeclarationNameInfo(D->getDirectiveName(), Record);
1996   Code = serialization::STMT_OMP_CRITICAL_DIRECTIVE;
1997 }
1998
1999 void ASTStmtWriter::VisitOMPParallelForDirective(OMPParallelForDirective *D) {
2000   VisitOMPLoopDirective(D);
2001   Code = serialization::STMT_OMP_PARALLEL_FOR_DIRECTIVE;
2002 }
2003
2004 void ASTStmtWriter::VisitOMPParallelForSimdDirective(
2005     OMPParallelForSimdDirective *D) {
2006   VisitOMPLoopDirective(D);
2007   Code = serialization::STMT_OMP_PARALLEL_FOR_SIMD_DIRECTIVE;
2008 }
2009
2010 void ASTStmtWriter::VisitOMPParallelSectionsDirective(
2011     OMPParallelSectionsDirective *D) {
2012   VisitStmt(D);
2013   Record.push_back(D->getNumClauses());
2014   VisitOMPExecutableDirective(D);
2015   Code = serialization::STMT_OMP_PARALLEL_SECTIONS_DIRECTIVE;
2016 }
2017
2018 void ASTStmtWriter::VisitOMPTaskDirective(OMPTaskDirective *D) {
2019   VisitStmt(D);
2020   Record.push_back(D->getNumClauses());
2021   VisitOMPExecutableDirective(D);
2022   Code = serialization::STMT_OMP_TASK_DIRECTIVE;
2023 }
2024
2025 void ASTStmtWriter::VisitOMPAtomicDirective(OMPAtomicDirective *D) {
2026   VisitStmt(D);
2027   Record.push_back(D->getNumClauses());
2028   VisitOMPExecutableDirective(D);
2029   Writer.AddStmt(D->getX());
2030   Writer.AddStmt(D->getV());
2031   Writer.AddStmt(D->getExpr());
2032   Writer.AddStmt(D->getUpdateExpr());
2033   Record.push_back(D->isXLHSInRHSPart() ? 1 : 0);
2034   Record.push_back(D->isPostfixUpdate() ? 1 : 0);
2035   Code = serialization::STMT_OMP_ATOMIC_DIRECTIVE;
2036 }
2037
2038 void ASTStmtWriter::VisitOMPTargetDirective(OMPTargetDirective *D) {
2039   VisitStmt(D);
2040   Record.push_back(D->getNumClauses());
2041   VisitOMPExecutableDirective(D);
2042   Code = serialization::STMT_OMP_TARGET_DIRECTIVE;
2043 }
2044
2045 void ASTStmtWriter::VisitOMPTaskyieldDirective(OMPTaskyieldDirective *D) {
2046   VisitStmt(D);
2047   VisitOMPExecutableDirective(D);
2048   Code = serialization::STMT_OMP_TASKYIELD_DIRECTIVE;
2049 }
2050
2051 void ASTStmtWriter::VisitOMPBarrierDirective(OMPBarrierDirective *D) {
2052   VisitStmt(D);
2053   VisitOMPExecutableDirective(D);
2054   Code = serialization::STMT_OMP_BARRIER_DIRECTIVE;
2055 }
2056
2057 void ASTStmtWriter::VisitOMPTaskwaitDirective(OMPTaskwaitDirective *D) {
2058   VisitStmt(D);
2059   VisitOMPExecutableDirective(D);
2060   Code = serialization::STMT_OMP_TASKWAIT_DIRECTIVE;
2061 }
2062
2063 void ASTStmtWriter::VisitOMPFlushDirective(OMPFlushDirective *D) {
2064   VisitStmt(D);
2065   Record.push_back(D->getNumClauses());
2066   VisitOMPExecutableDirective(D);
2067   Code = serialization::STMT_OMP_FLUSH_DIRECTIVE;
2068 }
2069
2070 void ASTStmtWriter::VisitOMPOrderedDirective(OMPOrderedDirective *D) {
2071   VisitStmt(D);
2072   VisitOMPExecutableDirective(D);
2073   Code = serialization::STMT_OMP_ORDERED_DIRECTIVE;
2074 }
2075
2076 void ASTStmtWriter::VisitOMPTeamsDirective(OMPTeamsDirective *D) {
2077   VisitStmt(D);
2078   Record.push_back(D->getNumClauses());
2079   VisitOMPExecutableDirective(D);
2080   Code = serialization::STMT_OMP_TEAMS_DIRECTIVE;
2081 }
2082
2083 //===----------------------------------------------------------------------===//
2084 // ASTWriter Implementation
2085 //===----------------------------------------------------------------------===//
2086
2087 unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
2088   assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
2089          "SwitchCase recorded twice");
2090   unsigned NextID = SwitchCaseIDs.size();
2091   SwitchCaseIDs[S] = NextID;
2092   return NextID;
2093 }
2094
2095 unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
2096   assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
2097          "SwitchCase hasn't been seen yet");
2098   return SwitchCaseIDs[S];
2099 }
2100
2101 void ASTWriter::ClearSwitchCaseIDs() {
2102   SwitchCaseIDs.clear();
2103 }
2104
2105 /// \brief Write the given substatement or subexpression to the
2106 /// bitstream.
2107 void ASTWriter::WriteSubStmt(Stmt *S,
2108                              llvm::DenseMap<Stmt *, uint64_t> &SubStmtEntries,
2109                              llvm::DenseSet<Stmt *> &ParentStmts) {
2110   RecordData Record;
2111   ASTStmtWriter Writer(*this, Record);
2112   ++NumStatements;
2113   
2114   if (!S) {
2115     Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
2116     return;
2117   }
2118
2119   llvm::DenseMap<Stmt *, uint64_t>::iterator I = SubStmtEntries.find(S);
2120   if (I != SubStmtEntries.end()) {
2121     Record.push_back(I->second);
2122     Stream.EmitRecord(serialization::STMT_REF_PTR, Record);
2123     return;
2124   }
2125
2126 #ifndef NDEBUG
2127   assert(!ParentStmts.count(S) && "There is a Stmt cycle!");
2128
2129   struct ParentStmtInserterRAII {
2130     Stmt *S;
2131     llvm::DenseSet<Stmt *> &ParentStmts;
2132
2133     ParentStmtInserterRAII(Stmt *S, llvm::DenseSet<Stmt *> &ParentStmts)
2134       : S(S), ParentStmts(ParentStmts) {
2135       ParentStmts.insert(S);
2136     }
2137     ~ParentStmtInserterRAII() {
2138       ParentStmts.erase(S);
2139     }
2140   };
2141
2142   ParentStmtInserterRAII ParentStmtInserter(S, ParentStmts);
2143 #endif
2144
2145   // Redirect ASTWriter::AddStmt to collect sub-stmts.
2146   SmallVector<Stmt *, 16> SubStmts;
2147   CollectedStmts = &SubStmts;
2148
2149   Writer.Code = serialization::STMT_NULL_PTR;
2150   Writer.AbbrevToUse = 0;
2151   Writer.Visit(S);
2152   
2153 #ifndef NDEBUG
2154   if (Writer.Code == serialization::STMT_NULL_PTR) {
2155     SourceManager &SrcMgr
2156       = DeclIDs.begin()->first->getASTContext().getSourceManager();
2157     S->dump(SrcMgr);
2158     llvm_unreachable("Unhandled sub-statement writing AST file");
2159   }
2160 #endif
2161
2162   // Revert ASTWriter::AddStmt.
2163   CollectedStmts = &StmtsToEmit;
2164
2165   // Write the sub-stmts in reverse order, last to first. When reading them back
2166   // we will read them in correct order by "pop"ing them from the Stmts stack.
2167   // This simplifies reading and allows to store a variable number of sub-stmts
2168   // without knowing it in advance.
2169   while (!SubStmts.empty())
2170     WriteSubStmt(SubStmts.pop_back_val(), SubStmtEntries, ParentStmts);
2171   
2172   Stream.EmitRecord(Writer.Code, Record, Writer.AbbrevToUse);
2173  
2174   SubStmtEntries[S] = Stream.GetCurrentBitNo();
2175 }
2176
2177 /// \brief Flush all of the statements that have been added to the
2178 /// queue via AddStmt().
2179 void ASTWriter::FlushStmts() {
2180   RecordData Record;
2181
2182   // We expect to be the only consumer of the two temporary statement maps,
2183   // assert that they are empty.
2184   assert(SubStmtEntries.empty() && "unexpected entries in sub-stmt map");
2185   assert(ParentStmts.empty() && "unexpected entries in parent stmt map");
2186
2187   for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
2188     WriteSubStmt(StmtsToEmit[I], SubStmtEntries, ParentStmts);
2189     
2190     assert(N == StmtsToEmit.size() &&
2191            "Substatement written via AddStmt rather than WriteSubStmt!");
2192
2193     // Note that we are at the end of a full expression. Any
2194     // expression records that follow this one are part of a different
2195     // expression.
2196     Stream.EmitRecord(serialization::STMT_STOP, Record);
2197
2198     SubStmtEntries.clear();
2199     ParentStmts.clear();
2200   }
2201
2202   StmtsToEmit.clear();
2203 }