blob: c61d71acada13d695eac73d062bfb1b7409c2d34 [file] [log] [blame]
Anders Carlssond76cead2008-01-26 01:36:00 +00001//===--- CGExprConstant.cpp - Emit LLVM Code from Constant Expressions ----===//
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 contains code to emit Constant Expr nodes as LLVM code.
11//
12//===----------------------------------------------------------------------===//
13
14#include "CodeGenFunction.h"
15#include "CodeGenModule.h"
16#include "clang/AST/AST.h"
17#include "llvm/Constants.h"
18#include "llvm/Function.h"
19#include "llvm/GlobalVariable.h"
20#include "llvm/Support/Compiler.h"
21using namespace clang;
22using namespace CodeGen;
23
24namespace {
Anders Carlssonde598292008-01-26 04:30:23 +000025class VISIBILITY_HIDDEN ConstExprEmitter :
26 public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
Anders Carlssond76cead2008-01-26 01:36:00 +000027 CodeGenModule &CGM;
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +000028 CodeGenFunction *CGF;
Anders Carlssond76cead2008-01-26 01:36:00 +000029public:
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +000030 ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
31 : CGM(cgm), CGF(cgf) {
Anders Carlssond76cead2008-01-26 01:36:00 +000032 }
33
34 //===--------------------------------------------------------------------===//
35 // Visitor Methods
36 //===--------------------------------------------------------------------===//
37
38 llvm::Constant *VisitStmt(Stmt *S) {
39 CGM.WarnUnsupported(S, "constant expression");
Chris Lattner8cd0e932008-03-05 18:54:05 +000040 QualType T = cast<Expr>(S)->getType();
41 return llvm::UndefValue::get(CGM.getTypes().ConvertType(T));
Anders Carlssond76cead2008-01-26 01:36:00 +000042 }
43
44 llvm::Constant *VisitParenExpr(ParenExpr *PE) {
45 return Visit(PE->getSubExpr());
46 }
47
48 // Leaves
49 llvm::Constant *VisitIntegerLiteral(const IntegerLiteral *E) {
50 return llvm::ConstantInt::get(E->getValue());
51 }
52 llvm::Constant *VisitFloatingLiteral(const FloatingLiteral *E) {
Chris Lattner70c38672008-04-20 00:45:53 +000053 return llvm::ConstantFP::get(E->getValue());
Anders Carlssond76cead2008-01-26 01:36:00 +000054 }
55 llvm::Constant *VisitCharacterLiteral(const CharacterLiteral *E) {
56 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
57 }
58 llvm::Constant *VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
59 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
60 }
61
62 llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
63 return Visit(E->getInitializer());
64 }
65
66 llvm::Constant *VisitCastExpr(const CastExpr* E) {
67 llvm::Constant *C = Visit(E->getSubExpr());
68
69 return EmitConversion(C, E->getSubExpr()->getType(), E->getType());
70 }
Devang Patel04520012008-02-05 02:39:50 +000071
Chris Lattner3e254fb2008-04-08 04:40:51 +000072 llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
73 return Visit(DAE->getExpr());
74 }
75
Devang Patel04520012008-02-05 02:39:50 +000076 llvm::Constant *EmitArrayInitialization(InitListExpr *ILE,
77 const llvm::ArrayType *AType) {
Eli Friedman8ac5b3a2008-02-21 17:45:41 +000078 std::vector<llvm::Constant*> Elts;
79 unsigned NumInitElements = ILE->getNumInits();
80 // FIXME: Check for wide strings
81 if (NumInitElements > 0 && isa<StringLiteral>(ILE->getInit(0)) &&
82 ILE->getType()->getAsArrayType()->getElementType()->isCharType())
83 return Visit(ILE->getInit(0));
Devang Patel04520012008-02-05 02:39:50 +000084 const llvm::Type *ElemTy = AType->getElementType();
85 unsigned NumElements = AType->getNumElements();
86
87 // Initialising an array requires us to automatically
88 // initialise any elements that have not been initialised explicitly
89 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
90
91 // Copy initializer elements.
92 unsigned i = 0;
93 for (; i < NumInitableElts; ++i) {
Devang Patel04520012008-02-05 02:39:50 +000094 llvm::Constant *C = Visit(ILE->getInit(i));
Devang Patel04520012008-02-05 02:39:50 +000095 assert (C && "Failed to create initializer expression");
96 Elts.push_back(C);
97 }
98
99 // Initialize remaining array elements.
100 for (; i < NumElements; ++i)
101 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
102
103 return llvm::ConstantArray::get(AType, Elts);
104 }
105
106 llvm::Constant *EmitStructInitialization(InitListExpr *ILE,
107 const llvm::StructType *SType) {
108
Eli Friedman0242cd22008-05-20 00:11:07 +0000109 RecordDecl *RD = ILE->getType()->getAsRecordType()->getDecl();
Devang Patel04520012008-02-05 02:39:50 +0000110 std::vector<llvm::Constant*> Elts;
Devang Patel04520012008-02-05 02:39:50 +0000111
Eli Friedman0242cd22008-05-20 00:11:07 +0000112 // Initialize the whole structure to zero.
113 for (unsigned i = 0; i < SType->getNumElements(); ++i) {
Devang Patel04520012008-02-05 02:39:50 +0000114 const llvm::Type *FieldTy = SType->getElementType(i);
115 Elts.push_back(llvm::Constant::getNullValue(FieldTy));
116 }
Eli Friedman0242cd22008-05-20 00:11:07 +0000117
118 // Copy initializer elements. Skip padding fields.
119 unsigned EltNo = 0; // Element no in ILE
120 int FieldNo = 0; // Field no in RecordDecl
121 while (EltNo < ILE->getNumInits() && FieldNo < RD->getNumMembers()) {
122 FieldDecl* curField = RD->getMember(FieldNo);
123 FieldNo++;
124 if (!curField->getIdentifier())
125 continue;
126
127 llvm::Constant *C = Visit(ILE->getInit(EltNo));
128 assert (C && "Failed to create initializer expression");
129
130 if (curField->isBitField()) {
131 CGM.WarnUnsupported(ILE->getInit(EltNo), "bitfield initialization");
132 } else {
133 Elts[CGM.getTypes().getLLVMFieldNo(curField)] = C;
134 }
135 EltNo++;
136 }
137
Devang Patel04520012008-02-05 02:39:50 +0000138 return llvm::ConstantStruct::get(SType, Elts);
139 }
140
141 llvm::Constant *EmitVectorInitialization(InitListExpr *ILE,
142 const llvm::VectorType *VType) {
143
144 std::vector<llvm::Constant*> Elts;
145 unsigned NumInitElements = ILE->getNumInits();
146 unsigned NumElements = VType->getNumElements();
147
148 assert (NumInitElements == NumElements
149 && "Unsufficient vector init elelments");
150 // Copy initializer elements.
151 unsigned i = 0;
152 for (; i < NumElements; ++i) {
Devang Patel04520012008-02-05 02:39:50 +0000153 llvm::Constant *C = Visit(ILE->getInit(i));
Devang Patel04520012008-02-05 02:39:50 +0000154 assert (C && "Failed to create initializer expression");
155 Elts.push_back(C);
156 }
157
158 return llvm::ConstantVector::get(VType, Elts);
159 }
160
Anders Carlssond76cead2008-01-26 01:36:00 +0000161 llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
Anders Carlsson35ab4f92008-01-29 01:15:48 +0000162 const llvm::CompositeType *CType =
163 dyn_cast<llvm::CompositeType>(ConvertType(ILE->getType()));
164
165 if (!CType) {
166 // We have a scalar in braces. Just use the first element.
167 return Visit(ILE->getInit(0));
Anders Carlssond76cead2008-01-26 01:36:00 +0000168 }
Anders Carlsson35ab4f92008-01-29 01:15:48 +0000169
Devang Patel04520012008-02-05 02:39:50 +0000170 if (const llvm::ArrayType *AType = dyn_cast<llvm::ArrayType>(CType))
171 return EmitArrayInitialization(ILE, AType);
Devang Pateld72d1012008-01-29 23:23:18 +0000172
Devang Patel04520012008-02-05 02:39:50 +0000173 if (const llvm::StructType *SType = dyn_cast<llvm::StructType>(CType))
174 return EmitStructInitialization(ILE, SType);
Devang Pateld72d1012008-01-29 23:23:18 +0000175
Devang Patel04520012008-02-05 02:39:50 +0000176 if (const llvm::VectorType *VType = dyn_cast<llvm::VectorType>(CType))
177 return EmitVectorInitialization(ILE, VType);
Anders Carlssond76cead2008-01-26 01:36:00 +0000178
179 // Make sure we have an array at this point
Devang Patel04520012008-02-05 02:39:50 +0000180 assert(0 && "Unable to handle InitListExpr");
Chris Lattnercbfb5512008-03-01 08:45:05 +0000181 // Get rid of control reaches end of void function warning.
182 // Not reached.
183 return 0;
Anders Carlssond76cead2008-01-26 01:36:00 +0000184 }
Eli Friedman1693a3f2008-02-21 17:57:49 +0000185
Anders Carlssond76cead2008-01-26 01:36:00 +0000186 llvm::Constant *VisitImplicitCastExpr(ImplicitCastExpr *ICExpr) {
Eli Friedman1693a3f2008-02-21 17:57:49 +0000187 Expr* SExpr = ICExpr->getSubExpr();
188 QualType SType = SExpr->getType();
189 llvm::Constant *C; // the intermediate expression
190 QualType T; // the type of the intermediate expression
191 if (SType->isArrayType()) {
192 // Arrays decay to a pointer to the first element
193 // VLAs would require special handling, but they can't occur here
194 C = EmitLValue(SExpr);
Anders Carlssond76cead2008-01-26 01:36:00 +0000195 llvm::Constant *Idx0 = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
Anders Carlssond76cead2008-01-26 01:36:00 +0000196 llvm::Constant *Ops[] = {Idx0, Idx0};
197 C = llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
Eli Friedman1693a3f2008-02-21 17:57:49 +0000198
199 QualType ElemType = SType->getAsArrayType()->getElementType();
200 T = CGM.getContext().getPointerType(ElemType);
201 } else if (SType->isFunctionType()) {
202 // Function types decay to a pointer to the function
203 C = EmitLValue(SExpr);
204 T = CGM.getContext().getPointerType(SType);
205 } else {
206 C = Visit(SExpr);
207 T = SType;
Anders Carlssond76cead2008-01-26 01:36:00 +0000208 }
Eli Friedman1693a3f2008-02-21 17:57:49 +0000209
210 // Perform the conversion; note that an implicit cast can both promote
211 // and convert an array/function
212 return EmitConversion(C, T, ICExpr->getType());
Anders Carlssond76cead2008-01-26 01:36:00 +0000213 }
Eli Friedman1693a3f2008-02-21 17:57:49 +0000214
Anders Carlssond76cead2008-01-26 01:36:00 +0000215 llvm::Constant *VisitStringLiteral(StringLiteral *E) {
216 const char *StrData = E->getStrData();
217 unsigned Len = E->getByteLength();
Chris Lattnera6dcce32008-02-11 00:02:17 +0000218 assert(!E->getType()->isPointerType() && "Strings are always arrays");
Anders Carlssond76cead2008-01-26 01:36:00 +0000219
220 // Otherwise this must be a string initializing an array in a static
221 // initializer. Don't emit it as the address of the string, emit the string
222 // data itself as an inline array.
223 const ConstantArrayType *CAT = E->getType()->getAsConstantArrayType();
224 assert(CAT && "String isn't pointer or array!");
225
226 std::string Str(StrData, StrData + Len);
227 // Null terminate the string before potentially truncating it.
228 // FIXME: What about wchar_t strings?
229 Str.push_back(0);
230
231 uint64_t RealLen = CAT->getSize().getZExtValue();
232 // String or grow the initializer to the required size.
233 if (RealLen != Str.size())
234 Str.resize(RealLen);
235
236 return llvm::ConstantArray::get(Str, false);
237 }
238
239 llvm::Constant *VisitDeclRefExpr(DeclRefExpr *E) {
240 const ValueDecl *Decl = E->getDecl();
Anders Carlsson544cbe52008-01-29 01:28:48 +0000241 if (const EnumConstantDecl *EC = dyn_cast<EnumConstantDecl>(Decl))
242 return llvm::ConstantInt::get(EC->getInitVal());
Anders Carlssond76cead2008-01-26 01:36:00 +0000243 assert(0 && "Unsupported decl ref type!");
244 return 0;
245 }
246
247 llvm::Constant *VisitSizeOfAlignOfTypeExpr(const SizeOfAlignOfTypeExpr *E) {
248 return EmitSizeAlignOf(E->getArgumentType(), E->getType(), E->isSizeOf());
249 }
250
251 // Unary operators
252 llvm::Constant *VisitUnaryPlus(const UnaryOperator *E) {
253 return Visit(E->getSubExpr());
254 }
255 llvm::Constant *VisitUnaryMinus(const UnaryOperator *E) {
256 return llvm::ConstantExpr::getNeg(Visit(E->getSubExpr()));
257 }
258 llvm::Constant *VisitUnaryNot(const UnaryOperator *E) {
259 return llvm::ConstantExpr::getNot(Visit(E->getSubExpr()));
260 }
261 llvm::Constant *VisitUnaryLNot(const UnaryOperator *E) {
262 llvm::Constant *SubExpr = Visit(E->getSubExpr());
263
264 if (E->getSubExpr()->getType()->isRealFloatingType()) {
265 // Compare against 0.0 for fp scalars.
266 llvm::Constant *Zero = llvm::Constant::getNullValue(SubExpr->getType());
267 SubExpr = llvm::ConstantExpr::getFCmp(llvm::FCmpInst::FCMP_UEQ, SubExpr,
268 Zero);
269 } else {
270 assert((E->getSubExpr()->getType()->isIntegerType() ||
271 E->getSubExpr()->getType()->isPointerType()) &&
272 "Unknown scalar type to convert");
273 // Compare against an integer or pointer null.
274 llvm::Constant *Zero = llvm::Constant::getNullValue(SubExpr->getType());
275 SubExpr = llvm::ConstantExpr::getICmp(llvm::ICmpInst::ICMP_EQ, SubExpr,
276 Zero);
277 }
278
279 return llvm::ConstantExpr::getZExt(SubExpr, ConvertType(E->getType()));
280 }
281 llvm::Constant *VisitUnarySizeOf(const UnaryOperator *E) {
282 return EmitSizeAlignOf(E->getSubExpr()->getType(), E->getType(), true);
283 }
284 llvm::Constant *VisitUnaryAlignOf(const UnaryOperator *E) {
285 return EmitSizeAlignOf(E->getSubExpr()->getType(), E->getType(), false);
286 }
Anders Carlsson00b837c2008-01-26 02:08:50 +0000287 llvm::Constant *VisitUnaryAddrOf(const UnaryOperator *E) {
288 return EmitLValue(E->getSubExpr());
289 }
Anders Carlsson52774ad2008-01-29 15:56:48 +0000290 llvm::Constant *VisitUnaryOffsetOf(const UnaryOperator *E) {
291 int64_t Val = E->evaluateOffsetOf(CGM.getContext());
292
293 assert(E->getType()->isIntegerType() && "Result type must be an integer!");
294
Chris Lattner8cd0e932008-03-05 18:54:05 +0000295 uint32_t ResultWidth =
296 static_cast<uint32_t>(CGM.getContext().getTypeSize(E->getType()));
Anders Carlsson52774ad2008-01-29 15:56:48 +0000297 return llvm::ConstantInt::get(llvm::APInt(ResultWidth, Val));
298 }
Eli Friedmana36799c2008-05-29 11:22:45 +0000299
300 llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
301 return Visit(E->getSubExpr());
302 }
Anders Carlssond76cead2008-01-26 01:36:00 +0000303
Anders Carlssoncc4855e2008-01-29 01:33:32 +0000304 // Binary operators
305 llvm::Constant *VisitBinOr(const BinaryOperator *E) {
306 llvm::Constant *LHS = Visit(E->getLHS());
307 llvm::Constant *RHS = Visit(E->getRHS());
308
309 return llvm::ConstantExpr::getOr(LHS, RHS);
310 }
Anders Carlsson4d852ee2008-02-05 20:54:21 +0000311 llvm::Constant *VisitBinSub(const BinaryOperator *E) {
312 llvm::Constant *LHS = Visit(E->getLHS());
313 llvm::Constant *RHS = Visit(E->getRHS());
314
315 if (!isa<llvm::PointerType>(RHS->getType())) {
316 // pointer - int
317 if (isa<llvm::PointerType>(LHS->getType())) {
318 llvm::Constant *Idx = llvm::ConstantExpr::getNeg(RHS);
319
320 return llvm::ConstantExpr::getGetElementPtr(LHS, &Idx, 1);
321 }
322
323 // int - int
324 return llvm::ConstantExpr::getSub(LHS, RHS);
325 }
326
327 assert(0 && "Unhandled bin sub case!");
328 return 0;
329 }
330
331 llvm::Constant *VisitBinShl(const BinaryOperator *E) {
332 llvm::Constant *LHS = Visit(E->getLHS());
333 llvm::Constant *RHS = Visit(E->getRHS());
Anders Carlssoncc4855e2008-01-29 01:33:32 +0000334
Anders Carlsson4d852ee2008-02-05 20:54:21 +0000335 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
336 // RHS to the same size as the LHS.
337 if (LHS->getType() != RHS->getType())
338 RHS = llvm::ConstantExpr::getIntegerCast(RHS, LHS->getType(), false);
339
340 return llvm::ConstantExpr::getShl(LHS, RHS);
341 }
342
343 llvm::Constant *VisitBinMul(const BinaryOperator *E) {
344 llvm::Constant *LHS = Visit(E->getLHS());
345 llvm::Constant *RHS = Visit(E->getRHS());
346
347 return llvm::ConstantExpr::getMul(LHS, RHS);
348 }
349
350 llvm::Constant *VisitBinDiv(const BinaryOperator *E) {
351 llvm::Constant *LHS = Visit(E->getLHS());
352 llvm::Constant *RHS = Visit(E->getRHS());
353
354 if (LHS->getType()->isFPOrFPVector())
355 return llvm::ConstantExpr::getFDiv(LHS, RHS);
356 else if (E->getType()->isUnsignedIntegerType())
357 return llvm::ConstantExpr::getUDiv(LHS, RHS);
358 else
359 return llvm::ConstantExpr::getSDiv(LHS, RHS);
360 }
361
362 llvm::Constant *VisitBinAdd(const BinaryOperator *E) {
363 llvm::Constant *LHS = Visit(E->getLHS());
364 llvm::Constant *RHS = Visit(E->getRHS());
365
366 if (!E->getType()->isPointerType())
367 return llvm::ConstantExpr::getAdd(LHS, RHS);
368
Anders Carlsson42da0682008-02-06 07:23:14 +0000369 llvm::Constant *Ptr, *Idx;
370 if (isa<llvm::PointerType>(LHS->getType())) { // pointer + int
371 Ptr = LHS;
372 Idx = RHS;
373 } else { // int + pointer
374 Ptr = RHS;
375 Idx = LHS;
376 }
377
378 return llvm::ConstantExpr::getGetElementPtr(Ptr, &Idx, 1);
Anders Carlsson4d852ee2008-02-05 20:54:21 +0000379 }
380
381 llvm::Constant *VisitBinAnd(const BinaryOperator *E) {
382 llvm::Constant *LHS = Visit(E->getLHS());
383 llvm::Constant *RHS = Visit(E->getRHS());
384
385 return llvm::ConstantExpr::getAnd(LHS, RHS);
386 }
Eli Friedmana36799c2008-05-29 11:22:45 +0000387
Eli Friedman8710b932008-05-29 15:17:49 +0000388 llvm::Constant *EmitCmp(const BinaryOperator *E,
389 llvm::CmpInst::Predicate SignedPred,
390 llvm::CmpInst::Predicate UnsignedPred,
391 llvm::CmpInst::Predicate FloatPred) {
Eli Friedmana36799c2008-05-29 11:22:45 +0000392 llvm::Constant *LHS = Visit(E->getLHS());
393 llvm::Constant *RHS = Visit(E->getRHS());
Eli Friedman8710b932008-05-29 15:17:49 +0000394 llvm::Constant *Result;
395 if (LHS->getType()->isInteger() ||
396 isa<llvm::PointerType>(LHS->getType())) {
397 if (E->getLHS()->getType()->isSignedIntegerType())
398 Result = llvm::ConstantExpr::getICmp(SignedPred, LHS, RHS);
399 else
400 Result = llvm::ConstantExpr::getICmp(UnsignedPred, LHS, RHS);
401 } else if (LHS->getType()->isFloatingPoint()) {
402 Result = llvm::ConstantExpr::getFCmp(FloatPred, LHS, RHS);
403 } else {
404 CGM.WarnUnsupported(E, "constant expression");
405 Result = llvm::ConstantInt::getFalse();
406 }
Eli Friedmana36799c2008-05-29 11:22:45 +0000407
408 const llvm::Type* ResultType = ConvertType(E->getType());
Eli Friedmana36799c2008-05-29 11:22:45 +0000409 return llvm::ConstantExpr::getZExtOrBitCast(Result, ResultType);
410 }
411
Eli Friedman8710b932008-05-29 15:17:49 +0000412 llvm::Constant *VisitBinNE(const BinaryOperator *E) {
413 return EmitCmp(E, llvm::CmpInst::ICMP_NE, llvm::CmpInst::ICMP_NE,
414 llvm::CmpInst::FCMP_ONE);
415 }
416
417 llvm::Constant *VisitBinEQ(const BinaryOperator *E) {
418 return EmitCmp(E, llvm::CmpInst::ICMP_EQ, llvm::CmpInst::ICMP_EQ,
419 llvm::CmpInst::FCMP_OEQ);
420 }
421
422 llvm::Constant *VisitBinLT(const BinaryOperator *E) {
423 return EmitCmp(E, llvm::CmpInst::ICMP_SLT, llvm::CmpInst::ICMP_ULT,
424 llvm::CmpInst::FCMP_OLT);
425 }
426
427 llvm::Constant *VisitBinLE(const BinaryOperator *E) {
428 return EmitCmp(E, llvm::CmpInst::ICMP_SLE, llvm::CmpInst::ICMP_ULE,
429 llvm::CmpInst::FCMP_OLE);
430 }
431
432 llvm::Constant *VisitBinGT(const BinaryOperator *E) {
433 return EmitCmp(E, llvm::CmpInst::ICMP_SGT, llvm::CmpInst::ICMP_UGT,
434 llvm::CmpInst::FCMP_OGT);
435 }
436
437 llvm::Constant *VisitBinGE(const BinaryOperator *E) {
438 return EmitCmp(E, llvm::CmpInst::ICMP_SGE, llvm::CmpInst::ICMP_SGE,
439 llvm::CmpInst::FCMP_OGE);
440 }
441
Eli Friedmana36799c2008-05-29 11:22:45 +0000442 llvm::Constant *VisitConditionalOperator(const ConditionalOperator *E) {
443 llvm::Constant *Cond = Visit(E->getCond());
444 llvm::Constant *CondVal = EmitConversionToBool(Cond, E->getType());
445 llvm::ConstantInt *CondValInt = dyn_cast<llvm::ConstantInt>(CondVal);
446 if (!CondValInt) {
447 CGM.WarnUnsupported(E, "constant expression");
448 return llvm::Constant::getNullValue(ConvertType(E->getType()));
449 }
450 if (CondValInt->isOne()) {
451 if (E->getLHS())
452 return Visit(E->getLHS());
453 return Cond;
454 }
455
456 return Visit(E->getRHS());
457 }
Anders Carlsson4d852ee2008-02-05 20:54:21 +0000458
Anders Carlssond76cead2008-01-26 01:36:00 +0000459 // Utility methods
460 const llvm::Type *ConvertType(QualType T) {
461 return CGM.getTypes().ConvertType(T);
462 }
463
464 llvm::Constant *EmitConversionToBool(llvm::Constant *Src, QualType SrcType) {
465 assert(SrcType->isCanonical() && "EmitConversion strips typedefs");
466
467 if (SrcType->isRealFloatingType()) {
468 // Compare against 0.0 for fp scalars.
469 llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
470 return llvm::ConstantExpr::getFCmp(llvm::FCmpInst::FCMP_UNE, Src, Zero);
471 }
472
473 assert((SrcType->isIntegerType() || SrcType->isPointerType()) &&
474 "Unknown scalar type to convert");
475
476 // Compare against an integer or pointer null.
477 llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
478 return llvm::ConstantExpr::getICmp(llvm::ICmpInst::ICMP_NE, Src, Zero);
479 }
480
481 llvm::Constant *EmitConversion(llvm::Constant *Src, QualType SrcType,
482 QualType DstType) {
483 SrcType = SrcType.getCanonicalType();
484 DstType = DstType.getCanonicalType();
485 if (SrcType == DstType) return Src;
486
487 // Handle conversions to bool first, they are special: comparisons against 0.
488 if (DstType->isBooleanType())
489 return EmitConversionToBool(Src, SrcType);
490
491 const llvm::Type *DstTy = ConvertType(DstType);
492
493 // Ignore conversions like int -> uint.
494 if (Src->getType() == DstTy)
495 return Src;
496
497 // Handle pointer conversions next: pointers can only be converted to/from
498 // other pointers and integers.
499 if (isa<PointerType>(DstType)) {
500 // The source value may be an integer, or a pointer.
501 if (isa<llvm::PointerType>(Src->getType()))
502 return llvm::ConstantExpr::getBitCast(Src, DstTy);
Anders Carlssonde598292008-01-26 04:30:23 +0000503 assert(SrcType->isIntegerType() &&"Not ptr->ptr or int->ptr conversion?");
Anders Carlssond76cead2008-01-26 01:36:00 +0000504 return llvm::ConstantExpr::getIntToPtr(Src, DstTy);
505 }
506
507 if (isa<PointerType>(SrcType)) {
508 // Must be an ptr to int cast.
509 assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
510 return llvm::ConstantExpr::getPtrToInt(Src, DstTy);
511 }
512
513 // A scalar source can be splatted to a vector of the same element type
514 if (isa<llvm::VectorType>(DstTy) && !isa<VectorType>(SrcType)) {
515 const llvm::VectorType *VT = cast<llvm::VectorType>(DstTy);
516 assert((VT->getElementType() == Src->getType()) &&
517 "Vector element type must match scalar type to splat.");
518 unsigned NumElements = DstType->getAsVectorType()->getNumElements();
519 llvm::SmallVector<llvm::Constant*, 16> Elements;
520 for (unsigned i = 0; i < NumElements; i++)
521 Elements.push_back(Src);
522
523 return llvm::ConstantVector::get(&Elements[0], NumElements);
524 }
525
526 if (isa<llvm::VectorType>(Src->getType()) ||
527 isa<llvm::VectorType>(DstTy)) {
528 return llvm::ConstantExpr::getBitCast(Src, DstTy);
529 }
530
531 // Finally, we have the arithmetic types: real int/float.
532 if (isa<llvm::IntegerType>(Src->getType())) {
533 bool InputSigned = SrcType->isSignedIntegerType();
534 if (isa<llvm::IntegerType>(DstTy))
535 return llvm::ConstantExpr::getIntegerCast(Src, DstTy, InputSigned);
536 else if (InputSigned)
537 return llvm::ConstantExpr::getSIToFP(Src, DstTy);
538 else
539 return llvm::ConstantExpr::getUIToFP(Src, DstTy);
540 }
541
542 assert(Src->getType()->isFloatingPoint() && "Unknown real conversion");
543 if (isa<llvm::IntegerType>(DstTy)) {
544 if (DstType->isSignedIntegerType())
545 return llvm::ConstantExpr::getFPToSI(Src, DstTy);
546 else
547 return llvm::ConstantExpr::getFPToUI(Src, DstTy);
548 }
549
550 assert(DstTy->isFloatingPoint() && "Unknown real conversion");
551 if (DstTy->getTypeID() < Src->getType()->getTypeID())
552 return llvm::ConstantExpr::getFPTrunc(Src, DstTy);
553 else
554 return llvm::ConstantExpr::getFPExtend(Src, DstTy);
555 }
556
557 llvm::Constant *EmitSizeAlignOf(QualType TypeToSize,
558 QualType RetType, bool isSizeOf) {
559 std::pair<uint64_t, unsigned> Info =
Chris Lattner8cd0e932008-03-05 18:54:05 +0000560 CGM.getContext().getTypeInfo(TypeToSize);
Anders Carlssond76cead2008-01-26 01:36:00 +0000561
562 uint64_t Val = isSizeOf ? Info.first : Info.second;
563 Val /= 8; // Return size in bytes, not bits.
564
565 assert(RetType->isIntegerType() && "Result type must be an integer!");
566
Chris Lattner8cd0e932008-03-05 18:54:05 +0000567 uint32_t ResultWidth =
568 static_cast<uint32_t>(CGM.getContext().getTypeSize(RetType));
Anders Carlssond76cead2008-01-26 01:36:00 +0000569 return llvm::ConstantInt::get(llvm::APInt(ResultWidth, Val));
570 }
Anders Carlssonde598292008-01-26 04:30:23 +0000571
572 llvm::Constant *EmitLValue(Expr *E) {
Anders Carlsson00b837c2008-01-26 02:08:50 +0000573 switch (E->getStmtClass()) {
Eli Friedmane493ea92008-02-11 00:23:10 +0000574 default: break;
Anders Carlsson00b837c2008-01-26 02:08:50 +0000575 case Expr::ParenExprClass:
576 // Elide parenthesis
577 return EmitLValue(cast<ParenExpr>(E)->getSubExpr());
578 case Expr::CompoundLiteralExprClass: {
579 // Note that due to the nature of compound literals, this is guaranteed
580 // to be the only use of the variable, so we just generate it here.
Anders Carlssonde598292008-01-26 04:30:23 +0000581 CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
582 llvm::Constant* C = Visit(CLE->getInitializer());
Chris Lattnera6dcce32008-02-11 00:02:17 +0000583 C = new llvm::GlobalVariable(C->getType(),E->getType().isConstQualified(),
Anders Carlsson00b837c2008-01-26 02:08:50 +0000584 llvm::GlobalValue::InternalLinkage,
585 C, ".compoundliteral", &CGM.getModule());
586 return C;
Anders Carlssonde598292008-01-26 04:30:23 +0000587 }
Anders Carlsson00b837c2008-01-26 02:08:50 +0000588 case Expr::DeclRefExprClass: {
Anders Carlssonde598292008-01-26 04:30:23 +0000589 ValueDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
Anders Carlsson00b837c2008-01-26 02:08:50 +0000590 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
591 return CGM.GetAddrOfFunctionDecl(FD, false);
Steve Naroff72a6ebc2008-04-15 22:42:06 +0000592 if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
593 if (VD->isFileVarDecl())
594 return CGM.GetAddrOfGlobalVar(VD, false);
595 else if (VD->isBlockVarDecl()) {
596 assert(CGF && "Can't access static local vars without CGF");
597 return CGF->GetAddrOfStaticLocalVar(VD);
598 }
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +0000599 }
Eli Friedmane493ea92008-02-11 00:23:10 +0000600 break;
Anders Carlsson00b837c2008-01-26 02:08:50 +0000601 }
602 case Expr::MemberExprClass: {
Anders Carlssonde598292008-01-26 04:30:23 +0000603 MemberExpr* ME = cast<MemberExpr>(E);
Anders Carlssonde598292008-01-26 04:30:23 +0000604 llvm::Constant *Base;
605 if (ME->isArrow())
606 Base = Visit(ME->getBase());
607 else
608 Base = EmitLValue(ME->getBase());
Lauro Ramos Venancio2f049552008-01-30 21:23:20 +0000609
610 unsigned FieldNumber = CGM.getTypes().getLLVMFieldNo(ME->getMemberDecl());
Anders Carlsson00b837c2008-01-26 02:08:50 +0000611 llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
612 llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
613 FieldNumber);
614 llvm::Value *Ops[] = {Zero, Idx};
615 return llvm::ConstantExpr::getGetElementPtr(Base, Ops, 2);
616 }
617 case Expr::ArraySubscriptExprClass: {
Anders Carlssonde598292008-01-26 04:30:23 +0000618 ArraySubscriptExpr* ASExpr = cast<ArraySubscriptExpr>(E);
619 llvm::Constant *Base = Visit(ASExpr->getBase());
620 llvm::Constant *Index = Visit(ASExpr->getIdx());
Anders Carlsson00b837c2008-01-26 02:08:50 +0000621 assert(!ASExpr->getBase()->getType()->isVectorType() &&
622 "Taking the address of a vector component is illegal!");
623 return llvm::ConstantExpr::getGetElementPtr(Base, &Index, 1);
624 }
625 case Expr::StringLiteralClass: {
Anders Carlssonde598292008-01-26 04:30:23 +0000626 StringLiteral *String = cast<StringLiteral>(E);
Anders Carlsson00b837c2008-01-26 02:08:50 +0000627 assert(!String->isWide() && "Cannot codegen wide strings yet");
Anders Carlssonde598292008-01-26 04:30:23 +0000628 const char *StrData = String->getStrData();
629 unsigned Len = String->getByteLength();
Anders Carlsson00b837c2008-01-26 02:08:50 +0000630
Anders Carlssonde598292008-01-26 04:30:23 +0000631 return CGM.GetAddrOfConstantString(std::string(StrData, StrData + Len));
632 }
633 case Expr::UnaryOperatorClass: {
634 UnaryOperator *Exp = cast<UnaryOperator>(E);
635 switch (Exp->getOpcode()) {
Eli Friedmane493ea92008-02-11 00:23:10 +0000636 default: break;
637 case UnaryOperator::Extension:
Anders Carlssonde598292008-01-26 04:30:23 +0000638 // Extension is just a wrapper for expressions
639 return EmitLValue(Exp->getSubExpr());
640 case UnaryOperator::Real:
641 case UnaryOperator::Imag: {
642 // The address of __real or __imag is just a GEP off the address
643 // of the internal expression
644 llvm::Constant* C = EmitLValue(Exp->getSubExpr());
645 llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
646 llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
Anders Carlsson00b837c2008-01-26 02:08:50 +0000647 Exp->getOpcode() == UnaryOperator::Imag);
Anders Carlssonde598292008-01-26 04:30:23 +0000648 llvm::Value *Ops[] = {Zero, Idx};
649 return llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
650 }
651 case UnaryOperator::Deref:
652 // The address of a deref is just the value of the expression
653 return Visit(Exp->getSubExpr());
654 }
Eli Friedmane493ea92008-02-11 00:23:10 +0000655 break;
Anders Carlsson00b837c2008-01-26 02:08:50 +0000656 }
Eli Friedmane493ea92008-02-11 00:23:10 +0000657 }
658 CGM.WarnUnsupported(E, "constant l-value expression");
659 llvm::Type *Ty = llvm::PointerType::getUnqual(ConvertType(E->getType()));
660 return llvm::UndefValue::get(Ty);
Anders Carlsson00b837c2008-01-26 02:08:50 +0000661 }
Anders Carlsson00b837c2008-01-26 02:08:50 +0000662
663};
664
Anders Carlssond76cead2008-01-26 01:36:00 +0000665} // end anonymous namespace.
666
667
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +0000668llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
669 CodeGenFunction *CGF)
Anders Carlssond76cead2008-01-26 01:36:00 +0000670{
671 QualType type = E->getType().getCanonicalType();
672
673 if (type->isIntegerType()) {
Chris Lattner8cd0e932008-03-05 18:54:05 +0000674 llvm::APSInt Value(static_cast<uint32_t>(Context.getTypeSize(type)));
Anders Carlssond76cead2008-01-26 01:36:00 +0000675 if (E->isIntegerConstantExpr(Value, Context)) {
676 return llvm::ConstantInt::get(Value);
677 }
678 }
679
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +0000680 return ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
Anders Carlssond76cead2008-01-26 01:36:00 +0000681}