blob: d664c46081ea883610ae8a3892a3c85bd4d12fc8 [file] [log] [blame]
Chris Lattner4b009652007-07-25 00:24:17 +00001//===--- Expr.cpp - Expression AST Node Implementation --------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by Chris Lattner and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the Expr class and subclasses.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/AST/Expr.h"
15#include "clang/AST/ASTContext.h"
16#include "clang/AST/StmtVisitor.h"
17#include "clang/Lex/IdentifierTable.h"
18using namespace clang;
19
20//===----------------------------------------------------------------------===//
21// Primary Expressions.
22//===----------------------------------------------------------------------===//
23
24StringLiteral::StringLiteral(const char *strData, unsigned byteLength,
25 bool Wide, QualType t, SourceLocation firstLoc,
26 SourceLocation lastLoc) :
27 Expr(StringLiteralClass, t) {
28 // OPTIMIZE: could allocate this appended to the StringLiteral.
29 char *AStrData = new char[byteLength];
30 memcpy(AStrData, strData, byteLength);
31 StrData = AStrData;
32 ByteLength = byteLength;
33 IsWide = Wide;
34 firstTokLoc = firstLoc;
35 lastTokLoc = lastLoc;
36}
37
38StringLiteral::~StringLiteral() {
39 delete[] StrData;
40}
41
42bool UnaryOperator::isPostfix(Opcode Op) {
43 switch (Op) {
44 case PostInc:
45 case PostDec:
46 return true;
47 default:
48 return false;
49 }
50}
51
52/// getOpcodeStr - Turn an Opcode enum value into the punctuation char it
53/// corresponds to, e.g. "sizeof" or "[pre]++".
54const char *UnaryOperator::getOpcodeStr(Opcode Op) {
55 switch (Op) {
56 default: assert(0 && "Unknown unary operator");
57 case PostInc: return "++";
58 case PostDec: return "--";
59 case PreInc: return "++";
60 case PreDec: return "--";
61 case AddrOf: return "&";
62 case Deref: return "*";
63 case Plus: return "+";
64 case Minus: return "-";
65 case Not: return "~";
66 case LNot: return "!";
67 case Real: return "__real";
68 case Imag: return "__imag";
69 case SizeOf: return "sizeof";
70 case AlignOf: return "alignof";
71 case Extension: return "__extension__";
Chris Lattner0d9bcea2007-08-30 17:45:32 +000072 case OffsetOf: return "__builtin_offsetof";
Chris Lattner4b009652007-07-25 00:24:17 +000073 }
74}
75
76//===----------------------------------------------------------------------===//
77// Postfix Operators.
78//===----------------------------------------------------------------------===//
79
80CallExpr::CallExpr(Expr *fn, Expr **args, unsigned numargs, QualType t,
81 SourceLocation rparenloc)
Ted Kremeneke4acb9c2007-08-24 18:13:47 +000082 : Expr(CallExprClass, t), NumArgs(numargs) {
83 SubExprs = new Expr*[numargs+1];
84 SubExprs[FN] = fn;
Chris Lattner4b009652007-07-25 00:24:17 +000085 for (unsigned i = 0; i != numargs; ++i)
Ted Kremeneke4acb9c2007-08-24 18:13:47 +000086 SubExprs[i+ARGS_START] = args[i];
Chris Lattner4b009652007-07-25 00:24:17 +000087 RParenLoc = rparenloc;
88}
89
Steve Naroff8d3b1702007-08-08 22:15:55 +000090bool CallExpr::isBuiltinClassifyType(llvm::APSInt &Result) const {
91 // The following enum mimics gcc's internal "typeclass.h" file.
92 enum gcc_type_class {
93 no_type_class = -1,
94 void_type_class, integer_type_class, char_type_class,
95 enumeral_type_class, boolean_type_class,
96 pointer_type_class, reference_type_class, offset_type_class,
97 real_type_class, complex_type_class,
98 function_type_class, method_type_class,
99 record_type_class, union_type_class,
100 array_type_class, string_type_class,
101 lang_type_class
102 };
103 Result.setIsSigned(true);
104
105 // All simple function calls (e.g. func()) are implicitly cast to pointer to
106 // function. As a result, we try and obtain the DeclRefExpr from the
107 // ImplicitCastExpr.
108 const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(getCallee());
109 if (!ICE) // FIXME: deal with more complex calls (e.g. (func)(), (*func)()).
110 return false;
111 const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(ICE->getSubExpr());
112 if (!DRE)
113 return false;
114
115 // We have a DeclRefExpr.
116 if (strcmp(DRE->getDecl()->getName(), "__builtin_classify_type") == 0) {
117 // If no argument was supplied, default to "no_type_class". This isn't
118 // ideal, however it's what gcc does.
119 Result = static_cast<uint64_t>(no_type_class);
120 if (NumArgs >= 1) {
121 QualType argType = getArg(0)->getType();
122
123 if (argType->isVoidType())
124 Result = void_type_class;
125 else if (argType->isEnumeralType())
126 Result = enumeral_type_class;
127 else if (argType->isBooleanType())
128 Result = boolean_type_class;
129 else if (argType->isCharType())
130 Result = string_type_class; // gcc doesn't appear to use char_type_class
131 else if (argType->isIntegerType())
132 Result = integer_type_class;
133 else if (argType->isPointerType())
134 Result = pointer_type_class;
135 else if (argType->isReferenceType())
136 Result = reference_type_class;
137 else if (argType->isRealType())
138 Result = real_type_class;
139 else if (argType->isComplexType())
140 Result = complex_type_class;
141 else if (argType->isFunctionType())
142 Result = function_type_class;
143 else if (argType->isStructureType())
144 Result = record_type_class;
145 else if (argType->isUnionType())
146 Result = union_type_class;
147 else if (argType->isArrayType())
148 Result = array_type_class;
149 else if (argType->isUnionType())
150 Result = union_type_class;
151 else // FIXME: offset_type_class, method_type_class, & lang_type_class?
152 assert(1 && "CallExpr::isBuiltinClassifyType(): unimplemented type");
153 }
154 return true;
155 }
156 return false;
157}
158
Chris Lattner4b009652007-07-25 00:24:17 +0000159/// getOpcodeStr - Turn an Opcode enum value into the punctuation char it
160/// corresponds to, e.g. "<<=".
161const char *BinaryOperator::getOpcodeStr(Opcode Op) {
162 switch (Op) {
163 default: assert(0 && "Unknown binary operator");
164 case Mul: return "*";
165 case Div: return "/";
166 case Rem: return "%";
167 case Add: return "+";
168 case Sub: return "-";
169 case Shl: return "<<";
170 case Shr: return ">>";
171 case LT: return "<";
172 case GT: return ">";
173 case LE: return "<=";
174 case GE: return ">=";
175 case EQ: return "==";
176 case NE: return "!=";
177 case And: return "&";
178 case Xor: return "^";
179 case Or: return "|";
180 case LAnd: return "&&";
181 case LOr: return "||";
182 case Assign: return "=";
183 case MulAssign: return "*=";
184 case DivAssign: return "/=";
185 case RemAssign: return "%=";
186 case AddAssign: return "+=";
187 case SubAssign: return "-=";
188 case ShlAssign: return "<<=";
189 case ShrAssign: return ">>=";
190 case AndAssign: return "&=";
191 case XorAssign: return "^=";
192 case OrAssign: return "|=";
193 case Comma: return ",";
194 }
195}
196
Anders Carlsson762b7c72007-08-31 04:56:16 +0000197InitListExpr::InitListExpr(SourceLocation lbraceloc,
198 Expr **initexprs, unsigned numinits,
199 SourceLocation rbraceloc)
200 : Expr(InitListExprClass, QualType())
201 , NumInits(numinits)
202 , LBraceLoc(lbraceloc)
203 , RBraceLoc(rbraceloc)
204{
205 InitExprs = new Expr*[numinits];
206 for (unsigned i = 0; i != numinits; i++)
207 InitExprs[i] = initexprs[i];
208}
Chris Lattner4b009652007-07-25 00:24:17 +0000209
210//===----------------------------------------------------------------------===//
211// Generic Expression Routines
212//===----------------------------------------------------------------------===//
213
214/// hasLocalSideEffect - Return true if this immediate expression has side
215/// effects, not counting any sub-expressions.
216bool Expr::hasLocalSideEffect() const {
217 switch (getStmtClass()) {
218 default:
219 return false;
220 case ParenExprClass:
221 return cast<ParenExpr>(this)->getSubExpr()->hasLocalSideEffect();
222 case UnaryOperatorClass: {
223 const UnaryOperator *UO = cast<UnaryOperator>(this);
224
225 switch (UO->getOpcode()) {
226 default: return false;
227 case UnaryOperator::PostInc:
228 case UnaryOperator::PostDec:
229 case UnaryOperator::PreInc:
230 case UnaryOperator::PreDec:
231 return true; // ++/--
232
233 case UnaryOperator::Deref:
234 // Dereferencing a volatile pointer is a side-effect.
235 return getType().isVolatileQualified();
236 case UnaryOperator::Real:
237 case UnaryOperator::Imag:
238 // accessing a piece of a volatile complex is a side-effect.
239 return UO->getSubExpr()->getType().isVolatileQualified();
240
241 case UnaryOperator::Extension:
242 return UO->getSubExpr()->hasLocalSideEffect();
243 }
244 }
245 case BinaryOperatorClass:
246 return cast<BinaryOperator>(this)->isAssignmentOp();
Chris Lattner06078d22007-08-25 02:00:02 +0000247 case CompoundAssignOperatorClass:
Chris Lattner9c0da3b2007-08-25 01:55:00 +0000248 return true;
Chris Lattner4b009652007-07-25 00:24:17 +0000249
250 case MemberExprClass:
251 case ArraySubscriptExprClass:
252 // If the base pointer or element is to a volatile pointer/field, accessing
253 // if is a side effect.
254 return getType().isVolatileQualified();
255
256 case CallExprClass:
257 // TODO: check attributes for pure/const. "void foo() { strlen("bar"); }"
258 // should warn.
259 return true;
260
261 case CastExprClass:
262 // If this is a cast to void, check the operand. Otherwise, the result of
263 // the cast is unused.
264 if (getType()->isVoidType())
265 return cast<CastExpr>(this)->getSubExpr()->hasLocalSideEffect();
266 return false;
267 }
268}
269
270/// isLvalue - C99 6.3.2.1: an lvalue is an expression with an object type or an
271/// incomplete type other than void. Nonarray expressions that can be lvalues:
272/// - name, where name must be a variable
273/// - e[i]
274/// - (e), where e must be an lvalue
275/// - e.name, where e must be an lvalue
276/// - e->name
277/// - *e, the type of e cannot be a function type
278/// - string-constant
279/// - reference type [C++ [expr]]
280///
281Expr::isLvalueResult Expr::isLvalue() const {
282 // first, check the type (C99 6.3.2.1)
283 if (TR->isFunctionType()) // from isObjectType()
284 return LV_NotObjectType;
285
286 if (TR->isVoidType())
287 return LV_IncompleteVoidType;
288
289 if (TR->isReferenceType()) // C++ [expr]
290 return LV_Valid;
291
292 // the type looks fine, now check the expression
293 switch (getStmtClass()) {
294 case StringLiteralClass: // C99 6.5.1p4
295 case ArraySubscriptExprClass: // C99 6.5.3p4 (e1[e2] == (*((e1)+(e2))))
296 // For vectors, make sure base is an lvalue (i.e. not a function call).
297 if (cast<ArraySubscriptExpr>(this)->getBase()->getType()->isVectorType())
298 return cast<ArraySubscriptExpr>(this)->getBase()->isLvalue();
299 return LV_Valid;
300 case DeclRefExprClass: // C99 6.5.1p2
301 if (isa<VarDecl>(cast<DeclRefExpr>(this)->getDecl()))
302 return LV_Valid;
303 break;
304 case MemberExprClass: { // C99 6.5.2.3p4
305 const MemberExpr *m = cast<MemberExpr>(this);
306 return m->isArrow() ? LV_Valid : m->getBase()->isLvalue();
307 }
308 case UnaryOperatorClass: // C99 6.5.3p4
309 if (cast<UnaryOperator>(this)->getOpcode() == UnaryOperator::Deref)
310 return LV_Valid;
311 break;
312 case ParenExprClass: // C99 6.5.1p5
313 return cast<ParenExpr>(this)->getSubExpr()->isLvalue();
Chris Lattnera0d03a72007-08-03 17:31:20 +0000314 case OCUVectorElementExprClass:
315 if (cast<OCUVectorElementExpr>(this)->containsDuplicateElements())
Steve Naroffba67f692007-07-30 03:29:09 +0000316 return LV_DuplicateVectorComponents;
317 return LV_Valid;
Chris Lattner4b009652007-07-25 00:24:17 +0000318 default:
319 break;
320 }
321 return LV_InvalidExpression;
322}
323
324/// isModifiableLvalue - C99 6.3.2.1: an lvalue that does not have array type,
325/// does not have an incomplete type, does not have a const-qualified type, and
326/// if it is a structure or union, does not have any member (including,
327/// recursively, any member or element of all contained aggregates or unions)
328/// with a const-qualified type.
329Expr::isModifiableLvalueResult Expr::isModifiableLvalue() const {
330 isLvalueResult lvalResult = isLvalue();
331
332 switch (lvalResult) {
333 case LV_Valid: break;
334 case LV_NotObjectType: return MLV_NotObjectType;
335 case LV_IncompleteVoidType: return MLV_IncompleteVoidType;
Steve Naroffba67f692007-07-30 03:29:09 +0000336 case LV_DuplicateVectorComponents: return MLV_DuplicateVectorComponents;
Chris Lattner4b009652007-07-25 00:24:17 +0000337 case LV_InvalidExpression: return MLV_InvalidExpression;
338 }
339 if (TR.isConstQualified())
340 return MLV_ConstQualified;
341 if (TR->isArrayType())
342 return MLV_ArrayType;
343 if (TR->isIncompleteType())
344 return MLV_IncompleteType;
345
346 if (const RecordType *r = dyn_cast<RecordType>(TR.getCanonicalType())) {
347 if (r->hasConstFields())
348 return MLV_ConstQualified;
349 }
350 return MLV_Valid;
351}
352
353/// isIntegerConstantExpr - this recursive routine will test if an expression is
354/// an integer constant expression. Note: With the introduction of VLA's in
355/// C99 the result of the sizeof operator is no longer always a constant
356/// expression. The generalization of the wording to include any subexpression
357/// that is not evaluated (C99 6.6p3) means that nonconstant subexpressions
358/// can appear as operands to other operators (e.g. &&, ||, ?:). For instance,
359/// "0 || f()" can be treated as a constant expression. In C90 this expression,
360/// occurring in a context requiring a constant, would have been a constraint
361/// violation. FIXME: This routine currently implements C90 semantics.
362/// To properly implement C99 semantics this routine will need to evaluate
363/// expressions involving operators previously mentioned.
364
365/// FIXME: Pass up a reason why! Invalid operation in i-c-e, division by zero,
366/// comma, etc
367///
368/// FIXME: This should ext-warn on overflow during evaluation! ISO C does not
369/// permit this.
370///
371/// FIXME: Handle offsetof. Two things to do: Handle GCC's __builtin_offsetof
372/// to support gcc 4.0+ and handle the idiom GCC recognizes with a null pointer
373/// cast+dereference.
374bool Expr::isIntegerConstantExpr(llvm::APSInt &Result, ASTContext &Ctx,
375 SourceLocation *Loc, bool isEvaluated) const {
376 switch (getStmtClass()) {
377 default:
378 if (Loc) *Loc = getLocStart();
379 return false;
380 case ParenExprClass:
381 return cast<ParenExpr>(this)->getSubExpr()->
382 isIntegerConstantExpr(Result, Ctx, Loc, isEvaluated);
383 case IntegerLiteralClass:
384 Result = cast<IntegerLiteral>(this)->getValue();
385 break;
386 case CharacterLiteralClass: {
387 const CharacterLiteral *CL = cast<CharacterLiteral>(this);
388 Result.zextOrTrunc(Ctx.getTypeSize(getType(), CL->getLoc()));
389 Result = CL->getValue();
390 Result.setIsUnsigned(!getType()->isSignedIntegerType());
391 break;
392 }
Steve Naroffc6f0fd32007-08-02 04:09:23 +0000393 case TypesCompatibleExprClass: {
394 const TypesCompatibleExpr *TCE = cast<TypesCompatibleExpr>(this);
395 Result.zextOrTrunc(Ctx.getTypeSize(getType(), TCE->getLocStart()));
396 Result = TCE->typesAreCompatible();
Steve Naroff1200b5a2007-08-02 00:13:27 +0000397 break;
Steve Naroffc6f0fd32007-08-02 04:09:23 +0000398 }
Steve Naroff8d3b1702007-08-08 22:15:55 +0000399 case CallExprClass: {
400 const CallExpr *CE = cast<CallExpr>(this);
401 Result.zextOrTrunc(Ctx.getTypeSize(getType(), CE->getLocStart()));
402 if (CE->isBuiltinClassifyType(Result))
403 break;
404 if (Loc) *Loc = getLocStart();
405 return false;
406 }
Chris Lattner4b009652007-07-25 00:24:17 +0000407 case DeclRefExprClass:
408 if (const EnumConstantDecl *D =
409 dyn_cast<EnumConstantDecl>(cast<DeclRefExpr>(this)->getDecl())) {
410 Result = D->getInitVal();
411 break;
412 }
413 if (Loc) *Loc = getLocStart();
414 return false;
415 case UnaryOperatorClass: {
416 const UnaryOperator *Exp = cast<UnaryOperator>(this);
417
418 // Get the operand value. If this is sizeof/alignof, do not evalute the
419 // operand. This affects C99 6.6p3.
Chris Lattner5a9b6242007-08-23 21:42:50 +0000420 if (!Exp->isSizeOfAlignOfOp() &&
421 !Exp->getSubExpr()->isIntegerConstantExpr(Result, Ctx, Loc,isEvaluated))
Chris Lattner4b009652007-07-25 00:24:17 +0000422 return false;
423
424 switch (Exp->getOpcode()) {
425 // Address, indirect, pre/post inc/dec, etc are not valid constant exprs.
426 // See C99 6.6p3.
427 default:
428 if (Loc) *Loc = Exp->getOperatorLoc();
429 return false;
430 case UnaryOperator::Extension:
431 return true; // FIXME: this is wrong.
432 case UnaryOperator::SizeOf:
433 case UnaryOperator::AlignOf:
434 // sizeof(vla) is not a constantexpr: C99 6.5.3.4p2.
435 if (!Exp->getSubExpr()->getType()->isConstantSizeType(Ctx, Loc))
436 return false;
437
438 // Return the result in the right width.
439 Result.zextOrTrunc(Ctx.getTypeSize(getType(), Exp->getOperatorLoc()));
440
441 // Get information about the size or align.
442 if (Exp->getOpcode() == UnaryOperator::SizeOf)
443 Result = Ctx.getTypeSize(Exp->getSubExpr()->getType(),
444 Exp->getOperatorLoc());
445 else
446 Result = Ctx.getTypeAlign(Exp->getSubExpr()->getType(),
447 Exp->getOperatorLoc());
448 break;
449 case UnaryOperator::LNot: {
450 bool Val = Result != 0;
451 Result.zextOrTrunc(Ctx.getTypeSize(getType(), Exp->getOperatorLoc()));
452 Result = Val;
453 break;
454 }
455 case UnaryOperator::Plus:
456 break;
457 case UnaryOperator::Minus:
458 Result = -Result;
459 break;
460 case UnaryOperator::Not:
461 Result = ~Result;
462 break;
463 }
464 break;
465 }
466 case SizeOfAlignOfTypeExprClass: {
467 const SizeOfAlignOfTypeExpr *Exp = cast<SizeOfAlignOfTypeExpr>(this);
468 // alignof always evaluates to a constant.
469 if (Exp->isSizeOf() && !Exp->getArgumentType()->isConstantSizeType(Ctx,Loc))
470 return false;
471
472 // Return the result in the right width.
473 Result.zextOrTrunc(Ctx.getTypeSize(getType(), Exp->getOperatorLoc()));
474
475 // Get information about the size or align.
476 if (Exp->isSizeOf())
477 Result = Ctx.getTypeSize(Exp->getArgumentType(), Exp->getOperatorLoc());
478 else
479 Result = Ctx.getTypeAlign(Exp->getArgumentType(), Exp->getOperatorLoc());
480 break;
481 }
482 case BinaryOperatorClass: {
483 const BinaryOperator *Exp = cast<BinaryOperator>(this);
484
485 // The LHS of a constant expr is always evaluated and needed.
486 if (!Exp->getLHS()->isIntegerConstantExpr(Result, Ctx, Loc, isEvaluated))
487 return false;
488
489 llvm::APSInt RHS(Result);
490
491 // The short-circuiting &&/|| operators don't necessarily evaluate their
492 // RHS. Make sure to pass isEvaluated down correctly.
493 if (Exp->isLogicalOp()) {
494 bool RHSEval;
495 if (Exp->getOpcode() == BinaryOperator::LAnd)
496 RHSEval = Result != 0;
497 else {
498 assert(Exp->getOpcode() == BinaryOperator::LOr &&"Unexpected logical");
499 RHSEval = Result == 0;
500 }
501
502 if (!Exp->getRHS()->isIntegerConstantExpr(RHS, Ctx, Loc,
503 isEvaluated & RHSEval))
504 return false;
505 } else {
506 if (!Exp->getRHS()->isIntegerConstantExpr(RHS, Ctx, Loc, isEvaluated))
507 return false;
508 }
509
510 switch (Exp->getOpcode()) {
511 default:
512 if (Loc) *Loc = getLocStart();
513 return false;
514 case BinaryOperator::Mul:
515 Result *= RHS;
516 break;
517 case BinaryOperator::Div:
518 if (RHS == 0) {
519 if (!isEvaluated) break;
520 if (Loc) *Loc = getLocStart();
521 return false;
522 }
523 Result /= RHS;
524 break;
525 case BinaryOperator::Rem:
526 if (RHS == 0) {
527 if (!isEvaluated) break;
528 if (Loc) *Loc = getLocStart();
529 return false;
530 }
531 Result %= RHS;
532 break;
533 case BinaryOperator::Add: Result += RHS; break;
534 case BinaryOperator::Sub: Result -= RHS; break;
535 case BinaryOperator::Shl:
536 Result <<= RHS.getLimitedValue(Result.getBitWidth()-1);
537 break;
538 case BinaryOperator::Shr:
539 Result >>= RHS.getLimitedValue(Result.getBitWidth()-1);
540 break;
541 case BinaryOperator::LT: Result = Result < RHS; break;
542 case BinaryOperator::GT: Result = Result > RHS; break;
543 case BinaryOperator::LE: Result = Result <= RHS; break;
544 case BinaryOperator::GE: Result = Result >= RHS; break;
545 case BinaryOperator::EQ: Result = Result == RHS; break;
546 case BinaryOperator::NE: Result = Result != RHS; break;
547 case BinaryOperator::And: Result &= RHS; break;
548 case BinaryOperator::Xor: Result ^= RHS; break;
549 case BinaryOperator::Or: Result |= RHS; break;
550 case BinaryOperator::LAnd:
551 Result = Result != 0 && RHS != 0;
552 break;
553 case BinaryOperator::LOr:
554 Result = Result != 0 || RHS != 0;
555 break;
556
557 case BinaryOperator::Comma:
558 // C99 6.6p3: "shall not contain assignment, ..., or comma operators,
559 // *except* when they are contained within a subexpression that is not
560 // evaluated". Note that Assignment can never happen due to constraints
561 // on the LHS subexpr, so we don't need to check it here.
562 if (isEvaluated) {
563 if (Loc) *Loc = getLocStart();
564 return false;
565 }
566
567 // The result of the constant expr is the RHS.
568 Result = RHS;
569 return true;
570 }
571
572 assert(!Exp->isAssignmentOp() && "LHS can't be a constant expr!");
573 break;
574 }
575 case ImplicitCastExprClass:
576 case CastExprClass: {
577 const Expr *SubExpr;
578 SourceLocation CastLoc;
579 if (const CastExpr *C = dyn_cast<CastExpr>(this)) {
580 SubExpr = C->getSubExpr();
581 CastLoc = C->getLParenLoc();
582 } else {
583 SubExpr = cast<ImplicitCastExpr>(this)->getSubExpr();
584 CastLoc = getLocStart();
585 }
586
587 // C99 6.6p6: shall only convert arithmetic types to integer types.
588 if (!SubExpr->getType()->isArithmeticType() ||
589 !getType()->isIntegerType()) {
590 if (Loc) *Loc = SubExpr->getLocStart();
591 return false;
592 }
593
594 // Handle simple integer->integer casts.
595 if (SubExpr->getType()->isIntegerType()) {
596 if (!SubExpr->isIntegerConstantExpr(Result, Ctx, Loc, isEvaluated))
597 return false;
598
599 // Figure out if this is a truncate, extend or noop cast.
600 unsigned DestWidth = Ctx.getTypeSize(getType(), CastLoc);
601
602 // If the input is signed, do a sign extend, noop, or truncate.
603 if (SubExpr->getType()->isSignedIntegerType())
604 Result.sextOrTrunc(DestWidth);
605 else // If the input is unsigned, do a zero extend, noop, or truncate.
606 Result.zextOrTrunc(DestWidth);
607 break;
608 }
609
610 // Allow floating constants that are the immediate operands of casts or that
611 // are parenthesized.
612 const Expr *Operand = SubExpr;
613 while (const ParenExpr *PE = dyn_cast<ParenExpr>(Operand))
614 Operand = PE->getSubExpr();
615
616 if (const FloatingLiteral *FL = dyn_cast<FloatingLiteral>(Operand)) {
617 // FIXME: Evaluate this correctly!
618 Result = (int)FL->getValue();
619 break;
620 }
621 if (Loc) *Loc = Operand->getLocStart();
622 return false;
623 }
624 case ConditionalOperatorClass: {
625 const ConditionalOperator *Exp = cast<ConditionalOperator>(this);
626
627 if (!Exp->getCond()->isIntegerConstantExpr(Result, Ctx, Loc, isEvaluated))
628 return false;
629
630 const Expr *TrueExp = Exp->getLHS();
631 const Expr *FalseExp = Exp->getRHS();
632 if (Result == 0) std::swap(TrueExp, FalseExp);
633
634 // Evaluate the false one first, discard the result.
635 if (!FalseExp->isIntegerConstantExpr(Result, Ctx, Loc, false))
636 return false;
637 // Evalute the true one, capture the result.
638 if (!TrueExp->isIntegerConstantExpr(Result, Ctx, Loc, isEvaluated))
639 return false;
640 break;
641 }
642 }
643
644 // Cases that are valid constant exprs fall through to here.
645 Result.setIsUnsigned(getType()->isUnsignedIntegerType());
646 return true;
647}
648
649
650/// isNullPointerConstant - C99 6.3.2.3p3 - Return true if this is either an
651/// integer constant expression with the value zero, or if this is one that is
652/// cast to void*.
653bool Expr::isNullPointerConstant(ASTContext &Ctx) const {
654 // Strip off a cast to void*, if it exists.
655 if (const CastExpr *CE = dyn_cast<CastExpr>(this)) {
656 // Check that it is a cast to void*.
657 if (const PointerType *PT = dyn_cast<PointerType>(CE->getType())) {
658 QualType Pointee = PT->getPointeeType();
659 if (Pointee.getQualifiers() == 0 && Pointee->isVoidType() && // to void*
660 CE->getSubExpr()->getType()->isIntegerType()) // from int.
661 return CE->getSubExpr()->isNullPointerConstant(Ctx);
662 }
Steve Naroff1d6b2472007-08-28 21:20:34 +0000663 } else if (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(this)) {
Steve Naroff3d052872007-08-29 00:00:02 +0000664 // Ignore the ImplicitCastExpr type entirely.
665 return ICE->getSubExpr()->isNullPointerConstant(Ctx);
Chris Lattner4b009652007-07-25 00:24:17 +0000666 } else if (const ParenExpr *PE = dyn_cast<ParenExpr>(this)) {
667 // Accept ((void*)0) as a null pointer constant, as many other
668 // implementations do.
669 return PE->getSubExpr()->isNullPointerConstant(Ctx);
670 }
671
672 // This expression must be an integer type.
673 if (!getType()->isIntegerType())
674 return false;
675
676 // If we have an integer constant expression, we need to *evaluate* it and
677 // test for the value 0.
678 llvm::APSInt Val(32);
679 return isIntegerConstantExpr(Val, Ctx, 0, true) && Val == 0;
680}
Steve Naroffc11705f2007-07-28 23:10:27 +0000681
Chris Lattnera0d03a72007-08-03 17:31:20 +0000682unsigned OCUVectorElementExpr::getNumElements() const {
Chris Lattner50547852007-08-03 16:00:20 +0000683 return strlen(Accessor.getName());
684}
685
686
Chris Lattnerf4bf5512007-08-02 21:47:28 +0000687/// getComponentType - Determine whether the components of this access are
688/// "point" "color" or "texture" elements.
Chris Lattnera0d03a72007-08-03 17:31:20 +0000689OCUVectorElementExpr::ElementType
690OCUVectorElementExpr::getElementType() const {
Steve Naroffc11705f2007-07-28 23:10:27 +0000691 // derive the component type, no need to waste space.
692 const char *compStr = Accessor.getName();
Chris Lattnerabf25b32007-08-02 22:20:00 +0000693
Chris Lattner9096b792007-08-02 22:33:49 +0000694 if (OCUVectorType::getPointAccessorIdx(*compStr) != -1) return Point;
695 if (OCUVectorType::getColorAccessorIdx(*compStr) != -1) return Color;
Chris Lattnerabf25b32007-08-02 22:20:00 +0000696
Chris Lattner9096b792007-08-02 22:33:49 +0000697 assert(OCUVectorType::getTextureAccessorIdx(*compStr) != -1 &&
Chris Lattnerabf25b32007-08-02 22:20:00 +0000698 "getComponentType(): Illegal accessor");
699 return Texture;
Steve Naroffc11705f2007-07-28 23:10:27 +0000700}
Steve Naroffba67f692007-07-30 03:29:09 +0000701
Chris Lattnera0d03a72007-08-03 17:31:20 +0000702/// containsDuplicateElements - Return true if any element access is
Chris Lattnerf4bf5512007-08-02 21:47:28 +0000703/// repeated.
Chris Lattnera0d03a72007-08-03 17:31:20 +0000704bool OCUVectorElementExpr::containsDuplicateElements() const {
Steve Naroffba67f692007-07-30 03:29:09 +0000705 const char *compStr = Accessor.getName();
706 unsigned length = strlen(compStr);
707
708 for (unsigned i = 0; i < length-1; i++) {
709 const char *s = compStr+i;
710 for (const char c = *s++; *s; s++)
711 if (c == *s)
712 return true;
713 }
714 return false;
715}
Chris Lattner42158e72007-08-02 23:36:59 +0000716
717/// getEncodedElementAccess - We encode fields with two bits per component.
Chris Lattnera0d03a72007-08-03 17:31:20 +0000718unsigned OCUVectorElementExpr::getEncodedElementAccess() const {
Chris Lattner42158e72007-08-02 23:36:59 +0000719 const char *compStr = Accessor.getName();
Chris Lattnera0d03a72007-08-03 17:31:20 +0000720 unsigned length = getNumElements();
Chris Lattner42158e72007-08-02 23:36:59 +0000721
722 unsigned Result = 0;
723
724 while (length--) {
725 Result <<= 2;
726 int Idx = OCUVectorType::getAccessorIdx(compStr[length]);
727 assert(Idx != -1 && "Invalid accessor letter");
728 Result |= Idx;
729 }
730 return Result;
731}
732
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000733//===----------------------------------------------------------------------===//
734// Child Iterators for iterating over subexpressions/substatements
735//===----------------------------------------------------------------------===//
736
737// DeclRefExpr
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000738Stmt::child_iterator DeclRefExpr::child_begin() { return NULL; }
739Stmt::child_iterator DeclRefExpr::child_end() { return NULL; }
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000740
741// PreDefinedExpr
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000742Stmt::child_iterator PreDefinedExpr::child_begin() { return NULL; }
743Stmt::child_iterator PreDefinedExpr::child_end() { return NULL; }
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000744
745// IntegerLiteral
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000746Stmt::child_iterator IntegerLiteral::child_begin() { return NULL; }
747Stmt::child_iterator IntegerLiteral::child_end() { return NULL; }
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000748
749// CharacterLiteral
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000750Stmt::child_iterator CharacterLiteral::child_begin() { return NULL; }
751Stmt::child_iterator CharacterLiteral::child_end() { return NULL; }
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000752
753// FloatingLiteral
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000754Stmt::child_iterator FloatingLiteral::child_begin() { return NULL; }
755Stmt::child_iterator FloatingLiteral::child_end() { return NULL; }
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000756
Chris Lattner1de66eb2007-08-26 03:42:43 +0000757// ImaginaryLiteral
758Stmt::child_iterator ImaginaryLiteral::child_begin() {
759 return reinterpret_cast<Stmt**>(&Val);
760}
761Stmt::child_iterator ImaginaryLiteral::child_end() {
762 return reinterpret_cast<Stmt**>(&Val)+1;
763}
764
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000765// StringLiteral
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000766Stmt::child_iterator StringLiteral::child_begin() { return NULL; }
767Stmt::child_iterator StringLiteral::child_end() { return NULL; }
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000768
769// ParenExpr
770Stmt::child_iterator ParenExpr::child_begin() {
771 return reinterpret_cast<Stmt**>(&Val);
772}
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000773Stmt::child_iterator ParenExpr::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000774 return reinterpret_cast<Stmt**>(&Val)+1;
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000775}
776
777// UnaryOperator
778Stmt::child_iterator UnaryOperator::child_begin() {
779 return reinterpret_cast<Stmt**>(&Val);
780}
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000781Stmt::child_iterator UnaryOperator::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000782 return reinterpret_cast<Stmt**>(&Val)+1;
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000783}
784
785// SizeOfAlignOfTypeExpr
Chris Lattner1de66eb2007-08-26 03:42:43 +0000786Stmt::child_iterator SizeOfAlignOfTypeExpr::child_begin() { return NULL; }
787Stmt::child_iterator SizeOfAlignOfTypeExpr::child_end() { return NULL; }
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000788
789// ArraySubscriptExpr
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000790Stmt::child_iterator ArraySubscriptExpr::child_begin() {
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000791 return reinterpret_cast<Stmt**>(&SubExprs);
792}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000793Stmt::child_iterator ArraySubscriptExpr::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000794 return reinterpret_cast<Stmt**>(&SubExprs)+END_EXPR;
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000795}
796
797// CallExpr
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000798Stmt::child_iterator CallExpr::child_begin() {
Ted Kremenek15ede502007-08-27 21:11:44 +0000799 return reinterpret_cast<Stmt**>(&SubExprs[0]);
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000800}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000801Stmt::child_iterator CallExpr::child_end() {
Ted Kremenek15ede502007-08-27 21:11:44 +0000802 return reinterpret_cast<Stmt**>(&SubExprs[NumArgs+ARGS_START]);
Ted Kremeneke4acb9c2007-08-24 18:13:47 +0000803}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000804
805// MemberExpr
806Stmt::child_iterator MemberExpr::child_begin() {
807 return reinterpret_cast<Stmt**>(&Base);
808}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000809Stmt::child_iterator MemberExpr::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000810 return reinterpret_cast<Stmt**>(&Base)+1;
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000811}
812
813// OCUVectorElementExpr
814Stmt::child_iterator OCUVectorElementExpr::child_begin() {
815 return reinterpret_cast<Stmt**>(&Base);
816}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000817Stmt::child_iterator OCUVectorElementExpr::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000818 return reinterpret_cast<Stmt**>(&Base)+1;
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000819}
820
821// CompoundLiteralExpr
822Stmt::child_iterator CompoundLiteralExpr::child_begin() {
823 return reinterpret_cast<Stmt**>(&Init);
824}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000825Stmt::child_iterator CompoundLiteralExpr::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000826 return reinterpret_cast<Stmt**>(&Init)+1;
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000827}
828
829// ImplicitCastExpr
830Stmt::child_iterator ImplicitCastExpr::child_begin() {
831 return reinterpret_cast<Stmt**>(&Op);
832}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000833Stmt::child_iterator ImplicitCastExpr::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000834 return reinterpret_cast<Stmt**>(&Op)+1;
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000835}
836
837// CastExpr
838Stmt::child_iterator CastExpr::child_begin() {
839 return reinterpret_cast<Stmt**>(&Op);
840}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000841Stmt::child_iterator CastExpr::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000842 return reinterpret_cast<Stmt**>(&Op)+1;
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000843}
844
845// BinaryOperator
846Stmt::child_iterator BinaryOperator::child_begin() {
847 return reinterpret_cast<Stmt**>(&SubExprs);
848}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000849Stmt::child_iterator BinaryOperator::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000850 return reinterpret_cast<Stmt**>(&SubExprs)+END_EXPR;
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000851}
852
853// ConditionalOperator
854Stmt::child_iterator ConditionalOperator::child_begin() {
855 return reinterpret_cast<Stmt**>(&SubExprs);
856}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000857Stmt::child_iterator ConditionalOperator::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000858 return reinterpret_cast<Stmt**>(&SubExprs)+END_EXPR;
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000859}
860
861// AddrLabelExpr
862Stmt::child_iterator AddrLabelExpr::child_begin() { return NULL; }
863Stmt::child_iterator AddrLabelExpr::child_end() { return NULL; }
864
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000865// StmtExpr
866Stmt::child_iterator StmtExpr::child_begin() {
867 return reinterpret_cast<Stmt**>(&SubStmt);
868}
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000869Stmt::child_iterator StmtExpr::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000870 return reinterpret_cast<Stmt**>(&SubStmt)+1;
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000871}
872
873// TypesCompatibleExpr
874Stmt::child_iterator TypesCompatibleExpr::child_begin() { return NULL; }
875Stmt::child_iterator TypesCompatibleExpr::child_end() { return NULL; }
876
877// ChooseExpr
878Stmt::child_iterator ChooseExpr::child_begin() {
879 return reinterpret_cast<Stmt**>(&SubExprs);
880}
881
882Stmt::child_iterator ChooseExpr::child_end() {
Chris Lattner1de66eb2007-08-26 03:42:43 +0000883 return reinterpret_cast<Stmt**>(&SubExprs)+END_EXPR;
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000884}
885
Anders Carlsson762b7c72007-08-31 04:56:16 +0000886// InitListExpr
887Stmt::child_iterator InitListExpr::child_begin() {
888 return reinterpret_cast<Stmt**>(&InitExprs[0]);
889}
890Stmt::child_iterator InitListExpr::child_end() {
891 return reinterpret_cast<Stmt**>(&InitExprs[NumInits]);
892}
893
Ted Kremenek9fdc8a92007-08-24 20:06:47 +0000894// ObjCStringLiteral
895Stmt::child_iterator ObjCStringLiteral::child_begin() { return NULL; }
896Stmt::child_iterator ObjCStringLiteral::child_end() { return NULL; }
897
898// ObjCEncodeExpr
899Stmt::child_iterator ObjCEncodeExpr::child_begin() { return NULL; }
900Stmt::child_iterator ObjCEncodeExpr::child_end() { return NULL; }
901