blob: 33c5a6124bdcc5592ed29ba3dc3ee01d2fc4b84b [file] [log] [blame]
Chris Lattner9fba49a2007-08-24 05:35:26 +00001//===--- CGExprScalar.cpp - Emit LLVM Code for Scalar Exprs ---------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner959e5be2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattner9fba49a2007-08-24 05:35:26 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This contains code to emit Expr nodes with scalar LLVM types as LLVM code.
11//
12//===----------------------------------------------------------------------===//
13
14#include "CodeGenFunction.h"
15#include "CodeGenModule.h"
Daniel Dunbareee5cd12008-08-11 05:00:27 +000016#include "clang/AST/ASTContext.h"
Daniel Dunbarfa456242008-08-12 05:08:18 +000017#include "clang/AST/DeclObjC.h"
Eli Friedmanccffea92009-01-24 22:38:55 +000018#include "clang/AST/RecordLayout.h"
Daniel Dunbareee5cd12008-08-11 05:00:27 +000019#include "clang/AST/StmtVisitor.h"
Chris Lattnerd54d1f22008-04-20 00:50:39 +000020#include "clang/Basic/TargetInfo.h"
Chris Lattner9fba49a2007-08-24 05:35:26 +000021#include "llvm/Constants.h"
22#include "llvm/Function.h"
Anders Carlsson36f07d82007-10-29 05:01:08 +000023#include "llvm/GlobalVariable.h"
Anders Carlsson36760332007-10-15 20:28:48 +000024#include "llvm/Intrinsics.h"
Chris Lattner9fba49a2007-08-24 05:35:26 +000025#include "llvm/Support/Compiler.h"
Chris Lattner7f80bb32008-11-12 08:38:24 +000026#include "llvm/Support/CFG.h"
Chris Lattnerc2126682008-01-03 07:05:49 +000027#include <cstdarg>
Ted Kremenek03cf4df2007-12-10 23:44:32 +000028
Chris Lattner9fba49a2007-08-24 05:35:26 +000029using namespace clang;
30using namespace CodeGen;
31using llvm::Value;
32
33//===----------------------------------------------------------------------===//
34// Scalar Expression Emitter
35//===----------------------------------------------------------------------===//
36
37struct BinOpInfo {
38 Value *LHS;
39 Value *RHS;
Chris Lattner660e31d2007-08-24 21:00:35 +000040 QualType Ty; // Computation Type.
Chris Lattner9fba49a2007-08-24 05:35:26 +000041 const BinaryOperator *E;
42};
43
44namespace {
45class VISIBILITY_HIDDEN ScalarExprEmitter
46 : public StmtVisitor<ScalarExprEmitter, Value*> {
47 CodeGenFunction &CGF;
Daniel Dunbard916e6e2008-11-01 01:53:16 +000048 CGBuilderTy &Builder;
Chris Lattnercbfb5512008-03-01 08:45:05 +000049
Chris Lattner9fba49a2007-08-24 05:35:26 +000050public:
51
Chris Lattnercbfb5512008-03-01 08:45:05 +000052 ScalarExprEmitter(CodeGenFunction &cgf) : CGF(cgf),
Daniel Dunbarf1f7f192008-08-20 00:28:19 +000053 Builder(CGF.Builder) {
Chris Lattner9fba49a2007-08-24 05:35:26 +000054 }
Chris Lattner9fba49a2007-08-24 05:35:26 +000055
56 //===--------------------------------------------------------------------===//
57 // Utilities
58 //===--------------------------------------------------------------------===//
59
60 const llvm::Type *ConvertType(QualType T) { return CGF.ConvertType(T); }
61 LValue EmitLValue(const Expr *E) { return CGF.EmitLValue(E); }
62
63 Value *EmitLoadOfLValue(LValue LV, QualType T) {
Chris Lattnere24c4cf2007-08-31 22:49:20 +000064 return CGF.EmitLoadOfLValue(LV, T).getScalarVal();
Chris Lattner9fba49a2007-08-24 05:35:26 +000065 }
66
67 /// EmitLoadOfLValue - Given an expression with complex type that represents a
68 /// value l-value, this method emits the address of the l-value, then loads
69 /// and returns the result.
70 Value *EmitLoadOfLValue(const Expr *E) {
71 // FIXME: Volatile
72 return EmitLoadOfLValue(EmitLValue(E), E->getType());
73 }
74
Chris Lattnerd8d44222007-08-26 16:42:57 +000075 /// EmitConversionToBool - Convert the specified expression value to a
Chris Lattner05942062007-08-26 17:25:57 +000076 /// boolean (i1) truth value. This is equivalent to "Val != 0".
Chris Lattnerd8d44222007-08-26 16:42:57 +000077 Value *EmitConversionToBool(Value *Src, QualType DstTy);
78
Chris Lattner4e05d1e2007-08-26 06:48:56 +000079 /// EmitScalarConversion - Emit a conversion from the specified type to the
80 /// specified destination type, both of which are LLVM scalar types.
Chris Lattnerfb182ee2007-08-26 16:34:22 +000081 Value *EmitScalarConversion(Value *Src, QualType SrcTy, QualType DstTy);
82
83 /// EmitComplexToScalarConversion - Emit a conversion from the specified
84 /// complex type to the specified destination type, where the destination
85 /// type is an LLVM scalar type.
86 Value *EmitComplexToScalarConversion(CodeGenFunction::ComplexPairTy Src,
87 QualType SrcTy, QualType DstTy);
Chris Lattner4e05d1e2007-08-26 06:48:56 +000088
Chris Lattner9fba49a2007-08-24 05:35:26 +000089 //===--------------------------------------------------------------------===//
90 // Visitor Methods
91 //===--------------------------------------------------------------------===//
92
93 Value *VisitStmt(Stmt *S) {
Ted Kremenekb3ee1932007-12-11 21:27:55 +000094 S->dump(CGF.getContext().getSourceManager());
Chris Lattner9fba49a2007-08-24 05:35:26 +000095 assert(0 && "Stmt can't have complex result type!");
96 return 0;
97 }
98 Value *VisitExpr(Expr *S);
99 Value *VisitParenExpr(ParenExpr *PE) { return Visit(PE->getSubExpr()); }
100
101 // Leaves.
102 Value *VisitIntegerLiteral(const IntegerLiteral *E) {
103 return llvm::ConstantInt::get(E->getValue());
104 }
105 Value *VisitFloatingLiteral(const FloatingLiteral *E) {
Chris Lattner70c38672008-04-20 00:45:53 +0000106 return llvm::ConstantFP::get(E->getValue());
Chris Lattner9fba49a2007-08-24 05:35:26 +0000107 }
108 Value *VisitCharacterLiteral(const CharacterLiteral *E) {
109 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
110 }
Nate Begemane9bfe6d2007-11-15 05:40:03 +0000111 Value *VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
112 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
113 }
Argiris Kirtzidis750eb972008-08-23 19:35:47 +0000114 Value *VisitCXXZeroInitValueExpr(const CXXZeroInitValueExpr *E) {
115 return llvm::Constant::getNullValue(ConvertType(E->getType()));
116 }
Anders Carlsson774f9c72008-12-21 22:39:40 +0000117 Value *VisitGNUNullExpr(const GNUNullExpr *E) {
118 return llvm::Constant::getNullValue(ConvertType(E->getType()));
119 }
Chris Lattner9fba49a2007-08-24 05:35:26 +0000120 Value *VisitTypesCompatibleExpr(const TypesCompatibleExpr *E) {
121 return llvm::ConstantInt::get(ConvertType(E->getType()),
Steve Naroff85f0dc52007-10-15 20:41:53 +0000122 CGF.getContext().typesAreCompatible(
123 E->getArgType1(), E->getArgType2()));
Chris Lattner9fba49a2007-08-24 05:35:26 +0000124 }
Sebastian Redl0cb7c872008-11-11 17:56:53 +0000125 Value *VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E);
Daniel Dunbar879788d2008-08-04 16:51:22 +0000126 Value *VisitAddrLabelExpr(const AddrLabelExpr *E) {
Daniel Dunbarb5fda0c2008-08-16 01:41:47 +0000127 llvm::Value *V =
128 llvm::ConstantInt::get(llvm::Type::Int32Ty,
129 CGF.GetIDForAddrOfLabel(E->getLabel()));
130
131 return Builder.CreateIntToPtr(V, ConvertType(E->getType()));
Daniel Dunbar879788d2008-08-04 16:51:22 +0000132 }
Chris Lattner9fba49a2007-08-24 05:35:26 +0000133
134 // l-values.
135 Value *VisitDeclRefExpr(DeclRefExpr *E) {
136 if (const EnumConstantDecl *EC = dyn_cast<EnumConstantDecl>(E->getDecl()))
137 return llvm::ConstantInt::get(EC->getInitVal());
138 return EmitLoadOfLValue(E);
139 }
Daniel Dunbar91cc4022008-08-27 06:57:25 +0000140 Value *VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
141 return CGF.EmitObjCSelectorExpr(E);
142 }
143 Value *VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
144 return CGF.EmitObjCProtocolExpr(E);
145 }
146 Value *VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
147 return EmitLoadOfLValue(E);
148 }
Daniel Dunbar5e105892008-08-23 10:51:21 +0000149 Value *VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
Daniel Dunbare6c31752008-08-29 08:11:39 +0000150 return EmitLoadOfLValue(E);
Daniel Dunbar91cc4022008-08-27 06:57:25 +0000151 }
Fariborz Jahanianb0973da2008-11-22 22:30:21 +0000152 Value *VisitObjCKVCRefExpr(ObjCKVCRefExpr *E) {
153 return EmitLoadOfLValue(E);
154 }
Daniel Dunbar91cc4022008-08-27 06:57:25 +0000155 Value *VisitObjCMessageExpr(ObjCMessageExpr *E) {
156 return CGF.EmitObjCMessageExpr(E).getScalarVal();
Daniel Dunbar5e105892008-08-23 10:51:21 +0000157 }
158
Chris Lattner9fba49a2007-08-24 05:35:26 +0000159 Value *VisitArraySubscriptExpr(ArraySubscriptExpr *E);
Eli Friedmand0e9d092008-05-14 19:38:39 +0000160 Value *VisitShuffleVectorExpr(ShuffleVectorExpr *E);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000161 Value *VisitMemberExpr(Expr *E) { return EmitLoadOfLValue(E); }
Nate Begemanaf6ed502008-04-18 23:10:10 +0000162 Value *VisitExtVectorElementExpr(Expr *E) { return EmitLoadOfLValue(E); }
Chris Lattnera9177982008-10-26 23:53:12 +0000163 Value *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
164 return EmitLoadOfLValue(E);
165 }
Chris Lattner9fba49a2007-08-24 05:35:26 +0000166 Value *VisitStringLiteral(Expr *E) { return EmitLValue(E).getAddress(); }
Chris Lattner69909292008-08-10 01:53:14 +0000167 Value *VisitPredefinedExpr(Expr *E) { return EmitLValue(E).getAddress(); }
Devang Patel01ab1302007-10-24 17:18:43 +0000168
169 Value *VisitInitListExpr(InitListExpr *E) {
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000170 unsigned NumInitElements = E->getNumInits();
171
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000172 const llvm::VectorType *VType =
Anders Carlsson35ab4f92008-01-29 01:15:48 +0000173 dyn_cast<llvm::VectorType>(ConvertType(E->getType()));
174
175 // We have a scalar in braces. Just use the first element.
176 if (!VType)
177 return Visit(E->getInit(0));
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000178
Chris Lattnera9177982008-10-26 23:53:12 +0000179 if (E->hadDesignators()) {
180 CGF.ErrorUnsupported(E, "initializer list with designators");
181 return llvm::UndefValue::get(CGF.ConvertType(E->getType()));
182 }
183
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000184 unsigned NumVectorElements = VType->getNumElements();
185 const llvm::Type *ElementType = VType->getElementType();
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000186
187 // Emit individual vector element stores.
188 llvm::Value *V = llvm::UndefValue::get(VType);
189
Anders Carlsson323d5682007-12-18 02:45:33 +0000190 // Emit initializers
191 unsigned i;
192 for (i = 0; i < NumInitElements; ++i) {
Devang Patel32c39832007-10-24 18:05:48 +0000193 Value *NewV = Visit(E->getInit(i));
194 Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, i);
195 V = Builder.CreateInsertElement(V, NewV, Idx);
Devang Patel01ab1302007-10-24 17:18:43 +0000196 }
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000197
198 // Emit remaining default initializers
199 for (/* Do not initialize i*/; i < NumVectorElements; ++i) {
200 Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, i);
201 llvm::Value *NewV = llvm::Constant::getNullValue(ElementType);
202 V = Builder.CreateInsertElement(V, NewV, Idx);
203 }
204
Devang Patel32c39832007-10-24 18:05:48 +0000205 return V;
Devang Patel01ab1302007-10-24 17:18:43 +0000206 }
Chris Lattner3e254fb2008-04-08 04:40:51 +0000207
Chris Lattner9fba49a2007-08-24 05:35:26 +0000208 Value *VisitImplicitCastExpr(const ImplicitCastExpr *E);
209 Value *VisitCastExpr(const CastExpr *E) {
210 return EmitCastExpr(E->getSubExpr(), E->getType());
211 }
212 Value *EmitCastExpr(const Expr *E, QualType T);
213
214 Value *VisitCallExpr(const CallExpr *E) {
Chris Lattnere24c4cf2007-08-31 22:49:20 +0000215 return CGF.EmitCallExpr(E).getScalarVal();
Chris Lattner9fba49a2007-08-24 05:35:26 +0000216 }
Daniel Dunbara04840b2008-08-23 03:46:30 +0000217
Chris Lattnerea6cdd72007-08-31 22:09:40 +0000218 Value *VisitStmtExpr(const StmtExpr *E);
219
Chris Lattner9fba49a2007-08-24 05:35:26 +0000220 // Unary Operators.
221 Value *VisitPrePostIncDec(const UnaryOperator *E, bool isInc, bool isPre);
222 Value *VisitUnaryPostDec(const UnaryOperator *E) {
223 return VisitPrePostIncDec(E, false, false);
224 }
225 Value *VisitUnaryPostInc(const UnaryOperator *E) {
226 return VisitPrePostIncDec(E, true, false);
227 }
228 Value *VisitUnaryPreDec(const UnaryOperator *E) {
229 return VisitPrePostIncDec(E, false, true);
230 }
231 Value *VisitUnaryPreInc(const UnaryOperator *E) {
232 return VisitPrePostIncDec(E, true, true);
233 }
234 Value *VisitUnaryAddrOf(const UnaryOperator *E) {
235 return EmitLValue(E->getSubExpr()).getAddress();
236 }
237 Value *VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); }
238 Value *VisitUnaryPlus(const UnaryOperator *E) {
239 return Visit(E->getSubExpr());
240 }
241 Value *VisitUnaryMinus (const UnaryOperator *E);
242 Value *VisitUnaryNot (const UnaryOperator *E);
243 Value *VisitUnaryLNot (const UnaryOperator *E);
Chris Lattner01211af2007-08-24 21:20:17 +0000244 Value *VisitUnaryReal (const UnaryOperator *E);
245 Value *VisitUnaryImag (const UnaryOperator *E);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000246 Value *VisitUnaryExtension(const UnaryOperator *E) {
247 return Visit(E->getSubExpr());
248 }
Anders Carlsson52774ad2008-01-29 15:56:48 +0000249 Value *VisitUnaryOffsetOf(const UnaryOperator *E);
Chris Lattner3e254fb2008-04-08 04:40:51 +0000250 Value *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
251 return Visit(DAE->getExpr());
252 }
Anders Carlsson52774ad2008-01-29 15:56:48 +0000253
Chris Lattner9fba49a2007-08-24 05:35:26 +0000254 // Binary Operators.
Chris Lattner9fba49a2007-08-24 05:35:26 +0000255 Value *EmitMul(const BinOpInfo &Ops) {
256 return Builder.CreateMul(Ops.LHS, Ops.RHS, "mul");
257 }
258 Value *EmitDiv(const BinOpInfo &Ops);
259 Value *EmitRem(const BinOpInfo &Ops);
260 Value *EmitAdd(const BinOpInfo &Ops);
261 Value *EmitSub(const BinOpInfo &Ops);
262 Value *EmitShl(const BinOpInfo &Ops);
263 Value *EmitShr(const BinOpInfo &Ops);
264 Value *EmitAnd(const BinOpInfo &Ops) {
265 return Builder.CreateAnd(Ops.LHS, Ops.RHS, "and");
266 }
267 Value *EmitXor(const BinOpInfo &Ops) {
268 return Builder.CreateXor(Ops.LHS, Ops.RHS, "xor");
269 }
270 Value *EmitOr (const BinOpInfo &Ops) {
271 return Builder.CreateOr(Ops.LHS, Ops.RHS, "or");
272 }
273
Chris Lattner660e31d2007-08-24 21:00:35 +0000274 BinOpInfo EmitBinOps(const BinaryOperator *E);
Chris Lattner0d965302007-08-26 21:41:21 +0000275 Value *EmitCompoundAssign(const CompoundAssignOperator *E,
Chris Lattner660e31d2007-08-24 21:00:35 +0000276 Value *(ScalarExprEmitter::*F)(const BinOpInfo &));
277
278 // Binary operators and binary compound assignment operators.
279#define HANDLEBINOP(OP) \
Chris Lattner0d965302007-08-26 21:41:21 +0000280 Value *VisitBin ## OP(const BinaryOperator *E) { \
281 return Emit ## OP(EmitBinOps(E)); \
282 } \
283 Value *VisitBin ## OP ## Assign(const CompoundAssignOperator *E) { \
284 return EmitCompoundAssign(E, &ScalarExprEmitter::Emit ## OP); \
Chris Lattner660e31d2007-08-24 21:00:35 +0000285 }
286 HANDLEBINOP(Mul);
287 HANDLEBINOP(Div);
288 HANDLEBINOP(Rem);
289 HANDLEBINOP(Add);
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000290 HANDLEBINOP(Sub);
Chris Lattner660e31d2007-08-24 21:00:35 +0000291 HANDLEBINOP(Shl);
292 HANDLEBINOP(Shr);
293 HANDLEBINOP(And);
294 HANDLEBINOP(Xor);
295 HANDLEBINOP(Or);
296#undef HANDLEBINOP
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000297
Chris Lattner9fba49a2007-08-24 05:35:26 +0000298 // Comparisons.
299 Value *EmitCompare(const BinaryOperator *E, unsigned UICmpOpc,
300 unsigned SICmpOpc, unsigned FCmpOpc);
301#define VISITCOMP(CODE, UI, SI, FP) \
302 Value *VisitBin##CODE(const BinaryOperator *E) { \
303 return EmitCompare(E, llvm::ICmpInst::UI, llvm::ICmpInst::SI, \
304 llvm::FCmpInst::FP); }
305 VISITCOMP(LT, ICMP_ULT, ICMP_SLT, FCMP_OLT);
306 VISITCOMP(GT, ICMP_UGT, ICMP_SGT, FCMP_OGT);
307 VISITCOMP(LE, ICMP_ULE, ICMP_SLE, FCMP_OLE);
308 VISITCOMP(GE, ICMP_UGE, ICMP_SGE, FCMP_OGE);
309 VISITCOMP(EQ, ICMP_EQ , ICMP_EQ , FCMP_OEQ);
310 VISITCOMP(NE, ICMP_NE , ICMP_NE , FCMP_UNE);
311#undef VISITCOMP
312
313 Value *VisitBinAssign (const BinaryOperator *E);
314
315 Value *VisitBinLAnd (const BinaryOperator *E);
316 Value *VisitBinLOr (const BinaryOperator *E);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000317 Value *VisitBinComma (const BinaryOperator *E);
318
319 // Other Operators.
Daniel Dunbarbf5eb6c2009-01-09 17:04:29 +0000320 Value *VisitBlockExpr(const BlockExpr *BE) {
321 CGF.ErrorUnsupported(BE, "block expression");
322 return llvm::UndefValue::get(CGF.ConvertType(BE->getType()));
323 }
324
Chris Lattner9fba49a2007-08-24 05:35:26 +0000325 Value *VisitConditionalOperator(const ConditionalOperator *CO);
326 Value *VisitChooseExpr(ChooseExpr *CE);
Nate Begeman9f3bfb72008-01-17 17:46:27 +0000327 Value *VisitOverloadExpr(OverloadExpr *OE);
Anders Carlsson36760332007-10-15 20:28:48 +0000328 Value *VisitVAArgExpr(VAArgExpr *VE);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000329 Value *VisitObjCStringLiteral(const ObjCStringLiteral *E) {
330 return CGF.EmitObjCStringLiteral(E);
331 }
Anders Carlsson36f07d82007-10-29 05:01:08 +0000332 Value *VisitObjCEncodeExpr(const ObjCEncodeExpr *E);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000333};
334} // end anonymous namespace.
335
336//===----------------------------------------------------------------------===//
337// Utilities
338//===----------------------------------------------------------------------===//
339
Chris Lattnerd8d44222007-08-26 16:42:57 +0000340/// EmitConversionToBool - Convert the specified expression value to a
Chris Lattner05942062007-08-26 17:25:57 +0000341/// boolean (i1) truth value. This is equivalent to "Val != 0".
Chris Lattnerd8d44222007-08-26 16:42:57 +0000342Value *ScalarExprEmitter::EmitConversionToBool(Value *Src, QualType SrcType) {
343 assert(SrcType->isCanonical() && "EmitScalarConversion strips typedefs");
344
345 if (SrcType->isRealFloatingType()) {
346 // Compare against 0.0 for fp scalars.
347 llvm::Value *Zero = llvm::Constant::getNullValue(Src->getType());
Chris Lattnerd8d44222007-08-26 16:42:57 +0000348 return Builder.CreateFCmpUNE(Src, Zero, "tobool");
349 }
350
Daniel Dunbar5d54eed2008-08-25 10:38:11 +0000351 assert((SrcType->isIntegerType() || isa<llvm::PointerType>(Src->getType())) &&
Chris Lattnerd8d44222007-08-26 16:42:57 +0000352 "Unknown scalar type to convert");
353
354 // Because of the type rules of C, we often end up computing a logical value,
355 // then zero extending it to int, then wanting it as a logical value again.
356 // Optimize this common case.
357 if (llvm::ZExtInst *ZI = dyn_cast<llvm::ZExtInst>(Src)) {
358 if (ZI->getOperand(0)->getType() == llvm::Type::Int1Ty) {
359 Value *Result = ZI->getOperand(0);
Eli Friedman24f33972008-01-29 18:13:51 +0000360 // If there aren't any more uses, zap the instruction to save space.
361 // Note that there can be more uses, for example if this
362 // is the result of an assignment.
363 if (ZI->use_empty())
364 ZI->eraseFromParent();
Chris Lattnerd8d44222007-08-26 16:42:57 +0000365 return Result;
366 }
367 }
368
369 // Compare against an integer or pointer null.
370 llvm::Value *Zero = llvm::Constant::getNullValue(Src->getType());
371 return Builder.CreateICmpNE(Src, Zero, "tobool");
372}
373
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000374/// EmitScalarConversion - Emit a conversion from the specified type to the
375/// specified destination type, both of which are LLVM scalar types.
Chris Lattnerfb182ee2007-08-26 16:34:22 +0000376Value *ScalarExprEmitter::EmitScalarConversion(Value *Src, QualType SrcType,
377 QualType DstType) {
Chris Lattnerc154ac12008-07-26 22:37:01 +0000378 SrcType = CGF.getContext().getCanonicalType(SrcType);
379 DstType = CGF.getContext().getCanonicalType(DstType);
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000380 if (SrcType == DstType) return Src;
Chris Lattnere133d7f2007-08-26 07:21:11 +0000381
382 if (DstType->isVoidType()) return 0;
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000383
384 // Handle conversions to bool first, they are special: comparisons against 0.
Chris Lattnerc39c3652007-08-26 16:52:28 +0000385 if (DstType->isBooleanType())
386 return EmitConversionToBool(Src, SrcType);
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000387
388 const llvm::Type *DstTy = ConvertType(DstType);
389
390 // Ignore conversions like int -> uint.
391 if (Src->getType() == DstTy)
392 return Src;
393
Daniel Dunbar238335f2008-08-25 09:51:32 +0000394 // Handle pointer conversions next: pointers can only be converted
395 // to/from other pointers and integers. Check for pointer types in
396 // terms of LLVM, as some native types (like Obj-C id) may map to a
397 // pointer type.
398 if (isa<llvm::PointerType>(DstTy)) {
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000399 // The source value may be an integer, or a pointer.
400 if (isa<llvm::PointerType>(Src->getType()))
401 return Builder.CreateBitCast(Src, DstTy, "conv");
402 assert(SrcType->isIntegerType() && "Not ptr->ptr or int->ptr conversion?");
403 return Builder.CreateIntToPtr(Src, DstTy, "conv");
404 }
405
Daniel Dunbar238335f2008-08-25 09:51:32 +0000406 if (isa<llvm::PointerType>(Src->getType())) {
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000407 // Must be an ptr to int cast.
408 assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
Anders Carlsson44db38f2007-10-31 23:18:02 +0000409 return Builder.CreatePtrToInt(Src, DstTy, "conv");
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000410 }
411
Nate Begemanaf6ed502008-04-18 23:10:10 +0000412 // A scalar can be splatted to an extended vector of the same element type
Nate Begeman7903d052009-01-18 06:42:49 +0000413 if (DstType->isExtVectorType() && !isa<VectorType>(SrcType)) {
414 // Cast the scalar to element type
415 QualType EltTy = DstType->getAsExtVectorType()->getElementType();
416 llvm::Value *Elt = EmitScalarConversion(Src, SrcType, EltTy);
417
418 // Insert the element in element zero of an undef vector
419 llvm::Value *UnV = llvm::UndefValue::get(DstTy);
420 llvm::Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
421 UnV = Builder.CreateInsertElement(UnV, Elt, Idx, "tmp");
422
423 // Splat the element across to all elements
424 llvm::SmallVector<llvm::Constant*, 16> Args;
425 unsigned NumElements = cast<llvm::VectorType>(DstTy)->getNumElements();
426 for (unsigned i = 0; i < NumElements; i++)
427 Args.push_back(llvm::ConstantInt::get(llvm::Type::Int32Ty, 0));
428
429 llvm::Constant *Mask = llvm::ConstantVector::get(&Args[0], NumElements);
430 llvm::Value *Yay = Builder.CreateShuffleVector(UnV, UnV, Mask, "splat");
431 return Yay;
432 }
Nate Begemanec2d1062007-12-30 02:59:45 +0000433
Chris Lattner4f025a42008-02-02 04:51:41 +0000434 // Allow bitcast from vector to integer/fp of the same size.
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000435 if (isa<llvm::VectorType>(Src->getType()) ||
Chris Lattner4f025a42008-02-02 04:51:41 +0000436 isa<llvm::VectorType>(DstTy))
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000437 return Builder.CreateBitCast(Src, DstTy, "conv");
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000438
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000439 // Finally, we have the arithmetic types: real int/float.
440 if (isa<llvm::IntegerType>(Src->getType())) {
441 bool InputSigned = SrcType->isSignedIntegerType();
Anders Carlsson4dac3f42007-12-26 18:20:19 +0000442 if (isa<llvm::IntegerType>(DstTy))
443 return Builder.CreateIntCast(Src, DstTy, InputSigned, "conv");
444 else if (InputSigned)
445 return Builder.CreateSIToFP(Src, DstTy, "conv");
446 else
447 return Builder.CreateUIToFP(Src, DstTy, "conv");
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000448 }
449
450 assert(Src->getType()->isFloatingPoint() && "Unknown real conversion");
451 if (isa<llvm::IntegerType>(DstTy)) {
Anders Carlsson4dac3f42007-12-26 18:20:19 +0000452 if (DstType->isSignedIntegerType())
453 return Builder.CreateFPToSI(Src, DstTy, "conv");
454 else
455 return Builder.CreateFPToUI(Src, DstTy, "conv");
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000456 }
457
458 assert(DstTy->isFloatingPoint() && "Unknown real conversion");
Anders Carlsson4dac3f42007-12-26 18:20:19 +0000459 if (DstTy->getTypeID() < Src->getType()->getTypeID())
460 return Builder.CreateFPTrunc(Src, DstTy, "conv");
461 else
462 return Builder.CreateFPExt(Src, DstTy, "conv");
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000463}
464
Chris Lattnerfb182ee2007-08-26 16:34:22 +0000465/// EmitComplexToScalarConversion - Emit a conversion from the specified
466/// complex type to the specified destination type, where the destination
467/// type is an LLVM scalar type.
468Value *ScalarExprEmitter::
469EmitComplexToScalarConversion(CodeGenFunction::ComplexPairTy Src,
470 QualType SrcTy, QualType DstTy) {
Chris Lattnerc39c3652007-08-26 16:52:28 +0000471 // Get the source element type.
Chris Lattnerc154ac12008-07-26 22:37:01 +0000472 SrcTy = SrcTy->getAsComplexType()->getElementType();
Chris Lattnerc39c3652007-08-26 16:52:28 +0000473
474 // Handle conversions to bool first, they are special: comparisons against 0.
475 if (DstTy->isBooleanType()) {
476 // Complex != 0 -> (Real != 0) | (Imag != 0)
477 Src.first = EmitScalarConversion(Src.first, SrcTy, DstTy);
478 Src.second = EmitScalarConversion(Src.second, SrcTy, DstTy);
479 return Builder.CreateOr(Src.first, Src.second, "tobool");
480 }
481
Chris Lattnerfb182ee2007-08-26 16:34:22 +0000482 // C99 6.3.1.7p2: "When a value of complex type is converted to a real type,
483 // the imaginary part of the complex value is discarded and the value of the
484 // real part is converted according to the conversion rules for the
485 // corresponding real type.
Chris Lattnerfb182ee2007-08-26 16:34:22 +0000486 return EmitScalarConversion(Src.first, SrcTy, DstTy);
487}
488
489
Chris Lattner9fba49a2007-08-24 05:35:26 +0000490//===----------------------------------------------------------------------===//
491// Visitor Methods
492//===----------------------------------------------------------------------===//
493
494Value *ScalarExprEmitter::VisitExpr(Expr *E) {
Daniel Dunbar9503b782008-08-16 00:56:44 +0000495 CGF.ErrorUnsupported(E, "scalar expression");
Chris Lattner9fba49a2007-08-24 05:35:26 +0000496 if (E->getType()->isVoidType())
497 return 0;
498 return llvm::UndefValue::get(CGF.ConvertType(E->getType()));
499}
500
Eli Friedmand0e9d092008-05-14 19:38:39 +0000501Value *ScalarExprEmitter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
502 llvm::SmallVector<llvm::Constant*, 32> indices;
503 for (unsigned i = 2; i < E->getNumSubExprs(); i++) {
504 indices.push_back(cast<llvm::Constant>(CGF.EmitScalarExpr(E->getExpr(i))));
505 }
506 Value* V1 = CGF.EmitScalarExpr(E->getExpr(0));
507 Value* V2 = CGF.EmitScalarExpr(E->getExpr(1));
508 Value* SV = llvm::ConstantVector::get(indices.begin(), indices.size());
509 return Builder.CreateShuffleVector(V1, V2, SV, "shuffle");
510}
511
Chris Lattner9fba49a2007-08-24 05:35:26 +0000512Value *ScalarExprEmitter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
513 // Emit subscript expressions in rvalue context's. For most cases, this just
514 // loads the lvalue formed by the subscript expr. However, we have to be
515 // careful, because the base of a vector subscript is occasionally an rvalue,
516 // so we can't get it as an lvalue.
517 if (!E->getBase()->getType()->isVectorType())
518 return EmitLoadOfLValue(E);
519
520 // Handle the vector case. The base must be a vector, the index must be an
521 // integer value.
522 Value *Base = Visit(E->getBase());
523 Value *Idx = Visit(E->getIdx());
524
525 // FIXME: Convert Idx to i32 type.
526 return Builder.CreateExtractElement(Base, Idx, "vecext");
527}
528
529/// VisitImplicitCastExpr - Implicit casts are the same as normal casts, but
530/// also handle things like function to pointer-to-function decay, and array to
531/// pointer decay.
532Value *ScalarExprEmitter::VisitImplicitCastExpr(const ImplicitCastExpr *E) {
533 const Expr *Op = E->getSubExpr();
534
535 // If this is due to array->pointer conversion, emit the array expression as
536 // an l-value.
537 if (Op->getType()->isArrayType()) {
538 // FIXME: For now we assume that all source arrays map to LLVM arrays. This
539 // will not true when we add support for VLAs.
Chris Lattnerfb182ee2007-08-26 16:34:22 +0000540 Value *V = EmitLValue(Op).getAddress(); // Bitfields can't be arrays.
Eli Friedman8fef47e2008-12-20 23:11:59 +0000541
542 if (!Op->getType()->isVariableArrayType()) {
543 assert(isa<llvm::PointerType>(V->getType()) && "Expected pointer");
544 assert(isa<llvm::ArrayType>(cast<llvm::PointerType>(V->getType())
545 ->getElementType()) &&
546 "Expected pointer to array");
547 V = Builder.CreateStructGEP(V, 0, "arraydecay");
Daniel Dunbar952f4732008-08-29 17:28:43 +0000548 }
Chris Lattnere54443b2007-12-12 04:13:20 +0000549
550 // The resultant pointer type can be implicitly casted to other pointer
Chris Lattner3b8f5c62008-07-23 06:31:27 +0000551 // types as well (e.g. void*) and can be implicitly converted to integer.
552 const llvm::Type *DestTy = ConvertType(E->getType());
553 if (V->getType() != DestTy) {
554 if (isa<llvm::PointerType>(DestTy))
555 V = Builder.CreateBitCast(V, DestTy, "ptrconv");
556 else {
557 assert(isa<llvm::IntegerType>(DestTy) && "Unknown array decay");
558 V = Builder.CreatePtrToInt(V, DestTy, "ptrconv");
559 }
560 }
Chris Lattnere54443b2007-12-12 04:13:20 +0000561 return V;
562
Anders Carlssoncebb8d62007-10-12 23:56:29 +0000563 } else if (E->getType()->isReferenceType()) {
Anders Carlssoncebb8d62007-10-12 23:56:29 +0000564 return EmitLValue(Op).getAddress();
Chris Lattner9fba49a2007-08-24 05:35:26 +0000565 }
566
567 return EmitCastExpr(Op, E->getType());
568}
569
570
571// VisitCastExpr - Emit code for an explicit or implicit cast. Implicit casts
572// have to handle a more broad range of conversions than explicit casts, as they
573// handle things like function to ptr-to-function decay etc.
574Value *ScalarExprEmitter::EmitCastExpr(const Expr *E, QualType DestTy) {
Chris Lattner82e10392007-08-26 07:26:12 +0000575 // Handle cases where the source is an non-complex type.
Chris Lattner77288792008-02-16 23:55:16 +0000576
577 if (!CGF.hasAggregateLLVMType(E->getType())) {
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000578 Value *Src = Visit(const_cast<Expr*>(E));
579
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000580 // Use EmitScalarConversion to perform the conversion.
581 return EmitScalarConversion(Src, E->getType(), DestTy);
582 }
Chris Lattner77288792008-02-16 23:55:16 +0000583
Chris Lattnerde0908b2008-04-04 16:54:41 +0000584 if (E->getType()->isAnyComplexType()) {
Chris Lattner77288792008-02-16 23:55:16 +0000585 // Handle cases where the source is a complex type.
586 return EmitComplexToScalarConversion(CGF.EmitComplexExpr(E), E->getType(),
587 DestTy);
588 }
Chris Lattnerd579f7f2007-08-26 07:16:41 +0000589
Chris Lattner77288792008-02-16 23:55:16 +0000590 // Okay, this is a cast from an aggregate. It must be a cast to void. Just
591 // evaluate the result and return.
592 CGF.EmitAggExpr(E, 0, false);
593 return 0;
Chris Lattner9fba49a2007-08-24 05:35:26 +0000594}
595
Chris Lattnerea6cdd72007-08-31 22:09:40 +0000596Value *ScalarExprEmitter::VisitStmtExpr(const StmtExpr *E) {
Chris Lattner09cee852008-07-26 20:23:23 +0000597 return CGF.EmitCompoundStmt(*E->getSubStmt(),
598 !E->getType()->isVoidType()).getScalarVal();
Chris Lattnerea6cdd72007-08-31 22:09:40 +0000599}
600
601
Chris Lattner9fba49a2007-08-24 05:35:26 +0000602//===----------------------------------------------------------------------===//
603// Unary Operators
604//===----------------------------------------------------------------------===//
605
606Value *ScalarExprEmitter::VisitPrePostIncDec(const UnaryOperator *E,
Chris Lattner855e3d72007-08-24 16:24:49 +0000607 bool isInc, bool isPre) {
Chris Lattner9fba49a2007-08-24 05:35:26 +0000608 LValue LV = EmitLValue(E->getSubExpr());
609 // FIXME: Handle volatile!
Chris Lattner0dc11f62007-08-26 05:10:16 +0000610 Value *InVal = CGF.EmitLoadOfLValue(LV, // false
Chris Lattnere24c4cf2007-08-31 22:49:20 +0000611 E->getSubExpr()->getType()).getScalarVal();
Chris Lattner9fba49a2007-08-24 05:35:26 +0000612
613 int AmountVal = isInc ? 1 : -1;
614
615 Value *NextVal;
Chris Lattner0dc11f62007-08-26 05:10:16 +0000616 if (isa<llvm::PointerType>(InVal->getType())) {
617 // FIXME: This isn't right for VLAs.
618 NextVal = llvm::ConstantInt::get(llvm::Type::Int32Ty, AmountVal);
Chris Lattner07307562008-03-19 05:19:41 +0000619 NextVal = Builder.CreateGEP(InVal, NextVal, "ptrincdec");
Chris Lattner0dc11f62007-08-26 05:10:16 +0000620 } else {
621 // Add the inc/dec to the real part.
622 if (isa<llvm::IntegerType>(InVal->getType()))
623 NextVal = llvm::ConstantInt::get(InVal->getType(), AmountVal);
Chris Lattnerb2a7dab2007-09-13 06:19:18 +0000624 else if (InVal->getType() == llvm::Type::FloatTy)
Devang Patel0f2a8fb2007-10-30 20:59:40 +0000625 NextVal =
Chris Lattner70c38672008-04-20 00:45:53 +0000626 llvm::ConstantFP::get(llvm::APFloat(static_cast<float>(AmountVal)));
Chris Lattnerd54d1f22008-04-20 00:50:39 +0000627 else if (InVal->getType() == llvm::Type::DoubleTy)
Devang Patel0f2a8fb2007-10-30 20:59:40 +0000628 NextVal =
Chris Lattner70c38672008-04-20 00:45:53 +0000629 llvm::ConstantFP::get(llvm::APFloat(static_cast<double>(AmountVal)));
Chris Lattnerd54d1f22008-04-20 00:50:39 +0000630 else {
631 llvm::APFloat F(static_cast<float>(AmountVal));
Dale Johannesen2461f612008-10-09 23:02:32 +0000632 bool ignored;
633 F.convert(CGF.Target.getLongDoubleFormat(), llvm::APFloat::rmTowardZero,
634 &ignored);
Chris Lattnerd54d1f22008-04-20 00:50:39 +0000635 NextVal = llvm::ConstantFP::get(F);
Chris Lattnerb2a7dab2007-09-13 06:19:18 +0000636 }
Chris Lattner0dc11f62007-08-26 05:10:16 +0000637 NextVal = Builder.CreateAdd(InVal, NextVal, isInc ? "inc" : "dec");
638 }
Chris Lattner9fba49a2007-08-24 05:35:26 +0000639
640 // Store the updated result through the lvalue.
641 CGF.EmitStoreThroughLValue(RValue::get(NextVal), LV,
642 E->getSubExpr()->getType());
643
644 // If this is a postinc, return the value read from memory, otherwise use the
645 // updated value.
646 return isPre ? NextVal : InVal;
647}
648
649
650Value *ScalarExprEmitter::VisitUnaryMinus(const UnaryOperator *E) {
651 Value *Op = Visit(E->getSubExpr());
652 return Builder.CreateNeg(Op, "neg");
653}
654
655Value *ScalarExprEmitter::VisitUnaryNot(const UnaryOperator *E) {
656 Value *Op = Visit(E->getSubExpr());
657 return Builder.CreateNot(Op, "neg");
658}
659
660Value *ScalarExprEmitter::VisitUnaryLNot(const UnaryOperator *E) {
661 // Compare operand to zero.
662 Value *BoolVal = CGF.EvaluateExprAsBool(E->getSubExpr());
663
664 // Invert value.
665 // TODO: Could dynamically modify easy computations here. For example, if
666 // the operand is an icmp ne, turn into icmp eq.
667 BoolVal = Builder.CreateNot(BoolVal, "lnot");
668
669 // ZExt result to int.
670 return Builder.CreateZExt(BoolVal, CGF.LLVMIntTy, "lnot.ext");
671}
672
Sebastian Redl0cb7c872008-11-11 17:56:53 +0000673/// VisitSizeOfAlignOfExpr - Return the size or alignment of the type of
674/// argument of the sizeof expression as an integer.
675Value *
676ScalarExprEmitter::VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E) {
Sebastian Redl0cb7c872008-11-11 17:56:53 +0000677 QualType TypeToSize = E->getTypeOfArgument();
Eli Friedman5a2c38f2009-01-24 22:19:05 +0000678 if (E->isSizeOf()) {
679 if (const VariableArrayType *VAT =
680 CGF.getContext().getAsVariableArrayType(TypeToSize)) {
681 if (E->isArgumentType()) {
682 // sizeof(type) - make sure to emit the VLA size.
683 CGF.EmitVLASize(TypeToSize);
684 }
685 return CGF.GetVLASize(VAT);
Anders Carlsson6cb99b72008-12-21 03:33:21 +0000686 }
Anders Carlsson9be6aaf2008-12-12 07:38:43 +0000687 }
Eli Friedman5a2c38f2009-01-24 22:19:05 +0000688
689 // If this isn't sizeof(vla), the result must be constant; use the
690 // constant folding logic so we don't have to duplicate it here.
691 Expr::EvalResult Result;
692 E->Evaluate(Result, CGF.getContext());
693 return llvm::ConstantInt::get(Result.Val.getInt());
Chris Lattner9fba49a2007-08-24 05:35:26 +0000694}
695
Chris Lattner01211af2007-08-24 21:20:17 +0000696Value *ScalarExprEmitter::VisitUnaryReal(const UnaryOperator *E) {
697 Expr *Op = E->getSubExpr();
Chris Lattnerde0908b2008-04-04 16:54:41 +0000698 if (Op->getType()->isAnyComplexType())
Chris Lattner01211af2007-08-24 21:20:17 +0000699 return CGF.EmitComplexExpr(Op).first;
700 return Visit(Op);
701}
702Value *ScalarExprEmitter::VisitUnaryImag(const UnaryOperator *E) {
703 Expr *Op = E->getSubExpr();
Chris Lattnerde0908b2008-04-04 16:54:41 +0000704 if (Op->getType()->isAnyComplexType())
Chris Lattner01211af2007-08-24 21:20:17 +0000705 return CGF.EmitComplexExpr(Op).second;
Chris Lattnerdb8a6c92007-08-26 05:29:21 +0000706
707 // __imag on a scalar returns zero. Emit it the subexpr to ensure side
708 // effects are evaluated.
709 CGF.EmitScalarExpr(Op);
710 return llvm::Constant::getNullValue(ConvertType(E->getType()));
Chris Lattner01211af2007-08-24 21:20:17 +0000711}
712
Anders Carlsson52774ad2008-01-29 15:56:48 +0000713Value *ScalarExprEmitter::VisitUnaryOffsetOf(const UnaryOperator *E)
714{
Eli Friedmanccffea92009-01-24 22:38:55 +0000715 const Expr* SubExpr = E->getSubExpr();
716 const llvm::Type* ResultType = ConvertType(E->getType());
717 llvm::Value* Result = llvm::Constant::getNullValue(ResultType);
718 while (!isa<CompoundLiteralExpr>(SubExpr)) {
719 if (const MemberExpr *ME = dyn_cast<MemberExpr>(SubExpr)) {
720 SubExpr = ME->getBase();
721 QualType Ty = SubExpr->getType();
722
723 RecordDecl *RD = Ty->getAsRecordType()->getDecl();
724 const ASTRecordLayout &RL = CGF.getContext().getASTRecordLayout(RD);
725 FieldDecl *FD = cast<FieldDecl>(ME->getMemberDecl());
726
727 // FIXME: This is linear time. And the fact that we're indexing
728 // into the layout by position in the record means that we're
729 // either stuck numbering the fields in the AST or we have to keep
730 // the linear search (yuck and yuck).
731 unsigned i = 0;
732 for (RecordDecl::field_iterator Field = RD->field_begin(),
733 FieldEnd = RD->field_end();
734 Field != FieldEnd; (void)++Field, ++i) {
735 if (*Field == FD)
736 break;
737 }
738
739 llvm::Value* Offset =
740 llvm::ConstantInt::get(ResultType, RL.getFieldOffset(i) / 8);
741 Result = Builder.CreateAdd(Result, Offset);
742 } else if (const ArraySubscriptExpr *ASE = dyn_cast<ArraySubscriptExpr>(SubExpr)) {
743 SubExpr = ASE->getBase();
744 int64_t size = CGF.getContext().getTypeSize(ASE->getType()) / 8;
745 llvm::Value* ElemSize = llvm::ConstantInt::get(ResultType, size);
746 llvm::Value* ElemIndex = CGF.EmitScalarExpr(ASE->getIdx());
747 bool IndexSigned = ASE->getIdx()->getType()->isSignedIntegerType();
748 ElemIndex = Builder.CreateIntCast(ElemIndex, ResultType, IndexSigned);
749 llvm::Value* Offset = Builder.CreateMul(ElemSize, ElemIndex);
750 Result = Builder.CreateAdd(Result, Offset);
751 } else {
752 assert(0 && "This should be impossible!");
753 }
754 }
755 return Result;
Anders Carlsson52774ad2008-01-29 15:56:48 +0000756}
Chris Lattner01211af2007-08-24 21:20:17 +0000757
Chris Lattner9fba49a2007-08-24 05:35:26 +0000758//===----------------------------------------------------------------------===//
759// Binary Operators
760//===----------------------------------------------------------------------===//
761
762BinOpInfo ScalarExprEmitter::EmitBinOps(const BinaryOperator *E) {
763 BinOpInfo Result;
764 Result.LHS = Visit(E->getLHS());
765 Result.RHS = Visit(E->getRHS());
Chris Lattner660e31d2007-08-24 21:00:35 +0000766 Result.Ty = E->getType();
Chris Lattner9fba49a2007-08-24 05:35:26 +0000767 Result.E = E;
768 return Result;
769}
770
Chris Lattner0d965302007-08-26 21:41:21 +0000771Value *ScalarExprEmitter::EmitCompoundAssign(const CompoundAssignOperator *E,
Chris Lattner660e31d2007-08-24 21:00:35 +0000772 Value *(ScalarExprEmitter::*Func)(const BinOpInfo &)) {
773 QualType LHSTy = E->getLHS()->getType(), RHSTy = E->getRHS()->getType();
774
775 BinOpInfo OpInfo;
776
777 // Load the LHS and RHS operands.
778 LValue LHSLV = EmitLValue(E->getLHS());
779 OpInfo.LHS = EmitLoadOfLValue(LHSLV, LHSTy);
Chris Lattner9c9f4bb2007-08-26 22:37:40 +0000780
781 // Determine the computation type. If the RHS is complex, then this is one of
782 // the add/sub/mul/div operators. All of these operators can be computed in
783 // with just their real component even though the computation domain really is
784 // complex.
Chris Lattner0d965302007-08-26 21:41:21 +0000785 QualType ComputeType = E->getComputationType();
Chris Lattner660e31d2007-08-24 21:00:35 +0000786
Chris Lattner9c9f4bb2007-08-26 22:37:40 +0000787 // If the computation type is complex, then the RHS is complex. Emit the RHS.
788 if (const ComplexType *CT = ComputeType->getAsComplexType()) {
789 ComputeType = CT->getElementType();
790
791 // Emit the RHS, only keeping the real component.
792 OpInfo.RHS = CGF.EmitComplexExpr(E->getRHS()).first;
793 RHSTy = RHSTy->getAsComplexType()->getElementType();
794 } else {
795 // Otherwise the RHS is a simple scalar value.
796 OpInfo.RHS = Visit(E->getRHS());
797 }
798
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000799 QualType LComputeTy, RComputeTy, ResultTy;
800
801 // Compound assignment does not contain enough information about all
802 // the types involved for pointer arithmetic cases. Figure it out
803 // here for now.
804 if (E->getLHS()->getType()->isPointerType()) {
805 // Pointer arithmetic cases: ptr +=,-= int and ptr -= ptr,
806 assert((E->getOpcode() == BinaryOperator::AddAssign ||
807 E->getOpcode() == BinaryOperator::SubAssign) &&
808 "Invalid compound assignment operator on pointer type.");
809 LComputeTy = E->getLHS()->getType();
810
811 if (E->getRHS()->getType()->isPointerType()) {
812 // Degenerate case of (ptr -= ptr) allowed by GCC implicit cast
813 // extension, the conversion from the pointer difference back to
814 // the LHS type is handled at the end.
815 assert(E->getOpcode() == BinaryOperator::SubAssign &&
816 "Invalid compound assignment operator on pointer type.");
817 RComputeTy = E->getLHS()->getType();
818 ResultTy = CGF.getContext().getPointerDiffType();
819 } else {
820 RComputeTy = E->getRHS()->getType();
821 ResultTy = LComputeTy;
822 }
823 } else if (E->getRHS()->getType()->isPointerType()) {
824 // Degenerate case of (int += ptr) allowed by GCC implicit cast
825 // extension.
826 assert(E->getOpcode() == BinaryOperator::AddAssign &&
827 "Invalid compound assignment operator on pointer type.");
828 LComputeTy = E->getLHS()->getType();
829 RComputeTy = E->getRHS()->getType();
830 ResultTy = RComputeTy;
831 } else {
832 LComputeTy = RComputeTy = ResultTy = ComputeType;
Chris Lattner660e31d2007-08-24 21:00:35 +0000833 }
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000834
835 // Convert the LHS/RHS values to the computation type.
836 OpInfo.LHS = EmitScalarConversion(OpInfo.LHS, LHSTy, LComputeTy);
837 OpInfo.RHS = EmitScalarConversion(OpInfo.RHS, RHSTy, RComputeTy);
838 OpInfo.Ty = ResultTy;
Chris Lattner660e31d2007-08-24 21:00:35 +0000839 OpInfo.E = E;
840
841 // Expand the binary operator.
842 Value *Result = (this->*Func)(OpInfo);
843
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000844 // Convert the result back to the LHS type.
845 Result = EmitScalarConversion(Result, ResultTy, LHSTy);
Chris Lattner660e31d2007-08-24 21:00:35 +0000846
Daniel Dunbar2668dd12008-11-19 09:36:46 +0000847 // Store the result value into the LHS lvalue. Bit-fields are
Daniel Dunbar2710fc92008-11-19 11:54:05 +0000848 // handled specially because the result is altered by the store,
849 // i.e., [C99 6.5.16p1] 'An assignment expression has the value of
850 // the left operand after the assignment...'.
Eli Friedmanf9b930c2008-05-25 14:13:57 +0000851 if (LHSLV.isBitfield())
Daniel Dunbar2668dd12008-11-19 09:36:46 +0000852 CGF.EmitStoreThroughBitfieldLValue(RValue::get(Result), LHSLV, LHSTy,
853 &Result);
854 else
855 CGF.EmitStoreThroughLValue(RValue::get(Result), LHSLV, LHSTy);
856
Chris Lattner660e31d2007-08-24 21:00:35 +0000857 return Result;
858}
859
860
Chris Lattner9fba49a2007-08-24 05:35:26 +0000861Value *ScalarExprEmitter::EmitDiv(const BinOpInfo &Ops) {
Nate Begemanaade3bf2007-12-30 01:28:16 +0000862 if (Ops.LHS->getType()->isFPOrFPVector())
Chris Lattner9fba49a2007-08-24 05:35:26 +0000863 return Builder.CreateFDiv(Ops.LHS, Ops.RHS, "div");
Chris Lattner660e31d2007-08-24 21:00:35 +0000864 else if (Ops.Ty->isUnsignedIntegerType())
Chris Lattner9fba49a2007-08-24 05:35:26 +0000865 return Builder.CreateUDiv(Ops.LHS, Ops.RHS, "div");
866 else
867 return Builder.CreateSDiv(Ops.LHS, Ops.RHS, "div");
868}
869
870Value *ScalarExprEmitter::EmitRem(const BinOpInfo &Ops) {
871 // Rem in C can't be a floating point type: C99 6.5.5p2.
Chris Lattner660e31d2007-08-24 21:00:35 +0000872 if (Ops.Ty->isUnsignedIntegerType())
Chris Lattner9fba49a2007-08-24 05:35:26 +0000873 return Builder.CreateURem(Ops.LHS, Ops.RHS, "rem");
874 else
875 return Builder.CreateSRem(Ops.LHS, Ops.RHS, "rem");
876}
877
878
879Value *ScalarExprEmitter::EmitAdd(const BinOpInfo &Ops) {
Chris Lattner660e31d2007-08-24 21:00:35 +0000880 if (!Ops.Ty->isPointerType())
Chris Lattner9fba49a2007-08-24 05:35:26 +0000881 return Builder.CreateAdd(Ops.LHS, Ops.RHS, "add");
Chris Lattner660e31d2007-08-24 21:00:35 +0000882
883 // FIXME: What about a pointer to a VLA?
Chris Lattner17c0cb02008-01-03 06:36:51 +0000884 Value *Ptr, *Idx;
885 Expr *IdxExp;
Daniel Dunbar4fd58ab2009-01-23 18:51:09 +0000886 const PointerType *PT;
887 if ((PT = Ops.E->getLHS()->getType()->getAsPointerType())) {
Chris Lattner17c0cb02008-01-03 06:36:51 +0000888 Ptr = Ops.LHS;
889 Idx = Ops.RHS;
890 IdxExp = Ops.E->getRHS();
891 } else { // int + pointer
Daniel Dunbar4fd58ab2009-01-23 18:51:09 +0000892 PT = Ops.E->getRHS()->getType()->getAsPointerType();
893 assert(PT && "Invalid add expr");
Chris Lattner17c0cb02008-01-03 06:36:51 +0000894 Ptr = Ops.RHS;
895 Idx = Ops.LHS;
896 IdxExp = Ops.E->getLHS();
897 }
898
899 unsigned Width = cast<llvm::IntegerType>(Idx->getType())->getBitWidth();
900 if (Width < CGF.LLVMPointerWidth) {
901 // Zero or sign extend the pointer value based on whether the index is
902 // signed or not.
903 const llvm::Type *IdxType = llvm::IntegerType::get(CGF.LLVMPointerWidth);
Chris Lattnerc154ac12008-07-26 22:37:01 +0000904 if (IdxExp->getType()->isSignedIntegerType())
Chris Lattner17c0cb02008-01-03 06:36:51 +0000905 Idx = Builder.CreateSExt(Idx, IdxType, "idx.ext");
906 else
907 Idx = Builder.CreateZExt(Idx, IdxType, "idx.ext");
908 }
Daniel Dunbar4fd58ab2009-01-23 18:51:09 +0000909
910 // Explicitly handle GNU void* and function pointer arithmetic
911 // extensions. The GNU void* casts amount to no-ops since our void*
912 // type is i8*, but this is future proof.
913 const QualType ElementType = PT->getPointeeType();
914 if (ElementType->isVoidType() || ElementType->isFunctionType()) {
915 const llvm::Type *i8Ty = llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
916 Value *Casted = Builder.CreateBitCast(Ptr, i8Ty);
917 Value *Res = Builder.CreateGEP(Casted, Idx, "sub.ptr");
918 return Builder.CreateBitCast(Res, Ptr->getType());
919 }
Chris Lattner17c0cb02008-01-03 06:36:51 +0000920
921 return Builder.CreateGEP(Ptr, Idx, "add.ptr");
Chris Lattner9fba49a2007-08-24 05:35:26 +0000922}
923
924Value *ScalarExprEmitter::EmitSub(const BinOpInfo &Ops) {
925 if (!isa<llvm::PointerType>(Ops.LHS->getType()))
926 return Builder.CreateSub(Ops.LHS, Ops.RHS, "sub");
Chris Lattner660e31d2007-08-24 21:00:35 +0000927
Daniel Dunbar4fd58ab2009-01-23 18:51:09 +0000928 const QualType LHSType = Ops.E->getLHS()->getType();
929 const QualType LHSElementType = LHSType->getAsPointerType()->getPointeeType();
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000930 if (!isa<llvm::PointerType>(Ops.RHS->getType())) {
931 // pointer - int
932 Value *Idx = Ops.RHS;
933 unsigned Width = cast<llvm::IntegerType>(Idx->getType())->getBitWidth();
934 if (Width < CGF.LLVMPointerWidth) {
935 // Zero or sign extend the pointer value based on whether the index is
936 // signed or not.
937 const llvm::Type *IdxType = llvm::IntegerType::get(CGF.LLVMPointerWidth);
938 if (Ops.E->getRHS()->getType()->isSignedIntegerType())
939 Idx = Builder.CreateSExt(Idx, IdxType, "idx.ext");
940 else
941 Idx = Builder.CreateZExt(Idx, IdxType, "idx.ext");
942 }
943 Idx = Builder.CreateNeg(Idx, "sub.ptr.neg");
944
945 // FIXME: The pointer could point to a VLA.
Daniel Dunbar4fd58ab2009-01-23 18:51:09 +0000946
947 // Explicitly handle GNU void* and function pointer arithmetic
948 // extensions. The GNU void* casts amount to no-ops since our
949 // void* type is i8*, but this is future proof.
950 if (LHSElementType->isVoidType() || LHSElementType->isFunctionType()) {
951 const llvm::Type *i8Ty = llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
952 Value *LHSCasted = Builder.CreateBitCast(Ops.LHS, i8Ty);
953 Value *Res = Builder.CreateGEP(LHSCasted, Idx, "sub.ptr");
954 return Builder.CreateBitCast(Res, Ops.LHS->getType());
955 }
956
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000957 return Builder.CreateGEP(Ops.LHS, Idx, "sub.ptr");
Daniel Dunbar0aac9f62008-08-05 00:47:03 +0000958 } else {
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000959 // pointer - pointer
960 Value *LHS = Ops.LHS;
961 Value *RHS = Ops.RHS;
Chris Lattner660e31d2007-08-24 21:00:35 +0000962
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000963 uint64_t ElementSize;
Daniel Dunbar0aac9f62008-08-05 00:47:03 +0000964
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000965 // Handle GCC extension for pointer arithmetic on void* types.
966 if (LHSElementType->isVoidType()) {
967 ElementSize = 1;
968 } else {
969 ElementSize = CGF.getContext().getTypeSize(LHSElementType) / 8;
970 }
971
972 const llvm::Type *ResultType = ConvertType(Ops.Ty);
973 LHS = Builder.CreatePtrToInt(LHS, ResultType, "sub.ptr.lhs.cast");
974 RHS = Builder.CreatePtrToInt(RHS, ResultType, "sub.ptr.rhs.cast");
975 Value *BytesBetween = Builder.CreateSub(LHS, RHS, "sub.ptr.sub");
976
977 // HACK: LLVM doesn't have an divide instruction that 'knows' there is no
978 // remainder. As such, we handle common power-of-two cases here to generate
979 // better code. See PR2247.
980 if (llvm::isPowerOf2_64(ElementSize)) {
981 Value *ShAmt =
982 llvm::ConstantInt::get(ResultType, llvm::Log2_64(ElementSize));
983 return Builder.CreateAShr(BytesBetween, ShAmt, "sub.ptr.shr");
984 }
985
986 // Otherwise, do a full sdiv.
987 Value *BytesPerElt = llvm::ConstantInt::get(ResultType, ElementSize);
988 return Builder.CreateSDiv(BytesBetween, BytesPerElt, "sub.ptr.div");
Chris Lattner9fba49a2007-08-24 05:35:26 +0000989 }
Chris Lattner9fba49a2007-08-24 05:35:26 +0000990}
991
992Value *ScalarExprEmitter::EmitShl(const BinOpInfo &Ops) {
993 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
994 // RHS to the same size as the LHS.
995 Value *RHS = Ops.RHS;
996 if (Ops.LHS->getType() != RHS->getType())
997 RHS = Builder.CreateIntCast(RHS, Ops.LHS->getType(), false, "sh_prom");
998
999 return Builder.CreateShl(Ops.LHS, RHS, "shl");
1000}
1001
1002Value *ScalarExprEmitter::EmitShr(const BinOpInfo &Ops) {
1003 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
1004 // RHS to the same size as the LHS.
1005 Value *RHS = Ops.RHS;
1006 if (Ops.LHS->getType() != RHS->getType())
1007 RHS = Builder.CreateIntCast(RHS, Ops.LHS->getType(), false, "sh_prom");
1008
Chris Lattner660e31d2007-08-24 21:00:35 +00001009 if (Ops.Ty->isUnsignedIntegerType())
Chris Lattner9fba49a2007-08-24 05:35:26 +00001010 return Builder.CreateLShr(Ops.LHS, RHS, "shr");
1011 return Builder.CreateAShr(Ops.LHS, RHS, "shr");
1012}
1013
1014Value *ScalarExprEmitter::EmitCompare(const BinaryOperator *E,unsigned UICmpOpc,
1015 unsigned SICmpOpc, unsigned FCmpOpc) {
Chris Lattnerfb182ee2007-08-26 16:34:22 +00001016 Value *Result;
Chris Lattner9fba49a2007-08-24 05:35:26 +00001017 QualType LHSTy = E->getLHS()->getType();
Nate Begeman1591bc52008-07-25 20:16:05 +00001018 if (!LHSTy->isAnyComplexType() && !LHSTy->isVectorType()) {
Chris Lattner9fba49a2007-08-24 05:35:26 +00001019 Value *LHS = Visit(E->getLHS());
1020 Value *RHS = Visit(E->getRHS());
1021
1022 if (LHS->getType()->isFloatingPoint()) {
Nate Begeman1591bc52008-07-25 20:16:05 +00001023 Result = Builder.CreateFCmp((llvm::CmpInst::Predicate)FCmpOpc,
Chris Lattner9fba49a2007-08-24 05:35:26 +00001024 LHS, RHS, "cmp");
Eli Friedman850ea372008-05-29 15:09:15 +00001025 } else if (LHSTy->isSignedIntegerType()) {
1026 Result = Builder.CreateICmp((llvm::ICmpInst::Predicate)SICmpOpc,
Chris Lattner9fba49a2007-08-24 05:35:26 +00001027 LHS, RHS, "cmp");
1028 } else {
Eli Friedman850ea372008-05-29 15:09:15 +00001029 // Unsigned integers and pointers.
1030 Result = Builder.CreateICmp((llvm::ICmpInst::Predicate)UICmpOpc,
Chris Lattner9fba49a2007-08-24 05:35:26 +00001031 LHS, RHS, "cmp");
1032 }
Nate Begeman1591bc52008-07-25 20:16:05 +00001033 } else if (LHSTy->isVectorType()) {
1034 Value *LHS = Visit(E->getLHS());
1035 Value *RHS = Visit(E->getRHS());
1036
1037 if (LHS->getType()->isFPOrFPVector()) {
1038 Result = Builder.CreateVFCmp((llvm::CmpInst::Predicate)FCmpOpc,
1039 LHS, RHS, "cmp");
1040 } else if (LHSTy->isUnsignedIntegerType()) {
1041 Result = Builder.CreateVICmp((llvm::CmpInst::Predicate)UICmpOpc,
1042 LHS, RHS, "cmp");
1043 } else {
1044 // Signed integers and pointers.
1045 Result = Builder.CreateVICmp((llvm::CmpInst::Predicate)SICmpOpc,
1046 LHS, RHS, "cmp");
1047 }
1048 return Result;
Chris Lattner9fba49a2007-08-24 05:35:26 +00001049 } else {
1050 // Complex Comparison: can only be an equality comparison.
1051 CodeGenFunction::ComplexPairTy LHS = CGF.EmitComplexExpr(E->getLHS());
1052 CodeGenFunction::ComplexPairTy RHS = CGF.EmitComplexExpr(E->getRHS());
1053
Chris Lattnerc154ac12008-07-26 22:37:01 +00001054 QualType CETy = LHSTy->getAsComplexType()->getElementType();
Chris Lattner9fba49a2007-08-24 05:35:26 +00001055
Chris Lattnerfb182ee2007-08-26 16:34:22 +00001056 Value *ResultR, *ResultI;
Chris Lattner9fba49a2007-08-24 05:35:26 +00001057 if (CETy->isRealFloatingType()) {
1058 ResultR = Builder.CreateFCmp((llvm::FCmpInst::Predicate)FCmpOpc,
1059 LHS.first, RHS.first, "cmp.r");
1060 ResultI = Builder.CreateFCmp((llvm::FCmpInst::Predicate)FCmpOpc,
1061 LHS.second, RHS.second, "cmp.i");
1062 } else {
1063 // Complex comparisons can only be equality comparisons. As such, signed
1064 // and unsigned opcodes are the same.
1065 ResultR = Builder.CreateICmp((llvm::ICmpInst::Predicate)UICmpOpc,
1066 LHS.first, RHS.first, "cmp.r");
1067 ResultI = Builder.CreateICmp((llvm::ICmpInst::Predicate)UICmpOpc,
1068 LHS.second, RHS.second, "cmp.i");
1069 }
1070
1071 if (E->getOpcode() == BinaryOperator::EQ) {
1072 Result = Builder.CreateAnd(ResultR, ResultI, "and.ri");
1073 } else {
1074 assert(E->getOpcode() == BinaryOperator::NE &&
1075 "Complex comparison other than == or != ?");
1076 Result = Builder.CreateOr(ResultR, ResultI, "or.ri");
1077 }
1078 }
Nuno Lopes92577002009-01-11 23:22:37 +00001079
1080 return EmitScalarConversion(Result, CGF.getContext().BoolTy, E->getType());
Chris Lattner9fba49a2007-08-24 05:35:26 +00001081}
1082
1083Value *ScalarExprEmitter::VisitBinAssign(const BinaryOperator *E) {
1084 LValue LHS = EmitLValue(E->getLHS());
1085 Value *RHS = Visit(E->getRHS());
1086
Daniel Dunbar2668dd12008-11-19 09:36:46 +00001087 // Store the value into the LHS. Bit-fields are handled specially
Daniel Dunbar2710fc92008-11-19 11:54:05 +00001088 // because the result is altered by the store, i.e., [C99 6.5.16p1]
1089 // 'An assignment expression has the value of the left operand after
1090 // the assignment...'.
Chris Lattner9fba49a2007-08-24 05:35:26 +00001091 // FIXME: Volatility!
Eli Friedmanf9b930c2008-05-25 14:13:57 +00001092 if (LHS.isBitfield())
Daniel Dunbar2668dd12008-11-19 09:36:46 +00001093 CGF.EmitStoreThroughBitfieldLValue(RValue::get(RHS), LHS, E->getType(),
1094 &RHS);
1095 else
1096 CGF.EmitStoreThroughLValue(RValue::get(RHS), LHS, E->getType());
Daniel Dunbare6c31752008-08-29 08:11:39 +00001097
Chris Lattner9fba49a2007-08-24 05:35:26 +00001098 // Return the RHS.
1099 return RHS;
1100}
1101
1102Value *ScalarExprEmitter::VisitBinLAnd(const BinaryOperator *E) {
Chris Lattner715c2a72008-11-12 08:26:50 +00001103 // If we have 0 && RHS, see if we can elide RHS, if so, just return 0.
1104 // If we have 1 && X, just emit X without inserting the control flow.
1105 if (int Cond = CGF.ConstantFoldsToSimpleInteger(E->getLHS())) {
1106 if (Cond == 1) { // If we have 1 && X, just emit X.
Chris Lattner3f73d0d2008-11-11 07:41:27 +00001107 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1108 // ZExt result to int.
1109 return Builder.CreateZExt(RHSCond, CGF.LLVMIntTy, "land.ext");
1110 }
Chris Lattner715c2a72008-11-12 08:26:50 +00001111
1112 // 0 && RHS: If it is safe, just elide the RHS, and return 0.
1113 if (!CGF.ContainsLabel(E->getRHS()))
1114 return llvm::Constant::getNullValue(CGF.LLVMIntTy);
Chris Lattner3f73d0d2008-11-11 07:41:27 +00001115 }
1116
Daniel Dunbar6e3a10c2008-11-13 01:38:36 +00001117 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("land.end");
1118 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("land.rhs");
Chris Lattner715c2a72008-11-12 08:26:50 +00001119
Chris Lattner7f80bb32008-11-12 08:38:24 +00001120 // Branch on the LHS first. If it is false, go to the failure (cont) block.
1121 CGF.EmitBranchOnBoolExpr(E->getLHS(), RHSBlock, ContBlock);
1122
1123 // Any edges into the ContBlock are now from an (indeterminate number of)
1124 // edges from this first condition. All of these values will be false. Start
1125 // setting up the PHI node in the Cont Block for this.
1126 llvm::PHINode *PN = llvm::PHINode::Create(llvm::Type::Int1Ty, "", ContBlock);
1127 PN->reserveOperandSpace(2); // Normal case, two inputs.
1128 for (llvm::pred_iterator PI = pred_begin(ContBlock), PE = pred_end(ContBlock);
1129 PI != PE; ++PI)
1130 PN->addIncoming(llvm::ConstantInt::getFalse(), *PI);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001131
1132 CGF.EmitBlock(RHSBlock);
1133 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1134
1135 // Reaquire the RHS block, as there may be subblocks inserted.
1136 RHSBlock = Builder.GetInsertBlock();
Chris Lattner7f80bb32008-11-12 08:38:24 +00001137
1138 // Emit an unconditional branch from this block to ContBlock. Insert an entry
1139 // into the phi node for the edge with the value of RHSCond.
Chris Lattner9fba49a2007-08-24 05:35:26 +00001140 CGF.EmitBlock(ContBlock);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001141 PN->addIncoming(RHSCond, RHSBlock);
1142
1143 // ZExt result to int.
1144 return Builder.CreateZExt(PN, CGF.LLVMIntTy, "land.ext");
1145}
1146
1147Value *ScalarExprEmitter::VisitBinLOr(const BinaryOperator *E) {
Chris Lattner715c2a72008-11-12 08:26:50 +00001148 // If we have 1 || RHS, see if we can elide RHS, if so, just return 1.
1149 // If we have 0 || X, just emit X without inserting the control flow.
1150 if (int Cond = CGF.ConstantFoldsToSimpleInteger(E->getLHS())) {
1151 if (Cond == -1) { // If we have 0 || X, just emit X.
Chris Lattner3f73d0d2008-11-11 07:41:27 +00001152 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1153 // ZExt result to int.
1154 return Builder.CreateZExt(RHSCond, CGF.LLVMIntTy, "lor.ext");
1155 }
Chris Lattner715c2a72008-11-12 08:26:50 +00001156
Eli Friedmanea137cd2008-12-02 16:02:46 +00001157 // 1 || RHS: If it is safe, just elide the RHS, and return 1.
Chris Lattner715c2a72008-11-12 08:26:50 +00001158 if (!CGF.ContainsLabel(E->getRHS()))
Eli Friedmanea137cd2008-12-02 16:02:46 +00001159 return llvm::ConstantInt::get(CGF.LLVMIntTy, 1);
Chris Lattner3f73d0d2008-11-11 07:41:27 +00001160 }
1161
Daniel Dunbar6e3a10c2008-11-13 01:38:36 +00001162 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("lor.end");
1163 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("lor.rhs");
Chris Lattner9fba49a2007-08-24 05:35:26 +00001164
Chris Lattner7f80bb32008-11-12 08:38:24 +00001165 // Branch on the LHS first. If it is true, go to the success (cont) block.
1166 CGF.EmitBranchOnBoolExpr(E->getLHS(), ContBlock, RHSBlock);
1167
1168 // Any edges into the ContBlock are now from an (indeterminate number of)
1169 // edges from this first condition. All of these values will be true. Start
1170 // setting up the PHI node in the Cont Block for this.
1171 llvm::PHINode *PN = llvm::PHINode::Create(llvm::Type::Int1Ty, "", ContBlock);
1172 PN->reserveOperandSpace(2); // Normal case, two inputs.
1173 for (llvm::pred_iterator PI = pred_begin(ContBlock), PE = pred_end(ContBlock);
1174 PI != PE; ++PI)
1175 PN->addIncoming(llvm::ConstantInt::getTrue(), *PI);
1176
1177 // Emit the RHS condition as a bool value.
Chris Lattner9fba49a2007-08-24 05:35:26 +00001178 CGF.EmitBlock(RHSBlock);
1179 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1180
1181 // Reaquire the RHS block, as there may be subblocks inserted.
1182 RHSBlock = Builder.GetInsertBlock();
Chris Lattner9fba49a2007-08-24 05:35:26 +00001183
Chris Lattner7f80bb32008-11-12 08:38:24 +00001184 // Emit an unconditional branch from this block to ContBlock. Insert an entry
1185 // into the phi node for the edge with the value of RHSCond.
1186 CGF.EmitBlock(ContBlock);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001187 PN->addIncoming(RHSCond, RHSBlock);
1188
1189 // ZExt result to int.
1190 return Builder.CreateZExt(PN, CGF.LLVMIntTy, "lor.ext");
1191}
1192
1193Value *ScalarExprEmitter::VisitBinComma(const BinaryOperator *E) {
1194 CGF.EmitStmt(E->getLHS());
Daniel Dunbar5aa22bc2008-11-11 23:11:34 +00001195 CGF.EnsureInsertPoint();
Chris Lattner9fba49a2007-08-24 05:35:26 +00001196 return Visit(E->getRHS());
1197}
1198
1199//===----------------------------------------------------------------------===//
1200// Other Operators
1201//===----------------------------------------------------------------------===//
1202
Chris Lattner504a5282008-11-12 08:55:54 +00001203/// isCheapEnoughToEvaluateUnconditionally - Return true if the specified
1204/// expression is cheap enough and side-effect-free enough to evaluate
1205/// unconditionally instead of conditionally. This is used to convert control
1206/// flow into selects in some cases.
1207static bool isCheapEnoughToEvaluateUnconditionally(const Expr *E) {
1208 if (const ParenExpr *PE = dyn_cast<ParenExpr>(E))
1209 return isCheapEnoughToEvaluateUnconditionally(PE->getSubExpr());
1210
1211 // TODO: Allow anything we can constant fold to an integer or fp constant.
1212 if (isa<IntegerLiteral>(E) || isa<CharacterLiteral>(E) ||
1213 isa<FloatingLiteral>(E))
1214 return true;
1215
1216 // Non-volatile automatic variables too, to get "cond ? X : Y" where
1217 // X and Y are local variables.
1218 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
1219 if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
1220 if (VD->hasLocalStorage() && !VD->getType().isVolatileQualified())
1221 return true;
1222
1223 return false;
1224}
1225
1226
Chris Lattner9fba49a2007-08-24 05:35:26 +00001227Value *ScalarExprEmitter::
1228VisitConditionalOperator(const ConditionalOperator *E) {
Chris Lattner3d6606b2008-11-12 08:04:58 +00001229 // If the condition constant folds and can be elided, try to avoid emitting
1230 // the condition and the dead arm.
1231 if (int Cond = CGF.ConstantFoldsToSimpleInteger(E->getCond())){
Chris Lattner044bffc2008-11-11 18:56:45 +00001232 Expr *Live = E->getLHS(), *Dead = E->getRHS();
Chris Lattner3d6606b2008-11-12 08:04:58 +00001233 if (Cond == -1)
Chris Lattner044bffc2008-11-11 18:56:45 +00001234 std::swap(Live, Dead);
Chris Lattner3d6606b2008-11-12 08:04:58 +00001235
1236 // If the dead side doesn't have labels we need, and if the Live side isn't
1237 // the gnu missing ?: extension (which we could handle, but don't bother
1238 // to), just emit the Live part.
1239 if ((!Dead || !CGF.ContainsLabel(Dead)) && // No labels in dead part
1240 Live) // Live part isn't missing.
1241 return Visit(Live);
Chris Lattner044bffc2008-11-11 18:56:45 +00001242 }
1243
Chris Lattner504a5282008-11-12 08:55:54 +00001244
1245 // If this is a really simple expression (like x ? 4 : 5), emit this as a
1246 // select instead of as control flow. We can only do this if it is cheap and
Chris Lattner1f11af22008-11-16 06:16:27 +00001247 // safe to evaluate the LHS and RHS unconditionally.
Chris Lattner504a5282008-11-12 08:55:54 +00001248 if (E->getLHS() && isCheapEnoughToEvaluateUnconditionally(E->getLHS()) &&
1249 isCheapEnoughToEvaluateUnconditionally(E->getRHS())) {
1250 llvm::Value *CondV = CGF.EvaluateExprAsBool(E->getCond());
1251 llvm::Value *LHS = Visit(E->getLHS());
1252 llvm::Value *RHS = Visit(E->getRHS());
1253 return Builder.CreateSelect(CondV, LHS, RHS, "cond");
1254 }
1255
1256
Daniel Dunbarb23e9922008-11-12 10:13:37 +00001257 llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true");
1258 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false");
Daniel Dunbar6e3a10c2008-11-13 01:38:36 +00001259 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end");
Chris Lattner67e22462008-11-12 08:08:13 +00001260 Value *CondVal = 0;
Chris Lattner3d6606b2008-11-12 08:04:58 +00001261
Chris Lattner67e22462008-11-12 08:08:13 +00001262 // If we have the GNU missing condition extension, evaluate the conditional
1263 // and then convert it to bool the hard way. We do this explicitly
1264 // because we need the unconverted value for the missing middle value of
1265 // the ?:.
1266 if (E->getLHS() == 0) {
1267 CondVal = CGF.EmitScalarExpr(E->getCond());
1268 Value *CondBoolVal =
1269 CGF.EmitScalarConversion(CondVal, E->getCond()->getType(),
1270 CGF.getContext().BoolTy);
1271 Builder.CreateCondBr(CondBoolVal, LHSBlock, RHSBlock);
1272 } else {
1273 // Otherwise, just use EmitBranchOnBoolExpr to get small and simple code for
1274 // the branch on bool.
1275 CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock);
1276 }
Chris Lattner9fba49a2007-08-24 05:35:26 +00001277
1278 CGF.EmitBlock(LHSBlock);
1279
1280 // Handle the GNU extension for missing LHS.
Chris Lattner98a425c2007-11-26 01:40:58 +00001281 Value *LHS;
1282 if (E->getLHS())
Eli Friedmance8d7032008-05-16 20:38:39 +00001283 LHS = Visit(E->getLHS());
Chris Lattner98a425c2007-11-26 01:40:58 +00001284 else // Perform promotions, to handle cases like "short ?: int"
1285 LHS = EmitScalarConversion(CondVal, E->getCond()->getType(), E->getType());
1286
Chris Lattner9fba49a2007-08-24 05:35:26 +00001287 LHSBlock = Builder.GetInsertBlock();
Daniel Dunbar5276caa2008-11-11 09:41:28 +00001288 CGF.EmitBranch(ContBlock);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001289
1290 CGF.EmitBlock(RHSBlock);
1291
Eli Friedmance8d7032008-05-16 20:38:39 +00001292 Value *RHS = Visit(E->getRHS());
Chris Lattner9fba49a2007-08-24 05:35:26 +00001293 RHSBlock = Builder.GetInsertBlock();
Daniel Dunbar5276caa2008-11-11 09:41:28 +00001294 CGF.EmitBranch(ContBlock);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001295
1296 CGF.EmitBlock(ContBlock);
1297
Nuno Lopesb62ff242008-06-04 19:15:45 +00001298 if (!LHS || !RHS) {
Chris Lattner307da022007-11-30 17:56:23 +00001299 assert(E->getType()->isVoidType() && "Non-void value should have a value");
1300 return 0;
1301 }
1302
Chris Lattner9fba49a2007-08-24 05:35:26 +00001303 // Create a PHI node for the real part.
1304 llvm::PHINode *PN = Builder.CreatePHI(LHS->getType(), "cond");
1305 PN->reserveOperandSpace(2);
1306 PN->addIncoming(LHS, LHSBlock);
1307 PN->addIncoming(RHS, RHSBlock);
1308 return PN;
1309}
1310
1311Value *ScalarExprEmitter::VisitChooseExpr(ChooseExpr *E) {
Chris Lattner9fba49a2007-08-24 05:35:26 +00001312 // Emit the LHS or RHS as appropriate.
Devang Patel0f2a8fb2007-10-30 20:59:40 +00001313 return
1314 Visit(E->isConditionTrue(CGF.getContext()) ? E->getLHS() : E->getRHS());
Chris Lattner9fba49a2007-08-24 05:35:26 +00001315}
1316
Nate Begeman9f3bfb72008-01-17 17:46:27 +00001317Value *ScalarExprEmitter::VisitOverloadExpr(OverloadExpr *E) {
Nate Begemanbd881ef2008-01-30 20:50:20 +00001318 return CGF.EmitCallExpr(E->getFn(), E->arg_begin(),
Ted Kremenek2719e982008-06-17 02:43:46 +00001319 E->arg_end(CGF.getContext())).getScalarVal();
Nate Begeman9f3bfb72008-01-17 17:46:27 +00001320}
1321
Chris Lattner307da022007-11-30 17:56:23 +00001322Value *ScalarExprEmitter::VisitVAArgExpr(VAArgExpr *VE) {
Eli Friedman8f5e8782009-01-20 17:46:04 +00001323 llvm::Value *ArgValue = CGF.EmitVAListRef(VE->getSubExpr());
Anders Carlsson36760332007-10-15 20:28:48 +00001324
Anders Carlsson285611e2008-11-04 05:30:00 +00001325 llvm::Value *ArgPtr = CGF.EmitVAArg(ArgValue, VE->getType());
1326
1327 // If EmitVAArg fails, we fall back to the LLVM instruction.
1328 if (!ArgPtr)
1329 return Builder.CreateVAArg(ArgValue, ConvertType(VE->getType()));
1330
1331 // FIXME: volatile?
1332 return Builder.CreateLoad(ArgPtr);
Anders Carlsson36760332007-10-15 20:28:48 +00001333}
1334
Chris Lattner307da022007-11-30 17:56:23 +00001335Value *ScalarExprEmitter::VisitObjCEncodeExpr(const ObjCEncodeExpr *E) {
Anders Carlsson36f07d82007-10-29 05:01:08 +00001336 std::string str;
Daniel Dunbarc9197cd2008-10-17 20:21:44 +00001337 CGF.getContext().getObjCEncodingForType(E->getEncodedType(), str);
Anders Carlsson36f07d82007-10-29 05:01:08 +00001338
1339 llvm::Constant *C = llvm::ConstantArray::get(str);
1340 C = new llvm::GlobalVariable(C->getType(), true,
1341 llvm::GlobalValue::InternalLinkage,
1342 C, ".str", &CGF.CGM.getModule());
1343 llvm::Constant *Zero = llvm::Constant::getNullValue(llvm::Type::Int32Ty);
1344 llvm::Constant *Zeros[] = { Zero, Zero };
1345 C = llvm::ConstantExpr::getGetElementPtr(C, Zeros, 2);
1346
1347 return C;
1348}
1349
Chris Lattner9fba49a2007-08-24 05:35:26 +00001350//===----------------------------------------------------------------------===//
1351// Entry Point into this File
1352//===----------------------------------------------------------------------===//
1353
1354/// EmitComplexExpr - Emit the computation of the specified expression of
1355/// complex type, ignoring the result.
1356Value *CodeGenFunction::EmitScalarExpr(const Expr *E) {
1357 assert(E && !hasAggregateLLVMType(E->getType()) &&
1358 "Invalid scalar expression to emit");
1359
1360 return ScalarExprEmitter(*this).Visit(const_cast<Expr*>(E));
1361}
Chris Lattner4e05d1e2007-08-26 06:48:56 +00001362
1363/// EmitScalarConversion - Emit a conversion from the specified type to the
1364/// specified destination type, both of which are LLVM scalar types.
Chris Lattnerfb182ee2007-08-26 16:34:22 +00001365Value *CodeGenFunction::EmitScalarConversion(Value *Src, QualType SrcTy,
1366 QualType DstTy) {
Chris Lattner4e05d1e2007-08-26 06:48:56 +00001367 assert(!hasAggregateLLVMType(SrcTy) && !hasAggregateLLVMType(DstTy) &&
1368 "Invalid scalar expression to emit");
1369 return ScalarExprEmitter(*this).EmitScalarConversion(Src, SrcTy, DstTy);
1370}
Chris Lattnerfb182ee2007-08-26 16:34:22 +00001371
1372/// EmitComplexToScalarConversion - Emit a conversion from the specified
1373/// complex type to the specified destination type, where the destination
1374/// type is an LLVM scalar type.
1375Value *CodeGenFunction::EmitComplexToScalarConversion(ComplexPairTy Src,
1376 QualType SrcTy,
1377 QualType DstTy) {
Chris Lattnerde0908b2008-04-04 16:54:41 +00001378 assert(SrcTy->isAnyComplexType() && !hasAggregateLLVMType(DstTy) &&
Chris Lattnerfb182ee2007-08-26 16:34:22 +00001379 "Invalid complex -> scalar conversion");
1380 return ScalarExprEmitter(*this).EmitComplexToScalarConversion(Src, SrcTy,
1381 DstTy);
1382}
Anders Carlssona9234fe2007-12-10 19:35:18 +00001383
1384Value *CodeGenFunction::EmitShuffleVector(Value* V1, Value *V2, ...) {
1385 assert(V1->getType() == V2->getType() &&
1386 "Vector operands must be of the same type");
Anders Carlssona9234fe2007-12-10 19:35:18 +00001387 unsigned NumElements =
1388 cast<llvm::VectorType>(V1->getType())->getNumElements();
1389
1390 va_list va;
1391 va_start(va, V2);
1392
1393 llvm::SmallVector<llvm::Constant*, 16> Args;
Anders Carlssona9234fe2007-12-10 19:35:18 +00001394 for (unsigned i = 0; i < NumElements; i++) {
1395 int n = va_arg(va, int);
Anders Carlssona9234fe2007-12-10 19:35:18 +00001396 assert(n >= 0 && n < (int)NumElements * 2 &&
1397 "Vector shuffle index out of bounds!");
Anders Carlssona9234fe2007-12-10 19:35:18 +00001398 Args.push_back(llvm::ConstantInt::get(llvm::Type::Int32Ty, n));
1399 }
1400
1401 const char *Name = va_arg(va, const char *);
1402 va_end(va);
1403
1404 llvm::Constant *Mask = llvm::ConstantVector::get(&Args[0], NumElements);
1405
1406 return Builder.CreateShuffleVector(V1, V2, Mask, Name);
1407}
1408
Anders Carlsson68b8be92007-12-15 21:23:30 +00001409llvm::Value *CodeGenFunction::EmitVector(llvm::Value * const *Vals,
Chris Lattnera23eb7b2008-07-26 20:15:14 +00001410 unsigned NumVals, bool isSplat) {
Anders Carlsson68b8be92007-12-15 21:23:30 +00001411 llvm::Value *Vec
Chris Lattnera23eb7b2008-07-26 20:15:14 +00001412 = llvm::UndefValue::get(llvm::VectorType::get(Vals[0]->getType(), NumVals));
Anders Carlsson68b8be92007-12-15 21:23:30 +00001413
Chris Lattnera23eb7b2008-07-26 20:15:14 +00001414 for (unsigned i = 0, e = NumVals; i != e; ++i) {
Nate Begemanec2d1062007-12-30 02:59:45 +00001415 llvm::Value *Val = isSplat ? Vals[0] : Vals[i];
Anders Carlsson68b8be92007-12-15 21:23:30 +00001416 llvm::Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, i);
Nate Begemanec2d1062007-12-30 02:59:45 +00001417 Vec = Builder.CreateInsertElement(Vec, Val, Idx, "tmp");
Anders Carlsson68b8be92007-12-15 21:23:30 +00001418 }
1419
1420 return Vec;
1421}