blob: 7f294e29965e08d9f4a22d6cf6cbd2987b203e64 [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"
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +000016#include "CGObjCRuntime.h"
Chris Lattnerf6b24ea2008-10-06 05:59:01 +000017#include "clang/AST/APValue.h"
Daniel Dunbarde7fb842008-08-11 05:00:27 +000018#include "clang/AST/ASTContext.h"
19#include "clang/AST/StmtVisitor.h"
Anders Carlsson3b1d57b2008-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 Friedmana8234002008-05-30 10:24:46 +000024#include "llvm/Target/TargetData.h"
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000025using namespace clang;
26using namespace CodeGen;
27
28namespace {
Anders Carlsson84005b42008-01-26 04:30:23 +000029class VISIBILITY_HIDDEN ConstExprEmitter :
30 public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000031 CodeGenModule &CGM;
Lauro Ramos Venancio81373352008-02-26 21:41:45 +000032 CodeGenFunction *CGF;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000033public:
Lauro Ramos Venancio81373352008-02-26 21:41:45 +000034 ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
35 : CGM(cgm), CGF(cgf) {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000036 }
37
38 //===--------------------------------------------------------------------===//
39 // Visitor Methods
40 //===--------------------------------------------------------------------===//
41
42 llvm::Constant *VisitStmt(Stmt *S) {
Daniel Dunbar488e9932008-08-16 00:56:44 +000043 CGM.ErrorUnsupported(S, "constant expression");
Chris Lattner98be4942008-03-05 18:54:05 +000044 QualType T = cast<Expr>(S)->getType();
45 return llvm::UndefValue::get(CGM.getTypes().ConvertType(T));
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000046 }
47
48 llvm::Constant *VisitParenExpr(ParenExpr *PE) {
49 return Visit(PE->getSubExpr());
50 }
51
Chris Lattner13fd7e52008-06-21 21:44:18 +000052 llvm::Constant *VisitObjCStringLiteral(const ObjCStringLiteral *E) {
Daniel Dunbarbbce49b2008-08-12 00:12:39 +000053 std::string S(E->getString()->getStrData(),
54 E->getString()->getByteLength());
55 llvm::Constant *C = CGM.getObjCRuntime().GenerateConstantString(S);
56 return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
Chris Lattner13fd7e52008-06-21 21:44:18 +000057 }
Anders Carlssonbcf62252008-11-22 02:34:39 +000058
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000059 llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
60 return Visit(E->getInitializer());
61 }
62
Argyrios Kyrtzidis0835a3c2008-08-18 23:01:59 +000063 llvm::Constant *VisitCastExpr(CastExpr* E) {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000064 llvm::Constant *C = Visit(E->getSubExpr());
65
66 return EmitConversion(C, E->getSubExpr()->getType(), E->getType());
67 }
Devang Pateleae15602008-02-05 02:39:50 +000068
Chris Lattner04421082008-04-08 04:40:51 +000069 llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
70 return Visit(DAE->getExpr());
71 }
72
Eli Friedmana8234002008-05-30 10:24:46 +000073 llvm::Constant *EmitArrayInitialization(InitListExpr *ILE) {
Eli Friedman4d1c0182008-02-21 17:45:41 +000074 std::vector<llvm::Constant*> Elts;
Eli Friedmana8234002008-05-30 10:24:46 +000075 const llvm::ArrayType *AType =
76 cast<llvm::ArrayType>(ConvertType(ILE->getType()));
Eli Friedman4d1c0182008-02-21 17:45:41 +000077 unsigned NumInitElements = ILE->getNumInits();
78 // FIXME: Check for wide strings
79 if (NumInitElements > 0 && isa<StringLiteral>(ILE->getInit(0)) &&
Chris Lattnerc63a1f22008-08-04 07:31:14 +000080 ILE->getType()->getArrayElementTypeNoTypeQual()->isCharType())
Eli Friedman4d1c0182008-02-21 17:45:41 +000081 return Visit(ILE->getInit(0));
Devang Pateleae15602008-02-05 02:39:50 +000082 const llvm::Type *ElemTy = AType->getElementType();
83 unsigned NumElements = AType->getNumElements();
84
85 // Initialising an array requires us to automatically
86 // initialise any elements that have not been initialised explicitly
87 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
88
89 // Copy initializer elements.
90 unsigned i = 0;
Eli Friedmana04a1532008-05-30 19:58:50 +000091 bool RewriteType = false;
Devang Pateleae15602008-02-05 02:39:50 +000092 for (; i < NumInitableElts; ++i) {
Anders Carlssonaa432562008-11-17 01:58:55 +000093 llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(i), CGF);
Eli Friedmana04a1532008-05-30 19:58:50 +000094 RewriteType |= (C->getType() != ElemTy);
Devang Pateleae15602008-02-05 02:39:50 +000095 Elts.push_back(C);
96 }
Eli Friedmana04a1532008-05-30 19:58:50 +000097
Devang Pateleae15602008-02-05 02:39:50 +000098 // Initialize remaining array elements.
99 for (; i < NumElements; ++i)
100 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
101
Eli Friedmana04a1532008-05-30 19:58:50 +0000102 if (RewriteType) {
103 // FIXME: Try to avoid packing the array
104 std::vector<const llvm::Type*> Types;
105 for (unsigned i = 0; i < Elts.size(); ++i)
106 Types.push_back(Elts[i]->getType());
107 const llvm::StructType *SType = llvm::StructType::get(Types, true);
108 return llvm::ConstantStruct::get(SType, Elts);
109 }
110
Devang Pateleae15602008-02-05 02:39:50 +0000111 return llvm::ConstantArray::get(AType, Elts);
112 }
113
Eli Friedmana8234002008-05-30 10:24:46 +0000114 void InsertBitfieldIntoStruct(std::vector<llvm::Constant*>& Elts,
115 FieldDecl* Field, Expr* E) {
116 // Calculate the value to insert
Anders Carlssonaa432562008-11-17 01:58:55 +0000117 llvm::Constant *C = CGM.EmitConstantExpr(E, CGF);
Eli Friedmana8234002008-05-30 10:24:46 +0000118 llvm::ConstantInt *CI = dyn_cast<llvm::ConstantInt>(C);
119 if (!CI) {
Daniel Dunbar488e9932008-08-16 00:56:44 +0000120 CGM.ErrorUnsupported(E, "bitfield initialization");
Eli Friedmana8234002008-05-30 10:24:46 +0000121 return;
122 }
123 llvm::APInt V = CI->getValue();
Devang Pateleae15602008-02-05 02:39:50 +0000124
Eli Friedmana8234002008-05-30 10:24:46 +0000125 // Calculate information about the relevant field
126 const llvm::Type* Ty = CI->getType();
Daniel Dunbar9ca8bb02008-08-10 07:00:24 +0000127 const llvm::TargetData &TD = CGM.getTypes().getTargetData();
128 unsigned size = TD.getTypeStoreSizeInBits(Ty);
Eli Friedmana8234002008-05-30 10:24:46 +0000129 unsigned fieldOffset = CGM.getTypes().getLLVMFieldNo(Field) * size;
130 CodeGenTypes::BitFieldInfo bitFieldInfo =
131 CGM.getTypes().getBitFieldInfo(Field);
132 fieldOffset += bitFieldInfo.Begin;
133
134 // Find where to start the insertion
135 // FIXME: This is O(n^2) in the number of bit-fields!
136 // FIXME: This won't work if the struct isn't completely packed!
137 unsigned offset = 0, i = 0;
138 while (offset < (fieldOffset & -8))
Daniel Dunbar9ca8bb02008-08-10 07:00:24 +0000139 offset += TD.getTypeStoreSizeInBits(Elts[i++]->getType());
140
141 // Advance over 0 sized elements (must terminate in bounds since
142 // the bitfield must have a size).
143 while (TD.getTypeStoreSizeInBits(Elts[i]->getType()) == 0)
144 ++i;
Eli Friedmana8234002008-05-30 10:24:46 +0000145
Daniel Dunbar1f3846e2008-07-17 21:07:48 +0000146 // Promote the size of V if necessary
147 // FIXME: This should never occur, but currently it can because
148 // initializer constants are cast to bool, and because clang is
149 // not enforcing bitfield width limits.
150 if (bitFieldInfo.Size > V.getBitWidth())
151 V.zext(bitFieldInfo.Size);
152
Eli Friedmana8234002008-05-30 10:24:46 +0000153 // Insert the bits into the struct
154 // FIXME: This algorthm is only correct on X86!
155 // FIXME: THis algorthm assumes bit-fields only have byte-size elements!
156 unsigned bitsToInsert = bitFieldInfo.Size;
157 unsigned curBits = std::min(8 - (fieldOffset & 7), bitsToInsert);
158 unsigned byte = V.getLoBits(curBits).getZExtValue() << (fieldOffset & 7);
159 do {
160 llvm::Constant* byteC = llvm::ConstantInt::get(llvm::Type::Int8Ty, byte);
161 Elts[i] = llvm::ConstantExpr::getOr(Elts[i], byteC);
162 ++i;
163 V = V.lshr(curBits);
164 bitsToInsert -= curBits;
165
166 if (!bitsToInsert)
167 break;
168
169 curBits = bitsToInsert > 8 ? 8 : bitsToInsert;
170 byte = V.getLoBits(curBits).getZExtValue();
171 } while (true);
172 }
173
174 llvm::Constant *EmitStructInitialization(InitListExpr *ILE) {
175 const llvm::StructType *SType =
176 cast<llvm::StructType>(ConvertType(ILE->getType()));
Eli Friedman017cbdf2008-05-20 00:11:07 +0000177 RecordDecl *RD = ILE->getType()->getAsRecordType()->getDecl();
Devang Pateleae15602008-02-05 02:39:50 +0000178 std::vector<llvm::Constant*> Elts;
Devang Pateleae15602008-02-05 02:39:50 +0000179
Eli Friedman017cbdf2008-05-20 00:11:07 +0000180 // Initialize the whole structure to zero.
181 for (unsigned i = 0; i < SType->getNumElements(); ++i) {
Devang Pateleae15602008-02-05 02:39:50 +0000182 const llvm::Type *FieldTy = SType->getElementType(i);
183 Elts.push_back(llvm::Constant::getNullValue(FieldTy));
184 }
Eli Friedman017cbdf2008-05-20 00:11:07 +0000185
186 // Copy initializer elements. Skip padding fields.
187 unsigned EltNo = 0; // Element no in ILE
188 int FieldNo = 0; // Field no in RecordDecl
Eli Friedmana04a1532008-05-30 19:58:50 +0000189 bool RewriteType = false;
Eli Friedman017cbdf2008-05-20 00:11:07 +0000190 while (EltNo < ILE->getNumInits() && FieldNo < RD->getNumMembers()) {
191 FieldDecl* curField = RD->getMember(FieldNo);
192 FieldNo++;
193 if (!curField->getIdentifier())
194 continue;
195
Eli Friedman017cbdf2008-05-20 00:11:07 +0000196 if (curField->isBitField()) {
Eli Friedmana8234002008-05-30 10:24:46 +0000197 InsertBitfieldIntoStruct(Elts, curField, ILE->getInit(EltNo));
Eli Friedman017cbdf2008-05-20 00:11:07 +0000198 } else {
Eli Friedmana04a1532008-05-30 19:58:50 +0000199 unsigned FieldNo = CGM.getTypes().getLLVMFieldNo(curField);
Anders Carlssonaa432562008-11-17 01:58:55 +0000200 llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(EltNo), CGF);
Eli Friedmana04a1532008-05-30 19:58:50 +0000201 RewriteType |= (C->getType() != Elts[FieldNo]->getType());
202 Elts[FieldNo] = C;
Eli Friedman017cbdf2008-05-20 00:11:07 +0000203 }
204 EltNo++;
205 }
206
Eli Friedmana04a1532008-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 Pateleae15602008-02-05 02:39:50 +0000216 return llvm::ConstantStruct::get(SType, Elts);
217 }
218
Eli Friedmana8234002008-05-30 10:24:46 +0000219 llvm::Constant *EmitUnionInitialization(InitListExpr *ILE) {
Eli Friedmana8234002008-05-30 10:24:46 +0000220 RecordDecl *RD = ILE->getType()->getAsRecordType()->getDecl();
221 const llvm::Type *Ty = ConvertType(ILE->getType());
222
223 // Find the field decl we're initializing, if any
224 int FieldNo = 0; // Field no in RecordDecl
Eli Friedmanc55f98d2008-06-06 20:12:37 +0000225 FieldDecl* curField = 0;
226 while (FieldNo < RD->getNumMembers()) {
Eli Friedmana8234002008-05-30 10:24:46 +0000227 curField = RD->getMember(FieldNo);
228 FieldNo++;
Eli Friedmanc55f98d2008-06-06 20:12:37 +0000229 if (curField->getIdentifier())
230 break;
231 }
Eli Friedmana8234002008-05-30 10:24:46 +0000232
Eli Friedmanc55f98d2008-06-06 20:12:37 +0000233 if (!curField || !curField->getIdentifier() || ILE->getNumInits() == 0)
Eli Friedmana8234002008-05-30 10:24:46 +0000234 return llvm::Constant::getNullValue(Ty);
235
236 if (curField->isBitField()) {
237 // Create a dummy struct for bit-field insertion
238 unsigned NumElts = CGM.getTargetData().getABITypeSize(Ty) / 8;
239 llvm::Constant* NV = llvm::Constant::getNullValue(llvm::Type::Int8Ty);
240 std::vector<llvm::Constant*> Elts(NumElts, NV);
241
242 InsertBitfieldIntoStruct(Elts, curField, ILE->getInit(0));
243 const llvm::ArrayType *RetTy =
244 llvm::ArrayType::get(NV->getType(), NumElts);
245 return llvm::ConstantArray::get(RetTy, Elts);
246 }
247
Anders Carlssonaa432562008-11-17 01:58:55 +0000248 llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(0), CGF);
Eli Friedmana8234002008-05-30 10:24:46 +0000249
250 // Build a struct with the union sub-element as the first member,
251 // and padded to the appropriate size
252 std::vector<llvm::Constant*> Elts;
253 std::vector<const llvm::Type*> Types;
254 Elts.push_back(C);
255 Types.push_back(C->getType());
256 unsigned CurSize = CGM.getTargetData().getTypeStoreSize(C->getType());
257 unsigned TotalSize = CGM.getTargetData().getTypeStoreSize(Ty);
258 while (CurSize < TotalSize) {
259 Elts.push_back(llvm::Constant::getNullValue(llvm::Type::Int8Ty));
260 Types.push_back(llvm::Type::Int8Ty);
261 CurSize++;
262 }
263
264 // This always generates a packed struct
265 // FIXME: Try to generate an unpacked struct when we can
266 llvm::StructType* STy = llvm::StructType::get(Types, true);
267 return llvm::ConstantStruct::get(STy, Elts);
268 }
269
270 llvm::Constant *EmitVectorInitialization(InitListExpr *ILE) {
271 const llvm::VectorType *VType =
272 cast<llvm::VectorType>(ConvertType(ILE->getType()));
Eli Friedmancfb313b2008-05-30 12:02:14 +0000273 const llvm::Type *ElemTy = VType->getElementType();
274 std::vector<llvm::Constant*> Elts;
Devang Pateleae15602008-02-05 02:39:50 +0000275 unsigned NumElements = VType->getNumElements();
Eli Friedmancfb313b2008-05-30 12:02:14 +0000276 unsigned NumInitElements = ILE->getNumInits();
Devang Pateleae15602008-02-05 02:39:50 +0000277
Eli Friedmancfb313b2008-05-30 12:02:14 +0000278 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
279
Devang Pateleae15602008-02-05 02:39:50 +0000280 // Copy initializer elements.
281 unsigned i = 0;
Eli Friedmancfb313b2008-05-30 12:02:14 +0000282 for (; i < NumInitableElts; ++i) {
Anders Carlssonaa432562008-11-17 01:58:55 +0000283 llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(i), CGF);
Devang Pateleae15602008-02-05 02:39:50 +0000284 Elts.push_back(C);
285 }
286
Eli Friedmancfb313b2008-05-30 12:02:14 +0000287 for (; i < NumElements; ++i)
288 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
289
Devang Pateleae15602008-02-05 02:39:50 +0000290 return llvm::ConstantVector::get(VType, Elts);
291 }
292
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000293 llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
Eli Friedmana8234002008-05-30 10:24:46 +0000294 if (ILE->getType()->isScalarType()) {
295 // We have a scalar in braces. Just use the first element.
296 if (ILE->getNumInits() > 0)
Anders Carlssonaa432562008-11-17 01:58:55 +0000297 return CGM.EmitConstantExpr(ILE->getInit(0), CGF);
Eli Friedmana8234002008-05-30 10:24:46 +0000298
299 const llvm::Type* RetTy = CGM.getTypes().ConvertType(ILE->getType());
300 return llvm::Constant::getNullValue(RetTy);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000301 }
Chris Lattnerbe20bb52008-10-26 23:53:12 +0000302
303 // FIXME: We don't codegen or sema designators yet.
304 if (ILE->hadDesignators()) {
305 CGM.ErrorUnsupported(ILE, "initializer list with designators");
306 return llvm::UndefValue::get(ConvertType(ILE->getType()));
307 }
Devang Patel47fb6972008-01-29 23:23:18 +0000308
Eli Friedmana8234002008-05-30 10:24:46 +0000309 if (ILE->getType()->isArrayType())
310 return EmitArrayInitialization(ILE);
Devang Patel47fb6972008-01-29 23:23:18 +0000311
Eli Friedmana8234002008-05-30 10:24:46 +0000312 if (ILE->getType()->isStructureType())
313 return EmitStructInitialization(ILE);
314
315 if (ILE->getType()->isUnionType())
316 return EmitUnionInitialization(ILE);
317
318 if (ILE->getType()->isVectorType())
319 return EmitVectorInitialization(ILE);
320
Devang Pateleae15602008-02-05 02:39:50 +0000321 assert(0 && "Unable to handle InitListExpr");
Chris Lattner2b94fe32008-03-01 08:45:05 +0000322 // Get rid of control reaches end of void function warning.
323 // Not reached.
324 return 0;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000325 }
Eli Friedman3b707e72008-02-21 17:57:49 +0000326
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000327 llvm::Constant *VisitImplicitCastExpr(ImplicitCastExpr *ICExpr) {
Eli Friedman3b707e72008-02-21 17:57:49 +0000328 Expr* SExpr = ICExpr->getSubExpr();
329 QualType SType = SExpr->getType();
330 llvm::Constant *C; // the intermediate expression
331 QualType T; // the type of the intermediate expression
332 if (SType->isArrayType()) {
333 // Arrays decay to a pointer to the first element
334 // VLAs would require special handling, but they can't occur here
335 C = EmitLValue(SExpr);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000336 llvm::Constant *Idx0 = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000337 llvm::Constant *Ops[] = {Idx0, Idx0};
338 C = llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
Chris Lattnerc63a1f22008-08-04 07:31:14 +0000339 T = CGM.getContext().getArrayDecayedType(SType);
Eli Friedman3b707e72008-02-21 17:57:49 +0000340 } else if (SType->isFunctionType()) {
341 // Function types decay to a pointer to the function
342 C = EmitLValue(SExpr);
343 T = CGM.getContext().getPointerType(SType);
344 } else {
345 C = Visit(SExpr);
346 T = SType;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000347 }
Eli Friedman3b707e72008-02-21 17:57:49 +0000348
349 // Perform the conversion; note that an implicit cast can both promote
350 // and convert an array/function
351 return EmitConversion(C, T, ICExpr->getType());
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000352 }
Eli Friedman3b707e72008-02-21 17:57:49 +0000353
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000354 llvm::Constant *VisitStringLiteral(StringLiteral *E) {
Chris Lattnera7ad98f2008-02-11 00:02:17 +0000355 assert(!E->getType()->isPointerType() && "Strings are always arrays");
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000356
357 // Otherwise this must be a string initializing an array in a static
358 // initializer. Don't emit it as the address of the string, emit the string
359 // data itself as an inline array.
Daniel Dunbar61432932008-08-13 23:20:05 +0000360 return llvm::ConstantArray::get(CGM.GetStringForStringLiteral(E), false);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000361 }
362
Daniel Dunbar54d19092008-08-16 01:41:47 +0000363 llvm::Constant *VisitAddrLabelExpr(const AddrLabelExpr *E) {
364 assert(CGF && "Invalid address of label expression outside function.");
365 llvm::Constant *C =
366 llvm::ConstantInt::get(llvm::Type::Int32Ty,
367 CGF->GetIDForAddrOfLabel(E->getLabel()));
368 return llvm::ConstantExpr::getIntToPtr(C, ConvertType(E->getType()));
369 }
370
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000371 llvm::Constant *VisitUnaryAddrOf(const UnaryOperator *E) {
372 return EmitLValue(E->getSubExpr());
373 }
Anders Carlsson5a1deb82008-01-29 15:56:48 +0000374 llvm::Constant *VisitUnaryOffsetOf(const UnaryOperator *E) {
375 int64_t Val = E->evaluateOffsetOf(CGM.getContext());
376
377 assert(E->getType()->isIntegerType() && "Result type must be an integer!");
378
Chris Lattner98be4942008-03-05 18:54:05 +0000379 uint32_t ResultWidth =
380 static_cast<uint32_t>(CGM.getContext().getTypeSize(E->getType()));
Anders Carlsson5a1deb82008-01-29 15:56:48 +0000381 return llvm::ConstantInt::get(llvm::APInt(ResultWidth, Val));
382 }
Eli Friedmanbc5ed6e2008-05-29 11:22:45 +0000383
384 llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
385 return Visit(E->getSubExpr());
386 }
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000387
Anders Carlssonce6237e2008-01-29 01:33:32 +0000388 // Binary operators
Anders Carlsson2ea2b5e2008-08-25 03:27:15 +0000389
390 llvm::Constant *VisitCallExpr(const CallExpr *E) {
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000391 Expr::EvalResult Result;
Chris Lattner6ee7aa12008-11-16 21:24:15 +0000392 if (E->Evaluate(Result, CGM.getContext())) {
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000393 if (Result.Val.isInt())
394 return llvm::ConstantInt::get(Result.Val.getInt());
395 if (Result.Val.isFloat())
396 return llvm::ConstantFP::get(Result.Val.getFloat());
Chris Lattnerf6b24ea2008-10-06 05:59:01 +0000397 }
Anders Carlsson2ea2b5e2008-08-25 03:27:15 +0000398
Chris Lattner506ff882008-10-06 07:26:43 +0000399 // Handle __builtin___CFStringMakeConstantString.
400 if (E->isBuiltinCall() ==Builtin::BI__builtin___CFStringMakeConstantString){
Anders Carlssond114ade2008-11-15 18:52:06 +0000401 const Expr *Arg = E->getArg(0)->IgnoreParenCasts();
Chris Lattner506ff882008-10-06 07:26:43 +0000402
403 const StringLiteral *Literal = cast<StringLiteral>(Arg);
404 std::string S(Literal->getStrData(), Literal->getByteLength());
405 return CGM.GetAddrOfConstantCFString(S);
406 }
407
Anders Carlsson2ea2b5e2008-08-25 03:27:15 +0000408 CGM.ErrorUnsupported(E, "constant call expression");
409 return llvm::Constant::getNullValue(ConvertType(E->getType()));
410 }
Anders Carlsson7be3d012008-02-05 20:54:21 +0000411
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000412 // Utility methods
413 const llvm::Type *ConvertType(QualType T) {
414 return CGM.getTypes().ConvertType(T);
415 }
416
417 llvm::Constant *EmitConversionToBool(llvm::Constant *Src, QualType SrcType) {
418 assert(SrcType->isCanonical() && "EmitConversion strips typedefs");
419
420 if (SrcType->isRealFloatingType()) {
421 // Compare against 0.0 for fp scalars.
422 llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
423 return llvm::ConstantExpr::getFCmp(llvm::FCmpInst::FCMP_UNE, Src, Zero);
424 }
425
426 assert((SrcType->isIntegerType() || SrcType->isPointerType()) &&
427 "Unknown scalar type to convert");
428
429 // Compare against an integer or pointer null.
430 llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
431 return llvm::ConstantExpr::getICmp(llvm::ICmpInst::ICMP_NE, Src, Zero);
432 }
433
434 llvm::Constant *EmitConversion(llvm::Constant *Src, QualType SrcType,
435 QualType DstType) {
Chris Lattnerd394cc62008-07-26 22:39:33 +0000436 SrcType = CGM.getContext().getCanonicalType(SrcType);
437 DstType = CGM.getContext().getCanonicalType(DstType);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000438 if (SrcType == DstType) return Src;
439
440 // Handle conversions to bool first, they are special: comparisons against 0.
441 if (DstType->isBooleanType())
442 return EmitConversionToBool(Src, SrcType);
443
444 const llvm::Type *DstTy = ConvertType(DstType);
445
446 // Ignore conversions like int -> uint.
447 if (Src->getType() == DstTy)
448 return Src;
449
450 // Handle pointer conversions next: pointers can only be converted to/from
451 // other pointers and integers.
Eli Friedmanc53296e2008-06-07 17:27:51 +0000452 if (isa<llvm::PointerType>(DstTy)) {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000453 // The source value may be an integer, or a pointer.
454 if (isa<llvm::PointerType>(Src->getType()))
455 return llvm::ConstantExpr::getBitCast(Src, DstTy);
Anders Carlsson84005b42008-01-26 04:30:23 +0000456 assert(SrcType->isIntegerType() &&"Not ptr->ptr or int->ptr conversion?");
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000457 return llvm::ConstantExpr::getIntToPtr(Src, DstTy);
458 }
459
Eli Friedmanc53296e2008-06-07 17:27:51 +0000460 if (isa<llvm::PointerType>(Src->getType())) {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000461 // Must be an ptr to int cast.
462 assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
463 return llvm::ConstantExpr::getPtrToInt(Src, DstTy);
464 }
465
466 // A scalar source can be splatted to a vector of the same element type
467 if (isa<llvm::VectorType>(DstTy) && !isa<VectorType>(SrcType)) {
Chris Lattner3304e552008-10-12 00:31:50 +0000468 assert((cast<llvm::VectorType>(DstTy)->getElementType()
469 == Src->getType()) &&
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000470 "Vector element type must match scalar type to splat.");
471 unsigned NumElements = DstType->getAsVectorType()->getNumElements();
472 llvm::SmallVector<llvm::Constant*, 16> Elements;
473 for (unsigned i = 0; i < NumElements; i++)
474 Elements.push_back(Src);
475
476 return llvm::ConstantVector::get(&Elements[0], NumElements);
477 }
478
479 if (isa<llvm::VectorType>(Src->getType()) ||
480 isa<llvm::VectorType>(DstTy)) {
481 return llvm::ConstantExpr::getBitCast(Src, DstTy);
482 }
483
484 // Finally, we have the arithmetic types: real int/float.
485 if (isa<llvm::IntegerType>(Src->getType())) {
486 bool InputSigned = SrcType->isSignedIntegerType();
487 if (isa<llvm::IntegerType>(DstTy))
488 return llvm::ConstantExpr::getIntegerCast(Src, DstTy, InputSigned);
489 else if (InputSigned)
490 return llvm::ConstantExpr::getSIToFP(Src, DstTy);
491 else
492 return llvm::ConstantExpr::getUIToFP(Src, DstTy);
493 }
494
495 assert(Src->getType()->isFloatingPoint() && "Unknown real conversion");
496 if (isa<llvm::IntegerType>(DstTy)) {
497 if (DstType->isSignedIntegerType())
498 return llvm::ConstantExpr::getFPToSI(Src, DstTy);
499 else
500 return llvm::ConstantExpr::getFPToUI(Src, DstTy);
501 }
502
503 assert(DstTy->isFloatingPoint() && "Unknown real conversion");
504 if (DstTy->getTypeID() < Src->getType()->getTypeID())
505 return llvm::ConstantExpr::getFPTrunc(Src, DstTy);
506 else
507 return llvm::ConstantExpr::getFPExtend(Src, DstTy);
508 }
509
510 llvm::Constant *EmitSizeAlignOf(QualType TypeToSize,
511 QualType RetType, bool isSizeOf) {
512 std::pair<uint64_t, unsigned> Info =
Chris Lattner98be4942008-03-05 18:54:05 +0000513 CGM.getContext().getTypeInfo(TypeToSize);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000514
515 uint64_t Val = isSizeOf ? Info.first : Info.second;
516 Val /= 8; // Return size in bytes, not bits.
517
518 assert(RetType->isIntegerType() && "Result type must be an integer!");
519
Chris Lattner98be4942008-03-05 18:54:05 +0000520 uint32_t ResultWidth =
521 static_cast<uint32_t>(CGM.getContext().getTypeSize(RetType));
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000522 return llvm::ConstantInt::get(llvm::APInt(ResultWidth, Val));
523 }
Anders Carlsson84005b42008-01-26 04:30:23 +0000524
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000525public:
Anders Carlsson84005b42008-01-26 04:30:23 +0000526 llvm::Constant *EmitLValue(Expr *E) {
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000527 switch (E->getStmtClass()) {
Eli Friedman546d94c2008-02-11 00:23:10 +0000528 default: break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000529 case Expr::ParenExprClass:
530 // Elide parenthesis
531 return EmitLValue(cast<ParenExpr>(E)->getSubExpr());
532 case Expr::CompoundLiteralExprClass: {
533 // Note that due to the nature of compound literals, this is guaranteed
534 // to be the only use of the variable, so we just generate it here.
Anders Carlsson84005b42008-01-26 04:30:23 +0000535 CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
536 llvm::Constant* C = Visit(CLE->getInitializer());
Chris Lattnera7ad98f2008-02-11 00:02:17 +0000537 C = new llvm::GlobalVariable(C->getType(),E->getType().isConstQualified(),
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000538 llvm::GlobalValue::InternalLinkage,
539 C, ".compoundliteral", &CGM.getModule());
540 return C;
Anders Carlsson84005b42008-01-26 04:30:23 +0000541 }
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000542 case Expr::DeclRefExprClass: {
Douglas Gregor8e9bebd2008-10-21 16:13:35 +0000543 NamedDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000544 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000545 return CGM.GetAddrOfFunction(FD);
Steve Naroff248a7532008-04-15 22:42:06 +0000546 if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
547 if (VD->isFileVarDecl())
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000548 return CGM.GetAddrOfGlobalVar(VD);
Steve Naroff248a7532008-04-15 22:42:06 +0000549 else if (VD->isBlockVarDecl()) {
550 assert(CGF && "Can't access static local vars without CGF");
551 return CGF->GetAddrOfStaticLocalVar(VD);
552 }
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000553 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000554 break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000555 }
556 case Expr::MemberExprClass: {
Anders Carlsson84005b42008-01-26 04:30:23 +0000557 MemberExpr* ME = cast<MemberExpr>(E);
Anders Carlsson84005b42008-01-26 04:30:23 +0000558 llvm::Constant *Base;
559 if (ME->isArrow())
560 Base = Visit(ME->getBase());
561 else
562 Base = EmitLValue(ME->getBase());
Lauro Ramos Venancio5bfe18c2008-01-30 21:23:20 +0000563
564 unsigned FieldNumber = CGM.getTypes().getLLVMFieldNo(ME->getMemberDecl());
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000565 llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
566 llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
567 FieldNumber);
568 llvm::Value *Ops[] = {Zero, Idx};
569 return llvm::ConstantExpr::getGetElementPtr(Base, Ops, 2);
570 }
571 case Expr::ArraySubscriptExprClass: {
Anders Carlsson84005b42008-01-26 04:30:23 +0000572 ArraySubscriptExpr* ASExpr = cast<ArraySubscriptExpr>(E);
573 llvm::Constant *Base = Visit(ASExpr->getBase());
574 llvm::Constant *Index = Visit(ASExpr->getIdx());
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000575 assert(!ASExpr->getBase()->getType()->isVectorType() &&
576 "Taking the address of a vector component is illegal!");
577 return llvm::ConstantExpr::getGetElementPtr(Base, &Index, 1);
578 }
Daniel Dunbar61432932008-08-13 23:20:05 +0000579 case Expr::StringLiteralClass:
580 return CGM.GetAddrOfConstantStringFromLiteral(cast<StringLiteral>(E));
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000581 case Expr::ObjCStringLiteralClass: {
582 ObjCStringLiteral* SL = cast<ObjCStringLiteral>(E);
583 std::string S(SL->getString()->getStrData(),
584 SL->getString()->getByteLength());
585 llvm::Constant *C = CGM.getObjCRuntime().GenerateConstantString(S);
586 return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
587 }
Anders Carlsson84005b42008-01-26 04:30:23 +0000588 case Expr::UnaryOperatorClass: {
589 UnaryOperator *Exp = cast<UnaryOperator>(E);
590 switch (Exp->getOpcode()) {
Eli Friedman546d94c2008-02-11 00:23:10 +0000591 default: break;
592 case UnaryOperator::Extension:
Anders Carlsson84005b42008-01-26 04:30:23 +0000593 // Extension is just a wrapper for expressions
594 return EmitLValue(Exp->getSubExpr());
595 case UnaryOperator::Real:
596 case UnaryOperator::Imag: {
597 // The address of __real or __imag is just a GEP off the address
598 // of the internal expression
599 llvm::Constant* C = EmitLValue(Exp->getSubExpr());
600 llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
601 llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000602 Exp->getOpcode() == UnaryOperator::Imag);
Anders Carlsson84005b42008-01-26 04:30:23 +0000603 llvm::Value *Ops[] = {Zero, Idx};
604 return llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
605 }
606 case UnaryOperator::Deref:
607 // The address of a deref is just the value of the expression
608 return Visit(Exp->getSubExpr());
609 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000610 break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000611 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000612 }
Daniel Dunbar488e9932008-08-16 00:56:44 +0000613 CGM.ErrorUnsupported(E, "constant l-value expression");
Eli Friedman546d94c2008-02-11 00:23:10 +0000614 llvm::Type *Ty = llvm::PointerType::getUnqual(ConvertType(E->getType()));
615 return llvm::UndefValue::get(Ty);
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000616 }
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000617};
618
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000619} // end anonymous namespace.
620
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000621llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
Chris Lattner96196622008-07-26 22:37:01 +0000622 CodeGenFunction *CGF) {
623 QualType type = Context.getCanonicalType(E->getType());
Eli Friedman7dfa6392008-06-01 15:31:44 +0000624
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000625 Expr::EvalResult Result;
626
627 if (E->Evaluate(Result, Context)) {
628 assert(!Result.HasSideEffects &&
629 "Constant expr should not have any side effects!");
630 switch (Result.Val.getKind()) {
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000631 default: assert(0 && "unhandled value kind!");
632 case APValue::LValue: {
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000633 llvm::Constant *Offset =
634 llvm::ConstantInt::get(llvm::Type::Int64Ty,
635 Result.Val.getLValueOffset());
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000636
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000637 if (const Expr *LVBase = Result.Val.getLValueBase()) {
Anders Carlssonaa432562008-11-17 01:58:55 +0000638 llvm::Constant *C =
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000639 ConstExprEmitter(*this, CGF).EmitLValue(const_cast<Expr*>(LVBase));
640
Anders Carlssonaa432562008-11-17 01:58:55 +0000641 const llvm::Type *Type =
642 llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
Eli Friedman7b30b1c2008-11-17 03:57:28 +0000643 const llvm::Type *DestType = getTypes().ConvertTypeForMem(E->getType());
Anders Carlssonaa432562008-11-17 01:58:55 +0000644
645 // FIXME: It's a little ugly that we need to cast to a pointer,
646 // apply the GEP and then cast back.
647 C = llvm::ConstantExpr::getBitCast(C, Type);
648 C = llvm::ConstantExpr::getGetElementPtr(C, &Offset, 1);
649
650 return llvm::ConstantExpr::getBitCast(C, DestType);
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000651 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000652
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000653 return llvm::ConstantExpr::getIntToPtr(Offset,
654 getTypes().ConvertType(type));
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000655 }
Eli Friedman7b30b1c2008-11-17 03:57:28 +0000656 case APValue::Int: {
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000657 llvm::Constant *C = llvm::ConstantInt::get(Result.Val.getInt());
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000658
659 if (C->getType() == llvm::Type::Int1Ty) {
660 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
661 C = llvm::ConstantExpr::getZExt(C, BoolTy);
662 }
663 return C;
Eli Friedman7b30b1c2008-11-17 03:57:28 +0000664 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000665 case APValue::Float:
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000666 return llvm::ConstantFP::get(Result.Val.getFloat());
Anders Carlssonaa432562008-11-17 01:58:55 +0000667 case APValue::ComplexFloat: {
668 llvm::Constant *Complex[2];
669
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000670 Complex[0] = llvm::ConstantFP::get(Result.Val.getComplexFloatReal());
671 Complex[1] = llvm::ConstantFP::get(Result.Val.getComplexFloatImag());
Anders Carlssonaa432562008-11-17 01:58:55 +0000672
673 return llvm::ConstantStruct::get(Complex, 2);
674 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000675 }
676 }
Eli Friedman7dfa6392008-06-01 15:31:44 +0000677
678 llvm::Constant* C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
679 if (C->getType() == llvm::Type::Int1Ty) {
680 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
681 C = llvm::ConstantExpr::getZExt(C, BoolTy);
682 }
683 return C;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000684}