blob: d1713626ef71a96d809fa9dfdf33d32bf64f2931 [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 Carlsson4c98efd2009-07-24 15:20:52 +000040 std::vector<llvm::Constant *> Elements;
41
42 ConstStructBuilder(CodeGenModule &CGM, CodeGenFunction *CGF)
43 : CGM(CGM), CGF(CGF), Packed(false), NextFieldOffsetInBytes(0) { }
44
Mike Stump1eb44332009-09-09 15:08:12 +000045 bool AppendField(const FieldDecl *Field, uint64_t FieldOffset,
Anders Carlsson4c98efd2009-07-24 15:20:52 +000046 const Expr *InitExpr) {
47 uint64_t FieldOffsetInBytes = FieldOffset / 8;
Mike Stump1eb44332009-09-09 15:08:12 +000048
49 assert(NextFieldOffsetInBytes <= FieldOffsetInBytes
Anders Carlsson4c98efd2009-07-24 15:20:52 +000050 && "Field offset mismatch!");
Mike Stump1eb44332009-09-09 15:08:12 +000051
Anders Carlsson4c98efd2009-07-24 15:20:52 +000052 // Emit the field.
53 llvm::Constant *C = CGM.EmitConstantExpr(InitExpr, Field->getType(), CGF);
54 if (!C)
55 return false;
56
57 unsigned FieldAlignment = getAlignment(C);
Mike Stump1eb44332009-09-09 15:08:12 +000058
Anders Carlsson4c98efd2009-07-24 15:20:52 +000059 // Round up the field offset to the alignment of the field type.
Mike Stump1eb44332009-09-09 15:08:12 +000060 uint64_t AlignedNextFieldOffsetInBytes =
Anders Carlsson4c98efd2009-07-24 15:20:52 +000061 llvm::RoundUpToAlignment(NextFieldOffsetInBytes, FieldAlignment);
Mike Stump1eb44332009-09-09 15:08:12 +000062
Anders Carlsson4c98efd2009-07-24 15:20:52 +000063 if (AlignedNextFieldOffsetInBytes > FieldOffsetInBytes) {
Anders Carlssonbea9c522009-07-25 01:40:29 +000064 std::vector<llvm::Constant *> PackedElements;
Mike Stump1eb44332009-09-09 15:08:12 +000065
Anders Carlssonbea9c522009-07-25 01:40:29 +000066 assert(!Packed && "Alignment is wrong even with a packed struct!");
Mike Stump1eb44332009-09-09 15:08:12 +000067
Anders Carlssonbea9c522009-07-25 01:40:29 +000068 // Convert the struct to a packed struct.
69 uint64_t ElementOffsetInBytes = 0;
Mike Stump1eb44332009-09-09 15:08:12 +000070
Anders Carlssonbea9c522009-07-25 01:40:29 +000071 for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
72 llvm::Constant *C = Elements[i];
Mike Stump1eb44332009-09-09 15:08:12 +000073
74 unsigned ElementAlign =
Anders Carlssonbea9c522009-07-25 01:40:29 +000075 CGM.getTargetData().getABITypeAlignment(C->getType());
Mike Stump1eb44332009-09-09 15:08:12 +000076 uint64_t AlignedElementOffsetInBytes =
Anders Carlssonbea9c522009-07-25 01:40:29 +000077 llvm::RoundUpToAlignment(ElementOffsetInBytes, ElementAlign);
Mike Stump1eb44332009-09-09 15:08:12 +000078
Anders Carlssonbea9c522009-07-25 01:40:29 +000079 if (AlignedElementOffsetInBytes > ElementOffsetInBytes) {
80 // We need some padding.
Mike Stump1eb44332009-09-09 15:08:12 +000081 uint64_t NumBytes =
Anders Carlssonbea9c522009-07-25 01:40:29 +000082 AlignedElementOffsetInBytes - ElementOffsetInBytes;
Mike Stump1eb44332009-09-09 15:08:12 +000083
Owen Anderson0032b272009-08-13 21:57:51 +000084 const llvm::Type *Ty = llvm::Type::getInt8Ty(CGF->getLLVMContext());
Mike Stump1eb44332009-09-09 15:08:12 +000085 if (NumBytes > 1)
Owen Anderson96e0fc72009-07-29 22:16:19 +000086 Ty = llvm::ArrayType::get(Ty, NumBytes);
Mike Stump1eb44332009-09-09 15:08:12 +000087
Owen Andersonc9c88b42009-07-31 20:28:54 +000088 llvm::Constant *Padding = llvm::Constant::getNullValue(Ty);
Anders Carlssonbea9c522009-07-25 01:40:29 +000089 PackedElements.push_back(Padding);
90 ElementOffsetInBytes += getSizeInBytes(Padding);
91 }
Mike Stump1eb44332009-09-09 15:08:12 +000092
Anders Carlssonbea9c522009-07-25 01:40:29 +000093 PackedElements.push_back(C);
94 ElementOffsetInBytes += getSizeInBytes(C);
95 }
96
97 assert(ElementOffsetInBytes == NextFieldOffsetInBytes &&
98 "Packing the struct changed its size!");
99
100 Elements = PackedElements;
101 Packed = true;
102 AlignedNextFieldOffsetInBytes = NextFieldOffsetInBytes;
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000103 }
104
105 if (AlignedNextFieldOffsetInBytes < FieldOffsetInBytes) {
106 // We need to append padding.
Anders Carlssonbea9c522009-07-25 01:40:29 +0000107 AppendPadding(FieldOffsetInBytes - NextFieldOffsetInBytes);
Mike Stump1eb44332009-09-09 15:08:12 +0000108
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000109 assert(NextFieldOffsetInBytes == FieldOffsetInBytes &&
110 "Did not add enough padding!");
Mike Stump1eb44332009-09-09 15:08:12 +0000111
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000112 AlignedNextFieldOffsetInBytes = NextFieldOffsetInBytes;
113 }
Mike Stump1eb44332009-09-09 15:08:12 +0000114
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000115 // Add the field.
116 Elements.push_back(C);
117 NextFieldOffsetInBytes = AlignedNextFieldOffsetInBytes + getSizeInBytes(C);
118
119 return true;
120 }
Mike Stump1eb44332009-09-09 15:08:12 +0000121
122 bool AppendBitField(const FieldDecl *Field, uint64_t FieldOffset,
Anders Carlssond24393b2009-07-27 01:23:51 +0000123 const Expr *InitExpr) {
Mike Stump1eb44332009-09-09 15:08:12 +0000124 llvm::ConstantInt *CI =
125 cast_or_null<llvm::ConstantInt>(CGM.EmitConstantExpr(InitExpr,
126 Field->getType(),
Anders Carlssond24393b2009-07-27 01:23:51 +0000127 CGF));
128 // FIXME: Can this ever happen?
129 if (!CI)
130 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000131
Anders Carlssond24393b2009-07-27 01:23:51 +0000132 if (FieldOffset > NextFieldOffsetInBytes * 8) {
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000133 // We need to add padding.
Mike Stump1eb44332009-09-09 15:08:12 +0000134 uint64_t NumBytes =
135 llvm::RoundUpToAlignment(FieldOffset -
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000136 NextFieldOffsetInBytes * 8, 8) / 8;
Mike Stump1eb44332009-09-09 15:08:12 +0000137
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000138 AppendPadding(NumBytes);
Anders Carlssond24393b2009-07-27 01:23:51 +0000139 }
140
Mike Stump1eb44332009-09-09 15:08:12 +0000141 uint64_t FieldSize =
Anders Carlssond24393b2009-07-27 01:23:51 +0000142 Field->getBitWidth()->EvaluateAsInt(CGM.getContext()).getZExtValue();
143
144 llvm::APInt FieldValue = CI->getValue();
Mike Stump1eb44332009-09-09 15:08:12 +0000145
Anders Carlssond24393b2009-07-27 01:23:51 +0000146 // Promote the size of FieldValue if necessary
147 // FIXME: This should never occur, but currently it can because initializer
148 // constants are cast to bool, and because clang is not enforcing bitfield
149 // width limits.
150 if (FieldSize > FieldValue.getBitWidth())
151 FieldValue.zext(FieldSize);
Mike Stump1eb44332009-09-09 15:08:12 +0000152
Anders Carlssond24393b2009-07-27 01:23:51 +0000153 // Truncate the size of FieldValue to the bit field size.
154 if (FieldSize < FieldValue.getBitWidth())
155 FieldValue.trunc(FieldSize);
156
157 if (FieldOffset < NextFieldOffsetInBytes * 8) {
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000158 // Either part of the field or the entire field can go into the previous
159 // byte.
160 assert(!Elements.empty() && "Elements can't be empty!");
Mike Stump1eb44332009-09-09 15:08:12 +0000161
162 unsigned BitsInPreviousByte =
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000163 NextFieldOffsetInBytes * 8 - FieldOffset;
Mike Stump1eb44332009-09-09 15:08:12 +0000164
165 bool FitsCompletelyInPreviousByte =
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000166 BitsInPreviousByte >= FieldValue.getBitWidth();
Mike Stump1eb44332009-09-09 15:08:12 +0000167
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000168 llvm::APInt Tmp = FieldValue;
Mike Stump1eb44332009-09-09 15:08:12 +0000169
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000170 if (!FitsCompletelyInPreviousByte) {
171 unsigned NewFieldWidth = FieldSize - BitsInPreviousByte;
Mike Stump1eb44332009-09-09 15:08:12 +0000172
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000173 if (CGM.getTargetData().isBigEndian()) {
174 Tmp = Tmp.lshr(NewFieldWidth);
175 Tmp.trunc(BitsInPreviousByte);
176
177 // We want the remaining high bits.
178 FieldValue.trunc(NewFieldWidth);
179 } else {
180 Tmp.trunc(BitsInPreviousByte);
181
182 // We want the remaining low bits.
183 FieldValue = FieldValue.lshr(BitsInPreviousByte);
184 FieldValue.trunc(NewFieldWidth);
185 }
186 }
Mike Stump1eb44332009-09-09 15:08:12 +0000187
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000188 Tmp.zext(8);
189 if (CGM.getTargetData().isBigEndian()) {
190 if (FitsCompletelyInPreviousByte)
191 Tmp = Tmp.shl(BitsInPreviousByte - FieldValue.getBitWidth());
192 } else {
193 Tmp = Tmp.shl(8 - BitsInPreviousByte);
194 }
195
196 // Or in the bits that go into the previous byte.
197 Tmp |= cast<llvm::ConstantInt>(Elements.back())->getValue();
198 Elements.back() = llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp);
Mike Stump1eb44332009-09-09 15:08:12 +0000199
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000200 if (FitsCompletelyInPreviousByte)
201 return true;
Anders Carlssond24393b2009-07-27 01:23:51 +0000202 }
Mike Stump1eb44332009-09-09 15:08:12 +0000203
Anders Carlssond24393b2009-07-27 01:23:51 +0000204 while (FieldValue.getBitWidth() > 8) {
205 llvm::APInt Tmp;
Mike Stump1eb44332009-09-09 15:08:12 +0000206
Anders Carlssond24393b2009-07-27 01:23:51 +0000207 if (CGM.getTargetData().isBigEndian()) {
Anders Carlssond24393b2009-07-27 01:23:51 +0000208 // We want the high bits.
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000209 Tmp = FieldValue;
210 Tmp = Tmp.lshr(Tmp.getBitWidth() - 8);
211 Tmp.trunc(8);
212 } else {
213 // We want the low bits.
214 Tmp = FieldValue;
215 Tmp.trunc(8);
Mike Stump1eb44332009-09-09 15:08:12 +0000216
Anders Carlsson7a86d4b2009-07-27 02:56:37 +0000217 FieldValue = FieldValue.lshr(8);
Anders Carlssond24393b2009-07-27 01:23:51 +0000218 }
Mike Stump1eb44332009-09-09 15:08:12 +0000219
Anders Carlssond24393b2009-07-27 01:23:51 +0000220 Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp));
221 NextFieldOffsetInBytes++;
Mike Stump1eb44332009-09-09 15:08:12 +0000222
Anders Carlssond24393b2009-07-27 01:23:51 +0000223 FieldValue.trunc(FieldValue.getBitWidth() - 8);
224 }
225
226 assert(FieldValue.getBitWidth() > 0 &&
227 "Should have at least one bit left!");
228 assert(FieldValue.getBitWidth() <= 8 &&
229 "Should not have more than a byte left!");
230
231 if (FieldValue.getBitWidth() < 8) {
232 if (CGM.getTargetData().isBigEndian()) {
233 unsigned BitWidth = FieldValue.getBitWidth();
Mike Stump1eb44332009-09-09 15:08:12 +0000234
Anders Carlssond24393b2009-07-27 01:23:51 +0000235 FieldValue.zext(8);
236 FieldValue = FieldValue << (8 - BitWidth);
Mike Stump1eb44332009-09-09 15:08:12 +0000237 } else
Anders Carlssond24393b2009-07-27 01:23:51 +0000238 FieldValue.zext(8);
239 }
240
241 // Append the last element.
242 Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(),
243 FieldValue));
244 NextFieldOffsetInBytes++;
245 return true;
246 }
Mike Stump1eb44332009-09-09 15:08:12 +0000247
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000248 void AppendPadding(uint64_t NumBytes) {
249 if (!NumBytes)
250 return;
251
Owen Anderson0032b272009-08-13 21:57:51 +0000252 const llvm::Type *Ty = llvm::Type::getInt8Ty(CGM.getLLVMContext());
Mike Stump1eb44332009-09-09 15:08:12 +0000253 if (NumBytes > 1)
Owen Anderson96e0fc72009-07-29 22:16:19 +0000254 Ty = llvm::ArrayType::get(Ty, NumBytes);
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000255
Owen Andersonc9c88b42009-07-31 20:28:54 +0000256 llvm::Constant *C = llvm::Constant::getNullValue(Ty);
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000257 Elements.push_back(C);
258 assert(getAlignment(C) == 1 && "Padding must have 1 byte alignment!");
Mike Stump1eb44332009-09-09 15:08:12 +0000259
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000260 NextFieldOffsetInBytes += getSizeInBytes(C);
261 }
262
263 void AppendTailPadding(uint64_t RecordSize) {
264 assert(RecordSize % 8 == 0 && "Invalid record size!");
265
266 uint64_t RecordSizeInBytes = RecordSize / 8;
267 assert(NextFieldOffsetInBytes <= RecordSizeInBytes && "Size mismatch!");
Mike Stump1eb44332009-09-09 15:08:12 +0000268
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000269 unsigned NumPadBytes = RecordSizeInBytes - NextFieldOffsetInBytes;
270 AppendPadding(NumPadBytes);
271 }
Mike Stump1eb44332009-09-09 15:08:12 +0000272
Eli Friedmanc0879bb2009-08-01 23:11:24 +0000273 bool Build(InitListExpr *ILE) {
Ted Kremenek6217b802009-07-29 21:53:49 +0000274 RecordDecl *RD = ILE->getType()->getAs<RecordType>()->getDecl();
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000275 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
Mike Stump1eb44332009-09-09 15:08:12 +0000276
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000277 unsigned FieldNo = 0;
278 unsigned ElementNo = 0;
Mike Stump1eb44332009-09-09 15:08:12 +0000279 for (RecordDecl::field_iterator Field = RD->field_begin(),
280 FieldEnd = RD->field_end();
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000281 ElementNo < ILE->getNumInits() && Field != FieldEnd;
282 ++Field, ++FieldNo) {
Eli Friedmanc0879bb2009-08-01 23:11:24 +0000283 if (RD->isUnion() && ILE->getInitializedFieldInUnion() != *Field)
284 continue;
285
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000286 if (Field->isBitField()) {
287 if (!Field->getIdentifier())
288 continue;
Mike Stump1eb44332009-09-09 15:08:12 +0000289
Anders Carlssond24393b2009-07-27 01:23:51 +0000290 if (!AppendBitField(*Field, Layout.getFieldOffset(FieldNo),
291 ILE->getInit(ElementNo)))
292 return false;
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000293 } else {
294 if (!AppendField(*Field, Layout.getFieldOffset(FieldNo),
295 ILE->getInit(ElementNo)))
296 return false;
297 }
Mike Stump1eb44332009-09-09 15:08:12 +0000298
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000299 ElementNo++;
300 }
Mike Stump1eb44332009-09-09 15:08:12 +0000301
Anders Carlsson4257bc62009-07-25 00:24:56 +0000302 uint64_t LayoutSizeInBytes = Layout.getSize() / 8;
Mike Stump1eb44332009-09-09 15:08:12 +0000303
Anders Carlsson4257bc62009-07-25 00:24:56 +0000304 if (NextFieldOffsetInBytes > LayoutSizeInBytes) {
305 // If the struct is bigger than the size of the record type,
306 // we must have a flexible array member at the end.
307 assert(RD->hasFlexibleArrayMember() &&
308 "Must have flexible array member if struct is bigger than type!");
Mike Stump1eb44332009-09-09 15:08:12 +0000309
Anders Carlsson4257bc62009-07-25 00:24:56 +0000310 // No tail padding is necessary.
311 return true;
312 }
Mike Stump1eb44332009-09-09 15:08:12 +0000313
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000314 // Append tail padding if necessary.
315 AppendTailPadding(Layout.getSize());
Mike Stump1eb44332009-09-09 15:08:12 +0000316
317 assert(Layout.getSize() / 8 == NextFieldOffsetInBytes &&
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000318 "Tail padding mismatch!");
Mike Stump1eb44332009-09-09 15:08:12 +0000319
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000320 return true;
321 }
Mike Stump1eb44332009-09-09 15:08:12 +0000322
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000323 unsigned getAlignment(const llvm::Constant *C) const {
324 if (Packed)
325 return 1;
Mike Stump1eb44332009-09-09 15:08:12 +0000326
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000327 return CGM.getTargetData().getABITypeAlignment(C->getType());
328 }
Mike Stump1eb44332009-09-09 15:08:12 +0000329
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000330 uint64_t getSizeInBytes(const llvm::Constant *C) const {
331 return CGM.getTargetData().getTypeAllocSize(C->getType());
332 }
Mike Stump1eb44332009-09-09 15:08:12 +0000333
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000334public:
335 static llvm::Constant *BuildStruct(CodeGenModule &CGM, CodeGenFunction *CGF,
Eli Friedmanc0879bb2009-08-01 23:11:24 +0000336 InitListExpr *ILE) {
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000337 ConstStructBuilder Builder(CGM, CGF);
Mike Stump1eb44332009-09-09 15:08:12 +0000338
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000339 if (!Builder.Build(ILE))
340 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +0000341
342 llvm::Constant *Result =
Daniel Dunbara10f7ea2009-08-06 01:24:27 +0000343 llvm::ConstantStruct::get(CGM.getLLVMContext(),
Owen Anderson47a434f2009-08-05 23:18:46 +0000344 Builder.Elements, Builder.Packed);
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000345
Anders Carlsson4257bc62009-07-25 00:24:56 +0000346 assert(llvm::RoundUpToAlignment(Builder.NextFieldOffsetInBytes,
Mike Stump1eb44332009-09-09 15:08:12 +0000347 Builder.getAlignment(Result)) ==
Anders Carlsson4257bc62009-07-25 00:24:56 +0000348 Builder.getSizeInBytes(Result) && "Size mismatch!");
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000349
Anders Carlsson56027192009-07-24 18:20:38 +0000350 return Result;
Anders Carlsson4c98efd2009-07-24 15:20:52 +0000351 }
352};
Mike Stump1eb44332009-09-09 15:08:12 +0000353
354class VISIBILITY_HIDDEN ConstExprEmitter :
Anders Carlsson84005b42008-01-26 04:30:23 +0000355 public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000356 CodeGenModule &CGM;
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000357 CodeGenFunction *CGF;
Owen Andersona1cf15f2009-07-14 23:10:40 +0000358 llvm::LLVMContext &VMContext;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000359public:
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000360 ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
Owen Andersona1cf15f2009-07-14 23:10:40 +0000361 : CGM(cgm), CGF(cgf), VMContext(cgm.getLLVMContext()) {
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000362 }
Mike Stump1eb44332009-09-09 15:08:12 +0000363
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000364 //===--------------------------------------------------------------------===//
365 // Visitor Methods
366 //===--------------------------------------------------------------------===//
Mike Stump1eb44332009-09-09 15:08:12 +0000367
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000368 llvm::Constant *VisitStmt(Stmt *S) {
Anders Carlsson069880e2009-03-03 16:43:34 +0000369 return 0;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000370 }
Mike Stump1eb44332009-09-09 15:08:12 +0000371
372 llvm::Constant *VisitParenExpr(ParenExpr *PE) {
373 return Visit(PE->getSubExpr());
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000374 }
Mike Stump1eb44332009-09-09 15:08:12 +0000375
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000376 llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
377 return Visit(E->getInitializer());
378 }
Mike Stump1eb44332009-09-09 15:08:12 +0000379
Argyrios Kyrtzidis0835a3c2008-08-18 23:01:59 +0000380 llvm::Constant *VisitCastExpr(CastExpr* E) {
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000381 switch (E->getCastKind()) {
382 case CastExpr::CK_ToUnion: {
383 // GCC cast to union extension
384 assert(E->getType()->isUnionType() &&
385 "Destination type is not union type!");
Nuno Lopes81e51e22009-01-17 00:48:48 +0000386 const llvm::Type *Ty = ConvertType(E->getType());
Anders Carlssone9352cc2009-04-08 04:48:15 +0000387 Expr *SubExpr = E->getSubExpr();
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000388
Mike Stump1eb44332009-09-09 15:08:12 +0000389 llvm::Constant *C =
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000390 CGM.EmitConstantExpr(SubExpr, SubExpr->getType(), CGF);
391 if (!C)
392 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +0000393
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000394 // Build a struct with the union sub-element as the first member,
395 // and padded to the appropriate size
396 std::vector<llvm::Constant*> Elts;
397 std::vector<const llvm::Type*> Types;
398 Elts.push_back(C);
399 Types.push_back(C->getType());
400 unsigned CurSize = CGM.getTargetData().getTypeAllocSize(C->getType());
401 unsigned TotalSize = CGM.getTargetData().getTypeAllocSize(Ty);
Mike Stump1eb44332009-09-09 15:08:12 +0000402
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000403 assert(CurSize <= TotalSize && "Union size mismatch!");
404 if (unsigned NumPadBytes = TotalSize - CurSize) {
Owen Anderson0032b272009-08-13 21:57:51 +0000405 const llvm::Type *Ty = llvm::Type::getInt8Ty(VMContext);
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000406 if (NumPadBytes > 1)
407 Ty = llvm::ArrayType::get(Ty, NumPadBytes);
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000408
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000409 Elts.push_back(llvm::Constant::getNullValue(Ty));
410 Types.push_back(Ty);
411 }
Mike Stump1eb44332009-09-09 15:08:12 +0000412
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000413 llvm::StructType* STy =
414 llvm::StructType::get(C->getType()->getContext(), Types, false);
Anders Carlssonaaed50d2009-07-31 21:38:39 +0000415 return llvm::ConstantStruct::get(STy, Elts);
Nuno Lopes81e51e22009-01-17 00:48:48 +0000416 }
Anders Carlssond1a9bac2009-08-23 00:02:11 +0000417 case CastExpr::CK_NullToMemberPointer:
418 return CGM.EmitNullConstant(E->getType());
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000419 default: {
420 // FIXME: This should be handled by the CK_NoOp cast kind.
421 // Explicit and implicit no-op casts
422 QualType Ty = E->getType(), SubTy = E->getSubExpr()->getType();
423 if (CGM.getContext().hasSameUnqualifiedType(Ty, SubTy))
424 return Visit(E->getSubExpr());
425 return 0;
Eli Friedmanddd2b6e2009-02-22 07:29:04 +0000426 }
Anders Carlsson0086ccb2009-08-22 23:54:44 +0000427 }
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000428 }
Devang Pateleae15602008-02-05 02:39:50 +0000429
Chris Lattner04421082008-04-08 04:40:51 +0000430 llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
431 return Visit(DAE->getExpr());
432 }
433
Eli Friedmana8234002008-05-30 10:24:46 +0000434 llvm::Constant *EmitArrayInitialization(InitListExpr *ILE) {
Eli Friedman4d1c0182008-02-21 17:45:41 +0000435 std::vector<llvm::Constant*> Elts;
Eli Friedmana8234002008-05-30 10:24:46 +0000436 const llvm::ArrayType *AType =
437 cast<llvm::ArrayType>(ConvertType(ILE->getType()));
Eli Friedman4d1c0182008-02-21 17:45:41 +0000438 unsigned NumInitElements = ILE->getNumInits();
439 // FIXME: Check for wide strings
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000440 // FIXME: Check for NumInitElements exactly equal to 1??
Mike Stump1eb44332009-09-09 15:08:12 +0000441 if (NumInitElements > 0 &&
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000442 (isa<StringLiteral>(ILE->getInit(0)) ||
443 isa<ObjCEncodeExpr>(ILE->getInit(0))) &&
Chris Lattnerc63a1f22008-08-04 07:31:14 +0000444 ILE->getType()->getArrayElementTypeNoTypeQual()->isCharType())
Eli Friedman4d1c0182008-02-21 17:45:41 +0000445 return Visit(ILE->getInit(0));
Devang Pateleae15602008-02-05 02:39:50 +0000446 const llvm::Type *ElemTy = AType->getElementType();
447 unsigned NumElements = AType->getNumElements();
448
Mike Stump1eb44332009-09-09 15:08:12 +0000449 // Initialising an array requires us to automatically
Devang Pateleae15602008-02-05 02:39:50 +0000450 // initialise any elements that have not been initialised explicitly
451 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
452
453 // Copy initializer elements.
454 unsigned i = 0;
Eli Friedmana04a1532008-05-30 19:58:50 +0000455 bool RewriteType = false;
Devang Pateleae15602008-02-05 02:39:50 +0000456 for (; i < NumInitableElts; ++i) {
Anders Carlssone9352cc2009-04-08 04:48:15 +0000457 Expr *Init = ILE->getInit(i);
458 llvm::Constant *C = CGM.EmitConstantExpr(Init, Init->getType(), CGF);
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000459 if (!C)
460 return 0;
Eli Friedmana04a1532008-05-30 19:58:50 +0000461 RewriteType |= (C->getType() != ElemTy);
Devang Pateleae15602008-02-05 02:39:50 +0000462 Elts.push_back(C);
463 }
Eli Friedmana04a1532008-05-30 19:58:50 +0000464
Devang Pateleae15602008-02-05 02:39:50 +0000465 // Initialize remaining array elements.
Eli Friedman0f593122009-04-13 21:47:26 +0000466 // FIXME: This doesn't handle member pointers correctly!
Devang Pateleae15602008-02-05 02:39:50 +0000467 for (; i < NumElements; ++i)
Owen Andersonc9c88b42009-07-31 20:28:54 +0000468 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
Devang Pateleae15602008-02-05 02:39:50 +0000469
Eli Friedmana04a1532008-05-30 19:58:50 +0000470 if (RewriteType) {
471 // FIXME: Try to avoid packing the array
472 std::vector<const llvm::Type*> Types;
473 for (unsigned i = 0; i < Elts.size(); ++i)
474 Types.push_back(Elts[i]->getType());
Mike Stump1eb44332009-09-09 15:08:12 +0000475 const llvm::StructType *SType = llvm::StructType::get(AType->getContext(),
Owen Anderson47a434f2009-08-05 23:18:46 +0000476 Types, true);
Owen Anderson08e25242009-07-27 22:29:56 +0000477 return llvm::ConstantStruct::get(SType, Elts);
Eli Friedmana04a1532008-05-30 19:58:50 +0000478 }
479
Mike Stump1eb44332009-09-09 15:08:12 +0000480 return llvm::ConstantArray::get(AType, Elts);
Devang Pateleae15602008-02-05 02:39:50 +0000481 }
482
Eli Friedmana8234002008-05-30 10:24:46 +0000483 llvm::Constant *EmitStructInitialization(InitListExpr *ILE) {
Anders Carlsson8dca3b32009-07-27 05:54:15 +0000484 return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
Devang Pateleae15602008-02-05 02:39:50 +0000485 }
Mike Stump1eb44332009-09-09 15:08:12 +0000486
Eli Friedmana8234002008-05-30 10:24:46 +0000487 llvm::Constant *EmitUnionInitialization(InitListExpr *ILE) {
Anders Carlssonf6c43962009-07-31 21:34:04 +0000488 return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
Eli Friedmana8234002008-05-30 10:24:46 +0000489 }
490
491 llvm::Constant *EmitVectorInitialization(InitListExpr *ILE) {
492 const llvm::VectorType *VType =
493 cast<llvm::VectorType>(ConvertType(ILE->getType()));
Eli Friedmancfb313b2008-05-30 12:02:14 +0000494 const llvm::Type *ElemTy = VType->getElementType();
495 std::vector<llvm::Constant*> Elts;
Devang Pateleae15602008-02-05 02:39:50 +0000496 unsigned NumElements = VType->getNumElements();
Eli Friedmancfb313b2008-05-30 12:02:14 +0000497 unsigned NumInitElements = ILE->getNumInits();
Devang Pateleae15602008-02-05 02:39:50 +0000498
Eli Friedmancfb313b2008-05-30 12:02:14 +0000499 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
500
Devang Pateleae15602008-02-05 02:39:50 +0000501 // Copy initializer elements.
502 unsigned i = 0;
Eli Friedmancfb313b2008-05-30 12:02:14 +0000503 for (; i < NumInitableElts; ++i) {
Anders Carlssone9352cc2009-04-08 04:48:15 +0000504 Expr *Init = ILE->getInit(i);
505 llvm::Constant *C = CGM.EmitConstantExpr(Init, Init->getType(), CGF);
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000506 if (!C)
507 return 0;
Devang Pateleae15602008-02-05 02:39:50 +0000508 Elts.push_back(C);
509 }
510
Eli Friedmancfb313b2008-05-30 12:02:14 +0000511 for (; i < NumElements; ++i)
Owen Andersonc9c88b42009-07-31 20:28:54 +0000512 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
Eli Friedmancfb313b2008-05-30 12:02:14 +0000513
Mike Stump1eb44332009-09-09 15:08:12 +0000514 return llvm::ConstantVector::get(VType, Elts);
Devang Pateleae15602008-02-05 02:39:50 +0000515 }
Mike Stump1eb44332009-09-09 15:08:12 +0000516
Anders Carlsson7d801ea2009-01-30 06:13:25 +0000517 llvm::Constant *VisitImplicitValueInitExpr(ImplicitValueInitExpr* E) {
Eli Friedman0f593122009-04-13 21:47:26 +0000518 return CGM.EmitNullConstant(E->getType());
Anders Carlsson7d801ea2009-01-30 06:13:25 +0000519 }
Mike Stump1eb44332009-09-09 15:08:12 +0000520
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000521 llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
Eli Friedmana8234002008-05-30 10:24:46 +0000522 if (ILE->getType()->isScalarType()) {
523 // We have a scalar in braces. Just use the first element.
Anders Carlssone9352cc2009-04-08 04:48:15 +0000524 if (ILE->getNumInits() > 0) {
525 Expr *Init = ILE->getInit(0);
526 return CGM.EmitConstantExpr(Init, Init->getType(), CGF);
527 }
Eli Friedman0f593122009-04-13 21:47:26 +0000528 return CGM.EmitNullConstant(ILE->getType());
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000529 }
Mike Stump1eb44332009-09-09 15:08:12 +0000530
Eli Friedmana8234002008-05-30 10:24:46 +0000531 if (ILE->getType()->isArrayType())
532 return EmitArrayInitialization(ILE);
Devang Patel47fb6972008-01-29 23:23:18 +0000533
Eli Friedmana8234002008-05-30 10:24:46 +0000534 if (ILE->getType()->isStructureType())
535 return EmitStructInitialization(ILE);
536
537 if (ILE->getType()->isUnionType())
538 return EmitUnionInitialization(ILE);
539
540 if (ILE->getType()->isVectorType())
541 return EmitVectorInitialization(ILE);
542
Devang Pateleae15602008-02-05 02:39:50 +0000543 assert(0 && "Unable to handle InitListExpr");
Chris Lattner2b94fe32008-03-01 08:45:05 +0000544 // Get rid of control reaches end of void function warning.
545 // Not reached.
546 return 0;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000547 }
Eli Friedman3b707e72008-02-21 17:57:49 +0000548
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000549 llvm::Constant *VisitStringLiteral(StringLiteral *E) {
Chris Lattnera7ad98f2008-02-11 00:02:17 +0000550 assert(!E->getType()->isPointerType() && "Strings are always arrays");
Mike Stump1eb44332009-09-09 15:08:12 +0000551
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000552 // This must be a string initializing an array in a static initializer.
553 // Don't emit it as the address of the string, emit the string data itself
554 // as an inline array.
Owen Anderson0032b272009-08-13 21:57:51 +0000555 return llvm::ConstantArray::get(VMContext,
556 CGM.GetStringForStringLiteral(E), false);
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000557 }
558
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000559 llvm::Constant *VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
560 // This must be an @encode initializing an array in a static initializer.
561 // Don't emit it as the address of the string, emit the string data itself
562 // as an inline array.
563 std::string Str;
564 CGM.getContext().getObjCEncodingForType(E->getEncodedType(), Str);
565 const ConstantArrayType *CAT = cast<ConstantArrayType>(E->getType());
Mike Stump1eb44332009-09-09 15:08:12 +0000566
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000567 // Resize the string to the right size, adding zeros at the end, or
568 // truncating as needed.
569 Str.resize(CAT->getSize().getZExtValue(), '\0');
Owen Anderson0032b272009-08-13 21:57:51 +0000570 return llvm::ConstantArray::get(VMContext, Str, false);
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000571 }
Mike Stump1eb44332009-09-09 15:08:12 +0000572
Eli Friedmanbc5ed6e2008-05-29 11:22:45 +0000573 llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
574 return Visit(E->getSubExpr());
575 }
Mike Stumpb83d2872009-02-19 22:01:56 +0000576
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000577 // Utility methods
578 const llvm::Type *ConvertType(QualType T) {
579 return CGM.getTypes().ConvertType(T);
580 }
Anders Carlsson84005b42008-01-26 04:30:23 +0000581
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000582public:
Anders Carlsson84005b42008-01-26 04:30:23 +0000583 llvm::Constant *EmitLValue(Expr *E) {
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000584 switch (E->getStmtClass()) {
Eli Friedman546d94c2008-02-11 00:23:10 +0000585 default: break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000586 case Expr::CompoundLiteralExprClass: {
587 // Note that due to the nature of compound literals, this is guaranteed
588 // to be the only use of the variable, so we just generate it here.
Anders Carlsson84005b42008-01-26 04:30:23 +0000589 CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
590 llvm::Constant* C = Visit(CLE->getInitializer());
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000591 // FIXME: "Leaked" on failure.
592 if (C)
Owen Anderson1c431b32009-07-08 19:05:04 +0000593 C = new llvm::GlobalVariable(CGM.getModule(), C->getType(),
Eli Friedmand15d8c62009-08-26 20:01:39 +0000594 E->getType().isConstant(CGM.getContext()),
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000595 llvm::GlobalValue::InternalLinkage,
Eli Friedmand15d8c62009-08-26 20:01:39 +0000596 C, ".compoundliteral", 0, false,
597 E->getType().getAddressSpace());
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000598 return C;
Anders Carlsson84005b42008-01-26 04:30:23 +0000599 }
Mike Stump1eb44332009-09-09 15:08:12 +0000600 case Expr::DeclRefExprClass:
Douglas Gregor1a49af92009-01-06 05:10:23 +0000601 case Expr::QualifiedDeclRefExprClass: {
Douglas Gregor8e9bebd2008-10-21 16:13:35 +0000602 NamedDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000603 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
Anders Carlsson555b4bb2009-09-10 23:43:36 +0000604 return CGM.GetAddrOfFunction(FD);
Steve Naroff248a7532008-04-15 22:42:06 +0000605 if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
Daniel Dunbard01b6692009-02-24 18:41:57 +0000606 // We can never refer to a variable with local storage.
Mike Stump1eb44332009-09-09 15:08:12 +0000607 if (!VD->hasLocalStorage()) {
Daniel Dunbard01b6692009-02-24 18:41:57 +0000608 if (VD->isFileVarDecl() || VD->hasExternalStorage())
609 return CGM.GetAddrOfGlobalVar(VD);
610 else if (VD->isBlockVarDecl()) {
611 assert(CGF && "Can't access static local vars without CGF");
612 return CGF->GetAddrOfStaticLocalVar(VD);
613 }
Steve Naroff248a7532008-04-15 22:42:06 +0000614 }
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000615 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000616 break;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000617 }
Daniel Dunbar61432932008-08-13 23:20:05 +0000618 case Expr::StringLiteralClass:
619 return CGM.GetAddrOfConstantStringFromLiteral(cast<StringLiteral>(E));
Chris Lattnereaf2bb82009-02-24 22:18:39 +0000620 case Expr::ObjCEncodeExprClass:
621 return CGM.GetAddrOfConstantStringFromObjCEncode(cast<ObjCEncodeExpr>(E));
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000622 case Expr::ObjCStringLiteralClass: {
623 ObjCStringLiteral* SL = cast<ObjCStringLiteral>(E);
Steve Naroff33fdb732009-03-31 16:53:37 +0000624 llvm::Constant *C = CGM.getObjCRuntime().GenerateConstantString(SL);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000625 return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000626 }
Chris Lattner35345642008-12-12 05:18:02 +0000627 case Expr::PredefinedExprClass: {
628 // __func__/__FUNCTION__ -> "". __PRETTY_FUNCTION__ -> "top level".
629 std::string Str;
Mike Stump1eb44332009-09-09 15:08:12 +0000630 if (cast<PredefinedExpr>(E)->getIdentType() ==
Chris Lattner35345642008-12-12 05:18:02 +0000631 PredefinedExpr::PrettyFunction)
632 Str = "top level";
Mike Stump1eb44332009-09-09 15:08:12 +0000633
Chris Lattner35345642008-12-12 05:18:02 +0000634 return CGM.GetAddrOfConstantCString(Str, ".tmp");
635 }
Eli Friedmanf0115892009-01-25 01:21:06 +0000636 case Expr::AddrLabelExprClass: {
637 assert(CGF && "Invalid address of label expression outside function.");
Owen Anderson0032b272009-08-13 21:57:51 +0000638 unsigned id =
639 CGF->GetIDForAddrOfLabel(cast<AddrLabelExpr>(E)->getLabel());
640 llvm::Constant *C =
641 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), id);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000642 return llvm::ConstantExpr::getIntToPtr(C, ConvertType(E->getType()));
Eli Friedmanf0115892009-01-25 01:21:06 +0000643 }
Eli Friedman3941b182009-01-25 01:54:01 +0000644 case Expr::CallExprClass: {
645 CallExpr* CE = cast<CallExpr>(E);
Mike Stump1eb44332009-09-09 15:08:12 +0000646 if (CE->isBuiltinCall(CGM.getContext()) !=
Douglas Gregor3c385e52009-02-14 18:57:46 +0000647 Builtin::BI__builtin___CFStringMakeConstantString)
Eli Friedman3941b182009-01-25 01:54:01 +0000648 break;
649 const Expr *Arg = CE->getArg(0)->IgnoreParenCasts();
650 const StringLiteral *Literal = cast<StringLiteral>(Arg);
Steve Naroff33fdb732009-03-31 16:53:37 +0000651 // FIXME: need to deal with UCN conversion issues.
Steve Naroff8d4141f2009-04-01 13:55:36 +0000652 return CGM.GetAddrOfConstantCFString(Literal);
Eli Friedman3941b182009-01-25 01:54:01 +0000653 }
Mike Stumpbd65cac2009-02-19 01:01:04 +0000654 case Expr::BlockExprClass: {
Anders Carlsson4de9fce2009-03-01 01:09:12 +0000655 std::string FunctionName;
656 if (CGF)
657 FunctionName = CGF->CurFn->getName();
658 else
659 FunctionName = "global";
660
661 return CGM.GetAddrOfGlobalBlock(cast<BlockExpr>(E), FunctionName.c_str());
Mike Stumpbd65cac2009-02-19 01:01:04 +0000662 }
Eli Friedman546d94c2008-02-11 00:23:10 +0000663 }
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000664
665 return 0;
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000666 }
Anders Carlssone3d3b7d2008-01-26 02:08:50 +0000667};
Mike Stump1eb44332009-09-09 15:08:12 +0000668
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000669} // end anonymous namespace.
670
Lauro Ramos Venancio81373352008-02-26 21:41:45 +0000671llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
Anders Carlssone9352cc2009-04-08 04:48:15 +0000672 QualType DestType,
Chris Lattner96196622008-07-26 22:37:01 +0000673 CodeGenFunction *CGF) {
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000674 Expr::EvalResult Result;
Mike Stump1eb44332009-09-09 15:08:12 +0000675
Anders Carlsson1e5dc6e2009-04-11 01:08:03 +0000676 bool Success = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000677
Eli Friedman50c39ea2009-05-27 06:04:58 +0000678 if (DestType->isReferenceType())
679 Success = E->EvaluateAsLValue(Result, Context);
Mike Stump1eb44332009-09-09 15:08:12 +0000680 else
Anders Carlsson1e5dc6e2009-04-11 01:08:03 +0000681 Success = E->Evaluate(Result, Context);
Mike Stump1eb44332009-09-09 15:08:12 +0000682
Anders Carlsson1e5dc6e2009-04-11 01:08:03 +0000683 if (Success) {
Mike Stump1eb44332009-09-09 15:08:12 +0000684 assert(!Result.HasSideEffects &&
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000685 "Constant expr should not have any side effects!");
686 switch (Result.Val.getKind()) {
Daniel Dunbara5fd07b2009-01-28 22:24:07 +0000687 case APValue::Uninitialized:
Daniel Dunbard60f2fb2009-02-17 18:43:32 +0000688 assert(0 && "Constant expressions should be initialized.");
689 return 0;
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000690 case APValue::LValue: {
Anders Carlsson1e5dc6e2009-04-11 01:08:03 +0000691 const llvm::Type *DestTy = getTypes().ConvertTypeForMem(DestType);
Mike Stump1eb44332009-09-09 15:08:12 +0000692 llvm::Constant *Offset =
693 llvm::ConstantInt::get(llvm::Type::getInt64Ty(VMContext),
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000694 Result.Val.getLValueOffset());
Mike Stump1eb44332009-09-09 15:08:12 +0000695
Daniel Dunbare20de512009-02-19 21:44:24 +0000696 llvm::Constant *C;
Anders Carlsson4b3f9c02008-12-01 02:42:14 +0000697 if (const Expr *LVBase = Result.Val.getLValueBase()) {
Daniel Dunbare20de512009-02-19 21:44:24 +0000698 C = ConstExprEmitter(*this, CGF).EmitLValue(const_cast<Expr*>(LVBase));
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000699
Daniel Dunbare20de512009-02-19 21:44:24 +0000700 // Apply offset if necessary.
701 if (!Offset->isNullValue()) {
Mike Stump1eb44332009-09-09 15:08:12 +0000702 const llvm::Type *Type =
Owen Anderson0032b272009-08-13 21:57:51 +0000703 llvm::PointerType::getUnqual(llvm::Type::getInt8Ty(VMContext));
Owen Anderson3c4972d2009-07-29 18:54:39 +0000704 llvm::Constant *Casted = llvm::ConstantExpr::getBitCast(C, Type);
705 Casted = llvm::ConstantExpr::getGetElementPtr(Casted, &Offset, 1);
706 C = llvm::ConstantExpr::getBitCast(Casted, C->getType());
Daniel Dunbare20de512009-02-19 21:44:24 +0000707 }
708
709 // Convert to the appropriate type; this could be an lvalue for
710 // an integer.
Anders Carlsson12e330c2009-04-10 04:59:59 +0000711 if (isa<llvm::PointerType>(DestTy))
Owen Anderson3c4972d2009-07-29 18:54:39 +0000712 return llvm::ConstantExpr::getBitCast(C, DestTy);
Daniel Dunbare20de512009-02-19 21:44:24 +0000713
Owen Anderson3c4972d2009-07-29 18:54:39 +0000714 return llvm::ConstantExpr::getPtrToInt(C, DestTy);
Daniel Dunbare20de512009-02-19 21:44:24 +0000715 } else {
716 C = Offset;
717
718 // Convert to the appropriate type; this could be an lvalue for
719 // an integer.
Anders Carlsson12e330c2009-04-10 04:59:59 +0000720 if (isa<llvm::PointerType>(DestTy))
Owen Anderson3c4972d2009-07-29 18:54:39 +0000721 return llvm::ConstantExpr::getIntToPtr(C, DestTy);
Daniel Dunbare20de512009-02-19 21:44:24 +0000722
723 // If the types don't match this should only be a truncate.
Anders Carlsson12e330c2009-04-10 04:59:59 +0000724 if (C->getType() != DestTy)
Owen Anderson3c4972d2009-07-29 18:54:39 +0000725 return llvm::ConstantExpr::getTrunc(C, DestTy);
Daniel Dunbare20de512009-02-19 21:44:24 +0000726
727 return C;
Anders Carlsson5ea7d072008-11-16 06:23:45 +0000728 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000729 }
Eli Friedman7b30b1c2008-11-17 03:57:28 +0000730 case APValue::Int: {
Mike Stump1eb44332009-09-09 15:08:12 +0000731 llvm::Constant *C = llvm::ConstantInt::get(VMContext,
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000732 Result.Val.getInt());
Mike Stump1eb44332009-09-09 15:08:12 +0000733
Owen Anderson0032b272009-08-13 21:57:51 +0000734 if (C->getType() == llvm::Type::getInt1Ty(VMContext)) {
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000735 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
Owen Anderson3c4972d2009-07-29 18:54:39 +0000736 C = llvm::ConstantExpr::getZExt(C, BoolTy);
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000737 }
738 return C;
Eli Friedman7b30b1c2008-11-17 03:57:28 +0000739 }
Daniel Dunbara5fd07b2009-01-28 22:24:07 +0000740 case APValue::ComplexInt: {
741 llvm::Constant *Complex[2];
Mike Stump1eb44332009-09-09 15:08:12 +0000742
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000743 Complex[0] = llvm::ConstantInt::get(VMContext,
744 Result.Val.getComplexIntReal());
Mike Stump1eb44332009-09-09 15:08:12 +0000745 Complex[1] = llvm::ConstantInt::get(VMContext,
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000746 Result.Val.getComplexIntImag());
Mike Stump1eb44332009-09-09 15:08:12 +0000747
Nick Lewycky0d36dd22009-09-19 20:00:52 +0000748 // FIXME: the target may want to specify that this is packed.
749 return llvm::ConstantStruct::get(VMContext, Complex, 2, false);
Daniel Dunbara5fd07b2009-01-28 22:24:07 +0000750 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000751 case APValue::Float:
Owen Andersonbc0a2222009-07-27 21:00:51 +0000752 return llvm::ConstantFP::get(VMContext, Result.Val.getFloat());
Anders Carlssonaa432562008-11-17 01:58:55 +0000753 case APValue::ComplexFloat: {
754 llvm::Constant *Complex[2];
Mike Stump1eb44332009-09-09 15:08:12 +0000755
756 Complex[0] = llvm::ConstantFP::get(VMContext,
Owen Andersonbc0a2222009-07-27 21:00:51 +0000757 Result.Val.getComplexFloatReal());
758 Complex[1] = llvm::ConstantFP::get(VMContext,
759 Result.Val.getComplexFloatImag());
Mike Stump1eb44332009-09-09 15:08:12 +0000760
Nick Lewycky0d36dd22009-09-19 20:00:52 +0000761 // FIXME: the target may want to specify that this is packed.
762 return llvm::ConstantStruct::get(VMContext, Complex, 2, false);
Anders Carlssonaa432562008-11-17 01:58:55 +0000763 }
Nate Begeman3d309f92009-01-18 01:01:34 +0000764 case APValue::Vector: {
765 llvm::SmallVector<llvm::Constant *, 4> Inits;
766 unsigned NumElts = Result.Val.getVectorLength();
Mike Stump1eb44332009-09-09 15:08:12 +0000767
Nate Begeman3d309f92009-01-18 01:01:34 +0000768 for (unsigned i = 0; i != NumElts; ++i) {
769 APValue &Elt = Result.Val.getVectorElt(i);
770 if (Elt.isInt())
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000771 Inits.push_back(llvm::ConstantInt::get(VMContext, Elt.getInt()));
Nate Begeman3d309f92009-01-18 01:01:34 +0000772 else
Owen Andersonbc0a2222009-07-27 21:00:51 +0000773 Inits.push_back(llvm::ConstantFP::get(VMContext, Elt.getFloat()));
Nate Begeman3d309f92009-01-18 01:01:34 +0000774 }
Owen Anderson4a289322009-07-28 21:22:35 +0000775 return llvm::ConstantVector::get(&Inits[0], Inits.size());
Nate Begeman3d309f92009-01-18 01:01:34 +0000776 }
Anders Carlsson5a9a4572008-11-15 20:45:50 +0000777 }
778 }
Eli Friedman7dfa6392008-06-01 15:31:44 +0000779
780 llvm::Constant* C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
Owen Anderson0032b272009-08-13 21:57:51 +0000781 if (C && C->getType() == llvm::Type::getInt1Ty(VMContext)) {
Eli Friedman7dfa6392008-06-01 15:31:44 +0000782 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
Owen Anderson3c4972d2009-07-29 18:54:39 +0000783 C = llvm::ConstantExpr::getZExt(C, BoolTy);
Eli Friedman7dfa6392008-06-01 15:31:44 +0000784 }
785 return C;
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000786}
Eli Friedman0f593122009-04-13 21:47:26 +0000787
788llvm::Constant *CodeGenModule::EmitNullConstant(QualType T) {
Anders Carlssonad3e7112009-08-24 17:16:23 +0000789 // No need to check for member pointers when not compiling C++.
790 if (!getContext().getLangOptions().CPlusPlus)
791 return llvm::Constant::getNullValue(getTypes().ConvertTypeForMem(T));
Mike Stump1eb44332009-09-09 15:08:12 +0000792
Anders Carlsson237957c2009-08-09 18:26:27 +0000793 if (const ConstantArrayType *CAT = Context.getAsConstantArrayType(T)) {
Mike Stump1eb44332009-09-09 15:08:12 +0000794
Anders Carlsson237957c2009-08-09 18:26:27 +0000795 QualType ElementTy = CAT->getElementType();
796
797 // FIXME: Handle arrays of structs that contain member pointers.
798 if (Context.getBaseElementType(ElementTy)->isMemberPointerType()) {
799 llvm::Constant *Element = EmitNullConstant(ElementTy);
800 uint64_t NumElements = CAT->getSize().getZExtValue();
801 std::vector<llvm::Constant *> Array(NumElements);
802 for (uint64_t i = 0; i != NumElements; ++i)
803 Array[i] = Element;
Mike Stump1eb44332009-09-09 15:08:12 +0000804
805 const llvm::ArrayType *ATy =
Anders Carlsson237957c2009-08-09 18:26:27 +0000806 cast<llvm::ArrayType>(getTypes().ConvertTypeForMem(T));
807 return llvm::ConstantArray::get(ATy, Array);
808 }
809 }
Anders Carlssonfc3eaa42009-08-23 01:25:01 +0000810
811 if (const RecordType *RT = T->getAs<RecordType>()) {
Anders Carlssonad3e7112009-08-24 17:16:23 +0000812 const RecordDecl *RD = RT->getDecl();
Mike Stump1eb44332009-09-09 15:08:12 +0000813 // FIXME: It would be better if there was a way to explicitly compute the
Anders Carlssonad3e7112009-08-24 17:16:23 +0000814 // record layout instead of converting to a type.
815 Types.ConvertTagDeclType(RD);
Mike Stump1eb44332009-09-09 15:08:12 +0000816
Anders Carlssonad3e7112009-08-24 17:16:23 +0000817 const CGRecordLayout &Layout = Types.getCGRecordLayout(RD);
818 if (Layout.containsMemberPointer()) {
Anders Carlssonfc3eaa42009-08-23 01:25:01 +0000819 assert(0 && "FIXME: No support for structs with member pointers yet!");
820 }
Anders Carlssonfc3eaa42009-08-23 01:25:01 +0000821 }
Mike Stump1eb44332009-09-09 15:08:12 +0000822
Anders Carlsson237957c2009-08-09 18:26:27 +0000823 // FIXME: Handle structs that contain member pointers.
Mike Stump1eb44332009-09-09 15:08:12 +0000824 if (T->isMemberPointerType())
Anders Carlsson237957c2009-08-09 18:26:27 +0000825 return llvm::Constant::getAllOnesValue(getTypes().ConvertTypeForMem(T));
Mike Stump1eb44332009-09-09 15:08:12 +0000826
Anders Carlssonb0d0ea02009-08-02 21:18:22 +0000827 return llvm::Constant::getNullValue(getTypes().ConvertTypeForMem(T));
Eli Friedman0f593122009-04-13 21:47:26 +0000828}