Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 1 | //===--- EvaluatedExprVisitor.h - Evaluated expression visitor --*- C++ -*-===// | 
| 2 | // | ||
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. | ||
| 4 | // See https://llvm.org/LICENSE.txt for license information. | ||
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
| 6 | // | ||
| 7 | //===----------------------------------------------------------------------===// | ||
| 8 | // | ||
| 9 | //  This file defines the EvaluatedExprVisitor class template, which visits | ||
| 10 | //  the potentially-evaluated subexpressions of a potentially-evaluated | ||
| 11 | //  expression. | ||
| 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 | /// Given a potentially-evaluated expression, this visitor visits all | ||
| 28 | /// of its potentially-evaluated subexpressions, recursively. | ||
| 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 |   // Return whether this visitor should recurse into discarded statements for a | ||
| 36 |   // 'constexpr-if'. | ||
| 37 | bool shouldVisitDiscardedStmt() const { return true; } | ||
| 38 | #define PTR(CLASS) typename Ptr<CLASS>::type | ||
| 39 | |||
| 40 | explicit EvaluatedExprVisitorBase(const ASTContext &Context) : Context(Context) { } | ||
| 41 | |||
| 42 |   // Expressions that have no potentially-evaluated subexpressions (but may have | ||
| 43 |   // other sub-expressions). | ||
| 44 | void VisitDeclRefExpr(PTR(DeclRefExpr) E) { } | ||
| 45 | void VisitOffsetOfExpr(PTR(OffsetOfExpr) E) { } | ||
| 46 | void VisitUnaryExprOrTypeTraitExpr(PTR(UnaryExprOrTypeTraitExpr) E) { } | ||
| 47 | void VisitExpressionTraitExpr(PTR(ExpressionTraitExpr) E) { } | ||
| 48 | void VisitBlockExpr(PTR(BlockExpr) E) { } | ||
| 49 | void VisitCXXUuidofExpr(PTR(CXXUuidofExpr) E) { } | ||
| 50 | void VisitCXXNoexceptExpr(PTR(CXXNoexceptExpr) E) { } | ||
| 51 | |||
| 52 | void VisitMemberExpr(PTR(MemberExpr) E) { | ||
| 53 |     // Only the base matters. | ||
| 54 | return this->Visit(E->getBase()); | ||
| 55 |   } | ||
| 56 | |||
| 57 | void VisitChooseExpr(PTR(ChooseExpr) E) { | ||
| 58 |     // Don't visit either child expression if the condition is dependent. | ||
| 59 | if (E->getCond()->isValueDependent()) | ||
| 60 | return; | ||
| 61 |     // Only the selected subexpression matters; the other one is not evaluated. | ||
| 62 | return this->Visit(E->getChosenSubExpr()); | ||
| 63 |   } | ||
| 64 | |||
| 65 | void VisitGenericSelectionExpr(PTR(GenericSelectionExpr) E) { | ||
| 66 |     // The controlling expression of a generic selection is not evaluated. | ||
| 67 | |||
| 68 |     // Don't visit either child expression if the condition is type-dependent. | ||
| 69 | if (E->isResultDependent()) | ||
| 70 | return; | ||
| 71 |     // Only the selected subexpression matters; the other subexpressions and the | ||
| 72 |     // controlling expression are not evaluated. | ||
| 73 | return this->Visit(E->getResultExpr()); | ||
| 74 |   } | ||
| 75 | |||
| 76 | void VisitDesignatedInitExpr(PTR(DesignatedInitExpr) E) { | ||
| 77 |     // Only the actual initializer matters; the designators are all constant | ||
| 78 |     // expressions. | ||
| 79 | return this->Visit(E->getInit()); | ||
| 80 |   } | ||
| 81 | |||
| 82 | void VisitCXXTypeidExpr(PTR(CXXTypeidExpr) E) { | ||
| 83 | if (E->isPotentiallyEvaluated()) | ||
| 84 | return this->Visit(E->getExprOperand()); | ||
| 85 |   } | ||
| 86 | |||
| 87 | void VisitCallExpr(PTR(CallExpr) CE) { | ||
| 88 | if (!CE->isUnevaluatedBuiltinCall(Context)) | ||
| 89 | return getDerived().VisitExpr(CE); | ||
| 90 |   } | ||
| 91 | |||
| 92 | void VisitLambdaExpr(PTR(LambdaExpr) LE) { | ||
| 93 |     // Only visit the capture initializers, and not the body. | ||
| 94 | for (LambdaExpr::const_capture_init_iterator I = LE->capture_init_begin(), | ||
| 95 | E = LE->capture_init_end(); | ||
| 96 | I != E; ++I) | ||
| 97 | if (*I) | ||
| 98 | this->Visit(*I); | ||
| 99 |   } | ||
| 100 | |||
| 101 |   /// The basis case walks all of the children of the statement or | ||
| 102 |   /// expression, assuming they are all potentially evaluated. | ||
| 103 | void VisitStmt(PTR(Stmt) S) { | ||
| 104 | for (auto *SubStmt : S->children()) | ||
| 105 | if (SubStmt) | ||
| 106 | this->Visit(SubStmt); | ||
| 107 |   } | ||
| 108 | |||
| 109 | void VisitIfStmt(PTR(IfStmt) If) { | ||
| 110 | if (!getDerived().shouldVisitDiscardedStmt()) { | ||
| 111 | if (auto SubStmt = If->getNondiscardedCase(Context)) { | ||
| 112 | if (*SubStmt) | ||
| 113 | this->Visit(*SubStmt); | ||
| 114 | return; | ||
| 115 |       } | ||
| 116 |     } | ||
| 117 | |||
| 118 | getDerived().VisitStmt(If); | ||
| 119 |   } | ||
| 120 | |||
| 121 | ImplClass &getDerived() { return *static_cast<ImplClass *>(this); } | ||
| 122 | |||
| 123 | #undef PTR | ||
| 124 | }; | ||
| 125 | |||
| 126 | /// EvaluatedExprVisitor - This class visits 'Expr *'s | ||
| 127 | template <typename ImplClass> | ||
| 128 | class EvaluatedExprVisitor | ||
| 129 | : public EvaluatedExprVisitorBase<std::add_pointer, ImplClass> { | ||
| 130 | public: | ||
| 131 | explicit EvaluatedExprVisitor(const ASTContext &Context) | ||
| 132 | : EvaluatedExprVisitorBase<std::add_pointer, ImplClass>(Context) {} | ||
| 133 | }; | ||
| 134 | |||
| 135 | /// ConstEvaluatedExprVisitor - This class visits 'const Expr *'s. | ||
| 136 | template <typename ImplClass> | ||
| 137 | class ConstEvaluatedExprVisitor | ||
| 138 | : public EvaluatedExprVisitorBase<llvm::make_const_ptr, ImplClass> { | ||
| 139 | public: | ||
| 140 | explicit ConstEvaluatedExprVisitor(const ASTContext &Context) | ||
| 141 | : EvaluatedExprVisitorBase<llvm::make_const_ptr, ImplClass>(Context) {} | ||
| 142 | }; | ||
| 143 | } | ||
| 144 | |||
| 145 | #endif // LLVM_CLANG_AST_EVALUATEDEXPRVISITOR_H |