1 //===--- ASTWriterStmt.cpp - Statement and Expression Serialization -------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 ///
9 /// \file
10 /// Implements serialization for Statements and Expressions.
11 ///
12 //===----------------------------------------------------------------------===//
13
14 #include "clang/AST/ASTConcept.h"
15 #include "clang/AST/ASTContext.h"
16 #include "clang/AST/DeclCXX.h"
17 #include "clang/AST/DeclObjC.h"
18 #include "clang/AST/DeclTemplate.h"
19 #include "clang/AST/ExprOpenMP.h"
20 #include "clang/AST/StmtVisitor.h"
21 #include "clang/Lex/Token.h"
22 #include "clang/Sema/DeclSpec.h"
23 #include "clang/Serialization/ASTRecordWriter.h"
24 #include "llvm/Bitstream/BitstreamWriter.h"
25 using namespace clang;
26
27 //===----------------------------------------------------------------------===//
28 // Statement/expression serialization
29 //===----------------------------------------------------------------------===//
30
31 namespace clang {
32
33 class ASTStmtWriter : public StmtVisitor<ASTStmtWriter, void> {
34 ASTWriter &Writer;
35 ASTRecordWriter Record;
36
37 serialization::StmtCode Code;
38 unsigned AbbrevToUse;
39
40 /// A helper that can help us to write a packed bit across function
41 /// calls. For example, we may write seperate bits in seperate functions:
42 ///
43 /// void VisitA(A* a) {
44 /// Record.push_back(a->isSomething());
45 /// }
46 ///
47 /// void Visitb(B *b) {
48 /// VisitA(b);
49 /// Record.push_back(b->isAnother());
50 /// }
51 ///
52 /// In such cases, it'll be better if we can pack these 2 bits. We achieve
53 /// this by writing a zero value in `VisitA` and recorded that first and add
54 /// the new bit to the recorded value.
55 class PakedBitsWriter {
56 public:
PakedBitsWriter(ASTRecordWriter & Record)57 PakedBitsWriter(ASTRecordWriter &Record) : RecordRef(Record) {}
~PakedBitsWriter()58 ~PakedBitsWriter() { assert(!CurrentIndex); }
59
addBit(bool Value)60 void addBit(bool Value) {
61 assert(CurrentIndex && "Writing Bits without recording first!");
62 PackingBits.addBit(Value);
63 }
addBits(uint32_t Value,uint32_t BitsWidth)64 void addBits(uint32_t Value, uint32_t BitsWidth) {
65 assert(CurrentIndex && "Writing Bits without recording first!");
66 PackingBits.addBits(Value, BitsWidth);
67 }
68
writeBits()69 void writeBits() {
70 if (!CurrentIndex)
71 return;
72
73 RecordRef[*CurrentIndex] = (uint32_t)PackingBits;
74 CurrentIndex = std::nullopt;
75 PackingBits.reset(0);
76 }
77
updateBits()78 void updateBits() {
79 writeBits();
80
81 CurrentIndex = RecordRef.size();
82 RecordRef.push_back(0);
83 }
84
85 private:
86 BitsPacker PackingBits;
87 ASTRecordWriter &RecordRef;
88 std::optional<unsigned> CurrentIndex;
89 };
90
91 PakedBitsWriter CurrentPackingBits;
92
93 public:
ASTStmtWriter(ASTWriter & Writer,ASTWriter::RecordData & Record)94 ASTStmtWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
95 : Writer(Writer), Record(Writer, Record),
96 Code(serialization::STMT_NULL_PTR), AbbrevToUse(0),
97 CurrentPackingBits(this->Record) {}
98
99 ASTStmtWriter(const ASTStmtWriter&) = delete;
100 ASTStmtWriter &operator=(const ASTStmtWriter &) = delete;
101
Emit()102 uint64_t Emit() {
103 CurrentPackingBits.writeBits();
104 assert(Code != serialization::STMT_NULL_PTR &&
105 "unhandled sub-statement writing AST file");
106 return Record.EmitStmt(Code, AbbrevToUse);
107 }
108
109 void AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo &ArgInfo,
110 const TemplateArgumentLoc *Args);
111
112 void VisitStmt(Stmt *S);
113 #define STMT(Type, Base) \
114 void Visit##Type(Type *);
115 #include "clang/AST/StmtNodes.inc"
116 };
117 }
118
AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo & ArgInfo,const TemplateArgumentLoc * Args)119 void ASTStmtWriter::AddTemplateKWAndArgsInfo(
120 const ASTTemplateKWAndArgsInfo &ArgInfo, const TemplateArgumentLoc *Args) {
121 Record.AddSourceLocation(ArgInfo.TemplateKWLoc);
122 Record.AddSourceLocation(ArgInfo.LAngleLoc);
123 Record.AddSourceLocation(ArgInfo.RAngleLoc);
124 for (unsigned i = 0; i != ArgInfo.NumTemplateArgs; ++i)
125 Record.AddTemplateArgumentLoc(Args[i]);
126 }
127
VisitStmt(Stmt * S)128 void ASTStmtWriter::VisitStmt(Stmt *S) {
129 }
130
VisitNullStmt(NullStmt * S)131 void ASTStmtWriter::VisitNullStmt(NullStmt *S) {
132 VisitStmt(S);
133 Record.AddSourceLocation(S->getSemiLoc());
134 Record.push_back(S->NullStmtBits.HasLeadingEmptyMacro);
135 Code = serialization::STMT_NULL;
136 }
137
VisitCompoundStmt(CompoundStmt * S)138 void ASTStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
139 VisitStmt(S);
140
141 Record.push_back(S->size());
142 Record.push_back(S->hasStoredFPFeatures());
143
144 for (auto *CS : S->body())
145 Record.AddStmt(CS);
146 if (S->hasStoredFPFeatures())
147 Record.push_back(S->getStoredFPFeatures().getAsOpaqueInt());
148 Record.AddSourceLocation(S->getLBracLoc());
149 Record.AddSourceLocation(S->getRBracLoc());
150
151 if (!S->hasStoredFPFeatures())
152 AbbrevToUse = Writer.getCompoundStmtAbbrev();
153
154 Code = serialization::STMT_COMPOUND;
155 }
156
VisitSwitchCase(SwitchCase * S)157 void ASTStmtWriter::VisitSwitchCase(SwitchCase *S) {
158 VisitStmt(S);
159 Record.push_back(Writer.getSwitchCaseID(S));
160 Record.AddSourceLocation(S->getKeywordLoc());
161 Record.AddSourceLocation(S->getColonLoc());
162 }
163
VisitCaseStmt(CaseStmt * S)164 void ASTStmtWriter::VisitCaseStmt(CaseStmt *S) {
165 VisitSwitchCase(S);
166 Record.push_back(S->caseStmtIsGNURange());
167 Record.AddStmt(S->getLHS());
168 Record.AddStmt(S->getSubStmt());
169 if (S->caseStmtIsGNURange()) {
170 Record.AddStmt(S->getRHS());
171 Record.AddSourceLocation(S->getEllipsisLoc());
172 }
173 Code = serialization::STMT_CASE;
174 }
175
VisitDefaultStmt(DefaultStmt * S)176 void ASTStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
177 VisitSwitchCase(S);
178 Record.AddStmt(S->getSubStmt());
179 Code = serialization::STMT_DEFAULT;
180 }
181
VisitLabelStmt(LabelStmt * S)182 void ASTStmtWriter::VisitLabelStmt(LabelStmt *S) {
183 VisitStmt(S);
184 Record.push_back(S->isSideEntry());
185 Record.AddDeclRef(S->getDecl());
186 Record.AddStmt(S->getSubStmt());
187 Record.AddSourceLocation(S->getIdentLoc());
188 Code = serialization::STMT_LABEL;
189 }
190
VisitAttributedStmt(AttributedStmt * S)191 void ASTStmtWriter::VisitAttributedStmt(AttributedStmt *S) {
192 VisitStmt(S);
193 Record.push_back(S->getAttrs().size());
194 Record.AddAttributes(S->getAttrs());
195 Record.AddStmt(S->getSubStmt());
196 Record.AddSourceLocation(S->getAttrLoc());
197 Code = serialization::STMT_ATTRIBUTED;
198 }
199
VisitIfStmt(IfStmt * S)200 void ASTStmtWriter::VisitIfStmt(IfStmt *S) {
201 VisitStmt(S);
202
203 bool HasElse = S->getElse() != nullptr;
204 bool HasVar = S->getConditionVariableDeclStmt() != nullptr;
205 bool HasInit = S->getInit() != nullptr;
206
207 CurrentPackingBits.updateBits();
208
209 CurrentPackingBits.addBit(HasElse);
210 CurrentPackingBits.addBit(HasVar);
211 CurrentPackingBits.addBit(HasInit);
212 Record.push_back(static_cast<uint64_t>(S->getStatementKind()));
213 Record.AddStmt(S->getCond());
214 Record.AddStmt(S->getThen());
215 if (HasElse)
216 Record.AddStmt(S->getElse());
217 if (HasVar)
218 Record.AddStmt(S->getConditionVariableDeclStmt());
219 if (HasInit)
220 Record.AddStmt(S->getInit());
221
222 Record.AddSourceLocation(S->getIfLoc());
223 Record.AddSourceLocation(S->getLParenLoc());
224 Record.AddSourceLocation(S->getRParenLoc());
225 if (HasElse)
226 Record.AddSourceLocation(S->getElseLoc());
227
228 Code = serialization::STMT_IF;
229 }
230
VisitSwitchStmt(SwitchStmt * S)231 void ASTStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
232 VisitStmt(S);
233
234 bool HasInit = S->getInit() != nullptr;
235 bool HasVar = S->getConditionVariableDeclStmt() != nullptr;
236 Record.push_back(HasInit);
237 Record.push_back(HasVar);
238 Record.push_back(S->isAllEnumCasesCovered());
239
240 Record.AddStmt(S->getCond());
241 Record.AddStmt(S->getBody());
242 if (HasInit)
243 Record.AddStmt(S->getInit());
244 if (HasVar)
245 Record.AddStmt(S->getConditionVariableDeclStmt());
246
247 Record.AddSourceLocation(S->getSwitchLoc());
248 Record.AddSourceLocation(S->getLParenLoc());
249 Record.AddSourceLocation(S->getRParenLoc());
250
251 for (SwitchCase *SC = S->getSwitchCaseList(); SC;
252 SC = SC->getNextSwitchCase())
253 Record.push_back(Writer.RecordSwitchCaseID(SC));
254 Code = serialization::STMT_SWITCH;
255 }
256
VisitWhileStmt(WhileStmt * S)257 void ASTStmtWriter::VisitWhileStmt(WhileStmt *S) {
258 VisitStmt(S);
259
260 bool HasVar = S->getConditionVariableDeclStmt() != nullptr;
261 Record.push_back(HasVar);
262
263 Record.AddStmt(S->getCond());
264 Record.AddStmt(S->getBody());
265 if (HasVar)
266 Record.AddStmt(S->getConditionVariableDeclStmt());
267
268 Record.AddSourceLocation(S->getWhileLoc());
269 Record.AddSourceLocation(S->getLParenLoc());
270 Record.AddSourceLocation(S->getRParenLoc());
271 Code = serialization::STMT_WHILE;
272 }
273
VisitDoStmt(DoStmt * S)274 void ASTStmtWriter::VisitDoStmt(DoStmt *S) {
275 VisitStmt(S);
276 Record.AddStmt(S->getCond());
277 Record.AddStmt(S->getBody());
278 Record.AddSourceLocation(S->getDoLoc());
279 Record.AddSourceLocation(S->getWhileLoc());
280 Record.AddSourceLocation(S->getRParenLoc());
281 Code = serialization::STMT_DO;
282 }
283
VisitForStmt(ForStmt * S)284 void ASTStmtWriter::VisitForStmt(ForStmt *S) {
285 VisitStmt(S);
286 Record.AddStmt(S->getInit());
287 Record.AddStmt(S->getCond());
288 Record.AddStmt(S->getConditionVariableDeclStmt());
289 Record.AddStmt(S->getInc());
290 Record.AddStmt(S->getBody());
291 Record.AddSourceLocation(S->getForLoc());
292 Record.AddSourceLocation(S->getLParenLoc());
293 Record.AddSourceLocation(S->getRParenLoc());
294 Code = serialization::STMT_FOR;
295 }
296
VisitGotoStmt(GotoStmt * S)297 void ASTStmtWriter::VisitGotoStmt(GotoStmt *S) {
298 VisitStmt(S);
299 Record.AddDeclRef(S->getLabel());
300 Record.AddSourceLocation(S->getGotoLoc());
301 Record.AddSourceLocation(S->getLabelLoc());
302 Code = serialization::STMT_GOTO;
303 }
304
VisitIndirectGotoStmt(IndirectGotoStmt * S)305 void ASTStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
306 VisitStmt(S);
307 Record.AddSourceLocation(S->getGotoLoc());
308 Record.AddSourceLocation(S->getStarLoc());
309 Record.AddStmt(S->getTarget());
310 Code = serialization::STMT_INDIRECT_GOTO;
311 }
312
VisitContinueStmt(ContinueStmt * S)313 void ASTStmtWriter::VisitContinueStmt(ContinueStmt *S) {
314 VisitStmt(S);
315 Record.AddSourceLocation(S->getContinueLoc());
316 Code = serialization::STMT_CONTINUE;
317 }
318
VisitBreakStmt(BreakStmt * S)319 void ASTStmtWriter::VisitBreakStmt(BreakStmt *S) {
320 VisitStmt(S);
321 Record.AddSourceLocation(S->getBreakLoc());
322 Code = serialization::STMT_BREAK;
323 }
324
VisitReturnStmt(ReturnStmt * S)325 void ASTStmtWriter::VisitReturnStmt(ReturnStmt *S) {
326 VisitStmt(S);
327
328 bool HasNRVOCandidate = S->getNRVOCandidate() != nullptr;
329 Record.push_back(HasNRVOCandidate);
330
331 Record.AddStmt(S->getRetValue());
332 if (HasNRVOCandidate)
333 Record.AddDeclRef(S->getNRVOCandidate());
334
335 Record.AddSourceLocation(S->getReturnLoc());
336 Code = serialization::STMT_RETURN;
337 }
338
VisitDeclStmt(DeclStmt * S)339 void ASTStmtWriter::VisitDeclStmt(DeclStmt *S) {
340 VisitStmt(S);
341 Record.AddSourceLocation(S->getBeginLoc());
342 Record.AddSourceLocation(S->getEndLoc());
343 DeclGroupRef DG = S->getDeclGroup();
344 for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
345 Record.AddDeclRef(*D);
346 Code = serialization::STMT_DECL;
347 }
348
VisitAsmStmt(AsmStmt * S)349 void ASTStmtWriter::VisitAsmStmt(AsmStmt *S) {
350 VisitStmt(S);
351 Record.push_back(S->getNumOutputs());
352 Record.push_back(S->getNumInputs());
353 Record.push_back(S->getNumClobbers());
354 Record.AddSourceLocation(S->getAsmLoc());
355 Record.push_back(S->isVolatile());
356 Record.push_back(S->isSimple());
357 }
358
VisitGCCAsmStmt(GCCAsmStmt * S)359 void ASTStmtWriter::VisitGCCAsmStmt(GCCAsmStmt *S) {
360 VisitAsmStmt(S);
361 Record.push_back(S->getNumLabels());
362 Record.AddSourceLocation(S->getRParenLoc());
363 Record.AddStmt(S->getAsmString());
364
365 // Outputs
366 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
367 Record.AddIdentifierRef(S->getOutputIdentifier(I));
368 Record.AddStmt(S->getOutputConstraintLiteral(I));
369 Record.AddStmt(S->getOutputExpr(I));
370 }
371
372 // Inputs
373 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
374 Record.AddIdentifierRef(S->getInputIdentifier(I));
375 Record.AddStmt(S->getInputConstraintLiteral(I));
376 Record.AddStmt(S->getInputExpr(I));
377 }
378
379 // Clobbers
380 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
381 Record.AddStmt(S->getClobberStringLiteral(I));
382
383 // Labels
384 for (unsigned I = 0, N = S->getNumLabels(); I != N; ++I) {
385 Record.AddIdentifierRef(S->getLabelIdentifier(I));
386 Record.AddStmt(S->getLabelExpr(I));
387 }
388
389 Code = serialization::STMT_GCCASM;
390 }
391
VisitMSAsmStmt(MSAsmStmt * S)392 void ASTStmtWriter::VisitMSAsmStmt(MSAsmStmt *S) {
393 VisitAsmStmt(S);
394 Record.AddSourceLocation(S->getLBraceLoc());
395 Record.AddSourceLocation(S->getEndLoc());
396 Record.push_back(S->getNumAsmToks());
397 Record.AddString(S->getAsmString());
398
399 // Tokens
400 for (unsigned I = 0, N = S->getNumAsmToks(); I != N; ++I) {
401 // FIXME: Move this to ASTRecordWriter?
402 Writer.AddToken(S->getAsmToks()[I], Record.getRecordData());
403 }
404
405 // Clobbers
406 for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I) {
407 Record.AddString(S->getClobber(I));
408 }
409
410 // Outputs
411 for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
412 Record.AddStmt(S->getOutputExpr(I));
413 Record.AddString(S->getOutputConstraint(I));
414 }
415
416 // Inputs
417 for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
418 Record.AddStmt(S->getInputExpr(I));
419 Record.AddString(S->getInputConstraint(I));
420 }
421
422 Code = serialization::STMT_MSASM;
423 }
424
VisitCoroutineBodyStmt(CoroutineBodyStmt * CoroStmt)425 void ASTStmtWriter::VisitCoroutineBodyStmt(CoroutineBodyStmt *CoroStmt) {
426 VisitStmt(CoroStmt);
427 Record.push_back(CoroStmt->getParamMoves().size());
428 for (Stmt *S : CoroStmt->children())
429 Record.AddStmt(S);
430 Code = serialization::STMT_COROUTINE_BODY;
431 }
432
VisitCoreturnStmt(CoreturnStmt * S)433 void ASTStmtWriter::VisitCoreturnStmt(CoreturnStmt *S) {
434 VisitStmt(S);
435 Record.AddSourceLocation(S->getKeywordLoc());
436 Record.AddStmt(S->getOperand());
437 Record.AddStmt(S->getPromiseCall());
438 Record.push_back(S->isImplicit());
439 Code = serialization::STMT_CORETURN;
440 }
441
VisitCoroutineSuspendExpr(CoroutineSuspendExpr * E)442 void ASTStmtWriter::VisitCoroutineSuspendExpr(CoroutineSuspendExpr *E) {
443 VisitExpr(E);
444 Record.AddSourceLocation(E->getKeywordLoc());
445 for (Stmt *S : E->children())
446 Record.AddStmt(S);
447 Record.AddStmt(E->getOpaqueValue());
448 }
449
VisitCoawaitExpr(CoawaitExpr * E)450 void ASTStmtWriter::VisitCoawaitExpr(CoawaitExpr *E) {
451 VisitCoroutineSuspendExpr(E);
452 Record.push_back(E->isImplicit());
453 Code = serialization::EXPR_COAWAIT;
454 }
455
VisitCoyieldExpr(CoyieldExpr * E)456 void ASTStmtWriter::VisitCoyieldExpr(CoyieldExpr *E) {
457 VisitCoroutineSuspendExpr(E);
458 Code = serialization::EXPR_COYIELD;
459 }
460
VisitDependentCoawaitExpr(DependentCoawaitExpr * E)461 void ASTStmtWriter::VisitDependentCoawaitExpr(DependentCoawaitExpr *E) {
462 VisitExpr(E);
463 Record.AddSourceLocation(E->getKeywordLoc());
464 for (Stmt *S : E->children())
465 Record.AddStmt(S);
466 Code = serialization::EXPR_DEPENDENT_COAWAIT;
467 }
468
469 static void
addConstraintSatisfaction(ASTRecordWriter & Record,const ASTConstraintSatisfaction & Satisfaction)470 addConstraintSatisfaction(ASTRecordWriter &Record,
471 const ASTConstraintSatisfaction &Satisfaction) {
472 Record.push_back(Satisfaction.IsSatisfied);
473 Record.push_back(Satisfaction.ContainsErrors);
474 if (!Satisfaction.IsSatisfied) {
475 Record.push_back(Satisfaction.NumRecords);
476 for (const auto &DetailRecord : Satisfaction) {
477 Record.AddStmt(const_cast<Expr *>(DetailRecord.first));
478 auto *E = DetailRecord.second.dyn_cast<Expr *>();
479 Record.push_back(E == nullptr);
480 if (E)
481 Record.AddStmt(E);
482 else {
483 auto *Diag = DetailRecord.second.get<std::pair<SourceLocation,
484 StringRef> *>();
485 Record.AddSourceLocation(Diag->first);
486 Record.AddString(Diag->second);
487 }
488 }
489 }
490 }
491
492 static void
addSubstitutionDiagnostic(ASTRecordWriter & Record,const concepts::Requirement::SubstitutionDiagnostic * D)493 addSubstitutionDiagnostic(
494 ASTRecordWriter &Record,
495 const concepts::Requirement::SubstitutionDiagnostic *D) {
496 Record.AddString(D->SubstitutedEntity);
497 Record.AddSourceLocation(D->DiagLoc);
498 Record.AddString(D->DiagMessage);
499 }
500
VisitConceptSpecializationExpr(ConceptSpecializationExpr * E)501 void ASTStmtWriter::VisitConceptSpecializationExpr(
502 ConceptSpecializationExpr *E) {
503 VisitExpr(E);
504 Record.AddDeclRef(E->getSpecializationDecl());
505 const ConceptReference *CR = E->getConceptReference();
506 Record.push_back(CR != nullptr);
507 if (CR)
508 Record.AddConceptReference(CR);
509 if (!E->isValueDependent())
510 addConstraintSatisfaction(Record, E->getSatisfaction());
511
512 Code = serialization::EXPR_CONCEPT_SPECIALIZATION;
513 }
514
VisitRequiresExpr(RequiresExpr * E)515 void ASTStmtWriter::VisitRequiresExpr(RequiresExpr *E) {
516 VisitExpr(E);
517 Record.push_back(E->getLocalParameters().size());
518 Record.push_back(E->getRequirements().size());
519 Record.AddSourceLocation(E->RequiresExprBits.RequiresKWLoc);
520 Record.push_back(E->RequiresExprBits.IsSatisfied);
521 Record.AddDeclRef(E->getBody());
522 for (ParmVarDecl *P : E->getLocalParameters())
523 Record.AddDeclRef(P);
524 for (concepts::Requirement *R : E->getRequirements()) {
525 if (auto *TypeReq = dyn_cast<concepts::TypeRequirement>(R)) {
526 Record.push_back(concepts::Requirement::RK_Type);
527 Record.push_back(TypeReq->Status);
528 if (TypeReq->Status == concepts::TypeRequirement::SS_SubstitutionFailure)
529 addSubstitutionDiagnostic(Record, TypeReq->getSubstitutionDiagnostic());
530 else
531 Record.AddTypeSourceInfo(TypeReq->getType());
532 } else if (auto *ExprReq = dyn_cast<concepts::ExprRequirement>(R)) {
533 Record.push_back(ExprReq->getKind());
534 Record.push_back(ExprReq->Status);
535 if (ExprReq->isExprSubstitutionFailure()) {
536 addSubstitutionDiagnostic(Record,
537 ExprReq->Value.get<concepts::Requirement::SubstitutionDiagnostic *>());
538 } else
539 Record.AddStmt(ExprReq->Value.get<Expr *>());
540 if (ExprReq->getKind() == concepts::Requirement::RK_Compound) {
541 Record.AddSourceLocation(ExprReq->NoexceptLoc);
542 const auto &RetReq = ExprReq->getReturnTypeRequirement();
543 if (RetReq.isSubstitutionFailure()) {
544 Record.push_back(2);
545 addSubstitutionDiagnostic(Record, RetReq.getSubstitutionDiagnostic());
546 } else if (RetReq.isTypeConstraint()) {
547 Record.push_back(1);
548 Record.AddTemplateParameterList(
549 RetReq.getTypeConstraintTemplateParameterList());
550 if (ExprReq->Status >=
551 concepts::ExprRequirement::SS_ConstraintsNotSatisfied)
552 Record.AddStmt(
553 ExprReq->getReturnTypeRequirementSubstitutedConstraintExpr());
554 } else {
555 assert(RetReq.isEmpty());
556 Record.push_back(0);
557 }
558 }
559 } else {
560 auto *NestedReq = cast<concepts::NestedRequirement>(R);
561 Record.push_back(concepts::Requirement::RK_Nested);
562 Record.push_back(NestedReq->hasInvalidConstraint());
563 if (NestedReq->hasInvalidConstraint()) {
564 Record.AddString(NestedReq->getInvalidConstraintEntity());
565 addConstraintSatisfaction(Record, *NestedReq->Satisfaction);
566 } else {
567 Record.AddStmt(NestedReq->getConstraintExpr());
568 if (!NestedReq->isDependent())
569 addConstraintSatisfaction(Record, *NestedReq->Satisfaction);
570 }
571 }
572 }
573 Record.AddSourceLocation(E->getLParenLoc());
574 Record.AddSourceLocation(E->getRParenLoc());
575 Record.AddSourceLocation(E->getEndLoc());
576
577 Code = serialization::EXPR_REQUIRES;
578 }
579
580
VisitCapturedStmt(CapturedStmt * S)581 void ASTStmtWriter::VisitCapturedStmt(CapturedStmt *S) {
582 VisitStmt(S);
583 // NumCaptures
584 Record.push_back(std::distance(S->capture_begin(), S->capture_end()));
585
586 // CapturedDecl and captured region kind
587 Record.AddDeclRef(S->getCapturedDecl());
588 Record.push_back(S->getCapturedRegionKind());
589
590 Record.AddDeclRef(S->getCapturedRecordDecl());
591
592 // Capture inits
593 for (auto *I : S->capture_inits())
594 Record.AddStmt(I);
595
596 // Body
597 Record.AddStmt(S->getCapturedStmt());
598
599 // Captures
600 for (const auto &I : S->captures()) {
601 if (I.capturesThis() || I.capturesVariableArrayType())
602 Record.AddDeclRef(nullptr);
603 else
604 Record.AddDeclRef(I.getCapturedVar());
605 Record.push_back(I.getCaptureKind());
606 Record.AddSourceLocation(I.getLocation());
607 }
608
609 Code = serialization::STMT_CAPTURED;
610 }
611
VisitExpr(Expr * E)612 void ASTStmtWriter::VisitExpr(Expr *E) {
613 VisitStmt(E);
614
615 CurrentPackingBits.updateBits();
616 CurrentPackingBits.addBits(E->getDependence(), /*BitsWidth=*/5);
617 CurrentPackingBits.addBits(E->getValueKind(), /*BitsWidth=*/2);
618 CurrentPackingBits.addBits(E->getObjectKind(), /*BitsWidth=*/3);
619
620 Record.AddTypeRef(E->getType());
621 }
622
VisitConstantExpr(ConstantExpr * E)623 void ASTStmtWriter::VisitConstantExpr(ConstantExpr *E) {
624 VisitExpr(E);
625 Record.push_back(E->ConstantExprBits.ResultKind);
626
627 Record.push_back(E->ConstantExprBits.APValueKind);
628 Record.push_back(E->ConstantExprBits.IsUnsigned);
629 Record.push_back(E->ConstantExprBits.BitWidth);
630 // HasCleanup not serialized since we can just query the APValue.
631 Record.push_back(E->ConstantExprBits.IsImmediateInvocation);
632
633 switch (E->getResultStorageKind()) {
634 case ConstantResultStorageKind::None:
635 break;
636 case ConstantResultStorageKind::Int64:
637 Record.push_back(E->Int64Result());
638 break;
639 case ConstantResultStorageKind::APValue:
640 Record.AddAPValue(E->APValueResult());
641 break;
642 }
643
644 Record.AddStmt(E->getSubExpr());
645 Code = serialization::EXPR_CONSTANT;
646 }
647
VisitSYCLUniqueStableNameExpr(SYCLUniqueStableNameExpr * E)648 void ASTStmtWriter::VisitSYCLUniqueStableNameExpr(SYCLUniqueStableNameExpr *E) {
649 VisitExpr(E);
650
651 Record.AddSourceLocation(E->getLocation());
652 Record.AddSourceLocation(E->getLParenLocation());
653 Record.AddSourceLocation(E->getRParenLocation());
654 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
655
656 Code = serialization::EXPR_SYCL_UNIQUE_STABLE_NAME;
657 }
658
VisitPredefinedExpr(PredefinedExpr * E)659 void ASTStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
660 VisitExpr(E);
661
662 bool HasFunctionName = E->getFunctionName() != nullptr;
663 Record.push_back(HasFunctionName);
664 Record.push_back(
665 llvm::to_underlying(E->getIdentKind())); // FIXME: stable encoding
666 Record.push_back(E->isTransparent());
667 Record.AddSourceLocation(E->getLocation());
668 if (HasFunctionName)
669 Record.AddStmt(E->getFunctionName());
670 Code = serialization::EXPR_PREDEFINED;
671 }
672
VisitDeclRefExpr(DeclRefExpr * E)673 void ASTStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
674 VisitExpr(E);
675
676 CurrentPackingBits.updateBits();
677
678 CurrentPackingBits.addBit(E->hadMultipleCandidates());
679 CurrentPackingBits.addBit(E->refersToEnclosingVariableOrCapture());
680 CurrentPackingBits.addBits(E->isNonOdrUse(), /*Width=*/2);
681 CurrentPackingBits.addBit(E->isImmediateEscalating());
682 CurrentPackingBits.addBit(E->getDecl() != E->getFoundDecl());
683 CurrentPackingBits.addBit(E->hasQualifier());
684 CurrentPackingBits.addBit(E->hasTemplateKWAndArgsInfo());
685
686 if (E->hasTemplateKWAndArgsInfo()) {
687 unsigned NumTemplateArgs = E->getNumTemplateArgs();
688 Record.push_back(NumTemplateArgs);
689 }
690
691 DeclarationName::NameKind nk = (E->getDecl()->getDeclName().getNameKind());
692
693 if ((!E->hasTemplateKWAndArgsInfo()) && (!E->hasQualifier()) &&
694 (E->getDecl() == E->getFoundDecl()) &&
695 nk == DeclarationName::Identifier && E->getObjectKind() == OK_Ordinary) {
696 AbbrevToUse = Writer.getDeclRefExprAbbrev();
697 }
698
699 if (E->hasQualifier())
700 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
701
702 if (E->getDecl() != E->getFoundDecl())
703 Record.AddDeclRef(E->getFoundDecl());
704
705 if (E->hasTemplateKWAndArgsInfo())
706 AddTemplateKWAndArgsInfo(*E->getTrailingObjects<ASTTemplateKWAndArgsInfo>(),
707 E->getTrailingObjects<TemplateArgumentLoc>());
708
709 Record.AddDeclRef(E->getDecl());
710 Record.AddSourceLocation(E->getLocation());
711 Record.AddDeclarationNameLoc(E->DNLoc, E->getDecl()->getDeclName());
712 Code = serialization::EXPR_DECL_REF;
713 }
714
VisitIntegerLiteral(IntegerLiteral * E)715 void ASTStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
716 VisitExpr(E);
717 Record.AddSourceLocation(E->getLocation());
718 Record.AddAPInt(E->getValue());
719
720 if (E->getValue().getBitWidth() == 32) {
721 AbbrevToUse = Writer.getIntegerLiteralAbbrev();
722 }
723
724 Code = serialization::EXPR_INTEGER_LITERAL;
725 }
726
VisitFixedPointLiteral(FixedPointLiteral * E)727 void ASTStmtWriter::VisitFixedPointLiteral(FixedPointLiteral *E) {
728 VisitExpr(E);
729 Record.AddSourceLocation(E->getLocation());
730 Record.push_back(E->getScale());
731 Record.AddAPInt(E->getValue());
732 Code = serialization::EXPR_FIXEDPOINT_LITERAL;
733 }
734
VisitFloatingLiteral(FloatingLiteral * E)735 void ASTStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
736 VisitExpr(E);
737 Record.push_back(E->getRawSemantics());
738 Record.push_back(E->isExact());
739 Record.AddAPFloat(E->getValue());
740 Record.AddSourceLocation(E->getLocation());
741 Code = serialization::EXPR_FLOATING_LITERAL;
742 }
743
VisitImaginaryLiteral(ImaginaryLiteral * E)744 void ASTStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
745 VisitExpr(E);
746 Record.AddStmt(E->getSubExpr());
747 Code = serialization::EXPR_IMAGINARY_LITERAL;
748 }
749
VisitStringLiteral(StringLiteral * E)750 void ASTStmtWriter::VisitStringLiteral(StringLiteral *E) {
751 VisitExpr(E);
752
753 // Store the various bits of data of StringLiteral.
754 Record.push_back(E->getNumConcatenated());
755 Record.push_back(E->getLength());
756 Record.push_back(E->getCharByteWidth());
757 Record.push_back(llvm::to_underlying(E->getKind()));
758 Record.push_back(E->isPascal());
759
760 // Store the trailing array of SourceLocation.
761 for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
762 Record.AddSourceLocation(E->getStrTokenLoc(I));
763
764 // Store the trailing array of char holding the string data.
765 StringRef StrData = E->getBytes();
766 for (unsigned I = 0, N = E->getByteLength(); I != N; ++I)
767 Record.push_back(StrData[I]);
768
769 Code = serialization::EXPR_STRING_LITERAL;
770 }
771
VisitCharacterLiteral(CharacterLiteral * E)772 void ASTStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
773 VisitExpr(E);
774 Record.push_back(E->getValue());
775 Record.AddSourceLocation(E->getLocation());
776 Record.push_back(llvm::to_underlying(E->getKind()));
777
778 AbbrevToUse = Writer.getCharacterLiteralAbbrev();
779
780 Code = serialization::EXPR_CHARACTER_LITERAL;
781 }
782
VisitParenExpr(ParenExpr * E)783 void ASTStmtWriter::VisitParenExpr(ParenExpr *E) {
784 VisitExpr(E);
785 Record.AddSourceLocation(E->getLParen());
786 Record.AddSourceLocation(E->getRParen());
787 Record.AddStmt(E->getSubExpr());
788 Code = serialization::EXPR_PAREN;
789 }
790
VisitParenListExpr(ParenListExpr * E)791 void ASTStmtWriter::VisitParenListExpr(ParenListExpr *E) {
792 VisitExpr(E);
793 Record.push_back(E->getNumExprs());
794 for (auto *SubStmt : E->exprs())
795 Record.AddStmt(SubStmt);
796 Record.AddSourceLocation(E->getLParenLoc());
797 Record.AddSourceLocation(E->getRParenLoc());
798 Code = serialization::EXPR_PAREN_LIST;
799 }
800
VisitUnaryOperator(UnaryOperator * E)801 void ASTStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
802 VisitExpr(E);
803 bool HasFPFeatures = E->hasStoredFPFeatures();
804 // Write this first for easy access when deserializing, as they affect the
805 // size of the UnaryOperator.
806 CurrentPackingBits.addBit(HasFPFeatures);
807 Record.AddStmt(E->getSubExpr());
808 CurrentPackingBits.addBits(E->getOpcode(),
809 /*Width=*/5); // FIXME: stable encoding
810 Record.AddSourceLocation(E->getOperatorLoc());
811 CurrentPackingBits.addBit(E->canOverflow());
812
813 if (HasFPFeatures)
814 Record.push_back(E->getStoredFPFeatures().getAsOpaqueInt());
815 Code = serialization::EXPR_UNARY_OPERATOR;
816 }
817
VisitOffsetOfExpr(OffsetOfExpr * E)818 void ASTStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
819 VisitExpr(E);
820 Record.push_back(E->getNumComponents());
821 Record.push_back(E->getNumExpressions());
822 Record.AddSourceLocation(E->getOperatorLoc());
823 Record.AddSourceLocation(E->getRParenLoc());
824 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
825 for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
826 const OffsetOfNode &ON = E->getComponent(I);
827 Record.push_back(ON.getKind()); // FIXME: Stable encoding
828 Record.AddSourceLocation(ON.getSourceRange().getBegin());
829 Record.AddSourceLocation(ON.getSourceRange().getEnd());
830 switch (ON.getKind()) {
831 case OffsetOfNode::Array:
832 Record.push_back(ON.getArrayExprIndex());
833 break;
834
835 case OffsetOfNode::Field:
836 Record.AddDeclRef(ON.getField());
837 break;
838
839 case OffsetOfNode::Identifier:
840 Record.AddIdentifierRef(ON.getFieldName());
841 break;
842
843 case OffsetOfNode::Base:
844 Record.AddCXXBaseSpecifier(*ON.getBase());
845 break;
846 }
847 }
848 for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
849 Record.AddStmt(E->getIndexExpr(I));
850 Code = serialization::EXPR_OFFSETOF;
851 }
852
VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr * E)853 void ASTStmtWriter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E) {
854 VisitExpr(E);
855 Record.push_back(E->getKind());
856 if (E->isArgumentType())
857 Record.AddTypeSourceInfo(E->getArgumentTypeInfo());
858 else {
859 Record.push_back(0);
860 Record.AddStmt(E->getArgumentExpr());
861 }
862 Record.AddSourceLocation(E->getOperatorLoc());
863 Record.AddSourceLocation(E->getRParenLoc());
864 Code = serialization::EXPR_SIZEOF_ALIGN_OF;
865 }
866
VisitArraySubscriptExpr(ArraySubscriptExpr * E)867 void ASTStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
868 VisitExpr(E);
869 Record.AddStmt(E->getLHS());
870 Record.AddStmt(E->getRHS());
871 Record.AddSourceLocation(E->getRBracketLoc());
872 Code = serialization::EXPR_ARRAY_SUBSCRIPT;
873 }
874
VisitMatrixSubscriptExpr(MatrixSubscriptExpr * E)875 void ASTStmtWriter::VisitMatrixSubscriptExpr(MatrixSubscriptExpr *E) {
876 VisitExpr(E);
877 Record.AddStmt(E->getBase());
878 Record.AddStmt(E->getRowIdx());
879 Record.AddStmt(E->getColumnIdx());
880 Record.AddSourceLocation(E->getRBracketLoc());
881 Code = serialization::EXPR_ARRAY_SUBSCRIPT;
882 }
883
VisitOMPArraySectionExpr(OMPArraySectionExpr * E)884 void ASTStmtWriter::VisitOMPArraySectionExpr(OMPArraySectionExpr *E) {
885 VisitExpr(E);
886 Record.AddStmt(E->getBase());
887 Record.AddStmt(E->getLowerBound());
888 Record.AddStmt(E->getLength());
889 Record.AddStmt(E->getStride());
890 Record.AddSourceLocation(E->getColonLocFirst());
891 Record.AddSourceLocation(E->getColonLocSecond());
892 Record.AddSourceLocation(E->getRBracketLoc());
893 Code = serialization::EXPR_OMP_ARRAY_SECTION;
894 }
895
VisitOMPArrayShapingExpr(OMPArrayShapingExpr * E)896 void ASTStmtWriter::VisitOMPArrayShapingExpr(OMPArrayShapingExpr *E) {
897 VisitExpr(E);
898 Record.push_back(E->getDimensions().size());
899 Record.AddStmt(E->getBase());
900 for (Expr *Dim : E->getDimensions())
901 Record.AddStmt(Dim);
902 for (SourceRange SR : E->getBracketsRanges())
903 Record.AddSourceRange(SR);
904 Record.AddSourceLocation(E->getLParenLoc());
905 Record.AddSourceLocation(E->getRParenLoc());
906 Code = serialization::EXPR_OMP_ARRAY_SHAPING;
907 }
908
VisitOMPIteratorExpr(OMPIteratorExpr * E)909 void ASTStmtWriter::VisitOMPIteratorExpr(OMPIteratorExpr *E) {
910 VisitExpr(E);
911 Record.push_back(E->numOfIterators());
912 Record.AddSourceLocation(E->getIteratorKwLoc());
913 Record.AddSourceLocation(E->getLParenLoc());
914 Record.AddSourceLocation(E->getRParenLoc());
915 for (unsigned I = 0, End = E->numOfIterators(); I < End; ++I) {
916 Record.AddDeclRef(E->getIteratorDecl(I));
917 Record.AddSourceLocation(E->getAssignLoc(I));
918 OMPIteratorExpr::IteratorRange Range = E->getIteratorRange(I);
919 Record.AddStmt(Range.Begin);
920 Record.AddStmt(Range.End);
921 Record.AddStmt(Range.Step);
922 Record.AddSourceLocation(E->getColonLoc(I));
923 if (Range.Step)
924 Record.AddSourceLocation(E->getSecondColonLoc(I));
925 // Serialize helpers
926 OMPIteratorHelperData &HD = E->getHelper(I);
927 Record.AddDeclRef(HD.CounterVD);
928 Record.AddStmt(HD.Upper);
929 Record.AddStmt(HD.Update);
930 Record.AddStmt(HD.CounterUpdate);
931 }
932 Code = serialization::EXPR_OMP_ITERATOR;
933 }
934
VisitCallExpr(CallExpr * E)935 void ASTStmtWriter::VisitCallExpr(CallExpr *E) {
936 VisitExpr(E);
937
938 Record.push_back(E->getNumArgs());
939 CurrentPackingBits.updateBits();
940 CurrentPackingBits.addBit(static_cast<bool>(E->getADLCallKind()));
941 CurrentPackingBits.addBit(E->hasStoredFPFeatures());
942
943 Record.AddSourceLocation(E->getRParenLoc());
944 Record.AddStmt(E->getCallee());
945 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
946 Arg != ArgEnd; ++Arg)
947 Record.AddStmt(*Arg);
948
949 if (E->hasStoredFPFeatures())
950 Record.push_back(E->getFPFeatures().getAsOpaqueInt());
951
952 if (!E->hasStoredFPFeatures() && !static_cast<bool>(E->getADLCallKind()) &&
953 E->getStmtClass() == Stmt::CallExprClass)
954 AbbrevToUse = Writer.getCallExprAbbrev();
955
956 Code = serialization::EXPR_CALL;
957 }
958
VisitRecoveryExpr(RecoveryExpr * E)959 void ASTStmtWriter::VisitRecoveryExpr(RecoveryExpr *E) {
960 VisitExpr(E);
961 Record.push_back(std::distance(E->children().begin(), E->children().end()));
962 Record.AddSourceLocation(E->getBeginLoc());
963 Record.AddSourceLocation(E->getEndLoc());
964 for (Stmt *Child : E->children())
965 Record.AddStmt(Child);
966 Code = serialization::EXPR_RECOVERY;
967 }
968
VisitMemberExpr(MemberExpr * E)969 void ASTStmtWriter::VisitMemberExpr(MemberExpr *E) {
970 VisitExpr(E);
971
972 bool HasQualifier = E->hasQualifier();
973 bool HasFoundDecl =
974 E->hasQualifierOrFoundDecl() &&
975 (E->getFoundDecl().getDecl() != E->getMemberDecl() ||
976 E->getFoundDecl().getAccess() != E->getMemberDecl()->getAccess());
977 bool HasTemplateInfo = E->hasTemplateKWAndArgsInfo();
978 unsigned NumTemplateArgs = E->getNumTemplateArgs();
979
980 // Write these first for easy access when deserializing, as they affect the
981 // size of the MemberExpr.
982 CurrentPackingBits.updateBits();
983 CurrentPackingBits.addBit(HasQualifier);
984 CurrentPackingBits.addBit(HasFoundDecl);
985 CurrentPackingBits.addBit(HasTemplateInfo);
986 Record.push_back(NumTemplateArgs);
987
988 Record.AddStmt(E->getBase());
989 Record.AddDeclRef(E->getMemberDecl());
990 Record.AddDeclarationNameLoc(E->MemberDNLoc,
991 E->getMemberDecl()->getDeclName());
992 Record.AddSourceLocation(E->getMemberLoc());
993 CurrentPackingBits.addBit(E->isArrow());
994 CurrentPackingBits.addBit(E->hadMultipleCandidates());
995 CurrentPackingBits.addBits(E->isNonOdrUse(), /*Width=*/2);
996 Record.AddSourceLocation(E->getOperatorLoc());
997
998 if (HasFoundDecl) {
999 DeclAccessPair FoundDecl = E->getFoundDecl();
1000 Record.AddDeclRef(FoundDecl.getDecl());
1001 CurrentPackingBits.addBits(FoundDecl.getAccess(), /*BitWidth=*/2);
1002 }
1003
1004 if (HasQualifier)
1005 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1006
1007 if (HasTemplateInfo)
1008 AddTemplateKWAndArgsInfo(*E->getTrailingObjects<ASTTemplateKWAndArgsInfo>(),
1009 E->getTrailingObjects<TemplateArgumentLoc>());
1010
1011 Code = serialization::EXPR_MEMBER;
1012 }
1013
VisitObjCIsaExpr(ObjCIsaExpr * E)1014 void ASTStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
1015 VisitExpr(E);
1016 Record.AddStmt(E->getBase());
1017 Record.AddSourceLocation(E->getIsaMemberLoc());
1018 Record.AddSourceLocation(E->getOpLoc());
1019 Record.push_back(E->isArrow());
1020 Code = serialization::EXPR_OBJC_ISA;
1021 }
1022
1023 void ASTStmtWriter::
VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr * E)1024 VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) {
1025 VisitExpr(E);
1026 Record.AddStmt(E->getSubExpr());
1027 Record.push_back(E->shouldCopy());
1028 Code = serialization::EXPR_OBJC_INDIRECT_COPY_RESTORE;
1029 }
1030
VisitObjCBridgedCastExpr(ObjCBridgedCastExpr * E)1031 void ASTStmtWriter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) {
1032 VisitExplicitCastExpr(E);
1033 Record.AddSourceLocation(E->getLParenLoc());
1034 Record.AddSourceLocation(E->getBridgeKeywordLoc());
1035 Record.push_back(E->getBridgeKind()); // FIXME: Stable encoding
1036 Code = serialization::EXPR_OBJC_BRIDGED_CAST;
1037 }
1038
VisitCastExpr(CastExpr * E)1039 void ASTStmtWriter::VisitCastExpr(CastExpr *E) {
1040 VisitExpr(E);
1041
1042 Record.push_back(E->path_size());
1043 CurrentPackingBits.updateBits();
1044 // 7 bits should be enough to store the casting kinds.
1045 CurrentPackingBits.addBits(E->getCastKind(), /*Width=*/7);
1046 CurrentPackingBits.addBit(E->hasStoredFPFeatures());
1047 Record.AddStmt(E->getSubExpr());
1048
1049 for (CastExpr::path_iterator
1050 PI = E->path_begin(), PE = E->path_end(); PI != PE; ++PI)
1051 Record.AddCXXBaseSpecifier(**PI);
1052
1053 if (E->hasStoredFPFeatures())
1054 Record.push_back(E->getFPFeatures().getAsOpaqueInt());
1055 }
1056
VisitBinaryOperator(BinaryOperator * E)1057 void ASTStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
1058 VisitExpr(E);
1059
1060 // Write this first for easy access when deserializing, as they affect the
1061 // size of the UnaryOperator.
1062 CurrentPackingBits.updateBits();
1063 CurrentPackingBits.addBits(E->getOpcode(), /*Width=*/6);
1064 bool HasFPFeatures = E->hasStoredFPFeatures();
1065 CurrentPackingBits.addBit(HasFPFeatures);
1066 Record.AddStmt(E->getLHS());
1067 Record.AddStmt(E->getRHS());
1068 Record.AddSourceLocation(E->getOperatorLoc());
1069 if (HasFPFeatures)
1070 Record.push_back(E->getStoredFPFeatures().getAsOpaqueInt());
1071
1072 if (!HasFPFeatures && E->getValueKind() == VK_PRValue &&
1073 E->getObjectKind() == OK_Ordinary)
1074 AbbrevToUse = Writer.getBinaryOperatorAbbrev();
1075
1076 Code = serialization::EXPR_BINARY_OPERATOR;
1077 }
1078
VisitCompoundAssignOperator(CompoundAssignOperator * E)1079 void ASTStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
1080 VisitBinaryOperator(E);
1081 Record.AddTypeRef(E->getComputationLHSType());
1082 Record.AddTypeRef(E->getComputationResultType());
1083
1084 if (!E->hasStoredFPFeatures() && E->getValueKind() == VK_PRValue &&
1085 E->getObjectKind() == OK_Ordinary)
1086 AbbrevToUse = Writer.getCompoundAssignOperatorAbbrev();
1087
1088 Code = serialization::EXPR_COMPOUND_ASSIGN_OPERATOR;
1089 }
1090
VisitConditionalOperator(ConditionalOperator * E)1091 void ASTStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
1092 VisitExpr(E);
1093 Record.AddStmt(E->getCond());
1094 Record.AddStmt(E->getLHS());
1095 Record.AddStmt(E->getRHS());
1096 Record.AddSourceLocation(E->getQuestionLoc());
1097 Record.AddSourceLocation(E->getColonLoc());
1098 Code = serialization::EXPR_CONDITIONAL_OPERATOR;
1099 }
1100
1101 void
VisitBinaryConditionalOperator(BinaryConditionalOperator * E)1102 ASTStmtWriter::VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
1103 VisitExpr(E);
1104 Record.AddStmt(E->getOpaqueValue());
1105 Record.AddStmt(E->getCommon());
1106 Record.AddStmt(E->getCond());
1107 Record.AddStmt(E->getTrueExpr());
1108 Record.AddStmt(E->getFalseExpr());
1109 Record.AddSourceLocation(E->getQuestionLoc());
1110 Record.AddSourceLocation(E->getColonLoc());
1111 Code = serialization::EXPR_BINARY_CONDITIONAL_OPERATOR;
1112 }
1113
VisitImplicitCastExpr(ImplicitCastExpr * E)1114 void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
1115 VisitCastExpr(E);
1116 CurrentPackingBits.addBit(E->isPartOfExplicitCast());
1117
1118 if (E->path_size() == 0 && !E->hasStoredFPFeatures())
1119 AbbrevToUse = Writer.getExprImplicitCastAbbrev();
1120
1121 Code = serialization::EXPR_IMPLICIT_CAST;
1122 }
1123
VisitExplicitCastExpr(ExplicitCastExpr * E)1124 void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
1125 VisitCastExpr(E);
1126 Record.AddTypeSourceInfo(E->getTypeInfoAsWritten());
1127 }
1128
VisitCStyleCastExpr(CStyleCastExpr * E)1129 void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
1130 VisitExplicitCastExpr(E);
1131 Record.AddSourceLocation(E->getLParenLoc());
1132 Record.AddSourceLocation(E->getRParenLoc());
1133 Code = serialization::EXPR_CSTYLE_CAST;
1134 }
1135
VisitCompoundLiteralExpr(CompoundLiteralExpr * E)1136 void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
1137 VisitExpr(E);
1138 Record.AddSourceLocation(E->getLParenLoc());
1139 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1140 Record.AddStmt(E->getInitializer());
1141 Record.push_back(E->isFileScope());
1142 Code = serialization::EXPR_COMPOUND_LITERAL;
1143 }
1144
VisitExtVectorElementExpr(ExtVectorElementExpr * E)1145 void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
1146 VisitExpr(E);
1147 Record.AddStmt(E->getBase());
1148 Record.AddIdentifierRef(&E->getAccessor());
1149 Record.AddSourceLocation(E->getAccessorLoc());
1150 Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
1151 }
1152
VisitInitListExpr(InitListExpr * E)1153 void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
1154 VisitExpr(E);
1155 // NOTE: only add the (possibly null) syntactic form.
1156 // No need to serialize the isSemanticForm flag and the semantic form.
1157 Record.AddStmt(E->getSyntacticForm());
1158 Record.AddSourceLocation(E->getLBraceLoc());
1159 Record.AddSourceLocation(E->getRBraceLoc());
1160 bool isArrayFiller = E->ArrayFillerOrUnionFieldInit.is<Expr*>();
1161 Record.push_back(isArrayFiller);
1162 if (isArrayFiller)
1163 Record.AddStmt(E->getArrayFiller());
1164 else
1165 Record.AddDeclRef(E->getInitializedFieldInUnion());
1166 Record.push_back(E->hadArrayRangeDesignator());
1167 Record.push_back(E->getNumInits());
1168 if (isArrayFiller) {
1169 // ArrayFiller may have filled "holes" due to designated initializer.
1170 // Replace them by 0 to indicate that the filler goes in that place.
1171 Expr *filler = E->getArrayFiller();
1172 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
1173 Record.AddStmt(E->getInit(I) != filler ? E->getInit(I) : nullptr);
1174 } else {
1175 for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
1176 Record.AddStmt(E->getInit(I));
1177 }
1178 Code = serialization::EXPR_INIT_LIST;
1179 }
1180
VisitDesignatedInitExpr(DesignatedInitExpr * E)1181 void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
1182 VisitExpr(E);
1183 Record.push_back(E->getNumSubExprs());
1184 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
1185 Record.AddStmt(E->getSubExpr(I));
1186 Record.AddSourceLocation(E->getEqualOrColonLoc());
1187 Record.push_back(E->usesGNUSyntax());
1188 for (const DesignatedInitExpr::Designator &D : E->designators()) {
1189 if (D.isFieldDesignator()) {
1190 if (FieldDecl *Field = D.getFieldDecl()) {
1191 Record.push_back(serialization::DESIG_FIELD_DECL);
1192 Record.AddDeclRef(Field);
1193 } else {
1194 Record.push_back(serialization::DESIG_FIELD_NAME);
1195 Record.AddIdentifierRef(D.getFieldName());
1196 }
1197 Record.AddSourceLocation(D.getDotLoc());
1198 Record.AddSourceLocation(D.getFieldLoc());
1199 } else if (D.isArrayDesignator()) {
1200 Record.push_back(serialization::DESIG_ARRAY);
1201 Record.push_back(D.getArrayIndex());
1202 Record.AddSourceLocation(D.getLBracketLoc());
1203 Record.AddSourceLocation(D.getRBracketLoc());
1204 } else {
1205 assert(D.isArrayRangeDesignator() && "Unknown designator");
1206 Record.push_back(serialization::DESIG_ARRAY_RANGE);
1207 Record.push_back(D.getArrayIndex());
1208 Record.AddSourceLocation(D.getLBracketLoc());
1209 Record.AddSourceLocation(D.getEllipsisLoc());
1210 Record.AddSourceLocation(D.getRBracketLoc());
1211 }
1212 }
1213 Code = serialization::EXPR_DESIGNATED_INIT;
1214 }
1215
VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr * E)1216 void ASTStmtWriter::VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr *E) {
1217 VisitExpr(E);
1218 Record.AddStmt(E->getBase());
1219 Record.AddStmt(E->getUpdater());
1220 Code = serialization::EXPR_DESIGNATED_INIT_UPDATE;
1221 }
1222
VisitNoInitExpr(NoInitExpr * E)1223 void ASTStmtWriter::VisitNoInitExpr(NoInitExpr *E) {
1224 VisitExpr(E);
1225 Code = serialization::EXPR_NO_INIT;
1226 }
1227
VisitArrayInitLoopExpr(ArrayInitLoopExpr * E)1228 void ASTStmtWriter::VisitArrayInitLoopExpr(ArrayInitLoopExpr *E) {
1229 VisitExpr(E);
1230 Record.AddStmt(E->SubExprs[0]);
1231 Record.AddStmt(E->SubExprs[1]);
1232 Code = serialization::EXPR_ARRAY_INIT_LOOP;
1233 }
1234
VisitArrayInitIndexExpr(ArrayInitIndexExpr * E)1235 void ASTStmtWriter::VisitArrayInitIndexExpr(ArrayInitIndexExpr *E) {
1236 VisitExpr(E);
1237 Code = serialization::EXPR_ARRAY_INIT_INDEX;
1238 }
1239
VisitImplicitValueInitExpr(ImplicitValueInitExpr * E)1240 void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
1241 VisitExpr(E);
1242 Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
1243 }
1244
VisitVAArgExpr(VAArgExpr * E)1245 void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
1246 VisitExpr(E);
1247 Record.AddStmt(E->getSubExpr());
1248 Record.AddTypeSourceInfo(E->getWrittenTypeInfo());
1249 Record.AddSourceLocation(E->getBuiltinLoc());
1250 Record.AddSourceLocation(E->getRParenLoc());
1251 Record.push_back(E->isMicrosoftABI());
1252 Code = serialization::EXPR_VA_ARG;
1253 }
1254
VisitSourceLocExpr(SourceLocExpr * E)1255 void ASTStmtWriter::VisitSourceLocExpr(SourceLocExpr *E) {
1256 VisitExpr(E);
1257 Record.AddDeclRef(cast_or_null<Decl>(E->getParentContext()));
1258 Record.AddSourceLocation(E->getBeginLoc());
1259 Record.AddSourceLocation(E->getEndLoc());
1260 Record.push_back(llvm::to_underlying(E->getIdentKind()));
1261 Code = serialization::EXPR_SOURCE_LOC;
1262 }
1263
VisitAddrLabelExpr(AddrLabelExpr * E)1264 void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
1265 VisitExpr(E);
1266 Record.AddSourceLocation(E->getAmpAmpLoc());
1267 Record.AddSourceLocation(E->getLabelLoc());
1268 Record.AddDeclRef(E->getLabel());
1269 Code = serialization::EXPR_ADDR_LABEL;
1270 }
1271
VisitStmtExpr(StmtExpr * E)1272 void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
1273 VisitExpr(E);
1274 Record.AddStmt(E->getSubStmt());
1275 Record.AddSourceLocation(E->getLParenLoc());
1276 Record.AddSourceLocation(E->getRParenLoc());
1277 Record.push_back(E->getTemplateDepth());
1278 Code = serialization::EXPR_STMT;
1279 }
1280
VisitChooseExpr(ChooseExpr * E)1281 void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
1282 VisitExpr(E);
1283 Record.AddStmt(E->getCond());
1284 Record.AddStmt(E->getLHS());
1285 Record.AddStmt(E->getRHS());
1286 Record.AddSourceLocation(E->getBuiltinLoc());
1287 Record.AddSourceLocation(E->getRParenLoc());
1288 Record.push_back(E->isConditionDependent() ? false : E->isConditionTrue());
1289 Code = serialization::EXPR_CHOOSE;
1290 }
1291
VisitGNUNullExpr(GNUNullExpr * E)1292 void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
1293 VisitExpr(E);
1294 Record.AddSourceLocation(E->getTokenLocation());
1295 Code = serialization::EXPR_GNU_NULL;
1296 }
1297
VisitShuffleVectorExpr(ShuffleVectorExpr * E)1298 void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
1299 VisitExpr(E);
1300 Record.push_back(E->getNumSubExprs());
1301 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
1302 Record.AddStmt(E->getExpr(I));
1303 Record.AddSourceLocation(E->getBuiltinLoc());
1304 Record.AddSourceLocation(E->getRParenLoc());
1305 Code = serialization::EXPR_SHUFFLE_VECTOR;
1306 }
1307
VisitConvertVectorExpr(ConvertVectorExpr * E)1308 void ASTStmtWriter::VisitConvertVectorExpr(ConvertVectorExpr *E) {
1309 VisitExpr(E);
1310 Record.AddSourceLocation(E->getBuiltinLoc());
1311 Record.AddSourceLocation(E->getRParenLoc());
1312 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1313 Record.AddStmt(E->getSrcExpr());
1314 Code = serialization::EXPR_CONVERT_VECTOR;
1315 }
1316
VisitBlockExpr(BlockExpr * E)1317 void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
1318 VisitExpr(E);
1319 Record.AddDeclRef(E->getBlockDecl());
1320 Code = serialization::EXPR_BLOCK;
1321 }
1322
VisitGenericSelectionExpr(GenericSelectionExpr * E)1323 void ASTStmtWriter::VisitGenericSelectionExpr(GenericSelectionExpr *E) {
1324 VisitExpr(E);
1325
1326 Record.push_back(E->getNumAssocs());
1327 Record.push_back(E->isExprPredicate());
1328 Record.push_back(E->ResultIndex);
1329 Record.AddSourceLocation(E->getGenericLoc());
1330 Record.AddSourceLocation(E->getDefaultLoc());
1331 Record.AddSourceLocation(E->getRParenLoc());
1332
1333 Stmt **Stmts = E->getTrailingObjects<Stmt *>();
1334 // Add 1 to account for the controlling expression which is the first
1335 // expression in the trailing array of Stmt *. This is not needed for
1336 // the trailing array of TypeSourceInfo *.
1337 for (unsigned I = 0, N = E->getNumAssocs() + 1; I < N; ++I)
1338 Record.AddStmt(Stmts[I]);
1339
1340 TypeSourceInfo **TSIs = E->getTrailingObjects<TypeSourceInfo *>();
1341 for (unsigned I = 0, N = E->getNumAssocs(); I < N; ++I)
1342 Record.AddTypeSourceInfo(TSIs[I]);
1343
1344 Code = serialization::EXPR_GENERIC_SELECTION;
1345 }
1346
VisitPseudoObjectExpr(PseudoObjectExpr * E)1347 void ASTStmtWriter::VisitPseudoObjectExpr(PseudoObjectExpr *E) {
1348 VisitExpr(E);
1349 Record.push_back(E->getNumSemanticExprs());
1350
1351 // Push the result index. Currently, this needs to exactly match
1352 // the encoding used internally for ResultIndex.
1353 unsigned result = E->getResultExprIndex();
1354 result = (result == PseudoObjectExpr::NoResult ? 0 : result + 1);
1355 Record.push_back(result);
1356
1357 Record.AddStmt(E->getSyntacticForm());
1358 for (PseudoObjectExpr::semantics_iterator
1359 i = E->semantics_begin(), e = E->semantics_end(); i != e; ++i) {
1360 Record.AddStmt(*i);
1361 }
1362 Code = serialization::EXPR_PSEUDO_OBJECT;
1363 }
1364
VisitAtomicExpr(AtomicExpr * E)1365 void ASTStmtWriter::VisitAtomicExpr(AtomicExpr *E) {
1366 VisitExpr(E);
1367 Record.push_back(E->getOp());
1368 for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
1369 Record.AddStmt(E->getSubExprs()[I]);
1370 Record.AddSourceLocation(E->getBuiltinLoc());
1371 Record.AddSourceLocation(E->getRParenLoc());
1372 Code = serialization::EXPR_ATOMIC;
1373 }
1374
1375 //===----------------------------------------------------------------------===//
1376 // Objective-C Expressions and Statements.
1377 //===----------------------------------------------------------------------===//
1378
VisitObjCStringLiteral(ObjCStringLiteral * E)1379 void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
1380 VisitExpr(E);
1381 Record.AddStmt(E->getString());
1382 Record.AddSourceLocation(E->getAtLoc());
1383 Code = serialization::EXPR_OBJC_STRING_LITERAL;
1384 }
1385
VisitObjCBoxedExpr(ObjCBoxedExpr * E)1386 void ASTStmtWriter::VisitObjCBoxedExpr(ObjCBoxedExpr *E) {
1387 VisitExpr(E);
1388 Record.AddStmt(E->getSubExpr());
1389 Record.AddDeclRef(E->getBoxingMethod());
1390 Record.AddSourceRange(E->getSourceRange());
1391 Code = serialization::EXPR_OBJC_BOXED_EXPRESSION;
1392 }
1393
VisitObjCArrayLiteral(ObjCArrayLiteral * E)1394 void ASTStmtWriter::VisitObjCArrayLiteral(ObjCArrayLiteral *E) {
1395 VisitExpr(E);
1396 Record.push_back(E->getNumElements());
1397 for (unsigned i = 0; i < E->getNumElements(); i++)
1398 Record.AddStmt(E->getElement(i));
1399 Record.AddDeclRef(E->getArrayWithObjectsMethod());
1400 Record.AddSourceRange(E->getSourceRange());
1401 Code = serialization::EXPR_OBJC_ARRAY_LITERAL;
1402 }
1403
VisitObjCDictionaryLiteral(ObjCDictionaryLiteral * E)1404 void ASTStmtWriter::VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) {
1405 VisitExpr(E);
1406 Record.push_back(E->getNumElements());
1407 Record.push_back(E->HasPackExpansions);
1408 for (unsigned i = 0; i < E->getNumElements(); i++) {
1409 ObjCDictionaryElement Element = E->getKeyValueElement(i);
1410 Record.AddStmt(Element.Key);
1411 Record.AddStmt(Element.Value);
1412 if (E->HasPackExpansions) {
1413 Record.AddSourceLocation(Element.EllipsisLoc);
1414 unsigned NumExpansions = 0;
1415 if (Element.NumExpansions)
1416 NumExpansions = *Element.NumExpansions + 1;
1417 Record.push_back(NumExpansions);
1418 }
1419 }
1420
1421 Record.AddDeclRef(E->getDictWithObjectsMethod());
1422 Record.AddSourceRange(E->getSourceRange());
1423 Code = serialization::EXPR_OBJC_DICTIONARY_LITERAL;
1424 }
1425
VisitObjCEncodeExpr(ObjCEncodeExpr * E)1426 void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
1427 VisitExpr(E);
1428 Record.AddTypeSourceInfo(E->getEncodedTypeSourceInfo());
1429 Record.AddSourceLocation(E->getAtLoc());
1430 Record.AddSourceLocation(E->getRParenLoc());
1431 Code = serialization::EXPR_OBJC_ENCODE;
1432 }
1433
VisitObjCSelectorExpr(ObjCSelectorExpr * E)1434 void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
1435 VisitExpr(E);
1436 Record.AddSelectorRef(E->getSelector());
1437 Record.AddSourceLocation(E->getAtLoc());
1438 Record.AddSourceLocation(E->getRParenLoc());
1439 Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
1440 }
1441
VisitObjCProtocolExpr(ObjCProtocolExpr * E)1442 void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
1443 VisitExpr(E);
1444 Record.AddDeclRef(E->getProtocol());
1445 Record.AddSourceLocation(E->getAtLoc());
1446 Record.AddSourceLocation(E->ProtoLoc);
1447 Record.AddSourceLocation(E->getRParenLoc());
1448 Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
1449 }
1450
VisitObjCIvarRefExpr(ObjCIvarRefExpr * E)1451 void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
1452 VisitExpr(E);
1453 Record.AddDeclRef(E->getDecl());
1454 Record.AddSourceLocation(E->getLocation());
1455 Record.AddSourceLocation(E->getOpLoc());
1456 Record.AddStmt(E->getBase());
1457 Record.push_back(E->isArrow());
1458 Record.push_back(E->isFreeIvar());
1459 Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
1460 }
1461
VisitObjCPropertyRefExpr(ObjCPropertyRefExpr * E)1462 void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
1463 VisitExpr(E);
1464 Record.push_back(E->SetterAndMethodRefFlags.getInt());
1465 Record.push_back(E->isImplicitProperty());
1466 if (E->isImplicitProperty()) {
1467 Record.AddDeclRef(E->getImplicitPropertyGetter());
1468 Record.AddDeclRef(E->getImplicitPropertySetter());
1469 } else {
1470 Record.AddDeclRef(E->getExplicitProperty());
1471 }
1472 Record.AddSourceLocation(E->getLocation());
1473 Record.AddSourceLocation(E->getReceiverLocation());
1474 if (E->isObjectReceiver()) {
1475 Record.push_back(0);
1476 Record.AddStmt(E->getBase());
1477 } else if (E->isSuperReceiver()) {
1478 Record.push_back(1);
1479 Record.AddTypeRef(E->getSuperReceiverType());
1480 } else {
1481 Record.push_back(2);
1482 Record.AddDeclRef(E->getClassReceiver());
1483 }
1484
1485 Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
1486 }
1487
VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr * E)1488 void ASTStmtWriter::VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr *E) {
1489 VisitExpr(E);
1490 Record.AddSourceLocation(E->getRBracket());
1491 Record.AddStmt(E->getBaseExpr());
1492 Record.AddStmt(E->getKeyExpr());
1493 Record.AddDeclRef(E->getAtIndexMethodDecl());
1494 Record.AddDeclRef(E->setAtIndexMethodDecl());
1495
1496 Code = serialization::EXPR_OBJC_SUBSCRIPT_REF_EXPR;
1497 }
1498
VisitObjCMessageExpr(ObjCMessageExpr * E)1499 void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
1500 VisitExpr(E);
1501 Record.push_back(E->getNumArgs());
1502 Record.push_back(E->getNumStoredSelLocs());
1503 Record.push_back(E->SelLocsKind);
1504 Record.push_back(E->isDelegateInitCall());
1505 Record.push_back(E->IsImplicit);
1506 Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
1507 switch (E->getReceiverKind()) {
1508 case ObjCMessageExpr::Instance:
1509 Record.AddStmt(E->getInstanceReceiver());
1510 break;
1511
1512 case ObjCMessageExpr::Class:
1513 Record.AddTypeSourceInfo(E->getClassReceiverTypeInfo());
1514 break;
1515
1516 case ObjCMessageExpr::SuperClass:
1517 case ObjCMessageExpr::SuperInstance:
1518 Record.AddTypeRef(E->getSuperType());
1519 Record.AddSourceLocation(E->getSuperLoc());
1520 break;
1521 }
1522
1523 if (E->getMethodDecl()) {
1524 Record.push_back(1);
1525 Record.AddDeclRef(E->getMethodDecl());
1526 } else {
1527 Record.push_back(0);
1528 Record.AddSelectorRef(E->getSelector());
1529 }
1530
1531 Record.AddSourceLocation(E->getLeftLoc());
1532 Record.AddSourceLocation(E->getRightLoc());
1533
1534 for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
1535 Arg != ArgEnd; ++Arg)
1536 Record.AddStmt(*Arg);
1537
1538 SourceLocation *Locs = E->getStoredSelLocs();
1539 for (unsigned i = 0, e = E->getNumStoredSelLocs(); i != e; ++i)
1540 Record.AddSourceLocation(Locs[i]);
1541
1542 Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
1543 }
1544
VisitObjCForCollectionStmt(ObjCForCollectionStmt * S)1545 void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
1546 VisitStmt(S);
1547 Record.AddStmt(S->getElement());
1548 Record.AddStmt(S->getCollection());
1549 Record.AddStmt(S->getBody());
1550 Record.AddSourceLocation(S->getForLoc());
1551 Record.AddSourceLocation(S->getRParenLoc());
1552 Code = serialization::STMT_OBJC_FOR_COLLECTION;
1553 }
1554
VisitObjCAtCatchStmt(ObjCAtCatchStmt * S)1555 void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
1556 VisitStmt(S);
1557 Record.AddStmt(S->getCatchBody());
1558 Record.AddDeclRef(S->getCatchParamDecl());
1559 Record.AddSourceLocation(S->getAtCatchLoc());
1560 Record.AddSourceLocation(S->getRParenLoc());
1561 Code = serialization::STMT_OBJC_CATCH;
1562 }
1563
VisitObjCAtFinallyStmt(ObjCAtFinallyStmt * S)1564 void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
1565 VisitStmt(S);
1566 Record.AddStmt(S->getFinallyBody());
1567 Record.AddSourceLocation(S->getAtFinallyLoc());
1568 Code = serialization::STMT_OBJC_FINALLY;
1569 }
1570
VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt * S)1571 void ASTStmtWriter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *S) {
1572 VisitStmt(S); // FIXME: no test coverage.
1573 Record.AddStmt(S->getSubStmt());
1574 Record.AddSourceLocation(S->getAtLoc());
1575 Code = serialization::STMT_OBJC_AUTORELEASE_POOL;
1576 }
1577
VisitObjCAtTryStmt(ObjCAtTryStmt * S)1578 void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
1579 VisitStmt(S);
1580 Record.push_back(S->getNumCatchStmts());
1581 Record.push_back(S->getFinallyStmt() != nullptr);
1582 Record.AddStmt(S->getTryBody());
1583 for (ObjCAtCatchStmt *C : S->catch_stmts())
1584 Record.AddStmt(C);
1585 if (S->getFinallyStmt())
1586 Record.AddStmt(S->getFinallyStmt());
1587 Record.AddSourceLocation(S->getAtTryLoc());
1588 Code = serialization::STMT_OBJC_AT_TRY;
1589 }
1590
VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt * S)1591 void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
1592 VisitStmt(S); // FIXME: no test coverage.
1593 Record.AddStmt(S->getSynchExpr());
1594 Record.AddStmt(S->getSynchBody());
1595 Record.AddSourceLocation(S->getAtSynchronizedLoc());
1596 Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
1597 }
1598
VisitObjCAtThrowStmt(ObjCAtThrowStmt * S)1599 void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
1600 VisitStmt(S); // FIXME: no test coverage.
1601 Record.AddStmt(S->getThrowExpr());
1602 Record.AddSourceLocation(S->getThrowLoc());
1603 Code = serialization::STMT_OBJC_AT_THROW;
1604 }
1605
VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr * E)1606 void ASTStmtWriter::VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr *E) {
1607 VisitExpr(E);
1608 Record.push_back(E->getValue());
1609 Record.AddSourceLocation(E->getLocation());
1610 Code = serialization::EXPR_OBJC_BOOL_LITERAL;
1611 }
1612
VisitObjCAvailabilityCheckExpr(ObjCAvailabilityCheckExpr * E)1613 void ASTStmtWriter::VisitObjCAvailabilityCheckExpr(ObjCAvailabilityCheckExpr *E) {
1614 VisitExpr(E);
1615 Record.AddSourceRange(E->getSourceRange());
1616 Record.AddVersionTuple(E->getVersion());
1617 Code = serialization::EXPR_OBJC_AVAILABILITY_CHECK;
1618 }
1619
1620 //===----------------------------------------------------------------------===//
1621 // C++ Expressions and Statements.
1622 //===----------------------------------------------------------------------===//
1623
VisitCXXCatchStmt(CXXCatchStmt * S)1624 void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
1625 VisitStmt(S);
1626 Record.AddSourceLocation(S->getCatchLoc());
1627 Record.AddDeclRef(S->getExceptionDecl());
1628 Record.AddStmt(S->getHandlerBlock());
1629 Code = serialization::STMT_CXX_CATCH;
1630 }
1631
VisitCXXTryStmt(CXXTryStmt * S)1632 void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
1633 VisitStmt(S);
1634 Record.push_back(S->getNumHandlers());
1635 Record.AddSourceLocation(S->getTryLoc());
1636 Record.AddStmt(S->getTryBlock());
1637 for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
1638 Record.AddStmt(S->getHandler(i));
1639 Code = serialization::STMT_CXX_TRY;
1640 }
1641
VisitCXXForRangeStmt(CXXForRangeStmt * S)1642 void ASTStmtWriter::VisitCXXForRangeStmt(CXXForRangeStmt *S) {
1643 VisitStmt(S);
1644 Record.AddSourceLocation(S->getForLoc());
1645 Record.AddSourceLocation(S->getCoawaitLoc());
1646 Record.AddSourceLocation(S->getColonLoc());
1647 Record.AddSourceLocation(S->getRParenLoc());
1648 Record.AddStmt(S->getInit());
1649 Record.AddStmt(S->getRangeStmt());
1650 Record.AddStmt(S->getBeginStmt());
1651 Record.AddStmt(S->getEndStmt());
1652 Record.AddStmt(S->getCond());
1653 Record.AddStmt(S->getInc());
1654 Record.AddStmt(S->getLoopVarStmt());
1655 Record.AddStmt(S->getBody());
1656 Code = serialization::STMT_CXX_FOR_RANGE;
1657 }
1658
VisitMSDependentExistsStmt(MSDependentExistsStmt * S)1659 void ASTStmtWriter::VisitMSDependentExistsStmt(MSDependentExistsStmt *S) {
1660 VisitStmt(S);
1661 Record.AddSourceLocation(S->getKeywordLoc());
1662 Record.push_back(S->isIfExists());
1663 Record.AddNestedNameSpecifierLoc(S->getQualifierLoc());
1664 Record.AddDeclarationNameInfo(S->getNameInfo());
1665 Record.AddStmt(S->getSubStmt());
1666 Code = serialization::STMT_MS_DEPENDENT_EXISTS;
1667 }
1668
VisitCXXOperatorCallExpr(CXXOperatorCallExpr * E)1669 void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
1670 VisitCallExpr(E);
1671 Record.push_back(E->getOperator());
1672 Record.AddSourceRange(E->Range);
1673
1674 if (!E->hasStoredFPFeatures() && !static_cast<bool>(E->getADLCallKind()))
1675 AbbrevToUse = Writer.getCXXOperatorCallExprAbbrev();
1676
1677 Code = serialization::EXPR_CXX_OPERATOR_CALL;
1678 }
1679
VisitCXXMemberCallExpr(CXXMemberCallExpr * E)1680 void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
1681 VisitCallExpr(E);
1682
1683 if (!E->hasStoredFPFeatures() && !static_cast<bool>(E->getADLCallKind()))
1684 AbbrevToUse = Writer.getCXXMemberCallExprAbbrev();
1685
1686 Code = serialization::EXPR_CXX_MEMBER_CALL;
1687 }
1688
VisitCXXRewrittenBinaryOperator(CXXRewrittenBinaryOperator * E)1689 void ASTStmtWriter::VisitCXXRewrittenBinaryOperator(
1690 CXXRewrittenBinaryOperator *E) {
1691 VisitExpr(E);
1692 Record.push_back(E->isReversed());
1693 Record.AddStmt(E->getSemanticForm());
1694 Code = serialization::EXPR_CXX_REWRITTEN_BINARY_OPERATOR;
1695 }
1696
VisitCXXConstructExpr(CXXConstructExpr * E)1697 void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
1698 VisitExpr(E);
1699
1700 Record.push_back(E->getNumArgs());
1701 Record.push_back(E->isElidable());
1702 Record.push_back(E->hadMultipleCandidates());
1703 Record.push_back(E->isListInitialization());
1704 Record.push_back(E->isStdInitListInitialization());
1705 Record.push_back(E->requiresZeroInitialization());
1706 Record.push_back(
1707 llvm::to_underlying(E->getConstructionKind())); // FIXME: stable encoding
1708 Record.push_back(E->isImmediateEscalating());
1709 Record.AddSourceLocation(E->getLocation());
1710 Record.AddDeclRef(E->getConstructor());
1711 Record.AddSourceRange(E->getParenOrBraceRange());
1712
1713 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
1714 Record.AddStmt(E->getArg(I));
1715
1716 Code = serialization::EXPR_CXX_CONSTRUCT;
1717 }
1718
VisitCXXInheritedCtorInitExpr(CXXInheritedCtorInitExpr * E)1719 void ASTStmtWriter::VisitCXXInheritedCtorInitExpr(CXXInheritedCtorInitExpr *E) {
1720 VisitExpr(E);
1721 Record.AddDeclRef(E->getConstructor());
1722 Record.AddSourceLocation(E->getLocation());
1723 Record.push_back(E->constructsVBase());
1724 Record.push_back(E->inheritedFromVBase());
1725 Code = serialization::EXPR_CXX_INHERITED_CTOR_INIT;
1726 }
1727
VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr * E)1728 void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
1729 VisitCXXConstructExpr(E);
1730 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1731 Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
1732 }
1733
VisitLambdaExpr(LambdaExpr * E)1734 void ASTStmtWriter::VisitLambdaExpr(LambdaExpr *E) {
1735 VisitExpr(E);
1736 Record.push_back(E->LambdaExprBits.NumCaptures);
1737 Record.AddSourceRange(E->IntroducerRange);
1738 Record.push_back(E->LambdaExprBits.CaptureDefault); // FIXME: stable encoding
1739 Record.AddSourceLocation(E->CaptureDefaultLoc);
1740 Record.push_back(E->LambdaExprBits.ExplicitParams);
1741 Record.push_back(E->LambdaExprBits.ExplicitResultType);
1742 Record.AddSourceLocation(E->ClosingBrace);
1743
1744 // Add capture initializers.
1745 for (LambdaExpr::capture_init_iterator C = E->capture_init_begin(),
1746 CEnd = E->capture_init_end();
1747 C != CEnd; ++C) {
1748 Record.AddStmt(*C);
1749 }
1750
1751 // Don't serialize the body. It belongs to the call operator declaration.
1752 // LambdaExpr only stores a copy of the Stmt *.
1753
1754 Code = serialization::EXPR_LAMBDA;
1755 }
1756
VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr * E)1757 void ASTStmtWriter::VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E) {
1758 VisitExpr(E);
1759 Record.AddStmt(E->getSubExpr());
1760 Code = serialization::EXPR_CXX_STD_INITIALIZER_LIST;
1761 }
1762
VisitCXXNamedCastExpr(CXXNamedCastExpr * E)1763 void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
1764 VisitExplicitCastExpr(E);
1765 Record.AddSourceRange(SourceRange(E->getOperatorLoc(), E->getRParenLoc()));
1766 CurrentPackingBits.addBit(E->getAngleBrackets().isValid());
1767 if (E->getAngleBrackets().isValid())
1768 Record.AddSourceRange(E->getAngleBrackets());
1769 }
1770
VisitCXXStaticCastExpr(CXXStaticCastExpr * E)1771 void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
1772 VisitCXXNamedCastExpr(E);
1773 Code = serialization::EXPR_CXX_STATIC_CAST;
1774 }
1775
VisitCXXDynamicCastExpr(CXXDynamicCastExpr * E)1776 void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
1777 VisitCXXNamedCastExpr(E);
1778 Code = serialization::EXPR_CXX_DYNAMIC_CAST;
1779 }
1780
VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr * E)1781 void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
1782 VisitCXXNamedCastExpr(E);
1783 Code = serialization::EXPR_CXX_REINTERPRET_CAST;
1784 }
1785
VisitCXXConstCastExpr(CXXConstCastExpr * E)1786 void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
1787 VisitCXXNamedCastExpr(E);
1788 Code = serialization::EXPR_CXX_CONST_CAST;
1789 }
1790
VisitCXXAddrspaceCastExpr(CXXAddrspaceCastExpr * E)1791 void ASTStmtWriter::VisitCXXAddrspaceCastExpr(CXXAddrspaceCastExpr *E) {
1792 VisitCXXNamedCastExpr(E);
1793 Code = serialization::EXPR_CXX_ADDRSPACE_CAST;
1794 }
1795
VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr * E)1796 void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
1797 VisitExplicitCastExpr(E);
1798 Record.AddSourceLocation(E->getLParenLoc());
1799 Record.AddSourceLocation(E->getRParenLoc());
1800 Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
1801 }
1802
VisitBuiltinBitCastExpr(BuiltinBitCastExpr * E)1803 void ASTStmtWriter::VisitBuiltinBitCastExpr(BuiltinBitCastExpr *E) {
1804 VisitExplicitCastExpr(E);
1805 Record.AddSourceLocation(E->getBeginLoc());
1806 Record.AddSourceLocation(E->getEndLoc());
1807 Code = serialization::EXPR_BUILTIN_BIT_CAST;
1808 }
1809
VisitUserDefinedLiteral(UserDefinedLiteral * E)1810 void ASTStmtWriter::VisitUserDefinedLiteral(UserDefinedLiteral *E) {
1811 VisitCallExpr(E);
1812 Record.AddSourceLocation(E->UDSuffixLoc);
1813 Code = serialization::EXPR_USER_DEFINED_LITERAL;
1814 }
1815
VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr * E)1816 void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
1817 VisitExpr(E);
1818 Record.push_back(E->getValue());
1819 Record.AddSourceLocation(E->getLocation());
1820 Code = serialization::EXPR_CXX_BOOL_LITERAL;
1821 }
1822
VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr * E)1823 void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
1824 VisitExpr(E);
1825 Record.AddSourceLocation(E->getLocation());
1826 Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
1827 }
1828
VisitCXXTypeidExpr(CXXTypeidExpr * E)1829 void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1830 VisitExpr(E);
1831 Record.AddSourceRange(E->getSourceRange());
1832 if (E->isTypeOperand()) {
1833 Record.AddTypeSourceInfo(E->getTypeOperandSourceInfo());
1834 Code = serialization::EXPR_CXX_TYPEID_TYPE;
1835 } else {
1836 Record.AddStmt(E->getExprOperand());
1837 Code = serialization::EXPR_CXX_TYPEID_EXPR;
1838 }
1839 }
1840
VisitCXXThisExpr(CXXThisExpr * E)1841 void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
1842 VisitExpr(E);
1843 Record.AddSourceLocation(E->getLocation());
1844 Record.push_back(E->isImplicit());
1845
1846 Code = serialization::EXPR_CXX_THIS;
1847 }
1848
VisitCXXThrowExpr(CXXThrowExpr * E)1849 void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
1850 VisitExpr(E);
1851 Record.AddSourceLocation(E->getThrowLoc());
1852 Record.AddStmt(E->getSubExpr());
1853 Record.push_back(E->isThrownVariableInScope());
1854 Code = serialization::EXPR_CXX_THROW;
1855 }
1856
VisitCXXDefaultArgExpr(CXXDefaultArgExpr * E)1857 void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
1858 VisitExpr(E);
1859 Record.AddDeclRef(E->getParam());
1860 Record.AddDeclRef(cast_or_null<Decl>(E->getUsedContext()));
1861 Record.AddSourceLocation(E->getUsedLocation());
1862 Record.push_back(E->hasRewrittenInit());
1863 if (E->hasRewrittenInit())
1864 Record.AddStmt(E->getRewrittenExpr());
1865 Code = serialization::EXPR_CXX_DEFAULT_ARG;
1866 }
1867
VisitCXXDefaultInitExpr(CXXDefaultInitExpr * E)1868 void ASTStmtWriter::VisitCXXDefaultInitExpr(CXXDefaultInitExpr *E) {
1869 VisitExpr(E);
1870 Record.push_back(E->hasRewrittenInit());
1871 Record.AddDeclRef(E->getField());
1872 Record.AddDeclRef(cast_or_null<Decl>(E->getUsedContext()));
1873 Record.AddSourceLocation(E->getExprLoc());
1874 if (E->hasRewrittenInit())
1875 Record.AddStmt(E->getRewrittenExpr());
1876 Code = serialization::EXPR_CXX_DEFAULT_INIT;
1877 }
1878
VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr * E)1879 void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
1880 VisitExpr(E);
1881 Record.AddCXXTemporary(E->getTemporary());
1882 Record.AddStmt(E->getSubExpr());
1883 Code = serialization::EXPR_CXX_BIND_TEMPORARY;
1884 }
1885
VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr * E)1886 void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
1887 VisitExpr(E);
1888 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1889 Record.AddSourceLocation(E->getRParenLoc());
1890 Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
1891 }
1892
VisitCXXNewExpr(CXXNewExpr * E)1893 void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
1894 VisitExpr(E);
1895
1896 Record.push_back(E->isArray());
1897 Record.push_back(E->hasInitializer());
1898 Record.push_back(E->getNumPlacementArgs());
1899 Record.push_back(E->isParenTypeId());
1900
1901 Record.push_back(E->isGlobalNew());
1902 Record.push_back(E->passAlignment());
1903 Record.push_back(E->doesUsualArrayDeleteWantSize());
1904 Record.push_back(E->CXXNewExprBits.HasInitializer);
1905 Record.push_back(E->CXXNewExprBits.StoredInitializationStyle);
1906
1907 Record.AddDeclRef(E->getOperatorNew());
1908 Record.AddDeclRef(E->getOperatorDelete());
1909 Record.AddTypeSourceInfo(E->getAllocatedTypeSourceInfo());
1910 if (E->isParenTypeId())
1911 Record.AddSourceRange(E->getTypeIdParens());
1912 Record.AddSourceRange(E->getSourceRange());
1913 Record.AddSourceRange(E->getDirectInitRange());
1914
1915 for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), N = E->raw_arg_end();
1916 I != N; ++I)
1917 Record.AddStmt(*I);
1918
1919 Code = serialization::EXPR_CXX_NEW;
1920 }
1921
VisitCXXDeleteExpr(CXXDeleteExpr * E)1922 void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
1923 VisitExpr(E);
1924 Record.push_back(E->isGlobalDelete());
1925 Record.push_back(E->isArrayForm());
1926 Record.push_back(E->isArrayFormAsWritten());
1927 Record.push_back(E->doesUsualArrayDeleteWantSize());
1928 Record.AddDeclRef(E->getOperatorDelete());
1929 Record.AddStmt(E->getArgument());
1930 Record.AddSourceLocation(E->getBeginLoc());
1931
1932 Code = serialization::EXPR_CXX_DELETE;
1933 }
1934
VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr * E)1935 void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1936 VisitExpr(E);
1937
1938 Record.AddStmt(E->getBase());
1939 Record.push_back(E->isArrow());
1940 Record.AddSourceLocation(E->getOperatorLoc());
1941 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1942 Record.AddTypeSourceInfo(E->getScopeTypeInfo());
1943 Record.AddSourceLocation(E->getColonColonLoc());
1944 Record.AddSourceLocation(E->getTildeLoc());
1945
1946 // PseudoDestructorTypeStorage.
1947 Record.AddIdentifierRef(E->getDestroyedTypeIdentifier());
1948 if (E->getDestroyedTypeIdentifier())
1949 Record.AddSourceLocation(E->getDestroyedTypeLoc());
1950 else
1951 Record.AddTypeSourceInfo(E->getDestroyedTypeInfo());
1952
1953 Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
1954 }
1955
VisitExprWithCleanups(ExprWithCleanups * E)1956 void ASTStmtWriter::VisitExprWithCleanups(ExprWithCleanups *E) {
1957 VisitExpr(E);
1958 Record.push_back(E->getNumObjects());
1959 for (auto &Obj : E->getObjects()) {
1960 if (auto *BD = Obj.dyn_cast<BlockDecl *>()) {
1961 Record.push_back(serialization::COK_Block);
1962 Record.AddDeclRef(BD);
1963 } else if (auto *CLE = Obj.dyn_cast<CompoundLiteralExpr *>()) {
1964 Record.push_back(serialization::COK_CompoundLiteral);
1965 Record.AddStmt(CLE);
1966 }
1967 }
1968
1969 Record.push_back(E->cleanupsHaveSideEffects());
1970 Record.AddStmt(E->getSubExpr());
1971 Code = serialization::EXPR_EXPR_WITH_CLEANUPS;
1972 }
1973
VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr * E)1974 void ASTStmtWriter::VisitCXXDependentScopeMemberExpr(
1975 CXXDependentScopeMemberExpr *E) {
1976 VisitExpr(E);
1977
1978 // Don't emit anything here (or if you do you will have to update
1979 // the corresponding deserialization function).
1980 Record.push_back(E->getNumTemplateArgs());
1981 CurrentPackingBits.updateBits();
1982 CurrentPackingBits.addBit(E->hasTemplateKWAndArgsInfo());
1983 CurrentPackingBits.addBit(E->hasFirstQualifierFoundInScope());
1984
1985 if (E->hasTemplateKWAndArgsInfo()) {
1986 const ASTTemplateKWAndArgsInfo &ArgInfo =
1987 *E->getTrailingObjects<ASTTemplateKWAndArgsInfo>();
1988 AddTemplateKWAndArgsInfo(ArgInfo,
1989 E->getTrailingObjects<TemplateArgumentLoc>());
1990 }
1991
1992 CurrentPackingBits.addBit(E->isArrow());
1993
1994 Record.AddTypeRef(E->getBaseType());
1995 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1996 CurrentPackingBits.addBit(!E->isImplicitAccess());
1997 if (!E->isImplicitAccess())
1998 Record.AddStmt(E->getBase());
1999
2000 Record.AddSourceLocation(E->getOperatorLoc());
2001
2002 if (E->hasFirstQualifierFoundInScope())
2003 Record.AddDeclRef(E->getFirstQualifierFoundInScope());
2004
2005 Record.AddDeclarationNameInfo(E->MemberNameInfo);
2006 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
2007 }
2008
2009 void
VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr * E)2010 ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
2011 VisitExpr(E);
2012
2013 // Don't emit anything here, HasTemplateKWAndArgsInfo must be
2014 // emitted first.
2015 CurrentPackingBits.addBit(
2016 E->DependentScopeDeclRefExprBits.HasTemplateKWAndArgsInfo);
2017
2018 if (E->DependentScopeDeclRefExprBits.HasTemplateKWAndArgsInfo) {
2019 const ASTTemplateKWAndArgsInfo &ArgInfo =
2020 *E->getTrailingObjects<ASTTemplateKWAndArgsInfo>();
2021 // 16 bits should be enought to store the number of args
2022 CurrentPackingBits.addBits(ArgInfo.NumTemplateArgs, /*Width=*/16);
2023 AddTemplateKWAndArgsInfo(ArgInfo,
2024 E->getTrailingObjects<TemplateArgumentLoc>());
2025 }
2026
2027 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
2028 Record.AddDeclarationNameInfo(E->NameInfo);
2029 Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
2030 }
2031
2032 void
VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr * E)2033 ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
2034 VisitExpr(E);
2035 Record.push_back(E->getNumArgs());
2036 for (CXXUnresolvedConstructExpr::arg_iterator
2037 ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
2038 Record.AddStmt(*ArgI);
2039 Record.AddTypeSourceInfo(E->getTypeSourceInfo());
2040 Record.AddSourceLocation(E->getLParenLoc());
2041 Record.AddSourceLocation(E->getRParenLoc());
2042 Record.push_back(E->isListInitialization());
2043 Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
2044 }
2045
VisitOverloadExpr(OverloadExpr * E)2046 void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
2047 VisitExpr(E);
2048
2049 Record.push_back(E->getNumDecls());
2050
2051 CurrentPackingBits.updateBits();
2052 CurrentPackingBits.addBit(E->hasTemplateKWAndArgsInfo());
2053 if (E->hasTemplateKWAndArgsInfo()) {
2054 const ASTTemplateKWAndArgsInfo &ArgInfo =
2055 *E->getTrailingASTTemplateKWAndArgsInfo();
2056 Record.push_back(ArgInfo.NumTemplateArgs);
2057 AddTemplateKWAndArgsInfo(ArgInfo, E->getTrailingTemplateArgumentLoc());
2058 }
2059
2060 for (OverloadExpr::decls_iterator OvI = E->decls_begin(),
2061 OvE = E->decls_end();
2062 OvI != OvE; ++OvI) {
2063 Record.AddDeclRef(OvI.getDecl());
2064 Record.push_back(OvI.getAccess());
2065 }
2066
2067 Record.AddDeclarationNameInfo(E->getNameInfo());
2068 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
2069 }
2070
VisitUnresolvedMemberExpr(UnresolvedMemberExpr * E)2071 void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
2072 VisitOverloadExpr(E);
2073 CurrentPackingBits.addBit(E->isArrow());
2074 CurrentPackingBits.addBit(E->hasUnresolvedUsing());
2075 CurrentPackingBits.addBit(!E->isImplicitAccess());
2076 if (!E->isImplicitAccess())
2077 Record.AddStmt(E->getBase());
2078
2079 Record.AddSourceLocation(E->getOperatorLoc());
2080
2081 Record.AddTypeRef(E->getBaseType());
2082 Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
2083 }
2084
VisitUnresolvedLookupExpr(UnresolvedLookupExpr * E)2085 void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
2086 VisitOverloadExpr(E);
2087 CurrentPackingBits.addBit(E->requiresADL());
2088 CurrentPackingBits.addBit(E->isOverloaded());
2089 Record.AddDeclRef(E->getNamingClass());
2090 Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
2091 }
2092
VisitTypeTraitExpr(TypeTraitExpr * E)2093 void ASTStmtWriter::VisitTypeTraitExpr(TypeTraitExpr *E) {
2094 VisitExpr(E);
2095 Record.push_back(E->TypeTraitExprBits.NumArgs);
2096 Record.push_back(E->TypeTraitExprBits.Kind); // FIXME: Stable encoding
2097 Record.push_back(E->TypeTraitExprBits.Value);
2098 Record.AddSourceRange(E->getSourceRange());
2099 for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
2100 Record.AddTypeSourceInfo(E->getArg(I));
2101 Code = serialization::EXPR_TYPE_TRAIT;
2102 }
2103
VisitArrayTypeTraitExpr(ArrayTypeTraitExpr * E)2104 void ASTStmtWriter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
2105 VisitExpr(E);
2106 Record.push_back(E->getTrait());
2107 Record.push_back(E->getValue());
2108 Record.AddSourceRange(E->getSourceRange());
2109 Record.AddTypeSourceInfo(E->getQueriedTypeSourceInfo());
2110 Record.AddStmt(E->getDimensionExpression());
2111 Code = serialization::EXPR_ARRAY_TYPE_TRAIT;
2112 }
2113
VisitExpressionTraitExpr(ExpressionTraitExpr * E)2114 void ASTStmtWriter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
2115 VisitExpr(E);
2116 Record.push_back(E->getTrait());
2117 Record.push_back(E->getValue());
2118 Record.AddSourceRange(E->getSourceRange());
2119 Record.AddStmt(E->getQueriedExpression());
2120 Code = serialization::EXPR_CXX_EXPRESSION_TRAIT;
2121 }
2122
VisitCXXNoexceptExpr(CXXNoexceptExpr * E)2123 void ASTStmtWriter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
2124 VisitExpr(E);
2125 Record.push_back(E->getValue());
2126 Record.AddSourceRange(E->getSourceRange());
2127 Record.AddStmt(E->getOperand());
2128 Code = serialization::EXPR_CXX_NOEXCEPT;
2129 }
2130
VisitPackExpansionExpr(PackExpansionExpr * E)2131 void ASTStmtWriter::VisitPackExpansionExpr(PackExpansionExpr *E) {
2132 VisitExpr(E);
2133 Record.AddSourceLocation(E->getEllipsisLoc());
2134 Record.push_back(E->NumExpansions);
2135 Record.AddStmt(E->getPattern());
2136 Code = serialization::EXPR_PACK_EXPANSION;
2137 }
2138
VisitSizeOfPackExpr(SizeOfPackExpr * E)2139 void ASTStmtWriter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
2140 VisitExpr(E);
2141 Record.push_back(E->isPartiallySubstituted() ? E->getPartialArguments().size()
2142 : 0);
2143 Record.AddSourceLocation(E->OperatorLoc);
2144 Record.AddSourceLocation(E->PackLoc);
2145 Record.AddSourceLocation(E->RParenLoc);
2146 Record.AddDeclRef(E->Pack);
2147 if (E->isPartiallySubstituted()) {
2148 for (const auto &TA : E->getPartialArguments())
2149 Record.AddTemplateArgument(TA);
2150 } else if (!E->isValueDependent()) {
2151 Record.push_back(E->getPackLength());
2152 }
2153 Code = serialization::EXPR_SIZEOF_PACK;
2154 }
2155
VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr * E)2156 void ASTStmtWriter::VisitSubstNonTypeTemplateParmExpr(
2157 SubstNonTypeTemplateParmExpr *E) {
2158 VisitExpr(E);
2159 Record.AddDeclRef(E->getAssociatedDecl());
2160 CurrentPackingBits.addBit(E->isReferenceParameter());
2161 CurrentPackingBits.addBits(E->getIndex(), /*Width=*/12);
2162 CurrentPackingBits.addBit((bool)E->getPackIndex());
2163 if (auto PackIndex = E->getPackIndex())
2164 Record.push_back(*PackIndex + 1);
2165
2166 Record.AddSourceLocation(E->getNameLoc());
2167 Record.AddStmt(E->getReplacement());
2168 Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM;
2169 }
2170
VisitSubstNonTypeTemplateParmPackExpr(SubstNonTypeTemplateParmPackExpr * E)2171 void ASTStmtWriter::VisitSubstNonTypeTemplateParmPackExpr(
2172 SubstNonTypeTemplateParmPackExpr *E) {
2173 VisitExpr(E);
2174 Record.AddDeclRef(E->getAssociatedDecl());
2175 Record.push_back(E->getIndex());
2176 Record.AddTemplateArgument(E->getArgumentPack());
2177 Record.AddSourceLocation(E->getParameterPackLocation());
2178 Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM_PACK;
2179 }
2180
VisitFunctionParmPackExpr(FunctionParmPackExpr * E)2181 void ASTStmtWriter::VisitFunctionParmPackExpr(FunctionParmPackExpr *E) {
2182 VisitExpr(E);
2183 Record.push_back(E->getNumExpansions());
2184 Record.AddDeclRef(E->getParameterPack());
2185 Record.AddSourceLocation(E->getParameterPackLocation());
2186 for (FunctionParmPackExpr::iterator I = E->begin(), End = E->end();
2187 I != End; ++I)
2188 Record.AddDeclRef(*I);
2189 Code = serialization::EXPR_FUNCTION_PARM_PACK;
2190 }
2191
VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr * E)2192 void ASTStmtWriter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E) {
2193 VisitExpr(E);
2194 Record.push_back(static_cast<bool>(E->getLifetimeExtendedTemporaryDecl()));
2195 if (E->getLifetimeExtendedTemporaryDecl())
2196 Record.AddDeclRef(E->getLifetimeExtendedTemporaryDecl());
2197 else
2198 Record.AddStmt(E->getSubExpr());
2199 Code = serialization::EXPR_MATERIALIZE_TEMPORARY;
2200 }
2201
VisitCXXFoldExpr(CXXFoldExpr * E)2202 void ASTStmtWriter::VisitCXXFoldExpr(CXXFoldExpr *E) {
2203 VisitExpr(E);
2204 Record.AddSourceLocation(E->LParenLoc);
2205 Record.AddSourceLocation(E->EllipsisLoc);
2206 Record.AddSourceLocation(E->RParenLoc);
2207 Record.push_back(E->NumExpansions);
2208 Record.AddStmt(E->SubExprs[0]);
2209 Record.AddStmt(E->SubExprs[1]);
2210 Record.AddStmt(E->SubExprs[2]);
2211 Record.push_back(E->Opcode);
2212 Code = serialization::EXPR_CXX_FOLD;
2213 }
2214
VisitCXXParenListInitExpr(CXXParenListInitExpr * E)2215 void ASTStmtWriter::VisitCXXParenListInitExpr(CXXParenListInitExpr *E) {
2216 VisitExpr(E);
2217 ArrayRef<Expr *> InitExprs = E->getInitExprs();
2218 Record.push_back(InitExprs.size());
2219 Record.push_back(E->getUserSpecifiedInitExprs().size());
2220 Record.AddSourceLocation(E->getInitLoc());
2221 Record.AddSourceLocation(E->getBeginLoc());
2222 Record.AddSourceLocation(E->getEndLoc());
2223 for (Expr *InitExpr : E->getInitExprs())
2224 Record.AddStmt(InitExpr);
2225 Expr *ArrayFiller = E->getArrayFiller();
2226 FieldDecl *UnionField = E->getInitializedFieldInUnion();
2227 bool HasArrayFillerOrUnionDecl = ArrayFiller || UnionField;
2228 Record.push_back(HasArrayFillerOrUnionDecl);
2229 if (HasArrayFillerOrUnionDecl) {
2230 Record.push_back(static_cast<bool>(ArrayFiller));
2231 if (ArrayFiller)
2232 Record.AddStmt(ArrayFiller);
2233 else
2234 Record.AddDeclRef(UnionField);
2235 }
2236 Code = serialization::EXPR_CXX_PAREN_LIST_INIT;
2237 }
2238
VisitOpaqueValueExpr(OpaqueValueExpr * E)2239 void ASTStmtWriter::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
2240 VisitExpr(E);
2241 Record.AddStmt(E->getSourceExpr());
2242 Record.AddSourceLocation(E->getLocation());
2243 Record.push_back(E->isUnique());
2244 Code = serialization::EXPR_OPAQUE_VALUE;
2245 }
2246
VisitTypoExpr(TypoExpr * E)2247 void ASTStmtWriter::VisitTypoExpr(TypoExpr *E) {
2248 VisitExpr(E);
2249 // TODO: Figure out sane writer behavior for a TypoExpr, if necessary
2250 llvm_unreachable("Cannot write TypoExpr nodes");
2251 }
2252
2253 //===----------------------------------------------------------------------===//
2254 // CUDA Expressions and Statements.
2255 //===----------------------------------------------------------------------===//
2256
VisitCUDAKernelCallExpr(CUDAKernelCallExpr * E)2257 void ASTStmtWriter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *E) {
2258 VisitCallExpr(E);
2259 Record.AddStmt(E->getConfig());
2260 Code = serialization::EXPR_CUDA_KERNEL_CALL;
2261 }
2262
2263 //===----------------------------------------------------------------------===//
2264 // OpenCL Expressions and Statements.
2265 //===----------------------------------------------------------------------===//
VisitAsTypeExpr(AsTypeExpr * E)2266 void ASTStmtWriter::VisitAsTypeExpr(AsTypeExpr *E) {
2267 VisitExpr(E);
2268 Record.AddSourceLocation(E->getBuiltinLoc());
2269 Record.AddSourceLocation(E->getRParenLoc());
2270 Record.AddStmt(E->getSrcExpr());
2271 Code = serialization::EXPR_ASTYPE;
2272 }
2273
2274 //===----------------------------------------------------------------------===//
2275 // Microsoft Expressions and Statements.
2276 //===----------------------------------------------------------------------===//
VisitMSPropertyRefExpr(MSPropertyRefExpr * E)2277 void ASTStmtWriter::VisitMSPropertyRefExpr(MSPropertyRefExpr *E) {
2278 VisitExpr(E);
2279 Record.push_back(E->isArrow());
2280 Record.AddStmt(E->getBaseExpr());
2281 Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
2282 Record.AddSourceLocation(E->getMemberLoc());
2283 Record.AddDeclRef(E->getPropertyDecl());
2284 Code = serialization::EXPR_CXX_PROPERTY_REF_EXPR;
2285 }
2286
VisitMSPropertySubscriptExpr(MSPropertySubscriptExpr * E)2287 void ASTStmtWriter::VisitMSPropertySubscriptExpr(MSPropertySubscriptExpr *E) {
2288 VisitExpr(E);
2289 Record.AddStmt(E->getBase());
2290 Record.AddStmt(E->getIdx());
2291 Record.AddSourceLocation(E->getRBracketLoc());
2292 Code = serialization::EXPR_CXX_PROPERTY_SUBSCRIPT_EXPR;
2293 }
2294
VisitCXXUuidofExpr(CXXUuidofExpr * E)2295 void ASTStmtWriter::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
2296 VisitExpr(E);
2297 Record.AddSourceRange(E->getSourceRange());
2298 Record.AddDeclRef(E->getGuidDecl());
2299 if (E->isTypeOperand()) {
2300 Record.AddTypeSourceInfo(E->getTypeOperandSourceInfo());
2301 Code = serialization::EXPR_CXX_UUIDOF_TYPE;
2302 } else {
2303 Record.AddStmt(E->getExprOperand());
2304 Code = serialization::EXPR_CXX_UUIDOF_EXPR;
2305 }
2306 }
2307
VisitSEHExceptStmt(SEHExceptStmt * S)2308 void ASTStmtWriter::VisitSEHExceptStmt(SEHExceptStmt *S) {
2309 VisitStmt(S);
2310 Record.AddSourceLocation(S->getExceptLoc());
2311 Record.AddStmt(S->getFilterExpr());
2312 Record.AddStmt(S->getBlock());
2313 Code = serialization::STMT_SEH_EXCEPT;
2314 }
2315
VisitSEHFinallyStmt(SEHFinallyStmt * S)2316 void ASTStmtWriter::VisitSEHFinallyStmt(SEHFinallyStmt *S) {
2317 VisitStmt(S);
2318 Record.AddSourceLocation(S->getFinallyLoc());
2319 Record.AddStmt(S->getBlock());
2320 Code = serialization::STMT_SEH_FINALLY;
2321 }
2322
VisitSEHTryStmt(SEHTryStmt * S)2323 void ASTStmtWriter::VisitSEHTryStmt(SEHTryStmt *S) {
2324 VisitStmt(S);
2325 Record.push_back(S->getIsCXXTry());
2326 Record.AddSourceLocation(S->getTryLoc());
2327 Record.AddStmt(S->getTryBlock());
2328 Record.AddStmt(S->getHandler());
2329 Code = serialization::STMT_SEH_TRY;
2330 }
2331
VisitSEHLeaveStmt(SEHLeaveStmt * S)2332 void ASTStmtWriter::VisitSEHLeaveStmt(SEHLeaveStmt *S) {
2333 VisitStmt(S);
2334 Record.AddSourceLocation(S->getLeaveLoc());
2335 Code = serialization::STMT_SEH_LEAVE;
2336 }
2337
2338 //===----------------------------------------------------------------------===//
2339 // OpenMP Directives.
2340 //===----------------------------------------------------------------------===//
2341
VisitOMPCanonicalLoop(OMPCanonicalLoop * S)2342 void ASTStmtWriter::VisitOMPCanonicalLoop(OMPCanonicalLoop *S) {
2343 VisitStmt(S);
2344 for (Stmt *SubStmt : S->SubStmts)
2345 Record.AddStmt(SubStmt);
2346 Code = serialization::STMT_OMP_CANONICAL_LOOP;
2347 }
2348
VisitOMPExecutableDirective(OMPExecutableDirective * E)2349 void ASTStmtWriter::VisitOMPExecutableDirective(OMPExecutableDirective *E) {
2350 Record.writeOMPChildren(E->Data);
2351 Record.AddSourceLocation(E->getBeginLoc());
2352 Record.AddSourceLocation(E->getEndLoc());
2353 Record.writeEnum(E->getMappedDirective());
2354 }
2355
VisitOMPLoopBasedDirective(OMPLoopBasedDirective * D)2356 void ASTStmtWriter::VisitOMPLoopBasedDirective(OMPLoopBasedDirective *D) {
2357 VisitStmt(D);
2358 Record.writeUInt32(D->getLoopsNumber());
2359 VisitOMPExecutableDirective(D);
2360 }
2361
VisitOMPLoopDirective(OMPLoopDirective * D)2362 void ASTStmtWriter::VisitOMPLoopDirective(OMPLoopDirective *D) {
2363 VisitOMPLoopBasedDirective(D);
2364 }
2365
VisitOMPMetaDirective(OMPMetaDirective * D)2366 void ASTStmtWriter::VisitOMPMetaDirective(OMPMetaDirective *D) {
2367 VisitStmt(D);
2368 Record.push_back(D->getNumClauses());
2369 VisitOMPExecutableDirective(D);
2370 Code = serialization::STMT_OMP_META_DIRECTIVE;
2371 }
2372
VisitOMPParallelDirective(OMPParallelDirective * D)2373 void ASTStmtWriter::VisitOMPParallelDirective(OMPParallelDirective *D) {
2374 VisitStmt(D);
2375 VisitOMPExecutableDirective(D);
2376 Record.writeBool(D->hasCancel());
2377 Code = serialization::STMT_OMP_PARALLEL_DIRECTIVE;
2378 }
2379
VisitOMPSimdDirective(OMPSimdDirective * D)2380 void ASTStmtWriter::VisitOMPSimdDirective(OMPSimdDirective *D) {
2381 VisitOMPLoopDirective(D);
2382 Code = serialization::STMT_OMP_SIMD_DIRECTIVE;
2383 }
2384
VisitOMPLoopTransformationDirective(OMPLoopTransformationDirective * D)2385 void ASTStmtWriter::VisitOMPLoopTransformationDirective(
2386 OMPLoopTransformationDirective *D) {
2387 VisitOMPLoopBasedDirective(D);
2388 Record.writeUInt32(D->getNumGeneratedLoops());
2389 }
2390
VisitOMPTileDirective(OMPTileDirective * D)2391 void ASTStmtWriter::VisitOMPTileDirective(OMPTileDirective *D) {
2392 VisitOMPLoopTransformationDirective(D);
2393 Code = serialization::STMT_OMP_TILE_DIRECTIVE;
2394 }
2395
VisitOMPUnrollDirective(OMPUnrollDirective * D)2396 void ASTStmtWriter::VisitOMPUnrollDirective(OMPUnrollDirective *D) {
2397 VisitOMPLoopTransformationDirective(D);
2398 Code = serialization::STMT_OMP_UNROLL_DIRECTIVE;
2399 }
2400
VisitOMPForDirective(OMPForDirective * D)2401 void ASTStmtWriter::VisitOMPForDirective(OMPForDirective *D) {
2402 VisitOMPLoopDirective(D);
2403 Record.writeBool(D->hasCancel());
2404 Code = serialization::STMT_OMP_FOR_DIRECTIVE;
2405 }
2406
VisitOMPForSimdDirective(OMPForSimdDirective * D)2407 void ASTStmtWriter::VisitOMPForSimdDirective(OMPForSimdDirective *D) {
2408 VisitOMPLoopDirective(D);
2409 Code = serialization::STMT_OMP_FOR_SIMD_DIRECTIVE;
2410 }
2411
VisitOMPSectionsDirective(OMPSectionsDirective * D)2412 void ASTStmtWriter::VisitOMPSectionsDirective(OMPSectionsDirective *D) {
2413 VisitStmt(D);
2414 VisitOMPExecutableDirective(D);
2415 Record.writeBool(D->hasCancel());
2416 Code = serialization::STMT_OMP_SECTIONS_DIRECTIVE;
2417 }
2418
VisitOMPSectionDirective(OMPSectionDirective * D)2419 void ASTStmtWriter::VisitOMPSectionDirective(OMPSectionDirective *D) {
2420 VisitStmt(D);
2421 VisitOMPExecutableDirective(D);
2422 Record.writeBool(D->hasCancel());
2423 Code = serialization::STMT_OMP_SECTION_DIRECTIVE;
2424 }
2425
VisitOMPScopeDirective(OMPScopeDirective * D)2426 void ASTStmtWriter::VisitOMPScopeDirective(OMPScopeDirective *D) {
2427 VisitStmt(D);
2428 VisitOMPExecutableDirective(D);
2429 Code = serialization::STMT_OMP_SCOPE_DIRECTIVE;
2430 }
2431
VisitOMPSingleDirective(OMPSingleDirective * D)2432 void ASTStmtWriter::VisitOMPSingleDirective(OMPSingleDirective *D) {
2433 VisitStmt(D);
2434 VisitOMPExecutableDirective(D);
2435 Code = serialization::STMT_OMP_SINGLE_DIRECTIVE;
2436 }
2437
VisitOMPMasterDirective(OMPMasterDirective * D)2438 void ASTStmtWriter::VisitOMPMasterDirective(OMPMasterDirective *D) {
2439 VisitStmt(D);
2440 VisitOMPExecutableDirective(D);
2441 Code = serialization::STMT_OMP_MASTER_DIRECTIVE;
2442 }
2443
VisitOMPCriticalDirective(OMPCriticalDirective * D)2444 void ASTStmtWriter::VisitOMPCriticalDirective(OMPCriticalDirective *D) {
2445 VisitStmt(D);
2446 VisitOMPExecutableDirective(D);
2447 Record.AddDeclarationNameInfo(D->getDirectiveName());
2448 Code = serialization::STMT_OMP_CRITICAL_DIRECTIVE;
2449 }
2450
VisitOMPParallelForDirective(OMPParallelForDirective * D)2451 void ASTStmtWriter::VisitOMPParallelForDirective(OMPParallelForDirective *D) {
2452 VisitOMPLoopDirective(D);
2453 Record.writeBool(D->hasCancel());
2454 Code = serialization::STMT_OMP_PARALLEL_FOR_DIRECTIVE;
2455 }
2456
VisitOMPParallelForSimdDirective(OMPParallelForSimdDirective * D)2457 void ASTStmtWriter::VisitOMPParallelForSimdDirective(
2458 OMPParallelForSimdDirective *D) {
2459 VisitOMPLoopDirective(D);
2460 Code = serialization::STMT_OMP_PARALLEL_FOR_SIMD_DIRECTIVE;
2461 }
2462
VisitOMPParallelMasterDirective(OMPParallelMasterDirective * D)2463 void ASTStmtWriter::VisitOMPParallelMasterDirective(
2464 OMPParallelMasterDirective *D) {
2465 VisitStmt(D);
2466 VisitOMPExecutableDirective(D);
2467 Code = serialization::STMT_OMP_PARALLEL_MASTER_DIRECTIVE;
2468 }
2469
VisitOMPParallelMaskedDirective(OMPParallelMaskedDirective * D)2470 void ASTStmtWriter::VisitOMPParallelMaskedDirective(
2471 OMPParallelMaskedDirective *D) {
2472 VisitStmt(D);
2473 VisitOMPExecutableDirective(D);
2474 Code = serialization::STMT_OMP_PARALLEL_MASKED_DIRECTIVE;
2475 }
2476
VisitOMPParallelSectionsDirective(OMPParallelSectionsDirective * D)2477 void ASTStmtWriter::VisitOMPParallelSectionsDirective(
2478 OMPParallelSectionsDirective *D) {
2479 VisitStmt(D);
2480 VisitOMPExecutableDirective(D);
2481 Record.writeBool(D->hasCancel());
2482 Code = serialization::STMT_OMP_PARALLEL_SECTIONS_DIRECTIVE;
2483 }
2484
VisitOMPTaskDirective(OMPTaskDirective * D)2485 void ASTStmtWriter::VisitOMPTaskDirective(OMPTaskDirective *D) {
2486 VisitStmt(D);
2487 VisitOMPExecutableDirective(D);
2488 Record.writeBool(D->hasCancel());
2489 Code = serialization::STMT_OMP_TASK_DIRECTIVE;
2490 }
2491
VisitOMPAtomicDirective(OMPAtomicDirective * D)2492 void ASTStmtWriter::VisitOMPAtomicDirective(OMPAtomicDirective *D) {
2493 VisitStmt(D);
2494 VisitOMPExecutableDirective(D);
2495 Record.writeBool(D->isXLHSInRHSPart());
2496 Record.writeBool(D->isPostfixUpdate());
2497 Record.writeBool(D->isFailOnly());
2498 Code = serialization::STMT_OMP_ATOMIC_DIRECTIVE;
2499 }
2500
VisitOMPTargetDirective(OMPTargetDirective * D)2501 void ASTStmtWriter::VisitOMPTargetDirective(OMPTargetDirective *D) {
2502 VisitStmt(D);
2503 VisitOMPExecutableDirective(D);
2504 Code = serialization::STMT_OMP_TARGET_DIRECTIVE;
2505 }
2506
VisitOMPTargetDataDirective(OMPTargetDataDirective * D)2507 void ASTStmtWriter::VisitOMPTargetDataDirective(OMPTargetDataDirective *D) {
2508 VisitStmt(D);
2509 VisitOMPExecutableDirective(D);
2510 Code = serialization::STMT_OMP_TARGET_DATA_DIRECTIVE;
2511 }
2512
VisitOMPTargetEnterDataDirective(OMPTargetEnterDataDirective * D)2513 void ASTStmtWriter::VisitOMPTargetEnterDataDirective(
2514 OMPTargetEnterDataDirective *D) {
2515 VisitStmt(D);
2516 VisitOMPExecutableDirective(D);
2517 Code = serialization::STMT_OMP_TARGET_ENTER_DATA_DIRECTIVE;
2518 }
2519
VisitOMPTargetExitDataDirective(OMPTargetExitDataDirective * D)2520 void ASTStmtWriter::VisitOMPTargetExitDataDirective(
2521 OMPTargetExitDataDirective *D) {
2522 VisitStmt(D);
2523 VisitOMPExecutableDirective(D);
2524 Code = serialization::STMT_OMP_TARGET_EXIT_DATA_DIRECTIVE;
2525 }
2526
VisitOMPTargetParallelDirective(OMPTargetParallelDirective * D)2527 void ASTStmtWriter::VisitOMPTargetParallelDirective(
2528 OMPTargetParallelDirective *D) {
2529 VisitStmt(D);
2530 VisitOMPExecutableDirective(D);
2531 Record.writeBool(D->hasCancel());
2532 Code = serialization::STMT_OMP_TARGET_PARALLEL_DIRECTIVE;
2533 }
2534
VisitOMPTargetParallelForDirective(OMPTargetParallelForDirective * D)2535 void ASTStmtWriter::VisitOMPTargetParallelForDirective(
2536 OMPTargetParallelForDirective *D) {
2537 VisitOMPLoopDirective(D);
2538 Record.writeBool(D->hasCancel());
2539 Code = serialization::STMT_OMP_TARGET_PARALLEL_FOR_DIRECTIVE;
2540 }
2541
VisitOMPTaskyieldDirective(OMPTaskyieldDirective * D)2542 void ASTStmtWriter::VisitOMPTaskyieldDirective(OMPTaskyieldDirective *D) {
2543 VisitStmt(D);
2544 VisitOMPExecutableDirective(D);
2545 Code = serialization::STMT_OMP_TASKYIELD_DIRECTIVE;
2546 }
2547
VisitOMPBarrierDirective(OMPBarrierDirective * D)2548 void ASTStmtWriter::VisitOMPBarrierDirective(OMPBarrierDirective *D) {
2549 VisitStmt(D);
2550 VisitOMPExecutableDirective(D);
2551 Code = serialization::STMT_OMP_BARRIER_DIRECTIVE;
2552 }
2553
VisitOMPTaskwaitDirective(OMPTaskwaitDirective * D)2554 void ASTStmtWriter::VisitOMPTaskwaitDirective(OMPTaskwaitDirective *D) {
2555 VisitStmt(D);
2556 Record.push_back(D->getNumClauses());
2557 VisitOMPExecutableDirective(D);
2558 Code = serialization::STMT_OMP_TASKWAIT_DIRECTIVE;
2559 }
2560
VisitOMPErrorDirective(OMPErrorDirective * D)2561 void ASTStmtWriter::VisitOMPErrorDirective(OMPErrorDirective *D) {
2562 VisitStmt(D);
2563 Record.push_back(D->getNumClauses());
2564 VisitOMPExecutableDirective(D);
2565 Code = serialization::STMT_OMP_ERROR_DIRECTIVE;
2566 }
2567
VisitOMPTaskgroupDirective(OMPTaskgroupDirective * D)2568 void ASTStmtWriter::VisitOMPTaskgroupDirective(OMPTaskgroupDirective *D) {
2569 VisitStmt(D);
2570 VisitOMPExecutableDirective(D);
2571 Code = serialization::STMT_OMP_TASKGROUP_DIRECTIVE;
2572 }
2573
VisitOMPFlushDirective(OMPFlushDirective * D)2574 void ASTStmtWriter::VisitOMPFlushDirective(OMPFlushDirective *D) {
2575 VisitStmt(D);
2576 VisitOMPExecutableDirective(D);
2577 Code = serialization::STMT_OMP_FLUSH_DIRECTIVE;
2578 }
2579
VisitOMPDepobjDirective(OMPDepobjDirective * D)2580 void ASTStmtWriter::VisitOMPDepobjDirective(OMPDepobjDirective *D) {
2581 VisitStmt(D);
2582 VisitOMPExecutableDirective(D);
2583 Code = serialization::STMT_OMP_DEPOBJ_DIRECTIVE;
2584 }
2585
VisitOMPScanDirective(OMPScanDirective * D)2586 void ASTStmtWriter::VisitOMPScanDirective(OMPScanDirective *D) {
2587 VisitStmt(D);
2588 VisitOMPExecutableDirective(D);
2589 Code = serialization::STMT_OMP_SCAN_DIRECTIVE;
2590 }
2591
VisitOMPOrderedDirective(OMPOrderedDirective * D)2592 void ASTStmtWriter::VisitOMPOrderedDirective(OMPOrderedDirective *D) {
2593 VisitStmt(D);
2594 VisitOMPExecutableDirective(D);
2595 Code = serialization::STMT_OMP_ORDERED_DIRECTIVE;
2596 }
2597
VisitOMPTeamsDirective(OMPTeamsDirective * D)2598 void ASTStmtWriter::VisitOMPTeamsDirective(OMPTeamsDirective *D) {
2599 VisitStmt(D);
2600 VisitOMPExecutableDirective(D);
2601 Code = serialization::STMT_OMP_TEAMS_DIRECTIVE;
2602 }
2603
VisitOMPCancellationPointDirective(OMPCancellationPointDirective * D)2604 void ASTStmtWriter::VisitOMPCancellationPointDirective(
2605 OMPCancellationPointDirective *D) {
2606 VisitStmt(D);
2607 VisitOMPExecutableDirective(D);
2608 Record.writeEnum(D->getCancelRegion());
2609 Code = serialization::STMT_OMP_CANCELLATION_POINT_DIRECTIVE;
2610 }
2611
VisitOMPCancelDirective(OMPCancelDirective * D)2612 void ASTStmtWriter::VisitOMPCancelDirective(OMPCancelDirective *D) {
2613 VisitStmt(D);
2614 VisitOMPExecutableDirective(D);
2615 Record.writeEnum(D->getCancelRegion());
2616 Code = serialization::STMT_OMP_CANCEL_DIRECTIVE;
2617 }
2618
VisitOMPTaskLoopDirective(OMPTaskLoopDirective * D)2619 void ASTStmtWriter::VisitOMPTaskLoopDirective(OMPTaskLoopDirective *D) {
2620 VisitOMPLoopDirective(D);
2621 Record.writeBool(D->hasCancel());
2622 Code = serialization::STMT_OMP_TASKLOOP_DIRECTIVE;
2623 }
2624
VisitOMPTaskLoopSimdDirective(OMPTaskLoopSimdDirective * D)2625 void ASTStmtWriter::VisitOMPTaskLoopSimdDirective(OMPTaskLoopSimdDirective *D) {
2626 VisitOMPLoopDirective(D);
2627 Code = serialization::STMT_OMP_TASKLOOP_SIMD_DIRECTIVE;
2628 }
2629
VisitOMPMasterTaskLoopDirective(OMPMasterTaskLoopDirective * D)2630 void ASTStmtWriter::VisitOMPMasterTaskLoopDirective(
2631 OMPMasterTaskLoopDirective *D) {
2632 VisitOMPLoopDirective(D);
2633 Record.writeBool(D->hasCancel());
2634 Code = serialization::STMT_OMP_MASTER_TASKLOOP_DIRECTIVE;
2635 }
2636
VisitOMPMaskedTaskLoopDirective(OMPMaskedTaskLoopDirective * D)2637 void ASTStmtWriter::VisitOMPMaskedTaskLoopDirective(
2638 OMPMaskedTaskLoopDirective *D) {
2639 VisitOMPLoopDirective(D);
2640 Record.writeBool(D->hasCancel());
2641 Code = serialization::STMT_OMP_MASKED_TASKLOOP_DIRECTIVE;
2642 }
2643
VisitOMPMasterTaskLoopSimdDirective(OMPMasterTaskLoopSimdDirective * D)2644 void ASTStmtWriter::VisitOMPMasterTaskLoopSimdDirective(
2645 OMPMasterTaskLoopSimdDirective *D) {
2646 VisitOMPLoopDirective(D);
2647 Code = serialization::STMT_OMP_MASTER_TASKLOOP_SIMD_DIRECTIVE;
2648 }
2649
VisitOMPMaskedTaskLoopSimdDirective(OMPMaskedTaskLoopSimdDirective * D)2650 void ASTStmtWriter::VisitOMPMaskedTaskLoopSimdDirective(
2651 OMPMaskedTaskLoopSimdDirective *D) {
2652 VisitOMPLoopDirective(D);
2653 Code = serialization::STMT_OMP_MASKED_TASKLOOP_SIMD_DIRECTIVE;
2654 }
2655
VisitOMPParallelMasterTaskLoopDirective(OMPParallelMasterTaskLoopDirective * D)2656 void ASTStmtWriter::VisitOMPParallelMasterTaskLoopDirective(
2657 OMPParallelMasterTaskLoopDirective *D) {
2658 VisitOMPLoopDirective(D);
2659 Record.writeBool(D->hasCancel());
2660 Code = serialization::STMT_OMP_PARALLEL_MASTER_TASKLOOP_DIRECTIVE;
2661 }
2662
VisitOMPParallelMaskedTaskLoopDirective(OMPParallelMaskedTaskLoopDirective * D)2663 void ASTStmtWriter::VisitOMPParallelMaskedTaskLoopDirective(
2664 OMPParallelMaskedTaskLoopDirective *D) {
2665 VisitOMPLoopDirective(D);
2666 Record.writeBool(D->hasCancel());
2667 Code = serialization::STMT_OMP_PARALLEL_MASKED_TASKLOOP_DIRECTIVE;
2668 }
2669
VisitOMPParallelMasterTaskLoopSimdDirective(OMPParallelMasterTaskLoopSimdDirective * D)2670 void ASTStmtWriter::VisitOMPParallelMasterTaskLoopSimdDirective(
2671 OMPParallelMasterTaskLoopSimdDirective *D) {
2672 VisitOMPLoopDirective(D);
2673 Code = serialization::STMT_OMP_PARALLEL_MASTER_TASKLOOP_SIMD_DIRECTIVE;
2674 }
2675
VisitOMPParallelMaskedTaskLoopSimdDirective(OMPParallelMaskedTaskLoopSimdDirective * D)2676 void ASTStmtWriter::VisitOMPParallelMaskedTaskLoopSimdDirective(
2677 OMPParallelMaskedTaskLoopSimdDirective *D) {
2678 VisitOMPLoopDirective(D);
2679 Code = serialization::STMT_OMP_PARALLEL_MASKED_TASKLOOP_SIMD_DIRECTIVE;
2680 }
2681
VisitOMPDistributeDirective(OMPDistributeDirective * D)2682 void ASTStmtWriter::VisitOMPDistributeDirective(OMPDistributeDirective *D) {
2683 VisitOMPLoopDirective(D);
2684 Code = serialization::STMT_OMP_DISTRIBUTE_DIRECTIVE;
2685 }
2686
VisitOMPTargetUpdateDirective(OMPTargetUpdateDirective * D)2687 void ASTStmtWriter::VisitOMPTargetUpdateDirective(OMPTargetUpdateDirective *D) {
2688 VisitStmt(D);
2689 VisitOMPExecutableDirective(D);
2690 Code = serialization::STMT_OMP_TARGET_UPDATE_DIRECTIVE;
2691 }
2692
VisitOMPDistributeParallelForDirective(OMPDistributeParallelForDirective * D)2693 void ASTStmtWriter::VisitOMPDistributeParallelForDirective(
2694 OMPDistributeParallelForDirective *D) {
2695 VisitOMPLoopDirective(D);
2696 Record.writeBool(D->hasCancel());
2697 Code = serialization::STMT_OMP_DISTRIBUTE_PARALLEL_FOR_DIRECTIVE;
2698 }
2699
VisitOMPDistributeParallelForSimdDirective(OMPDistributeParallelForSimdDirective * D)2700 void ASTStmtWriter::VisitOMPDistributeParallelForSimdDirective(
2701 OMPDistributeParallelForSimdDirective *D) {
2702 VisitOMPLoopDirective(D);
2703 Code = serialization::STMT_OMP_DISTRIBUTE_PARALLEL_FOR_SIMD_DIRECTIVE;
2704 }
2705
VisitOMPDistributeSimdDirective(OMPDistributeSimdDirective * D)2706 void ASTStmtWriter::VisitOMPDistributeSimdDirective(
2707 OMPDistributeSimdDirective *D) {
2708 VisitOMPLoopDirective(D);
2709 Code = serialization::STMT_OMP_DISTRIBUTE_SIMD_DIRECTIVE;
2710 }
2711
VisitOMPTargetParallelForSimdDirective(OMPTargetParallelForSimdDirective * D)2712 void ASTStmtWriter::VisitOMPTargetParallelForSimdDirective(
2713 OMPTargetParallelForSimdDirective *D) {
2714 VisitOMPLoopDirective(D);
2715 Code = serialization::STMT_OMP_TARGET_PARALLEL_FOR_SIMD_DIRECTIVE;
2716 }
2717
VisitOMPTargetSimdDirective(OMPTargetSimdDirective * D)2718 void ASTStmtWriter::VisitOMPTargetSimdDirective(OMPTargetSimdDirective *D) {
2719 VisitOMPLoopDirective(D);
2720 Code = serialization::STMT_OMP_TARGET_SIMD_DIRECTIVE;
2721 }
2722
VisitOMPTeamsDistributeDirective(OMPTeamsDistributeDirective * D)2723 void ASTStmtWriter::VisitOMPTeamsDistributeDirective(
2724 OMPTeamsDistributeDirective *D) {
2725 VisitOMPLoopDirective(D);
2726 Code = serialization::STMT_OMP_TEAMS_DISTRIBUTE_DIRECTIVE;
2727 }
2728
VisitOMPTeamsDistributeSimdDirective(OMPTeamsDistributeSimdDirective * D)2729 void ASTStmtWriter::VisitOMPTeamsDistributeSimdDirective(
2730 OMPTeamsDistributeSimdDirective *D) {
2731 VisitOMPLoopDirective(D);
2732 Code = serialization::STMT_OMP_TEAMS_DISTRIBUTE_SIMD_DIRECTIVE;
2733 }
2734
VisitOMPTeamsDistributeParallelForSimdDirective(OMPTeamsDistributeParallelForSimdDirective * D)2735 void ASTStmtWriter::VisitOMPTeamsDistributeParallelForSimdDirective(
2736 OMPTeamsDistributeParallelForSimdDirective *D) {
2737 VisitOMPLoopDirective(D);
2738 Code = serialization::STMT_OMP_TEAMS_DISTRIBUTE_PARALLEL_FOR_SIMD_DIRECTIVE;
2739 }
2740
VisitOMPTeamsDistributeParallelForDirective(OMPTeamsDistributeParallelForDirective * D)2741 void ASTStmtWriter::VisitOMPTeamsDistributeParallelForDirective(
2742 OMPTeamsDistributeParallelForDirective *D) {
2743 VisitOMPLoopDirective(D);
2744 Record.writeBool(D->hasCancel());
2745 Code = serialization::STMT_OMP_TEAMS_DISTRIBUTE_PARALLEL_FOR_DIRECTIVE;
2746 }
2747
VisitOMPTargetTeamsDirective(OMPTargetTeamsDirective * D)2748 void ASTStmtWriter::VisitOMPTargetTeamsDirective(OMPTargetTeamsDirective *D) {
2749 VisitStmt(D);
2750 VisitOMPExecutableDirective(D);
2751 Code = serialization::STMT_OMP_TARGET_TEAMS_DIRECTIVE;
2752 }
2753
VisitOMPTargetTeamsDistributeDirective(OMPTargetTeamsDistributeDirective * D)2754 void ASTStmtWriter::VisitOMPTargetTeamsDistributeDirective(
2755 OMPTargetTeamsDistributeDirective *D) {
2756 VisitOMPLoopDirective(D);
2757 Code = serialization::STMT_OMP_TARGET_TEAMS_DISTRIBUTE_DIRECTIVE;
2758 }
2759
VisitOMPTargetTeamsDistributeParallelForDirective(OMPTargetTeamsDistributeParallelForDirective * D)2760 void ASTStmtWriter::VisitOMPTargetTeamsDistributeParallelForDirective(
2761 OMPTargetTeamsDistributeParallelForDirective *D) {
2762 VisitOMPLoopDirective(D);
2763 Record.writeBool(D->hasCancel());
2764 Code = serialization::STMT_OMP_TARGET_TEAMS_DISTRIBUTE_PARALLEL_FOR_DIRECTIVE;
2765 }
2766
VisitOMPTargetTeamsDistributeParallelForSimdDirective(OMPTargetTeamsDistributeParallelForSimdDirective * D)2767 void ASTStmtWriter::VisitOMPTargetTeamsDistributeParallelForSimdDirective(
2768 OMPTargetTeamsDistributeParallelForSimdDirective *D) {
2769 VisitOMPLoopDirective(D);
2770 Code = serialization::
2771 STMT_OMP_TARGET_TEAMS_DISTRIBUTE_PARALLEL_FOR_SIMD_DIRECTIVE;
2772 }
2773
VisitOMPTargetTeamsDistributeSimdDirective(OMPTargetTeamsDistributeSimdDirective * D)2774 void ASTStmtWriter::VisitOMPTargetTeamsDistributeSimdDirective(
2775 OMPTargetTeamsDistributeSimdDirective *D) {
2776 VisitOMPLoopDirective(D);
2777 Code = serialization::STMT_OMP_TARGET_TEAMS_DISTRIBUTE_SIMD_DIRECTIVE;
2778 }
2779
VisitOMPInteropDirective(OMPInteropDirective * D)2780 void ASTStmtWriter::VisitOMPInteropDirective(OMPInteropDirective *D) {
2781 VisitStmt(D);
2782 VisitOMPExecutableDirective(D);
2783 Code = serialization::STMT_OMP_INTEROP_DIRECTIVE;
2784 }
2785
VisitOMPDispatchDirective(OMPDispatchDirective * D)2786 void ASTStmtWriter::VisitOMPDispatchDirective(OMPDispatchDirective *D) {
2787 VisitStmt(D);
2788 VisitOMPExecutableDirective(D);
2789 Record.AddSourceLocation(D->getTargetCallLoc());
2790 Code = serialization::STMT_OMP_DISPATCH_DIRECTIVE;
2791 }
2792
VisitOMPMaskedDirective(OMPMaskedDirective * D)2793 void ASTStmtWriter::VisitOMPMaskedDirective(OMPMaskedDirective *D) {
2794 VisitStmt(D);
2795 VisitOMPExecutableDirective(D);
2796 Code = serialization::STMT_OMP_MASKED_DIRECTIVE;
2797 }
2798
VisitOMPGenericLoopDirective(OMPGenericLoopDirective * D)2799 void ASTStmtWriter::VisitOMPGenericLoopDirective(OMPGenericLoopDirective *D) {
2800 VisitOMPLoopDirective(D);
2801 Code = serialization::STMT_OMP_GENERIC_LOOP_DIRECTIVE;
2802 }
2803
VisitOMPTeamsGenericLoopDirective(OMPTeamsGenericLoopDirective * D)2804 void ASTStmtWriter::VisitOMPTeamsGenericLoopDirective(
2805 OMPTeamsGenericLoopDirective *D) {
2806 VisitOMPLoopDirective(D);
2807 Code = serialization::STMT_OMP_TEAMS_GENERIC_LOOP_DIRECTIVE;
2808 }
2809
VisitOMPTargetTeamsGenericLoopDirective(OMPTargetTeamsGenericLoopDirective * D)2810 void ASTStmtWriter::VisitOMPTargetTeamsGenericLoopDirective(
2811 OMPTargetTeamsGenericLoopDirective *D) {
2812 VisitOMPLoopDirective(D);
2813 Code = serialization::STMT_OMP_TARGET_TEAMS_GENERIC_LOOP_DIRECTIVE;
2814 }
2815
VisitOMPParallelGenericLoopDirective(OMPParallelGenericLoopDirective * D)2816 void ASTStmtWriter::VisitOMPParallelGenericLoopDirective(
2817 OMPParallelGenericLoopDirective *D) {
2818 VisitOMPLoopDirective(D);
2819 Code = serialization::STMT_OMP_PARALLEL_GENERIC_LOOP_DIRECTIVE;
2820 }
2821
VisitOMPTargetParallelGenericLoopDirective(OMPTargetParallelGenericLoopDirective * D)2822 void ASTStmtWriter::VisitOMPTargetParallelGenericLoopDirective(
2823 OMPTargetParallelGenericLoopDirective *D) {
2824 VisitOMPLoopDirective(D);
2825 Code = serialization::STMT_OMP_TARGET_PARALLEL_GENERIC_LOOP_DIRECTIVE;
2826 }
2827
2828 //===----------------------------------------------------------------------===//
2829 // ASTWriter Implementation
2830 //===----------------------------------------------------------------------===//
2831
RecordSwitchCaseID(SwitchCase * S)2832 unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
2833 assert(!SwitchCaseIDs.contains(S) && "SwitchCase recorded twice");
2834 unsigned NextID = SwitchCaseIDs.size();
2835 SwitchCaseIDs[S] = NextID;
2836 return NextID;
2837 }
2838
getSwitchCaseID(SwitchCase * S)2839 unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
2840 assert(SwitchCaseIDs.contains(S) && "SwitchCase hasn't been seen yet");
2841 return SwitchCaseIDs[S];
2842 }
2843
ClearSwitchCaseIDs()2844 void ASTWriter::ClearSwitchCaseIDs() {
2845 SwitchCaseIDs.clear();
2846 }
2847
2848 /// Write the given substatement or subexpression to the
2849 /// bitstream.
WriteSubStmt(Stmt * S)2850 void ASTWriter::WriteSubStmt(Stmt *S) {
2851 RecordData Record;
2852 ASTStmtWriter Writer(*this, Record);
2853 ++NumStatements;
2854
2855 if (!S) {
2856 Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
2857 return;
2858 }
2859
2860 llvm::DenseMap<Stmt *, uint64_t>::iterator I = SubStmtEntries.find(S);
2861 if (I != SubStmtEntries.end()) {
2862 Record.push_back(I->second);
2863 Stream.EmitRecord(serialization::STMT_REF_PTR, Record);
2864 return;
2865 }
2866
2867 #ifndef NDEBUG
2868 assert(!ParentStmts.count(S) && "There is a Stmt cycle!");
2869
2870 struct ParentStmtInserterRAII {
2871 Stmt *S;
2872 llvm::DenseSet<Stmt *> &ParentStmts;
2873
2874 ParentStmtInserterRAII(Stmt *S, llvm::DenseSet<Stmt *> &ParentStmts)
2875 : S(S), ParentStmts(ParentStmts) {
2876 ParentStmts.insert(S);
2877 }
2878 ~ParentStmtInserterRAII() {
2879 ParentStmts.erase(S);
2880 }
2881 };
2882
2883 ParentStmtInserterRAII ParentStmtInserter(S, ParentStmts);
2884 #endif
2885
2886 Writer.Visit(S);
2887
2888 uint64_t Offset = Writer.Emit();
2889 SubStmtEntries[S] = Offset;
2890 }
2891
2892 /// Flush all of the statements that have been added to the
2893 /// queue via AddStmt().
FlushStmts()2894 void ASTRecordWriter::FlushStmts() {
2895 // We expect to be the only consumer of the two temporary statement maps,
2896 // assert that they are empty.
2897 assert(Writer->SubStmtEntries.empty() && "unexpected entries in sub-stmt map");
2898 assert(Writer->ParentStmts.empty() && "unexpected entries in parent stmt map");
2899
2900 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
2901 Writer->WriteSubStmt(StmtsToEmit[I]);
2902
2903 assert(N == StmtsToEmit.size() && "record modified while being written!");
2904
2905 // Note that we are at the end of a full expression. Any
2906 // expression records that follow this one are part of a different
2907 // expression.
2908 Writer->Stream.EmitRecord(serialization::STMT_STOP, ArrayRef<uint32_t>());
2909
2910 Writer->SubStmtEntries.clear();
2911 Writer->ParentStmts.clear();
2912 }
2913
2914 StmtsToEmit.clear();
2915 }
2916
FlushSubStmts()2917 void ASTRecordWriter::FlushSubStmts() {
2918 // For a nested statement, write out the substatements in reverse order (so
2919 // that a simple stack machine can be used when loading), and don't emit a
2920 // STMT_STOP after each one.
2921 for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
2922 Writer->WriteSubStmt(StmtsToEmit[N - I - 1]);
2923 assert(N == StmtsToEmit.size() && "record modified while being written!");
2924 }
2925
2926 StmtsToEmit.clear();
2927 }
2928