blob: a642fd236d38eb673c7d26ab486ba49cca22e6d6 [file] [log] [blame]
Anders Carlsson3b1d57b2008-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 Carlsson84005b42008-01-26 04:30:23 +000025class VISIBILITY_HIDDEN ConstExprEmitter :
26 public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000027 CodeGenModule &CGM;
Lauro Ramos Venancio81373352008-02-26 21:41:45 +000028 CodeGenFunction *CGF;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000029public:
Lauro Ramos Venancio81373352008-02-26 21:41:45 +000030 ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
31 : CGM(cgm), CGF(cgf) {
Anders Carlsson3b1d57b2008-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 Lattner98be4942008-03-05 18:54:05 +000040 QualType T = cast<Expr>(S)->getType();
41 return llvm::UndefValue::get(CGM.getTypes().ConvertType(T));
Anders Carlsson3b1d57b2008-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 Lattner59138ba2008-04-20 00:45:53 +000053 return llvm::ConstantFP::get(E->getValue());
Anders Carlsson3b1d57b2008-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 Pateleae15602008-02-05 02:39:50 +000071
Chris Lattner04421082008-04-08 04:40:51 +000072 llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
73 return Visit(DAE->getExpr());
74 }
75
Devang Pateleae15602008-02-05 02:39:50 +000076 llvm::Constant *EmitArrayInitialization(InitListExpr *ILE,
77 const llvm::ArrayType *AType) {
Eli Friedman4d1c0182008-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 Pateleae15602008-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 Pateleae15602008-02-05 02:39:50 +000094 llvm::Constant *C = Visit(ILE->getInit(i));
Devang Pateleae15602008-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 Friedman017cbdf2008-05-20 00:11:07 +0000109 RecordDecl *RD = ILE->getType()->getAsRecordType()->getDecl();
Devang Pateleae15602008-02-05 02:39:50 +0000110 std::vector<llvm::Constant*> Elts;
Devang Pateleae15602008-02-05 02:39:50 +0000111
Eli Friedman017cbdf2008-05-20 00:11:07 +0000112 // Initialize the whole structure to zero.
113 for (unsigned i = 0; i < SType->getNumElements(); ++i) {
Devang Pateleae15602008-02-05 02:39:50 +0000114 const llvm::Type *FieldTy = SType->getElementType(i);
115 Elts.push_back(llvm::Constant::getNullValue(FieldTy));
116 }
Eli Friedman017cbdf2008-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 Pateleae15602008-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 Pateleae15602008-02-05 02:39:50 +0000153 llvm::Constant *C = Visit(ILE->getInit(i));
Devang Pateleae15602008-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 Carlsson3b1d57b2008-01-26 01:36:00 +0000161 llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
Anders Carlssonf6884ac2008-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 Carlsson3b1d57b2008-01-26 01:36:00 +0000168 }
Anders Carlssonf6884ac2008-01-29 01:15:48 +0000169
Devang Pateleae15602008-02-05 02:39:50 +0000170 if (const llvm::ArrayType *AType = dyn_cast<llvm::ArrayType>(CType))
171 return EmitArrayInitialization(ILE, AType);
Devang Patel47fb6972008-01-29 23:23:18 +0000172
Devang Pateleae15602008-02-05 02:39:50 +0000173 if (const llvm::StructType *SType = dyn_cast<llvm::StructType>(CType))
174 return EmitStructInitialization(ILE, SType);
Devang Patel47fb6972008-01-29 23:23:18 +0000175
Devang Pateleae15602008-02-05 02:39:50 +0000176 if (const llvm::VectorType *VType = dyn_cast<llvm::VectorType>(CType))
177 return EmitVectorInitialization(ILE, VType);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000178
179 // Make sure we have an array at this point
Devang Pateleae15602008-02-05 02:39:50 +0000180 assert(0 && "Unable to handle InitListExpr");
Chris Lattner2b94fe32008-03-01 08:45:05 +0000181 // Get rid of control reaches end of void function warning.
182 // Not reached.
183 return 0;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000184 }
Eli Friedman3b707e72008-02-21 17:57:49 +0000185
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000186 llvm::Constant *VisitImplicitCastExpr(ImplicitCastExpr *ICExpr) {
Eli Friedman3b707e72008-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 Carlsson3b1d57b2008-01-26 01:36:00 +0000195 llvm::Constant *Idx0 = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000196 llvm::Constant *Ops[] = {Idx0, Idx0};
197 C = llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
Eli Friedman3b707e72008-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 Carlsson3b1d57b2008-01-26 01:36:00 +0000208 }
Eli Friedman3b707e72008-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 Carlsson3b1d57b2008-01-26 01:36:00 +0000213 }
Eli Friedman3b707e72008-02-21 17:57:49 +0000214
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000215 llvm::Constant *VisitStringLiteral(StringLiteral *E) {
216 const char *StrData = E->getStrData();
217 unsigned Len = E->getByteLength();
Chris Lattnera7ad98f2008-02-11 00:02:17 +0000218 assert(!E->getType()->isPointerType() && "Strings are always arrays");
Anders Carlsson3b1d57b2008-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 Carlssona3881fc2008-01-29 01:28:48 +0000241 if (const EnumConstantDecl *EC = dyn_cast<EnumConstantDecl>(Decl))
242 return llvm::ConstantInt::get(EC->getInitVal());
Anders Carlsson3b1d57b2008-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 Carlssone3d3b7d2008-01-26 02:08:50 +0000287 llvm::Constant *VisitUnaryAddrOf(const UnaryOperator *E) {
288 return EmitLValue(E->getSubExpr());
289 }
Anders Carlsson5a1deb82008-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 Lattner98be4942008-03-05 18:54:05 +0000295 uint32_t ResultWidth =
296 static_cast<uint32_t>(CGM.getContext().getTypeSize(E->getType()));
Anders Carlsson5a1deb82008-01-29 15:56:48 +0000297 return llvm::ConstantInt::get(llvm::APInt(ResultWidth, Val));
298 }
Eli Friedmanbc5ed6e2008-05-29 11:22:45 +0000299
300 llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
301 return Visit(E->getSubExpr());
302 }
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000303
Anders Carlssonce6237e2008-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 Carlsson7be3d012008-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 Carlssonce6237e2008-01-29 01:33:32 +0000334
Anders Carlsson7be3d012008-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 Carlsson87a98752008-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 Carlsson7be3d012008-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 Friedmanbc5ed6e2008-05-29 11:22:45 +0000387
388 llvm::Constant *VisitBinNE(const BinaryOperator *E) {
389 llvm::Constant *LHS = Visit(E->getLHS());
390 llvm::Constant *RHS = Visit(E->getRHS());
391
392 const llvm::Type* ResultType = ConvertType(E->getType());
393 if (!ResultType->isInteger()) {
394 CGM.WarnUnsupported(E, "constant expression");
395 return llvm::Constant::getNullValue(ConvertType(E->getType()));
396 }
397 llvm::Constant *Result =
398 llvm::ConstantExpr::getICmp(llvm::ICmpInst::ICMP_NE, LHS, RHS);
399 return llvm::ConstantExpr::getZExtOrBitCast(Result, ResultType);
400 }
401
402 llvm::Constant *VisitConditionalOperator(const ConditionalOperator *E) {
403 llvm::Constant *Cond = Visit(E->getCond());
404 llvm::Constant *CondVal = EmitConversionToBool(Cond, E->getType());
405 llvm::ConstantInt *CondValInt = dyn_cast<llvm::ConstantInt>(CondVal);
406 if (!CondValInt) {
407 CGM.WarnUnsupported(E, "constant expression");
408 return llvm::Constant::getNullValue(ConvertType(E->getType()));
409 }
410 if (CondValInt->isOne()) {
411 if (E->getLHS())
412 return Visit(E->getLHS());
413 return Cond;
414 }
415
416 return Visit(E->getRHS());
417 }
Anders Carlsson7be3d012008-02-05 20:54:21 +0000418
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000419 // Utility methods
420 const llvm::Type *ConvertType(QualType T) {
421 return CGM.getTypes().ConvertType(T);
422 }
423
424 llvm::Constant *EmitConversionToBool(llvm::Constant *Src, QualType SrcType) {
425 assert(SrcType->isCanonical() && "EmitConversion strips typedefs");
426
427 if (SrcType->isRealFloatingType()) {
428 // Compare against 0.0 for fp scalars.
429 llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
430 return llvm::ConstantExpr::getFCmp(llvm::FCmpInst::FCMP_UNE, Src, Zero);
431 }
432
433 assert((SrcType->isIntegerType() || SrcType->isPointerType()) &&
434 "Unknown scalar type to convert");
435
436 // Compare against an integer or pointer null.
437 llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
438 return llvm::ConstantExpr::getICmp(llvm::ICmpInst::ICMP_NE, Src, Zero);
439 }
440
441 llvm::Constant *EmitConversion(llvm::Constant *Src, QualType SrcType,
442 QualType DstType) {
443 SrcType = SrcType.getCanonicalType();
444 DstType = DstType.getCanonicalType();
445 if (SrcType == DstType) return Src;
446
447 // Handle conversions to bool first, they are special: comparisons against 0.
448 if (DstType->isBooleanType())
449 return EmitConversionToBool(Src, SrcType);
450
451 const llvm::Type *DstTy = ConvertType(DstType);
452
453 // Ignore conversions like int -> uint.
454 if (Src->getType() == DstTy)
455 return Src;
456
457 // Handle pointer conversions next: pointers can only be converted to/from
458 // other pointers and integers.
459 if (isa<PointerType>(DstType)) {
460 // The source value may be an integer, or a pointer.
461 if (isa<llvm::PointerType>(Src->getType()))
462 return llvm::ConstantExpr::getBitCast(Src, DstTy);
Anders Carlsson84005b42008-01-26 04:30:23 +0000463 assert(SrcType->isIntegerType() &&"Not ptr->ptr or int->ptr conversion?");
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000464 return llvm::ConstantExpr::getIntToPtr(Src, DstTy);
465 }
466
467 if (isa<PointerType>(SrcType)) {
468 // Must be an ptr to int cast.
469 assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
470 return llvm::ConstantExpr::getPtrToInt(Src, DstTy);
471 }
472
473 // A scalar source can be splatted to a vector of the same element type
474 if (isa<llvm::VectorType>(DstTy) && !isa<VectorType>(SrcType)) {
475 const llvm::VectorType *VT = cast<llvm::VectorType>(DstTy);
476 assert((VT->getElementType() == Src->getType()) &&
477 "Vector element type must match scalar type to splat.");
478 unsigned NumElements = DstType->getAsVectorType()->getNumElements();
479 llvm::SmallVector<llvm::Constant*, 16> Elements;
480 for (unsigned i = 0; i < NumElements; i++)
481 Elements.push_back(Src);
482
483 return llvm::ConstantVector::get(&Elements[0], NumElements);
484 }
485
486 if (isa<llvm::VectorType>(Src->getType()) ||
487 isa<llvm::VectorType>(DstTy)) {
488 return llvm::ConstantExpr::getBitCast(Src, DstTy);
489 }
490
491 // Finally, we have the arithmetic types: real int/float.
492 if (isa<llvm::IntegerType>(Src->getType())) {
493 bool InputSigned = SrcType->isSignedIntegerType();
494 if (isa<llvm::IntegerType>(DstTy))
495 return llvm::ConstantExpr::getIntegerCast(Src, DstTy, InputSigned);
496 else if (InputSigned)
497 return llvm::ConstantExpr::getSIToFP(Src, DstTy);
498 else
499 return llvm::ConstantExpr::getUIToFP(Src, DstTy);
500 }
501
502 assert(Src->getType()->isFloatingPoint() && "Unknown real conversion");
503 if (isa<llvm::IntegerType>(DstTy)) {
504 if (DstType->isSignedIntegerType())
505 return llvm::ConstantExpr::getFPToSI(Src, DstTy);
506 else
507 return llvm::ConstantExpr::getFPToUI(Src, DstTy);
508 }
509
510 assert(DstTy->isFloatingPoint() && "Unknown real conversion");
511 if (DstTy->getTypeID() < Src->getType()->getTypeID())
512 return llvm::ConstantExpr::getFPTrunc(Src, DstTy);
513 else
514 return llvm::ConstantExpr::getFPExtend(Src, DstTy);
515 }
516
517 llvm::Constant *EmitSizeAlignOf(QualType TypeToSize,
518 QualType RetType, bool isSizeOf) {
519 std::pair<uint64_t, unsigned> Info =
Chris Lattner98be4942008-03-05 18:54:05 +0000520 CGM.getContext().getTypeInfo(TypeToSize);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000521
522 uint64_t Val = isSizeOf ? Info.first : Info.second;
523 Val /= 8; // Return size in bytes, not bits.
524
525 assert(RetType->isIntegerType() && "Result type must be an integer!");
526
Chris Lattner98be4942008-03-05 18:54:05 +0000527 uint32_t ResultWidth =
528 static_cast<uint32_t>(CGM.getContext().getTypeSize(RetType));
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000529 return llvm::ConstantInt::get(llvm::APInt(ResultWidth, Val));
530 }
Anders Carlsson84005b42008-01-26 04:30:23 +0000531
532 llvm::Constant *EmitLValue(Expr *E) {
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000533 switch (E->getStmtClass()) {
Eli Friedman546d94c2008-02-11 00:23:10 +0000534 default: break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000535 case Expr::ParenExprClass:
536 // Elide parenthesis
537 return EmitLValue(cast<ParenExpr>(E)->getSubExpr());
538 case Expr::CompoundLiteralExprClass: {
539 // Note that due to the nature of compound literals, this is guaranteed
540 // to be the only use of the variable, so we just generate it here.
Anders Carlsson84005b42008-01-26 04:30:23 +0000541 CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
542 llvm::Constant* C = Visit(CLE->getInitializer());
Chris Lattnera7ad98f2008-02-11 00:02:17 +0000543 C = new llvm::GlobalVariable(C->getType(),E->getType().isConstQualified(),
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000544 llvm::GlobalValue::InternalLinkage,
545 C, ".compoundliteral", &CGM.getModule());
546 return C;
Anders Carlsson84005b42008-01-26 04:30:23 +0000547 }
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000548 case Expr::DeclRefExprClass: {
Anders Carlsson84005b42008-01-26 04:30:23 +0000549 ValueDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000550 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
551 return CGM.GetAddrOfFunctionDecl(FD, false);
Steve Naroff248a7532008-04-15 22:42:06 +0000552 if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
553 if (VD->isFileVarDecl())
554 return CGM.GetAddrOfGlobalVar(VD, false);
555 else if (VD->isBlockVarDecl()) {
556 assert(CGF && "Can't access static local vars without CGF");
557 return CGF->GetAddrOfStaticLocalVar(VD);
558 }
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000559 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000560 break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000561 }
562 case Expr::MemberExprClass: {
Anders Carlsson84005b42008-01-26 04:30:23 +0000563 MemberExpr* ME = cast<MemberExpr>(E);
Anders Carlsson84005b42008-01-26 04:30:23 +0000564 llvm::Constant *Base;
565 if (ME->isArrow())
566 Base = Visit(ME->getBase());
567 else
568 Base = EmitLValue(ME->getBase());
Lauro Ramos Venancio5bfe18c2008-01-30 21:23:20 +0000569
570 unsigned FieldNumber = CGM.getTypes().getLLVMFieldNo(ME->getMemberDecl());
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000571 llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
572 llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
573 FieldNumber);
574 llvm::Value *Ops[] = {Zero, Idx};
575 return llvm::ConstantExpr::getGetElementPtr(Base, Ops, 2);
576 }
577 case Expr::ArraySubscriptExprClass: {
Anders Carlsson84005b42008-01-26 04:30:23 +0000578 ArraySubscriptExpr* ASExpr = cast<ArraySubscriptExpr>(E);
579 llvm::Constant *Base = Visit(ASExpr->getBase());
580 llvm::Constant *Index = Visit(ASExpr->getIdx());
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000581 assert(!ASExpr->getBase()->getType()->isVectorType() &&
582 "Taking the address of a vector component is illegal!");
583 return llvm::ConstantExpr::getGetElementPtr(Base, &Index, 1);
584 }
585 case Expr::StringLiteralClass: {
Anders Carlsson84005b42008-01-26 04:30:23 +0000586 StringLiteral *String = cast<StringLiteral>(E);
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000587 assert(!String->isWide() && "Cannot codegen wide strings yet");
Anders Carlsson84005b42008-01-26 04:30:23 +0000588 const char *StrData = String->getStrData();
589 unsigned Len = String->getByteLength();
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000590
Anders Carlsson84005b42008-01-26 04:30:23 +0000591 return CGM.GetAddrOfConstantString(std::string(StrData, StrData + Len));
592 }
593 case Expr::UnaryOperatorClass: {
594 UnaryOperator *Exp = cast<UnaryOperator>(E);
595 switch (Exp->getOpcode()) {
Eli Friedman546d94c2008-02-11 00:23:10 +0000596 default: break;
597 case UnaryOperator::Extension:
Anders Carlsson84005b42008-01-26 04:30:23 +0000598 // Extension is just a wrapper for expressions
599 return EmitLValue(Exp->getSubExpr());
600 case UnaryOperator::Real:
601 case UnaryOperator::Imag: {
602 // The address of __real or __imag is just a GEP off the address
603 // of the internal expression
604 llvm::Constant* C = EmitLValue(Exp->getSubExpr());
605 llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
606 llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000607 Exp->getOpcode() == UnaryOperator::Imag);
Anders Carlsson84005b42008-01-26 04:30:23 +0000608 llvm::Value *Ops[] = {Zero, Idx};
609 return llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
610 }
611 case UnaryOperator::Deref:
612 // The address of a deref is just the value of the expression
613 return Visit(Exp->getSubExpr());
614 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000615 break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000616 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000617 }
618 CGM.WarnUnsupported(E, "constant l-value expression");
619 llvm::Type *Ty = llvm::PointerType::getUnqual(ConvertType(E->getType()));
620 return llvm::UndefValue::get(Ty);
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000621 }
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000622
623};
624
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000625} // end anonymous namespace.
626
627
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000628llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
629 CodeGenFunction *CGF)
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000630{
631 QualType type = E->getType().getCanonicalType();
632
633 if (type->isIntegerType()) {
Chris Lattner98be4942008-03-05 18:54:05 +0000634 llvm::APSInt Value(static_cast<uint32_t>(Context.getTypeSize(type)));
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000635 if (E->isIntegerConstantExpr(Value, Context)) {
636 return llvm::ConstantInt::get(Value);
637 }
638 }
639
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000640 return ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000641}