blob: 651a283baa898135b4512932f38d6a2cef76c7c3 [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 }
Anders Carlssonbcf62252008-11-22 02:34:39 +000051
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000052 llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
53 return Visit(E->getInitializer());
54 }
55
Argyrios Kyrtzidis0835a3c2008-08-18 23:01:59 +000056 llvm::Constant *VisitCastExpr(CastExpr* E) {
Nuno Lopes81e51e22009-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 Carlsson3b1d57b2008-01-26 01:36:00 +000063 llvm::Constant *C = Visit(E->getSubExpr());
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000064 return EmitConversion(C, E->getSubExpr()->getType(), E->getType());
65 }
Devang Pateleae15602008-02-05 02:39:50 +000066
Chris Lattner04421082008-04-08 04:40:51 +000067 llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
68 return Visit(DAE->getExpr());
69 }
70
Eli Friedmana8234002008-05-30 10:24:46 +000071 llvm::Constant *EmitArrayInitialization(InitListExpr *ILE) {
Eli Friedman4d1c0182008-02-21 17:45:41 +000072 std::vector<llvm::Constant*> Elts;
Eli Friedmana8234002008-05-30 10:24:46 +000073 const llvm::ArrayType *AType =
74 cast<llvm::ArrayType>(ConvertType(ILE->getType()));
Eli Friedman4d1c0182008-02-21 17:45:41 +000075 unsigned NumInitElements = ILE->getNumInits();
76 // FIXME: Check for wide strings
77 if (NumInitElements > 0 && isa<StringLiteral>(ILE->getInit(0)) &&
Chris Lattnerc63a1f22008-08-04 07:31:14 +000078 ILE->getType()->getArrayElementTypeNoTypeQual()->isCharType())
Eli Friedman4d1c0182008-02-21 17:45:41 +000079 return Visit(ILE->getInit(0));
Devang Pateleae15602008-02-05 02:39:50 +000080 const llvm::Type *ElemTy = AType->getElementType();
81 unsigned NumElements = AType->getNumElements();
82
83 // Initialising an array requires us to automatically
84 // initialise any elements that have not been initialised explicitly
85 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
86
87 // Copy initializer elements.
88 unsigned i = 0;
Eli Friedmana04a1532008-05-30 19:58:50 +000089 bool RewriteType = false;
Devang Pateleae15602008-02-05 02:39:50 +000090 for (; i < NumInitableElts; ++i) {
Anders Carlssonaa432562008-11-17 01:58:55 +000091 llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(i), CGF);
Daniel Dunbard60f2fb2009-02-17 18:43:32 +000092 if (!C)
93 return 0;
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);
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000118 if (!C)
119 return;
120
Eli Friedmana8234002008-05-30 10:24:46 +0000121 llvm::ConstantInt *CI = dyn_cast<llvm::ConstantInt>(C);
122 if (!CI) {
Daniel Dunbar488e9932008-08-16 00:56:44 +0000123 CGM.ErrorUnsupported(E, "bitfield initialization");
Eli Friedmana8234002008-05-30 10:24:46 +0000124 return;
125 }
126 llvm::APInt V = CI->getValue();
Devang Pateleae15602008-02-05 02:39:50 +0000127
Eli Friedmana8234002008-05-30 10:24:46 +0000128 // Calculate information about the relevant field
129 const llvm::Type* Ty = CI->getType();
Daniel Dunbar9ca8bb02008-08-10 07:00:24 +0000130 const llvm::TargetData &TD = CGM.getTypes().getTargetData();
Eli Friedman7f5221b2009-02-01 08:12:19 +0000131 unsigned size = TD.getTypePaddedSizeInBits(Ty);
Eli Friedmana8234002008-05-30 10:24:46 +0000132 unsigned fieldOffset = CGM.getTypes().getLLVMFieldNo(Field) * size;
133 CodeGenTypes::BitFieldInfo bitFieldInfo =
134 CGM.getTypes().getBitFieldInfo(Field);
135 fieldOffset += bitFieldInfo.Begin;
136
137 // Find where to start the insertion
138 // FIXME: This is O(n^2) in the number of bit-fields!
139 // FIXME: This won't work if the struct isn't completely packed!
140 unsigned offset = 0, i = 0;
141 while (offset < (fieldOffset & -8))
Eli Friedman7f5221b2009-02-01 08:12:19 +0000142 offset += TD.getTypePaddedSizeInBits(Elts[i++]->getType());
Daniel Dunbar9ca8bb02008-08-10 07:00:24 +0000143
144 // Advance over 0 sized elements (must terminate in bounds since
145 // the bitfield must have a size).
Eli Friedman7f5221b2009-02-01 08:12:19 +0000146 while (TD.getTypePaddedSizeInBits(Elts[i]->getType()) == 0)
Daniel Dunbar9ca8bb02008-08-10 07:00:24 +0000147 ++i;
Eli Friedmana8234002008-05-30 10:24:46 +0000148
Daniel Dunbar1f3846e2008-07-17 21:07:48 +0000149 // Promote the size of V if necessary
150 // FIXME: This should never occur, but currently it can because
151 // initializer constants are cast to bool, and because clang is
152 // not enforcing bitfield width limits.
153 if (bitFieldInfo.Size > V.getBitWidth())
154 V.zext(bitFieldInfo.Size);
155
Eli Friedmana8234002008-05-30 10:24:46 +0000156 // Insert the bits into the struct
157 // FIXME: This algorthm is only correct on X86!
158 // FIXME: THis algorthm assumes bit-fields only have byte-size elements!
159 unsigned bitsToInsert = bitFieldInfo.Size;
160 unsigned curBits = std::min(8 - (fieldOffset & 7), bitsToInsert);
161 unsigned byte = V.getLoBits(curBits).getZExtValue() << (fieldOffset & 7);
162 do {
163 llvm::Constant* byteC = llvm::ConstantInt::get(llvm::Type::Int8Ty, byte);
164 Elts[i] = llvm::ConstantExpr::getOr(Elts[i], byteC);
165 ++i;
166 V = V.lshr(curBits);
167 bitsToInsert -= curBits;
168
169 if (!bitsToInsert)
170 break;
171
172 curBits = bitsToInsert > 8 ? 8 : bitsToInsert;
173 byte = V.getLoBits(curBits).getZExtValue();
174 } while (true);
175 }
176
177 llvm::Constant *EmitStructInitialization(InitListExpr *ILE) {
178 const llvm::StructType *SType =
179 cast<llvm::StructType>(ConvertType(ILE->getType()));
Eli Friedman017cbdf2008-05-20 00:11:07 +0000180 RecordDecl *RD = ILE->getType()->getAsRecordType()->getDecl();
Devang Pateleae15602008-02-05 02:39:50 +0000181 std::vector<llvm::Constant*> Elts;
Devang Pateleae15602008-02-05 02:39:50 +0000182
Eli Friedman017cbdf2008-05-20 00:11:07 +0000183 // Initialize the whole structure to zero.
184 for (unsigned i = 0; i < SType->getNumElements(); ++i) {
Devang Pateleae15602008-02-05 02:39:50 +0000185 const llvm::Type *FieldTy = SType->getElementType(i);
186 Elts.push_back(llvm::Constant::getNullValue(FieldTy));
187 }
Eli Friedman017cbdf2008-05-20 00:11:07 +0000188
189 // Copy initializer elements. Skip padding fields.
190 unsigned EltNo = 0; // Element no in ILE
191 int FieldNo = 0; // Field no in RecordDecl
Eli Friedmana04a1532008-05-30 19:58:50 +0000192 bool RewriteType = false;
Douglas Gregor44b43212008-12-11 16:49:14 +0000193 for (RecordDecl::field_iterator Field = RD->field_begin(),
194 FieldEnd = RD->field_end();
195 EltNo < ILE->getNumInits() && Field != FieldEnd; ++Field) {
Eli Friedman017cbdf2008-05-20 00:11:07 +0000196 FieldNo++;
Douglas Gregor44b43212008-12-11 16:49:14 +0000197 if (!Field->getIdentifier())
Eli Friedman017cbdf2008-05-20 00:11:07 +0000198 continue;
199
Douglas Gregor44b43212008-12-11 16:49:14 +0000200 if (Field->isBitField()) {
201 InsertBitfieldIntoStruct(Elts, *Field, ILE->getInit(EltNo));
Eli Friedman017cbdf2008-05-20 00:11:07 +0000202 } else {
Douglas Gregor44b43212008-12-11 16:49:14 +0000203 unsigned FieldNo = CGM.getTypes().getLLVMFieldNo(*Field);
Anders Carlssonaa432562008-11-17 01:58:55 +0000204 llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(EltNo), CGF);
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000205 if (!C) return 0;
Eli Friedmana04a1532008-05-30 19:58:50 +0000206 RewriteType |= (C->getType() != Elts[FieldNo]->getType());
207 Elts[FieldNo] = C;
Eli Friedman017cbdf2008-05-20 00:11:07 +0000208 }
209 EltNo++;
210 }
211
Eli Friedmana04a1532008-05-30 19:58:50 +0000212 if (RewriteType) {
213 // FIXME: Make this work for non-packed structs
214 assert(SType->isPacked() && "Cannot recreate unpacked structs");
215 std::vector<const llvm::Type*> Types;
216 for (unsigned i = 0; i < Elts.size(); ++i)
217 Types.push_back(Elts[i]->getType());
218 SType = llvm::StructType::get(Types, true);
219 }
220
Devang Pateleae15602008-02-05 02:39:50 +0000221 return llvm::ConstantStruct::get(SType, Elts);
222 }
223
Nuno Lopes81e51e22009-01-17 00:48:48 +0000224 llvm::Constant *EmitUnion(llvm::Constant *C, const llvm::Type *Ty) {
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000225 if (!C)
226 return 0;
227
Nuno Lopes81e51e22009-01-17 00:48:48 +0000228 // Build a struct with the union sub-element as the first member,
229 // and padded to the appropriate size
230 std::vector<llvm::Constant*> Elts;
231 std::vector<const llvm::Type*> Types;
232 Elts.push_back(C);
233 Types.push_back(C->getType());
Eli Friedman7f5221b2009-02-01 08:12:19 +0000234 unsigned CurSize = CGM.getTargetData().getTypePaddedSize(C->getType());
235 unsigned TotalSize = CGM.getTargetData().getTypePaddedSize(Ty);
Nuno Lopes81e51e22009-01-17 00:48:48 +0000236 while (CurSize < TotalSize) {
237 Elts.push_back(llvm::Constant::getNullValue(llvm::Type::Int8Ty));
238 Types.push_back(llvm::Type::Int8Ty);
239 CurSize++;
240 }
241
242 // This always generates a packed struct
243 // FIXME: Try to generate an unpacked struct when we can
244 llvm::StructType* STy = llvm::StructType::get(Types, true);
245 return llvm::ConstantStruct::get(STy, Elts);
246 }
247
Eli Friedmana8234002008-05-30 10:24:46 +0000248 llvm::Constant *EmitUnionInitialization(InitListExpr *ILE) {
Eli Friedmana8234002008-05-30 10:24:46 +0000249 const llvm::Type *Ty = ConvertType(ILE->getType());
250
Douglas Gregor0bb76892009-01-29 16:53:55 +0000251 // If this is an empty initializer list, we value-initialize the
252 // union.
253 if (ILE->getNumInits() == 0)
254 return llvm::Constant::getNullValue(Ty);
Douglas Gregor34e79462009-01-28 23:36:17 +0000255
Douglas Gregor0bb76892009-01-29 16:53:55 +0000256 FieldDecl* curField = ILE->getInitializedFieldInUnion();
257 if (!curField) {
Douglas Gregor0bb76892009-01-29 16:53:55 +0000258 // There's no field to initialize, so value-initialize the union.
259#ifndef NDEBUG
260 // Make sure that it's really an empty and not a failure of
261 // semantic analysis.
Douglas Gregor71d81742009-02-11 16:17:49 +0000262 RecordDecl *RD = ILE->getType()->getAsRecordType()->getDecl();
Douglas Gregor0bb76892009-01-29 16:53:55 +0000263 for (RecordDecl::field_iterator Field = RD->field_begin(),
264 FieldEnd = RD->field_end();
265 Field != FieldEnd; ++Field)
266 assert(Field->isUnnamedBitfield() && "Only unnamed bitfields allowed");
267#endif
Eli Friedmana8234002008-05-30 10:24:46 +0000268 return llvm::Constant::getNullValue(Ty);
Douglas Gregor0bb76892009-01-29 16:53:55 +0000269 }
Eli Friedmana8234002008-05-30 10:24:46 +0000270
271 if (curField->isBitField()) {
272 // Create a dummy struct for bit-field insertion
Daniel Dunbar491c7b72009-01-12 21:08:18 +0000273 unsigned NumElts = CGM.getTargetData().getTypePaddedSize(Ty) / 8;
Eli Friedmana8234002008-05-30 10:24:46 +0000274 llvm::Constant* NV = llvm::Constant::getNullValue(llvm::Type::Int8Ty);
275 std::vector<llvm::Constant*> Elts(NumElts, NV);
276
277 InsertBitfieldIntoStruct(Elts, curField, ILE->getInit(0));
278 const llvm::ArrayType *RetTy =
279 llvm::ArrayType::get(NV->getType(), NumElts);
280 return llvm::ConstantArray::get(RetTy, Elts);
281 }
282
Nuno Lopes81e51e22009-01-17 00:48:48 +0000283 return EmitUnion(CGM.EmitConstantExpr(ILE->getInit(0), CGF), Ty);
Eli Friedmana8234002008-05-30 10:24:46 +0000284 }
285
286 llvm::Constant *EmitVectorInitialization(InitListExpr *ILE) {
287 const llvm::VectorType *VType =
288 cast<llvm::VectorType>(ConvertType(ILE->getType()));
Eli Friedmancfb313b2008-05-30 12:02:14 +0000289 const llvm::Type *ElemTy = VType->getElementType();
290 std::vector<llvm::Constant*> Elts;
Devang Pateleae15602008-02-05 02:39:50 +0000291 unsigned NumElements = VType->getNumElements();
Eli Friedmancfb313b2008-05-30 12:02:14 +0000292 unsigned NumInitElements = ILE->getNumInits();
Devang Pateleae15602008-02-05 02:39:50 +0000293
Eli Friedmancfb313b2008-05-30 12:02:14 +0000294 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
295
Devang Pateleae15602008-02-05 02:39:50 +0000296 // Copy initializer elements.
297 unsigned i = 0;
Eli Friedmancfb313b2008-05-30 12:02:14 +0000298 for (; i < NumInitableElts; ++i) {
Anders Carlssonaa432562008-11-17 01:58:55 +0000299 llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(i), CGF);
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000300 if (!C)
301 return 0;
Devang Pateleae15602008-02-05 02:39:50 +0000302 Elts.push_back(C);
303 }
304
Eli Friedmancfb313b2008-05-30 12:02:14 +0000305 for (; i < NumElements; ++i)
306 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
307
Devang Pateleae15602008-02-05 02:39:50 +0000308 return llvm::ConstantVector::get(VType, Elts);
309 }
Anders Carlsson7d801ea2009-01-30 06:13:25 +0000310
311 llvm::Constant *VisitImplicitValueInitExpr(ImplicitValueInitExpr* E) {
312 const llvm::Type* RetTy = CGM.getTypes().ConvertType(E->getType());
313 return llvm::Constant::getNullValue(RetTy);
314 }
315
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000316 llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
Eli Friedmana8234002008-05-30 10:24:46 +0000317 if (ILE->getType()->isScalarType()) {
318 // We have a scalar in braces. Just use the first element.
319 if (ILE->getNumInits() > 0)
Anders Carlssonaa432562008-11-17 01:58:55 +0000320 return CGM.EmitConstantExpr(ILE->getInit(0), CGF);
Eli Friedmana8234002008-05-30 10:24:46 +0000321
322 const llvm::Type* RetTy = CGM.getTypes().ConvertType(ILE->getType());
323 return llvm::Constant::getNullValue(RetTy);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000324 }
Chris Lattnerbe20bb52008-10-26 23:53:12 +0000325
Eli Friedmana8234002008-05-30 10:24:46 +0000326 if (ILE->getType()->isArrayType())
327 return EmitArrayInitialization(ILE);
Devang Patel47fb6972008-01-29 23:23:18 +0000328
Eli Friedmana8234002008-05-30 10:24:46 +0000329 if (ILE->getType()->isStructureType())
330 return EmitStructInitialization(ILE);
331
332 if (ILE->getType()->isUnionType())
333 return EmitUnionInitialization(ILE);
334
335 if (ILE->getType()->isVectorType())
336 return EmitVectorInitialization(ILE);
337
Devang Pateleae15602008-02-05 02:39:50 +0000338 assert(0 && "Unable to handle InitListExpr");
Chris Lattner2b94fe32008-03-01 08:45:05 +0000339 // Get rid of control reaches end of void function warning.
340 // Not reached.
341 return 0;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000342 }
Eli Friedman3b707e72008-02-21 17:57:49 +0000343
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000344 llvm::Constant *VisitImplicitCastExpr(ImplicitCastExpr *ICExpr) {
Eli Friedman3b707e72008-02-21 17:57:49 +0000345 Expr* SExpr = ICExpr->getSubExpr();
346 QualType SType = SExpr->getType();
347 llvm::Constant *C; // the intermediate expression
348 QualType T; // the type of the intermediate expression
349 if (SType->isArrayType()) {
350 // Arrays decay to a pointer to the first element
351 // VLAs would require special handling, but they can't occur here
352 C = EmitLValue(SExpr);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000353 llvm::Constant *Idx0 = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000354 llvm::Constant *Ops[] = {Idx0, Idx0};
355 C = llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
Chris Lattnerc63a1f22008-08-04 07:31:14 +0000356 T = CGM.getContext().getArrayDecayedType(SType);
Eli Friedman3b707e72008-02-21 17:57:49 +0000357 } else if (SType->isFunctionType()) {
358 // Function types decay to a pointer to the function
359 C = EmitLValue(SExpr);
360 T = CGM.getContext().getPointerType(SType);
361 } else {
362 C = Visit(SExpr);
363 T = SType;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000364 }
Eli Friedman3b707e72008-02-21 17:57:49 +0000365
366 // Perform the conversion; note that an implicit cast can both promote
367 // and convert an array/function
368 return EmitConversion(C, T, ICExpr->getType());
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000369 }
Eli Friedman3b707e72008-02-21 17:57:49 +0000370
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000371 llvm::Constant *VisitStringLiteral(StringLiteral *E) {
Chris Lattnera7ad98f2008-02-11 00:02:17 +0000372 assert(!E->getType()->isPointerType() && "Strings are always arrays");
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000373
374 // Otherwise this must be a string initializing an array in a static
375 // initializer. Don't emit it as the address of the string, emit the string
376 // data itself as an inline array.
Daniel Dunbar61432932008-08-13 23:20:05 +0000377 return llvm::ConstantArray::get(CGM.GetStringForStringLiteral(E), false);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000378 }
379
Eli Friedmanbc5ed6e2008-05-29 11:22:45 +0000380 llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
381 return Visit(E->getSubExpr());
382 }
Anders Carlsson7be3d012008-02-05 20:54:21 +0000383
Anders Carlssond5cab542009-02-12 17:55:02 +0000384 llvm::Constant *VisitBlockExpr(const BlockExpr *E) {
Mike Stump67a64482009-02-14 22:16:35 +0000385 const char *Name = "";
386 if (const NamedDecl *ND = dyn_cast<NamedDecl>(CGF->CurFuncDecl))
387 Name = ND->getNameAsString().c_str();
388 return CGM.GetAddrOfGlobalBlock(E, Name);
Anders Carlssond5cab542009-02-12 17:55:02 +0000389 }
390
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000391 // Utility methods
392 const llvm::Type *ConvertType(QualType T) {
393 return CGM.getTypes().ConvertType(T);
394 }
395
396 llvm::Constant *EmitConversionToBool(llvm::Constant *Src, QualType SrcType) {
397 assert(SrcType->isCanonical() && "EmitConversion strips typedefs");
398
399 if (SrcType->isRealFloatingType()) {
400 // Compare against 0.0 for fp scalars.
401 llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
402 return llvm::ConstantExpr::getFCmp(llvm::FCmpInst::FCMP_UNE, Src, Zero);
403 }
404
405 assert((SrcType->isIntegerType() || SrcType->isPointerType()) &&
406 "Unknown scalar type to convert");
407
408 // Compare against an integer or pointer null.
409 llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
410 return llvm::ConstantExpr::getICmp(llvm::ICmpInst::ICMP_NE, Src, Zero);
411 }
412
413 llvm::Constant *EmitConversion(llvm::Constant *Src, QualType SrcType,
414 QualType DstType) {
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000415 if (!Src)
416 return 0;
417
Chris Lattnerd394cc62008-07-26 22:39:33 +0000418 SrcType = CGM.getContext().getCanonicalType(SrcType);
419 DstType = CGM.getContext().getCanonicalType(DstType);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000420 if (SrcType == DstType) return Src;
421
422 // Handle conversions to bool first, they are special: comparisons against 0.
423 if (DstType->isBooleanType())
424 return EmitConversionToBool(Src, SrcType);
425
426 const llvm::Type *DstTy = ConvertType(DstType);
427
428 // Ignore conversions like int -> uint.
429 if (Src->getType() == DstTy)
430 return Src;
431
432 // Handle pointer conversions next: pointers can only be converted to/from
433 // other pointers and integers.
Eli Friedmanc53296e2008-06-07 17:27:51 +0000434 if (isa<llvm::PointerType>(DstTy)) {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000435 // The source value may be an integer, or a pointer.
436 if (isa<llvm::PointerType>(Src->getType()))
437 return llvm::ConstantExpr::getBitCast(Src, DstTy);
Anders Carlsson84005b42008-01-26 04:30:23 +0000438 assert(SrcType->isIntegerType() &&"Not ptr->ptr or int->ptr conversion?");
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000439 return llvm::ConstantExpr::getIntToPtr(Src, DstTy);
440 }
441
Eli Friedmanc53296e2008-06-07 17:27:51 +0000442 if (isa<llvm::PointerType>(Src->getType())) {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000443 // Must be an ptr to int cast.
444 assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
445 return llvm::ConstantExpr::getPtrToInt(Src, DstTy);
446 }
447
448 // A scalar source can be splatted to a vector of the same element type
449 if (isa<llvm::VectorType>(DstTy) && !isa<VectorType>(SrcType)) {
Chris Lattner3304e552008-10-12 00:31:50 +0000450 assert((cast<llvm::VectorType>(DstTy)->getElementType()
451 == Src->getType()) &&
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000452 "Vector element type must match scalar type to splat.");
453 unsigned NumElements = DstType->getAsVectorType()->getNumElements();
454 llvm::SmallVector<llvm::Constant*, 16> Elements;
455 for (unsigned i = 0; i < NumElements; i++)
456 Elements.push_back(Src);
457
458 return llvm::ConstantVector::get(&Elements[0], NumElements);
459 }
460
461 if (isa<llvm::VectorType>(Src->getType()) ||
462 isa<llvm::VectorType>(DstTy)) {
463 return llvm::ConstantExpr::getBitCast(Src, DstTy);
464 }
465
466 // Finally, we have the arithmetic types: real int/float.
467 if (isa<llvm::IntegerType>(Src->getType())) {
468 bool InputSigned = SrcType->isSignedIntegerType();
469 if (isa<llvm::IntegerType>(DstTy))
470 return llvm::ConstantExpr::getIntegerCast(Src, DstTy, InputSigned);
471 else if (InputSigned)
472 return llvm::ConstantExpr::getSIToFP(Src, DstTy);
473 else
474 return llvm::ConstantExpr::getUIToFP(Src, DstTy);
475 }
476
477 assert(Src->getType()->isFloatingPoint() && "Unknown real conversion");
478 if (isa<llvm::IntegerType>(DstTy)) {
479 if (DstType->isSignedIntegerType())
480 return llvm::ConstantExpr::getFPToSI(Src, DstTy);
481 else
482 return llvm::ConstantExpr::getFPToUI(Src, DstTy);
483 }
484
485 assert(DstTy->isFloatingPoint() && "Unknown real conversion");
486 if (DstTy->getTypeID() < Src->getType()->getTypeID())
487 return llvm::ConstantExpr::getFPTrunc(Src, DstTy);
488 else
489 return llvm::ConstantExpr::getFPExtend(Src, DstTy);
490 }
Anders Carlsson84005b42008-01-26 04:30:23 +0000491
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000492public:
Anders Carlsson84005b42008-01-26 04:30:23 +0000493 llvm::Constant *EmitLValue(Expr *E) {
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000494 switch (E->getStmtClass()) {
Eli Friedman546d94c2008-02-11 00:23:10 +0000495 default: break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000496 case Expr::ParenExprClass:
497 // Elide parenthesis
498 return EmitLValue(cast<ParenExpr>(E)->getSubExpr());
499 case Expr::CompoundLiteralExprClass: {
500 // Note that due to the nature of compound literals, this is guaranteed
501 // to be the only use of the variable, so we just generate it here.
Anders Carlsson84005b42008-01-26 04:30:23 +0000502 CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
503 llvm::Constant* C = Visit(CLE->getInitializer());
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000504 // FIXME: "Leaked" on failure.
505 if (C)
506 C = new llvm::GlobalVariable(C->getType(),
507 E->getType().isConstQualified(),
508 llvm::GlobalValue::InternalLinkage,
509 C, ".compoundliteral", &CGM.getModule());
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000510 return C;
Anders Carlsson84005b42008-01-26 04:30:23 +0000511 }
Douglas Gregor1a49af92009-01-06 05:10:23 +0000512 case Expr::DeclRefExprClass:
513 case Expr::QualifiedDeclRefExprClass: {
Douglas Gregor8e9bebd2008-10-21 16:13:35 +0000514 NamedDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000515 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000516 return CGM.GetAddrOfFunction(FD);
Steve Naroff248a7532008-04-15 22:42:06 +0000517 if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
518 if (VD->isFileVarDecl())
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000519 return CGM.GetAddrOfGlobalVar(VD);
Steve Naroff248a7532008-04-15 22:42:06 +0000520 else if (VD->isBlockVarDecl()) {
521 assert(CGF && "Can't access static local vars without CGF");
522 return CGF->GetAddrOfStaticLocalVar(VD);
523 }
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000524 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000525 break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000526 }
527 case Expr::MemberExprClass: {
Anders Carlsson84005b42008-01-26 04:30:23 +0000528 MemberExpr* ME = cast<MemberExpr>(E);
Anders Carlsson84005b42008-01-26 04:30:23 +0000529 llvm::Constant *Base;
530 if (ME->isArrow())
531 Base = Visit(ME->getBase());
532 else
533 Base = EmitLValue(ME->getBase());
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000534 if (!Base)
535 return 0;
Lauro Ramos Venancio5bfe18c2008-01-30 21:23:20 +0000536
Douglas Gregor86f19402008-12-20 23:49:58 +0000537 FieldDecl *Field = dyn_cast<FieldDecl>(ME->getMemberDecl());
538 // FIXME: Handle other kinds of member expressions.
539 assert(Field && "No code generation for non-field member expressions");
540 unsigned FieldNumber = CGM.getTypes().getLLVMFieldNo(Field);
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000541 llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
542 llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
543 FieldNumber);
544 llvm::Value *Ops[] = {Zero, Idx};
545 return llvm::ConstantExpr::getGetElementPtr(Base, Ops, 2);
546 }
547 case Expr::ArraySubscriptExprClass: {
Anders Carlsson84005b42008-01-26 04:30:23 +0000548 ArraySubscriptExpr* ASExpr = cast<ArraySubscriptExpr>(E);
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000549 assert(!ASExpr->getBase()->getType()->isVectorType() &&
550 "Taking the address of a vector component is illegal!");
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000551
552 llvm::Constant *Base = Visit(ASExpr->getBase());
553 llvm::Constant *Index = Visit(ASExpr->getIdx());
554 if (!Base || !Index)
555 return 0;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000556 return llvm::ConstantExpr::getGetElementPtr(Base, &Index, 1);
557 }
Daniel Dunbar61432932008-08-13 23:20:05 +0000558 case Expr::StringLiteralClass:
559 return CGM.GetAddrOfConstantStringFromLiteral(cast<StringLiteral>(E));
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000560 case Expr::ObjCStringLiteralClass: {
561 ObjCStringLiteral* SL = cast<ObjCStringLiteral>(E);
562 std::string S(SL->getString()->getStrData(),
563 SL->getString()->getByteLength());
564 llvm::Constant *C = CGM.getObjCRuntime().GenerateConstantString(S);
565 return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
566 }
Anders Carlsson84005b42008-01-26 04:30:23 +0000567 case Expr::UnaryOperatorClass: {
568 UnaryOperator *Exp = cast<UnaryOperator>(E);
569 switch (Exp->getOpcode()) {
Eli Friedman546d94c2008-02-11 00:23:10 +0000570 default: break;
571 case UnaryOperator::Extension:
Anders Carlsson84005b42008-01-26 04:30:23 +0000572 // Extension is just a wrapper for expressions
573 return EmitLValue(Exp->getSubExpr());
574 case UnaryOperator::Real:
575 case UnaryOperator::Imag: {
576 // The address of __real or __imag is just a GEP off the address
577 // of the internal expression
578 llvm::Constant* C = EmitLValue(Exp->getSubExpr());
579 llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
580 llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000581 Exp->getOpcode() == UnaryOperator::Imag);
Anders Carlsson84005b42008-01-26 04:30:23 +0000582 llvm::Value *Ops[] = {Zero, Idx};
583 return llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
584 }
585 case UnaryOperator::Deref:
586 // The address of a deref is just the value of the expression
587 return Visit(Exp->getSubExpr());
588 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000589 break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000590 }
Chris Lattner35345642008-12-12 05:18:02 +0000591
592 case Expr::PredefinedExprClass: {
593 // __func__/__FUNCTION__ -> "". __PRETTY_FUNCTION__ -> "top level".
594 std::string Str;
595 if (cast<PredefinedExpr>(E)->getIdentType() ==
596 PredefinedExpr::PrettyFunction)
597 Str = "top level";
598
599 return CGM.GetAddrOfConstantCString(Str, ".tmp");
600 }
Eli Friedmanf0115892009-01-25 01:21:06 +0000601 case Expr::AddrLabelExprClass: {
602 assert(CGF && "Invalid address of label expression outside function.");
603 unsigned id = CGF->GetIDForAddrOfLabel(cast<AddrLabelExpr>(E)->getLabel());
604 llvm::Constant *C = llvm::ConstantInt::get(llvm::Type::Int32Ty, id);
605 return llvm::ConstantExpr::getIntToPtr(C, ConvertType(E->getType()));
606 }
Eli Friedman3941b182009-01-25 01:54:01 +0000607 case Expr::CallExprClass: {
608 CallExpr* CE = cast<CallExpr>(E);
Douglas Gregor3c385e52009-02-14 18:57:46 +0000609 if (CE->isBuiltinCall(CGM.getContext()) !=
610 Builtin::BI__builtin___CFStringMakeConstantString)
Eli Friedman3941b182009-01-25 01:54:01 +0000611 break;
612 const Expr *Arg = CE->getArg(0)->IgnoreParenCasts();
613 const StringLiteral *Literal = cast<StringLiteral>(Arg);
614 std::string S(Literal->getStrData(), Literal->getByteLength());
615 return CGM.GetAddrOfConstantCFString(S);
616 }
Mike Stumpbd65cac2009-02-19 01:01:04 +0000617 case Expr::BlockExprClass: {
618 return CGF->BuildBlockLiteralTmp(cast<BlockExpr>(E));
619 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000620 }
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000621
622 return 0;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000623 }
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000624};
625
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000626} // end anonymous namespace.
627
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000628llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
Chris Lattner96196622008-07-26 22:37:01 +0000629 CodeGenFunction *CGF) {
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000630 Expr::EvalResult Result;
631
632 if (E->Evaluate(Result, Context)) {
633 assert(!Result.HasSideEffects &&
634 "Constant expr should not have any side effects!");
635 switch (Result.Val.getKind()) {
Daniel Dunbara5fd07b2009-01-28 22:24:07 +0000636 case APValue::Uninitialized:
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000637 assert(0 && "Constant expressions should be initialized.");
638 return 0;
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000639 case APValue::LValue: {
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000640 llvm::Constant *Offset =
641 llvm::ConstantInt::get(llvm::Type::Int64Ty,
642 Result.Val.getLValueOffset());
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000643
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000644 if (const Expr *LVBase = Result.Val.getLValueBase()) {
Anders Carlssonaa432562008-11-17 01:58:55 +0000645 llvm::Constant *C =
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000646 ConstExprEmitter(*this, CGF).EmitLValue(const_cast<Expr*>(LVBase));
647
Anders Carlssonaa432562008-11-17 01:58:55 +0000648 const llvm::Type *Type =
649 llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
Eli Friedman7b30b1c2008-11-17 03:57:28 +0000650 const llvm::Type *DestType = getTypes().ConvertTypeForMem(E->getType());
Anders Carlssonaa432562008-11-17 01:58:55 +0000651
652 // FIXME: It's a little ugly that we need to cast to a pointer,
653 // apply the GEP and then cast back.
654 C = llvm::ConstantExpr::getBitCast(C, Type);
655 C = llvm::ConstantExpr::getGetElementPtr(C, &Offset, 1);
656
657 return llvm::ConstantExpr::getBitCast(C, DestType);
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000658 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000659
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000660 return llvm::ConstantExpr::getIntToPtr(Offset,
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000661 getTypes().ConvertType(E->getType()));
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000662 }
Eli Friedman7b30b1c2008-11-17 03:57:28 +0000663 case APValue::Int: {
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000664 llvm::Constant *C = llvm::ConstantInt::get(Result.Val.getInt());
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000665
666 if (C->getType() == llvm::Type::Int1Ty) {
667 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
668 C = llvm::ConstantExpr::getZExt(C, BoolTy);
669 }
670 return C;
Eli Friedman7b30b1c2008-11-17 03:57:28 +0000671 }
Daniel Dunbara5fd07b2009-01-28 22:24:07 +0000672 case APValue::ComplexInt: {
673 llvm::Constant *Complex[2];
674
675 Complex[0] = llvm::ConstantInt::get(Result.Val.getComplexIntReal());
676 Complex[1] = llvm::ConstantInt::get(Result.Val.getComplexIntImag());
677
678 return llvm::ConstantStruct::get(Complex, 2);
679 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000680 case APValue::Float:
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000681 return llvm::ConstantFP::get(Result.Val.getFloat());
Anders Carlssonaa432562008-11-17 01:58:55 +0000682 case APValue::ComplexFloat: {
683 llvm::Constant *Complex[2];
684
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000685 Complex[0] = llvm::ConstantFP::get(Result.Val.getComplexFloatReal());
686 Complex[1] = llvm::ConstantFP::get(Result.Val.getComplexFloatImag());
Anders Carlssonaa432562008-11-17 01:58:55 +0000687
688 return llvm::ConstantStruct::get(Complex, 2);
689 }
Nate Begeman3d309f92009-01-18 01:01:34 +0000690 case APValue::Vector: {
691 llvm::SmallVector<llvm::Constant *, 4> Inits;
692 unsigned NumElts = Result.Val.getVectorLength();
693
694 for (unsigned i = 0; i != NumElts; ++i) {
695 APValue &Elt = Result.Val.getVectorElt(i);
696 if (Elt.isInt())
697 Inits.push_back(llvm::ConstantInt::get(Elt.getInt()));
698 else
699 Inits.push_back(llvm::ConstantFP::get(Elt.getFloat()));
700 }
701 return llvm::ConstantVector::get(&Inits[0], Inits.size());
702 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000703 }
704 }
Eli Friedman7dfa6392008-06-01 15:31:44 +0000705
706 llvm::Constant* C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000707 if (C && C->getType() == llvm::Type::Int1Ty) {
Eli Friedman7dfa6392008-06-01 15:31:44 +0000708 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
709 C = llvm::ConstantExpr::getZExt(C, BoolTy);
710 }
711 return C;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000712}