1 //===--- ASTWriterStmt.cpp - Statement and Expression Serialization -------===//
2 //
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 /// \file
11 /// \brief Implements serialization for Statements and Expressions.
12 ///
13 //===----------------------------------------------------------------------===//
14
15 #include "clang/Serialization/ASTWriter.h"
16 #include "clang/AST/ASTContext.h"
17 #include "clang/AST/DeclCXX.h"
18 #include "clang/AST/DeclObjC.h"
19 #include "clang/AST/DeclTemplate.h"
20 #include "clang/AST/StmtVisitor.h"
21 #include "clang/Lex/Token.h"
22 #include "llvm/Bitcode/BitstreamWriter.h"
23 using namespace clang;
24
25 //===----------------------------------------------------------------------===//
26 // Statement/expression serialization
27 //===----------------------------------------------------------------------===//
28
29 namespace clang {
30
31 class ASTStmtWriter : public StmtVisitor<ASTStmtWriter, void> {
32 ASTWriter &Writer;
33 ASTRecordWriter Record;
34
35 serialization::StmtCode Code;
36 unsigned AbbrevToUse;
37
38 public:
ASTStmtWriter(ASTWriter & Writer,ASTWriter::RecordData & Record)39 ASTStmtWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
40 : Writer(Writer), Record(Writer, Record),
41 Code(serialization::STMT_NULL_PTR), AbbrevToUse(0) {}
42
43 ASTStmtWriter(const ASTStmtWriter&) = delete;
44
Emit()45 uint64_t Emit() {
46 assert(Code != serialization::STMT_NULL_PTR &&
47 "unhandled sub-statement writing AST file");
48 return Record.EmitStmt(Code, AbbrevToUse);
49 }
50
51 void AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo &ArgInfo,
52 const TemplateArgumentLoc *Args);
53
54 void VisitStmt(Stmt *S);
55 #define STMT(Type, Base) \
56 void Visit##Type(Type *);
57 #include "clang/AST/StmtNodes.inc"
58 };
59 }
60
AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo & ArgInfo,const TemplateArgumentLoc * Args)61 void ASTStmtWriter::AddTemplateKWAndArgsInfo(
62 const ASTTemplateKWAndArgsInfo &ArgInfo, const TemplateArgumentLoc *Args) {
63 Record.AddSourceLocation(ArgInfo.TemplateKWLoc);
64 Record.AddSourceLocation(ArgInfo.LAngleLoc);
65 Record.AddSourceLocation(ArgInfo.RAngleLoc);
66 for (unsigned i = 0; i != ArgInfo.NumTemplateArgs; ++i)
67 Record.AddTemplateArgumentLoc(Args[i]);
68 }
69
VisitStmt(Stmt * S)70 void ASTStmtWriter::VisitStmt(Stmt *S) {
71 }
72
VisitNullStmt(NullStmt * S)73 void ASTStmtWriter::VisitNullStmt(NullStmt *S) {
74 VisitStmt(S);
75 Record.AddSourceLocation(S->getSemiLoc());
76 Record.push_back(S->HasLeadingEmptyMacro);
77 Code = serialization::STMT_NULL;
78 }
79
VisitCompoundStmt(CompoundStmt * S)80 void ASTStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
81 VisitStmt(S);
82 Record.push_back(S->size());
83 for (auto *CS : S->body())
84 Record.AddStmt(CS);
85 Record.AddSourceLocation(S->getLBracLoc());
86 Record.AddSourceLocation(S->getRBracLoc());
87 Code = serialization::STMT_COMPOUND;
88 }
89
VisitSwitchCase(SwitchCase * S)90 void ASTStmtWriter::VisitSwitchCase(SwitchCase *S) {
91 VisitStmt(S);
92 Record.push_back(Writer.getSwitchCaseID(S));
93 Record.AddSourceLocation(S->getKeywordLoc());
94 Record.AddSourceLocation(S->getColonLoc());
95 }
96
VisitCaseStmt(CaseStmt * S)97 void ASTStmtWriter::VisitCaseStmt(CaseStmt *S) {
98 VisitSwitchCase(S);
99 Record.AddStmt(S->getLHS());
100 Record.AddStmt(S->getRHS());
101 Record.AddStmt(S->getSubStmt());
102 Record.AddSourceLocation(S->getEllipsisLoc());
103 Code = serialization::STMT_CASE;
104 }
105
VisitDefaultStmt(DefaultStmt * S)106 void ASTStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
107 VisitSwitchCase(S);
108 Record.AddStmt(S->getSubStmt());
109 Code = serialization::STMT_DEFAULT;
110 }
111
VisitLabelStmt(LabelStmt * S)112 void ASTStmtWriter::VisitLabelStmt(LabelStmt *S) {
113 VisitStmt(S);
114 Record.AddDeclRef(S->getDecl());
115 Record.AddStmt(S->getSubStmt());
116 Record.AddSourceLocation(S->getIdentLoc());
117 Code = serialization::STMT_LABEL;
118 }
119
VisitAttributedStmt(AttributedStmt * S)120 void ASTStmtWriter::VisitAttributedStmt(AttributedStmt *S) {
121 VisitStmt(S);
122 Record.push_back(S->getAttrs().size());
123 Record.AddAttributes(S->getAttrs());
124 Record.AddStmt(S->getSubStmt());
125 Record.AddSourceLocation(S->getAttrLoc());
126 Code = serialization::STMT_ATTRIBUTED;
127 }
128
VisitIfStmt(IfStmt * S)129 void ASTStmtWriter::VisitIfStmt(IfStmt *S) {
130 VisitStmt(S);
131 Record.push_back(S->isConstexpr());
132 Record.AddStmt(S->getInit());
133 Record.AddDeclRef(S->getConditionVariable());
134 Record.AddStmt(S->getCond());
135 Record.AddStmt(S->getThen());
136 Record.AddStmt(S->getElse());
137 Record.AddSourceLocation(S->getIfLoc());
138 Record.AddSourceLocation(S->getElseLoc());
139 Code = serialization::STMT_IF;
140 }
141
VisitSwitchStmt(SwitchStmt * S)142 void ASTStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
143 VisitStmt(S);
144 Record.AddStmt(S->getInit());
145 Record.AddDeclRef(S->getConditionVariable());
146 Record.AddStmt(S->getCond());
147 Record.AddStmt(S->getBody());
148 Record.AddSourceLocation(S->getSwitchLoc());
149 Record.push_back(S->isAllEnumCasesCovered());
150 for (SwitchCase *SC = S->getSwitchCaseList(); SC;
151 SC = SC->getNextSwitchCase())
152 Record.push_back(Writer.RecordSwitchCaseID(SC));
153 Code = serialization::STMT_SWITCH;
154 }
155
VisitWhileStmt(WhileStmt * S)156 void ASTStmtWriter::VisitWhileStmt(WhileStmt *S) {
157 VisitStmt(S);
158 Record.AddDeclRef(S->getConditionVariable());
159 Record.AddStmt(S->getCond());
160 Record.AddStmt(S->getBody());
161 Record.AddSourceLocation(S->getWhileLoc());
162 Code = serialization::STMT_WHILE;
163 }
164
VisitDoStmt(DoStmt * S)165 void ASTStmtWriter::VisitDoStmt(DoStmt *S) {
166 VisitStmt(S);
167 Record.AddStmt(S->getCond());
168 Record.AddStmt(S->getBody());
169 Record.AddSourceLocation(S->getDoLoc());
170 Record.AddSourceLocation(S->getWhileLoc());
171 Record.AddSourceLocation(S->getRParenLoc());
172 Code = serialization::STMT_DO;
173 }
174
VisitForStmt(ForStmt * S)175 void ASTStmtWriter::VisitForStmt(ForStmt *S) {
176 VisitStmt(S);
177 Record.AddStmt(S->getInit());
178 Record.AddStmt(S->getCond());
179 Record.AddDeclRef(S->getConditionVariable());
180 Record.AddStmt(S->getInc());
181 Record.AddStmt(S->getBody());
182 Record.AddSourceLocation(S->getForLoc());
183 Record.AddSourceLocation(S->getLParenLoc());
184 Record.AddSourceLocation(S->getRParenLoc());
185 Code = serialization::STMT_FOR;
186 }
187
VisitGotoStmt(GotoStmt * S)188 void ASTStmtWriter::VisitGotoStmt(GotoStmt *S) {
189 VisitStmt(S);
190 Record.AddDeclRef(S->getLabel());
191 Record.AddSourceLocation(S->getGotoLoc());
192 Record.AddSourceLocation(S->getLabelLoc());
193 Code = serialization::STMT_GOTO;
194 }
195
VisitIndirectGotoStmt(IndirectGotoStmt * S)196 void ASTStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
197 VisitStmt(S);
198 Record.AddSourceLocation(S->getGotoLoc());
199 Record.AddSourceLocation(S->getStarLoc());
200 Record.AddStmt(S->getTarget());
201 Code = serialization::STMT_INDIRECT_GOTO;
202 }
203
VisitContinueStmt(ContinueStmt * S)204 void ASTStmtWriter::VisitContinueStmt(ContinueStmt *S) {
205 VisitStmt(S);
206 Record.AddSourceLocation(S->getContinueLoc());
207 Code = serialization::STMT_CONTINUE;
208 }
209
VisitBreakStmt(BreakStmt * S)210 void ASTStmtWriter::VisitBreakStmt(BreakStmt *S) {
211 VisitStmt(S);
212 Record.AddSourceLocation(S->getBreakLoc());
213 Code = serialization::STMT_BREAK;
214 }
215
VisitReturnStmt(ReturnStmt * S)216 void ASTStmtWriter::VisitReturnStmt(ReturnStmt *S) {
217 VisitStmt(S);
218 Record.AddStmt(S->getRetValue());
219 Record.AddSourceLocation(S->getReturnLoc());
220 Record.AddDeclRef(S->getNRVOCandidate());
221 Code = serialization::STMT_RETURN;
222 }
223
VisitDeclStmt(DeclStmt * S)224 void ASTStmtWriter::VisitDeclStmt(DeclStmt *S) {
225 VisitStmt(S);
226 Record.AddSourceLocation(S->getStartLoc());
227 Record.AddSourceLocation(S->getEndLoc());
228 DeclGroupRef DG = S->getDeclGroup();
229 for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
230 Record.AddDeclRef(*D);
231 Code = serialization::STMT_DECL;
232 }
233
VisitAsmStmt(AsmStmt * S)234 void ASTStmtWriter::VisitAsmStmt(AsmStmt *S) {
235 VisitStmt(S);
236 Record.push_back(S->getNumOutputs());
237 Record.push_back(S->getNumInputs());
238 Record.push_back(S->getNumClobbers());
239 Record.AddSourceLocation(S->getAsmLoc());
240 Record.push_back(S->isVolatile());
241 Record.push_back(S->isSimple());
242 }
243
VisitGCCAsmStmt(GCCAsmStmt * S)244 void ASTStmtWriter::VisitGCCAsmStmt(GCCAsmStmt *S) {
245 VisitAsmStmt(S);
246 Record.AddSourceLocation(S->getRParenLoc());
247 Record.AddStmt(S->getAsmString());
248
249 // Outputs
250 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
251 Record.AddIdentifierRef(S->getOutputIdentifier(I));
252 Record.AddStmt(S->getOutputConstraintLiteral(I));
253 Record.AddStmt(S->getOutputExpr(I));
254 }
255
256 // Inputs
257 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
258 Record.AddIdentifierRef(S->getInputIdentifier(I));
259 Record.AddStmt(S->getInputConstraintLiteral(I));
260 Record.AddStmt(S->getInputExpr(I));
261 }
262
263 // Clobbers
264 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
265 Record.AddStmt(S->getClobberStringLiteral(I));
266
267 Code = serialization::STMT_GCCASM;
268 }
269
VisitMSAsmStmt(MSAsmStmt * S)270 void ASTStmtWriter::VisitMSAsmStmt(MSAsmStmt *S) {
271 VisitAsmStmt(S);
272 Record.AddSourceLocation(S->getLBraceLoc());
273 Record.AddSourceLocation(S->getEndLoc());
274 Record.push_back(S->getNumAsmToks());
275 Record.AddString(S->getAsmString());
276
277 // Tokens
278 for (unsigned I = 0, N = S->getNumAsmToks(); I != N; ++I) {
279 // FIXME: Move this to ASTRecordWriter?
280 Writer.AddToken(S->getAsmToks()[I], Record.getRecordData());
281 }
282
283 // Clobbers
284 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I) {
285 Record.AddString(S->getClobber(I));
286 }
287
288 // Outputs
289 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
290 Record.AddStmt(S->getOutputExpr(I));
291 Record.AddString(S->getOutputConstraint(I));
292 }
293
294 // Inputs
295 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
296 Record.AddStmt(S->getInputExpr(I));
297 Record.AddString(S->getInputConstraint(I));
298 }
299
300 Code = serialization::STMT_MSASM;
301 }
302
VisitCoroutineBodyStmt(CoroutineBodyStmt * S)303 void ASTStmtWriter::VisitCoroutineBodyStmt(CoroutineBodyStmt *S) {
304 // FIXME: Implement coroutine serialization.
305 llvm_unreachable("unimplemented");
306 }
307
VisitCoreturnStmt(CoreturnStmt * S)308 void ASTStmtWriter::VisitCoreturnStmt(CoreturnStmt *S) {
309 // FIXME: Implement coroutine serialization.
310 llvm_unreachable("unimplemented");
311 }
312
VisitCoawaitExpr(CoawaitExpr * S)313 void ASTStmtWriter::VisitCoawaitExpr(CoawaitExpr *S) {
314 // FIXME: Implement coroutine serialization.
315 llvm_unreachable("unimplemented");
316 }
317
VisitCoyieldExpr(CoyieldExpr * S)318 void ASTStmtWriter::VisitCoyieldExpr(CoyieldExpr *S) {
319 // FIXME: Implement coroutine serialization.
320 llvm_unreachable("unimplemented");
321 }
322
VisitCapturedStmt(CapturedStmt * S)323 void ASTStmtWriter::VisitCapturedStmt(CapturedStmt *S) {
324 VisitStmt(S);
325 // NumCaptures
326 Record.push_back(std::distance(S->capture_begin(), S->capture_end()));
327
328 // CapturedDecl and captured region kind
329 Record.AddDeclRef(S->getCapturedDecl());
330 Record.push_back(S->getCapturedRegionKind());
331
332 Record.AddDeclRef(S->getCapturedRecordDecl());
333
334 // Capture inits
335 for (auto *I : S->capture_inits())
336 Record.AddStmt(I);
337
338 // Body
339 Record.AddStmt(S->getCapturedStmt());
340
341 // Captures
342 for (const auto &I : S->captures()) {
343 if (I.capturesThis() || I.capturesVariableArrayType())
344 Record.AddDeclRef(nullptr);
345 else
346 Record.AddDeclRef(I.getCapturedVar());
347 Record.push_back(I.getCaptureKind());
348 Record.AddSourceLocation(I.getLocation());
349 }
350
351 Code = serialization::STMT_CAPTURED;
352 }
353
VisitExpr(Expr * E)354 void ASTStmtWriter::VisitExpr(Expr *E) {
355 VisitStmt(E);
356 Record.AddTypeRef(E->getType());
357 Record.push_back(E->isTypeDependent());
358 Record.push_back(E->isValueDependent());
359 Record.push_back(E->isInstantiationDependent());
360 Record.push_back(E->containsUnexpandedParameterPack());
361 Record.push_back(E->getValueKind());
362 Record.push_back(E->getObjectKind());
363 }
364
VisitPredefinedExpr(PredefinedExpr * E)365 void ASTStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
366 VisitExpr(E);
367 Record.AddSourceLocation(E->getLocation());
368 Record.push_back(E->getIdentType()); // FIXME: stable encoding
369 Record.AddStmt(E->getFunctionName());
370 Code = serialization::EXPR_PREDEFINED;
371 }
372
VisitDeclRefExpr(DeclRefExpr * E)373 void ASTStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
374 VisitExpr(E);
375
376 Record.push_back(E->hasQualifier());
377 Record.push_back(E->getDecl() != E->getFoundDecl());
378 Record.push_back(E->hasTemplateKWAndArgsInfo());
379 Record.push_back(E->hadMultipleCandidates());
380 Record.push_back(E->refersToEnclosingVariableOrCapture());
381
382 if (E->hasTemplateKWAndArgsInfo()) {
383 unsigned NumTemplateArgs = E->getNumTemplateArgs();
384 Record.push_back(NumTemplateArgs);
385 }
386
387 DeclarationName::NameKind nk = (E->getDecl()->getDeclName().getNameKind());
388
389 if ((!E->hasTemplateKWAndArgsInfo()) && (!E->hasQualifier()) &&
390 (E->getDecl() == E->getFoundDecl()) &&
391 nk == DeclarationName::Identifier) {
392 AbbrevToUse = Writer.getDeclRefExprAbbrev();
393 }
394
395 if (E->hasQualifier())
396 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
397
398 if (E->getDecl() != E->getFoundDecl())
399 Record.AddDeclRef(E->getFoundDecl());
400
401 if (E->hasTemplateKWAndArgsInfo())
402 AddTemplateKWAndArgsInfo(*E->getTrailingObjects<ASTTemplateKWAndArgsInfo>(),
403 E->getTrailingObjects<TemplateArgumentLoc>());
404
405 Record.AddDeclRef(E->getDecl());
406 Record.AddSourceLocation(E->getLocation());
407 Record.AddDeclarationNameLoc(E->DNLoc, E->getDecl()->getDeclName());
408 Code = serialization::EXPR_DECL_REF;
409 }
410
VisitIntegerLiteral(IntegerLiteral * E)411 void ASTStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
412 VisitExpr(E);
413 Record.AddSourceLocation(E->getLocation());
414 Record.AddAPInt(E->getValue());
415
416 if (E->getValue().getBitWidth() == 32) {
417 AbbrevToUse = Writer.getIntegerLiteralAbbrev();
418 }
419
420 Code = serialization::EXPR_INTEGER_LITERAL;
421 }
422
VisitFloatingLiteral(FloatingLiteral * E)423 void ASTStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
424 VisitExpr(E);
425 Record.push_back(E->getRawSemantics());
426 Record.push_back(E->isExact());
427 Record.AddAPFloat(E->getValue());
428 Record.AddSourceLocation(E->getLocation());
429 Code = serialization::EXPR_FLOATING_LITERAL;
430 }
431
VisitImaginaryLiteral(ImaginaryLiteral * E)432 void ASTStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
433 VisitExpr(E);
434 Record.AddStmt(E->getSubExpr());
435 Code = serialization::EXPR_IMAGINARY_LITERAL;
436 }
437
VisitStringLiteral(StringLiteral * E)438 void ASTStmtWriter::VisitStringLiteral(StringLiteral *E) {
439 VisitExpr(E);
440 Record.push_back(E->getByteLength());
441 Record.push_back(E->getNumConcatenated());
442 Record.push_back(E->getKind());
443 Record.push_back(E->isPascal());
444 // FIXME: String data should be stored as a blob at the end of the
445 // StringLiteral. However, we can't do so now because we have no
446 // provision for coping with abbreviations when we're jumping around
447 // the AST file during deserialization.
448 Record.append(E->getBytes().begin(), E->getBytes().end());
449 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
450 Record.AddSourceLocation(E->getStrTokenLoc(I));
451 Code = serialization::EXPR_STRING_LITERAL;
452 }
453
VisitCharacterLiteral(CharacterLiteral * E)454 void ASTStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
455 VisitExpr(E);
456 Record.push_back(E->getValue());
457 Record.AddSourceLocation(E->getLocation());
458 Record.push_back(E->getKind());
459
460 AbbrevToUse = Writer.getCharacterLiteralAbbrev();
461
462 Code = serialization::EXPR_CHARACTER_LITERAL;
463 }
464
VisitParenExpr(ParenExpr * E)465 void ASTStmtWriter::VisitParenExpr(ParenExpr *E) {
466 VisitExpr(E);
467 Record.AddSourceLocation(E->getLParen());
468 Record.AddSourceLocation(E->getRParen());
469 Record.AddStmt(E->getSubExpr());
470 Code = serialization::EXPR_PAREN;
471 }
472
VisitParenListExpr(ParenListExpr * E)473 void ASTStmtWriter::VisitParenListExpr(ParenListExpr *E) {
474 VisitExpr(E);
475 Record.push_back(E->NumExprs);
476 for (unsigned i=0; i != E->NumExprs; ++i)
477 Record.AddStmt(E->Exprs[i]);
478 Record.AddSourceLocation(E->LParenLoc);
479 Record.AddSourceLocation(E->RParenLoc);
480 Code = serialization::EXPR_PAREN_LIST;
481 }
482
VisitUnaryOperator(UnaryOperator * E)483 void ASTStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
484 VisitExpr(E);
485 Record.AddStmt(E->getSubExpr());
486 Record.push_back(E->getOpcode()); // FIXME: stable encoding
487 Record.AddSourceLocation(E->getOperatorLoc());
488 Code = serialization::EXPR_UNARY_OPERATOR;
489 }
490
VisitOffsetOfExpr(OffsetOfExpr * E)491 void ASTStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
492 VisitExpr(E);
493 Record.push_back(E->getNumComponents());
494 Record.push_back(E->getNumExpressions());
495 Record.AddSourceLocation(E->getOperatorLoc());
496 Record.AddSourceLocation(E->getRParenLoc());
497 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
498 for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
499 const OffsetOfNode &ON = E->getComponent(I);
500 Record.push_back(ON.getKind()); // FIXME: Stable encoding
501 Record.AddSourceLocation(ON.getSourceRange().getBegin());
502 Record.AddSourceLocation(ON.getSourceRange().getEnd());
503 switch (ON.getKind()) {
504 case OffsetOfNode::Array:
505 Record.push_back(ON.getArrayExprIndex());
506 break;
507
508 case OffsetOfNode::Field:
509 Record.AddDeclRef(ON.getField());
510 break;
511
512 case OffsetOfNode::Identifier:
513 Record.AddIdentifierRef(ON.getFieldName());
514 break;
515
516 case OffsetOfNode::Base:
517 Record.AddCXXBaseSpecifier(*ON.getBase());
518 break;
519 }
520 }
521 for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
522 Record.AddStmt(E->getIndexExpr(I));
523 Code = serialization::EXPR_OFFSETOF;
524 }
525
VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr * E)526 void ASTStmtWriter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E) {
527 VisitExpr(E);
528 Record.push_back(E->getKind());
529 if (E->isArgumentType())
530 Record.AddTypeSourceInfo(E->getArgumentTypeInfo());
531 else {
532 Record.push_back(0);
533 Record.AddStmt(E->getArgumentExpr());
534 }
535 Record.AddSourceLocation(E->getOperatorLoc());
536 Record.AddSourceLocation(E->getRParenLoc());
537 Code = serialization::EXPR_SIZEOF_ALIGN_OF;
538 }
539
VisitArraySubscriptExpr(ArraySubscriptExpr * E)540 void ASTStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
541 VisitExpr(E);
542 Record.AddStmt(E->getLHS());
543 Record.AddStmt(E->getRHS());
544 Record.AddSourceLocation(E->getRBracketLoc());
545 Code = serialization::EXPR_ARRAY_SUBSCRIPT;
546 }
547
VisitOMPArraySectionExpr(OMPArraySectionExpr * E)548 void ASTStmtWriter::VisitOMPArraySectionExpr(OMPArraySectionExpr *E) {
549 VisitExpr(E);
550 Record.AddStmt(E->getBase());
551 Record.AddStmt(E->getLowerBound());
552 Record.AddStmt(E->getLength());
553 Record.AddSourceLocation(E->getColonLoc());
554 Record.AddSourceLocation(E->getRBracketLoc());
555 Code = serialization::EXPR_OMP_ARRAY_SECTION;
556 }
557
VisitCallExpr(CallExpr * E)558 void ASTStmtWriter::VisitCallExpr(CallExpr *E) {
559 VisitExpr(E);
560 Record.push_back(E->getNumArgs());
561 Record.AddSourceLocation(E->getRParenLoc());
562 Record.AddStmt(E->getCallee());
563 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
564 Arg != ArgEnd; ++Arg)
565 Record.AddStmt(*Arg);
566 Code = serialization::EXPR_CALL;
567 }
568
VisitMemberExpr(MemberExpr * E)569 void ASTStmtWriter::VisitMemberExpr(MemberExpr *E) {
570 // Don't call VisitExpr, we'll write everything here.
571
572 Record.push_back(E->hasQualifier());
573 if (E->hasQualifier())
574 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
575
576 Record.push_back(E->HasTemplateKWAndArgsInfo);
577 if (E->HasTemplateKWAndArgsInfo) {
578 Record.AddSourceLocation(E->getTemplateKeywordLoc());
579 unsigned NumTemplateArgs = E->getNumTemplateArgs();
580 Record.push_back(NumTemplateArgs);
581 Record.AddSourceLocation(E->getLAngleLoc());
582 Record.AddSourceLocation(E->getRAngleLoc());
583 for (unsigned i=0; i != NumTemplateArgs; ++i)
584 Record.AddTemplateArgumentLoc(E->getTemplateArgs()[i]);
585 }
586
587 Record.push_back(E->hadMultipleCandidates());
588
589 DeclAccessPair FoundDecl = E->getFoundDecl();
590 Record.AddDeclRef(FoundDecl.getDecl());
591 Record.push_back(FoundDecl.getAccess());
592
593 Record.AddTypeRef(E->getType());
594 Record.push_back(E->getValueKind());
595 Record.push_back(E->getObjectKind());
596 Record.AddStmt(E->getBase());
597 Record.AddDeclRef(E->getMemberDecl());
598 Record.AddSourceLocation(E->getMemberLoc());
599 Record.push_back(E->isArrow());
600 Record.AddSourceLocation(E->getOperatorLoc());
601 Record.AddDeclarationNameLoc(E->MemberDNLoc,
602 E->getMemberDecl()->getDeclName());
603 Code = serialization::EXPR_MEMBER;
604 }
605
VisitObjCIsaExpr(ObjCIsaExpr * E)606 void ASTStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
607 VisitExpr(E);
608 Record.AddStmt(E->getBase());
609 Record.AddSourceLocation(E->getIsaMemberLoc());
610 Record.AddSourceLocation(E->getOpLoc());
611 Record.push_back(E->isArrow());
612 Code = serialization::EXPR_OBJC_ISA;
613 }
614
615 void ASTStmtWriter::
VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr * E)616 VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) {
617 VisitExpr(E);
618 Record.AddStmt(E->getSubExpr());
619 Record.push_back(E->shouldCopy());
620 Code = serialization::EXPR_OBJC_INDIRECT_COPY_RESTORE;
621 }
622
VisitObjCBridgedCastExpr(ObjCBridgedCastExpr * E)623 void ASTStmtWriter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) {
624 VisitExplicitCastExpr(E);
625 Record.AddSourceLocation(E->getLParenLoc());
626 Record.AddSourceLocation(E->getBridgeKeywordLoc());
627 Record.push_back(E->getBridgeKind()); // FIXME: Stable encoding
628 Code = serialization::EXPR_OBJC_BRIDGED_CAST;
629 }
630
VisitCastExpr(CastExpr * E)631 void ASTStmtWriter::VisitCastExpr(CastExpr *E) {
632 VisitExpr(E);
633 Record.push_back(E->path_size());
634 Record.AddStmt(E->getSubExpr());
635 Record.push_back(E->getCastKind()); // FIXME: stable encoding
636
637 for (CastExpr::path_iterator
638 PI = E->path_begin(), PE = E->path_end(); PI != PE; ++PI)
639 Record.AddCXXBaseSpecifier(**PI);
640 }
641
VisitBinaryOperator(BinaryOperator * E)642 void ASTStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
643 VisitExpr(E);
644 Record.AddStmt(E->getLHS());
645 Record.AddStmt(E->getRHS());
646 Record.push_back(E->getOpcode()); // FIXME: stable encoding
647 Record.AddSourceLocation(E->getOperatorLoc());
648 Record.push_back(E->isFPContractable());
649 Code = serialization::EXPR_BINARY_OPERATOR;
650 }
651
VisitCompoundAssignOperator(CompoundAssignOperator * E)652 void ASTStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
653 VisitBinaryOperator(E);
654 Record.AddTypeRef(E->getComputationLHSType());
655 Record.AddTypeRef(E->getComputationResultType());
656 Code = serialization::EXPR_COMPOUND_ASSIGN_OPERATOR;
657 }
658
VisitConditionalOperator(ConditionalOperator * E)659 void ASTStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
660 VisitExpr(E);
661 Record.AddStmt(E->getCond());
662 Record.AddStmt(E->getLHS());
663 Record.AddStmt(E->getRHS());
664 Record.AddSourceLocation(E->getQuestionLoc());
665 Record.AddSourceLocation(E->getColonLoc());
666 Code = serialization::EXPR_CONDITIONAL_OPERATOR;
667 }
668
669 void
VisitBinaryConditionalOperator(BinaryConditionalOperator * E)670 ASTStmtWriter::VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
671 VisitExpr(E);
672 Record.AddStmt(E->getOpaqueValue());
673 Record.AddStmt(E->getCommon());
674 Record.AddStmt(E->getCond());
675 Record.AddStmt(E->getTrueExpr());
676 Record.AddStmt(E->getFalseExpr());
677 Record.AddSourceLocation(E->getQuestionLoc());
678 Record.AddSourceLocation(E->getColonLoc());
679 Code = serialization::EXPR_BINARY_CONDITIONAL_OPERATOR;
680 }
681
VisitImplicitCastExpr(ImplicitCastExpr * E)682 void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
683 VisitCastExpr(E);
684
685 if (E->path_size() == 0)
686 AbbrevToUse = Writer.getExprImplicitCastAbbrev();
687
688 Code = serialization::EXPR_IMPLICIT_CAST;
689 }
690
VisitExplicitCastExpr(ExplicitCastExpr * E)691 void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
692 VisitCastExpr(E);
693 Record.AddTypeSourceInfo(E->getTypeInfoAsWritten());
694 }
695
VisitCStyleCastExpr(CStyleCastExpr * E)696 void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
697 VisitExplicitCastExpr(E);
698 Record.AddSourceLocation(E->getLParenLoc());
699 Record.AddSourceLocation(E->getRParenLoc());
700 Code = serialization::EXPR_CSTYLE_CAST;
701 }
702
VisitCompoundLiteralExpr(CompoundLiteralExpr * E)703 void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
704 VisitExpr(E);
705 Record.AddSourceLocation(E->getLParenLoc());
706 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
707 Record.AddStmt(E->getInitializer());
708 Record.push_back(E->isFileScope());
709 Code = serialization::EXPR_COMPOUND_LITERAL;
710 }
711
VisitExtVectorElementExpr(ExtVectorElementExpr * E)712 void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
713 VisitExpr(E);
714 Record.AddStmt(E->getBase());
715 Record.AddIdentifierRef(&E->getAccessor());
716 Record.AddSourceLocation(E->getAccessorLoc());
717 Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
718 }
719
VisitInitListExpr(InitListExpr * E)720 void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
721 VisitExpr(E);
722 // NOTE: only add the (possibly null) syntactic form.
723 // No need to serialize the isSemanticForm flag and the semantic form.
724 Record.AddStmt(E->getSyntacticForm());
725 Record.AddSourceLocation(E->getLBraceLoc());
726 Record.AddSourceLocation(E->getRBraceLoc());
727 bool isArrayFiller = E->ArrayFillerOrUnionFieldInit.is<Expr*>();
728 Record.push_back(isArrayFiller);
729 if (isArrayFiller)
730 Record.AddStmt(E->getArrayFiller());
731 else
732 Record.AddDeclRef(E->getInitializedFieldInUnion());
733 Record.push_back(E->hadArrayRangeDesignator());
734 Record.push_back(E->getNumInits());
735 if (isArrayFiller) {
736 // ArrayFiller may have filled "holes" due to designated initializer.
737 // Replace them by 0 to indicate that the filler goes in that place.
738 Expr *filler = E->getArrayFiller();
739 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
740 Record.AddStmt(E->getInit(I) != filler ? E->getInit(I) : nullptr);
741 } else {
742 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
743 Record.AddStmt(E->getInit(I));
744 }
745 Code = serialization::EXPR_INIT_LIST;
746 }
747
VisitDesignatedInitExpr(DesignatedInitExpr * E)748 void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
749 VisitExpr(E);
750 Record.push_back(E->getNumSubExprs());
751 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
752 Record.AddStmt(E->getSubExpr(I));
753 Record.AddSourceLocation(E->getEqualOrColonLoc());
754 Record.push_back(E->usesGNUSyntax());
755 for (const DesignatedInitExpr::Designator &D : E->designators()) {
756 if (D.isFieldDesignator()) {
757 if (FieldDecl *Field = D.getField()) {
758 Record.push_back(serialization::DESIG_FIELD_DECL);
759 Record.AddDeclRef(Field);
760 } else {
761 Record.push_back(serialization::DESIG_FIELD_NAME);
762 Record.AddIdentifierRef(D.getFieldName());
763 }
764 Record.AddSourceLocation(D.getDotLoc());
765 Record.AddSourceLocation(D.getFieldLoc());
766 } else if (D.isArrayDesignator()) {
767 Record.push_back(serialization::DESIG_ARRAY);
768 Record.push_back(D.getFirstExprIndex());
769 Record.AddSourceLocation(D.getLBracketLoc());
770 Record.AddSourceLocation(D.getRBracketLoc());
771 } else {
772 assert(D.isArrayRangeDesignator() && "Unknown designator");
773 Record.push_back(serialization::DESIG_ARRAY_RANGE);
774 Record.push_back(D.getFirstExprIndex());
775 Record.AddSourceLocation(D.getLBracketLoc());
776 Record.AddSourceLocation(D.getEllipsisLoc());
777 Record.AddSourceLocation(D.getRBracketLoc());
778 }
779 }
780 Code = serialization::EXPR_DESIGNATED_INIT;
781 }
782
VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr * E)783 void ASTStmtWriter::VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr *E) {
784 VisitExpr(E);
785 Record.AddStmt(E->getBase());
786 Record.AddStmt(E->getUpdater());
787 Code = serialization::EXPR_DESIGNATED_INIT_UPDATE;
788 }
789
VisitNoInitExpr(NoInitExpr * E)790 void ASTStmtWriter::VisitNoInitExpr(NoInitExpr *E) {
791 VisitExpr(E);
792 Code = serialization::EXPR_NO_INIT;
793 }
794
VisitImplicitValueInitExpr(ImplicitValueInitExpr * E)795 void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
796 VisitExpr(E);
797 Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
798 }
799
VisitVAArgExpr(VAArgExpr * E)800 void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
801 VisitExpr(E);
802 Record.AddStmt(E->getSubExpr());
803 Record.AddTypeSourceInfo(E->getWrittenTypeInfo());
804 Record.AddSourceLocation(E->getBuiltinLoc());
805 Record.AddSourceLocation(E->getRParenLoc());
806 Record.push_back(E->isMicrosoftABI());
807 Code = serialization::EXPR_VA_ARG;
808 }
809
VisitAddrLabelExpr(AddrLabelExpr * E)810 void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
811 VisitExpr(E);
812 Record.AddSourceLocation(E->getAmpAmpLoc());
813 Record.AddSourceLocation(E->getLabelLoc());
814 Record.AddDeclRef(E->getLabel());
815 Code = serialization::EXPR_ADDR_LABEL;
816 }
817
VisitStmtExpr(StmtExpr * E)818 void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
819 VisitExpr(E);
820 Record.AddStmt(E->getSubStmt());
821 Record.AddSourceLocation(E->getLParenLoc());
822 Record.AddSourceLocation(E->getRParenLoc());
823 Code = serialization::EXPR_STMT;
824 }
825
VisitChooseExpr(ChooseExpr * E)826 void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
827 VisitExpr(E);
828 Record.AddStmt(E->getCond());
829 Record.AddStmt(E->getLHS());
830 Record.AddStmt(E->getRHS());
831 Record.AddSourceLocation(E->getBuiltinLoc());
832 Record.AddSourceLocation(E->getRParenLoc());
833 Record.push_back(E->isConditionDependent() ? false : E->isConditionTrue());
834 Code = serialization::EXPR_CHOOSE;
835 }
836
VisitGNUNullExpr(GNUNullExpr * E)837 void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
838 VisitExpr(E);
839 Record.AddSourceLocation(E->getTokenLocation());
840 Code = serialization::EXPR_GNU_NULL;
841 }
842
VisitShuffleVectorExpr(ShuffleVectorExpr * E)843 void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
844 VisitExpr(E);
845 Record.push_back(E->getNumSubExprs());
846 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
847 Record.AddStmt(E->getExpr(I));
848 Record.AddSourceLocation(E->getBuiltinLoc());
849 Record.AddSourceLocation(E->getRParenLoc());
850 Code = serialization::EXPR_SHUFFLE_VECTOR;
851 }
852
VisitConvertVectorExpr(ConvertVectorExpr * E)853 void ASTStmtWriter::VisitConvertVectorExpr(ConvertVectorExpr *E) {
854 VisitExpr(E);
855 Record.AddSourceLocation(E->getBuiltinLoc());
856 Record.AddSourceLocation(E->getRParenLoc());
857 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
858 Record.AddStmt(E->getSrcExpr());
859 Code = serialization::EXPR_CONVERT_VECTOR;
860 }
861
VisitBlockExpr(BlockExpr * E)862 void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
863 VisitExpr(E);
864 Record.AddDeclRef(E->getBlockDecl());
865 Code = serialization::EXPR_BLOCK;
866 }
867
VisitGenericSelectionExpr(GenericSelectionExpr * E)868 void ASTStmtWriter::VisitGenericSelectionExpr(GenericSelectionExpr *E) {
869 VisitExpr(E);
870 Record.push_back(E->getNumAssocs());
871
872 Record.AddStmt(E->getControllingExpr());
873 for (unsigned I = 0, N = E->getNumAssocs(); I != N; ++I) {
874 Record.AddTypeSourceInfo(E->getAssocTypeSourceInfo(I));
875 Record.AddStmt(E->getAssocExpr(I));
876 }
877 Record.push_back(E->isResultDependent() ? -1U : E->getResultIndex());
878
879 Record.AddSourceLocation(E->getGenericLoc());
880 Record.AddSourceLocation(E->getDefaultLoc());
881 Record.AddSourceLocation(E->getRParenLoc());
882 Code = serialization::EXPR_GENERIC_SELECTION;
883 }
884
VisitPseudoObjectExpr(PseudoObjectExpr * E)885 void ASTStmtWriter::VisitPseudoObjectExpr(PseudoObjectExpr *E) {
886 VisitExpr(E);
887 Record.push_back(E->getNumSemanticExprs());
888
889 // Push the result index. Currently, this needs to exactly match
890 // the encoding used internally for ResultIndex.
891 unsigned result = E->getResultExprIndex();
892 result = (result == PseudoObjectExpr::NoResult ? 0 : result + 1);
893 Record.push_back(result);
894
895 Record.AddStmt(E->getSyntacticForm());
896 for (PseudoObjectExpr::semantics_iterator
897 i = E->semantics_begin(), e = E->semantics_end(); i != e; ++i) {
898 Record.AddStmt(*i);
899 }
900 Code = serialization::EXPR_PSEUDO_OBJECT;
901 }
902
VisitAtomicExpr(AtomicExpr * E)903 void ASTStmtWriter::VisitAtomicExpr(AtomicExpr *E) {
904 VisitExpr(E);
905 Record.push_back(E->getOp());
906 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
907 Record.AddStmt(E->getSubExprs()[I]);
908 Record.AddSourceLocation(E->getBuiltinLoc());
909 Record.AddSourceLocation(E->getRParenLoc());
910 Code = serialization::EXPR_ATOMIC;
911 }
912
913 //===----------------------------------------------------------------------===//
914 // Objective-C Expressions and Statements.
915 //===----------------------------------------------------------------------===//
916
VisitObjCStringLiteral(ObjCStringLiteral * E)917 void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
918 VisitExpr(E);
919 Record.AddStmt(E->getString());
920 Record.AddSourceLocation(E->getAtLoc());
921 Code = serialization::EXPR_OBJC_STRING_LITERAL;
922 }
923
VisitObjCBoxedExpr(ObjCBoxedExpr * E)924 void ASTStmtWriter::VisitObjCBoxedExpr(ObjCBoxedExpr *E) {
925 VisitExpr(E);
926 Record.AddStmt(E->getSubExpr());
927 Record.AddDeclRef(E->getBoxingMethod());
928 Record.AddSourceRange(E->getSourceRange());
929 Code = serialization::EXPR_OBJC_BOXED_EXPRESSION;
930 }
931
VisitObjCArrayLiteral(ObjCArrayLiteral * E)932 void ASTStmtWriter::VisitObjCArrayLiteral(ObjCArrayLiteral *E) {
933 VisitExpr(E);
934 Record.push_back(E->getNumElements());
935 for (unsigned i = 0; i < E->getNumElements(); i++)
936 Record.AddStmt(E->getElement(i));
937 Record.AddDeclRef(E->getArrayWithObjectsMethod());
938 Record.AddSourceRange(E->getSourceRange());
939 Code = serialization::EXPR_OBJC_ARRAY_LITERAL;
940 }
941
VisitObjCDictionaryLiteral(ObjCDictionaryLiteral * E)942 void ASTStmtWriter::VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) {
943 VisitExpr(E);
944 Record.push_back(E->getNumElements());
945 Record.push_back(E->HasPackExpansions);
946 for (unsigned i = 0; i < E->getNumElements(); i++) {
947 ObjCDictionaryElement Element = E->getKeyValueElement(i);
948 Record.AddStmt(Element.Key);
949 Record.AddStmt(Element.Value);
950 if (E->HasPackExpansions) {
951 Record.AddSourceLocation(Element.EllipsisLoc);
952 unsigned NumExpansions = 0;
953 if (Element.NumExpansions)
954 NumExpansions = *Element.NumExpansions + 1;
955 Record.push_back(NumExpansions);
956 }
957 }
958
959 Record.AddDeclRef(E->getDictWithObjectsMethod());
960 Record.AddSourceRange(E->getSourceRange());
961 Code = serialization::EXPR_OBJC_DICTIONARY_LITERAL;
962 }
963
VisitObjCEncodeExpr(ObjCEncodeExpr * E)964 void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
965 VisitExpr(E);
966 Record.AddTypeSourceInfo(E->getEncodedTypeSourceInfo());
967 Record.AddSourceLocation(E->getAtLoc());
968 Record.AddSourceLocation(E->getRParenLoc());
969 Code = serialization::EXPR_OBJC_ENCODE;
970 }
971
VisitObjCSelectorExpr(ObjCSelectorExpr * E)972 void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
973 VisitExpr(E);
974 Record.AddSelectorRef(E->getSelector());
975 Record.AddSourceLocation(E->getAtLoc());
976 Record.AddSourceLocation(E->getRParenLoc());
977 Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
978 }
979
VisitObjCProtocolExpr(ObjCProtocolExpr * E)980 void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
981 VisitExpr(E);
982 Record.AddDeclRef(E->getProtocol());
983 Record.AddSourceLocation(E->getAtLoc());
984 Record.AddSourceLocation(E->ProtoLoc);
985 Record.AddSourceLocation(E->getRParenLoc());
986 Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
987 }
988
VisitObjCIvarRefExpr(ObjCIvarRefExpr * E)989 void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
990 VisitExpr(E);
991 Record.AddDeclRef(E->getDecl());
992 Record.AddSourceLocation(E->getLocation());
993 Record.AddSourceLocation(E->getOpLoc());
994 Record.AddStmt(E->getBase());
995 Record.push_back(E->isArrow());
996 Record.push_back(E->isFreeIvar());
997 Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
998 }
999
VisitObjCPropertyRefExpr(ObjCPropertyRefExpr * E)1000 void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
1001 VisitExpr(E);
1002 Record.push_back(E->SetterAndMethodRefFlags.getInt());
1003 Record.push_back(E->isImplicitProperty());
1004 if (E->isImplicitProperty()) {
1005 Record.AddDeclRef(E->getImplicitPropertyGetter());
1006 Record.AddDeclRef(E->getImplicitPropertySetter());
1007 } else {
1008 Record.AddDeclRef(E->getExplicitProperty());
1009 }
1010 Record.AddSourceLocation(E->getLocation());
1011 Record.AddSourceLocation(E->getReceiverLocation());
1012 if (E->isObjectReceiver()) {
1013 Record.push_back(0);
1014 Record.AddStmt(E->getBase());
1015 } else if (E->isSuperReceiver()) {
1016 Record.push_back(1);
1017 Record.AddTypeRef(E->getSuperReceiverType());
1018 } else {
1019 Record.push_back(2);
1020 Record.AddDeclRef(E->getClassReceiver());
1021 }
1022
1023 Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
1024 }
1025
VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr * E)1026 void ASTStmtWriter::VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr *E) {
1027 VisitExpr(E);
1028 Record.AddSourceLocation(E->getRBracket());
1029 Record.AddStmt(E->getBaseExpr());
1030 Record.AddStmt(E->getKeyExpr());
1031 Record.AddDeclRef(E->getAtIndexMethodDecl());
1032 Record.AddDeclRef(E->setAtIndexMethodDecl());
1033
1034 Code = serialization::EXPR_OBJC_SUBSCRIPT_REF_EXPR;
1035 }
1036
VisitObjCMessageExpr(ObjCMessageExpr * E)1037 void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
1038 VisitExpr(E);
1039 Record.push_back(E->getNumArgs());
1040 Record.push_back(E->getNumStoredSelLocs());
1041 Record.push_back(E->SelLocsKind);
1042 Record.push_back(E->isDelegateInitCall());
1043 Record.push_back(E->IsImplicit);
1044 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
1045 switch (E->getReceiverKind()) {
1046 case ObjCMessageExpr::Instance:
1047 Record.AddStmt(E->getInstanceReceiver());
1048 break;
1049
1050 case ObjCMessageExpr::Class:
1051 Record.AddTypeSourceInfo(E->getClassReceiverTypeInfo());
1052 break;
1053
1054 case ObjCMessageExpr::SuperClass:
1055 case ObjCMessageExpr::SuperInstance:
1056 Record.AddTypeRef(E->getSuperType());
1057 Record.AddSourceLocation(E->getSuperLoc());
1058 break;
1059 }
1060
1061 if (E->getMethodDecl()) {
1062 Record.push_back(1);
1063 Record.AddDeclRef(E->getMethodDecl());
1064 } else {
1065 Record.push_back(0);
1066 Record.AddSelectorRef(E->getSelector());
1067 }
1068
1069 Record.AddSourceLocation(E->getLeftLoc());
1070 Record.AddSourceLocation(E->getRightLoc());
1071
1072 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
1073 Arg != ArgEnd; ++Arg)
1074 Record.AddStmt(*Arg);
1075
1076 SourceLocation *Locs = E->getStoredSelLocs();
1077 for (unsigned i = 0, e = E->getNumStoredSelLocs(); i != e; ++i)
1078 Record.AddSourceLocation(Locs[i]);
1079
1080 Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
1081 }
1082
VisitObjCForCollectionStmt(ObjCForCollectionStmt * S)1083 void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
1084 VisitStmt(S);
1085 Record.AddStmt(S->getElement());
1086 Record.AddStmt(S->getCollection());
1087 Record.AddStmt(S->getBody());
1088 Record.AddSourceLocation(S->getForLoc());
1089 Record.AddSourceLocation(S->getRParenLoc());
1090 Code = serialization::STMT_OBJC_FOR_COLLECTION;
1091 }
1092
VisitObjCAtCatchStmt(ObjCAtCatchStmt * S)1093 void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
1094 Record.AddStmt(S->getCatchBody());
1095 Record.AddDeclRef(S->getCatchParamDecl());
1096 Record.AddSourceLocation(S->getAtCatchLoc());
1097 Record.AddSourceLocation(S->getRParenLoc());
1098 Code = serialization::STMT_OBJC_CATCH;
1099 }
1100
VisitObjCAtFinallyStmt(ObjCAtFinallyStmt * S)1101 void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
1102 Record.AddStmt(S->getFinallyBody());
1103 Record.AddSourceLocation(S->getAtFinallyLoc());
1104 Code = serialization::STMT_OBJC_FINALLY;
1105 }
1106
VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt * S)1107 void ASTStmtWriter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *S) {
1108 Record.AddStmt(S->getSubStmt());
1109 Record.AddSourceLocation(S->getAtLoc());
1110 Code = serialization::STMT_OBJC_AUTORELEASE_POOL;
1111 }
1112
VisitObjCAtTryStmt(ObjCAtTryStmt * S)1113 void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
1114 Record.push_back(S->getNumCatchStmts());
1115 Record.push_back(S->getFinallyStmt() != nullptr);
1116 Record.AddStmt(S->getTryBody());
1117 for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
1118 Record.AddStmt(S->getCatchStmt(I));
1119 if (S->getFinallyStmt())
1120 Record.AddStmt(S->getFinallyStmt());
1121 Record.AddSourceLocation(S->getAtTryLoc());
1122 Code = serialization::STMT_OBJC_AT_TRY;
1123 }
1124
VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt * S)1125 void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
1126 Record.AddStmt(S->getSynchExpr());
1127 Record.AddStmt(S->getSynchBody());
1128 Record.AddSourceLocation(S->getAtSynchronizedLoc());
1129 Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
1130 }
1131
VisitObjCAtThrowStmt(ObjCAtThrowStmt * S)1132 void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
1133 Record.AddStmt(S->getThrowExpr());
1134 Record.AddSourceLocation(S->getThrowLoc());
1135 Code = serialization::STMT_OBJC_AT_THROW;
1136 }
1137
VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr * E)1138 void ASTStmtWriter::VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr *E) {
1139 VisitExpr(E);
1140 Record.push_back(E->getValue());
1141 Record.AddSourceLocation(E->getLocation());
1142 Code = serialization::EXPR_OBJC_BOOL_LITERAL;
1143 }
1144
1145 //===----------------------------------------------------------------------===//
1146 // C++ Expressions and Statements.
1147 //===----------------------------------------------------------------------===//
1148
VisitCXXCatchStmt(CXXCatchStmt * S)1149 void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
1150 VisitStmt(S);
1151 Record.AddSourceLocation(S->getCatchLoc());
1152 Record.AddDeclRef(S->getExceptionDecl());
1153 Record.AddStmt(S->getHandlerBlock());
1154 Code = serialization::STMT_CXX_CATCH;
1155 }
1156
VisitCXXTryStmt(CXXTryStmt * S)1157 void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
1158 VisitStmt(S);
1159 Record.push_back(S->getNumHandlers());
1160 Record.AddSourceLocation(S->getTryLoc());
1161 Record.AddStmt(S->getTryBlock());
1162 for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
1163 Record.AddStmt(S->getHandler(i));
1164 Code = serialization::STMT_CXX_TRY;
1165 }
1166
VisitCXXForRangeStmt(CXXForRangeStmt * S)1167 void ASTStmtWriter::VisitCXXForRangeStmt(CXXForRangeStmt *S) {
1168 VisitStmt(S);
1169 Record.AddSourceLocation(S->getForLoc());
1170 Record.AddSourceLocation(S->getCoawaitLoc());
1171 Record.AddSourceLocation(S->getColonLoc());
1172 Record.AddSourceLocation(S->getRParenLoc());
1173 Record.AddStmt(S->getRangeStmt());
1174 Record.AddStmt(S->getBeginStmt());
1175 Record.AddStmt(S->getEndStmt());
1176 Record.AddStmt(S->getCond());
1177 Record.AddStmt(S->getInc());
1178 Record.AddStmt(S->getLoopVarStmt());
1179 Record.AddStmt(S->getBody());
1180 Code = serialization::STMT_CXX_FOR_RANGE;
1181 }
1182
VisitMSDependentExistsStmt(MSDependentExistsStmt * S)1183 void ASTStmtWriter::VisitMSDependentExistsStmt(MSDependentExistsStmt *S) {
1184 VisitStmt(S);
1185 Record.AddSourceLocation(S->getKeywordLoc());
1186 Record.push_back(S->isIfExists());
1187 Record.AddNestedNameSpecifierLoc(S->getQualifierLoc());
1188 Record.AddDeclarationNameInfo(S->getNameInfo());
1189 Record.AddStmt(S->getSubStmt());
1190 Code = serialization::STMT_MS_DEPENDENT_EXISTS;
1191 }
1192
VisitCXXOperatorCallExpr(CXXOperatorCallExpr * E)1193 void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
1194 VisitCallExpr(E);
1195 Record.push_back(E->getOperator());
1196 Record.AddSourceRange(E->Range);
1197 Record.push_back(E->isFPContractable());
1198 Code = serialization::EXPR_CXX_OPERATOR_CALL;
1199 }
1200
VisitCXXMemberCallExpr(CXXMemberCallExpr * E)1201 void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
1202 VisitCallExpr(E);
1203 Code = serialization::EXPR_CXX_MEMBER_CALL;
1204 }
1205
VisitCXXConstructExpr(CXXConstructExpr * E)1206 void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
1207 VisitExpr(E);
1208 Record.push_back(E->getNumArgs());
1209 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
1210 Record.AddStmt(E->getArg(I));
1211 Record.AddDeclRef(E->getConstructor());
1212 Record.AddSourceLocation(E->getLocation());
1213 Record.push_back(E->isElidable());
1214 Record.push_back(E->hadMultipleCandidates());
1215 Record.push_back(E->isListInitialization());
1216 Record.push_back(E->isStdInitListInitialization());
1217 Record.push_back(E->requiresZeroInitialization());
1218 Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
1219 Record.AddSourceRange(E->getParenOrBraceRange());
1220 Code = serialization::EXPR_CXX_CONSTRUCT;
1221 }
1222
VisitCXXInheritedCtorInitExpr(CXXInheritedCtorInitExpr * E)1223 void ASTStmtWriter::VisitCXXInheritedCtorInitExpr(CXXInheritedCtorInitExpr *E) {
1224 VisitExpr(E);
1225 Record.AddDeclRef(E->getConstructor());
1226 Record.AddSourceLocation(E->getLocation());
1227 Record.push_back(E->constructsVBase());
1228 Record.push_back(E->inheritedFromVBase());
1229 Code = serialization::EXPR_CXX_INHERITED_CTOR_INIT;
1230 }
1231
VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr * E)1232 void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
1233 VisitCXXConstructExpr(E);
1234 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1235 Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
1236 }
1237
VisitLambdaExpr(LambdaExpr * E)1238 void ASTStmtWriter::VisitLambdaExpr(LambdaExpr *E) {
1239 VisitExpr(E);
1240 Record.push_back(E->NumCaptures);
1241 unsigned NumArrayIndexVars = 0;
1242 if (E->HasArrayIndexVars)
1243 NumArrayIndexVars = E->getArrayIndexStarts()[E->NumCaptures];
1244 Record.push_back(NumArrayIndexVars);
1245 Record.AddSourceRange(E->IntroducerRange);
1246 Record.push_back(E->CaptureDefault); // FIXME: stable encoding
1247 Record.AddSourceLocation(E->CaptureDefaultLoc);
1248 Record.push_back(E->ExplicitParams);
1249 Record.push_back(E->ExplicitResultType);
1250 Record.AddSourceLocation(E->ClosingBrace);
1251
1252 // Add capture initializers.
1253 for (LambdaExpr::capture_init_iterator C = E->capture_init_begin(),
1254 CEnd = E->capture_init_end();
1255 C != CEnd; ++C) {
1256 Record.AddStmt(*C);
1257 }
1258
1259 // Add array index variables, if any.
1260 if (NumArrayIndexVars) {
1261 Record.append(E->getArrayIndexStarts(),
1262 E->getArrayIndexStarts() + E->NumCaptures + 1);
1263 VarDecl **ArrayIndexVars = E->getArrayIndexVars();
1264 for (unsigned I = 0; I != NumArrayIndexVars; ++I)
1265 Record.AddDeclRef(ArrayIndexVars[I]);
1266 }
1267
1268 Code = serialization::EXPR_LAMBDA;
1269 }
1270
VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr * E)1271 void ASTStmtWriter::VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E) {
1272 VisitExpr(E);
1273 Record.AddStmt(E->getSubExpr());
1274 Code = serialization::EXPR_CXX_STD_INITIALIZER_LIST;
1275 }
1276
VisitCXXNamedCastExpr(CXXNamedCastExpr * E)1277 void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
1278 VisitExplicitCastExpr(E);
1279 Record.AddSourceRange(SourceRange(E->getOperatorLoc(), E->getRParenLoc()));
1280 Record.AddSourceRange(E->getAngleBrackets());
1281 }
1282
VisitCXXStaticCastExpr(CXXStaticCastExpr * E)1283 void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
1284 VisitCXXNamedCastExpr(E);
1285 Code = serialization::EXPR_CXX_STATIC_CAST;
1286 }
1287
VisitCXXDynamicCastExpr(CXXDynamicCastExpr * E)1288 void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
1289 VisitCXXNamedCastExpr(E);
1290 Code = serialization::EXPR_CXX_DYNAMIC_CAST;
1291 }
1292
VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr * E)1293 void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
1294 VisitCXXNamedCastExpr(E);
1295 Code = serialization::EXPR_CXX_REINTERPRET_CAST;
1296 }
1297
VisitCXXConstCastExpr(CXXConstCastExpr * E)1298 void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
1299 VisitCXXNamedCastExpr(E);
1300 Code = serialization::EXPR_CXX_CONST_CAST;
1301 }
1302
VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr * E)1303 void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
1304 VisitExplicitCastExpr(E);
1305 Record.AddSourceLocation(E->getLParenLoc());
1306 Record.AddSourceLocation(E->getRParenLoc());
1307 Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
1308 }
1309
VisitUserDefinedLiteral(UserDefinedLiteral * E)1310 void ASTStmtWriter::VisitUserDefinedLiteral(UserDefinedLiteral *E) {
1311 VisitCallExpr(E);
1312 Record.AddSourceLocation(E->UDSuffixLoc);
1313 Code = serialization::EXPR_USER_DEFINED_LITERAL;
1314 }
1315
VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr * E)1316 void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
1317 VisitExpr(E);
1318 Record.push_back(E->getValue());
1319 Record.AddSourceLocation(E->getLocation());
1320 Code = serialization::EXPR_CXX_BOOL_LITERAL;
1321 }
1322
VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr * E)1323 void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
1324 VisitExpr(E);
1325 Record.AddSourceLocation(E->getLocation());
1326 Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
1327 }
1328
VisitCXXTypeidExpr(CXXTypeidExpr * E)1329 void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1330 VisitExpr(E);
1331 Record.AddSourceRange(E->getSourceRange());
1332 if (E->isTypeOperand()) {
1333 Record.AddTypeSourceInfo(E->getTypeOperandSourceInfo());
1334 Code = serialization::EXPR_CXX_TYPEID_TYPE;
1335 } else {
1336 Record.AddStmt(E->getExprOperand());
1337 Code = serialization::EXPR_CXX_TYPEID_EXPR;
1338 }
1339 }
1340
VisitCXXThisExpr(CXXThisExpr * E)1341 void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
1342 VisitExpr(E);
1343 Record.AddSourceLocation(E->getLocation());
1344 Record.push_back(E->isImplicit());
1345 Code = serialization::EXPR_CXX_THIS;
1346 }
1347
VisitCXXThrowExpr(CXXThrowExpr * E)1348 void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
1349 VisitExpr(E);
1350 Record.AddSourceLocation(E->getThrowLoc());
1351 Record.AddStmt(E->getSubExpr());
1352 Record.push_back(E->isThrownVariableInScope());
1353 Code = serialization::EXPR_CXX_THROW;
1354 }
1355
VisitCXXDefaultArgExpr(CXXDefaultArgExpr * E)1356 void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
1357 VisitExpr(E);
1358 Record.AddDeclRef(E->getParam());
1359 Record.AddSourceLocation(E->getUsedLocation());
1360 Code = serialization::EXPR_CXX_DEFAULT_ARG;
1361 }
1362
VisitCXXDefaultInitExpr(CXXDefaultInitExpr * E)1363 void ASTStmtWriter::VisitCXXDefaultInitExpr(CXXDefaultInitExpr *E) {
1364 VisitExpr(E);
1365 Record.AddDeclRef(E->getField());
1366 Record.AddSourceLocation(E->getExprLoc());
1367 Code = serialization::EXPR_CXX_DEFAULT_INIT;
1368 }
1369
VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr * E)1370 void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
1371 VisitExpr(E);
1372 Record.AddCXXTemporary(E->getTemporary());
1373 Record.AddStmt(E->getSubExpr());
1374 Code = serialization::EXPR_CXX_BIND_TEMPORARY;
1375 }
1376
VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr * E)1377 void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
1378 VisitExpr(E);
1379 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1380 Record.AddSourceLocation(E->getRParenLoc());
1381 Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
1382 }
1383
VisitCXXNewExpr(CXXNewExpr * E)1384 void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
1385 VisitExpr(E);
1386 Record.push_back(E->isGlobalNew());
1387 Record.push_back(E->isArray());
1388 Record.push_back(E->doesUsualArrayDeleteWantSize());
1389 Record.push_back(E->getNumPlacementArgs());
1390 Record.push_back(E->StoredInitializationStyle);
1391 Record.AddDeclRef(E->getOperatorNew());
1392 Record.AddDeclRef(E->getOperatorDelete());
1393 Record.AddTypeSourceInfo(E->getAllocatedTypeSourceInfo());
1394 Record.AddSourceRange(E->getTypeIdParens());
1395 Record.AddSourceRange(E->getSourceRange());
1396 Record.AddSourceRange(E->getDirectInitRange());
1397 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), e = E->raw_arg_end();
1398 I != e; ++I)
1399 Record.AddStmt(*I);
1400
1401 Code = serialization::EXPR_CXX_NEW;
1402 }
1403
VisitCXXDeleteExpr(CXXDeleteExpr * E)1404 void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
1405 VisitExpr(E);
1406 Record.push_back(E->isGlobalDelete());
1407 Record.push_back(E->isArrayForm());
1408 Record.push_back(E->isArrayFormAsWritten());
1409 Record.push_back(E->doesUsualArrayDeleteWantSize());
1410 Record.AddDeclRef(E->getOperatorDelete());
1411 Record.AddStmt(E->getArgument());
1412 Record.AddSourceLocation(E->getSourceRange().getBegin());
1413
1414 Code = serialization::EXPR_CXX_DELETE;
1415 }
1416
VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr * E)1417 void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1418 VisitExpr(E);
1419
1420 Record.AddStmt(E->getBase());
1421 Record.push_back(E->isArrow());
1422 Record.AddSourceLocation(E->getOperatorLoc());
1423 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1424 Record.AddTypeSourceInfo(E->getScopeTypeInfo());
1425 Record.AddSourceLocation(E->getColonColonLoc());
1426 Record.AddSourceLocation(E->getTildeLoc());
1427
1428 // PseudoDestructorTypeStorage.
1429 Record.AddIdentifierRef(E->getDestroyedTypeIdentifier());
1430 if (E->getDestroyedTypeIdentifier())
1431 Record.AddSourceLocation(E->getDestroyedTypeLoc());
1432 else
1433 Record.AddTypeSourceInfo(E->getDestroyedTypeInfo());
1434
1435 Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
1436 }
1437
VisitExprWithCleanups(ExprWithCleanups * E)1438 void ASTStmtWriter::VisitExprWithCleanups(ExprWithCleanups *E) {
1439 VisitExpr(E);
1440 Record.push_back(E->getNumObjects());
1441 for (unsigned i = 0, e = E->getNumObjects(); i != e; ++i)
1442 Record.AddDeclRef(E->getObject(i));
1443
1444 Record.push_back(E->cleanupsHaveSideEffects());
1445 Record.AddStmt(E->getSubExpr());
1446 Code = serialization::EXPR_EXPR_WITH_CLEANUPS;
1447 }
1448
1449 void
VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr * E)1450 ASTStmtWriter::VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E){
1451 VisitExpr(E);
1452
1453 // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1454 // emitted first.
1455
1456 Record.push_back(E->HasTemplateKWAndArgsInfo);
1457 if (E->HasTemplateKWAndArgsInfo) {
1458 const ASTTemplateKWAndArgsInfo &ArgInfo =
1459 *E->getTrailingObjects<ASTTemplateKWAndArgsInfo>();
1460 Record.push_back(ArgInfo.NumTemplateArgs);
1461 AddTemplateKWAndArgsInfo(ArgInfo,
1462 E->getTrailingObjects<TemplateArgumentLoc>());
1463 }
1464
1465 if (!E->isImplicitAccess())
1466 Record.AddStmt(E->getBase());
1467 else
1468 Record.AddStmt(nullptr);
1469 Record.AddTypeRef(E->getBaseType());
1470 Record.push_back(E->isArrow());
1471 Record.AddSourceLocation(E->getOperatorLoc());
1472 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1473 Record.AddDeclRef(E->getFirstQualifierFoundInScope());
1474 Record.AddDeclarationNameInfo(E->MemberNameInfo);
1475 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
1476 }
1477
1478 void
VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr * E)1479 ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1480 VisitExpr(E);
1481
1482 // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1483 // emitted first.
1484
1485 Record.push_back(E->HasTemplateKWAndArgsInfo);
1486 if (E->HasTemplateKWAndArgsInfo) {
1487 const ASTTemplateKWAndArgsInfo &ArgInfo =
1488 *E->getTrailingObjects<ASTTemplateKWAndArgsInfo>();
1489 Record.push_back(ArgInfo.NumTemplateArgs);
1490 AddTemplateKWAndArgsInfo(ArgInfo,
1491 E->getTrailingObjects<TemplateArgumentLoc>());
1492 }
1493
1494 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1495 Record.AddDeclarationNameInfo(E->NameInfo);
1496 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
1497 }
1498
1499 void
VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr * E)1500 ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
1501 VisitExpr(E);
1502 Record.push_back(E->arg_size());
1503 for (CXXUnresolvedConstructExpr::arg_iterator
1504 ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
1505 Record.AddStmt(*ArgI);
1506 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1507 Record.AddSourceLocation(E->getLParenLoc());
1508 Record.AddSourceLocation(E->getRParenLoc());
1509 Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
1510 }
1511
VisitOverloadExpr(OverloadExpr * E)1512 void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
1513 VisitExpr(E);
1514
1515 // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1516 // emitted first.
1517
1518 Record.push_back(E->HasTemplateKWAndArgsInfo);
1519 if (E->HasTemplateKWAndArgsInfo) {
1520 const ASTTemplateKWAndArgsInfo &ArgInfo =
1521 *E->getTrailingASTTemplateKWAndArgsInfo();
1522 Record.push_back(ArgInfo.NumTemplateArgs);
1523 AddTemplateKWAndArgsInfo(ArgInfo, E->getTrailingTemplateArgumentLoc());
1524 }
1525
1526 Record.push_back(E->getNumDecls());
1527 for (OverloadExpr::decls_iterator
1528 OvI = E->decls_begin(), OvE = E->decls_end(); OvI != OvE; ++OvI) {
1529 Record.AddDeclRef(OvI.getDecl());
1530 Record.push_back(OvI.getAccess());
1531 }
1532
1533 Record.AddDeclarationNameInfo(E->NameInfo);
1534 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1535 }
1536
VisitUnresolvedMemberExpr(UnresolvedMemberExpr * E)1537 void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
1538 VisitOverloadExpr(E);
1539 Record.push_back(E->isArrow());
1540 Record.push_back(E->hasUnresolvedUsing());
1541 Record.AddStmt(!E->isImplicitAccess() ? E->getBase() : nullptr);
1542 Record.AddTypeRef(E->getBaseType());
1543 Record.AddSourceLocation(E->getOperatorLoc());
1544 Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
1545 }
1546
VisitUnresolvedLookupExpr(UnresolvedLookupExpr * E)1547 void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
1548 VisitOverloadExpr(E);
1549 Record.push_back(E->requiresADL());
1550 Record.push_back(E->isOverloaded());
1551 Record.AddDeclRef(E->getNamingClass());
1552 Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
1553 }
1554
VisitTypeTraitExpr(TypeTraitExpr * E)1555 void ASTStmtWriter::VisitTypeTraitExpr(TypeTraitExpr *E) {
1556 VisitExpr(E);
1557 Record.push_back(E->TypeTraitExprBits.NumArgs);
1558 Record.push_back(E->TypeTraitExprBits.Kind); // FIXME: Stable encoding
1559 Record.push_back(E->TypeTraitExprBits.Value);
1560 Record.AddSourceRange(E->getSourceRange());
1561 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
1562 Record.AddTypeSourceInfo(E->getArg(I));
1563 Code = serialization::EXPR_TYPE_TRAIT;
1564 }
1565
VisitArrayTypeTraitExpr(ArrayTypeTraitExpr * E)1566 void ASTStmtWriter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
1567 VisitExpr(E);
1568 Record.push_back(E->getTrait());
1569 Record.push_back(E->getValue());
1570 Record.AddSourceRange(E->getSourceRange());
1571 Record.AddTypeSourceInfo(E->getQueriedTypeSourceInfo());
1572 Code = serialization::EXPR_ARRAY_TYPE_TRAIT;
1573 }
1574
VisitExpressionTraitExpr(ExpressionTraitExpr * E)1575 void ASTStmtWriter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
1576 VisitExpr(E);
1577 Record.push_back(E->getTrait());
1578 Record.push_back(E->getValue());
1579 Record.AddSourceRange(E->getSourceRange());
1580 Record.AddStmt(E->getQueriedExpression());
1581 Code = serialization::EXPR_CXX_EXPRESSION_TRAIT;
1582 }
1583
VisitCXXNoexceptExpr(CXXNoexceptExpr * E)1584 void ASTStmtWriter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
1585 VisitExpr(E);
1586 Record.push_back(E->getValue());
1587 Record.AddSourceRange(E->getSourceRange());
1588 Record.AddStmt(E->getOperand());
1589 Code = serialization::EXPR_CXX_NOEXCEPT;
1590 }
1591
VisitPackExpansionExpr(PackExpansionExpr * E)1592 void ASTStmtWriter::VisitPackExpansionExpr(PackExpansionExpr *E) {
1593 VisitExpr(E);
1594 Record.AddSourceLocation(E->getEllipsisLoc());
1595 Record.push_back(E->NumExpansions);
1596 Record.AddStmt(E->getPattern());
1597 Code = serialization::EXPR_PACK_EXPANSION;
1598 }
1599
VisitSizeOfPackExpr(SizeOfPackExpr * E)1600 void ASTStmtWriter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
1601 VisitExpr(E);
1602 Record.push_back(E->isPartiallySubstituted() ? E->getPartialArguments().size()
1603 : 0);
1604 Record.AddSourceLocation(E->OperatorLoc);
1605 Record.AddSourceLocation(E->PackLoc);
1606 Record.AddSourceLocation(E->RParenLoc);
1607 Record.AddDeclRef(E->Pack);
1608 if (E->isPartiallySubstituted()) {
1609 for (const auto &TA : E->getPartialArguments())
1610 Record.AddTemplateArgument(TA);
1611 } else if (!E->isValueDependent()) {
1612 Record.push_back(E->getPackLength());
1613 }
1614 Code = serialization::EXPR_SIZEOF_PACK;
1615 }
1616
VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr * E)1617 void ASTStmtWriter::VisitSubstNonTypeTemplateParmExpr(
1618 SubstNonTypeTemplateParmExpr *E) {
1619 VisitExpr(E);
1620 Record.AddDeclRef(E->getParameter());
1621 Record.AddSourceLocation(E->getNameLoc());
1622 Record.AddStmt(E->getReplacement());
1623 Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM;
1624 }
1625
VisitSubstNonTypeTemplateParmPackExpr(SubstNonTypeTemplateParmPackExpr * E)1626 void ASTStmtWriter::VisitSubstNonTypeTemplateParmPackExpr(
1627 SubstNonTypeTemplateParmPackExpr *E) {
1628 VisitExpr(E);
1629 Record.AddDeclRef(E->getParameterPack());
1630 Record.AddTemplateArgument(E->getArgumentPack());
1631 Record.AddSourceLocation(E->getParameterPackLocation());
1632 Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM_PACK;
1633 }
1634
VisitFunctionParmPackExpr(FunctionParmPackExpr * E)1635 void ASTStmtWriter::VisitFunctionParmPackExpr(FunctionParmPackExpr *E) {
1636 VisitExpr(E);
1637 Record.push_back(E->getNumExpansions());
1638 Record.AddDeclRef(E->getParameterPack());
1639 Record.AddSourceLocation(E->getParameterPackLocation());
1640 for (FunctionParmPackExpr::iterator I = E->begin(), End = E->end();
1641 I != End; ++I)
1642 Record.AddDeclRef(*I);
1643 Code = serialization::EXPR_FUNCTION_PARM_PACK;
1644 }
1645
VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr * E)1646 void ASTStmtWriter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E) {
1647 VisitExpr(E);
1648 Record.AddStmt(E->getTemporary());
1649 Record.AddDeclRef(E->getExtendingDecl());
1650 Record.push_back(E->getManglingNumber());
1651 Code = serialization::EXPR_MATERIALIZE_TEMPORARY;
1652 }
1653
VisitCXXFoldExpr(CXXFoldExpr * E)1654 void ASTStmtWriter::VisitCXXFoldExpr(CXXFoldExpr *E) {
1655 VisitExpr(E);
1656 Record.AddSourceLocation(E->LParenLoc);
1657 Record.AddSourceLocation(E->EllipsisLoc);
1658 Record.AddSourceLocation(E->RParenLoc);
1659 Record.AddStmt(E->SubExprs[0]);
1660 Record.AddStmt(E->SubExprs[1]);
1661 Record.push_back(E->Opcode);
1662 Code = serialization::EXPR_CXX_FOLD;
1663 }
1664
VisitOpaqueValueExpr(OpaqueValueExpr * E)1665 void ASTStmtWriter::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
1666 VisitExpr(E);
1667 Record.AddStmt(E->getSourceExpr());
1668 Record.AddSourceLocation(E->getLocation());
1669 Code = serialization::EXPR_OPAQUE_VALUE;
1670 }
1671
VisitTypoExpr(TypoExpr * E)1672 void ASTStmtWriter::VisitTypoExpr(TypoExpr *E) {
1673 VisitExpr(E);
1674 // TODO: Figure out sane writer behavior for a TypoExpr, if necessary
1675 llvm_unreachable("Cannot write TypoExpr nodes");
1676 }
1677
1678 //===----------------------------------------------------------------------===//
1679 // CUDA Expressions and Statements.
1680 //===----------------------------------------------------------------------===//
1681
VisitCUDAKernelCallExpr(CUDAKernelCallExpr * E)1682 void ASTStmtWriter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *E) {
1683 VisitCallExpr(E);
1684 Record.AddStmt(E->getConfig());
1685 Code = serialization::EXPR_CUDA_KERNEL_CALL;
1686 }
1687
1688 //===----------------------------------------------------------------------===//
1689 // OpenCL Expressions and Statements.
1690 //===----------------------------------------------------------------------===//
VisitAsTypeExpr(AsTypeExpr * E)1691 void ASTStmtWriter::VisitAsTypeExpr(AsTypeExpr *E) {
1692 VisitExpr(E);
1693 Record.AddSourceLocation(E->getBuiltinLoc());
1694 Record.AddSourceLocation(E->getRParenLoc());
1695 Record.AddStmt(E->getSrcExpr());
1696 Code = serialization::EXPR_ASTYPE;
1697 }
1698
1699 //===----------------------------------------------------------------------===//
1700 // Microsoft Expressions and Statements.
1701 //===----------------------------------------------------------------------===//
VisitMSPropertyRefExpr(MSPropertyRefExpr * E)1702 void ASTStmtWriter::VisitMSPropertyRefExpr(MSPropertyRefExpr *E) {
1703 VisitExpr(E);
1704 Record.push_back(E->isArrow());
1705 Record.AddStmt(E->getBaseExpr());
1706 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1707 Record.AddSourceLocation(E->getMemberLoc());
1708 Record.AddDeclRef(E->getPropertyDecl());
1709 Code = serialization::EXPR_CXX_PROPERTY_REF_EXPR;
1710 }
1711
VisitMSPropertySubscriptExpr(MSPropertySubscriptExpr * E)1712 void ASTStmtWriter::VisitMSPropertySubscriptExpr(MSPropertySubscriptExpr *E) {
1713 VisitExpr(E);
1714 Record.AddStmt(E->getBase());
1715 Record.AddStmt(E->getIdx());
1716 Record.AddSourceLocation(E->getRBracketLoc());
1717 Code = serialization::EXPR_CXX_PROPERTY_SUBSCRIPT_EXPR;
1718 }
1719
VisitCXXUuidofExpr(CXXUuidofExpr * E)1720 void ASTStmtWriter::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
1721 VisitExpr(E);
1722 Record.AddSourceRange(E->getSourceRange());
1723 Record.AddString(E->getUuidStr());
1724 if (E->isTypeOperand()) {
1725 Record.AddTypeSourceInfo(E->getTypeOperandSourceInfo());
1726 Code = serialization::EXPR_CXX_UUIDOF_TYPE;
1727 } else {
1728 Record.AddStmt(E->getExprOperand());
1729 Code = serialization::EXPR_CXX_UUIDOF_EXPR;
1730 }
1731 }
1732
VisitSEHExceptStmt(SEHExceptStmt * S)1733 void ASTStmtWriter::VisitSEHExceptStmt(SEHExceptStmt *S) {
1734 VisitStmt(S);
1735 Record.AddSourceLocation(S->getExceptLoc());
1736 Record.AddStmt(S->getFilterExpr());
1737 Record.AddStmt(S->getBlock());
1738 Code = serialization::STMT_SEH_EXCEPT;
1739 }
1740
VisitSEHFinallyStmt(SEHFinallyStmt * S)1741 void ASTStmtWriter::VisitSEHFinallyStmt(SEHFinallyStmt *S) {
1742 VisitStmt(S);
1743 Record.AddSourceLocation(S->getFinallyLoc());
1744 Record.AddStmt(S->getBlock());
1745 Code = serialization::STMT_SEH_FINALLY;
1746 }
1747
VisitSEHTryStmt(SEHTryStmt * S)1748 void ASTStmtWriter::VisitSEHTryStmt(SEHTryStmt *S) {
1749 VisitStmt(S);
1750 Record.push_back(S->getIsCXXTry());
1751 Record.AddSourceLocation(S->getTryLoc());
1752 Record.AddStmt(S->getTryBlock());
1753 Record.AddStmt(S->getHandler());
1754 Code = serialization::STMT_SEH_TRY;
1755 }
1756
VisitSEHLeaveStmt(SEHLeaveStmt * S)1757 void ASTStmtWriter::VisitSEHLeaveStmt(SEHLeaveStmt *S) {
1758 VisitStmt(S);
1759 Record.AddSourceLocation(S->getLeaveLoc());
1760 Code = serialization::STMT_SEH_LEAVE;
1761 }
1762
1763 //===----------------------------------------------------------------------===//
1764 // OpenMP Clauses.
1765 //===----------------------------------------------------------------------===//
1766
1767 namespace clang {
1768 class OMPClauseWriter : public OMPClauseVisitor<OMPClauseWriter> {
1769 ASTRecordWriter &Record;
1770 public:
OMPClauseWriter(ASTRecordWriter & Record)1771 OMPClauseWriter(ASTRecordWriter &Record) : Record(Record) {}
1772 #define OPENMP_CLAUSE(Name, Class) \
1773 void Visit##Class(Class *S);
1774 #include "clang/Basic/OpenMPKinds.def"
1775 void writeClause(OMPClause *C);
1776 void VisitOMPClauseWithPreInit(OMPClauseWithPreInit *C);
1777 void VisitOMPClauseWithPostUpdate(OMPClauseWithPostUpdate *C);
1778 };
1779 }
1780
writeClause(OMPClause * C)1781 void OMPClauseWriter::writeClause(OMPClause *C) {
1782 Record.push_back(C->getClauseKind());
1783 Visit(C);
1784 Record.AddSourceLocation(C->getLocStart());
1785 Record.AddSourceLocation(C->getLocEnd());
1786 }
1787
VisitOMPClauseWithPreInit(OMPClauseWithPreInit * C)1788 void OMPClauseWriter::VisitOMPClauseWithPreInit(OMPClauseWithPreInit *C) {
1789 Record.AddStmt(C->getPreInitStmt());
1790 }
1791
VisitOMPClauseWithPostUpdate(OMPClauseWithPostUpdate * C)1792 void OMPClauseWriter::VisitOMPClauseWithPostUpdate(OMPClauseWithPostUpdate *C) {
1793 VisitOMPClauseWithPreInit(C);
1794 Record.AddStmt(C->getPostUpdateExpr());
1795 }
1796
VisitOMPIfClause(OMPIfClause * C)1797 void OMPClauseWriter::VisitOMPIfClause(OMPIfClause *C) {
1798 Record.push_back(C->getNameModifier());
1799 Record.AddSourceLocation(C->getNameModifierLoc());
1800 Record.AddSourceLocation(C->getColonLoc());
1801 Record.AddStmt(C->getCondition());
1802 Record.AddSourceLocation(C->getLParenLoc());
1803 }
1804
VisitOMPFinalClause(OMPFinalClause * C)1805 void OMPClauseWriter::VisitOMPFinalClause(OMPFinalClause *C) {
1806 Record.AddStmt(C->getCondition());
1807 Record.AddSourceLocation(C->getLParenLoc());
1808 }
1809
VisitOMPNumThreadsClause(OMPNumThreadsClause * C)1810 void OMPClauseWriter::VisitOMPNumThreadsClause(OMPNumThreadsClause *C) {
1811 Record.AddStmt(C->getNumThreads());
1812 Record.AddSourceLocation(C->getLParenLoc());
1813 }
1814
VisitOMPSafelenClause(OMPSafelenClause * C)1815 void OMPClauseWriter::VisitOMPSafelenClause(OMPSafelenClause *C) {
1816 Record.AddStmt(C->getSafelen());
1817 Record.AddSourceLocation(C->getLParenLoc());
1818 }
1819
VisitOMPSimdlenClause(OMPSimdlenClause * C)1820 void OMPClauseWriter::VisitOMPSimdlenClause(OMPSimdlenClause *C) {
1821 Record.AddStmt(C->getSimdlen());
1822 Record.AddSourceLocation(C->getLParenLoc());
1823 }
1824
VisitOMPCollapseClause(OMPCollapseClause * C)1825 void OMPClauseWriter::VisitOMPCollapseClause(OMPCollapseClause *C) {
1826 Record.AddStmt(C->getNumForLoops());
1827 Record.AddSourceLocation(C->getLParenLoc());
1828 }
1829
VisitOMPDefaultClause(OMPDefaultClause * C)1830 void OMPClauseWriter::VisitOMPDefaultClause(OMPDefaultClause *C) {
1831 Record.push_back(C->getDefaultKind());
1832 Record.AddSourceLocation(C->getLParenLoc());
1833 Record.AddSourceLocation(C->getDefaultKindKwLoc());
1834 }
1835
VisitOMPProcBindClause(OMPProcBindClause * C)1836 void OMPClauseWriter::VisitOMPProcBindClause(OMPProcBindClause *C) {
1837 Record.push_back(C->getProcBindKind());
1838 Record.AddSourceLocation(C->getLParenLoc());
1839 Record.AddSourceLocation(C->getProcBindKindKwLoc());
1840 }
1841
VisitOMPScheduleClause(OMPScheduleClause * C)1842 void OMPClauseWriter::VisitOMPScheduleClause(OMPScheduleClause *C) {
1843 VisitOMPClauseWithPreInit(C);
1844 Record.push_back(C->getScheduleKind());
1845 Record.push_back(C->getFirstScheduleModifier());
1846 Record.push_back(C->getSecondScheduleModifier());
1847 Record.AddStmt(C->getChunkSize());
1848 Record.AddSourceLocation(C->getLParenLoc());
1849 Record.AddSourceLocation(C->getFirstScheduleModifierLoc());
1850 Record.AddSourceLocation(C->getSecondScheduleModifierLoc());
1851 Record.AddSourceLocation(C->getScheduleKindLoc());
1852 Record.AddSourceLocation(C->getCommaLoc());
1853 }
1854
VisitOMPOrderedClause(OMPOrderedClause * C)1855 void OMPClauseWriter::VisitOMPOrderedClause(OMPOrderedClause *C) {
1856 Record.AddStmt(C->getNumForLoops());
1857 Record.AddSourceLocation(C->getLParenLoc());
1858 }
1859
VisitOMPNowaitClause(OMPNowaitClause *)1860 void OMPClauseWriter::VisitOMPNowaitClause(OMPNowaitClause *) {}
1861
VisitOMPUntiedClause(OMPUntiedClause *)1862 void OMPClauseWriter::VisitOMPUntiedClause(OMPUntiedClause *) {}
1863
VisitOMPMergeableClause(OMPMergeableClause *)1864 void OMPClauseWriter::VisitOMPMergeableClause(OMPMergeableClause *) {}
1865
VisitOMPReadClause(OMPReadClause *)1866 void OMPClauseWriter::VisitOMPReadClause(OMPReadClause *) {}
1867
VisitOMPWriteClause(OMPWriteClause *)1868 void OMPClauseWriter::VisitOMPWriteClause(OMPWriteClause *) {}
1869
VisitOMPUpdateClause(OMPUpdateClause *)1870 void OMPClauseWriter::VisitOMPUpdateClause(OMPUpdateClause *) {}
1871
VisitOMPCaptureClause(OMPCaptureClause *)1872 void OMPClauseWriter::VisitOMPCaptureClause(OMPCaptureClause *) {}
1873
VisitOMPSeqCstClause(OMPSeqCstClause *)1874 void OMPClauseWriter::VisitOMPSeqCstClause(OMPSeqCstClause *) {}
1875
VisitOMPThreadsClause(OMPThreadsClause *)1876 void OMPClauseWriter::VisitOMPThreadsClause(OMPThreadsClause *) {}
1877
VisitOMPSIMDClause(OMPSIMDClause *)1878 void OMPClauseWriter::VisitOMPSIMDClause(OMPSIMDClause *) {}
1879
VisitOMPNogroupClause(OMPNogroupClause *)1880 void OMPClauseWriter::VisitOMPNogroupClause(OMPNogroupClause *) {}
1881
VisitOMPPrivateClause(OMPPrivateClause * C)1882 void OMPClauseWriter::VisitOMPPrivateClause(OMPPrivateClause *C) {
1883 Record.push_back(C->varlist_size());
1884 Record.AddSourceLocation(C->getLParenLoc());
1885 for (auto *VE : C->varlists()) {
1886 Record.AddStmt(VE);
1887 }
1888 for (auto *VE : C->private_copies()) {
1889 Record.AddStmt(VE);
1890 }
1891 }
1892
VisitOMPFirstprivateClause(OMPFirstprivateClause * C)1893 void OMPClauseWriter::VisitOMPFirstprivateClause(OMPFirstprivateClause *C) {
1894 Record.push_back(C->varlist_size());
1895 VisitOMPClauseWithPreInit(C);
1896 Record.AddSourceLocation(C->getLParenLoc());
1897 for (auto *VE : C->varlists()) {
1898 Record.AddStmt(VE);
1899 }
1900 for (auto *VE : C->private_copies()) {
1901 Record.AddStmt(VE);
1902 }
1903 for (auto *VE : C->inits()) {
1904 Record.AddStmt(VE);
1905 }
1906 }
1907
VisitOMPLastprivateClause(OMPLastprivateClause * C)1908 void OMPClauseWriter::VisitOMPLastprivateClause(OMPLastprivateClause *C) {
1909 Record.push_back(C->varlist_size());
1910 VisitOMPClauseWithPostUpdate(C);
1911 Record.AddSourceLocation(C->getLParenLoc());
1912 for (auto *VE : C->varlists())
1913 Record.AddStmt(VE);
1914 for (auto *E : C->private_copies())
1915 Record.AddStmt(E);
1916 for (auto *E : C->source_exprs())
1917 Record.AddStmt(E);
1918 for (auto *E : C->destination_exprs())
1919 Record.AddStmt(E);
1920 for (auto *E : C->assignment_ops())
1921 Record.AddStmt(E);
1922 }
1923
VisitOMPSharedClause(OMPSharedClause * C)1924 void OMPClauseWriter::VisitOMPSharedClause(OMPSharedClause *C) {
1925 Record.push_back(C->varlist_size());
1926 Record.AddSourceLocation(C->getLParenLoc());
1927 for (auto *VE : C->varlists())
1928 Record.AddStmt(VE);
1929 }
1930
VisitOMPReductionClause(OMPReductionClause * C)1931 void OMPClauseWriter::VisitOMPReductionClause(OMPReductionClause *C) {
1932 Record.push_back(C->varlist_size());
1933 VisitOMPClauseWithPostUpdate(C);
1934 Record.AddSourceLocation(C->getLParenLoc());
1935 Record.AddSourceLocation(C->getColonLoc());
1936 Record.AddNestedNameSpecifierLoc(C->getQualifierLoc());
1937 Record.AddDeclarationNameInfo(C->getNameInfo());
1938 for (auto *VE : C->varlists())
1939 Record.AddStmt(VE);
1940 for (auto *VE : C->privates())
1941 Record.AddStmt(VE);
1942 for (auto *E : C->lhs_exprs())
1943 Record.AddStmt(E);
1944 for (auto *E : C->rhs_exprs())
1945 Record.AddStmt(E);
1946 for (auto *E : C->reduction_ops())
1947 Record.AddStmt(E);
1948 }
1949
VisitOMPLinearClause(OMPLinearClause * C)1950 void OMPClauseWriter::VisitOMPLinearClause(OMPLinearClause *C) {
1951 Record.push_back(C->varlist_size());
1952 VisitOMPClauseWithPostUpdate(C);
1953 Record.AddSourceLocation(C->getLParenLoc());
1954 Record.AddSourceLocation(C->getColonLoc());
1955 Record.push_back(C->getModifier());
1956 Record.AddSourceLocation(C->getModifierLoc());
1957 for (auto *VE : C->varlists()) {
1958 Record.AddStmt(VE);
1959 }
1960 for (auto *VE : C->privates()) {
1961 Record.AddStmt(VE);
1962 }
1963 for (auto *VE : C->inits()) {
1964 Record.AddStmt(VE);
1965 }
1966 for (auto *VE : C->updates()) {
1967 Record.AddStmt(VE);
1968 }
1969 for (auto *VE : C->finals()) {
1970 Record.AddStmt(VE);
1971 }
1972 Record.AddStmt(C->getStep());
1973 Record.AddStmt(C->getCalcStep());
1974 }
1975
VisitOMPAlignedClause(OMPAlignedClause * C)1976 void OMPClauseWriter::VisitOMPAlignedClause(OMPAlignedClause *C) {
1977 Record.push_back(C->varlist_size());
1978 Record.AddSourceLocation(C->getLParenLoc());
1979 Record.AddSourceLocation(C->getColonLoc());
1980 for (auto *VE : C->varlists())
1981 Record.AddStmt(VE);
1982 Record.AddStmt(C->getAlignment());
1983 }
1984
VisitOMPCopyinClause(OMPCopyinClause * C)1985 void OMPClauseWriter::VisitOMPCopyinClause(OMPCopyinClause *C) {
1986 Record.push_back(C->varlist_size());
1987 Record.AddSourceLocation(C->getLParenLoc());
1988 for (auto *VE : C->varlists())
1989 Record.AddStmt(VE);
1990 for (auto *E : C->source_exprs())
1991 Record.AddStmt(E);
1992 for (auto *E : C->destination_exprs())
1993 Record.AddStmt(E);
1994 for (auto *E : C->assignment_ops())
1995 Record.AddStmt(E);
1996 }
1997
VisitOMPCopyprivateClause(OMPCopyprivateClause * C)1998 void OMPClauseWriter::VisitOMPCopyprivateClause(OMPCopyprivateClause *C) {
1999 Record.push_back(C->varlist_size());
2000 Record.AddSourceLocation(C->getLParenLoc());
2001 for (auto *VE : C->varlists())
2002 Record.AddStmt(VE);
2003 for (auto *E : C->source_exprs())
2004 Record.AddStmt(E);
2005 for (auto *E : C->destination_exprs())
2006 Record.AddStmt(E);
2007 for (auto *E : C->assignment_ops())
2008 Record.AddStmt(E);
2009 }
2010
VisitOMPFlushClause(OMPFlushClause * C)2011 void OMPClauseWriter::VisitOMPFlushClause(OMPFlushClause *C) {
2012 Record.push_back(C->varlist_size());
2013 Record.AddSourceLocation(C->getLParenLoc());
2014 for (auto *VE : C->varlists())
2015 Record.AddStmt(VE);
2016 }
2017
VisitOMPDependClause(OMPDependClause * C)2018 void OMPClauseWriter::VisitOMPDependClause(OMPDependClause *C) {
2019 Record.push_back(C->varlist_size());
2020 Record.AddSourceLocation(C->getLParenLoc());
2021 Record.push_back(C->getDependencyKind());
2022 Record.AddSourceLocation(C->getDependencyLoc());
2023 Record.AddSourceLocation(C->getColonLoc());
2024 for (auto *VE : C->varlists())
2025 Record.AddStmt(VE);
2026 Record.AddStmt(C->getCounterValue());
2027 }
2028
VisitOMPDeviceClause(OMPDeviceClause * C)2029 void OMPClauseWriter::VisitOMPDeviceClause(OMPDeviceClause *C) {
2030 Record.AddStmt(C->getDevice());
2031 Record.AddSourceLocation(C->getLParenLoc());
2032 }
2033
VisitOMPMapClause(OMPMapClause * C)2034 void OMPClauseWriter::VisitOMPMapClause(OMPMapClause *C) {
2035 Record.push_back(C->varlist_size());
2036 Record.push_back(C->getUniqueDeclarationsNum());
2037 Record.push_back(C->getTotalComponentListNum());
2038 Record.push_back(C->getTotalComponentsNum());
2039 Record.AddSourceLocation(C->getLParenLoc());
2040 Record.push_back(C->getMapTypeModifier());
2041 Record.push_back(C->getMapType());
2042 Record.AddSourceLocation(C->getMapLoc());
2043 Record.AddSourceLocation(C->getColonLoc());
2044 for (auto *E : C->varlists())
2045 Record.AddStmt(E);
2046 for (auto *D : C->all_decls())
2047 Record.AddDeclRef(D);
2048 for (auto N : C->all_num_lists())
2049 Record.push_back(N);
2050 for (auto N : C->all_lists_sizes())
2051 Record.push_back(N);
2052 for (auto &M : C->all_components()) {
2053 Record.AddStmt(M.getAssociatedExpression());
2054 Record.AddDeclRef(M.getAssociatedDeclaration());
2055 }
2056 }
2057
VisitOMPNumTeamsClause(OMPNumTeamsClause * C)2058 void OMPClauseWriter::VisitOMPNumTeamsClause(OMPNumTeamsClause *C) {
2059 Record.AddStmt(C->getNumTeams());
2060 Record.AddSourceLocation(C->getLParenLoc());
2061 }
2062
VisitOMPThreadLimitClause(OMPThreadLimitClause * C)2063 void OMPClauseWriter::VisitOMPThreadLimitClause(OMPThreadLimitClause *C) {
2064 Record.AddStmt(C->getThreadLimit());
2065 Record.AddSourceLocation(C->getLParenLoc());
2066 }
2067
VisitOMPPriorityClause(OMPPriorityClause * C)2068 void OMPClauseWriter::VisitOMPPriorityClause(OMPPriorityClause *C) {
2069 Record.AddStmt(C->getPriority());
2070 Record.AddSourceLocation(C->getLParenLoc());
2071 }
2072
VisitOMPGrainsizeClause(OMPGrainsizeClause * C)2073 void OMPClauseWriter::VisitOMPGrainsizeClause(OMPGrainsizeClause *C) {
2074 Record.AddStmt(C->getGrainsize());
2075 Record.AddSourceLocation(C->getLParenLoc());
2076 }
2077
VisitOMPNumTasksClause(OMPNumTasksClause * C)2078 void OMPClauseWriter::VisitOMPNumTasksClause(OMPNumTasksClause *C) {
2079 Record.AddStmt(C->getNumTasks());
2080 Record.AddSourceLocation(C->getLParenLoc());
2081 }
2082
VisitOMPHintClause(OMPHintClause * C)2083 void OMPClauseWriter::VisitOMPHintClause(OMPHintClause *C) {
2084 Record.AddStmt(C->getHint());
2085 Record.AddSourceLocation(C->getLParenLoc());
2086 }
2087
VisitOMPDistScheduleClause(OMPDistScheduleClause * C)2088 void OMPClauseWriter::VisitOMPDistScheduleClause(OMPDistScheduleClause *C) {
2089 VisitOMPClauseWithPreInit(C);
2090 Record.push_back(C->getDistScheduleKind());
2091 Record.AddStmt(C->getChunkSize());
2092 Record.AddSourceLocation(C->getLParenLoc());
2093 Record.AddSourceLocation(C->getDistScheduleKindLoc());
2094 Record.AddSourceLocation(C->getCommaLoc());
2095 }
2096
VisitOMPDefaultmapClause(OMPDefaultmapClause * C)2097 void OMPClauseWriter::VisitOMPDefaultmapClause(OMPDefaultmapClause *C) {
2098 Record.push_back(C->getDefaultmapKind());
2099 Record.push_back(C->getDefaultmapModifier());
2100 Record.AddSourceLocation(C->getLParenLoc());
2101 Record.AddSourceLocation(C->getDefaultmapModifierLoc());
2102 Record.AddSourceLocation(C->getDefaultmapKindLoc());
2103 }
2104
VisitOMPToClause(OMPToClause * C)2105 void OMPClauseWriter::VisitOMPToClause(OMPToClause *C) {
2106 Record.push_back(C->varlist_size());
2107 Record.push_back(C->getUniqueDeclarationsNum());
2108 Record.push_back(C->getTotalComponentListNum());
2109 Record.push_back(C->getTotalComponentsNum());
2110 Record.AddSourceLocation(C->getLParenLoc());
2111 for (auto *E : C->varlists())
2112 Record.AddStmt(E);
2113 for (auto *D : C->all_decls())
2114 Record.AddDeclRef(D);
2115 for (auto N : C->all_num_lists())
2116 Record.push_back(N);
2117 for (auto N : C->all_lists_sizes())
2118 Record.push_back(N);
2119 for (auto &M : C->all_components()) {
2120 Record.AddStmt(M.getAssociatedExpression());
2121 Record.AddDeclRef(M.getAssociatedDeclaration());
2122 }
2123 }
2124
VisitOMPFromClause(OMPFromClause * C)2125 void OMPClauseWriter::VisitOMPFromClause(OMPFromClause *C) {
2126 Record.push_back(C->varlist_size());
2127 Record.push_back(C->getUniqueDeclarationsNum());
2128 Record.push_back(C->getTotalComponentListNum());
2129 Record.push_back(C->getTotalComponentsNum());
2130 Record.AddSourceLocation(C->getLParenLoc());
2131 for (auto *E : C->varlists())
2132 Record.AddStmt(E);
2133 for (auto *D : C->all_decls())
2134 Record.AddDeclRef(D);
2135 for (auto N : C->all_num_lists())
2136 Record.push_back(N);
2137 for (auto N : C->all_lists_sizes())
2138 Record.push_back(N);
2139 for (auto &M : C->all_components()) {
2140 Record.AddStmt(M.getAssociatedExpression());
2141 Record.AddDeclRef(M.getAssociatedDeclaration());
2142 }
2143 }
2144
VisitOMPUseDevicePtrClause(OMPUseDevicePtrClause * C)2145 void OMPClauseWriter::VisitOMPUseDevicePtrClause(OMPUseDevicePtrClause *C) {
2146 Record.push_back(C->varlist_size());
2147 Record.AddSourceLocation(C->getLParenLoc());
2148 for (auto *VE : C->varlists()) {
2149 Record.AddStmt(VE);
2150 }
2151 }
2152
VisitOMPIsDevicePtrClause(OMPIsDevicePtrClause * C)2153 void OMPClauseWriter::VisitOMPIsDevicePtrClause(OMPIsDevicePtrClause *C) {
2154 Record.push_back(C->varlist_size());
2155 Record.AddSourceLocation(C->getLParenLoc());
2156 for (auto *VE : C->varlists()) {
2157 Record.AddStmt(VE);
2158 }
2159 }
2160
2161 //===----------------------------------------------------------------------===//
2162 // OpenMP Directives.
2163 //===----------------------------------------------------------------------===//
VisitOMPExecutableDirective(OMPExecutableDirective * E)2164 void ASTStmtWriter::VisitOMPExecutableDirective(OMPExecutableDirective *E) {
2165 Record.AddSourceLocation(E->getLocStart());
2166 Record.AddSourceLocation(E->getLocEnd());
2167 OMPClauseWriter ClauseWriter(Record);
2168 for (unsigned i = 0; i < E->getNumClauses(); ++i) {
2169 ClauseWriter.writeClause(E->getClause(i));
2170 }
2171 if (E->hasAssociatedStmt())
2172 Record.AddStmt(E->getAssociatedStmt());
2173 }
2174
VisitOMPLoopDirective(OMPLoopDirective * D)2175 void ASTStmtWriter::VisitOMPLoopDirective(OMPLoopDirective *D) {
2176 VisitStmt(D);
2177 Record.push_back(D->getNumClauses());
2178 Record.push_back(D->getCollapsedNumber());
2179 VisitOMPExecutableDirective(D);
2180 Record.AddStmt(D->getIterationVariable());
2181 Record.AddStmt(D->getLastIteration());
2182 Record.AddStmt(D->getCalcLastIteration());
2183 Record.AddStmt(D->getPreCond());
2184 Record.AddStmt(D->getCond());
2185 Record.AddStmt(D->getInit());
2186 Record.AddStmt(D->getInc());
2187 Record.AddStmt(D->getPreInits());
2188 if (isOpenMPWorksharingDirective(D->getDirectiveKind()) ||
2189 isOpenMPTaskLoopDirective(D->getDirectiveKind()) ||
2190 isOpenMPDistributeDirective(D->getDirectiveKind())) {
2191 Record.AddStmt(D->getIsLastIterVariable());
2192 Record.AddStmt(D->getLowerBoundVariable());
2193 Record.AddStmt(D->getUpperBoundVariable());
2194 Record.AddStmt(D->getStrideVariable());
2195 Record.AddStmt(D->getEnsureUpperBound());
2196 Record.AddStmt(D->getNextLowerBound());
2197 Record.AddStmt(D->getNextUpperBound());
2198 Record.AddStmt(D->getNumIterations());
2199 }
2200 if (isOpenMPLoopBoundSharingDirective(D->getDirectiveKind())) {
2201 Record.AddStmt(D->getPrevLowerBoundVariable());
2202 Record.AddStmt(D->getPrevUpperBoundVariable());
2203 }
2204 for (auto I : D->counters()) {
2205 Record.AddStmt(I);
2206 }
2207 for (auto I : D->private_counters()) {
2208 Record.AddStmt(I);
2209 }
2210 for (auto I : D->inits()) {
2211 Record.AddStmt(I);
2212 }
2213 for (auto I : D->updates()) {
2214 Record.AddStmt(I);
2215 }
2216 for (auto I : D->finals()) {
2217 Record.AddStmt(I);
2218 }
2219 }
2220
VisitOMPParallelDirective(OMPParallelDirective * D)2221 void ASTStmtWriter::VisitOMPParallelDirective(OMPParallelDirective *D) {
2222 VisitStmt(D);
2223 Record.push_back(D->getNumClauses());
2224 VisitOMPExecutableDirective(D);
2225 Record.push_back(D->hasCancel() ? 1 : 0);
2226 Code = serialization::STMT_OMP_PARALLEL_DIRECTIVE;
2227 }
2228
VisitOMPSimdDirective(OMPSimdDirective * D)2229 void ASTStmtWriter::VisitOMPSimdDirective(OMPSimdDirective *D) {
2230 VisitOMPLoopDirective(D);
2231 Code = serialization::STMT_OMP_SIMD_DIRECTIVE;
2232 }
2233
VisitOMPForDirective(OMPForDirective * D)2234 void ASTStmtWriter::VisitOMPForDirective(OMPForDirective *D) {
2235 VisitOMPLoopDirective(D);
2236 Record.push_back(D->hasCancel() ? 1 : 0);
2237 Code = serialization::STMT_OMP_FOR_DIRECTIVE;
2238 }
2239
VisitOMPForSimdDirective(OMPForSimdDirective * D)2240 void ASTStmtWriter::VisitOMPForSimdDirective(OMPForSimdDirective *D) {
2241 VisitOMPLoopDirective(D);
2242 Code = serialization::STMT_OMP_FOR_SIMD_DIRECTIVE;
2243 }
2244
VisitOMPSectionsDirective(OMPSectionsDirective * D)2245 void ASTStmtWriter::VisitOMPSectionsDirective(OMPSectionsDirective *D) {
2246 VisitStmt(D);
2247 Record.push_back(D->getNumClauses());
2248 VisitOMPExecutableDirective(D);
2249 Record.push_back(D->hasCancel() ? 1 : 0);
2250 Code = serialization::STMT_OMP_SECTIONS_DIRECTIVE;
2251 }
2252
VisitOMPSectionDirective(OMPSectionDirective * D)2253 void ASTStmtWriter::VisitOMPSectionDirective(OMPSectionDirective *D) {
2254 VisitStmt(D);
2255 VisitOMPExecutableDirective(D);
2256 Record.push_back(D->hasCancel() ? 1 : 0);
2257 Code = serialization::STMT_OMP_SECTION_DIRECTIVE;
2258 }
2259
VisitOMPSingleDirective(OMPSingleDirective * D)2260 void ASTStmtWriter::VisitOMPSingleDirective(OMPSingleDirective *D) {
2261 VisitStmt(D);
2262 Record.push_back(D->getNumClauses());
2263 VisitOMPExecutableDirective(D);
2264 Code = serialization::STMT_OMP_SINGLE_DIRECTIVE;
2265 }
2266
VisitOMPMasterDirective(OMPMasterDirective * D)2267 void ASTStmtWriter::VisitOMPMasterDirective(OMPMasterDirective *D) {
2268 VisitStmt(D);
2269 VisitOMPExecutableDirective(D);
2270 Code = serialization::STMT_OMP_MASTER_DIRECTIVE;
2271 }
2272
VisitOMPCriticalDirective(OMPCriticalDirective * D)2273 void ASTStmtWriter::VisitOMPCriticalDirective(OMPCriticalDirective *D) {
2274 VisitStmt(D);
2275 Record.push_back(D->getNumClauses());
2276 VisitOMPExecutableDirective(D);
2277 Record.AddDeclarationNameInfo(D->getDirectiveName());
2278 Code = serialization::STMT_OMP_CRITICAL_DIRECTIVE;
2279 }
2280
VisitOMPParallelForDirective(OMPParallelForDirective * D)2281 void ASTStmtWriter::VisitOMPParallelForDirective(OMPParallelForDirective *D) {
2282 VisitOMPLoopDirective(D);
2283 Record.push_back(D->hasCancel() ? 1 : 0);
2284 Code = serialization::STMT_OMP_PARALLEL_FOR_DIRECTIVE;
2285 }
2286
VisitOMPParallelForSimdDirective(OMPParallelForSimdDirective * D)2287 void ASTStmtWriter::VisitOMPParallelForSimdDirective(
2288 OMPParallelForSimdDirective *D) {
2289 VisitOMPLoopDirective(D);
2290 Code = serialization::STMT_OMP_PARALLEL_FOR_SIMD_DIRECTIVE;
2291 }
2292
VisitOMPParallelSectionsDirective(OMPParallelSectionsDirective * D)2293 void ASTStmtWriter::VisitOMPParallelSectionsDirective(
2294 OMPParallelSectionsDirective *D) {
2295 VisitStmt(D);
2296 Record.push_back(D->getNumClauses());
2297 VisitOMPExecutableDirective(D);
2298 Record.push_back(D->hasCancel() ? 1 : 0);
2299 Code = serialization::STMT_OMP_PARALLEL_SECTIONS_DIRECTIVE;
2300 }
2301
VisitOMPTaskDirective(OMPTaskDirective * D)2302 void ASTStmtWriter::VisitOMPTaskDirective(OMPTaskDirective *D) {
2303 VisitStmt(D);
2304 Record.push_back(D->getNumClauses());
2305 VisitOMPExecutableDirective(D);
2306 Record.push_back(D->hasCancel() ? 1 : 0);
2307 Code = serialization::STMT_OMP_TASK_DIRECTIVE;
2308 }
2309
VisitOMPAtomicDirective(OMPAtomicDirective * D)2310 void ASTStmtWriter::VisitOMPAtomicDirective(OMPAtomicDirective *D) {
2311 VisitStmt(D);
2312 Record.push_back(D->getNumClauses());
2313 VisitOMPExecutableDirective(D);
2314 Record.AddStmt(D->getX());
2315 Record.AddStmt(D->getV());
2316 Record.AddStmt(D->getExpr());
2317 Record.AddStmt(D->getUpdateExpr());
2318 Record.push_back(D->isXLHSInRHSPart() ? 1 : 0);
2319 Record.push_back(D->isPostfixUpdate() ? 1 : 0);
2320 Code = serialization::STMT_OMP_ATOMIC_DIRECTIVE;
2321 }
2322
VisitOMPTargetDirective(OMPTargetDirective * D)2323 void ASTStmtWriter::VisitOMPTargetDirective(OMPTargetDirective *D) {
2324 VisitStmt(D);
2325 Record.push_back(D->getNumClauses());
2326 VisitOMPExecutableDirective(D);
2327 Code = serialization::STMT_OMP_TARGET_DIRECTIVE;
2328 }
2329
VisitOMPTargetDataDirective(OMPTargetDataDirective * D)2330 void ASTStmtWriter::VisitOMPTargetDataDirective(OMPTargetDataDirective *D) {
2331 VisitStmt(D);
2332 Record.push_back(D->getNumClauses());
2333 VisitOMPExecutableDirective(D);
2334 Code = serialization::STMT_OMP_TARGET_DATA_DIRECTIVE;
2335 }
2336
VisitOMPTargetEnterDataDirective(OMPTargetEnterDataDirective * D)2337 void ASTStmtWriter::VisitOMPTargetEnterDataDirective(
2338 OMPTargetEnterDataDirective *D) {
2339 VisitStmt(D);
2340 Record.push_back(D->getNumClauses());
2341 VisitOMPExecutableDirective(D);
2342 Code = serialization::STMT_OMP_TARGET_ENTER_DATA_DIRECTIVE;
2343 }
2344
VisitOMPTargetExitDataDirective(OMPTargetExitDataDirective * D)2345 void ASTStmtWriter::VisitOMPTargetExitDataDirective(
2346 OMPTargetExitDataDirective *D) {
2347 VisitStmt(D);
2348 Record.push_back(D->getNumClauses());
2349 VisitOMPExecutableDirective(D);
2350 Code = serialization::STMT_OMP_TARGET_EXIT_DATA_DIRECTIVE;
2351 }
2352
VisitOMPTargetParallelDirective(OMPTargetParallelDirective * D)2353 void ASTStmtWriter::VisitOMPTargetParallelDirective(
2354 OMPTargetParallelDirective *D) {
2355 VisitStmt(D);
2356 Record.push_back(D->getNumClauses());
2357 VisitOMPExecutableDirective(D);
2358 Code = serialization::STMT_OMP_TARGET_PARALLEL_DIRECTIVE;
2359 }
2360
VisitOMPTargetParallelForDirective(OMPTargetParallelForDirective * D)2361 void ASTStmtWriter::VisitOMPTargetParallelForDirective(
2362 OMPTargetParallelForDirective *D) {
2363 VisitOMPLoopDirective(D);
2364 Record.push_back(D->hasCancel() ? 1 : 0);
2365 Code = serialization::STMT_OMP_TARGET_PARALLEL_FOR_DIRECTIVE;
2366 }
2367
VisitOMPTaskyieldDirective(OMPTaskyieldDirective * D)2368 void ASTStmtWriter::VisitOMPTaskyieldDirective(OMPTaskyieldDirective *D) {
2369 VisitStmt(D);
2370 VisitOMPExecutableDirective(D);
2371 Code = serialization::STMT_OMP_TASKYIELD_DIRECTIVE;
2372 }
2373
VisitOMPBarrierDirective(OMPBarrierDirective * D)2374 void ASTStmtWriter::VisitOMPBarrierDirective(OMPBarrierDirective *D) {
2375 VisitStmt(D);
2376 VisitOMPExecutableDirective(D);
2377 Code = serialization::STMT_OMP_BARRIER_DIRECTIVE;
2378 }
2379
VisitOMPTaskwaitDirective(OMPTaskwaitDirective * D)2380 void ASTStmtWriter::VisitOMPTaskwaitDirective(OMPTaskwaitDirective *D) {
2381 VisitStmt(D);
2382 VisitOMPExecutableDirective(D);
2383 Code = serialization::STMT_OMP_TASKWAIT_DIRECTIVE;
2384 }
2385
VisitOMPTaskgroupDirective(OMPTaskgroupDirective * D)2386 void ASTStmtWriter::VisitOMPTaskgroupDirective(OMPTaskgroupDirective *D) {
2387 VisitStmt(D);
2388 VisitOMPExecutableDirective(D);
2389 Code = serialization::STMT_OMP_TASKGROUP_DIRECTIVE;
2390 }
2391
VisitOMPFlushDirective(OMPFlushDirective * D)2392 void ASTStmtWriter::VisitOMPFlushDirective(OMPFlushDirective *D) {
2393 VisitStmt(D);
2394 Record.push_back(D->getNumClauses());
2395 VisitOMPExecutableDirective(D);
2396 Code = serialization::STMT_OMP_FLUSH_DIRECTIVE;
2397 }
2398
VisitOMPOrderedDirective(OMPOrderedDirective * D)2399 void ASTStmtWriter::VisitOMPOrderedDirective(OMPOrderedDirective *D) {
2400 VisitStmt(D);
2401 Record.push_back(D->getNumClauses());
2402 VisitOMPExecutableDirective(D);
2403 Code = serialization::STMT_OMP_ORDERED_DIRECTIVE;
2404 }
2405
VisitOMPTeamsDirective(OMPTeamsDirective * D)2406 void ASTStmtWriter::VisitOMPTeamsDirective(OMPTeamsDirective *D) {
2407 VisitStmt(D);
2408 Record.push_back(D->getNumClauses());
2409 VisitOMPExecutableDirective(D);
2410 Code = serialization::STMT_OMP_TEAMS_DIRECTIVE;
2411 }
2412
VisitOMPCancellationPointDirective(OMPCancellationPointDirective * D)2413 void ASTStmtWriter::VisitOMPCancellationPointDirective(
2414 OMPCancellationPointDirective *D) {
2415 VisitStmt(D);
2416 VisitOMPExecutableDirective(D);
2417 Record.push_back(D->getCancelRegion());
2418 Code = serialization::STMT_OMP_CANCELLATION_POINT_DIRECTIVE;
2419 }
2420
VisitOMPCancelDirective(OMPCancelDirective * D)2421 void ASTStmtWriter::VisitOMPCancelDirective(OMPCancelDirective *D) {
2422 VisitStmt(D);
2423 Record.push_back(D->getNumClauses());
2424 VisitOMPExecutableDirective(D);
2425 Record.push_back(D->getCancelRegion());
2426 Code = serialization::STMT_OMP_CANCEL_DIRECTIVE;
2427 }
2428
VisitOMPTaskLoopDirective(OMPTaskLoopDirective * D)2429 void ASTStmtWriter::VisitOMPTaskLoopDirective(OMPTaskLoopDirective *D) {
2430 VisitOMPLoopDirective(D);
2431 Code = serialization::STMT_OMP_TASKLOOP_DIRECTIVE;
2432 }
2433
VisitOMPTaskLoopSimdDirective(OMPTaskLoopSimdDirective * D)2434 void ASTStmtWriter::VisitOMPTaskLoopSimdDirective(OMPTaskLoopSimdDirective *D) {
2435 VisitOMPLoopDirective(D);
2436 Code = serialization::STMT_OMP_TASKLOOP_SIMD_DIRECTIVE;
2437 }
2438
VisitOMPDistributeDirective(OMPDistributeDirective * D)2439 void ASTStmtWriter::VisitOMPDistributeDirective(OMPDistributeDirective *D) {
2440 VisitOMPLoopDirective(D);
2441 Code = serialization::STMT_OMP_DISTRIBUTE_DIRECTIVE;
2442 }
2443
VisitOMPTargetUpdateDirective(OMPTargetUpdateDirective * D)2444 void ASTStmtWriter::VisitOMPTargetUpdateDirective(OMPTargetUpdateDirective *D) {
2445 VisitStmt(D);
2446 Record.push_back(D->getNumClauses());
2447 VisitOMPExecutableDirective(D);
2448 Code = serialization::STMT_OMP_TARGET_UPDATE_DIRECTIVE;
2449 }
2450
VisitOMPDistributeParallelForDirective(OMPDistributeParallelForDirective * D)2451 void ASTStmtWriter::VisitOMPDistributeParallelForDirective(
2452 OMPDistributeParallelForDirective *D) {
2453 VisitOMPLoopDirective(D);
2454 Code = serialization::STMT_OMP_DISTRIBUTE_PARALLEL_FOR_DIRECTIVE;
2455 }
2456
VisitOMPDistributeParallelForSimdDirective(OMPDistributeParallelForSimdDirective * D)2457 void ASTStmtWriter::VisitOMPDistributeParallelForSimdDirective(
2458 OMPDistributeParallelForSimdDirective *D) {
2459 VisitOMPLoopDirective(D);
2460 Code = serialization::STMT_OMP_DISTRIBUTE_PARALLEL_FOR_SIMD_DIRECTIVE;
2461 }
2462
VisitOMPDistributeSimdDirective(OMPDistributeSimdDirective * D)2463 void ASTStmtWriter::VisitOMPDistributeSimdDirective(
2464 OMPDistributeSimdDirective *D) {
2465 VisitOMPLoopDirective(D);
2466 Code = serialization::STMT_OMP_DISTRIBUTE_SIMD_DIRECTIVE;
2467 }
2468
VisitOMPTargetParallelForSimdDirective(OMPTargetParallelForSimdDirective * D)2469 void ASTStmtWriter::VisitOMPTargetParallelForSimdDirective(
2470 OMPTargetParallelForSimdDirective *D) {
2471 VisitOMPLoopDirective(D);
2472 Code = serialization::STMT_OMP_TARGET_PARALLEL_FOR_SIMD_DIRECTIVE;
2473 }
2474
2475 //===----------------------------------------------------------------------===//
2476 // ASTWriter Implementation
2477 //===----------------------------------------------------------------------===//
2478
RecordSwitchCaseID(SwitchCase * S)2479 unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
2480 assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
2481 "SwitchCase recorded twice");
2482 unsigned NextID = SwitchCaseIDs.size();
2483 SwitchCaseIDs[S] = NextID;
2484 return NextID;
2485 }
2486
getSwitchCaseID(SwitchCase * S)2487 unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
2488 assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
2489 "SwitchCase hasn't been seen yet");
2490 return SwitchCaseIDs[S];
2491 }
2492
ClearSwitchCaseIDs()2493 void ASTWriter::ClearSwitchCaseIDs() {
2494 SwitchCaseIDs.clear();
2495 }
2496
2497 /// \brief Write the given substatement or subexpression to the
2498 /// bitstream.
WriteSubStmt(Stmt * S)2499 void ASTWriter::WriteSubStmt(Stmt *S) {
2500 RecordData Record;
2501 ASTStmtWriter Writer(*this, Record);
2502 ++NumStatements;
2503
2504 if (!S) {
2505 Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
2506 return;
2507 }
2508
2509 llvm::DenseMap<Stmt *, uint64_t>::iterator I = SubStmtEntries.find(S);
2510 if (I != SubStmtEntries.end()) {
2511 Record.push_back(I->second);
2512 Stream.EmitRecord(serialization::STMT_REF_PTR, Record);
2513 return;
2514 }
2515
2516 #ifndef NDEBUG
2517 assert(!ParentStmts.count(S) && "There is a Stmt cycle!");
2518
2519 struct ParentStmtInserterRAII {
2520 Stmt *S;
2521 llvm::DenseSet<Stmt *> &ParentStmts;
2522
2523 ParentStmtInserterRAII(Stmt *S, llvm::DenseSet<Stmt *> &ParentStmts)
2524 : S(S), ParentStmts(ParentStmts) {
2525 ParentStmts.insert(S);
2526 }
2527 ~ParentStmtInserterRAII() {
2528 ParentStmts.erase(S);
2529 }
2530 };
2531
2532 ParentStmtInserterRAII ParentStmtInserter(S, ParentStmts);
2533 #endif
2534
2535 Writer.Visit(S);
2536
2537 uint64_t Offset = Writer.Emit();
2538 SubStmtEntries[S] = Offset;
2539 }
2540
2541 /// \brief Flush all of the statements that have been added to the
2542 /// queue via AddStmt().
FlushStmts()2543 void ASTRecordWriter::FlushStmts() {
2544 // We expect to be the only consumer of the two temporary statement maps,
2545 // assert that they are empty.
2546 assert(Writer->SubStmtEntries.empty() && "unexpected entries in sub-stmt map");
2547 assert(Writer->ParentStmts.empty() && "unexpected entries in parent stmt map");
2548
2549 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
2550 Writer->WriteSubStmt(StmtsToEmit[I]);
2551
2552 assert(N == StmtsToEmit.size() && "record modified while being written!");
2553
2554 // Note that we are at the end of a full expression. Any
2555 // expression records that follow this one are part of a different
2556 // expression.
2557 Writer->Stream.EmitRecord(serialization::STMT_STOP, ArrayRef<uint32_t>());
2558
2559 Writer->SubStmtEntries.clear();
2560 Writer->ParentStmts.clear();
2561 }
2562
2563 StmtsToEmit.clear();
2564 }
2565
FlushSubStmts()2566 void ASTRecordWriter::FlushSubStmts() {
2567 // For a nested statement, write out the substatements in reverse order (so
2568 // that a simple stack machine can be used when loading), and don't emit a
2569 // STMT_STOP after each one.
2570 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
2571 Writer->WriteSubStmt(StmtsToEmit[N - I - 1]);
2572 assert(N == StmtsToEmit.size() && "record modified while being written!");
2573 }
2574
2575 StmtsToEmit.clear();
2576 }
2577