ExprConstant.cpp revision c6ed729f669044f5072a49d79041f455d971ece3
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"
16199c3d6cd16aebbb9c7f0d42af9d922c9628bf70Ken Dyck#include "clang/AST/CharUnits.h"
1719cc4abea06a9b49e0e16a50d335c064cd723572Anders Carlsson#include "clang/AST/RecordLayout.h"
180fe52e1bcaa69ba127f1bda036f057fec1f478deSeo Sanghyeon#include "clang/AST/StmtVisitor.h"
198ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor#include "clang/AST/TypeLoc.h"
20500d3297d2a21edeac4d46cbcbe21bc2352c2a28Chris Lattner#include "clang/AST/ASTDiagnostic.h"
218ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor#include "clang/AST/Expr.h"
221b63e4f732dbc73d90abf886b4d21f8e3a165f6dChris Lattner#include "clang/Basic/Builtins.h"
2306a3675627e3b3c47b49c689c8e404a33144194aAnders Carlsson#include "clang/Basic/TargetInfo.h"
247462b39a9bccaf4392687831036713f09f9c0681Mike Stump#include "llvm/ADT/SmallString.h"
254572baba9d18c275968ac113fd73b0e3c77cccb8Mike Stump#include <cstring>
264572baba9d18c275968ac113fd73b0e3c77cccb8Mike Stump
27c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlssonusing namespace clang;
28f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnerusing llvm::APSInt;
29d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanusing llvm::APFloat;
30c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson
3187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// EvalInfo - This is a private struct used by the evaluator to capture
3287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// information about a subexpression as it is folded.  It retains information
3387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// about the AST context, but also maintains information about the folded
3487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// expression.
3587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner///
3687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// If an expression could be evaluated, it is still possible it is not a C
3787eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// "integer constant expression" or constant expression.  If not, this struct
3887eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// captures information about how and why not.
3987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner///
4087eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// One bit of information passed *into* the request for constant folding
4187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// indicates whether the subexpression is "evaluated" or not according to C
4287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// rules.  For example, the RHS of (0 && foo()) is not evaluated.  We can
4387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// evaluate the expression regardless of what the RHS is, but C only allows
4487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// certain things in certain situations.
4587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstruct EvalInfo {
4687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  ASTContext &Ctx;
471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4854da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson  /// EvalResult - Contains information about the evaluation.
4954da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson  Expr::EvalResult &EvalResult;
50f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson
5142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  EvalInfo(ASTContext &ctx, Expr::EvalResult& evalresult)
5242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    : Ctx(ctx), EvalResult(evalresult) {}
5387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner};
5487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner
55f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallnamespace {
56f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  struct ComplexValue {
57f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  private:
58f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    bool IsInt;
59f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
60f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  public:
61f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APSInt IntReal, IntImag;
62f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APFloat FloatReal, FloatImag;
63f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
64f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue() : FloatReal(APFloat::Bogus), FloatImag(APFloat::Bogus) {}
65f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
66f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    void makeComplexFloat() { IsInt = false; }
67f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    bool isComplexFloat() const { return !IsInt; }
68f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APFloat &getComplexFloatReal() { return FloatReal; }
69f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APFloat &getComplexFloatImag() { return FloatImag; }
70f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
71f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    void makeComplexInt() { IsInt = true; }
72f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    bool isComplexInt() const { return IsInt; }
73f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APSInt &getComplexIntReal() { return IntReal; }
74f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APSInt &getComplexIntImag() { return IntImag; }
75f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
76f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    void moveInto(APValue &v) {
77f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (isComplexFloat())
78f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        v = APValue(FloatReal, FloatImag);
79f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      else
80f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        v = APValue(IntReal, IntImag);
81f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    }
82f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  };
83efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
84efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  struct LValue {
85efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Expr *Base;
86efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    CharUnits Offset;
87efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
88efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Expr *getLValueBase() { return Base; }
89efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    CharUnits getLValueOffset() { return Offset; }
90efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
91efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    void moveInto(APValue &v) {
92efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      v = APValue(Base, Offset);
93efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    }
94efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  };
95f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall}
9687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner
97efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluateLValue(const Expr *E, LValue &Result, EvalInfo &Info);
98efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluatePointer(const Expr *E, LValue &Result, EvalInfo &Info);
9987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstatic bool EvaluateInteger(const Expr *E, APSInt  &Result, EvalInfo &Info);
100d9becd1846e2c72bf6ad283faa1b048f33dd3afeChris Lattnerstatic bool EvaluateIntegerOrLValue(const Expr *E, APValue  &Result,
101d9becd1846e2c72bf6ad283faa1b048f33dd3afeChris Lattner                                    EvalInfo &Info);
102d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanstatic bool EvaluateFloat(const Expr *E, APFloat &Result, EvalInfo &Info);
103f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallstatic bool EvaluateComplex(const Expr *E, ComplexValue &Res, EvalInfo &Info);
104f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
105f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
1064efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman// Misc utilities
1074efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
1084efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
109e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnarastatic bool IsGlobalLValue(const Expr* E) {
11042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!E) return true;
11142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
11242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
11342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (isa<FunctionDecl>(DRE->getDecl()))
11442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return true;
11542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
11642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return VD->hasGlobalStorage();
11742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
11842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  }
11942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
12042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (const CompoundLiteralExpr *CLE = dyn_cast<CompoundLiteralExpr>(E))
12142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return CLE->isFileScope();
12242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
12342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return true;
12442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
12542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
126efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvalPointerValueAsBool(LValue& Value, bool& Result) {
127efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  const Expr* Base = Value.Base;
128a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1293554283157190e67918fad4221a5e6faf9317362John McCall  // A null base expression indicates a null pointer.  These are always
1303554283157190e67918fad4221a5e6faf9317362John McCall  // evaluatable, and they are false unless the offset is zero.
1313554283157190e67918fad4221a5e6faf9317362John McCall  if (!Base) {
1323554283157190e67918fad4221a5e6faf9317362John McCall    Result = !Value.Offset.isZero();
1333554283157190e67918fad4221a5e6faf9317362John McCall    return true;
1343554283157190e67918fad4221a5e6faf9317362John McCall  }
1353554283157190e67918fad4221a5e6faf9317362John McCall
13642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // Require the base expression to be a global l-value.
137e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  if (!IsGlobalLValue(Base)) return false;
13842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
1393554283157190e67918fad4221a5e6faf9317362John McCall  // We have a non-null base expression.  These are generally known to
1403554283157190e67918fad4221a5e6faf9317362John McCall  // be true, but if it'a decl-ref to a weak symbol it can be null at
1413554283157190e67918fad4221a5e6faf9317362John McCall  // runtime.
1423554283157190e67918fad4221a5e6faf9317362John McCall  Result = true;
143a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1443554283157190e67918fad4221a5e6faf9317362John McCall  const DeclRefExpr* DeclRef = dyn_cast<DeclRefExpr>(Base);
145a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola  if (!DeclRef)
146a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola    return true;
147a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1483554283157190e67918fad4221a5e6faf9317362John McCall  // If it's a weak symbol, it isn't constant-evaluable.
149a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola  const ValueDecl* Decl = DeclRef->getDecl();
150a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola  if (Decl->hasAttr<WeakAttr>() ||
151a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola      Decl->hasAttr<WeakRefAttr>() ||
152a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola      Decl->hasAttr<WeakImportAttr>())
153a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola    return false;
154a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1555bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman  return true;
1565bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman}
1575bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman
158cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCallstatic bool HandleConversionToBool(const Expr* E, bool& Result,
159cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall                                   EvalInfo &Info) {
1602ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (E->getType()->isIntegralOrEnumerationType()) {
1614efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APSInt IntResult;
1624efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluateInteger(E, IntResult, Info))
1634efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1644efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Result = IntResult != 0;
1654efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
1664efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  } else if (E->getType()->isRealFloatingType()) {
1674efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APFloat FloatResult(0.0);
1684efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluateFloat(E, FloatResult, Info))
1694efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1704efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Result = !FloatResult.isZero();
1714efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
172a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman  } else if (E->getType()->hasPointerRepresentation()) {
173efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue PointerResult;
1744efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluatePointer(E, PointerResult, Info))
1754efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1765bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman    return EvalPointerValueAsBool(PointerResult, Result);
177a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman  } else if (E->getType()->isAnyComplexType()) {
178f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue ComplexResult;
179a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    if (!EvaluateComplex(E, ComplexResult, Info))
180a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman      return false;
181a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    if (ComplexResult.isComplexFloat()) {
182a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman      Result = !ComplexResult.getComplexFloatReal().isZero() ||
183a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman               !ComplexResult.getComplexFloatImag().isZero();
184a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    } else {
185a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman      Result = ComplexResult.getComplexIntReal().getBoolValue() ||
186a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman               ComplexResult.getComplexIntImag().getBoolValue();
187a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    }
188a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    return true;
1894efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
1904efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1914efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return false;
1924efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
1934efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APSInt HandleFloatToIntCast(QualType DestType, QualType SrcType,
195a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                   APFloat &Value, ASTContext &Ctx) {
196a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  unsigned DestWidth = Ctx.getIntWidth(DestType);
197a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // Determine whether we are converting to unsigned or signed.
198a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool DestSigned = DestType->isSignedIntegerType();
1991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
200a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // FIXME: Warning for overflow.
201a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  uint64_t Space[4];
202a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool ignored;
203a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  (void)Value.convertToInteger(Space, DestWidth, DestSigned,
204a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                               llvm::APFloat::rmTowardZero, &ignored);
205a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return APSInt(llvm::APInt(DestWidth, 4, Space), !DestSigned);
206a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
207a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APFloat HandleFloatToFloatCast(QualType DestType, QualType SrcType,
209a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                      APFloat &Value, ASTContext &Ctx) {
210a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool ignored;
211a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APFloat Result = Value;
2121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Result.convert(Ctx.getFloatTypeSemantics(DestType),
213a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                 APFloat::rmNearestTiesToEven, &ignored);
214a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
215a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
216a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APSInt HandleIntToIntCast(QualType DestType, QualType SrcType,
218a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                 APSInt &Value, ASTContext &Ctx) {
219a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  unsigned DestWidth = Ctx.getIntWidth(DestType);
220a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APSInt Result = Value;
221a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // Figure out if this is a truncate, extend or noop cast.
222a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // If the input is signed, do a sign extend, noop, or truncate.
2239f71a8f4c7a182a5236da9e747d57cc1d1bd24c2Jay Foad  Result = Result.extOrTrunc(DestWidth);
224a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.setIsUnsigned(DestType->isUnsignedIntegerType());
225a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
226a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
227a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APFloat HandleIntToFloatCast(QualType DestType, QualType SrcType,
229a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                    APSInt &Value, ASTContext &Ctx) {
230a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
231a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APFloat Result(Ctx.getFloatTypeSemantics(DestType), 1);
232a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.convertFromAPInt(Value, Value.isSigned(),
233a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                          APFloat::rmNearestTiesToEven);
234a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
235a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
236a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
237c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stumpnamespace {
238770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass HasSideEffect
239c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  : public StmtVisitor<HasSideEffect, bool> {
240c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  EvalInfo &Info;
241c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stumppublic:
242c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
243c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  HasSideEffect(EvalInfo &info) : Info(info) {}
244c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
245c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  // Unhandled nodes conservatively default to having side effects.
246c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitStmt(Stmt *S) {
247c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    return true;
248c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  }
249c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
250c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
251c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitDeclRefExpr(DeclRefExpr *E) {
252df317bf71653eeb235da8337b1e8e790f9653aa4Mike Stump    if (Info.Ctx.getCanonicalType(E->getType()).isVolatileQualified())
253c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump      return true;
254c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    return false;
255c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  }
256c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  // We don't want to evaluate BlockExprs multiple times, as they generate
257c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  // a ton of code.
258c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitBlockExpr(BlockExpr *E) { return true; }
259c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitPredefinedExpr(PredefinedExpr *E) { return false; }
260c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitCompoundLiteralExpr(CompoundLiteralExpr *E)
261c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    { return Visit(E->getInitializer()); }
262c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitMemberExpr(MemberExpr *E) { return Visit(E->getBase()); }
263c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitIntegerLiteral(IntegerLiteral *E) { return false; }
264c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitFloatingLiteral(FloatingLiteral *E) { return false; }
265c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitStringLiteral(StringLiteral *E) { return false; }
266c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitCharacterLiteral(CharacterLiteral *E) { return false; }
267c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) { return false; }
268c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitArraySubscriptExpr(ArraySubscriptExpr *E)
269980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump    { return Visit(E->getLHS()) || Visit(E->getRHS()); }
270c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitChooseExpr(ChooseExpr *E)
271c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
272c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitCastExpr(CastExpr *E) { return Visit(E->getSubExpr()); }
273c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitBinAssign(BinaryOperator *E) { return true; }
2743f0147e161df4725ff15fbb731f4f727afcc229fMike Stump  bool VisitCompoundAssignOperator(BinaryOperator *E) { return true; }
275980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump  bool VisitBinaryOperator(BinaryOperator *E)
276980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump  { return Visit(E->getLHS()) || Visit(E->getRHS()); }
277c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPreInc(UnaryOperator *E) { return true; }
278c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPostInc(UnaryOperator *E) { return true; }
279c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPreDec(UnaryOperator *E) { return true; }
280c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPostDec(UnaryOperator *E) { return true; }
281c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryDeref(UnaryOperator *E) {
282df317bf71653eeb235da8337b1e8e790f9653aa4Mike Stump    if (Info.Ctx.getCanonicalType(E->getType()).isVolatileQualified())
283c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump      return true;
284980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump    return Visit(E->getSubExpr());
285c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  }
286c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryOperator(UnaryOperator *E) { return Visit(E->getSubExpr()); }
287363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner
288363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner  // Has side effects if any element does.
289363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner  bool VisitInitListExpr(InitListExpr *E) {
290363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner    for (unsigned i = 0, e = E->getNumInits(); i != e; ++i)
291363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner      if (Visit(E->getInit(i))) return true;
292363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner    return false;
293363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner  }
294c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump};
295c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
296c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump} // end anonymous namespace
297c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
2984efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
2994efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman// LValue Evaluation
3004efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
3014efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmannamespace {
302770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass LValueExprEvaluator
303efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  : public StmtVisitor<LValueExprEvaluator, bool> {
3044efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  EvalInfo &Info;
305efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValue &Result;
306efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
307efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool Success(Expr *E) {
308efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Base = E;
309efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset = CharUnits::Zero();
310efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return true;
311efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  }
3124efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanpublic:
3131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
314efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValueExprEvaluator(EvalInfo &info, LValue &Result) :
315efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Info(info), Result(Result) {}
3164efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
317efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitStmt(Stmt *S) {
318efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
3194efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
3208ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
321efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
322efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitDeclRefExpr(DeclRefExpr *E);
323efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitPredefinedExpr(PredefinedExpr *E) { return Success(E); }
324efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
325efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitMemberExpr(MemberExpr *E);
326efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitStringLiteral(StringLiteral *E) { return Success(E); }
327efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitObjCEncodeExpr(ObjCEncodeExpr *E) { return Success(E); }
328efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitArraySubscriptExpr(ArraySubscriptExpr *E);
329efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryDeref(UnaryOperator *E);
330efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryExtension(const UnaryOperator *E)
331ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getSubExpr()); }
332efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitChooseExpr(const ChooseExpr *E)
333ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
33426bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson
335efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCastExpr(CastExpr *E) {
33626bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    switch (E->getCastKind()) {
33726bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    default:
338efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return false;
33926bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson
3402de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case CK_NoOp:
34126bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson      return Visit(E->getSubExpr());
34226bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    }
34326bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson  }
344ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  // FIXME: Missing: __real__, __imag__
3454efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman};
3464efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman} // end anonymous namespace
3474efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
348efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluateLValue(const Expr* E, LValue& Result, EvalInfo &Info) {
349efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return LValueExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
3504efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
3514efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
352efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitDeclRefExpr(DeclRefExpr *E) {
35350c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman  if (isa<FunctionDecl>(E->getDecl())) {
354efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return Success(E);
35550c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman  } else if (VarDecl* VD = dyn_cast<VarDecl>(E->getDecl())) {
35650c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman    if (!VD->getType()->isReferenceType())
357efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return Success(E);
358761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth    // Reference parameters can refer to anything even if they have an
359761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth    // "initializer" in the form of a default argument.
360761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth    if (isa<ParmVarDecl>(VD))
361761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth      return false;
362d933a0198f3ccce9c73bf2951625315b911d37bfEli Friedman    // FIXME: Check whether VD might be overridden!
36331310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (const Expr *Init = VD->getAnyInitializer())
364cf3293eaeb3853d12cff47e648bbe835004e929fDouglas Gregor      return Visit(const_cast<Expr *>(Init));
36550c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman  }
36650c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman
367efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
36835873c49adad211ff466e34342a52665742794f5Anders Carlsson}
36935873c49adad211ff466e34342a52665742794f5Anders Carlsson
370efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
371efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return Success(E);
3724efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
3734efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
374efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitMemberExpr(MemberExpr *E) {
3754efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  QualType Ty;
3764efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (E->isArrow()) {
377efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluatePointer(E->getBase(), Result, Info))
378efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return false;
3796217b80b7a1379b74cced1c076338262c3c980b3Ted Kremenek    Ty = E->getBase()->getType()->getAs<PointerType>()->getPointeeType();
3804efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  } else {
381efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!Visit(E->getBase()))
382efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return false;
3834efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Ty = E->getBase()->getType();
3844efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
3854efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
3866217b80b7a1379b74cced1c076338262c3c980b3Ted Kremenek  RecordDecl *RD = Ty->getAs<RecordType>()->getDecl();
3874efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
38886f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor
38986f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor  FieldDecl *FD = dyn_cast<FieldDecl>(E->getMemberDecl());
39086f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor  if (!FD) // FIXME: deal with other kinds of member expressions
391efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
3922be586108bb401019647791feca19ea03fd477ceEli Friedman
3932be586108bb401019647791feca19ea03fd477ceEli Friedman  if (FD->getType()->isReferenceType())
394efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
3952be586108bb401019647791feca19ea03fd477ceEli Friedman
3964efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  // FIXME: This is linear time.
39744b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor  unsigned i = 0;
39817945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis  for (RecordDecl::field_iterator Field = RD->field_begin(),
39917945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis                               FieldEnd = RD->field_end();
40044b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor       Field != FieldEnd; (void)++Field, ++i) {
40144b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor    if (*Field == FD)
4024efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      break;
4034efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
4044efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
405efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  Result.Offset += CharUnits::fromQuantity(RL.getFieldOffset(i) / 8);
406efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return true;
4074efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
4084efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
409efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
4103068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  if (!EvaluatePointer(E->getBase(), Result, Info))
411efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4133068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  APSInt Index;
4143068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  if (!EvaluateInteger(E->getIdx(), Index, Info))
415efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4163068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
417199c3d6cd16aebbb9c7f0d42af9d922c9628bf70Ken Dyck  CharUnits ElementSize = Info.Ctx.getTypeSizeInChars(E->getType());
418efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  Result.Offset += Index.getSExtValue() * ElementSize;
419efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return true;
4203068d117951a8df54bae9db039b56201ab10962bAnders Carlsson}
4214efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
422efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitUnaryDeref(UnaryOperator *E) {
423efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return EvaluatePointer(E->getSubExpr(), Result, Info);
424e8761c8fe2ee6b628104a0885f49fd3c21c08a4fEli Friedman}
425e8761c8fe2ee6b628104a0885f49fd3c21c08a4fEli Friedman
4264efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
427f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner// Pointer Evaluation
428f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
429f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
430c754aa62643e66ab967ca32ae8b0b3fc419bba25Anders Carlssonnamespace {
431770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass PointerExprEvaluator
432efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  : public StmtVisitor<PointerExprEvaluator, bool> {
43387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  EvalInfo &Info;
434efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValue &Result;
435efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
436efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool Success(Expr *E) {
437efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Base = E;
438efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset = CharUnits::Zero();
439efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return true;
440efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  }
4412bad1687fe6f00e10767a691a33b070b151902b6Anders Carlssonpublic:
4421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
443efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  PointerExprEvaluator(EvalInfo &info, LValue &Result)
444efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    : Info(info), Result(Result) {}
445f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
446efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitStmt(Stmt *S) {
447efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4482bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  }
4492bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
450efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
4512bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
452efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitBinaryOperator(const BinaryOperator *E);
453efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCastExpr(CastExpr* E);
454efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryExtension(const UnaryOperator *E)
4552217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman      { return Visit(E->getSubExpr()); }
456efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryAddrOf(const UnaryOperator *E);
457efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitObjCStringLiteral(ObjCStringLiteral *E)
458efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success(E); }
459efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitAddrLabelExpr(AddrLabelExpr *E)
460efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success(E); }
461efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCallExpr(CallExpr *E);
462efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitBlockExpr(BlockExpr *E) {
463b83d287bc7f47d36fb0751a481e2ef9308b37252Mike Stump    if (!E->hasBlockDeclRefExprs())
464efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return Success(E);
465efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
466b83d287bc7f47d36fb0751a481e2ef9308b37252Mike Stump  }
467efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitImplicitValueInitExpr(ImplicitValueInitExpr *E)
468efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success((Expr*)0); }
469efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitConditionalOperator(ConditionalOperator *E);
470efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitChooseExpr(ChooseExpr *E)
4716e8ed16ffef02b82995a90bdcf10ffff7d63839aSebastian Redl      { return Visit(E->getChosenSubExpr(Info.Ctx)); }
472efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E)
473efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success((Expr*)0); }
474ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  // FIXME: Missing: @protocol, @selector
4752bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson};
476f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner} // end anonymous namespace
4772bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
478efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluatePointer(const Expr* E, LValue& Result, EvalInfo &Info) {
4797db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall  assert(E->getType()->hasPointerRepresentation());
480efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return PointerExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
481f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner}
482650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
483efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
4842de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() != BO_Add &&
4852de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      E->getOpcode() != BO_Sub)
486efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
488650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  const Expr *PExp = E->getLHS();
489650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  const Expr *IExp = E->getRHS();
490650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (IExp->getType()->isPointerType())
491f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner    std::swap(PExp, IExp);
4921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
493efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  if (!EvaluatePointer(PExp, Result, Info))
494efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
496efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  llvm::APSInt Offset;
497efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  if (!EvaluateInteger(IExp, Offset, Info))
498efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
499efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  int64_t AdditionalOffset
500efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    = Offset.isSigned() ? Offset.getSExtValue()
501efdb83e26f9a1fd2566afe54461216cd84814d42John McCall                        : static_cast<int64_t>(Offset.getZExtValue());
502650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
503e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar  // Compute the new offset in the appropriate width.
504e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar
505e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar  QualType PointeeType =
506e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar    PExp->getType()->getAs<PointerType>()->getPointeeType();
507efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  CharUnits SizeOfPointee;
5081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5094d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  // Explicitly handle GNU void* and function pointer arithmetic extensions.
5104d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  if (PointeeType->isVoidType() || PointeeType->isFunctionType())
511efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    SizeOfPointee = CharUnits::One();
5124d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  else
513efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    SizeOfPointee = Info.Ctx.getTypeSizeInChars(PointeeType);
5144efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
5152de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == BO_Add)
516efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset += AdditionalOffset * SizeOfPointee;
517650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  else
518efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset -= AdditionalOffset * SizeOfPointee;
5194efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
520efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return true;
521650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson}
5224efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
523efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitUnaryAddrOf(const UnaryOperator *E) {
524efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return EvaluateLValue(E->getSubExpr(), Result, Info);
5254efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
5261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
527650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
528efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitCastExpr(CastExpr* E) {
52909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  Expr* SubExpr = E->getSubExpr();
530650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
53109a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  switch (E->getCastKind()) {
53209a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  default:
53309a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    break;
53409a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
5352de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_NoOp:
5362de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_BitCast:
5372de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_LValueBitCast:
5382de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_AnyPointerToObjCPointerCast:
5392de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_AnyPointerToBlockPointerCast:
54009a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    return Visit(SubExpr);
54109a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
5425c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson  case CK_DerivedToBase:
5435c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson  case CK_UncheckedDerivedToBase: {
5445c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    LValue BaseLV;
5455c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    if (!EvaluatePointer(E->getSubExpr(), BaseLV, Info))
5465c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson      return false;
5475c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson
5485c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    // Now figure out the necessary offset to add to the baseLV to get from
5495c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    // the derived class to the base class.
5505c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    uint64_t Offset = 0;
5515c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson
5525c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    QualType Ty = E->getSubExpr()->getType();
5535c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    const CXXRecordDecl *DerivedDecl =
5545c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson      Ty->getAs<PointerType>()->getPointeeType()->getAsCXXRecordDecl();
5555c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson
5565c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    for (CastExpr::path_const_iterator PathI = E->path_begin(),
5575c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson         PathE = E->path_end(); PathI != PathE; ++PathI) {
5585c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson      const CXXBaseSpecifier *Base = *PathI;
5595c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson
5605c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson      // FIXME: If the base is virtual, we'd need to determine the type of the
5615c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson      // most derived class and we don't support that right now.
5625c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson      if (Base->isVirtual())
5635c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson        return false;
5645c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson
5655c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson      const CXXRecordDecl *BaseDecl = Base->getType()->getAsCXXRecordDecl();
5665c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson      const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(DerivedDecl);
5675c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson
568a14f5979572aa25c03d24750ee4724d2031d4edeAnders Carlsson      Offset += Layout.getBaseClassOffsetInBits(BaseDecl);
5695c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson      DerivedDecl = BaseDecl;
5705c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    }
5715c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson
5725c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    Result.Base = BaseLV.getLValueBase();
5735c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    Result.Offset = BaseLV.getLValueOffset() +
5745c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson      CharUnits::fromQuantity(Offset / Info.Ctx.getCharWidth());
5755c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson    return true;
5765c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson  }
5775c5a764fcd256df6f6cfbce5cdd2a2dfb2c45e95Anders Carlsson
578404cd1669c3ba138a9ae0a619bd689cce5aae271John McCall  case CK_NullToPointer: {
579404cd1669c3ba138a9ae0a619bd689cce5aae271John McCall    Result.Base = 0;
580404cd1669c3ba138a9ae0a619bd689cce5aae271John McCall    Result.Offset = CharUnits::Zero();
581404cd1669c3ba138a9ae0a619bd689cce5aae271John McCall    return true;
582404cd1669c3ba138a9ae0a619bd689cce5aae271John McCall  }
583404cd1669c3ba138a9ae0a619bd689cce5aae271John McCall
5842de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_IntegralToPointer: {
585efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    APValue Value;
586efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateIntegerOrLValue(SubExpr, Value, Info))
58709a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman      break;
58869ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar
589efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (Value.isInt()) {
5909f71a8f4c7a182a5236da9e747d57cc1d1bd24c2Jay Foad      Value.getInt() = Value.getInt().extOrTrunc((unsigned)Info.Ctx.getTypeSize(E->getType()));
591efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Base = 0;
592efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Offset = CharUnits::fromQuantity(Value.getInt().getZExtValue());
593efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return true;
594efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    } else {
595efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      // Cast is of an lvalue, no need to change value.
596efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Base = Value.getLValueBase();
597efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Offset = Value.getLValueOffset();
598efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return true;
599650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson    }
600650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  }
6012de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_ArrayToPointerDecay:
6022de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_FunctionToPointerDecay:
603efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return EvaluateLValue(SubExpr, Result, Info);
6044efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
6054efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
606efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
6071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump}
608650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
609efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitCallExpr(CallExpr *E) {
6101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  if (E->isBuiltinCall(Info.Ctx) ==
6110d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall        Builtin::BI__builtin___CFStringMakeConstantString ||
6120d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall      E->isBuiltinCall(Info.Ctx) ==
6130d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall        Builtin::BI__builtin___NSStringMakeConstantString)
614efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return Success(E);
615efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
6163941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman}
6173941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman
618efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitConditionalOperator(ConditionalOperator *E) {
6194efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  bool BoolResult;
6204efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (!HandleConversionToBool(E->getCond(), BoolResult, Info))
621efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
6224efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
6234efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Expr* EvalExpr = BoolResult ? E->getTrueExpr() : E->getFalseExpr();
624efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return Visit(EvalExpr);
6254efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
626f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
627f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
62859b5da6d853b4368b984700315adf7b37de05764Nate Begeman// Vector Evaluation
62959b5da6d853b4368b984700315adf7b37de05764Nate Begeman//===----------------------------------------------------------------------===//
63059b5da6d853b4368b984700315adf7b37de05764Nate Begeman
63159b5da6d853b4368b984700315adf7b37de05764Nate Begemannamespace {
632770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramer  class VectorExprEvaluator
63359b5da6d853b4368b984700315adf7b37de05764Nate Begeman  : public StmtVisitor<VectorExprEvaluator, APValue> {
63459b5da6d853b4368b984700315adf7b37de05764Nate Begeman    EvalInfo &Info;
63591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue GetZeroVector(QualType VecType);
63659b5da6d853b4368b984700315adf7b37de05764Nate Begeman  public:
6371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
63859b5da6d853b4368b984700315adf7b37de05764Nate Begeman    VectorExprEvaluator(EvalInfo &info) : Info(info) {}
6391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
64059b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitStmt(Stmt *S) {
64159b5da6d853b4368b984700315adf7b37de05764Nate Begeman      return APValue();
64259b5da6d853b4368b984700315adf7b37de05764Nate Begeman    }
6431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
64491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitParenExpr(ParenExpr *E)
64591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman        { return Visit(E->getSubExpr()); }
64691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryExtension(const UnaryOperator *E)
64791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return Visit(E->getSubExpr()); }
64891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryPlus(const UnaryOperator *E)
64991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return Visit(E->getSubExpr()); }
65091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryReal(const UnaryOperator *E)
65191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return Visit(E->getSubExpr()); }
65291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E)
65391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return GetZeroVector(E->getType()); }
65459b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitCastExpr(const CastExpr* E);
65559b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitCompoundLiteralExpr(const CompoundLiteralExpr *E);
65659b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitInitListExpr(const InitListExpr *E);
65791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitConditionalOperator(const ConditionalOperator *E);
658ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    APValue VisitChooseExpr(const ChooseExpr *E)
659ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman      { return Visit(E->getChosenSubExpr(Info.Ctx)); }
66091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryImag(const UnaryOperator *E);
66191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    // FIXME: Missing: unary -, unary ~, binary add/sub/mul/div,
6622217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman    //                 binary comparisons, binary and/or/xor,
66391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    //                 shufflevector, ExtVectorElementExpr
66491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    //        (Note that these require implementing conversions
66591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    //         between vector types.)
66659b5da6d853b4368b984700315adf7b37de05764Nate Begeman  };
66759b5da6d853b4368b984700315adf7b37de05764Nate Begeman} // end anonymous namespace
66859b5da6d853b4368b984700315adf7b37de05764Nate Begeman
66959b5da6d853b4368b984700315adf7b37de05764Nate Begemanstatic bool EvaluateVector(const Expr* E, APValue& Result, EvalInfo &Info) {
67059b5da6d853b4368b984700315adf7b37de05764Nate Begeman  if (!E->getType()->isVectorType())
67159b5da6d853b4368b984700315adf7b37de05764Nate Begeman    return false;
67259b5da6d853b4368b984700315adf7b37de05764Nate Begeman  Result = VectorExprEvaluator(Info).Visit(const_cast<Expr*>(E));
67359b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return !Result.isUninit();
67459b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
67559b5da6d853b4368b984700315adf7b37de05764Nate Begeman
67659b5da6d853b4368b984700315adf7b37de05764Nate BegemanAPValue VectorExprEvaluator::VisitCastExpr(const CastExpr* E) {
677183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall  const VectorType *VTy = E->getType()->getAs<VectorType>();
678c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  QualType EltTy = VTy->getElementType();
679c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  unsigned NElts = VTy->getNumElements();
680c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  unsigned EltWidth = Info.Ctx.getTypeSize(EltTy);
6811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
68259b5da6d853b4368b984700315adf7b37de05764Nate Begeman  const Expr* SE = E->getSubExpr();
683e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  QualType SETy = SE->getType();
684e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  APValue Result = APValue();
68559b5da6d853b4368b984700315adf7b37de05764Nate Begeman
686e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  // Check for vector->vector bitcast and scalar->vector splat.
687e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  if (SETy->isVectorType()) {
68859b5da6d853b4368b984700315adf7b37de05764Nate Begeman    return this->Visit(const_cast<Expr*>(SE));
689e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  } else if (SETy->isIntegerType()) {
690e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman    APSInt IntResult;
691d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    if (!EvaluateInteger(SE, IntResult, Info))
692d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar      return APValue();
693d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    Result = APValue(IntResult);
694e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  } else if (SETy->isRealFloatingType()) {
695e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman    APFloat F(0.0);
696d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    if (!EvaluateFloat(SE, F, Info))
697d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar      return APValue();
698d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    Result = APValue(F);
699d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar  } else
700c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    return APValue();
70159b5da6d853b4368b984700315adf7b37de05764Nate Begeman
702c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // For casts of a scalar to ExtVector, convert the scalar to the element type
703c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // and splat it to all elements.
704c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  if (E->getType()->isExtVectorType()) {
705c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    if (EltTy->isIntegerType() && Result.isInt())
706c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleIntToIntCast(EltTy, SETy, Result.getInt(),
707c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                          Info.Ctx));
708c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isIntegerType())
709c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleFloatToIntCast(EltTy, SETy, Result.getFloat(),
710c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                            Info.Ctx));
711c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isRealFloatingType() && Result.isInt())
712c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleIntToFloatCast(EltTy, SETy, Result.getInt(),
713c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                            Info.Ctx));
714c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isRealFloatingType())
715c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleFloatToFloatCast(EltTy, SETy, Result.getFloat(),
716c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                              Info.Ctx));
717c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else
718c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      return APValue();
719c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman
720c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    // Splat and create vector APValue.
721c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    llvm::SmallVector<APValue, 4> Elts(NElts, Result);
722c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    return APValue(&Elts[0], Elts.size());
723e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  }
724c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman
725c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // For casts of a scalar to regular gcc-style vector type, bitcast the scalar
726c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // to the vector. To construct the APValue vector initializer, bitcast the
727c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // initializing value to an APInt, and shift out the bits pertaining to each
728c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // element.
729c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  APSInt Init;
730c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  Init = Result.isInt() ? Result.getInt() : Result.getFloat().bitcastToAPInt();
7311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
732c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  llvm::SmallVector<APValue, 4> Elts;
733c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  for (unsigned i = 0; i != NElts; ++i) {
7349f71a8f4c7a182a5236da9e747d57cc1d1bd24c2Jay Foad    APSInt Tmp = Init.extOrTrunc(EltWidth);
7351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
736c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    if (EltTy->isIntegerType())
737c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Elts.push_back(APValue(Tmp));
738c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isRealFloatingType())
739c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Elts.push_back(APValue(APFloat(Tmp)));
740c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else
741c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      return APValue();
742c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman
743c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    Init >>= EltWidth;
744c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  }
745c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  return APValue(&Elts[0], Elts.size());
74659b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
74759b5da6d853b4368b984700315adf7b37de05764Nate Begeman
7481eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpAPValue
74959b5da6d853b4368b984700315adf7b37de05764Nate BegemanVectorExprEvaluator::VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
75059b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return this->Visit(const_cast<Expr*>(E->getInitializer()));
75159b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
75259b5da6d853b4368b984700315adf7b37de05764Nate Begeman
7531eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpAPValue
75459b5da6d853b4368b984700315adf7b37de05764Nate BegemanVectorExprEvaluator::VisitInitListExpr(const InitListExpr *E) {
755183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall  const VectorType *VT = E->getType()->getAs<VectorType>();
75659b5da6d853b4368b984700315adf7b37de05764Nate Begeman  unsigned NumInits = E->getNumInits();
75791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  unsigned NumElements = VT->getNumElements();
7581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
75959b5da6d853b4368b984700315adf7b37de05764Nate Begeman  QualType EltTy = VT->getElementType();
76059b5da6d853b4368b984700315adf7b37de05764Nate Begeman  llvm::SmallVector<APValue, 4> Elements;
76159b5da6d853b4368b984700315adf7b37de05764Nate Begeman
762a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  // If a vector is initialized with a single element, that value
763a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  // becomes every element of the vector, not just the first.
764a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  // This is the behavior described in the IBM AltiVec documentation.
765a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  if (NumInits == 1) {
766a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    APValue InitValue;
76759b5da6d853b4368b984700315adf7b37de05764Nate Begeman    if (EltTy->isIntegerType()) {
76859b5da6d853b4368b984700315adf7b37de05764Nate Begeman      llvm::APSInt sInt(32);
769a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      if (!EvaluateInteger(E->getInit(0), sInt, Info))
770a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        return APValue();
771a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      InitValue = APValue(sInt);
77259b5da6d853b4368b984700315adf7b37de05764Nate Begeman    } else {
77359b5da6d853b4368b984700315adf7b37de05764Nate Begeman      llvm::APFloat f(0.0);
774a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      if (!EvaluateFloat(E->getInit(0), f, Info))
775a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        return APValue();
776a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      InitValue = APValue(f);
777a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    }
778a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    for (unsigned i = 0; i < NumElements; i++) {
779a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      Elements.push_back(InitValue);
780a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    }
781a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  } else {
782a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    for (unsigned i = 0; i < NumElements; i++) {
783a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      if (EltTy->isIntegerType()) {
784a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        llvm::APSInt sInt(32);
785a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        if (i < NumInits) {
786a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          if (!EvaluateInteger(E->getInit(i), sInt, Info))
787a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall            return APValue();
788a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        } else {
789a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          sInt = Info.Ctx.MakeIntValue(0, EltTy);
790a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        }
791a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        Elements.push_back(APValue(sInt));
79291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      } else {
793a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        llvm::APFloat f(0.0);
794a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        if (i < NumInits) {
795a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          if (!EvaluateFloat(E->getInit(i), f, Info))
796a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall            return APValue();
797a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        } else {
798a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          f = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(EltTy));
799a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        }
800a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        Elements.push_back(APValue(f));
80191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      }
80259b5da6d853b4368b984700315adf7b37de05764Nate Begeman    }
80359b5da6d853b4368b984700315adf7b37de05764Nate Begeman  }
80459b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return APValue(&Elements[0], Elements.size());
80559b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
80659b5da6d853b4368b984700315adf7b37de05764Nate Begeman
8071eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpAPValue
80891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli FriedmanVectorExprEvaluator::GetZeroVector(QualType T) {
809183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall  const VectorType *VT = T->getAs<VectorType>();
81091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  QualType EltTy = VT->getElementType();
81191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  APValue ZeroElement;
81291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (EltTy->isIntegerType())
81391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    ZeroElement = APValue(Info.Ctx.MakeIntValue(0, EltTy));
81491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  else
81591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    ZeroElement =
81691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman        APValue(APFloat::getZero(Info.Ctx.getFloatTypeSemantics(EltTy)));
81791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
81891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  llvm::SmallVector<APValue, 4> Elements(VT->getNumElements(), ZeroElement);
81991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  return APValue(&Elements[0], Elements.size());
82091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman}
82191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
82291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli FriedmanAPValue VectorExprEvaluator::VisitConditionalOperator(const ConditionalOperator *E) {
82391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  bool BoolResult;
82491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (!HandleConversionToBool(E->getCond(), BoolResult, Info))
82591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    return APValue();
82691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
82791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  Expr* EvalExpr = BoolResult ? E->getTrueExpr() : E->getFalseExpr();
82891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
82991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  APValue Result;
83091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (EvaluateVector(EvalExpr, Result, Info))
83191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    return Result;
83291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  return APValue();
83391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman}
83491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
83591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli FriedmanAPValue VectorExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
83691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
83791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    Info.EvalResult.HasSideEffects = true;
83891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  return GetZeroVector(E->getType());
83991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman}
84091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
84159b5da6d853b4368b984700315adf7b37de05764Nate Begeman//===----------------------------------------------------------------------===//
842f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner// Integer Evaluation
843f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
844f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
845f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnernamespace {
846770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass IntExprEvaluator
847b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  : public StmtVisitor<IntExprEvaluator, bool> {
84887eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  EvalInfo &Info;
84930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  APValue &Result;
850f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnerpublic:
85130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  IntExprEvaluator(EvalInfo &info, APValue &result)
85287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    : Info(info), Result(result) {}
853f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
8543f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  bool Success(const llvm::APSInt &SI, const Expr *E) {
8552ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    assert(E->getType()->isIntegralOrEnumerationType() &&
8562ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor           "Invalid evaluation result.");
85730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(SI.isSigned() == E->getType()->isSignedIntegerType() &&
8583f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
85930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(SI.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
8603f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
86130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(SI);
8623f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
8633f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
8643f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar
865131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  bool Success(const llvm::APInt &I, const Expr *E) {
8662ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    assert(E->getType()->isIntegralOrEnumerationType() &&
8672ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor           "Invalid evaluation result.");
86830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(I.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
8693f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
87030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(APSInt(I));
87130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result.getInt().setIsUnsigned(E->getType()->isUnsignedIntegerType());
872131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return true;
873131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  }
874131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
875131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  bool Success(uint64_t Value, const Expr *E) {
8762ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    assert(E->getType()->isIntegralOrEnumerationType() &&
8772ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor           "Invalid evaluation result.");
87830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(Info.Ctx.MakeIntValue(Value, E->getType()));
879131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return true;
880131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  }
881131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
88282206e267ce6cc709797127616f64672d255b310Anders Carlsson  bool Error(SourceLocation L, diag::kind D, const Expr *E) {
88332fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    // Take the first error.
88454da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson    if (Info.EvalResult.Diag == 0) {
88554da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson      Info.EvalResult.DiagLoc = L;
88654da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson      Info.EvalResult.Diag = D;
88782206e267ce6cc709797127616f64672d255b310Anders Carlsson      Info.EvalResult.DiagExpr = E;
88832fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    }
88954176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    return false;
8907a76778e218da57a3b10c80066a0a938f28987b6Chris Lattner  }
8911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
892f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //===--------------------------------------------------------------------===//
893f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //                            Visitor Methods
894f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //===--------------------------------------------------------------------===//
8951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
89632fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  bool VisitStmt(Stmt *) {
89732fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    assert(0 && "This should be called on integers, stmts are not integers");
89832fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    return false;
89932fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  }
9001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
90132fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  bool VisitExpr(Expr *E) {
9020e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
903f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  }
9041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
905b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
906f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
9074c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitIntegerLiteral(const IntegerLiteral *E) {
908131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
9094c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
9104c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitCharacterLiteral(const CharacterLiteral *E) {
911131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
9124c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
913043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman
914043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  bool CheckReferencedDecl(const Expr *E, const Decl *D);
915043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  bool VisitDeclRefExpr(const DeclRefExpr *E) {
916043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    return CheckReferencedDecl(E, E->getDecl());
917043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  }
918043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  bool VisitMemberExpr(const MemberExpr *E) {
919043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    if (CheckReferencedDecl(E, E->getMemberDecl())) {
920043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman      // Conservatively assume a MemberExpr will have side-effects
921043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman      Info.EvalResult.HasSideEffects = true;
922043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman      return true;
923043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    }
924043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    return false;
925043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  }
926043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman
927c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman  bool VisitCallExpr(CallExpr *E);
928b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitBinaryOperator(const BinaryOperator *E);
9298ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  bool VisitOffsetOfExpr(const OffsetOfExpr *E);
930b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitUnaryOperator(const UnaryOperator *E);
931ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes  bool VisitConditionalOperator(const ConditionalOperator *E);
932f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
933a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool VisitCastExpr(CastExpr* E);
9340518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl  bool VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E);
9350518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl
9363068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  bool VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
937131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
9383068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
9391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9403f70456b8adb0405ef2a47d51f9fc2d5937ae8aeAnders Carlsson  bool VisitGNUNullExpr(const GNUNullExpr *E) {
941131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(0, E);
9423f70456b8adb0405ef2a47d51f9fc2d5937ae8aeAnders Carlsson  }
9431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
944ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  bool VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E) {
945131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(0, E);
9463068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
9473068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
948664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  bool VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E) {
949664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman    return Success(0, E);
950664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  }
951664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman
95264b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl  bool VisitUnaryTypeTraitExpr(const UnaryTypeTraitExpr *E) {
9530dfd848fa4c9664852ba8c929a8bd3fce93ddca2Sebastian Redl    return Success(E->getValue(), E);
95464b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl  }
95564b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl
9566ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  bool VisitBinaryTypeTraitExpr(const BinaryTypeTraitExpr *E) {
9576ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet    return Success(E->getValue(), E);
9586ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  }
9596ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
960ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitChooseExpr(const ChooseExpr *E) {
961ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    return Visit(E->getChosenSubExpr(Info.Ctx));
962ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  }
963ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman
964722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  bool VisitUnaryReal(const UnaryOperator *E);
965664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  bool VisitUnaryImag(const UnaryOperator *E);
966664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman
967295995c9c3196416372c9cd35d9cedb6da37bd3dSebastian Redl  bool VisitCXXNoexceptExpr(const CXXNoexceptExpr *E);
968295995c9c3196416372c9cd35d9cedb6da37bd3dSebastian Redl
969fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattnerprivate:
9708b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck  CharUnits GetAlignOfExpr(const Expr *E);
9718b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck  CharUnits GetAlignOfType(QualType T);
97242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  static QualType GetObjectType(const Expr *E);
97342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  bool TryEvaluateBuiltinObjectSize(CallExpr *E);
974664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  // FIXME: Missing: array subscript of vector, member of vector
975f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner};
976f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner} // end anonymous namespace
977f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
97869ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbarstatic bool EvaluateIntegerOrLValue(const Expr* E, APValue &Result, EvalInfo &Info) {
9792ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  assert(E->getType()->isIntegralOrEnumerationType());
98069ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar  return IntExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
98169ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar}
98269ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar
98369ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbarstatic bool EvaluateInteger(const Expr* E, APSInt &Result, EvalInfo &Info) {
9842ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  assert(E->getType()->isIntegralOrEnumerationType());
9857db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall
98630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  APValue Val;
98769ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar  if (!EvaluateIntegerOrLValue(E, Val, Info) || !Val.isInt())
98830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return false;
98930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  Result = Val.getInt();
99030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  return true;
991f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner}
992f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
993043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedmanbool IntExprEvaluator::CheckReferencedDecl(const Expr* E, const Decl* D) {
9944c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Enums are integer constant exprs.
99529a7f3342c3c6dd15d914c61ae22246c36d51ce7Eli Friedman  if (const EnumConstantDecl *ECD = dyn_cast<EnumConstantDecl>(D))
99629a7f3342c3c6dd15d914c61ae22246c36d51ce7Eli Friedman    return Success(ECD->getInitVal(), E);
997b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl
998b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl  // In C++, const, non-volatile integers initialized with ICEs are ICEs.
999e1646da3bb0fe97b372e5fe8cefc537b22048fc4Eli Friedman  // In C, they can also be folded, although they are not ICEs.
1000cf3293eaeb3853d12cff47e648bbe835004e929fDouglas Gregor  if (Info.Ctx.getCanonicalType(E->getType()).getCVRQualifiers()
1001cf3293eaeb3853d12cff47e648bbe835004e929fDouglas Gregor                                                        == Qualifiers::Const) {
1002f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson
1003f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson    if (isa<ParmVarDecl>(D))
1004f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson      return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
1005f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson
1006043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
100731310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl      if (const Expr *Init = VD->getAnyInitializer()) {
1008c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        if (APValue *V = VD->getEvaluatedValue()) {
1009c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          if (V->isInt())
1010c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman            return Success(V->getInt(), E);
1011c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
1012c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        }
1013c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman
1014c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        if (VD->isEvaluatingValue())
1015c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
1016c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman
1017c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        VD->setEvaluatingValue();
1018c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman
1019a7dedf7004e8ac3f947d6817370b2778fa648c2bEli Friedman        Expr::EvalResult EResult;
1020a7dedf7004e8ac3f947d6817370b2778fa648c2bEli Friedman        if (Init->Evaluate(EResult, Info.Ctx) && !EResult.HasSideEffects &&
1021a7dedf7004e8ac3f947d6817370b2778fa648c2bEli Friedman            EResult.Val.isInt()) {
102278d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor          // Cache the evaluated value in the variable declaration.
1023a7dedf7004e8ac3f947d6817370b2778fa648c2bEli Friedman          Result = EResult.Val;
1024c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          VD->setEvaluatedValue(Result);
102578d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor          return true;
102678d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor        }
102778d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor
1028c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        VD->setEvaluatedValue(APValue());
102978d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor      }
1030b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl    }
1031b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl  }
1032b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl
10334c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Otherwise, random variable references are not constants.
10340e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson  return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
10354c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner}
10364c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner
1037a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner/// EvaluateBuiltinClassifyType - Evaluate __builtin_classify_type the same way
1038a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner/// as GCC.
1039a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattnerstatic int EvaluateBuiltinClassifyType(const CallExpr *E) {
1040a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // The following enum mimics the values returned by GCC.
10417c80bd64032e610c0dbd74fc0ef6ea334447f2fdSebastian Redl  // FIXME: Does GCC differ between lvalue and rvalue references here?
1042a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  enum gcc_type_class {
1043a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    no_type_class = -1,
1044a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    void_type_class, integer_type_class, char_type_class,
1045a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    enumeral_type_class, boolean_type_class,
1046a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    pointer_type_class, reference_type_class, offset_type_class,
1047a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    real_type_class, complex_type_class,
1048a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    function_type_class, method_type_class,
1049a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    record_type_class, union_type_class,
1050a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    array_type_class, string_type_class,
1051a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    lang_type_class
1052a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  };
10531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // If no argument was supplied, default to "no_type_class". This isn't
1055a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // ideal, however it is what gcc does.
1056a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  if (E->getNumArgs() == 0)
1057a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return no_type_class;
10581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1059a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  QualType ArgTy = E->getArg(0)->getType();
1060a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  if (ArgTy->isVoidType())
1061a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return void_type_class;
1062a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isEnumeralType())
1063a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return enumeral_type_class;
1064a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isBooleanType())
1065a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return boolean_type_class;
1066a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isCharType())
1067a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return string_type_class; // gcc doesn't appear to use char_type_class
1068a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isIntegerType())
1069a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return integer_type_class;
1070a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isPointerType())
1071a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return pointer_type_class;
1072a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isReferenceType())
1073a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return reference_type_class;
1074a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isRealType())
1075a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return real_type_class;
1076a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isComplexType())
1077a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return complex_type_class;
1078a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isFunctionType())
1079a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return function_type_class;
1080fb87b89fc9eb103e19fb8e4b925c23f0bd091b99Douglas Gregor  else if (ArgTy->isStructureOrClassType())
1081a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return record_type_class;
1082a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isUnionType())
1083a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return union_type_class;
1084a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isArrayType())
1085a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return array_type_class;
1086a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isUnionType())
1087a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return union_type_class;
1088a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else  // FIXME: offset_type_class, method_type_class, & lang_type_class?
1089a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    assert(0 && "CallExpr::isBuiltinClassifyType(): unimplemented type");
1090a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  return -1;
1091a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner}
1092a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner
109342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// Retrieves the "underlying object type" of the given expression,
109442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// as used by __builtin_object_size.
109542c8f87eb60958170c46767273bf93e6c96125bfJohn McCallQualType IntExprEvaluator::GetObjectType(const Expr *E) {
109642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
109742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
109842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return VD->getType();
109942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  } else if (isa<CompoundLiteralExpr>(E)) {
110042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return E->getType();
110142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  }
110242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
110342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return QualType();
110442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
110542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
110642c8f87eb60958170c46767273bf93e6c96125bfJohn McCallbool IntExprEvaluator::TryEvaluateBuiltinObjectSize(CallExpr *E) {
110742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // TODO: Perhaps we should let LLVM lower this?
110842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  LValue Base;
110942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!EvaluatePointer(E->getArg(0), Base, Info))
111042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
111142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
111242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // If we can prove the base is null, lower to zero now.
111342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  const Expr *LVBase = Base.getLValueBase();
111442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!LVBase) return Success(0, E);
111542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
111642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  QualType T = GetObjectType(LVBase);
111742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (T.isNull() ||
111842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isIncompleteType() ||
11191357869bc5983cdfbc986db1f3d18265bb34cb0eEli Friedman      T->isFunctionType() ||
112042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isVariablyModifiedType() ||
112142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isDependentType())
112242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
112342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
112442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  CharUnits Size = Info.Ctx.getTypeSizeInChars(T);
112542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  CharUnits Offset = Base.getLValueOffset();
112642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
112742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!Offset.isNegative() && Offset <= Size)
112842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    Size -= Offset;
112942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  else
113042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    Size = CharUnits::Zero();
113142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return Success(Size.getQuantity(), E);
113242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
113342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
1134c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedmanbool IntExprEvaluator::VisitCallExpr(CallExpr *E) {
11353c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  switch (E->isBuiltinCall(Info.Ctx)) {
1136019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  default:
11370e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
113864eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
113964eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump  case Builtin::BI__builtin_object_size: {
114042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (TryEvaluateBuiltinObjectSize(E))
114142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return true;
114264eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
1143b2aaf51ed73a4774322a39a0dd59d0fe7e3258c0Eric Christopher    // If evaluating the argument has side-effects we can't determine
1144b2aaf51ed73a4774322a39a0dd59d0fe7e3258c0Eric Christopher    // the size of the object and lower it to unknown now.
1145393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian    if (E->getArg(0)->HasSideEffects(Info.Ctx)) {
11463f27b384801de26ce7efaa395699b42719372f24Benjamin Kramer      if (E->getArg(1)->EvaluateAsInt(Info.Ctx).getZExtValue() <= 1)
1147cf184655319cf7a5b811067cff9d26a5741fd161Chris Lattner        return Success(-1ULL, E);
114864eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump      return Success(0, E);
114964eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump    }
1150c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
115164eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
115264eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump  }
115364eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
1154019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_classify_type:
1155131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(EvaluateBuiltinClassifyType(E), E);
11561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11574bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson  case Builtin::BI__builtin_constant_p:
1158019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner    // __builtin_constant_p always has one operand: it returns true if that
1159019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner    // operand can be folded, false otherwise.
1160131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getArg(0)->isEvaluatable(Info.Ctx), E);
116121fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner
116221fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner  case Builtin::BI__builtin_eh_return_data_regno: {
116321fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    int Operand = E->getArg(0)->EvaluateAsInt(Info.Ctx).getZExtValue();
116421fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    Operand = Info.Ctx.Target.getEHDataRegisterNumber(Operand);
116521fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    return Success(Operand, E);
116621fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner  }
1167c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman
1168c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman  case Builtin::BI__builtin_expect:
1169c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman    return Visit(E->getArg(0));
11705726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor
11715726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor  case Builtin::BIstrlen:
11725726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor  case Builtin::BI__builtin_strlen:
11735726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor    // As an extension, we support strlen() and __builtin_strlen() as constant
11745726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor    // expressions when the argument is a string literal.
11755726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor    if (StringLiteral *S
11765726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor               = dyn_cast<StringLiteral>(E->getArg(0)->IgnoreParenImpCasts())) {
11775726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      // The string literal may have embedded null characters. Find the first
11785726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      // one and truncate there.
11795726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      llvm::StringRef Str = S->getString();
11805726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      llvm::StringRef::size_type Pos = Str.find(0);
11815726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      if (Pos != llvm::StringRef::npos)
11825726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor        Str = Str.substr(0, Pos);
11835726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor
11845726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      return Success(Str.size(), E);
11855726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor    }
11865726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor
11875726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
1188019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  }
11894c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner}
1190f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
1191b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattnerbool IntExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
11922de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == BO_Comma) {
1193027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson    if (!Visit(E->getRHS()))
1194027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson      return false;
11954fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson
119633ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    // If we can't evaluate the LHS, it might have side effects;
119733ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    // conservatively mark it.
119833ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    if (!E->getLHS()->isEvaluatable(Info.Ctx))
119933ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman      Info.EvalResult.HasSideEffects = true;
1200c8cc9ccc7b87a7ed1749b074f6b670bcec49abc1Chris Lattner
1201027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson    return true;
1202a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1203a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1204a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->isLogicalOp()) {
1205a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // These need to be handled specially because the operands aren't
1206a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // necessarily integral
1207fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson    bool lhsResult, rhsResult;
12081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1209fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson    if (HandleConversionToBool(E->getLHS(), lhsResult, Info)) {
121051fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson      // We were able to evaluate the LHS, see if we can get away with not
121151fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson      // evaluating the RHS: 0 && X -> 0, 1 || X -> 1
12122de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (lhsResult == (E->getOpcode() == BO_LOr))
12133f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar        return Success(lhsResult, E);
1214a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1215fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
12162de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        if (E->getOpcode() == BO_LOr)
1217131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(lhsResult || rhsResult, E);
12184bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        else
1219131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(lhsResult && rhsResult, E);
12204bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson      }
12214bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson    } else {
1222fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
12234bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        // We can't evaluate the LHS; however, sometimes the result
12244bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        // is determined by the RHS: X && 0 -> 0, X || 1 -> 1.
12252de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        if (rhsResult == (E->getOpcode() == BO_LOr) ||
12262de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall            !rhsResult == (E->getOpcode() == BO_LAnd)) {
1227131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          // Since we weren't able to evaluate the left hand side, it
1228fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          // must have had side effects.
1229fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          Info.EvalResult.HasSideEffects = true;
1230131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
1231131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(rhsResult, E);
12324bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        }
12334bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson      }
1234a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    }
1235a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1236a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false;
1237c8cc9ccc7b87a7ed1749b074f6b670bcec49abc1Chris Lattner  }
123854176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner
1239286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  QualType LHSTy = E->getLHS()->getType();
1240286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  QualType RHSTy = E->getRHS()->getType();
12414087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12424087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar  if (LHSTy->isAnyComplexType()) {
12434087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    assert(RHSTy->isAnyComplexType() && "Invalid comparison");
1244f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LHS, RHS;
12454087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12464087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (!EvaluateComplex(E->getLHS(), LHS, Info))
12474087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      return false;
12484087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12494087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (!EvaluateComplex(E->getRHS(), RHS, Info))
12504087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      return false;
12514087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12524087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (LHS.isComplexFloat()) {
12531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      APFloat::cmpResult CR_r =
12544087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar        LHS.getComplexFloatReal().compare(RHS.getComplexFloatReal());
12551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      APFloat::cmpResult CR_i =
12564087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar        LHS.getComplexFloatImag().compare(RHS.getComplexFloatImag());
12574087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12582de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (E->getOpcode() == BO_EQ)
1259131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((CR_r == APFloat::cmpEqual &&
1260131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        CR_i == APFloat::cmpEqual), E);
1261131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      else {
12622de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        assert(E->getOpcode() == BO_NE &&
1263131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar               "Invalid complex comparison.");
12641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        return Success(((CR_r == APFloat::cmpGreaterThan ||
1265fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_r == APFloat::cmpLessThan ||
1266fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_r == APFloat::cmpUnordered) ||
12671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                        (CR_i == APFloat::cmpGreaterThan ||
1268fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_i == APFloat::cmpLessThan ||
1269fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_i == APFloat::cmpUnordered)), E);
1270131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      }
12714087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    } else {
12722de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (E->getOpcode() == BO_EQ)
1273131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((LHS.getComplexIntReal() == RHS.getComplexIntReal() &&
1274131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        LHS.getComplexIntImag() == RHS.getComplexIntImag()), E);
1275131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      else {
12762de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        assert(E->getOpcode() == BO_NE &&
1277131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar               "Invalid compex comparison.");
1278131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((LHS.getComplexIntReal() != RHS.getComplexIntReal() ||
1279131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        LHS.getComplexIntImag() != RHS.getComplexIntImag()), E);
1280131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      }
12814087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    }
12824087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar  }
12831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1284286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  if (LHSTy->isRealFloatingType() &&
1285286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      RHSTy->isRealFloatingType()) {
1286286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    APFloat RHS(0.0), LHS(0.0);
12871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1288286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    if (!EvaluateFloat(E->getRHS(), RHS, Info))
1289286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      return false;
12901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1291286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    if (!EvaluateFloat(E->getLHS(), LHS, Info))
1292286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      return false;
12931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1294286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    APFloat::cmpResult CR = LHS.compare(RHS);
1295529569e68d10b0fd3750fd2124faf742249b846bAnders Carlsson
1296286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    switch (E->getOpcode()) {
1297286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    default:
1298286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      assert(0 && "Invalid binary operator!");
12992de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LT:
1300131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpLessThan, E);
13012de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_GT:
1302131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpGreaterThan, E);
13032de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LE:
1304131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpLessThan || CR == APFloat::cmpEqual, E);
13052de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_GE:
13061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      return Success(CR == APFloat::cmpGreaterThan || CR == APFloat::cmpEqual,
1307131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                     E);
13082de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_EQ:
1309131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpEqual, E);
13102de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_NE:
13111eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      return Success(CR == APFloat::cmpGreaterThan
1312fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                     || CR == APFloat::cmpLessThan
1313fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                     || CR == APFloat::cmpUnordered, E);
1314286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    }
1315286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  }
13161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1317ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman  if (LHSTy->isPointerType() && RHSTy->isPointerType()) {
13182de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    if (E->getOpcode() == BO_Sub || E->isEqualityOp()) {
1319efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LValue LHSValue;
13203068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (!EvaluatePointer(E->getLHS(), LHSValue, Info))
13213068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
1322a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1323efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LValue RHSValue;
13243068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (!EvaluatePointer(E->getRHS(), RHSValue, Info))
13253068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
1326a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
13275bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      // Reject any bases from the normal codepath; we special-case comparisons
13285bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      // to null.
13295bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      if (LHSValue.getLValueBase()) {
13305bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!E->isEqualityOp())
13315bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
1332a73058324197b7bdfd19307965954f626e26199dKen Dyck        if (RHSValue.getLValueBase() || !RHSValue.getLValueOffset().isZero())
13335bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13345bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        bool bres;
13355bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!EvalPointerValueAsBool(LHSValue, bres))
13365bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13372de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        return Success(bres ^ (E->getOpcode() == BO_EQ), E);
13385bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      } else if (RHSValue.getLValueBase()) {
13395bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!E->isEqualityOp())
13405bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
1341a73058324197b7bdfd19307965954f626e26199dKen Dyck        if (LHSValue.getLValueBase() || !LHSValue.getLValueOffset().isZero())
13425bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13435bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        bool bres;
13445bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!EvalPointerValueAsBool(RHSValue, bres))
13455bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13462de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        return Success(bres ^ (E->getOpcode() == BO_EQ), E);
13475bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      }
1348a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
13492de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (E->getOpcode() == BO_Sub) {
13504992bdde387c5f033bb450a716eaabc0fda52688Chris Lattner        QualType Type = E->getLHS()->getType();
13514992bdde387c5f033bb450a716eaabc0fda52688Chris Lattner        QualType ElementType = Type->getAs<PointerType>()->getPointeeType();
13523068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
1353a73058324197b7bdfd19307965954f626e26199dKen Dyck        CharUnits ElementSize = CharUnits::One();
1354ce1bca73aef9bbf6359ab8420278203dda81a054Eli Friedman        if (!ElementType->isVoidType() && !ElementType->isFunctionType())
1355a73058324197b7bdfd19307965954f626e26199dKen Dyck          ElementSize = Info.Ctx.getTypeSizeInChars(ElementType);
1356a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1357a73058324197b7bdfd19307965954f626e26199dKen Dyck        CharUnits Diff = LHSValue.getLValueOffset() -
1358a73058324197b7bdfd19307965954f626e26199dKen Dyck                             RHSValue.getLValueOffset();
1359a73058324197b7bdfd19307965954f626e26199dKen Dyck        return Success(Diff / ElementSize, E);
1360ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      }
1361ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      bool Result;
13622de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (E->getOpcode() == BO_EQ) {
1363ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman        Result = LHSValue.getLValueOffset() == RHSValue.getLValueOffset();
1364267c0ab1b9a15768f3f15abbfc40ce344751c78bEli Friedman      } else {
1365ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman        Result = LHSValue.getLValueOffset() != RHSValue.getLValueOffset();
1366ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      }
1367ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      return Success(Result, E);
13683068d117951a8df54bae9db039b56201ab10962bAnders Carlsson    }
13693068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
13702ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!LHSTy->isIntegralOrEnumerationType() ||
13712ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor      !RHSTy->isIntegralOrEnumerationType()) {
1372a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // We can't continue from here for non-integral types, and they
1373a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // could potentially confuse the following operations.
1374a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false;
1375a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1376a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1377a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  // The LHS of a constant expr is always evaluated and needed.
137830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  if (!Visit(E->getLHS()))
1379a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false; // error in subexpression.
1380d9f4bcda18bfbf79341edd9d381d4b6a3cffe655Eli Friedman
138142edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  APValue RHSVal;
138242edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (!EvaluateIntegerOrLValue(E->getRHS(), RHSVal, Info))
138330c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return false;
138442edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
138542edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // Handle cases like (unsigned long)&a + 4.
138642edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (E->isAdditiveOp() && Result.isLValue() && RHSVal.isInt()) {
1387a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits Offset = Result.getLValueOffset();
1388a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits AdditionalOffset = CharUnits::fromQuantity(
1389a73058324197b7bdfd19307965954f626e26199dKen Dyck                                     RHSVal.getInt().getZExtValue());
13902de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    if (E->getOpcode() == BO_Add)
1391a73058324197b7bdfd19307965954f626e26199dKen Dyck      Offset += AdditionalOffset;
139242edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    else
1393a73058324197b7bdfd19307965954f626e26199dKen Dyck      Offset -= AdditionalOffset;
1394a73058324197b7bdfd19307965954f626e26199dKen Dyck    Result = APValue(Result.getLValueBase(), Offset);
139542edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    return true;
139642edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  }
139742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
139842edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // Handle cases like 4 + (unsigned long)&a
13992de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == BO_Add &&
140042edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman        RHSVal.isLValue() && Result.isInt()) {
1401a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits Offset = RHSVal.getLValueOffset();
1402a73058324197b7bdfd19307965954f626e26199dKen Dyck    Offset += CharUnits::fromQuantity(Result.getInt().getZExtValue());
1403a73058324197b7bdfd19307965954f626e26199dKen Dyck    Result = APValue(RHSVal.getLValueBase(), Offset);
140442edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    return true;
140542edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  }
140642edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
140742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // All the following cases expect both operands to be an integer
140842edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (!Result.isInt() || !RHSVal.isInt())
1409b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner    return false;
1410a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
141142edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  APSInt& RHS = RHSVal.getInt();
141242edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
1413a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  switch (E->getOpcode()) {
141432fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  default:
14150e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
14162de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Mul: return Success(Result.getInt() * RHS, E);
14172de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Add: return Success(Result.getInt() + RHS, E);
14182de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Sub: return Success(Result.getInt() - RHS, E);
14192de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_And: return Success(Result.getInt() & RHS, E);
14202de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Xor: return Success(Result.getInt() ^ RHS, E);
14212de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Or:  return Success(Result.getInt() | RHS, E);
14222de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Div:
142354176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    if (RHS == 0)
14240e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
142530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() / RHS, E);
14262de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Rem:
142754176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    if (RHS == 0)
14280e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
142930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() % RHS, E);
14302de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Shl: {
1431091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall    // During constant-folding, a negative shift is an opposite shift.
1432091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall    if (RHS.isSigned() && RHS.isNegative()) {
1433091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall      RHS = -RHS;
1434091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall      goto shift_right;
1435091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall    }
1436091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall
1437091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall  shift_left:
1438091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall    unsigned SA
1439091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall      = (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
144030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() << SA, E);
14413f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
14422de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Shr: {
1443091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall    // During constant-folding, a negative shift is an opposite shift.
1444091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall    if (RHS.isSigned() && RHS.isNegative()) {
1445091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall      RHS = -RHS;
1446091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall      goto shift_left;
1447091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall    }
1448091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall
1449091f23f1d6d4bcffd6641cda72a6831e08c02ea7John McCall  shift_right:
14501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    unsigned SA =
145130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
145230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() >> SA, E);
14533f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
14541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
14552de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_LT: return Success(Result.getInt() < RHS, E);
14562de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_GT: return Success(Result.getInt() > RHS, E);
14572de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_LE: return Success(Result.getInt() <= RHS, E);
14582de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_GE: return Success(Result.getInt() >= RHS, E);
14592de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_EQ: return Success(Result.getInt() == RHS, E);
14602de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_NE: return Success(Result.getInt() != RHS, E);
1461b11e77836dd0867955c5abf32baf1c3e6c7f81e1Eli Friedman  }
1462a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
1463a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
1464ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopesbool IntExprEvaluator::VisitConditionalOperator(const ConditionalOperator *E) {
1465a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  bool Cond;
1466a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  if (!HandleConversionToBool(E->getCond(), Cond, Info))
1467ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes    return false;
1468ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes
1469a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  return Visit(Cond ? E->getTrueExpr() : E->getFalseExpr());
1470ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes}
1471ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes
14728b752f10c394b140f9ef89e049cbad1a7676fc25Ken DyckCharUnits IntExprEvaluator::GetAlignOfType(QualType T) {
14735d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
14745d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   the result is the size of the referenced type."
14755d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.alignof]p3: "When alignof is applied to a reference type, the
14765d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   result shall be the alignment of the referenced type."
14775d484e8cf710207010720589d89602233de61d01Sebastian Redl  if (const ReferenceType *Ref = T->getAs<ReferenceType>())
14785d484e8cf710207010720589d89602233de61d01Sebastian Redl    T = Ref->getPointeeType();
14795d484e8cf710207010720589d89602233de61d01Sebastian Redl
1480e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Get information about the alignment.
1481e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  unsigned CharSize = Info.Ctx.Target.getCharWidth();
148218857644059c45da6776f1a288eec7b4cf3a844aDouglas Gregor
14832be586108bb401019647791feca19ea03fd477ceEli Friedman  // __alignof is defined to return the preferred alignment.
14848b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck  return CharUnits::fromQuantity(
14858b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      Info.Ctx.getPreferredTypeAlign(T.getTypePtr()) / CharSize);
1486e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner}
1487e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
14888b752f10c394b140f9ef89e049cbad1a7676fc25Ken DyckCharUnits IntExprEvaluator::GetAlignOfExpr(const Expr *E) {
1489af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  E = E->IgnoreParens();
1490af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
1491af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  // alignof decl is always accepted, even if it doesn't make sense: we default
14921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // to 1 in those cases.
1493af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
14948b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck    return Info.Ctx.getDeclAlign(DRE->getDecl(),
14958b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck                                 /*RefAsPointee*/true);
1496a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1497af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  if (const MemberExpr *ME = dyn_cast<MemberExpr>(E))
14988b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck    return Info.Ctx.getDeclAlign(ME->getMemberDecl(),
14998b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck                                 /*RefAsPointee*/true);
1500af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
1501e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  return GetAlignOfType(E->getType());
1502e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner}
1503e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
1504e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
15050518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl/// VisitSizeAlignOfExpr - Evaluate a sizeof or alignof with a result as the
15060518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl/// expression's type.
15070518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redlbool IntExprEvaluator::VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E) {
1508e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Handle alignof separately.
1509e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (!E->isSizeOf()) {
1510e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    if (E->isArgumentType())
15118b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      return Success(GetAlignOfType(E->getArgumentType()).getQuantity(), E);
1512e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    else
15138b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      return Success(GetAlignOfExpr(E->getArgumentExpr()).getQuantity(), E);
1514e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  }
1515a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
15160518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl  QualType SrcTy = E->getTypeOfArgument();
15175d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
15185d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   the result is the size of the referenced type."
15195d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.alignof]p3: "When alignof is applied to a reference type, the
15205d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   result shall be the alignment of the referenced type."
15215d484e8cf710207010720589d89602233de61d01Sebastian Redl  if (const ReferenceType *Ref = SrcTy->getAs<ReferenceType>())
15225d484e8cf710207010720589d89602233de61d01Sebastian Redl    SrcTy = Ref->getPointeeType();
15230518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl
1524131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  // sizeof(void), __alignof__(void), sizeof(function) = 1 as a gcc
1525131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  // extension.
1526131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  if (SrcTy->isVoidType() || SrcTy->isFunctionType())
1527131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(1, E);
1528a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1529fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner  // sizeof(vla) is not a constantexpr: C99 6.5.3.4p2.
1530e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (!SrcTy->isConstantSizeType())
1531fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner    return false;
1532f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman
1533e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Get information about the size.
1534199c3d6cd16aebbb9c7f0d42af9d922c9628bf70Ken Dyck  return Success(Info.Ctx.getTypeSizeInChars(SrcTy).getQuantity(), E);
1535fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner}
1536fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner
15378ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregorbool IntExprEvaluator::VisitOffsetOfExpr(const OffsetOfExpr *E) {
15388ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  CharUnits Result;
15398ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  unsigned n = E->getNumComponents();
15408ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  OffsetOfExpr* OOE = const_cast<OffsetOfExpr*>(E);
15418ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  if (n == 0)
15428ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    return false;
15438ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  QualType CurrentType = E->getTypeSourceInfo()->getType();
15448ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  for (unsigned i = 0; i != n; ++i) {
15458ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    OffsetOfExpr::OffsetOfNode ON = OOE->getComponent(i);
15468ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    switch (ON.getKind()) {
15478ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Array: {
15488ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      Expr *Idx = OOE->getIndexExpr(ON.getArrayExprIndex());
15498ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      APSInt IdxResult;
15508ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!EvaluateInteger(Idx, IdxResult, Info))
15518ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15528ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const ArrayType *AT = Info.Ctx.getAsArrayType(CurrentType);
15538ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!AT)
15548ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15558ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CurrentType = AT->getElementType();
15568ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CharUnits ElementSize = Info.Ctx.getTypeSizeInChars(CurrentType);
15578ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      Result += IdxResult.getSExtValue() * ElementSize;
15588ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        break;
15598ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15608ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
15618ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Field: {
15628ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      FieldDecl *MemberDecl = ON.getField();
15638ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const RecordType *RT = CurrentType->getAs<RecordType>();
15648ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!RT)
15658ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15668ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      RecordDecl *RD = RT->getDecl();
15678ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
15688ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      unsigned i = 0;
15698ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      // FIXME: It would be nice if we didn't have to loop here!
15708ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      for (RecordDecl::field_iterator Field = RD->field_begin(),
15718ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor                                      FieldEnd = RD->field_end();
15728ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor           Field != FieldEnd; (void)++Field, ++i) {
15738ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        if (*Field == MemberDecl)
15748ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor          break;
15758ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      }
1576cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      assert(i < RL.getFieldCount() && "offsetof field in wrong type");
1577cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      Result += CharUnits::fromQuantity(
1578cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                           RL.getFieldOffset(i) / Info.Ctx.getCharWidth());
15798ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CurrentType = MemberDecl->getType().getNonReferenceType();
15808ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      break;
15818ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15828ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
15838ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Identifier:
15848ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      llvm_unreachable("dependent __builtin_offsetof");
1585cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      return false;
1586cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1587cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Base: {
1588cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      CXXBaseSpecifier *BaseSpec = ON.getBase();
1589cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (BaseSpec->isVirtual())
1590cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1591cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1592cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Find the layout of the class whose base we are looking into.
1593cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const RecordType *RT = CurrentType->getAs<RecordType>();
1594cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (!RT)
1595cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1596cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      RecordDecl *RD = RT->getDecl();
1597cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
1598cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1599cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Find the base class itself.
1600cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      CurrentType = BaseSpec->getType();
1601cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const RecordType *BaseRT = CurrentType->getAs<RecordType>();
1602cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (!BaseRT)
1603cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1604cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1605cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Add the offset to the base.
1606cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      Result += CharUnits::fromQuantity(
1607a14f5979572aa25c03d24750ee4724d2031d4edeAnders Carlsson             RL.getBaseClassOffsetInBits(cast<CXXRecordDecl>(BaseRT->getDecl()))
1608cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                                        / Info.Ctx.getCharWidth());
1609cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      break;
1610cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor    }
16118ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
16128ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  }
16138ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  return Success(Result.getQuantity(), E);
16148ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor}
16158ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
1616b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattnerbool IntExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
16172de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == UO_LNot) {
1618a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // LNot's operand isn't necessarily an integer, so we handle it specially.
1619a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    bool bres;
1620a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    if (!HandleConversionToBool(E->getSubExpr(), bres, Info))
1621a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman      return false;
1622131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(!bres, E);
1623a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1624a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
16254fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar  // Only handle integral operations...
16262ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!E->getSubExpr()->getType()->isIntegralOrEnumerationType())
16274fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar    return false;
16284fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar
162987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  // Get the operand value into 'Result'.
163087eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  if (!Visit(E->getSubExpr()))
163175a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    return false;
1632a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
163375a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  switch (E->getOpcode()) {
16344c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  default:
163575a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    // Address, indirect, pre/post inc/dec, etc are not valid constant exprs.
163675a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    // See C99 6.6p3.
16370e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
16382de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Extension:
16394c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // FIXME: Should extension allow i-c-e extension expressions in its scope?
16404c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // If so, we could clear the diagnostic ID.
16413f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
16422de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Plus:
16431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // The result is always just the subexpr.
16443f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
16452de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Minus:
164630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt()) return false;
164730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(-Result.getInt(), E);
16482de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Not:
164930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt()) return false;
165030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(~Result.getInt(), E);
165106a3675627e3b3c47b49c689c8e404a33144194aAnders Carlsson  }
1652a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
16531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1654732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner/// HandleCast - This is used to evaluate implicit or explicit casts where the
1655732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner/// result type is integer.
1656a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbarbool IntExprEvaluator::VisitCastExpr(CastExpr *E) {
165782206e267ce6cc709797127616f64672d255b310Anders Carlsson  Expr *SubExpr = E->getSubExpr();
165882206e267ce6cc709797127616f64672d255b310Anders Carlsson  QualType DestType = E->getType();
1659b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  QualType SrcType = SubExpr->getType();
166082206e267ce6cc709797127616f64672d255b310Anders Carlsson
16614efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (DestType->isBooleanType()) {
16624efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    bool BoolResult;
16634efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!HandleConversionToBool(SubExpr, BoolResult, Info))
16644efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1665131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(BoolResult, E);
16664efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
16674efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1668a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  // Handle simple integer->integer casts.
16692ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (SrcType->isIntegralOrEnumerationType()) {
1670732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner    if (!Visit(SubExpr))
1671b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
1672a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
1673be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (!Result.isInt()) {
1674be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      // Only allow casts of lvalues if they are lossless.
1675be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return Info.Ctx.getTypeSize(DestType) == Info.Ctx.getTypeSize(SrcType);
1676be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    }
167730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar
1678dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    return Success(HandleIntToIntCast(DestType, SrcType,
167930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar                                      Result.getInt(), Info.Ctx), E);
1680732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  }
16811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1682732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  // FIXME: Clean this up!
1683b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  if (SrcType->isPointerType()) {
1684efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
168587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    if (!EvaluatePointer(SubExpr, LV, Info))
1686b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
16874efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1688dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    if (LV.getLValueBase()) {
1689dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      // Only allow based lvalue casts if they are lossless.
1690dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      if (Info.Ctx.getTypeSize(DestType) != Info.Ctx.getTypeSize(SrcType))
1691dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar        return false;
1692dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar
1693efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LV.moveInto(Result);
1694dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      return true;
1695dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    }
16964efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1697a73058324197b7bdfd19307965954f626e26199dKen Dyck    APSInt AsInt = Info.Ctx.MakeIntValue(LV.getLValueOffset().getQuantity(),
1698a73058324197b7bdfd19307965954f626e26199dKen Dyck                                         SrcType);
1699dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    return Success(HandleIntToIntCast(DestType, SrcType, AsInt, Info.Ctx), E);
17002bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  }
17014efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1702be26570e3faa009bdcefedfaf04473e518940520Eli Friedman  if (SrcType->isArrayType() || SrcType->isFunctionType()) {
1703be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    // This handles double-conversion cases, where there's both
1704be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    // an l-value promotion and an implicit conversion to int.
1705efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
1706be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (!EvaluateLValue(SubExpr, LV, Info))
1707be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return false;
1708be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
1709be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (Info.Ctx.getTypeSize(DestType) != Info.Ctx.getTypeSize(Info.Ctx.VoidPtrTy))
1710be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return false;
1711be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
1712efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Result);
1713be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    return true;
1714be26570e3faa009bdcefedfaf04473e518940520Eli Friedman  }
1715be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
17161725f683432715e5afe34d476024bd6f16eac3fcEli Friedman  if (SrcType->isAnyComplexType()) {
1717f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue C;
17181725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    if (!EvaluateComplex(SubExpr, C, Info))
17191725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return false;
17201725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    if (C.isComplexFloat())
17211725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return Success(HandleFloatToIntCast(DestType, SrcType,
17221725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                                          C.getComplexFloatReal(), Info.Ctx),
17231725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                     E);
17241725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    else
17251725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return Success(HandleIntToIntCast(DestType, SrcType,
17261725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                                        C.getComplexIntReal(), Info.Ctx), E);
17271725f683432715e5afe34d476024bd6f16eac3fcEli Friedman  }
17282217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman  // FIXME: Handle vectors
17292217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman
1730b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  if (!SrcType->isRealFloatingType())
17310e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1732732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner
1733d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  APFloat F(0.0);
1734d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(SubExpr, F, Info))
17350e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
17361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1737b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  return Success(HandleFloatToIntCast(DestType, SrcType, F, Info.Ctx), E);
1738a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
17392bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
1740722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedmanbool IntExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
1741722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  if (E->getSubExpr()->getType()->isAnyComplexType()) {
1742f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LV;
1743722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), LV, Info) || !LV.isComplexInt())
1744722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman      return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1745722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    return Success(LV.getComplexIntReal(), E);
1746722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  }
1747722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1748722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  return Visit(E->getSubExpr());
1749722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman}
1750722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1751664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedmanbool IntExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
1752722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  if (E->getSubExpr()->getType()->isComplexIntegerType()) {
1753f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LV;
1754722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), LV, Info) || !LV.isComplexInt())
1755722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman      return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1756722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    return Success(LV.getComplexIntImag(), E);
1757722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  }
1758722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1759664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
1760664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman    Info.EvalResult.HasSideEffects = true;
1761664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  return Success(0, E);
1762664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman}
1763664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman
1764295995c9c3196416372c9cd35d9cedb6da37bd3dSebastian Redlbool IntExprEvaluator::VisitCXXNoexceptExpr(const CXXNoexceptExpr *E) {
1765295995c9c3196416372c9cd35d9cedb6da37bd3dSebastian Redl  return Success(E->getValue(), E);
1766295995c9c3196416372c9cd35d9cedb6da37bd3dSebastian Redl}
1767295995c9c3196416372c9cd35d9cedb6da37bd3dSebastian Redl
1768f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
1769d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman// Float Evaluation
1770d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman//===----------------------------------------------------------------------===//
1771d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1772d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmannamespace {
1773770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass FloatExprEvaluator
1774d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  : public StmtVisitor<FloatExprEvaluator, bool> {
1775d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  EvalInfo &Info;
1776d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  APFloat &Result;
1777d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanpublic:
1778d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  FloatExprEvaluator(EvalInfo &info, APFloat &result)
1779d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    : Info(info), Result(result) {}
1780d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1781d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitStmt(Stmt *S) {
1782d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1783d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  }
1784d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1785d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
1786019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  bool VisitCallExpr(const CallExpr *E);
1787d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
17885db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  bool VisitUnaryOperator(const UnaryOperator *E);
1789d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitBinaryOperator(const BinaryOperator *E);
1790d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitFloatingLiteral(const FloatingLiteral *E);
17914efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  bool VisitCastExpr(CastExpr *E);
1792ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  bool VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
179367f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  bool VisitConditionalOperator(ConditionalOperator *E);
17942217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman
1795ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitChooseExpr(const ChooseExpr *E)
1796ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
1797ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitUnaryExtension(const UnaryOperator *E)
1798ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getSubExpr()); }
1799abd3a857ace59100305790545d1baae5877b8945John McCall  bool VisitUnaryReal(const UnaryOperator *E);
1800abd3a857ace59100305790545d1baae5877b8945John McCall  bool VisitUnaryImag(const UnaryOperator *E);
1801ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman
1802189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  bool VisitDeclRefExpr(const DeclRefExpr *E);
1803189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall
1804abd3a857ace59100305790545d1baae5877b8945John McCall  // FIXME: Missing: array subscript of vector, member of vector,
1805abd3a857ace59100305790545d1baae5877b8945John McCall  //                 ImplicitValueInitExpr
1806d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman};
1807d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman} // end anonymous namespace
1808d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1809d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanstatic bool EvaluateFloat(const Expr* E, APFloat& Result, EvalInfo &Info) {
18107db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall  assert(E->getType()->isRealFloatingType());
1811d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  return FloatExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
1812d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1813d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1814db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCallstatic bool TryEvaluateBuiltinNaN(ASTContext &Context,
1815db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  QualType ResultTy,
1816db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  const Expr *Arg,
1817db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  bool SNaN,
1818db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  llvm::APFloat &Result) {
1819db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  const StringLiteral *S = dyn_cast<StringLiteral>(Arg->IgnoreParenCasts());
1820db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (!S) return false;
1821db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1822db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  const llvm::fltSemantics &Sem = Context.getFloatTypeSemantics(ResultTy);
1823db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1824db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  llvm::APInt fill;
1825db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1826db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  // Treat empty strings as if they were zero.
1827db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (S->getString().empty())
1828db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    fill = llvm::APInt(32, 0);
1829db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  else if (S->getString().getAsInteger(0, fill))
1830db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return false;
1831db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1832db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (SNaN)
1833db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    Result = llvm::APFloat::getSNaN(Sem, false, &fill);
1834db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  else
1835db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    Result = llvm::APFloat::getQNaN(Sem, false, &fill);
1836db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  return true;
1837db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall}
1838db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1839019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattnerbool FloatExprEvaluator::VisitCallExpr(const CallExpr *E) {
18403c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  switch (E->isBuiltinCall(Info.Ctx)) {
184134a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner  default: return false;
1842019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_val:
1843019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_valf:
1844019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_vall:
1845019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_inf:
1846019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_inff:
18477cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar  case Builtin::BI__builtin_infl: {
18487cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar    const llvm::fltSemantics &Sem =
18497cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar      Info.Ctx.getFloatTypeSemantics(E->getType());
185034a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner    Result = llvm::APFloat::getInf(Sem);
185134a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner    return true;
18527cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar  }
18531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1854db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nans:
1855db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nansf:
1856db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nansl:
1857db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
1858db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                 true, Result);
1859db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
18609e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nan:
18619e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nanf:
18629e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nanl:
18634572baba9d18c275968ac113fd73b0e3c77cccb8Mike Stump    // If this is __builtin_nan() turn this into a nan, otherwise we
18649e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner    // can't constant fold it.
1865db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
1866db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                 false, Result);
18675db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18685db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabs:
18695db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabsf:
18705db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabsl:
18715db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (!EvaluateFloat(E->getArg(0), Result, Info))
18725db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      return false;
18731eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
18745db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (Result.isNegative())
18755db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      Result.changeSign();
18765db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18775db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  case Builtin::BI__builtin_copysign:
18791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  case Builtin::BI__builtin_copysignf:
18805db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_copysignl: {
18815db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    APFloat RHS(0.);
18825db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (!EvaluateFloat(E->getArg(0), Result, Info) ||
18835db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar        !EvaluateFloat(E->getArg(1), RHS, Info))
18845db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      return false;
18855db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    Result.copySign(RHS);
18865db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18875db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  }
1888019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  }
1889019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner}
1890019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
1891189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCallbool FloatExprEvaluator::VisitDeclRefExpr(const DeclRefExpr *E) {
1892189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  const Decl *D = E->getDecl();
1893189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  if (!isa<VarDecl>(D) || isa<ParmVarDecl>(D)) return false;
1894189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  const VarDecl *VD = cast<VarDecl>(D);
1895189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall
1896189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  // Require the qualifiers to be const and not volatile.
1897189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  CanQualType T = Info.Ctx.getCanonicalType(E->getType());
1898189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  if (!T.isConstQualified() || T.isVolatileQualified())
1899189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall    return false;
1900189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall
1901189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  const Expr *Init = VD->getAnyInitializer();
1902189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  if (!Init) return false;
1903189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall
1904189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  if (APValue *V = VD->getEvaluatedValue()) {
1905189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall    if (V->isFloat()) {
1906189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall      Result = V->getFloat();
1907189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall      return true;
1908189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall    }
1909189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall    return false;
1910189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  }
1911189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall
1912189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  if (VD->isEvaluatingValue())
1913189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall    return false;
1914189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall
1915189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  VD->setEvaluatingValue();
1916189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall
1917189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  Expr::EvalResult InitResult;
1918189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  if (Init->Evaluate(InitResult, Info.Ctx) && !InitResult.HasSideEffects &&
1919189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall      InitResult.Val.isFloat()) {
1920189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall    // Cache the evaluated value in the variable declaration.
1921189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall    Result = InitResult.Val.getFloat();
1922189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall    VD->setEvaluatedValue(InitResult.Val);
1923189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall    return true;
1924189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  }
1925189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall
1926189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  VD->setEvaluatedValue(APValue());
1927189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall  return false;
1928189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall}
1929189d6ef40eff11b83b2cda941d5ed89a5cef09b2John McCall
1930abd3a857ace59100305790545d1baae5877b8945John McCallbool FloatExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
193143efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  if (E->getSubExpr()->getType()->isAnyComplexType()) {
193243efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    ComplexValue CV;
193343efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), CV, Info))
193443efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman      return false;
193543efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    Result = CV.FloatReal;
193643efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    return true;
193743efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  }
193843efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman
193943efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  return Visit(E->getSubExpr());
1940abd3a857ace59100305790545d1baae5877b8945John McCall}
1941abd3a857ace59100305790545d1baae5877b8945John McCall
1942abd3a857ace59100305790545d1baae5877b8945John McCallbool FloatExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
194343efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  if (E->getSubExpr()->getType()->isAnyComplexType()) {
194443efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    ComplexValue CV;
194543efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), CV, Info))
194643efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman      return false;
194743efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    Result = CV.FloatImag;
194843efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    return true;
194943efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  }
195043efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman
195143efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
195243efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    Info.EvalResult.HasSideEffects = true;
195343efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  const llvm::fltSemantics &Sem = Info.Ctx.getFloatTypeSemantics(E->getType());
195443efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  Result = llvm::APFloat::getZero(Sem);
1955abd3a857ace59100305790545d1baae5877b8945John McCall  return true;
1956abd3a857ace59100305790545d1baae5877b8945John McCall}
1957abd3a857ace59100305790545d1baae5877b8945John McCall
19585db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbarbool FloatExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
19592de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == UO_Deref)
1960a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes    return false;
1961a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes
19625db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  if (!EvaluateFloat(E->getSubExpr(), Result, Info))
19635db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return false;
19645db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
19655db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  switch (E->getOpcode()) {
19665db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  default: return false;
19672de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Plus:
19685db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
19692de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Minus:
19705db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    Result.changeSign();
19715db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
19725db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  }
19735db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar}
1974019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
1975d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanbool FloatExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
19762de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == BO_Comma) {
19777f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    if (!EvaluateFloat(E->getRHS(), Result, Info))
19787f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman      return false;
19797f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
19807f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    // If we can't evaluate the LHS, it might have side effects;
19817f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    // conservatively mark it.
19827f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    if (!E->getLHS()->isEvaluatable(Info.Ctx))
19837f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman      Info.EvalResult.HasSideEffects = true;
19847f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
19857f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    return true;
19867f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman  }
19877f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
198896e93660124c8028a4c3bcc038ab0cdd18cd7ab2Anders Carlsson  // We can't evaluate pointer-to-member operations.
198996e93660124c8028a4c3bcc038ab0cdd18cd7ab2Anders Carlsson  if (E->isPtrMemOp())
199096e93660124c8028a4c3bcc038ab0cdd18cd7ab2Anders Carlsson    return false;
199196e93660124c8028a4c3bcc038ab0cdd18cd7ab2Anders Carlsson
1992d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  // FIXME: Diagnostics?  I really don't understand how the warnings
1993d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  // and errors are supposed to work.
19945db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  APFloat RHS(0.0);
1995d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(E->getLHS(), Result, Info))
1996d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1997d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(E->getRHS(), RHS, Info))
1998d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1999d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
2000d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  switch (E->getOpcode()) {
2001d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  default: return false;
20022de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Mul:
2003d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.multiply(RHS, APFloat::rmNearestTiesToEven);
2004d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
20052de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Add:
2006d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.add(RHS, APFloat::rmNearestTiesToEven);
2007d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
20082de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Sub:
2009d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.subtract(RHS, APFloat::rmNearestTiesToEven);
2010d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
20112de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Div:
2012d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.divide(RHS, APFloat::rmNearestTiesToEven);
2013d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
2014d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  }
2015d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
2016d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
2017d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanbool FloatExprEvaluator::VisitFloatingLiteral(const FloatingLiteral *E) {
2018d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  Result = E->getValue();
2019d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  return true;
2020d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
2021d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
20224efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanbool FloatExprEvaluator::VisitCastExpr(CastExpr *E) {
20234efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Expr* SubExpr = E->getSubExpr();
20241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
20252ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (SubExpr->getType()->isIntegralOrEnumerationType()) {
20264efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APSInt IntResult;
20273f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    if (!EvaluateInteger(SubExpr, IntResult, Info))
20284efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
20291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    Result = HandleIntToFloatCast(E->getType(), SubExpr->getType(),
2030a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                  IntResult, Info.Ctx);
20314efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
20324efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
20334efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (SubExpr->getType()->isRealFloatingType()) {
20344efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!Visit(SubExpr))
20354efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
2036a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    Result = HandleFloatToFloatCast(E->getType(), SubExpr->getType(),
2037a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                    Result, Info.Ctx);
20384efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
20394efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
2040f3ea8cfe6b1c2ef0702efe130561e9e66708d799John McCall
2041f3ea8cfe6b1c2ef0702efe130561e9e66708d799John McCall  if (E->getCastKind() == CK_FloatingComplexToReal) {
2042f3ea8cfe6b1c2ef0702efe130561e9e66708d799John McCall    ComplexValue V;
2043f3ea8cfe6b1c2ef0702efe130561e9e66708d799John McCall    if (!EvaluateComplex(SubExpr, V, Info))
2044f3ea8cfe6b1c2ef0702efe130561e9e66708d799John McCall      return false;
2045f3ea8cfe6b1c2ef0702efe130561e9e66708d799John McCall    Result = V.getComplexFloatReal();
2046f3ea8cfe6b1c2ef0702efe130561e9e66708d799John McCall    return true;
2047f3ea8cfe6b1c2ef0702efe130561e9e66708d799John McCall  }
20484efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
20494efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return false;
20504efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
20514efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
2052ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregorbool FloatExprEvaluator::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
20534efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Result = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(E->getType()));
20544efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return true;
20554efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
20564efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
205767f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedmanbool FloatExprEvaluator::VisitConditionalOperator(ConditionalOperator *E) {
205867f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  bool Cond;
205967f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  if (!HandleConversionToBool(E->getCond(), Cond, Info))
206067f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman    return false;
206167f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman
206267f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  return Visit(Cond ? E->getTrueExpr() : E->getFalseExpr());
206367f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman}
206467f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman
2065d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman//===----------------------------------------------------------------------===//
2066a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar// Complex Evaluation (for float and integer)
20679ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson//===----------------------------------------------------------------------===//
20689ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
20699ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlssonnamespace {
2070770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass ComplexExprEvaluator
2071f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  : public StmtVisitor<ComplexExprEvaluator, bool> {
20729ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  EvalInfo &Info;
2073f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexValue &Result;
20741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
20759ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlssonpublic:
2076f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexExprEvaluator(EvalInfo &info, ComplexValue &Result)
2077f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    : Info(info), Result(Result) {}
20781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
20799ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //===--------------------------------------------------------------------===//
20809ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //                            Visitor Methods
20819ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //===--------------------------------------------------------------------===//
20829ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2083f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitStmt(Stmt *S) {
2084f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
20859ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  }
20861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2087f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
20889ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2089b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  bool VisitImaginaryLiteral(ImaginaryLiteral *E);
2090a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
2091b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  bool VisitCastExpr(CastExpr *E);
2092b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2093b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  bool VisitBinaryOperator(const BinaryOperator *E);
209496fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  bool VisitUnaryOperator(const UnaryOperator *E);
209596fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  bool VisitConditionalOperator(const ConditionalOperator *E);
2096b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  bool VisitChooseExpr(const ChooseExpr *E)
2097b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
2098b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  bool VisitUnaryExtension(const UnaryOperator *E)
2099b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    { return Visit(E->getSubExpr()); }
210096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  // FIXME Missing: ImplicitValueInitExpr
2101b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman};
2102b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman} // end anonymous namespace
21031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2104b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedmanstatic bool EvaluateComplex(const Expr *E, ComplexValue &Result,
2105b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                            EvalInfo &Info) {
2106b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  assert(E->getType()->isAnyComplexType());
2107b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  return ComplexExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
2108b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman}
2109b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2110b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedmanbool ComplexExprEvaluator::VisitImaginaryLiteral(ImaginaryLiteral *E) {
2111b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  Expr* SubExpr = E->getSubExpr();
2112b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2113b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  if (SubExpr->getType()->isRealFloatingType()) {
2114b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    Result.makeComplexFloat();
2115b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    APFloat &Imag = Result.FloatImag;
2116b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    if (!EvaluateFloat(SubExpr, Imag, Info))
2117b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return false;
2118b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2119b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    Result.FloatReal = APFloat(Imag.getSemantics());
2120b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    return true;
2121b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  } else {
2122b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    assert(SubExpr->getType()->isIntegerType() &&
2123b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman           "Unexpected imaginary literal.");
2124b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2125b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    Result.makeComplexInt();
2126b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    APSInt &Imag = Result.IntImag;
2127b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    if (!EvaluateInteger(SubExpr, Imag, Info))
2128b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return false;
2129b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2130b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    Result.IntReal = APSInt(Imag.getBitWidth(), !Imag.isSigned());
2131b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    return true;
2132b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  }
2133b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman}
2134b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2135b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedmanbool ComplexExprEvaluator::VisitCastExpr(CastExpr *E) {
2136b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
21378786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  switch (E->getCastKind()) {
21388786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_BitCast:
21398786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_LValueBitCast:
21408786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_BaseToDerived:
21418786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_DerivedToBase:
21428786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_UncheckedDerivedToBase:
21438786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_Dynamic:
21448786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_ToUnion:
21458786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_ArrayToPointerDecay:
21468786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_FunctionToPointerDecay:
21478786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_NullToPointer:
21488786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_NullToMemberPointer:
21498786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_BaseToDerivedMemberPointer:
21508786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_DerivedToBaseMemberPointer:
21518786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_MemberPointerToBoolean:
21528786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_ConstructorConversion:
21538786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_IntegralToPointer:
21548786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_PointerToIntegral:
21558786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_PointerToBoolean:
21568786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_ToVoid:
21578786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_VectorSplat:
21588786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_IntegralCast:
21598786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_IntegralToBoolean:
21608786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_IntegralToFloating:
21618786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_FloatingToIntegral:
21628786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_FloatingToBoolean:
21638786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_FloatingCast:
21648786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_AnyPointerToObjCPointerCast:
21658786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_AnyPointerToBlockPointerCast:
21668786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_ObjCObjectLValueCast:
21678786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_FloatingComplexToReal:
21688786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_FloatingComplexToBoolean:
21698786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_IntegralComplexToReal:
21708786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_IntegralComplexToBoolean:
21718786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    llvm_unreachable("invalid cast kind for complex value");
21728786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
21738786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_LValueToRValue:
21748786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_NoOp:
21758786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    return Visit(E->getSubExpr());
21762bb5d00fcf71a7b4d478d478be778fff0494aff6John McCall
21778786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_Dependent:
21788786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_GetObjCProperty:
21798786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_UserDefinedConversion:
21808786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    return false;
21818786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
21828786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_FloatingRealToComplex: {
2183b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    APFloat &Real = Result.FloatReal;
21848786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    if (!EvaluateFloat(E->getSubExpr(), Real, Info))
2185b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return false;
2186b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
21878786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.makeComplexFloat();
21888786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.FloatImag = APFloat(Real.getSemantics());
21898786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    return true;
21908786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  }
21918786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
21928786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_FloatingComplexCast: {
21938786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    if (!Visit(E->getSubExpr()))
21948786da77984e81d48e0e1b2bd339809b1efc19f3John McCall      return false;
21958786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
21968786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    QualType To = E->getType()->getAs<ComplexType>()->getElementType();
21978786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    QualType From
21988786da77984e81d48e0e1b2bd339809b1efc19f3John McCall      = E->getSubExpr()->getType()->getAs<ComplexType>()->getElementType();
21998786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
22008786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.FloatReal
22018786da77984e81d48e0e1b2bd339809b1efc19f3John McCall      = HandleFloatToFloatCast(To, From, Result.FloatReal, Info.Ctx);
22028786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.FloatImag
22038786da77984e81d48e0e1b2bd339809b1efc19f3John McCall      = HandleFloatToFloatCast(To, From, Result.FloatImag, Info.Ctx);
22048786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    return true;
22058786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  }
22068786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
22078786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_FloatingComplexToIntegralComplex: {
22088786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    if (!Visit(E->getSubExpr()))
22098786da77984e81d48e0e1b2bd339809b1efc19f3John McCall      return false;
22108786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
22118786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    QualType To = E->getType()->getAs<ComplexType>()->getElementType();
22128786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    QualType From
22138786da77984e81d48e0e1b2bd339809b1efc19f3John McCall      = E->getSubExpr()->getType()->getAs<ComplexType>()->getElementType();
22148786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.makeComplexInt();
22158786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.IntReal = HandleFloatToIntCast(To, From, Result.FloatReal, Info.Ctx);
22168786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.IntImag = HandleFloatToIntCast(To, From, Result.FloatImag, Info.Ctx);
22178786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    return true;
22188786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  }
22198786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
22208786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_IntegralRealToComplex: {
2221b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    APSInt &Real = Result.IntReal;
22228786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    if (!EvaluateInteger(E->getSubExpr(), Real, Info))
2223b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return false;
22249ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
22258786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.makeComplexInt();
22268786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.IntImag = APSInt(Real.getBitWidth(), !Real.isSigned());
22278786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    return true;
22288786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  }
22298786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
22308786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_IntegralComplexCast: {
22318786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    if (!Visit(E->getSubExpr()))
2232b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return false;
2233ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
22348786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    QualType To = E->getType()->getAs<ComplexType>()->getElementType();
22358786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    QualType From
22368786da77984e81d48e0e1b2bd339809b1efc19f3John McCall      = E->getSubExpr()->getType()->getAs<ComplexType>()->getElementType();
22371725f683432715e5afe34d476024bd6f16eac3fcEli Friedman
22388786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.IntReal = HandleIntToIntCast(To, From, Result.IntReal, Info.Ctx);
22398786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.IntImag = HandleIntToIntCast(To, From, Result.IntImag, Info.Ctx);
22408786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    return true;
22418786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  }
22428786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
22438786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  case CK_IntegralComplexToFloatingComplex: {
22448786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    if (!Visit(E->getSubExpr()))
22458786da77984e81d48e0e1b2bd339809b1efc19f3John McCall      return false;
22468786da77984e81d48e0e1b2bd339809b1efc19f3John McCall
22478786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    QualType To = E->getType()->getAs<ComplexType>()->getElementType();
22488786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    QualType From
22498786da77984e81d48e0e1b2bd339809b1efc19f3John McCall      = E->getSubExpr()->getType()->getAs<ComplexType>()->getElementType();
22508786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.makeComplexFloat();
22518786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.FloatReal = HandleIntToFloatCast(To, From, Result.IntReal, Info.Ctx);
22528786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    Result.FloatImag = HandleIntToFloatCast(To, From, Result.IntImag, Info.Ctx);
22538786da77984e81d48e0e1b2bd339809b1efc19f3John McCall    return true;
22548786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  }
2255ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  }
22561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
22578786da77984e81d48e0e1b2bd339809b1efc19f3John McCall  llvm_unreachable("unknown cast resulting in complex value");
2258b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  return false;
22599ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson}
22609ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2261f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallbool ComplexExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
226296fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  if (E->getOpcode() == BO_Comma) {
226396fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    if (!Visit(E->getRHS()))
226496fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      return false;
226596fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara
226696fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    // If we can't evaluate the LHS, it might have side effects;
226796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    // conservatively mark it.
226896fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    if (!E->getLHS()->isEvaluatable(Info.Ctx))
226996fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Info.EvalResult.HasSideEffects = true;
227096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara
227196fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    return true;
227296fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  }
2273f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  if (!Visit(E->getLHS()))
2274f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
22751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2276f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexValue RHS;
2277a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  if (!EvaluateComplex(E->getRHS(), RHS, Info))
2278f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
2279a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
22803f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar  assert(Result.isComplexFloat() == RHS.isComplexFloat() &&
22813f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         "Invalid operands to binary operator.");
2282ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  switch (E->getOpcode()) {
2283f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  default: return false;
22842de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Add:
2285a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (Result.isComplexFloat()) {
2286a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatReal().add(RHS.getComplexFloatReal(),
2287a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                       APFloat::rmNearestTiesToEven);
2288a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatImag().add(RHS.getComplexFloatImag(),
2289a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                       APFloat::rmNearestTiesToEven);
2290a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
2291a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntReal() += RHS.getComplexIntReal();
2292a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntImag() += RHS.getComplexIntImag();
2293a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
22943f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
22952de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Sub:
2296a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (Result.isComplexFloat()) {
2297a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatReal().subtract(RHS.getComplexFloatReal(),
2298a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                            APFloat::rmNearestTiesToEven);
2299a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatImag().subtract(RHS.getComplexFloatImag(),
2300a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                            APFloat::rmNearestTiesToEven);
2301a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
2302a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntReal() -= RHS.getComplexIntReal();
2303a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntImag() -= RHS.getComplexIntImag();
2304a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
23053f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
23062de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Mul:
23073f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    if (Result.isComplexFloat()) {
2308f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      ComplexValue LHS = Result;
23093f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &LHS_r = LHS.getComplexFloatReal();
23103f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &LHS_i = LHS.getComplexFloatImag();
23113f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &RHS_r = RHS.getComplexFloatReal();
23123f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &RHS_i = RHS.getComplexFloatImag();
23131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
23143f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat Tmp = LHS_r;
23153f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
23163f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatReal() = Tmp;
23173f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_i;
23183f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
23193f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatReal().subtract(Tmp, APFloat::rmNearestTiesToEven);
23203f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar
23213f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_r;
23223f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
23233f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatImag() = Tmp;
23243f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_i;
23253f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
23263f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatImag().add(Tmp, APFloat::rmNearestTiesToEven);
23273f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    } else {
2328f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      ComplexValue LHS = Result;
23291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      Result.getComplexIntReal() =
23303f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar        (LHS.getComplexIntReal() * RHS.getComplexIntReal() -
23313f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         LHS.getComplexIntImag() * RHS.getComplexIntImag());
23321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      Result.getComplexIntImag() =
23333f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar        (LHS.getComplexIntReal() * RHS.getComplexIntImag() +
23343f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         LHS.getComplexIntImag() * RHS.getComplexIntReal());
23353f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    }
23363f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
233796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  case BO_Div:
233896fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    if (Result.isComplexFloat()) {
233996fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      ComplexValue LHS = Result;
234096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      APFloat &LHS_r = LHS.getComplexFloatReal();
234196fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      APFloat &LHS_i = LHS.getComplexFloatImag();
234296fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      APFloat &RHS_r = RHS.getComplexFloatReal();
234396fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      APFloat &RHS_i = RHS.getComplexFloatImag();
234496fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      APFloat &Res_r = Result.getComplexFloatReal();
234596fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      APFloat &Res_i = Result.getComplexFloatImag();
234696fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara
234796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      APFloat Den = RHS_r;
234896fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Den.multiply(RHS_r, APFloat::rmNearestTiesToEven);
234996fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      APFloat Tmp = RHS_i;
235096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
235196fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Den.add(Tmp, APFloat::rmNearestTiesToEven);
235296fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara
235396fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Res_r = LHS_r;
235496fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Res_r.multiply(RHS_r, APFloat::rmNearestTiesToEven);
235596fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Tmp = LHS_i;
235696fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
235796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Res_r.add(Tmp, APFloat::rmNearestTiesToEven);
235896fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Res_r.divide(Den, APFloat::rmNearestTiesToEven);
235996fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara
236096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Res_i = LHS_i;
236196fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Res_i.multiply(RHS_r, APFloat::rmNearestTiesToEven);
236296fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Tmp = LHS_r;
236396fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
236496fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Res_i.subtract(Tmp, APFloat::rmNearestTiesToEven);
236596fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Res_i.divide(Den, APFloat::rmNearestTiesToEven);
236696fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    } else {
236796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      if (RHS.getComplexIntReal() == 0 && RHS.getComplexIntImag() == 0) {
236896fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara        // FIXME: what about diagnostics?
236996fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara        return false;
237096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      }
237196fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      ComplexValue LHS = Result;
237296fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      APSInt Den = RHS.getComplexIntReal() * RHS.getComplexIntReal() +
237396fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara        RHS.getComplexIntImag() * RHS.getComplexIntImag();
237496fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Result.getComplexIntReal() =
237596fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara        (LHS.getComplexIntReal() * RHS.getComplexIntReal() +
237696fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara         LHS.getComplexIntImag() * RHS.getComplexIntImag()) / Den;
237796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Result.getComplexIntImag() =
237896fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara        (LHS.getComplexIntImag() * RHS.getComplexIntReal() -
237996fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara         LHS.getComplexIntReal() * RHS.getComplexIntImag()) / Den;
238096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    }
238196fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    break;
2382ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  }
2383ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
2384f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  return true;
2385ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson}
2386ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
238796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnarabool ComplexExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
238896fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  // Get the operand value into 'Result'.
238996fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  if (!Visit(E->getSubExpr()))
239096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    return false;
239196fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara
239296fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  switch (E->getOpcode()) {
239396fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  default:
239496fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    // FIXME: what about diagnostics?
239596fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    return false;
239696fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  case UO_Extension:
239796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    return true;
239896fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  case UO_Plus:
239996fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    // The result is always just the subexpr.
240096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    return true;
240196fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  case UO_Minus:
240296fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    if (Result.isComplexFloat()) {
240396fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Result.getComplexFloatReal().changeSign();
240496fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Result.getComplexFloatImag().changeSign();
240596fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    }
240696fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    else {
240796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Result.getComplexIntReal() = -Result.getComplexIntReal();
240896fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Result.getComplexIntImag() = -Result.getComplexIntImag();
240996fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    }
241096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    return true;
241196fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  case UO_Not:
241296fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    if (Result.isComplexFloat())
241396fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Result.getComplexFloatImag().changeSign();
241496fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    else
241596fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara      Result.getComplexIntImag() = -Result.getComplexIntImag();
241696fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    return true;
241796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  }
241896fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara}
241996fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara
242096fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnarabool ComplexExprEvaluator::VisitConditionalOperator(const ConditionalOperator *E) {
242196fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  bool Cond;
242296fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  if (!HandleConversionToBool(E->getCond(), Cond, Info))
242396fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara    return false;
242496fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara
242596fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara  return Visit(Cond ? E->getTrueExpr() : E->getFalseExpr());
242696fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara}
242796fc8e4086df323c49f17cac594db1d2f066a2e9Abramo Bagnara
24289ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson//===----------------------------------------------------------------------===//
24296ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner// Top level Expr::Evaluate method.
2430f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
2431f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
243242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// Evaluate - Return true if this is a constant which we can fold using
243342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// any crazy technique (that has nothing to do with language standards) that
243442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// we want to.  If this function returns true, it returns the folded constant
243542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// in Result.
243642c8f87eb60958170c46767273bf93e6c96125bfJohn McCallbool Expr::Evaluate(EvalResult &Result, ASTContext &Ctx) const {
243742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  const Expr *E = this;
243842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  EvalInfo Info(Ctx, Result);
2439efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  if (E->getType()->isVectorType()) {
2440efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateVector(E, Info.EvalResult.Val, Info))
244159b5da6d853b4368b984700315adf7b37de05764Nate Begeman      return false;
2442efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isIntegerType()) {
2443efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!IntExprEvaluator(Info, Info.EvalResult.Val).Visit(const_cast<Expr*>(E)))
24446dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
24450f2b692bb10be35fdc60d0a72a847bdd73124670John McCall    if (Result.Val.isLValue() && !IsGlobalLValue(Result.Val.getLValueBase()))
24460f2b692bb10be35fdc60d0a72a847bdd73124670John McCall      return false;
2447efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->hasPointerRepresentation()) {
2448efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
2449efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluatePointer(E, LV, Info))
24506dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
2451e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    if (!IsGlobalLValue(LV.Base))
245242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return false;
2453efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Info.EvalResult.Val);
2454efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isRealFloatingType()) {
2455efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    llvm::APFloat F(0.0);
2456efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateFloat(E, F, Info))
24576dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
24581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2459efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Info.EvalResult.Val = APValue(F);
2460efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isAnyComplexType()) {
2461efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    ComplexValue C;
2462efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateComplex(E, C, Info))
2463660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump      return false;
2464efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    C.moveInto(Info.EvalResult.Val);
2465660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump  } else
2466660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump    return false;
2467660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump
2468660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump  return true;
2469660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump}
2470660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump
2471cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCallbool Expr::EvaluateAsBooleanCondition(bool &Result, ASTContext &Ctx) const {
2472cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  EvalResult Scratch;
2473cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  EvalInfo Info(Ctx, Scratch);
2474cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall
2475cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  return HandleConversionToBool(this, Result, Info);
2476cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall}
2477cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall
24781b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlssonbool Expr::EvaluateAsLValue(EvalResult &Result, ASTContext &Ctx) const {
24791b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlsson  EvalInfo Info(Ctx, Result);
24801b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlsson
2481efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValue LV;
248242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (EvaluateLValue(this, LV, Info) &&
248342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      !Result.HasSideEffects &&
2484e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara      IsGlobalLValue(LV.Base)) {
2485e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    LV.moveInto(Result.Val);
2486e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    return true;
2487e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  }
2488e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  return false;
2489e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara}
2490e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2491e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnarabool Expr::EvaluateAsAnyLValue(EvalResult &Result, ASTContext &Ctx) const {
2492e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  EvalInfo Info(Ctx, Result);
2493e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2494e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  LValue LV;
2495e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  if (EvaluateLValue(this, LV, Info)) {
2496efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Result.Val);
2497efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return true;
2498efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  }
2499efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
2500b2f295c8050fb8c141bf2cf38eed0a56e99d0092Eli Friedman}
2501b2f295c8050fb8c141bf2cf38eed0a56e99d0092Eli Friedman
25026ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner/// isEvaluatable - Call Evaluate to see if this expression can be constant
250345b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattner/// folded, but discard the result.
250445b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattnerbool Expr::isEvaluatable(ASTContext &Ctx) const {
25054fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson  EvalResult Result;
25064fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson  return Evaluate(Result, Ctx) && !Result.HasSideEffects;
250745b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattner}
250851fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
2509393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanianbool Expr::HasSideEffects(ASTContext &Ctx) const {
2510393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  Expr::EvalResult Result;
2511393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  EvalInfo Info(Ctx, Result);
2512393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  return HasSideEffect(Info).Visit(const_cast<Expr*>(this));
2513393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian}
2514393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian
251551fe996231b1d7199f76e4005ff4c943d5deeecdAnders CarlssonAPSInt Expr::EvaluateAsInt(ASTContext &Ctx) const {
25161c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  EvalResult EvalResult;
25171c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  bool Result = Evaluate(EvalResult, Ctx);
2518c6ed729f669044f5072a49d79041f455d971ece3Jeffrey Yasskin  (void)Result;
251951fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson  assert(Result && "Could not evaluate expression");
25201c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  assert(EvalResult.Val.isInt() && "Expression did not evaluate to integer");
252151fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
25221c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  return EvalResult.Val.getInt();
252351fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson}
2524d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2525e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara bool Expr::EvalResult::isGlobalLValue() const {
2526e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara   assert(Val.isLValue());
2527e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara   return IsGlobalLValue(Val.getLValueBase());
2528e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara }
2529e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2530e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2531d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// isIntegerConstantExpr - this recursive routine will test if an expression is
2532d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// an integer constant expression.
2533d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2534d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// FIXME: Pass up a reason why! Invalid operation in i-c-e, division by zero,
2535d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// comma, etc
2536d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall///
2537d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// FIXME: Handle offsetof.  Two things to do:  Handle GCC's __builtin_offsetof
2538d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// to support gcc 4.0+  and handle the idiom GCC recognizes with a null pointer
2539d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// cast+dereference.
2540d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2541d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// CheckICE - This function does the fundamental ICE checking: the returned
2542d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// ICEDiag contains a Val of 0, 1, or 2, and a possibly null SourceLocation.
2543d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// Note that to reduce code duplication, this helper does no evaluation
2544d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// itself; the caller checks whether the expression is evaluatable, and
2545d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// in the rare cases where CheckICE actually cares about the evaluated
2546d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// value, it calls into Evalute.
2547d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//
2548d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// Meanings of Val:
2549d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 0: This expression is an ICE if it can be evaluated by Evaluate.
2550d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 1: This expression is not an ICE, but if it isn't evaluated, it's
2551d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//    a legal subexpression for an ICE. This return value is used to handle
2552d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//    the comma operator in C99 mode.
2553d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 2: This expression is not an ICE, and is not a legal subexpression for one.
2554d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
25553c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohmannamespace {
25563c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohman
2557d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstruct ICEDiag {
2558d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  unsigned Val;
2559d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  SourceLocation Loc;
2560d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2561d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  public:
2562d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag(unsigned v, SourceLocation l) : Val(v), Loc(l) {}
2563d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag() : Val(0) {}
2564d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall};
2565d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
25663c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohman}
25673c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohman
25683c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohmanstatic ICEDiag NoDiag() { return ICEDiag(); }
2569d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2570d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstatic ICEDiag CheckEvalInICE(const Expr* E, ASTContext &Ctx) {
2571d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  Expr::EvalResult EVResult;
2572d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (!E->Evaluate(EVResult, Ctx) || EVResult.HasSideEffects ||
2573d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      !EVResult.Val.isInt()) {
2574d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2575d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2576d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return NoDiag();
2577d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2578d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2579d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstatic ICEDiag CheckICE(const Expr* E, ASTContext &Ctx) {
2580d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(!E->isValueDependent() && "Should not see value dependent exprs!");
25812ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!E->getType()->isIntegralOrEnumerationType()) {
2582d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2583d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2584d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2585d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  switch (E->getStmtClass()) {
2586d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#define STMT(Node, Base) case Expr::Node##Class:
2587d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#define EXPR(Node, Base)
2588d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#include "clang/AST/StmtNodes.inc"
2589d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::PredefinedExprClass:
2590d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::FloatingLiteralClass:
2591d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImaginaryLiteralClass:
2592d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::StringLiteralClass:
2593d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ArraySubscriptExprClass:
2594d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::MemberExprClass:
2595d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CompoundAssignOperatorClass:
2596d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CompoundLiteralExprClass:
2597d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ExtVectorElementExprClass:
2598d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::InitListExprClass:
2599d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DesignatedInitExprClass:
2600d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImplicitValueInitExprClass:
2601d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ParenListExprClass:
2602d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::VAArgExprClass:
2603d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::AddrLabelExprClass:
2604d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::StmtExprClass:
2605d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXMemberCallExprClass:
2606d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDynamicCastExprClass:
2607d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXTypeidExprClass:
26089be88403e965cc49af76c9d33d818781d44b333eFrancois Pichet  case Expr::CXXUuidofExprClass:
2609d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXNullPtrLiteralExprClass:
2610d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXThisExprClass:
2611d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXThrowExprClass:
2612d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXNewExprClass:
2613d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDeleteExprClass:
2614d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXPseudoDestructorExprClass:
2615d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnresolvedLookupExprClass:
2616d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DependentScopeDeclRefExprClass:
2617d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXConstructExprClass:
2618d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXBindTemporaryExprClass:
26194765fa05b5652fcc4356371c2f481d0ea9a1b007John McCall  case Expr::ExprWithCleanupsClass:
2620d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXTemporaryObjectExprClass:
2621d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXUnresolvedConstructExprClass:
2622d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDependentScopeMemberExprClass:
2623d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnresolvedMemberExprClass:
2624d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCStringLiteralClass:
2625d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCEncodeExprClass:
2626d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCMessageExprClass:
2627d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCSelectorExprClass:
2628d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCProtocolExprClass:
2629d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCIvarRefExprClass:
2630d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCPropertyRefExprClass:
2631d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCIsaExprClass:
2632d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ShuffleVectorExprClass:
2633d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BlockExprClass:
2634d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BlockDeclRefExprClass:
2635d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::NoStmtClass:
26367cd7d1ad33fdf49eef83942e8855fe20d95aa1b9John McCall  case Expr::OpaqueValueExprClass:
2637d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2638d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2639d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::GNUNullExprClass:
2640d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // GCC considers the GNU __null value to be an integral constant expression.
2641d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2642d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2643d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ParenExprClass:
2644d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<ParenExpr>(E)->getSubExpr(), Ctx);
2645d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::IntegerLiteralClass:
2646d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CharacterLiteralClass:
2647d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXBoolLiteralExprClass:
2648ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  case Expr::CXXScalarValueInitExprClass:
2649d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnaryTypeTraitExprClass:
26506ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  case Expr::BinaryTypeTraitExprClass:
26512e156225a29407a50dd19041aa5750171ad44ea3Sebastian Redl  case Expr::CXXNoexceptExprClass:
2652d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2653d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CallExprClass:
26546cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt  case Expr::CXXOperatorCallExprClass: {
2655d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const CallExpr *CE = cast<CallExpr>(E);
2656d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CE->isBuiltinCall(Ctx))
2657d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckEvalInICE(E, Ctx);
2658d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2659d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2660d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DeclRefExprClass:
2661d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (isa<EnumConstantDecl>(cast<DeclRefExpr>(E)->getDecl()))
2662d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2663d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Ctx.getLangOptions().CPlusPlus &&
2664d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        E->getType().getCVRQualifiers() == Qualifiers::Const) {
2665d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      const NamedDecl *D = cast<DeclRefExpr>(E)->getDecl();
2666d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2667d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Parameter variables are never constants.  Without this check,
2668d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // getAnyInitializer() can find a default argument, which leads
2669d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // to chaos.
2670d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (isa<ParmVarDecl>(D))
2671d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2672d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2673d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // C++ 7.1.5.1p2
2674d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      //   A variable of non-volatile const-qualified integral or enumeration
2675d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      //   type initialized by an ICE can be used in ICEs.
2676d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (const VarDecl *Dcl = dyn_cast<VarDecl>(D)) {
2677d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        Qualifiers Quals = Ctx.getCanonicalType(Dcl->getType()).getQualifiers();
2678d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Quals.hasVolatile() || !Quals.hasConst())
2679d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2680d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2681d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Look for a declaration of this variable that has an initializer.
2682d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        const VarDecl *ID = 0;
2683d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        const Expr *Init = Dcl->getAnyInitializer(ID);
2684d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Init) {
2685d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (ID->isInitKnownICE()) {
2686d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            // We have already checked whether this subexpression is an
2687d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            // integral constant expression.
2688d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            if (ID->isInitICE())
2689d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return NoDiag();
2690d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            else
2691d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2692d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2693d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2694d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // It's an ICE whether or not the definition we found is
2695d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // out-of-line.  See DR 721 and the discussion in Clang PR
2696d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // 6206 for details.
2697d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2698d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (Dcl->isCheckingICE()) {
2699d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2700d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2701d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2702d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          Dcl->setCheckingICE();
2703d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          ICEDiag Result = CheckICE(Init, Ctx);
2704d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // Cache the result of the ICE test.
2705d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          Dcl->setInitKnownICE(Result.Val == 0);
2706d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return Result;
2707d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2708d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2709d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2710d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2711d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnaryOperatorClass: {
2712d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const UnaryOperator *Exp = cast<UnaryOperator>(E);
2713d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    switch (Exp->getOpcode()) {
27142de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_PostInc:
27152de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_PostDec:
27162de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_PreInc:
27172de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_PreDec:
27182de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_AddrOf:
27192de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Deref:
2720d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
27212de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Extension:
27222de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_LNot:
27232de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Plus:
27242de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Minus:
27252de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Not:
27262de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Real:
27272de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Imag:
2728d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckICE(Exp->getSubExpr(), Ctx);
2729d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2730d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2731d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // OffsetOf falls through here.
2732d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2733d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::OffsetOfExprClass: {
2734d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Note that per C99, offsetof must be an ICE. And AFAIK, using
2735d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Evaluate matches the proposed gcc behavior for cases like
2736d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // "offsetof(struct s{int x[4];}, x[!.0])".  This doesn't affect
2737d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // compliance: we should warn earlier for offsetof expressions with
2738d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // array subscripts that aren't ICEs, and if the array subscripts
2739d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // are ICEs, the value of the offsetof must be an integer constant.
2740d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckEvalInICE(E, Ctx);
2741d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2742d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::SizeOfAlignOfExprClass: {
2743d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const SizeOfAlignOfExpr *Exp = cast<SizeOfAlignOfExpr>(E);
2744d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Exp->isSizeOf() && Exp->getTypeOfArgument()->isVariableArrayType())
2745d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
2746d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2747d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2748d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BinaryOperatorClass: {
2749d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const BinaryOperator *Exp = cast<BinaryOperator>(E);
2750d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    switch (Exp->getOpcode()) {
27512de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_PtrMemD:
27522de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_PtrMemI:
27532de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Assign:
27542de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_MulAssign:
27552de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_DivAssign:
27562de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_RemAssign:
27572de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_AddAssign:
27582de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_SubAssign:
27592de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_ShlAssign:
27602de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_ShrAssign:
27612de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_AndAssign:
27622de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_XorAssign:
27632de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_OrAssign:
2764d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
2765d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
27662de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Mul:
27672de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Div:
27682de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Rem:
27692de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Add:
27702de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Sub:
27712de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Shl:
27722de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Shr:
27732de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LT:
27742de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_GT:
27752de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LE:
27762de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_GE:
27772de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_EQ:
27782de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_NE:
27792de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_And:
27802de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Xor:
27812de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Or:
27822de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Comma: {
2783d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
2784d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
27852de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (Exp->getOpcode() == BO_Div ||
27862de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall          Exp->getOpcode() == BO_Rem) {
2787d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Evaluate gives an error for undefined Div/Rem, so make sure
2788d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // we don't evaluate one.
2789d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (LHSResult.Val != 2 && RHSResult.Val != 2) {
2790d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          llvm::APSInt REval = Exp->getRHS()->EvaluateAsInt(Ctx);
2791d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (REval == 0)
2792d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(1, E->getLocStart());
2793d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (REval.isSigned() && REval.isAllOnesValue()) {
2794d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            llvm::APSInt LEval = Exp->getLHS()->EvaluateAsInt(Ctx);
2795d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            if (LEval.isMinSignedValue())
2796d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return ICEDiag(1, E->getLocStart());
2797d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2798d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2799d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
28002de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (Exp->getOpcode() == BO_Comma) {
2801d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Ctx.getLangOptions().C99) {
2802d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // C99 6.6p3 introduces a strange edge case: comma can be in an ICE
2803d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // if it isn't evaluated.
2804d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (LHSResult.Val == 0 && RHSResult.Val == 0)
2805d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(1, E->getLocStart());
2806d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        } else {
2807d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // In both C89 and C++, commas in ICEs are illegal.
2808d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, E->getLocStart());
2809d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2810d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2811d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val >= RHSResult.Val)
2812d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return LHSResult;
2813d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return RHSResult;
2814d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
28152de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LAnd:
28162de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LOr: {
2817d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
2818d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
2819d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val == 0 && RHSResult.Val == 1) {
2820d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Rare case where the RHS has a comma "side-effect"; we need
2821d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // to actually check the condition to see whether the side
2822d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // with the comma is evaluated.
28232de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        if ((Exp->getOpcode() == BO_LAnd) !=
2824d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            (Exp->getLHS()->EvaluateAsInt(Ctx) == 0))
2825d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return RHSResult;
2826d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return NoDiag();
2827d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2828d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2829d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val >= RHSResult.Val)
2830d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return LHSResult;
2831d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return RHSResult;
2832d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2833d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2834d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2835d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImplicitCastExprClass:
2836d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CStyleCastExprClass:
2837d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXFunctionalCastExprClass:
2838d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXStaticCastExprClass:
2839d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXReinterpretCastExprClass:
2840d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXConstCastExprClass: {
2841d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const Expr *SubExpr = cast<CastExpr>(E)->getSubExpr();
28422ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    if (SubExpr->getType()->isIntegralOrEnumerationType())
2843d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckICE(SubExpr, Ctx);
2844d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (isa<FloatingLiteral>(SubExpr->IgnoreParens()))
2845d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2846d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2847d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2848d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ConditionalOperatorClass: {
2849d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const ConditionalOperator *Exp = cast<ConditionalOperator>(E);
2850d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // If the condition (ignoring parens) is a __builtin_constant_p call,
2851d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // then only the true side is actually considered in an integer constant
2852d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // expression, and it is fully evaluated.  This is an important GNU
2853d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // extension.  See GCC PR38377 for discussion.
2854d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (const CallExpr *CallCE
2855d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        = dyn_cast<CallExpr>(Exp->getCond()->IgnoreParenCasts()))
2856d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (CallCE->isBuiltinCall(Ctx) == Builtin::BI__builtin_constant_p) {
2857d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        Expr::EvalResult EVResult;
2858d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (!E->Evaluate(EVResult, Ctx) || EVResult.HasSideEffects ||
2859d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            !EVResult.Val.isInt()) {
2860d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, E->getLocStart());
2861d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2862d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return NoDiag();
2863d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2864d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag CondResult = CheckICE(Exp->getCond(), Ctx);
2865d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag TrueResult = CheckICE(Exp->getTrueExpr(), Ctx);
2866d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag FalseResult = CheckICE(Exp->getFalseExpr(), Ctx);
2867d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CondResult.Val == 2)
2868d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CondResult;
2869d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (TrueResult.Val == 2)
2870d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return TrueResult;
2871d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (FalseResult.Val == 2)
2872d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return FalseResult;
2873d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CondResult.Val == 1)
2874d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CondResult;
2875d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (TrueResult.Val == 0 && FalseResult.Val == 0)
2876d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2877d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // Rare case where the diagnostics depend on which side is evaluated
2878d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // Note that if we get here, CondResult is 0, and at least one of
2879d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // TrueResult and FalseResult is non-zero.
2880d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Exp->getCond()->EvaluateAsInt(Ctx) == 0) {
2881d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return FalseResult;
2882d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2883d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return TrueResult;
2884d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2885d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDefaultArgExprClass:
2886d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<CXXDefaultArgExpr>(E)->getExpr(), Ctx);
2887d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ChooseExprClass: {
2888d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<ChooseExpr>(E)->getChosenSubExpr(Ctx), Ctx);
2889d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2890d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2891d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2892d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  // Silence a GCC warning
2893d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return ICEDiag(2, E->getLocStart());
2894d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2895d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2896d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallbool Expr::isIntegerConstantExpr(llvm::APSInt &Result, ASTContext &Ctx,
2897d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall                                 SourceLocation *Loc, bool isEvaluated) const {
2898d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag d = CheckICE(this, Ctx);
2899d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (d.Val != 0) {
2900d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Loc) *Loc = d.Loc;
2901d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return false;
2902d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2903d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  EvalResult EvalResult;
2904d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (!Evaluate(EvalResult, Ctx))
2905d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    llvm_unreachable("ICE cannot be evaluated!");
2906d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(!EvalResult.HasSideEffects && "ICE with side effects!");
2907d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(EvalResult.Val.isInt() && "ICE that isn't integer!");
2908d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  Result = EvalResult.Val.getInt();
2909d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return true;
2910d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2911