contrib/llvm/tools/clang/lib/AST/StmtProfile.cpp

   1 //===---- StmtProfile.cpp - Profile implementation for Stmt ASTs ----------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements the Stmt::Profile method, which builds a unique bit
  11 // representation that identifies a statement/expression.
  12 //
  13 //===----------------------------------------------------------------------===//
  14 #include "clang/AST/ASTContext.h"
  15 #include "clang/AST/DeclCXX.h"
  16 #include "clang/AST/DeclObjC.h"
  17 #include "clang/AST/DeclTemplate.h"
  18 #include "clang/AST/Expr.h"
  19 #include "clang/AST/ExprCXX.h"
  20 #include "clang/AST/ExprObjC.h"
  21 #include "clang/AST/ExprOpenMP.h"
  22 #include "clang/AST/StmtVisitor.h"
  23 #include "llvm/ADT/FoldingSet.h"
  24 using namespace clang;
  25
  26 namespace {
  27   class StmtProfiler : public ConstStmtVisitor<StmtProfiler> {
  28     llvm::FoldingSetNodeID &ID;
  29     const ASTContext &Context;
  30     bool Canonical;
  31
  32   public:
  33     StmtProfiler(llvm::FoldingSetNodeID &ID, const ASTContext &Context,
  34                  bool Canonical)
  35       : ID(ID), Context(Context), Canonical(Canonical) { }
  36
  37     void VisitStmt(const Stmt *S);
  38
  39 #define STMT(Node, Base) void Visit##Node(const Node *S);
  40 #include "clang/AST/StmtNodes.inc"
  41
  42     /// \brief Visit a declaration that is referenced within an expression
  43     /// or statement.
  44     void VisitDecl(const Decl *D);
  45
  46     /// \brief Visit a type that is referenced within an expression or
  47     /// statement.
  48     void VisitType(QualType T);
  49
  50     /// \brief Visit a name that occurs within an expression or statement.
  51     void VisitName(DeclarationName Name);
  52
  53     /// \brief Visit a nested-name-specifier that occurs within an expression
  54     /// or statement.
  55     void VisitNestedNameSpecifier(NestedNameSpecifier *NNS);
  56
  57     /// \brief Visit a template name that occurs within an expression or
  58     /// statement.
  59     void VisitTemplateName(TemplateName Name);
  60
  61     /// \brief Visit template arguments that occur within an expression or
  62     /// statement.
  63     void VisitTemplateArguments(const TemplateArgumentLoc *Args,
  64                                 unsigned NumArgs);
  65
  66     /// \brief Visit a single template argument.
  67     void VisitTemplateArgument(const TemplateArgument &Arg);
  68   };
  69 }
  70
  71 void StmtProfiler::VisitStmt(const Stmt *S) {
  72   assert(S && "Requires non-null Stmt pointer");
  73   ID.AddInteger(S->getStmtClass());
  74   for (const Stmt *SubStmt : S->children()) {
  75     if (SubStmt)
  76       Visit(SubStmt);
  77     else
  78       ID.AddInteger(0);
  79   }
  80 }
  81
  82 void StmtProfiler::VisitDeclStmt(const DeclStmt *S) {
  83   VisitStmt(S);
  84   for (const auto *D : S->decls())
  85     VisitDecl(D);
  86 }
  87
  88 void StmtProfiler::VisitNullStmt(const NullStmt *S) {
  89   VisitStmt(S);
  90 }
  91
  92 void StmtProfiler::VisitCompoundStmt(const CompoundStmt *S) {
  93   VisitStmt(S);
  94 }
  95
  96 void StmtProfiler::VisitCaseStmt(const CaseStmt *S) {
  97   VisitStmt(S);
  98 }
  99
 100 void StmtProfiler::VisitDefaultStmt(const DefaultStmt *S) {
 101   VisitStmt(S);
 102 }
 103
 104 void StmtProfiler::VisitLabelStmt(const LabelStmt *S) {
 105   VisitStmt(S);
 106   VisitDecl(S->getDecl());
 107 }
 108
 109 void StmtProfiler::VisitAttributedStmt(const AttributedStmt *S) {
 110   VisitStmt(S);
 111   // TODO: maybe visit attributes?
 112 }
 113
 114 void StmtProfiler::VisitIfStmt(const IfStmt *S) {
 115   VisitStmt(S);
 116   VisitDecl(S->getConditionVariable());
 117 }
 118
 119 void StmtProfiler::VisitSwitchStmt(const SwitchStmt *S) {
 120   VisitStmt(S);
 121   VisitDecl(S->getConditionVariable());
 122 }
 123
 124 void StmtProfiler::VisitWhileStmt(const WhileStmt *S) {
 125   VisitStmt(S);
 126   VisitDecl(S->getConditionVariable());
 127 }
 128
 129 void StmtProfiler::VisitDoStmt(const DoStmt *S) {
 130   VisitStmt(S);
 131 }
 132
 133 void StmtProfiler::VisitForStmt(const ForStmt *S) {
 134   VisitStmt(S);
 135 }
 136
 137 void StmtProfiler::VisitGotoStmt(const GotoStmt *S) {
 138   VisitStmt(S);
 139   VisitDecl(S->getLabel());
 140 }
 141
 142 void StmtProfiler::VisitIndirectGotoStmt(const IndirectGotoStmt *S) {
 143   VisitStmt(S);
 144 }
 145
 146 void StmtProfiler::VisitContinueStmt(const ContinueStmt *S) {
 147   VisitStmt(S);
 148 }
 149
 150 void StmtProfiler::VisitBreakStmt(const BreakStmt *S) {
 151   VisitStmt(S);
 152 }
 153
 154 void StmtProfiler::VisitReturnStmt(const ReturnStmt *S) {
 155   VisitStmt(S);
 156 }
 157
 158 void StmtProfiler::VisitGCCAsmStmt(const GCCAsmStmt *S) {
 159   VisitStmt(S);
 160   ID.AddBoolean(S->isVolatile());
 161   ID.AddBoolean(S->isSimple());
 162   VisitStringLiteral(S->getAsmString());
 163   ID.AddInteger(S->getNumOutputs());
 164   for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
 165     ID.AddString(S->getOutputName(I));
 166     VisitStringLiteral(S->getOutputConstraintLiteral(I));
 167   }
 168   ID.AddInteger(S->getNumInputs());
 169   for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
 170     ID.AddString(S->getInputName(I));
 171     VisitStringLiteral(S->getInputConstraintLiteral(I));
 172   }
 173   ID.AddInteger(S->getNumClobbers());
 174   for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
 175     VisitStringLiteral(S->getClobberStringLiteral(I));
 176 }
 177
 178 void StmtProfiler::VisitMSAsmStmt(const MSAsmStmt *S) {
 179   // FIXME: Implement MS style inline asm statement profiler.
 180   VisitStmt(S);
 181 }
 182
 183 void StmtProfiler::VisitCXXCatchStmt(const CXXCatchStmt *S) {
 184   VisitStmt(S);
 185   VisitType(S->getCaughtType());
 186 }
 187
 188 void StmtProfiler::VisitCXXTryStmt(const CXXTryStmt *S) {
 189   VisitStmt(S);
 190 }
 191
 192 void StmtProfiler::VisitCXXForRangeStmt(const CXXForRangeStmt *S) {
 193   VisitStmt(S);
 194 }
 195
 196 void StmtProfiler::VisitMSDependentExistsStmt(const MSDependentExistsStmt *S) {
 197   VisitStmt(S);
 198   ID.AddBoolean(S->isIfExists());
 199   VisitNestedNameSpecifier(S->getQualifierLoc().getNestedNameSpecifier());
 200   VisitName(S->getNameInfo().getName());
 201 }
 202
 203 void StmtProfiler::VisitSEHTryStmt(const SEHTryStmt *S) {
 204   VisitStmt(S);
 205 }
 206
 207 void StmtProfiler::VisitSEHFinallyStmt(const SEHFinallyStmt *S) {
 208   VisitStmt(S);
 209 }
 210
 211 void StmtProfiler::VisitSEHExceptStmt(const SEHExceptStmt *S) {
 212   VisitStmt(S);
 213 }
 214
 215 void StmtProfiler::VisitSEHLeaveStmt(const SEHLeaveStmt *S) {
 216   VisitStmt(S);
 217 }
 218
 219 void StmtProfiler::VisitCapturedStmt(const CapturedStmt *S) {
 220   VisitStmt(S);
 221 }
 222
 223 void StmtProfiler::VisitObjCForCollectionStmt(const ObjCForCollectionStmt *S) {
 224   VisitStmt(S);
 225 }
 226
 227 void StmtProfiler::VisitObjCAtCatchStmt(const ObjCAtCatchStmt *S) {
 228   VisitStmt(S);
 229   ID.AddBoolean(S->hasEllipsis());
 230   if (S->getCatchParamDecl())
 231     VisitType(S->getCatchParamDecl()->getType());
 232 }
 233
 234 void StmtProfiler::VisitObjCAtFinallyStmt(const ObjCAtFinallyStmt *S) {
 235   VisitStmt(S);
 236 }
 237
 238 void StmtProfiler::VisitObjCAtTryStmt(const ObjCAtTryStmt *S) {
 239   VisitStmt(S);
 240 }
 241
 242 void
 243 StmtProfiler::VisitObjCAtSynchronizedStmt(const ObjCAtSynchronizedStmt *S) {
 244   VisitStmt(S);
 245 }
 246
 247 void StmtProfiler::VisitObjCAtThrowStmt(const ObjCAtThrowStmt *S) {
 248   VisitStmt(S);
 249 }
 250
 251 void
 252 StmtProfiler::VisitObjCAutoreleasePoolStmt(const ObjCAutoreleasePoolStmt *S) {
 253   VisitStmt(S);
 254 }
 255
 256 namespace {
 257 class OMPClauseProfiler : public ConstOMPClauseVisitor<OMPClauseProfiler> {
 258   StmtProfiler *Profiler;
 259   /// \brief Process clauses with list of variables.
 260   template <typename T>
 261   void VisitOMPClauseList(T *Node);
 262
 263 public:
 264   OMPClauseProfiler(StmtProfiler *P) : Profiler(P) { }
 265 #define OPENMP_CLAUSE(Name, Class)                                             \
 266   void Visit##Class(const Class *C);
 267 #include "clang/Basic/OpenMPKinds.def"
 268   void VistOMPClauseWithPreInit(const OMPClauseWithPreInit *C);
 269   void VistOMPClauseWithPostUpdate(const OMPClauseWithPostUpdate *C);
 270 };
 271
 272 void OMPClauseProfiler::VistOMPClauseWithPreInit(
 273     const OMPClauseWithPreInit *C) {
 274   if (auto *S = C->getPreInitStmt())
 275     Profiler->VisitStmt(S);
 276 }
 277
 278 void OMPClauseProfiler::VistOMPClauseWithPostUpdate(
 279     const OMPClauseWithPostUpdate *C) {
 280   VistOMPClauseWithPreInit(C);
 281   if (auto *E = C->getPostUpdateExpr())
 282     Profiler->VisitStmt(E);
 283 }
 284
 285 void OMPClauseProfiler::VisitOMPIfClause(const OMPIfClause *C) {
 286   if (C->getCondition())
 287     Profiler->VisitStmt(C->getCondition());
 288 }
 289
 290 void OMPClauseProfiler::VisitOMPFinalClause(const OMPFinalClause *C) {
 291   if (C->getCondition())
 292     Profiler->VisitStmt(C->getCondition());
 293 }
 294
 295 void OMPClauseProfiler::VisitOMPNumThreadsClause(const OMPNumThreadsClause *C) {
 296   if (C->getNumThreads())
 297     Profiler->VisitStmt(C->getNumThreads());
 298 }
 299
 300 void OMPClauseProfiler::VisitOMPSafelenClause(const OMPSafelenClause *C) {
 301   if (C->getSafelen())
 302     Profiler->VisitStmt(C->getSafelen());
 303 }
 304
 305 void OMPClauseProfiler::VisitOMPSimdlenClause(const OMPSimdlenClause *C) {
 306   if (C->getSimdlen())
 307     Profiler->VisitStmt(C->getSimdlen());
 308 }
 309
 310 void OMPClauseProfiler::VisitOMPCollapseClause(const OMPCollapseClause *C) {
 311   if (C->getNumForLoops())
 312     Profiler->VisitStmt(C->getNumForLoops());
 313 }
 314
 315 void OMPClauseProfiler::VisitOMPDefaultClause(const OMPDefaultClause *C) { }
 316
 317 void OMPClauseProfiler::VisitOMPProcBindClause(const OMPProcBindClause *C) { }
 318
 319 void OMPClauseProfiler::VisitOMPScheduleClause(const OMPScheduleClause *C) {
 320   VistOMPClauseWithPreInit(C);
 321   if (auto *S = C->getChunkSize())
 322     Profiler->VisitStmt(S);
 323 }
 324
 325 void OMPClauseProfiler::VisitOMPOrderedClause(const OMPOrderedClause *C) {
 326   if (auto *Num = C->getNumForLoops())
 327     Profiler->VisitStmt(Num);
 328 }
 329
 330 void OMPClauseProfiler::VisitOMPNowaitClause(const OMPNowaitClause *) {}
 331
 332 void OMPClauseProfiler::VisitOMPUntiedClause(const OMPUntiedClause *) {}
 333
 334 void OMPClauseProfiler::VisitOMPMergeableClause(const OMPMergeableClause *) {}
 335
 336 void OMPClauseProfiler::VisitOMPReadClause(const OMPReadClause *) {}
 337
 338 void OMPClauseProfiler::VisitOMPWriteClause(const OMPWriteClause *) {}
 339
 340 void OMPClauseProfiler::VisitOMPUpdateClause(const OMPUpdateClause *) {}
 341
 342 void OMPClauseProfiler::VisitOMPCaptureClause(const OMPCaptureClause *) {}
 343
 344 void OMPClauseProfiler::VisitOMPSeqCstClause(const OMPSeqCstClause *) {}
 345
 346 void OMPClauseProfiler::VisitOMPThreadsClause(const OMPThreadsClause *) {}
 347
 348 void OMPClauseProfiler::VisitOMPSIMDClause(const OMPSIMDClause *) {}
 349
 350 void OMPClauseProfiler::VisitOMPNogroupClause(const OMPNogroupClause *) {}
 351
 352 template<typename T>
 353 void OMPClauseProfiler::VisitOMPClauseList(T *Node) {
 354   for (auto *E : Node->varlists()) {
 355     if (E)
 356       Profiler->VisitStmt(E);
 357   }
 358 }
 359
 360 void OMPClauseProfiler::VisitOMPPrivateClause(const OMPPrivateClause *C) {
 361   VisitOMPClauseList(C);
 362   for (auto *E : C->private_copies()) {
 363     if (E)
 364       Profiler->VisitStmt(E);
 365   }
 366 }
 367 void
 368 OMPClauseProfiler::VisitOMPFirstprivateClause(const OMPFirstprivateClause *C) {
 369   VisitOMPClauseList(C);
 370   VistOMPClauseWithPreInit(C);
 371   for (auto *E : C->private_copies()) {
 372     if (E)
 373       Profiler->VisitStmt(E);
 374   }
 375   for (auto *E : C->inits()) {
 376     if (E)
 377       Profiler->VisitStmt(E);
 378   }
 379 }
 380 void
 381 OMPClauseProfiler::VisitOMPLastprivateClause(const OMPLastprivateClause *C) {
 382   VisitOMPClauseList(C);
 383   VistOMPClauseWithPostUpdate(C);
 384   for (auto *E : C->source_exprs()) {
 385     if (E)
 386       Profiler->VisitStmt(E);
 387   }
 388   for (auto *E : C->destination_exprs()) {
 389     if (E)
 390       Profiler->VisitStmt(E);
 391   }
 392   for (auto *E : C->assignment_ops()) {
 393     if (E)
 394       Profiler->VisitStmt(E);
 395   }
 396 }
 397 void OMPClauseProfiler::VisitOMPSharedClause(const OMPSharedClause *C) {
 398   VisitOMPClauseList(C);
 399 }
 400 void OMPClauseProfiler::VisitOMPReductionClause(
 401                                          const OMPReductionClause *C) {
 402   Profiler->VisitNestedNameSpecifier(
 403       C->getQualifierLoc().getNestedNameSpecifier());
 404   Profiler->VisitName(C->getNameInfo().getName());
 405   VisitOMPClauseList(C);
 406   VistOMPClauseWithPostUpdate(C);
 407   for (auto *E : C->privates()) {
 408     if (E)
 409       Profiler->VisitStmt(E);
 410   }
 411   for (auto *E : C->lhs_exprs()) {
 412     if (E)
 413       Profiler->VisitStmt(E);
 414   }
 415   for (auto *E : C->rhs_exprs()) {
 416     if (E)
 417       Profiler->VisitStmt(E);
 418   }
 419   for (auto *E : C->reduction_ops()) {
 420     if (E)
 421       Profiler->VisitStmt(E);
 422   }
 423 }
 424 void OMPClauseProfiler::VisitOMPLinearClause(const OMPLinearClause *C) {
 425   VisitOMPClauseList(C);
 426   VistOMPClauseWithPostUpdate(C);
 427   for (auto *E : C->privates()) {
 428     if (E)
 429       Profiler->VisitStmt(E);
 430   }
 431   for (auto *E : C->inits()) {
 432     if (E)
 433       Profiler->VisitStmt(E);
 434   }
 435   for (auto *E : C->updates()) {
 436     if (E)
 437       Profiler->VisitStmt(E);
 438   }
 439   for (auto *E : C->finals()) {
 440     if (E)
 441       Profiler->VisitStmt(E);
 442   }
 443   if (C->getStep())
 444     Profiler->VisitStmt(C->getStep());
 445   if (C->getCalcStep())
 446     Profiler->VisitStmt(C->getCalcStep());
 447 }
 448 void OMPClauseProfiler::VisitOMPAlignedClause(const OMPAlignedClause *C) {
 449   VisitOMPClauseList(C);
 450   if (C->getAlignment())
 451     Profiler->VisitStmt(C->getAlignment());
 452 }
 453 void OMPClauseProfiler::VisitOMPCopyinClause(const OMPCopyinClause *C) {
 454   VisitOMPClauseList(C);
 455   for (auto *E : C->source_exprs()) {
 456     if (E)
 457       Profiler->VisitStmt(E);
 458   }
 459   for (auto *E : C->destination_exprs()) {
 460     if (E)
 461       Profiler->VisitStmt(E);
 462   }
 463   for (auto *E : C->assignment_ops()) {
 464     if (E)
 465       Profiler->VisitStmt(E);
 466   }
 467 }
 468 void
 469 OMPClauseProfiler::VisitOMPCopyprivateClause(const OMPCopyprivateClause *C) {
 470   VisitOMPClauseList(C);
 471   for (auto *E : C->source_exprs()) {
 472     if (E)
 473       Profiler->VisitStmt(E);
 474   }
 475   for (auto *E : C->destination_exprs()) {
 476     if (E)
 477       Profiler->VisitStmt(E);
 478   }
 479   for (auto *E : C->assignment_ops()) {
 480     if (E)
 481       Profiler->VisitStmt(E);
 482   }
 483 }
 484 void OMPClauseProfiler::VisitOMPFlushClause(const OMPFlushClause *C) {
 485   VisitOMPClauseList(C);
 486 }
 487 void OMPClauseProfiler::VisitOMPDependClause(const OMPDependClause *C) {
 488   VisitOMPClauseList(C);
 489 }
 490 void OMPClauseProfiler::VisitOMPDeviceClause(const OMPDeviceClause *C) {
 491   if (C->getDevice())
 492     Profiler->VisitStmt(C->getDevice());
 493 }
 494 void OMPClauseProfiler::VisitOMPMapClause(const OMPMapClause *C) {
 495   VisitOMPClauseList(C);
 496 }
 497 void OMPClauseProfiler::VisitOMPNumTeamsClause(const OMPNumTeamsClause *C) {
 498   if (C->getNumTeams())
 499     Profiler->VisitStmt(C->getNumTeams());
 500 }
 501 void OMPClauseProfiler::VisitOMPThreadLimitClause(
 502     const OMPThreadLimitClause *C) {
 503   if (C->getThreadLimit())
 504     Profiler->VisitStmt(C->getThreadLimit());
 505 }
 506 void OMPClauseProfiler::VisitOMPPriorityClause(const OMPPriorityClause *C) {
 507   if (C->getPriority())
 508     Profiler->VisitStmt(C->getPriority());
 509 }
 510 void OMPClauseProfiler::VisitOMPGrainsizeClause(const OMPGrainsizeClause *C) {
 511   if (C->getGrainsize())
 512     Profiler->VisitStmt(C->getGrainsize());
 513 }
 514 void OMPClauseProfiler::VisitOMPNumTasksClause(const OMPNumTasksClause *C) {
 515   if (C->getNumTasks())
 516     Profiler->VisitStmt(C->getNumTasks());
 517 }
 518 void OMPClauseProfiler::VisitOMPHintClause(const OMPHintClause *C) {
 519   if (C->getHint())
 520     Profiler->VisitStmt(C->getHint());
 521 }
 522 void OMPClauseProfiler::VisitOMPToClause(const OMPToClause *C) {
 523   VisitOMPClauseList(C);
 524 }
 525 void OMPClauseProfiler::VisitOMPFromClause(const OMPFromClause *C) {
 526   VisitOMPClauseList(C);
 527 }
 528 void OMPClauseProfiler::VisitOMPUseDevicePtrClause(
 529     const OMPUseDevicePtrClause *C) {
 530   VisitOMPClauseList(C);
 531 }
 532 void OMPClauseProfiler::VisitOMPIsDevicePtrClause(
 533     const OMPIsDevicePtrClause *C) {
 534   VisitOMPClauseList(C);
 535 }
 536 }
 537
 538 void
 539 StmtProfiler::VisitOMPExecutableDirective(const OMPExecutableDirective *S) {
 540   VisitStmt(S);
 541   OMPClauseProfiler P(this);
 542   ArrayRef<OMPClause *> Clauses = S->clauses();
 543   for (ArrayRef<OMPClause *>::iterator I = Clauses.begin(), E = Clauses.end();
 544        I != E; ++I)
 545     if (*I)
 546       P.Visit(*I);
 547 }
 548
 549 void StmtProfiler::VisitOMPLoopDirective(const OMPLoopDirective *S) {
 550   VisitOMPExecutableDirective(S);
 551 }
 552
 553 void StmtProfiler::VisitOMPParallelDirective(const OMPParallelDirective *S) {
 554   VisitOMPExecutableDirective(S);
 555 }
 556
 557 void StmtProfiler::VisitOMPSimdDirective(const OMPSimdDirective *S) {
 558   VisitOMPLoopDirective(S);
 559 }
 560
 561 void StmtProfiler::VisitOMPForDirective(const OMPForDirective *S) {
 562   VisitOMPLoopDirective(S);
 563 }
 564
 565 void StmtProfiler::VisitOMPForSimdDirective(const OMPForSimdDirective *S) {
 566   VisitOMPLoopDirective(S);
 567 }
 568
 569 void StmtProfiler::VisitOMPSectionsDirective(const OMPSectionsDirective *S) {
 570   VisitOMPExecutableDirective(S);
 571 }
 572
 573 void StmtProfiler::VisitOMPSectionDirective(const OMPSectionDirective *S) {
 574   VisitOMPExecutableDirective(S);
 575 }
 576
 577 void StmtProfiler::VisitOMPSingleDirective(const OMPSingleDirective *S) {
 578   VisitOMPExecutableDirective(S);
 579 }
 580
 581 void StmtProfiler::VisitOMPMasterDirective(const OMPMasterDirective *S) {
 582   VisitOMPExecutableDirective(S);
 583 }
 584
 585 void StmtProfiler::VisitOMPCriticalDirective(const OMPCriticalDirective *S) {
 586   VisitOMPExecutableDirective(S);
 587   VisitName(S->getDirectiveName().getName());
 588 }
 589
 590 void
 591 StmtProfiler::VisitOMPParallelForDirective(const OMPParallelForDirective *S) {
 592   VisitOMPLoopDirective(S);
 593 }
 594
 595 void StmtProfiler::VisitOMPParallelForSimdDirective(
 596     const OMPParallelForSimdDirective *S) {
 597   VisitOMPLoopDirective(S);
 598 }
 599
 600 void StmtProfiler::VisitOMPParallelSectionsDirective(
 601     const OMPParallelSectionsDirective *S) {
 602   VisitOMPExecutableDirective(S);
 603 }
 604
 605 void StmtProfiler::VisitOMPTaskDirective(const OMPTaskDirective *S) {
 606   VisitOMPExecutableDirective(S);
 607 }
 608
 609 void StmtProfiler::VisitOMPTaskyieldDirective(const OMPTaskyieldDirective *S) {
 610   VisitOMPExecutableDirective(S);
 611 }
 612
 613 void StmtProfiler::VisitOMPBarrierDirective(const OMPBarrierDirective *S) {
 614   VisitOMPExecutableDirective(S);
 615 }
 616
 617 void StmtProfiler::VisitOMPTaskwaitDirective(const OMPTaskwaitDirective *S) {
 618   VisitOMPExecutableDirective(S);
 619 }
 620
 621 void StmtProfiler::VisitOMPTaskgroupDirective(const OMPTaskgroupDirective *S) {
 622   VisitOMPExecutableDirective(S);
 623 }
 624
 625 void StmtProfiler::VisitOMPFlushDirective(const OMPFlushDirective *S) {
 626   VisitOMPExecutableDirective(S);
 627 }
 628
 629 void StmtProfiler::VisitOMPOrderedDirective(const OMPOrderedDirective *S) {
 630   VisitOMPExecutableDirective(S);
 631 }
 632
 633 void StmtProfiler::VisitOMPAtomicDirective(const OMPAtomicDirective *S) {
 634   VisitOMPExecutableDirective(S);
 635 }
 636
 637 void StmtProfiler::VisitOMPTargetDirective(const OMPTargetDirective *S) {
 638   VisitOMPExecutableDirective(S);
 639 }
 640
 641 void StmtProfiler::VisitOMPTargetDataDirective(const OMPTargetDataDirective *S) {
 642   VisitOMPExecutableDirective(S);
 643 }
 644
 645 void StmtProfiler::VisitOMPTargetEnterDataDirective(
 646     const OMPTargetEnterDataDirective *S) {
 647   VisitOMPExecutableDirective(S);
 648 }
 649
 650 void StmtProfiler::VisitOMPTargetExitDataDirective(
 651     const OMPTargetExitDataDirective *S) {
 652   VisitOMPExecutableDirective(S);
 653 }
 654
 655 void StmtProfiler::VisitOMPTargetParallelDirective(
 656     const OMPTargetParallelDirective *S) {
 657   VisitOMPExecutableDirective(S);
 658 }
 659
 660 void StmtProfiler::VisitOMPTargetParallelForDirective(
 661     const OMPTargetParallelForDirective *S) {
 662   VisitOMPExecutableDirective(S);
 663 }
 664
 665 void StmtProfiler::VisitOMPTeamsDirective(const OMPTeamsDirective *S) {
 666   VisitOMPExecutableDirective(S);
 667 }
 668
 669 void StmtProfiler::VisitOMPCancellationPointDirective(
 670     const OMPCancellationPointDirective *S) {
 671   VisitOMPExecutableDirective(S);
 672 }
 673
 674 void StmtProfiler::VisitOMPCancelDirective(const OMPCancelDirective *S) {
 675   VisitOMPExecutableDirective(S);
 676 }
 677
 678 void StmtProfiler::VisitOMPTaskLoopDirective(const OMPTaskLoopDirective *S) {
 679   VisitOMPLoopDirective(S);
 680 }
 681
 682 void StmtProfiler::VisitOMPTaskLoopSimdDirective(
 683     const OMPTaskLoopSimdDirective *S) {
 684   VisitOMPLoopDirective(S);
 685 }
 686
 687 void StmtProfiler::VisitOMPDistributeDirective(
 688     const OMPDistributeDirective *S) {
 689   VisitOMPLoopDirective(S);
 690 }
 691
 692 void OMPClauseProfiler::VisitOMPDistScheduleClause(
 693     const OMPDistScheduleClause *C) {
 694   VistOMPClauseWithPreInit(C);
 695   if (auto *S = C->getChunkSize())
 696     Profiler->VisitStmt(S);
 697 }
 698
 699 void OMPClauseProfiler::VisitOMPDefaultmapClause(const OMPDefaultmapClause *) {}
 700
 701 void StmtProfiler::VisitOMPTargetUpdateDirective(
 702     const OMPTargetUpdateDirective *S) {
 703   VisitOMPExecutableDirective(S);
 704 }
 705
 706 void StmtProfiler::VisitOMPDistributeParallelForDirective(
 707     const OMPDistributeParallelForDirective *S) {
 708   VisitOMPLoopDirective(S);
 709 }
 710
 711 void StmtProfiler::VisitOMPDistributeParallelForSimdDirective(
 712     const OMPDistributeParallelForSimdDirective *S) {
 713   VisitOMPLoopDirective(S);
 714 }
 715
 716 void StmtProfiler::VisitOMPDistributeSimdDirective(
 717     const OMPDistributeSimdDirective *S) {
 718   VisitOMPLoopDirective(S);
 719 }
 720
 721 void StmtProfiler::VisitOMPTargetParallelForSimdDirective(
 722     const OMPTargetParallelForSimdDirective *S) {
 723   VisitOMPLoopDirective(S);
 724 }
 725
 726 void StmtProfiler::VisitOMPTargetSimdDirective(
 727     const OMPTargetSimdDirective *S) {
 728   VisitOMPLoopDirective(S);
 729 }
 730
 731 void StmtProfiler::VisitOMPTeamsDistributeDirective(
 732     const OMPTeamsDistributeDirective *S) {
 733   VisitOMPLoopDirective(S);
 734 }
 735
 736 void StmtProfiler::VisitOMPTeamsDistributeSimdDirective(
 737     const OMPTeamsDistributeSimdDirective *S) {
 738   VisitOMPLoopDirective(S);
 739 }
 740
 741 void StmtProfiler::VisitOMPTeamsDistributeParallelForSimdDirective(
 742     const OMPTeamsDistributeParallelForSimdDirective *S) {
 743   VisitOMPLoopDirective(S);
 744 }
 745
 746 void StmtProfiler::VisitOMPTeamsDistributeParallelForDirective(
 747     const OMPTeamsDistributeParallelForDirective *S) {
 748   VisitOMPLoopDirective(S);
 749 }
 750
 751 void StmtProfiler::VisitOMPTargetTeamsDirective(
 752     const OMPTargetTeamsDirective *S) {
 753   VisitOMPExecutableDirective(S);
 754 }
 755
 756 void StmtProfiler::VisitOMPTargetTeamsDistributeDirective(
 757     const OMPTargetTeamsDistributeDirective *S) {
 758   VisitOMPLoopDirective(S);
 759 }
 760
 761 void StmtProfiler::VisitOMPTargetTeamsDistributeParallelForDirective(
 762     const OMPTargetTeamsDistributeParallelForDirective *S) {
 763   VisitOMPLoopDirective(S);
 764 }
 765
 766 void StmtProfiler::VisitOMPTargetTeamsDistributeParallelForSimdDirective(
 767     const OMPTargetTeamsDistributeParallelForSimdDirective *S) {
 768   VisitOMPLoopDirective(S);
 769 }
 770
 771 void StmtProfiler::VisitOMPTargetTeamsDistributeSimdDirective(
 772     const OMPTargetTeamsDistributeSimdDirective *S) {
 773   VisitOMPLoopDirective(S);
 774 }
 775
 776 void StmtProfiler::VisitExpr(const Expr *S) {
 777   VisitStmt(S);
 778 }
 779
 780 void StmtProfiler::VisitDeclRefExpr(const DeclRefExpr *S) {
 781   VisitExpr(S);
 782   if (!Canonical)
 783     VisitNestedNameSpecifier(S->getQualifier());
 784   VisitDecl(S->getDecl());
 785   if (!Canonical)
 786     VisitTemplateArguments(S->getTemplateArgs(), S->getNumTemplateArgs());
 787 }
 788
 789 void StmtProfiler::VisitPredefinedExpr(const PredefinedExpr *S) {
 790   VisitExpr(S);
 791   ID.AddInteger(S->getIdentType());
 792 }
 793
 794 void StmtProfiler::VisitIntegerLiteral(const IntegerLiteral *S) {
 795   VisitExpr(S);
 796   S->getValue().Profile(ID);
 797   ID.AddInteger(S->getType()->castAs<BuiltinType>()->getKind());
 798 }
 799
 800 void StmtProfiler::VisitCharacterLiteral(const CharacterLiteral *S) {
 801   VisitExpr(S);
 802   ID.AddInteger(S->getKind());
 803   ID.AddInteger(S->getValue());
 804 }
 805
 806 void StmtProfiler::VisitFloatingLiteral(const FloatingLiteral *S) {
 807   VisitExpr(S);
 808   S->getValue().Profile(ID);
 809   ID.AddBoolean(S->isExact());
 810   ID.AddInteger(S->getType()->castAs<BuiltinType>()->getKind());
 811 }
 812
 813 void StmtProfiler::VisitImaginaryLiteral(const ImaginaryLiteral *S) {
 814   VisitExpr(S);
 815 }
 816
 817 void StmtProfiler::VisitStringLiteral(const StringLiteral *S) {
 818   VisitExpr(S);
 819   ID.AddString(S->getBytes());
 820   ID.AddInteger(S->getKind());
 821 }
 822
 823 void StmtProfiler::VisitParenExpr(const ParenExpr *S) {
 824   VisitExpr(S);
 825 }
 826
 827 void StmtProfiler::VisitParenListExpr(const ParenListExpr *S) {
 828   VisitExpr(S);
 829 }
 830
 831 void StmtProfiler::VisitUnaryOperator(const UnaryOperator *S) {
 832   VisitExpr(S);
 833   ID.AddInteger(S->getOpcode());
 834 }
 835
 836 void StmtProfiler::VisitOffsetOfExpr(const OffsetOfExpr *S) {
 837   VisitType(S->getTypeSourceInfo()->getType());
 838   unsigned n = S->getNumComponents();
 839   for (unsigned i = 0; i < n; ++i) {
 840     const OffsetOfNode &ON = S->getComponent(i);
 841     ID.AddInteger(ON.getKind());
 842     switch (ON.getKind()) {
 843     case OffsetOfNode::Array:
 844       // Expressions handled below.
 845       break;
 846
 847     case OffsetOfNode::Field:
 848       VisitDecl(ON.getField());
 849       break;
 850
 851     case OffsetOfNode::Identifier:
 852       ID.AddPointer(ON.getFieldName());
 853       break;
 854
 855     case OffsetOfNode::Base:
 856       // These nodes are implicit, and therefore don't need profiling.
 857       break;
 858     }
 859   }
 860
 861   VisitExpr(S);
 862 }
 863
 864 void
 865 StmtProfiler::VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr *S) {
 866   VisitExpr(S);
 867   ID.AddInteger(S->getKind());
 868   if (S->isArgumentType())
 869     VisitType(S->getArgumentType());
 870 }
 871
 872 void StmtProfiler::VisitArraySubscriptExpr(const ArraySubscriptExpr *S) {
 873   VisitExpr(S);
 874 }
 875
 876 void StmtProfiler::VisitOMPArraySectionExpr(const OMPArraySectionExpr *S) {
 877   VisitExpr(S);
 878 }
 879
 880 void StmtProfiler::VisitCallExpr(const CallExpr *S) {
 881   VisitExpr(S);
 882 }
 883
 884 void StmtProfiler::VisitMemberExpr(const MemberExpr *S) {
 885   VisitExpr(S);
 886   VisitDecl(S->getMemberDecl());
 887   if (!Canonical)
 888     VisitNestedNameSpecifier(S->getQualifier());
 889   ID.AddBoolean(S->isArrow());
 890 }
 891
 892 void StmtProfiler::VisitCompoundLiteralExpr(const CompoundLiteralExpr *S) {
 893   VisitExpr(S);
 894   ID.AddBoolean(S->isFileScope());
 895 }
 896
 897 void StmtProfiler::VisitCastExpr(const CastExpr *S) {
 898   VisitExpr(S);
 899 }
 900
 901 void StmtProfiler::VisitImplicitCastExpr(const ImplicitCastExpr *S) {
 902   VisitCastExpr(S);
 903   ID.AddInteger(S->getValueKind());
 904 }
 905
 906 void StmtProfiler::VisitExplicitCastExpr(const ExplicitCastExpr *S) {
 907   VisitCastExpr(S);
 908   VisitType(S->getTypeAsWritten());
 909 }
 910
 911 void StmtProfiler::VisitCStyleCastExpr(const CStyleCastExpr *S) {
 912   VisitExplicitCastExpr(S);
 913 }
 914
 915 void StmtProfiler::VisitBinaryOperator(const BinaryOperator *S) {
 916   VisitExpr(S);
 917   ID.AddInteger(S->getOpcode());
 918 }
 919
 920 void
 921 StmtProfiler::VisitCompoundAssignOperator(const CompoundAssignOperator *S) {
 922   VisitBinaryOperator(S);
 923 }
 924
 925 void StmtProfiler::VisitConditionalOperator(const ConditionalOperator *S) {
 926   VisitExpr(S);
 927 }
 928
 929 void StmtProfiler::VisitBinaryConditionalOperator(
 930     const BinaryConditionalOperator *S) {
 931   VisitExpr(S);
 932 }
 933
 934 void StmtProfiler::VisitAddrLabelExpr(const AddrLabelExpr *S) {
 935   VisitExpr(S);
 936   VisitDecl(S->getLabel());
 937 }
 938
 939 void StmtProfiler::VisitStmtExpr(const StmtExpr *S) {
 940   VisitExpr(S);
 941 }
 942
 943 void StmtProfiler::VisitShuffleVectorExpr(const ShuffleVectorExpr *S) {
 944   VisitExpr(S);
 945 }
 946
 947 void StmtProfiler::VisitConvertVectorExpr(const ConvertVectorExpr *S) {
 948   VisitExpr(S);
 949 }
 950
 951 void StmtProfiler::VisitChooseExpr(const ChooseExpr *S) {
 952   VisitExpr(S);
 953 }
 954
 955 void StmtProfiler::VisitGNUNullExpr(const GNUNullExpr *S) {
 956   VisitExpr(S);
 957 }
 958
 959 void StmtProfiler::VisitVAArgExpr(const VAArgExpr *S) {
 960   VisitExpr(S);
 961 }
 962
 963 void StmtProfiler::VisitInitListExpr(const InitListExpr *S) {
 964   if (S->getSyntacticForm()) {
 965     VisitInitListExpr(S->getSyntacticForm());
 966     return;
 967   }
 968
 969   VisitExpr(S);
 970 }
 971
 972 void StmtProfiler::VisitDesignatedInitExpr(const DesignatedInitExpr *S) {
 973   VisitExpr(S);
 974   ID.AddBoolean(S->usesGNUSyntax());
 975   for (const DesignatedInitExpr::Designator &D : S->designators()) {
 976     if (D.isFieldDesignator()) {
 977       ID.AddInteger(0);
 978       VisitName(D.getFieldName());
 979       continue;
 980     }
 981
 982     if (D.isArrayDesignator()) {
 983       ID.AddInteger(1);
 984     } else {
 985       assert(D.isArrayRangeDesignator());
 986       ID.AddInteger(2);
 987     }
 988     ID.AddInteger(D.getFirstExprIndex());
 989   }
 990 }
 991
 992 // Seems that if VisitInitListExpr() only works on the syntactic form of an
 993 // InitListExpr, then a DesignatedInitUpdateExpr is not encountered.
 994 void StmtProfiler::VisitDesignatedInitUpdateExpr(
 995     const DesignatedInitUpdateExpr *S) {
 996   llvm_unreachable("Unexpected DesignatedInitUpdateExpr in syntactic form of "
 997                    "initializer");
 998 }
 999
1000 void StmtProfiler::VisitArrayInitLoopExpr(const ArrayInitLoopExpr *S) {
1001   VisitExpr(S);
1002 }
1003
1004 void StmtProfiler::VisitArrayInitIndexExpr(const ArrayInitIndexExpr *S) {
1005   VisitExpr(S);
1006 }
1007
1008 void StmtProfiler::VisitNoInitExpr(const NoInitExpr *S) {
1009   llvm_unreachable("Unexpected NoInitExpr in syntactic form of initializer");
1010 }
1011
1012 void StmtProfiler::VisitImplicitValueInitExpr(const ImplicitValueInitExpr *S) {
1013   VisitExpr(S);
1014 }
1015
1016 void StmtProfiler::VisitExtVectorElementExpr(const ExtVectorElementExpr *S) {
1017   VisitExpr(S);
1018   VisitName(&S->getAccessor());
1019 }
1020
1021 void StmtProfiler::VisitBlockExpr(const BlockExpr *S) {
1022   VisitExpr(S);
1023   VisitDecl(S->getBlockDecl());
1024 }
1025
1026 void StmtProfiler::VisitGenericSelectionExpr(const GenericSelectionExpr *S) {
1027   VisitExpr(S);
1028   for (unsigned i = 0; i != S->getNumAssocs(); ++i) {
1029     QualType T = S->getAssocType(i);
1030     if (T.isNull())
1031       ID.AddPointer(nullptr);
1032     else
1033       VisitType(T);
1034     VisitExpr(S->getAssocExpr(i));
1035   }
1036 }
1037
1038 void StmtProfiler::VisitPseudoObjectExpr(const PseudoObjectExpr *S) {
1039   VisitExpr(S);
1040   for (PseudoObjectExpr::const_semantics_iterator
1041          i = S->semantics_begin(), e = S->semantics_end(); i != e; ++i)
1042     // Normally, we would not profile the source expressions of OVEs.
1043     if (const OpaqueValueExpr *OVE = dyn_cast<OpaqueValueExpr>(*i))
1044       Visit(OVE->getSourceExpr());
1045 }
1046
1047 void StmtProfiler::VisitAtomicExpr(const AtomicExpr *S) {
1048   VisitExpr(S);
1049   ID.AddInteger(S->getOp());
1050 }
1051
1052 static Stmt::StmtClass DecodeOperatorCall(const CXXOperatorCallExpr *S,
1053                                           UnaryOperatorKind &UnaryOp,
1054                                           BinaryOperatorKind &BinaryOp) {
1055   switch (S->getOperator()) {
1056   case OO_None:
1057   case OO_New:
1058   case OO_Delete:
1059   case OO_Array_New:
1060   case OO_Array_Delete:
1061   case OO_Arrow:
1062   case OO_Call:
1063   case OO_Conditional:
1064   case OO_Coawait:
1065   case NUM_OVERLOADED_OPERATORS:
1066     llvm_unreachable("Invalid operator call kind");
1067
1068   case OO_Plus:
1069     if (S->getNumArgs() == 1) {
1070       UnaryOp = UO_Plus;
1071       return Stmt::UnaryOperatorClass;
1072     }
1073
1074     BinaryOp = BO_Add;
1075     return Stmt::BinaryOperatorClass;
1076
1077   case OO_Minus:
1078     if (S->getNumArgs() == 1) {
1079       UnaryOp = UO_Minus;
1080       return Stmt::UnaryOperatorClass;
1081     }
1082
1083     BinaryOp = BO_Sub;
1084     return Stmt::BinaryOperatorClass;
1085
1086   case OO_Star:
1087     if (S->getNumArgs() == 1) {
1088       UnaryOp = UO_Deref;
1089       return Stmt::UnaryOperatorClass;
1090     }
1091
1092     BinaryOp = BO_Mul;
1093     return Stmt::BinaryOperatorClass;
1094
1095   case OO_Slash:
1096     BinaryOp = BO_Div;
1097     return Stmt::BinaryOperatorClass;
1098
1099   case OO_Percent:
1100     BinaryOp = BO_Rem;
1101     return Stmt::BinaryOperatorClass;
1102
1103   case OO_Caret:
1104     BinaryOp = BO_Xor;
1105     return Stmt::BinaryOperatorClass;
1106
1107   case OO_Amp:
1108     if (S->getNumArgs() == 1) {
1109       UnaryOp = UO_AddrOf;
1110       return Stmt::UnaryOperatorClass;
1111     }
1112
1113     BinaryOp = BO_And;
1114     return Stmt::BinaryOperatorClass;
1115
1116   case OO_Pipe:
1117     BinaryOp = BO_Or;
1118     return Stmt::BinaryOperatorClass;
1119
1120   case OO_Tilde:
1121     UnaryOp = UO_Not;
1122     return Stmt::UnaryOperatorClass;
1123
1124   case OO_Exclaim:
1125     UnaryOp = UO_LNot;
1126     return Stmt::UnaryOperatorClass;
1127
1128   case OO_Equal:
1129     BinaryOp = BO_Assign;
1130     return Stmt::BinaryOperatorClass;
1131
1132   case OO_Less:
1133     BinaryOp = BO_LT;
1134     return Stmt::BinaryOperatorClass;
1135
1136   case OO_Greater:
1137     BinaryOp = BO_GT;
1138     return Stmt::BinaryOperatorClass;
1139
1140   case OO_PlusEqual:
1141     BinaryOp = BO_AddAssign;
1142     return Stmt::CompoundAssignOperatorClass;
1143
1144   case OO_MinusEqual:
1145     BinaryOp = BO_SubAssign;
1146     return Stmt::CompoundAssignOperatorClass;
1147
1148   case OO_StarEqual:
1149     BinaryOp = BO_MulAssign;
1150     return Stmt::CompoundAssignOperatorClass;
1151
1152   case OO_SlashEqual:
1153     BinaryOp = BO_DivAssign;
1154     return Stmt::CompoundAssignOperatorClass;
1155
1156   case OO_PercentEqual:
1157     BinaryOp = BO_RemAssign;
1158     return Stmt::CompoundAssignOperatorClass;
1159
1160   case OO_CaretEqual:
1161     BinaryOp = BO_XorAssign;
1162     return Stmt::CompoundAssignOperatorClass;
1163
1164   case OO_AmpEqual:
1165     BinaryOp = BO_AndAssign;
1166     return Stmt::CompoundAssignOperatorClass;
1167
1168   case OO_PipeEqual:
1169     BinaryOp = BO_OrAssign;
1170     return Stmt::CompoundAssignOperatorClass;
1171
1172   case OO_LessLess:
1173     BinaryOp = BO_Shl;
1174     return Stmt::BinaryOperatorClass;
1175
1176   case OO_GreaterGreater:
1177     BinaryOp = BO_Shr;
1178     return Stmt::BinaryOperatorClass;
1179
1180   case OO_LessLessEqual:
1181     BinaryOp = BO_ShlAssign;
1182     return Stmt::CompoundAssignOperatorClass;
1183
1184   case OO_GreaterGreaterEqual:
1185     BinaryOp = BO_ShrAssign;
1186     return Stmt::CompoundAssignOperatorClass;
1187
1188   case OO_EqualEqual:
1189     BinaryOp = BO_EQ;
1190     return Stmt::BinaryOperatorClass;
1191
1192   case OO_ExclaimEqual:
1193     BinaryOp = BO_NE;
1194     return Stmt::BinaryOperatorClass;
1195
1196   case OO_LessEqual:
1197     BinaryOp = BO_LE;
1198     return Stmt::BinaryOperatorClass;
1199
1200   case OO_GreaterEqual:
1201     BinaryOp = BO_GE;
1202     return Stmt::BinaryOperatorClass;
1203
1204   case OO_AmpAmp:
1205     BinaryOp = BO_LAnd;
1206     return Stmt::BinaryOperatorClass;
1207
1208   case OO_PipePipe:
1209     BinaryOp = BO_LOr;
1210     return Stmt::BinaryOperatorClass;
1211
1212   case OO_PlusPlus:
1213     UnaryOp = S->getNumArgs() == 1? UO_PreInc
1214                                   : UO_PostInc;
1215     return Stmt::UnaryOperatorClass;
1216
1217   case OO_MinusMinus:
1218     UnaryOp = S->getNumArgs() == 1? UO_PreDec
1219                                   : UO_PostDec;
1220     return Stmt::UnaryOperatorClass;
1221
1222   case OO_Comma:
1223     BinaryOp = BO_Comma;
1224     return Stmt::BinaryOperatorClass;
1225
1226   case OO_ArrowStar:
1227     BinaryOp = BO_PtrMemI;
1228     return Stmt::BinaryOperatorClass;
1229
1230   case OO_Subscript:
1231     return Stmt::ArraySubscriptExprClass;
1232   }
1233
1234   llvm_unreachable("Invalid overloaded operator expression");
1235 }
1236
1237 void StmtProfiler::VisitCXXOperatorCallExpr(const CXXOperatorCallExpr *S) {
1238   if (S->isTypeDependent()) {
1239     // Type-dependent operator calls are profiled like their underlying
1240     // syntactic operator.
1241     //
1242     // An operator call to operator-> is always implicit, so just skip it. The
1243     // enclosing MemberExpr will profile the actual member access.
1244     if (S->getOperator() == OO_Arrow)
1245       return Visit(S->getArg(0));
1246
1247     UnaryOperatorKind UnaryOp = UO_Extension;
1248     BinaryOperatorKind BinaryOp = BO_Comma;
1249     Stmt::StmtClass SC = DecodeOperatorCall(S, UnaryOp, BinaryOp);
1250
1251     ID.AddInteger(SC);
1252     for (unsigned I = 0, N = S->getNumArgs(); I != N; ++I)
1253       Visit(S->getArg(I));
1254     if (SC == Stmt::UnaryOperatorClass)
1255       ID.AddInteger(UnaryOp);
1256     else if (SC == Stmt::BinaryOperatorClass ||
1257              SC == Stmt::CompoundAssignOperatorClass)
1258       ID.AddInteger(BinaryOp);
1259     else
1260       assert(SC == Stmt::ArraySubscriptExprClass);
1261
1262     return;
1263   }
1264
1265   VisitCallExpr(S);
1266   ID.AddInteger(S->getOperator());
1267 }
1268
1269 void StmtProfiler::VisitCXXMemberCallExpr(const CXXMemberCallExpr *S) {
1270   VisitCallExpr(S);
1271 }
1272
1273 void StmtProfiler::VisitCUDAKernelCallExpr(const CUDAKernelCallExpr *S) {
1274   VisitCallExpr(S);
1275 }
1276
1277 void StmtProfiler::VisitAsTypeExpr(const AsTypeExpr *S) {
1278   VisitExpr(S);
1279 }
1280
1281 void StmtProfiler::VisitCXXNamedCastExpr(const CXXNamedCastExpr *S) {
1282   VisitExplicitCastExpr(S);
1283 }
1284
1285 void StmtProfiler::VisitCXXStaticCastExpr(const CXXStaticCastExpr *S) {
1286   VisitCXXNamedCastExpr(S);
1287 }
1288
1289 void StmtProfiler::VisitCXXDynamicCastExpr(const CXXDynamicCastExpr *S) {
1290   VisitCXXNamedCastExpr(S);
1291 }
1292
1293 void
1294 StmtProfiler::VisitCXXReinterpretCastExpr(const CXXReinterpretCastExpr *S) {
1295   VisitCXXNamedCastExpr(S);
1296 }
1297
1298 void StmtProfiler::VisitCXXConstCastExpr(const CXXConstCastExpr *S) {
1299   VisitCXXNamedCastExpr(S);
1300 }
1301
1302 void StmtProfiler::VisitUserDefinedLiteral(const UserDefinedLiteral *S) {
1303   VisitCallExpr(S);
1304 }
1305
1306 void StmtProfiler::VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *S) {
1307   VisitExpr(S);
1308   ID.AddBoolean(S->getValue());
1309 }
1310
1311 void StmtProfiler::VisitCXXNullPtrLiteralExpr(const CXXNullPtrLiteralExpr *S) {
1312   VisitExpr(S);
1313 }
1314
1315 void StmtProfiler::VisitCXXStdInitializerListExpr(
1316     const CXXStdInitializerListExpr *S) {
1317   VisitExpr(S);
1318 }
1319
1320 void StmtProfiler::VisitCXXTypeidExpr(const CXXTypeidExpr *S) {
1321   VisitExpr(S);
1322   if (S->isTypeOperand())
1323     VisitType(S->getTypeOperandSourceInfo()->getType());
1324 }
1325
1326 void StmtProfiler::VisitCXXUuidofExpr(const CXXUuidofExpr *S) {
1327   VisitExpr(S);
1328   if (S->isTypeOperand())
1329     VisitType(S->getTypeOperandSourceInfo()->getType());
1330 }
1331
1332 void StmtProfiler::VisitMSPropertyRefExpr(const MSPropertyRefExpr *S) {
1333   VisitExpr(S);
1334   VisitDecl(S->getPropertyDecl());
1335 }
1336
1337 void StmtProfiler::VisitMSPropertySubscriptExpr(
1338     const MSPropertySubscriptExpr *S) {
1339   VisitExpr(S);
1340 }
1341
1342 void StmtProfiler::VisitCXXThisExpr(const CXXThisExpr *S) {
1343   VisitExpr(S);
1344   ID.AddBoolean(S->isImplicit());
1345 }
1346
1347 void StmtProfiler::VisitCXXThrowExpr(const CXXThrowExpr *S) {
1348   VisitExpr(S);
1349 }
1350
1351 void StmtProfiler::VisitCXXDefaultArgExpr(const CXXDefaultArgExpr *S) {
1352   VisitExpr(S);
1353   VisitDecl(S->getParam());
1354 }
1355
1356 void StmtProfiler::VisitCXXDefaultInitExpr(const CXXDefaultInitExpr *S) {
1357   VisitExpr(S);
1358   VisitDecl(S->getField());
1359 }
1360
1361 void StmtProfiler::VisitCXXBindTemporaryExpr(const CXXBindTemporaryExpr *S) {
1362   VisitExpr(S);
1363   VisitDecl(
1364          const_cast<CXXDestructorDecl *>(S->getTemporary()->getDestructor()));
1365 }
1366
1367 void StmtProfiler::VisitCXXConstructExpr(const CXXConstructExpr *S) {
1368   VisitExpr(S);
1369   VisitDecl(S->getConstructor());
1370   ID.AddBoolean(S->isElidable());
1371 }
1372
1373 void StmtProfiler::VisitCXXInheritedCtorInitExpr(
1374     const CXXInheritedCtorInitExpr *S) {
1375   VisitExpr(S);
1376   VisitDecl(S->getConstructor());
1377 }
1378
1379 void StmtProfiler::VisitCXXFunctionalCastExpr(const CXXFunctionalCastExpr *S) {
1380   VisitExplicitCastExpr(S);
1381 }
1382
1383 void
1384 StmtProfiler::VisitCXXTemporaryObjectExpr(const CXXTemporaryObjectExpr *S) {
1385   VisitCXXConstructExpr(S);
1386 }
1387
1388 void
1389 StmtProfiler::VisitLambdaExpr(const LambdaExpr *S) {
1390   VisitExpr(S);
1391   for (LambdaExpr::capture_iterator C = S->explicit_capture_begin(),
1392                                  CEnd = S->explicit_capture_end();
1393        C != CEnd; ++C) {
1394     ID.AddInteger(C->getCaptureKind());
1395     switch (C->getCaptureKind()) {
1396     case LCK_StarThis:
1397     case LCK_This:
1398       break;
1399     case LCK_ByRef:
1400     case LCK_ByCopy:
1401       VisitDecl(C->getCapturedVar());
1402       ID.AddBoolean(C->isPackExpansion());
1403       break;
1404     case LCK_VLAType:
1405       llvm_unreachable("VLA type in explicit captures.");
1406     }
1407   }
1408   // Note: If we actually needed to be able to match lambda
1409   // expressions, we would have to consider parameters and return type
1410   // here, among other things.
1411   VisitStmt(S->getBody());
1412 }
1413
1414 void
1415 StmtProfiler::VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *S) {
1416   VisitExpr(S);
1417 }
1418
1419 void StmtProfiler::VisitCXXDeleteExpr(const CXXDeleteExpr *S) {
1420   VisitExpr(S);
1421   ID.AddBoolean(S->isGlobalDelete());
1422   ID.AddBoolean(S->isArrayForm());
1423   VisitDecl(S->getOperatorDelete());
1424 }
1425
1426 void StmtProfiler::VisitCXXNewExpr(const CXXNewExpr *S) {
1427   VisitExpr(S);
1428   VisitType(S->getAllocatedType());
1429   VisitDecl(S->getOperatorNew());
1430   VisitDecl(S->getOperatorDelete());
1431   ID.AddBoolean(S->isArray());
1432   ID.AddInteger(S->getNumPlacementArgs());
1433   ID.AddBoolean(S->isGlobalNew());
1434   ID.AddBoolean(S->isParenTypeId());
1435   ID.AddInteger(S->getInitializationStyle());
1436 }
1437
1438 void
1439 StmtProfiler::VisitCXXPseudoDestructorExpr(const CXXPseudoDestructorExpr *S) {
1440   VisitExpr(S);
1441   ID.AddBoolean(S->isArrow());
1442   VisitNestedNameSpecifier(S->getQualifier());
1443   ID.AddBoolean(S->getScopeTypeInfo() != nullptr);
1444   if (S->getScopeTypeInfo())
1445     VisitType(S->getScopeTypeInfo()->getType());
1446   ID.AddBoolean(S->getDestroyedTypeInfo() != nullptr);
1447   if (S->getDestroyedTypeInfo())
1448     VisitType(S->getDestroyedType());
1449   else
1450     ID.AddPointer(S->getDestroyedTypeIdentifier());
1451 }
1452
1453 void StmtProfiler::VisitOverloadExpr(const OverloadExpr *S) {
1454   VisitExpr(S);
1455   VisitNestedNameSpecifier(S->getQualifier());
1456   VisitName(S->getName());
1457   ID.AddBoolean(S->hasExplicitTemplateArgs());
1458   if (S->hasExplicitTemplateArgs())
1459     VisitTemplateArguments(S->getTemplateArgs(), S->getNumTemplateArgs());
1460 }
1461
1462 void
1463 StmtProfiler::VisitUnresolvedLookupExpr(const UnresolvedLookupExpr *S) {
1464   VisitOverloadExpr(S);
1465 }
1466
1467 void StmtProfiler::VisitTypeTraitExpr(const TypeTraitExpr *S) {
1468   VisitExpr(S);
1469   ID.AddInteger(S->getTrait());
1470   ID.AddInteger(S->getNumArgs());
1471   for (unsigned I = 0, N = S->getNumArgs(); I != N; ++I)
1472     VisitType(S->getArg(I)->getType());
1473 }
1474
1475 void StmtProfiler::VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *S) {
1476   VisitExpr(S);
1477   ID.AddInteger(S->getTrait());
1478   VisitType(S->getQueriedType());
1479 }
1480
1481 void StmtProfiler::VisitExpressionTraitExpr(const ExpressionTraitExpr *S) {
1482   VisitExpr(S);
1483   ID.AddInteger(S->getTrait());
1484   VisitExpr(S->getQueriedExpression());
1485 }
1486
1487 void StmtProfiler::VisitDependentScopeDeclRefExpr(
1488     const DependentScopeDeclRefExpr *S) {
1489   VisitExpr(S);
1490   VisitName(S->getDeclName());
1491   VisitNestedNameSpecifier(S->getQualifier());
1492   ID.AddBoolean(S->hasExplicitTemplateArgs());
1493   if (S->hasExplicitTemplateArgs())
1494     VisitTemplateArguments(S->getTemplateArgs(), S->getNumTemplateArgs());
1495 }
1496
1497 void StmtProfiler::VisitExprWithCleanups(const ExprWithCleanups *S) {
1498   VisitExpr(S);
1499 }
1500
1501 void StmtProfiler::VisitCXXUnresolvedConstructExpr(
1502     const CXXUnresolvedConstructExpr *S) {
1503   VisitExpr(S);
1504   VisitType(S->getTypeAsWritten());
1505 }
1506
1507 void StmtProfiler::VisitCXXDependentScopeMemberExpr(
1508     const CXXDependentScopeMemberExpr *S) {
1509   ID.AddBoolean(S->isImplicitAccess());
1510   if (!S->isImplicitAccess()) {
1511     VisitExpr(S);
1512     ID.AddBoolean(S->isArrow());
1513   }
1514   VisitNestedNameSpecifier(S->getQualifier());
1515   VisitName(S->getMember());
1516   ID.AddBoolean(S->hasExplicitTemplateArgs());
1517   if (S->hasExplicitTemplateArgs())
1518     VisitTemplateArguments(S->getTemplateArgs(), S->getNumTemplateArgs());
1519 }
1520
1521 void StmtProfiler::VisitUnresolvedMemberExpr(const UnresolvedMemberExpr *S) {
1522   ID.AddBoolean(S->isImplicitAccess());
1523   if (!S->isImplicitAccess()) {
1524     VisitExpr(S);
1525     ID.AddBoolean(S->isArrow());
1526   }
1527   VisitNestedNameSpecifier(S->getQualifier());
1528   VisitName(S->getMemberName());
1529   ID.AddBoolean(S->hasExplicitTemplateArgs());
1530   if (S->hasExplicitTemplateArgs())
1531     VisitTemplateArguments(S->getTemplateArgs(), S->getNumTemplateArgs());
1532 }
1533
1534 void StmtProfiler::VisitCXXNoexceptExpr(const CXXNoexceptExpr *S) {
1535   VisitExpr(S);
1536 }
1537
1538 void StmtProfiler::VisitPackExpansionExpr(const PackExpansionExpr *S) {
1539   VisitExpr(S);
1540 }
1541
1542 void StmtProfiler::VisitSizeOfPackExpr(const SizeOfPackExpr *S) {
1543   VisitExpr(S);
1544   VisitDecl(S->getPack());
1545   if (S->isPartiallySubstituted()) {
1546     auto Args = S->getPartialArguments();
1547     ID.AddInteger(Args.size());
1548     for (const auto &TA : Args)
1549       VisitTemplateArgument(TA);
1550   } else {
1551     ID.AddInteger(0);
1552   }
1553 }
1554
1555 void StmtProfiler::VisitSubstNonTypeTemplateParmPackExpr(
1556     const SubstNonTypeTemplateParmPackExpr *S) {
1557   VisitExpr(S);
1558   VisitDecl(S->getParameterPack());
1559   VisitTemplateArgument(S->getArgumentPack());
1560 }
1561
1562 void StmtProfiler::VisitSubstNonTypeTemplateParmExpr(
1563     const SubstNonTypeTemplateParmExpr *E) {
1564   // Profile exactly as the replacement expression.
1565   Visit(E->getReplacement());
1566 }
1567
1568 void StmtProfiler::VisitFunctionParmPackExpr(const FunctionParmPackExpr *S) {
1569   VisitExpr(S);
1570   VisitDecl(S->getParameterPack());
1571   ID.AddInteger(S->getNumExpansions());
1572   for (FunctionParmPackExpr::iterator I = S->begin(), E = S->end(); I != E; ++I)
1573     VisitDecl(*I);
1574 }
1575
1576 void StmtProfiler::VisitMaterializeTemporaryExpr(
1577                                            const MaterializeTemporaryExpr *S) {
1578   VisitExpr(S);
1579 }
1580
1581 void StmtProfiler::VisitCXXFoldExpr(const CXXFoldExpr *S) {
1582   VisitExpr(S);
1583   ID.AddInteger(S->getOperator());
1584 }
1585
1586 void StmtProfiler::VisitCoroutineBodyStmt(const CoroutineBodyStmt *S) {
1587   VisitStmt(S);
1588 }
1589
1590 void StmtProfiler::VisitCoreturnStmt(const CoreturnStmt *S) {
1591   VisitStmt(S);
1592 }
1593
1594 void StmtProfiler::VisitCoawaitExpr(const CoawaitExpr *S) {
1595   VisitExpr(S);
1596 }
1597
1598 void StmtProfiler::VisitCoyieldExpr(const CoyieldExpr *S) {
1599   VisitExpr(S);
1600 }
1601
1602 void StmtProfiler::VisitOpaqueValueExpr(const OpaqueValueExpr *E) {
1603   VisitExpr(E);
1604 }
1605
1606 void StmtProfiler::VisitTypoExpr(const TypoExpr *E) {
1607   VisitExpr(E);
1608 }
1609
1610 void StmtProfiler::VisitObjCStringLiteral(const ObjCStringLiteral *S) {
1611   VisitExpr(S);
1612 }
1613
1614 void StmtProfiler::VisitObjCBoxedExpr(const ObjCBoxedExpr *E) {
1615   VisitExpr(E);
1616 }
1617
1618 void StmtProfiler::VisitObjCArrayLiteral(const ObjCArrayLiteral *E) {
1619   VisitExpr(E);
1620 }
1621
1622 void StmtProfiler::VisitObjCDictionaryLiteral(const ObjCDictionaryLiteral *E) {
1623   VisitExpr(E);
1624 }
1625
1626 void StmtProfiler::VisitObjCEncodeExpr(const ObjCEncodeExpr *S) {
1627   VisitExpr(S);
1628   VisitType(S->getEncodedType());
1629 }
1630
1631 void StmtProfiler::VisitObjCSelectorExpr(const ObjCSelectorExpr *S) {
1632   VisitExpr(S);
1633   VisitName(S->getSelector());
1634 }
1635
1636 void StmtProfiler::VisitObjCProtocolExpr(const ObjCProtocolExpr *S) {
1637   VisitExpr(S);
1638   VisitDecl(S->getProtocol());
1639 }
1640
1641 void StmtProfiler::VisitObjCIvarRefExpr(const ObjCIvarRefExpr *S) {
1642   VisitExpr(S);
1643   VisitDecl(S->getDecl());
1644   ID.AddBoolean(S->isArrow());
1645   ID.AddBoolean(S->isFreeIvar());
1646 }
1647
1648 void StmtProfiler::VisitObjCPropertyRefExpr(const ObjCPropertyRefExpr *S) {
1649   VisitExpr(S);
1650   if (S->isImplicitProperty()) {
1651     VisitDecl(S->getImplicitPropertyGetter());
1652     VisitDecl(S->getImplicitPropertySetter());
1653   } else {
1654     VisitDecl(S->getExplicitProperty());
1655   }
1656   if (S->isSuperReceiver()) {
1657     ID.AddBoolean(S->isSuperReceiver());
1658     VisitType(S->getSuperReceiverType());
1659   }
1660 }
1661
1662 void StmtProfiler::VisitObjCSubscriptRefExpr(const ObjCSubscriptRefExpr *S) {
1663   VisitExpr(S);
1664   VisitDecl(S->getAtIndexMethodDecl());
1665   VisitDecl(S->setAtIndexMethodDecl());
1666 }
1667
1668 void StmtProfiler::VisitObjCMessageExpr(const ObjCMessageExpr *S) {
1669   VisitExpr(S);
1670   VisitName(S->getSelector());
1671   VisitDecl(S->getMethodDecl());
1672 }
1673
1674 void StmtProfiler::VisitObjCIsaExpr(const ObjCIsaExpr *S) {
1675   VisitExpr(S);
1676   ID.AddBoolean(S->isArrow());
1677 }
1678
1679 void StmtProfiler::VisitObjCBoolLiteralExpr(const ObjCBoolLiteralExpr *S) {
1680   VisitExpr(S);
1681   ID.AddBoolean(S->getValue());
1682 }
1683
1684 void StmtProfiler::VisitObjCIndirectCopyRestoreExpr(
1685     const ObjCIndirectCopyRestoreExpr *S) {
1686   VisitExpr(S);
1687   ID.AddBoolean(S->shouldCopy());
1688 }
1689
1690 void StmtProfiler::VisitObjCBridgedCastExpr(const ObjCBridgedCastExpr *S) {
1691   VisitExplicitCastExpr(S);
1692   ID.AddBoolean(S->getBridgeKind());
1693 }
1694
1695 void StmtProfiler::VisitObjCAvailabilityCheckExpr(
1696     const ObjCAvailabilityCheckExpr *S) {
1697   VisitExpr(S);
1698 }
1699
1700 void StmtProfiler::VisitDecl(const Decl *D) {
1701   ID.AddInteger(D? D->getKind() : 0);
1702
1703   if (Canonical && D) {
1704     if (const NonTypeTemplateParmDecl *NTTP =
1705           dyn_cast<NonTypeTemplateParmDecl>(D)) {
1706       ID.AddInteger(NTTP->getDepth());
1707       ID.AddInteger(NTTP->getIndex());
1708       ID.AddBoolean(NTTP->isParameterPack());
1709       VisitType(NTTP->getType());
1710       return;
1711     }
1712
1713     if (const ParmVarDecl *Parm = dyn_cast<ParmVarDecl>(D)) {
1714       // The Itanium C++ ABI uses the type, scope depth, and scope
1715       // index of a parameter when mangling expressions that involve
1716       // function parameters, so we will use the parameter's type for
1717       // establishing function parameter identity. That way, our
1718       // definition of "equivalent" (per C++ [temp.over.link]) is at
1719       // least as strong as the definition of "equivalent" used for
1720       // name mangling.
1721       VisitType(Parm->getType());
1722       ID.AddInteger(Parm->getFunctionScopeDepth());
1723       ID.AddInteger(Parm->getFunctionScopeIndex());
1724       return;
1725     }
1726
1727     if (const TemplateTypeParmDecl *TTP =
1728           dyn_cast<TemplateTypeParmDecl>(D)) {
1729       ID.AddInteger(TTP->getDepth());
1730       ID.AddInteger(TTP->getIndex());
1731       ID.AddBoolean(TTP->isParameterPack());
1732       return;
1733     }
1734
1735     if (const TemplateTemplateParmDecl *TTP =
1736           dyn_cast<TemplateTemplateParmDecl>(D)) {
1737       ID.AddInteger(TTP->getDepth());
1738       ID.AddInteger(TTP->getIndex());
1739       ID.AddBoolean(TTP->isParameterPack());
1740       return;
1741     }
1742   }
1743
1744   ID.AddPointer(D? D->getCanonicalDecl() : nullptr);
1745 }
1746
1747 void StmtProfiler::VisitType(QualType T) {
1748   if (Canonical)
1749     T = Context.getCanonicalType(T);
1750
1751   ID.AddPointer(T.getAsOpaquePtr());
1752 }
1753
1754 void StmtProfiler::VisitName(DeclarationName Name) {
1755   ID.AddPointer(Name.getAsOpaquePtr());
1756 }
1757
1758 void StmtProfiler::VisitNestedNameSpecifier(NestedNameSpecifier *NNS) {
1759   if (Canonical)
1760     NNS = Context.getCanonicalNestedNameSpecifier(NNS);
1761   ID.AddPointer(NNS);
1762 }
1763
1764 void StmtProfiler::VisitTemplateName(TemplateName Name) {
1765   if (Canonical)
1766     Name = Context.getCanonicalTemplateName(Name);
1767
1768   Name.Profile(ID);
1769 }
1770
1771 void StmtProfiler::VisitTemplateArguments(const TemplateArgumentLoc *Args,
1772                                           unsigned NumArgs) {
1773   ID.AddInteger(NumArgs);
1774   for (unsigned I = 0; I != NumArgs; ++I)
1775     VisitTemplateArgument(Args[I].getArgument());
1776 }
1777
1778 void StmtProfiler::VisitTemplateArgument(const TemplateArgument &Arg) {
1779   // Mostly repetitive with TemplateArgument::Profile!
1780   ID.AddInteger(Arg.getKind());
1781   switch (Arg.getKind()) {
1782   case TemplateArgument::Null:
1783     break;
1784
1785   case TemplateArgument::Type:
1786     VisitType(Arg.getAsType());
1787     break;
1788
1789   case TemplateArgument::Template:
1790   case TemplateArgument::TemplateExpansion:
1791     VisitTemplateName(Arg.getAsTemplateOrTemplatePattern());
1792     break;
1793
1794   case TemplateArgument::Declaration:
1795     VisitDecl(Arg.getAsDecl());
1796     break;
1797
1798   case TemplateArgument::NullPtr:
1799     VisitType(Arg.getNullPtrType());
1800     break;
1801
1802   case TemplateArgument::Integral:
1803     Arg.getAsIntegral().Profile(ID);
1804     VisitType(Arg.getIntegralType());
1805     break;
1806
1807   case TemplateArgument::Expression:
1808     Visit(Arg.getAsExpr());
1809     break;
1810
1811   case TemplateArgument::Pack:
1812     for (const auto &P : Arg.pack_elements())
1813       VisitTemplateArgument(P);
1814     break;
1815   }
1816 }
1817
1818 void Stmt::Profile(llvm::FoldingSetNodeID &ID, const ASTContext &Context,
1819                    bool Canonical) const {
1820   StmtProfiler Profiler(ID, Context, Canonical);
1821   Profiler.Visit(this);
1822 }