blob: 1bb9d1934fa6a5dd0b9c76b3cbedb3838c27865a [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);
Chris Lattner2fbdfcd2010-05-09 06:15:05 +0000134 void VisitCXXThisExpr(CXXThisExpr *E);
135 void VisitCXXThrowExpr(CXXThrowExpr *E);
Chris Lattner030854b2010-05-09 06:40:08 +0000136 void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E);
Chris Lattnerd2598362010-05-10 00:25:06 +0000137 void VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E);
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +0000138 void VisitCXXBindReferenceExpr(CXXBindReferenceExpr *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());
Douglas Gregor47e1f7c2009-08-26 14:37:04 +0000601 Writer.AddSourceLocation(E->getQuestionLoc(), Record);
602 Writer.AddSourceLocation(E->getColonLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000603 Code = serialization::EXPR_CONDITIONAL_OPERATOR;
Chris Lattner63377d52009-04-27 06:20:01 +0000604}
605
Sebastian Redl3397c552010-08-18 23:56:27 +0000606void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000607 VisitCastExpr(E);
Sebastian Redl906082e2010-07-20 04:20:21 +0000608 Record.push_back(E->getCategory());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000609 Code = serialization::EXPR_IMPLICIT_CAST;
Chris Lattner63377d52009-04-27 06:20:01 +0000610}
611
Sebastian Redl3397c552010-08-18 23:56:27 +0000612void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000613 VisitCastExpr(E);
John McCall9d125032010-01-15 18:39:57 +0000614 Writer.AddTypeSourceInfo(E->getTypeInfoAsWritten(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000615}
616
Sebastian Redl3397c552010-08-18 23:56:27 +0000617void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000618 VisitExplicitCastExpr(E);
619 Writer.AddSourceLocation(E->getLParenLoc(), Record);
620 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000621 Code = serialization::EXPR_CSTYLE_CAST;
Chris Lattner63377d52009-04-27 06:20:01 +0000622}
623
Sebastian Redl3397c552010-08-18 23:56:27 +0000624void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000625 VisitExpr(E);
626 Writer.AddSourceLocation(E->getLParenLoc(), Record);
John McCall42f56b52010-01-18 19:35:47 +0000627 Writer.AddTypeSourceInfo(E->getTypeSourceInfo(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000628 Writer.AddStmt(E->getInitializer());
Chris Lattner63377d52009-04-27 06:20:01 +0000629 Record.push_back(E->isFileScope());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000630 Code = serialization::EXPR_COMPOUND_LITERAL;
Chris Lattner63377d52009-04-27 06:20:01 +0000631}
632
Sebastian Redl3397c552010-08-18 23:56:27 +0000633void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000634 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000635 Writer.AddStmt(E->getBase());
Chris Lattner63377d52009-04-27 06:20:01 +0000636 Writer.AddIdentifierRef(&E->getAccessor(), Record);
637 Writer.AddSourceLocation(E->getAccessorLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000638 Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
Chris Lattner63377d52009-04-27 06:20:01 +0000639}
640
Sebastian Redl3397c552010-08-18 23:56:27 +0000641void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000642 VisitExpr(E);
643 Record.push_back(E->getNumInits());
644 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000645 Writer.AddStmt(E->getInit(I));
646 Writer.AddStmt(E->getSyntacticForm());
Chris Lattner63377d52009-04-27 06:20:01 +0000647 Writer.AddSourceLocation(E->getLBraceLoc(), Record);
648 Writer.AddSourceLocation(E->getRBraceLoc(), Record);
649 Writer.AddDeclRef(E->getInitializedFieldInUnion(), Record);
650 Record.push_back(E->hadArrayRangeDesignator());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000651 Code = serialization::EXPR_INIT_LIST;
Chris Lattner63377d52009-04-27 06:20:01 +0000652}
653
Sebastian Redl3397c552010-08-18 23:56:27 +0000654void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000655 VisitExpr(E);
656 Record.push_back(E->getNumSubExprs());
657 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000658 Writer.AddStmt(E->getSubExpr(I));
Chris Lattner63377d52009-04-27 06:20:01 +0000659 Writer.AddSourceLocation(E->getEqualOrColonLoc(), Record);
660 Record.push_back(E->usesGNUSyntax());
661 for (DesignatedInitExpr::designators_iterator D = E->designators_begin(),
662 DEnd = E->designators_end();
663 D != DEnd; ++D) {
664 if (D->isFieldDesignator()) {
665 if (FieldDecl *Field = D->getField()) {
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000666 Record.push_back(serialization::DESIG_FIELD_DECL);
Chris Lattner63377d52009-04-27 06:20:01 +0000667 Writer.AddDeclRef(Field, Record);
668 } else {
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000669 Record.push_back(serialization::DESIG_FIELD_NAME);
Chris Lattner63377d52009-04-27 06:20:01 +0000670 Writer.AddIdentifierRef(D->getFieldName(), Record);
671 }
672 Writer.AddSourceLocation(D->getDotLoc(), Record);
673 Writer.AddSourceLocation(D->getFieldLoc(), Record);
674 } else if (D->isArrayDesignator()) {
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000675 Record.push_back(serialization::DESIG_ARRAY);
Chris Lattner63377d52009-04-27 06:20:01 +0000676 Record.push_back(D->getFirstExprIndex());
677 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
678 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
679 } else {
680 assert(D->isArrayRangeDesignator() && "Unknown designator");
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000681 Record.push_back(serialization::DESIG_ARRAY_RANGE);
Chris Lattner63377d52009-04-27 06:20:01 +0000682 Record.push_back(D->getFirstExprIndex());
683 Writer.AddSourceLocation(D->getLBracketLoc(), Record);
684 Writer.AddSourceLocation(D->getEllipsisLoc(), Record);
685 Writer.AddSourceLocation(D->getRBracketLoc(), Record);
686 }
687 }
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000688 Code = serialization::EXPR_DESIGNATED_INIT;
Chris Lattner63377d52009-04-27 06:20:01 +0000689}
690
Sebastian Redl3397c552010-08-18 23:56:27 +0000691void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000692 VisitExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000693 Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
Chris Lattner63377d52009-04-27 06:20:01 +0000694}
695
Sebastian Redl3397c552010-08-18 23:56:27 +0000696void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000697 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000698 Writer.AddStmt(E->getSubExpr());
Abramo Bagnara2cad9002010-08-10 10:06:15 +0000699 Writer.AddTypeSourceInfo(E->getWrittenTypeInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000700 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
701 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000702 Code = serialization::EXPR_VA_ARG;
Chris Lattner63377d52009-04-27 06:20:01 +0000703}
704
Sebastian Redl3397c552010-08-18 23:56:27 +0000705void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000706 VisitExpr(E);
707 Writer.AddSourceLocation(E->getAmpAmpLoc(), Record);
708 Writer.AddSourceLocation(E->getLabelLoc(), Record);
709 Record.push_back(Writer.GetLabelID(E->getLabel()));
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000710 Code = serialization::EXPR_ADDR_LABEL;
Chris Lattner63377d52009-04-27 06:20:01 +0000711}
712
Sebastian Redl3397c552010-08-18 23:56:27 +0000713void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000714 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000715 Writer.AddStmt(E->getSubStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000716 Writer.AddSourceLocation(E->getLParenLoc(), Record);
717 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000718 Code = serialization::EXPR_STMT;
Chris Lattner63377d52009-04-27 06:20:01 +0000719}
720
Sebastian Redl3397c552010-08-18 23:56:27 +0000721void ASTStmtWriter::VisitTypesCompatibleExpr(TypesCompatibleExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000722 VisitExpr(E);
Abramo Bagnara3fcb73d2010-08-10 08:50:03 +0000723 Writer.AddTypeSourceInfo(E->getArgTInfo1(), Record);
724 Writer.AddTypeSourceInfo(E->getArgTInfo2(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000725 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
726 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000727 Code = serialization::EXPR_TYPES_COMPATIBLE;
Chris Lattner63377d52009-04-27 06:20:01 +0000728}
729
Sebastian Redl3397c552010-08-18 23:56:27 +0000730void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000731 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000732 Writer.AddStmt(E->getCond());
733 Writer.AddStmt(E->getLHS());
734 Writer.AddStmt(E->getRHS());
Chris Lattner63377d52009-04-27 06:20:01 +0000735 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
736 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000737 Code = serialization::EXPR_CHOOSE;
Chris Lattner63377d52009-04-27 06:20:01 +0000738}
739
Sebastian Redl3397c552010-08-18 23:56:27 +0000740void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000741 VisitExpr(E);
742 Writer.AddSourceLocation(E->getTokenLocation(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000743 Code = serialization::EXPR_GNU_NULL;
Chris Lattner63377d52009-04-27 06:20:01 +0000744}
745
Sebastian Redl3397c552010-08-18 23:56:27 +0000746void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000747 VisitExpr(E);
748 Record.push_back(E->getNumSubExprs());
749 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000750 Writer.AddStmt(E->getExpr(I));
Chris Lattner63377d52009-04-27 06:20:01 +0000751 Writer.AddSourceLocation(E->getBuiltinLoc(), Record);
752 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000753 Code = serialization::EXPR_SHUFFLE_VECTOR;
Chris Lattner63377d52009-04-27 06:20:01 +0000754}
755
Sebastian Redl3397c552010-08-18 23:56:27 +0000756void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000757 VisitExpr(E);
758 Writer.AddDeclRef(E->getBlockDecl(), Record);
759 Record.push_back(E->hasBlockDeclRefExprs());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000760 Code = serialization::EXPR_BLOCK;
Chris Lattner63377d52009-04-27 06:20:01 +0000761}
762
Sebastian Redl3397c552010-08-18 23:56:27 +0000763void ASTStmtWriter::VisitBlockDeclRefExpr(BlockDeclRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000764 VisitExpr(E);
765 Writer.AddDeclRef(E->getDecl(), Record);
766 Writer.AddSourceLocation(E->getLocation(), Record);
767 Record.push_back(E->isByRef());
Fariborz Jahanian9b0b57c2009-06-20 00:02:26 +0000768 Record.push_back(E->isConstQualAdded());
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000769 Writer.AddStmt(E->getCopyConstructorExpr());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000770 Code = serialization::EXPR_BLOCK_DECL_REF;
Chris Lattner63377d52009-04-27 06:20:01 +0000771}
772
773//===----------------------------------------------------------------------===//
774// Objective-C Expressions and Statements.
775//===----------------------------------------------------------------------===//
776
Sebastian Redl3397c552010-08-18 23:56:27 +0000777void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000778 VisitExpr(E);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000779 Writer.AddStmt(E->getString());
Chris Lattner63377d52009-04-27 06:20:01 +0000780 Writer.AddSourceLocation(E->getAtLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000781 Code = serialization::EXPR_OBJC_STRING_LITERAL;
Chris Lattner63377d52009-04-27 06:20:01 +0000782}
783
Sebastian Redl3397c552010-08-18 23:56:27 +0000784void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000785 VisitExpr(E);
Douglas Gregor81d34662010-04-20 15:39:42 +0000786 Writer.AddTypeSourceInfo(E->getEncodedTypeSourceInfo(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000787 Writer.AddSourceLocation(E->getAtLoc(), Record);
788 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000789 Code = serialization::EXPR_OBJC_ENCODE;
Chris Lattner63377d52009-04-27 06:20:01 +0000790}
791
Sebastian Redl3397c552010-08-18 23:56:27 +0000792void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000793 VisitExpr(E);
794 Writer.AddSelectorRef(E->getSelector(), Record);
795 Writer.AddSourceLocation(E->getAtLoc(), Record);
796 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000797 Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000798}
799
Sebastian Redl3397c552010-08-18 23:56:27 +0000800void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000801 VisitExpr(E);
802 Writer.AddDeclRef(E->getProtocol(), Record);
803 Writer.AddSourceLocation(E->getAtLoc(), Record);
804 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000805 Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000806}
807
Sebastian Redl3397c552010-08-18 23:56:27 +0000808void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000809 VisitExpr(E);
810 Writer.AddDeclRef(E->getDecl(), Record);
811 Writer.AddSourceLocation(E->getLocation(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000812 Writer.AddStmt(E->getBase());
Chris Lattner63377d52009-04-27 06:20:01 +0000813 Record.push_back(E->isArrow());
814 Record.push_back(E->isFreeIvar());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000815 Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000816}
817
Sebastian Redl3397c552010-08-18 23:56:27 +0000818void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000819 VisitExpr(E);
820 Writer.AddDeclRef(E->getProperty(), Record);
821 Writer.AddSourceLocation(E->getLocation(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000822 Writer.AddStmt(E->getBase());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000823 Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000824}
825
Sebastian Redl3397c552010-08-18 23:56:27 +0000826void ASTStmtWriter::VisitObjCImplicitSetterGetterRefExpr(
Fariborz Jahanian09105f52009-08-20 17:02:02 +0000827 ObjCImplicitSetterGetterRefExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000828 VisitExpr(E);
829 Writer.AddDeclRef(E->getGetterMethod(), Record);
830 Writer.AddDeclRef(E->getSetterMethod(), Record);
Mike Stump1eb44332009-09-09 15:08:12 +0000831
Fariborz Jahaniand2ae5aa2009-08-18 21:37:33 +0000832 // NOTE: InterfaceDecl and Base are mutually exclusive.
833 Writer.AddDeclRef(E->getInterfaceDecl(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000834 Writer.AddStmt(E->getBase());
Chris Lattner63377d52009-04-27 06:20:01 +0000835 Writer.AddSourceLocation(E->getLocation(), Record);
836 Writer.AddSourceLocation(E->getClassLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000837 Code = serialization::EXPR_OBJC_KVC_REF_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000838}
839
Sebastian Redl3397c552010-08-18 23:56:27 +0000840void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000841 VisitExpr(E);
842 Record.push_back(E->getNumArgs());
Douglas Gregor04badcf2010-04-21 00:45:42 +0000843 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
844 switch (E->getReceiverKind()) {
845 case ObjCMessageExpr::Instance:
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000846 Writer.AddStmt(E->getInstanceReceiver());
Douglas Gregor04badcf2010-04-21 00:45:42 +0000847 break;
848
849 case ObjCMessageExpr::Class:
850 Writer.AddTypeSourceInfo(E->getClassReceiverTypeInfo(), Record);
851 break;
852
853 case ObjCMessageExpr::SuperClass:
854 case ObjCMessageExpr::SuperInstance:
855 Writer.AddTypeRef(E->getSuperType(), Record);
856 Writer.AddSourceLocation(E->getSuperLoc(), Record);
857 break;
858 }
859
860 if (E->getMethodDecl()) {
861 Record.push_back(1);
862 Writer.AddDeclRef(E->getMethodDecl(), Record);
863 } else {
864 Record.push_back(0);
865 Writer.AddSelectorRef(E->getSelector(), Record);
866 }
867
Chris Lattner63377d52009-04-27 06:20:01 +0000868 Writer.AddSourceLocation(E->getLeftLoc(), Record);
869 Writer.AddSourceLocation(E->getRightLoc(), Record);
Chris Lattner63377d52009-04-27 06:20:01 +0000870
871 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
872 Arg != ArgEnd; ++Arg)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000873 Writer.AddStmt(*Arg);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000874 Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000875}
876
Sebastian Redl3397c552010-08-18 23:56:27 +0000877void ASTStmtWriter::VisitObjCSuperExpr(ObjCSuperExpr *E) {
Chris Lattner63377d52009-04-27 06:20:01 +0000878 VisitExpr(E);
879 Writer.AddSourceLocation(E->getLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000880 Code = serialization::EXPR_OBJC_SUPER_EXPR;
Chris Lattner63377d52009-04-27 06:20:01 +0000881}
882
Sebastian Redl3397c552010-08-18 23:56:27 +0000883void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +0000884 VisitStmt(S);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000885 Writer.AddStmt(S->getElement());
886 Writer.AddStmt(S->getCollection());
887 Writer.AddStmt(S->getBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000888 Writer.AddSourceLocation(S->getForLoc(), Record);
889 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000890 Code = serialization::STMT_OBJC_FOR_COLLECTION;
Chris Lattner63377d52009-04-27 06:20:01 +0000891}
892
Sebastian Redl3397c552010-08-18 23:56:27 +0000893void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000894 Writer.AddStmt(S->getCatchBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000895 Writer.AddDeclRef(S->getCatchParamDecl(), Record);
896 Writer.AddSourceLocation(S->getAtCatchLoc(), Record);
897 Writer.AddSourceLocation(S->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000898 Code = serialization::STMT_OBJC_CATCH;
Chris Lattner63377d52009-04-27 06:20:01 +0000899}
900
Sebastian Redl3397c552010-08-18 23:56:27 +0000901void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000902 Writer.AddStmt(S->getFinallyBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000903 Writer.AddSourceLocation(S->getAtFinallyLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000904 Code = serialization::STMT_OBJC_FINALLY;
Chris Lattner63377d52009-04-27 06:20:01 +0000905}
906
Sebastian Redl3397c552010-08-18 23:56:27 +0000907void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000908 Record.push_back(S->getNumCatchStmts());
909 Record.push_back(S->getFinallyStmt() != 0);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000910 Writer.AddStmt(S->getTryBody());
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000911 for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000912 Writer.AddStmt(S->getCatchStmt(I));
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +0000913 if (S->getFinallyStmt())
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000914 Writer.AddStmt(S->getFinallyStmt());
Chris Lattner63377d52009-04-27 06:20:01 +0000915 Writer.AddSourceLocation(S->getAtTryLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000916 Code = serialization::STMT_OBJC_AT_TRY;
Chris Lattner63377d52009-04-27 06:20:01 +0000917}
918
Sebastian Redl3397c552010-08-18 23:56:27 +0000919void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000920 Writer.AddStmt(S->getSynchExpr());
921 Writer.AddStmt(S->getSynchBody());
Chris Lattner63377d52009-04-27 06:20:01 +0000922 Writer.AddSourceLocation(S->getAtSynchronizedLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000923 Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
Chris Lattner63377d52009-04-27 06:20:01 +0000924}
925
Sebastian Redl3397c552010-08-18 23:56:27 +0000926void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000927 Writer.AddStmt(S->getThrowExpr());
Chris Lattner63377d52009-04-27 06:20:01 +0000928 Writer.AddSourceLocation(S->getThrowLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000929 Code = serialization::STMT_OBJC_AT_THROW;
Chris Lattner63377d52009-04-27 06:20:01 +0000930}
931
932//===----------------------------------------------------------------------===//
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000933// C++ Expressions and Statements.
934//===----------------------------------------------------------------------===//
935
Sebastian Redl3397c552010-08-18 23:56:27 +0000936void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
Argyrios Kyrtzidis7cb45e32010-07-22 16:03:56 +0000937 VisitStmt(S);
938 Writer.AddSourceLocation(S->getCatchLoc(), Record);
939 Writer.AddDeclRef(S->getExceptionDecl(), Record);
940 Writer.AddStmt(S->getHandlerBlock());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000941 Code = serialization::STMT_CXX_CATCH;
Argyrios Kyrtzidis7cb45e32010-07-22 16:03:56 +0000942}
943
Sebastian Redl3397c552010-08-18 23:56:27 +0000944void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
Argyrios Kyrtzidis7cb45e32010-07-22 16:03:56 +0000945 VisitStmt(S);
946 Record.push_back(S->getNumHandlers());
947 Writer.AddSourceLocation(S->getTryLoc(), Record);
948 Writer.AddStmt(S->getTryBlock());
949 for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
950 Writer.AddStmt(S->getHandler(i));
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000951 Code = serialization::STMT_CXX_TRY;
Argyrios Kyrtzidis7cb45e32010-07-22 16:03:56 +0000952}
953
Sebastian Redl3397c552010-08-18 23:56:27 +0000954void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000955 VisitCallExpr(E);
956 Record.push_back(E->getOperator());
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000957 Code = serialization::EXPR_CXX_OPERATOR_CALL;
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +0000958}
959
Sebastian Redl3397c552010-08-18 23:56:27 +0000960void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
Chris Lattner1817bd42010-05-09 05:36:05 +0000961 VisitCallExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000962 Code = serialization::EXPR_CXX_MEMBER_CALL;
Chris Lattner1817bd42010-05-09 05:36:05 +0000963}
964
Sebastian Redl3397c552010-08-18 23:56:27 +0000965void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
Douglas Gregor39da0b82009-09-09 23:08:42 +0000966 VisitExpr(E);
Argyrios Kyrtzidis5e1b7c22010-06-24 08:57:09 +0000967 Record.push_back(E->getNumArgs());
968 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +0000969 Writer.AddStmt(E->getArg(I));
Douglas Gregor39da0b82009-09-09 23:08:42 +0000970 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor99a2e602009-12-16 01:38:02 +0000971 Writer.AddSourceLocation(E->getLocation(), Record);
Douglas Gregor39da0b82009-09-09 23:08:42 +0000972 Record.push_back(E->isElidable());
Douglas Gregor16006c92009-12-16 18:50:27 +0000973 Record.push_back(E->requiresZeroInitialization());
Douglas Gregor3c9034c2010-05-15 00:13:29 +0000974 Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000975 Code = serialization::EXPR_CXX_CONSTRUCT;
Douglas Gregor39da0b82009-09-09 23:08:42 +0000976}
977
Sebastian Redl3397c552010-08-18 23:56:27 +0000978void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +0000979 VisitCXXConstructExpr(E);
980 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
981 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000982 Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +0000983}
984
Sebastian Redl3397c552010-08-18 23:56:27 +0000985void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +0000986 VisitExplicitCastExpr(E);
987 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
988}
989
Sebastian Redl3397c552010-08-18 23:56:27 +0000990void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +0000991 VisitCXXNamedCastExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000992 Code = serialization::EXPR_CXX_STATIC_CAST;
Sam Weinigce757a72010-01-16 21:21:01 +0000993}
994
Sebastian Redl3397c552010-08-18 23:56:27 +0000995void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +0000996 VisitCXXNamedCastExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +0000997 Code = serialization::EXPR_CXX_DYNAMIC_CAST;
Sam Weinigce757a72010-01-16 21:21:01 +0000998}
999
Sebastian Redl3397c552010-08-18 23:56:27 +00001000void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +00001001 VisitCXXNamedCastExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001002 Code = serialization::EXPR_CXX_REINTERPRET_CAST;
Sam Weinigce757a72010-01-16 21:21:01 +00001003}
1004
Sebastian Redl3397c552010-08-18 23:56:27 +00001005void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +00001006 VisitCXXNamedCastExpr(E);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001007 Code = serialization::EXPR_CXX_CONST_CAST;
Sam Weinigce757a72010-01-16 21:21:01 +00001008}
1009
Sebastian Redl3397c552010-08-18 23:56:27 +00001010void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
Sam Weinigce757a72010-01-16 21:21:01 +00001011 VisitExplicitCastExpr(E);
1012 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1013 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001014 Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
Sam Weinigce757a72010-01-16 21:21:01 +00001015}
1016
Sebastian Redl3397c552010-08-18 23:56:27 +00001017void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
Sam Weinigeb7f9612010-02-07 06:32:43 +00001018 VisitExpr(E);
1019 Record.push_back(E->getValue());
1020 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001021 Code = serialization::EXPR_CXX_BOOL_LITERAL;
Sam Weinigeb7f9612010-02-07 06:32:43 +00001022}
1023
Sebastian Redl3397c552010-08-18 23:56:27 +00001024void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
Sam Weinigeb7f9612010-02-07 06:32:43 +00001025 VisitExpr(E);
1026 Writer.AddSourceLocation(E->getLocation(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001027 Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
Sam Weinigeb7f9612010-02-07 06:32:43 +00001028}
1029
Sebastian Redl3397c552010-08-18 23:56:27 +00001030void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
Chris Lattner14ab24f2010-05-09 06:03:39 +00001031 VisitExpr(E);
1032 Writer.AddSourceRange(E->getSourceRange(), Record);
1033 if (E->isTypeOperand()) {
1034 Writer.AddTypeSourceInfo(E->getTypeOperandSourceInfo(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001035 Code = serialization::EXPR_CXX_TYPEID_TYPE;
Chris Lattner14ab24f2010-05-09 06:03:39 +00001036 } else {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001037 Writer.AddStmt(E->getExprOperand());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001038 Code = serialization::EXPR_CXX_TYPEID_EXPR;
Chris Lattner14ab24f2010-05-09 06:03:39 +00001039 }
1040}
1041
Sebastian Redl3397c552010-08-18 23:56:27 +00001042void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001043 VisitExpr(E);
1044 Writer.AddSourceLocation(E->getLocation(), Record);
1045 Record.push_back(E->isImplicit());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001046 Code = serialization::EXPR_CXX_THIS;
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001047}
1048
Sebastian Redl3397c552010-08-18 23:56:27 +00001049void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
Chris Lattner030854b2010-05-09 06:40:08 +00001050 VisitExpr(E);
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001051 Writer.AddSourceLocation(E->getThrowLoc(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001052 Writer.AddStmt(E->getSubExpr());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001053 Code = serialization::EXPR_CXX_THROW;
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001054}
1055
Sebastian Redl3397c552010-08-18 23:56:27 +00001056void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
Chris Lattner030854b2010-05-09 06:40:08 +00001057 VisitExpr(E);
Argyrios Kyrtzidis8a507332010-07-02 23:30:15 +00001058
1059 bool HasOtherExprStored = E->Param.getInt();
1060 // Store these first, the reader reads them before creation.
1061 Record.push_back(HasOtherExprStored);
1062 if (HasOtherExprStored)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001063 Writer.AddStmt(E->getExpr());
Argyrios Kyrtzidis8a507332010-07-02 23:30:15 +00001064 Writer.AddDeclRef(E->getParam(), Record);
1065 Writer.AddSourceLocation(E->getUsedLocation(), Record);
Chris Lattner030854b2010-05-09 06:40:08 +00001066
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001067 Code = serialization::EXPR_CXX_DEFAULT_ARG;
Chris Lattner030854b2010-05-09 06:40:08 +00001068}
1069
Sebastian Redl3397c552010-08-18 23:56:27 +00001070void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
Chris Lattnerd2598362010-05-10 00:25:06 +00001071 VisitExpr(E);
1072 Writer.AddCXXTemporary(E->getTemporary(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001073 Writer.AddStmt(E->getSubExpr());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001074 Code = serialization::EXPR_CXX_BIND_TEMPORARY;
Chris Lattnerd2598362010-05-10 00:25:06 +00001075}
1076
Sebastian Redl3397c552010-08-18 23:56:27 +00001077void ASTStmtWriter::VisitCXXBindReferenceExpr(CXXBindReferenceExpr *E) {
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +00001078 VisitExpr(E);
1079 Writer.AddStmt(E->getSubExpr());
1080 Record.push_back(E->extendsLifetime());
1081 Record.push_back(E->requiresTemporaryCopy());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001082 Code = serialization::EXPR_CXX_BIND_REFERENCE;
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +00001083}
1084
Sebastian Redl3397c552010-08-18 23:56:27 +00001085void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
Chris Lattner59218632010-05-10 01:22:27 +00001086 VisitExpr(E);
1087 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1088 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001089 Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
Chris Lattner59218632010-05-10 01:22:27 +00001090}
1091
Sebastian Redl3397c552010-08-18 23:56:27 +00001092void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
Chris Lattner59218632010-05-10 01:22:27 +00001093 VisitExpr(E);
1094 Record.push_back(E->isGlobalNew());
Chris Lattner59218632010-05-10 01:22:27 +00001095 Record.push_back(E->hasInitializer());
1096 Record.push_back(E->isArray());
1097 Record.push_back(E->getNumPlacementArgs());
1098 Record.push_back(E->getNumConstructorArgs());
1099 Writer.AddDeclRef(E->getOperatorNew(), Record);
1100 Writer.AddDeclRef(E->getOperatorDelete(), Record);
1101 Writer.AddDeclRef(E->getConstructor(), Record);
Douglas Gregor4bd40312010-07-13 15:54:32 +00001102 Writer.AddSourceRange(E->getTypeIdParens(), Record);
Chris Lattner59218632010-05-10 01:22:27 +00001103 Writer.AddSourceLocation(E->getStartLoc(), Record);
1104 Writer.AddSourceLocation(E->getEndLoc(), Record);
1105 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
1106 I != e; ++I)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001107 Writer.AddStmt(*I);
Chris Lattner59218632010-05-10 01:22:27 +00001108
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001109 Code = serialization::EXPR_CXX_NEW;
Chris Lattner59218632010-05-10 01:22:27 +00001110}
1111
Sebastian Redl3397c552010-08-18 23:56:27 +00001112void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
Argyrios Kyrtzidis95fc98c2010-06-22 17:07:59 +00001113 VisitExpr(E);
1114 Record.push_back(E->isGlobalDelete());
1115 Record.push_back(E->isArrayForm());
1116 Writer.AddDeclRef(E->getOperatorDelete(), Record);
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001117 Writer.AddStmt(E->getArgument());
Argyrios Kyrtzidis95fc98c2010-06-22 17:07:59 +00001118 Writer.AddSourceLocation(E->getSourceRange().getBegin(), Record);
1119
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001120 Code = serialization::EXPR_CXX_DELETE;
Argyrios Kyrtzidis95fc98c2010-06-22 17:07:59 +00001121}
Chris Lattner59218632010-05-10 01:22:27 +00001122
Sebastian Redl3397c552010-08-18 23:56:27 +00001123void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
Argyrios Kyrtzidisde4bd182010-06-28 09:32:03 +00001124 VisitExpr(E);
1125
1126 Writer.AddStmt(E->getBase());
1127 Record.push_back(E->isArrow());
1128 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1129 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1130 Writer.AddSourceRange(E->getQualifierRange(), Record);
1131 Writer.AddTypeSourceInfo(E->getScopeTypeInfo(), Record);
1132 Writer.AddSourceLocation(E->getColonColonLoc(), Record);
1133 Writer.AddSourceLocation(E->getTildeLoc(), Record);
1134
1135 // PseudoDestructorTypeStorage.
1136 Writer.AddIdentifierRef(E->getDestroyedTypeIdentifier(), Record);
1137 if (E->getDestroyedTypeIdentifier())
1138 Writer.AddSourceLocation(E->getDestroyedTypeLoc(), Record);
1139 else
1140 Writer.AddTypeSourceInfo(E->getDestroyedTypeInfo(), Record);
1141
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001142 Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
Argyrios Kyrtzidisde4bd182010-06-28 09:32:03 +00001143}
1144
Sebastian Redl3397c552010-08-18 23:56:27 +00001145void ASTStmtWriter::VisitCXXExprWithTemporaries(CXXExprWithTemporaries *E) {
Chris Lattnerd2598362010-05-10 00:25:06 +00001146 VisitExpr(E);
1147 Record.push_back(E->getNumTemporaries());
1148 for (unsigned i = 0, e = E->getNumTemporaries(); i != e; ++i)
1149 Writer.AddCXXTemporary(E->getTemporary(i), Record);
1150
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001151 Writer.AddStmt(E->getSubExpr());
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001152 Code = serialization::EXPR_CXX_EXPR_WITH_TEMPORARIES;
Chris Lattnerd2598362010-05-10 00:25:06 +00001153}
1154
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001155void
Sebastian Redl3397c552010-08-18 23:56:27 +00001156ASTStmtWriter::VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E){
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001157 VisitExpr(E);
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001158
1159 // Don't emit anything here, NumTemplateArgs must be emitted first.
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001160
1161 if (E->hasExplicitTemplateArgs()) {
John McCall096832c2010-08-19 23:49:38 +00001162 const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001163 assert(Args.NumTemplateArgs &&
Sebastian Redl3397c552010-08-18 23:56:27 +00001164 "Num of template args was zero! AST reading will mess up!");
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001165 Record.push_back(Args.NumTemplateArgs);
1166 AddExplicitTemplateArgumentList(Args);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001167 } else {
1168 Record.push_back(0);
1169 }
1170
1171 if (!E->isImplicitAccess())
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001172 Writer.AddStmt(E->getBase());
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001173 else
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001174 Writer.AddStmt(0);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001175 Writer.AddTypeRef(E->getBaseType(), Record);
1176 Record.push_back(E->isArrow());
1177 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
1178 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1179 Writer.AddSourceRange(E->getQualifierRange(), Record);
1180 Writer.AddDeclRef(E->getFirstQualifierFoundInScope(), Record);
Abramo Bagnara25777432010-08-11 22:01:17 +00001181 // FIXME: write whole DeclarationNameInfo.
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001182 Writer.AddDeclarationName(E->getMember(), Record);
1183 Writer.AddSourceLocation(E->getMemberLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001184 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001185}
1186
1187void
Sebastian Redl3397c552010-08-18 23:56:27 +00001188ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
Argyrios Kyrtzidis12dffcd2010-06-28 09:31:56 +00001189 VisitExpr(E);
1190
1191 // Don't emit anything here, NumTemplateArgs must be emitted first.
1192
1193 if (E->hasExplicitTemplateArgs()) {
1194 const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
1195 assert(Args.NumTemplateArgs &&
Sebastian Redl3397c552010-08-18 23:56:27 +00001196 "Num of template args was zero! AST reading will mess up!");
Argyrios Kyrtzidis12dffcd2010-06-28 09:31:56 +00001197 Record.push_back(Args.NumTemplateArgs);
1198 AddExplicitTemplateArgumentList(Args);
1199 } else {
1200 Record.push_back(0);
1201 }
1202
Abramo Bagnara25777432010-08-11 22:01:17 +00001203 // FIXME: write whole DeclarationNameInfo.
Argyrios Kyrtzidis12dffcd2010-06-28 09:31:56 +00001204 Writer.AddDeclarationName(E->getDeclName(), Record);
1205 Writer.AddSourceLocation(E->getLocation(), Record);
1206 Writer.AddSourceRange(E->getQualifierRange(), Record);
1207 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001208 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
Argyrios Kyrtzidis12dffcd2010-06-28 09:31:56 +00001209}
1210
1211void
Sebastian Redl3397c552010-08-18 23:56:27 +00001212ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001213 VisitExpr(E);
1214 Record.push_back(E->arg_size());
1215 for (CXXUnresolvedConstructExpr::arg_iterator
1216 ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001217 Writer.AddStmt(*ArgI);
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001218 Writer.AddSourceLocation(E->getTypeBeginLoc(), Record);
1219 Writer.AddTypeRef(E->getTypeAsWritten(), Record);
1220 Writer.AddSourceLocation(E->getLParenLoc(), Record);
1221 Writer.AddSourceLocation(E->getRParenLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001222 Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
Argyrios Kyrtzidis8dfbd8b2010-06-24 08:57:31 +00001223}
Chris Lattner2fbdfcd2010-05-09 06:15:05 +00001224
Sebastian Redl3397c552010-08-18 23:56:27 +00001225void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001226 VisitExpr(E);
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001227
1228 // Don't emit anything here, NumTemplateArgs must be emitted first.
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001229
1230 if (E->hasExplicitTemplateArgs()) {
1231 const ExplicitTemplateArgumentList &Args = E->getExplicitTemplateArgs();
1232 assert(Args.NumTemplateArgs &&
Sebastian Redl3397c552010-08-18 23:56:27 +00001233 "Num of template args was zero! AST reading will mess up!");
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001234 Record.push_back(Args.NumTemplateArgs);
Argyrios Kyrtzidis36c76f02010-06-28 09:31:48 +00001235 AddExplicitTemplateArgumentList(Args);
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001236 } else {
1237 Record.push_back(0);
1238 }
1239
1240 Record.push_back(E->getNumDecls());
1241 for (OverloadExpr::decls_iterator
1242 OvI = E->decls_begin(), OvE = E->decls_end(); OvI != OvE; ++OvI) {
1243 Writer.AddDeclRef(OvI.getDecl(), Record);
1244 Record.push_back(OvI.getAccess());
1245 }
1246
Abramo Bagnara25777432010-08-11 22:01:17 +00001247 // FIXME: write whole DeclarationNameInfo.
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001248 Writer.AddDeclarationName(E->getName(), Record);
1249 Writer.AddNestedNameSpecifier(E->getQualifier(), Record);
1250 Writer.AddSourceRange(E->getQualifierRange(), Record);
1251 Writer.AddSourceLocation(E->getNameLoc(), Record);
1252}
1253
Sebastian Redl3397c552010-08-18 23:56:27 +00001254void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001255 VisitOverloadExpr(E);
1256 Record.push_back(E->isArrow());
1257 Record.push_back(E->hasUnresolvedUsing());
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001258 Writer.AddStmt(!E->isImplicitAccess() ? E->getBase() : 0);
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001259 Writer.AddTypeRef(E->getBaseType(), Record);
1260 Writer.AddSourceLocation(E->getOperatorLoc(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001261 Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
Argyrios Kyrtzidisa77eb082010-06-25 09:03:26 +00001262}
1263
Sebastian Redl3397c552010-08-18 23:56:27 +00001264void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
Argyrios Kyrtzidisbd65bb52010-06-25 09:03:34 +00001265 VisitOverloadExpr(E);
1266 Record.push_back(E->requiresADL());
1267 Record.push_back(E->isOverloaded());
1268 Writer.AddDeclRef(E->getNamingClass(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001269 Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
Argyrios Kyrtzidisbd65bb52010-06-25 09:03:34 +00001270}
1271
Sebastian Redl3397c552010-08-18 23:56:27 +00001272void ASTStmtWriter::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +00001273 VisitExpr(E);
1274 Record.push_back(E->getTrait());
1275 Writer.AddSourceRange(E->getSourceRange(), Record);
1276 Writer.AddTypeRef(E->getQueriedType(), Record);
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001277 Code = serialization::EXPR_CXX_UNARY_TYPE_TRAIT;
Argyrios Kyrtzidis6d00c132010-07-10 11:46:15 +00001278}
1279
Argyrios Kyrtzidisba0a9002009-07-14 03:19:21 +00001280//===----------------------------------------------------------------------===//
Sebastian Redla4232eb2010-08-18 23:56:21 +00001281// ASTWriter Implementation
Chris Lattner63377d52009-04-27 06:20:01 +00001282//===----------------------------------------------------------------------===//
1283
Sebastian Redla4232eb2010-08-18 23:56:21 +00001284unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001285 assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001286 "SwitchCase recorded twice");
1287 unsigned NextID = SwitchCaseIDs.size();
1288 SwitchCaseIDs[S] = NextID;
1289 return NextID;
1290}
1291
Sebastian Redla4232eb2010-08-18 23:56:21 +00001292unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
Mike Stump1eb44332009-09-09 15:08:12 +00001293 assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001294 "SwitchCase hasn't been seen yet");
1295 return SwitchCaseIDs[S];
1296}
1297
1298/// \brief Retrieve the ID for the given label statement, which may
1299/// or may not have been emitted yet.
Sebastian Redla4232eb2010-08-18 23:56:21 +00001300unsigned ASTWriter::GetLabelID(LabelStmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +00001301 std::map<LabelStmt *, unsigned>::iterator Pos = LabelIDs.find(S);
1302 if (Pos != LabelIDs.end())
1303 return Pos->second;
Mike Stump1eb44332009-09-09 15:08:12 +00001304
Chris Lattner63377d52009-04-27 06:20:01 +00001305 unsigned NextID = LabelIDs.size();
1306 LabelIDs[S] = NextID;
1307 return NextID;
1308}
1309
1310/// \brief Write the given substatement or subexpression to the
1311/// bitstream.
Sebastian Redla4232eb2010-08-18 23:56:21 +00001312void ASTWriter::WriteSubStmt(Stmt *S) {
Chris Lattner63377d52009-04-27 06:20:01 +00001313 RecordData Record;
Sebastian Redl3397c552010-08-18 23:56:27 +00001314 ASTStmtWriter Writer(*this, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001315 ++NumStatements;
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001316
Chris Lattner63377d52009-04-27 06:20:01 +00001317 if (!S) {
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001318 Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001319 return;
1320 }
Mike Stump1eb44332009-09-09 15:08:12 +00001321
Sebastian Redla4232eb2010-08-18 23:56:21 +00001322 // Redirect ASTWriter::AddStmt to collect sub stmts.
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001323 llvm::SmallVector<Stmt *, 16> SubStmts;
1324 CollectedStmts = &SubStmts;
1325
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001326 Writer.Code = serialization::STMT_NULL_PTR;
Chris Lattner63377d52009-04-27 06:20:01 +00001327 Writer.Visit(S);
Chris Lattner59218632010-05-10 01:22:27 +00001328
1329#ifndef NDEBUG
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001330 if (Writer.Code == serialization::STMT_NULL_PTR) {
Argyrios Kyrtzidis12bf28f2010-06-25 09:03:12 +00001331 SourceManager &SrcMgr
1332 = DeclIDs.begin()->first->getASTContext().getSourceManager();
1333 S->dump(SrcMgr);
Sebastian Redl3397c552010-08-18 23:56:27 +00001334 assert(0 && "Unhandled sub statement writing AST file");
Chris Lattner59218632010-05-10 01:22:27 +00001335 }
1336#endif
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001337
Sebastian Redla4232eb2010-08-18 23:56:21 +00001338 // Revert ASTWriter::AddStmt.
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001339 CollectedStmts = &StmtsToEmit;
1340
1341 // Write the sub stmts in reverse order, last to first. When reading them back
1342 // we will read them in correct order by "pop"ing them from the Stmts stack.
1343 // This simplifies reading and allows to store a variable number of sub stmts
1344 // without knowing it in advance.
1345 while (!SubStmts.empty())
1346 WriteSubStmt(SubStmts.pop_back_val());
1347
Mike Stump1eb44332009-09-09 15:08:12 +00001348 Stream.EmitRecord(Writer.Code, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001349}
1350
1351/// \brief Flush all of the statements that have been added to the
1352/// queue via AddStmt().
Sebastian Redla4232eb2010-08-18 23:56:21 +00001353void ASTWriter::FlushStmts() {
Chris Lattner63377d52009-04-27 06:20:01 +00001354 RecordData Record;
Mike Stump1eb44332009-09-09 15:08:12 +00001355
Chris Lattner63377d52009-04-27 06:20:01 +00001356 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
Argyrios Kyrtzidis919e6932010-06-28 22:28:35 +00001357 WriteSubStmt(StmtsToEmit[I]);
1358
Mike Stump1eb44332009-09-09 15:08:12 +00001359 assert(N == StmtsToEmit.size() &&
Chris Lattner63377d52009-04-27 06:20:01 +00001360 "Substatement writen via AddStmt rather than WriteSubStmt!");
Mike Stump1eb44332009-09-09 15:08:12 +00001361
Chris Lattner63377d52009-04-27 06:20:01 +00001362 // Note that we are at the end of a full expression. Any
1363 // expression records that follow this one are part of a different
1364 // expression.
Sebastian Redl8538e8d2010-08-18 23:57:32 +00001365 Stream.EmitRecord(serialization::STMT_STOP, Record);
Chris Lattner63377d52009-04-27 06:20:01 +00001366 }
Mike Stump1eb44332009-09-09 15:08:12 +00001367
Chris Lattner63377d52009-04-27 06:20:01 +00001368 StmtsToEmit.clear();
Chris Lattner63377d52009-04-27 06:20:01 +00001369}