Some minor improvements to Evaluate.

llvm-svn: 65613

1 files changed, 23 insertions, 3 deletions

diff --git a/clang/lib/AST/ExprConstant.cpp b/clang/lib/AST/ExprConstant.cpp
index 5cfc6e6ab31..4f55f8d7642 100644
--- a/clang/lib/AST/ExprConstant.cpp
+++ b/clang/lib/AST/ExprConstant.cpp
@@ -636,16 +636,24 @@ public:
     return Success(0, E);
   }
 
+  bool VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E) {
+    return Success(0, E);
+  }
+
   bool VisitUnaryTypeTraitExpr(const UnaryTypeTraitExpr *E) {
     return Success(E->EvaluateTrait(), E);
   }
 
+  bool VisitChooseExpr(const ChooseExpr *E);
+  bool VisitUnaryReal(const UnaryOperator *E) {
+    return Visit(E->getSubExpr());
+  }
+  bool VisitUnaryImag(const UnaryOperator *E);
+
 private:
   unsigned GetAlignOfExpr(const Expr *E);
   unsigned GetAlignOfType(QualType T);
-  // FIXME: Missing: __real__/__imag__, array subscript of vector,
-  //                 member of vector, __builtin_choose_expr,
-  //                 ImplicitValueInitExpr
+  // FIXME: Missing: array subscript of vector, member of vector
 };
 } // end anonymous namespace
 
@@ -1170,6 +1178,18 @@ bool IntExprEvaluator::VisitCastExpr(CastExpr *E) {
   return Success(HandleFloatToIntCast(DestType, SrcType, F, Info.Ctx), E);
 }
 
+bool IntExprEvaluator::VisitChooseExpr(const ChooseExpr *E) {
+  Expr* EvalExpr = E->isConditionTrue(Info.Ctx) ? E->getLHS() : E->getRHS();
+
+  return Visit(EvalExpr);
+}
+
+bool IntExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
+  if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
+    Info.EvalResult.HasSideEffects = true;
+  return Success(0, E);
+}
+
 //===----------------------------------------------------------------------===//
 // Float Evaluation
 //===----------------------------------------------------------------------===//