blob: fae0e369a972dc1e34d59e0b0c587f11c86434a8 [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"
Anders Carlsson4c98efd2009-07-24 15:20:52 +000019#include "clang/AST/RecordLayout.h"
Daniel Dunbarde7fb842008-08-11 05:00:27 +000020#include "clang/AST/StmtVisitor.h"
Chris Lattner1b63e4f2009-06-14 01:54:56 +000021#include "clang/Basic/Builtins.h"
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000022#include "llvm/Constants.h"
23#include "llvm/Function.h"
24#include "llvm/GlobalVariable.h"
25#include "llvm/Support/Compiler.h"
Eli Friedmana8234002008-05-30 10:24:46 +000026#include "llvm/Target/TargetData.h"
Anders Carlsson3b1d57b2008-01-26 01:36:00 +000027using namespace clang;
28using namespace CodeGen;
29
30namespace {
Anders Carlsson4c98efd2009-07-24 15:20:52 +000031
32class VISIBILITY_HIDDEN ConstStructBuilder {
33 CodeGenModule &CGM;
34 CodeGenFunction *CGF;
35
Mike Stump1eb44332009-09-09 15:08:12 +000036 bool Packed;
Anders Carlsson4257bc62009-07-25 00:24:56 +000037
Anders Carlsson4c98efd2009-07-24 15:20:52 +000038 unsigned NextFieldOffsetInBytes;
Mike Stump1eb44332009-09-09 15:08:12 +000039
Anders Carlsson31f2f9c2009-10-02 02:15:20 +000040 unsigned LLVMStructAlignment;
41
Anders Carlsson4c98efd2009-07-24 15:20:52 +000042 std::vector<llvm::Constant *> Elements;
43
44 ConstStructBuilder(CodeGenModule &CGM, CodeGenFunction *CGF)
Anders Carlsson31f2f9c2009-10-02 02:15:20 +000045 : CGM(CGM), CGF(CGF), Packed(false), NextFieldOffsetInBytes(0),
46 LLVMStructAlignment(1) { }
Anders Carlsson4c98efd2009-07-24 15:20:52 +000047
Mike Stump1eb44332009-09-09 15:08:12 +000048 bool AppendField(const FieldDecl *Field, uint64_t FieldOffset,
Anders Carlsson4c98efd2009-07-24 15:20:52 +000049 const Expr *InitExpr) {
50 uint64_t FieldOffsetInBytes = FieldOffset / 8;
Mike Stump1eb44332009-09-09 15:08:12 +000051
52 assert(NextFieldOffsetInBytes <= FieldOffsetInBytes
Anders Carlsson4c98efd2009-07-24 15:20:52 +000053 && "Field offset mismatch!");
Mike Stump1eb44332009-09-09 15:08:12 +000054
Anders Carlsson4c98efd2009-07-24 15:20:52 +000055 // Emit the field.
56 llvm::Constant *C = CGM.EmitConstantExpr(InitExpr, Field->getType(), CGF);
57 if (!C)
58 return false;
59
60 unsigned FieldAlignment = getAlignment(C);
Mike Stump1eb44332009-09-09 15:08:12 +000061
Anders Carlsson4c98efd2009-07-24 15:20:52 +000062 // Round up the field offset to the alignment of the field type.
Mike Stump1eb44332009-09-09 15:08:12 +000063 uint64_t AlignedNextFieldOffsetInBytes =
Anders Carlsson4c98efd2009-07-24 15:20:52 +000064 llvm::RoundUpToAlignment(NextFieldOffsetInBytes, FieldAlignment);
Mike Stump1eb44332009-09-09 15:08:12 +000065
Anders Carlsson4c98efd2009-07-24 15:20:52 +000066 if (AlignedNextFieldOffsetInBytes > FieldOffsetInBytes) {
Anders Carlssonbea9c522009-07-25 01:40:29 +000067 assert(!Packed && "Alignment is wrong even with a packed struct!");
Mike Stump1eb44332009-09-09 15:08:12 +000068
Anders Carlssonbea9c522009-07-25 01:40:29 +000069 // Convert the struct to a packed struct.
Anders Carlsson31f2f9c2009-10-02 02:15:20 +000070 ConvertStructToPacked();
71
Anders Carlssonbea9c522009-07-25 01:40:29 +000072 AlignedNextFieldOffsetInBytes = NextFieldOffsetInBytes;
Anders Carlsson4c98efd2009-07-24 15:20:52 +000073 }
74
75 if (AlignedNextFieldOffsetInBytes < FieldOffsetInBytes) {
76 // We need to append padding.
Anders Carlssonbea9c522009-07-25 01:40:29 +000077 AppendPadding(FieldOffsetInBytes - NextFieldOffsetInBytes);
Mike Stump1eb44332009-09-09 15:08:12 +000078
Anders Carlsson4c98efd2009-07-24 15:20:52 +000079 assert(NextFieldOffsetInBytes == FieldOffsetInBytes &&
80 "Did not add enough padding!");
Mike Stump1eb44332009-09-09 15:08:12 +000081
Anders Carlsson4c98efd2009-07-24 15:20:52 +000082 AlignedNextFieldOffsetInBytes = NextFieldOffsetInBytes;
83 }
Mike Stump1eb44332009-09-09 15:08:12 +000084
Anders Carlsson4c98efd2009-07-24 15:20:52 +000085 // Add the field.
86 Elements.push_back(C);
87 NextFieldOffsetInBytes = AlignedNextFieldOffsetInBytes + getSizeInBytes(C);
Anders Carlsson31f2f9c2009-10-02 02:15:20 +000088 LLVMStructAlignment = std::max(LLVMStructAlignment, FieldAlignment);
Anders Carlsson4c98efd2009-07-24 15:20:52 +000089 return true;
90 }
Mike Stump1eb44332009-09-09 15:08:12 +000091
92 bool AppendBitField(const FieldDecl *Field, uint64_t FieldOffset,
Anders Carlssond24393b2009-07-27 01:23:51 +000093 const Expr *InitExpr) {
Mike Stump1eb44332009-09-09 15:08:12 +000094 llvm::ConstantInt *CI =
95 cast_or_null<llvm::ConstantInt>(CGM.EmitConstantExpr(InitExpr,
96 Field->getType(),
Anders Carlssond24393b2009-07-27 01:23:51 +000097 CGF));
98 // FIXME: Can this ever happen?
99 if (!CI)
100 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000101
Anders Carlssond24393b2009-07-27 01:23:51 +0000102 if (FieldOffset > NextFieldOffsetInBytes * 8) {
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000103 // We need to add padding.
Mike Stump1eb44332009-09-09 15:08:12 +0000104 uint64_t NumBytes =
105 llvm::RoundUpToAlignment(FieldOffset -
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000106 NextFieldOffsetInBytes * 8, 8) / 8;
Mike Stump1eb44332009-09-09 15:08:12 +0000107
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000108 AppendPadding(NumBytes);
Anders Carlssond24393b2009-07-27 01:23:51 +0000109 }
110
Mike Stump1eb44332009-09-09 15:08:12 +0000111 uint64_t FieldSize =
Anders Carlssond24393b2009-07-27 01:23:51 +0000112 Field->getBitWidth()->EvaluateAsInt(CGM.getContext()).getZExtValue();
113
114 llvm::APInt FieldValue = CI->getValue();
Mike Stump1eb44332009-09-09 15:08:12 +0000115
Anders Carlssond24393b2009-07-27 01:23:51 +0000116 // Promote the size of FieldValue if necessary
117 // FIXME: This should never occur, but currently it can because initializer
118 // constants are cast to bool, and because clang is not enforcing bitfield
119 // width limits.
120 if (FieldSize > FieldValue.getBitWidth())
121 FieldValue.zext(FieldSize);
Mike Stump1eb44332009-09-09 15:08:12 +0000122
Anders Carlssond24393b2009-07-27 01:23:51 +0000123 // Truncate the size of FieldValue to the bit field size.
124 if (FieldSize < FieldValue.getBitWidth())
125 FieldValue.trunc(FieldSize);
126
127 if (FieldOffset < NextFieldOffsetInBytes * 8) {
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000128 // Either part of the field or the entire field can go into the previous
129 // byte.
130 assert(!Elements.empty() && "Elements can't be empty!");
Mike Stump1eb44332009-09-09 15:08:12 +0000131
132 unsigned BitsInPreviousByte =
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000133 NextFieldOffsetInBytes * 8 - FieldOffset;
Mike Stump1eb44332009-09-09 15:08:12 +0000134
135 bool FitsCompletelyInPreviousByte =
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000136 BitsInPreviousByte >= FieldValue.getBitWidth();
Mike Stump1eb44332009-09-09 15:08:12 +0000137
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000138 llvm::APInt Tmp = FieldValue;
Mike Stump1eb44332009-09-09 15:08:12 +0000139
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000140 if (!FitsCompletelyInPreviousByte) {
141 unsigned NewFieldWidth = FieldSize - BitsInPreviousByte;
Mike Stump1eb44332009-09-09 15:08:12 +0000142
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000143 if (CGM.getTargetData().isBigEndian()) {
144 Tmp = Tmp.lshr(NewFieldWidth);
145 Tmp.trunc(BitsInPreviousByte);
146
147 // We want the remaining high bits.
148 FieldValue.trunc(NewFieldWidth);
149 } else {
150 Tmp.trunc(BitsInPreviousByte);
151
152 // We want the remaining low bits.
153 FieldValue = FieldValue.lshr(BitsInPreviousByte);
154 FieldValue.trunc(NewFieldWidth);
155 }
156 }
Mike Stump1eb44332009-09-09 15:08:12 +0000157
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000158 Tmp.zext(8);
159 if (CGM.getTargetData().isBigEndian()) {
160 if (FitsCompletelyInPreviousByte)
161 Tmp = Tmp.shl(BitsInPreviousByte - FieldValue.getBitWidth());
162 } else {
163 Tmp = Tmp.shl(8 - BitsInPreviousByte);
164 }
165
166 // Or in the bits that go into the previous byte.
167 Tmp |= cast<llvm::ConstantInt>(Elements.back())->getValue();
168 Elements.back() = llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp);
Mike Stump1eb44332009-09-09 15:08:12 +0000169
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000170 if (FitsCompletelyInPreviousByte)
171 return true;
Anders Carlssond24393b2009-07-27 01:23:51 +0000172 }
Mike Stump1eb44332009-09-09 15:08:12 +0000173
Anders Carlssond24393b2009-07-27 01:23:51 +0000174 while (FieldValue.getBitWidth() > 8) {
175 llvm::APInt Tmp;
Mike Stump1eb44332009-09-09 15:08:12 +0000176
Anders Carlssond24393b2009-07-27 01:23:51 +0000177 if (CGM.getTargetData().isBigEndian()) {
Anders Carlssond24393b2009-07-27 01:23:51 +0000178 // We want the high bits.
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000179 Tmp = FieldValue;
180 Tmp = Tmp.lshr(Tmp.getBitWidth() - 8);
181 Tmp.trunc(8);
182 } else {
183 // We want the low bits.
184 Tmp = FieldValue;
185 Tmp.trunc(8);
Mike Stump1eb44332009-09-09 15:08:12 +0000186
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000187 FieldValue = FieldValue.lshr(8);
Anders Carlssond24393b2009-07-27 01:23:51 +0000188 }
Mike Stump1eb44332009-09-09 15:08:12 +0000189
Anders Carlssond24393b2009-07-27 01:23:51 +0000190 Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp));
191 NextFieldOffsetInBytes++;
Mike Stump1eb44332009-09-09 15:08:12 +0000192
Anders Carlssond24393b2009-07-27 01:23:51 +0000193 FieldValue.trunc(FieldValue.getBitWidth() - 8);
194 }
195
196 assert(FieldValue.getBitWidth() > 0 &&
197 "Should have at least one bit left!");
198 assert(FieldValue.getBitWidth() <= 8 &&
199 "Should not have more than a byte left!");
200
201 if (FieldValue.getBitWidth() < 8) {
202 if (CGM.getTargetData().isBigEndian()) {
203 unsigned BitWidth = FieldValue.getBitWidth();
Mike Stump1eb44332009-09-09 15:08:12 +0000204
Anders Carlssond24393b2009-07-27 01:23:51 +0000205 FieldValue.zext(8);
206 FieldValue = FieldValue << (8 - BitWidth);
Mike Stump1eb44332009-09-09 15:08:12 +0000207 } else
Anders Carlssond24393b2009-07-27 01:23:51 +0000208 FieldValue.zext(8);
209 }
210
211 // Append the last element.
212 Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(),
213 FieldValue));
214 NextFieldOffsetInBytes++;
215 return true;
216 }
Mike Stump1eb44332009-09-09 15:08:12 +0000217
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000218 void AppendPadding(uint64_t NumBytes) {
219 if (!NumBytes)
220 return;
221
Owen Anderson0032b272009-08-13 21:57:51 +0000222 const llvm::Type *Ty = llvm::Type::getInt8Ty(CGM.getLLVMContext());
Mike Stump1eb44332009-09-09 15:08:12 +0000223 if (NumBytes > 1)
Owen Anderson96e0fc72009-07-29 22:16:19 +0000224 Ty = llvm::ArrayType::get(Ty, NumBytes);
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000225
Owen Andersonc9c88b42009-07-31 20:28:54 +0000226 llvm::Constant *C = llvm::Constant::getNullValue(Ty);
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000227 Elements.push_back(C);
228 assert(getAlignment(C) == 1 && "Padding must have 1 byte alignment!");
Mike Stump1eb44332009-09-09 15:08:12 +0000229
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000230 NextFieldOffsetInBytes += getSizeInBytes(C);
231 }
232
233 void AppendTailPadding(uint64_t RecordSize) {
234 assert(RecordSize % 8 == 0 && "Invalid record size!");
235
236 uint64_t RecordSizeInBytes = RecordSize / 8;
237 assert(NextFieldOffsetInBytes <= RecordSizeInBytes && "Size mismatch!");
Mike Stump1eb44332009-09-09 15:08:12 +0000238
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000239 unsigned NumPadBytes = RecordSizeInBytes - NextFieldOffsetInBytes;
240 AppendPadding(NumPadBytes);
241 }
Mike Stump1eb44332009-09-09 15:08:12 +0000242
Anders Carlsson31f2f9c2009-10-02 02:15:20 +0000243 void ConvertStructToPacked() {
244 std::vector<llvm::Constant *> PackedElements;
245 uint64_t ElementOffsetInBytes = 0;
246
247 for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
248 llvm::Constant *C = Elements[i];
249
250 unsigned ElementAlign =
251 CGM.getTargetData().getABITypeAlignment(C->getType());
252 uint64_t AlignedElementOffsetInBytes =
253 llvm::RoundUpToAlignment(ElementOffsetInBytes, ElementAlign);
254
255 if (AlignedElementOffsetInBytes > ElementOffsetInBytes) {
256 // We need some padding.
257 uint64_t NumBytes =
258 AlignedElementOffsetInBytes - ElementOffsetInBytes;
259
260 const llvm::Type *Ty = llvm::Type::getInt8Ty(CGF->getLLVMContext());
261 if (NumBytes > 1)
262 Ty = llvm::ArrayType::get(Ty, NumBytes);
263
264 llvm::Constant *Padding = llvm::Constant::getNullValue(Ty);
265 PackedElements.push_back(Padding);
266 ElementOffsetInBytes += getSizeInBytes(Padding);
267 }
268
269 PackedElements.push_back(C);
270 ElementOffsetInBytes += getSizeInBytes(C);
271 }
272
273 assert(ElementOffsetInBytes == NextFieldOffsetInBytes &&
274 "Packing the struct changed its size!");
275
276 Elements = PackedElements;
277 LLVMStructAlignment = 1;
278 Packed = true;
279 }
280
Eli Friedmanc0879bb2009-08-01 23:11:24 +0000281 bool Build(InitListExpr *ILE) {
Ted Kremenek6217b802009-07-29 21:53:49 +0000282 RecordDecl *RD = ILE->getType()->getAs<RecordType>()->getDecl();
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000283 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
Mike Stump1eb44332009-09-09 15:08:12 +0000284
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000285 unsigned FieldNo = 0;
286 unsigned ElementNo = 0;
Mike Stump1eb44332009-09-09 15:08:12 +0000287 for (RecordDecl::field_iterator Field = RD->field_begin(),
288 FieldEnd = RD->field_end();
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000289 ElementNo < ILE->getNumInits() && Field != FieldEnd;
290 ++Field, ++FieldNo) {
Eli Friedmanc0879bb2009-08-01 23:11:24 +0000291 if (RD->isUnion() && ILE->getInitializedFieldInUnion() != *Field)
292 continue;
293
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000294 if (Field->isBitField()) {
295 if (!Field->getIdentifier())
296 continue;
Mike Stump1eb44332009-09-09 15:08:12 +0000297
Anders Carlssond24393b2009-07-27 01:23:51 +0000298 if (!AppendBitField(*Field, Layout.getFieldOffset(FieldNo),
299 ILE->getInit(ElementNo)))
300 return false;
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000301 } else {
302 if (!AppendField(*Field, Layout.getFieldOffset(FieldNo),
303 ILE->getInit(ElementNo)))
304 return false;
305 }
Mike Stump1eb44332009-09-09 15:08:12 +0000306
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000307 ElementNo++;
308 }
Mike Stump1eb44332009-09-09 15:08:12 +0000309
Anders Carlsson4257bc62009-07-25 00:24:56 +0000310 uint64_t LayoutSizeInBytes = Layout.getSize() / 8;
Mike Stump1eb44332009-09-09 15:08:12 +0000311
Anders Carlsson4257bc62009-07-25 00:24:56 +0000312 if (NextFieldOffsetInBytes > LayoutSizeInBytes) {
313 // If the struct is bigger than the size of the record type,
314 // we must have a flexible array member at the end.
315 assert(RD->hasFlexibleArrayMember() &&
316 "Must have flexible array member if struct is bigger than type!");
Anders Carlsson31f2f9c2009-10-02 02:15:20 +0000317
Anders Carlsson4257bc62009-07-25 00:24:56 +0000318 // No tail padding is necessary.
319 return true;
320 }
Mike Stump1eb44332009-09-09 15:08:12 +0000321
Anders Carlsson31f2f9c2009-10-02 02:15:20 +0000322 uint64_t LLVMSizeInBytes = llvm::RoundUpToAlignment(NextFieldOffsetInBytes,
323 LLVMStructAlignment);
324
325 // Check if we need to convert the struct to a packed struct.
326 if (NextFieldOffsetInBytes <= LayoutSizeInBytes &&
327 LLVMSizeInBytes > LayoutSizeInBytes) {
328 assert(!Packed && "Size mismatch!");
329
330 ConvertStructToPacked();
331 assert(NextFieldOffsetInBytes == LayoutSizeInBytes &&
332 "Converting to packed did not help!");
333 }
334
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000335 // Append tail padding if necessary.
336 AppendTailPadding(Layout.getSize());
Mike Stump1eb44332009-09-09 15:08:12 +0000337
338 assert(Layout.getSize() / 8 == NextFieldOffsetInBytes &&
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000339 "Tail padding mismatch!");
Mike Stump1eb44332009-09-09 15:08:12 +0000340
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000341 return true;
342 }
Mike Stump1eb44332009-09-09 15:08:12 +0000343
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000344 unsigned getAlignment(const llvm::Constant *C) const {
345 if (Packed)
346 return 1;
Mike Stump1eb44332009-09-09 15:08:12 +0000347
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000348 return CGM.getTargetData().getABITypeAlignment(C->getType());
349 }
Mike Stump1eb44332009-09-09 15:08:12 +0000350
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000351 uint64_t getSizeInBytes(const llvm::Constant *C) const {
352 return CGM.getTargetData().getTypeAllocSize(C->getType());
353 }
Mike Stump1eb44332009-09-09 15:08:12 +0000354
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000355public:
356 static llvm::Constant *BuildStruct(CodeGenModule &CGM, CodeGenFunction *CGF,
Eli Friedmanc0879bb2009-08-01 23:11:24 +0000357 InitListExpr *ILE) {
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000358 ConstStructBuilder Builder(CGM, CGF);
Mike Stump1eb44332009-09-09 15:08:12 +0000359
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000360 if (!Builder.Build(ILE))
361 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +0000362
363 llvm::Constant *Result =
Daniel Dunbara10f7ea2009-08-06 01:24:27 +0000364 llvm::ConstantStruct::get(CGM.getLLVMContext(),
Owen Anderson47a434f2009-08-05 23:18:46 +0000365 Builder.Elements, Builder.Packed);
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000366
Anders Carlsson4257bc62009-07-25 00:24:56 +0000367 assert(llvm::RoundUpToAlignment(Builder.NextFieldOffsetInBytes,
Mike Stump1eb44332009-09-09 15:08:12 +0000368 Builder.getAlignment(Result)) ==
Anders Carlsson4257bc62009-07-25 00:24:56 +0000369 Builder.getSizeInBytes(Result) && "Size mismatch!");
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000370
Anders Carlsson56027192009-07-24 18:20:38 +0000371 return Result;
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000372 }
373};
Mike Stump1eb44332009-09-09 15:08:12 +0000374
375class VISIBILITY_HIDDEN ConstExprEmitter :
Anders Carlsson84005b42008-01-26 04:30:23 +0000376 public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000377 CodeGenModule &CGM;
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000378 CodeGenFunction *CGF;
Owen Andersona1cf15f2009-07-14 23:10:40 +0000379 llvm::LLVMContext &VMContext;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000380public:
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000381 ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
Owen Andersona1cf15f2009-07-14 23:10:40 +0000382 : CGM(cgm), CGF(cgf), VMContext(cgm.getLLVMContext()) {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000383 }
Mike Stump1eb44332009-09-09 15:08:12 +0000384
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000385 //===--------------------------------------------------------------------===//
386 // Visitor Methods
387 //===--------------------------------------------------------------------===//
Mike Stump1eb44332009-09-09 15:08:12 +0000388
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000389 llvm::Constant *VisitStmt(Stmt *S) {
Anders Carlsson069880e2009-03-03 16:43:34 +0000390 return 0;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000391 }
Mike Stump1eb44332009-09-09 15:08:12 +0000392
393 llvm::Constant *VisitParenExpr(ParenExpr *PE) {
394 return Visit(PE->getSubExpr());
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000395 }
Mike Stump1eb44332009-09-09 15:08:12 +0000396
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000397 llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
398 return Visit(E->getInitializer());
399 }
Mike Stump1eb44332009-09-09 15:08:12 +0000400
Argyrios Kyrtzidis0835a3c2008-08-18 23:01:59 +0000401 llvm::Constant *VisitCastExpr(CastExpr* E) {
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000402 switch (E->getCastKind()) {
403 case CastExpr::CK_ToUnion: {
404 // GCC cast to union extension
405 assert(E->getType()->isUnionType() &&
406 "Destination type is not union type!");
Nuno Lopes81e51e22009-01-17 00:48:48 +0000407 const llvm::Type *Ty = ConvertType(E->getType());
Anders Carlssone9352cc2009-04-08 04:48:15 +0000408 Expr *SubExpr = E->getSubExpr();
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000409
Mike Stump1eb44332009-09-09 15:08:12 +0000410 llvm::Constant *C =
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000411 CGM.EmitConstantExpr(SubExpr, SubExpr->getType(), CGF);
412 if (!C)
413 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +0000414
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000415 // Build a struct with the union sub-element as the first member,
416 // and padded to the appropriate size
417 std::vector<llvm::Constant*> Elts;
418 std::vector<const llvm::Type*> Types;
419 Elts.push_back(C);
420 Types.push_back(C->getType());
421 unsigned CurSize = CGM.getTargetData().getTypeAllocSize(C->getType());
422 unsigned TotalSize = CGM.getTargetData().getTypeAllocSize(Ty);
Mike Stump1eb44332009-09-09 15:08:12 +0000423
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000424 assert(CurSize <= TotalSize && "Union size mismatch!");
425 if (unsigned NumPadBytes = TotalSize - CurSize) {
Owen Anderson0032b272009-08-13 21:57:51 +0000426 const llvm::Type *Ty = llvm::Type::getInt8Ty(VMContext);
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000427 if (NumPadBytes > 1)
428 Ty = llvm::ArrayType::get(Ty, NumPadBytes);
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000429
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000430 Elts.push_back(llvm::Constant::getNullValue(Ty));
431 Types.push_back(Ty);
432 }
Mike Stump1eb44332009-09-09 15:08:12 +0000433
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000434 llvm::StructType* STy =
435 llvm::StructType::get(C->getType()->getContext(), Types, false);
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000436 return llvm::ConstantStruct::get(STy, Elts);
Nuno Lopes81e51e22009-01-17 00:48:48 +0000437 }
Anders Carlssond1a9bac2009-08-23 00:02:11 +0000438 case CastExpr::CK_NullToMemberPointer:
439 return CGM.EmitNullConstant(E->getType());
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000440 default: {
441 // FIXME: This should be handled by the CK_NoOp cast kind.
442 // Explicit and implicit no-op casts
443 QualType Ty = E->getType(), SubTy = E->getSubExpr()->getType();
444 if (CGM.getContext().hasSameUnqualifiedType(Ty, SubTy))
445 return Visit(E->getSubExpr());
446 return 0;
Eli Friedmanddd2b6e2009-02-22 07:29:04 +0000447 }
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000448 }
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000449 }
Devang Pateleae15602008-02-05 02:39:50 +0000450
Chris Lattner04421082008-04-08 04:40:51 +0000451 llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
452 return Visit(DAE->getExpr());
453 }
454
Eli Friedmana8234002008-05-30 10:24:46 +0000455 llvm::Constant *EmitArrayInitialization(InitListExpr *ILE) {
Eli Friedman4d1c0182008-02-21 17:45:41 +0000456 std::vector<llvm::Constant*> Elts;
Eli Friedmana8234002008-05-30 10:24:46 +0000457 const llvm::ArrayType *AType =
458 cast<llvm::ArrayType>(ConvertType(ILE->getType()));
Eli Friedman4d1c0182008-02-21 17:45:41 +0000459 unsigned NumInitElements = ILE->getNumInits();
460 // FIXME: Check for wide strings
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000461 // FIXME: Check for NumInitElements exactly equal to 1??
Mike Stump1eb44332009-09-09 15:08:12 +0000462 if (NumInitElements > 0 &&
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000463 (isa<StringLiteral>(ILE->getInit(0)) ||
464 isa<ObjCEncodeExpr>(ILE->getInit(0))) &&
Chris Lattnerc63a1f22008-08-04 07:31:14 +0000465 ILE->getType()->getArrayElementTypeNoTypeQual()->isCharType())
Eli Friedman4d1c0182008-02-21 17:45:41 +0000466 return Visit(ILE->getInit(0));
Devang Pateleae15602008-02-05 02:39:50 +0000467 const llvm::Type *ElemTy = AType->getElementType();
468 unsigned NumElements = AType->getNumElements();
469
Mike Stump1eb44332009-09-09 15:08:12 +0000470 // Initialising an array requires us to automatically
Devang Pateleae15602008-02-05 02:39:50 +0000471 // initialise any elements that have not been initialised explicitly
472 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
473
474 // Copy initializer elements.
475 unsigned i = 0;
Eli Friedmana04a1532008-05-30 19:58:50 +0000476 bool RewriteType = false;
Devang Pateleae15602008-02-05 02:39:50 +0000477 for (; i < NumInitableElts; ++i) {
Anders Carlssone9352cc2009-04-08 04:48:15 +0000478 Expr *Init = ILE->getInit(i);
479 llvm::Constant *C = CGM.EmitConstantExpr(Init, Init->getType(), CGF);
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000480 if (!C)
481 return 0;
Eli Friedmana04a1532008-05-30 19:58:50 +0000482 RewriteType |= (C->getType() != ElemTy);
Devang Pateleae15602008-02-05 02:39:50 +0000483 Elts.push_back(C);
484 }
Eli Friedmana04a1532008-05-30 19:58:50 +0000485
Devang Pateleae15602008-02-05 02:39:50 +0000486 // Initialize remaining array elements.
Eli Friedman0f593122009-04-13 21:47:26 +0000487 // FIXME: This doesn't handle member pointers correctly!
Devang Pateleae15602008-02-05 02:39:50 +0000488 for (; i < NumElements; ++i)
Owen Andersonc9c88b42009-07-31 20:28:54 +0000489 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
Devang Pateleae15602008-02-05 02:39:50 +0000490
Eli Friedmana04a1532008-05-30 19:58:50 +0000491 if (RewriteType) {
492 // FIXME: Try to avoid packing the array
493 std::vector<const llvm::Type*> Types;
494 for (unsigned i = 0; i < Elts.size(); ++i)
495 Types.push_back(Elts[i]->getType());
Mike Stump1eb44332009-09-09 15:08:12 +0000496 const llvm::StructType *SType = llvm::StructType::get(AType->getContext(),
Owen Anderson47a434f2009-08-05 23:18:46 +0000497 Types, true);
Owen Anderson08e25242009-07-27 22:29:56 +0000498 return llvm::ConstantStruct::get(SType, Elts);
Eli Friedmana04a1532008-05-30 19:58:50 +0000499 }
500
Mike Stump1eb44332009-09-09 15:08:12 +0000501 return llvm::ConstantArray::get(AType, Elts);
Devang Pateleae15602008-02-05 02:39:50 +0000502 }
503
Eli Friedmana8234002008-05-30 10:24:46 +0000504 llvm::Constant *EmitStructInitialization(InitListExpr *ILE) {
Anders Carlsson8dca3b32009-07-27 05:54:15 +0000505 return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
Devang Pateleae15602008-02-05 02:39:50 +0000506 }
Mike Stump1eb44332009-09-09 15:08:12 +0000507
Eli Friedmana8234002008-05-30 10:24:46 +0000508 llvm::Constant *EmitUnionInitialization(InitListExpr *ILE) {
Anders Carlssonf6c43962009-07-31 21:34:04 +0000509 return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
Eli Friedmana8234002008-05-30 10:24:46 +0000510 }
511
512 llvm::Constant *EmitVectorInitialization(InitListExpr *ILE) {
513 const llvm::VectorType *VType =
514 cast<llvm::VectorType>(ConvertType(ILE->getType()));
Eli Friedmancfb313b2008-05-30 12:02:14 +0000515 const llvm::Type *ElemTy = VType->getElementType();
516 std::vector<llvm::Constant*> Elts;
Devang Pateleae15602008-02-05 02:39:50 +0000517 unsigned NumElements = VType->getNumElements();
Eli Friedmancfb313b2008-05-30 12:02:14 +0000518 unsigned NumInitElements = ILE->getNumInits();
Devang Pateleae15602008-02-05 02:39:50 +0000519
Eli Friedmancfb313b2008-05-30 12:02:14 +0000520 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
521
Devang Pateleae15602008-02-05 02:39:50 +0000522 // Copy initializer elements.
523 unsigned i = 0;
Eli Friedmancfb313b2008-05-30 12:02:14 +0000524 for (; i < NumInitableElts; ++i) {
Anders Carlssone9352cc2009-04-08 04:48:15 +0000525 Expr *Init = ILE->getInit(i);
526 llvm::Constant *C = CGM.EmitConstantExpr(Init, Init->getType(), CGF);
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000527 if (!C)
528 return 0;
Devang Pateleae15602008-02-05 02:39:50 +0000529 Elts.push_back(C);
530 }
531
Eli Friedmancfb313b2008-05-30 12:02:14 +0000532 for (; i < NumElements; ++i)
Owen Andersonc9c88b42009-07-31 20:28:54 +0000533 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
Eli Friedmancfb313b2008-05-30 12:02:14 +0000534
Mike Stump1eb44332009-09-09 15:08:12 +0000535 return llvm::ConstantVector::get(VType, Elts);
Devang Pateleae15602008-02-05 02:39:50 +0000536 }
Mike Stump1eb44332009-09-09 15:08:12 +0000537
Anders Carlsson7d801ea2009-01-30 06:13:25 +0000538 llvm::Constant *VisitImplicitValueInitExpr(ImplicitValueInitExpr* E) {
Eli Friedman0f593122009-04-13 21:47:26 +0000539 return CGM.EmitNullConstant(E->getType());
Anders Carlsson7d801ea2009-01-30 06:13:25 +0000540 }
Mike Stump1eb44332009-09-09 15:08:12 +0000541
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000542 llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
Eli Friedmana8234002008-05-30 10:24:46 +0000543 if (ILE->getType()->isScalarType()) {
544 // We have a scalar in braces. Just use the first element.
Anders Carlssone9352cc2009-04-08 04:48:15 +0000545 if (ILE->getNumInits() > 0) {
546 Expr *Init = ILE->getInit(0);
547 return CGM.EmitConstantExpr(Init, Init->getType(), CGF);
548 }
Eli Friedman0f593122009-04-13 21:47:26 +0000549 return CGM.EmitNullConstant(ILE->getType());
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000550 }
Mike Stump1eb44332009-09-09 15:08:12 +0000551
Eli Friedmana8234002008-05-30 10:24:46 +0000552 if (ILE->getType()->isArrayType())
553 return EmitArrayInitialization(ILE);
Devang Patel47fb6972008-01-29 23:23:18 +0000554
Eli Friedmana8234002008-05-30 10:24:46 +0000555 if (ILE->getType()->isStructureType())
556 return EmitStructInitialization(ILE);
557
558 if (ILE->getType()->isUnionType())
559 return EmitUnionInitialization(ILE);
560
561 if (ILE->getType()->isVectorType())
562 return EmitVectorInitialization(ILE);
563
Devang Pateleae15602008-02-05 02:39:50 +0000564 assert(0 && "Unable to handle InitListExpr");
Chris Lattner2b94fe32008-03-01 08:45:05 +0000565 // Get rid of control reaches end of void function warning.
566 // Not reached.
567 return 0;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000568 }
Eli Friedman3b707e72008-02-21 17:57:49 +0000569
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000570 llvm::Constant *VisitStringLiteral(StringLiteral *E) {
Chris Lattnera7ad98f2008-02-11 00:02:17 +0000571 assert(!E->getType()->isPointerType() && "Strings are always arrays");
Mike Stump1eb44332009-09-09 15:08:12 +0000572
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000573 // This must be a string initializing an array in a static initializer.
574 // Don't emit it as the address of the string, emit the string data itself
575 // as an inline array.
Owen Anderson0032b272009-08-13 21:57:51 +0000576 return llvm::ConstantArray::get(VMContext,
577 CGM.GetStringForStringLiteral(E), false);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000578 }
579
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000580 llvm::Constant *VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
581 // This must be an @encode initializing an array in a static initializer.
582 // Don't emit it as the address of the string, emit the string data itself
583 // as an inline array.
584 std::string Str;
585 CGM.getContext().getObjCEncodingForType(E->getEncodedType(), Str);
586 const ConstantArrayType *CAT = cast<ConstantArrayType>(E->getType());
Mike Stump1eb44332009-09-09 15:08:12 +0000587
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000588 // Resize the string to the right size, adding zeros at the end, or
589 // truncating as needed.
590 Str.resize(CAT->getSize().getZExtValue(), '\0');
Owen Anderson0032b272009-08-13 21:57:51 +0000591 return llvm::ConstantArray::get(VMContext, Str, false);
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000592 }
Mike Stump1eb44332009-09-09 15:08:12 +0000593
Eli Friedmanbc5ed6e2008-05-29 11:22:45 +0000594 llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
595 return Visit(E->getSubExpr());
596 }
Mike Stumpb83d2872009-02-19 22:01:56 +0000597
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000598 // Utility methods
599 const llvm::Type *ConvertType(QualType T) {
600 return CGM.getTypes().ConvertType(T);
601 }
Anders Carlsson84005b42008-01-26 04:30:23 +0000602
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000603public:
Anders Carlsson84005b42008-01-26 04:30:23 +0000604 llvm::Constant *EmitLValue(Expr *E) {
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000605 switch (E->getStmtClass()) {
Eli Friedman546d94c2008-02-11 00:23:10 +0000606 default: break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000607 case Expr::CompoundLiteralExprClass: {
608 // Note that due to the nature of compound literals, this is guaranteed
609 // to be the only use of the variable, so we just generate it here.
Anders Carlsson84005b42008-01-26 04:30:23 +0000610 CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
611 llvm::Constant* C = Visit(CLE->getInitializer());
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000612 // FIXME: "Leaked" on failure.
613 if (C)
Owen Anderson1c431b32009-07-08 19:05:04 +0000614 C = new llvm::GlobalVariable(CGM.getModule(), C->getType(),
Eli Friedmand15d8c62009-08-26 20:01:39 +0000615 E->getType().isConstant(CGM.getContext()),
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000616 llvm::GlobalValue::InternalLinkage,
Eli Friedmand15d8c62009-08-26 20:01:39 +0000617 C, ".compoundliteral", 0, false,
618 E->getType().getAddressSpace());
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000619 return C;
Anders Carlsson84005b42008-01-26 04:30:23 +0000620 }
Mike Stump1eb44332009-09-09 15:08:12 +0000621 case Expr::DeclRefExprClass:
Douglas Gregor1a49af92009-01-06 05:10:23 +0000622 case Expr::QualifiedDeclRefExprClass: {
Douglas Gregor8e9bebd2008-10-21 16:13:35 +0000623 NamedDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000624 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
Anders Carlsson555b4bb2009-09-10 23:43:36 +0000625 return CGM.GetAddrOfFunction(FD);
Steve Naroff248a7532008-04-15 22:42:06 +0000626 if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
Daniel Dunbard01b6692009-02-24 18:41:57 +0000627 // We can never refer to a variable with local storage.
Mike Stump1eb44332009-09-09 15:08:12 +0000628 if (!VD->hasLocalStorage()) {
Daniel Dunbard01b6692009-02-24 18:41:57 +0000629 if (VD->isFileVarDecl() || VD->hasExternalStorage())
630 return CGM.GetAddrOfGlobalVar(VD);
631 else if (VD->isBlockVarDecl()) {
632 assert(CGF && "Can't access static local vars without CGF");
633 return CGF->GetAddrOfStaticLocalVar(VD);
634 }
Steve Naroff248a7532008-04-15 22:42:06 +0000635 }
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000636 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000637 break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000638 }
Daniel Dunbar61432932008-08-13 23:20:05 +0000639 case Expr::StringLiteralClass:
640 return CGM.GetAddrOfConstantStringFromLiteral(cast<StringLiteral>(E));
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000641 case Expr::ObjCEncodeExprClass:
642 return CGM.GetAddrOfConstantStringFromObjCEncode(cast<ObjCEncodeExpr>(E));
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000643 case Expr::ObjCStringLiteralClass: {
644 ObjCStringLiteral* SL = cast<ObjCStringLiteral>(E);
Steve Naroff33fdb732009-03-31 16:53:37 +0000645 llvm::Constant *C = CGM.getObjCRuntime().GenerateConstantString(SL);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000646 return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000647 }
Chris Lattner35345642008-12-12 05:18:02 +0000648 case Expr::PredefinedExprClass: {
649 // __func__/__FUNCTION__ -> "". __PRETTY_FUNCTION__ -> "top level".
650 std::string Str;
Mike Stump1eb44332009-09-09 15:08:12 +0000651 if (cast<PredefinedExpr>(E)->getIdentType() ==
Chris Lattner35345642008-12-12 05:18:02 +0000652 PredefinedExpr::PrettyFunction)
653 Str = "top level";
Mike Stump1eb44332009-09-09 15:08:12 +0000654
Chris Lattner35345642008-12-12 05:18:02 +0000655 return CGM.GetAddrOfConstantCString(Str, ".tmp");
656 }
Eli Friedmanf0115892009-01-25 01:21:06 +0000657 case Expr::AddrLabelExprClass: {
658 assert(CGF && "Invalid address of label expression outside function.");
Owen Anderson0032b272009-08-13 21:57:51 +0000659 unsigned id =
660 CGF->GetIDForAddrOfLabel(cast<AddrLabelExpr>(E)->getLabel());
661 llvm::Constant *C =
662 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), id);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000663 return llvm::ConstantExpr::getIntToPtr(C, ConvertType(E->getType()));
Eli Friedmanf0115892009-01-25 01:21:06 +0000664 }
Eli Friedman3941b182009-01-25 01:54:01 +0000665 case Expr::CallExprClass: {
666 CallExpr* CE = cast<CallExpr>(E);
Mike Stump1eb44332009-09-09 15:08:12 +0000667 if (CE->isBuiltinCall(CGM.getContext()) !=
Douglas Gregor3c385e52009-02-14 18:57:46 +0000668 Builtin::BI__builtin___CFStringMakeConstantString)
Eli Friedman3941b182009-01-25 01:54:01 +0000669 break;
670 const Expr *Arg = CE->getArg(0)->IgnoreParenCasts();
671 const StringLiteral *Literal = cast<StringLiteral>(Arg);
Steve Naroff33fdb732009-03-31 16:53:37 +0000672 // FIXME: need to deal with UCN conversion issues.
Steve Naroff8d4141f2009-04-01 13:55:36 +0000673 return CGM.GetAddrOfConstantCFString(Literal);
Eli Friedman3941b182009-01-25 01:54:01 +0000674 }
Mike Stumpbd65cac2009-02-19 01:01:04 +0000675 case Expr::BlockExprClass: {
Anders Carlsson4de9fce2009-03-01 01:09:12 +0000676 std::string FunctionName;
677 if (CGF)
678 FunctionName = CGF->CurFn->getName();
679 else
680 FunctionName = "global";
681
682 return CGM.GetAddrOfGlobalBlock(cast<BlockExpr>(E), FunctionName.c_str());
Mike Stumpbd65cac2009-02-19 01:01:04 +0000683 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000684 }
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000685
686 return 0;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000687 }
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000688};
Mike Stump1eb44332009-09-09 15:08:12 +0000689
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000690} // end anonymous namespace.
691
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000692llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
Anders Carlssone9352cc2009-04-08 04:48:15 +0000693 QualType DestType,
Chris Lattner96196622008-07-26 22:37:01 +0000694 CodeGenFunction *CGF) {
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000695 Expr::EvalResult Result;
Mike Stump1eb44332009-09-09 15:08:12 +0000696
Anders Carlsson1e5dc6e2009-04-11 01:08:03 +0000697 bool Success = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000698
Eli Friedman50c39ea2009-05-27 06:04:58 +0000699 if (DestType->isReferenceType())
700 Success = E->EvaluateAsLValue(Result, Context);
Mike Stump1eb44332009-09-09 15:08:12 +0000701 else
Anders Carlsson1e5dc6e2009-04-11 01:08:03 +0000702 Success = E->Evaluate(Result, Context);
Mike Stump1eb44332009-09-09 15:08:12 +0000703
Anders Carlsson1e5dc6e2009-04-11 01:08:03 +0000704 if (Success) {
Mike Stump1eb44332009-09-09 15:08:12 +0000705 assert(!Result.HasSideEffects &&
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000706 "Constant expr should not have any side effects!");
707 switch (Result.Val.getKind()) {
Daniel Dunbara5fd07b2009-01-28 22:24:07 +0000708 case APValue::Uninitialized:
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000709 assert(0 && "Constant expressions should be initialized.");
710 return 0;
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000711 case APValue::LValue: {
Anders Carlsson1e5dc6e2009-04-11 01:08:03 +0000712 const llvm::Type *DestTy = getTypes().ConvertTypeForMem(DestType);
Mike Stump1eb44332009-09-09 15:08:12 +0000713 llvm::Constant *Offset =
714 llvm::ConstantInt::get(llvm::Type::getInt64Ty(VMContext),
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000715 Result.Val.getLValueOffset());
Mike Stump1eb44332009-09-09 15:08:12 +0000716
Daniel Dunbare20de512009-02-19 21:44:24 +0000717 llvm::Constant *C;
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000718 if (const Expr *LVBase = Result.Val.getLValueBase()) {
Daniel Dunbare20de512009-02-19 21:44:24 +0000719 C = ConstExprEmitter(*this, CGF).EmitLValue(const_cast<Expr*>(LVBase));
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000720
Daniel Dunbare20de512009-02-19 21:44:24 +0000721 // Apply offset if necessary.
722 if (!Offset->isNullValue()) {
Mike Stump1eb44332009-09-09 15:08:12 +0000723 const llvm::Type *Type =
Owen Anderson0032b272009-08-13 21:57:51 +0000724 llvm::PointerType::getUnqual(llvm::Type::getInt8Ty(VMContext));
Owen Anderson3c4972d2009-07-29 18:54:39 +0000725 llvm::Constant *Casted = llvm::ConstantExpr::getBitCast(C, Type);
726 Casted = llvm::ConstantExpr::getGetElementPtr(Casted, &Offset, 1);
727 C = llvm::ConstantExpr::getBitCast(Casted, C->getType());
Daniel Dunbare20de512009-02-19 21:44:24 +0000728 }
729
730 // Convert to the appropriate type; this could be an lvalue for
731 // an integer.
Anders Carlsson12e330c2009-04-10 04:59:59 +0000732 if (isa<llvm::PointerType>(DestTy))
Owen Anderson3c4972d2009-07-29 18:54:39 +0000733 return llvm::ConstantExpr::getBitCast(C, DestTy);
Daniel Dunbare20de512009-02-19 21:44:24 +0000734
Owen Anderson3c4972d2009-07-29 18:54:39 +0000735 return llvm::ConstantExpr::getPtrToInt(C, DestTy);
Daniel Dunbare20de512009-02-19 21:44:24 +0000736 } else {
737 C = Offset;
738
739 // Convert to the appropriate type; this could be an lvalue for
740 // an integer.
Anders Carlsson12e330c2009-04-10 04:59:59 +0000741 if (isa<llvm::PointerType>(DestTy))
Owen Anderson3c4972d2009-07-29 18:54:39 +0000742 return llvm::ConstantExpr::getIntToPtr(C, DestTy);
Daniel Dunbare20de512009-02-19 21:44:24 +0000743
744 // If the types don't match this should only be a truncate.
Anders Carlsson12e330c2009-04-10 04:59:59 +0000745 if (C->getType() != DestTy)
Owen Anderson3c4972d2009-07-29 18:54:39 +0000746 return llvm::ConstantExpr::getTrunc(C, DestTy);
Daniel Dunbare20de512009-02-19 21:44:24 +0000747
748 return C;
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000749 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000750 }
Eli Friedman7b30b1c2008-11-17 03:57:28 +0000751 case APValue::Int: {
Mike Stump1eb44332009-09-09 15:08:12 +0000752 llvm::Constant *C = llvm::ConstantInt::get(VMContext,
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000753 Result.Val.getInt());
Mike Stump1eb44332009-09-09 15:08:12 +0000754
Owen Anderson0032b272009-08-13 21:57:51 +0000755 if (C->getType() == llvm::Type::getInt1Ty(VMContext)) {
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000756 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
Owen Anderson3c4972d2009-07-29 18:54:39 +0000757 C = llvm::ConstantExpr::getZExt(C, BoolTy);
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000758 }
759 return C;
Eli Friedman7b30b1c2008-11-17 03:57:28 +0000760 }
Daniel Dunbara5fd07b2009-01-28 22:24:07 +0000761 case APValue::ComplexInt: {
762 llvm::Constant *Complex[2];
Mike Stump1eb44332009-09-09 15:08:12 +0000763
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000764 Complex[0] = llvm::ConstantInt::get(VMContext,
765 Result.Val.getComplexIntReal());
Mike Stump1eb44332009-09-09 15:08:12 +0000766 Complex[1] = llvm::ConstantInt::get(VMContext,
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000767 Result.Val.getComplexIntImag());
Mike Stump1eb44332009-09-09 15:08:12 +0000768
Nick Lewycky0d36dd22009-09-19 20:00:52 +0000769 // FIXME: the target may want to specify that this is packed.
770 return llvm::ConstantStruct::get(VMContext, Complex, 2, false);
Daniel Dunbara5fd07b2009-01-28 22:24:07 +0000771 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000772 case APValue::Float:
Owen Andersonbc0a2222009-07-27 21:00:51 +0000773 return llvm::ConstantFP::get(VMContext, Result.Val.getFloat());
Anders Carlssonaa432562008-11-17 01:58:55 +0000774 case APValue::ComplexFloat: {
775 llvm::Constant *Complex[2];
Mike Stump1eb44332009-09-09 15:08:12 +0000776
777 Complex[0] = llvm::ConstantFP::get(VMContext,
Owen Andersonbc0a2222009-07-27 21:00:51 +0000778 Result.Val.getComplexFloatReal());
779 Complex[1] = llvm::ConstantFP::get(VMContext,
780 Result.Val.getComplexFloatImag());
Mike Stump1eb44332009-09-09 15:08:12 +0000781
Nick Lewycky0d36dd22009-09-19 20:00:52 +0000782 // FIXME: the target may want to specify that this is packed.
783 return llvm::ConstantStruct::get(VMContext, Complex, 2, false);
Anders Carlssonaa432562008-11-17 01:58:55 +0000784 }
Nate Begeman3d309f92009-01-18 01:01:34 +0000785 case APValue::Vector: {
786 llvm::SmallVector<llvm::Constant *, 4> Inits;
787 unsigned NumElts = Result.Val.getVectorLength();
Mike Stump1eb44332009-09-09 15:08:12 +0000788
Nate Begeman3d309f92009-01-18 01:01:34 +0000789 for (unsigned i = 0; i != NumElts; ++i) {
790 APValue &Elt = Result.Val.getVectorElt(i);
791 if (Elt.isInt())
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000792 Inits.push_back(llvm::ConstantInt::get(VMContext, Elt.getInt()));
Nate Begeman3d309f92009-01-18 01:01:34 +0000793 else
Owen Andersonbc0a2222009-07-27 21:00:51 +0000794 Inits.push_back(llvm::ConstantFP::get(VMContext, Elt.getFloat()));
Nate Begeman3d309f92009-01-18 01:01:34 +0000795 }
Owen Anderson4a289322009-07-28 21:22:35 +0000796 return llvm::ConstantVector::get(&Inits[0], Inits.size());
Nate Begeman3d309f92009-01-18 01:01:34 +0000797 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000798 }
799 }
Eli Friedman7dfa6392008-06-01 15:31:44 +0000800
801 llvm::Constant* C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
Owen Anderson0032b272009-08-13 21:57:51 +0000802 if (C && C->getType() == llvm::Type::getInt1Ty(VMContext)) {
Eli Friedman7dfa6392008-06-01 15:31:44 +0000803 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
Owen Anderson3c4972d2009-07-29 18:54:39 +0000804 C = llvm::ConstantExpr::getZExt(C, BoolTy);
Eli Friedman7dfa6392008-06-01 15:31:44 +0000805 }
806 return C;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000807}
Eli Friedman0f593122009-04-13 21:47:26 +0000808
Anders Carlssone9d34dc2009-09-29 02:09:01 +0000809static inline bool isDataMemberPointerType(QualType T) {
810 if (const MemberPointerType *MPT = T->getAs<MemberPointerType>())
811 return !MPT->getPointeeType()->isFunctionType();
812
813 return false;
814}
815
Eli Friedman0f593122009-04-13 21:47:26 +0000816llvm::Constant *CodeGenModule::EmitNullConstant(QualType T) {
Anders Carlssonad3e7112009-08-24 17:16:23 +0000817 // No need to check for member pointers when not compiling C++.
818 if (!getContext().getLangOptions().CPlusPlus)
819 return llvm::Constant::getNullValue(getTypes().ConvertTypeForMem(T));
Mike Stump1eb44332009-09-09 15:08:12 +0000820
Anders Carlsson237957c2009-08-09 18:26:27 +0000821 if (const ConstantArrayType *CAT = Context.getAsConstantArrayType(T)) {
Mike Stump1eb44332009-09-09 15:08:12 +0000822
Anders Carlsson237957c2009-08-09 18:26:27 +0000823 QualType ElementTy = CAT->getElementType();
824
825 // FIXME: Handle arrays of structs that contain member pointers.
Anders Carlssone9d34dc2009-09-29 02:09:01 +0000826 if (isDataMemberPointerType(Context.getBaseElementType(ElementTy))) {
Anders Carlsson237957c2009-08-09 18:26:27 +0000827 llvm::Constant *Element = EmitNullConstant(ElementTy);
828 uint64_t NumElements = CAT->getSize().getZExtValue();
829 std::vector<llvm::Constant *> Array(NumElements);
830 for (uint64_t i = 0; i != NumElements; ++i)
831 Array[i] = Element;
Mike Stump1eb44332009-09-09 15:08:12 +0000832
833 const llvm::ArrayType *ATy =
Anders Carlsson237957c2009-08-09 18:26:27 +0000834 cast<llvm::ArrayType>(getTypes().ConvertTypeForMem(T));
835 return llvm::ConstantArray::get(ATy, Array);
836 }
837 }
Anders Carlssonfc3eaa42009-08-23 01:25:01 +0000838
839 if (const RecordType *RT = T->getAs<RecordType>()) {
Anders Carlssonad3e7112009-08-24 17:16:23 +0000840 const RecordDecl *RD = RT->getDecl();
Mike Stump1eb44332009-09-09 15:08:12 +0000841 // FIXME: It would be better if there was a way to explicitly compute the
Anders Carlssonad3e7112009-08-24 17:16:23 +0000842 // record layout instead of converting to a type.
843 Types.ConvertTagDeclType(RD);
Mike Stump1eb44332009-09-09 15:08:12 +0000844
Anders Carlssonad3e7112009-08-24 17:16:23 +0000845 const CGRecordLayout &Layout = Types.getCGRecordLayout(RD);
846 if (Layout.containsMemberPointer()) {
Anders Carlssonfc3eaa42009-08-23 01:25:01 +0000847 assert(0 && "FIXME: No support for structs with member pointers yet!");
848 }
Anders Carlssonfc3eaa42009-08-23 01:25:01 +0000849 }
Mike Stump1eb44332009-09-09 15:08:12 +0000850
Anders Carlsson237957c2009-08-09 18:26:27 +0000851 // FIXME: Handle structs that contain member pointers.
Anders Carlssone9d34dc2009-09-29 02:09:01 +0000852 if (isDataMemberPointerType(T))
Anders Carlsson237957c2009-08-09 18:26:27 +0000853 return llvm::Constant::getAllOnesValue(getTypes().ConvertTypeForMem(T));
Mike Stump1eb44332009-09-09 15:08:12 +0000854
Anders Carlssonb0d0ea02009-08-02 21:18:22 +0000855 return llvm::Constant::getNullValue(getTypes().ConvertTypeForMem(T));
Eli Friedman0f593122009-04-13 21:47:26 +0000856}