StmtProfile.cpp revision 261991 
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/StmtVisitor.h"
22#include "llvm/ADT/FoldingSet.h"
23using namespace clang;
24
25namespace {
26  class StmtProfiler : public ConstStmtVisitor<StmtProfiler> {
27    llvm::FoldingSetNodeID &ID;
28    const ASTContext &Context;
29    bool Canonical;
30
31  public:
32    StmtProfiler(llvm::FoldingSetNodeID &ID, const ASTContext &Context,
33                 bool Canonical)
34      : ID(ID), Context(Context), Canonical(Canonical) { }
35
36    void VisitStmt(const Stmt *S);
37
38#define STMT(Node, Base) void Visit##Node(const Node *S);
39#include "clang/AST/StmtNodes.inc"
40
41    /// \brief Visit a declaration that is referenced within an expression
42    /// or statement.
43    void VisitDecl(const Decl *D);
44
45    /// \brief Visit a type that is referenced within an expression or
46    /// statement.
47    void VisitType(QualType T);
48
49    /// \brief Visit a name that occurs within an expression or statement.
50    void VisitName(DeclarationName Name);
51
52    /// \brief Visit a nested-name-specifier that occurs within an expression
53    /// or statement.
54    void VisitNestedNameSpecifier(NestedNameSpecifier *NNS);
55
56    /// \brief Visit a template name that occurs within an expression or
57    /// statement.
58    void VisitTemplateName(TemplateName Name);
59
60    /// \brief Visit template arguments that occur within an expression or
61    /// statement.
62    void VisitTemplateArguments(const TemplateArgumentLoc *Args,
63                                unsigned NumArgs);
64
65    /// \brief Visit a single template argument.
66    void VisitTemplateArgument(const TemplateArgument &Arg);
67  };
68}
69
70void StmtProfiler::VisitStmt(const Stmt *S) {
71  ID.AddInteger(S->getStmtClass());
72  for (Stmt::const_child_range C = S->children(); C; ++C) {
73    if (*C)
74      Visit(*C);
75    else
76      ID.AddInteger(0);
77  }
78}
79
80void StmtProfiler::VisitDeclStmt(const DeclStmt *S) {
81  VisitStmt(S);
82  for (DeclStmt::const_decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
83       D != DEnd; ++D)
84    VisitDecl(*D);
85}
86
87void StmtProfiler::VisitNullStmt(const NullStmt *S) {
88  VisitStmt(S);
89}
90
91void StmtProfiler::VisitCompoundStmt(const CompoundStmt *S) {
92  VisitStmt(S);
93}
94
95void StmtProfiler::VisitSwitchCase(const SwitchCase *S) {
96  VisitStmt(S);
97}
98
99void StmtProfiler::VisitCaseStmt(const CaseStmt *S) {
100  VisitStmt(S);
101}
102
103void StmtProfiler::VisitDefaultStmt(const DefaultStmt *S) {
104  VisitStmt(S);
105}
106
107void StmtProfiler::VisitLabelStmt(const LabelStmt *S) {
108  VisitStmt(S);
109  VisitDecl(S->getDecl());
110}
111
112void StmtProfiler::VisitAttributedStmt(const AttributedStmt *S) {
113  VisitStmt(S);
114  // TODO: maybe visit attributes?
115}
116
117void StmtProfiler::VisitIfStmt(const IfStmt *S) {
118  VisitStmt(S);
119  VisitDecl(S->getConditionVariable());
120}
121
122void StmtProfiler::VisitSwitchStmt(const SwitchStmt *S) {
123  VisitStmt(S);
124  VisitDecl(S->getConditionVariable());
125}
126
127void StmtProfiler::VisitWhileStmt(const WhileStmt *S) {
128  VisitStmt(S);
129  VisitDecl(S->getConditionVariable());
130}
131
132void StmtProfiler::VisitDoStmt(const DoStmt *S) {
133  VisitStmt(S);
134}
135
136void StmtProfiler::VisitForStmt(const ForStmt *S) {
137  VisitStmt(S);
138}
139
140void StmtProfiler::VisitGotoStmt(const GotoStmt *S) {
141  VisitStmt(S);
142  VisitDecl(S->getLabel());
143}
144
145void StmtProfiler::VisitIndirectGotoStmt(const IndirectGotoStmt *S) {
146  VisitStmt(S);
147}
148
149void StmtProfiler::VisitContinueStmt(const ContinueStmt *S) {
150  VisitStmt(S);
151}
152
153void StmtProfiler::VisitBreakStmt(const BreakStmt *S) {
154  VisitStmt(S);
155}
156
157void StmtProfiler::VisitReturnStmt(const ReturnStmt *S) {
158  VisitStmt(S);
159}
160
161void StmtProfiler::VisitGCCAsmStmt(const GCCAsmStmt *S) {
162  VisitStmt(S);
163  ID.AddBoolean(S->isVolatile());
164  ID.AddBoolean(S->isSimple());
165  VisitStringLiteral(S->getAsmString());
166  ID.AddInteger(S->getNumOutputs());
167  for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
168    ID.AddString(S->getOutputName(I));
169    VisitStringLiteral(S->getOutputConstraintLiteral(I));
170  }
171  ID.AddInteger(S->getNumInputs());
172  for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
173    ID.AddString(S->getInputName(I));
174    VisitStringLiteral(S->getInputConstraintLiteral(I));
175  }
176  ID.AddInteger(S->getNumClobbers());
177  for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
178    VisitStringLiteral(S->getClobberStringLiteral(I));
179}
180
181void StmtProfiler::VisitMSAsmStmt(const MSAsmStmt *S) {
182  // FIXME: Implement MS style inline asm statement profiler.
183  VisitStmt(S);
184}
185
186void StmtProfiler::VisitCXXCatchStmt(const CXXCatchStmt *S) {
187  VisitStmt(S);
188  VisitType(S->getCaughtType());
189}
190
191void StmtProfiler::VisitCXXTryStmt(const CXXTryStmt *S) {
192  VisitStmt(S);
193}
194
195void StmtProfiler::VisitCXXForRangeStmt(const CXXForRangeStmt *S) {
196  VisitStmt(S);
197}
198
199void StmtProfiler::VisitMSDependentExistsStmt(const MSDependentExistsStmt *S) {
200  VisitStmt(S);
201  ID.AddBoolean(S->isIfExists());
202  VisitNestedNameSpecifier(S->getQualifierLoc().getNestedNameSpecifier());
203  VisitName(S->getNameInfo().getName());
204}
205
206void StmtProfiler::VisitSEHTryStmt(const SEHTryStmt *S) {
207  VisitStmt(S);
208}
209
210void StmtProfiler::VisitSEHFinallyStmt(const SEHFinallyStmt *S) {
211  VisitStmt(S);
212}
213
214void StmtProfiler::VisitSEHExceptStmt(const SEHExceptStmt *S) {
215  VisitStmt(S);
216}
217
218void StmtProfiler::VisitCapturedStmt(const CapturedStmt *S) {
219  VisitStmt(S);
220}
221
222void StmtProfiler::VisitObjCForCollectionStmt(const ObjCForCollectionStmt *S) {
223  VisitStmt(S);
224}
225
226void StmtProfiler::VisitObjCAtCatchStmt(const ObjCAtCatchStmt *S) {
227  VisitStmt(S);
228  ID.AddBoolean(S->hasEllipsis());
229  if (S->getCatchParamDecl())
230    VisitType(S->getCatchParamDecl()->getType());
231}
232
233void StmtProfiler::VisitObjCAtFinallyStmt(const ObjCAtFinallyStmt *S) {
234  VisitStmt(S);
235}
236
237void StmtProfiler::VisitObjCAtTryStmt(const ObjCAtTryStmt *S) {
238  VisitStmt(S);
239}
240
241void
242StmtProfiler::VisitObjCAtSynchronizedStmt(const ObjCAtSynchronizedStmt *S) {
243  VisitStmt(S);
244}
245
246void StmtProfiler::VisitObjCAtThrowStmt(const ObjCAtThrowStmt *S) {
247  VisitStmt(S);
248}
249
250void
251StmtProfiler::VisitObjCAutoreleasePoolStmt(const ObjCAutoreleasePoolStmt *S) {
252  VisitStmt(S);
253}
254
255namespace {
256class OMPClauseProfiler : public ConstOMPClauseVisitor<OMPClauseProfiler> {
257  StmtProfiler *Profiler;
258  /// \brief Process clauses with list of variables.
259  template <typename T>
260  void VisitOMPClauseList(T *Node);
261public:
262  OMPClauseProfiler(StmtProfiler *P) : Profiler(P) { }
263#define OPENMP_CLAUSE(Name, Class)                                             \
264  void Visit##Class(const Class *C);
265#include "clang/Basic/OpenMPKinds.def"
266};
267
268void OMPClauseProfiler::VisitOMPDefaultClause(const OMPDefaultClause *C) { }
269
270template<typename T>
271void OMPClauseProfiler::VisitOMPClauseList(T *Node) {
272  for (typename T::varlist_const_iterator I = Node->varlist_begin(),
273                                          E = Node->varlist_end();
274         I != E; ++I)
275    Profiler->VisitStmt(*I);
276}
277
278void OMPClauseProfiler::VisitOMPPrivateClause(const OMPPrivateClause *C) {
279  VisitOMPClauseList(C);
280}
281void OMPClauseProfiler::VisitOMPFirstprivateClause(
282                                         const OMPFirstprivateClause *C) {
283  VisitOMPClauseList(C);
284}
285void OMPClauseProfiler::VisitOMPSharedClause(const OMPSharedClause *C) {
286  VisitOMPClauseList(C);
287}
288}
289
290void
291StmtProfiler::VisitOMPParallelDirective(const OMPParallelDirective *S) {
292  VisitStmt(S);
293  OMPClauseProfiler P(this);
294  ArrayRef<OMPClause *> Clauses = S->clauses();
295  for (ArrayRef<OMPClause *>::iterator I = Clauses.begin(), E = Clauses.end();
296       I != E; ++I)
297    if (*I)
298      P.Visit(*I);
299}
300
301void StmtProfiler::VisitExpr(const Expr *S) {
302  VisitStmt(S);
303}
304
305void StmtProfiler::VisitDeclRefExpr(const DeclRefExpr *S) {
306  VisitExpr(S);
307  if (!Canonical)
308    VisitNestedNameSpecifier(S->getQualifier());
309  VisitDecl(S->getDecl());
310  if (!Canonical)
311    VisitTemplateArguments(S->getTemplateArgs(), S->getNumTemplateArgs());
312}
313
314void StmtProfiler::VisitPredefinedExpr(const PredefinedExpr *S) {
315  VisitExpr(S);
316  ID.AddInteger(S->getIdentType());
317}
318
319void StmtProfiler::VisitIntegerLiteral(const IntegerLiteral *S) {
320  VisitExpr(S);
321  S->getValue().Profile(ID);
322}
323
324void StmtProfiler::VisitCharacterLiteral(const CharacterLiteral *S) {
325  VisitExpr(S);
326  ID.AddInteger(S->getKind());
327  ID.AddInteger(S->getValue());
328}
329
330void StmtProfiler::VisitFloatingLiteral(const FloatingLiteral *S) {
331  VisitExpr(S);
332  S->getValue().Profile(ID);
333  ID.AddBoolean(S->isExact());
334}
335
336void StmtProfiler::VisitImaginaryLiteral(const ImaginaryLiteral *S) {
337  VisitExpr(S);
338}
339
340void StmtProfiler::VisitStringLiteral(const StringLiteral *S) {
341  VisitExpr(S);
342  ID.AddString(S->getBytes());
343  ID.AddInteger(S->getKind());
344}
345
346void StmtProfiler::VisitParenExpr(const ParenExpr *S) {
347  VisitExpr(S);
348}
349
350void StmtProfiler::VisitParenListExpr(const ParenListExpr *S) {
351  VisitExpr(S);
352}
353
354void StmtProfiler::VisitUnaryOperator(const UnaryOperator *S) {
355  VisitExpr(S);
356  ID.AddInteger(S->getOpcode());
357}
358
359void StmtProfiler::VisitOffsetOfExpr(const OffsetOfExpr *S) {
360  VisitType(S->getTypeSourceInfo()->getType());
361  unsigned n = S->getNumComponents();
362  for (unsigned i = 0; i < n; ++i) {
363    const OffsetOfExpr::OffsetOfNode& ON = S->getComponent(i);
364    ID.AddInteger(ON.getKind());
365    switch (ON.getKind()) {
366    case OffsetOfExpr::OffsetOfNode::Array:
367      // Expressions handled below.
368      break;
369
370    case OffsetOfExpr::OffsetOfNode::Field:
371      VisitDecl(ON.getField());
372      break;
373
374    case OffsetOfExpr::OffsetOfNode::Identifier:
375      ID.AddPointer(ON.getFieldName());
376      break;
377
378    case OffsetOfExpr::OffsetOfNode::Base:
379      // These nodes are implicit, and therefore don't need profiling.
380      break;
381    }
382  }
383
384  VisitExpr(S);
385}
386
387void
388StmtProfiler::VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr *S) {
389  VisitExpr(S);
390  ID.AddInteger(S->getKind());
391  if (S->isArgumentType())
392    VisitType(S->getArgumentType());
393}
394
395void StmtProfiler::VisitArraySubscriptExpr(const ArraySubscriptExpr *S) {
396  VisitExpr(S);
397}
398
399void StmtProfiler::VisitCallExpr(const CallExpr *S) {
400  VisitExpr(S);
401}
402
403void StmtProfiler::VisitMemberExpr(const MemberExpr *S) {
404  VisitExpr(S);
405  VisitDecl(S->getMemberDecl());
406  if (!Canonical)
407    VisitNestedNameSpecifier(S->getQualifier());
408  ID.AddBoolean(S->isArrow());
409}
410
411void StmtProfiler::VisitCompoundLiteralExpr(const CompoundLiteralExpr *S) {
412  VisitExpr(S);
413  ID.AddBoolean(S->isFileScope());
414}
415
416void StmtProfiler::VisitCastExpr(const CastExpr *S) {
417  VisitExpr(S);
418}
419
420void StmtProfiler::VisitImplicitCastExpr(const ImplicitCastExpr *S) {
421  VisitCastExpr(S);
422  ID.AddInteger(S->getValueKind());
423}
424
425void StmtProfiler::VisitExplicitCastExpr(const ExplicitCastExpr *S) {
426  VisitCastExpr(S);
427  VisitType(S->getTypeAsWritten());
428}
429
430void StmtProfiler::VisitCStyleCastExpr(const CStyleCastExpr *S) {
431  VisitExplicitCastExpr(S);
432}
433
434void StmtProfiler::VisitBinaryOperator(const BinaryOperator *S) {
435  VisitExpr(S);
436  ID.AddInteger(S->getOpcode());
437}
438
439void
440StmtProfiler::VisitCompoundAssignOperator(const CompoundAssignOperator *S) {
441  VisitBinaryOperator(S);
442}
443
444void StmtProfiler::VisitConditionalOperator(const ConditionalOperator *S) {
445  VisitExpr(S);
446}
447
448void StmtProfiler::VisitBinaryConditionalOperator(
449    const BinaryConditionalOperator *S) {
450  VisitExpr(S);
451}
452
453void StmtProfiler::VisitAddrLabelExpr(const AddrLabelExpr *S) {
454  VisitExpr(S);
455  VisitDecl(S->getLabel());
456}
457
458void StmtProfiler::VisitStmtExpr(const StmtExpr *S) {
459  VisitExpr(S);
460}
461
462void StmtProfiler::VisitShuffleVectorExpr(const ShuffleVectorExpr *S) {
463  VisitExpr(S);
464}
465
466void StmtProfiler::VisitConvertVectorExpr(const ConvertVectorExpr *S) {
467  VisitExpr(S);
468}
469
470void StmtProfiler::VisitChooseExpr(const ChooseExpr *S) {
471  VisitExpr(S);
472}
473
474void StmtProfiler::VisitGNUNullExpr(const GNUNullExpr *S) {
475  VisitExpr(S);
476}
477
478void StmtProfiler::VisitVAArgExpr(const VAArgExpr *S) {
479  VisitExpr(S);
480}
481
482void StmtProfiler::VisitInitListExpr(const InitListExpr *S) {
483  if (S->getSyntacticForm()) {
484    VisitInitListExpr(S->getSyntacticForm());
485    return;
486  }
487
488  VisitExpr(S);
489}
490
491void StmtProfiler::VisitDesignatedInitExpr(const DesignatedInitExpr *S) {
492  VisitExpr(S);
493  ID.AddBoolean(S->usesGNUSyntax());
494  for (DesignatedInitExpr::const_designators_iterator D =
495         S->designators_begin(), DEnd = S->designators_end();
496       D != DEnd; ++D) {
497    if (D->isFieldDesignator()) {
498      ID.AddInteger(0);
499      VisitName(D->getFieldName());
500      continue;
501    }
502
503    if (D->isArrayDesignator()) {
504      ID.AddInteger(1);
505    } else {
506      assert(D->isArrayRangeDesignator());
507      ID.AddInteger(2);
508    }
509    ID.AddInteger(D->getFirstExprIndex());
510  }
511}
512
513void StmtProfiler::VisitImplicitValueInitExpr(const ImplicitValueInitExpr *S) {
514  VisitExpr(S);
515}
516
517void StmtProfiler::VisitExtVectorElementExpr(const ExtVectorElementExpr *S) {
518  VisitExpr(S);
519  VisitName(&S->getAccessor());
520}
521
522void StmtProfiler::VisitBlockExpr(const BlockExpr *S) {
523  VisitExpr(S);
524  VisitDecl(S->getBlockDecl());
525}
526
527void StmtProfiler::VisitGenericSelectionExpr(const GenericSelectionExpr *S) {
528  VisitExpr(S);
529  for (unsigned i = 0; i != S->getNumAssocs(); ++i) {
530    QualType T = S->getAssocType(i);
531    if (T.isNull())
532      ID.AddPointer(0);
533    else
534      VisitType(T);
535    VisitExpr(S->getAssocExpr(i));
536  }
537}
538
539void StmtProfiler::VisitPseudoObjectExpr(const PseudoObjectExpr *S) {
540  VisitExpr(S);
541  for (PseudoObjectExpr::const_semantics_iterator
542         i = S->semantics_begin(), e = S->semantics_end(); i != e; ++i)
543    // Normally, we would not profile the source expressions of OVEs.
544    if (const OpaqueValueExpr *OVE = dyn_cast<OpaqueValueExpr>(*i))
545      Visit(OVE->getSourceExpr());
546}
547
548void StmtProfiler::VisitAtomicExpr(const AtomicExpr *S) {
549  VisitExpr(S);
550  ID.AddInteger(S->getOp());
551}
552
553static Stmt::StmtClass DecodeOperatorCall(const CXXOperatorCallExpr *S,
554                                          UnaryOperatorKind &UnaryOp,
555                                          BinaryOperatorKind &BinaryOp) {
556  switch (S->getOperator()) {
557  case OO_None:
558  case OO_New:
559  case OO_Delete:
560  case OO_Array_New:
561  case OO_Array_Delete:
562  case OO_Arrow:
563  case OO_Call:
564  case OO_Conditional:
565  case NUM_OVERLOADED_OPERATORS:
566    llvm_unreachable("Invalid operator call kind");
567
568  case OO_Plus:
569    if (S->getNumArgs() == 1) {
570      UnaryOp = UO_Plus;
571      return Stmt::UnaryOperatorClass;
572    }
573
574    BinaryOp = BO_Add;
575    return Stmt::BinaryOperatorClass;
576
577  case OO_Minus:
578    if (S->getNumArgs() == 1) {
579      UnaryOp = UO_Minus;
580      return Stmt::UnaryOperatorClass;
581    }
582
583    BinaryOp = BO_Sub;
584    return Stmt::BinaryOperatorClass;
585
586  case OO_Star:
587    if (S->getNumArgs() == 1) {
588      UnaryOp = UO_Minus;
589      return Stmt::UnaryOperatorClass;
590    }
591
592    BinaryOp = BO_Sub;
593    return Stmt::BinaryOperatorClass;
594
595  case OO_Slash:
596    BinaryOp = BO_Div;
597    return Stmt::BinaryOperatorClass;
598
599  case OO_Percent:
600    BinaryOp = BO_Rem;
601    return Stmt::BinaryOperatorClass;
602
603  case OO_Caret:
604    BinaryOp = BO_Xor;
605    return Stmt::BinaryOperatorClass;
606
607  case OO_Amp:
608    if (S->getNumArgs() == 1) {
609      UnaryOp = UO_AddrOf;
610      return Stmt::UnaryOperatorClass;
611    }
612
613    BinaryOp = BO_And;
614    return Stmt::BinaryOperatorClass;
615
616  case OO_Pipe:
617    BinaryOp = BO_Or;
618    return Stmt::BinaryOperatorClass;
619
620  case OO_Tilde:
621    UnaryOp = UO_Not;
622    return Stmt::UnaryOperatorClass;
623
624  case OO_Exclaim:
625    UnaryOp = UO_LNot;
626    return Stmt::UnaryOperatorClass;
627
628  case OO_Equal:
629    BinaryOp = BO_Assign;
630    return Stmt::BinaryOperatorClass;
631
632  case OO_Less:
633    BinaryOp = BO_LT;
634    return Stmt::BinaryOperatorClass;
635
636  case OO_Greater:
637    BinaryOp = BO_GT;
638    return Stmt::BinaryOperatorClass;
639
640  case OO_PlusEqual:
641    BinaryOp = BO_AddAssign;
642    return Stmt::CompoundAssignOperatorClass;
643
644  case OO_MinusEqual:
645    BinaryOp = BO_SubAssign;
646    return Stmt::CompoundAssignOperatorClass;
647
648  case OO_StarEqual:
649    BinaryOp = BO_MulAssign;
650    return Stmt::CompoundAssignOperatorClass;
651
652  case OO_SlashEqual:
653    BinaryOp = BO_DivAssign;
654    return Stmt::CompoundAssignOperatorClass;
655
656  case OO_PercentEqual:
657    BinaryOp = BO_RemAssign;
658    return Stmt::CompoundAssignOperatorClass;
659
660  case OO_CaretEqual:
661    BinaryOp = BO_XorAssign;
662    return Stmt::CompoundAssignOperatorClass;
663
664  case OO_AmpEqual:
665    BinaryOp = BO_AndAssign;
666    return Stmt::CompoundAssignOperatorClass;
667
668  case OO_PipeEqual:
669    BinaryOp = BO_OrAssign;
670    return Stmt::CompoundAssignOperatorClass;
671
672  case OO_LessLess:
673    BinaryOp = BO_Shl;
674    return Stmt::BinaryOperatorClass;
675
676  case OO_GreaterGreater:
677    BinaryOp = BO_Shr;
678    return Stmt::BinaryOperatorClass;
679
680  case OO_LessLessEqual:
681    BinaryOp = BO_ShlAssign;
682    return Stmt::CompoundAssignOperatorClass;
683
684  case OO_GreaterGreaterEqual:
685    BinaryOp = BO_ShrAssign;
686    return Stmt::CompoundAssignOperatorClass;
687
688  case OO_EqualEqual:
689    BinaryOp = BO_EQ;
690    return Stmt::BinaryOperatorClass;
691
692  case OO_ExclaimEqual:
693    BinaryOp = BO_NE;
694    return Stmt::BinaryOperatorClass;
695
696  case OO_LessEqual:
697    BinaryOp = BO_LE;
698    return Stmt::BinaryOperatorClass;
699
700  case OO_GreaterEqual:
701    BinaryOp = BO_GE;
702    return Stmt::BinaryOperatorClass;
703
704  case OO_AmpAmp:
705    BinaryOp = BO_LAnd;
706    return Stmt::BinaryOperatorClass;
707
708  case OO_PipePipe:
709    BinaryOp = BO_LOr;
710    return Stmt::BinaryOperatorClass;
711
712  case OO_PlusPlus:
713    UnaryOp = S->getNumArgs() == 1? UO_PreInc
714                                  : UO_PostInc;
715    return Stmt::UnaryOperatorClass;
716
717  case OO_MinusMinus:
718    UnaryOp = S->getNumArgs() == 1? UO_PreDec
719                                  : UO_PostDec;
720    return Stmt::UnaryOperatorClass;
721
722  case OO_Comma:
723    BinaryOp = BO_Comma;
724    return Stmt::BinaryOperatorClass;
725
726
727  case OO_ArrowStar:
728    BinaryOp = BO_PtrMemI;
729    return Stmt::BinaryOperatorClass;
730
731  case OO_Subscript:
732    return Stmt::ArraySubscriptExprClass;
733  }
734
735  llvm_unreachable("Invalid overloaded operator expression");
736}
737
738
739void StmtProfiler::VisitCXXOperatorCallExpr(const CXXOperatorCallExpr *S) {
740  if (S->isTypeDependent()) {
741    // Type-dependent operator calls are profiled like their underlying
742    // syntactic operator.
743    UnaryOperatorKind UnaryOp = UO_Extension;
744    BinaryOperatorKind BinaryOp = BO_Comma;
745    Stmt::StmtClass SC = DecodeOperatorCall(S, UnaryOp, BinaryOp);
746
747    ID.AddInteger(SC);
748    for (unsigned I = 0, N = S->getNumArgs(); I != N; ++I)
749      Visit(S->getArg(I));
750    if (SC == Stmt::UnaryOperatorClass)
751      ID.AddInteger(UnaryOp);
752    else if (SC == Stmt::BinaryOperatorClass ||
753             SC == Stmt::CompoundAssignOperatorClass)
754      ID.AddInteger(BinaryOp);
755    else
756      assert(SC == Stmt::ArraySubscriptExprClass);
757
758    return;
759  }
760
761  VisitCallExpr(S);
762  ID.AddInteger(S->getOperator());
763}
764
765void StmtProfiler::VisitCXXMemberCallExpr(const CXXMemberCallExpr *S) {
766  VisitCallExpr(S);
767}
768
769void StmtProfiler::VisitCUDAKernelCallExpr(const CUDAKernelCallExpr *S) {
770  VisitCallExpr(S);
771}
772
773void StmtProfiler::VisitAsTypeExpr(const AsTypeExpr *S) {
774  VisitExpr(S);
775}
776
777void StmtProfiler::VisitCXXNamedCastExpr(const CXXNamedCastExpr *S) {
778  VisitExplicitCastExpr(S);
779}
780
781void StmtProfiler::VisitCXXStaticCastExpr(const CXXStaticCastExpr *S) {
782  VisitCXXNamedCastExpr(S);
783}
784
785void StmtProfiler::VisitCXXDynamicCastExpr(const CXXDynamicCastExpr *S) {
786  VisitCXXNamedCastExpr(S);
787}
788
789void
790StmtProfiler::VisitCXXReinterpretCastExpr(const CXXReinterpretCastExpr *S) {
791  VisitCXXNamedCastExpr(S);
792}
793
794void StmtProfiler::VisitCXXConstCastExpr(const CXXConstCastExpr *S) {
795  VisitCXXNamedCastExpr(S);
796}
797
798void StmtProfiler::VisitUserDefinedLiteral(const UserDefinedLiteral *S) {
799  VisitCallExpr(S);
800}
801
802void StmtProfiler::VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *S) {
803  VisitExpr(S);
804  ID.AddBoolean(S->getValue());
805}
806
807void StmtProfiler::VisitCXXNullPtrLiteralExpr(const CXXNullPtrLiteralExpr *S) {
808  VisitExpr(S);
809}
810
811void StmtProfiler::VisitCXXStdInitializerListExpr(
812    const CXXStdInitializerListExpr *S) {
813  VisitExpr(S);
814}
815
816void StmtProfiler::VisitCXXTypeidExpr(const CXXTypeidExpr *S) {
817  VisitExpr(S);
818  if (S->isTypeOperand())
819    VisitType(S->getTypeOperandSourceInfo()->getType());
820}
821
822void StmtProfiler::VisitCXXUuidofExpr(const CXXUuidofExpr *S) {
823  VisitExpr(S);
824  if (S->isTypeOperand())
825    VisitType(S->getTypeOperandSourceInfo()->getType());
826}
827
828void StmtProfiler::VisitMSPropertyRefExpr(const MSPropertyRefExpr *S) {
829  VisitExpr(S);
830  VisitDecl(S->getPropertyDecl());
831}
832
833void StmtProfiler::VisitCXXThisExpr(const CXXThisExpr *S) {
834  VisitExpr(S);
835  ID.AddBoolean(S->isImplicit());
836}
837
838void StmtProfiler::VisitCXXThrowExpr(const CXXThrowExpr *S) {
839  VisitExpr(S);
840}
841
842void StmtProfiler::VisitCXXDefaultArgExpr(const CXXDefaultArgExpr *S) {
843  VisitExpr(S);
844  VisitDecl(S->getParam());
845}
846
847void StmtProfiler::VisitCXXDefaultInitExpr(const CXXDefaultInitExpr *S) {
848  VisitExpr(S);
849  VisitDecl(S->getField());
850}
851
852void StmtProfiler::VisitCXXBindTemporaryExpr(const CXXBindTemporaryExpr *S) {
853  VisitExpr(S);
854  VisitDecl(
855         const_cast<CXXDestructorDecl *>(S->getTemporary()->getDestructor()));
856}
857
858void StmtProfiler::VisitCXXConstructExpr(const CXXConstructExpr *S) {
859  VisitExpr(S);
860  VisitDecl(S->getConstructor());
861  ID.AddBoolean(S->isElidable());
862}
863
864void StmtProfiler::VisitCXXFunctionalCastExpr(const CXXFunctionalCastExpr *S) {
865  VisitExplicitCastExpr(S);
866}
867
868void
869StmtProfiler::VisitCXXTemporaryObjectExpr(const CXXTemporaryObjectExpr *S) {
870  VisitCXXConstructExpr(S);
871}
872
873void
874StmtProfiler::VisitLambdaExpr(const LambdaExpr *S) {
875  VisitExpr(S);
876  for (LambdaExpr::capture_iterator C = S->explicit_capture_begin(),
877                                 CEnd = S->explicit_capture_end();
878       C != CEnd; ++C) {
879    ID.AddInteger(C->getCaptureKind());
880    switch (C->getCaptureKind()) {
881    case LCK_This:
882      break;
883    case LCK_ByRef:
884    case LCK_ByCopy:
885      VisitDecl(C->getCapturedVar());
886      ID.AddBoolean(C->isPackExpansion());
887      break;
888    }
889  }
890  // Note: If we actually needed to be able to match lambda
891  // expressions, we would have to consider parameters and return type
892  // here, among other things.
893  VisitStmt(S->getBody());
894}
895
896void
897StmtProfiler::VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *S) {
898  VisitExpr(S);
899}
900
901void StmtProfiler::VisitCXXDeleteExpr(const CXXDeleteExpr *S) {
902  VisitExpr(S);
903  ID.AddBoolean(S->isGlobalDelete());
904  ID.AddBoolean(S->isArrayForm());
905  VisitDecl(S->getOperatorDelete());
906}
907
908
909void StmtProfiler::VisitCXXNewExpr(const CXXNewExpr *S) {
910  VisitExpr(S);
911  VisitType(S->getAllocatedType());
912  VisitDecl(S->getOperatorNew());
913  VisitDecl(S->getOperatorDelete());
914  ID.AddBoolean(S->isArray());
915  ID.AddInteger(S->getNumPlacementArgs());
916  ID.AddBoolean(S->isGlobalNew());
917  ID.AddBoolean(S->isParenTypeId());
918  ID.AddInteger(S->getInitializationStyle());
919}
920
921void
922StmtProfiler::VisitCXXPseudoDestructorExpr(const CXXPseudoDestructorExpr *S) {
923  VisitExpr(S);
924  ID.AddBoolean(S->isArrow());
925  VisitNestedNameSpecifier(S->getQualifier());
926  ID.AddBoolean(S->getScopeTypeInfo() != 0);
927  if (S->getScopeTypeInfo())
928    VisitType(S->getScopeTypeInfo()->getType());
929  ID.AddBoolean(S->getDestroyedTypeInfo() != 0);
930  if (S->getDestroyedTypeInfo())
931    VisitType(S->getDestroyedType());
932  else
933    ID.AddPointer(S->getDestroyedTypeIdentifier());
934}
935
936void StmtProfiler::VisitOverloadExpr(const OverloadExpr *S) {
937  VisitExpr(S);
938  VisitNestedNameSpecifier(S->getQualifier());
939  VisitName(S->getName());
940  ID.AddBoolean(S->hasExplicitTemplateArgs());
941  if (S->hasExplicitTemplateArgs())
942    VisitTemplateArguments(S->getExplicitTemplateArgs().getTemplateArgs(),
943                           S->getExplicitTemplateArgs().NumTemplateArgs);
944}
945
946void
947StmtProfiler::VisitUnresolvedLookupExpr(const UnresolvedLookupExpr *S) {
948  VisitOverloadExpr(S);
949}
950
951void StmtProfiler::VisitUnaryTypeTraitExpr(const UnaryTypeTraitExpr *S) {
952  VisitExpr(S);
953  ID.AddInteger(S->getTrait());
954  VisitType(S->getQueriedType());
955}
956
957void StmtProfiler::VisitBinaryTypeTraitExpr(const BinaryTypeTraitExpr *S) {
958  VisitExpr(S);
959  ID.AddInteger(S->getTrait());
960  VisitType(S->getLhsType());
961  VisitType(S->getRhsType());
962}
963
964void StmtProfiler::VisitTypeTraitExpr(const TypeTraitExpr *S) {
965  VisitExpr(S);
966  ID.AddInteger(S->getTrait());
967  ID.AddInteger(S->getNumArgs());
968  for (unsigned I = 0, N = S->getNumArgs(); I != N; ++I)
969    VisitType(S->getArg(I)->getType());
970}
971
972void StmtProfiler::VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *S) {
973  VisitExpr(S);
974  ID.AddInteger(S->getTrait());
975  VisitType(S->getQueriedType());
976}
977
978void StmtProfiler::VisitExpressionTraitExpr(const ExpressionTraitExpr *S) {
979  VisitExpr(S);
980  ID.AddInteger(S->getTrait());
981  VisitExpr(S->getQueriedExpression());
982}
983
984void StmtProfiler::VisitDependentScopeDeclRefExpr(
985    const DependentScopeDeclRefExpr *S) {
986  VisitExpr(S);
987  VisitName(S->getDeclName());
988  VisitNestedNameSpecifier(S->getQualifier());
989  ID.AddBoolean(S->hasExplicitTemplateArgs());
990  if (S->hasExplicitTemplateArgs())
991    VisitTemplateArguments(S->getTemplateArgs(), S->getNumTemplateArgs());
992}
993
994void StmtProfiler::VisitExprWithCleanups(const ExprWithCleanups *S) {
995  VisitExpr(S);
996}
997
998void StmtProfiler::VisitCXXUnresolvedConstructExpr(
999    const CXXUnresolvedConstructExpr *S) {
1000  VisitExpr(S);
1001  VisitType(S->getTypeAsWritten());
1002}
1003
1004void StmtProfiler::VisitCXXDependentScopeMemberExpr(
1005    const CXXDependentScopeMemberExpr *S) {
1006  ID.AddBoolean(S->isImplicitAccess());
1007  if (!S->isImplicitAccess()) {
1008    VisitExpr(S);
1009    ID.AddBoolean(S->isArrow());
1010  }
1011  VisitNestedNameSpecifier(S->getQualifier());
1012  VisitName(S->getMember());
1013  ID.AddBoolean(S->hasExplicitTemplateArgs());
1014  if (S->hasExplicitTemplateArgs())
1015    VisitTemplateArguments(S->getTemplateArgs(), S->getNumTemplateArgs());
1016}
1017
1018void StmtProfiler::VisitUnresolvedMemberExpr(const UnresolvedMemberExpr *S) {
1019  ID.AddBoolean(S->isImplicitAccess());
1020  if (!S->isImplicitAccess()) {
1021    VisitExpr(S);
1022    ID.AddBoolean(S->isArrow());
1023  }
1024  VisitNestedNameSpecifier(S->getQualifier());
1025  VisitName(S->getMemberName());
1026  ID.AddBoolean(S->hasExplicitTemplateArgs());
1027  if (S->hasExplicitTemplateArgs())
1028    VisitTemplateArguments(S->getTemplateArgs(), S->getNumTemplateArgs());
1029}
1030
1031void StmtProfiler::VisitCXXNoexceptExpr(const CXXNoexceptExpr *S) {
1032  VisitExpr(S);
1033}
1034
1035void StmtProfiler::VisitPackExpansionExpr(const PackExpansionExpr *S) {
1036  VisitExpr(S);
1037}
1038
1039void StmtProfiler::VisitSizeOfPackExpr(const SizeOfPackExpr *S) {
1040  VisitExpr(S);
1041  VisitDecl(S->getPack());
1042}
1043
1044void StmtProfiler::VisitSubstNonTypeTemplateParmPackExpr(
1045    const SubstNonTypeTemplateParmPackExpr *S) {
1046  VisitExpr(S);
1047  VisitDecl(S->getParameterPack());
1048  VisitTemplateArgument(S->getArgumentPack());
1049}
1050
1051void StmtProfiler::VisitSubstNonTypeTemplateParmExpr(
1052    const SubstNonTypeTemplateParmExpr *E) {
1053  // Profile exactly as the replacement expression.
1054  Visit(E->getReplacement());
1055}
1056
1057void StmtProfiler::VisitFunctionParmPackExpr(const FunctionParmPackExpr *S) {
1058  VisitExpr(S);
1059  VisitDecl(S->getParameterPack());
1060  ID.AddInteger(S->getNumExpansions());
1061  for (FunctionParmPackExpr::iterator I = S->begin(), E = S->end(); I != E; ++I)
1062    VisitDecl(*I);
1063}
1064
1065void StmtProfiler::VisitMaterializeTemporaryExpr(
1066                                           const MaterializeTemporaryExpr *S) {
1067  VisitExpr(S);
1068}
1069
1070void StmtProfiler::VisitOpaqueValueExpr(const OpaqueValueExpr *E) {
1071  VisitExpr(E);
1072}
1073
1074void StmtProfiler::VisitObjCStringLiteral(const ObjCStringLiteral *S) {
1075  VisitExpr(S);
1076}
1077
1078void StmtProfiler::VisitObjCBoxedExpr(const ObjCBoxedExpr *E) {
1079  VisitExpr(E);
1080}
1081
1082void StmtProfiler::VisitObjCArrayLiteral(const ObjCArrayLiteral *E) {
1083  VisitExpr(E);
1084}
1085
1086void StmtProfiler::VisitObjCDictionaryLiteral(const ObjCDictionaryLiteral *E) {
1087  VisitExpr(E);
1088}
1089
1090void StmtProfiler::VisitObjCEncodeExpr(const ObjCEncodeExpr *S) {
1091  VisitExpr(S);
1092  VisitType(S->getEncodedType());
1093}
1094
1095void StmtProfiler::VisitObjCSelectorExpr(const ObjCSelectorExpr *S) {
1096  VisitExpr(S);
1097  VisitName(S->getSelector());
1098}
1099
1100void StmtProfiler::VisitObjCProtocolExpr(const ObjCProtocolExpr *S) {
1101  VisitExpr(S);
1102  VisitDecl(S->getProtocol());
1103}
1104
1105void StmtProfiler::VisitObjCIvarRefExpr(const ObjCIvarRefExpr *S) {
1106  VisitExpr(S);
1107  VisitDecl(S->getDecl());
1108  ID.AddBoolean(S->isArrow());
1109  ID.AddBoolean(S->isFreeIvar());
1110}
1111
1112void StmtProfiler::VisitObjCPropertyRefExpr(const ObjCPropertyRefExpr *S) {
1113  VisitExpr(S);
1114  if (S->isImplicitProperty()) {
1115    VisitDecl(S->getImplicitPropertyGetter());
1116    VisitDecl(S->getImplicitPropertySetter());
1117  } else {
1118    VisitDecl(S->getExplicitProperty());
1119  }
1120  if (S->isSuperReceiver()) {
1121    ID.AddBoolean(S->isSuperReceiver());
1122    VisitType(S->getSuperReceiverType());
1123  }
1124}
1125
1126void StmtProfiler::VisitObjCSubscriptRefExpr(const ObjCSubscriptRefExpr *S) {
1127  VisitExpr(S);
1128  VisitDecl(S->getAtIndexMethodDecl());
1129  VisitDecl(S->setAtIndexMethodDecl());
1130}
1131
1132void StmtProfiler::VisitObjCMessageExpr(const ObjCMessageExpr *S) {
1133  VisitExpr(S);
1134  VisitName(S->getSelector());
1135  VisitDecl(S->getMethodDecl());
1136}
1137
1138void StmtProfiler::VisitObjCIsaExpr(const ObjCIsaExpr *S) {
1139  VisitExpr(S);
1140  ID.AddBoolean(S->isArrow());
1141}
1142
1143void StmtProfiler::VisitObjCBoolLiteralExpr(const ObjCBoolLiteralExpr *S) {
1144  VisitExpr(S);
1145  ID.AddBoolean(S->getValue());
1146}
1147
1148void StmtProfiler::VisitObjCIndirectCopyRestoreExpr(
1149    const ObjCIndirectCopyRestoreExpr *S) {
1150  VisitExpr(S);
1151  ID.AddBoolean(S->shouldCopy());
1152}
1153
1154void StmtProfiler::VisitObjCBridgedCastExpr(const ObjCBridgedCastExpr *S) {
1155  VisitExplicitCastExpr(S);
1156  ID.AddBoolean(S->getBridgeKind());
1157}
1158
1159void StmtProfiler::VisitDecl(const Decl *D) {
1160  ID.AddInteger(D? D->getKind() : 0);
1161
1162  if (Canonical && D) {
1163    if (const NonTypeTemplateParmDecl *NTTP =
1164          dyn_cast<NonTypeTemplateParmDecl>(D)) {
1165      ID.AddInteger(NTTP->getDepth());
1166      ID.AddInteger(NTTP->getIndex());
1167      ID.AddBoolean(NTTP->isParameterPack());
1168      VisitType(NTTP->getType());
1169      return;
1170    }
1171
1172    if (const ParmVarDecl *Parm = dyn_cast<ParmVarDecl>(D)) {
1173      // The Itanium C++ ABI uses the type, scope depth, and scope
1174      // index of a parameter when mangling expressions that involve
1175      // function parameters, so we will use the parameter's type for
1176      // establishing function parameter identity. That way, our
1177      // definition of "equivalent" (per C++ [temp.over.link]) is at
1178      // least as strong as the definition of "equivalent" used for
1179      // name mangling.
1180      VisitType(Parm->getType());
1181      ID.AddInteger(Parm->getFunctionScopeDepth());
1182      ID.AddInteger(Parm->getFunctionScopeIndex());
1183      return;
1184    }
1185
1186    if (const TemplateTypeParmDecl *TTP =
1187          dyn_cast<TemplateTypeParmDecl>(D)) {
1188      ID.AddInteger(TTP->getDepth());
1189      ID.AddInteger(TTP->getIndex());
1190      ID.AddBoolean(TTP->isParameterPack());
1191      return;
1192    }
1193
1194    if (const TemplateTemplateParmDecl *TTP =
1195          dyn_cast<TemplateTemplateParmDecl>(D)) {
1196      ID.AddInteger(TTP->getDepth());
1197      ID.AddInteger(TTP->getIndex());
1198      ID.AddBoolean(TTP->isParameterPack());
1199      return;
1200    }
1201  }
1202
1203  ID.AddPointer(D? D->getCanonicalDecl() : 0);
1204}
1205
1206void StmtProfiler::VisitType(QualType T) {
1207  if (Canonical)
1208    T = Context.getCanonicalType(T);
1209
1210  ID.AddPointer(T.getAsOpaquePtr());
1211}
1212
1213void StmtProfiler::VisitName(DeclarationName Name) {
1214  ID.AddPointer(Name.getAsOpaquePtr());
1215}
1216
1217void StmtProfiler::VisitNestedNameSpecifier(NestedNameSpecifier *NNS) {
1218  if (Canonical)
1219    NNS = Context.getCanonicalNestedNameSpecifier(NNS);
1220  ID.AddPointer(NNS);
1221}
1222
1223void StmtProfiler::VisitTemplateName(TemplateName Name) {
1224  if (Canonical)
1225    Name = Context.getCanonicalTemplateName(Name);
1226
1227  Name.Profile(ID);
1228}
1229
1230void StmtProfiler::VisitTemplateArguments(const TemplateArgumentLoc *Args,
1231                                          unsigned NumArgs) {
1232  ID.AddInteger(NumArgs);
1233  for (unsigned I = 0; I != NumArgs; ++I)
1234    VisitTemplateArgument(Args[I].getArgument());
1235}
1236
1237void StmtProfiler::VisitTemplateArgument(const TemplateArgument &Arg) {
1238  // Mostly repetitive with TemplateArgument::Profile!
1239  ID.AddInteger(Arg.getKind());
1240  switch (Arg.getKind()) {
1241  case TemplateArgument::Null:
1242    break;
1243
1244  case TemplateArgument::Type:
1245    VisitType(Arg.getAsType());
1246    break;
1247
1248  case TemplateArgument::Template:
1249  case TemplateArgument::TemplateExpansion:
1250    VisitTemplateName(Arg.getAsTemplateOrTemplatePattern());
1251    break;
1252
1253  case TemplateArgument::Declaration:
1254    VisitDecl(Arg.getAsDecl());
1255    break;
1256
1257  case TemplateArgument::NullPtr:
1258    VisitType(Arg.getNullPtrType());
1259    break;
1260
1261  case TemplateArgument::Integral:
1262    Arg.getAsIntegral().Profile(ID);
1263    VisitType(Arg.getIntegralType());
1264    break;
1265
1266  case TemplateArgument::Expression:
1267    Visit(Arg.getAsExpr());
1268    break;
1269
1270  case TemplateArgument::Pack:
1271    const TemplateArgument *Pack = Arg.pack_begin();
1272    for (unsigned i = 0, e = Arg.pack_size(); i != e; ++i)
1273      VisitTemplateArgument(Pack[i]);
1274    break;
1275  }
1276}
1277
1278void Stmt::Profile(llvm::FoldingSetNodeID &ID, const ASTContext &Context,
1279                   bool Canonical) const {
1280  StmtProfiler Profiler(ID, Context, Canonical);
1281  Profiler.Visit(this);
1282}
1283