1 | |
2 | |
3 | |
4 | |
5 | |
6 | |
7 | |
8 | |
9 | |
10 | |
11 | |
12 | |
13 | |
14 | #ifndef LLVM_CLANG_AST_EVALUATEDEXPRVISITOR_H |
15 | #define LLVM_CLANG_AST_EVALUATEDEXPRVISITOR_H |
16 | |
17 | #include "clang/AST/DeclCXX.h" |
18 | #include "clang/AST/Expr.h" |
19 | #include "clang/AST/ExprCXX.h" |
20 | #include "clang/AST/StmtVisitor.h" |
21 | #include "llvm/ADT/STLExtras.h" |
22 | |
23 | namespace clang { |
24 | |
25 | class ASTContext; |
26 | |
27 | |
28 | |
29 | template<template <typename> class Ptr, typename ImplClass> |
30 | class EvaluatedExprVisitorBase : public StmtVisitorBase<Ptr, ImplClass, void> { |
31 | protected: |
32 | const ASTContext &Context; |
33 | |
34 | public: |
35 | #define PTR(CLASS) typename Ptr<CLASS>::type |
36 | |
37 | explicit EvaluatedExprVisitorBase(const ASTContext &Context) : Context(Context) { } |
38 | |
39 | |
40 | |
41 | void VisitDeclRefExpr(PTR(DeclRefExpr) E) { } |
42 | void VisitOffsetOfExpr(PTR(OffsetOfExpr) E) { } |
43 | void VisitUnaryExprOrTypeTraitExpr(PTR(UnaryExprOrTypeTraitExpr) E) { } |
44 | void VisitExpressionTraitExpr(PTR(ExpressionTraitExpr) E) { } |
45 | void VisitBlockExpr(PTR(BlockExpr) E) { } |
46 | void VisitCXXUuidofExpr(PTR(CXXUuidofExpr) E) { } |
47 | void VisitCXXNoexceptExpr(PTR(CXXNoexceptExpr) E) { } |
48 | |
49 | void VisitMemberExpr(PTR(MemberExpr) E) { |
50 | |
51 | return this->Visit(E->getBase()); |
52 | } |
53 | |
54 | void VisitChooseExpr(PTR(ChooseExpr) E) { |
55 | |
56 | if (E->getCond()->isValueDependent()) |
57 | return; |
58 | |
59 | return this->Visit(E->getChosenSubExpr()); |
60 | } |
61 | |
62 | void VisitGenericSelectionExpr(PTR(GenericSelectionExpr) E) { |
63 | |
64 | |
65 | |
66 | if (E->isResultDependent()) |
67 | return; |
68 | |
69 | |
70 | return this->Visit(E->getResultExpr()); |
71 | } |
72 | |
73 | void VisitDesignatedInitExpr(PTR(DesignatedInitExpr) E) { |
74 | |
75 | |
76 | return this->Visit(E->getInit()); |
77 | } |
78 | |
79 | void VisitCXXTypeidExpr(PTR(CXXTypeidExpr) E) { |
80 | if (E->isPotentiallyEvaluated()) |
81 | return this->Visit(E->getExprOperand()); |
82 | } |
83 | |
84 | void VisitCallExpr(PTR(CallExpr) CE) { |
85 | if (!CE->isUnevaluatedBuiltinCall(Context)) |
86 | return static_cast<ImplClass*>(this)->VisitExpr(CE); |
87 | } |
88 | |
89 | void VisitLambdaExpr(PTR(LambdaExpr) LE) { |
90 | |
91 | for (LambdaExpr::const_capture_init_iterator I = LE->capture_init_begin(), |
92 | E = LE->capture_init_end(); |
93 | I != E; ++I) |
94 | if (*I) |
95 | this->Visit(*I); |
96 | } |
97 | |
98 | |
99 | |
100 | void VisitStmt(PTR(Stmt) S) { |
101 | for (auto *SubStmt : S->children()) |
102 | if (SubStmt) |
103 | this->Visit(SubStmt); |
104 | } |
105 | |
106 | #undef PTR |
107 | }; |
108 | |
109 | |
110 | template <typename ImplClass> |
111 | class EvaluatedExprVisitor |
112 | : public EvaluatedExprVisitorBase<std::add_pointer, ImplClass> { |
113 | public: |
114 | explicit EvaluatedExprVisitor(const ASTContext &Context) |
115 | : EvaluatedExprVisitorBase<std::add_pointer, ImplClass>(Context) {} |
116 | }; |
117 | |
118 | |
119 | template <typename ImplClass> |
120 | class ConstEvaluatedExprVisitor |
121 | : public EvaluatedExprVisitorBase<llvm::make_const_ptr, ImplClass> { |
122 | public: |
123 | explicit ConstEvaluatedExprVisitor(const ASTContext &Context) |
124 | : EvaluatedExprVisitorBase<llvm::make_const_ptr, ImplClass>(Context) {} |
125 | }; |
126 | } |
127 | |
128 | #endif |
129 | |