blob: 3cfab776d024ee88a54760309a4bdd500d8e38c0 [file] [log] [blame]
Sebastian Redld6522cf2010-08-18 23:56:31 +00001//===--- ASTWriterStmt.cpp - Statement and Expression Serialization -------===//
Chris Lattner1f551822009-04-27 06:20:01 +00002//
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
Sebastian Redl1914c6f2010-08-18 23:56:37 +000014#include "clang/Serialization/ASTWriter.h"
Douglas Gregor5d3507d2009-09-09 23:08:42 +000015#include "clang/AST/DeclCXX.h"
Chris Lattner1f551822009-04-27 06:20:01 +000016#include "clang/AST/DeclObjC.h"
Douglas Gregorcdbc5392011-01-15 01:15:58 +000017#include "clang/AST/DeclTemplate.h"
Chris Lattner1f551822009-04-27 06:20:01 +000018#include "clang/AST/StmtVisitor.h"
19#include "llvm/Bitcode/BitstreamWriter.h"
20using namespace clang;
21
22//===----------------------------------------------------------------------===//
23// Statement/expression serialization
24//===----------------------------------------------------------------------===//
25
Argyrios Kyrtzidisf9f47c82010-06-30 08:49:18 +000026namespace clang {
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000027 class ASTStmtWriter : public StmtVisitor<ASTStmtWriter, void> {
Sebastian Redl55c0ad52010-08-18 23:56:21 +000028 ASTWriter &Writer;
29 ASTWriter::RecordData &Record;
Chris Lattner1f551822009-04-27 06:20:01 +000030
31 public:
Sebastian Redl539c5062010-08-18 23:57:32 +000032 serialization::StmtCode Code;
Douglas Gregor03412ba2011-06-03 02:27:19 +000033 unsigned AbbrevToUse;
Chris Lattner1f551822009-04-27 06:20:01 +000034
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000035 ASTStmtWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
Chris Lattner1f551822009-04-27 06:20:01 +000036 : Writer(Writer), Record(Record) { }
Abramo Bagnara7945c982012-01-27 09:46:47 +000037
38 void AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo &Args);
Chris Lattner1f551822009-04-27 06:20:01 +000039
40 void VisitStmt(Stmt *S);
John McCallfa194042011-07-15 07:00:14 +000041#define STMT(Type, Base) \
42 void Visit##Type(Type *);
43#include "clang/AST/StmtNodes.inc"
Chris Lattner1f551822009-04-27 06:20:01 +000044 };
45}
46
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000047void ASTStmtWriter::
Abramo Bagnara7945c982012-01-27 09:46:47 +000048AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo &Args) {
49 Writer.AddSourceLocation(Args.getTemplateKeywordLoc(), Record);
Argyrios Kyrtzidisb5288de2010-06-28 09:31:48 +000050 Writer.AddSourceLocation(Args.LAngleLoc, Record);
51 Writer.AddSourceLocation(Args.RAngleLoc, Record);
52 for (unsigned i=0; i != Args.NumTemplateArgs; ++i)
53 Writer.AddTemplateArgumentLoc(Args.getTemplateArgs()[i], Record);
54}
55
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000056void ASTStmtWriter::VisitStmt(Stmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +000057}
58
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000059void ASTStmtWriter::VisitNullStmt(NullStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +000060 VisitStmt(S);
61 Writer.AddSourceLocation(S->getSemiLoc(), Record);
Argyrios Kyrtzidis43ea78b2011-09-01 21:53:45 +000062 Record.push_back(S->HasLeadingEmptyMacro);
Sebastian Redl539c5062010-08-18 23:57:32 +000063 Code = serialization::STMT_NULL;
Chris Lattner1f551822009-04-27 06:20:01 +000064}
65
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000066void ASTStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +000067 VisitStmt(S);
68 Record.push_back(S->size());
69 for (CompoundStmt::body_iterator CS = S->body_begin(), CSEnd = S->body_end();
70 CS != CSEnd; ++CS)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +000071 Writer.AddStmt(*CS);
Chris Lattner1f551822009-04-27 06:20:01 +000072 Writer.AddSourceLocation(S->getLBracLoc(), Record);
73 Writer.AddSourceLocation(S->getRBracLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +000074 Code = serialization::STMT_COMPOUND;
Chris Lattner1f551822009-04-27 06:20:01 +000075}
76
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000077void ASTStmtWriter::VisitSwitchCase(SwitchCase *S) {
Chris Lattner1f551822009-04-27 06:20:01 +000078 VisitStmt(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +000079 Record.push_back(Writer.getSwitchCaseID(S));
Chris Lattner1f551822009-04-27 06:20:01 +000080}
81
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000082void ASTStmtWriter::VisitCaseStmt(CaseStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +000083 VisitSwitchCase(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +000084 Writer.AddStmt(S->getLHS());
85 Writer.AddStmt(S->getRHS());
86 Writer.AddStmt(S->getSubStmt());
Chris Lattner1f551822009-04-27 06:20:01 +000087 Writer.AddSourceLocation(S->getCaseLoc(), Record);
Douglas Gregor2a2d00f2009-05-15 23:57:33 +000088 Writer.AddSourceLocation(S->getEllipsisLoc(), Record);
89 Writer.AddSourceLocation(S->getColonLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +000090 Code = serialization::STMT_CASE;
Chris Lattner1f551822009-04-27 06:20:01 +000091}
92
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000093void ASTStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +000094 VisitSwitchCase(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +000095 Writer.AddStmt(S->getSubStmt());
Chris Lattner1f551822009-04-27 06:20:01 +000096 Writer.AddSourceLocation(S->getDefaultLoc(), Record);
Douglas Gregor2a2d00f2009-05-15 23:57:33 +000097 Writer.AddSourceLocation(S->getColonLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +000098 Code = serialization::STMT_DEFAULT;
Chris Lattner1f551822009-04-27 06:20:01 +000099}
100
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000101void ASTStmtWriter::VisitLabelStmt(LabelStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000102 VisitStmt(S);
Chris Lattnerc8e630e2011-02-17 07:39:24 +0000103 Writer.AddDeclRef(S->getDecl(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000104 Writer.AddStmt(S->getSubStmt());
Chris Lattner1f551822009-04-27 06:20:01 +0000105 Writer.AddSourceLocation(S->getIdentLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000106 Code = serialization::STMT_LABEL;
Chris Lattner1f551822009-04-27 06:20:01 +0000107}
108
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000109void ASTStmtWriter::VisitIfStmt(IfStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000110 VisitStmt(S);
Douglas Gregor633caca2009-11-23 23:44:04 +0000111 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000112 Writer.AddStmt(S->getCond());
113 Writer.AddStmt(S->getThen());
114 Writer.AddStmt(S->getElse());
Chris Lattner1f551822009-04-27 06:20:01 +0000115 Writer.AddSourceLocation(S->getIfLoc(), Record);
Douglas Gregor9d73cab2009-05-15 18:53:42 +0000116 Writer.AddSourceLocation(S->getElseLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000117 Code = serialization::STMT_IF;
Chris Lattner1f551822009-04-27 06:20:01 +0000118}
119
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000120void ASTStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000121 VisitStmt(S);
Douglas Gregordcf19622009-11-24 17:07:59 +0000122 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000123 Writer.AddStmt(S->getCond());
124 Writer.AddStmt(S->getBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000125 Writer.AddSourceLocation(S->getSwitchLoc(), Record);
Ted Kremenekc42f3452010-09-09 00:05:53 +0000126 Record.push_back(S->isAllEnumCasesCovered());
Mike Stump11289f42009-09-09 15:08:12 +0000127 for (SwitchCase *SC = S->getSwitchCaseList(); SC;
Chris Lattner1f551822009-04-27 06:20:01 +0000128 SC = SC->getNextSwitchCase())
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000129 Record.push_back(Writer.RecordSwitchCaseID(SC));
Sebastian Redl539c5062010-08-18 23:57:32 +0000130 Code = serialization::STMT_SWITCH;
Chris Lattner1f551822009-04-27 06:20:01 +0000131}
132
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000133void ASTStmtWriter::VisitWhileStmt(WhileStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000134 VisitStmt(S);
Douglas Gregor680f8612009-11-24 21:15:44 +0000135 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000136 Writer.AddStmt(S->getCond());
137 Writer.AddStmt(S->getBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000138 Writer.AddSourceLocation(S->getWhileLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000139 Code = serialization::STMT_WHILE;
Chris Lattner1f551822009-04-27 06:20:01 +0000140}
141
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000142void ASTStmtWriter::VisitDoStmt(DoStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000143 VisitStmt(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000144 Writer.AddStmt(S->getCond());
145 Writer.AddStmt(S->getBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000146 Writer.AddSourceLocation(S->getDoLoc(), Record);
Douglas Gregor3daa82d2009-05-15 21:56:04 +0000147 Writer.AddSourceLocation(S->getWhileLoc(), Record);
Chris Lattner815b70e2009-06-12 23:04:47 +0000148 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000149 Code = serialization::STMT_DO;
Chris Lattner1f551822009-04-27 06:20:01 +0000150}
151
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000152void ASTStmtWriter::VisitForStmt(ForStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000153 VisitStmt(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000154 Writer.AddStmt(S->getInit());
155 Writer.AddStmt(S->getCond());
Douglas Gregor7bab5ff2009-11-25 00:27:52 +0000156 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000157 Writer.AddStmt(S->getInc());
158 Writer.AddStmt(S->getBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000159 Writer.AddSourceLocation(S->getForLoc(), Record);
Douglas Gregor5d138682009-05-15 22:12:32 +0000160 Writer.AddSourceLocation(S->getLParenLoc(), Record);
161 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000162 Code = serialization::STMT_FOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000163}
164
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000165void ASTStmtWriter::VisitGotoStmt(GotoStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000166 VisitStmt(S);
Chris Lattnerc8e630e2011-02-17 07:39:24 +0000167 Writer.AddDeclRef(S->getLabel(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000168 Writer.AddSourceLocation(S->getGotoLoc(), Record);
169 Writer.AddSourceLocation(S->getLabelLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000170 Code = serialization::STMT_GOTO;
Chris Lattner1f551822009-04-27 06:20:01 +0000171}
172
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000173void ASTStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000174 VisitStmt(S);
175 Writer.AddSourceLocation(S->getGotoLoc(), Record);
Douglas Gregor30776d42009-05-16 00:20:29 +0000176 Writer.AddSourceLocation(S->getStarLoc(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000177 Writer.AddStmt(S->getTarget());
Sebastian Redl539c5062010-08-18 23:57:32 +0000178 Code = serialization::STMT_INDIRECT_GOTO;
Chris Lattner1f551822009-04-27 06:20:01 +0000179}
180
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000181void ASTStmtWriter::VisitContinueStmt(ContinueStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000182 VisitStmt(S);
183 Writer.AddSourceLocation(S->getContinueLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000184 Code = serialization::STMT_CONTINUE;
Chris Lattner1f551822009-04-27 06:20:01 +0000185}
186
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000187void ASTStmtWriter::VisitBreakStmt(BreakStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000188 VisitStmt(S);
189 Writer.AddSourceLocation(S->getBreakLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000190 Code = serialization::STMT_BREAK;
Chris Lattner1f551822009-04-27 06:20:01 +0000191}
192
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000193void ASTStmtWriter::VisitReturnStmt(ReturnStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000194 VisitStmt(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000195 Writer.AddStmt(S->getRetValue());
Chris Lattner1f551822009-04-27 06:20:01 +0000196 Writer.AddSourceLocation(S->getReturnLoc(), Record);
Douglas Gregor6fd1b182010-05-15 06:01:05 +0000197 Writer.AddDeclRef(S->getNRVOCandidate(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000198 Code = serialization::STMT_RETURN;
Chris Lattner1f551822009-04-27 06:20:01 +0000199}
200
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000201void ASTStmtWriter::VisitDeclStmt(DeclStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000202 VisitStmt(S);
203 Writer.AddSourceLocation(S->getStartLoc(), Record);
204 Writer.AddSourceLocation(S->getEndLoc(), Record);
205 DeclGroupRef DG = S->getDeclGroup();
206 for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
207 Writer.AddDeclRef(*D, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000208 Code = serialization::STMT_DECL;
Chris Lattner1f551822009-04-27 06:20:01 +0000209}
210
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000211void ASTStmtWriter::VisitAsmStmt(AsmStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000212 VisitStmt(S);
213 Record.push_back(S->getNumOutputs());
214 Record.push_back(S->getNumInputs());
215 Record.push_back(S->getNumClobbers());
216 Writer.AddSourceLocation(S->getAsmLoc(), Record);
217 Writer.AddSourceLocation(S->getRParenLoc(), Record);
218 Record.push_back(S->isVolatile());
219 Record.push_back(S->isSimple());
Mike Stump90be58a2010-01-04 22:37:17 +0000220 Record.push_back(S->isMSAsm());
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000221 Writer.AddStmt(S->getAsmString());
Chris Lattner1f551822009-04-27 06:20:01 +0000222
223 // Outputs
Anders Carlsson9a020f92010-01-30 22:25:16 +0000224 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
225 Writer.AddIdentifierRef(S->getOutputIdentifier(I), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000226 Writer.AddStmt(S->getOutputConstraintLiteral(I));
227 Writer.AddStmt(S->getOutputExpr(I));
Chris Lattner1f551822009-04-27 06:20:01 +0000228 }
229
230 // Inputs
231 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
Anders Carlsson9a020f92010-01-30 22:25:16 +0000232 Writer.AddIdentifierRef(S->getInputIdentifier(I), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000233 Writer.AddStmt(S->getInputConstraintLiteral(I));
234 Writer.AddStmt(S->getInputExpr(I));
Chris Lattner1f551822009-04-27 06:20:01 +0000235 }
236
237 // Clobbers
238 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000239 Writer.AddStmt(S->getClobber(I));
Chris Lattner1f551822009-04-27 06:20:01 +0000240
Sebastian Redl539c5062010-08-18 23:57:32 +0000241 Code = serialization::STMT_ASM;
Chris Lattner1f551822009-04-27 06:20:01 +0000242}
243
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000244void ASTStmtWriter::VisitExpr(Expr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000245 VisitStmt(E);
246 Writer.AddTypeRef(E->getType(), Record);
247 Record.push_back(E->isTypeDependent());
248 Record.push_back(E->isValueDependent());
Douglas Gregor678d76c2011-07-01 01:22:09 +0000249 Record.push_back(E->isInstantiationDependent());
Douglas Gregor506bd562010-12-13 22:49:22 +0000250 Record.push_back(E->containsUnexpandedParameterPack());
John McCall7decc9e2010-11-18 06:31:45 +0000251 Record.push_back(E->getValueKind());
252 Record.push_back(E->getObjectKind());
Chris Lattner1f551822009-04-27 06:20:01 +0000253}
254
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000255void ASTStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000256 VisitExpr(E);
257 Writer.AddSourceLocation(E->getLocation(), Record);
258 Record.push_back(E->getIdentType()); // FIXME: stable encoding
Sebastian Redl539c5062010-08-18 23:57:32 +0000259 Code = serialization::EXPR_PREDEFINED;
Chris Lattner1f551822009-04-27 06:20:01 +0000260}
261
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000262void ASTStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000263 VisitExpr(E);
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000264
265 Record.push_back(E->hasQualifier());
Chandler Carruth8d26bb02011-05-01 23:48:14 +0000266 Record.push_back(E->getDecl() != E->getFoundDecl());
Abramo Bagnara7945c982012-01-27 09:46:47 +0000267 Record.push_back(E->hasTemplateKWAndArgsInfo());
Abramo Bagnara635ed24e2011-10-05 07:56:41 +0000268 Record.push_back(E->hadMultipleCandidates());
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000269
Abramo Bagnara7945c982012-01-27 09:46:47 +0000270 if (E->hasTemplateKWAndArgsInfo()) {
Douglas Gregor87866ce2011-02-04 12:01:24 +0000271 unsigned NumTemplateArgs = E->getNumTemplateArgs();
272 Record.push_back(NumTemplateArgs);
Douglas Gregor87866ce2011-02-04 12:01:24 +0000273 }
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000274
Douglas Gregor03412ba2011-06-03 02:27:19 +0000275 DeclarationName::NameKind nk = (E->getDecl()->getDeclName().getNameKind());
276
Abramo Bagnara7945c982012-01-27 09:46:47 +0000277 if ((!E->hasTemplateKWAndArgsInfo()) && (!E->hasQualifier()) &&
Douglas Gregor03412ba2011-06-03 02:27:19 +0000278 (E->getDecl() == E->getFoundDecl()) &&
279 nk == DeclarationName::Identifier) {
280 AbbrevToUse = Writer.getDeclRefExprAbbrev();
281 }
282
Anders Carlsson80756f62011-03-06 18:19:42 +0000283 if (E->hasQualifier())
284 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
285
Chandler Carruth8d26bb02011-05-01 23:48:14 +0000286 if (E->getDecl() != E->getFoundDecl())
287 Writer.AddDeclRef(E->getFoundDecl(), Record);
288
Abramo Bagnara7945c982012-01-27 09:46:47 +0000289 if (E->hasTemplateKWAndArgsInfo())
290 AddTemplateKWAndArgsInfo(*E->getTemplateKWAndArgsInfo());
Anders Carlsson80756f62011-03-06 18:19:42 +0000291
Chris Lattner1f551822009-04-27 06:20:01 +0000292 Writer.AddDeclRef(E->getDecl(), Record);
293 Writer.AddSourceLocation(E->getLocation(), Record);
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +0000294 Writer.AddDeclarationNameLoc(E->DNLoc, E->getDecl()->getDeclName(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000295 Code = serialization::EXPR_DECL_REF;
Chris Lattner1f551822009-04-27 06:20:01 +0000296}
297
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000298void ASTStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000299 VisitExpr(E);
300 Writer.AddSourceLocation(E->getLocation(), Record);
301 Writer.AddAPInt(E->getValue(), Record);
Douglas Gregor03412ba2011-06-03 02:27:19 +0000302
303 if (E->getValue().getBitWidth() == 32) {
304 AbbrevToUse = Writer.getIntegerLiteralAbbrev();
305 }
306
Sebastian Redl539c5062010-08-18 23:57:32 +0000307 Code = serialization::EXPR_INTEGER_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000308}
309
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000310void ASTStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000311 VisitExpr(E);
312 Writer.AddAPFloat(E->getValue(), Record);
313 Record.push_back(E->isExact());
314 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000315 Code = serialization::EXPR_FLOATING_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000316}
317
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000318void ASTStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000319 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000320 Writer.AddStmt(E->getSubExpr());
Sebastian Redl539c5062010-08-18 23:57:32 +0000321 Code = serialization::EXPR_IMAGINARY_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000322}
323
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000324void ASTStmtWriter::VisitStringLiteral(StringLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000325 VisitExpr(E);
326 Record.push_back(E->getByteLength());
327 Record.push_back(E->getNumConcatenated());
Douglas Gregorfb65e592011-07-27 05:40:30 +0000328 Record.push_back(E->getKind());
Anders Carlsson75245402011-04-14 00:40:03 +0000329 Record.push_back(E->isPascal());
Chris Lattner1f551822009-04-27 06:20:01 +0000330 // FIXME: String data should be stored as a blob at the end of the
331 // StringLiteral. However, we can't do so now because we have no
332 // provision for coping with abbreviations when we're jumping around
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000333 // the AST file during deserialization.
Eli Friedmanfcec6302011-11-01 02:23:42 +0000334 Record.append(E->getBytes().begin(), E->getBytes().end());
Chris Lattner1f551822009-04-27 06:20:01 +0000335 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
336 Writer.AddSourceLocation(E->getStrTokenLoc(I), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000337 Code = serialization::EXPR_STRING_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000338}
339
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000340void ASTStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000341 VisitExpr(E);
342 Record.push_back(E->getValue());
Chris Lattnere6434cb2009-08-24 17:39:36 +0000343 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregorfb65e592011-07-27 05:40:30 +0000344 Record.push_back(E->getKind());
Douglas Gregor03412ba2011-06-03 02:27:19 +0000345
346 AbbrevToUse = Writer.getCharacterLiteralAbbrev();
347
Sebastian Redl539c5062010-08-18 23:57:32 +0000348 Code = serialization::EXPR_CHARACTER_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000349}
350
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000351void ASTStmtWriter::VisitParenExpr(ParenExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000352 VisitExpr(E);
353 Writer.AddSourceLocation(E->getLParen(), Record);
354 Writer.AddSourceLocation(E->getRParen(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000355 Writer.AddStmt(E->getSubExpr());
Sebastian Redl539c5062010-08-18 23:57:32 +0000356 Code = serialization::EXPR_PAREN;
Chris Lattner1f551822009-04-27 06:20:01 +0000357}
358
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000359void ASTStmtWriter::VisitParenListExpr(ParenListExpr *E) {
Argyrios Kyrtzidisf9f47c82010-06-30 08:49:18 +0000360 VisitExpr(E);
361 Record.push_back(E->NumExprs);
362 for (unsigned i=0; i != E->NumExprs; ++i)
363 Writer.AddStmt(E->Exprs[i]);
364 Writer.AddSourceLocation(E->LParenLoc, Record);
365 Writer.AddSourceLocation(E->RParenLoc, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000366 Code = serialization::EXPR_PAREN_LIST;
Argyrios Kyrtzidisf9f47c82010-06-30 08:49:18 +0000367}
368
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000369void ASTStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000370 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000371 Writer.AddStmt(E->getSubExpr());
Chris Lattner1f551822009-04-27 06:20:01 +0000372 Record.push_back(E->getOpcode()); // FIXME: stable encoding
373 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000374 Code = serialization::EXPR_UNARY_OPERATOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000375}
376
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000377void ASTStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
Douglas Gregor882211c2010-04-28 22:16:22 +0000378 VisitExpr(E);
379 Record.push_back(E->getNumComponents());
380 Record.push_back(E->getNumExpressions());
381 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
382 Writer.AddSourceLocation(E->getRParenLoc(), Record);
383 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
384 for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
385 const OffsetOfExpr::OffsetOfNode &ON = E->getComponent(I);
386 Record.push_back(ON.getKind()); // FIXME: Stable encoding
Abramo Bagnara6b6f0512011-03-12 09:45:03 +0000387 Writer.AddSourceLocation(ON.getSourceRange().getBegin(), Record);
388 Writer.AddSourceLocation(ON.getSourceRange().getEnd(), Record);
Douglas Gregor882211c2010-04-28 22:16:22 +0000389 switch (ON.getKind()) {
390 case OffsetOfExpr::OffsetOfNode::Array:
391 Record.push_back(ON.getArrayExprIndex());
392 break;
393
394 case OffsetOfExpr::OffsetOfNode::Field:
395 Writer.AddDeclRef(ON.getField(), Record);
396 break;
397
398 case OffsetOfExpr::OffsetOfNode::Identifier:
399 Writer.AddIdentifierRef(ON.getFieldName(), Record);
400 break;
Douglas Gregord1702062010-04-29 00:18:15 +0000401
402 case OffsetOfExpr::OffsetOfNode::Base:
Argyrios Kyrtzidisd67d4cc2010-07-29 18:16:10 +0000403 Writer.AddCXXBaseSpecifier(*ON.getBase(), Record);
Douglas Gregord1702062010-04-29 00:18:15 +0000404 break;
Douglas Gregor882211c2010-04-28 22:16:22 +0000405 }
406 }
407 for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000408 Writer.AddStmt(E->getIndexExpr(I));
Sebastian Redl539c5062010-08-18 23:57:32 +0000409 Code = serialization::EXPR_OFFSETOF;
Douglas Gregor882211c2010-04-28 22:16:22 +0000410}
411
Peter Collingbournee190dee2011-03-11 19:24:49 +0000412void ASTStmtWriter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000413 VisitExpr(E);
Peter Collingbournee190dee2011-03-11 19:24:49 +0000414 Record.push_back(E->getKind());
Chris Lattner1f551822009-04-27 06:20:01 +0000415 if (E->isArgumentType())
John McCallbcd03502009-12-07 02:54:59 +0000416 Writer.AddTypeSourceInfo(E->getArgumentTypeInfo(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000417 else {
418 Record.push_back(0);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000419 Writer.AddStmt(E->getArgumentExpr());
Chris Lattner1f551822009-04-27 06:20:01 +0000420 }
421 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
422 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000423 Code = serialization::EXPR_SIZEOF_ALIGN_OF;
Chris Lattner1f551822009-04-27 06:20:01 +0000424}
425
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000426void ASTStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000427 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000428 Writer.AddStmt(E->getLHS());
429 Writer.AddStmt(E->getRHS());
Chris Lattner1f551822009-04-27 06:20:01 +0000430 Writer.AddSourceLocation(E->getRBracketLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000431 Code = serialization::EXPR_ARRAY_SUBSCRIPT;
Chris Lattner1f551822009-04-27 06:20:01 +0000432}
433
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000434void ASTStmtWriter::VisitCallExpr(CallExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000435 VisitExpr(E);
436 Record.push_back(E->getNumArgs());
437 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000438 Writer.AddStmt(E->getCallee());
Chris Lattner1f551822009-04-27 06:20:01 +0000439 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
440 Arg != ArgEnd; ++Arg)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000441 Writer.AddStmt(*Arg);
Sebastian Redl539c5062010-08-18 23:57:32 +0000442 Code = serialization::EXPR_CALL;
Chris Lattner1f551822009-04-27 06:20:01 +0000443}
444
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000445void ASTStmtWriter::VisitMemberExpr(MemberExpr *E) {
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000446 // Don't call VisitExpr, we'll write everything here.
447
448 Record.push_back(E->hasQualifier());
Douglas Gregorea972d32011-02-28 21:54:11 +0000449 if (E->hasQualifier())
450 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000451
Abramo Bagnara7945c982012-01-27 09:46:47 +0000452 Record.push_back(E->HasTemplateKWAndArgsInfo);
453 if (E->HasTemplateKWAndArgsInfo) {
454 Writer.AddSourceLocation(E->getTemplateKeywordLoc(), Record);
Douglas Gregor87866ce2011-02-04 12:01:24 +0000455 unsigned NumTemplateArgs = E->getNumTemplateArgs();
456 Record.push_back(NumTemplateArgs);
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000457 Writer.AddSourceLocation(E->getLAngleLoc(), Record);
458 Writer.AddSourceLocation(E->getRAngleLoc(), Record);
459 for (unsigned i=0; i != NumTemplateArgs; ++i)
460 Writer.AddTemplateArgumentLoc(E->getTemplateArgs()[i], Record);
461 }
Abramo Bagnara635ed24e2011-10-05 07:56:41 +0000462
463 Record.push_back(E->hadMultipleCandidates());
464
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000465 DeclAccessPair FoundDecl = E->getFoundDecl();
466 Writer.AddDeclRef(FoundDecl.getDecl(), Record);
467 Record.push_back(FoundDecl.getAccess());
468
469 Writer.AddTypeRef(E->getType(), Record);
John McCall7decc9e2010-11-18 06:31:45 +0000470 Record.push_back(E->getValueKind());
471 Record.push_back(E->getObjectKind());
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000472 Writer.AddStmt(E->getBase());
Chris Lattner1f551822009-04-27 06:20:01 +0000473 Writer.AddDeclRef(E->getMemberDecl(), Record);
474 Writer.AddSourceLocation(E->getMemberLoc(), Record);
475 Record.push_back(E->isArrow());
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +0000476 Writer.AddDeclarationNameLoc(E->MemberDNLoc,
477 E->getMemberDecl()->getDeclName(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000478 Code = serialization::EXPR_MEMBER;
Chris Lattner1f551822009-04-27 06:20:01 +0000479}
480
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000481void ASTStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
Steve Naroffe87026a2009-07-24 17:54:45 +0000482 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000483 Writer.AddStmt(E->getBase());
Steve Naroffe87026a2009-07-24 17:54:45 +0000484 Writer.AddSourceLocation(E->getIsaMemberLoc(), Record);
485 Record.push_back(E->isArrow());
Sebastian Redl539c5062010-08-18 23:57:32 +0000486 Code = serialization::EXPR_OBJC_ISA;
Steve Naroffe87026a2009-07-24 17:54:45 +0000487}
488
John McCall31168b02011-06-15 23:02:42 +0000489void ASTStmtWriter::
490VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) {
491 VisitExpr(E);
492 Writer.AddStmt(E->getSubExpr());
493 Record.push_back(E->shouldCopy());
494 Code = serialization::EXPR_OBJC_INDIRECT_COPY_RESTORE;
495}
496
497void ASTStmtWriter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) {
498 VisitExplicitCastExpr(E);
499 Writer.AddSourceLocation(E->getLParenLoc(), Record);
500 Writer.AddSourceLocation(E->getBridgeKeywordLoc(), Record);
501 Record.push_back(E->getBridgeKind()); // FIXME: Stable encoding
502 Code = serialization::EXPR_OBJC_BRIDGED_CAST;
503}
504
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000505void ASTStmtWriter::VisitCastExpr(CastExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000506 VisitExpr(E);
John McCallcf142162010-08-07 06:22:56 +0000507 Record.push_back(E->path_size());
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000508 Writer.AddStmt(E->getSubExpr());
Anders Carlssona2615922009-07-31 00:48:10 +0000509 Record.push_back(E->getCastKind()); // FIXME: stable encoding
John McCallcf142162010-08-07 06:22:56 +0000510
511 for (CastExpr::path_iterator
512 PI = E->path_begin(), PE = E->path_end(); PI != PE; ++PI)
513 Writer.AddCXXBaseSpecifier(**PI, Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000514}
515
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000516void ASTStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000517 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000518 Writer.AddStmt(E->getLHS());
519 Writer.AddStmt(E->getRHS());
Chris Lattner1f551822009-04-27 06:20:01 +0000520 Record.push_back(E->getOpcode()); // FIXME: stable encoding
521 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000522 Code = serialization::EXPR_BINARY_OPERATOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000523}
524
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000525void ASTStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000526 VisitBinaryOperator(E);
527 Writer.AddTypeRef(E->getComputationLHSType(), Record);
528 Writer.AddTypeRef(E->getComputationResultType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000529 Code = serialization::EXPR_COMPOUND_ASSIGN_OPERATOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000530}
531
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000532void ASTStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000533 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000534 Writer.AddStmt(E->getCond());
535 Writer.AddStmt(E->getLHS());
536 Writer.AddStmt(E->getRHS());
Douglas Gregor7e112b02009-08-26 14:37:04 +0000537 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
538 Writer.AddSourceLocation(E->getColonLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000539 Code = serialization::EXPR_CONDITIONAL_OPERATOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000540}
541
John McCallc07a0c72011-02-17 10:25:35 +0000542void
543ASTStmtWriter::VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
544 VisitExpr(E);
545 Writer.AddStmt(E->getOpaqueValue());
546 Writer.AddStmt(E->getCommon());
547 Writer.AddStmt(E->getCond());
548 Writer.AddStmt(E->getTrueExpr());
549 Writer.AddStmt(E->getFalseExpr());
550 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
551 Writer.AddSourceLocation(E->getColonLoc(), Record);
552 Code = serialization::EXPR_BINARY_CONDITIONAL_OPERATOR;
553}
554
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000555void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000556 VisitCastExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +0000557 Code = serialization::EXPR_IMPLICIT_CAST;
Chris Lattner1f551822009-04-27 06:20:01 +0000558}
559
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000560void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000561 VisitCastExpr(E);
John McCall97513962010-01-15 18:39:57 +0000562 Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000563}
564
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000565void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000566 VisitExplicitCastExpr(E);
567 Writer.AddSourceLocation(E->getLParenLoc(), Record);
568 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000569 Code = serialization::EXPR_CSTYLE_CAST;
Chris Lattner1f551822009-04-27 06:20:01 +0000570}
571
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000572void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000573 VisitExpr(E);
574 Writer.AddSourceLocation(E->getLParenLoc(), Record);
John McCalle15bbff2010-01-18 19:35:47 +0000575 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000576 Writer.AddStmt(E->getInitializer());
Chris Lattner1f551822009-04-27 06:20:01 +0000577 Record.push_back(E->isFileScope());
Sebastian Redl539c5062010-08-18 23:57:32 +0000578 Code = serialization::EXPR_COMPOUND_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000579}
580
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000581void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000582 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000583 Writer.AddStmt(E->getBase());
Chris Lattner1f551822009-04-27 06:20:01 +0000584 Writer.AddIdentifierRef(&E->getAccessor(), Record);
585 Writer.AddSourceLocation(E->getAccessorLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000586 Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
Chris Lattner1f551822009-04-27 06:20:01 +0000587}
588
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000589void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000590 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000591 Writer.AddStmt(E->getSyntacticForm());
Chris Lattner1f551822009-04-27 06:20:01 +0000592 Writer.AddSourceLocation(E->getLBraceLoc(), Record);
593 Writer.AddSourceLocation(E->getRBraceLoc(), Record);
Argyrios Kyrtzidisb2ed28e2011-04-21 00:27:41 +0000594 bool isArrayFiller = E->ArrayFillerOrUnionFieldInit.is<Expr*>();
595 Record.push_back(isArrayFiller);
596 if (isArrayFiller)
597 Writer.AddStmt(E->getArrayFiller());
598 else
599 Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000600 Record.push_back(E->hadArrayRangeDesignator());
Argyrios Kyrtzidisbbcefa72011-04-22 05:29:30 +0000601 Record.push_back(E->getNumInits());
602 if (isArrayFiller) {
603 // ArrayFiller may have filled "holes" due to designated initializer.
604 // Replace them by 0 to indicate that the filler goes in that place.
605 Expr *filler = E->getArrayFiller();
606 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
607 Writer.AddStmt(E->getInit(I) != filler ? E->getInit(I) : 0);
608 } else {
609 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
610 Writer.AddStmt(E->getInit(I));
611 }
Sebastian Redl539c5062010-08-18 23:57:32 +0000612 Code = serialization::EXPR_INIT_LIST;
Chris Lattner1f551822009-04-27 06:20:01 +0000613}
614
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000615void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000616 VisitExpr(E);
617 Record.push_back(E->getNumSubExprs());
618 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000619 Writer.AddStmt(E->getSubExpr(I));
Chris Lattner1f551822009-04-27 06:20:01 +0000620 Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
621 Record.push_back(E->usesGNUSyntax());
622 for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
623 DEnd = E->designators_end();
624 D != DEnd; ++D) {
625 if (D->isFieldDesignator()) {
626 if (FieldDecl *Field = D->getField()) {
Sebastian Redl539c5062010-08-18 23:57:32 +0000627 Record.push_back(serialization::DESIG_FIELD_DECL);
Chris Lattner1f551822009-04-27 06:20:01 +0000628 Writer.AddDeclRef(Field, Record);
629 } else {
Sebastian Redl539c5062010-08-18 23:57:32 +0000630 Record.push_back(serialization::DESIG_FIELD_NAME);
Chris Lattner1f551822009-04-27 06:20:01 +0000631 Writer.AddIdentifierRef(D->getFieldName(), Record);
632 }
633 Writer.AddSourceLocation(D->getDotLoc(), Record);
634 Writer.AddSourceLocation(D->getFieldLoc(), Record);
635 } else if (D->isArrayDesignator()) {
Sebastian Redl539c5062010-08-18 23:57:32 +0000636 Record.push_back(serialization::DESIG_ARRAY);
Chris Lattner1f551822009-04-27 06:20:01 +0000637 Record.push_back(D->getFirstExprIndex());
638 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
639 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
640 } else {
641 assert(D->isArrayRangeDesignator() && "Unknown designator");
Sebastian Redl539c5062010-08-18 23:57:32 +0000642 Record.push_back(serialization::DESIG_ARRAY_RANGE);
Chris Lattner1f551822009-04-27 06:20:01 +0000643 Record.push_back(D->getFirstExprIndex());
644 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
645 Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
646 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
647 }
648 }
Sebastian Redl539c5062010-08-18 23:57:32 +0000649 Code = serialization::EXPR_DESIGNATED_INIT;
Chris Lattner1f551822009-04-27 06:20:01 +0000650}
651
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000652void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000653 VisitExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +0000654 Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
Chris Lattner1f551822009-04-27 06:20:01 +0000655}
656
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000657void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000658 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000659 Writer.AddStmt(E->getSubExpr());
Abramo Bagnara27db2392010-08-10 10:06:15 +0000660 Writer.AddTypeSourceInfo(E->getWrittenTypeInfo(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000661 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
662 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000663 Code = serialization::EXPR_VA_ARG;
Chris Lattner1f551822009-04-27 06:20:01 +0000664}
665
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000666void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000667 VisitExpr(E);
668 Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
669 Writer.AddSourceLocation(E->getLabelLoc(), Record);
Chris Lattnerc8e630e2011-02-17 07:39:24 +0000670 Writer.AddDeclRef(E->getLabel(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000671 Code = serialization::EXPR_ADDR_LABEL;
Chris Lattner1f551822009-04-27 06:20:01 +0000672}
673
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000674void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000675 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000676 Writer.AddStmt(E->getSubStmt());
Chris Lattner1f551822009-04-27 06:20:01 +0000677 Writer.AddSourceLocation(E->getLParenLoc(), Record);
678 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000679 Code = serialization::EXPR_STMT;
Chris Lattner1f551822009-04-27 06:20:01 +0000680}
681
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000682void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000683 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000684 Writer.AddStmt(E->getCond());
685 Writer.AddStmt(E->getLHS());
686 Writer.AddStmt(E->getRHS());
Chris Lattner1f551822009-04-27 06:20:01 +0000687 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
688 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000689 Code = serialization::EXPR_CHOOSE;
Chris Lattner1f551822009-04-27 06:20:01 +0000690}
691
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000692void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000693 VisitExpr(E);
694 Writer.AddSourceLocation(E->getTokenLocation(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000695 Code = serialization::EXPR_GNU_NULL;
Chris Lattner1f551822009-04-27 06:20:01 +0000696}
697
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000698void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000699 VisitExpr(E);
700 Record.push_back(E->getNumSubExprs());
701 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000702 Writer.AddStmt(E->getExpr(I));
Chris Lattner1f551822009-04-27 06:20:01 +0000703 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
704 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000705 Code = serialization::EXPR_SHUFFLE_VECTOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000706}
707
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000708void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000709 VisitExpr(E);
710 Writer.AddDeclRef(E->getBlockDecl(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000711 Code = serialization::EXPR_BLOCK;
Chris Lattner1f551822009-04-27 06:20:01 +0000712}
713
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000714void ASTStmtWriter::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000715 VisitExpr(E);
716 Writer.AddDeclRef(E->getDecl(), Record);
717 Writer.AddSourceLocation(E->getLocation(), Record);
718 Record.push_back(E->isByRef());
Fariborz Jahaniane8918d52009-06-20 00:02:26 +0000719 Record.push_back(E->isConstQualAdded());
Sebastian Redl539c5062010-08-18 23:57:32 +0000720 Code = serialization::EXPR_BLOCK_DECL_REF;
Chris Lattner1f551822009-04-27 06:20:01 +0000721}
722
Peter Collingbourne91147592011-04-15 00:35:48 +0000723void ASTStmtWriter::VisitGenericSelectionExpr(GenericSelectionExpr *E) {
724 VisitExpr(E);
725 Record.push_back(E->getNumAssocs());
726
727 Writer.AddStmt(E->getControllingExpr());
728 for (unsigned I = 0, N = E->getNumAssocs(); I != N; ++I) {
729 Writer.AddTypeSourceInfo(E->getAssocTypeSourceInfo(I), Record);
730 Writer.AddStmt(E->getAssocExpr(I));
731 }
732 Record.push_back(E->isResultDependent() ? -1U : E->getResultIndex());
733
734 Writer.AddSourceLocation(E->getGenericLoc(), Record);
735 Writer.AddSourceLocation(E->getDefaultLoc(), Record);
736 Writer.AddSourceLocation(E->getRParenLoc(), Record);
737 Code = serialization::EXPR_GENERIC_SELECTION;
738}
739
John McCallfe96e0b2011-11-06 09:01:30 +0000740void ASTStmtWriter::VisitPseudoObjectExpr(PseudoObjectExpr *E) {
741 VisitExpr(E);
742 Record.push_back(E->getNumSemanticExprs());
743
744 // Push the result index. Currently, this needs to exactly match
745 // the encoding used internally for ResultIndex.
746 unsigned result = E->getResultExprIndex();
747 result = (result == PseudoObjectExpr::NoResult ? 0 : result + 1);
748 Record.push_back(result);
749
750 Writer.AddStmt(E->getSyntacticForm());
751 for (PseudoObjectExpr::semantics_iterator
752 i = E->semantics_begin(), e = E->semantics_end(); i != e; ++i) {
753 Writer.AddStmt(*i);
John McCallfe96e0b2011-11-06 09:01:30 +0000754 }
Argyrios Kyrtzidis59386502011-11-15 06:20:27 +0000755
756 Code = serialization::EXPR_PSEUDO_OBJECT;
John McCallfe96e0b2011-11-06 09:01:30 +0000757}
758
Eli Friedmandf14b3a2011-10-11 02:20:01 +0000759void ASTStmtWriter::VisitAtomicExpr(AtomicExpr *E) {
760 VisitExpr(E);
761 Record.push_back(E->getOp());
762 Writer.AddStmt(E->getPtr());
763 Writer.AddStmt(E->getOrder());
764 if (E->getOp() != AtomicExpr::Load)
765 Writer.AddStmt(E->getVal1());
766 if (E->isCmpXChg()) {
767 Writer.AddStmt(E->getOrderFail());
768 Writer.AddStmt(E->getVal2());
769 }
770 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
771 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Argyrios Kyrtzidis59386502011-11-15 06:20:27 +0000772
773 Code = serialization::EXPR_ATOMIC;
Eli Friedmandf14b3a2011-10-11 02:20:01 +0000774}
775
Chris Lattner1f551822009-04-27 06:20:01 +0000776//===----------------------------------------------------------------------===//
777// Objective-C Expressions and Statements.
778//===----------------------------------------------------------------------===//
779
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000780void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000781 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000782 Writer.AddStmt(E->getString());
Chris Lattner1f551822009-04-27 06:20:01 +0000783 Writer.AddSourceLocation(E->getAtLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000784 Code = serialization::EXPR_OBJC_STRING_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000785}
786
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000787void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000788 VisitExpr(E);
Douglas Gregorabd9e962010-04-20 15:39:42 +0000789 Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000790 Writer.AddSourceLocation(E->getAtLoc(), Record);
791 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000792 Code = serialization::EXPR_OBJC_ENCODE;
Chris Lattner1f551822009-04-27 06:20:01 +0000793}
794
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000795void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000796 VisitExpr(E);
797 Writer.AddSelectorRef(E->getSelector(), Record);
798 Writer.AddSourceLocation(E->getAtLoc(), Record);
799 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000800 Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
Chris Lattner1f551822009-04-27 06:20:01 +0000801}
802
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000803void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000804 VisitExpr(E);
805 Writer.AddDeclRef(E->getProtocol(), Record);
806 Writer.AddSourceLocation(E->getAtLoc(), Record);
807 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000808 Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
Chris Lattner1f551822009-04-27 06:20:01 +0000809}
810
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000811void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000812 VisitExpr(E);
813 Writer.AddDeclRef(E->getDecl(), Record);
814 Writer.AddSourceLocation(E->getLocation(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000815 Writer.AddStmt(E->getBase());
Chris Lattner1f551822009-04-27 06:20:01 +0000816 Record.push_back(E->isArrow());
817 Record.push_back(E->isFreeIvar());
Sebastian Redl539c5062010-08-18 23:57:32 +0000818 Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
Chris Lattner1f551822009-04-27 06:20:01 +0000819}
820
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000821void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000822 VisitExpr(E);
John McCallb7bd14f2010-12-02 01:19:52 +0000823 Record.push_back(E->isImplicitProperty());
824 if (E->isImplicitProperty()) {
825 Writer.AddDeclRef(E->getImplicitPropertyGetter(), Record);
826 Writer.AddDeclRef(E->getImplicitPropertySetter(), Record);
827 } else {
828 Writer.AddDeclRef(E->getExplicitProperty(), Record);
829 }
Chris Lattner1f551822009-04-27 06:20:01 +0000830 Writer.AddSourceLocation(E->getLocation(), Record);
John McCallb7bd14f2010-12-02 01:19:52 +0000831 Writer.AddSourceLocation(E->getReceiverLocation(), Record);
832 if (E->isObjectReceiver()) {
833 Record.push_back(0);
Fariborz Jahanian681c0752010-10-14 16:04:05 +0000834 Writer.AddStmt(E->getBase());
John McCallb7bd14f2010-12-02 01:19:52 +0000835 } else if (E->isSuperReceiver()) {
836 Record.push_back(1);
837 Writer.AddTypeRef(E->getSuperReceiverType(), Record);
838 } else {
839 Record.push_back(2);
840 Writer.AddDeclRef(E->getClassReceiver(), Record);
841 }
Fariborz Jahanian681c0752010-10-14 16:04:05 +0000842
Sebastian Redl539c5062010-08-18 23:57:32 +0000843 Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
Chris Lattner1f551822009-04-27 06:20:01 +0000844}
845
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000846void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000847 VisitExpr(E);
848 Record.push_back(E->getNumArgs());
Argyrios Kyrtzidisa6011e22011-10-03 06:36:51 +0000849 Record.push_back(E->getNumStoredSelLocs());
850 Record.push_back(E->SelLocsKind);
John McCall31168b02011-06-15 23:02:42 +0000851 Record.push_back(E->isDelegateInitCall());
Argyrios Kyrtzidisa80f1bf2012-01-12 02:34:39 +0000852 Record.push_back(E->IsImplicit);
Douglas Gregor9a129192010-04-21 00:45:42 +0000853 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
854 switch (E->getReceiverKind()) {
855 case ObjCMessageExpr::Instance:
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000856 Writer.AddStmt(E->getInstanceReceiver());
Douglas Gregor9a129192010-04-21 00:45:42 +0000857 break;
858
859 case ObjCMessageExpr::Class:
860 Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
861 break;
862
863 case ObjCMessageExpr::SuperClass:
864 case ObjCMessageExpr::SuperInstance:
865 Writer.AddTypeRef(E->getSuperType(), Record);
866 Writer.AddSourceLocation(E->getSuperLoc(), Record);
867 break;
868 }
869
870 if (E->getMethodDecl()) {
871 Record.push_back(1);
872 Writer.AddDeclRef(E->getMethodDecl(), Record);
873 } else {
874 Record.push_back(0);
875 Writer.AddSelectorRef(E->getSelector(), Record);
876 }
877
Chris Lattner1f551822009-04-27 06:20:01 +0000878 Writer.AddSourceLocation(E->getLeftLoc(), Record);
879 Writer.AddSourceLocation(E->getRightLoc(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000880
881 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
882 Arg != ArgEnd; ++Arg)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000883 Writer.AddStmt(*Arg);
Argyrios Kyrtzidisa6011e22011-10-03 06:36:51 +0000884
885 SourceLocation *Locs = E->getStoredSelLocs();
886 for (unsigned i = 0, e = E->getNumStoredSelLocs(); i != e; ++i)
887 Writer.AddSourceLocation(Locs[i], Record);
888
Sebastian Redl539c5062010-08-18 23:57:32 +0000889 Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
Chris Lattner1f551822009-04-27 06:20:01 +0000890}
891
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000892void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000893 VisitStmt(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000894 Writer.AddStmt(S->getElement());
895 Writer.AddStmt(S->getCollection());
896 Writer.AddStmt(S->getBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000897 Writer.AddSourceLocation(S->getForLoc(), Record);
898 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000899 Code = serialization::STMT_OBJC_FOR_COLLECTION;
Chris Lattner1f551822009-04-27 06:20:01 +0000900}
901
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000902void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000903 Writer.AddStmt(S->getCatchBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000904 Writer.AddDeclRef(S->getCatchParamDecl(), Record);
905 Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
906 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000907 Code = serialization::STMT_OBJC_CATCH;
Chris Lattner1f551822009-04-27 06:20:01 +0000908}
909
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000910void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000911 Writer.AddStmt(S->getFinallyBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000912 Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000913 Code = serialization::STMT_OBJC_FINALLY;
Chris Lattner1f551822009-04-27 06:20:01 +0000914}
915
John McCall31168b02011-06-15 23:02:42 +0000916void ASTStmtWriter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *S) {
917 Writer.AddStmt(S->getSubStmt());
918 Writer.AddSourceLocation(S->getAtLoc(), Record);
919 Code = serialization::STMT_OBJC_AUTORELEASE_POOL;
920}
921
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000922void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
Douglas Gregor96c79492010-04-23 22:50:49 +0000923 Record.push_back(S->getNumCatchStmts());
924 Record.push_back(S->getFinallyStmt() != 0);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000925 Writer.AddStmt(S->getTryBody());
Douglas Gregor96c79492010-04-23 22:50:49 +0000926 for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000927 Writer.AddStmt(S->getCatchStmt(I));
Douglas Gregor96c79492010-04-23 22:50:49 +0000928 if (S->getFinallyStmt())
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000929 Writer.AddStmt(S->getFinallyStmt());
Chris Lattner1f551822009-04-27 06:20:01 +0000930 Writer.AddSourceLocation(S->getAtTryLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000931 Code = serialization::STMT_OBJC_AT_TRY;
Chris Lattner1f551822009-04-27 06:20:01 +0000932}
933
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000934void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000935 Writer.AddStmt(S->getSynchExpr());
936 Writer.AddStmt(S->getSynchBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000937 Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000938 Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
Chris Lattner1f551822009-04-27 06:20:01 +0000939}
940
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000941void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000942 Writer.AddStmt(S->getThrowExpr());
Chris Lattner1f551822009-04-27 06:20:01 +0000943 Writer.AddSourceLocation(S->getThrowLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000944 Code = serialization::STMT_OBJC_AT_THROW;
Chris Lattner1f551822009-04-27 06:20:01 +0000945}
946
947//===----------------------------------------------------------------------===//
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +0000948// C++ Expressions and Statements.
949//===----------------------------------------------------------------------===//
950
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000951void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000952 VisitStmt(S);
953 Writer.AddSourceLocation(S->getCatchLoc(), Record);
954 Writer.AddDeclRef(S->getExceptionDecl(), Record);
955 Writer.AddStmt(S->getHandlerBlock());
Sebastian Redl539c5062010-08-18 23:57:32 +0000956 Code = serialization::STMT_CXX_CATCH;
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000957}
958
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000959void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000960 VisitStmt(S);
961 Record.push_back(S->getNumHandlers());
962 Writer.AddSourceLocation(S->getTryLoc(), Record);
963 Writer.AddStmt(S->getTryBlock());
964 for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
965 Writer.AddStmt(S->getHandler(i));
Sebastian Redl539c5062010-08-18 23:57:32 +0000966 Code = serialization::STMT_CXX_TRY;
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000967}
968
Richard Smith02e85f32011-04-14 22:09:26 +0000969void ASTStmtWriter::VisitCXXForRangeStmt(CXXForRangeStmt *S) {
970 VisitStmt(S);
971 Writer.AddSourceLocation(S->getForLoc(), Record);
972 Writer.AddSourceLocation(S->getColonLoc(), Record);
973 Writer.AddSourceLocation(S->getRParenLoc(), Record);
974 Writer.AddStmt(S->getRangeStmt());
975 Writer.AddStmt(S->getBeginEndStmt());
976 Writer.AddStmt(S->getCond());
977 Writer.AddStmt(S->getInc());
978 Writer.AddStmt(S->getLoopVarStmt());
979 Writer.AddStmt(S->getBody());
980 Code = serialization::STMT_CXX_FOR_RANGE;
981}
982
Douglas Gregordeb4a2be2011-10-25 01:33:02 +0000983void ASTStmtWriter::VisitMSDependentExistsStmt(MSDependentExistsStmt *S) {
984 VisitStmt(S);
985 Writer.AddSourceLocation(S->getKeywordLoc(), Record);
986 Record.push_back(S->isIfExists());
987 Writer.AddNestedNameSpecifierLoc(S->getQualifierLoc(), Record);
988 Writer.AddDeclarationNameInfo(S->getNameInfo(), Record);
989 Writer.AddStmt(S->getSubStmt());
990 Code = serialization::STMT_MS_DEPENDENT_EXISTS;
991}
992
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000993void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +0000994 VisitCallExpr(E);
995 Record.push_back(E->getOperator());
Sebastian Redl539c5062010-08-18 23:57:32 +0000996 Code = serialization::EXPR_CXX_OPERATOR_CALL;
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +0000997}
998
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000999void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
Chris Lattnerb7e7f722010-05-09 05:36:05 +00001000 VisitCallExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +00001001 Code = serialization::EXPR_CXX_MEMBER_CALL;
Chris Lattnerb7e7f722010-05-09 05:36:05 +00001002}
1003
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001004void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
Douglas Gregor5d3507d2009-09-09 23:08:42 +00001005 VisitExpr(E);
Argyrios Kyrtzidis30d98f32010-06-24 08:57:09 +00001006 Record.push_back(E->getNumArgs());
1007 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001008 Writer.AddStmt(E->getArg(I));
Douglas Gregor5d3507d2009-09-09 23:08:42 +00001009 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor85dabae2009-12-16 01:38:02 +00001010 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregor5d3507d2009-09-09 23:08:42 +00001011 Record.push_back(E->isElidable());
Abramo Bagnara635ed24e2011-10-05 07:56:41 +00001012 Record.push_back(E->hadMultipleCandidates());
Douglas Gregor4f4b1862009-12-16 18:50:27 +00001013 Record.push_back(E->requiresZeroInitialization());
Douglas Gregor222cf0e2010-05-15 00:13:29 +00001014 Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
Chandler Carruth01718152010-10-25 08:47:36 +00001015 Writer.AddSourceRange(E->getParenRange(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001016 Code = serialization::EXPR_CXX_CONSTRUCT;
Douglas Gregor5d3507d2009-09-09 23:08:42 +00001017}
1018
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001019void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +00001020 VisitCXXConstructExpr(E);
Douglas Gregor2b88c112010-09-08 00:15:04 +00001021 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001022 Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +00001023}
1024
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001025void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001026 VisitExplicitCastExpr(E);
Douglas Gregor4478f852011-01-12 22:41:29 +00001027 Writer.AddSourceRange(SourceRange(E->getOperatorLoc(), E->getRParenLoc()),
1028 Record);
Sam Weinigd01101e2010-01-16 21:21:01 +00001029}
1030
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001031void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001032 VisitCXXNamedCastExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +00001033 Code = serialization::EXPR_CXX_STATIC_CAST;
Sam Weinigd01101e2010-01-16 21:21:01 +00001034}
1035
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001036void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001037 VisitCXXNamedCastExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +00001038 Code = serialization::EXPR_CXX_DYNAMIC_CAST;
Sam Weinigd01101e2010-01-16 21:21:01 +00001039}
1040
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001041void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001042 VisitCXXNamedCastExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +00001043 Code = serialization::EXPR_CXX_REINTERPRET_CAST;
Sam Weinigd01101e2010-01-16 21:21:01 +00001044}
1045
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001046void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001047 VisitCXXNamedCastExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +00001048 Code = serialization::EXPR_CXX_CONST_CAST;
Sam Weinigd01101e2010-01-16 21:21:01 +00001049}
1050
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001051void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001052 VisitExplicitCastExpr(E);
1053 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1054 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001055 Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
Sam Weinigd01101e2010-01-16 21:21:01 +00001056}
1057
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001058void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
Sam Weinige83b3ac2010-02-07 06:32:43 +00001059 VisitExpr(E);
1060 Record.push_back(E->getValue());
1061 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001062 Code = serialization::EXPR_CXX_BOOL_LITERAL;
Sam Weinige83b3ac2010-02-07 06:32:43 +00001063}
1064
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001065void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
Sam Weinige83b3ac2010-02-07 06:32:43 +00001066 VisitExpr(E);
1067 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001068 Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
Sam Weinige83b3ac2010-02-07 06:32:43 +00001069}
1070
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001071void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
Chris Lattner13a5ecc2010-05-09 06:03:39 +00001072 VisitExpr(E);
1073 Writer.AddSourceRange(E->getSourceRange(), Record);
1074 if (E->isTypeOperand()) {
1075 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001076 Code = serialization::EXPR_CXX_TYPEID_TYPE;
Chris Lattner13a5ecc2010-05-09 06:03:39 +00001077 } else {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001078 Writer.AddStmt(E->getExprOperand());
Sebastian Redl539c5062010-08-18 23:57:32 +00001079 Code = serialization::EXPR_CXX_TYPEID_EXPR;
Chris Lattner13a5ecc2010-05-09 06:03:39 +00001080 }
1081}
1082
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001083void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
Chris Lattner98267332010-05-09 06:15:05 +00001084 VisitExpr(E);
1085 Writer.AddSourceLocation(E->getLocation(), Record);
1086 Record.push_back(E->isImplicit());
Sebastian Redl539c5062010-08-18 23:57:32 +00001087 Code = serialization::EXPR_CXX_THIS;
Chris Lattner98267332010-05-09 06:15:05 +00001088}
1089
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001090void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
Chris Lattnere2437f42010-05-09 06:40:08 +00001091 VisitExpr(E);
Chris Lattner98267332010-05-09 06:15:05 +00001092 Writer.AddSourceLocation(E->getThrowLoc(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001093 Writer.AddStmt(E->getSubExpr());
Douglas Gregor53e191ed2011-07-06 22:04:06 +00001094 Record.push_back(E->isThrownVariableInScope());
Sebastian Redl539c5062010-08-18 23:57:32 +00001095 Code = serialization::EXPR_CXX_THROW;
Chris Lattner98267332010-05-09 06:15:05 +00001096}
1097
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001098void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
Chris Lattnere2437f42010-05-09 06:40:08 +00001099 VisitExpr(E);
Argyrios Kyrtzidis4259ebc2010-07-02 23:30:15 +00001100
1101 bool HasOtherExprStored = E->Param.getInt();
1102 // Store these first, the reader reads them before creation.
1103 Record.push_back(HasOtherExprStored);
1104 if (HasOtherExprStored)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001105 Writer.AddStmt(E->getExpr());
Argyrios Kyrtzidis4259ebc2010-07-02 23:30:15 +00001106 Writer.AddDeclRef(E->getParam(), Record);
1107 Writer.AddSourceLocation(E->getUsedLocation(), Record);
Chris Lattnere2437f42010-05-09 06:40:08 +00001108
Sebastian Redl539c5062010-08-18 23:57:32 +00001109 Code = serialization::EXPR_CXX_DEFAULT_ARG;
Chris Lattnere2437f42010-05-09 06:40:08 +00001110}
1111
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001112void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
Chris Lattnercba86142010-05-10 00:25:06 +00001113 VisitExpr(E);
1114 Writer.AddCXXTemporary(E->getTemporary(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001115 Writer.AddStmt(E->getSubExpr());
Sebastian Redl539c5062010-08-18 23:57:32 +00001116 Code = serialization::EXPR_CXX_BIND_TEMPORARY;
Chris Lattnercba86142010-05-10 00:25:06 +00001117}
1118
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001119void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001120 VisitExpr(E);
Douglas Gregor2b88c112010-09-08 00:15:04 +00001121 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001122 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001123 Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001124}
1125
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001126void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001127 VisitExpr(E);
1128 Record.push_back(E->isGlobalNew());
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001129 Record.push_back(E->hasInitializer());
John McCall284c48f2011-01-27 09:37:56 +00001130 Record.push_back(E->doesUsualArrayDeleteWantSize());
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001131 Record.push_back(E->isArray());
Abramo Bagnara635ed24e2011-10-05 07:56:41 +00001132 Record.push_back(E->hadMultipleCandidates());
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001133 Record.push_back(E->getNumPlacementArgs());
1134 Record.push_back(E->getNumConstructorArgs());
1135 Writer.AddDeclRef(E->getOperatorNew(), Record);
1136 Writer.AddDeclRef(E->getOperatorDelete(), Record);
1137 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor0744ef62010-09-07 21:49:58 +00001138 Writer.AddTypeSourceInfo(E->getAllocatedTypeSourceInfo(), Record);
Douglas Gregorf2753b32010-07-13 15:54:32 +00001139 Writer.AddSourceRange(E->getTypeIdParens(), Record);
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001140 Writer.AddSourceLocation(E->getStartLoc(), Record);
1141 Writer.AddSourceLocation(E->getEndLoc(), Record);
Chandler Carruth01718152010-10-25 08:47:36 +00001142 Writer.AddSourceLocation(E->getConstructorLParen(), Record);
1143 Writer.AddSourceLocation(E->getConstructorRParen(), Record);
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001144 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
1145 I != e; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001146 Writer.AddStmt(*I);
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001147
Sebastian Redl539c5062010-08-18 23:57:32 +00001148 Code = serialization::EXPR_CXX_NEW;
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001149}
1150
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001151void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
Argyrios Kyrtzidis6e57c352010-06-22 17:07:59 +00001152 VisitExpr(E);
1153 Record.push_back(E->isGlobalDelete());
1154 Record.push_back(E->isArrayForm());
Argyrios Kyrtzidis14ec9f62010-09-13 20:15:54 +00001155 Record.push_back(E->isArrayFormAsWritten());
John McCall284c48f2011-01-27 09:37:56 +00001156 Record.push_back(E->doesUsualArrayDeleteWantSize());
Argyrios Kyrtzidis6e57c352010-06-22 17:07:59 +00001157 Writer.AddDeclRef(E->getOperatorDelete(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001158 Writer.AddStmt(E->getArgument());
Argyrios Kyrtzidis6e57c352010-06-22 17:07:59 +00001159 Writer.AddSourceLocation(E->getSourceRange().getBegin(), Record);
1160
Sebastian Redl539c5062010-08-18 23:57:32 +00001161 Code = serialization::EXPR_CXX_DELETE;
Argyrios Kyrtzidis6e57c352010-06-22 17:07:59 +00001162}
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001163
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001164void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
Argyrios Kyrtzidis99a226d2010-06-28 09:32:03 +00001165 VisitExpr(E);
1166
1167 Writer.AddStmt(E->getBase());
1168 Record.push_back(E->isArrow());
1169 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Douglas Gregora6ce6082011-02-25 18:19:59 +00001170 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
Argyrios Kyrtzidis99a226d2010-06-28 09:32:03 +00001171 Writer.AddTypeSourceInfo(E->getScopeTypeInfo(), Record);
1172 Writer.AddSourceLocation(E->getColonColonLoc(), Record);
1173 Writer.AddSourceLocation(E->getTildeLoc(), Record);
1174
1175 // PseudoDestructorTypeStorage.
1176 Writer.AddIdentifierRef(E->getDestroyedTypeIdentifier(), Record);
1177 if (E->getDestroyedTypeIdentifier())
1178 Writer.AddSourceLocation(E->getDestroyedTypeLoc(), Record);
1179 else
1180 Writer.AddTypeSourceInfo(E->getDestroyedTypeInfo(), Record);
1181
Sebastian Redl539c5062010-08-18 23:57:32 +00001182 Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
Argyrios Kyrtzidis99a226d2010-06-28 09:32:03 +00001183}
1184
John McCall5d413782010-12-06 08:20:24 +00001185void ASTStmtWriter::VisitExprWithCleanups(ExprWithCleanups *E) {
Chris Lattnercba86142010-05-10 00:25:06 +00001186 VisitExpr(E);
John McCall28fc7092011-11-10 05:35:25 +00001187 Record.push_back(E->getNumObjects());
1188 for (unsigned i = 0, e = E->getNumObjects(); i != e; ++i)
1189 Writer.AddDeclRef(E->getObject(i), Record);
Chris Lattnercba86142010-05-10 00:25:06 +00001190
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001191 Writer.AddStmt(E->getSubExpr());
John McCall5d413782010-12-06 08:20:24 +00001192 Code = serialization::EXPR_EXPR_WITH_CLEANUPS;
Chris Lattnercba86142010-05-10 00:25:06 +00001193}
1194
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001195void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001196ASTStmtWriter::VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E){
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001197 VisitExpr(E);
Abramo Bagnara7945c982012-01-27 09:46:47 +00001198
1199 // Don't emit anything here, HasTemplateKWAndArgsInfo must be
Douglas Gregor87866ce2011-02-04 12:01:24 +00001200 // emitted first.
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001201
Abramo Bagnara7945c982012-01-27 09:46:47 +00001202 Record.push_back(E->HasTemplateKWAndArgsInfo);
1203 if (E->HasTemplateKWAndArgsInfo) {
1204 const ASTTemplateKWAndArgsInfo &Args = *E->getTemplateKWAndArgsInfo();
Argyrios Kyrtzidisb5288de2010-06-28 09:31:48 +00001205 Record.push_back(Args.NumTemplateArgs);
Abramo Bagnara7945c982012-01-27 09:46:47 +00001206 AddTemplateKWAndArgsInfo(Args);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001207 }
Abramo Bagnara7945c982012-01-27 09:46:47 +00001208
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001209 if (!E->isImplicitAccess())
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001210 Writer.AddStmt(E->getBase());
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001211 else
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001212 Writer.AddStmt(0);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001213 Writer.AddTypeRef(E->getBaseType(), Record);
1214 Record.push_back(E->isArrow());
1215 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Douglas Gregore16af532011-02-28 18:50:33 +00001216 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001217 Writer.AddDeclRef(E->getFirstQualifierFoundInScope(), Record);
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +00001218 Writer.AddDeclarationNameInfo(E->MemberNameInfo, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001219 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001220}
1221
1222void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001223ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
Argyrios Kyrtzidiscd444d1a2010-06-28 09:31:56 +00001224 VisitExpr(E);
Abramo Bagnara7945c982012-01-27 09:46:47 +00001225
1226 // Don't emit anything here, HasTemplateKWAndArgsInfo must be
Douglas Gregor87866ce2011-02-04 12:01:24 +00001227 // emitted first.
Abramo Bagnara7945c982012-01-27 09:46:47 +00001228
1229 Record.push_back(E->HasTemplateKWAndArgsInfo);
1230 if (E->HasTemplateKWAndArgsInfo) {
1231 const ASTTemplateKWAndArgsInfo &Args = *E->getTemplateKWAndArgsInfo();
Argyrios Kyrtzidiscd444d1a2010-06-28 09:31:56 +00001232 Record.push_back(Args.NumTemplateArgs);
Abramo Bagnara7945c982012-01-27 09:46:47 +00001233 AddTemplateKWAndArgsInfo(Args);
Argyrios Kyrtzidiscd444d1a2010-06-28 09:31:56 +00001234 }
1235
Douglas Gregor3a43fd62011-02-25 20:49:16 +00001236 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +00001237 Writer.AddDeclarationNameInfo(E->NameInfo, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001238 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
Argyrios Kyrtzidiscd444d1a2010-06-28 09:31:56 +00001239}
1240
1241void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001242ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001243 VisitExpr(E);
1244 Record.push_back(E->arg_size());
1245 for (CXXUnresolvedConstructExpr::arg_iterator
1246 ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001247 Writer.AddStmt(*ArgI);
Douglas Gregor2b88c112010-09-08 00:15:04 +00001248 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001249 Writer.AddSourceLocation(E->getLParenLoc(), Record);
1250 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001251 Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001252}
Chris Lattner98267332010-05-09 06:15:05 +00001253
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001254void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001255 VisitExpr(E);
Abramo Bagnara7945c982012-01-27 09:46:47 +00001256
1257 // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1258 // emitted first.
1259
1260 Record.push_back(E->HasTemplateKWAndArgsInfo);
1261 if (E->HasTemplateKWAndArgsInfo) {
1262 const ASTTemplateKWAndArgsInfo &Args = *E->getTemplateKWAndArgsInfo();
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001263 Record.push_back(Args.NumTemplateArgs);
Abramo Bagnara7945c982012-01-27 09:46:47 +00001264 AddTemplateKWAndArgsInfo(Args);
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001265 }
1266
1267 Record.push_back(E->getNumDecls());
1268 for (OverloadExpr::decls_iterator
1269 OvI = E->decls_begin(), OvE = E->decls_end(); OvI != OvE; ++OvI) {
1270 Writer.AddDeclRef(OvI.getDecl(), Record);
1271 Record.push_back(OvI.getAccess());
1272 }
1273
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +00001274 Writer.AddDeclarationNameInfo(E->NameInfo, Record);
Douglas Gregor0da1d432011-02-28 20:01:57 +00001275 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001276}
1277
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001278void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001279 VisitOverloadExpr(E);
1280 Record.push_back(E->isArrow());
1281 Record.push_back(E->hasUnresolvedUsing());
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001282 Writer.AddStmt(!E->isImplicitAccess() ? E->getBase() : 0);
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001283 Writer.AddTypeRef(E->getBaseType(), Record);
1284 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001285 Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001286}
1287
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001288void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
Argyrios Kyrtzidis58e01ad2010-06-25 09:03:34 +00001289 VisitOverloadExpr(E);
1290 Record.push_back(E->requiresADL());
Richard Smith02e85f32011-04-14 22:09:26 +00001291 if (E->requiresADL())
1292 Record.push_back(E->isStdAssociatedNamespace());
Argyrios Kyrtzidis58e01ad2010-06-25 09:03:34 +00001293 Record.push_back(E->isOverloaded());
1294 Writer.AddDeclRef(E->getNamingClass(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001295 Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
Argyrios Kyrtzidis58e01ad2010-06-25 09:03:34 +00001296}
1297
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001298void ASTStmtWriter::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +00001299 VisitExpr(E);
1300 Record.push_back(E->getTrait());
Sebastian Redl8eb06f12010-09-13 20:56:31 +00001301 Record.push_back(E->getValue());
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +00001302 Writer.AddSourceRange(E->getSourceRange(), Record);
Douglas Gregor54e5b132010-09-09 16:14:44 +00001303 Writer.AddTypeSourceInfo(E->getQueriedTypeSourceInfo(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001304 Code = serialization::EXPR_CXX_UNARY_TYPE_TRAIT;
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +00001305}
1306
Francois Pichet9dfa3ce2010-12-07 00:08:36 +00001307void ASTStmtWriter::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
1308 VisitExpr(E);
1309 Record.push_back(E->getTrait());
1310 Record.push_back(E->getValue());
1311 Writer.AddSourceRange(E->getSourceRange(), Record);
1312 Writer.AddTypeSourceInfo(E->getLhsTypeSourceInfo(), Record);
1313 Writer.AddTypeSourceInfo(E->getRhsTypeSourceInfo(), Record);
Francois Pichet34b21132010-12-08 22:35:30 +00001314 Code = serialization::EXPR_BINARY_TYPE_TRAIT;
Francois Pichet9dfa3ce2010-12-07 00:08:36 +00001315}
1316
John Wiegley6242b6a2011-04-28 00:16:57 +00001317void ASTStmtWriter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
1318 VisitExpr(E);
1319 Record.push_back(E->getTrait());
1320 Record.push_back(E->getValue());
1321 Writer.AddSourceRange(E->getSourceRange(), Record);
1322 Writer.AddTypeSourceInfo(E->getQueriedTypeSourceInfo(), Record);
1323 Code = serialization::EXPR_ARRAY_TYPE_TRAIT;
1324}
1325
John Wiegleyf9f65842011-04-25 06:54:41 +00001326void ASTStmtWriter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
1327 VisitExpr(E);
1328 Record.push_back(E->getTrait());
1329 Record.push_back(E->getValue());
1330 Writer.AddSourceRange(E->getSourceRange(), Record);
1331 Writer.AddStmt(E->getQueriedExpression());
1332 Code = serialization::EXPR_CXX_EXPRESSION_TRAIT;
1333}
1334
Sebastian Redl9ac55dd2010-09-10 20:55:54 +00001335void ASTStmtWriter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
1336 VisitExpr(E);
1337 Record.push_back(E->getValue());
1338 Writer.AddSourceRange(E->getSourceRange(), Record);
1339 Writer.AddStmt(E->getOperand());
1340 Code = serialization::EXPR_CXX_NOEXCEPT;
1341}
1342
Douglas Gregore8e9dd62011-01-03 17:17:50 +00001343void ASTStmtWriter::VisitPackExpansionExpr(PackExpansionExpr *E) {
1344 VisitExpr(E);
1345 Writer.AddSourceLocation(E->getEllipsisLoc(), Record);
Douglas Gregorb8840002011-01-14 21:20:45 +00001346 Record.push_back(E->NumExpansions);
Douglas Gregore8e9dd62011-01-03 17:17:50 +00001347 Writer.AddStmt(E->getPattern());
1348 Code = serialization::EXPR_PACK_EXPANSION;
1349}
1350
Douglas Gregor820ba7b2011-01-04 17:33:58 +00001351void ASTStmtWriter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
1352 VisitExpr(E);
1353 Writer.AddSourceLocation(E->OperatorLoc, Record);
1354 Writer.AddSourceLocation(E->PackLoc, Record);
1355 Writer.AddSourceLocation(E->RParenLoc, Record);
1356 Record.push_back(E->Length);
1357 Writer.AddDeclRef(E->Pack, Record);
1358 Code = serialization::EXPR_SIZEOF_PACK;
1359}
1360
John McCallfa194042011-07-15 07:00:14 +00001361void ASTStmtWriter::VisitSubstNonTypeTemplateParmExpr(
1362 SubstNonTypeTemplateParmExpr *E) {
1363 VisitExpr(E);
1364 Writer.AddDeclRef(E->getParameter(), Record);
1365 Writer.AddSourceLocation(E->getNameLoc(), Record);
1366 Writer.AddStmt(E->getReplacement());
1367 Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM;
1368}
1369
Douglas Gregorcdbc5392011-01-15 01:15:58 +00001370void ASTStmtWriter::VisitSubstNonTypeTemplateParmPackExpr(
1371 SubstNonTypeTemplateParmPackExpr *E) {
1372 VisitExpr(E);
John McCallfa194042011-07-15 07:00:14 +00001373 Writer.AddDeclRef(E->getParameterPack(), Record);
Douglas Gregorcdbc5392011-01-15 01:15:58 +00001374 Writer.AddTemplateArgument(E->getArgumentPack(), Record);
John McCallfa194042011-07-15 07:00:14 +00001375 Writer.AddSourceLocation(E->getParameterPackLocation(), Record);
Douglas Gregorcdbc5392011-01-15 01:15:58 +00001376 Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM_PACK;
1377}
1378
Douglas Gregorfe314812011-06-21 17:03:29 +00001379void ASTStmtWriter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E) {
1380 VisitExpr(E);
1381 Writer.AddStmt(E->Temporary);
1382 Code = serialization::EXPR_MATERIALIZE_TEMPORARY;
1383}
1384
John McCall8d69a212010-11-15 23:31:06 +00001385void ASTStmtWriter::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
1386 VisitExpr(E);
Argyrios Kyrtzidisb2b07952011-12-03 03:49:52 +00001387 Writer.AddStmt(E->getSourceExpr());
Douglas Gregorc03a1082011-01-28 02:26:04 +00001388 Writer.AddSourceLocation(E->getLocation(), Record);
John McCall8d69a212010-11-15 23:31:06 +00001389 Code = serialization::EXPR_OPAQUE_VALUE;
1390}
1391
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +00001392//===----------------------------------------------------------------------===//
Peter Collingbourne41f85462011-02-09 21:07:24 +00001393// CUDA Expressions and Statements.
1394//===----------------------------------------------------------------------===//
1395
1396void ASTStmtWriter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *E) {
1397 VisitCallExpr(E);
1398 Writer.AddStmt(E->getConfig());
1399 Code = serialization::EXPR_CUDA_KERNEL_CALL;
1400}
1401
1402//===----------------------------------------------------------------------===//
Tanya Lattner55808c12011-06-04 00:47:47 +00001403// OpenCL Expressions and Statements.
1404//===----------------------------------------------------------------------===//
1405void ASTStmtWriter::VisitAsTypeExpr(AsTypeExpr *E) {
1406 VisitExpr(E);
John McCallfa194042011-07-15 07:00:14 +00001407 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
1408 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Tanya Lattner55808c12011-06-04 00:47:47 +00001409 Writer.AddStmt(E->getSrcExpr());
1410 Code = serialization::EXPR_ASTYPE;
1411}
1412
1413//===----------------------------------------------------------------------===//
John McCallfa194042011-07-15 07:00:14 +00001414// Microsoft Expressions and Statements.
1415//===----------------------------------------------------------------------===//
1416void ASTStmtWriter::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
1417 VisitExpr(E);
1418 Writer.AddSourceRange(E->getSourceRange(), Record);
1419 if (E->isTypeOperand()) {
1420 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1421 Code = serialization::EXPR_CXX_UUIDOF_TYPE;
1422 } else {
1423 Writer.AddStmt(E->getExprOperand());
1424 Code = serialization::EXPR_CXX_UUIDOF_EXPR;
1425 }
1426}
1427
1428void ASTStmtWriter::VisitSEHExceptStmt(SEHExceptStmt *S) {
1429 VisitStmt(S);
1430 Writer.AddSourceLocation(S->getExceptLoc(), Record);
1431 Writer.AddStmt(S->getFilterExpr());
1432 Writer.AddStmt(S->getBlock());
1433 Code = serialization::STMT_SEH_EXCEPT;
1434}
1435
1436void ASTStmtWriter::VisitSEHFinallyStmt(SEHFinallyStmt *S) {
1437 VisitStmt(S);
1438 Writer.AddSourceLocation(S->getFinallyLoc(), Record);
1439 Writer.AddStmt(S->getBlock());
1440 Code = serialization::STMT_SEH_FINALLY;
1441}
1442
1443void ASTStmtWriter::VisitSEHTryStmt(SEHTryStmt *S) {
1444 VisitStmt(S);
1445 Record.push_back(S->getIsCXXTry());
1446 Writer.AddSourceLocation(S->getTryLoc(), Record);
1447 Writer.AddStmt(S->getTryBlock());
1448 Writer.AddStmt(S->getHandler());
1449 Code = serialization::STMT_SEH_TRY;
1450}
1451
1452//===----------------------------------------------------------------------===//
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001453// ASTWriter Implementation
Chris Lattner1f551822009-04-27 06:20:01 +00001454//===----------------------------------------------------------------------===//
1455
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001456unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
Mike Stump11289f42009-09-09 15:08:12 +00001457 assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
Chris Lattner1f551822009-04-27 06:20:01 +00001458 "SwitchCase recorded twice");
1459 unsigned NextID = SwitchCaseIDs.size();
1460 SwitchCaseIDs[S] = NextID;
1461 return NextID;
1462}
1463
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001464unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
Mike Stump11289f42009-09-09 15:08:12 +00001465 assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
Chris Lattner1f551822009-04-27 06:20:01 +00001466 "SwitchCase hasn't been seen yet");
1467 return SwitchCaseIDs[S];
1468}
1469
Argyrios Kyrtzidisd9f526f2010-10-28 09:29:32 +00001470void ASTWriter::ClearSwitchCaseIDs() {
1471 SwitchCaseIDs.clear();
1472}
1473
Chris Lattner1f551822009-04-27 06:20:01 +00001474/// \brief Write the given substatement or subexpression to the
1475/// bitstream.
Argyrios Kyrtzidis6a598972011-10-21 23:02:28 +00001476void ASTWriter::WriteSubStmt(Stmt *S,
1477 llvm::DenseMap<Stmt *, uint64_t> &SubStmtEntries,
1478 llvm::DenseSet<Stmt *> &ParentStmts) {
Chris Lattner1f551822009-04-27 06:20:01 +00001479 RecordData Record;
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001480 ASTStmtWriter Writer(*this, Record);
Chris Lattner1f551822009-04-27 06:20:01 +00001481 ++NumStatements;
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001482
Chris Lattner1f551822009-04-27 06:20:01 +00001483 if (!S) {
Sebastian Redl539c5062010-08-18 23:57:32 +00001484 Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
Chris Lattner1f551822009-04-27 06:20:01 +00001485 return;
1486 }
Mike Stump11289f42009-09-09 15:08:12 +00001487
Argyrios Kyrtzidis6a598972011-10-21 23:02:28 +00001488 llvm::DenseMap<Stmt *, uint64_t>::iterator I = SubStmtEntries.find(S);
1489 if (I != SubStmtEntries.end()) {
1490 Record.push_back(I->second);
1491 Stream.EmitRecord(serialization::STMT_REF_PTR, Record);
1492 return;
1493 }
1494
1495#ifndef NDEBUG
1496 assert(!ParentStmts.count(S) && "There is a Stmt cycle!");
1497
1498 struct ParentStmtInserterRAII {
1499 Stmt *S;
1500 llvm::DenseSet<Stmt *> &ParentStmts;
1501
1502 ParentStmtInserterRAII(Stmt *S, llvm::DenseSet<Stmt *> &ParentStmts)
1503 : S(S), ParentStmts(ParentStmts) {
1504 ParentStmts.insert(S);
1505 }
1506 ~ParentStmtInserterRAII() {
1507 ParentStmts.erase(S);
1508 }
1509 };
1510
1511 ParentStmtInserterRAII ParentStmtInserter(S, ParentStmts);
1512#endif
1513
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001514 // Redirect ASTWriter::AddStmt to collect sub stmts.
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001515 SmallVector<Stmt *, 16> SubStmts;
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001516 CollectedStmts = &SubStmts;
1517
Sebastian Redl539c5062010-08-18 23:57:32 +00001518 Writer.Code = serialization::STMT_NULL_PTR;
Douglas Gregor03412ba2011-06-03 02:27:19 +00001519 Writer.AbbrevToUse = 0;
Chris Lattner1f551822009-04-27 06:20:01 +00001520 Writer.Visit(S);
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001521
1522#ifndef NDEBUG
Sebastian Redl539c5062010-08-18 23:57:32 +00001523 if (Writer.Code == serialization::STMT_NULL_PTR) {
Argyrios Kyrtzidis6ba1bb52010-06-25 09:03:12 +00001524 SourceManager &SrcMgr
1525 = DeclIDs.begin()->first->getASTContext().getSourceManager();
1526 S->dump(SrcMgr);
David Blaikie83d382b2011-09-23 05:06:16 +00001527 llvm_unreachable("Unhandled sub statement writing AST file");
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001528 }
1529#endif
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001530
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001531 // Revert ASTWriter::AddStmt.
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001532 CollectedStmts = &StmtsToEmit;
1533
1534 // Write the sub stmts in reverse order, last to first. When reading them back
1535 // we will read them in correct order by "pop"ing them from the Stmts stack.
1536 // This simplifies reading and allows to store a variable number of sub stmts
1537 // without knowing it in advance.
1538 while (!SubStmts.empty())
Argyrios Kyrtzidis6a598972011-10-21 23:02:28 +00001539 WriteSubStmt(SubStmts.pop_back_val(), SubStmtEntries, ParentStmts);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001540
Douglas Gregor03412ba2011-06-03 02:27:19 +00001541 Stream.EmitRecord(Writer.Code, Record, Writer.AbbrevToUse);
Argyrios Kyrtzidis6a598972011-10-21 23:02:28 +00001542
1543 SubStmtEntries[S] = Stream.GetCurrentBitNo();
Chris Lattner1f551822009-04-27 06:20:01 +00001544}
1545
1546/// \brief Flush all of the statements that have been added to the
1547/// queue via AddStmt().
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001548void ASTWriter::FlushStmts() {
Chris Lattner1f551822009-04-27 06:20:01 +00001549 RecordData Record;
Mike Stump11289f42009-09-09 15:08:12 +00001550
Argyrios Kyrtzidis6a598972011-10-21 23:02:28 +00001551 /// \brief Set of parent Stmts for the currently serializing sub stmt.
1552 llvm::DenseSet<Stmt *> ParentStmts;
1553 /// \brief Offsets of sub stmts already serialized. The offset points
1554 /// just after the stmt record.
1555 llvm::DenseMap<Stmt *, uint64_t> SubStmtEntries;
1556
Chris Lattner1f551822009-04-27 06:20:01 +00001557 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
Argyrios Kyrtzidis6a598972011-10-21 23:02:28 +00001558 WriteSubStmt(StmtsToEmit[I], SubStmtEntries, ParentStmts);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001559
Mike Stump11289f42009-09-09 15:08:12 +00001560 assert(N == StmtsToEmit.size() &&
Chris Lattner57540c52011-04-15 05:22:18 +00001561 "Substatement written via AddStmt rather than WriteSubStmt!");
Mike Stump11289f42009-09-09 15:08:12 +00001562
Chris Lattner1f551822009-04-27 06:20:01 +00001563 // Note that we are at the end of a full expression. Any
1564 // expression records that follow this one are part of a different
1565 // expression.
Sebastian Redl539c5062010-08-18 23:57:32 +00001566 Stream.EmitRecord(serialization::STMT_STOP, Record);
Argyrios Kyrtzidis6a598972011-10-21 23:02:28 +00001567
1568 SubStmtEntries.clear();
1569 ParentStmts.clear();
Chris Lattner1f551822009-04-27 06:20:01 +00001570 }
Mike Stump11289f42009-09-09 15:08:12 +00001571
Chris Lattner1f551822009-04-27 06:20:01 +00001572 StmtsToEmit.clear();
Chris Lattner1f551822009-04-27 06:20:01 +00001573}