1 //===--- ASTWriterStmt.cpp - Statement and Expression Serialization -------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file implements serialization for Statements and Expressions.
12 //===----------------------------------------------------------------------===//
14 #include "clang/Serialization/ASTWriter.h"
15 #include "clang/AST/DeclCXX.h"
16 #include "clang/AST/DeclObjC.h"
17 #include "clang/AST/DeclTemplate.h"
18 #include "clang/AST/StmtVisitor.h"
19 #include "llvm/Bitcode/BitstreamWriter.h"
20 using namespace clang;
22 //===----------------------------------------------------------------------===//
23 // Statement/expression serialization
24 //===----------------------------------------------------------------------===//
27 class ASTStmtWriter : public StmtVisitor<ASTStmtWriter, void> {
29 ASTWriter::RecordData &Record;
32 serialization::StmtCode Code;
35 ASTStmtWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
36 : Writer(Writer), Record(Record) { }
38 void AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo &Args);
40 void VisitStmt(Stmt *S);
41 #define STMT(Type, Base) \
42 void Visit##Type(Type *);
43 #include "clang/AST/StmtNodes.inc"
48 AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo &Args) {
49 Writer.AddSourceLocation(Args.getTemplateKeywordLoc(), Record);
50 Writer.AddSourceLocation(Args.LAngleLoc, Record);
51 Writer.AddSourceLocation(Args.RAngleLoc, Record);
52 for (unsigned i=0; i != Args.NumTemplateArgs; ++i)
53 Writer.AddTemplateArgumentLoc(Args.getTemplateArgs()[i], Record);
56 void ASTStmtWriter::VisitStmt(Stmt *S) {
59 void ASTStmtWriter::VisitNullStmt(NullStmt *S) {
61 Writer.AddSourceLocation(S->getSemiLoc(), Record);
62 Record.push_back(S->HasLeadingEmptyMacro);
63 Code = serialization::STMT_NULL;
66 void ASTStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
68 Record.push_back(S->size());
69 for (CompoundStmt::body_iterator CS = S->body_begin(), CSEnd = S->body_end();
72 Writer.AddSourceLocation(S->getLBracLoc(), Record);
73 Writer.AddSourceLocation(S->getRBracLoc(), Record);
74 Code = serialization::STMT_COMPOUND;
77 void ASTStmtWriter::VisitSwitchCase(SwitchCase *S) {
79 Record.push_back(Writer.getSwitchCaseID(S));
82 void ASTStmtWriter::VisitCaseStmt(CaseStmt *S) {
84 Writer.AddStmt(S->getLHS());
85 Writer.AddStmt(S->getRHS());
86 Writer.AddStmt(S->getSubStmt());
87 Writer.AddSourceLocation(S->getCaseLoc(), Record);
88 Writer.AddSourceLocation(S->getEllipsisLoc(), Record);
89 Writer.AddSourceLocation(S->getColonLoc(), Record);
90 Code = serialization::STMT_CASE;
93 void ASTStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
95 Writer.AddStmt(S->getSubStmt());
96 Writer.AddSourceLocation(S->getDefaultLoc(), Record);
97 Writer.AddSourceLocation(S->getColonLoc(), Record);
98 Code = serialization::STMT_DEFAULT;
101 void ASTStmtWriter::VisitLabelStmt(LabelStmt *S) {
103 Writer.AddDeclRef(S->getDecl(), Record);
104 Writer.AddStmt(S->getSubStmt());
105 Writer.AddSourceLocation(S->getIdentLoc(), Record);
106 Code = serialization::STMT_LABEL;
109 void ASTStmtWriter::VisitAttributedStmt(AttributedStmt *S) {
111 Writer.WriteAttributes(S->getAttrs(), Record);
112 Writer.AddStmt(S->getSubStmt());
113 Writer.AddSourceLocation(S->getAttrLoc(), Record);
114 Code = serialization::STMT_ATTRIBUTED;
117 void ASTStmtWriter::VisitIfStmt(IfStmt *S) {
119 Writer.AddDeclRef(S->getConditionVariable(), Record);
120 Writer.AddStmt(S->getCond());
121 Writer.AddStmt(S->getThen());
122 Writer.AddStmt(S->getElse());
123 Writer.AddSourceLocation(S->getIfLoc(), Record);
124 Writer.AddSourceLocation(S->getElseLoc(), Record);
125 Code = serialization::STMT_IF;
128 void ASTStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
130 Writer.AddDeclRef(S->getConditionVariable(), Record);
131 Writer.AddStmt(S->getCond());
132 Writer.AddStmt(S->getBody());
133 Writer.AddSourceLocation(S->getSwitchLoc(), Record);
134 Record.push_back(S->isAllEnumCasesCovered());
135 for (SwitchCase *SC = S->getSwitchCaseList(); SC;
136 SC = SC->getNextSwitchCase())
137 Record.push_back(Writer.RecordSwitchCaseID(SC));
138 Code = serialization::STMT_SWITCH;
141 void ASTStmtWriter::VisitWhileStmt(WhileStmt *S) {
143 Writer.AddDeclRef(S->getConditionVariable(), Record);
144 Writer.AddStmt(S->getCond());
145 Writer.AddStmt(S->getBody());
146 Writer.AddSourceLocation(S->getWhileLoc(), Record);
147 Code = serialization::STMT_WHILE;
150 void ASTStmtWriter::VisitDoStmt(DoStmt *S) {
152 Writer.AddStmt(S->getCond());
153 Writer.AddStmt(S->getBody());
154 Writer.AddSourceLocation(S->getDoLoc(), Record);
155 Writer.AddSourceLocation(S->getWhileLoc(), Record);
156 Writer.AddSourceLocation(S->getRParenLoc(), Record);
157 Code = serialization::STMT_DO;
160 void ASTStmtWriter::VisitForStmt(ForStmt *S) {
162 Writer.AddStmt(S->getInit());
163 Writer.AddStmt(S->getCond());
164 Writer.AddDeclRef(S->getConditionVariable(), Record);
165 Writer.AddStmt(S->getInc());
166 Writer.AddStmt(S->getBody());
167 Writer.AddSourceLocation(S->getForLoc(), Record);
168 Writer.AddSourceLocation(S->getLParenLoc(), Record);
169 Writer.AddSourceLocation(S->getRParenLoc(), Record);
170 Code = serialization::STMT_FOR;
173 void ASTStmtWriter::VisitGotoStmt(GotoStmt *S) {
175 Writer.AddDeclRef(S->getLabel(), Record);
176 Writer.AddSourceLocation(S->getGotoLoc(), Record);
177 Writer.AddSourceLocation(S->getLabelLoc(), Record);
178 Code = serialization::STMT_GOTO;
181 void ASTStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
183 Writer.AddSourceLocation(S->getGotoLoc(), Record);
184 Writer.AddSourceLocation(S->getStarLoc(), Record);
185 Writer.AddStmt(S->getTarget());
186 Code = serialization::STMT_INDIRECT_GOTO;
189 void ASTStmtWriter::VisitContinueStmt(ContinueStmt *S) {
191 Writer.AddSourceLocation(S->getContinueLoc(), Record);
192 Code = serialization::STMT_CONTINUE;
195 void ASTStmtWriter::VisitBreakStmt(BreakStmt *S) {
197 Writer.AddSourceLocation(S->getBreakLoc(), Record);
198 Code = serialization::STMT_BREAK;
201 void ASTStmtWriter::VisitReturnStmt(ReturnStmt *S) {
203 Writer.AddStmt(S->getRetValue());
204 Writer.AddSourceLocation(S->getReturnLoc(), Record);
205 Writer.AddDeclRef(S->getNRVOCandidate(), Record);
206 Code = serialization::STMT_RETURN;
209 void ASTStmtWriter::VisitDeclStmt(DeclStmt *S) {
211 Writer.AddSourceLocation(S->getStartLoc(), Record);
212 Writer.AddSourceLocation(S->getEndLoc(), Record);
213 DeclGroupRef DG = S->getDeclGroup();
214 for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
215 Writer.AddDeclRef(*D, Record);
216 Code = serialization::STMT_DECL;
219 void ASTStmtWriter::VisitAsmStmt(AsmStmt *S) {
221 Record.push_back(S->getNumOutputs());
222 Record.push_back(S->getNumInputs());
223 Record.push_back(S->getNumClobbers());
224 Writer.AddSourceLocation(S->getAsmLoc(), Record);
225 Writer.AddSourceLocation(S->getRParenLoc(), Record);
226 Record.push_back(S->isVolatile());
227 Record.push_back(S->isSimple());
228 Record.push_back(S->isMSAsm());
229 Writer.AddStmt(S->getAsmString());
232 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
233 Writer.AddIdentifierRef(S->getOutputIdentifier(I), Record);
234 Writer.AddStmt(S->getOutputConstraintLiteral(I));
235 Writer.AddStmt(S->getOutputExpr(I));
239 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
240 Writer.AddIdentifierRef(S->getInputIdentifier(I), Record);
241 Writer.AddStmt(S->getInputConstraintLiteral(I));
242 Writer.AddStmt(S->getInputExpr(I));
246 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
247 Writer.AddStmt(S->getClobber(I));
249 Code = serialization::STMT_ASM;
252 void ASTStmtWriter::VisitExpr(Expr *E) {
254 Writer.AddTypeRef(E->getType(), Record);
255 Record.push_back(E->isTypeDependent());
256 Record.push_back(E->isValueDependent());
257 Record.push_back(E->isInstantiationDependent());
258 Record.push_back(E->containsUnexpandedParameterPack());
259 Record.push_back(E->getValueKind());
260 Record.push_back(E->getObjectKind());
263 void ASTStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
265 Writer.AddSourceLocation(E->getLocation(), Record);
266 Record.push_back(E->getIdentType()); // FIXME: stable encoding
267 Code = serialization::EXPR_PREDEFINED;
270 void ASTStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
273 Record.push_back(E->hasQualifier());
274 Record.push_back(E->getDecl() != E->getFoundDecl());
275 Record.push_back(E->hasTemplateKWAndArgsInfo());
276 Record.push_back(E->hadMultipleCandidates());
277 Record.push_back(E->refersToEnclosingLocal());
279 if (E->hasTemplateKWAndArgsInfo()) {
280 unsigned NumTemplateArgs = E->getNumTemplateArgs();
281 Record.push_back(NumTemplateArgs);
284 DeclarationName::NameKind nk = (E->getDecl()->getDeclName().getNameKind());
286 if ((!E->hasTemplateKWAndArgsInfo()) && (!E->hasQualifier()) &&
287 (E->getDecl() == E->getFoundDecl()) &&
288 nk == DeclarationName::Identifier) {
289 AbbrevToUse = Writer.getDeclRefExprAbbrev();
292 if (E->hasQualifier())
293 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
295 if (E->getDecl() != E->getFoundDecl())
296 Writer.AddDeclRef(E->getFoundDecl(), Record);
298 if (E->hasTemplateKWAndArgsInfo())
299 AddTemplateKWAndArgsInfo(*E->getTemplateKWAndArgsInfo());
301 Writer.AddDeclRef(E->getDecl(), Record);
302 Writer.AddSourceLocation(E->getLocation(), Record);
303 Writer.AddDeclarationNameLoc(E->DNLoc, E->getDecl()->getDeclName(), Record);
304 Code = serialization::EXPR_DECL_REF;
307 void ASTStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
309 Writer.AddSourceLocation(E->getLocation(), Record);
310 Writer.AddAPInt(E->getValue(), Record);
312 if (E->getValue().getBitWidth() == 32) {
313 AbbrevToUse = Writer.getIntegerLiteralAbbrev();
316 Code = serialization::EXPR_INTEGER_LITERAL;
319 void ASTStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
321 Writer.AddAPFloat(E->getValue(), Record);
322 Record.push_back(E->isExact());
323 Writer.AddSourceLocation(E->getLocation(), Record);
324 Code = serialization::EXPR_FLOATING_LITERAL;
327 void ASTStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
329 Writer.AddStmt(E->getSubExpr());
330 Code = serialization::EXPR_IMAGINARY_LITERAL;
333 void ASTStmtWriter::VisitStringLiteral(StringLiteral *E) {
335 Record.push_back(E->getByteLength());
336 Record.push_back(E->getNumConcatenated());
337 Record.push_back(E->getKind());
338 Record.push_back(E->isPascal());
339 // FIXME: String data should be stored as a blob at the end of the
340 // StringLiteral. However, we can't do so now because we have no
341 // provision for coping with abbreviations when we're jumping around
342 // the AST file during deserialization.
343 Record.append(E->getBytes().begin(), E->getBytes().end());
344 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
345 Writer.AddSourceLocation(E->getStrTokenLoc(I), Record);
346 Code = serialization::EXPR_STRING_LITERAL;
349 void ASTStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
351 Record.push_back(E->getValue());
352 Writer.AddSourceLocation(E->getLocation(), Record);
353 Record.push_back(E->getKind());
355 AbbrevToUse = Writer.getCharacterLiteralAbbrev();
357 Code = serialization::EXPR_CHARACTER_LITERAL;
360 void ASTStmtWriter::VisitParenExpr(ParenExpr *E) {
362 Writer.AddSourceLocation(E->getLParen(), Record);
363 Writer.AddSourceLocation(E->getRParen(), Record);
364 Writer.AddStmt(E->getSubExpr());
365 Code = serialization::EXPR_PAREN;
368 void ASTStmtWriter::VisitParenListExpr(ParenListExpr *E) {
370 Record.push_back(E->NumExprs);
371 for (unsigned i=0; i != E->NumExprs; ++i)
372 Writer.AddStmt(E->Exprs[i]);
373 Writer.AddSourceLocation(E->LParenLoc, Record);
374 Writer.AddSourceLocation(E->RParenLoc, Record);
375 Code = serialization::EXPR_PAREN_LIST;
378 void ASTStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
380 Writer.AddStmt(E->getSubExpr());
381 Record.push_back(E->getOpcode()); // FIXME: stable encoding
382 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
383 Code = serialization::EXPR_UNARY_OPERATOR;
386 void ASTStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
388 Record.push_back(E->getNumComponents());
389 Record.push_back(E->getNumExpressions());
390 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
391 Writer.AddSourceLocation(E->getRParenLoc(), Record);
392 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
393 for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
394 const OffsetOfExpr::OffsetOfNode &ON = E->getComponent(I);
395 Record.push_back(ON.getKind()); // FIXME: Stable encoding
396 Writer.AddSourceLocation(ON.getSourceRange().getBegin(), Record);
397 Writer.AddSourceLocation(ON.getSourceRange().getEnd(), Record);
398 switch (ON.getKind()) {
399 case OffsetOfExpr::OffsetOfNode::Array:
400 Record.push_back(ON.getArrayExprIndex());
403 case OffsetOfExpr::OffsetOfNode::Field:
404 Writer.AddDeclRef(ON.getField(), Record);
407 case OffsetOfExpr::OffsetOfNode::Identifier:
408 Writer.AddIdentifierRef(ON.getFieldName(), Record);
411 case OffsetOfExpr::OffsetOfNode::Base:
412 Writer.AddCXXBaseSpecifier(*ON.getBase(), Record);
416 for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
417 Writer.AddStmt(E->getIndexExpr(I));
418 Code = serialization::EXPR_OFFSETOF;
421 void ASTStmtWriter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E) {
423 Record.push_back(E->getKind());
424 if (E->isArgumentType())
425 Writer.AddTypeSourceInfo(E->getArgumentTypeInfo(), Record);
428 Writer.AddStmt(E->getArgumentExpr());
430 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
431 Writer.AddSourceLocation(E->getRParenLoc(), Record);
432 Code = serialization::EXPR_SIZEOF_ALIGN_OF;
435 void ASTStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
437 Writer.AddStmt(E->getLHS());
438 Writer.AddStmt(E->getRHS());
439 Writer.AddSourceLocation(E->getRBracketLoc(), Record);
440 Code = serialization::EXPR_ARRAY_SUBSCRIPT;
443 void ASTStmtWriter::VisitCallExpr(CallExpr *E) {
445 Record.push_back(E->getNumArgs());
446 Writer.AddSourceLocation(E->getRParenLoc(), Record);
447 Writer.AddStmt(E->getCallee());
448 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
449 Arg != ArgEnd; ++Arg)
450 Writer.AddStmt(*Arg);
451 Code = serialization::EXPR_CALL;
454 void ASTStmtWriter::VisitMemberExpr(MemberExpr *E) {
455 // Don't call VisitExpr, we'll write everything here.
457 Record.push_back(E->hasQualifier());
458 if (E->hasQualifier())
459 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
461 Record.push_back(E->HasTemplateKWAndArgsInfo);
462 if (E->HasTemplateKWAndArgsInfo) {
463 Writer.AddSourceLocation(E->getTemplateKeywordLoc(), Record);
464 unsigned NumTemplateArgs = E->getNumTemplateArgs();
465 Record.push_back(NumTemplateArgs);
466 Writer.AddSourceLocation(E->getLAngleLoc(), Record);
467 Writer.AddSourceLocation(E->getRAngleLoc(), Record);
468 for (unsigned i=0; i != NumTemplateArgs; ++i)
469 Writer.AddTemplateArgumentLoc(E->getTemplateArgs()[i], Record);
472 Record.push_back(E->hadMultipleCandidates());
474 DeclAccessPair FoundDecl = E->getFoundDecl();
475 Writer.AddDeclRef(FoundDecl.getDecl(), Record);
476 Record.push_back(FoundDecl.getAccess());
478 Writer.AddTypeRef(E->getType(), Record);
479 Record.push_back(E->getValueKind());
480 Record.push_back(E->getObjectKind());
481 Writer.AddStmt(E->getBase());
482 Writer.AddDeclRef(E->getMemberDecl(), Record);
483 Writer.AddSourceLocation(E->getMemberLoc(), Record);
484 Record.push_back(E->isArrow());
485 Writer.AddDeclarationNameLoc(E->MemberDNLoc,
486 E->getMemberDecl()->getDeclName(), Record);
487 Code = serialization::EXPR_MEMBER;
490 void ASTStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
492 Writer.AddStmt(E->getBase());
493 Writer.AddSourceLocation(E->getIsaMemberLoc(), Record);
494 Record.push_back(E->isArrow());
495 Code = serialization::EXPR_OBJC_ISA;
499 VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) {
501 Writer.AddStmt(E->getSubExpr());
502 Record.push_back(E->shouldCopy());
503 Code = serialization::EXPR_OBJC_INDIRECT_COPY_RESTORE;
506 void ASTStmtWriter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) {
507 VisitExplicitCastExpr(E);
508 Writer.AddSourceLocation(E->getLParenLoc(), Record);
509 Writer.AddSourceLocation(E->getBridgeKeywordLoc(), Record);
510 Record.push_back(E->getBridgeKind()); // FIXME: Stable encoding
511 Code = serialization::EXPR_OBJC_BRIDGED_CAST;
514 void ASTStmtWriter::VisitCastExpr(CastExpr *E) {
516 Record.push_back(E->path_size());
517 Writer.AddStmt(E->getSubExpr());
518 Record.push_back(E->getCastKind()); // FIXME: stable encoding
520 for (CastExpr::path_iterator
521 PI = E->path_begin(), PE = E->path_end(); PI != PE; ++PI)
522 Writer.AddCXXBaseSpecifier(**PI, Record);
525 void ASTStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
527 Writer.AddStmt(E->getLHS());
528 Writer.AddStmt(E->getRHS());
529 Record.push_back(E->getOpcode()); // FIXME: stable encoding
530 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
531 Code = serialization::EXPR_BINARY_OPERATOR;
534 void ASTStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
535 VisitBinaryOperator(E);
536 Writer.AddTypeRef(E->getComputationLHSType(), Record);
537 Writer.AddTypeRef(E->getComputationResultType(), Record);
538 Code = serialization::EXPR_COMPOUND_ASSIGN_OPERATOR;
541 void ASTStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
543 Writer.AddStmt(E->getCond());
544 Writer.AddStmt(E->getLHS());
545 Writer.AddStmt(E->getRHS());
546 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
547 Writer.AddSourceLocation(E->getColonLoc(), Record);
548 Code = serialization::EXPR_CONDITIONAL_OPERATOR;
552 ASTStmtWriter::VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
554 Writer.AddStmt(E->getOpaqueValue());
555 Writer.AddStmt(E->getCommon());
556 Writer.AddStmt(E->getCond());
557 Writer.AddStmt(E->getTrueExpr());
558 Writer.AddStmt(E->getFalseExpr());
559 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
560 Writer.AddSourceLocation(E->getColonLoc(), Record);
561 Code = serialization::EXPR_BINARY_CONDITIONAL_OPERATOR;
564 void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
566 Code = serialization::EXPR_IMPLICIT_CAST;
569 void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
571 Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
574 void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
575 VisitExplicitCastExpr(E);
576 Writer.AddSourceLocation(E->getLParenLoc(), Record);
577 Writer.AddSourceLocation(E->getRParenLoc(), Record);
578 Code = serialization::EXPR_CSTYLE_CAST;
581 void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
583 Writer.AddSourceLocation(E->getLParenLoc(), Record);
584 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
585 Writer.AddStmt(E->getInitializer());
586 Record.push_back(E->isFileScope());
587 Code = serialization::EXPR_COMPOUND_LITERAL;
590 void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
592 Writer.AddStmt(E->getBase());
593 Writer.AddIdentifierRef(&E->getAccessor(), Record);
594 Writer.AddSourceLocation(E->getAccessorLoc(), Record);
595 Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
598 void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
600 Writer.AddStmt(E->getSyntacticForm());
601 Writer.AddSourceLocation(E->getLBraceLoc(), Record);
602 Writer.AddSourceLocation(E->getRBraceLoc(), Record);
603 bool isArrayFiller = E->ArrayFillerOrUnionFieldInit.is<Expr*>();
604 Record.push_back(isArrayFiller);
606 Writer.AddStmt(E->getArrayFiller());
608 Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
609 Record.push_back(E->hadArrayRangeDesignator());
610 Record.push_back(E->initializesStdInitializerList());
611 Record.push_back(E->getNumInits());
613 // ArrayFiller may have filled "holes" due to designated initializer.
614 // Replace them by 0 to indicate that the filler goes in that place.
615 Expr *filler = E->getArrayFiller();
616 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
617 Writer.AddStmt(E->getInit(I) != filler ? E->getInit(I) : 0);
619 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
620 Writer.AddStmt(E->getInit(I));
622 Code = serialization::EXPR_INIT_LIST;
625 void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
627 Record.push_back(E->getNumSubExprs());
628 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
629 Writer.AddStmt(E->getSubExpr(I));
630 Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
631 Record.push_back(E->usesGNUSyntax());
632 for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
633 DEnd = E->designators_end();
635 if (D->isFieldDesignator()) {
636 if (FieldDecl *Field = D->getField()) {
637 Record.push_back(serialization::DESIG_FIELD_DECL);
638 Writer.AddDeclRef(Field, Record);
640 Record.push_back(serialization::DESIG_FIELD_NAME);
641 Writer.AddIdentifierRef(D->getFieldName(), Record);
643 Writer.AddSourceLocation(D->getDotLoc(), Record);
644 Writer.AddSourceLocation(D->getFieldLoc(), Record);
645 } else if (D->isArrayDesignator()) {
646 Record.push_back(serialization::DESIG_ARRAY);
647 Record.push_back(D->getFirstExprIndex());
648 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
649 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
651 assert(D->isArrayRangeDesignator() && "Unknown designator");
652 Record.push_back(serialization::DESIG_ARRAY_RANGE);
653 Record.push_back(D->getFirstExprIndex());
654 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
655 Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
656 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
659 Code = serialization::EXPR_DESIGNATED_INIT;
662 void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
664 Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
667 void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
669 Writer.AddStmt(E->getSubExpr());
670 Writer.AddTypeSourceInfo(E->getWrittenTypeInfo(), Record);
671 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
672 Writer.AddSourceLocation(E->getRParenLoc(), Record);
673 Code = serialization::EXPR_VA_ARG;
676 void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
678 Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
679 Writer.AddSourceLocation(E->getLabelLoc(), Record);
680 Writer.AddDeclRef(E->getLabel(), Record);
681 Code = serialization::EXPR_ADDR_LABEL;
684 void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
686 Writer.AddStmt(E->getSubStmt());
687 Writer.AddSourceLocation(E->getLParenLoc(), Record);
688 Writer.AddSourceLocation(E->getRParenLoc(), Record);
689 Code = serialization::EXPR_STMT;
692 void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
694 Writer.AddStmt(E->getCond());
695 Writer.AddStmt(E->getLHS());
696 Writer.AddStmt(E->getRHS());
697 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
698 Writer.AddSourceLocation(E->getRParenLoc(), Record);
699 Code = serialization::EXPR_CHOOSE;
702 void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
704 Writer.AddSourceLocation(E->getTokenLocation(), Record);
705 Code = serialization::EXPR_GNU_NULL;
708 void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
710 Record.push_back(E->getNumSubExprs());
711 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
712 Writer.AddStmt(E->getExpr(I));
713 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
714 Writer.AddSourceLocation(E->getRParenLoc(), Record);
715 Code = serialization::EXPR_SHUFFLE_VECTOR;
718 void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
720 Writer.AddDeclRef(E->getBlockDecl(), Record);
721 Code = serialization::EXPR_BLOCK;
724 void ASTStmtWriter::VisitGenericSelectionExpr(GenericSelectionExpr *E) {
726 Record.push_back(E->getNumAssocs());
728 Writer.AddStmt(E->getControllingExpr());
729 for (unsigned I = 0, N = E->getNumAssocs(); I != N; ++I) {
730 Writer.AddTypeSourceInfo(E->getAssocTypeSourceInfo(I), Record);
731 Writer.AddStmt(E->getAssocExpr(I));
733 Record.push_back(E->isResultDependent() ? -1U : E->getResultIndex());
735 Writer.AddSourceLocation(E->getGenericLoc(), Record);
736 Writer.AddSourceLocation(E->getDefaultLoc(), Record);
737 Writer.AddSourceLocation(E->getRParenLoc(), Record);
738 Code = serialization::EXPR_GENERIC_SELECTION;
741 void ASTStmtWriter::VisitPseudoObjectExpr(PseudoObjectExpr *E) {
743 Record.push_back(E->getNumSemanticExprs());
745 // Push the result index. Currently, this needs to exactly match
746 // the encoding used internally for ResultIndex.
747 unsigned result = E->getResultExprIndex();
748 result = (result == PseudoObjectExpr::NoResult ? 0 : result + 1);
749 Record.push_back(result);
751 Writer.AddStmt(E->getSyntacticForm());
752 for (PseudoObjectExpr::semantics_iterator
753 i = E->semantics_begin(), e = E->semantics_end(); i != e; ++i) {
756 Code = serialization::EXPR_PSEUDO_OBJECT;
759 void ASTStmtWriter::VisitAtomicExpr(AtomicExpr *E) {
761 Record.push_back(E->getOp());
762 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
763 Writer.AddStmt(E->getSubExprs()[I]);
764 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
765 Writer.AddSourceLocation(E->getRParenLoc(), Record);
766 Code = serialization::EXPR_ATOMIC;
769 //===----------------------------------------------------------------------===//
770 // Objective-C Expressions and Statements.
771 //===----------------------------------------------------------------------===//
773 void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
775 Writer.AddStmt(E->getString());
776 Writer.AddSourceLocation(E->getAtLoc(), Record);
777 Code = serialization::EXPR_OBJC_STRING_LITERAL;
780 void ASTStmtWriter::VisitObjCNumericLiteral(ObjCNumericLiteral *E) {
782 Writer.AddStmt(E->getNumber());
783 Writer.AddDeclRef(E->getObjCNumericLiteralMethod(), Record);
784 Writer.AddSourceLocation(E->getAtLoc(), Record);
785 Code = serialization::EXPR_OBJC_NUMERIC_LITERAL;
788 void ASTStmtWriter::VisitObjCArrayLiteral(ObjCArrayLiteral *E) {
790 Record.push_back(E->getNumElements());
791 for (unsigned i = 0; i < E->getNumElements(); i++)
792 Writer.AddStmt(E->getElement(i));
793 Writer.AddDeclRef(E->getArrayWithObjectsMethod(), Record);
794 Writer.AddSourceRange(E->getSourceRange(), Record);
795 Code = serialization::EXPR_OBJC_ARRAY_LITERAL;
798 void ASTStmtWriter::VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) {
800 Record.push_back(E->getNumElements());
801 Record.push_back(E->HasPackExpansions);
802 for (unsigned i = 0; i < E->getNumElements(); i++) {
803 ObjCDictionaryElement Element = E->getKeyValueElement(i);
804 Writer.AddStmt(Element.Key);
805 Writer.AddStmt(Element.Value);
806 if (E->HasPackExpansions) {
807 Writer.AddSourceLocation(Element.EllipsisLoc, Record);
808 unsigned NumExpansions = 0;
809 if (Element.NumExpansions)
810 NumExpansions = *Element.NumExpansions + 1;
811 Record.push_back(NumExpansions);
815 Writer.AddDeclRef(E->getDictWithObjectsMethod(), Record);
816 Writer.AddSourceRange(E->getSourceRange(), Record);
817 Code = serialization::EXPR_OBJC_DICTIONARY_LITERAL;
820 void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
822 Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
823 Writer.AddSourceLocation(E->getAtLoc(), Record);
824 Writer.AddSourceLocation(E->getRParenLoc(), Record);
825 Code = serialization::EXPR_OBJC_ENCODE;
828 void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
830 Writer.AddSelectorRef(E->getSelector(), Record);
831 Writer.AddSourceLocation(E->getAtLoc(), Record);
832 Writer.AddSourceLocation(E->getRParenLoc(), Record);
833 Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
836 void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
838 Writer.AddDeclRef(E->getProtocol(), Record);
839 Writer.AddSourceLocation(E->getAtLoc(), Record);
840 Writer.AddSourceLocation(E->getRParenLoc(), Record);
841 Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
844 void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
846 Writer.AddDeclRef(E->getDecl(), Record);
847 Writer.AddSourceLocation(E->getLocation(), Record);
848 Writer.AddStmt(E->getBase());
849 Record.push_back(E->isArrow());
850 Record.push_back(E->isFreeIvar());
851 Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
854 void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
856 Record.push_back(E->SetterAndMethodRefFlags.getInt());
857 Record.push_back(E->isImplicitProperty());
858 if (E->isImplicitProperty()) {
859 Writer.AddDeclRef(E->getImplicitPropertyGetter(), Record);
860 Writer.AddDeclRef(E->getImplicitPropertySetter(), Record);
862 Writer.AddDeclRef(E->getExplicitProperty(), Record);
864 Writer.AddSourceLocation(E->getLocation(), Record);
865 Writer.AddSourceLocation(E->getReceiverLocation(), Record);
866 if (E->isObjectReceiver()) {
868 Writer.AddStmt(E->getBase());
869 } else if (E->isSuperReceiver()) {
871 Writer.AddTypeRef(E->getSuperReceiverType(), Record);
874 Writer.AddDeclRef(E->getClassReceiver(), Record);
877 Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
880 void ASTStmtWriter::VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr *E) {
882 Writer.AddSourceLocation(E->getRBracket(), Record);
883 Writer.AddStmt(E->getBaseExpr());
884 Writer.AddStmt(E->getKeyExpr());
885 Writer.AddDeclRef(E->getAtIndexMethodDecl(), Record);
886 Writer.AddDeclRef(E->setAtIndexMethodDecl(), Record);
888 Code = serialization::EXPR_OBJC_SUBSCRIPT_REF_EXPR;
891 void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
893 Record.push_back(E->getNumArgs());
894 Record.push_back(E->getNumStoredSelLocs());
895 Record.push_back(E->SelLocsKind);
896 Record.push_back(E->isDelegateInitCall());
897 Record.push_back(E->IsImplicit);
898 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
899 switch (E->getReceiverKind()) {
900 case ObjCMessageExpr::Instance:
901 Writer.AddStmt(E->getInstanceReceiver());
904 case ObjCMessageExpr::Class:
905 Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
908 case ObjCMessageExpr::SuperClass:
909 case ObjCMessageExpr::SuperInstance:
910 Writer.AddTypeRef(E->getSuperType(), Record);
911 Writer.AddSourceLocation(E->getSuperLoc(), Record);
915 if (E->getMethodDecl()) {
917 Writer.AddDeclRef(E->getMethodDecl(), Record);
920 Writer.AddSelectorRef(E->getSelector(), Record);
923 Writer.AddSourceLocation(E->getLeftLoc(), Record);
924 Writer.AddSourceLocation(E->getRightLoc(), Record);
926 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
927 Arg != ArgEnd; ++Arg)
928 Writer.AddStmt(*Arg);
930 SourceLocation *Locs = E->getStoredSelLocs();
931 for (unsigned i = 0, e = E->getNumStoredSelLocs(); i != e; ++i)
932 Writer.AddSourceLocation(Locs[i], Record);
934 Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
937 void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
939 Writer.AddStmt(S->getElement());
940 Writer.AddStmt(S->getCollection());
941 Writer.AddStmt(S->getBody());
942 Writer.AddSourceLocation(S->getForLoc(), Record);
943 Writer.AddSourceLocation(S->getRParenLoc(), Record);
944 Code = serialization::STMT_OBJC_FOR_COLLECTION;
947 void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
948 Writer.AddStmt(S->getCatchBody());
949 Writer.AddDeclRef(S->getCatchParamDecl(), Record);
950 Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
951 Writer.AddSourceLocation(S->getRParenLoc(), Record);
952 Code = serialization::STMT_OBJC_CATCH;
955 void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
956 Writer.AddStmt(S->getFinallyBody());
957 Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
958 Code = serialization::STMT_OBJC_FINALLY;
961 void ASTStmtWriter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *S) {
962 Writer.AddStmt(S->getSubStmt());
963 Writer.AddSourceLocation(S->getAtLoc(), Record);
964 Code = serialization::STMT_OBJC_AUTORELEASE_POOL;
967 void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
968 Record.push_back(S->getNumCatchStmts());
969 Record.push_back(S->getFinallyStmt() != 0);
970 Writer.AddStmt(S->getTryBody());
971 for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
972 Writer.AddStmt(S->getCatchStmt(I));
973 if (S->getFinallyStmt())
974 Writer.AddStmt(S->getFinallyStmt());
975 Writer.AddSourceLocation(S->getAtTryLoc(), Record);
976 Code = serialization::STMT_OBJC_AT_TRY;
979 void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
980 Writer.AddStmt(S->getSynchExpr());
981 Writer.AddStmt(S->getSynchBody());
982 Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
983 Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
986 void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
987 Writer.AddStmt(S->getThrowExpr());
988 Writer.AddSourceLocation(S->getThrowLoc(), Record);
989 Code = serialization::STMT_OBJC_AT_THROW;
992 void ASTStmtWriter::VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr *E) {
994 Record.push_back(E->getValue());
995 Writer.AddSourceLocation(E->getLocation(), Record);
996 Code = serialization::EXPR_OBJC_BOOL_LITERAL;
999 //===----------------------------------------------------------------------===//
1000 // C++ Expressions and Statements.
1001 //===----------------------------------------------------------------------===//
1003 void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
1005 Writer.AddSourceLocation(S->getCatchLoc(), Record);
1006 Writer.AddDeclRef(S->getExceptionDecl(), Record);
1007 Writer.AddStmt(S->getHandlerBlock());
1008 Code = serialization::STMT_CXX_CATCH;
1011 void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
1013 Record.push_back(S->getNumHandlers());
1014 Writer.AddSourceLocation(S->getTryLoc(), Record);
1015 Writer.AddStmt(S->getTryBlock());
1016 for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
1017 Writer.AddStmt(S->getHandler(i));
1018 Code = serialization::STMT_CXX_TRY;
1021 void ASTStmtWriter::VisitCXXForRangeStmt(CXXForRangeStmt *S) {
1023 Writer.AddSourceLocation(S->getForLoc(), Record);
1024 Writer.AddSourceLocation(S->getColonLoc(), Record);
1025 Writer.AddSourceLocation(S->getRParenLoc(), Record);
1026 Writer.AddStmt(S->getRangeStmt());
1027 Writer.AddStmt(S->getBeginEndStmt());
1028 Writer.AddStmt(S->getCond());
1029 Writer.AddStmt(S->getInc());
1030 Writer.AddStmt(S->getLoopVarStmt());
1031 Writer.AddStmt(S->getBody());
1032 Code = serialization::STMT_CXX_FOR_RANGE;
1035 void ASTStmtWriter::VisitMSDependentExistsStmt(MSDependentExistsStmt *S) {
1037 Writer.AddSourceLocation(S->getKeywordLoc(), Record);
1038 Record.push_back(S->isIfExists());
1039 Writer.AddNestedNameSpecifierLoc(S->getQualifierLoc(), Record);
1040 Writer.AddDeclarationNameInfo(S->getNameInfo(), Record);
1041 Writer.AddStmt(S->getSubStmt());
1042 Code = serialization::STMT_MS_DEPENDENT_EXISTS;
1045 void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
1047 Record.push_back(E->getOperator());
1048 Code = serialization::EXPR_CXX_OPERATOR_CALL;
1051 void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
1053 Code = serialization::EXPR_CXX_MEMBER_CALL;
1056 void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
1058 Record.push_back(E->getNumArgs());
1059 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
1060 Writer.AddStmt(E->getArg(I));
1061 Writer.AddDeclRef(E->getConstructor(), Record);
1062 Writer.AddSourceLocation(E->getLocation(), Record);
1063 Record.push_back(E->isElidable());
1064 Record.push_back(E->hadMultipleCandidates());
1065 Record.push_back(E->requiresZeroInitialization());
1066 Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
1067 Writer.AddSourceRange(E->getParenRange(), Record);
1068 Code = serialization::EXPR_CXX_CONSTRUCT;
1071 void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
1072 VisitCXXConstructExpr(E);
1073 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
1074 Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
1077 void ASTStmtWriter::VisitLambdaExpr(LambdaExpr *E) {
1079 Record.push_back(E->NumCaptures);
1080 unsigned NumArrayIndexVars = 0;
1081 if (E->HasArrayIndexVars)
1082 NumArrayIndexVars = E->getArrayIndexStarts()[E->NumCaptures];
1083 Record.push_back(NumArrayIndexVars);
1084 Writer.AddSourceRange(E->IntroducerRange, Record);
1085 Record.push_back(E->CaptureDefault); // FIXME: stable encoding
1086 Record.push_back(E->ExplicitParams);
1087 Record.push_back(E->ExplicitResultType);
1088 Writer.AddSourceLocation(E->ClosingBrace, Record);
1090 // Add capture initializers.
1091 for (LambdaExpr::capture_init_iterator C = E->capture_init_begin(),
1092 CEnd = E->capture_init_end();
1097 // Add array index variables, if any.
1098 if (NumArrayIndexVars) {
1099 Record.append(E->getArrayIndexStarts(),
1100 E->getArrayIndexStarts() + E->NumCaptures + 1);
1101 VarDecl **ArrayIndexVars = E->getArrayIndexVars();
1102 for (unsigned I = 0; I != NumArrayIndexVars; ++I)
1103 Writer.AddDeclRef(ArrayIndexVars[I], Record);
1106 Code = serialization::EXPR_LAMBDA;
1109 void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
1110 VisitExplicitCastExpr(E);
1111 Writer.AddSourceRange(SourceRange(E->getOperatorLoc(), E->getRParenLoc()),
1115 void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
1116 VisitCXXNamedCastExpr(E);
1117 Code = serialization::EXPR_CXX_STATIC_CAST;
1120 void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
1121 VisitCXXNamedCastExpr(E);
1122 Code = serialization::EXPR_CXX_DYNAMIC_CAST;
1125 void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
1126 VisitCXXNamedCastExpr(E);
1127 Code = serialization::EXPR_CXX_REINTERPRET_CAST;
1130 void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
1131 VisitCXXNamedCastExpr(E);
1132 Code = serialization::EXPR_CXX_CONST_CAST;
1135 void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
1136 VisitExplicitCastExpr(E);
1137 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1138 Writer.AddSourceLocation(E->getRParenLoc(), Record);
1139 Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
1142 void ASTStmtWriter::VisitUserDefinedLiteral(UserDefinedLiteral *E) {
1144 Writer.AddSourceLocation(E->UDSuffixLoc, Record);
1145 Code = serialization::EXPR_USER_DEFINED_LITERAL;
1148 void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
1150 Record.push_back(E->getValue());
1151 Writer.AddSourceLocation(E->getLocation(), Record);
1152 Code = serialization::EXPR_CXX_BOOL_LITERAL;
1155 void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
1157 Writer.AddSourceLocation(E->getLocation(), Record);
1158 Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
1161 void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1163 Writer.AddSourceRange(E->getSourceRange(), Record);
1164 if (E->isTypeOperand()) {
1165 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1166 Code = serialization::EXPR_CXX_TYPEID_TYPE;
1168 Writer.AddStmt(E->getExprOperand());
1169 Code = serialization::EXPR_CXX_TYPEID_EXPR;
1173 void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
1175 Writer.AddSourceLocation(E->getLocation(), Record);
1176 Record.push_back(E->isImplicit());
1177 Code = serialization::EXPR_CXX_THIS;
1180 void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
1182 Writer.AddSourceLocation(E->getThrowLoc(), Record);
1183 Writer.AddStmt(E->getSubExpr());
1184 Record.push_back(E->isThrownVariableInScope());
1185 Code = serialization::EXPR_CXX_THROW;
1188 void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
1191 bool HasOtherExprStored = E->Param.getInt();
1192 // Store these first, the reader reads them before creation.
1193 Record.push_back(HasOtherExprStored);
1194 if (HasOtherExprStored)
1195 Writer.AddStmt(E->getExpr());
1196 Writer.AddDeclRef(E->getParam(), Record);
1197 Writer.AddSourceLocation(E->getUsedLocation(), Record);
1199 Code = serialization::EXPR_CXX_DEFAULT_ARG;
1202 void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
1204 Writer.AddCXXTemporary(E->getTemporary(), Record);
1205 Writer.AddStmt(E->getSubExpr());
1206 Code = serialization::EXPR_CXX_BIND_TEMPORARY;
1209 void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
1211 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
1212 Writer.AddSourceLocation(E->getRParenLoc(), Record);
1213 Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
1216 void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
1218 Record.push_back(E->isGlobalNew());
1219 Record.push_back(E->isArray());
1220 Record.push_back(E->doesUsualArrayDeleteWantSize());
1221 Record.push_back(E->getNumPlacementArgs());
1222 Record.push_back(E->StoredInitializationStyle);
1223 Writer.AddDeclRef(E->getOperatorNew(), Record);
1224 Writer.AddDeclRef(E->getOperatorDelete(), Record);
1225 Writer.AddTypeSourceInfo(E->getAllocatedTypeSourceInfo(), Record);
1226 Writer.AddSourceRange(E->getTypeIdParens(), Record);
1227 Writer.AddSourceLocation(E->getStartLoc(), Record);
1228 Writer.AddSourceRange(E->getDirectInitRange(), Record);
1229 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
1233 Code = serialization::EXPR_CXX_NEW;
1236 void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
1238 Record.push_back(E->isGlobalDelete());
1239 Record.push_back(E->isArrayForm());
1240 Record.push_back(E->isArrayFormAsWritten());
1241 Record.push_back(E->doesUsualArrayDeleteWantSize());
1242 Writer.AddDeclRef(E->getOperatorDelete(), Record);
1243 Writer.AddStmt(E->getArgument());
1244 Writer.AddSourceLocation(E->getSourceRange().getBegin(), Record);
1246 Code = serialization::EXPR_CXX_DELETE;
1249 void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1252 Writer.AddStmt(E->getBase());
1253 Record.push_back(E->isArrow());
1254 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1255 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1256 Writer.AddTypeSourceInfo(E->getScopeTypeInfo(), Record);
1257 Writer.AddSourceLocation(E->getColonColonLoc(), Record);
1258 Writer.AddSourceLocation(E->getTildeLoc(), Record);
1260 // PseudoDestructorTypeStorage.
1261 Writer.AddIdentifierRef(E->getDestroyedTypeIdentifier(), Record);
1262 if (E->getDestroyedTypeIdentifier())
1263 Writer.AddSourceLocation(E->getDestroyedTypeLoc(), Record);
1265 Writer.AddTypeSourceInfo(E->getDestroyedTypeInfo(), Record);
1267 Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
1270 void ASTStmtWriter::VisitExprWithCleanups(ExprWithCleanups *E) {
1272 Record.push_back(E->getNumObjects());
1273 for (unsigned i = 0, e = E->getNumObjects(); i != e; ++i)
1274 Writer.AddDeclRef(E->getObject(i), Record);
1276 Writer.AddStmt(E->getSubExpr());
1277 Code = serialization::EXPR_EXPR_WITH_CLEANUPS;
1281 ASTStmtWriter::VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E){
1284 // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1287 Record.push_back(E->HasTemplateKWAndArgsInfo);
1288 if (E->HasTemplateKWAndArgsInfo) {
1289 const ASTTemplateKWAndArgsInfo &Args = *E->getTemplateKWAndArgsInfo();
1290 Record.push_back(Args.NumTemplateArgs);
1291 AddTemplateKWAndArgsInfo(Args);
1294 if (!E->isImplicitAccess())
1295 Writer.AddStmt(E->getBase());
1298 Writer.AddTypeRef(E->getBaseType(), Record);
1299 Record.push_back(E->isArrow());
1300 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1301 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1302 Writer.AddDeclRef(E->getFirstQualifierFoundInScope(), Record);
1303 Writer.AddDeclarationNameInfo(E->MemberNameInfo, Record);
1304 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
1308 ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1311 // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1314 Record.push_back(E->HasTemplateKWAndArgsInfo);
1315 if (E->HasTemplateKWAndArgsInfo) {
1316 const ASTTemplateKWAndArgsInfo &Args = *E->getTemplateKWAndArgsInfo();
1317 Record.push_back(Args.NumTemplateArgs);
1318 AddTemplateKWAndArgsInfo(Args);
1321 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1322 Writer.AddDeclarationNameInfo(E->NameInfo, Record);
1323 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
1327 ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
1329 Record.push_back(E->arg_size());
1330 for (CXXUnresolvedConstructExpr::arg_iterator
1331 ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
1332 Writer.AddStmt(*ArgI);
1333 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
1334 Writer.AddSourceLocation(E->getLParenLoc(), Record);
1335 Writer.AddSourceLocation(E->getRParenLoc(), Record);
1336 Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
1339 void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
1342 // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1345 Record.push_back(E->HasTemplateKWAndArgsInfo);
1346 if (E->HasTemplateKWAndArgsInfo) {
1347 const ASTTemplateKWAndArgsInfo &Args = *E->getTemplateKWAndArgsInfo();
1348 Record.push_back(Args.NumTemplateArgs);
1349 AddTemplateKWAndArgsInfo(Args);
1352 Record.push_back(E->getNumDecls());
1353 for (OverloadExpr::decls_iterator
1354 OvI = E->decls_begin(), OvE = E->decls_end(); OvI != OvE; ++OvI) {
1355 Writer.AddDeclRef(OvI.getDecl(), Record);
1356 Record.push_back(OvI.getAccess());
1359 Writer.AddDeclarationNameInfo(E->NameInfo, Record);
1360 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
1363 void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
1364 VisitOverloadExpr(E);
1365 Record.push_back(E->isArrow());
1366 Record.push_back(E->hasUnresolvedUsing());
1367 Writer.AddStmt(!E->isImplicitAccess() ? E->getBase() : 0);
1368 Writer.AddTypeRef(E->getBaseType(), Record);
1369 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1370 Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
1373 void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
1374 VisitOverloadExpr(E);
1375 Record.push_back(E->requiresADL());
1376 if (E->requiresADL())
1377 Record.push_back(E->isStdAssociatedNamespace());
1378 Record.push_back(E->isOverloaded());
1379 Writer.AddDeclRef(E->getNamingClass(), Record);
1380 Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
1383 void ASTStmtWriter::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
1385 Record.push_back(E->getTrait());
1386 Record.push_back(E->getValue());
1387 Writer.AddSourceRange(E->getSourceRange(), Record);
1388 Writer.AddTypeSourceInfo(E->getQueriedTypeSourceInfo(), Record);
1389 Code = serialization::EXPR_CXX_UNARY_TYPE_TRAIT;
1392 void ASTStmtWriter::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
1394 Record.push_back(E->getTrait());
1395 Record.push_back(E->getValue());
1396 Writer.AddSourceRange(E->getSourceRange(), Record);
1397 Writer.AddTypeSourceInfo(E->getLhsTypeSourceInfo(), Record);
1398 Writer.AddTypeSourceInfo(E->getRhsTypeSourceInfo(), Record);
1399 Code = serialization::EXPR_BINARY_TYPE_TRAIT;
1402 void ASTStmtWriter::VisitTypeTraitExpr(TypeTraitExpr *E) {
1404 Record.push_back(E->TypeTraitExprBits.NumArgs);
1405 Record.push_back(E->TypeTraitExprBits.Kind); // FIXME: Stable encoding
1406 Record.push_back(E->TypeTraitExprBits.Value);
1407 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
1408 Writer.AddTypeSourceInfo(E->getArg(I), Record);
1409 Code = serialization::EXPR_TYPE_TRAIT;
1412 void ASTStmtWriter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
1414 Record.push_back(E->getTrait());
1415 Record.push_back(E->getValue());
1416 Writer.AddSourceRange(E->getSourceRange(), Record);
1417 Writer.AddTypeSourceInfo(E->getQueriedTypeSourceInfo(), Record);
1418 Code = serialization::EXPR_ARRAY_TYPE_TRAIT;
1421 void ASTStmtWriter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
1423 Record.push_back(E->getTrait());
1424 Record.push_back(E->getValue());
1425 Writer.AddSourceRange(E->getSourceRange(), Record);
1426 Writer.AddStmt(E->getQueriedExpression());
1427 Code = serialization::EXPR_CXX_EXPRESSION_TRAIT;
1430 void ASTStmtWriter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
1432 Record.push_back(E->getValue());
1433 Writer.AddSourceRange(E->getSourceRange(), Record);
1434 Writer.AddStmt(E->getOperand());
1435 Code = serialization::EXPR_CXX_NOEXCEPT;
1438 void ASTStmtWriter::VisitPackExpansionExpr(PackExpansionExpr *E) {
1440 Writer.AddSourceLocation(E->getEllipsisLoc(), Record);
1441 Record.push_back(E->NumExpansions);
1442 Writer.AddStmt(E->getPattern());
1443 Code = serialization::EXPR_PACK_EXPANSION;
1446 void ASTStmtWriter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
1448 Writer.AddSourceLocation(E->OperatorLoc, Record);
1449 Writer.AddSourceLocation(E->PackLoc, Record);
1450 Writer.AddSourceLocation(E->RParenLoc, Record);
1451 Record.push_back(E->Length);
1452 Writer.AddDeclRef(E->Pack, Record);
1453 Code = serialization::EXPR_SIZEOF_PACK;
1456 void ASTStmtWriter::VisitSubstNonTypeTemplateParmExpr(
1457 SubstNonTypeTemplateParmExpr *E) {
1459 Writer.AddDeclRef(E->getParameter(), Record);
1460 Writer.AddSourceLocation(E->getNameLoc(), Record);
1461 Writer.AddStmt(E->getReplacement());
1462 Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM;
1465 void ASTStmtWriter::VisitSubstNonTypeTemplateParmPackExpr(
1466 SubstNonTypeTemplateParmPackExpr *E) {
1468 Writer.AddDeclRef(E->getParameterPack(), Record);
1469 Writer.AddTemplateArgument(E->getArgumentPack(), Record);
1470 Writer.AddSourceLocation(E->getParameterPackLocation(), Record);
1471 Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM_PACK;
1474 void ASTStmtWriter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E) {
1476 Writer.AddStmt(E->Temporary);
1477 Code = serialization::EXPR_MATERIALIZE_TEMPORARY;
1480 void ASTStmtWriter::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
1482 Writer.AddStmt(E->getSourceExpr());
1483 Writer.AddSourceLocation(E->getLocation(), Record);
1484 Code = serialization::EXPR_OPAQUE_VALUE;
1487 //===----------------------------------------------------------------------===//
1488 // CUDA Expressions and Statements.
1489 //===----------------------------------------------------------------------===//
1491 void ASTStmtWriter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *E) {
1493 Writer.AddStmt(E->getConfig());
1494 Code = serialization::EXPR_CUDA_KERNEL_CALL;
1497 //===----------------------------------------------------------------------===//
1498 // OpenCL Expressions and Statements.
1499 //===----------------------------------------------------------------------===//
1500 void ASTStmtWriter::VisitAsTypeExpr(AsTypeExpr *E) {
1502 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
1503 Writer.AddSourceLocation(E->getRParenLoc(), Record);
1504 Writer.AddStmt(E->getSrcExpr());
1505 Code = serialization::EXPR_ASTYPE;
1508 //===----------------------------------------------------------------------===//
1509 // Microsoft Expressions and Statements.
1510 //===----------------------------------------------------------------------===//
1511 void ASTStmtWriter::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
1513 Writer.AddSourceRange(E->getSourceRange(), Record);
1514 if (E->isTypeOperand()) {
1515 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1516 Code = serialization::EXPR_CXX_UUIDOF_TYPE;
1518 Writer.AddStmt(E->getExprOperand());
1519 Code = serialization::EXPR_CXX_UUIDOF_EXPR;
1523 void ASTStmtWriter::VisitSEHExceptStmt(SEHExceptStmt *S) {
1525 Writer.AddSourceLocation(S->getExceptLoc(), Record);
1526 Writer.AddStmt(S->getFilterExpr());
1527 Writer.AddStmt(S->getBlock());
1528 Code = serialization::STMT_SEH_EXCEPT;
1531 void ASTStmtWriter::VisitSEHFinallyStmt(SEHFinallyStmt *S) {
1533 Writer.AddSourceLocation(S->getFinallyLoc(), Record);
1534 Writer.AddStmt(S->getBlock());
1535 Code = serialization::STMT_SEH_FINALLY;
1538 void ASTStmtWriter::VisitSEHTryStmt(SEHTryStmt *S) {
1540 Record.push_back(S->getIsCXXTry());
1541 Writer.AddSourceLocation(S->getTryLoc(), Record);
1542 Writer.AddStmt(S->getTryBlock());
1543 Writer.AddStmt(S->getHandler());
1544 Code = serialization::STMT_SEH_TRY;
1547 //===----------------------------------------------------------------------===//
1548 // ASTWriter Implementation
1549 //===----------------------------------------------------------------------===//
1551 unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
1552 assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
1553 "SwitchCase recorded twice");
1554 unsigned NextID = SwitchCaseIDs.size();
1555 SwitchCaseIDs[S] = NextID;
1559 unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
1560 assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
1561 "SwitchCase hasn't been seen yet");
1562 return SwitchCaseIDs[S];
1565 void ASTWriter::ClearSwitchCaseIDs() {
1566 SwitchCaseIDs.clear();
1569 /// \brief Write the given substatement or subexpression to the
1571 void ASTWriter::WriteSubStmt(Stmt *S,
1572 llvm::DenseMap<Stmt *, uint64_t> &SubStmtEntries,
1573 llvm::DenseSet<Stmt *> &ParentStmts) {
1575 ASTStmtWriter Writer(*this, Record);
1579 Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
1583 llvm::DenseMap<Stmt *, uint64_t>::iterator I = SubStmtEntries.find(S);
1584 if (I != SubStmtEntries.end()) {
1585 Record.push_back(I->second);
1586 Stream.EmitRecord(serialization::STMT_REF_PTR, Record);
1591 assert(!ParentStmts.count(S) && "There is a Stmt cycle!");
1593 struct ParentStmtInserterRAII {
1595 llvm::DenseSet<Stmt *> &ParentStmts;
1597 ParentStmtInserterRAII(Stmt *S, llvm::DenseSet<Stmt *> &ParentStmts)
1598 : S(S), ParentStmts(ParentStmts) {
1599 ParentStmts.insert(S);
1601 ~ParentStmtInserterRAII() {
1602 ParentStmts.erase(S);
1606 ParentStmtInserterRAII ParentStmtInserter(S, ParentStmts);
1609 // Redirect ASTWriter::AddStmt to collect sub stmts.
1610 SmallVector<Stmt *, 16> SubStmts;
1611 CollectedStmts = &SubStmts;
1613 Writer.Code = serialization::STMT_NULL_PTR;
1614 Writer.AbbrevToUse = 0;
1618 if (Writer.Code == serialization::STMT_NULL_PTR) {
1619 SourceManager &SrcMgr
1620 = DeclIDs.begin()->first->getASTContext().getSourceManager();
1622 llvm_unreachable("Unhandled sub statement writing AST file");
1626 // Revert ASTWriter::AddStmt.
1627 CollectedStmts = &StmtsToEmit;
1629 // Write the sub stmts in reverse order, last to first. When reading them back
1630 // we will read them in correct order by "pop"ing them from the Stmts stack.
1631 // This simplifies reading and allows to store a variable number of sub stmts
1632 // without knowing it in advance.
1633 while (!SubStmts.empty())
1634 WriteSubStmt(SubStmts.pop_back_val(), SubStmtEntries, ParentStmts);
1636 Stream.EmitRecord(Writer.Code, Record, Writer.AbbrevToUse);
1638 SubStmtEntries[S] = Stream.GetCurrentBitNo();
1641 /// \brief Flush all of the statements that have been added to the
1642 /// queue via AddStmt().
1643 void ASTWriter::FlushStmts() {
1646 // We expect to be the only consumer of the two temporary statement maps,
1647 // assert that they are empty.
1648 assert(SubStmtEntries.empty() && "unexpected entries in sub stmt map");
1649 assert(ParentStmts.empty() && "unexpected entries in parent stmt map");
1651 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
1652 WriteSubStmt(StmtsToEmit[I], SubStmtEntries, ParentStmts);
1654 assert(N == StmtsToEmit.size() &&
1655 "Substatement written via AddStmt rather than WriteSubStmt!");
1657 // Note that we are at the end of a full expression. Any
1658 // expression records that follow this one are part of a different
1660 Stream.EmitRecord(serialization::STMT_STOP, Record);
1662 SubStmtEntries.clear();
1663 ParentStmts.clear();
1666 StmtsToEmit.clear();