Minor refactoring to introduce a new transformation function,
TransformStmt, into the tree transformer.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@78347 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Sema/TreeTransform.h b/lib/Sema/TreeTransform.h
index 1f630b8..c935f81 100644
--- a/lib/Sema/TreeTransform.h
+++ b/lib/Sema/TreeTransform.h
@@ -15,6 +15,7 @@
#include "Sema.h"
#include "clang/Sema/SemaDiagnostic.h"
+#include "clang/AST/Expr.h"
#include <algorithm>
namespace clang {
@@ -149,9 +150,16 @@
/// other clients.
QualType AddTypeQualifiers(QualType T, unsigned CVRQualifiers);
- /// \brief Transform the given expression.
+ /// \brief Transform the given statement.
///
/// FIXME: At the moment, subclasses must override this.
+ Sema::OwningStmtResult TransformStmt(Stmt *S);
+
+ /// \brief Transform the given expression.
+ ///
+ /// By default, invokes the derived class's TransformStmt() and downcasts
+ /// the result. Subclasses may override this function to provide alternate
+ /// behavior.
Sema::OwningExprResult TransformExpr(Expr *E);
/// \brief Transform the given declaration, which is referenced from a type
@@ -468,6 +476,15 @@
};
template<typename Derived>
+Sema::OwningExprResult TreeTransform<Derived>::TransformExpr(Expr *E) {
+ Sema::OwningStmtResult Result = getDerived().TransformStmt(E);
+ if (Result.isInvalid())
+ return SemaRef.ExprError();
+
+ return SemaRef.Owned(cast_or_null<Stmt>(Result.takeAs<Stmt>()));
+}
+
+template<typename Derived>
NestedNameSpecifier *
TreeTransform<Derived>::TransformNestedNameSpecifier(NestedNameSpecifier *NNS,
SourceRange Range) {