ExprConstant.cpp revision 30c37f4d2ee5811e85f692c22fb67d74ddc88079
1b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner//===--- ExprConstant.cpp - Expression Constant Evaluator -----------------===//
2c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
3c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//                     The LLVM Compiler Infrastructure
4c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
5c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson// This file is distributed under the University of Illinois Open Source
6c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson// License. See LICENSE.TXT for details.
7c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
8c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//===----------------------------------------------------------------------===//
9c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
10c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson// This file implements the Expr constant evaluator.
11c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
12c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//===----------------------------------------------------------------------===//
13c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson
14c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson#include "clang/AST/APValue.h"
15c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson#include "clang/AST/ASTContext.h"
164efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman#include "clang/AST/RecordLayout.h"
170fe52e1bcaa69ba127f1bda036f057fec1f478deSeo Sanghyeon#include "clang/AST/StmtVisitor.h"
18500d3297d2a21edeac4d46cbcbe21bc2352c2a28Chris Lattner#include "clang/AST/ASTDiagnostic.h"
1906a3675627e3b3c47b49c689c8e404a33144194aAnders Carlsson#include "clang/Basic/TargetInfo.h"
20c754aa62643e66ab967ca32ae8b0b3fc419bba25Anders Carlsson#include "llvm/Support/Compiler.h"
21c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlssonusing namespace clang;
22f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnerusing llvm::APSInt;
23d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanusing llvm::APFloat;
24c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson
2587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// EvalInfo - This is a private struct used by the evaluator to capture
2687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// information about a subexpression as it is folded.  It retains information
2787eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// about the AST context, but also maintains information about the folded
2887eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// expression.
2987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner///
3087eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// If an expression could be evaluated, it is still possible it is not a C
3187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// "integer constant expression" or constant expression.  If not, this struct
3287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// captures information about how and why not.
3387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner///
3487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// One bit of information passed *into* the request for constant folding
3587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// indicates whether the subexpression is "evaluated" or not according to C
3687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// rules.  For example, the RHS of (0 && foo()) is not evaluated.  We can
3787eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// evaluate the expression regardless of what the RHS is, but C only allows
3887eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// certain things in certain situations.
3987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstruct EvalInfo {
4087eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  ASTContext &Ctx;
4187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner
4254da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson  /// EvalResult - Contains information about the evaluation.
4354da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson  Expr::EvalResult &EvalResult;
44f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson
45f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson  /// ShortCircuit - will be greater than zero if the current subexpression has
46f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson  /// will not be evaluated because it's short-circuited (according to C rules).
47f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson  unsigned ShortCircuit;
4854da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson
4954da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson  EvalInfo(ASTContext &ctx, Expr::EvalResult& evalresult) : Ctx(ctx),
50f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson           EvalResult(evalresult), ShortCircuit(0) {}
5187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner};
5287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner
5387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner
544efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanstatic bool EvaluateLValue(const Expr *E, APValue &Result, EvalInfo &Info);
5587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstatic bool EvaluatePointer(const Expr *E, APValue &Result, EvalInfo &Info);
5687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstatic bool EvaluateInteger(const Expr *E, APSInt  &Result, EvalInfo &Info);
57d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanstatic bool EvaluateFloat(const Expr *E, APFloat &Result, EvalInfo &Info);
58a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbarstatic bool EvaluateComplex(const Expr *E, APValue &Result, EvalInfo &Info);
59f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
60f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
614efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman// Misc utilities
624efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
634efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
644efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanstatic bool HandleConversionToBool(Expr* E, bool& Result, EvalInfo &Info) {
654efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (E->getType()->isIntegralType()) {
664efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APSInt IntResult;
674efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluateInteger(E, IntResult, Info))
684efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
694efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Result = IntResult != 0;
704efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
714efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  } else if (E->getType()->isRealFloatingType()) {
724efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APFloat FloatResult(0.0);
734efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluateFloat(E, FloatResult, Info))
744efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
754efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Result = !FloatResult.isZero();
764efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
774efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  } else if (E->getType()->isPointerType()) {
784efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APValue PointerResult;
794efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluatePointer(E, PointerResult, Info))
804efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
814efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    // FIXME: Is this accurate for all kinds of bases?  If not, what would
824efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    // the check look like?
834efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Result = PointerResult.getLValueBase() || PointerResult.getLValueOffset();
844efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
854efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
864efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
874efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return false;
884efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
894efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
90a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbarstatic APSInt HandleFloatToIntCast(QualType DestType, QualType SrcType,
91a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                   APFloat &Value, ASTContext &Ctx) {
92a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  unsigned DestWidth = Ctx.getIntWidth(DestType);
93a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // Determine whether we are converting to unsigned or signed.
94a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool DestSigned = DestType->isSignedIntegerType();
95a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
96a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // FIXME: Warning for overflow.
97a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  uint64_t Space[4];
98a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool ignored;
99a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  (void)Value.convertToInteger(Space, DestWidth, DestSigned,
100a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                               llvm::APFloat::rmTowardZero, &ignored);
101a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return APSInt(llvm::APInt(DestWidth, 4, Space), !DestSigned);
102a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
103a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
104a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbarstatic APFloat HandleFloatToFloatCast(QualType DestType, QualType SrcType,
105a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                      APFloat &Value, ASTContext &Ctx) {
106a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool ignored;
107a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APFloat Result = Value;
108a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.convert(Ctx.getFloatTypeSemantics(DestType),
109a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                 APFloat::rmNearestTiesToEven, &ignored);
110a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
111a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
112a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
113a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbarstatic APSInt HandleIntToIntCast(QualType DestType, QualType SrcType,
114a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                 APSInt &Value, ASTContext &Ctx) {
115a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  unsigned DestWidth = Ctx.getIntWidth(DestType);
116a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APSInt Result = Value;
117a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // Figure out if this is a truncate, extend or noop cast.
118a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // If the input is signed, do a sign extend, noop, or truncate.
119a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.extOrTrunc(DestWidth);
120a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.setIsUnsigned(DestType->isUnsignedIntegerType());
121a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
122a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
123a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
124a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbarstatic APFloat HandleIntToFloatCast(QualType DestType, QualType SrcType,
125a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                    APSInt &Value, ASTContext &Ctx) {
126a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
127a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APFloat Result(Ctx.getFloatTypeSemantics(DestType), 1);
128a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.convertFromAPInt(Value, Value.isSigned(),
129a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                          APFloat::rmNearestTiesToEven);
130a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
131a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
132a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
1334efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
1344efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman// LValue Evaluation
1354efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
1364efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmannamespace {
1374efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanclass VISIBILITY_HIDDEN LValueExprEvaluator
1384efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  : public StmtVisitor<LValueExprEvaluator, APValue> {
1394efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  EvalInfo &Info;
1404efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanpublic:
1414efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1424efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  LValueExprEvaluator(EvalInfo &info) : Info(info) {}
1434efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1444efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue VisitStmt(Stmt *S) {
1458a7b7c6d4c96ace3667204bed4a48bad91b7c0f0Daniel Dunbar#if 0
1464efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    // FIXME: Remove this when we support more expressions.
1474efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    printf("Unhandled pointer statement\n");
1484efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    S->dump();
1498a7b7c6d4c96ace3667204bed4a48bad91b7c0f0Daniel Dunbar#endif
1504efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return APValue();
1514efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
1524efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1534efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
15435873c49adad211ff466e34342a52665742794f5Anders Carlsson  APValue VisitDeclRefExpr(DeclRefExpr *E);
1554efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue VisitPredefinedExpr(PredefinedExpr *E) { return APValue(E, 0); }
1564efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1574efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue VisitMemberExpr(MemberExpr *E);
1584efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue VisitStringLiteral(StringLiteral *E) { return APValue(E, 0); }
1593068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  APValue VisitArraySubscriptExpr(ArraySubscriptExpr *E);
1604efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman};
1614efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman} // end anonymous namespace
1624efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1634efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanstatic bool EvaluateLValue(const Expr* E, APValue& Result, EvalInfo &Info) {
1644efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Result = LValueExprEvaluator(Info).Visit(const_cast<Expr*>(E));
1654efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return Result.isLValue();
1664efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
1674efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
16835873c49adad211ff466e34342a52665742794f5Anders CarlssonAPValue LValueExprEvaluator::VisitDeclRefExpr(DeclRefExpr *E)
16935873c49adad211ff466e34342a52665742794f5Anders Carlsson{
17035873c49adad211ff466e34342a52665742794f5Anders Carlsson  if (!E->hasGlobalStorage())
17135873c49adad211ff466e34342a52665742794f5Anders Carlsson    return APValue();
17235873c49adad211ff466e34342a52665742794f5Anders Carlsson
17335873c49adad211ff466e34342a52665742794f5Anders Carlsson  return APValue(E, 0);
17435873c49adad211ff466e34342a52665742794f5Anders Carlsson}
17535873c49adad211ff466e34342a52665742794f5Anders Carlsson
1764efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli FriedmanAPValue LValueExprEvaluator::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
1774efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (E->isFileScope())
1784efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return APValue(E, 0);
1794efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return APValue();
1804efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
1814efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1824efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli FriedmanAPValue LValueExprEvaluator::VisitMemberExpr(MemberExpr *E) {
1834efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue result;
1844efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  QualType Ty;
1854efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (E->isArrow()) {
1864efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluatePointer(E->getBase(), result, Info))
1874efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return APValue();
1884efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Ty = E->getBase()->getType()->getAsPointerType()->getPointeeType();
1894efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  } else {
1904efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    result = Visit(E->getBase());
1914efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (result.isUninit())
1924efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return APValue();
1934efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Ty = E->getBase()->getType();
1944efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
1954efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1964efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  RecordDecl *RD = Ty->getAsRecordType()->getDecl();
1974efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
19886f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor
19986f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor  FieldDecl *FD = dyn_cast<FieldDecl>(E->getMemberDecl());
20086f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor  if (!FD) // FIXME: deal with other kinds of member expressions
20186f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor    return APValue();
2024efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
2034efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  // FIXME: This is linear time.
20444b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor  unsigned i = 0;
20544b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor  for (RecordDecl::field_iterator Field = RD->field_begin(),
20644b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor                               FieldEnd = RD->field_end();
20744b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor       Field != FieldEnd; (void)++Field, ++i) {
20844b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor    if (*Field == FD)
2094efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      break;
2104efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
2114efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
2124efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  result.setLValue(result.getLValueBase(),
2134efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman                   result.getLValueOffset() + RL.getFieldOffset(i) / 8);
2144efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
2154efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return result;
2164efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
2174efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
2183068d117951a8df54bae9db039b56201ab10962bAnders CarlssonAPValue LValueExprEvaluator::VisitArraySubscriptExpr(ArraySubscriptExpr *E)
2193068d117951a8df54bae9db039b56201ab10962bAnders Carlsson{
2203068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  APValue Result;
2213068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
2223068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  if (!EvaluatePointer(E->getBase(), Result, Info))
2233068d117951a8df54bae9db039b56201ab10962bAnders Carlsson    return APValue();
2243068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
2253068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  APSInt Index;
2263068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  if (!EvaluateInteger(E->getIdx(), Index, Info))
2273068d117951a8df54bae9db039b56201ab10962bAnders Carlsson    return APValue();
2283068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
2293068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  uint64_t ElementSize = Info.Ctx.getTypeSize(E->getType()) / 8;
2303068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
2313068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  uint64_t Offset = Index.getSExtValue() * ElementSize;
2323068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  Result.setLValue(Result.getLValueBase(),
2333068d117951a8df54bae9db039b56201ab10962bAnders Carlsson                   Result.getLValueOffset() + Offset);
2343068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  return Result;
2353068d117951a8df54bae9db039b56201ab10962bAnders Carlsson}
2364efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
2374efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
238f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner// Pointer Evaluation
239f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
240f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
241c754aa62643e66ab967ca32ae8b0b3fc419bba25Anders Carlssonnamespace {
2422bad1687fe6f00e10767a691a33b070b151902b6Anders Carlssonclass VISIBILITY_HIDDEN PointerExprEvaluator
2432bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  : public StmtVisitor<PointerExprEvaluator, APValue> {
24487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  EvalInfo &Info;
2452bad1687fe6f00e10767a691a33b070b151902b6Anders Carlssonpublic:
2462bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
24787eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  PointerExprEvaluator(EvalInfo &info) : Info(info) {}
248f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
2492bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  APValue VisitStmt(Stmt *S) {
2502bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson    return APValue();
2512bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  }
2522bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
2532bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  APValue VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
2542bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
255650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  APValue VisitBinaryOperator(const BinaryOperator *E);
256650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  APValue VisitCastExpr(const CastExpr* E);
2574efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue VisitUnaryOperator(const UnaryOperator *E);
2584efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue VisitObjCStringLiteral(ObjCStringLiteral *E)
2594efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      { return APValue(E, 0); }
260f01158941ba4560c63150032073bb231ce38999eEli Friedman  APValue VisitAddrLabelExpr(AddrLabelExpr *E)
261f01158941ba4560c63150032073bb231ce38999eEli Friedman      { return APValue(E, 0); }
2623941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman  APValue VisitCallExpr(CallExpr *E);
263bd65cac8de63d108a681035782a71d42954b03abMike Stump  APValue VisitBlockExpr(BlockExpr *E) { return APValue(E, 0); }
2644efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue VisitConditionalOperator(ConditionalOperator *E);
2652bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson};
266f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner} // end anonymous namespace
2672bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
26887eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstatic bool EvaluatePointer(const Expr* E, APValue& Result, EvalInfo &Info) {
269ca2f3fdeedc4e4beea4f7572d6d975188d0bb5efMike Stump  if (!E->getType()->isPointerType()
270ca2f3fdeedc4e4beea4f7572d6d975188d0bb5efMike Stump      && !E->getType()->isBlockPointerType())
271f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner    return false;
27287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  Result = PointerExprEvaluator(Info).Visit(const_cast<Expr*>(E));
273f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  return Result.isLValue();
274f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner}
275650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
276f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris LattnerAPValue PointerExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
277650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (E->getOpcode() != BinaryOperator::Add &&
278650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson      E->getOpcode() != BinaryOperator::Sub)
279650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson    return APValue();
280650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
281650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  const Expr *PExp = E->getLHS();
282650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  const Expr *IExp = E->getRHS();
283650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (IExp->getType()->isPointerType())
284f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner    std::swap(PExp, IExp);
285650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
286650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  APValue ResultLValue;
28787eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  if (!EvaluatePointer(PExp, ResultLValue, Info))
288650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson    return APValue();
289f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
290650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  llvm::APSInt AdditionalOffset(32);
29187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  if (!EvaluateInteger(IExp, AdditionalOffset, Info))
292650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson    return APValue();
293650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
2944efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  QualType PointeeType = PExp->getType()->getAsPointerType()->getPointeeType();
2954d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  uint64_t SizeOfPointee;
2964d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson
2974d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  // Explicitly handle GNU void* and function pointer arithmetic extensions.
2984d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  if (PointeeType->isVoidType() || PointeeType->isFunctionType())
2994d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson    SizeOfPointee = 1;
3004d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  else
3014d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson    SizeOfPointee = Info.Ctx.getTypeSize(PointeeType) / 8;
3024efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
303650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  uint64_t Offset = ResultLValue.getLValueOffset();
3044efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
305650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (E->getOpcode() == BinaryOperator::Add)
3064efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Offset += AdditionalOffset.getLimitedValue() * SizeOfPointee;
307650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  else
3084efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Offset -= AdditionalOffset.getLimitedValue() * SizeOfPointee;
3094efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
310650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  return APValue(ResultLValue.getLValueBase(), Offset);
311650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson}
3124efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
3134efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli FriedmanAPValue PointerExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
3144efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (E->getOpcode() == UnaryOperator::Extension) {
3154efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    // FIXME: Deal with warnings?
3164efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return Visit(E->getSubExpr());
3174efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
3184efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
3194efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (E->getOpcode() == UnaryOperator::AddrOf) {
3204efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APValue result;
3214efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (EvaluateLValue(E->getSubExpr(), result, Info))
3224efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return result;
3234efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
3244efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
3254efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return APValue();
3264efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
327d407a7619904c81d38b41a9ee850de413105084bAnders Carlsson
328650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
329b542afe02d317411d53b3541946f9f2a8f509a11Chris LattnerAPValue PointerExprEvaluator::VisitCastExpr(const CastExpr* E) {
330650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  const Expr* SubExpr = E->getSubExpr();
331650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
332650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson   // Check for pointer->pointer cast
333650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (SubExpr->getType()->isPointerType()) {
334650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson    APValue Result;
33587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    if (EvaluatePointer(SubExpr, Result, Info))
336650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson      return Result;
337f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner    return APValue();
338650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  }
339650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
340d9f4bcda18bfbf79341edd9d381d4b6a3cffe655Eli Friedman  if (SubExpr->getType()->isIntegralType()) {
341650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson    llvm::APSInt Result(32);
34287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    if (EvaluateInteger(SubExpr, Result, Info)) {
34387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner      Result.extOrTrunc((unsigned)Info.Ctx.getTypeSize(E->getType()));
344650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson      return APValue(0, Result.getZExtValue());
345650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson    }
346650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  }
3474efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
3484efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (SubExpr->getType()->isFunctionType() ||
3494efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      SubExpr->getType()->isArrayType()) {
3504efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APValue Result;
3514efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (EvaluateLValue(SubExpr, Result, Info))
3524efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return Result;
3534efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return APValue();
3544efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
3554efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
3564efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  //assert(0 && "Unhandled cast");
357650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  return APValue();
358650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson}
359650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
3603941b18b8e441c8c466efecd557de60b9a32d10bEli FriedmanAPValue PointerExprEvaluator::VisitCallExpr(CallExpr *E) {
3613c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  if (E->isBuiltinCall(Info.Ctx) ==
3623c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor        Builtin::BI__builtin___CFStringMakeConstantString)
3633941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman    return APValue(E, 0);
3643941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman  return APValue();
3653941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman}
3663941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman
3674efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli FriedmanAPValue PointerExprEvaluator::VisitConditionalOperator(ConditionalOperator *E) {
3684efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  bool BoolResult;
3694efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (!HandleConversionToBool(E->getCond(), BoolResult, Info))
3704efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return APValue();
3714efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
3724efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Expr* EvalExpr = BoolResult ? E->getTrueExpr() : E->getFalseExpr();
3734efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
3744efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  APValue Result;
3754efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (EvaluatePointer(EvalExpr, Result, Info))
3764efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return Result;
3774efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return APValue();
3784efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
379f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
380f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
38159b5da6d853b4368b984700315adf7b37de05764Nate Begeman// Vector Evaluation
38259b5da6d853b4368b984700315adf7b37de05764Nate Begeman//===----------------------------------------------------------------------===//
38359b5da6d853b4368b984700315adf7b37de05764Nate Begeman
38459b5da6d853b4368b984700315adf7b37de05764Nate Begemannamespace {
38559b5da6d853b4368b984700315adf7b37de05764Nate Begeman  class VISIBILITY_HIDDEN VectorExprEvaluator
38659b5da6d853b4368b984700315adf7b37de05764Nate Begeman  : public StmtVisitor<VectorExprEvaluator, APValue> {
38759b5da6d853b4368b984700315adf7b37de05764Nate Begeman    EvalInfo &Info;
38859b5da6d853b4368b984700315adf7b37de05764Nate Begeman  public:
38959b5da6d853b4368b984700315adf7b37de05764Nate Begeman
39059b5da6d853b4368b984700315adf7b37de05764Nate Begeman    VectorExprEvaluator(EvalInfo &info) : Info(info) {}
39159b5da6d853b4368b984700315adf7b37de05764Nate Begeman
39259b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitStmt(Stmt *S) {
39359b5da6d853b4368b984700315adf7b37de05764Nate Begeman      return APValue();
39459b5da6d853b4368b984700315adf7b37de05764Nate Begeman    }
39559b5da6d853b4368b984700315adf7b37de05764Nate Begeman
39659b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
39759b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitCastExpr(const CastExpr* E);
39859b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitCompoundLiteralExpr(const CompoundLiteralExpr *E);
39959b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitInitListExpr(const InitListExpr *E);
40059b5da6d853b4368b984700315adf7b37de05764Nate Begeman  };
40159b5da6d853b4368b984700315adf7b37de05764Nate Begeman} // end anonymous namespace
40259b5da6d853b4368b984700315adf7b37de05764Nate Begeman
40359b5da6d853b4368b984700315adf7b37de05764Nate Begemanstatic bool EvaluateVector(const Expr* E, APValue& Result, EvalInfo &Info) {
40459b5da6d853b4368b984700315adf7b37de05764Nate Begeman  if (!E->getType()->isVectorType())
40559b5da6d853b4368b984700315adf7b37de05764Nate Begeman    return false;
40659b5da6d853b4368b984700315adf7b37de05764Nate Begeman  Result = VectorExprEvaluator(Info).Visit(const_cast<Expr*>(E));
40759b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return !Result.isUninit();
40859b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
40959b5da6d853b4368b984700315adf7b37de05764Nate Begeman
41059b5da6d853b4368b984700315adf7b37de05764Nate BegemanAPValue VectorExprEvaluator::VisitCastExpr(const CastExpr* E) {
41159b5da6d853b4368b984700315adf7b37de05764Nate Begeman  const Expr* SE = E->getSubExpr();
41259b5da6d853b4368b984700315adf7b37de05764Nate Begeman
41359b5da6d853b4368b984700315adf7b37de05764Nate Begeman  // Check for vector->vector bitcast.
41459b5da6d853b4368b984700315adf7b37de05764Nate Begeman  if (SE->getType()->isVectorType())
41559b5da6d853b4368b984700315adf7b37de05764Nate Begeman    return this->Visit(const_cast<Expr*>(SE));
41659b5da6d853b4368b984700315adf7b37de05764Nate Begeman
41759b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return APValue();
41859b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
41959b5da6d853b4368b984700315adf7b37de05764Nate Begeman
42059b5da6d853b4368b984700315adf7b37de05764Nate BegemanAPValue
42159b5da6d853b4368b984700315adf7b37de05764Nate BegemanVectorExprEvaluator::VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
42259b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return this->Visit(const_cast<Expr*>(E->getInitializer()));
42359b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
42459b5da6d853b4368b984700315adf7b37de05764Nate Begeman
42559b5da6d853b4368b984700315adf7b37de05764Nate BegemanAPValue
42659b5da6d853b4368b984700315adf7b37de05764Nate BegemanVectorExprEvaluator::VisitInitListExpr(const InitListExpr *E) {
42759b5da6d853b4368b984700315adf7b37de05764Nate Begeman  const VectorType *VT = E->getType()->getAsVectorType();
42859b5da6d853b4368b984700315adf7b37de05764Nate Begeman  unsigned NumInits = E->getNumInits();
42959b5da6d853b4368b984700315adf7b37de05764Nate Begeman
43059b5da6d853b4368b984700315adf7b37de05764Nate Begeman  if (!VT || VT->getNumElements() != NumInits)
43159b5da6d853b4368b984700315adf7b37de05764Nate Begeman    return APValue();
43259b5da6d853b4368b984700315adf7b37de05764Nate Begeman
43359b5da6d853b4368b984700315adf7b37de05764Nate Begeman  QualType EltTy = VT->getElementType();
43459b5da6d853b4368b984700315adf7b37de05764Nate Begeman  llvm::SmallVector<APValue, 4> Elements;
43559b5da6d853b4368b984700315adf7b37de05764Nate Begeman
43659b5da6d853b4368b984700315adf7b37de05764Nate Begeman  for (unsigned i = 0; i < NumInits; i++) {
43759b5da6d853b4368b984700315adf7b37de05764Nate Begeman    if (EltTy->isIntegerType()) {
43859b5da6d853b4368b984700315adf7b37de05764Nate Begeman      llvm::APSInt sInt(32);
43959b5da6d853b4368b984700315adf7b37de05764Nate Begeman      if (!EvaluateInteger(E->getInit(i), sInt, Info))
44059b5da6d853b4368b984700315adf7b37de05764Nate Begeman        return APValue();
44159b5da6d853b4368b984700315adf7b37de05764Nate Begeman      Elements.push_back(APValue(sInt));
44259b5da6d853b4368b984700315adf7b37de05764Nate Begeman    } else {
44359b5da6d853b4368b984700315adf7b37de05764Nate Begeman      llvm::APFloat f(0.0);
44459b5da6d853b4368b984700315adf7b37de05764Nate Begeman      if (!EvaluateFloat(E->getInit(i), f, Info))
44559b5da6d853b4368b984700315adf7b37de05764Nate Begeman        return APValue();
44659b5da6d853b4368b984700315adf7b37de05764Nate Begeman      Elements.push_back(APValue(f));
44759b5da6d853b4368b984700315adf7b37de05764Nate Begeman    }
44859b5da6d853b4368b984700315adf7b37de05764Nate Begeman  }
44959b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return APValue(&Elements[0], Elements.size());
45059b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
45159b5da6d853b4368b984700315adf7b37de05764Nate Begeman
45259b5da6d853b4368b984700315adf7b37de05764Nate Begeman//===----------------------------------------------------------------------===//
453f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner// Integer Evaluation
454f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
455f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
456f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnernamespace {
457f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnerclass VISIBILITY_HIDDEN IntExprEvaluator
458b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  : public StmtVisitor<IntExprEvaluator, bool> {
45987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  EvalInfo &Info;
46030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  APValue &Result;
461f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnerpublic:
46230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  IntExprEvaluator(EvalInfo &info, APValue &result)
46387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    : Info(info), Result(result) {}
464f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
46582206e267ce6cc709797127616f64672d255b310Anders Carlsson  bool Extension(SourceLocation L, diag::kind D, const Expr *E) {
46654da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson    Info.EvalResult.DiagLoc = L;
46754da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson    Info.EvalResult.Diag = D;
46882206e267ce6cc709797127616f64672d255b310Anders Carlsson    Info.EvalResult.DiagExpr = E;
46954176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    return true;  // still a constant.
47054176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner  }
471131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
4723f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  bool Success(const llvm::APSInt &SI, const Expr *E) {
47330c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(SI.isSigned() == E->getType()->isSignedIntegerType() &&
4743f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
47530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(SI.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
4763f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
47730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(SI);
4783f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
4793f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
4803f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar
481131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  bool Success(const llvm::APInt &I, const Expr *E) {
48230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(I.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
4833f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
48430c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(APSInt(I));
48530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result.getInt().setIsUnsigned(E->getType()->isUnsignedIntegerType());
486131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return true;
487131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  }
488131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
489131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  bool Success(uint64_t Value, const Expr *E) {
49030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(Info.Ctx.MakeIntValue(Value, E->getType()));
491131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return true;
492131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  }
493131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
49482206e267ce6cc709797127616f64672d255b310Anders Carlsson  bool Error(SourceLocation L, diag::kind D, const Expr *E) {
49554176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    // If this is in an unevaluated portion of the subexpression, ignore the
49654176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    // error.
497f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson    if (Info.ShortCircuit) {
49832fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner      // If error is ignored because the value isn't evaluated, get the real
49932fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner      // type at least to prevent errors downstream.
500131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(0, E);
50132fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    }
50254176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner
50332fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    // Take the first error.
50430c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar
50530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    // FIXME: This is wrong if we happen to have already emitted an
50630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    // extension diagnostic; in that case we should report this error.
50754da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson    if (Info.EvalResult.Diag == 0) {
50854da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson      Info.EvalResult.DiagLoc = L;
50954da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson      Info.EvalResult.Diag = D;
51082206e267ce6cc709797127616f64672d255b310Anders Carlsson      Info.EvalResult.DiagExpr = E;
51132fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    }
51254176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    return false;
5137a76778e218da57a3b10c80066a0a938f28987b6Chris Lattner  }
5147a76778e218da57a3b10c80066a0a938f28987b6Chris Lattner
515f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //===--------------------------------------------------------------------===//
516f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //                            Visitor Methods
517f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //===--------------------------------------------------------------------===//
51832fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner
51932fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  bool VisitStmt(Stmt *) {
52032fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    assert(0 && "This should be called on integers, stmts are not integers");
52132fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    return false;
52232fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  }
5237a76778e218da57a3b10c80066a0a938f28987b6Chris Lattner
52432fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  bool VisitExpr(Expr *E) {
5250e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
526f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  }
527f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
528b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
529f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
5304c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitIntegerLiteral(const IntegerLiteral *E) {
531131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
5324c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
5334c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitCharacterLiteral(const CharacterLiteral *E) {
534131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
5354c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
5364c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitTypesCompatibleExpr(const TypesCompatibleExpr *E) {
537ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    // Per gcc docs "this built-in function ignores top level
538ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    // qualifiers".  We need to use the canonical version to properly
539ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    // be able to strip CRV qualifiers from the type.
540ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    QualType T0 = Info.Ctx.getCanonicalType(E->getArgType1());
541ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    QualType T1 = Info.Ctx.getCanonicalType(E->getArgType2());
542131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(Info.Ctx.typesAreCompatible(T0.getUnqualifiedType(),
543131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                                               T1.getUnqualifiedType()),
544131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                   E);
5454c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
5464c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitDeclRefExpr(const DeclRefExpr *E);
5474c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitCallExpr(const CallExpr *E);
548b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitBinaryOperator(const BinaryOperator *E);
549b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitUnaryOperator(const UnaryOperator *E);
550ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes  bool VisitConditionalOperator(const ConditionalOperator *E);
551f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
552a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool VisitCastExpr(CastExpr* E);
5530518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl  bool VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E);
5540518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl
5553068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  bool VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
556131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
5573068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
5583068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
5593f70456b8adb0405ef2a47d51f9fc2d5937ae8aeAnders Carlsson  bool VisitGNUNullExpr(const GNUNullExpr *E) {
560131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(0, E);
5613f70456b8adb0405ef2a47d51f9fc2d5937ae8aeAnders Carlsson  }
5623f70456b8adb0405ef2a47d51f9fc2d5937ae8aeAnders Carlsson
5633068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  bool VisitCXXZeroInitValueExpr(const CXXZeroInitValueExpr *E) {
564131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(0, E);
5653068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
5663068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
56764b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl  bool VisitUnaryTypeTraitExpr(const UnaryTypeTraitExpr *E) {
568131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->EvaluateTrait(), E);
56964b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl  }
57064b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl
571fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattnerprivate:
572af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  unsigned GetAlignOfExpr(const Expr *E);
573af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  unsigned GetAlignOfType(QualType T);
574f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner};
575f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner} // end anonymous namespace
576f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
57787eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstatic bool EvaluateInteger(const Expr* E, APSInt &Result, EvalInfo &Info) {
5783f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  if (!E->getType()->isIntegralType())
5793f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return false;
58030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar
58130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  APValue Val;
58230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  if (!IntExprEvaluator(Info, Val).Visit(const_cast<Expr*>(E)) ||
58330c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      !Val.isInt())
58430c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return false;
58530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar
58630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  Result = Val.getInt();
58730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  return true;
588f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner}
589f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
5904c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattnerbool IntExprEvaluator::VisitDeclRefExpr(const DeclRefExpr *E) {
5914c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Enums are integer constant exprs.
5924c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  if (const EnumConstantDecl *D = dyn_cast<EnumConstantDecl>(E->getDecl())) {
593e9a0f43927041e2bcc89140c338dfae207ad3c3bEli Friedman    // FIXME: This is an ugly hack around the fact that enums don't set their
5943f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    // signedness consistently; see PR3173.
5953f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    APSInt SI = D->getInitVal();
5963f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    SI.setIsUnsigned(!E->getType()->isSignedIntegerType());
5973f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    // FIXME: This is an ugly hack around the fact that enums don't
5983f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    // set their width (!?!) consistently; see PR3173.
5993f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    SI.extOrTrunc(Info.Ctx.getIntWidth(E->getType()));
6003f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return Success(SI, E);
6014c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
602b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl
603b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl  // In C++, const, non-volatile integers initialized with ICEs are ICEs.
604b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl  if (Info.Ctx.getLangOptions().CPlusPlus &&
605b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl      E->getType().getCVRQualifiers() == QualType::Const) {
606b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl    if (const VarDecl *D = dyn_cast<VarDecl>(E->getDecl())) {
607b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl      if (const Expr *Init = D->getInit())
608b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl        return Visit(const_cast<Expr*>(Init));
609b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl    }
610b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl  }
611b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl
6124c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Otherwise, random variable references are not constants.
6130e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson  return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
6144c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner}
6154c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner
616a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner/// EvaluateBuiltinClassifyType - Evaluate __builtin_classify_type the same way
617a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner/// as GCC.
618a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattnerstatic int EvaluateBuiltinClassifyType(const CallExpr *E) {
619a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // The following enum mimics the values returned by GCC.
620a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  enum gcc_type_class {
621a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    no_type_class = -1,
622a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    void_type_class, integer_type_class, char_type_class,
623a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    enumeral_type_class, boolean_type_class,
624a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    pointer_type_class, reference_type_class, offset_type_class,
625a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    real_type_class, complex_type_class,
626a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    function_type_class, method_type_class,
627a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    record_type_class, union_type_class,
628a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    array_type_class, string_type_class,
629a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    lang_type_class
630a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  };
631a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner
632a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // If no argument was supplied, default to "no_type_class". This isn't
633a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // ideal, however it is what gcc does.
634a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  if (E->getNumArgs() == 0)
635a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return no_type_class;
636a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner
637a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  QualType ArgTy = E->getArg(0)->getType();
638a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  if (ArgTy->isVoidType())
639a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return void_type_class;
640a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isEnumeralType())
641a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return enumeral_type_class;
642a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isBooleanType())
643a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return boolean_type_class;
644a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isCharType())
645a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return string_type_class; // gcc doesn't appear to use char_type_class
646a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isIntegerType())
647a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return integer_type_class;
648a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isPointerType())
649a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return pointer_type_class;
650a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isReferenceType())
651a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return reference_type_class;
652a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isRealType())
653a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return real_type_class;
654a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isComplexType())
655a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return complex_type_class;
656a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isFunctionType())
657a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return function_type_class;
658a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isStructureType())
659a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return record_type_class;
660a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isUnionType())
661a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return union_type_class;
662a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isArrayType())
663a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return array_type_class;
664a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isUnionType())
665a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return union_type_class;
666a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else  // FIXME: offset_type_class, method_type_class, & lang_type_class?
667a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    assert(0 && "CallExpr::isBuiltinClassifyType(): unimplemented type");
668a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  return -1;
669a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner}
670a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner
6714c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattnerbool IntExprEvaluator::VisitCallExpr(const CallExpr *E) {
6723c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  switch (E->isBuiltinCall(Info.Ctx)) {
673019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  default:
6740e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
675019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_classify_type:
676131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(EvaluateBuiltinClassifyType(E), E);
677019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
6784bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson  case Builtin::BI__builtin_constant_p:
679019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner    // __builtin_constant_p always has one operand: it returns true if that
680019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner    // operand can be folded, false otherwise.
681131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getArg(0)->isEvaluatable(Info.Ctx), E);
682019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  }
6834c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner}
684f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
685b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattnerbool IntExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
686a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->getOpcode() == BinaryOperator::Comma) {
687027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson    if (!Visit(E->getRHS()))
688027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson      return false;
6894fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson
6904fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson    if (!Info.ShortCircuit) {
6914fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson      // If we can't evaluate the LHS, it must be because it has
6924fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson      // side effects.
6934fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson      if (!E->getLHS()->isEvaluatable(Info.Ctx))
6944fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson        Info.EvalResult.HasSideEffects = true;
6954fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson
6960e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Extension(E->getOperatorLoc(), diag::note_comma_in_ice, E);
6974fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson    }
698c8cc9ccc7b87a7ed1749b074f6b670bcec49abc1Chris Lattner
699027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson    return true;
700a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
701a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
702a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->isLogicalOp()) {
703a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // These need to be handled specially because the operands aren't
704a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // necessarily integral
705fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson    bool lhsResult, rhsResult;
70651fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
707fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson    if (HandleConversionToBool(E->getLHS(), lhsResult, Info)) {
70851fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson      // We were able to evaluate the LHS, see if we can get away with not
70951fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson      // evaluating the RHS: 0 && X -> 0, 1 || X -> 1
710fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (lhsResult == (E->getOpcode() == BinaryOperator::LOr) ||
711fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          !lhsResult == (E->getOpcode() == BinaryOperator::LAnd)) {
712f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson        Info.ShortCircuit++;
713fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson        bool rhsEvaluated = HandleConversionToBool(E->getRHS(), rhsResult, Info);
714f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson        Info.ShortCircuit--;
715f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson
716fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson        // FIXME: Return an extension warning saying that the RHS could not be
717fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson        // evaluated.
7183f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar        // if (!rhsEvaluated) ...
7193f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar        (void) rhsEvaluated;
7203f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar
7213f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar        return Success(lhsResult, E);
722a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman      }
723a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
724fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
7254bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        if (E->getOpcode() == BinaryOperator::LOr)
726131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(lhsResult || rhsResult, E);
7274bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        else
728131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(lhsResult && rhsResult, E);
7294bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson      }
7304bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson    } else {
731fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
7324bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        // We can't evaluate the LHS; however, sometimes the result
7334bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        // is determined by the RHS: X && 0 -> 0, X || 1 -> 1.
734fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson        if (rhsResult == (E->getOpcode() == BinaryOperator::LOr) ||
735fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson            !rhsResult == (E->getOpcode() == BinaryOperator::LAnd)) {
736131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          // Since we weren't able to evaluate the left hand side, it
737fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          // must have had side effects.
738fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          Info.EvalResult.HasSideEffects = true;
739131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
740131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(rhsResult, E);
7414bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        }
7424bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson      }
743a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    }
744a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
745a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false;
746c8cc9ccc7b87a7ed1749b074f6b670bcec49abc1Chris Lattner  }
74754176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner
748286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  QualType LHSTy = E->getLHS()->getType();
749286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  QualType RHSTy = E->getRHS()->getType();
7504087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
7514087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar  if (LHSTy->isAnyComplexType()) {
7524087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    assert(RHSTy->isAnyComplexType() && "Invalid comparison");
7534087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    APValue LHS, RHS;
7544087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
7554087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (!EvaluateComplex(E->getLHS(), LHS, Info))
7564087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      return false;
7574087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
7584087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (!EvaluateComplex(E->getRHS(), RHS, Info))
7594087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      return false;
7604087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
7614087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (LHS.isComplexFloat()) {
7624087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      APFloat::cmpResult CR_r =
7634087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar        LHS.getComplexFloatReal().compare(RHS.getComplexFloatReal());
7644087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      APFloat::cmpResult CR_i =
7654087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar        LHS.getComplexFloatImag().compare(RHS.getComplexFloatImag());
7664087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
7674087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      if (E->getOpcode() == BinaryOperator::EQ)
768131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((CR_r == APFloat::cmpEqual &&
769131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        CR_i == APFloat::cmpEqual), E);
770131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      else {
771131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        assert(E->getOpcode() == BinaryOperator::NE &&
772131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar               "Invalid complex comparison.");
773131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success(((CR_r == APFloat::cmpGreaterThan ||
774131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                         CR_r == APFloat::cmpLessThan) &&
775131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        (CR_i == APFloat::cmpGreaterThan ||
776131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                         CR_i == APFloat::cmpLessThan)), E);
777131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      }
7784087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    } else {
7794087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      if (E->getOpcode() == BinaryOperator::EQ)
780131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((LHS.getComplexIntReal() == RHS.getComplexIntReal() &&
781131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        LHS.getComplexIntImag() == RHS.getComplexIntImag()), E);
782131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      else {
783131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        assert(E->getOpcode() == BinaryOperator::NE &&
784131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar               "Invalid compex comparison.");
785131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((LHS.getComplexIntReal() != RHS.getComplexIntReal() ||
786131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        LHS.getComplexIntImag() != RHS.getComplexIntImag()), E);
787131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      }
7884087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    }
7894087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar  }
790286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson
791286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  if (LHSTy->isRealFloatingType() &&
792286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      RHSTy->isRealFloatingType()) {
793286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    APFloat RHS(0.0), LHS(0.0);
794286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson
795286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    if (!EvaluateFloat(E->getRHS(), RHS, Info))
796286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      return false;
797286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson
798286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    if (!EvaluateFloat(E->getLHS(), LHS, Info))
799286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      return false;
800286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson
801286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    APFloat::cmpResult CR = LHS.compare(RHS);
802529569e68d10b0fd3750fd2124faf742249b846bAnders Carlsson
803286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    switch (E->getOpcode()) {
804286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    default:
805286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      assert(0 && "Invalid binary operator!");
806286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::LT:
807131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpLessThan, E);
808286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::GT:
809131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpGreaterThan, E);
810286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::LE:
811131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpLessThan || CR == APFloat::cmpEqual, E);
812286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::GE:
813131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpGreaterThan || CR == APFloat::cmpEqual,
814131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                     E);
815286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::EQ:
816131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpEqual, E);
817286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::NE:
818131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpGreaterThan
819131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                     || CR == APFloat::cmpLessThan, E);
820286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    }
821286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  }
822286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson
8233068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  if (E->getOpcode() == BinaryOperator::Sub) {
824529569e68d10b0fd3750fd2124faf742249b846bAnders Carlsson    if (LHSTy->isPointerType() && RHSTy->isPointerType()) {
8253068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      APValue LHSValue;
8263068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (!EvaluatePointer(E->getLHS(), LHSValue, Info))
8273068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
8283068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
8293068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      APValue RHSValue;
8303068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (!EvaluatePointer(E->getRHS(), RHSValue, Info))
8313068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
8323068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
8333068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      // FIXME: Is this correct? What if only one of the operands has a base?
8343068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (LHSValue.getLValueBase() || RHSValue.getLValueBase())
8353068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
8363068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
8373068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      const QualType Type = E->getLHS()->getType();
8383068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      const QualType ElementType = Type->getAsPointerType()->getPointeeType();
8393068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
8403068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      uint64_t D = LHSValue.getLValueOffset() - RHSValue.getLValueOffset();
8413068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      D /= Info.Ctx.getTypeSize(ElementType) / 8;
8423068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
843131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(D, E);
8443068d117951a8df54bae9db039b56201ab10962bAnders Carlsson    }
8453068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
846286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  if (!LHSTy->isIntegralType() ||
847286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      !RHSTy->isIntegralType()) {
848a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // We can't continue from here for non-integral types, and they
849a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // could potentially confuse the following operations.
850a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // FIXME: Deal with EQ and friends.
851a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false;
852a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
853a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
854a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  // The LHS of a constant expr is always evaluated and needed.
85530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  if (!Visit(E->getLHS()))
856a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false; // error in subexpression.
857d9f4bcda18bfbf79341edd9d381d4b6a3cffe655Eli Friedman
85830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  // Only support arithmetic on integers for now.
85930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  if (!Result.isInt())
86030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return false;
86130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar
862131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  llvm::APSInt RHS;
86354176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner  if (!EvaluateInteger(E->getRHS(), RHS, Info))
864b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner    return false;
865a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
866a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  switch (E->getOpcode()) {
86732fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  default:
8680e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
86930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Mul: return Success(Result.getInt() * RHS, E);
87030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Add: return Success(Result.getInt() + RHS, E);
87130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Sub: return Success(Result.getInt() - RHS, E);
87230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::And: return Success(Result.getInt() & RHS, E);
87330c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Xor: return Success(Result.getInt() ^ RHS, E);
87430c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Or:  return Success(Result.getInt() | RHS, E);
875a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  case BinaryOperator::Div:
87654176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    if (RHS == 0)
8770e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
87830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() / RHS, E);
879a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  case BinaryOperator::Rem:
88054176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    if (RHS == 0)
8810e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
88230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() % RHS, E);
8833f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  case BinaryOperator::Shl: {
88454176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    // FIXME: Warn about out of range shift amounts!
88530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    unsigned SA =
88630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
88730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() << SA, E);
8883f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
8893f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  case BinaryOperator::Shr: {
89030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    unsigned SA =
89130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
89230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() >> SA, E);
8933f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
894b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner
89530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::LT: return Success(Result.getInt() < RHS, E);
89630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::GT: return Success(Result.getInt() > RHS, E);
89730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::LE: return Success(Result.getInt() <= RHS, E);
89830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::GE: return Success(Result.getInt() >= RHS, E);
89930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::EQ: return Success(Result.getInt() == RHS, E);
90030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::NE: return Success(Result.getInt() != RHS, E);
901b11e77836dd0867955c5abf32baf1c3e6c7f81e1Eli Friedman  }
902a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
903a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
904ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopesbool IntExprEvaluator::VisitConditionalOperator(const ConditionalOperator *E) {
905a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  bool Cond;
906a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  if (!HandleConversionToBool(E->getCond(), Cond, Info))
907ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes    return false;
908ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes
909a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  return Visit(Cond ? E->getTrueExpr() : E->getFalseExpr());
910ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes}
911ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes
912af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattnerunsigned IntExprEvaluator::GetAlignOfType(QualType T) {
913e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  const Type *Ty = Info.Ctx.getCanonicalType(T).getTypePtr();
914e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
915e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // __alignof__(void) = 1 as a gcc extension.
916e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (Ty->isVoidType())
917e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    return 1;
918e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
919e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // GCC extension: alignof(function) = 4.
920e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // FIXME: AlignOf shouldn't be unconditionally 4!  It should listen to the
921e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // attribute(align) directive.
922e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (Ty->isFunctionType())
923e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    return 4;
924e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
925f11284ac87daa613bc7b30db9f54bd716d123222Fariborz Jahanian  if (const ExtQualType *EXTQT = dyn_cast<ExtQualType>(Ty))
926f11284ac87daa613bc7b30db9f54bd716d123222Fariborz Jahanian    return GetAlignOfType(QualType(EXTQT->getBaseType(), 0));
927e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
928e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // alignof VLA/incomplete array.
929e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (const ArrayType *VAT = dyn_cast<ArrayType>(Ty))
930e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    return GetAlignOfType(VAT->getElementType());
931e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
932e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // sizeof (objc class)?
933e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (isa<ObjCInterfaceType>(Ty))
934e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    return 1;  // FIXME: This probably isn't right.
935e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
936e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Get information about the alignment.
937e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  unsigned CharSize = Info.Ctx.Target.getCharWidth();
938e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  return Info.Ctx.getTypeAlign(Ty) / CharSize;
939e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner}
940e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
941af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattnerunsigned IntExprEvaluator::GetAlignOfExpr(const Expr *E) {
942af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  E = E->IgnoreParens();
943af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
944af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  // alignof decl is always accepted, even if it doesn't make sense: we default
945af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  // to 1 in those cases.
946af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
947b7d0844c887a72064b624dc6df12cbe1441f69d0Daniel Dunbar    return Info.Ctx.getDeclAlignInBytes(DRE->getDecl());
948af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
949af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  if (const MemberExpr *ME = dyn_cast<MemberExpr>(E))
950b7d0844c887a72064b624dc6df12cbe1441f69d0Daniel Dunbar    return Info.Ctx.getDeclAlignInBytes(ME->getMemberDecl());
951af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
952e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  return GetAlignOfType(E->getType());
953e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner}
954e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
955e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
9560518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl/// VisitSizeAlignOfExpr - Evaluate a sizeof or alignof with a result as the
9570518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl/// expression's type.
9580518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redlbool IntExprEvaluator::VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E) {
9590518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl  QualType DstTy = E->getType();
960fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner
961e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Handle alignof separately.
962e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (!E->isSizeOf()) {
963e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    if (E->isArgumentType())
964131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(GetAlignOfType(E->getArgumentType()), E);
965e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    else
966131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(GetAlignOfExpr(E->getArgumentExpr()), E);
967e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  }
968e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
9690518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl  QualType SrcTy = E->getTypeOfArgument();
9700518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl
971131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  // sizeof(void), __alignof__(void), sizeof(function) = 1 as a gcc
972131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  // extension.
973131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  if (SrcTy->isVoidType() || SrcTy->isFunctionType())
974131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(1, E);
975fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner
976fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner  // sizeof(vla) is not a constantexpr: C99 6.5.3.4p2.
977e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (!SrcTy->isConstantSizeType())
978fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner    return false;
979f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman
980f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman  if (SrcTy->isObjCInterfaceType()) {
981f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman    // Slightly unusual case: the size of an ObjC interface type is the
982f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman    // size of the class.  This code intentionally falls through to the normal
983f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman    // case.
984f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman    ObjCInterfaceDecl *OI = SrcTy->getAsObjCInterfaceType()->getDecl();
985f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman    RecordDecl *RD = const_cast<RecordDecl*>(Info.Ctx.addRecordToClass(OI));
986f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman    SrcTy = Info.Ctx.getTagDeclType(static_cast<TagDecl*>(RD));
987f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman  }
988f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman
989e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Get information about the size.
99087eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  unsigned CharSize = Info.Ctx.Target.getCharWidth();
991131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  return Success(Info.Ctx.getTypeSize(SrcTy) / CharSize, E);
992fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner}
993fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner
994b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattnerbool IntExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
9954c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Special case unary operators that do not need their subexpression
9964c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // evaluated.  offsetof/sizeof/alignof are all special.
997131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  if (E->isOffsetOfOp())
998131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->evaluateOffsetOf(Info.Ctx), E);
999a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1000a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->getOpcode() == UnaryOperator::LNot) {
1001a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // LNot's operand isn't necessarily an integer, so we handle it specially.
1002a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    bool bres;
1003a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    if (!HandleConversionToBool(E->getSubExpr(), bres, Info))
1004a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman      return false;
1005131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(!bres, E);
1006a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1007a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
100887eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  // Get the operand value into 'Result'.
100987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  if (!Visit(E->getSubExpr()))
101075a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    return false;
1011a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
101275a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  switch (E->getOpcode()) {
10134c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  default:
101475a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    // Address, indirect, pre/post inc/dec, etc are not valid constant exprs.
101575a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    // See C99 6.6p3.
10160e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
101775a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Extension:
10184c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // FIXME: Should extension allow i-c-e extension expressions in its scope?
10194c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // If so, we could clear the diagnostic ID.
10203f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
102175a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Plus:
10224c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // The result is always just the subexpr.
10233f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
102475a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Minus:
102530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt()) return false;
102630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(-Result.getInt(), E);
102775a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Not:
102830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt()) return false;
102930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(~Result.getInt(), E);
103006a3675627e3b3c47b49c689c8e404a33144194aAnders Carlsson  }
1031a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
1032a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
1033732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner/// HandleCast - This is used to evaluate implicit or explicit casts where the
1034732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner/// result type is integer.
1035a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbarbool IntExprEvaluator::VisitCastExpr(CastExpr *E) {
103682206e267ce6cc709797127616f64672d255b310Anders Carlsson  Expr *SubExpr = E->getSubExpr();
103782206e267ce6cc709797127616f64672d255b310Anders Carlsson  QualType DestType = E->getType();
103882206e267ce6cc709797127616f64672d255b310Anders Carlsson
10394efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (DestType->isBooleanType()) {
10404efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    bool BoolResult;
10414efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!HandleConversionToBool(SubExpr, BoolResult, Info))
10424efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1043131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(BoolResult, E);
10444efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
10454efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1046a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  // Handle simple integer->integer casts.
1047a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (SubExpr->getType()->isIntegralType()) {
1048732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner    if (!Visit(SubExpr))
1049b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
1050a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
105130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    // FIXME: Support cast on LValue results.
105230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt())
105330c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      return false;
105430c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar
10553f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return Success(HandleIntToIntCast(DestType, SubExpr->getType(),
105630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar                                      Result.getInt(), Info.Ctx), E);
1057732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  }
1058732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner
1059732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  // FIXME: Clean this up!
1060732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  if (SubExpr->getType()->isPointerType()) {
1061a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson    APValue LV;
106287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    if (!EvaluatePointer(SubExpr, LV, Info))
1063b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
10644efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1065a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson    if (LV.getLValueBase())
1066b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
10674efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1068131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(LV.getLValueOffset(), E);
10692bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  }
10704efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1071732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  if (!SubExpr->getType()->isRealFloatingType())
10720e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1073732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner
1074d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  APFloat F(0.0);
1075d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(SubExpr, F, Info))
10760e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1077732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner
10783f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  return Success(HandleFloatToIntCast(DestType, SubExpr->getType(),
10793f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar                                      F, Info.Ctx), E);
1080a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
10812bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
1082f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
1083d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman// Float Evaluation
1084d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman//===----------------------------------------------------------------------===//
1085d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1086d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmannamespace {
1087d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanclass VISIBILITY_HIDDEN FloatExprEvaluator
1088d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  : public StmtVisitor<FloatExprEvaluator, bool> {
1089d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  EvalInfo &Info;
1090d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  APFloat &Result;
1091d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanpublic:
1092d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  FloatExprEvaluator(EvalInfo &info, APFloat &result)
1093d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    : Info(info), Result(result) {}
1094d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1095d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitStmt(Stmt *S) {
1096d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1097d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  }
1098d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1099d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
1100019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  bool VisitCallExpr(const CallExpr *E);
1101d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
11025db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  bool VisitUnaryOperator(const UnaryOperator *E);
1103d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitBinaryOperator(const BinaryOperator *E);
1104d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitFloatingLiteral(const FloatingLiteral *E);
11054efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  bool VisitCastExpr(CastExpr *E);
11064efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  bool VisitCXXZeroInitValueExpr(CXXZeroInitValueExpr *E);
1107d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman};
1108d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman} // end anonymous namespace
1109d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1110d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanstatic bool EvaluateFloat(const Expr* E, APFloat& Result, EvalInfo &Info) {
1111d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  return FloatExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
1112d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1113d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1114019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattnerbool FloatExprEvaluator::VisitCallExpr(const CallExpr *E) {
11153c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  switch (E->isBuiltinCall(Info.Ctx)) {
111634a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner  default: return false;
1117019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_val:
1118019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_valf:
1119019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_vall:
1120019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_inf:
1121019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_inff:
11227cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar  case Builtin::BI__builtin_infl: {
11237cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar    const llvm::fltSemantics &Sem =
11247cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar      Info.Ctx.getFloatTypeSemantics(E->getType());
112534a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner    Result = llvm::APFloat::getInf(Sem);
112634a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner    return true;
11277cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar  }
11289e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner
11299e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nan:
11309e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nanf:
11319e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nanl:
11329e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner    // If this is __builtin_nan("") turn this into a simple nan, otherwise we
11339e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner    // can't constant fold it.
11349e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner    if (const StringLiteral *S =
11359e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner        dyn_cast<StringLiteral>(E->getArg(0)->IgnoreParenCasts())) {
11369e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner      if (!S->isWide() && S->getByteLength() == 0) { // empty string.
11377cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar        const llvm::fltSemantics &Sem =
11387cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar          Info.Ctx.getFloatTypeSemantics(E->getType());
11399e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner        Result = llvm::APFloat::getNaN(Sem);
11409e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner        return true;
11419e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner      }
11429e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner    }
11439e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner    return false;
11445db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
11455db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabs:
11465db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabsf:
11475db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabsl:
11485db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (!EvaluateFloat(E->getArg(0), Result, Info))
11495db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      return false;
11505db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
11515db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (Result.isNegative())
11525db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      Result.changeSign();
11535db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
11545db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
11555db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_copysign:
11565db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_copysignf:
11575db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_copysignl: {
11585db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    APFloat RHS(0.);
11595db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (!EvaluateFloat(E->getArg(0), Result, Info) ||
11605db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar        !EvaluateFloat(E->getArg(1), RHS, Info))
11615db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      return false;
11625db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    Result.copySign(RHS);
11635db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
11645db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  }
1165019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  }
1166019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner}
1167019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
11685db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbarbool FloatExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
1169a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes  if (E->getOpcode() == UnaryOperator::Deref)
1170a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes    return false;
1171a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes
11725db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  if (!EvaluateFloat(E->getSubExpr(), Result, Info))
11735db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return false;
11745db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
11755db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  switch (E->getOpcode()) {
11765db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  default: return false;
11775db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case UnaryOperator::Plus:
11785db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
11795db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case UnaryOperator::Minus:
11805db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    Result.changeSign();
11815db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
11825db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  }
11835db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar}
1184019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
1185d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanbool FloatExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
1186d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  // FIXME: Diagnostics?  I really don't understand how the warnings
1187d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  // and errors are supposed to work.
11885db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  APFloat RHS(0.0);
1189d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(E->getLHS(), Result, Info))
1190d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1191d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(E->getRHS(), RHS, Info))
1192d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1193d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1194d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  switch (E->getOpcode()) {
1195d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  default: return false;
1196d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Mul:
1197d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.multiply(RHS, APFloat::rmNearestTiesToEven);
1198d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1199d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Add:
1200d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.add(RHS, APFloat::rmNearestTiesToEven);
1201d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1202d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Sub:
1203d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.subtract(RHS, APFloat::rmNearestTiesToEven);
1204d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1205d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Div:
1206d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.divide(RHS, APFloat::rmNearestTiesToEven);
1207d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1208d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Rem:
1209d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.mod(RHS, APFloat::rmNearestTiesToEven);
1210d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1211d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  }
1212d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1213d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1214d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanbool FloatExprEvaluator::VisitFloatingLiteral(const FloatingLiteral *E) {
1215d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  Result = E->getValue();
1216d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  return true;
1217d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1218d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
12194efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanbool FloatExprEvaluator::VisitCastExpr(CastExpr *E) {
12204efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Expr* SubExpr = E->getSubExpr();
122159b5da6d853b4368b984700315adf7b37de05764Nate Begeman
12224efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (SubExpr->getType()->isIntegralType()) {
12234efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APSInt IntResult;
12243f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    if (!EvaluateInteger(SubExpr, IntResult, Info))
12254efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1226a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    Result = HandleIntToFloatCast(E->getType(), SubExpr->getType(),
1227a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                  IntResult, Info.Ctx);
12284efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
12294efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
12304efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (SubExpr->getType()->isRealFloatingType()) {
12314efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!Visit(SubExpr))
12324efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1233a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    Result = HandleFloatToFloatCast(E->getType(), SubExpr->getType(),
1234a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                    Result, Info.Ctx);
12354efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
12364efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
12374efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
12384efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return false;
12394efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
12404efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
12414efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanbool FloatExprEvaluator::VisitCXXZeroInitValueExpr(CXXZeroInitValueExpr *E) {
12424efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Result = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(E->getType()));
12434efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return true;
12444efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
12454efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1246d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman//===----------------------------------------------------------------------===//
1247a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar// Complex Evaluation (for float and integer)
12489ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson//===----------------------------------------------------------------------===//
12499ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
12509ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlssonnamespace {
1251a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbarclass VISIBILITY_HIDDEN ComplexExprEvaluator
1252a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  : public StmtVisitor<ComplexExprEvaluator, APValue> {
12539ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  EvalInfo &Info;
12549ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
12559ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlssonpublic:
1256a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  ComplexExprEvaluator(EvalInfo &info) : Info(info) {}
12579ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
12589ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //===--------------------------------------------------------------------===//
12599ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //                            Visitor Methods
12609ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //===--------------------------------------------------------------------===//
12619ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
12629ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  APValue VisitStmt(Stmt *S) {
12639ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson    return APValue();
12649ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  }
12659ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
12669ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  APValue VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
12679ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
12689ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  APValue VisitImaginaryLiteral(ImaginaryLiteral *E) {
1269a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    Expr* SubExpr = E->getSubExpr();
1270a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
1271a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (SubExpr->getType()->isRealFloatingType()) {
1272a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      APFloat Result(0.0);
1273a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
1274a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      if (!EvaluateFloat(SubExpr, Result, Info))
1275a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar        return APValue();
12769ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
12773f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      return APValue(APFloat(Result.getSemantics(), APFloat::fcZero, false),
1278a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                     Result);
1279a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
1280a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      assert(SubExpr->getType()->isIntegerType() &&
1281a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar             "Unexpected imaginary literal.");
1282a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
1283a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      llvm::APSInt Result;
1284a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      if (!EvaluateInteger(SubExpr, Result, Info))
1285a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar        return APValue();
1286a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
1287a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      llvm::APSInt Zero(Result.getBitWidth(), !Result.isSigned());
1288a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Zero = 0;
1289a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      return APValue(Zero, Result);
1290a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
12919ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  }
12929ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
1293ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  APValue VisitCastExpr(CastExpr *E) {
1294ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    Expr* SubExpr = E->getSubExpr();
1295a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    QualType EltType = E->getType()->getAsComplexType()->getElementType();
1296a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    QualType SubType = SubExpr->getType();
1297ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
1298a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    if (SubType->isRealFloatingType()) {
1299ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson      APFloat Result(0.0);
1300ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
1301ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson      if (!EvaluateFloat(SubExpr, Result, Info))
1302ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson        return APValue();
1303ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
1304a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      // Apply float conversion if necessary.
1305a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      Result = HandleFloatToFloatCast(EltType, SubType, Result, Info.Ctx);
13068f826f0e0c90b05ab04e84c29157d2e965713288Daniel Dunbar      return APValue(Result,
13073f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar                     APFloat(Result.getSemantics(), APFloat::fcZero, false));
1308a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    } else if (SubType->isIntegerType()) {
1309a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      APSInt Result;
1310a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
1311a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      if (!EvaluateInteger(SubExpr, Result, Info))
1312a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar        return APValue();
1313a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
1314a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      // Apply integer conversion if necessary.
1315a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      Result = HandleIntToIntCast(EltType, SubType, Result, Info.Ctx);
1316a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      llvm::APSInt Zero(Result.getBitWidth(), !Result.isSigned());
1317a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Zero = 0;
1318a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      return APValue(Result, Zero);
1319a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    } else if (const ComplexType *CT = SubType->getAsComplexType()) {
1320a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      APValue Src;
1321a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
1322a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      if (!EvaluateComplex(SubExpr, Src, Info))
1323a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        return APValue();
1324a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
1325a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      QualType SrcType = CT->getElementType();
1326a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
1327a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      if (Src.isComplexFloat()) {
1328a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        if (EltType->isRealFloatingType()) {
1329a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar          return APValue(HandleFloatToFloatCast(EltType, SrcType,
1330a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                                Src.getComplexFloatReal(),
1331a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                                Info.Ctx),
1332a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                         HandleFloatToFloatCast(EltType, SrcType,
1333a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                                Src.getComplexFloatImag(),
1334a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                                Info.Ctx));
1335a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        } else {
1336a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar          return APValue(HandleFloatToIntCast(EltType, SrcType,
1337a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                              Src.getComplexFloatReal(),
1338a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                              Info.Ctx),
1339a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                         HandleFloatToIntCast(EltType, SrcType,
1340a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                              Src.getComplexFloatImag(),
1341a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                              Info.Ctx));
1342a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        }
1343a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      } else {
1344a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        assert(Src.isComplexInt() && "Invalid evaluate result.");
1345a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        if (EltType->isRealFloatingType()) {
1346a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar          return APValue(HandleIntToFloatCast(EltType, SrcType,
1347a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                              Src.getComplexIntReal(),
1348a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                              Info.Ctx),
1349a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                         HandleIntToFloatCast(EltType, SrcType,
1350a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                              Src.getComplexIntImag(),
1351a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                              Info.Ctx));
1352a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        } else {
1353a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar          return APValue(HandleIntToIntCast(EltType, SrcType,
1354a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                            Src.getComplexIntReal(),
1355a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                            Info.Ctx),
1356a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                         HandleIntToIntCast(EltType, SrcType,
1357a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                            Src.getComplexIntImag(),
1358a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                            Info.Ctx));
1359a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        }
1360a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      }
1361ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    }
1362ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
1363ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    // FIXME: Handle more casts.
1364ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    return APValue();
1365ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  }
1366ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
1367ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  APValue VisitBinaryOperator(const BinaryOperator *E);
1368ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
13699ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson};
13709ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson} // end anonymous namespace
13719ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
1372a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbarstatic bool EvaluateComplex(const Expr *E, APValue &Result, EvalInfo &Info)
13739ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson{
1374a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  Result = ComplexExprEvaluator(Info).Visit(const_cast<Expr*>(E));
1375a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  assert((!Result.isComplexFloat() ||
1376a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar          (&Result.getComplexFloatReal().getSemantics() ==
1377a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar           &Result.getComplexFloatImag().getSemantics())) &&
1378a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar         "Invalid complex evaluation.");
1379a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  return Result.isComplexFloat() || Result.isComplexInt();
13809ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson}
13819ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
1382a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel DunbarAPValue ComplexExprEvaluator::VisitBinaryOperator(const BinaryOperator *E)
1383ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson{
1384ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  APValue Result, RHS;
1385ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
1386a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  if (!EvaluateComplex(E->getLHS(), Result, Info))
1387ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    return APValue();
1388ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
1389a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  if (!EvaluateComplex(E->getRHS(), RHS, Info))
1390ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    return APValue();
1391a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
13923f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar  assert(Result.isComplexFloat() == RHS.isComplexFloat() &&
13933f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         "Invalid operands to binary operator.");
1394ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  switch (E->getOpcode()) {
1395ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  default: return APValue();
1396ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  case BinaryOperator::Add:
1397a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (Result.isComplexFloat()) {
1398a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatReal().add(RHS.getComplexFloatReal(),
1399a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                       APFloat::rmNearestTiesToEven);
1400a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatImag().add(RHS.getComplexFloatImag(),
1401a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                       APFloat::rmNearestTiesToEven);
1402a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
1403a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntReal() += RHS.getComplexIntReal();
1404a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntImag() += RHS.getComplexIntImag();
1405a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
14063f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
1407ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  case BinaryOperator::Sub:
1408a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (Result.isComplexFloat()) {
1409a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatReal().subtract(RHS.getComplexFloatReal(),
1410a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                            APFloat::rmNearestTiesToEven);
1411a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatImag().subtract(RHS.getComplexFloatImag(),
1412a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                            APFloat::rmNearestTiesToEven);
1413a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
1414a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntReal() -= RHS.getComplexIntReal();
1415a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntImag() -= RHS.getComplexIntImag();
1416a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
14173f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
14183f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar  case BinaryOperator::Mul:
14193f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    if (Result.isComplexFloat()) {
14203f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APValue LHS = Result;
14213f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &LHS_r = LHS.getComplexFloatReal();
14223f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &LHS_i = LHS.getComplexFloatImag();
14233f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &RHS_r = RHS.getComplexFloatReal();
14243f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &RHS_i = RHS.getComplexFloatImag();
14253f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar
14263f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat Tmp = LHS_r;
14273f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
14283f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatReal() = Tmp;
14293f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_i;
14303f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
14313f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatReal().subtract(Tmp, APFloat::rmNearestTiesToEven);
14323f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar
14333f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_r;
14343f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
14353f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatImag() = Tmp;
14363f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_i;
14373f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
14383f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatImag().add(Tmp, APFloat::rmNearestTiesToEven);
14393f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    } else {
14403f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APValue LHS = Result;
14413f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexIntReal() =
14423f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar        (LHS.getComplexIntReal() * RHS.getComplexIntReal() -
14433f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         LHS.getComplexIntImag() * RHS.getComplexIntImag());
14443f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexIntImag() =
14453f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar        (LHS.getComplexIntReal() * RHS.getComplexIntImag() +
14463f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         LHS.getComplexIntImag() * RHS.getComplexIntReal());
14473f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    }
14483f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
1449ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  }
1450ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
1451ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  return Result;
1452ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson}
1453ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
14549ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson//===----------------------------------------------------------------------===//
14556ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner// Top level Expr::Evaluate method.
1456f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
1457f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
14586ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner/// Evaluate - Return true if this is a constant which we can fold using
1459019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner/// any crazy technique (that has nothing to do with language standards) that
1460019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner/// we want to.  If this function returns true, it returns the folded constant
1461019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner/// in Result.
14625b45d4ef1ea3f04ec863daf8aa29be6c6e021750Anders Carlssonbool Expr::Evaluate(EvalResult &Result, ASTContext &Ctx) const {
14635b45d4ef1ea3f04ec863daf8aa29be6c6e021750Anders Carlsson  EvalInfo Info(Ctx, Result);
146454da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson
146559b5da6d853b4368b984700315adf7b37de05764Nate Begeman  if (getType()->isVectorType()) {
146659b5da6d853b4368b984700315adf7b37de05764Nate Begeman    if (!EvaluateVector(this, Result.Val, Info))
146759b5da6d853b4368b984700315adf7b37de05764Nate Begeman      return false;
146859b5da6d853b4368b984700315adf7b37de05764Nate Begeman  } else if (getType()->isIntegerType()) {
146930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!IntExprEvaluator(Info, Result.Val).Visit(const_cast<Expr*>(this)))
14706dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
1471ca2f3fdeedc4e4beea4f7572d6d975188d0bb5efMike Stump  } else if (getType()->isPointerType()
1472ca2f3fdeedc4e4beea4f7572d6d975188d0bb5efMike Stump             || getType()->isBlockPointerType()) {
14735b45d4ef1ea3f04ec863daf8aa29be6c6e021750Anders Carlsson    if (!EvaluatePointer(this, Result.Val, Info))
14746dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
1475d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  } else if (getType()->isRealFloatingType()) {
1476d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    llvm::APFloat f(0.0);
14776dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson    if (!EvaluateFloat(this, f, Info))
14786dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
14796dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson
14805b45d4ef1ea3f04ec863daf8aa29be6c6e021750Anders Carlsson    Result.Val = APValue(f);
1481a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  } else if (getType()->isAnyComplexType()) {
1482a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (!EvaluateComplex(this, Result.Val, Info))
14836dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
1484a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  } else
14859d4c157baaada8945671dc98363a40ed00cbd824Anders Carlsson    return false;
14866dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson
14875b45d4ef1ea3f04ec863daf8aa29be6c6e021750Anders Carlsson  return true;
14885b45d4ef1ea3f04ec863daf8aa29be6c6e021750Anders Carlsson}
14895b45d4ef1ea3f04ec863daf8aa29be6c6e021750Anders Carlsson
14906ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner/// isEvaluatable - Call Evaluate to see if this expression can be constant
149145b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattner/// folded, but discard the result.
149245b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattnerbool Expr::isEvaluatable(ASTContext &Ctx) const {
14934fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson  EvalResult Result;
14944fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson  return Evaluate(Result, Ctx) && !Result.HasSideEffects;
149545b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattner}
149651fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
149751fe996231b1d7199f76e4005ff4c943d5deeecdAnders CarlssonAPSInt Expr::EvaluateAsInt(ASTContext &Ctx) const {
14981c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  EvalResult EvalResult;
14991c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  bool Result = Evaluate(EvalResult, Ctx);
1500f185319f25efd6094870f287030270fad26085baDaniel Dunbar  Result = Result;
150151fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson  assert(Result && "Could not evaluate expression");
15021c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  assert(EvalResult.Val.isInt() && "Expression did not evaluate to integer");
150351fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
15041c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  return EvalResult.Val.getInt();
150551fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson}
1506