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