blob: aa5d37dd1f0fbda3336e2109a17520a7ff72207a [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"
Daniel Dunbar84bb85f2008-08-13 00:59:25 +000016#include "CGObjCRuntime.h"
Chris Lattner8ce618c2008-10-06 05:59:01 +000017#include "clang/AST/APValue.h"
Daniel Dunbareee5cd12008-08-11 05:00:27 +000018#include "clang/AST/ASTContext.h"
19#include "clang/AST/StmtVisitor.h"
Anders Carlssond76cead2008-01-26 01:36:00 +000020#include "llvm/Constants.h"
21#include "llvm/Function.h"
22#include "llvm/GlobalVariable.h"
23#include "llvm/Support/Compiler.h"
Eli Friedman7529fd62008-05-30 10:24:46 +000024#include "llvm/Target/TargetData.h"
Anders Carlssond76cead2008-01-26 01:36:00 +000025using namespace clang;
26using namespace CodeGen;
27
28namespace {
Anders Carlssonde598292008-01-26 04:30:23 +000029class VISIBILITY_HIDDEN ConstExprEmitter :
30 public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
Anders Carlssond76cead2008-01-26 01:36:00 +000031 CodeGenModule &CGM;
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +000032 CodeGenFunction *CGF;
Anders Carlssond76cead2008-01-26 01:36:00 +000033public:
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +000034 ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
35 : CGM(cgm), CGF(cgf) {
Anders Carlssond76cead2008-01-26 01:36:00 +000036 }
37
38 //===--------------------------------------------------------------------===//
39 // Visitor Methods
40 //===--------------------------------------------------------------------===//
41
42 llvm::Constant *VisitStmt(Stmt *S) {
Daniel Dunbar9503b782008-08-16 00:56:44 +000043 CGM.ErrorUnsupported(S, "constant expression");
Chris Lattner8cd0e932008-03-05 18:54:05 +000044 QualType T = cast<Expr>(S)->getType();
45 return llvm::UndefValue::get(CGM.getTypes().ConvertType(T));
Anders Carlssond76cead2008-01-26 01:36:00 +000046 }
47
48 llvm::Constant *VisitParenExpr(ParenExpr *PE) {
49 return Visit(PE->getSubExpr());
50 }
Anders Carlsson2502a382008-11-22 02:34:39 +000051
Anders Carlssond76cead2008-01-26 01:36:00 +000052 llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
53 return Visit(E->getInitializer());
54 }
55
Argiris Kirtzidisc45e2fb2008-08-18 23:01:59 +000056 llvm::Constant *VisitCastExpr(CastExpr* E) {
Nuno Lopes1a835322009-01-17 00:48:48 +000057 // GCC cast to union extension
58 if (E->getType()->isUnionType()) {
59 const llvm::Type *Ty = ConvertType(E->getType());
60 return EmitUnion(CGM.EmitConstantExpr(E->getSubExpr(), CGF), Ty);
61 }
62
Anders Carlssond76cead2008-01-26 01:36:00 +000063 llvm::Constant *C = Visit(E->getSubExpr());
64
65 return EmitConversion(C, E->getSubExpr()->getType(), E->getType());
66 }
Devang Patel04520012008-02-05 02:39:50 +000067
Chris Lattner3e254fb2008-04-08 04:40:51 +000068 llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
69 return Visit(DAE->getExpr());
70 }
71
Eli Friedman7529fd62008-05-30 10:24:46 +000072 llvm::Constant *EmitArrayInitialization(InitListExpr *ILE) {
Eli Friedman8ac5b3a2008-02-21 17:45:41 +000073 std::vector<llvm::Constant*> Elts;
Eli Friedman7529fd62008-05-30 10:24:46 +000074 const llvm::ArrayType *AType =
75 cast<llvm::ArrayType>(ConvertType(ILE->getType()));
Eli Friedman8ac5b3a2008-02-21 17:45:41 +000076 unsigned NumInitElements = ILE->getNumInits();
77 // FIXME: Check for wide strings
78 if (NumInitElements > 0 && isa<StringLiteral>(ILE->getInit(0)) &&
Chris Lattnera1923f62008-08-04 07:31:14 +000079 ILE->getType()->getArrayElementTypeNoTypeQual()->isCharType())
Eli Friedman8ac5b3a2008-02-21 17:45:41 +000080 return Visit(ILE->getInit(0));
Devang Patel04520012008-02-05 02:39:50 +000081 const llvm::Type *ElemTy = AType->getElementType();
82 unsigned NumElements = AType->getNumElements();
83
84 // Initialising an array requires us to automatically
85 // initialise any elements that have not been initialised explicitly
86 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
87
88 // Copy initializer elements.
89 unsigned i = 0;
Eli Friedman42fab002008-05-30 19:58:50 +000090 bool RewriteType = false;
Devang Patel04520012008-02-05 02:39:50 +000091 for (; i < NumInitableElts; ++i) {
Anders Carlsson806e9932008-11-17 01:58:55 +000092 llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(i), CGF);
Eli Friedman42fab002008-05-30 19:58:50 +000093 RewriteType |= (C->getType() != ElemTy);
Devang Patel04520012008-02-05 02:39:50 +000094 Elts.push_back(C);
95 }
Eli Friedman42fab002008-05-30 19:58:50 +000096
Devang Patel04520012008-02-05 02:39:50 +000097 // Initialize remaining array elements.
98 for (; i < NumElements; ++i)
99 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
100
Eli Friedman42fab002008-05-30 19:58:50 +0000101 if (RewriteType) {
102 // FIXME: Try to avoid packing the array
103 std::vector<const llvm::Type*> Types;
104 for (unsigned i = 0; i < Elts.size(); ++i)
105 Types.push_back(Elts[i]->getType());
106 const llvm::StructType *SType = llvm::StructType::get(Types, true);
107 return llvm::ConstantStruct::get(SType, Elts);
108 }
109
Devang Patel04520012008-02-05 02:39:50 +0000110 return llvm::ConstantArray::get(AType, Elts);
111 }
112
Eli Friedman7529fd62008-05-30 10:24:46 +0000113 void InsertBitfieldIntoStruct(std::vector<llvm::Constant*>& Elts,
114 FieldDecl* Field, Expr* E) {
115 // Calculate the value to insert
Anders Carlsson806e9932008-11-17 01:58:55 +0000116 llvm::Constant *C = CGM.EmitConstantExpr(E, CGF);
Eli Friedman7529fd62008-05-30 10:24:46 +0000117 llvm::ConstantInt *CI = dyn_cast<llvm::ConstantInt>(C);
118 if (!CI) {
Daniel Dunbar9503b782008-08-16 00:56:44 +0000119 CGM.ErrorUnsupported(E, "bitfield initialization");
Eli Friedman7529fd62008-05-30 10:24:46 +0000120 return;
121 }
122 llvm::APInt V = CI->getValue();
Devang Patel04520012008-02-05 02:39:50 +0000123
Eli Friedman7529fd62008-05-30 10:24:46 +0000124 // Calculate information about the relevant field
125 const llvm::Type* Ty = CI->getType();
Daniel Dunbar8324cb62008-08-10 07:00:24 +0000126 const llvm::TargetData &TD = CGM.getTypes().getTargetData();
Eli Friedmanc021c7f2009-02-01 08:12:19 +0000127 unsigned size = TD.getTypePaddedSizeInBits(Ty);
Eli Friedman7529fd62008-05-30 10:24:46 +0000128 unsigned fieldOffset = CGM.getTypes().getLLVMFieldNo(Field) * size;
129 CodeGenTypes::BitFieldInfo bitFieldInfo =
130 CGM.getTypes().getBitFieldInfo(Field);
131 fieldOffset += bitFieldInfo.Begin;
132
133 // Find where to start the insertion
134 // FIXME: This is O(n^2) in the number of bit-fields!
135 // FIXME: This won't work if the struct isn't completely packed!
136 unsigned offset = 0, i = 0;
137 while (offset < (fieldOffset & -8))
Eli Friedmanc021c7f2009-02-01 08:12:19 +0000138 offset += TD.getTypePaddedSizeInBits(Elts[i++]->getType());
Daniel Dunbar8324cb62008-08-10 07:00:24 +0000139
140 // Advance over 0 sized elements (must terminate in bounds since
141 // the bitfield must have a size).
Eli Friedmanc021c7f2009-02-01 08:12:19 +0000142 while (TD.getTypePaddedSizeInBits(Elts[i]->getType()) == 0)
Daniel Dunbar8324cb62008-08-10 07:00:24 +0000143 ++i;
Eli Friedman7529fd62008-05-30 10:24:46 +0000144
Daniel Dunbar3b9a77e2008-07-17 21:07:48 +0000145 // Promote the size of V if necessary
146 // FIXME: This should never occur, but currently it can because
147 // initializer constants are cast to bool, and because clang is
148 // not enforcing bitfield width limits.
149 if (bitFieldInfo.Size > V.getBitWidth())
150 V.zext(bitFieldInfo.Size);
151
Eli Friedman7529fd62008-05-30 10:24:46 +0000152 // Insert the bits into the struct
153 // FIXME: This algorthm is only correct on X86!
154 // FIXME: THis algorthm assumes bit-fields only have byte-size elements!
155 unsigned bitsToInsert = bitFieldInfo.Size;
156 unsigned curBits = std::min(8 - (fieldOffset & 7), bitsToInsert);
157 unsigned byte = V.getLoBits(curBits).getZExtValue() << (fieldOffset & 7);
158 do {
159 llvm::Constant* byteC = llvm::ConstantInt::get(llvm::Type::Int8Ty, byte);
160 Elts[i] = llvm::ConstantExpr::getOr(Elts[i], byteC);
161 ++i;
162 V = V.lshr(curBits);
163 bitsToInsert -= curBits;
164
165 if (!bitsToInsert)
166 break;
167
168 curBits = bitsToInsert > 8 ? 8 : bitsToInsert;
169 byte = V.getLoBits(curBits).getZExtValue();
170 } while (true);
171 }
172
173 llvm::Constant *EmitStructInitialization(InitListExpr *ILE) {
174 const llvm::StructType *SType =
175 cast<llvm::StructType>(ConvertType(ILE->getType()));
Eli Friedman0242cd22008-05-20 00:11:07 +0000176 RecordDecl *RD = ILE->getType()->getAsRecordType()->getDecl();
Devang Patel04520012008-02-05 02:39:50 +0000177 std::vector<llvm::Constant*> Elts;
Devang Patel04520012008-02-05 02:39:50 +0000178
Eli Friedman0242cd22008-05-20 00:11:07 +0000179 // Initialize the whole structure to zero.
180 for (unsigned i = 0; i < SType->getNumElements(); ++i) {
Devang Patel04520012008-02-05 02:39:50 +0000181 const llvm::Type *FieldTy = SType->getElementType(i);
182 Elts.push_back(llvm::Constant::getNullValue(FieldTy));
183 }
Eli Friedman0242cd22008-05-20 00:11:07 +0000184
185 // Copy initializer elements. Skip padding fields.
186 unsigned EltNo = 0; // Element no in ILE
187 int FieldNo = 0; // Field no in RecordDecl
Eli Friedman42fab002008-05-30 19:58:50 +0000188 bool RewriteType = false;
Douglas Gregor8acb7272008-12-11 16:49:14 +0000189 for (RecordDecl::field_iterator Field = RD->field_begin(),
190 FieldEnd = RD->field_end();
191 EltNo < ILE->getNumInits() && Field != FieldEnd; ++Field) {
Eli Friedman0242cd22008-05-20 00:11:07 +0000192 FieldNo++;
Douglas Gregor8acb7272008-12-11 16:49:14 +0000193 if (!Field->getIdentifier())
Eli Friedman0242cd22008-05-20 00:11:07 +0000194 continue;
195
Douglas Gregor8acb7272008-12-11 16:49:14 +0000196 if (Field->isBitField()) {
197 InsertBitfieldIntoStruct(Elts, *Field, ILE->getInit(EltNo));
Eli Friedman0242cd22008-05-20 00:11:07 +0000198 } else {
Douglas Gregor8acb7272008-12-11 16:49:14 +0000199 unsigned FieldNo = CGM.getTypes().getLLVMFieldNo(*Field);
Anders Carlsson806e9932008-11-17 01:58:55 +0000200 llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(EltNo), CGF);
Eli Friedman42fab002008-05-30 19:58:50 +0000201 RewriteType |= (C->getType() != Elts[FieldNo]->getType());
202 Elts[FieldNo] = C;
Eli Friedman0242cd22008-05-20 00:11:07 +0000203 }
204 EltNo++;
205 }
206
Eli Friedman42fab002008-05-30 19:58:50 +0000207 if (RewriteType) {
208 // FIXME: Make this work for non-packed structs
209 assert(SType->isPacked() && "Cannot recreate unpacked structs");
210 std::vector<const llvm::Type*> Types;
211 for (unsigned i = 0; i < Elts.size(); ++i)
212 Types.push_back(Elts[i]->getType());
213 SType = llvm::StructType::get(Types, true);
214 }
215
Devang Patel04520012008-02-05 02:39:50 +0000216 return llvm::ConstantStruct::get(SType, Elts);
217 }
218
Nuno Lopes1a835322009-01-17 00:48:48 +0000219 llvm::Constant *EmitUnion(llvm::Constant *C, const llvm::Type *Ty) {
220 // Build a struct with the union sub-element as the first member,
221 // and padded to the appropriate size
222 std::vector<llvm::Constant*> Elts;
223 std::vector<const llvm::Type*> Types;
224 Elts.push_back(C);
225 Types.push_back(C->getType());
Eli Friedmanc021c7f2009-02-01 08:12:19 +0000226 unsigned CurSize = CGM.getTargetData().getTypePaddedSize(C->getType());
227 unsigned TotalSize = CGM.getTargetData().getTypePaddedSize(Ty);
Nuno Lopes1a835322009-01-17 00:48:48 +0000228 while (CurSize < TotalSize) {
229 Elts.push_back(llvm::Constant::getNullValue(llvm::Type::Int8Ty));
230 Types.push_back(llvm::Type::Int8Ty);
231 CurSize++;
232 }
233
234 // This always generates a packed struct
235 // FIXME: Try to generate an unpacked struct when we can
236 llvm::StructType* STy = llvm::StructType::get(Types, true);
237 return llvm::ConstantStruct::get(STy, Elts);
238 }
239
Eli Friedman7529fd62008-05-30 10:24:46 +0000240 llvm::Constant *EmitUnionInitialization(InitListExpr *ILE) {
Eli Friedman7529fd62008-05-30 10:24:46 +0000241 const llvm::Type *Ty = ConvertType(ILE->getType());
242
Douglas Gregor82462762009-01-29 16:53:55 +0000243 // If this is an empty initializer list, we value-initialize the
244 // union.
245 if (ILE->getNumInits() == 0)
246 return llvm::Constant::getNullValue(Ty);
Douglas Gregor36dd0c52009-01-28 23:36:17 +0000247
Douglas Gregor82462762009-01-29 16:53:55 +0000248 FieldDecl* curField = ILE->getInitializedFieldInUnion();
249 if (!curField) {
Douglas Gregor82462762009-01-29 16:53:55 +0000250 // There's no field to initialize, so value-initialize the union.
251#ifndef NDEBUG
252 // Make sure that it's really an empty and not a failure of
253 // semantic analysis.
Douglas Gregorec85e812009-02-11 16:17:49 +0000254 RecordDecl *RD = ILE->getType()->getAsRecordType()->getDecl();
Douglas Gregor82462762009-01-29 16:53:55 +0000255 for (RecordDecl::field_iterator Field = RD->field_begin(),
256 FieldEnd = RD->field_end();
257 Field != FieldEnd; ++Field)
258 assert(Field->isUnnamedBitfield() && "Only unnamed bitfields allowed");
259#endif
Eli Friedman7529fd62008-05-30 10:24:46 +0000260 return llvm::Constant::getNullValue(Ty);
Douglas Gregor82462762009-01-29 16:53:55 +0000261 }
Eli Friedman7529fd62008-05-30 10:24:46 +0000262
263 if (curField->isBitField()) {
264 // Create a dummy struct for bit-field insertion
Daniel Dunbard8439f22009-01-12 21:08:18 +0000265 unsigned NumElts = CGM.getTargetData().getTypePaddedSize(Ty) / 8;
Eli Friedman7529fd62008-05-30 10:24:46 +0000266 llvm::Constant* NV = llvm::Constant::getNullValue(llvm::Type::Int8Ty);
267 std::vector<llvm::Constant*> Elts(NumElts, NV);
268
269 InsertBitfieldIntoStruct(Elts, curField, ILE->getInit(0));
270 const llvm::ArrayType *RetTy =
271 llvm::ArrayType::get(NV->getType(), NumElts);
272 return llvm::ConstantArray::get(RetTy, Elts);
273 }
274
Nuno Lopes1a835322009-01-17 00:48:48 +0000275 return EmitUnion(CGM.EmitConstantExpr(ILE->getInit(0), CGF), Ty);
Eli Friedman7529fd62008-05-30 10:24:46 +0000276 }
277
278 llvm::Constant *EmitVectorInitialization(InitListExpr *ILE) {
279 const llvm::VectorType *VType =
280 cast<llvm::VectorType>(ConvertType(ILE->getType()));
Eli Friedman6c221712008-05-30 12:02:14 +0000281 const llvm::Type *ElemTy = VType->getElementType();
282 std::vector<llvm::Constant*> Elts;
Devang Patel04520012008-02-05 02:39:50 +0000283 unsigned NumElements = VType->getNumElements();
Eli Friedman6c221712008-05-30 12:02:14 +0000284 unsigned NumInitElements = ILE->getNumInits();
Devang Patel04520012008-02-05 02:39:50 +0000285
Eli Friedman6c221712008-05-30 12:02:14 +0000286 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
287
Devang Patel04520012008-02-05 02:39:50 +0000288 // Copy initializer elements.
289 unsigned i = 0;
Eli Friedman6c221712008-05-30 12:02:14 +0000290 for (; i < NumInitableElts; ++i) {
Anders Carlsson806e9932008-11-17 01:58:55 +0000291 llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(i), CGF);
Devang Patel04520012008-02-05 02:39:50 +0000292 Elts.push_back(C);
293 }
294
Eli Friedman6c221712008-05-30 12:02:14 +0000295 for (; i < NumElements; ++i)
296 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
297
Devang Patel04520012008-02-05 02:39:50 +0000298 return llvm::ConstantVector::get(VType, Elts);
299 }
Anders Carlsson1ddb85f2009-01-30 06:13:25 +0000300
301 llvm::Constant *VisitImplicitValueInitExpr(ImplicitValueInitExpr* E) {
302 const llvm::Type* RetTy = CGM.getTypes().ConvertType(E->getType());
303 return llvm::Constant::getNullValue(RetTy);
304 }
305
Anders Carlssond76cead2008-01-26 01:36:00 +0000306 llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
Eli Friedman7529fd62008-05-30 10:24:46 +0000307 if (ILE->getType()->isScalarType()) {
308 // We have a scalar in braces. Just use the first element.
309 if (ILE->getNumInits() > 0)
Anders Carlsson806e9932008-11-17 01:58:55 +0000310 return CGM.EmitConstantExpr(ILE->getInit(0), CGF);
Eli Friedman7529fd62008-05-30 10:24:46 +0000311
312 const llvm::Type* RetTy = CGM.getTypes().ConvertType(ILE->getType());
313 return llvm::Constant::getNullValue(RetTy);
Anders Carlssond76cead2008-01-26 01:36:00 +0000314 }
Chris Lattnera9177982008-10-26 23:53:12 +0000315
Eli Friedman7529fd62008-05-30 10:24:46 +0000316 if (ILE->getType()->isArrayType())
317 return EmitArrayInitialization(ILE);
Devang Pateld72d1012008-01-29 23:23:18 +0000318
Eli Friedman7529fd62008-05-30 10:24:46 +0000319 if (ILE->getType()->isStructureType())
320 return EmitStructInitialization(ILE);
321
322 if (ILE->getType()->isUnionType())
323 return EmitUnionInitialization(ILE);
324
325 if (ILE->getType()->isVectorType())
326 return EmitVectorInitialization(ILE);
327
Devang Patel04520012008-02-05 02:39:50 +0000328 assert(0 && "Unable to handle InitListExpr");
Chris Lattnercbfb5512008-03-01 08:45:05 +0000329 // Get rid of control reaches end of void function warning.
330 // Not reached.
331 return 0;
Anders Carlssond76cead2008-01-26 01:36:00 +0000332 }
Eli Friedman1693a3f2008-02-21 17:57:49 +0000333
Anders Carlssond76cead2008-01-26 01:36:00 +0000334 llvm::Constant *VisitImplicitCastExpr(ImplicitCastExpr *ICExpr) {
Eli Friedman1693a3f2008-02-21 17:57:49 +0000335 Expr* SExpr = ICExpr->getSubExpr();
336 QualType SType = SExpr->getType();
337 llvm::Constant *C; // the intermediate expression
338 QualType T; // the type of the intermediate expression
339 if (SType->isArrayType()) {
340 // Arrays decay to a pointer to the first element
341 // VLAs would require special handling, but they can't occur here
342 C = EmitLValue(SExpr);
Anders Carlssond76cead2008-01-26 01:36:00 +0000343 llvm::Constant *Idx0 = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
Anders Carlssond76cead2008-01-26 01:36:00 +0000344 llvm::Constant *Ops[] = {Idx0, Idx0};
345 C = llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
Chris Lattnera1923f62008-08-04 07:31:14 +0000346 T = CGM.getContext().getArrayDecayedType(SType);
Eli Friedman1693a3f2008-02-21 17:57:49 +0000347 } else if (SType->isFunctionType()) {
348 // Function types decay to a pointer to the function
349 C = EmitLValue(SExpr);
350 T = CGM.getContext().getPointerType(SType);
351 } else {
352 C = Visit(SExpr);
353 T = SType;
Anders Carlssond76cead2008-01-26 01:36:00 +0000354 }
Eli Friedman1693a3f2008-02-21 17:57:49 +0000355
356 // Perform the conversion; note that an implicit cast can both promote
357 // and convert an array/function
358 return EmitConversion(C, T, ICExpr->getType());
Anders Carlssond76cead2008-01-26 01:36:00 +0000359 }
Eli Friedman1693a3f2008-02-21 17:57:49 +0000360
Anders Carlssond76cead2008-01-26 01:36:00 +0000361 llvm::Constant *VisitStringLiteral(StringLiteral *E) {
Chris Lattnera6dcce32008-02-11 00:02:17 +0000362 assert(!E->getType()->isPointerType() && "Strings are always arrays");
Anders Carlssond76cead2008-01-26 01:36:00 +0000363
364 // Otherwise this must be a string initializing an array in a static
365 // initializer. Don't emit it as the address of the string, emit the string
366 // data itself as an inline array.
Daniel Dunbar31fe9c32008-08-13 23:20:05 +0000367 return llvm::ConstantArray::get(CGM.GetStringForStringLiteral(E), false);
Anders Carlssond76cead2008-01-26 01:36:00 +0000368 }
369
Eli Friedmana36799c2008-05-29 11:22:45 +0000370 llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
371 return Visit(E->getSubExpr());
372 }
Anders Carlsson4d852ee2008-02-05 20:54:21 +0000373
Anders Carlsson1f1cd392009-02-12 17:55:02 +0000374 llvm::Constant *VisitBlockExpr(const BlockExpr *E) {
Mike Stump084ba462009-02-14 22:16:35 +0000375 const char *Name = "";
376 if (const NamedDecl *ND = dyn_cast<NamedDecl>(CGF->CurFuncDecl))
377 Name = ND->getNameAsString().c_str();
378 return CGM.GetAddrOfGlobalBlock(E, Name);
Anders Carlsson1f1cd392009-02-12 17:55:02 +0000379 }
380
Anders Carlssond76cead2008-01-26 01:36:00 +0000381 // Utility methods
382 const llvm::Type *ConvertType(QualType T) {
383 return CGM.getTypes().ConvertType(T);
384 }
385
386 llvm::Constant *EmitConversionToBool(llvm::Constant *Src, QualType SrcType) {
387 assert(SrcType->isCanonical() && "EmitConversion strips typedefs");
388
389 if (SrcType->isRealFloatingType()) {
390 // Compare against 0.0 for fp scalars.
391 llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
392 return llvm::ConstantExpr::getFCmp(llvm::FCmpInst::FCMP_UNE, Src, Zero);
393 }
394
395 assert((SrcType->isIntegerType() || SrcType->isPointerType()) &&
396 "Unknown scalar type to convert");
397
398 // Compare against an integer or pointer null.
399 llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
400 return llvm::ConstantExpr::getICmp(llvm::ICmpInst::ICMP_NE, Src, Zero);
401 }
402
403 llvm::Constant *EmitConversion(llvm::Constant *Src, QualType SrcType,
404 QualType DstType) {
Chris Lattner1793d982008-07-26 22:39:33 +0000405 SrcType = CGM.getContext().getCanonicalType(SrcType);
406 DstType = CGM.getContext().getCanonicalType(DstType);
Anders Carlssond76cead2008-01-26 01:36:00 +0000407 if (SrcType == DstType) return Src;
408
409 // Handle conversions to bool first, they are special: comparisons against 0.
410 if (DstType->isBooleanType())
411 return EmitConversionToBool(Src, SrcType);
412
413 const llvm::Type *DstTy = ConvertType(DstType);
414
415 // Ignore conversions like int -> uint.
416 if (Src->getType() == DstTy)
417 return Src;
418
419 // Handle pointer conversions next: pointers can only be converted to/from
420 // other pointers and integers.
Eli Friedman9f4c8a32008-06-07 17:27:51 +0000421 if (isa<llvm::PointerType>(DstTy)) {
Anders Carlssond76cead2008-01-26 01:36:00 +0000422 // The source value may be an integer, or a pointer.
423 if (isa<llvm::PointerType>(Src->getType()))
424 return llvm::ConstantExpr::getBitCast(Src, DstTy);
Anders Carlssonde598292008-01-26 04:30:23 +0000425 assert(SrcType->isIntegerType() &&"Not ptr->ptr or int->ptr conversion?");
Anders Carlssond76cead2008-01-26 01:36:00 +0000426 return llvm::ConstantExpr::getIntToPtr(Src, DstTy);
427 }
428
Eli Friedman9f4c8a32008-06-07 17:27:51 +0000429 if (isa<llvm::PointerType>(Src->getType())) {
Anders Carlssond76cead2008-01-26 01:36:00 +0000430 // Must be an ptr to int cast.
431 assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
432 return llvm::ConstantExpr::getPtrToInt(Src, DstTy);
433 }
434
435 // A scalar source can be splatted to a vector of the same element type
436 if (isa<llvm::VectorType>(DstTy) && !isa<VectorType>(SrcType)) {
Chris Lattner94c4b2d2008-10-12 00:31:50 +0000437 assert((cast<llvm::VectorType>(DstTy)->getElementType()
438 == Src->getType()) &&
Anders Carlssond76cead2008-01-26 01:36:00 +0000439 "Vector element type must match scalar type to splat.");
440 unsigned NumElements = DstType->getAsVectorType()->getNumElements();
441 llvm::SmallVector<llvm::Constant*, 16> Elements;
442 for (unsigned i = 0; i < NumElements; i++)
443 Elements.push_back(Src);
444
445 return llvm::ConstantVector::get(&Elements[0], NumElements);
446 }
447
448 if (isa<llvm::VectorType>(Src->getType()) ||
449 isa<llvm::VectorType>(DstTy)) {
450 return llvm::ConstantExpr::getBitCast(Src, DstTy);
451 }
452
453 // Finally, we have the arithmetic types: real int/float.
454 if (isa<llvm::IntegerType>(Src->getType())) {
455 bool InputSigned = SrcType->isSignedIntegerType();
456 if (isa<llvm::IntegerType>(DstTy))
457 return llvm::ConstantExpr::getIntegerCast(Src, DstTy, InputSigned);
458 else if (InputSigned)
459 return llvm::ConstantExpr::getSIToFP(Src, DstTy);
460 else
461 return llvm::ConstantExpr::getUIToFP(Src, DstTy);
462 }
463
464 assert(Src->getType()->isFloatingPoint() && "Unknown real conversion");
465 if (isa<llvm::IntegerType>(DstTy)) {
466 if (DstType->isSignedIntegerType())
467 return llvm::ConstantExpr::getFPToSI(Src, DstTy);
468 else
469 return llvm::ConstantExpr::getFPToUI(Src, DstTy);
470 }
471
472 assert(DstTy->isFloatingPoint() && "Unknown real conversion");
473 if (DstTy->getTypeID() < Src->getType()->getTypeID())
474 return llvm::ConstantExpr::getFPTrunc(Src, DstTy);
475 else
476 return llvm::ConstantExpr::getFPExtend(Src, DstTy);
477 }
Anders Carlssonde598292008-01-26 04:30:23 +0000478
Anders Carlsson85b43812008-11-16 06:23:45 +0000479public:
Anders Carlssonde598292008-01-26 04:30:23 +0000480 llvm::Constant *EmitLValue(Expr *E) {
Anders Carlsson00b837c2008-01-26 02:08:50 +0000481 switch (E->getStmtClass()) {
Eli Friedmane493ea92008-02-11 00:23:10 +0000482 default: break;
Anders Carlsson00b837c2008-01-26 02:08:50 +0000483 case Expr::ParenExprClass:
484 // Elide parenthesis
485 return EmitLValue(cast<ParenExpr>(E)->getSubExpr());
486 case Expr::CompoundLiteralExprClass: {
487 // Note that due to the nature of compound literals, this is guaranteed
488 // to be the only use of the variable, so we just generate it here.
Anders Carlssonde598292008-01-26 04:30:23 +0000489 CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
490 llvm::Constant* C = Visit(CLE->getInitializer());
Chris Lattnera6dcce32008-02-11 00:02:17 +0000491 C = new llvm::GlobalVariable(C->getType(),E->getType().isConstQualified(),
Anders Carlsson00b837c2008-01-26 02:08:50 +0000492 llvm::GlobalValue::InternalLinkage,
493 C, ".compoundliteral", &CGM.getModule());
494 return C;
Anders Carlssonde598292008-01-26 04:30:23 +0000495 }
Douglas Gregor566782a2009-01-06 05:10:23 +0000496 case Expr::DeclRefExprClass:
497 case Expr::QualifiedDeclRefExprClass: {
Douglas Gregord2baafd2008-10-21 16:13:35 +0000498 NamedDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
Anders Carlsson00b837c2008-01-26 02:08:50 +0000499 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
Daniel Dunbar7bf5b3d2008-07-29 23:18:29 +0000500 return CGM.GetAddrOfFunction(FD);
Steve Naroff72a6ebc2008-04-15 22:42:06 +0000501 if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
502 if (VD->isFileVarDecl())
Daniel Dunbar7bf5b3d2008-07-29 23:18:29 +0000503 return CGM.GetAddrOfGlobalVar(VD);
Steve Naroff72a6ebc2008-04-15 22:42:06 +0000504 else if (VD->isBlockVarDecl()) {
505 assert(CGF && "Can't access static local vars without CGF");
506 return CGF->GetAddrOfStaticLocalVar(VD);
507 }
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +0000508 }
Eli Friedmane493ea92008-02-11 00:23:10 +0000509 break;
Anders Carlsson00b837c2008-01-26 02:08:50 +0000510 }
511 case Expr::MemberExprClass: {
Anders Carlssonde598292008-01-26 04:30:23 +0000512 MemberExpr* ME = cast<MemberExpr>(E);
Anders Carlssonde598292008-01-26 04:30:23 +0000513 llvm::Constant *Base;
514 if (ME->isArrow())
515 Base = Visit(ME->getBase());
516 else
517 Base = EmitLValue(ME->getBase());
Lauro Ramos Venancio2f049552008-01-30 21:23:20 +0000518
Douglas Gregor82d44772008-12-20 23:49:58 +0000519 FieldDecl *Field = dyn_cast<FieldDecl>(ME->getMemberDecl());
520 // FIXME: Handle other kinds of member expressions.
521 assert(Field && "No code generation for non-field member expressions");
522 unsigned FieldNumber = CGM.getTypes().getLLVMFieldNo(Field);
Anders Carlsson00b837c2008-01-26 02:08:50 +0000523 llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
524 llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
525 FieldNumber);
526 llvm::Value *Ops[] = {Zero, Idx};
527 return llvm::ConstantExpr::getGetElementPtr(Base, Ops, 2);
528 }
529 case Expr::ArraySubscriptExprClass: {
Anders Carlssonde598292008-01-26 04:30:23 +0000530 ArraySubscriptExpr* ASExpr = cast<ArraySubscriptExpr>(E);
531 llvm::Constant *Base = Visit(ASExpr->getBase());
532 llvm::Constant *Index = Visit(ASExpr->getIdx());
Anders Carlsson00b837c2008-01-26 02:08:50 +0000533 assert(!ASExpr->getBase()->getType()->isVectorType() &&
534 "Taking the address of a vector component is illegal!");
535 return llvm::ConstantExpr::getGetElementPtr(Base, &Index, 1);
536 }
Daniel Dunbar31fe9c32008-08-13 23:20:05 +0000537 case Expr::StringLiteralClass:
538 return CGM.GetAddrOfConstantStringFromLiteral(cast<StringLiteral>(E));
Anders Carlsson85b43812008-11-16 06:23:45 +0000539 case Expr::ObjCStringLiteralClass: {
540 ObjCStringLiteral* SL = cast<ObjCStringLiteral>(E);
541 std::string S(SL->getString()->getStrData(),
542 SL->getString()->getByteLength());
543 llvm::Constant *C = CGM.getObjCRuntime().GenerateConstantString(S);
544 return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
545 }
Anders Carlssonde598292008-01-26 04:30:23 +0000546 case Expr::UnaryOperatorClass: {
547 UnaryOperator *Exp = cast<UnaryOperator>(E);
548 switch (Exp->getOpcode()) {
Eli Friedmane493ea92008-02-11 00:23:10 +0000549 default: break;
550 case UnaryOperator::Extension:
Anders Carlssonde598292008-01-26 04:30:23 +0000551 // Extension is just a wrapper for expressions
552 return EmitLValue(Exp->getSubExpr());
553 case UnaryOperator::Real:
554 case UnaryOperator::Imag: {
555 // The address of __real or __imag is just a GEP off the address
556 // of the internal expression
557 llvm::Constant* C = EmitLValue(Exp->getSubExpr());
558 llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
559 llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
Anders Carlsson00b837c2008-01-26 02:08:50 +0000560 Exp->getOpcode() == UnaryOperator::Imag);
Anders Carlssonde598292008-01-26 04:30:23 +0000561 llvm::Value *Ops[] = {Zero, Idx};
562 return llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
563 }
564 case UnaryOperator::Deref:
565 // The address of a deref is just the value of the expression
566 return Visit(Exp->getSubExpr());
567 }
Eli Friedmane493ea92008-02-11 00:23:10 +0000568 break;
Anders Carlsson00b837c2008-01-26 02:08:50 +0000569 }
Chris Lattnerfc370182008-12-12 05:18:02 +0000570
571 case Expr::PredefinedExprClass: {
572 // __func__/__FUNCTION__ -> "". __PRETTY_FUNCTION__ -> "top level".
573 std::string Str;
574 if (cast<PredefinedExpr>(E)->getIdentType() ==
575 PredefinedExpr::PrettyFunction)
576 Str = "top level";
577
578 return CGM.GetAddrOfConstantCString(Str, ".tmp");
579 }
Eli Friedmanf1941132009-01-25 01:21:06 +0000580 case Expr::AddrLabelExprClass: {
581 assert(CGF && "Invalid address of label expression outside function.");
582 unsigned id = CGF->GetIDForAddrOfLabel(cast<AddrLabelExpr>(E)->getLabel());
583 llvm::Constant *C = llvm::ConstantInt::get(llvm::Type::Int32Ty, id);
584 return llvm::ConstantExpr::getIntToPtr(C, ConvertType(E->getType()));
585 }
Eli Friedman67f4ac52009-01-25 01:54:01 +0000586 case Expr::CallExprClass: {
587 CallExpr* CE = cast<CallExpr>(E);
Douglas Gregorb5af7382009-02-14 18:57:46 +0000588 if (CE->isBuiltinCall(CGM.getContext()) !=
589 Builtin::BI__builtin___CFStringMakeConstantString)
Eli Friedman67f4ac52009-01-25 01:54:01 +0000590 break;
591 const Expr *Arg = CE->getArg(0)->IgnoreParenCasts();
592 const StringLiteral *Literal = cast<StringLiteral>(Arg);
593 std::string S(Literal->getStrData(), Literal->getByteLength());
594 return CGM.GetAddrOfConstantCFString(S);
595 }
Eli Friedmane493ea92008-02-11 00:23:10 +0000596 }
Daniel Dunbar9503b782008-08-16 00:56:44 +0000597 CGM.ErrorUnsupported(E, "constant l-value expression");
Eli Friedmane493ea92008-02-11 00:23:10 +0000598 llvm::Type *Ty = llvm::PointerType::getUnqual(ConvertType(E->getType()));
599 return llvm::UndefValue::get(Ty);
Anders Carlsson00b837c2008-01-26 02:08:50 +0000600 }
Anders Carlsson00b837c2008-01-26 02:08:50 +0000601};
602
Anders Carlssond76cead2008-01-26 01:36:00 +0000603} // end anonymous namespace.
604
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +0000605llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
Chris Lattnerc154ac12008-07-26 22:37:01 +0000606 CodeGenFunction *CGF) {
607 QualType type = Context.getCanonicalType(E->getType());
Eli Friedmane2636632008-06-01 15:31:44 +0000608
Anders Carlssond6e823b2008-12-01 02:42:14 +0000609 Expr::EvalResult Result;
610
611 if (E->Evaluate(Result, Context)) {
612 assert(!Result.HasSideEffects &&
613 "Constant expr should not have any side effects!");
614 switch (Result.Val.getKind()) {
Daniel Dunbaraf781bb2009-01-28 22:24:07 +0000615 case APValue::Uninitialized:
616 assert(0 && "Constant expressions should be uninitialized.");
617 return llvm::UndefValue::get(getTypes().ConvertType(type));
Anders Carlsson54483902008-11-15 20:45:50 +0000618 case APValue::LValue: {
Anders Carlssond6e823b2008-12-01 02:42:14 +0000619 llvm::Constant *Offset =
620 llvm::ConstantInt::get(llvm::Type::Int64Ty,
621 Result.Val.getLValueOffset());
Anders Carlsson54483902008-11-15 20:45:50 +0000622
Anders Carlssond6e823b2008-12-01 02:42:14 +0000623 if (const Expr *LVBase = Result.Val.getLValueBase()) {
Anders Carlsson806e9932008-11-17 01:58:55 +0000624 llvm::Constant *C =
Anders Carlsson85b43812008-11-16 06:23:45 +0000625 ConstExprEmitter(*this, CGF).EmitLValue(const_cast<Expr*>(LVBase));
626
Anders Carlsson806e9932008-11-17 01:58:55 +0000627 const llvm::Type *Type =
628 llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
Eli Friedman62301c02008-11-17 03:57:28 +0000629 const llvm::Type *DestType = getTypes().ConvertTypeForMem(E->getType());
Anders Carlsson806e9932008-11-17 01:58:55 +0000630
631 // FIXME: It's a little ugly that we need to cast to a pointer,
632 // apply the GEP and then cast back.
633 C = llvm::ConstantExpr::getBitCast(C, Type);
634 C = llvm::ConstantExpr::getGetElementPtr(C, &Offset, 1);
635
636 return llvm::ConstantExpr::getBitCast(C, DestType);
Anders Carlsson85b43812008-11-16 06:23:45 +0000637 }
Anders Carlsson54483902008-11-15 20:45:50 +0000638
Anders Carlsson85b43812008-11-16 06:23:45 +0000639 return llvm::ConstantExpr::getIntToPtr(Offset,
640 getTypes().ConvertType(type));
Anders Carlsson54483902008-11-15 20:45:50 +0000641 }
Eli Friedman62301c02008-11-17 03:57:28 +0000642 case APValue::Int: {
Anders Carlssond6e823b2008-12-01 02:42:14 +0000643 llvm::Constant *C = llvm::ConstantInt::get(Result.Val.getInt());
Anders Carlsson54483902008-11-15 20:45:50 +0000644
645 if (C->getType() == llvm::Type::Int1Ty) {
646 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
647 C = llvm::ConstantExpr::getZExt(C, BoolTy);
648 }
649 return C;
Eli Friedman62301c02008-11-17 03:57:28 +0000650 }
Daniel Dunbaraf781bb2009-01-28 22:24:07 +0000651 case APValue::ComplexInt: {
652 llvm::Constant *Complex[2];
653
654 Complex[0] = llvm::ConstantInt::get(Result.Val.getComplexIntReal());
655 Complex[1] = llvm::ConstantInt::get(Result.Val.getComplexIntImag());
656
657 return llvm::ConstantStruct::get(Complex, 2);
658 }
Anders Carlsson54483902008-11-15 20:45:50 +0000659 case APValue::Float:
Anders Carlssond6e823b2008-12-01 02:42:14 +0000660 return llvm::ConstantFP::get(Result.Val.getFloat());
Anders Carlsson806e9932008-11-17 01:58:55 +0000661 case APValue::ComplexFloat: {
662 llvm::Constant *Complex[2];
663
Anders Carlssond6e823b2008-12-01 02:42:14 +0000664 Complex[0] = llvm::ConstantFP::get(Result.Val.getComplexFloatReal());
665 Complex[1] = llvm::ConstantFP::get(Result.Val.getComplexFloatImag());
Anders Carlsson806e9932008-11-17 01:58:55 +0000666
667 return llvm::ConstantStruct::get(Complex, 2);
668 }
Nate Begemanc45d4e82009-01-18 01:01:34 +0000669 case APValue::Vector: {
670 llvm::SmallVector<llvm::Constant *, 4> Inits;
671 unsigned NumElts = Result.Val.getVectorLength();
672
673 for (unsigned i = 0; i != NumElts; ++i) {
674 APValue &Elt = Result.Val.getVectorElt(i);
675 if (Elt.isInt())
676 Inits.push_back(llvm::ConstantInt::get(Elt.getInt()));
677 else
678 Inits.push_back(llvm::ConstantFP::get(Elt.getFloat()));
679 }
680 return llvm::ConstantVector::get(&Inits[0], Inits.size());
681 }
Anders Carlsson54483902008-11-15 20:45:50 +0000682 }
683 }
Eli Friedmane2636632008-06-01 15:31:44 +0000684
685 llvm::Constant* C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
686 if (C->getType() == llvm::Type::Int1Ty) {
687 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
688 C = llvm::ConstantExpr::getZExt(C, BoolTy);
689 }
690 return C;
Anders Carlssond76cead2008-01-26 01:36:00 +0000691}