Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 1 | //===--- ExprConstant.cpp - Expression Constant Evaluator -----------------===// |
Anders Carlsson | c44eec6 | 2008-07-03 04:20:39 +0000 | [diff] [blame] | 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file implements the Expr constant evaluator. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
| 14 | #include "clang/AST/APValue.h" |
| 15 | #include "clang/AST/ASTContext.h" |
Seo Sanghyeon | 0fe52e1 | 2008-07-08 07:23:12 +0000 | [diff] [blame] | 16 | #include "clang/AST/StmtVisitor.h" |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 17 | #include "clang/Basic/Diagnostic.h" |
Anders Carlsson | 06a3675 | 2008-07-08 05:49:43 +0000 | [diff] [blame] | 18 | #include "clang/Basic/TargetInfo.h" |
Anders Carlsson | c754aa6 | 2008-07-08 05:13:58 +0000 | [diff] [blame] | 19 | #include "llvm/Support/Compiler.h" |
Anders Carlsson | c44eec6 | 2008-07-03 04:20:39 +0000 | [diff] [blame] | 20 | using namespace clang; |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 21 | using llvm::APSInt; |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 22 | using llvm::APFloat; |
Anders Carlsson | c44eec6 | 2008-07-03 04:20:39 +0000 | [diff] [blame] | 23 | |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 24 | /// EvalInfo - This is a private struct used by the evaluator to capture |
| 25 | /// information about a subexpression as it is folded. It retains information |
| 26 | /// about the AST context, but also maintains information about the folded |
| 27 | /// expression. |
| 28 | /// |
| 29 | /// If an expression could be evaluated, it is still possible it is not a C |
| 30 | /// "integer constant expression" or constant expression. If not, this struct |
| 31 | /// captures information about how and why not. |
| 32 | /// |
| 33 | /// One bit of information passed *into* the request for constant folding |
| 34 | /// indicates whether the subexpression is "evaluated" or not according to C |
| 35 | /// rules. For example, the RHS of (0 && foo()) is not evaluated. We can |
| 36 | /// evaluate the expression regardless of what the RHS is, but C only allows |
| 37 | /// certain things in certain situations. |
| 38 | struct EvalInfo { |
| 39 | ASTContext &Ctx; |
| 40 | |
| 41 | /// isEvaluated - True if the subexpression is required to be evaluated, false |
| 42 | /// if it is short-circuited (according to C rules). |
| 43 | bool isEvaluated; |
| 44 | |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 45 | /// ICEDiag - If the expression is unfoldable, then ICEDiag contains the |
| 46 | /// error diagnostic indicating why it is not foldable and DiagLoc indicates a |
| 47 | /// caret position for the error. If it is foldable, but the expression is |
| 48 | /// not an integer constant expression, ICEDiag contains the extension |
| 49 | /// diagnostic to emit which describes why it isn't an integer constant |
| 50 | /// expression. If this expression *is* an integer-constant-expr, then |
| 51 | /// ICEDiag is zero. |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 52 | /// |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 53 | /// The caller can choose to emit this diagnostic or not, depending on whether |
| 54 | /// they require an i-c-e or a constant or not. DiagLoc indicates the caret |
| 55 | /// position for the report. |
| 56 | /// |
| 57 | /// If ICEDiag is zero, then this expression is an i-c-e. |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 58 | unsigned ICEDiag; |
| 59 | SourceLocation DiagLoc; |
| 60 | |
| 61 | EvalInfo(ASTContext &ctx) : Ctx(ctx), isEvaluated(true), ICEDiag(0) {} |
| 62 | }; |
| 63 | |
| 64 | |
| 65 | static bool EvaluatePointer(const Expr *E, APValue &Result, EvalInfo &Info); |
| 66 | static bool EvaluateInteger(const Expr *E, APSInt &Result, EvalInfo &Info); |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 67 | static bool EvaluateFloat(const Expr *E, APFloat &Result, EvalInfo &Info); |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 68 | |
| 69 | //===----------------------------------------------------------------------===// |
| 70 | // Pointer Evaluation |
| 71 | //===----------------------------------------------------------------------===// |
| 72 | |
Anders Carlsson | c754aa6 | 2008-07-08 05:13:58 +0000 | [diff] [blame] | 73 | namespace { |
Anders Carlsson | 2bad168 | 2008-07-08 14:30:00 +0000 | [diff] [blame] | 74 | class VISIBILITY_HIDDEN PointerExprEvaluator |
| 75 | : public StmtVisitor<PointerExprEvaluator, APValue> { |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 76 | EvalInfo &Info; |
Anders Carlsson | 2bad168 | 2008-07-08 14:30:00 +0000 | [diff] [blame] | 77 | public: |
Anders Carlsson | 2bad168 | 2008-07-08 14:30:00 +0000 | [diff] [blame] | 78 | |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 79 | PointerExprEvaluator(EvalInfo &info) : Info(info) {} |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 80 | |
Anders Carlsson | 2bad168 | 2008-07-08 14:30:00 +0000 | [diff] [blame] | 81 | APValue VisitStmt(Stmt *S) { |
Anders Carlsson | 2bad168 | 2008-07-08 14:30:00 +0000 | [diff] [blame] | 82 | return APValue(); |
| 83 | } |
| 84 | |
| 85 | APValue VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); } |
| 86 | |
Anders Carlsson | 650c92f | 2008-07-08 15:34:11 +0000 | [diff] [blame] | 87 | APValue VisitBinaryOperator(const BinaryOperator *E); |
| 88 | APValue VisitCastExpr(const CastExpr* E); |
Anders Carlsson | 650c92f | 2008-07-08 15:34:11 +0000 | [diff] [blame] | 89 | }; |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 90 | } // end anonymous namespace |
Anders Carlsson | 650c92f | 2008-07-08 15:34:11 +0000 | [diff] [blame] | 91 | |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 92 | static bool EvaluatePointer(const Expr* E, APValue& Result, EvalInfo &Info) { |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 93 | if (!E->getType()->isPointerType()) |
| 94 | return false; |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 95 | Result = PointerExprEvaluator(Info).Visit(const_cast<Expr*>(E)); |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 96 | return Result.isLValue(); |
| 97 | } |
| 98 | |
| 99 | APValue PointerExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) { |
| 100 | if (E->getOpcode() != BinaryOperator::Add && |
| 101 | E->getOpcode() != BinaryOperator::Sub) |
| 102 | return APValue(); |
| 103 | |
| 104 | const Expr *PExp = E->getLHS(); |
| 105 | const Expr *IExp = E->getRHS(); |
| 106 | if (IExp->getType()->isPointerType()) |
| 107 | std::swap(PExp, IExp); |
| 108 | |
| 109 | APValue ResultLValue; |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 110 | if (!EvaluatePointer(PExp, ResultLValue, Info)) |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 111 | return APValue(); |
| 112 | |
| 113 | llvm::APSInt AdditionalOffset(32); |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 114 | if (!EvaluateInteger(IExp, AdditionalOffset, Info)) |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 115 | return APValue(); |
| 116 | |
| 117 | uint64_t Offset = ResultLValue.getLValueOffset(); |
| 118 | if (E->getOpcode() == BinaryOperator::Add) |
| 119 | Offset += AdditionalOffset.getZExtValue(); |
| 120 | else |
| 121 | Offset -= AdditionalOffset.getZExtValue(); |
| 122 | |
| 123 | return APValue(ResultLValue.getLValueBase(), Offset); |
| 124 | } |
| 125 | |
| 126 | |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 127 | APValue PointerExprEvaluator::VisitCastExpr(const CastExpr* E) { |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 128 | const Expr* SubExpr = E->getSubExpr(); |
| 129 | |
| 130 | // Check for pointer->pointer cast |
| 131 | if (SubExpr->getType()->isPointerType()) { |
| 132 | APValue Result; |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 133 | if (EvaluatePointer(SubExpr, Result, Info)) |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 134 | return Result; |
| 135 | return APValue(); |
| 136 | } |
| 137 | |
Eli Friedman | d9f4bcd | 2008-07-27 05:46:18 +0000 | [diff] [blame] | 138 | if (SubExpr->getType()->isIntegralType()) { |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 139 | llvm::APSInt Result(32); |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 140 | if (EvaluateInteger(SubExpr, Result, Info)) { |
| 141 | Result.extOrTrunc((unsigned)Info.Ctx.getTypeSize(E->getType())); |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 142 | return APValue(0, Result.getZExtValue()); |
| 143 | } |
| 144 | } |
| 145 | |
| 146 | assert(0 && "Unhandled cast"); |
| 147 | return APValue(); |
| 148 | } |
| 149 | |
| 150 | |
| 151 | //===----------------------------------------------------------------------===// |
| 152 | // Integer Evaluation |
| 153 | //===----------------------------------------------------------------------===// |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 154 | |
| 155 | namespace { |
Anders Carlsson | c754aa6 | 2008-07-08 05:13:58 +0000 | [diff] [blame] | 156 | class VISIBILITY_HIDDEN IntExprEvaluator |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 157 | : public StmtVisitor<IntExprEvaluator, bool> { |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 158 | EvalInfo &Info; |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 159 | APSInt &Result; |
Anders Carlsson | c754aa6 | 2008-07-08 05:13:58 +0000 | [diff] [blame] | 160 | public: |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 161 | IntExprEvaluator(EvalInfo &info, APSInt &result) |
| 162 | : Info(info), Result(result) {} |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 163 | |
Chris Lattner | 7a76778 | 2008-07-11 19:24:49 +0000 | [diff] [blame] | 164 | unsigned getIntTypeSizeInBits(QualType T) const { |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 165 | return (unsigned)Info.Ctx.getIntWidth(T); |
| 166 | } |
| 167 | |
| 168 | bool Extension(SourceLocation L, diag::kind D) { |
| 169 | Info.DiagLoc = L; |
| 170 | Info.ICEDiag = D; |
| 171 | return true; // still a constant. |
| 172 | } |
| 173 | |
| 174 | bool Error(SourceLocation L, diag::kind D) { |
| 175 | // If this is in an unevaluated portion of the subexpression, ignore the |
| 176 | // error. |
| 177 | if (!Info.isEvaluated) |
| 178 | return true; |
| 179 | |
| 180 | Info.DiagLoc = L; |
| 181 | Info.ICEDiag = D; |
| 182 | return false; |
Chris Lattner | 7a76778 | 2008-07-11 19:24:49 +0000 | [diff] [blame] | 183 | } |
| 184 | |
Anders Carlsson | c754aa6 | 2008-07-08 05:13:58 +0000 | [diff] [blame] | 185 | //===--------------------------------------------------------------------===// |
| 186 | // Visitor Methods |
| 187 | //===--------------------------------------------------------------------===// |
Chris Lattner | 7a76778 | 2008-07-11 19:24:49 +0000 | [diff] [blame] | 188 | |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 189 | bool VisitStmt(Stmt *S) { |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 190 | return Error(S->getLocStart(), diag::err_expr_not_constant); |
Anders Carlsson | c754aa6 | 2008-07-08 05:13:58 +0000 | [diff] [blame] | 191 | } |
| 192 | |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 193 | bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); } |
Anders Carlsson | c754aa6 | 2008-07-08 05:13:58 +0000 | [diff] [blame] | 194 | |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 195 | bool VisitIntegerLiteral(const IntegerLiteral *E) { |
| 196 | Result = E->getValue(); |
| 197 | Result.setIsUnsigned(E->getType()->isUnsignedIntegerType()); |
| 198 | return true; |
| 199 | } |
| 200 | bool VisitCharacterLiteral(const CharacterLiteral *E) { |
| 201 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 202 | Result = E->getValue(); |
| 203 | Result.setIsUnsigned(E->getType()->isUnsignedIntegerType()); |
| 204 | return true; |
| 205 | } |
| 206 | bool VisitTypesCompatibleExpr(const TypesCompatibleExpr *E) { |
| 207 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 208 | Result = Info.Ctx.typesAreCompatible(E->getArgType1(), E->getArgType2()); |
| 209 | return true; |
| 210 | } |
| 211 | bool VisitDeclRefExpr(const DeclRefExpr *E); |
| 212 | bool VisitCallExpr(const CallExpr *E); |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 213 | bool VisitBinaryOperator(const BinaryOperator *E); |
| 214 | bool VisitUnaryOperator(const UnaryOperator *E); |
Anders Carlsson | 06a3675 | 2008-07-08 05:49:43 +0000 | [diff] [blame] | 215 | |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 216 | bool VisitCastExpr(CastExpr* E) { |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 217 | return HandleCast(E->getLocStart(), E->getSubExpr(), E->getType()); |
Anders Carlsson | 650c92f | 2008-07-08 15:34:11 +0000 | [diff] [blame] | 218 | } |
Chris Lattner | fcee001 | 2008-07-11 21:24:13 +0000 | [diff] [blame] | 219 | bool VisitSizeOfAlignOfTypeExpr(const SizeOfAlignOfTypeExpr *E) { |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 220 | return EvaluateSizeAlignOf(E->isSizeOf(), E->getArgumentType(), |
| 221 | E->getType()); |
Chris Lattner | fcee001 | 2008-07-11 21:24:13 +0000 | [diff] [blame] | 222 | } |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 223 | |
Chris Lattner | fcee001 | 2008-07-11 21:24:13 +0000 | [diff] [blame] | 224 | private: |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 225 | bool HandleCast(SourceLocation CastLoc, Expr *SubExpr, QualType DestType); |
Chris Lattner | fcee001 | 2008-07-11 21:24:13 +0000 | [diff] [blame] | 226 | bool EvaluateSizeAlignOf(bool isSizeOf, QualType SrcTy, QualType DstTy); |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 227 | }; |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 228 | } // end anonymous namespace |
Anders Carlsson | 650c92f | 2008-07-08 15:34:11 +0000 | [diff] [blame] | 229 | |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 230 | static bool EvaluateInteger(const Expr* E, APSInt &Result, EvalInfo &Info) { |
| 231 | return IntExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E)); |
Anders Carlsson | 650c92f | 2008-07-08 15:34:11 +0000 | [diff] [blame] | 232 | } |
Anders Carlsson | 650c92f | 2008-07-08 15:34:11 +0000 | [diff] [blame] | 233 | |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 234 | bool IntExprEvaluator::VisitDeclRefExpr(const DeclRefExpr *E) { |
| 235 | // Enums are integer constant exprs. |
| 236 | if (const EnumConstantDecl *D = dyn_cast<EnumConstantDecl>(E->getDecl())) { |
| 237 | Result = D->getInitVal(); |
| 238 | return true; |
| 239 | } |
| 240 | |
| 241 | // Otherwise, random variable references are not constants. |
| 242 | return Error(E->getLocStart(), diag::err_expr_not_constant); |
| 243 | } |
| 244 | |
Chris Lattner | a4d55d8 | 2008-10-06 06:40:35 +0000 | [diff] [blame] | 245 | /// EvaluateBuiltinClassifyType - Evaluate __builtin_classify_type the same way |
| 246 | /// as GCC. |
| 247 | static int EvaluateBuiltinClassifyType(const CallExpr *E) { |
| 248 | // The following enum mimics the values returned by GCC. |
| 249 | enum gcc_type_class { |
| 250 | no_type_class = -1, |
| 251 | void_type_class, integer_type_class, char_type_class, |
| 252 | enumeral_type_class, boolean_type_class, |
| 253 | pointer_type_class, reference_type_class, offset_type_class, |
| 254 | real_type_class, complex_type_class, |
| 255 | function_type_class, method_type_class, |
| 256 | record_type_class, union_type_class, |
| 257 | array_type_class, string_type_class, |
| 258 | lang_type_class |
| 259 | }; |
| 260 | |
| 261 | // If no argument was supplied, default to "no_type_class". This isn't |
| 262 | // ideal, however it is what gcc does. |
| 263 | if (E->getNumArgs() == 0) |
| 264 | return no_type_class; |
| 265 | |
| 266 | QualType ArgTy = E->getArg(0)->getType(); |
| 267 | if (ArgTy->isVoidType()) |
| 268 | return void_type_class; |
| 269 | else if (ArgTy->isEnumeralType()) |
| 270 | return enumeral_type_class; |
| 271 | else if (ArgTy->isBooleanType()) |
| 272 | return boolean_type_class; |
| 273 | else if (ArgTy->isCharType()) |
| 274 | return string_type_class; // gcc doesn't appear to use char_type_class |
| 275 | else if (ArgTy->isIntegerType()) |
| 276 | return integer_type_class; |
| 277 | else if (ArgTy->isPointerType()) |
| 278 | return pointer_type_class; |
| 279 | else if (ArgTy->isReferenceType()) |
| 280 | return reference_type_class; |
| 281 | else if (ArgTy->isRealType()) |
| 282 | return real_type_class; |
| 283 | else if (ArgTy->isComplexType()) |
| 284 | return complex_type_class; |
| 285 | else if (ArgTy->isFunctionType()) |
| 286 | return function_type_class; |
| 287 | else if (ArgTy->isStructureType()) |
| 288 | return record_type_class; |
| 289 | else if (ArgTy->isUnionType()) |
| 290 | return union_type_class; |
| 291 | else if (ArgTy->isArrayType()) |
| 292 | return array_type_class; |
| 293 | else if (ArgTy->isUnionType()) |
| 294 | return union_type_class; |
| 295 | else // FIXME: offset_type_class, method_type_class, & lang_type_class? |
| 296 | assert(0 && "CallExpr::isBuiltinClassifyType(): unimplemented type"); |
| 297 | return -1; |
| 298 | } |
| 299 | |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 300 | bool IntExprEvaluator::VisitCallExpr(const CallExpr *E) { |
| 301 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 302 | |
Chris Lattner | 019f4e8 | 2008-10-06 05:28:25 +0000 | [diff] [blame] | 303 | switch (E->isBuiltinCall()) { |
| 304 | default: |
| 305 | return Error(E->getLocStart(), diag::err_expr_not_constant); |
| 306 | case Builtin::BI__builtin_classify_type: |
Chris Lattner | a4d55d8 | 2008-10-06 06:40:35 +0000 | [diff] [blame] | 307 | Result.setIsSigned(true); |
| 308 | Result = EvaluateBuiltinClassifyType(E); |
Chris Lattner | 019f4e8 | 2008-10-06 05:28:25 +0000 | [diff] [blame] | 309 | return true; |
| 310 | |
| 311 | case Builtin::BI__builtin_constant_p: { |
| 312 | // __builtin_constant_p always has one operand: it returns true if that |
| 313 | // operand can be folded, false otherwise. |
| 314 | APValue Res; |
| 315 | Result = E->getArg(0)->tryEvaluate(Res, Info.Ctx); |
| 316 | return true; |
| 317 | } |
| 318 | } |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 319 | } |
Anders Carlsson | 650c92f | 2008-07-08 15:34:11 +0000 | [diff] [blame] | 320 | |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 321 | bool IntExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) { |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 322 | // The LHS of a constant expr is always evaluated and needed. |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 323 | llvm::APSInt RHS(32); |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 324 | if (!Visit(E->getLHS())) |
| 325 | return false; // error in subexpression. |
| 326 | |
| 327 | bool OldEval = Info.isEvaluated; |
| 328 | |
| 329 | // The short-circuiting &&/|| operators don't necessarily evaluate their |
| 330 | // RHS. Make sure to pass isEvaluated down correctly. |
| 331 | if ((E->getOpcode() == BinaryOperator::LAnd && Result == 0) || |
| 332 | (E->getOpcode() == BinaryOperator::LOr && Result != 0)) |
| 333 | Info.isEvaluated = false; |
Eli Friedman | d9f4bcd | 2008-07-27 05:46:18 +0000 | [diff] [blame] | 334 | |
| 335 | // FIXME: Handle pointer subtraction |
| 336 | |
| 337 | // FIXME Maybe we want to succeed even where we can't evaluate the |
| 338 | // right side of LAnd/LOr? |
| 339 | // For example, see http://llvm.org/bugs/show_bug.cgi?id=2525 |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 340 | if (!EvaluateInteger(E->getRHS(), RHS, Info)) |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 341 | return false; |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 342 | Info.isEvaluated = OldEval; |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 343 | |
| 344 | switch (E->getOpcode()) { |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 345 | default: return Error(E->getOperatorLoc(), diag::err_expr_not_constant); |
| 346 | case BinaryOperator::Mul: Result *= RHS; return true; |
| 347 | case BinaryOperator::Add: Result += RHS; return true; |
| 348 | case BinaryOperator::Sub: Result -= RHS; return true; |
| 349 | case BinaryOperator::And: Result &= RHS; return true; |
| 350 | case BinaryOperator::Xor: Result ^= RHS; return true; |
| 351 | case BinaryOperator::Or: Result |= RHS; return true; |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 352 | case BinaryOperator::Div: |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 353 | if (RHS == 0) |
| 354 | return Error(E->getOperatorLoc(), diag::err_expr_divide_by_zero); |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 355 | Result /= RHS; |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 356 | return true; |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 357 | case BinaryOperator::Rem: |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 358 | if (RHS == 0) |
| 359 | return Error(E->getOperatorLoc(), diag::err_expr_divide_by_zero); |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 360 | Result %= RHS; |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 361 | return true; |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 362 | case BinaryOperator::Shl: |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 363 | // FIXME: Warn about out of range shift amounts! |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 364 | Result <<= (unsigned)RHS.getLimitedValue(Result.getBitWidth()-1); |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 365 | break; |
| 366 | case BinaryOperator::Shr: |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 367 | Result >>= (unsigned)RHS.getLimitedValue(Result.getBitWidth()-1); |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 368 | break; |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 369 | |
Chris Lattner | ac7cb60 | 2008-07-11 19:29:32 +0000 | [diff] [blame] | 370 | case BinaryOperator::LT: |
| 371 | Result = Result < RHS; |
| 372 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 373 | break; |
| 374 | case BinaryOperator::GT: |
| 375 | Result = Result > RHS; |
| 376 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 377 | break; |
| 378 | case BinaryOperator::LE: |
| 379 | Result = Result <= RHS; |
| 380 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 381 | break; |
| 382 | case BinaryOperator::GE: |
| 383 | Result = Result >= RHS; |
| 384 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 385 | break; |
| 386 | case BinaryOperator::EQ: |
| 387 | Result = Result == RHS; |
| 388 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 389 | break; |
| 390 | case BinaryOperator::NE: |
| 391 | Result = Result != RHS; |
| 392 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 393 | break; |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 394 | case BinaryOperator::LAnd: |
| 395 | Result = Result != 0 && RHS != 0; |
| 396 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 397 | break; |
| 398 | case BinaryOperator::LOr: |
| 399 | Result = Result != 0 || RHS != 0; |
| 400 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 401 | break; |
| 402 | |
Anders Carlsson | 06a3675 | 2008-07-08 05:49:43 +0000 | [diff] [blame] | 403 | |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 404 | case BinaryOperator::Comma: |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 405 | // Result of the comma is just the result of the RHS. |
| 406 | Result = RHS; |
| 407 | |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 408 | // C99 6.6p3: "shall not contain assignment, ..., or comma operators, |
| 409 | // *except* when they are contained within a subexpression that is not |
| 410 | // evaluated". Note that Assignment can never happen due to constraints |
| 411 | // on the LHS subexpr, so we don't need to check it here. |
Chris Lattner | 54176fd | 2008-07-12 00:14:42 +0000 | [diff] [blame] | 412 | if (!Info.isEvaluated) |
| 413 | return true; |
| 414 | |
| 415 | // If the value is evaluated, we can accept it as an extension. |
| 416 | return Extension(E->getOperatorLoc(), diag::ext_comma_in_constant_expr); |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 417 | } |
| 418 | |
| 419 | Result.setIsUnsigned(E->getType()->isUnsignedIntegerType()); |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 420 | return true; |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 421 | } |
| 422 | |
Chris Lattner | fcee001 | 2008-07-11 21:24:13 +0000 | [diff] [blame] | 423 | /// EvaluateSizeAlignOf - Evaluate sizeof(SrcTy) or alignof(SrcTy) with a result |
| 424 | /// as a DstTy type. |
| 425 | bool IntExprEvaluator::EvaluateSizeAlignOf(bool isSizeOf, QualType SrcTy, |
| 426 | QualType DstTy) { |
| 427 | // Return the result in the right width. |
| 428 | Result.zextOrTrunc(getIntTypeSizeInBits(DstTy)); |
| 429 | Result.setIsUnsigned(DstTy->isUnsignedIntegerType()); |
| 430 | |
| 431 | // sizeof(void) and __alignof__(void) = 1 as a gcc extension. |
| 432 | if (SrcTy->isVoidType()) |
| 433 | Result = 1; |
| 434 | |
| 435 | // sizeof(vla) is not a constantexpr: C99 6.5.3.4p2. |
| 436 | if (!SrcTy->isConstantSizeType()) { |
| 437 | // FIXME: Should we attempt to evaluate this? |
| 438 | return false; |
| 439 | } |
| 440 | |
| 441 | // GCC extension: sizeof(function) = 1. |
| 442 | if (SrcTy->isFunctionType()) { |
| 443 | // FIXME: AlignOf shouldn't be unconditionally 4! |
| 444 | Result = isSizeOf ? 1 : 4; |
| 445 | return true; |
| 446 | } |
| 447 | |
| 448 | // Get information about the size or align. |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 449 | unsigned CharSize = Info.Ctx.Target.getCharWidth(); |
Chris Lattner | fcee001 | 2008-07-11 21:24:13 +0000 | [diff] [blame] | 450 | if (isSizeOf) |
| 451 | Result = getIntTypeSizeInBits(SrcTy) / CharSize; |
| 452 | else |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 453 | Result = Info.Ctx.getTypeAlign(SrcTy) / CharSize; |
Chris Lattner | fcee001 | 2008-07-11 21:24:13 +0000 | [diff] [blame] | 454 | return true; |
| 455 | } |
| 456 | |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 457 | bool IntExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) { |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 458 | // Special case unary operators that do not need their subexpression |
| 459 | // evaluated. offsetof/sizeof/alignof are all special. |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 460 | if (E->isOffsetOfOp()) { |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 461 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 462 | Result = E->evaluateOffsetOf(Info.Ctx); |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 463 | Result.setIsUnsigned(E->getType()->isUnsignedIntegerType()); |
| 464 | return true; |
| 465 | } |
| 466 | |
| 467 | if (E->isSizeOfAlignOfOp()) |
Chris Lattner | fcee001 | 2008-07-11 21:24:13 +0000 | [diff] [blame] | 468 | return EvaluateSizeAlignOf(E->getOpcode() == UnaryOperator::SizeOf, |
| 469 | E->getSubExpr()->getType(), E->getType()); |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 470 | |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 471 | // Get the operand value into 'Result'. |
| 472 | if (!Visit(E->getSubExpr())) |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 473 | return false; |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 474 | |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 475 | switch (E->getOpcode()) { |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 476 | default: |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 477 | // Address, indirect, pre/post inc/dec, etc are not valid constant exprs. |
| 478 | // See C99 6.6p3. |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 479 | return Error(E->getOperatorLoc(), diag::err_expr_not_constant); |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 480 | case UnaryOperator::LNot: { |
| 481 | bool Val = Result == 0; |
| 482 | Result.zextOrTrunc(getIntTypeSizeInBits(E->getType())); |
| 483 | Result = Val; |
| 484 | break; |
| 485 | } |
| 486 | case UnaryOperator::Extension: |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 487 | // FIXME: Should extension allow i-c-e extension expressions in its scope? |
| 488 | // If so, we could clear the diagnostic ID. |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 489 | case UnaryOperator::Plus: |
Chris Lattner | 4c4867e | 2008-07-12 00:38:25 +0000 | [diff] [blame] | 490 | // The result is always just the subexpr. |
Chris Lattner | 75a4881 | 2008-07-11 22:15:16 +0000 | [diff] [blame] | 491 | break; |
| 492 | case UnaryOperator::Minus: |
| 493 | Result = -Result; |
| 494 | break; |
| 495 | case UnaryOperator::Not: |
| 496 | Result = ~Result; |
| 497 | break; |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 498 | } |
| 499 | |
| 500 | Result.setIsUnsigned(E->getType()->isUnsignedIntegerType()); |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 501 | return true; |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 502 | } |
| 503 | |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 504 | /// HandleCast - This is used to evaluate implicit or explicit casts where the |
| 505 | /// result type is integer. |
| 506 | bool IntExprEvaluator::HandleCast(SourceLocation CastLoc, |
| 507 | Expr *SubExpr, QualType DestType) { |
Chris Lattner | 7a76778 | 2008-07-11 19:24:49 +0000 | [diff] [blame] | 508 | unsigned DestWidth = getIntTypeSizeInBits(DestType); |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 509 | |
| 510 | // Handle simple integer->integer casts. |
| 511 | if (SubExpr->getType()->isIntegerType()) { |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 512 | if (!Visit(SubExpr)) |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 513 | return false; |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 514 | |
| 515 | // Figure out if this is a truncate, extend or noop cast. |
| 516 | // If the input is signed, do a sign extend, noop, or truncate. |
| 517 | if (DestType->isBooleanType()) { |
| 518 | // Conversion to bool compares against zero. |
| 519 | Result = Result != 0; |
| 520 | Result.zextOrTrunc(DestWidth); |
Chris Lattner | 7a76778 | 2008-07-11 19:24:49 +0000 | [diff] [blame] | 521 | } else |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 522 | Result.extOrTrunc(DestWidth); |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 523 | Result.setIsUnsigned(DestType->isUnsignedIntegerType()); |
| 524 | return true; |
| 525 | } |
| 526 | |
| 527 | // FIXME: Clean this up! |
| 528 | if (SubExpr->getType()->isPointerType()) { |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 529 | APValue LV; |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 530 | if (!EvaluatePointer(SubExpr, LV, Info)) |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 531 | return false; |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 532 | if (LV.getLValueBase()) |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 533 | return false; |
Anders Carlsson | 06a3675 | 2008-07-08 05:49:43 +0000 | [diff] [blame] | 534 | |
Anders Carlsson | 559e56b | 2008-07-08 16:49:00 +0000 | [diff] [blame] | 535 | Result.extOrTrunc(DestWidth); |
| 536 | Result = LV.getLValueOffset(); |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 537 | Result.setIsUnsigned(DestType->isUnsignedIntegerType()); |
| 538 | return true; |
Anders Carlsson | 2bad168 | 2008-07-08 14:30:00 +0000 | [diff] [blame] | 539 | } |
| 540 | |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 541 | if (!SubExpr->getType()->isRealFloatingType()) |
| 542 | return Error(CastLoc, diag::err_expr_not_constant); |
| 543 | |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 544 | APFloat F(0.0); |
| 545 | if (!EvaluateFloat(SubExpr, F, Info)) |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 546 | return Error(CastLoc, diag::err_expr_not_constant); |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 547 | |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 548 | // If the destination is boolean, compare against zero. |
| 549 | if (DestType->isBooleanType()) { |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 550 | Result = !F.isZero(); |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 551 | Result.zextOrTrunc(DestWidth); |
| 552 | Result.setIsUnsigned(DestType->isUnsignedIntegerType()); |
| 553 | return true; |
| 554 | } |
| 555 | |
| 556 | // Determine whether we are converting to unsigned or signed. |
| 557 | bool DestSigned = DestType->isSignedIntegerType(); |
| 558 | |
| 559 | // FIXME: Warning for overflow. |
Dale Johannesen | ee5a700 | 2008-10-09 23:02:32 +0000 | [diff] [blame] | 560 | uint64_t Space[4]; |
| 561 | bool ignored; |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 562 | (void)F.convertToInteger(Space, DestWidth, DestSigned, |
Dale Johannesen | ee5a700 | 2008-10-09 23:02:32 +0000 | [diff] [blame] | 563 | llvm::APFloat::rmTowardZero, &ignored); |
Chris Lattner | 732b223 | 2008-07-12 01:15:53 +0000 | [diff] [blame] | 564 | Result = llvm::APInt(DestWidth, 4, Space); |
| 565 | Result.setIsUnsigned(!DestSigned); |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 566 | return true; |
Anders Carlsson | a25ae3d | 2008-07-08 14:35:21 +0000 | [diff] [blame] | 567 | } |
Anders Carlsson | 2bad168 | 2008-07-08 14:30:00 +0000 | [diff] [blame] | 568 | |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 569 | //===----------------------------------------------------------------------===// |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 570 | // Float Evaluation |
| 571 | //===----------------------------------------------------------------------===// |
| 572 | |
| 573 | namespace { |
| 574 | class VISIBILITY_HIDDEN FloatExprEvaluator |
| 575 | : public StmtVisitor<FloatExprEvaluator, bool> { |
| 576 | EvalInfo &Info; |
| 577 | APFloat &Result; |
| 578 | public: |
| 579 | FloatExprEvaluator(EvalInfo &info, APFloat &result) |
| 580 | : Info(info), Result(result) {} |
| 581 | |
| 582 | bool VisitStmt(Stmt *S) { |
| 583 | return false; |
| 584 | } |
| 585 | |
| 586 | bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); } |
Chris Lattner | 019f4e8 | 2008-10-06 05:28:25 +0000 | [diff] [blame] | 587 | bool VisitCallExpr(const CallExpr *E); |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 588 | |
Daniel Dunbar | 5db4b3f | 2008-10-16 03:51:50 +0000 | [diff] [blame^] | 589 | bool VisitUnaryOperator(const UnaryOperator *E); |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 590 | bool VisitBinaryOperator(const BinaryOperator *E); |
| 591 | bool VisitFloatingLiteral(const FloatingLiteral *E); |
| 592 | }; |
| 593 | } // end anonymous namespace |
| 594 | |
| 595 | static bool EvaluateFloat(const Expr* E, APFloat& Result, EvalInfo &Info) { |
| 596 | return FloatExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E)); |
| 597 | } |
| 598 | |
Chris Lattner | 019f4e8 | 2008-10-06 05:28:25 +0000 | [diff] [blame] | 599 | bool FloatExprEvaluator::VisitCallExpr(const CallExpr *E) { |
Chris Lattner | 019f4e8 | 2008-10-06 05:28:25 +0000 | [diff] [blame] | 600 | switch (E->isBuiltinCall()) { |
Chris Lattner | 34a74ab | 2008-10-06 05:53:16 +0000 | [diff] [blame] | 601 | default: return false; |
Chris Lattner | 019f4e8 | 2008-10-06 05:28:25 +0000 | [diff] [blame] | 602 | case Builtin::BI__builtin_huge_val: |
| 603 | case Builtin::BI__builtin_huge_valf: |
| 604 | case Builtin::BI__builtin_huge_vall: |
| 605 | case Builtin::BI__builtin_inf: |
| 606 | case Builtin::BI__builtin_inff: |
Daniel Dunbar | 7cbed03 | 2008-10-14 05:41:12 +0000 | [diff] [blame] | 607 | case Builtin::BI__builtin_infl: { |
| 608 | const llvm::fltSemantics &Sem = |
| 609 | Info.Ctx.getFloatTypeSemantics(E->getType()); |
Chris Lattner | 34a74ab | 2008-10-06 05:53:16 +0000 | [diff] [blame] | 610 | Result = llvm::APFloat::getInf(Sem); |
| 611 | return true; |
Daniel Dunbar | 7cbed03 | 2008-10-14 05:41:12 +0000 | [diff] [blame] | 612 | } |
Chris Lattner | 9e62171 | 2008-10-06 06:31:58 +0000 | [diff] [blame] | 613 | |
| 614 | case Builtin::BI__builtin_nan: |
| 615 | case Builtin::BI__builtin_nanf: |
| 616 | case Builtin::BI__builtin_nanl: |
| 617 | // If this is __builtin_nan("") turn this into a simple nan, otherwise we |
| 618 | // can't constant fold it. |
| 619 | if (const StringLiteral *S = |
| 620 | dyn_cast<StringLiteral>(E->getArg(0)->IgnoreParenCasts())) { |
| 621 | if (!S->isWide() && S->getByteLength() == 0) { // empty string. |
Daniel Dunbar | 7cbed03 | 2008-10-14 05:41:12 +0000 | [diff] [blame] | 622 | const llvm::fltSemantics &Sem = |
| 623 | Info.Ctx.getFloatTypeSemantics(E->getType()); |
Chris Lattner | 9e62171 | 2008-10-06 06:31:58 +0000 | [diff] [blame] | 624 | Result = llvm::APFloat::getNaN(Sem); |
| 625 | return true; |
| 626 | } |
| 627 | } |
| 628 | return false; |
Daniel Dunbar | 5db4b3f | 2008-10-16 03:51:50 +0000 | [diff] [blame^] | 629 | |
| 630 | case Builtin::BI__builtin_fabs: |
| 631 | case Builtin::BI__builtin_fabsf: |
| 632 | case Builtin::BI__builtin_fabsl: |
| 633 | if (!EvaluateFloat(E->getArg(0), Result, Info)) |
| 634 | return false; |
| 635 | |
| 636 | if (Result.isNegative()) |
| 637 | Result.changeSign(); |
| 638 | return true; |
| 639 | |
| 640 | case Builtin::BI__builtin_copysign: |
| 641 | case Builtin::BI__builtin_copysignf: |
| 642 | case Builtin::BI__builtin_copysignl: { |
| 643 | APFloat RHS(0.); |
| 644 | if (!EvaluateFloat(E->getArg(0), Result, Info) || |
| 645 | !EvaluateFloat(E->getArg(1), RHS, Info)) |
| 646 | return false; |
| 647 | Result.copySign(RHS); |
| 648 | return true; |
| 649 | } |
Chris Lattner | 019f4e8 | 2008-10-06 05:28:25 +0000 | [diff] [blame] | 650 | } |
| 651 | } |
| 652 | |
Daniel Dunbar | 5db4b3f | 2008-10-16 03:51:50 +0000 | [diff] [blame^] | 653 | bool FloatExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) { |
| 654 | if (!EvaluateFloat(E->getSubExpr(), Result, Info)) |
| 655 | return false; |
| 656 | |
| 657 | switch (E->getOpcode()) { |
| 658 | default: return false; |
| 659 | case UnaryOperator::Plus: |
| 660 | return true; |
| 661 | case UnaryOperator::Minus: |
| 662 | Result.changeSign(); |
| 663 | return true; |
| 664 | } |
| 665 | } |
Chris Lattner | 019f4e8 | 2008-10-06 05:28:25 +0000 | [diff] [blame] | 666 | |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 667 | bool FloatExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) { |
| 668 | // FIXME: Diagnostics? I really don't understand how the warnings |
| 669 | // and errors are supposed to work. |
Daniel Dunbar | 5db4b3f | 2008-10-16 03:51:50 +0000 | [diff] [blame^] | 670 | APFloat RHS(0.0); |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 671 | if (!EvaluateFloat(E->getLHS(), Result, Info)) |
| 672 | return false; |
| 673 | if (!EvaluateFloat(E->getRHS(), RHS, Info)) |
| 674 | return false; |
| 675 | |
| 676 | switch (E->getOpcode()) { |
| 677 | default: return false; |
| 678 | case BinaryOperator::Mul: |
| 679 | Result.multiply(RHS, APFloat::rmNearestTiesToEven); |
| 680 | return true; |
| 681 | case BinaryOperator::Add: |
| 682 | Result.add(RHS, APFloat::rmNearestTiesToEven); |
| 683 | return true; |
| 684 | case BinaryOperator::Sub: |
| 685 | Result.subtract(RHS, APFloat::rmNearestTiesToEven); |
| 686 | return true; |
| 687 | case BinaryOperator::Div: |
| 688 | Result.divide(RHS, APFloat::rmNearestTiesToEven); |
| 689 | return true; |
| 690 | case BinaryOperator::Rem: |
| 691 | Result.mod(RHS, APFloat::rmNearestTiesToEven); |
| 692 | return true; |
| 693 | } |
| 694 | } |
| 695 | |
| 696 | bool FloatExprEvaluator::VisitFloatingLiteral(const FloatingLiteral *E) { |
| 697 | Result = E->getValue(); |
| 698 | return true; |
| 699 | } |
| 700 | |
| 701 | //===----------------------------------------------------------------------===// |
Chris Lattner | f5eeb05 | 2008-07-11 18:11:29 +0000 | [diff] [blame] | 702 | // Top level TryEvaluate. |
| 703 | //===----------------------------------------------------------------------===// |
| 704 | |
Chris Lattner | 019f4e8 | 2008-10-06 05:28:25 +0000 | [diff] [blame] | 705 | /// tryEvaluate - Return true if this is a constant which we can fold using |
| 706 | /// any crazy technique (that has nothing to do with language standards) that |
| 707 | /// we want to. If this function returns true, it returns the folded constant |
| 708 | /// in Result. |
Chris Lattner | b542afe | 2008-07-11 19:10:17 +0000 | [diff] [blame] | 709 | bool Expr::tryEvaluate(APValue &Result, ASTContext &Ctx) const { |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 710 | EvalInfo Info(Ctx); |
Anders Carlsson | 06a3675 | 2008-07-08 05:49:43 +0000 | [diff] [blame] | 711 | if (getType()->isIntegerType()) { |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 712 | llvm::APSInt sInt(32); |
Chris Lattner | 87eae5e | 2008-07-11 22:52:41 +0000 | [diff] [blame] | 713 | if (EvaluateInteger(this, sInt, Info)) { |
Anders Carlsson | 06a3675 | 2008-07-08 05:49:43 +0000 | [diff] [blame] | 714 | Result = APValue(sInt); |
| 715 | return true; |
| 716 | } |
Eli Friedman | d8bfe7f | 2008-08-22 00:06:13 +0000 | [diff] [blame] | 717 | } else if (getType()->isPointerType()) { |
| 718 | if (EvaluatePointer(this, Result, Info)) { |
| 719 | return true; |
| 720 | } |
| 721 | } else if (getType()->isRealFloatingType()) { |
| 722 | llvm::APFloat f(0.0); |
| 723 | if (EvaluateFloat(this, f, Info)) { |
| 724 | Result = APValue(f); |
| 725 | return true; |
| 726 | } |
| 727 | } |
Anders Carlsson | 165a70f | 2008-08-10 17:03:01 +0000 | [diff] [blame] | 728 | |
Anders Carlsson | c44eec6 | 2008-07-03 04:20:39 +0000 | [diff] [blame] | 729 | return false; |
| 730 | } |
Chris Lattner | 45b6b9d | 2008-10-06 06:49:02 +0000 | [diff] [blame] | 731 | |
| 732 | /// isEvaluatable - Call tryEvaluate to see if this expression can be constant |
| 733 | /// folded, but discard the result. |
| 734 | bool Expr::isEvaluatable(ASTContext &Ctx) const { |
| 735 | APValue V; |
| 736 | return tryEvaluate(V, Ctx); |
| 737 | } |