blob: 4fe3bae91a27aae3d01f5eb6ca1348ac0ca695a7 [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;
Chris Lattner1f551822009-04-27 06:20:01 +000033
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000034 ASTStmtWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
Chris Lattner1f551822009-04-27 06:20:01 +000035 : Writer(Writer), Record(Record) { }
Argyrios Kyrtzidisb5288de2010-06-28 09:31:48 +000036
37 void
38 AddExplicitTemplateArgumentList(const ExplicitTemplateArgumentList &Args);
Chris Lattner1f551822009-04-27 06:20:01 +000039
40 void VisitStmt(Stmt *S);
41 void VisitNullStmt(NullStmt *S);
42 void VisitCompoundStmt(CompoundStmt *S);
43 void VisitSwitchCase(SwitchCase *S);
44 void VisitCaseStmt(CaseStmt *S);
45 void VisitDefaultStmt(DefaultStmt *S);
46 void VisitLabelStmt(LabelStmt *S);
47 void VisitIfStmt(IfStmt *S);
48 void VisitSwitchStmt(SwitchStmt *S);
49 void VisitWhileStmt(WhileStmt *S);
50 void VisitDoStmt(DoStmt *S);
51 void VisitForStmt(ForStmt *S);
52 void VisitGotoStmt(GotoStmt *S);
53 void VisitIndirectGotoStmt(IndirectGotoStmt *S);
54 void VisitContinueStmt(ContinueStmt *S);
55 void VisitBreakStmt(BreakStmt *S);
56 void VisitReturnStmt(ReturnStmt *S);
57 void VisitDeclStmt(DeclStmt *S);
58 void VisitAsmStmt(AsmStmt *S);
59 void VisitExpr(Expr *E);
60 void VisitPredefinedExpr(PredefinedExpr *E);
61 void VisitDeclRefExpr(DeclRefExpr *E);
62 void VisitIntegerLiteral(IntegerLiteral *E);
63 void VisitFloatingLiteral(FloatingLiteral *E);
64 void VisitImaginaryLiteral(ImaginaryLiteral *E);
65 void VisitStringLiteral(StringLiteral *E);
66 void VisitCharacterLiteral(CharacterLiteral *E);
67 void VisitParenExpr(ParenExpr *E);
Argyrios Kyrtzidisf9f47c82010-06-30 08:49:18 +000068 void VisitParenListExpr(ParenListExpr *E);
Chris Lattner1f551822009-04-27 06:20:01 +000069 void VisitUnaryOperator(UnaryOperator *E);
Douglas Gregor882211c2010-04-28 22:16:22 +000070 void VisitOffsetOfExpr(OffsetOfExpr *E);
Peter Collingbournee190dee2011-03-11 19:24:49 +000071 void VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E);
Chris Lattner1f551822009-04-27 06:20:01 +000072 void VisitArraySubscriptExpr(ArraySubscriptExpr *E);
73 void VisitCallExpr(CallExpr *E);
74 void VisitMemberExpr(MemberExpr *E);
75 void VisitCastExpr(CastExpr *E);
76 void VisitBinaryOperator(BinaryOperator *E);
77 void VisitCompoundAssignOperator(CompoundAssignOperator *E);
78 void VisitConditionalOperator(ConditionalOperator *E);
John McCallc07a0c72011-02-17 10:25:35 +000079 void VisitBinaryConditionalOperator(BinaryConditionalOperator *E);
Chris Lattner1f551822009-04-27 06:20:01 +000080 void VisitImplicitCastExpr(ImplicitCastExpr *E);
81 void VisitExplicitCastExpr(ExplicitCastExpr *E);
82 void VisitCStyleCastExpr(CStyleCastExpr *E);
83 void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
84 void VisitExtVectorElementExpr(ExtVectorElementExpr *E);
85 void VisitInitListExpr(InitListExpr *E);
86 void VisitDesignatedInitExpr(DesignatedInitExpr *E);
87 void VisitImplicitValueInitExpr(ImplicitValueInitExpr *E);
88 void VisitVAArgExpr(VAArgExpr *E);
89 void VisitAddrLabelExpr(AddrLabelExpr *E);
90 void VisitStmtExpr(StmtExpr *E);
Chris Lattner1f551822009-04-27 06:20:01 +000091 void VisitChooseExpr(ChooseExpr *E);
92 void VisitGNUNullExpr(GNUNullExpr *E);
93 void VisitShuffleVectorExpr(ShuffleVectorExpr *E);
94 void VisitBlockExpr(BlockExpr *E);
95 void VisitBlockDeclRefExpr(BlockDeclRefExpr *E);
Peter Collingbourne91147592011-04-15 00:35:48 +000096 void VisitGenericSelectionExpr(GenericSelectionExpr *E);
Mike Stump11289f42009-09-09 15:08:12 +000097
Chris Lattner1f551822009-04-27 06:20:01 +000098 // Objective-C Expressions
99 void VisitObjCStringLiteral(ObjCStringLiteral *E);
100 void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
101 void VisitObjCSelectorExpr(ObjCSelectorExpr *E);
102 void VisitObjCProtocolExpr(ObjCProtocolExpr *E);
103 void VisitObjCIvarRefExpr(ObjCIvarRefExpr *E);
104 void VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E);
Chris Lattner1f551822009-04-27 06:20:01 +0000105 void VisitObjCMessageExpr(ObjCMessageExpr *E);
Steve Naroffe87026a2009-07-24 17:54:45 +0000106 void VisitObjCIsaExpr(ObjCIsaExpr *E);
Mike Stump11289f42009-09-09 15:08:12 +0000107
108 // Objective-C Statements
Chris Lattner1f551822009-04-27 06:20:01 +0000109 void VisitObjCForCollectionStmt(ObjCForCollectionStmt *);
110 void VisitObjCAtCatchStmt(ObjCAtCatchStmt *);
111 void VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *);
112 void VisitObjCAtTryStmt(ObjCAtTryStmt *);
113 void VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *);
114 void VisitObjCAtThrowStmt(ObjCAtThrowStmt *);
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +0000115
Mike Stump11289f42009-09-09 15:08:12 +0000116 // C++ Statements
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000117 void VisitCXXCatchStmt(CXXCatchStmt *S);
118 void VisitCXXTryStmt(CXXTryStmt *S);
Richard Smith02e85f32011-04-14 22:09:26 +0000119 void VisitCXXForRangeStmt(CXXForRangeStmt *);
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000120
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +0000121 void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
Chris Lattnerb7e7f722010-05-09 05:36:05 +0000122 void VisitCXXMemberCallExpr(CXXMemberCallExpr *E);
Douglas Gregor5d3507d2009-09-09 23:08:42 +0000123 void VisitCXXConstructExpr(CXXConstructExpr *E);
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +0000124 void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
Sam Weinigd01101e2010-01-16 21:21:01 +0000125 void VisitCXXNamedCastExpr(CXXNamedCastExpr *E);
126 void VisitCXXStaticCastExpr(CXXStaticCastExpr *E);
127 void VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E);
128 void VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E);
129 void VisitCXXConstCastExpr(CXXConstCastExpr *E);
130 void VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E);
Sam Weinige83b3ac2010-02-07 06:32:43 +0000131 void VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E);
132 void VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E);
Chris Lattner13a5ecc2010-05-09 06:03:39 +0000133 void VisitCXXTypeidExpr(CXXTypeidExpr *E);
Francois Pichet9f4f2072010-09-08 12:20:18 +0000134 void VisitCXXUuidofExpr(CXXUuidofExpr *E);
Chris Lattner98267332010-05-09 06:15:05 +0000135 void VisitCXXThisExpr(CXXThisExpr *E);
136 void VisitCXXThrowExpr(CXXThrowExpr *E);
Chris Lattnere2437f42010-05-09 06:40:08 +0000137 void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E);
Chris Lattnercba86142010-05-10 00:25:06 +0000138 void VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +0000139
Douglas Gregor747eb782010-07-08 06:14:04 +0000140 void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
Chris Lattnerabfb58d2010-05-10 01:22:27 +0000141 void VisitCXXNewExpr(CXXNewExpr *E);
Argyrios Kyrtzidis6e57c352010-06-22 17:07:59 +0000142 void VisitCXXDeleteExpr(CXXDeleteExpr *E);
Argyrios Kyrtzidis99a226d2010-06-28 09:32:03 +0000143 void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +0000144
John McCall5d413782010-12-06 08:20:24 +0000145 void VisitExprWithCleanups(ExprWithCleanups *E);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +0000146 void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
Argyrios Kyrtzidiscd444d1a2010-06-28 09:31:56 +0000147 void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +0000148 void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +0000149
150 void VisitOverloadExpr(OverloadExpr *E);
151 void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E);
Argyrios Kyrtzidis58e01ad2010-06-25 09:03:34 +0000152 void VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E);
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +0000153
154 void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
Francois Pichet9dfa3ce2010-12-07 00:08:36 +0000155 void VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E);
John Wiegley6242b6a2011-04-28 00:16:57 +0000156 void VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E);
John Wiegleyf9f65842011-04-25 06:54:41 +0000157 void VisitExpressionTraitExpr(ExpressionTraitExpr *E);
Sebastian Redl9ac55dd2010-09-10 20:55:54 +0000158 void VisitCXXNoexceptExpr(CXXNoexceptExpr *E);
Douglas Gregore8e9dd62011-01-03 17:17:50 +0000159 void VisitPackExpansionExpr(PackExpansionExpr *E);
Douglas Gregor820ba7b2011-01-04 17:33:58 +0000160 void VisitSizeOfPackExpr(SizeOfPackExpr *E);
Douglas Gregorcdbc5392011-01-15 01:15:58 +0000161 void VisitSubstNonTypeTemplateParmPackExpr(
162 SubstNonTypeTemplateParmPackExpr *E);
John McCall8d69a212010-11-15 23:31:06 +0000163 void VisitOpaqueValueExpr(OpaqueValueExpr *E);
Peter Collingbourne41f85462011-02-09 21:07:24 +0000164
165 // CUDA Expressions
166 void VisitCUDAKernelCallExpr(CUDAKernelCallExpr *E);
Chris Lattner1f551822009-04-27 06:20:01 +0000167 };
168}
169
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000170void ASTStmtWriter::
Argyrios Kyrtzidisb5288de2010-06-28 09:31:48 +0000171AddExplicitTemplateArgumentList(const ExplicitTemplateArgumentList &Args) {
Argyrios Kyrtzidisb5288de2010-06-28 09:31:48 +0000172 Writer.AddSourceLocation(Args.LAngleLoc, Record);
173 Writer.AddSourceLocation(Args.RAngleLoc, Record);
174 for (unsigned i=0; i != Args.NumTemplateArgs; ++i)
175 Writer.AddTemplateArgumentLoc(Args.getTemplateArgs()[i], Record);
176}
177
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000178void ASTStmtWriter::VisitStmt(Stmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000179}
180
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000181void ASTStmtWriter::VisitNullStmt(NullStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000182 VisitStmt(S);
183 Writer.AddSourceLocation(S->getSemiLoc(), Record);
Argyrios Kyrtzidisf7620e42011-04-27 05:04:02 +0000184 Writer.AddSourceLocation(S->LeadingEmptyMacro, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000185 Code = serialization::STMT_NULL;
Chris Lattner1f551822009-04-27 06:20:01 +0000186}
187
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000188void ASTStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000189 VisitStmt(S);
190 Record.push_back(S->size());
191 for (CompoundStmt::body_iterator CS = S->body_begin(), CSEnd = S->body_end();
192 CS != CSEnd; ++CS)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000193 Writer.AddStmt(*CS);
Chris Lattner1f551822009-04-27 06:20:01 +0000194 Writer.AddSourceLocation(S->getLBracLoc(), Record);
195 Writer.AddSourceLocation(S->getRBracLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000196 Code = serialization::STMT_COMPOUND;
Chris Lattner1f551822009-04-27 06:20:01 +0000197}
198
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000199void ASTStmtWriter::VisitSwitchCase(SwitchCase *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000200 VisitStmt(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000201 Record.push_back(Writer.getSwitchCaseID(S));
Chris Lattner1f551822009-04-27 06:20:01 +0000202}
203
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000204void ASTStmtWriter::VisitCaseStmt(CaseStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000205 VisitSwitchCase(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000206 Writer.AddStmt(S->getLHS());
207 Writer.AddStmt(S->getRHS());
208 Writer.AddStmt(S->getSubStmt());
Chris Lattner1f551822009-04-27 06:20:01 +0000209 Writer.AddSourceLocation(S->getCaseLoc(), Record);
Douglas Gregor2a2d00f2009-05-15 23:57:33 +0000210 Writer.AddSourceLocation(S->getEllipsisLoc(), Record);
211 Writer.AddSourceLocation(S->getColonLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000212 Code = serialization::STMT_CASE;
Chris Lattner1f551822009-04-27 06:20:01 +0000213}
214
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000215void ASTStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000216 VisitSwitchCase(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000217 Writer.AddStmt(S->getSubStmt());
Chris Lattner1f551822009-04-27 06:20:01 +0000218 Writer.AddSourceLocation(S->getDefaultLoc(), Record);
Douglas Gregor2a2d00f2009-05-15 23:57:33 +0000219 Writer.AddSourceLocation(S->getColonLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000220 Code = serialization::STMT_DEFAULT;
Chris Lattner1f551822009-04-27 06:20:01 +0000221}
222
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000223void ASTStmtWriter::VisitLabelStmt(LabelStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000224 VisitStmt(S);
Chris Lattnerc8e630e2011-02-17 07:39:24 +0000225 Writer.AddDeclRef(S->getDecl(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000226 Writer.AddStmt(S->getSubStmt());
Chris Lattner1f551822009-04-27 06:20:01 +0000227 Writer.AddSourceLocation(S->getIdentLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000228 Code = serialization::STMT_LABEL;
Chris Lattner1f551822009-04-27 06:20:01 +0000229}
230
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000231void ASTStmtWriter::VisitIfStmt(IfStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000232 VisitStmt(S);
Douglas Gregor633caca2009-11-23 23:44:04 +0000233 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000234 Writer.AddStmt(S->getCond());
235 Writer.AddStmt(S->getThen());
236 Writer.AddStmt(S->getElse());
Chris Lattner1f551822009-04-27 06:20:01 +0000237 Writer.AddSourceLocation(S->getIfLoc(), Record);
Douglas Gregor9d73cab2009-05-15 18:53:42 +0000238 Writer.AddSourceLocation(S->getElseLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000239 Code = serialization::STMT_IF;
Chris Lattner1f551822009-04-27 06:20:01 +0000240}
241
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000242void ASTStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000243 VisitStmt(S);
Douglas Gregordcf19622009-11-24 17:07:59 +0000244 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000245 Writer.AddStmt(S->getCond());
246 Writer.AddStmt(S->getBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000247 Writer.AddSourceLocation(S->getSwitchLoc(), Record);
Ted Kremenekc42f3452010-09-09 00:05:53 +0000248 Record.push_back(S->isAllEnumCasesCovered());
Mike Stump11289f42009-09-09 15:08:12 +0000249 for (SwitchCase *SC = S->getSwitchCaseList(); SC;
Chris Lattner1f551822009-04-27 06:20:01 +0000250 SC = SC->getNextSwitchCase())
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000251 Record.push_back(Writer.RecordSwitchCaseID(SC));
Sebastian Redl539c5062010-08-18 23:57:32 +0000252 Code = serialization::STMT_SWITCH;
Chris Lattner1f551822009-04-27 06:20:01 +0000253}
254
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000255void ASTStmtWriter::VisitWhileStmt(WhileStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000256 VisitStmt(S);
Douglas Gregor680f8612009-11-24 21:15:44 +0000257 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000258 Writer.AddStmt(S->getCond());
259 Writer.AddStmt(S->getBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000260 Writer.AddSourceLocation(S->getWhileLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000261 Code = serialization::STMT_WHILE;
Chris Lattner1f551822009-04-27 06:20:01 +0000262}
263
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000264void ASTStmtWriter::VisitDoStmt(DoStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000265 VisitStmt(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000266 Writer.AddStmt(S->getCond());
267 Writer.AddStmt(S->getBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000268 Writer.AddSourceLocation(S->getDoLoc(), Record);
Douglas Gregor3daa82d2009-05-15 21:56:04 +0000269 Writer.AddSourceLocation(S->getWhileLoc(), Record);
Chris Lattner815b70e2009-06-12 23:04:47 +0000270 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000271 Code = serialization::STMT_DO;
Chris Lattner1f551822009-04-27 06:20:01 +0000272}
273
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000274void ASTStmtWriter::VisitForStmt(ForStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000275 VisitStmt(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000276 Writer.AddStmt(S->getInit());
277 Writer.AddStmt(S->getCond());
Douglas Gregor7bab5ff2009-11-25 00:27:52 +0000278 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000279 Writer.AddStmt(S->getInc());
280 Writer.AddStmt(S->getBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000281 Writer.AddSourceLocation(S->getForLoc(), Record);
Douglas Gregor5d138682009-05-15 22:12:32 +0000282 Writer.AddSourceLocation(S->getLParenLoc(), Record);
283 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000284 Code = serialization::STMT_FOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000285}
286
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000287void ASTStmtWriter::VisitGotoStmt(GotoStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000288 VisitStmt(S);
Chris Lattnerc8e630e2011-02-17 07:39:24 +0000289 Writer.AddDeclRef(S->getLabel(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000290 Writer.AddSourceLocation(S->getGotoLoc(), Record);
291 Writer.AddSourceLocation(S->getLabelLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000292 Code = serialization::STMT_GOTO;
Chris Lattner1f551822009-04-27 06:20:01 +0000293}
294
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000295void ASTStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000296 VisitStmt(S);
297 Writer.AddSourceLocation(S->getGotoLoc(), Record);
Douglas Gregor30776d42009-05-16 00:20:29 +0000298 Writer.AddSourceLocation(S->getStarLoc(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000299 Writer.AddStmt(S->getTarget());
Sebastian Redl539c5062010-08-18 23:57:32 +0000300 Code = serialization::STMT_INDIRECT_GOTO;
Chris Lattner1f551822009-04-27 06:20:01 +0000301}
302
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000303void ASTStmtWriter::VisitContinueStmt(ContinueStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000304 VisitStmt(S);
305 Writer.AddSourceLocation(S->getContinueLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000306 Code = serialization::STMT_CONTINUE;
Chris Lattner1f551822009-04-27 06:20:01 +0000307}
308
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000309void ASTStmtWriter::VisitBreakStmt(BreakStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000310 VisitStmt(S);
311 Writer.AddSourceLocation(S->getBreakLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000312 Code = serialization::STMT_BREAK;
Chris Lattner1f551822009-04-27 06:20:01 +0000313}
314
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000315void ASTStmtWriter::VisitReturnStmt(ReturnStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000316 VisitStmt(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000317 Writer.AddStmt(S->getRetValue());
Chris Lattner1f551822009-04-27 06:20:01 +0000318 Writer.AddSourceLocation(S->getReturnLoc(), Record);
Douglas Gregor6fd1b182010-05-15 06:01:05 +0000319 Writer.AddDeclRef(S->getNRVOCandidate(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000320 Code = serialization::STMT_RETURN;
Chris Lattner1f551822009-04-27 06:20:01 +0000321}
322
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000323void ASTStmtWriter::VisitDeclStmt(DeclStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000324 VisitStmt(S);
325 Writer.AddSourceLocation(S->getStartLoc(), Record);
326 Writer.AddSourceLocation(S->getEndLoc(), Record);
327 DeclGroupRef DG = S->getDeclGroup();
328 for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
329 Writer.AddDeclRef(*D, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000330 Code = serialization::STMT_DECL;
Chris Lattner1f551822009-04-27 06:20:01 +0000331}
332
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000333void ASTStmtWriter::VisitAsmStmt(AsmStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000334 VisitStmt(S);
335 Record.push_back(S->getNumOutputs());
336 Record.push_back(S->getNumInputs());
337 Record.push_back(S->getNumClobbers());
338 Writer.AddSourceLocation(S->getAsmLoc(), Record);
339 Writer.AddSourceLocation(S->getRParenLoc(), Record);
340 Record.push_back(S->isVolatile());
341 Record.push_back(S->isSimple());
Mike Stump90be58a2010-01-04 22:37:17 +0000342 Record.push_back(S->isMSAsm());
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000343 Writer.AddStmt(S->getAsmString());
Chris Lattner1f551822009-04-27 06:20:01 +0000344
345 // Outputs
Anders Carlsson9a020f92010-01-30 22:25:16 +0000346 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
347 Writer.AddIdentifierRef(S->getOutputIdentifier(I), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000348 Writer.AddStmt(S->getOutputConstraintLiteral(I));
349 Writer.AddStmt(S->getOutputExpr(I));
Chris Lattner1f551822009-04-27 06:20:01 +0000350 }
351
352 // Inputs
353 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
Anders Carlsson9a020f92010-01-30 22:25:16 +0000354 Writer.AddIdentifierRef(S->getInputIdentifier(I), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000355 Writer.AddStmt(S->getInputConstraintLiteral(I));
356 Writer.AddStmt(S->getInputExpr(I));
Chris Lattner1f551822009-04-27 06:20:01 +0000357 }
358
359 // Clobbers
360 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000361 Writer.AddStmt(S->getClobber(I));
Chris Lattner1f551822009-04-27 06:20:01 +0000362
Sebastian Redl539c5062010-08-18 23:57:32 +0000363 Code = serialization::STMT_ASM;
Chris Lattner1f551822009-04-27 06:20:01 +0000364}
365
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000366void ASTStmtWriter::VisitExpr(Expr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000367 VisitStmt(E);
368 Writer.AddTypeRef(E->getType(), Record);
369 Record.push_back(E->isTypeDependent());
370 Record.push_back(E->isValueDependent());
Douglas Gregor506bd562010-12-13 22:49:22 +0000371 Record.push_back(E->containsUnexpandedParameterPack());
John McCall7decc9e2010-11-18 06:31:45 +0000372 Record.push_back(E->getValueKind());
373 Record.push_back(E->getObjectKind());
Chris Lattner1f551822009-04-27 06:20:01 +0000374}
375
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000376void ASTStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000377 VisitExpr(E);
378 Writer.AddSourceLocation(E->getLocation(), Record);
379 Record.push_back(E->getIdentType()); // FIXME: stable encoding
Sebastian Redl539c5062010-08-18 23:57:32 +0000380 Code = serialization::EXPR_PREDEFINED;
Chris Lattner1f551822009-04-27 06:20:01 +0000381}
382
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000383void ASTStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000384 VisitExpr(E);
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000385
386 Record.push_back(E->hasQualifier());
Douglas Gregor87866ce2011-02-04 12:01:24 +0000387 Record.push_back(E->hasExplicitTemplateArgs());
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000388
Douglas Gregor87866ce2011-02-04 12:01:24 +0000389 if (E->hasExplicitTemplateArgs()) {
390 unsigned NumTemplateArgs = E->getNumTemplateArgs();
391 Record.push_back(NumTemplateArgs);
Douglas Gregor87866ce2011-02-04 12:01:24 +0000392 }
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000393
Anders Carlsson80756f62011-03-06 18:19:42 +0000394 if (E->hasQualifier())
395 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
396
397 if (E->hasExplicitTemplateArgs())
398 AddExplicitTemplateArgumentList(E->getExplicitTemplateArgs());
399
Chris Lattner1f551822009-04-27 06:20:01 +0000400 Writer.AddDeclRef(E->getDecl(), Record);
401 Writer.AddSourceLocation(E->getLocation(), Record);
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +0000402 Writer.AddDeclarationNameLoc(E->DNLoc, E->getDecl()->getDeclName(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000403 Code = serialization::EXPR_DECL_REF;
Chris Lattner1f551822009-04-27 06:20:01 +0000404}
405
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000406void ASTStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000407 VisitExpr(E);
408 Writer.AddSourceLocation(E->getLocation(), Record);
409 Writer.AddAPInt(E->getValue(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000410 Code = serialization::EXPR_INTEGER_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000411}
412
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000413void ASTStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000414 VisitExpr(E);
415 Writer.AddAPFloat(E->getValue(), Record);
416 Record.push_back(E->isExact());
417 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000418 Code = serialization::EXPR_FLOATING_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000419}
420
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000421void ASTStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000422 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000423 Writer.AddStmt(E->getSubExpr());
Sebastian Redl539c5062010-08-18 23:57:32 +0000424 Code = serialization::EXPR_IMAGINARY_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000425}
426
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000427void ASTStmtWriter::VisitStringLiteral(StringLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000428 VisitExpr(E);
429 Record.push_back(E->getByteLength());
430 Record.push_back(E->getNumConcatenated());
431 Record.push_back(E->isWide());
Anders Carlsson75245402011-04-14 00:40:03 +0000432 Record.push_back(E->isPascal());
Chris Lattner1f551822009-04-27 06:20:01 +0000433 // FIXME: String data should be stored as a blob at the end of the
434 // StringLiteral. However, we can't do so now because we have no
435 // provision for coping with abbreviations when we're jumping around
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000436 // the AST file during deserialization.
Benjamin Kramer35b077e2010-08-17 12:54:38 +0000437 Record.append(E->getString().begin(), E->getString().end());
Chris Lattner1f551822009-04-27 06:20:01 +0000438 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
439 Writer.AddSourceLocation(E->getStrTokenLoc(I), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000440 Code = serialization::EXPR_STRING_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000441}
442
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000443void ASTStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000444 VisitExpr(E);
445 Record.push_back(E->getValue());
Chris Lattnere6434cb2009-08-24 17:39:36 +0000446 Writer.AddSourceLocation(E->getLocation(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000447 Record.push_back(E->isWide());
Sebastian Redl539c5062010-08-18 23:57:32 +0000448 Code = serialization::EXPR_CHARACTER_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000449}
450
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000451void ASTStmtWriter::VisitParenExpr(ParenExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000452 VisitExpr(E);
453 Writer.AddSourceLocation(E->getLParen(), Record);
454 Writer.AddSourceLocation(E->getRParen(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000455 Writer.AddStmt(E->getSubExpr());
Sebastian Redl539c5062010-08-18 23:57:32 +0000456 Code = serialization::EXPR_PAREN;
Chris Lattner1f551822009-04-27 06:20:01 +0000457}
458
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000459void ASTStmtWriter::VisitParenListExpr(ParenListExpr *E) {
Argyrios Kyrtzidisf9f47c82010-06-30 08:49:18 +0000460 VisitExpr(E);
461 Record.push_back(E->NumExprs);
462 for (unsigned i=0; i != E->NumExprs; ++i)
463 Writer.AddStmt(E->Exprs[i]);
464 Writer.AddSourceLocation(E->LParenLoc, Record);
465 Writer.AddSourceLocation(E->RParenLoc, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000466 Code = serialization::EXPR_PAREN_LIST;
Argyrios Kyrtzidisf9f47c82010-06-30 08:49:18 +0000467}
468
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000469void ASTStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000470 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000471 Writer.AddStmt(E->getSubExpr());
Chris Lattner1f551822009-04-27 06:20:01 +0000472 Record.push_back(E->getOpcode()); // FIXME: stable encoding
473 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000474 Code = serialization::EXPR_UNARY_OPERATOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000475}
476
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000477void ASTStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
Douglas Gregor882211c2010-04-28 22:16:22 +0000478 VisitExpr(E);
479 Record.push_back(E->getNumComponents());
480 Record.push_back(E->getNumExpressions());
481 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
482 Writer.AddSourceLocation(E->getRParenLoc(), Record);
483 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
484 for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
485 const OffsetOfExpr::OffsetOfNode &ON = E->getComponent(I);
486 Record.push_back(ON.getKind()); // FIXME: Stable encoding
Abramo Bagnara6b6f0512011-03-12 09:45:03 +0000487 Writer.AddSourceLocation(ON.getSourceRange().getBegin(), Record);
488 Writer.AddSourceLocation(ON.getSourceRange().getEnd(), Record);
Douglas Gregor882211c2010-04-28 22:16:22 +0000489 switch (ON.getKind()) {
490 case OffsetOfExpr::OffsetOfNode::Array:
491 Record.push_back(ON.getArrayExprIndex());
492 break;
493
494 case OffsetOfExpr::OffsetOfNode::Field:
495 Writer.AddDeclRef(ON.getField(), Record);
496 break;
497
498 case OffsetOfExpr::OffsetOfNode::Identifier:
499 Writer.AddIdentifierRef(ON.getFieldName(), Record);
500 break;
Douglas Gregord1702062010-04-29 00:18:15 +0000501
502 case OffsetOfExpr::OffsetOfNode::Base:
Argyrios Kyrtzidisd67d4cc2010-07-29 18:16:10 +0000503 Writer.AddCXXBaseSpecifier(*ON.getBase(), Record);
Douglas Gregord1702062010-04-29 00:18:15 +0000504 break;
Douglas Gregor882211c2010-04-28 22:16:22 +0000505 }
506 }
507 for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000508 Writer.AddStmt(E->getIndexExpr(I));
Sebastian Redl539c5062010-08-18 23:57:32 +0000509 Code = serialization::EXPR_OFFSETOF;
Douglas Gregor882211c2010-04-28 22:16:22 +0000510}
511
Peter Collingbournee190dee2011-03-11 19:24:49 +0000512void ASTStmtWriter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000513 VisitExpr(E);
Peter Collingbournee190dee2011-03-11 19:24:49 +0000514 Record.push_back(E->getKind());
Chris Lattner1f551822009-04-27 06:20:01 +0000515 if (E->isArgumentType())
John McCallbcd03502009-12-07 02:54:59 +0000516 Writer.AddTypeSourceInfo(E->getArgumentTypeInfo(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000517 else {
518 Record.push_back(0);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000519 Writer.AddStmt(E->getArgumentExpr());
Chris Lattner1f551822009-04-27 06:20:01 +0000520 }
521 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
522 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000523 Code = serialization::EXPR_SIZEOF_ALIGN_OF;
Chris Lattner1f551822009-04-27 06:20:01 +0000524}
525
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000526void ASTStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000527 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000528 Writer.AddStmt(E->getLHS());
529 Writer.AddStmt(E->getRHS());
Chris Lattner1f551822009-04-27 06:20:01 +0000530 Writer.AddSourceLocation(E->getRBracketLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000531 Code = serialization::EXPR_ARRAY_SUBSCRIPT;
Chris Lattner1f551822009-04-27 06:20:01 +0000532}
533
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000534void ASTStmtWriter::VisitCallExpr(CallExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000535 VisitExpr(E);
536 Record.push_back(E->getNumArgs());
537 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000538 Writer.AddStmt(E->getCallee());
Chris Lattner1f551822009-04-27 06:20:01 +0000539 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
540 Arg != ArgEnd; ++Arg)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000541 Writer.AddStmt(*Arg);
Sebastian Redl539c5062010-08-18 23:57:32 +0000542 Code = serialization::EXPR_CALL;
Chris Lattner1f551822009-04-27 06:20:01 +0000543}
544
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000545void ASTStmtWriter::VisitMemberExpr(MemberExpr *E) {
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000546 // Don't call VisitExpr, we'll write everything here.
547
548 Record.push_back(E->hasQualifier());
Douglas Gregorea972d32011-02-28 21:54:11 +0000549 if (E->hasQualifier())
550 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000551
Douglas Gregor87866ce2011-02-04 12:01:24 +0000552 Record.push_back(E->hasExplicitTemplateArgs());
553 if (E->hasExplicitTemplateArgs()) {
554 unsigned NumTemplateArgs = E->getNumTemplateArgs();
555 Record.push_back(NumTemplateArgs);
Argyrios Kyrtzidis1985bb32010-07-08 13:09:47 +0000556 Writer.AddSourceLocation(E->getLAngleLoc(), Record);
557 Writer.AddSourceLocation(E->getRAngleLoc(), Record);
558 for (unsigned i=0; i != NumTemplateArgs; ++i)
559 Writer.AddTemplateArgumentLoc(E->getTemplateArgs()[i], Record);
560 }
561
562 DeclAccessPair FoundDecl = E->getFoundDecl();
563 Writer.AddDeclRef(FoundDecl.getDecl(), Record);
564 Record.push_back(FoundDecl.getAccess());
565
566 Writer.AddTypeRef(E->getType(), Record);
John McCall7decc9e2010-11-18 06:31:45 +0000567 Record.push_back(E->getValueKind());
568 Record.push_back(E->getObjectKind());
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000569 Writer.AddStmt(E->getBase());
Chris Lattner1f551822009-04-27 06:20:01 +0000570 Writer.AddDeclRef(E->getMemberDecl(), Record);
571 Writer.AddSourceLocation(E->getMemberLoc(), Record);
572 Record.push_back(E->isArrow());
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +0000573 Writer.AddDeclarationNameLoc(E->MemberDNLoc,
574 E->getMemberDecl()->getDeclName(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000575 Code = serialization::EXPR_MEMBER;
Chris Lattner1f551822009-04-27 06:20:01 +0000576}
577
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000578void ASTStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
Steve Naroffe87026a2009-07-24 17:54:45 +0000579 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000580 Writer.AddStmt(E->getBase());
Steve Naroffe87026a2009-07-24 17:54:45 +0000581 Writer.AddSourceLocation(E->getIsaMemberLoc(), Record);
582 Record.push_back(E->isArrow());
Sebastian Redl539c5062010-08-18 23:57:32 +0000583 Code = serialization::EXPR_OBJC_ISA;
Steve Naroffe87026a2009-07-24 17:54:45 +0000584}
585
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000586void ASTStmtWriter::VisitCastExpr(CastExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000587 VisitExpr(E);
John McCallcf142162010-08-07 06:22:56 +0000588 Record.push_back(E->path_size());
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000589 Writer.AddStmt(E->getSubExpr());
Anders Carlssona2615922009-07-31 00:48:10 +0000590 Record.push_back(E->getCastKind()); // FIXME: stable encoding
John McCallcf142162010-08-07 06:22:56 +0000591
592 for (CastExpr::path_iterator
593 PI = E->path_begin(), PE = E->path_end(); PI != PE; ++PI)
594 Writer.AddCXXBaseSpecifier(**PI, Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000595}
596
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000597void ASTStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000598 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000599 Writer.AddStmt(E->getLHS());
600 Writer.AddStmt(E->getRHS());
Chris Lattner1f551822009-04-27 06:20:01 +0000601 Record.push_back(E->getOpcode()); // FIXME: stable encoding
602 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000603 Code = serialization::EXPR_BINARY_OPERATOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000604}
605
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000606void ASTStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000607 VisitBinaryOperator(E);
608 Writer.AddTypeRef(E->getComputationLHSType(), Record);
609 Writer.AddTypeRef(E->getComputationResultType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000610 Code = serialization::EXPR_COMPOUND_ASSIGN_OPERATOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000611}
612
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000613void ASTStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000614 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000615 Writer.AddStmt(E->getCond());
616 Writer.AddStmt(E->getLHS());
617 Writer.AddStmt(E->getRHS());
Douglas Gregor7e112b02009-08-26 14:37:04 +0000618 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
619 Writer.AddSourceLocation(E->getColonLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000620 Code = serialization::EXPR_CONDITIONAL_OPERATOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000621}
622
John McCallc07a0c72011-02-17 10:25:35 +0000623void
624ASTStmtWriter::VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
625 VisitExpr(E);
626 Writer.AddStmt(E->getOpaqueValue());
627 Writer.AddStmt(E->getCommon());
628 Writer.AddStmt(E->getCond());
629 Writer.AddStmt(E->getTrueExpr());
630 Writer.AddStmt(E->getFalseExpr());
631 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
632 Writer.AddSourceLocation(E->getColonLoc(), Record);
633 Code = serialization::EXPR_BINARY_CONDITIONAL_OPERATOR;
634}
635
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000636void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000637 VisitCastExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +0000638 Code = serialization::EXPR_IMPLICIT_CAST;
Chris Lattner1f551822009-04-27 06:20:01 +0000639}
640
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000641void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000642 VisitCastExpr(E);
John McCall97513962010-01-15 18:39:57 +0000643 Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000644}
645
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000646void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000647 VisitExplicitCastExpr(E);
648 Writer.AddSourceLocation(E->getLParenLoc(), Record);
649 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000650 Code = serialization::EXPR_CSTYLE_CAST;
Chris Lattner1f551822009-04-27 06:20:01 +0000651}
652
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000653void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000654 VisitExpr(E);
655 Writer.AddSourceLocation(E->getLParenLoc(), Record);
John McCalle15bbff2010-01-18 19:35:47 +0000656 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000657 Writer.AddStmt(E->getInitializer());
Chris Lattner1f551822009-04-27 06:20:01 +0000658 Record.push_back(E->isFileScope());
Sebastian Redl539c5062010-08-18 23:57:32 +0000659 Code = serialization::EXPR_COMPOUND_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000660}
661
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000662void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000663 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000664 Writer.AddStmt(E->getBase());
Chris Lattner1f551822009-04-27 06:20:01 +0000665 Writer.AddIdentifierRef(&E->getAccessor(), Record);
666 Writer.AddSourceLocation(E->getAccessorLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000667 Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
Chris Lattner1f551822009-04-27 06:20:01 +0000668}
669
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000670void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000671 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000672 Writer.AddStmt(E->getSyntacticForm());
Chris Lattner1f551822009-04-27 06:20:01 +0000673 Writer.AddSourceLocation(E->getLBraceLoc(), Record);
674 Writer.AddSourceLocation(E->getRBraceLoc(), Record);
Argyrios Kyrtzidisb2ed28e2011-04-21 00:27:41 +0000675 bool isArrayFiller = E->ArrayFillerOrUnionFieldInit.is<Expr*>();
676 Record.push_back(isArrayFiller);
677 if (isArrayFiller)
678 Writer.AddStmt(E->getArrayFiller());
679 else
680 Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000681 Record.push_back(E->hadArrayRangeDesignator());
Argyrios Kyrtzidisbbcefa72011-04-22 05:29:30 +0000682 Record.push_back(E->getNumInits());
683 if (isArrayFiller) {
684 // ArrayFiller may have filled "holes" due to designated initializer.
685 // Replace them by 0 to indicate that the filler goes in that place.
686 Expr *filler = E->getArrayFiller();
687 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
688 Writer.AddStmt(E->getInit(I) != filler ? E->getInit(I) : 0);
689 } else {
690 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
691 Writer.AddStmt(E->getInit(I));
692 }
Sebastian Redl539c5062010-08-18 23:57:32 +0000693 Code = serialization::EXPR_INIT_LIST;
Chris Lattner1f551822009-04-27 06:20:01 +0000694}
695
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000696void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000697 VisitExpr(E);
698 Record.push_back(E->getNumSubExprs());
699 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000700 Writer.AddStmt(E->getSubExpr(I));
Chris Lattner1f551822009-04-27 06:20:01 +0000701 Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
702 Record.push_back(E->usesGNUSyntax());
703 for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
704 DEnd = E->designators_end();
705 D != DEnd; ++D) {
706 if (D->isFieldDesignator()) {
707 if (FieldDecl *Field = D->getField()) {
Sebastian Redl539c5062010-08-18 23:57:32 +0000708 Record.push_back(serialization::DESIG_FIELD_DECL);
Chris Lattner1f551822009-04-27 06:20:01 +0000709 Writer.AddDeclRef(Field, Record);
710 } else {
Sebastian Redl539c5062010-08-18 23:57:32 +0000711 Record.push_back(serialization::DESIG_FIELD_NAME);
Chris Lattner1f551822009-04-27 06:20:01 +0000712 Writer.AddIdentifierRef(D->getFieldName(), Record);
713 }
714 Writer.AddSourceLocation(D->getDotLoc(), Record);
715 Writer.AddSourceLocation(D->getFieldLoc(), Record);
716 } else if (D->isArrayDesignator()) {
Sebastian Redl539c5062010-08-18 23:57:32 +0000717 Record.push_back(serialization::DESIG_ARRAY);
Chris Lattner1f551822009-04-27 06:20:01 +0000718 Record.push_back(D->getFirstExprIndex());
719 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
720 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
721 } else {
722 assert(D->isArrayRangeDesignator() && "Unknown designator");
Sebastian Redl539c5062010-08-18 23:57:32 +0000723 Record.push_back(serialization::DESIG_ARRAY_RANGE);
Chris Lattner1f551822009-04-27 06:20:01 +0000724 Record.push_back(D->getFirstExprIndex());
725 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
726 Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
727 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
728 }
729 }
Sebastian Redl539c5062010-08-18 23:57:32 +0000730 Code = serialization::EXPR_DESIGNATED_INIT;
Chris Lattner1f551822009-04-27 06:20:01 +0000731}
732
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000733void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000734 VisitExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +0000735 Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
Chris Lattner1f551822009-04-27 06:20:01 +0000736}
737
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000738void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000739 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000740 Writer.AddStmt(E->getSubExpr());
Abramo Bagnara27db2392010-08-10 10:06:15 +0000741 Writer.AddTypeSourceInfo(E->getWrittenTypeInfo(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000742 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
743 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000744 Code = serialization::EXPR_VA_ARG;
Chris Lattner1f551822009-04-27 06:20:01 +0000745}
746
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000747void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000748 VisitExpr(E);
749 Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
750 Writer.AddSourceLocation(E->getLabelLoc(), Record);
Chris Lattnerc8e630e2011-02-17 07:39:24 +0000751 Writer.AddDeclRef(E->getLabel(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000752 Code = serialization::EXPR_ADDR_LABEL;
Chris Lattner1f551822009-04-27 06:20:01 +0000753}
754
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000755void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000756 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000757 Writer.AddStmt(E->getSubStmt());
Chris Lattner1f551822009-04-27 06:20:01 +0000758 Writer.AddSourceLocation(E->getLParenLoc(), Record);
759 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000760 Code = serialization::EXPR_STMT;
Chris Lattner1f551822009-04-27 06:20:01 +0000761}
762
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000763void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000764 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000765 Writer.AddStmt(E->getCond());
766 Writer.AddStmt(E->getLHS());
767 Writer.AddStmt(E->getRHS());
Chris Lattner1f551822009-04-27 06:20:01 +0000768 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
769 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000770 Code = serialization::EXPR_CHOOSE;
Chris Lattner1f551822009-04-27 06:20:01 +0000771}
772
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000773void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000774 VisitExpr(E);
775 Writer.AddSourceLocation(E->getTokenLocation(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000776 Code = serialization::EXPR_GNU_NULL;
Chris Lattner1f551822009-04-27 06:20:01 +0000777}
778
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000779void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000780 VisitExpr(E);
781 Record.push_back(E->getNumSubExprs());
782 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000783 Writer.AddStmt(E->getExpr(I));
Chris Lattner1f551822009-04-27 06:20:01 +0000784 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
785 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000786 Code = serialization::EXPR_SHUFFLE_VECTOR;
Chris Lattner1f551822009-04-27 06:20:01 +0000787}
788
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000789void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000790 VisitExpr(E);
791 Writer.AddDeclRef(E->getBlockDecl(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000792 Code = serialization::EXPR_BLOCK;
Chris Lattner1f551822009-04-27 06:20:01 +0000793}
794
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000795void ASTStmtWriter::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000796 VisitExpr(E);
797 Writer.AddDeclRef(E->getDecl(), Record);
798 Writer.AddSourceLocation(E->getLocation(), Record);
799 Record.push_back(E->isByRef());
Fariborz Jahaniane8918d52009-06-20 00:02:26 +0000800 Record.push_back(E->isConstQualAdded());
Sebastian Redl539c5062010-08-18 23:57:32 +0000801 Code = serialization::EXPR_BLOCK_DECL_REF;
Chris Lattner1f551822009-04-27 06:20:01 +0000802}
803
Peter Collingbourne91147592011-04-15 00:35:48 +0000804void ASTStmtWriter::VisitGenericSelectionExpr(GenericSelectionExpr *E) {
805 VisitExpr(E);
806 Record.push_back(E->getNumAssocs());
807
808 Writer.AddStmt(E->getControllingExpr());
809 for (unsigned I = 0, N = E->getNumAssocs(); I != N; ++I) {
810 Writer.AddTypeSourceInfo(E->getAssocTypeSourceInfo(I), Record);
811 Writer.AddStmt(E->getAssocExpr(I));
812 }
813 Record.push_back(E->isResultDependent() ? -1U : E->getResultIndex());
814
815 Writer.AddSourceLocation(E->getGenericLoc(), Record);
816 Writer.AddSourceLocation(E->getDefaultLoc(), Record);
817 Writer.AddSourceLocation(E->getRParenLoc(), Record);
818 Code = serialization::EXPR_GENERIC_SELECTION;
819}
820
Chris Lattner1f551822009-04-27 06:20:01 +0000821//===----------------------------------------------------------------------===//
822// Objective-C Expressions and Statements.
823//===----------------------------------------------------------------------===//
824
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000825void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000826 VisitExpr(E);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000827 Writer.AddStmt(E->getString());
Chris Lattner1f551822009-04-27 06:20:01 +0000828 Writer.AddSourceLocation(E->getAtLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000829 Code = serialization::EXPR_OBJC_STRING_LITERAL;
Chris Lattner1f551822009-04-27 06:20:01 +0000830}
831
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000832void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000833 VisitExpr(E);
Douglas Gregorabd9e962010-04-20 15:39:42 +0000834 Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000835 Writer.AddSourceLocation(E->getAtLoc(), Record);
836 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000837 Code = serialization::EXPR_OBJC_ENCODE;
Chris Lattner1f551822009-04-27 06:20:01 +0000838}
839
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000840void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000841 VisitExpr(E);
842 Writer.AddSelectorRef(E->getSelector(), Record);
843 Writer.AddSourceLocation(E->getAtLoc(), Record);
844 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000845 Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
Chris Lattner1f551822009-04-27 06:20:01 +0000846}
847
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000848void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000849 VisitExpr(E);
850 Writer.AddDeclRef(E->getProtocol(), Record);
851 Writer.AddSourceLocation(E->getAtLoc(), Record);
852 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000853 Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
Chris Lattner1f551822009-04-27 06:20:01 +0000854}
855
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000856void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000857 VisitExpr(E);
858 Writer.AddDeclRef(E->getDecl(), Record);
859 Writer.AddSourceLocation(E->getLocation(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000860 Writer.AddStmt(E->getBase());
Chris Lattner1f551822009-04-27 06:20:01 +0000861 Record.push_back(E->isArrow());
862 Record.push_back(E->isFreeIvar());
Sebastian Redl539c5062010-08-18 23:57:32 +0000863 Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
Chris Lattner1f551822009-04-27 06:20:01 +0000864}
865
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000866void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000867 VisitExpr(E);
John McCallb7bd14f2010-12-02 01:19:52 +0000868 Record.push_back(E->isImplicitProperty());
869 if (E->isImplicitProperty()) {
870 Writer.AddDeclRef(E->getImplicitPropertyGetter(), Record);
871 Writer.AddDeclRef(E->getImplicitPropertySetter(), Record);
872 } else {
873 Writer.AddDeclRef(E->getExplicitProperty(), Record);
874 }
Chris Lattner1f551822009-04-27 06:20:01 +0000875 Writer.AddSourceLocation(E->getLocation(), Record);
John McCallb7bd14f2010-12-02 01:19:52 +0000876 Writer.AddSourceLocation(E->getReceiverLocation(), Record);
877 if (E->isObjectReceiver()) {
878 Record.push_back(0);
Fariborz Jahanian681c0752010-10-14 16:04:05 +0000879 Writer.AddStmt(E->getBase());
John McCallb7bd14f2010-12-02 01:19:52 +0000880 } else if (E->isSuperReceiver()) {
881 Record.push_back(1);
882 Writer.AddTypeRef(E->getSuperReceiverType(), Record);
883 } else {
884 Record.push_back(2);
885 Writer.AddDeclRef(E->getClassReceiver(), Record);
886 }
Fariborz Jahanian681c0752010-10-14 16:04:05 +0000887
Sebastian Redl539c5062010-08-18 23:57:32 +0000888 Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
Chris Lattner1f551822009-04-27 06:20:01 +0000889}
890
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000891void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
Chris Lattner1f551822009-04-27 06:20:01 +0000892 VisitExpr(E);
893 Record.push_back(E->getNumArgs());
Douglas Gregor9a129192010-04-21 00:45:42 +0000894 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
895 switch (E->getReceiverKind()) {
896 case ObjCMessageExpr::Instance:
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000897 Writer.AddStmt(E->getInstanceReceiver());
Douglas Gregor9a129192010-04-21 00:45:42 +0000898 break;
899
900 case ObjCMessageExpr::Class:
901 Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
902 break;
903
904 case ObjCMessageExpr::SuperClass:
905 case ObjCMessageExpr::SuperInstance:
906 Writer.AddTypeRef(E->getSuperType(), Record);
907 Writer.AddSourceLocation(E->getSuperLoc(), Record);
908 break;
909 }
910
911 if (E->getMethodDecl()) {
912 Record.push_back(1);
913 Writer.AddDeclRef(E->getMethodDecl(), Record);
914 } else {
915 Record.push_back(0);
916 Writer.AddSelectorRef(E->getSelector(), Record);
917 }
918
Chris Lattner1f551822009-04-27 06:20:01 +0000919 Writer.AddSourceLocation(E->getLeftLoc(), Record);
920 Writer.AddSourceLocation(E->getRightLoc(), Record);
Argyrios Kyrtzidisd0039e52010-12-10 20:08:27 +0000921 Writer.AddSourceLocation(E->getSelectorLoc(), Record);
Chris Lattner1f551822009-04-27 06:20:01 +0000922
923 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
924 Arg != ArgEnd; ++Arg)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000925 Writer.AddStmt(*Arg);
Sebastian Redl539c5062010-08-18 23:57:32 +0000926 Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
Chris Lattner1f551822009-04-27 06:20:01 +0000927}
928
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000929void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +0000930 VisitStmt(S);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000931 Writer.AddStmt(S->getElement());
932 Writer.AddStmt(S->getCollection());
933 Writer.AddStmt(S->getBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000934 Writer.AddSourceLocation(S->getForLoc(), Record);
935 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000936 Code = serialization::STMT_OBJC_FOR_COLLECTION;
Chris Lattner1f551822009-04-27 06:20:01 +0000937}
938
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000939void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000940 Writer.AddStmt(S->getCatchBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000941 Writer.AddDeclRef(S->getCatchParamDecl(), Record);
942 Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
943 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000944 Code = serialization::STMT_OBJC_CATCH;
Chris Lattner1f551822009-04-27 06:20:01 +0000945}
946
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000947void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000948 Writer.AddStmt(S->getFinallyBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000949 Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000950 Code = serialization::STMT_OBJC_FINALLY;
Chris Lattner1f551822009-04-27 06:20:01 +0000951}
952
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000953void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
Douglas Gregor96c79492010-04-23 22:50:49 +0000954 Record.push_back(S->getNumCatchStmts());
955 Record.push_back(S->getFinallyStmt() != 0);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000956 Writer.AddStmt(S->getTryBody());
Douglas Gregor96c79492010-04-23 22:50:49 +0000957 for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000958 Writer.AddStmt(S->getCatchStmt(I));
Douglas Gregor96c79492010-04-23 22:50:49 +0000959 if (S->getFinallyStmt())
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000960 Writer.AddStmt(S->getFinallyStmt());
Chris Lattner1f551822009-04-27 06:20:01 +0000961 Writer.AddSourceLocation(S->getAtTryLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000962 Code = serialization::STMT_OBJC_AT_TRY;
Chris Lattner1f551822009-04-27 06:20:01 +0000963}
964
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000965void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000966 Writer.AddStmt(S->getSynchExpr());
967 Writer.AddStmt(S->getSynchBody());
Chris Lattner1f551822009-04-27 06:20:01 +0000968 Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000969 Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
Chris Lattner1f551822009-04-27 06:20:01 +0000970}
971
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000972void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +0000973 Writer.AddStmt(S->getThrowExpr());
Chris Lattner1f551822009-04-27 06:20:01 +0000974 Writer.AddSourceLocation(S->getThrowLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000975 Code = serialization::STMT_OBJC_AT_THROW;
Chris Lattner1f551822009-04-27 06:20:01 +0000976}
977
978//===----------------------------------------------------------------------===//
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +0000979// C++ Expressions and Statements.
980//===----------------------------------------------------------------------===//
981
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000982void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000983 VisitStmt(S);
984 Writer.AddSourceLocation(S->getCatchLoc(), Record);
985 Writer.AddDeclRef(S->getExceptionDecl(), Record);
986 Writer.AddStmt(S->getHandlerBlock());
Sebastian Redl539c5062010-08-18 23:57:32 +0000987 Code = serialization::STMT_CXX_CATCH;
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000988}
989
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000990void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000991 VisitStmt(S);
992 Record.push_back(S->getNumHandlers());
993 Writer.AddSourceLocation(S->getTryLoc(), Record);
994 Writer.AddStmt(S->getTryBlock());
995 for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
996 Writer.AddStmt(S->getHandler(i));
Sebastian Redl539c5062010-08-18 23:57:32 +0000997 Code = serialization::STMT_CXX_TRY;
Argyrios Kyrtzidis47cd7a92010-07-22 16:03:56 +0000998}
999
Richard Smith02e85f32011-04-14 22:09:26 +00001000void ASTStmtWriter::VisitCXXForRangeStmt(CXXForRangeStmt *S) {
1001 VisitStmt(S);
1002 Writer.AddSourceLocation(S->getForLoc(), Record);
1003 Writer.AddSourceLocation(S->getColonLoc(), Record);
1004 Writer.AddSourceLocation(S->getRParenLoc(), Record);
1005 Writer.AddStmt(S->getRangeStmt());
1006 Writer.AddStmt(S->getBeginEndStmt());
1007 Writer.AddStmt(S->getCond());
1008 Writer.AddStmt(S->getInc());
1009 Writer.AddStmt(S->getLoopVarStmt());
1010 Writer.AddStmt(S->getBody());
1011 Code = serialization::STMT_CXX_FOR_RANGE;
1012}
1013
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001014void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +00001015 VisitCallExpr(E);
1016 Record.push_back(E->getOperator());
Sebastian Redl539c5062010-08-18 23:57:32 +00001017 Code = serialization::EXPR_CXX_OPERATOR_CALL;
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +00001018}
1019
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001020void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
Chris Lattnerb7e7f722010-05-09 05:36:05 +00001021 VisitCallExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +00001022 Code = serialization::EXPR_CXX_MEMBER_CALL;
Chris Lattnerb7e7f722010-05-09 05:36:05 +00001023}
1024
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001025void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
Douglas Gregor5d3507d2009-09-09 23:08:42 +00001026 VisitExpr(E);
Argyrios Kyrtzidis30d98f32010-06-24 08:57:09 +00001027 Record.push_back(E->getNumArgs());
1028 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001029 Writer.AddStmt(E->getArg(I));
Douglas Gregor5d3507d2009-09-09 23:08:42 +00001030 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor85dabae2009-12-16 01:38:02 +00001031 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregor5d3507d2009-09-09 23:08:42 +00001032 Record.push_back(E->isElidable());
Douglas Gregor4f4b1862009-12-16 18:50:27 +00001033 Record.push_back(E->requiresZeroInitialization());
Douglas Gregor222cf0e2010-05-15 00:13:29 +00001034 Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
Chandler Carruth01718152010-10-25 08:47:36 +00001035 Writer.AddSourceRange(E->getParenRange(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001036 Code = serialization::EXPR_CXX_CONSTRUCT;
Douglas Gregor5d3507d2009-09-09 23:08:42 +00001037}
1038
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001039void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +00001040 VisitCXXConstructExpr(E);
Douglas Gregor2b88c112010-09-08 00:15:04 +00001041 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001042 Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +00001043}
1044
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001045void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001046 VisitExplicitCastExpr(E);
Douglas Gregor4478f852011-01-12 22:41:29 +00001047 Writer.AddSourceRange(SourceRange(E->getOperatorLoc(), E->getRParenLoc()),
1048 Record);
Sam Weinigd01101e2010-01-16 21:21:01 +00001049}
1050
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001051void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001052 VisitCXXNamedCastExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +00001053 Code = serialization::EXPR_CXX_STATIC_CAST;
Sam Weinigd01101e2010-01-16 21:21:01 +00001054}
1055
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001056void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001057 VisitCXXNamedCastExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +00001058 Code = serialization::EXPR_CXX_DYNAMIC_CAST;
Sam Weinigd01101e2010-01-16 21:21:01 +00001059}
1060
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001061void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001062 VisitCXXNamedCastExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +00001063 Code = serialization::EXPR_CXX_REINTERPRET_CAST;
Sam Weinigd01101e2010-01-16 21:21:01 +00001064}
1065
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001066void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001067 VisitCXXNamedCastExpr(E);
Sebastian Redl539c5062010-08-18 23:57:32 +00001068 Code = serialization::EXPR_CXX_CONST_CAST;
Sam Weinigd01101e2010-01-16 21:21:01 +00001069}
1070
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001071void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
Sam Weinigd01101e2010-01-16 21:21:01 +00001072 VisitExplicitCastExpr(E);
1073 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1074 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001075 Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
Sam Weinigd01101e2010-01-16 21:21:01 +00001076}
1077
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001078void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
Sam Weinige83b3ac2010-02-07 06:32:43 +00001079 VisitExpr(E);
1080 Record.push_back(E->getValue());
1081 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001082 Code = serialization::EXPR_CXX_BOOL_LITERAL;
Sam Weinige83b3ac2010-02-07 06:32:43 +00001083}
1084
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001085void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
Sam Weinige83b3ac2010-02-07 06:32:43 +00001086 VisitExpr(E);
1087 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001088 Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
Sam Weinige83b3ac2010-02-07 06:32:43 +00001089}
1090
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001091void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
Chris Lattner13a5ecc2010-05-09 06:03:39 +00001092 VisitExpr(E);
1093 Writer.AddSourceRange(E->getSourceRange(), Record);
1094 if (E->isTypeOperand()) {
1095 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001096 Code = serialization::EXPR_CXX_TYPEID_TYPE;
Chris Lattner13a5ecc2010-05-09 06:03:39 +00001097 } else {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001098 Writer.AddStmt(E->getExprOperand());
Sebastian Redl539c5062010-08-18 23:57:32 +00001099 Code = serialization::EXPR_CXX_TYPEID_EXPR;
Chris Lattner13a5ecc2010-05-09 06:03:39 +00001100 }
1101}
1102
Francois Pichet9f4f2072010-09-08 12:20:18 +00001103void ASTStmtWriter::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
1104 VisitExpr(E);
1105 Writer.AddSourceRange(E->getSourceRange(), Record);
1106 if (E->isTypeOperand()) {
1107 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1108 Code = serialization::EXPR_CXX_UUIDOF_TYPE;
1109 } else {
1110 Writer.AddStmt(E->getExprOperand());
1111 Code = serialization::EXPR_CXX_UUIDOF_EXPR;
1112 }
1113}
1114
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001115void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
Chris Lattner98267332010-05-09 06:15:05 +00001116 VisitExpr(E);
1117 Writer.AddSourceLocation(E->getLocation(), Record);
1118 Record.push_back(E->isImplicit());
Sebastian Redl539c5062010-08-18 23:57:32 +00001119 Code = serialization::EXPR_CXX_THIS;
Chris Lattner98267332010-05-09 06:15:05 +00001120}
1121
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001122void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
Chris Lattnere2437f42010-05-09 06:40:08 +00001123 VisitExpr(E);
Chris Lattner98267332010-05-09 06:15:05 +00001124 Writer.AddSourceLocation(E->getThrowLoc(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001125 Writer.AddStmt(E->getSubExpr());
Sebastian Redl539c5062010-08-18 23:57:32 +00001126 Code = serialization::EXPR_CXX_THROW;
Chris Lattner98267332010-05-09 06:15:05 +00001127}
1128
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001129void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
Chris Lattnere2437f42010-05-09 06:40:08 +00001130 VisitExpr(E);
Argyrios Kyrtzidis4259ebc2010-07-02 23:30:15 +00001131
1132 bool HasOtherExprStored = E->Param.getInt();
1133 // Store these first, the reader reads them before creation.
1134 Record.push_back(HasOtherExprStored);
1135 if (HasOtherExprStored)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001136 Writer.AddStmt(E->getExpr());
Argyrios Kyrtzidis4259ebc2010-07-02 23:30:15 +00001137 Writer.AddDeclRef(E->getParam(), Record);
1138 Writer.AddSourceLocation(E->getUsedLocation(), Record);
Chris Lattnere2437f42010-05-09 06:40:08 +00001139
Sebastian Redl539c5062010-08-18 23:57:32 +00001140 Code = serialization::EXPR_CXX_DEFAULT_ARG;
Chris Lattnere2437f42010-05-09 06:40:08 +00001141}
1142
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001143void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
Chris Lattnercba86142010-05-10 00:25:06 +00001144 VisitExpr(E);
1145 Writer.AddCXXTemporary(E->getTemporary(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001146 Writer.AddStmt(E->getSubExpr());
Sebastian Redl539c5062010-08-18 23:57:32 +00001147 Code = serialization::EXPR_CXX_BIND_TEMPORARY;
Chris Lattnercba86142010-05-10 00:25:06 +00001148}
1149
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001150void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001151 VisitExpr(E);
Douglas Gregor2b88c112010-09-08 00:15:04 +00001152 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001153 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001154 Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001155}
1156
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001157void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001158 VisitExpr(E);
1159 Record.push_back(E->isGlobalNew());
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001160 Record.push_back(E->hasInitializer());
John McCall284c48f2011-01-27 09:37:56 +00001161 Record.push_back(E->doesUsualArrayDeleteWantSize());
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001162 Record.push_back(E->isArray());
1163 Record.push_back(E->getNumPlacementArgs());
1164 Record.push_back(E->getNumConstructorArgs());
1165 Writer.AddDeclRef(E->getOperatorNew(), Record);
1166 Writer.AddDeclRef(E->getOperatorDelete(), Record);
1167 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor0744ef62010-09-07 21:49:58 +00001168 Writer.AddTypeSourceInfo(E->getAllocatedTypeSourceInfo(), Record);
Douglas Gregorf2753b32010-07-13 15:54:32 +00001169 Writer.AddSourceRange(E->getTypeIdParens(), Record);
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001170 Writer.AddSourceLocation(E->getStartLoc(), Record);
1171 Writer.AddSourceLocation(E->getEndLoc(), Record);
Chandler Carruth01718152010-10-25 08:47:36 +00001172 Writer.AddSourceLocation(E->getConstructorLParen(), Record);
1173 Writer.AddSourceLocation(E->getConstructorRParen(), Record);
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001174 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
1175 I != e; ++I)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001176 Writer.AddStmt(*I);
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001177
Sebastian Redl539c5062010-08-18 23:57:32 +00001178 Code = serialization::EXPR_CXX_NEW;
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001179}
1180
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001181void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
Argyrios Kyrtzidis6e57c352010-06-22 17:07:59 +00001182 VisitExpr(E);
1183 Record.push_back(E->isGlobalDelete());
1184 Record.push_back(E->isArrayForm());
Argyrios Kyrtzidis14ec9f62010-09-13 20:15:54 +00001185 Record.push_back(E->isArrayFormAsWritten());
John McCall284c48f2011-01-27 09:37:56 +00001186 Record.push_back(E->doesUsualArrayDeleteWantSize());
Argyrios Kyrtzidis6e57c352010-06-22 17:07:59 +00001187 Writer.AddDeclRef(E->getOperatorDelete(), Record);
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001188 Writer.AddStmt(E->getArgument());
Argyrios Kyrtzidis6e57c352010-06-22 17:07:59 +00001189 Writer.AddSourceLocation(E->getSourceRange().getBegin(), Record);
1190
Sebastian Redl539c5062010-08-18 23:57:32 +00001191 Code = serialization::EXPR_CXX_DELETE;
Argyrios Kyrtzidis6e57c352010-06-22 17:07:59 +00001192}
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001193
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001194void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
Argyrios Kyrtzidis99a226d2010-06-28 09:32:03 +00001195 VisitExpr(E);
1196
1197 Writer.AddStmt(E->getBase());
1198 Record.push_back(E->isArrow());
1199 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Douglas Gregora6ce6082011-02-25 18:19:59 +00001200 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
Argyrios Kyrtzidis99a226d2010-06-28 09:32:03 +00001201 Writer.AddTypeSourceInfo(E->getScopeTypeInfo(), Record);
1202 Writer.AddSourceLocation(E->getColonColonLoc(), Record);
1203 Writer.AddSourceLocation(E->getTildeLoc(), Record);
1204
1205 // PseudoDestructorTypeStorage.
1206 Writer.AddIdentifierRef(E->getDestroyedTypeIdentifier(), Record);
1207 if (E->getDestroyedTypeIdentifier())
1208 Writer.AddSourceLocation(E->getDestroyedTypeLoc(), Record);
1209 else
1210 Writer.AddTypeSourceInfo(E->getDestroyedTypeInfo(), Record);
1211
Sebastian Redl539c5062010-08-18 23:57:32 +00001212 Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
Argyrios Kyrtzidis99a226d2010-06-28 09:32:03 +00001213}
1214
John McCall5d413782010-12-06 08:20:24 +00001215void ASTStmtWriter::VisitExprWithCleanups(ExprWithCleanups *E) {
Chris Lattnercba86142010-05-10 00:25:06 +00001216 VisitExpr(E);
1217 Record.push_back(E->getNumTemporaries());
1218 for (unsigned i = 0, e = E->getNumTemporaries(); i != e; ++i)
1219 Writer.AddCXXTemporary(E->getTemporary(i), Record);
1220
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001221 Writer.AddStmt(E->getSubExpr());
John McCall5d413782010-12-06 08:20:24 +00001222 Code = serialization::EXPR_EXPR_WITH_CLEANUPS;
Chris Lattnercba86142010-05-10 00:25:06 +00001223}
1224
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001225void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001226ASTStmtWriter::VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E){
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001227 VisitExpr(E);
Argyrios Kyrtzidisb5288de2010-06-28 09:31:48 +00001228
Douglas Gregor87866ce2011-02-04 12:01:24 +00001229 // Don't emit anything here, hasExplicitTemplateArgs() must be
1230 // emitted first.
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001231
Douglas Gregor87866ce2011-02-04 12:01:24 +00001232 Record.push_back(E->hasExplicitTemplateArgs());
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001233 if (E->hasExplicitTemplateArgs()) {
John McCallb3774b52010-08-19 23:49:38 +00001234 const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
Argyrios Kyrtzidisb5288de2010-06-28 09:31:48 +00001235 Record.push_back(Args.NumTemplateArgs);
1236 AddExplicitTemplateArgumentList(Args);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001237 }
1238
1239 if (!E->isImplicitAccess())
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001240 Writer.AddStmt(E->getBase());
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001241 else
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001242 Writer.AddStmt(0);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001243 Writer.AddTypeRef(E->getBaseType(), Record);
1244 Record.push_back(E->isArrow());
1245 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Douglas Gregore16af532011-02-28 18:50:33 +00001246 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001247 Writer.AddDeclRef(E->getFirstQualifierFoundInScope(), Record);
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +00001248 Writer.AddDeclarationNameInfo(E->MemberNameInfo, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001249 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001250}
1251
1252void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001253ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
Argyrios Kyrtzidiscd444d1a2010-06-28 09:31:56 +00001254 VisitExpr(E);
1255
Douglas Gregor87866ce2011-02-04 12:01:24 +00001256 // Don't emit anything here, hasExplicitTemplateArgs() must be
1257 // emitted first.
1258 Record.push_back(E->hasExplicitTemplateArgs());
Argyrios Kyrtzidiscd444d1a2010-06-28 09:31:56 +00001259 if (E->hasExplicitTemplateArgs()) {
1260 const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
Argyrios Kyrtzidiscd444d1a2010-06-28 09:31:56 +00001261 Record.push_back(Args.NumTemplateArgs);
1262 AddExplicitTemplateArgumentList(Args);
Argyrios Kyrtzidiscd444d1a2010-06-28 09:31:56 +00001263 }
1264
Douglas Gregor3a43fd62011-02-25 20:49:16 +00001265 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +00001266 Writer.AddDeclarationNameInfo(E->NameInfo, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001267 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
Argyrios Kyrtzidiscd444d1a2010-06-28 09:31:56 +00001268}
1269
1270void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001271ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001272 VisitExpr(E);
1273 Record.push_back(E->arg_size());
1274 for (CXXUnresolvedConstructExpr::arg_iterator
1275 ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001276 Writer.AddStmt(*ArgI);
Douglas Gregor2b88c112010-09-08 00:15:04 +00001277 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001278 Writer.AddSourceLocation(E->getLParenLoc(), Record);
1279 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001280 Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +00001281}
Chris Lattner98267332010-05-09 06:15:05 +00001282
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001283void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001284 VisitExpr(E);
Argyrios Kyrtzidisb5288de2010-06-28 09:31:48 +00001285
Douglas Gregor87866ce2011-02-04 12:01:24 +00001286 // Don't emit anything here, hasExplicitTemplateArgs() must be emitted first.
1287 Record.push_back(E->hasExplicitTemplateArgs());
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001288 if (E->hasExplicitTemplateArgs()) {
1289 const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001290 Record.push_back(Args.NumTemplateArgs);
Argyrios Kyrtzidisb5288de2010-06-28 09:31:48 +00001291 AddExplicitTemplateArgumentList(Args);
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001292 }
1293
1294 Record.push_back(E->getNumDecls());
1295 for (OverloadExpr::decls_iterator
1296 OvI = E->decls_begin(), OvE = E->decls_end(); OvI != OvE; ++OvI) {
1297 Writer.AddDeclRef(OvI.getDecl(), Record);
1298 Record.push_back(OvI.getAccess());
1299 }
1300
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +00001301 Writer.AddDeclarationNameInfo(E->NameInfo, Record);
Douglas Gregor0da1d432011-02-28 20:01:57 +00001302 Writer.AddNestedNameSpecifierLoc(E->getQualifierLoc(), Record);
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001303}
1304
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001305void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001306 VisitOverloadExpr(E);
1307 Record.push_back(E->isArrow());
1308 Record.push_back(E->hasUnresolvedUsing());
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001309 Writer.AddStmt(!E->isImplicitAccess() ? E->getBase() : 0);
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001310 Writer.AddTypeRef(E->getBaseType(), Record);
1311 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001312 Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
Argyrios Kyrtzidisb8d3c632010-06-25 09:03:26 +00001313}
1314
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001315void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
Argyrios Kyrtzidis58e01ad2010-06-25 09:03:34 +00001316 VisitOverloadExpr(E);
1317 Record.push_back(E->requiresADL());
Richard Smith02e85f32011-04-14 22:09:26 +00001318 if (E->requiresADL())
1319 Record.push_back(E->isStdAssociatedNamespace());
Argyrios Kyrtzidis58e01ad2010-06-25 09:03:34 +00001320 Record.push_back(E->isOverloaded());
1321 Writer.AddDeclRef(E->getNamingClass(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001322 Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
Argyrios Kyrtzidis58e01ad2010-06-25 09:03:34 +00001323}
1324
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001325void ASTStmtWriter::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +00001326 VisitExpr(E);
1327 Record.push_back(E->getTrait());
Sebastian Redl8eb06f12010-09-13 20:56:31 +00001328 Record.push_back(E->getValue());
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +00001329 Writer.AddSourceRange(E->getSourceRange(), Record);
Douglas Gregor54e5b132010-09-09 16:14:44 +00001330 Writer.AddTypeSourceInfo(E->getQueriedTypeSourceInfo(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001331 Code = serialization::EXPR_CXX_UNARY_TYPE_TRAIT;
Argyrios Kyrtzidisb8d77eb2010-07-10 11:46:15 +00001332}
1333
Francois Pichet9dfa3ce2010-12-07 00:08:36 +00001334void ASTStmtWriter::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
1335 VisitExpr(E);
1336 Record.push_back(E->getTrait());
1337 Record.push_back(E->getValue());
1338 Writer.AddSourceRange(E->getSourceRange(), Record);
1339 Writer.AddTypeSourceInfo(E->getLhsTypeSourceInfo(), Record);
1340 Writer.AddTypeSourceInfo(E->getRhsTypeSourceInfo(), Record);
Francois Pichet34b21132010-12-08 22:35:30 +00001341 Code = serialization::EXPR_BINARY_TYPE_TRAIT;
Francois Pichet9dfa3ce2010-12-07 00:08:36 +00001342}
1343
John Wiegley6242b6a2011-04-28 00:16:57 +00001344void ASTStmtWriter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
1345 VisitExpr(E);
1346 Record.push_back(E->getTrait());
1347 Record.push_back(E->getValue());
1348 Writer.AddSourceRange(E->getSourceRange(), Record);
1349 Writer.AddTypeSourceInfo(E->getQueriedTypeSourceInfo(), Record);
1350 Code = serialization::EXPR_ARRAY_TYPE_TRAIT;
1351}
1352
John Wiegleyf9f65842011-04-25 06:54:41 +00001353void ASTStmtWriter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
1354 VisitExpr(E);
1355 Record.push_back(E->getTrait());
1356 Record.push_back(E->getValue());
1357 Writer.AddSourceRange(E->getSourceRange(), Record);
1358 Writer.AddStmt(E->getQueriedExpression());
1359 Code = serialization::EXPR_CXX_EXPRESSION_TRAIT;
1360}
1361
Sebastian Redl9ac55dd2010-09-10 20:55:54 +00001362void ASTStmtWriter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
1363 VisitExpr(E);
1364 Record.push_back(E->getValue());
1365 Writer.AddSourceRange(E->getSourceRange(), Record);
1366 Writer.AddStmt(E->getOperand());
1367 Code = serialization::EXPR_CXX_NOEXCEPT;
1368}
1369
Douglas Gregore8e9dd62011-01-03 17:17:50 +00001370void ASTStmtWriter::VisitPackExpansionExpr(PackExpansionExpr *E) {
1371 VisitExpr(E);
1372 Writer.AddSourceLocation(E->getEllipsisLoc(), Record);
Douglas Gregorb8840002011-01-14 21:20:45 +00001373 Record.push_back(E->NumExpansions);
Douglas Gregore8e9dd62011-01-03 17:17:50 +00001374 Writer.AddStmt(E->getPattern());
1375 Code = serialization::EXPR_PACK_EXPANSION;
1376}
1377
Douglas Gregor820ba7b2011-01-04 17:33:58 +00001378void ASTStmtWriter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
1379 VisitExpr(E);
1380 Writer.AddSourceLocation(E->OperatorLoc, Record);
1381 Writer.AddSourceLocation(E->PackLoc, Record);
1382 Writer.AddSourceLocation(E->RParenLoc, Record);
1383 Record.push_back(E->Length);
1384 Writer.AddDeclRef(E->Pack, Record);
1385 Code = serialization::EXPR_SIZEOF_PACK;
1386}
1387
Douglas Gregorcdbc5392011-01-15 01:15:58 +00001388void ASTStmtWriter::VisitSubstNonTypeTemplateParmPackExpr(
1389 SubstNonTypeTemplateParmPackExpr *E) {
1390 VisitExpr(E);
1391 Writer.AddDeclRef(E->Param, Record);
1392 Writer.AddTemplateArgument(E->getArgumentPack(), Record);
1393 Writer.AddSourceLocation(E->NameLoc, Record);
1394 Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM_PACK;
1395}
1396
John McCall8d69a212010-11-15 23:31:06 +00001397void ASTStmtWriter::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
1398 VisitExpr(E);
John McCallc07a0c72011-02-17 10:25:35 +00001399 Record.push_back(Writer.getOpaqueValueID(E));
Douglas Gregorc03a1082011-01-28 02:26:04 +00001400 Writer.AddSourceLocation(E->getLocation(), Record);
John McCall8d69a212010-11-15 23:31:06 +00001401 Code = serialization::EXPR_OPAQUE_VALUE;
1402}
1403
Argyrios Kyrtzidiseeaaead2009-07-14 03:19:21 +00001404//===----------------------------------------------------------------------===//
Peter Collingbourne41f85462011-02-09 21:07:24 +00001405// CUDA Expressions and Statements.
1406//===----------------------------------------------------------------------===//
1407
1408void ASTStmtWriter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *E) {
1409 VisitCallExpr(E);
1410 Writer.AddStmt(E->getConfig());
1411 Code = serialization::EXPR_CUDA_KERNEL_CALL;
1412}
1413
1414//===----------------------------------------------------------------------===//
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001415// ASTWriter Implementation
Chris Lattner1f551822009-04-27 06:20:01 +00001416//===----------------------------------------------------------------------===//
1417
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001418unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
Mike Stump11289f42009-09-09 15:08:12 +00001419 assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
Chris Lattner1f551822009-04-27 06:20:01 +00001420 "SwitchCase recorded twice");
1421 unsigned NextID = SwitchCaseIDs.size();
1422 SwitchCaseIDs[S] = NextID;
1423 return NextID;
1424}
1425
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001426unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
Mike Stump11289f42009-09-09 15:08:12 +00001427 assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
Chris Lattner1f551822009-04-27 06:20:01 +00001428 "SwitchCase hasn't been seen yet");
1429 return SwitchCaseIDs[S];
1430}
1431
Argyrios Kyrtzidisd9f526f2010-10-28 09:29:32 +00001432void ASTWriter::ClearSwitchCaseIDs() {
1433 SwitchCaseIDs.clear();
1434}
1435
John McCallc07a0c72011-02-17 10:25:35 +00001436unsigned ASTWriter::getOpaqueValueID(OpaqueValueExpr *e) {
1437 unsigned &entry = OpaqueValues[e];
1438 if (!entry) entry = OpaqueValues.size();
1439 return entry;
1440}
1441
Chris Lattner1f551822009-04-27 06:20:01 +00001442/// \brief Write the given substatement or subexpression to the
1443/// bitstream.
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001444void ASTWriter::WriteSubStmt(Stmt *S) {
Chris Lattner1f551822009-04-27 06:20:01 +00001445 RecordData Record;
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001446 ASTStmtWriter Writer(*this, Record);
Chris Lattner1f551822009-04-27 06:20:01 +00001447 ++NumStatements;
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001448
Chris Lattner1f551822009-04-27 06:20:01 +00001449 if (!S) {
Sebastian Redl539c5062010-08-18 23:57:32 +00001450 Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
Chris Lattner1f551822009-04-27 06:20:01 +00001451 return;
1452 }
Mike Stump11289f42009-09-09 15:08:12 +00001453
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001454 // Redirect ASTWriter::AddStmt to collect sub stmts.
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001455 llvm::SmallVector<Stmt *, 16> SubStmts;
1456 CollectedStmts = &SubStmts;
1457
Sebastian Redl539c5062010-08-18 23:57:32 +00001458 Writer.Code = serialization::STMT_NULL_PTR;
Chris Lattner1f551822009-04-27 06:20:01 +00001459 Writer.Visit(S);
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001460
1461#ifndef NDEBUG
Sebastian Redl539c5062010-08-18 23:57:32 +00001462 if (Writer.Code == serialization::STMT_NULL_PTR) {
Argyrios Kyrtzidis6ba1bb52010-06-25 09:03:12 +00001463 SourceManager &SrcMgr
1464 = DeclIDs.begin()->first->getASTContext().getSourceManager();
1465 S->dump(SrcMgr);
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001466 assert(0 && "Unhandled sub statement writing AST file");
Chris Lattnerabfb58d2010-05-10 01:22:27 +00001467 }
1468#endif
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001469
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001470 // Revert ASTWriter::AddStmt.
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001471 CollectedStmts = &StmtsToEmit;
1472
1473 // Write the sub stmts in reverse order, last to first. When reading them back
1474 // we will read them in correct order by "pop"ing them from the Stmts stack.
1475 // This simplifies reading and allows to store a variable number of sub stmts
1476 // without knowing it in advance.
1477 while (!SubStmts.empty())
1478 WriteSubStmt(SubStmts.pop_back_val());
1479
Mike Stump11289f42009-09-09 15:08:12 +00001480 Stream.EmitRecord(Writer.Code, Record);
Chris Lattner1f551822009-04-27 06:20:01 +00001481}
1482
1483/// \brief Flush all of the statements that have been added to the
1484/// queue via AddStmt().
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001485void ASTWriter::FlushStmts() {
Chris Lattner1f551822009-04-27 06:20:01 +00001486 RecordData Record;
Mike Stump11289f42009-09-09 15:08:12 +00001487
Chris Lattner1f551822009-04-27 06:20:01 +00001488 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
Argyrios Kyrtzidisd0795b22010-06-28 22:28:35 +00001489 WriteSubStmt(StmtsToEmit[I]);
1490
Mike Stump11289f42009-09-09 15:08:12 +00001491 assert(N == StmtsToEmit.size() &&
Chris Lattner57540c52011-04-15 05:22:18 +00001492 "Substatement written via AddStmt rather than WriteSubStmt!");
Mike Stump11289f42009-09-09 15:08:12 +00001493
Chris Lattner1f551822009-04-27 06:20:01 +00001494 // Note that we are at the end of a full expression. Any
1495 // expression records that follow this one are part of a different
1496 // expression.
Sebastian Redl539c5062010-08-18 23:57:32 +00001497 Stream.EmitRecord(serialization::STMT_STOP, Record);
Chris Lattner1f551822009-04-27 06:20:01 +00001498 }
Mike Stump11289f42009-09-09 15:08:12 +00001499
Chris Lattner1f551822009-04-27 06:20:01 +00001500 StmtsToEmit.clear();
Chris Lattner1f551822009-04-27 06:20:01 +00001501}