blob: 403311b8c1af8ecf41abe533059bfc9b8a11b493 [file] [log] [blame]
Sebastian Redl4ee2ad02010-08-18 23:56:31 +00001//===--- ASTWriterStmt.cpp - Statement and Expression Serialization -------===//
Chris Lattner63377d52009-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 Redl7faa2ec2010-08-18 23:56:37 +000014#include "clang/Serialization/ASTWriter.h"
Douglas Gregor39da0b82009-09-09 23:08:42 +000015#include "clang/AST/DeclCXX.h"
Chris Lattner63377d52009-04-27 06:20:01 +000016#include "clang/AST/DeclObjC.h"
17#include "clang/AST/StmtVisitor.h"
18#include "llvm/Bitcode/BitstreamWriter.h"
19using namespace clang;
20
21//===----------------------------------------------------------------------===//
22// Statement/expression serialization
23//===----------------------------------------------------------------------===//
24
Argyrios Kyrtzidis37bdfe22010-06-30 08:49:18 +000025namespace clang {
Sebastian Redl3397c552010-08-18 23:56:27 +000026 class ASTStmtWriter : public StmtVisitor<ASTStmtWriter, void> {
Sebastian Redla4232eb2010-08-18 23:56:21 +000027 ASTWriter &Writer;
28 ASTWriter::RecordData &Record;
Chris Lattner63377d52009-04-27 06:20:01 +000029
30 public:
Sebastian Redl8538e8d2010-08-18 23:57:32 +000031 serialization::StmtCode Code;
Chris Lattner63377d52009-04-27 06:20:01 +000032
Sebastian Redl3397c552010-08-18 23:56:27 +000033 ASTStmtWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
Chris Lattner63377d52009-04-27 06:20:01 +000034 : Writer(Writer), Record(Record) { }
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +000035
36 void
37 AddExplicitTemplateArgumentList(const ExplicitTemplateArgumentList &Args);
Chris Lattner63377d52009-04-27 06:20:01 +000038
39 void VisitStmt(Stmt *S);
40 void VisitNullStmt(NullStmt *S);
41 void VisitCompoundStmt(CompoundStmt *S);
42 void VisitSwitchCase(SwitchCase *S);
43 void VisitCaseStmt(CaseStmt *S);
44 void VisitDefaultStmt(DefaultStmt *S);
45 void VisitLabelStmt(LabelStmt *S);
46 void VisitIfStmt(IfStmt *S);
47 void VisitSwitchStmt(SwitchStmt *S);
48 void VisitWhileStmt(WhileStmt *S);
49 void VisitDoStmt(DoStmt *S);
50 void VisitForStmt(ForStmt *S);
51 void VisitGotoStmt(GotoStmt *S);
52 void VisitIndirectGotoStmt(IndirectGotoStmt *S);
53 void VisitContinueStmt(ContinueStmt *S);
54 void VisitBreakStmt(BreakStmt *S);
55 void VisitReturnStmt(ReturnStmt *S);
56 void VisitDeclStmt(DeclStmt *S);
57 void VisitAsmStmt(AsmStmt *S);
58 void VisitExpr(Expr *E);
59 void VisitPredefinedExpr(PredefinedExpr *E);
60 void VisitDeclRefExpr(DeclRefExpr *E);
61 void VisitIntegerLiteral(IntegerLiteral *E);
62 void VisitFloatingLiteral(FloatingLiteral *E);
63 void VisitImaginaryLiteral(ImaginaryLiteral *E);
64 void VisitStringLiteral(StringLiteral *E);
65 void VisitCharacterLiteral(CharacterLiteral *E);
66 void VisitParenExpr(ParenExpr *E);
Argyrios Kyrtzidis37bdfe22010-06-30 08:49:18 +000067 void VisitParenListExpr(ParenListExpr *E);
Chris Lattner63377d52009-04-27 06:20:01 +000068 void VisitUnaryOperator(UnaryOperator *E);
Douglas Gregor8ecdb652010-04-28 22:16:22 +000069 void VisitOffsetOfExpr(OffsetOfExpr *E);
Chris Lattner63377d52009-04-27 06:20:01 +000070 void VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E);
71 void VisitArraySubscriptExpr(ArraySubscriptExpr *E);
72 void VisitCallExpr(CallExpr *E);
73 void VisitMemberExpr(MemberExpr *E);
74 void VisitCastExpr(CastExpr *E);
75 void VisitBinaryOperator(BinaryOperator *E);
76 void VisitCompoundAssignOperator(CompoundAssignOperator *E);
77 void VisitConditionalOperator(ConditionalOperator *E);
78 void VisitImplicitCastExpr(ImplicitCastExpr *E);
79 void VisitExplicitCastExpr(ExplicitCastExpr *E);
80 void VisitCStyleCastExpr(CStyleCastExpr *E);
81 void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
82 void VisitExtVectorElementExpr(ExtVectorElementExpr *E);
83 void VisitInitListExpr(InitListExpr *E);
84 void VisitDesignatedInitExpr(DesignatedInitExpr *E);
85 void VisitImplicitValueInitExpr(ImplicitValueInitExpr *E);
86 void VisitVAArgExpr(VAArgExpr *E);
87 void VisitAddrLabelExpr(AddrLabelExpr *E);
88 void VisitStmtExpr(StmtExpr *E);
89 void VisitTypesCompatibleExpr(TypesCompatibleExpr *E);
90 void VisitChooseExpr(ChooseExpr *E);
91 void VisitGNUNullExpr(GNUNullExpr *E);
92 void VisitShuffleVectorExpr(ShuffleVectorExpr *E);
93 void VisitBlockExpr(BlockExpr *E);
94 void VisitBlockDeclRefExpr(BlockDeclRefExpr *E);
Mike Stump1eb44332009-09-09 15:08:12 +000095
Chris Lattner63377d52009-04-27 06:20:01 +000096 // Objective-C Expressions
97 void VisitObjCStringLiteral(ObjCStringLiteral *E);
98 void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
99 void VisitObjCSelectorExpr(ObjCSelectorExpr *E);
100 void VisitObjCProtocolExpr(ObjCProtocolExpr *E);
101 void VisitObjCIvarRefExpr(ObjCIvarRefExpr *E);
102 void VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E);
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000103 void VisitObjCImplicitSetterGetterRefExpr(
104 ObjCImplicitSetterGetterRefExpr *E);
Chris Lattner63377d52009-04-27 06:20:01 +0000105 void VisitObjCMessageExpr(ObjCMessageExpr *E);
106 void VisitObjCSuperExpr(ObjCSuperExpr *E);
Steve Narofff242b1b2009-07-24 17:54:45 +0000107 void VisitObjCIsaExpr(ObjCIsaExpr *E);
Mike Stump1eb44332009-09-09 15:08:12 +0000108
109 // Objective-C Statements
Chris Lattner63377d52009-04-27 06:20:01 +0000110 void VisitObjCForCollectionStmt(ObjCForCollectionStmt *);
111 void VisitObjCAtCatchStmt(ObjCAtCatchStmt *);
112 void VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *);
113 void VisitObjCAtTryStmt(ObjCAtTryStmt *);
114 void VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *);
115 void VisitObjCAtThrowStmt(ObjCAtThrowStmt *);
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000116
Mike Stump1eb44332009-09-09 15:08:12 +0000117 // C++ Statements
Argyrios Kyrtzidis7cb45e32010-07-22 16:03:56 +0000118 void VisitCXXCatchStmt(CXXCatchStmt *S);
119 void VisitCXXTryStmt(CXXTryStmt *S);
120
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000121 void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
Chris Lattner1817bd42010-05-09 05:36:05 +0000122 void VisitCXXMemberCallExpr(CXXMemberCallExpr *E);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000123 void VisitCXXConstructExpr(CXXConstructExpr *E);
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +0000124 void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
Sam Weinigce757a72010-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 Weinigeb7f9612010-02-07 06:32:43 +0000131 void VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E);
132 void VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E);
Chris Lattner14ab24f2010-05-09 06:03:39 +0000133 void VisitCXXTypeidExpr(CXXTypeidExpr *E);
Francois Pichet01b7c302010-09-08 12:20:18 +0000134 void VisitCXXUuidofExpr(CXXUuidofExpr *E);
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000135 void VisitCXXThisExpr(CXXThisExpr *E);
136 void VisitCXXThrowExpr(CXXThrowExpr *E);
Chris Lattner030854b2010-05-09 06:40:08 +0000137 void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E);
Chris Lattnerd2598362010-05-10 00:25:06 +0000138 void VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +0000139
Douglas Gregored8abf12010-07-08 06:14:04 +0000140 void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
Chris Lattner59218632010-05-10 01:22:27 +0000141 void VisitCXXNewExpr(CXXNewExpr *E);
Argyrios Kyrtzidis95fc98c2010-06-22 17:07:59 +0000142 void VisitCXXDeleteExpr(CXXDeleteExpr *E);
Argyrios Kyrtzidisde4bd182010-06-28 09:32:03 +0000143 void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +0000144
Chris Lattnerd2598362010-05-10 00:25:06 +0000145 void VisitCXXExprWithTemporaries(CXXExprWithTemporaries *E);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +0000146 void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
Argyrios Kyrtzidis12dffcd2010-06-28 09:31:56 +0000147 void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +0000148 void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +0000149
150 void VisitOverloadExpr(OverloadExpr *E);
151 void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E);
Argyrios Kyrtzidisbd65bb52010-06-25 09:03:34 +0000152 void VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E);
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +0000153
154 void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
Chris Lattner63377d52009-04-27 06:20:01 +0000155 };
156}
157
Sebastian Redl3397c552010-08-18 23:56:27 +0000158void ASTStmtWriter::
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +0000159AddExplicitTemplateArgumentList(const ExplicitTemplateArgumentList &Args) {
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +0000160 Writer.AddSourceLocation(Args.LAngleLoc, Record);
161 Writer.AddSourceLocation(Args.RAngleLoc, Record);
162 for (unsigned i=0; i != Args.NumTemplateArgs; ++i)
163 Writer.AddTemplateArgumentLoc(Args.getTemplateArgs()[i], Record);
164}
165
Sebastian Redl3397c552010-08-18 23:56:27 +0000166void ASTStmtWriter::VisitStmt(Stmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000167}
168
Sebastian Redl3397c552010-08-18 23:56:27 +0000169void ASTStmtWriter::VisitNullStmt(NullStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000170 VisitStmt(S);
171 Writer.AddSourceLocation(S->getSemiLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000172 Code = serialization::STMT_NULL;
Chris Lattner63377d52009-04-27 06:20:01 +0000173}
174
Sebastian Redl3397c552010-08-18 23:56:27 +0000175void ASTStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000176 VisitStmt(S);
177 Record.push_back(S->size());
178 for (CompoundStmt::body_iterator CS = S->body_begin(), CSEnd = S->body_end();
179 CS != CSEnd; ++CS)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000180 Writer.AddStmt(*CS);
Chris Lattner63377d52009-04-27 06:20:01 +0000181 Writer.AddSourceLocation(S->getLBracLoc(), Record);
182 Writer.AddSourceLocation(S->getRBracLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000183 Code = serialization::STMT_COMPOUND;
Chris Lattner63377d52009-04-27 06:20:01 +0000184}
185
Sebastian Redl3397c552010-08-18 23:56:27 +0000186void ASTStmtWriter::VisitSwitchCase(SwitchCase *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000187 VisitStmt(S);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000188 Record.push_back(Writer.getSwitchCaseID(S));
Chris Lattner63377d52009-04-27 06:20:01 +0000189}
190
Sebastian Redl3397c552010-08-18 23:56:27 +0000191void ASTStmtWriter::VisitCaseStmt(CaseStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000192 VisitSwitchCase(S);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000193 Writer.AddStmt(S->getLHS());
194 Writer.AddStmt(S->getRHS());
195 Writer.AddStmt(S->getSubStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000196 Writer.AddSourceLocation(S->getCaseLoc(), Record);
Douglas Gregordbb26db2009-05-15 23:57:33 +0000197 Writer.AddSourceLocation(S->getEllipsisLoc(), Record);
198 Writer.AddSourceLocation(S->getColonLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000199 Code = serialization::STMT_CASE;
Chris Lattner63377d52009-04-27 06:20:01 +0000200}
201
Sebastian Redl3397c552010-08-18 23:56:27 +0000202void ASTStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000203 VisitSwitchCase(S);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000204 Writer.AddStmt(S->getSubStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000205 Writer.AddSourceLocation(S->getDefaultLoc(), Record);
Douglas Gregordbb26db2009-05-15 23:57:33 +0000206 Writer.AddSourceLocation(S->getColonLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000207 Code = serialization::STMT_DEFAULT;
Chris Lattner63377d52009-04-27 06:20:01 +0000208}
209
Sebastian Redl3397c552010-08-18 23:56:27 +0000210void ASTStmtWriter::VisitLabelStmt(LabelStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000211 VisitStmt(S);
212 Writer.AddIdentifierRef(S->getID(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000213 Writer.AddStmt(S->getSubStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000214 Writer.AddSourceLocation(S->getIdentLoc(), Record);
215 Record.push_back(Writer.GetLabelID(S));
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000216 Code = serialization::STMT_LABEL;
Chris Lattner63377d52009-04-27 06:20:01 +0000217}
218
Sebastian Redl3397c552010-08-18 23:56:27 +0000219void ASTStmtWriter::VisitIfStmt(IfStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000220 VisitStmt(S);
Douglas Gregor8cfe5a72009-11-23 23:44:04 +0000221 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000222 Writer.AddStmt(S->getCond());
223 Writer.AddStmt(S->getThen());
224 Writer.AddStmt(S->getElse());
Chris Lattner63377d52009-04-27 06:20:01 +0000225 Writer.AddSourceLocation(S->getIfLoc(), Record);
Douglas Gregord06f6ca2009-05-15 18:53:42 +0000226 Writer.AddSourceLocation(S->getElseLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000227 Code = serialization::STMT_IF;
Chris Lattner63377d52009-04-27 06:20:01 +0000228}
229
Sebastian Redl3397c552010-08-18 23:56:27 +0000230void ASTStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000231 VisitStmt(S);
Douglas Gregord3d53012009-11-24 17:07:59 +0000232 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000233 Writer.AddStmt(S->getCond());
234 Writer.AddStmt(S->getBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000235 Writer.AddSourceLocation(S->getSwitchLoc(), Record);
Mike Stump1eb44332009-09-09 15:08:12 +0000236 for (SwitchCase *SC = S->getSwitchCaseList(); SC;
Chris Lattner63377d52009-04-27 06:20:01 +0000237 SC = SC->getNextSwitchCase())
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000238 Record.push_back(Writer.RecordSwitchCaseID(SC));
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000239 Code = serialization::STMT_SWITCH;
Chris Lattner63377d52009-04-27 06:20:01 +0000240}
241
Sebastian Redl3397c552010-08-18 23:56:27 +0000242void ASTStmtWriter::VisitWhileStmt(WhileStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000243 VisitStmt(S);
Douglas Gregor5656e142009-11-24 21:15:44 +0000244 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000245 Writer.AddStmt(S->getCond());
246 Writer.AddStmt(S->getBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000247 Writer.AddSourceLocation(S->getWhileLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000248 Code = serialization::STMT_WHILE;
Chris Lattner63377d52009-04-27 06:20:01 +0000249}
250
Sebastian Redl3397c552010-08-18 23:56:27 +0000251void ASTStmtWriter::VisitDoStmt(DoStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000252 VisitStmt(S);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000253 Writer.AddStmt(S->getCond());
254 Writer.AddStmt(S->getBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000255 Writer.AddSourceLocation(S->getDoLoc(), Record);
Douglas Gregor9f3ca2a2009-05-15 21:56:04 +0000256 Writer.AddSourceLocation(S->getWhileLoc(), Record);
Chris Lattner98913592009-06-12 23:04:47 +0000257 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000258 Code = serialization::STMT_DO;
Chris Lattner63377d52009-04-27 06:20:01 +0000259}
260
Sebastian Redl3397c552010-08-18 23:56:27 +0000261void ASTStmtWriter::VisitForStmt(ForStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000262 VisitStmt(S);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000263 Writer.AddStmt(S->getInit());
264 Writer.AddStmt(S->getCond());
Douglas Gregor99e9b4d2009-11-25 00:27:52 +0000265 Writer.AddDeclRef(S->getConditionVariable(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000266 Writer.AddStmt(S->getInc());
267 Writer.AddStmt(S->getBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000268 Writer.AddSourceLocation(S->getForLoc(), Record);
Douglas Gregor5831c6a2009-05-15 22:12:32 +0000269 Writer.AddSourceLocation(S->getLParenLoc(), Record);
270 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000271 Code = serialization::STMT_FOR;
Chris Lattner63377d52009-04-27 06:20:01 +0000272}
273
Sebastian Redl3397c552010-08-18 23:56:27 +0000274void ASTStmtWriter::VisitGotoStmt(GotoStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000275 VisitStmt(S);
276 Record.push_back(Writer.GetLabelID(S->getLabel()));
277 Writer.AddSourceLocation(S->getGotoLoc(), Record);
278 Writer.AddSourceLocation(S->getLabelLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000279 Code = serialization::STMT_GOTO;
Chris Lattner63377d52009-04-27 06:20:01 +0000280}
281
Sebastian Redl3397c552010-08-18 23:56:27 +0000282void ASTStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000283 VisitStmt(S);
284 Writer.AddSourceLocation(S->getGotoLoc(), Record);
Douglas Gregor5f1b9e62009-05-16 00:20:29 +0000285 Writer.AddSourceLocation(S->getStarLoc(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000286 Writer.AddStmt(S->getTarget());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000287 Code = serialization::STMT_INDIRECT_GOTO;
Chris Lattner63377d52009-04-27 06:20:01 +0000288}
289
Sebastian Redl3397c552010-08-18 23:56:27 +0000290void ASTStmtWriter::VisitContinueStmt(ContinueStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000291 VisitStmt(S);
292 Writer.AddSourceLocation(S->getContinueLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000293 Code = serialization::STMT_CONTINUE;
Chris Lattner63377d52009-04-27 06:20:01 +0000294}
295
Sebastian Redl3397c552010-08-18 23:56:27 +0000296void ASTStmtWriter::VisitBreakStmt(BreakStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000297 VisitStmt(S);
298 Writer.AddSourceLocation(S->getBreakLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000299 Code = serialization::STMT_BREAK;
Chris Lattner63377d52009-04-27 06:20:01 +0000300}
301
Sebastian Redl3397c552010-08-18 23:56:27 +0000302void ASTStmtWriter::VisitReturnStmt(ReturnStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000303 VisitStmt(S);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000304 Writer.AddStmt(S->getRetValue());
Chris Lattner63377d52009-04-27 06:20:01 +0000305 Writer.AddSourceLocation(S->getReturnLoc(), Record);
Douglas Gregor5077c382010-05-15 06:01:05 +0000306 Writer.AddDeclRef(S->getNRVOCandidate(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000307 Code = serialization::STMT_RETURN;
Chris Lattner63377d52009-04-27 06:20:01 +0000308}
309
Sebastian Redl3397c552010-08-18 23:56:27 +0000310void ASTStmtWriter::VisitDeclStmt(DeclStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000311 VisitStmt(S);
312 Writer.AddSourceLocation(S->getStartLoc(), Record);
313 Writer.AddSourceLocation(S->getEndLoc(), Record);
314 DeclGroupRef DG = S->getDeclGroup();
315 for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
316 Writer.AddDeclRef(*D, Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000317 Code = serialization::STMT_DECL;
Chris Lattner63377d52009-04-27 06:20:01 +0000318}
319
Sebastian Redl3397c552010-08-18 23:56:27 +0000320void ASTStmtWriter::VisitAsmStmt(AsmStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000321 VisitStmt(S);
322 Record.push_back(S->getNumOutputs());
323 Record.push_back(S->getNumInputs());
324 Record.push_back(S->getNumClobbers());
325 Writer.AddSourceLocation(S->getAsmLoc(), Record);
326 Writer.AddSourceLocation(S->getRParenLoc(), Record);
327 Record.push_back(S->isVolatile());
328 Record.push_back(S->isSimple());
Mike Stump3b11fd32010-01-04 22:37:17 +0000329 Record.push_back(S->isMSAsm());
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000330 Writer.AddStmt(S->getAsmString());
Chris Lattner63377d52009-04-27 06:20:01 +0000331
332 // Outputs
Anders Carlssonff93dbd2010-01-30 22:25:16 +0000333 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
334 Writer.AddIdentifierRef(S->getOutputIdentifier(I), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000335 Writer.AddStmt(S->getOutputConstraintLiteral(I));
336 Writer.AddStmt(S->getOutputExpr(I));
Chris Lattner63377d52009-04-27 06:20:01 +0000337 }
338
339 // Inputs
340 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
Anders Carlssonff93dbd2010-01-30 22:25:16 +0000341 Writer.AddIdentifierRef(S->getInputIdentifier(I), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000342 Writer.AddStmt(S->getInputConstraintLiteral(I));
343 Writer.AddStmt(S->getInputExpr(I));
Chris Lattner63377d52009-04-27 06:20:01 +0000344 }
345
346 // Clobbers
347 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000348 Writer.AddStmt(S->getClobber(I));
Chris Lattner63377d52009-04-27 06:20:01 +0000349
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000350 Code = serialization::STMT_ASM;
Chris Lattner63377d52009-04-27 06:20:01 +0000351}
352
Sebastian Redl3397c552010-08-18 23:56:27 +0000353void ASTStmtWriter::VisitExpr(Expr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000354 VisitStmt(E);
355 Writer.AddTypeRef(E->getType(), Record);
356 Record.push_back(E->isTypeDependent());
357 Record.push_back(E->isValueDependent());
358}
359
Sebastian Redl3397c552010-08-18 23:56:27 +0000360void ASTStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000361 VisitExpr(E);
362 Writer.AddSourceLocation(E->getLocation(), Record);
363 Record.push_back(E->getIdentType()); // FIXME: stable encoding
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000364 Code = serialization::EXPR_PREDEFINED;
Chris Lattner63377d52009-04-27 06:20:01 +0000365}
366
Sebastian Redl3397c552010-08-18 23:56:27 +0000367void ASTStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000368 VisitExpr(E);
Argyrios Kyrtzidis663e3802010-07-08 13:09:47 +0000369
370 Record.push_back(E->hasQualifier());
371 unsigned NumTemplateArgs = E->getNumTemplateArgs();
John McCall096832c2010-08-19 23:49:38 +0000372 assert((NumTemplateArgs != 0) == E->hasExplicitTemplateArgs() &&
Argyrios Kyrtzidis663e3802010-07-08 13:09:47 +0000373 "Template args list with no args ?");
374 Record.push_back(NumTemplateArgs);
375
376 if (E->hasQualifier()) {
377 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
378 Writer.AddSourceRange(E->getQualifierRange(), Record);
379 }
380
381 if (NumTemplateArgs)
John McCall096832c2010-08-19 23:49:38 +0000382 AddExplicitTemplateArgumentList(E->getExplicitTemplateArgs());
Argyrios Kyrtzidis663e3802010-07-08 13:09:47 +0000383
Chris Lattner63377d52009-04-27 06:20:01 +0000384 Writer.AddDeclRef(E->getDecl(), Record);
Abramo Bagnara25777432010-08-11 22:01:17 +0000385 // FIXME: write DeclarationNameLoc.
Chris Lattner63377d52009-04-27 06:20:01 +0000386 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000387 Code = serialization::EXPR_DECL_REF;
Chris Lattner63377d52009-04-27 06:20:01 +0000388}
389
Sebastian Redl3397c552010-08-18 23:56:27 +0000390void ASTStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000391 VisitExpr(E);
392 Writer.AddSourceLocation(E->getLocation(), Record);
393 Writer.AddAPInt(E->getValue(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000394 Code = serialization::EXPR_INTEGER_LITERAL;
Chris Lattner63377d52009-04-27 06:20:01 +0000395}
396
Sebastian Redl3397c552010-08-18 23:56:27 +0000397void ASTStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000398 VisitExpr(E);
399 Writer.AddAPFloat(E->getValue(), Record);
400 Record.push_back(E->isExact());
401 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000402 Code = serialization::EXPR_FLOATING_LITERAL;
Chris Lattner63377d52009-04-27 06:20:01 +0000403}
404
Sebastian Redl3397c552010-08-18 23:56:27 +0000405void ASTStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000406 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000407 Writer.AddStmt(E->getSubExpr());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000408 Code = serialization::EXPR_IMAGINARY_LITERAL;
Chris Lattner63377d52009-04-27 06:20:01 +0000409}
410
Sebastian Redl3397c552010-08-18 23:56:27 +0000411void ASTStmtWriter::VisitStringLiteral(StringLiteral *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000412 VisitExpr(E);
413 Record.push_back(E->getByteLength());
414 Record.push_back(E->getNumConcatenated());
415 Record.push_back(E->isWide());
416 // FIXME: String data should be stored as a blob at the end of the
417 // StringLiteral. However, we can't do so now because we have no
418 // provision for coping with abbreviations when we're jumping around
Sebastian Redl3397c552010-08-18 23:56:27 +0000419 // the AST file during deserialization.
Benjamin Kramer2f4eaef2010-08-17 12:54:38 +0000420 Record.append(E->getString().begin(), E->getString().end());
Chris Lattner63377d52009-04-27 06:20:01 +0000421 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
422 Writer.AddSourceLocation(E->getStrTokenLoc(I), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000423 Code = serialization::EXPR_STRING_LITERAL;
Chris Lattner63377d52009-04-27 06:20:01 +0000424}
425
Sebastian Redl3397c552010-08-18 23:56:27 +0000426void ASTStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000427 VisitExpr(E);
428 Record.push_back(E->getValue());
Chris Lattner018d8e02009-08-24 17:39:36 +0000429 Writer.AddSourceLocation(E->getLocation(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000430 Record.push_back(E->isWide());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000431 Code = serialization::EXPR_CHARACTER_LITERAL;
Chris Lattner63377d52009-04-27 06:20:01 +0000432}
433
Sebastian Redl3397c552010-08-18 23:56:27 +0000434void ASTStmtWriter::VisitParenExpr(ParenExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000435 VisitExpr(E);
436 Writer.AddSourceLocation(E->getLParen(), Record);
437 Writer.AddSourceLocation(E->getRParen(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000438 Writer.AddStmt(E->getSubExpr());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000439 Code = serialization::EXPR_PAREN;
Chris Lattner63377d52009-04-27 06:20:01 +0000440}
441
Sebastian Redl3397c552010-08-18 23:56:27 +0000442void ASTStmtWriter::VisitParenListExpr(ParenListExpr *E) {
Argyrios Kyrtzidis37bdfe22010-06-30 08:49:18 +0000443 VisitExpr(E);
444 Record.push_back(E->NumExprs);
445 for (unsigned i=0; i != E->NumExprs; ++i)
446 Writer.AddStmt(E->Exprs[i]);
447 Writer.AddSourceLocation(E->LParenLoc, Record);
448 Writer.AddSourceLocation(E->RParenLoc, Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000449 Code = serialization::EXPR_PAREN_LIST;
Argyrios Kyrtzidis37bdfe22010-06-30 08:49:18 +0000450}
451
Sebastian Redl3397c552010-08-18 23:56:27 +0000452void ASTStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000453 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000454 Writer.AddStmt(E->getSubExpr());
Chris Lattner63377d52009-04-27 06:20:01 +0000455 Record.push_back(E->getOpcode()); // FIXME: stable encoding
456 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000457 Code = serialization::EXPR_UNARY_OPERATOR;
Chris Lattner63377d52009-04-27 06:20:01 +0000458}
459
Sebastian Redl3397c552010-08-18 23:56:27 +0000460void ASTStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
Douglas Gregor8ecdb652010-04-28 22:16:22 +0000461 VisitExpr(E);
462 Record.push_back(E->getNumComponents());
463 Record.push_back(E->getNumExpressions());
464 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
465 Writer.AddSourceLocation(E->getRParenLoc(), Record);
466 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
467 for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
468 const OffsetOfExpr::OffsetOfNode &ON = E->getComponent(I);
469 Record.push_back(ON.getKind()); // FIXME: Stable encoding
470 Writer.AddSourceLocation(ON.getRange().getBegin(), Record);
471 Writer.AddSourceLocation(ON.getRange().getEnd(), Record);
472 switch (ON.getKind()) {
473 case OffsetOfExpr::OffsetOfNode::Array:
474 Record.push_back(ON.getArrayExprIndex());
475 break;
476
477 case OffsetOfExpr::OffsetOfNode::Field:
478 Writer.AddDeclRef(ON.getField(), Record);
479 break;
480
481 case OffsetOfExpr::OffsetOfNode::Identifier:
482 Writer.AddIdentifierRef(ON.getFieldName(), Record);
483 break;
Douglas Gregorcc8a5d52010-04-29 00:18:15 +0000484
485 case OffsetOfExpr::OffsetOfNode::Base:
Argyrios Kyrtzidisafbf3122010-07-29 18:16:10 +0000486 Writer.AddCXXBaseSpecifier(*ON.getBase(), Record);
Douglas Gregorcc8a5d52010-04-29 00:18:15 +0000487 break;
Douglas Gregor8ecdb652010-04-28 22:16:22 +0000488 }
489 }
490 for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000491 Writer.AddStmt(E->getIndexExpr(I));
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000492 Code = serialization::EXPR_OFFSETOF;
Douglas Gregor8ecdb652010-04-28 22:16:22 +0000493}
494
Sebastian Redl3397c552010-08-18 23:56:27 +0000495void ASTStmtWriter::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000496 VisitExpr(E);
497 Record.push_back(E->isSizeOf());
498 if (E->isArgumentType())
John McCalla93c9342009-12-07 02:54:59 +0000499 Writer.AddTypeSourceInfo(E->getArgumentTypeInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000500 else {
501 Record.push_back(0);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000502 Writer.AddStmt(E->getArgumentExpr());
Chris Lattner63377d52009-04-27 06:20:01 +0000503 }
504 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
505 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000506 Code = serialization::EXPR_SIZEOF_ALIGN_OF;
Chris Lattner63377d52009-04-27 06:20:01 +0000507}
508
Sebastian Redl3397c552010-08-18 23:56:27 +0000509void ASTStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000510 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000511 Writer.AddStmt(E->getLHS());
512 Writer.AddStmt(E->getRHS());
Chris Lattner63377d52009-04-27 06:20:01 +0000513 Writer.AddSourceLocation(E->getRBracketLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000514 Code = serialization::EXPR_ARRAY_SUBSCRIPT;
Chris Lattner63377d52009-04-27 06:20:01 +0000515}
516
Sebastian Redl3397c552010-08-18 23:56:27 +0000517void ASTStmtWriter::VisitCallExpr(CallExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000518 VisitExpr(E);
519 Record.push_back(E->getNumArgs());
520 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000521 Writer.AddStmt(E->getCallee());
Chris Lattner63377d52009-04-27 06:20:01 +0000522 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
523 Arg != ArgEnd; ++Arg)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000524 Writer.AddStmt(*Arg);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000525 Code = serialization::EXPR_CALL;
Chris Lattner63377d52009-04-27 06:20:01 +0000526}
527
Sebastian Redl3397c552010-08-18 23:56:27 +0000528void ASTStmtWriter::VisitMemberExpr(MemberExpr *E) {
Argyrios Kyrtzidis663e3802010-07-08 13:09:47 +0000529 // Don't call VisitExpr, we'll write everything here.
530
531 Record.push_back(E->hasQualifier());
532 if (E->hasQualifier()) {
533 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
534 Writer.AddSourceRange(E->getQualifierRange(), Record);
535 }
536
537 unsigned NumTemplateArgs = E->getNumTemplateArgs();
John McCall096832c2010-08-19 23:49:38 +0000538 assert((NumTemplateArgs != 0) == E->hasExplicitTemplateArgs() &&
Argyrios Kyrtzidis663e3802010-07-08 13:09:47 +0000539 "Template args list with no args ?");
540 Record.push_back(NumTemplateArgs);
541 if (NumTemplateArgs) {
542 Writer.AddSourceLocation(E->getLAngleLoc(), Record);
543 Writer.AddSourceLocation(E->getRAngleLoc(), Record);
544 for (unsigned i=0; i != NumTemplateArgs; ++i)
545 Writer.AddTemplateArgumentLoc(E->getTemplateArgs()[i], Record);
546 }
547
548 DeclAccessPair FoundDecl = E->getFoundDecl();
549 Writer.AddDeclRef(FoundDecl.getDecl(), Record);
550 Record.push_back(FoundDecl.getAccess());
551
552 Writer.AddTypeRef(E->getType(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000553 Writer.AddStmt(E->getBase());
Chris Lattner63377d52009-04-27 06:20:01 +0000554 Writer.AddDeclRef(E->getMemberDecl(), Record);
Abramo Bagnara25777432010-08-11 22:01:17 +0000555 // FIXME: write DeclarationNameLoc.
Chris Lattner63377d52009-04-27 06:20:01 +0000556 Writer.AddSourceLocation(E->getMemberLoc(), Record);
557 Record.push_back(E->isArrow());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000558 Code = serialization::EXPR_MEMBER;
Chris Lattner63377d52009-04-27 06:20:01 +0000559}
560
Sebastian Redl3397c552010-08-18 23:56:27 +0000561void ASTStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
Steve Narofff242b1b2009-07-24 17:54:45 +0000562 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000563 Writer.AddStmt(E->getBase());
Steve Narofff242b1b2009-07-24 17:54:45 +0000564 Writer.AddSourceLocation(E->getIsaMemberLoc(), Record);
565 Record.push_back(E->isArrow());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000566 Code = serialization::EXPR_OBJC_ISA;
Steve Narofff242b1b2009-07-24 17:54:45 +0000567}
568
Sebastian Redl3397c552010-08-18 23:56:27 +0000569void ASTStmtWriter::VisitCastExpr(CastExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000570 VisitExpr(E);
John McCallf871d0c2010-08-07 06:22:56 +0000571 Record.push_back(E->path_size());
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000572 Writer.AddStmt(E->getSubExpr());
Anders Carlssoncdef2b72009-07-31 00:48:10 +0000573 Record.push_back(E->getCastKind()); // FIXME: stable encoding
John McCallf871d0c2010-08-07 06:22:56 +0000574
575 for (CastExpr::path_iterator
576 PI = E->path_begin(), PE = E->path_end(); PI != PE; ++PI)
577 Writer.AddCXXBaseSpecifier(**PI, Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000578}
579
Sebastian Redl3397c552010-08-18 23:56:27 +0000580void ASTStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000581 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000582 Writer.AddStmt(E->getLHS());
583 Writer.AddStmt(E->getRHS());
Chris Lattner63377d52009-04-27 06:20:01 +0000584 Record.push_back(E->getOpcode()); // FIXME: stable encoding
585 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000586 Code = serialization::EXPR_BINARY_OPERATOR;
Chris Lattner63377d52009-04-27 06:20:01 +0000587}
588
Sebastian Redl3397c552010-08-18 23:56:27 +0000589void ASTStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000590 VisitBinaryOperator(E);
591 Writer.AddTypeRef(E->getComputationLHSType(), Record);
592 Writer.AddTypeRef(E->getComputationResultType(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000593 Code = serialization::EXPR_COMPOUND_ASSIGN_OPERATOR;
Chris Lattner63377d52009-04-27 06:20:01 +0000594}
595
Sebastian Redl3397c552010-08-18 23:56:27 +0000596void ASTStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000597 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000598 Writer.AddStmt(E->getCond());
599 Writer.AddStmt(E->getLHS());
600 Writer.AddStmt(E->getRHS());
Fariborz Jahanianf9b949f2010-08-31 18:02:20 +0000601 Writer.AddStmt(E->getSAVE());
Douglas Gregor47e1f7c2009-08-26 14:37:04 +0000602 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
603 Writer.AddSourceLocation(E->getColonLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000604 Code = serialization::EXPR_CONDITIONAL_OPERATOR;
Chris Lattner63377d52009-04-27 06:20:01 +0000605}
606
Sebastian Redl3397c552010-08-18 23:56:27 +0000607void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000608 VisitCastExpr(E);
John McCall5baba9d2010-08-25 10:28:54 +0000609 Record.push_back(E->getValueKind());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000610 Code = serialization::EXPR_IMPLICIT_CAST;
Chris Lattner63377d52009-04-27 06:20:01 +0000611}
612
Sebastian Redl3397c552010-08-18 23:56:27 +0000613void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000614 VisitCastExpr(E);
John McCall9d125032010-01-15 18:39:57 +0000615 Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000616}
617
Sebastian Redl3397c552010-08-18 23:56:27 +0000618void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000619 VisitExplicitCastExpr(E);
620 Writer.AddSourceLocation(E->getLParenLoc(), Record);
621 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000622 Code = serialization::EXPR_CSTYLE_CAST;
Chris Lattner63377d52009-04-27 06:20:01 +0000623}
624
Sebastian Redl3397c552010-08-18 23:56:27 +0000625void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000626 VisitExpr(E);
627 Writer.AddSourceLocation(E->getLParenLoc(), Record);
John McCall42f56b52010-01-18 19:35:47 +0000628 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000629 Writer.AddStmt(E->getInitializer());
Chris Lattner63377d52009-04-27 06:20:01 +0000630 Record.push_back(E->isFileScope());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000631 Code = serialization::EXPR_COMPOUND_LITERAL;
Chris Lattner63377d52009-04-27 06:20:01 +0000632}
633
Sebastian Redl3397c552010-08-18 23:56:27 +0000634void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000635 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000636 Writer.AddStmt(E->getBase());
Chris Lattner63377d52009-04-27 06:20:01 +0000637 Writer.AddIdentifierRef(&E->getAccessor(), Record);
638 Writer.AddSourceLocation(E->getAccessorLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000639 Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
Chris Lattner63377d52009-04-27 06:20:01 +0000640}
641
Sebastian Redl3397c552010-08-18 23:56:27 +0000642void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000643 VisitExpr(E);
644 Record.push_back(E->getNumInits());
645 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000646 Writer.AddStmt(E->getInit(I));
647 Writer.AddStmt(E->getSyntacticForm());
Chris Lattner63377d52009-04-27 06:20:01 +0000648 Writer.AddSourceLocation(E->getLBraceLoc(), Record);
649 Writer.AddSourceLocation(E->getRBraceLoc(), Record);
650 Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
651 Record.push_back(E->hadArrayRangeDesignator());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000652 Code = serialization::EXPR_INIT_LIST;
Chris Lattner63377d52009-04-27 06:20:01 +0000653}
654
Sebastian Redl3397c552010-08-18 23:56:27 +0000655void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000656 VisitExpr(E);
657 Record.push_back(E->getNumSubExprs());
658 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000659 Writer.AddStmt(E->getSubExpr(I));
Chris Lattner63377d52009-04-27 06:20:01 +0000660 Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
661 Record.push_back(E->usesGNUSyntax());
662 for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
663 DEnd = E->designators_end();
664 D != DEnd; ++D) {
665 if (D->isFieldDesignator()) {
666 if (FieldDecl *Field = D->getField()) {
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000667 Record.push_back(serialization::DESIG_FIELD_DECL);
Chris Lattner63377d52009-04-27 06:20:01 +0000668 Writer.AddDeclRef(Field, Record);
669 } else {
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000670 Record.push_back(serialization::DESIG_FIELD_NAME);
Chris Lattner63377d52009-04-27 06:20:01 +0000671 Writer.AddIdentifierRef(D->getFieldName(), Record);
672 }
673 Writer.AddSourceLocation(D->getDotLoc(), Record);
674 Writer.AddSourceLocation(D->getFieldLoc(), Record);
675 } else if (D->isArrayDesignator()) {
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000676 Record.push_back(serialization::DESIG_ARRAY);
Chris Lattner63377d52009-04-27 06:20:01 +0000677 Record.push_back(D->getFirstExprIndex());
678 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
679 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
680 } else {
681 assert(D->isArrayRangeDesignator() && "Unknown designator");
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000682 Record.push_back(serialization::DESIG_ARRAY_RANGE);
Chris Lattner63377d52009-04-27 06:20:01 +0000683 Record.push_back(D->getFirstExprIndex());
684 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
685 Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
686 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
687 }
688 }
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000689 Code = serialization::EXPR_DESIGNATED_INIT;
Chris Lattner63377d52009-04-27 06:20:01 +0000690}
691
Sebastian Redl3397c552010-08-18 23:56:27 +0000692void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000693 VisitExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000694 Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
Chris Lattner63377d52009-04-27 06:20:01 +0000695}
696
Sebastian Redl3397c552010-08-18 23:56:27 +0000697void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000698 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000699 Writer.AddStmt(E->getSubExpr());
Abramo Bagnara2cad9002010-08-10 10:06:15 +0000700 Writer.AddTypeSourceInfo(E->getWrittenTypeInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000701 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
702 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000703 Code = serialization::EXPR_VA_ARG;
Chris Lattner63377d52009-04-27 06:20:01 +0000704}
705
Sebastian Redl3397c552010-08-18 23:56:27 +0000706void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000707 VisitExpr(E);
708 Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
709 Writer.AddSourceLocation(E->getLabelLoc(), Record);
710 Record.push_back(Writer.GetLabelID(E->getLabel()));
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000711 Code = serialization::EXPR_ADDR_LABEL;
Chris Lattner63377d52009-04-27 06:20:01 +0000712}
713
Sebastian Redl3397c552010-08-18 23:56:27 +0000714void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000715 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000716 Writer.AddStmt(E->getSubStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000717 Writer.AddSourceLocation(E->getLParenLoc(), Record);
718 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000719 Code = serialization::EXPR_STMT;
Chris Lattner63377d52009-04-27 06:20:01 +0000720}
721
Sebastian Redl3397c552010-08-18 23:56:27 +0000722void ASTStmtWriter::VisitTypesCompatibleExpr(TypesCompatibleExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000723 VisitExpr(E);
Abramo Bagnara3fcb73d2010-08-10 08:50:03 +0000724 Writer.AddTypeSourceInfo(E->getArgTInfo1(), Record);
725 Writer.AddTypeSourceInfo(E->getArgTInfo2(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000726 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
727 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000728 Code = serialization::EXPR_TYPES_COMPATIBLE;
Chris Lattner63377d52009-04-27 06:20:01 +0000729}
730
Sebastian Redl3397c552010-08-18 23:56:27 +0000731void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000732 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000733 Writer.AddStmt(E->getCond());
734 Writer.AddStmt(E->getLHS());
735 Writer.AddStmt(E->getRHS());
Chris Lattner63377d52009-04-27 06:20:01 +0000736 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
737 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000738 Code = serialization::EXPR_CHOOSE;
Chris Lattner63377d52009-04-27 06:20:01 +0000739}
740
Sebastian Redl3397c552010-08-18 23:56:27 +0000741void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000742 VisitExpr(E);
743 Writer.AddSourceLocation(E->getTokenLocation(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000744 Code = serialization::EXPR_GNU_NULL;
Chris Lattner63377d52009-04-27 06:20:01 +0000745}
746
Sebastian Redl3397c552010-08-18 23:56:27 +0000747void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000748 VisitExpr(E);
749 Record.push_back(E->getNumSubExprs());
750 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000751 Writer.AddStmt(E->getExpr(I));
Chris Lattner63377d52009-04-27 06:20:01 +0000752 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
753 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000754 Code = serialization::EXPR_SHUFFLE_VECTOR;
Chris Lattner63377d52009-04-27 06:20:01 +0000755}
756
Sebastian Redl3397c552010-08-18 23:56:27 +0000757void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000758 VisitExpr(E);
759 Writer.AddDeclRef(E->getBlockDecl(), Record);
760 Record.push_back(E->hasBlockDeclRefExprs());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000761 Code = serialization::EXPR_BLOCK;
Chris Lattner63377d52009-04-27 06:20:01 +0000762}
763
Sebastian Redl3397c552010-08-18 23:56:27 +0000764void ASTStmtWriter::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000765 VisitExpr(E);
766 Writer.AddDeclRef(E->getDecl(), Record);
767 Writer.AddSourceLocation(E->getLocation(), Record);
768 Record.push_back(E->isByRef());
Fariborz Jahanian9b0b57c2009-06-20 00:02:26 +0000769 Record.push_back(E->isConstQualAdded());
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000770 Writer.AddStmt(E->getCopyConstructorExpr());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000771 Code = serialization::EXPR_BLOCK_DECL_REF;
Chris Lattner63377d52009-04-27 06:20:01 +0000772}
773
774//===----------------------------------------------------------------------===//
775// Objective-C Expressions and Statements.
776//===----------------------------------------------------------------------===//
777
Sebastian Redl3397c552010-08-18 23:56:27 +0000778void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000779 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000780 Writer.AddStmt(E->getString());
Chris Lattner63377d52009-04-27 06:20:01 +0000781 Writer.AddSourceLocation(E->getAtLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000782 Code = serialization::EXPR_OBJC_STRING_LITERAL;
Chris Lattner63377d52009-04-27 06:20:01 +0000783}
784
Sebastian Redl3397c552010-08-18 23:56:27 +0000785void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000786 VisitExpr(E);
Douglas Gregor81d34662010-04-20 15:39:42 +0000787 Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000788 Writer.AddSourceLocation(E->getAtLoc(), Record);
789 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000790 Code = serialization::EXPR_OBJC_ENCODE;
Chris Lattner63377d52009-04-27 06:20:01 +0000791}
792
Sebastian Redl3397c552010-08-18 23:56:27 +0000793void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000794 VisitExpr(E);
795 Writer.AddSelectorRef(E->getSelector(), Record);
796 Writer.AddSourceLocation(E->getAtLoc(), Record);
797 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000798 Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000799}
800
Sebastian Redl3397c552010-08-18 23:56:27 +0000801void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000802 VisitExpr(E);
803 Writer.AddDeclRef(E->getProtocol(), Record);
804 Writer.AddSourceLocation(E->getAtLoc(), Record);
805 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000806 Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000807}
808
Sebastian Redl3397c552010-08-18 23:56:27 +0000809void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000810 VisitExpr(E);
811 Writer.AddDeclRef(E->getDecl(), Record);
812 Writer.AddSourceLocation(E->getLocation(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000813 Writer.AddStmt(E->getBase());
Chris Lattner63377d52009-04-27 06:20:01 +0000814 Record.push_back(E->isArrow());
815 Record.push_back(E->isFreeIvar());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000816 Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000817}
818
Sebastian Redl3397c552010-08-18 23:56:27 +0000819void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000820 VisitExpr(E);
821 Writer.AddDeclRef(E->getProperty(), Record);
822 Writer.AddSourceLocation(E->getLocation(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000823 Writer.AddStmt(E->getBase());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000824 Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000825}
826
Sebastian Redl3397c552010-08-18 23:56:27 +0000827void ASTStmtWriter::VisitObjCImplicitSetterGetterRefExpr(
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000828 ObjCImplicitSetterGetterRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000829 VisitExpr(E);
830 Writer.AddDeclRef(E->getGetterMethod(), Record);
831 Writer.AddDeclRef(E->getSetterMethod(), Record);
Mike Stump1eb44332009-09-09 15:08:12 +0000832
Fariborz Jahaniand2ae5aa2009-08-18 21:37:33 +0000833 // NOTE: InterfaceDecl and Base are mutually exclusive.
834 Writer.AddDeclRef(E->getInterfaceDecl(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000835 Writer.AddStmt(E->getBase());
Chris Lattner63377d52009-04-27 06:20:01 +0000836 Writer.AddSourceLocation(E->getLocation(), Record);
837 Writer.AddSourceLocation(E->getClassLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000838 Code = serialization::EXPR_OBJC_KVC_REF_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000839}
840
Sebastian Redl3397c552010-08-18 23:56:27 +0000841void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000842 VisitExpr(E);
843 Record.push_back(E->getNumArgs());
Douglas Gregor04badcf2010-04-21 00:45:42 +0000844 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
845 switch (E->getReceiverKind()) {
846 case ObjCMessageExpr::Instance:
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000847 Writer.AddStmt(E->getInstanceReceiver());
Douglas Gregor04badcf2010-04-21 00:45:42 +0000848 break;
849
850 case ObjCMessageExpr::Class:
851 Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
852 break;
853
854 case ObjCMessageExpr::SuperClass:
855 case ObjCMessageExpr::SuperInstance:
856 Writer.AddTypeRef(E->getSuperType(), Record);
857 Writer.AddSourceLocation(E->getSuperLoc(), Record);
858 break;
859 }
860
861 if (E->getMethodDecl()) {
862 Record.push_back(1);
863 Writer.AddDeclRef(E->getMethodDecl(), Record);
864 } else {
865 Record.push_back(0);
866 Writer.AddSelectorRef(E->getSelector(), Record);
867 }
868
Chris Lattner63377d52009-04-27 06:20:01 +0000869 Writer.AddSourceLocation(E->getLeftLoc(), Record);
870 Writer.AddSourceLocation(E->getRightLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000871
872 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
873 Arg != ArgEnd; ++Arg)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000874 Writer.AddStmt(*Arg);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000875 Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000876}
877
Sebastian Redl3397c552010-08-18 23:56:27 +0000878void ASTStmtWriter::VisitObjCSuperExpr(ObjCSuperExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000879 VisitExpr(E);
880 Writer.AddSourceLocation(E->getLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000881 Code = serialization::EXPR_OBJC_SUPER_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000882}
883
Sebastian Redl3397c552010-08-18 23:56:27 +0000884void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000885 VisitStmt(S);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000886 Writer.AddStmt(S->getElement());
887 Writer.AddStmt(S->getCollection());
888 Writer.AddStmt(S->getBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000889 Writer.AddSourceLocation(S->getForLoc(), Record);
890 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000891 Code = serialization::STMT_OBJC_FOR_COLLECTION;
Chris Lattner63377d52009-04-27 06:20:01 +0000892}
893
Sebastian Redl3397c552010-08-18 23:56:27 +0000894void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000895 Writer.AddStmt(S->getCatchBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000896 Writer.AddDeclRef(S->getCatchParamDecl(), Record);
897 Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
898 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000899 Code = serialization::STMT_OBJC_CATCH;
Chris Lattner63377d52009-04-27 06:20:01 +0000900}
901
Sebastian Redl3397c552010-08-18 23:56:27 +0000902void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000903 Writer.AddStmt(S->getFinallyBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000904 Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000905 Code = serialization::STMT_OBJC_FINALLY;
Chris Lattner63377d52009-04-27 06:20:01 +0000906}
907
Sebastian Redl3397c552010-08-18 23:56:27 +0000908void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000909 Record.push_back(S->getNumCatchStmts());
910 Record.push_back(S->getFinallyStmt() != 0);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000911 Writer.AddStmt(S->getTryBody());
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000912 for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000913 Writer.AddStmt(S->getCatchStmt(I));
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000914 if (S->getFinallyStmt())
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000915 Writer.AddStmt(S->getFinallyStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000916 Writer.AddSourceLocation(S->getAtTryLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000917 Code = serialization::STMT_OBJC_AT_TRY;
Chris Lattner63377d52009-04-27 06:20:01 +0000918}
919
Sebastian Redl3397c552010-08-18 23:56:27 +0000920void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000921 Writer.AddStmt(S->getSynchExpr());
922 Writer.AddStmt(S->getSynchBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000923 Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000924 Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
Chris Lattner63377d52009-04-27 06:20:01 +0000925}
926
Sebastian Redl3397c552010-08-18 23:56:27 +0000927void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000928 Writer.AddStmt(S->getThrowExpr());
Chris Lattner63377d52009-04-27 06:20:01 +0000929 Writer.AddSourceLocation(S->getThrowLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000930 Code = serialization::STMT_OBJC_AT_THROW;
Chris Lattner63377d52009-04-27 06:20:01 +0000931}
932
933//===----------------------------------------------------------------------===//
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000934// C++ Expressions and Statements.
935//===----------------------------------------------------------------------===//
936
Sebastian Redl3397c552010-08-18 23:56:27 +0000937void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
Argyrios Kyrtzidis7cb45e32010-07-22 16:03:56 +0000938 VisitStmt(S);
939 Writer.AddSourceLocation(S->getCatchLoc(), Record);
940 Writer.AddDeclRef(S->getExceptionDecl(), Record);
941 Writer.AddStmt(S->getHandlerBlock());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000942 Code = serialization::STMT_CXX_CATCH;
Argyrios Kyrtzidis7cb45e32010-07-22 16:03:56 +0000943}
944
Sebastian Redl3397c552010-08-18 23:56:27 +0000945void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
Argyrios Kyrtzidis7cb45e32010-07-22 16:03:56 +0000946 VisitStmt(S);
947 Record.push_back(S->getNumHandlers());
948 Writer.AddSourceLocation(S->getTryLoc(), Record);
949 Writer.AddStmt(S->getTryBlock());
950 for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
951 Writer.AddStmt(S->getHandler(i));
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000952 Code = serialization::STMT_CXX_TRY;
Argyrios Kyrtzidis7cb45e32010-07-22 16:03:56 +0000953}
954
Sebastian Redl3397c552010-08-18 23:56:27 +0000955void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000956 VisitCallExpr(E);
957 Record.push_back(E->getOperator());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000958 Code = serialization::EXPR_CXX_OPERATOR_CALL;
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000959}
960
Sebastian Redl3397c552010-08-18 23:56:27 +0000961void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
Chris Lattner1817bd42010-05-09 05:36:05 +0000962 VisitCallExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000963 Code = serialization::EXPR_CXX_MEMBER_CALL;
Chris Lattner1817bd42010-05-09 05:36:05 +0000964}
965
Sebastian Redl3397c552010-08-18 23:56:27 +0000966void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
Douglas Gregor39da0b82009-09-09 23:08:42 +0000967 VisitExpr(E);
Argyrios Kyrtzidis5e1b7c22010-06-24 08:57:09 +0000968 Record.push_back(E->getNumArgs());
969 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000970 Writer.AddStmt(E->getArg(I));
Douglas Gregor39da0b82009-09-09 23:08:42 +0000971 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor99a2e602009-12-16 01:38:02 +0000972 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000973 Record.push_back(E->isElidable());
Douglas Gregor16006c92009-12-16 18:50:27 +0000974 Record.push_back(E->requiresZeroInitialization());
Douglas Gregor3c9034c2010-05-15 00:13:29 +0000975 Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000976 Code = serialization::EXPR_CXX_CONSTRUCT;
Douglas Gregor39da0b82009-09-09 23:08:42 +0000977}
978
Sebastian Redl3397c552010-08-18 23:56:27 +0000979void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +0000980 VisitCXXConstructExpr(E);
Douglas Gregorab6677e2010-09-08 00:15:04 +0000981 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +0000982 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000983 Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +0000984}
985
Sebastian Redl3397c552010-08-18 23:56:27 +0000986void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +0000987 VisitExplicitCastExpr(E);
988 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
989}
990
Sebastian Redl3397c552010-08-18 23:56:27 +0000991void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +0000992 VisitCXXNamedCastExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000993 Code = serialization::EXPR_CXX_STATIC_CAST;
Sam Weinigce757a72010-01-16 21:21:01 +0000994}
995
Sebastian Redl3397c552010-08-18 23:56:27 +0000996void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +0000997 VisitCXXNamedCastExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000998 Code = serialization::EXPR_CXX_DYNAMIC_CAST;
Sam Weinigce757a72010-01-16 21:21:01 +0000999}
1000
Sebastian Redl3397c552010-08-18 23:56:27 +00001001void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +00001002 VisitCXXNamedCastExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001003 Code = serialization::EXPR_CXX_REINTERPRET_CAST;
Sam Weinigce757a72010-01-16 21:21:01 +00001004}
1005
Sebastian Redl3397c552010-08-18 23:56:27 +00001006void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +00001007 VisitCXXNamedCastExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001008 Code = serialization::EXPR_CXX_CONST_CAST;
Sam Weinigce757a72010-01-16 21:21:01 +00001009}
1010
Sebastian Redl3397c552010-08-18 23:56:27 +00001011void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +00001012 VisitExplicitCastExpr(E);
1013 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1014 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001015 Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
Sam Weinigce757a72010-01-16 21:21:01 +00001016}
1017
Sebastian Redl3397c552010-08-18 23:56:27 +00001018void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
Sam Weinigeb7f9612010-02-07 06:32:43 +00001019 VisitExpr(E);
1020 Record.push_back(E->getValue());
1021 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001022 Code = serialization::EXPR_CXX_BOOL_LITERAL;
Sam Weinigeb7f9612010-02-07 06:32:43 +00001023}
1024
Sebastian Redl3397c552010-08-18 23:56:27 +00001025void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
Sam Weinigeb7f9612010-02-07 06:32:43 +00001026 VisitExpr(E);
1027 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001028 Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
Sam Weinigeb7f9612010-02-07 06:32:43 +00001029}
1030
Sebastian Redl3397c552010-08-18 23:56:27 +00001031void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
Chris Lattner14ab24f2010-05-09 06:03:39 +00001032 VisitExpr(E);
1033 Writer.AddSourceRange(E->getSourceRange(), Record);
1034 if (E->isTypeOperand()) {
1035 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001036 Code = serialization::EXPR_CXX_TYPEID_TYPE;
Chris Lattner14ab24f2010-05-09 06:03:39 +00001037 } else {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001038 Writer.AddStmt(E->getExprOperand());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001039 Code = serialization::EXPR_CXX_TYPEID_EXPR;
Chris Lattner14ab24f2010-05-09 06:03:39 +00001040 }
1041}
1042
Francois Pichet01b7c302010-09-08 12:20:18 +00001043void ASTStmtWriter::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
1044 VisitExpr(E);
1045 Writer.AddSourceRange(E->getSourceRange(), Record);
1046 if (E->isTypeOperand()) {
1047 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
1048 Code = serialization::EXPR_CXX_UUIDOF_TYPE;
1049 } else {
1050 Writer.AddStmt(E->getExprOperand());
1051 Code = serialization::EXPR_CXX_UUIDOF_EXPR;
1052 }
1053}
1054
Sebastian Redl3397c552010-08-18 23:56:27 +00001055void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001056 VisitExpr(E);
1057 Writer.AddSourceLocation(E->getLocation(), Record);
1058 Record.push_back(E->isImplicit());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001059 Code = serialization::EXPR_CXX_THIS;
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001060}
1061
Sebastian Redl3397c552010-08-18 23:56:27 +00001062void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
Chris Lattner030854b2010-05-09 06:40:08 +00001063 VisitExpr(E);
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001064 Writer.AddSourceLocation(E->getThrowLoc(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001065 Writer.AddStmt(E->getSubExpr());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001066 Code = serialization::EXPR_CXX_THROW;
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001067}
1068
Sebastian Redl3397c552010-08-18 23:56:27 +00001069void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
Chris Lattner030854b2010-05-09 06:40:08 +00001070 VisitExpr(E);
Argyrios Kyrtzidis8a507332010-07-02 23:30:15 +00001071
1072 bool HasOtherExprStored = E->Param.getInt();
1073 // Store these first, the reader reads them before creation.
1074 Record.push_back(HasOtherExprStored);
1075 if (HasOtherExprStored)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001076 Writer.AddStmt(E->getExpr());
Argyrios Kyrtzidis8a507332010-07-02 23:30:15 +00001077 Writer.AddDeclRef(E->getParam(), Record);
1078 Writer.AddSourceLocation(E->getUsedLocation(), Record);
Chris Lattner030854b2010-05-09 06:40:08 +00001079
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001080 Code = serialization::EXPR_CXX_DEFAULT_ARG;
Chris Lattner030854b2010-05-09 06:40:08 +00001081}
1082
Sebastian Redl3397c552010-08-18 23:56:27 +00001083void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
Chris Lattnerd2598362010-05-10 00:25:06 +00001084 VisitExpr(E);
1085 Writer.AddCXXTemporary(E->getTemporary(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001086 Writer.AddStmt(E->getSubExpr());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001087 Code = serialization::EXPR_CXX_BIND_TEMPORARY;
Chris Lattnerd2598362010-05-10 00:25:06 +00001088}
1089
Sebastian Redl3397c552010-08-18 23:56:27 +00001090void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
Chris Lattner59218632010-05-10 01:22:27 +00001091 VisitExpr(E);
Douglas Gregorab6677e2010-09-08 00:15:04 +00001092 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Chris Lattner59218632010-05-10 01:22:27 +00001093 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001094 Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
Chris Lattner59218632010-05-10 01:22:27 +00001095}
1096
Sebastian Redl3397c552010-08-18 23:56:27 +00001097void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
Chris Lattner59218632010-05-10 01:22:27 +00001098 VisitExpr(E);
1099 Record.push_back(E->isGlobalNew());
Chris Lattner59218632010-05-10 01:22:27 +00001100 Record.push_back(E->hasInitializer());
1101 Record.push_back(E->isArray());
1102 Record.push_back(E->getNumPlacementArgs());
1103 Record.push_back(E->getNumConstructorArgs());
1104 Writer.AddDeclRef(E->getOperatorNew(), Record);
1105 Writer.AddDeclRef(E->getOperatorDelete(), Record);
1106 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor1bb2a932010-09-07 21:49:58 +00001107 Writer.AddTypeSourceInfo(E->getAllocatedTypeSourceInfo(), Record);
Douglas Gregor4bd40312010-07-13 15:54:32 +00001108 Writer.AddSourceRange(E->getTypeIdParens(), Record);
Chris Lattner59218632010-05-10 01:22:27 +00001109 Writer.AddSourceLocation(E->getStartLoc(), Record);
1110 Writer.AddSourceLocation(E->getEndLoc(), Record);
1111 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
1112 I != e; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001113 Writer.AddStmt(*I);
Chris Lattner59218632010-05-10 01:22:27 +00001114
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001115 Code = serialization::EXPR_CXX_NEW;
Chris Lattner59218632010-05-10 01:22:27 +00001116}
1117
Sebastian Redl3397c552010-08-18 23:56:27 +00001118void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
Argyrios Kyrtzidis95fc98c2010-06-22 17:07:59 +00001119 VisitExpr(E);
1120 Record.push_back(E->isGlobalDelete());
1121 Record.push_back(E->isArrayForm());
1122 Writer.AddDeclRef(E->getOperatorDelete(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001123 Writer.AddStmt(E->getArgument());
Argyrios Kyrtzidis95fc98c2010-06-22 17:07:59 +00001124 Writer.AddSourceLocation(E->getSourceRange().getBegin(), Record);
1125
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001126 Code = serialization::EXPR_CXX_DELETE;
Argyrios Kyrtzidis95fc98c2010-06-22 17:07:59 +00001127}
Chris Lattner59218632010-05-10 01:22:27 +00001128
Sebastian Redl3397c552010-08-18 23:56:27 +00001129void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
Argyrios Kyrtzidisde4bd182010-06-28 09:32:03 +00001130 VisitExpr(E);
1131
1132 Writer.AddStmt(E->getBase());
1133 Record.push_back(E->isArrow());
1134 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1135 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1136 Writer.AddSourceRange(E->getQualifierRange(), Record);
1137 Writer.AddTypeSourceInfo(E->getScopeTypeInfo(), Record);
1138 Writer.AddSourceLocation(E->getColonColonLoc(), Record);
1139 Writer.AddSourceLocation(E->getTildeLoc(), Record);
1140
1141 // PseudoDestructorTypeStorage.
1142 Writer.AddIdentifierRef(E->getDestroyedTypeIdentifier(), Record);
1143 if (E->getDestroyedTypeIdentifier())
1144 Writer.AddSourceLocation(E->getDestroyedTypeLoc(), Record);
1145 else
1146 Writer.AddTypeSourceInfo(E->getDestroyedTypeInfo(), Record);
1147
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001148 Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
Argyrios Kyrtzidisde4bd182010-06-28 09:32:03 +00001149}
1150
Sebastian Redl3397c552010-08-18 23:56:27 +00001151void ASTStmtWriter::VisitCXXExprWithTemporaries(CXXExprWithTemporaries *E) {
Chris Lattnerd2598362010-05-10 00:25:06 +00001152 VisitExpr(E);
1153 Record.push_back(E->getNumTemporaries());
1154 for (unsigned i = 0, e = E->getNumTemporaries(); i != e; ++i)
1155 Writer.AddCXXTemporary(E->getTemporary(i), Record);
1156
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001157 Writer.AddStmt(E->getSubExpr());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001158 Code = serialization::EXPR_CXX_EXPR_WITH_TEMPORARIES;
Chris Lattnerd2598362010-05-10 00:25:06 +00001159}
1160
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001161void
Sebastian Redl3397c552010-08-18 23:56:27 +00001162ASTStmtWriter::VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E){
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001163 VisitExpr(E);
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001164
1165 // Don't emit anything here, NumTemplateArgs must be emitted first.
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001166
1167 if (E->hasExplicitTemplateArgs()) {
John McCall096832c2010-08-19 23:49:38 +00001168 const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001169 assert(Args.NumTemplateArgs &&
Sebastian Redl3397c552010-08-18 23:56:27 +00001170 "Num of template args was zero! AST reading will mess up!");
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001171 Record.push_back(Args.NumTemplateArgs);
1172 AddExplicitTemplateArgumentList(Args);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001173 } else {
1174 Record.push_back(0);
1175 }
1176
1177 if (!E->isImplicitAccess())
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001178 Writer.AddStmt(E->getBase());
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001179 else
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001180 Writer.AddStmt(0);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001181 Writer.AddTypeRef(E->getBaseType(), Record);
1182 Record.push_back(E->isArrow());
1183 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1184 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1185 Writer.AddSourceRange(E->getQualifierRange(), Record);
1186 Writer.AddDeclRef(E->getFirstQualifierFoundInScope(), Record);
Abramo Bagnara25777432010-08-11 22:01:17 +00001187 // FIXME: write whole DeclarationNameInfo.
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001188 Writer.AddDeclarationName(E->getMember(), Record);
1189 Writer.AddSourceLocation(E->getMemberLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001190 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001191}
1192
1193void
Sebastian Redl3397c552010-08-18 23:56:27 +00001194ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
Argyrios Kyrtzidis12dffcd2010-06-28 09:31:56 +00001195 VisitExpr(E);
1196
1197 // Don't emit anything here, NumTemplateArgs must be emitted first.
1198
1199 if (E->hasExplicitTemplateArgs()) {
1200 const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
1201 assert(Args.NumTemplateArgs &&
Sebastian Redl3397c552010-08-18 23:56:27 +00001202 "Num of template args was zero! AST reading will mess up!");
Argyrios Kyrtzidis12dffcd2010-06-28 09:31:56 +00001203 Record.push_back(Args.NumTemplateArgs);
1204 AddExplicitTemplateArgumentList(Args);
1205 } else {
1206 Record.push_back(0);
1207 }
1208
Abramo Bagnara25777432010-08-11 22:01:17 +00001209 // FIXME: write whole DeclarationNameInfo.
Argyrios Kyrtzidis12dffcd2010-06-28 09:31:56 +00001210 Writer.AddDeclarationName(E->getDeclName(), Record);
1211 Writer.AddSourceLocation(E->getLocation(), Record);
1212 Writer.AddSourceRange(E->getQualifierRange(), Record);
1213 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001214 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
Argyrios Kyrtzidis12dffcd2010-06-28 09:31:56 +00001215}
1216
1217void
Sebastian Redl3397c552010-08-18 23:56:27 +00001218ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001219 VisitExpr(E);
1220 Record.push_back(E->arg_size());
1221 for (CXXUnresolvedConstructExpr::arg_iterator
1222 ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001223 Writer.AddStmt(*ArgI);
Douglas Gregorab6677e2010-09-08 00:15:04 +00001224 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001225 Writer.AddSourceLocation(E->getLParenLoc(), Record);
1226 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001227 Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001228}
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001229
Sebastian Redl3397c552010-08-18 23:56:27 +00001230void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001231 VisitExpr(E);
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001232
1233 // Don't emit anything here, NumTemplateArgs must be emitted first.
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001234
1235 if (E->hasExplicitTemplateArgs()) {
1236 const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
1237 assert(Args.NumTemplateArgs &&
Sebastian Redl3397c552010-08-18 23:56:27 +00001238 "Num of template args was zero! AST reading will mess up!");
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001239 Record.push_back(Args.NumTemplateArgs);
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001240 AddExplicitTemplateArgumentList(Args);
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001241 } else {
1242 Record.push_back(0);
1243 }
1244
1245 Record.push_back(E->getNumDecls());
1246 for (OverloadExpr::decls_iterator
1247 OvI = E->decls_begin(), OvE = E->decls_end(); OvI != OvE; ++OvI) {
1248 Writer.AddDeclRef(OvI.getDecl(), Record);
1249 Record.push_back(OvI.getAccess());
1250 }
1251
Abramo Bagnara25777432010-08-11 22:01:17 +00001252 // FIXME: write whole DeclarationNameInfo.
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001253 Writer.AddDeclarationName(E->getName(), Record);
1254 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1255 Writer.AddSourceRange(E->getQualifierRange(), Record);
1256 Writer.AddSourceLocation(E->getNameLoc(), Record);
1257}
1258
Sebastian Redl3397c552010-08-18 23:56:27 +00001259void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001260 VisitOverloadExpr(E);
1261 Record.push_back(E->isArrow());
1262 Record.push_back(E->hasUnresolvedUsing());
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001263 Writer.AddStmt(!E->isImplicitAccess() ? E->getBase() : 0);
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001264 Writer.AddTypeRef(E->getBaseType(), Record);
1265 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001266 Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001267}
1268
Sebastian Redl3397c552010-08-18 23:56:27 +00001269void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
Argyrios Kyrtzidisbd65bb52010-06-25 09:03:34 +00001270 VisitOverloadExpr(E);
1271 Record.push_back(E->requiresADL());
1272 Record.push_back(E->isOverloaded());
1273 Writer.AddDeclRef(E->getNamingClass(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001274 Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
Argyrios Kyrtzidisbd65bb52010-06-25 09:03:34 +00001275}
1276
Sebastian Redl3397c552010-08-18 23:56:27 +00001277void ASTStmtWriter::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +00001278 VisitExpr(E);
1279 Record.push_back(E->getTrait());
1280 Writer.AddSourceRange(E->getSourceRange(), Record);
1281 Writer.AddTypeRef(E->getQueriedType(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001282 Code = serialization::EXPR_CXX_UNARY_TYPE_TRAIT;
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +00001283}
1284
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001285//===----------------------------------------------------------------------===//
Sebastian Redla4232eb2010-08-18 23:56:21 +00001286// ASTWriter Implementation
Chris Lattner63377d52009-04-27 06:20:01 +00001287//===----------------------------------------------------------------------===//
1288
Sebastian Redla4232eb2010-08-18 23:56:21 +00001289unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001290 assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001291 "SwitchCase recorded twice");
1292 unsigned NextID = SwitchCaseIDs.size();
1293 SwitchCaseIDs[S] = NextID;
1294 return NextID;
1295}
1296
Sebastian Redla4232eb2010-08-18 23:56:21 +00001297unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001298 assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001299 "SwitchCase hasn't been seen yet");
1300 return SwitchCaseIDs[S];
1301}
1302
1303/// \brief Retrieve the ID for the given label statement, which may
1304/// or may not have been emitted yet.
Sebastian Redla4232eb2010-08-18 23:56:21 +00001305unsigned ASTWriter::GetLabelID(LabelStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +00001306 std::map<LabelStmt *, unsigned>::iterator Pos = LabelIDs.find(S);
1307 if (Pos != LabelIDs.end())
1308 return Pos->second;
Mike Stump1eb44332009-09-09 15:08:12 +00001309
Chris Lattner63377d52009-04-27 06:20:01 +00001310 unsigned NextID = LabelIDs.size();
1311 LabelIDs[S] = NextID;
1312 return NextID;
1313}
1314
1315/// \brief Write the given substatement or subexpression to the
1316/// bitstream.
Sebastian Redla4232eb2010-08-18 23:56:21 +00001317void ASTWriter::WriteSubStmt(Stmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +00001318 RecordData Record;
Sebastian Redl3397c552010-08-18 23:56:27 +00001319 ASTStmtWriter Writer(*this, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001320 ++NumStatements;
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001321
Chris Lattner63377d52009-04-27 06:20:01 +00001322 if (!S) {
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001323 Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001324 return;
1325 }
Mike Stump1eb44332009-09-09 15:08:12 +00001326
Sebastian Redla4232eb2010-08-18 23:56:21 +00001327 // Redirect ASTWriter::AddStmt to collect sub stmts.
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001328 llvm::SmallVector<Stmt *, 16> SubStmts;
1329 CollectedStmts = &SubStmts;
1330
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001331 Writer.Code = serialization::STMT_NULL_PTR;
Chris Lattner63377d52009-04-27 06:20:01 +00001332 Writer.Visit(S);
Chris Lattner59218632010-05-10 01:22:27 +00001333
1334#ifndef NDEBUG
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001335 if (Writer.Code == serialization::STMT_NULL_PTR) {
Argyrios Kyrtzidis12bf28f2010-06-25 09:03:12 +00001336 SourceManager &SrcMgr
1337 = DeclIDs.begin()->first->getASTContext().getSourceManager();
1338 S->dump(SrcMgr);
Sebastian Redl3397c552010-08-18 23:56:27 +00001339 assert(0 && "Unhandled sub statement writing AST file");
Chris Lattner59218632010-05-10 01:22:27 +00001340 }
1341#endif
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001342
Sebastian Redla4232eb2010-08-18 23:56:21 +00001343 // Revert ASTWriter::AddStmt.
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001344 CollectedStmts = &StmtsToEmit;
1345
1346 // Write the sub stmts in reverse order, last to first. When reading them back
1347 // we will read them in correct order by "pop"ing them from the Stmts stack.
1348 // This simplifies reading and allows to store a variable number of sub stmts
1349 // without knowing it in advance.
1350 while (!SubStmts.empty())
1351 WriteSubStmt(SubStmts.pop_back_val());
1352
Mike Stump1eb44332009-09-09 15:08:12 +00001353 Stream.EmitRecord(Writer.Code, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001354}
1355
1356/// \brief Flush all of the statements that have been added to the
1357/// queue via AddStmt().
Sebastian Redla4232eb2010-08-18 23:56:21 +00001358void ASTWriter::FlushStmts() {
Chris Lattner63377d52009-04-27 06:20:01 +00001359 RecordData Record;
Mike Stump1eb44332009-09-09 15:08:12 +00001360
Chris Lattner63377d52009-04-27 06:20:01 +00001361 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001362 WriteSubStmt(StmtsToEmit[I]);
1363
Mike Stump1eb44332009-09-09 15:08:12 +00001364 assert(N == StmtsToEmit.size() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001365 "Substatement writen via AddStmt rather than WriteSubStmt!");
Mike Stump1eb44332009-09-09 15:08:12 +00001366
Chris Lattner63377d52009-04-27 06:20:01 +00001367 // Note that we are at the end of a full expression. Any
1368 // expression records that follow this one are part of a different
1369 // expression.
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001370 Stream.EmitRecord(serialization::STMT_STOP, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001371 }
Mike Stump1eb44332009-09-09 15:08:12 +00001372
Chris Lattner63377d52009-04-27 06:20:01 +00001373 StmtsToEmit.clear();
Chris Lattner63377d52009-04-27 06:20:01 +00001374}