blob: 1d2da0ddf89d0cca8d5b345861c423c9a1d00064 [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"
Mike Stumpdb789912009-04-01 20:28:16 +000025#include "llvm/Module.h"
Chris Lattner9fba49a2007-08-24 05:35:26 +000026#include "llvm/Support/Compiler.h"
Chris Lattner7f80bb32008-11-12 08:38:24 +000027#include "llvm/Support/CFG.h"
Mike Stumpfca5da02009-02-21 20:00:35 +000028#include "llvm/Target/TargetData.h"
Chris Lattnerc2126682008-01-03 07:05:49 +000029#include <cstdarg>
Ted Kremenek03cf4df2007-12-10 23:44:32 +000030
Chris Lattner9fba49a2007-08-24 05:35:26 +000031using namespace clang;
32using namespace CodeGen;
33using llvm::Value;
34
35//===----------------------------------------------------------------------===//
36// Scalar Expression Emitter
37//===----------------------------------------------------------------------===//
38
39struct BinOpInfo {
40 Value *LHS;
41 Value *RHS;
Chris Lattner660e31d2007-08-24 21:00:35 +000042 QualType Ty; // Computation Type.
Chris Lattner9fba49a2007-08-24 05:35:26 +000043 const BinaryOperator *E;
44};
45
46namespace {
47class VISIBILITY_HIDDEN ScalarExprEmitter
48 : public StmtVisitor<ScalarExprEmitter, Value*> {
49 CodeGenFunction &CGF;
Daniel Dunbard916e6e2008-11-01 01:53:16 +000050 CGBuilderTy &Builder;
Chris Lattnercbfb5512008-03-01 08:45:05 +000051
Chris Lattner9fba49a2007-08-24 05:35:26 +000052public:
53
Chris Lattnercbfb5512008-03-01 08:45:05 +000054 ScalarExprEmitter(CodeGenFunction &cgf) : CGF(cgf),
Daniel Dunbarf1f7f192008-08-20 00:28:19 +000055 Builder(CGF.Builder) {
Chris Lattner9fba49a2007-08-24 05:35:26 +000056 }
Chris Lattner9fba49a2007-08-24 05:35:26 +000057
58 //===--------------------------------------------------------------------===//
59 // Utilities
60 //===--------------------------------------------------------------------===//
61
62 const llvm::Type *ConvertType(QualType T) { return CGF.ConvertType(T); }
63 LValue EmitLValue(const Expr *E) { return CGF.EmitLValue(E); }
64
65 Value *EmitLoadOfLValue(LValue LV, QualType T) {
Chris Lattnere24c4cf2007-08-31 22:49:20 +000066 return CGF.EmitLoadOfLValue(LV, T).getScalarVal();
Chris Lattner9fba49a2007-08-24 05:35:26 +000067 }
68
69 /// EmitLoadOfLValue - Given an expression with complex type that represents a
70 /// value l-value, this method emits the address of the l-value, then loads
71 /// and returns the result.
72 Value *EmitLoadOfLValue(const Expr *E) {
Chris Lattner9fba49a2007-08-24 05:35:26 +000073 return EmitLoadOfLValue(EmitLValue(E), E->getType());
74 }
75
Chris Lattnerd8d44222007-08-26 16:42:57 +000076 /// EmitConversionToBool - Convert the specified expression value to a
Chris Lattner05942062007-08-26 17:25:57 +000077 /// boolean (i1) truth value. This is equivalent to "Val != 0".
Chris Lattnerd8d44222007-08-26 16:42:57 +000078 Value *EmitConversionToBool(Value *Src, QualType DstTy);
79
Chris Lattner4e05d1e2007-08-26 06:48:56 +000080 /// EmitScalarConversion - Emit a conversion from the specified type to the
81 /// specified destination type, both of which are LLVM scalar types.
Chris Lattnerfb182ee2007-08-26 16:34:22 +000082 Value *EmitScalarConversion(Value *Src, QualType SrcTy, QualType DstTy);
83
84 /// EmitComplexToScalarConversion - Emit a conversion from the specified
85 /// complex type to the specified destination type, where the destination
86 /// type is an LLVM scalar type.
87 Value *EmitComplexToScalarConversion(CodeGenFunction::ComplexPairTy Src,
88 QualType SrcTy, QualType DstTy);
Mike Stump4eb81dc2009-02-12 18:29:15 +000089
Chris Lattner9fba49a2007-08-24 05:35:26 +000090 //===--------------------------------------------------------------------===//
91 // Visitor Methods
92 //===--------------------------------------------------------------------===//
93
94 Value *VisitStmt(Stmt *S) {
Ted Kremenekb3ee1932007-12-11 21:27:55 +000095 S->dump(CGF.getContext().getSourceManager());
Chris Lattner9fba49a2007-08-24 05:35:26 +000096 assert(0 && "Stmt can't have complex result type!");
97 return 0;
98 }
99 Value *VisitExpr(Expr *S);
100 Value *VisitParenExpr(ParenExpr *PE) { return Visit(PE->getSubExpr()); }
101
102 // Leaves.
103 Value *VisitIntegerLiteral(const IntegerLiteral *E) {
104 return llvm::ConstantInt::get(E->getValue());
105 }
106 Value *VisitFloatingLiteral(const FloatingLiteral *E) {
Chris Lattner70c38672008-04-20 00:45:53 +0000107 return llvm::ConstantFP::get(E->getValue());
Chris Lattner9fba49a2007-08-24 05:35:26 +0000108 }
109 Value *VisitCharacterLiteral(const CharacterLiteral *E) {
110 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
111 }
Nate Begemane9bfe6d2007-11-15 05:40:03 +0000112 Value *VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
113 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
114 }
Argiris Kirtzidis750eb972008-08-23 19:35:47 +0000115 Value *VisitCXXZeroInitValueExpr(const CXXZeroInitValueExpr *E) {
116 return llvm::Constant::getNullValue(ConvertType(E->getType()));
117 }
Anders Carlsson774f9c72008-12-21 22:39:40 +0000118 Value *VisitGNUNullExpr(const GNUNullExpr *E) {
119 return llvm::Constant::getNullValue(ConvertType(E->getType()));
120 }
Chris Lattner9fba49a2007-08-24 05:35:26 +0000121 Value *VisitTypesCompatibleExpr(const TypesCompatibleExpr *E) {
122 return llvm::ConstantInt::get(ConvertType(E->getType()),
Steve Naroff85f0dc52007-10-15 20:41:53 +0000123 CGF.getContext().typesAreCompatible(
124 E->getArgType1(), E->getArgType2()));
Chris Lattner9fba49a2007-08-24 05:35:26 +0000125 }
Sebastian Redl0cb7c872008-11-11 17:56:53 +0000126 Value *VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E);
Daniel Dunbar879788d2008-08-04 16:51:22 +0000127 Value *VisitAddrLabelExpr(const AddrLabelExpr *E) {
Daniel Dunbarb5fda0c2008-08-16 01:41:47 +0000128 llvm::Value *V =
129 llvm::ConstantInt::get(llvm::Type::Int32Ty,
130 CGF.GetIDForAddrOfLabel(E->getLabel()));
131
132 return Builder.CreateIntToPtr(V, ConvertType(E->getType()));
Daniel Dunbar879788d2008-08-04 16:51:22 +0000133 }
Chris Lattner9fba49a2007-08-24 05:35:26 +0000134
135 // l-values.
136 Value *VisitDeclRefExpr(DeclRefExpr *E) {
137 if (const EnumConstantDecl *EC = dyn_cast<EnumConstantDecl>(E->getDecl()))
138 return llvm::ConstantInt::get(EC->getInitVal());
139 return EmitLoadOfLValue(E);
140 }
Daniel Dunbar91cc4022008-08-27 06:57:25 +0000141 Value *VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
142 return CGF.EmitObjCSelectorExpr(E);
143 }
144 Value *VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
145 return CGF.EmitObjCProtocolExpr(E);
146 }
147 Value *VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
148 return EmitLoadOfLValue(E);
149 }
Daniel Dunbar5e105892008-08-23 10:51:21 +0000150 Value *VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
Daniel Dunbare6c31752008-08-29 08:11:39 +0000151 return EmitLoadOfLValue(E);
Daniel Dunbar91cc4022008-08-27 06:57:25 +0000152 }
Fariborz Jahanianb0973da2008-11-22 22:30:21 +0000153 Value *VisitObjCKVCRefExpr(ObjCKVCRefExpr *E) {
154 return EmitLoadOfLValue(E);
155 }
Daniel Dunbar91cc4022008-08-27 06:57:25 +0000156 Value *VisitObjCMessageExpr(ObjCMessageExpr *E) {
157 return CGF.EmitObjCMessageExpr(E).getScalarVal();
Daniel Dunbar5e105892008-08-23 10:51:21 +0000158 }
159
Chris Lattner9fba49a2007-08-24 05:35:26 +0000160 Value *VisitArraySubscriptExpr(ArraySubscriptExpr *E);
Eli Friedmand0e9d092008-05-14 19:38:39 +0000161 Value *VisitShuffleVectorExpr(ShuffleVectorExpr *E);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000162 Value *VisitMemberExpr(Expr *E) { return EmitLoadOfLValue(E); }
Nate Begemanaf6ed502008-04-18 23:10:10 +0000163 Value *VisitExtVectorElementExpr(Expr *E) { return EmitLoadOfLValue(E); }
Chris Lattnera9177982008-10-26 23:53:12 +0000164 Value *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
165 return EmitLoadOfLValue(E);
166 }
Chris Lattner9fba49a2007-08-24 05:35:26 +0000167 Value *VisitStringLiteral(Expr *E) { return EmitLValue(E).getAddress(); }
Chris Lattnerc5d32632009-02-24 22:18:39 +0000168 Value *VisitObjCEncodeExpr(const ObjCEncodeExpr *E) {
169 return EmitLValue(E).getAddress();
170 }
171
Chris Lattner69909292008-08-10 01:53:14 +0000172 Value *VisitPredefinedExpr(Expr *E) { return EmitLValue(E).getAddress(); }
Devang Patel01ab1302007-10-24 17:18:43 +0000173
174 Value *VisitInitListExpr(InitListExpr *E) {
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000175 unsigned NumInitElements = E->getNumInits();
176
Douglas Gregor9fddded2009-01-29 19:42:23 +0000177 if (E->hadArrayRangeDesignator()) {
178 CGF.ErrorUnsupported(E, "GNU array range designator extension");
179 }
180
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000181 const llvm::VectorType *VType =
Anders Carlsson35ab4f92008-01-29 01:15:48 +0000182 dyn_cast<llvm::VectorType>(ConvertType(E->getType()));
183
184 // We have a scalar in braces. Just use the first element.
185 if (!VType)
186 return Visit(E->getInit(0));
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000187
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000188 unsigned NumVectorElements = VType->getNumElements();
189 const llvm::Type *ElementType = VType->getElementType();
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000190
191 // Emit individual vector element stores.
192 llvm::Value *V = llvm::UndefValue::get(VType);
193
Anders Carlsson323d5682007-12-18 02:45:33 +0000194 // Emit initializers
195 unsigned i;
196 for (i = 0; i < NumInitElements; ++i) {
Devang Patel32c39832007-10-24 18:05:48 +0000197 Value *NewV = Visit(E->getInit(i));
198 Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, i);
199 V = Builder.CreateInsertElement(V, NewV, Idx);
Devang Patel01ab1302007-10-24 17:18:43 +0000200 }
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000201
202 // Emit remaining default initializers
203 for (/* Do not initialize i*/; i < NumVectorElements; ++i) {
204 Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, i);
205 llvm::Value *NewV = llvm::Constant::getNullValue(ElementType);
206 V = Builder.CreateInsertElement(V, NewV, Idx);
207 }
208
Devang Patel32c39832007-10-24 18:05:48 +0000209 return V;
Devang Patel01ab1302007-10-24 17:18:43 +0000210 }
Chris Lattner3e254fb2008-04-08 04:40:51 +0000211
Douglas Gregorc9e012a2009-01-29 17:44:32 +0000212 Value *VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E) {
213 return llvm::Constant::getNullValue(ConvertType(E->getType()));
214 }
Chris Lattner9fba49a2007-08-24 05:35:26 +0000215 Value *VisitImplicitCastExpr(const ImplicitCastExpr *E);
216 Value *VisitCastExpr(const CastExpr *E) {
217 return EmitCastExpr(E->getSubExpr(), E->getType());
218 }
219 Value *EmitCastExpr(const Expr *E, QualType T);
220
221 Value *VisitCallExpr(const CallExpr *E) {
Chris Lattnere24c4cf2007-08-31 22:49:20 +0000222 return CGF.EmitCallExpr(E).getScalarVal();
Chris Lattner9fba49a2007-08-24 05:35:26 +0000223 }
Daniel Dunbara04840b2008-08-23 03:46:30 +0000224
Chris Lattnerea6cdd72007-08-31 22:09:40 +0000225 Value *VisitStmtExpr(const StmtExpr *E);
Mike Stumpfca5da02009-02-21 20:00:35 +0000226
Mike Stump2b6933f2009-02-28 09:07:16 +0000227 Value *VisitBlockDeclRefExpr(const BlockDeclRefExpr *E);
Chris Lattnerea6cdd72007-08-31 22:09:40 +0000228
Chris Lattner9fba49a2007-08-24 05:35:26 +0000229 // Unary Operators.
230 Value *VisitPrePostIncDec(const UnaryOperator *E, bool isInc, bool isPre);
231 Value *VisitUnaryPostDec(const UnaryOperator *E) {
232 return VisitPrePostIncDec(E, false, false);
233 }
234 Value *VisitUnaryPostInc(const UnaryOperator *E) {
235 return VisitPrePostIncDec(E, true, false);
236 }
237 Value *VisitUnaryPreDec(const UnaryOperator *E) {
238 return VisitPrePostIncDec(E, false, true);
239 }
240 Value *VisitUnaryPreInc(const UnaryOperator *E) {
241 return VisitPrePostIncDec(E, true, true);
242 }
243 Value *VisitUnaryAddrOf(const UnaryOperator *E) {
244 return EmitLValue(E->getSubExpr()).getAddress();
245 }
246 Value *VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); }
247 Value *VisitUnaryPlus(const UnaryOperator *E) {
248 return Visit(E->getSubExpr());
249 }
250 Value *VisitUnaryMinus (const UnaryOperator *E);
251 Value *VisitUnaryNot (const UnaryOperator *E);
252 Value *VisitUnaryLNot (const UnaryOperator *E);
Chris Lattner01211af2007-08-24 21:20:17 +0000253 Value *VisitUnaryReal (const UnaryOperator *E);
254 Value *VisitUnaryImag (const UnaryOperator *E);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000255 Value *VisitUnaryExtension(const UnaryOperator *E) {
256 return Visit(E->getSubExpr());
257 }
Anders Carlsson52774ad2008-01-29 15:56:48 +0000258 Value *VisitUnaryOffsetOf(const UnaryOperator *E);
Chris Lattner3e254fb2008-04-08 04:40:51 +0000259 Value *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
260 return Visit(DAE->getExpr());
261 }
Anders Carlsson52774ad2008-01-29 15:56:48 +0000262
Chris Lattner9fba49a2007-08-24 05:35:26 +0000263 // Binary Operators.
Chris Lattner9fba49a2007-08-24 05:35:26 +0000264 Value *EmitMul(const BinOpInfo &Ops) {
Mike Stump0f595bb2009-04-02 01:03:55 +0000265 if (CGF.getContext().getLangOptions().UnsignedOverflowChecking
266 || (CGF.getContext().getLangOptions().OverflowChecking
267 && Ops.Ty->isSignedIntegerType()))
Mike Stumpdb789912009-04-01 20:28:16 +0000268 return EmitOverflowCheckedBinOp(Ops);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000269 return Builder.CreateMul(Ops.LHS, Ops.RHS, "mul");
270 }
Mike Stumpdb789912009-04-01 20:28:16 +0000271 /// Create a binary op that checks for overflow.
272 /// Currently only supports +, - and *.
273 Value *EmitOverflowCheckedBinOp(const BinOpInfo &Ops);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000274 Value *EmitDiv(const BinOpInfo &Ops);
275 Value *EmitRem(const BinOpInfo &Ops);
276 Value *EmitAdd(const BinOpInfo &Ops);
277 Value *EmitSub(const BinOpInfo &Ops);
278 Value *EmitShl(const BinOpInfo &Ops);
279 Value *EmitShr(const BinOpInfo &Ops);
280 Value *EmitAnd(const BinOpInfo &Ops) {
281 return Builder.CreateAnd(Ops.LHS, Ops.RHS, "and");
282 }
283 Value *EmitXor(const BinOpInfo &Ops) {
284 return Builder.CreateXor(Ops.LHS, Ops.RHS, "xor");
285 }
286 Value *EmitOr (const BinOpInfo &Ops) {
287 return Builder.CreateOr(Ops.LHS, Ops.RHS, "or");
288 }
289
Chris Lattner660e31d2007-08-24 21:00:35 +0000290 BinOpInfo EmitBinOps(const BinaryOperator *E);
Chris Lattner0d965302007-08-26 21:41:21 +0000291 Value *EmitCompoundAssign(const CompoundAssignOperator *E,
Chris Lattner660e31d2007-08-24 21:00:35 +0000292 Value *(ScalarExprEmitter::*F)(const BinOpInfo &));
293
294 // Binary operators and binary compound assignment operators.
295#define HANDLEBINOP(OP) \
Chris Lattner0d965302007-08-26 21:41:21 +0000296 Value *VisitBin ## OP(const BinaryOperator *E) { \
297 return Emit ## OP(EmitBinOps(E)); \
298 } \
299 Value *VisitBin ## OP ## Assign(const CompoundAssignOperator *E) { \
300 return EmitCompoundAssign(E, &ScalarExprEmitter::Emit ## OP); \
Chris Lattner660e31d2007-08-24 21:00:35 +0000301 }
302 HANDLEBINOP(Mul);
303 HANDLEBINOP(Div);
304 HANDLEBINOP(Rem);
305 HANDLEBINOP(Add);
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000306 HANDLEBINOP(Sub);
Chris Lattner660e31d2007-08-24 21:00:35 +0000307 HANDLEBINOP(Shl);
308 HANDLEBINOP(Shr);
309 HANDLEBINOP(And);
310 HANDLEBINOP(Xor);
311 HANDLEBINOP(Or);
312#undef HANDLEBINOP
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000313
Chris Lattner9fba49a2007-08-24 05:35:26 +0000314 // Comparisons.
315 Value *EmitCompare(const BinaryOperator *E, unsigned UICmpOpc,
316 unsigned SICmpOpc, unsigned FCmpOpc);
317#define VISITCOMP(CODE, UI, SI, FP) \
318 Value *VisitBin##CODE(const BinaryOperator *E) { \
319 return EmitCompare(E, llvm::ICmpInst::UI, llvm::ICmpInst::SI, \
320 llvm::FCmpInst::FP); }
321 VISITCOMP(LT, ICMP_ULT, ICMP_SLT, FCMP_OLT);
322 VISITCOMP(GT, ICMP_UGT, ICMP_SGT, FCMP_OGT);
323 VISITCOMP(LE, ICMP_ULE, ICMP_SLE, FCMP_OLE);
324 VISITCOMP(GE, ICMP_UGE, ICMP_SGE, FCMP_OGE);
325 VISITCOMP(EQ, ICMP_EQ , ICMP_EQ , FCMP_OEQ);
326 VISITCOMP(NE, ICMP_NE , ICMP_NE , FCMP_UNE);
327#undef VISITCOMP
328
329 Value *VisitBinAssign (const BinaryOperator *E);
330
331 Value *VisitBinLAnd (const BinaryOperator *E);
332 Value *VisitBinLOr (const BinaryOperator *E);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000333 Value *VisitBinComma (const BinaryOperator *E);
334
335 // Other Operators.
Mike Stump4eb81dc2009-02-12 18:29:15 +0000336 Value *VisitBlockExpr(const BlockExpr *BE);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000337 Value *VisitConditionalOperator(const ConditionalOperator *CO);
338 Value *VisitChooseExpr(ChooseExpr *CE);
Anders Carlsson36760332007-10-15 20:28:48 +0000339 Value *VisitVAArgExpr(VAArgExpr *VE);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000340 Value *VisitObjCStringLiteral(const ObjCStringLiteral *E) {
341 return CGF.EmitObjCStringLiteral(E);
342 }
343};
344} // end anonymous namespace.
345
346//===----------------------------------------------------------------------===//
347// Utilities
348//===----------------------------------------------------------------------===//
349
Chris Lattnerd8d44222007-08-26 16:42:57 +0000350/// EmitConversionToBool - Convert the specified expression value to a
Chris Lattner05942062007-08-26 17:25:57 +0000351/// boolean (i1) truth value. This is equivalent to "Val != 0".
Chris Lattnerd8d44222007-08-26 16:42:57 +0000352Value *ScalarExprEmitter::EmitConversionToBool(Value *Src, QualType SrcType) {
353 assert(SrcType->isCanonical() && "EmitScalarConversion strips typedefs");
354
355 if (SrcType->isRealFloatingType()) {
356 // Compare against 0.0 for fp scalars.
357 llvm::Value *Zero = llvm::Constant::getNullValue(Src->getType());
Chris Lattnerd8d44222007-08-26 16:42:57 +0000358 return Builder.CreateFCmpUNE(Src, Zero, "tobool");
359 }
360
Daniel Dunbar5d54eed2008-08-25 10:38:11 +0000361 assert((SrcType->isIntegerType() || isa<llvm::PointerType>(Src->getType())) &&
Chris Lattnerd8d44222007-08-26 16:42:57 +0000362 "Unknown scalar type to convert");
363
364 // Because of the type rules of C, we often end up computing a logical value,
365 // then zero extending it to int, then wanting it as a logical value again.
366 // Optimize this common case.
367 if (llvm::ZExtInst *ZI = dyn_cast<llvm::ZExtInst>(Src)) {
368 if (ZI->getOperand(0)->getType() == llvm::Type::Int1Ty) {
369 Value *Result = ZI->getOperand(0);
Eli Friedman24f33972008-01-29 18:13:51 +0000370 // If there aren't any more uses, zap the instruction to save space.
371 // Note that there can be more uses, for example if this
372 // is the result of an assignment.
373 if (ZI->use_empty())
374 ZI->eraseFromParent();
Chris Lattnerd8d44222007-08-26 16:42:57 +0000375 return Result;
376 }
377 }
378
379 // Compare against an integer or pointer null.
380 llvm::Value *Zero = llvm::Constant::getNullValue(Src->getType());
381 return Builder.CreateICmpNE(Src, Zero, "tobool");
382}
383
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000384/// EmitScalarConversion - Emit a conversion from the specified type to the
385/// specified destination type, both of which are LLVM scalar types.
Chris Lattnerfb182ee2007-08-26 16:34:22 +0000386Value *ScalarExprEmitter::EmitScalarConversion(Value *Src, QualType SrcType,
387 QualType DstType) {
Chris Lattnerc154ac12008-07-26 22:37:01 +0000388 SrcType = CGF.getContext().getCanonicalType(SrcType);
389 DstType = CGF.getContext().getCanonicalType(DstType);
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000390 if (SrcType == DstType) return Src;
Chris Lattnere133d7f2007-08-26 07:21:11 +0000391
392 if (DstType->isVoidType()) return 0;
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000393
394 // Handle conversions to bool first, they are special: comparisons against 0.
Chris Lattnerc39c3652007-08-26 16:52:28 +0000395 if (DstType->isBooleanType())
396 return EmitConversionToBool(Src, SrcType);
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000397
398 const llvm::Type *DstTy = ConvertType(DstType);
399
400 // Ignore conversions like int -> uint.
401 if (Src->getType() == DstTy)
402 return Src;
403
Daniel Dunbar238335f2008-08-25 09:51:32 +0000404 // Handle pointer conversions next: pointers can only be converted
405 // to/from other pointers and integers. Check for pointer types in
406 // terms of LLVM, as some native types (like Obj-C id) may map to a
407 // pointer type.
408 if (isa<llvm::PointerType>(DstTy)) {
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000409 // The source value may be an integer, or a pointer.
410 if (isa<llvm::PointerType>(Src->getType()))
411 return Builder.CreateBitCast(Src, DstTy, "conv");
412 assert(SrcType->isIntegerType() && "Not ptr->ptr or int->ptr conversion?");
Eli Friedman35bcec82009-03-04 04:02:35 +0000413 // First, convert to the correct width so that we control the kind of
414 // extension.
415 const llvm::Type *MiddleTy = llvm::IntegerType::get(CGF.LLVMPointerWidth);
416 bool InputSigned = SrcType->isSignedIntegerType();
417 llvm::Value* IntResult =
418 Builder.CreateIntCast(Src, MiddleTy, InputSigned, "conv");
419 // Then, cast to pointer.
420 return Builder.CreateIntToPtr(IntResult, DstTy, "conv");
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000421 }
422
Daniel Dunbar238335f2008-08-25 09:51:32 +0000423 if (isa<llvm::PointerType>(Src->getType())) {
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000424 // Must be an ptr to int cast.
425 assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
Anders Carlsson44db38f2007-10-31 23:18:02 +0000426 return Builder.CreatePtrToInt(Src, DstTy, "conv");
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000427 }
428
Nate Begemanaf6ed502008-04-18 23:10:10 +0000429 // A scalar can be splatted to an extended vector of the same element type
Nate Begeman7903d052009-01-18 06:42:49 +0000430 if (DstType->isExtVectorType() && !isa<VectorType>(SrcType)) {
431 // Cast the scalar to element type
432 QualType EltTy = DstType->getAsExtVectorType()->getElementType();
433 llvm::Value *Elt = EmitScalarConversion(Src, SrcType, EltTy);
434
435 // Insert the element in element zero of an undef vector
436 llvm::Value *UnV = llvm::UndefValue::get(DstTy);
437 llvm::Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
438 UnV = Builder.CreateInsertElement(UnV, Elt, Idx, "tmp");
439
440 // Splat the element across to all elements
441 llvm::SmallVector<llvm::Constant*, 16> Args;
442 unsigned NumElements = cast<llvm::VectorType>(DstTy)->getNumElements();
443 for (unsigned i = 0; i < NumElements; i++)
444 Args.push_back(llvm::ConstantInt::get(llvm::Type::Int32Ty, 0));
445
446 llvm::Constant *Mask = llvm::ConstantVector::get(&Args[0], NumElements);
447 llvm::Value *Yay = Builder.CreateShuffleVector(UnV, UnV, Mask, "splat");
448 return Yay;
449 }
Nate Begemanec2d1062007-12-30 02:59:45 +0000450
Chris Lattner4f025a42008-02-02 04:51:41 +0000451 // Allow bitcast from vector to integer/fp of the same size.
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000452 if (isa<llvm::VectorType>(Src->getType()) ||
Chris Lattner4f025a42008-02-02 04:51:41 +0000453 isa<llvm::VectorType>(DstTy))
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000454 return Builder.CreateBitCast(Src, DstTy, "conv");
Anders Carlsson4513ecb2007-12-05 07:36:10 +0000455
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000456 // Finally, we have the arithmetic types: real int/float.
457 if (isa<llvm::IntegerType>(Src->getType())) {
458 bool InputSigned = SrcType->isSignedIntegerType();
Anders Carlsson4dac3f42007-12-26 18:20:19 +0000459 if (isa<llvm::IntegerType>(DstTy))
460 return Builder.CreateIntCast(Src, DstTy, InputSigned, "conv");
461 else if (InputSigned)
462 return Builder.CreateSIToFP(Src, DstTy, "conv");
463 else
464 return Builder.CreateUIToFP(Src, DstTy, "conv");
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000465 }
466
467 assert(Src->getType()->isFloatingPoint() && "Unknown real conversion");
468 if (isa<llvm::IntegerType>(DstTy)) {
Anders Carlsson4dac3f42007-12-26 18:20:19 +0000469 if (DstType->isSignedIntegerType())
470 return Builder.CreateFPToSI(Src, DstTy, "conv");
471 else
472 return Builder.CreateFPToUI(Src, DstTy, "conv");
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000473 }
474
475 assert(DstTy->isFloatingPoint() && "Unknown real conversion");
Anders Carlsson4dac3f42007-12-26 18:20:19 +0000476 if (DstTy->getTypeID() < Src->getType()->getTypeID())
477 return Builder.CreateFPTrunc(Src, DstTy, "conv");
478 else
479 return Builder.CreateFPExt(Src, DstTy, "conv");
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000480}
481
Chris Lattnerfb182ee2007-08-26 16:34:22 +0000482/// EmitComplexToScalarConversion - Emit a conversion from the specified
483/// complex type to the specified destination type, where the destination
484/// type is an LLVM scalar type.
485Value *ScalarExprEmitter::
486EmitComplexToScalarConversion(CodeGenFunction::ComplexPairTy Src,
487 QualType SrcTy, QualType DstTy) {
Chris Lattnerc39c3652007-08-26 16:52:28 +0000488 // Get the source element type.
Chris Lattnerc154ac12008-07-26 22:37:01 +0000489 SrcTy = SrcTy->getAsComplexType()->getElementType();
Chris Lattnerc39c3652007-08-26 16:52:28 +0000490
491 // Handle conversions to bool first, they are special: comparisons against 0.
492 if (DstTy->isBooleanType()) {
493 // Complex != 0 -> (Real != 0) | (Imag != 0)
494 Src.first = EmitScalarConversion(Src.first, SrcTy, DstTy);
495 Src.second = EmitScalarConversion(Src.second, SrcTy, DstTy);
496 return Builder.CreateOr(Src.first, Src.second, "tobool");
497 }
498
Chris Lattnerfb182ee2007-08-26 16:34:22 +0000499 // C99 6.3.1.7p2: "When a value of complex type is converted to a real type,
500 // the imaginary part of the complex value is discarded and the value of the
501 // real part is converted according to the conversion rules for the
502 // corresponding real type.
Chris Lattnerfb182ee2007-08-26 16:34:22 +0000503 return EmitScalarConversion(Src.first, SrcTy, DstTy);
504}
505
506
Chris Lattner9fba49a2007-08-24 05:35:26 +0000507//===----------------------------------------------------------------------===//
508// Visitor Methods
509//===----------------------------------------------------------------------===//
510
511Value *ScalarExprEmitter::VisitExpr(Expr *E) {
Daniel Dunbar9503b782008-08-16 00:56:44 +0000512 CGF.ErrorUnsupported(E, "scalar expression");
Chris Lattner9fba49a2007-08-24 05:35:26 +0000513 if (E->getType()->isVoidType())
514 return 0;
515 return llvm::UndefValue::get(CGF.ConvertType(E->getType()));
516}
517
Eli Friedmand0e9d092008-05-14 19:38:39 +0000518Value *ScalarExprEmitter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
519 llvm::SmallVector<llvm::Constant*, 32> indices;
520 for (unsigned i = 2; i < E->getNumSubExprs(); i++) {
521 indices.push_back(cast<llvm::Constant>(CGF.EmitScalarExpr(E->getExpr(i))));
522 }
523 Value* V1 = CGF.EmitScalarExpr(E->getExpr(0));
524 Value* V2 = CGF.EmitScalarExpr(E->getExpr(1));
525 Value* SV = llvm::ConstantVector::get(indices.begin(), indices.size());
526 return Builder.CreateShuffleVector(V1, V2, SV, "shuffle");
527}
528
Chris Lattner9fba49a2007-08-24 05:35:26 +0000529Value *ScalarExprEmitter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
530 // Emit subscript expressions in rvalue context's. For most cases, this just
531 // loads the lvalue formed by the subscript expr. However, we have to be
532 // careful, because the base of a vector subscript is occasionally an rvalue,
533 // so we can't get it as an lvalue.
534 if (!E->getBase()->getType()->isVectorType())
535 return EmitLoadOfLValue(E);
536
537 // Handle the vector case. The base must be a vector, the index must be an
538 // integer value.
539 Value *Base = Visit(E->getBase());
540 Value *Idx = Visit(E->getIdx());
Eli Friedman4a0073b2009-03-28 02:45:41 +0000541 bool IdxSigned = E->getIdx()->getType()->isSignedIntegerType();
Eli Friedmand4531942009-03-28 03:27:06 +0000542 Idx = Builder.CreateIntCast(Idx, llvm::Type::Int32Ty, IdxSigned,
543 "vecidxcast");
Chris Lattner9fba49a2007-08-24 05:35:26 +0000544 return Builder.CreateExtractElement(Base, Idx, "vecext");
545}
546
547/// VisitImplicitCastExpr - Implicit casts are the same as normal casts, but
548/// also handle things like function to pointer-to-function decay, and array to
549/// pointer decay.
550Value *ScalarExprEmitter::VisitImplicitCastExpr(const ImplicitCastExpr *E) {
551 const Expr *Op = E->getSubExpr();
552
553 // If this is due to array->pointer conversion, emit the array expression as
554 // an l-value.
555 if (Op->getType()->isArrayType()) {
Chris Lattnerfb182ee2007-08-26 16:34:22 +0000556 Value *V = EmitLValue(Op).getAddress(); // Bitfields can't be arrays.
Eli Friedman8fef47e2008-12-20 23:11:59 +0000557
Eli Friedman4a0073b2009-03-28 02:45:41 +0000558 // Note that VLA pointers are always decayed, so we don't need to do
559 // anything here.
Eli Friedman8fef47e2008-12-20 23:11:59 +0000560 if (!Op->getType()->isVariableArrayType()) {
561 assert(isa<llvm::PointerType>(V->getType()) && "Expected pointer");
562 assert(isa<llvm::ArrayType>(cast<llvm::PointerType>(V->getType())
563 ->getElementType()) &&
564 "Expected pointer to array");
565 V = Builder.CreateStructGEP(V, 0, "arraydecay");
Daniel Dunbar952f4732008-08-29 17:28:43 +0000566 }
Chris Lattnere54443b2007-12-12 04:13:20 +0000567
568 // The resultant pointer type can be implicitly casted to other pointer
Chris Lattner3b8f5c62008-07-23 06:31:27 +0000569 // types as well (e.g. void*) and can be implicitly converted to integer.
570 const llvm::Type *DestTy = ConvertType(E->getType());
571 if (V->getType() != DestTy) {
572 if (isa<llvm::PointerType>(DestTy))
573 V = Builder.CreateBitCast(V, DestTy, "ptrconv");
574 else {
575 assert(isa<llvm::IntegerType>(DestTy) && "Unknown array decay");
576 V = Builder.CreatePtrToInt(V, DestTy, "ptrconv");
577 }
578 }
Chris Lattnere54443b2007-12-12 04:13:20 +0000579 return V;
Chris Lattner9fba49a2007-08-24 05:35:26 +0000580 }
Eli Friedman4a0073b2009-03-28 02:45:41 +0000581
Chris Lattner9fba49a2007-08-24 05:35:26 +0000582 return EmitCastExpr(Op, E->getType());
583}
584
585
586// VisitCastExpr - Emit code for an explicit or implicit cast. Implicit casts
587// have to handle a more broad range of conversions than explicit casts, as they
588// handle things like function to ptr-to-function decay etc.
589Value *ScalarExprEmitter::EmitCastExpr(const Expr *E, QualType DestTy) {
Chris Lattner82e10392007-08-26 07:26:12 +0000590 // Handle cases where the source is an non-complex type.
Chris Lattner77288792008-02-16 23:55:16 +0000591
592 if (!CGF.hasAggregateLLVMType(E->getType())) {
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000593 Value *Src = Visit(const_cast<Expr*>(E));
594
Chris Lattner4e05d1e2007-08-26 06:48:56 +0000595 // Use EmitScalarConversion to perform the conversion.
596 return EmitScalarConversion(Src, E->getType(), DestTy);
597 }
Chris Lattner77288792008-02-16 23:55:16 +0000598
Chris Lattnerde0908b2008-04-04 16:54:41 +0000599 if (E->getType()->isAnyComplexType()) {
Chris Lattner77288792008-02-16 23:55:16 +0000600 // Handle cases where the source is a complex type.
601 return EmitComplexToScalarConversion(CGF.EmitComplexExpr(E), E->getType(),
602 DestTy);
603 }
Chris Lattnerd579f7f2007-08-26 07:16:41 +0000604
Chris Lattner77288792008-02-16 23:55:16 +0000605 // Okay, this is a cast from an aggregate. It must be a cast to void. Just
606 // evaluate the result and return.
607 CGF.EmitAggExpr(E, 0, false);
608 return 0;
Chris Lattner9fba49a2007-08-24 05:35:26 +0000609}
610
Chris Lattnerea6cdd72007-08-31 22:09:40 +0000611Value *ScalarExprEmitter::VisitStmtExpr(const StmtExpr *E) {
Chris Lattner09cee852008-07-26 20:23:23 +0000612 return CGF.EmitCompoundStmt(*E->getSubStmt(),
613 !E->getType()->isVoidType()).getScalarVal();
Chris Lattnerea6cdd72007-08-31 22:09:40 +0000614}
615
Mike Stump2b6933f2009-02-28 09:07:16 +0000616Value *ScalarExprEmitter::VisitBlockDeclRefExpr(const BlockDeclRefExpr *E) {
617 return Builder.CreateLoad(CGF.GetAddrOfBlockDecl(E), false, "tmp");
Mike Stumpfca5da02009-02-21 20:00:35 +0000618}
Chris Lattnerea6cdd72007-08-31 22:09:40 +0000619
Chris Lattner9fba49a2007-08-24 05:35:26 +0000620//===----------------------------------------------------------------------===//
621// Unary Operators
622//===----------------------------------------------------------------------===//
623
624Value *ScalarExprEmitter::VisitPrePostIncDec(const UnaryOperator *E,
Chris Lattner855e3d72007-08-24 16:24:49 +0000625 bool isInc, bool isPre) {
Chris Lattner9fba49a2007-08-24 05:35:26 +0000626 LValue LV = EmitLValue(E->getSubExpr());
Eli Friedman6a259872009-03-23 03:00:06 +0000627 QualType ValTy = E->getSubExpr()->getType();
628 Value *InVal = CGF.EmitLoadOfLValue(LV, ValTy).getScalarVal();
Chris Lattner9fba49a2007-08-24 05:35:26 +0000629
630 int AmountVal = isInc ? 1 : -1;
Eli Friedman4a0073b2009-03-28 02:45:41 +0000631
632 if (ValTy->isPointerType() &&
633 ValTy->getAsPointerType()->isVariableArrayType()) {
634 // The amount of the addition/subtraction needs to account for the VLA size
635 CGF.ErrorUnsupported(E, "VLA pointer inc/dec");
636 }
637
Chris Lattner9fba49a2007-08-24 05:35:26 +0000638 Value *NextVal;
Chris Lattner8360c612009-03-18 04:25:13 +0000639 if (const llvm::PointerType *PT =
640 dyn_cast<llvm::PointerType>(InVal->getType())) {
Chris Lattner8360c612009-03-18 04:25:13 +0000641 llvm::Constant *Inc =llvm::ConstantInt::get(llvm::Type::Int32Ty, AmountVal);
642 if (!isa<llvm::FunctionType>(PT->getElementType())) {
643 NextVal = Builder.CreateGEP(InVal, Inc, "ptrincdec");
644 } else {
645 const llvm::Type *i8Ty = llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
646 NextVal = Builder.CreateBitCast(InVal, i8Ty, "tmp");
647 NextVal = Builder.CreateGEP(NextVal, Inc, "ptrincdec");
648 NextVal = Builder.CreateBitCast(NextVal, InVal->getType());
649 }
Chris Lattner49083172009-02-11 07:40:06 +0000650 } else if (InVal->getType() == llvm::Type::Int1Ty && isInc) {
651 // Bool++ is an interesting case, due to promotion rules, we get:
652 // Bool++ -> Bool = Bool+1 -> Bool = (int)Bool+1 ->
653 // Bool = ((int)Bool+1) != 0
654 // An interesting aspect of this is that increment is always true.
655 // Decrement does not have this property.
656 NextVal = llvm::ConstantInt::getTrue();
Chris Lattner0dc11f62007-08-26 05:10:16 +0000657 } else {
658 // Add the inc/dec to the real part.
659 if (isa<llvm::IntegerType>(InVal->getType()))
660 NextVal = llvm::ConstantInt::get(InVal->getType(), AmountVal);
Chris Lattnerb2a7dab2007-09-13 06:19:18 +0000661 else if (InVal->getType() == llvm::Type::FloatTy)
Devang Patel0f2a8fb2007-10-30 20:59:40 +0000662 NextVal =
Chris Lattner70c38672008-04-20 00:45:53 +0000663 llvm::ConstantFP::get(llvm::APFloat(static_cast<float>(AmountVal)));
Chris Lattnerd54d1f22008-04-20 00:50:39 +0000664 else if (InVal->getType() == llvm::Type::DoubleTy)
Devang Patel0f2a8fb2007-10-30 20:59:40 +0000665 NextVal =
Chris Lattner70c38672008-04-20 00:45:53 +0000666 llvm::ConstantFP::get(llvm::APFloat(static_cast<double>(AmountVal)));
Chris Lattnerd54d1f22008-04-20 00:50:39 +0000667 else {
668 llvm::APFloat F(static_cast<float>(AmountVal));
Dale Johannesen2461f612008-10-09 23:02:32 +0000669 bool ignored;
670 F.convert(CGF.Target.getLongDoubleFormat(), llvm::APFloat::rmTowardZero,
671 &ignored);
Chris Lattnerd54d1f22008-04-20 00:50:39 +0000672 NextVal = llvm::ConstantFP::get(F);
Chris Lattnerb2a7dab2007-09-13 06:19:18 +0000673 }
Chris Lattner0dc11f62007-08-26 05:10:16 +0000674 NextVal = Builder.CreateAdd(InVal, NextVal, isInc ? "inc" : "dec");
675 }
Chris Lattner9fba49a2007-08-24 05:35:26 +0000676
677 // Store the updated result through the lvalue.
Eli Friedman6a259872009-03-23 03:00:06 +0000678 if (LV.isBitfield())
679 CGF.EmitStoreThroughBitfieldLValue(RValue::get(NextVal), LV, ValTy,
680 &NextVal);
681 else
682 CGF.EmitStoreThroughLValue(RValue::get(NextVal), LV, ValTy);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000683
684 // If this is a postinc, return the value read from memory, otherwise use the
685 // updated value.
686 return isPre ? NextVal : InVal;
687}
688
689
690Value *ScalarExprEmitter::VisitUnaryMinus(const UnaryOperator *E) {
691 Value *Op = Visit(E->getSubExpr());
692 return Builder.CreateNeg(Op, "neg");
693}
694
695Value *ScalarExprEmitter::VisitUnaryNot(const UnaryOperator *E) {
696 Value *Op = Visit(E->getSubExpr());
697 return Builder.CreateNot(Op, "neg");
698}
699
700Value *ScalarExprEmitter::VisitUnaryLNot(const UnaryOperator *E) {
701 // Compare operand to zero.
702 Value *BoolVal = CGF.EvaluateExprAsBool(E->getSubExpr());
703
704 // Invert value.
705 // TODO: Could dynamically modify easy computations here. For example, if
706 // the operand is an icmp ne, turn into icmp eq.
707 BoolVal = Builder.CreateNot(BoolVal, "lnot");
708
709 // ZExt result to int.
710 return Builder.CreateZExt(BoolVal, CGF.LLVMIntTy, "lnot.ext");
711}
712
Sebastian Redl0cb7c872008-11-11 17:56:53 +0000713/// VisitSizeOfAlignOfExpr - Return the size or alignment of the type of
714/// argument of the sizeof expression as an integer.
715Value *
716ScalarExprEmitter::VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E) {
Sebastian Redl0cb7c872008-11-11 17:56:53 +0000717 QualType TypeToSize = E->getTypeOfArgument();
Eli Friedman5a2c38f2009-01-24 22:19:05 +0000718 if (E->isSizeOf()) {
719 if (const VariableArrayType *VAT =
720 CGF.getContext().getAsVariableArrayType(TypeToSize)) {
721 if (E->isArgumentType()) {
722 // sizeof(type) - make sure to emit the VLA size.
723 CGF.EmitVLASize(TypeToSize);
724 }
Anders Carlssond309f572009-01-30 16:41:04 +0000725
Anders Carlsson8f30de92009-02-05 19:43:10 +0000726 return CGF.GetVLASize(VAT);
Anders Carlsson6cb99b72008-12-21 03:33:21 +0000727 }
Anders Carlsson9be6aaf2008-12-12 07:38:43 +0000728 }
Eli Friedman5a2c38f2009-01-24 22:19:05 +0000729
730 // If this isn't sizeof(vla), the result must be constant; use the
731 // constant folding logic so we don't have to duplicate it here.
732 Expr::EvalResult Result;
733 E->Evaluate(Result, CGF.getContext());
734 return llvm::ConstantInt::get(Result.Val.getInt());
Chris Lattner9fba49a2007-08-24 05:35:26 +0000735}
736
Chris Lattner01211af2007-08-24 21:20:17 +0000737Value *ScalarExprEmitter::VisitUnaryReal(const UnaryOperator *E) {
738 Expr *Op = E->getSubExpr();
Chris Lattnerde0908b2008-04-04 16:54:41 +0000739 if (Op->getType()->isAnyComplexType())
Chris Lattner01211af2007-08-24 21:20:17 +0000740 return CGF.EmitComplexExpr(Op).first;
741 return Visit(Op);
742}
743Value *ScalarExprEmitter::VisitUnaryImag(const UnaryOperator *E) {
744 Expr *Op = E->getSubExpr();
Chris Lattnerde0908b2008-04-04 16:54:41 +0000745 if (Op->getType()->isAnyComplexType())
Chris Lattner01211af2007-08-24 21:20:17 +0000746 return CGF.EmitComplexExpr(Op).second;
Chris Lattnerdb8a6c92007-08-26 05:29:21 +0000747
748 // __imag on a scalar returns zero. Emit it the subexpr to ensure side
749 // effects are evaluated.
750 CGF.EmitScalarExpr(Op);
751 return llvm::Constant::getNullValue(ConvertType(E->getType()));
Chris Lattner01211af2007-08-24 21:20:17 +0000752}
753
Anders Carlsson52774ad2008-01-29 15:56:48 +0000754Value *ScalarExprEmitter::VisitUnaryOffsetOf(const UnaryOperator *E)
755{
Eli Friedman342d9432009-02-27 06:44:11 +0000756 Value* ResultAsPtr = EmitLValue(E->getSubExpr()).getAddress();
Eli Friedmanccffea92009-01-24 22:38:55 +0000757 const llvm::Type* ResultType = ConvertType(E->getType());
Eli Friedman342d9432009-02-27 06:44:11 +0000758 return Builder.CreatePtrToInt(ResultAsPtr, ResultType, "offsetof");
Anders Carlsson52774ad2008-01-29 15:56:48 +0000759}
Chris Lattner01211af2007-08-24 21:20:17 +0000760
Chris Lattner9fba49a2007-08-24 05:35:26 +0000761//===----------------------------------------------------------------------===//
762// Binary Operators
763//===----------------------------------------------------------------------===//
764
765BinOpInfo ScalarExprEmitter::EmitBinOps(const BinaryOperator *E) {
766 BinOpInfo Result;
767 Result.LHS = Visit(E->getLHS());
768 Result.RHS = Visit(E->getRHS());
Chris Lattner660e31d2007-08-24 21:00:35 +0000769 Result.Ty = E->getType();
Chris Lattner9fba49a2007-08-24 05:35:26 +0000770 Result.E = E;
771 return Result;
772}
773
Chris Lattner0d965302007-08-26 21:41:21 +0000774Value *ScalarExprEmitter::EmitCompoundAssign(const CompoundAssignOperator *E,
Chris Lattner660e31d2007-08-24 21:00:35 +0000775 Value *(ScalarExprEmitter::*Func)(const BinOpInfo &)) {
776 QualType LHSTy = E->getLHS()->getType(), RHSTy = E->getRHS()->getType();
777
778 BinOpInfo OpInfo;
779
Eli Friedman3cd92882009-03-28 01:22:36 +0000780 if (E->getComputationResultType()->isAnyComplexType()) {
Eli Friedman4a0073b2009-03-28 02:45:41 +0000781 // This needs to go through the complex expression emitter, but
Eli Friedman3cd92882009-03-28 01:22:36 +0000782 // it's a tad complicated to do that... I'm leaving it out for now.
783 // (Note that we do actually need the imaginary part of the RHS for
784 // multiplication and division.)
785 CGF.ErrorUnsupported(E, "complex compound assignment");
786 return llvm::UndefValue::get(CGF.ConvertType(E->getType()));
787 }
788
789 // Load/convert the LHS.
Chris Lattner660e31d2007-08-24 21:00:35 +0000790 LValue LHSLV = EmitLValue(E->getLHS());
791 OpInfo.LHS = EmitLoadOfLValue(LHSLV, LHSTy);
Eli Friedman3cd92882009-03-28 01:22:36 +0000792 OpInfo.LHS = EmitScalarConversion(OpInfo.LHS, LHSTy,
793 E->getComputationLHSType());
794 // Emit the RHS.
795 OpInfo.RHS = Visit(E->getRHS());
796 OpInfo.Ty = E->getComputationResultType();
Chris Lattner660e31d2007-08-24 21:00:35 +0000797 OpInfo.E = E;
798
799 // Expand the binary operator.
800 Value *Result = (this->*Func)(OpInfo);
801
Daniel Dunbar5d7d0382008-08-06 02:00:38 +0000802 // Convert the result back to the LHS type.
Eli Friedman3cd92882009-03-28 01:22:36 +0000803 Result = EmitScalarConversion(Result, E->getComputationResultType(), LHSTy);
804
Daniel Dunbar2668dd12008-11-19 09:36:46 +0000805 // Store the result value into the LHS lvalue. Bit-fields are
Daniel Dunbar2710fc92008-11-19 11:54:05 +0000806 // handled specially because the result is altered by the store,
807 // i.e., [C99 6.5.16p1] 'An assignment expression has the value of
808 // the left operand after the assignment...'.
Eli Friedmanf9b930c2008-05-25 14:13:57 +0000809 if (LHSLV.isBitfield())
Daniel Dunbar2668dd12008-11-19 09:36:46 +0000810 CGF.EmitStoreThroughBitfieldLValue(RValue::get(Result), LHSLV, LHSTy,
811 &Result);
812 else
813 CGF.EmitStoreThroughLValue(RValue::get(Result), LHSLV, LHSTy);
814
Chris Lattner660e31d2007-08-24 21:00:35 +0000815 return Result;
816}
817
818
Chris Lattner9fba49a2007-08-24 05:35:26 +0000819Value *ScalarExprEmitter::EmitDiv(const BinOpInfo &Ops) {
Nate Begemanaade3bf2007-12-30 01:28:16 +0000820 if (Ops.LHS->getType()->isFPOrFPVector())
Chris Lattner9fba49a2007-08-24 05:35:26 +0000821 return Builder.CreateFDiv(Ops.LHS, Ops.RHS, "div");
Chris Lattner660e31d2007-08-24 21:00:35 +0000822 else if (Ops.Ty->isUnsignedIntegerType())
Chris Lattner9fba49a2007-08-24 05:35:26 +0000823 return Builder.CreateUDiv(Ops.LHS, Ops.RHS, "div");
824 else
825 return Builder.CreateSDiv(Ops.LHS, Ops.RHS, "div");
826}
827
828Value *ScalarExprEmitter::EmitRem(const BinOpInfo &Ops) {
829 // Rem in C can't be a floating point type: C99 6.5.5p2.
Chris Lattner660e31d2007-08-24 21:00:35 +0000830 if (Ops.Ty->isUnsignedIntegerType())
Chris Lattner9fba49a2007-08-24 05:35:26 +0000831 return Builder.CreateURem(Ops.LHS, Ops.RHS, "rem");
832 else
833 return Builder.CreateSRem(Ops.LHS, Ops.RHS, "rem");
834}
835
Mike Stumpdb789912009-04-01 20:28:16 +0000836Value *ScalarExprEmitter::EmitOverflowCheckedBinOp(const BinOpInfo &Ops) {
837 unsigned IID;
838 unsigned OpID = 0;
Mike Stump0f595bb2009-04-02 01:03:55 +0000839
Mike Stumpdb789912009-04-01 20:28:16 +0000840 if (Ops.Ty->isSignedIntegerType()) {
841 switch (Ops.E->getOpcode()) {
842 case BinaryOperator::Add:
Mike Stump0f595bb2009-04-02 01:03:55 +0000843 case BinaryOperator::AddAssign:
Mike Stumpdb789912009-04-01 20:28:16 +0000844 OpID = 1;
845 IID = llvm::Intrinsic::sadd_with_overflow;
846 break;
847 case BinaryOperator::Sub:
Mike Stump0f595bb2009-04-02 01:03:55 +0000848 case BinaryOperator::SubAssign:
Mike Stumpdb789912009-04-01 20:28:16 +0000849 OpID = 2;
850 IID = llvm::Intrinsic::ssub_with_overflow;
851 break;
852 case BinaryOperator::Mul:
Mike Stump0f595bb2009-04-02 01:03:55 +0000853 case BinaryOperator::MulAssign:
Mike Stumpdb789912009-04-01 20:28:16 +0000854 OpID = 3;
855 IID = llvm::Intrinsic::smul_with_overflow;
856 break;
857 default:
Mike Stump0f595bb2009-04-02 01:03:55 +0000858 fprintf(stderr, "Opcode: %d\n", Ops.E->getOpcode());
Mike Stumpdb789912009-04-01 20:28:16 +0000859 assert(false && "Unsupported operation for overflow detection");
860 }
861 OpID <<= 1;
862 OpID |= 1;
863 }
864 else {
865 assert(Ops.Ty->isUnsignedIntegerType() &&
866 "Must be either a signed or unsigned integer op");
867 switch (Ops.E->getOpcode()) {
868 case BinaryOperator::Add:
Mike Stump0f595bb2009-04-02 01:03:55 +0000869 case BinaryOperator::AddAssign:
Mike Stumpdb789912009-04-01 20:28:16 +0000870 OpID = 1;
871 IID = llvm::Intrinsic::uadd_with_overflow;
872 break;
873 case BinaryOperator::Sub:
Mike Stump0f595bb2009-04-02 01:03:55 +0000874 case BinaryOperator::SubAssign:
Mike Stumpdb789912009-04-01 20:28:16 +0000875 OpID = 2;
876 IID = llvm::Intrinsic::usub_with_overflow;
877 break;
878 case BinaryOperator::Mul:
Mike Stump0f595bb2009-04-02 01:03:55 +0000879 case BinaryOperator::MulAssign:
Mike Stumpdb789912009-04-01 20:28:16 +0000880 OpID = 3;
881 IID = llvm::Intrinsic::umul_with_overflow;
882 break;
883 default:
884 assert(false && "Unsupported operation for overflow detection");
885 }
886 OpID <<= 1;
887 }
888 const llvm::Type *opTy = CGF.CGM.getTypes().ConvertType(Ops.Ty);
889
890 llvm::Function *intrinsic = CGF.CGM.getIntrinsic(IID, &opTy, 1);
891
892 Value *resultAndOverflow = Builder.CreateCall2(intrinsic, Ops.LHS, Ops.RHS);
893 Value *result = Builder.CreateExtractValue(resultAndOverflow, 0);
894 Value *overflow = Builder.CreateExtractValue(resultAndOverflow, 1);
895
896 // Branch in case of overflow.
897 llvm::BasicBlock *initialBB = Builder.GetInsertBlock();
898 llvm::BasicBlock *overflowBB =
899 CGF.createBasicBlock("overflow", CGF.CurFn);
900 llvm::BasicBlock *continueBB =
901 CGF.createBasicBlock("overflow.continue", CGF.CurFn);
902
903 Builder.CreateCondBr(overflow, overflowBB, continueBB);
904
905 // Handle overflow
906
907 Builder.SetInsertPoint(overflowBB);
908
909 // Handler is:
910 // long long *__overflow_handler)(long long a, long long b, char op,
911 // char width)
912 std::vector<const llvm::Type*> handerArgTypes;
913 handerArgTypes.push_back(llvm::Type::Int64Ty);
914 handerArgTypes.push_back(llvm::Type::Int64Ty);
915 handerArgTypes.push_back(llvm::Type::Int8Ty);
916 handerArgTypes.push_back(llvm::Type::Int8Ty);
917 llvm::FunctionType *handlerTy = llvm::FunctionType::get(llvm::Type::Int64Ty,
918 handerArgTypes, false);
919 llvm::Value *handlerFunction =
920 CGF.CGM.getModule().getOrInsertGlobal("__overflow_handler",
921 llvm::PointerType::getUnqual(handlerTy));
922 handlerFunction = Builder.CreateLoad(handlerFunction);
923
924 llvm::Value *handlerResult = Builder.CreateCall4(handlerFunction,
925 Builder.CreateSExt(Ops.LHS, llvm::Type::Int64Ty),
926 Builder.CreateSExt(Ops.RHS, llvm::Type::Int64Ty),
927 llvm::ConstantInt::get(llvm::Type::Int8Ty, OpID),
928 llvm::ConstantInt::get(llvm::Type::Int8Ty,
929 cast<llvm::IntegerType>(opTy)->getBitWidth()));
930
931 handlerResult = Builder.CreateTrunc(handlerResult, opTy);
932
933 Builder.CreateBr(continueBB);
934
935 // Set up the continuation
936 Builder.SetInsertPoint(continueBB);
937 // Get the correct result
938 llvm::PHINode *phi = Builder.CreatePHI(opTy);
939 phi->reserveOperandSpace(2);
940 phi->addIncoming(result, initialBB);
941 phi->addIncoming(handlerResult, overflowBB);
942
943 return phi;
944}
Chris Lattner9fba49a2007-08-24 05:35:26 +0000945
946Value *ScalarExprEmitter::EmitAdd(const BinOpInfo &Ops) {
Mike Stumpdb789912009-04-01 20:28:16 +0000947 if (!Ops.Ty->isPointerType()) {
Mike Stump0f595bb2009-04-02 01:03:55 +0000948 if (CGF.getContext().getLangOptions().UnsignedOverflowChecking
949 || (CGF.getContext().getLangOptions().OverflowChecking
950 && Ops.Ty->isSignedIntegerType()))
Mike Stumpdb789912009-04-01 20:28:16 +0000951 return EmitOverflowCheckedBinOp(Ops);
Chris Lattner9fba49a2007-08-24 05:35:26 +0000952 return Builder.CreateAdd(Ops.LHS, Ops.RHS, "add");
Mike Stumpdb789912009-04-01 20:28:16 +0000953 }
Eli Friedman4a0073b2009-03-28 02:45:41 +0000954
955 if (Ops.Ty->getAsPointerType()->isVariableArrayType()) {
956 // The amount of the addition needs to account for the VLA size
957 CGF.ErrorUnsupported(Ops.E, "VLA pointer addition");
958 }
Chris Lattner17c0cb02008-01-03 06:36:51 +0000959 Value *Ptr, *Idx;
960 Expr *IdxExp;
Daniel Dunbar4fd58ab2009-01-23 18:51:09 +0000961 const PointerType *PT;
962 if ((PT = Ops.E->getLHS()->getType()->getAsPointerType())) {
Chris Lattner17c0cb02008-01-03 06:36:51 +0000963 Ptr = Ops.LHS;
964 Idx = Ops.RHS;
965 IdxExp = Ops.E->getRHS();
966 } else { // int + pointer
Daniel Dunbar4fd58ab2009-01-23 18:51:09 +0000967 PT = Ops.E->getRHS()->getType()->getAsPointerType();
968 assert(PT && "Invalid add expr");
Chris Lattner17c0cb02008-01-03 06:36:51 +0000969 Ptr = Ops.RHS;
970 Idx = Ops.LHS;
971 IdxExp = Ops.E->getLHS();
972 }
973
974 unsigned Width = cast<llvm::IntegerType>(Idx->getType())->getBitWidth();
975 if (Width < CGF.LLVMPointerWidth) {
976 // Zero or sign extend the pointer value based on whether the index is
977 // signed or not.
978 const llvm::Type *IdxType = llvm::IntegerType::get(CGF.LLVMPointerWidth);
Chris Lattnerc154ac12008-07-26 22:37:01 +0000979 if (IdxExp->getType()->isSignedIntegerType())
Chris Lattner17c0cb02008-01-03 06:36:51 +0000980 Idx = Builder.CreateSExt(Idx, IdxType, "idx.ext");
981 else
982 Idx = Builder.CreateZExt(Idx, IdxType, "idx.ext");
983 }
Daniel Dunbar4fd58ab2009-01-23 18:51:09 +0000984
985 // Explicitly handle GNU void* and function pointer arithmetic
986 // extensions. The GNU void* casts amount to no-ops since our void*
987 // type is i8*, but this is future proof.
988 const QualType ElementType = PT->getPointeeType();
989 if (ElementType->isVoidType() || ElementType->isFunctionType()) {
990 const llvm::Type *i8Ty = llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
991 Value *Casted = Builder.CreateBitCast(Ptr, i8Ty);
992 Value *Res = Builder.CreateGEP(Casted, Idx, "sub.ptr");
993 return Builder.CreateBitCast(Res, Ptr->getType());
994 }
Chris Lattner17c0cb02008-01-03 06:36:51 +0000995
996 return Builder.CreateGEP(Ptr, Idx, "add.ptr");
Chris Lattner9fba49a2007-08-24 05:35:26 +0000997}
998
999Value *ScalarExprEmitter::EmitSub(const BinOpInfo &Ops) {
Mike Stumpdb789912009-04-01 20:28:16 +00001000 if (!isa<llvm::PointerType>(Ops.LHS->getType())) {
Mike Stump0f595bb2009-04-02 01:03:55 +00001001 if (CGF.getContext().getLangOptions().UnsignedOverflowChecking
1002 || (CGF.getContext().getLangOptions().OverflowChecking
1003 && Ops.Ty->isSignedIntegerType()))
Mike Stumpdb789912009-04-01 20:28:16 +00001004 return EmitOverflowCheckedBinOp(Ops);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001005 return Builder.CreateSub(Ops.LHS, Ops.RHS, "sub");
Mike Stumpdb789912009-04-01 20:28:16 +00001006 }
Chris Lattner660e31d2007-08-24 21:00:35 +00001007
Eli Friedman4a0073b2009-03-28 02:45:41 +00001008 if (Ops.E->getLHS()->getType()->getAsPointerType()->isVariableArrayType()) {
1009 // The amount of the addition needs to account for the VLA size for
1010 // ptr-int
1011 // The amount of the division needs to account for the VLA size for
1012 // ptr-ptr.
1013 CGF.ErrorUnsupported(Ops.E, "VLA pointer subtraction");
1014 }
1015
Daniel Dunbar4fd58ab2009-01-23 18:51:09 +00001016 const QualType LHSType = Ops.E->getLHS()->getType();
1017 const QualType LHSElementType = LHSType->getAsPointerType()->getPointeeType();
Daniel Dunbar5d7d0382008-08-06 02:00:38 +00001018 if (!isa<llvm::PointerType>(Ops.RHS->getType())) {
1019 // pointer - int
1020 Value *Idx = Ops.RHS;
1021 unsigned Width = cast<llvm::IntegerType>(Idx->getType())->getBitWidth();
1022 if (Width < CGF.LLVMPointerWidth) {
1023 // Zero or sign extend the pointer value based on whether the index is
1024 // signed or not.
1025 const llvm::Type *IdxType = llvm::IntegerType::get(CGF.LLVMPointerWidth);
1026 if (Ops.E->getRHS()->getType()->isSignedIntegerType())
1027 Idx = Builder.CreateSExt(Idx, IdxType, "idx.ext");
1028 else
1029 Idx = Builder.CreateZExt(Idx, IdxType, "idx.ext");
1030 }
1031 Idx = Builder.CreateNeg(Idx, "sub.ptr.neg");
Daniel Dunbar4fd58ab2009-01-23 18:51:09 +00001032
1033 // Explicitly handle GNU void* and function pointer arithmetic
1034 // extensions. The GNU void* casts amount to no-ops since our
1035 // void* type is i8*, but this is future proof.
1036 if (LHSElementType->isVoidType() || LHSElementType->isFunctionType()) {
1037 const llvm::Type *i8Ty = llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
1038 Value *LHSCasted = Builder.CreateBitCast(Ops.LHS, i8Ty);
1039 Value *Res = Builder.CreateGEP(LHSCasted, Idx, "sub.ptr");
1040 return Builder.CreateBitCast(Res, Ops.LHS->getType());
1041 }
1042
Daniel Dunbar5d7d0382008-08-06 02:00:38 +00001043 return Builder.CreateGEP(Ops.LHS, Idx, "sub.ptr");
Daniel Dunbar0aac9f62008-08-05 00:47:03 +00001044 } else {
Daniel Dunbar5d7d0382008-08-06 02:00:38 +00001045 // pointer - pointer
1046 Value *LHS = Ops.LHS;
1047 Value *RHS = Ops.RHS;
Chris Lattner660e31d2007-08-24 21:00:35 +00001048
Daniel Dunbar5d7d0382008-08-06 02:00:38 +00001049 uint64_t ElementSize;
Daniel Dunbar0aac9f62008-08-05 00:47:03 +00001050
Chris Lattner6d2e3492009-02-11 07:21:43 +00001051 // Handle GCC extension for pointer arithmetic on void* and function pointer
1052 // types.
1053 if (LHSElementType->isVoidType() || LHSElementType->isFunctionType()) {
Daniel Dunbar5d7d0382008-08-06 02:00:38 +00001054 ElementSize = 1;
1055 } else {
1056 ElementSize = CGF.getContext().getTypeSize(LHSElementType) / 8;
1057 }
1058
1059 const llvm::Type *ResultType = ConvertType(Ops.Ty);
1060 LHS = Builder.CreatePtrToInt(LHS, ResultType, "sub.ptr.lhs.cast");
1061 RHS = Builder.CreatePtrToInt(RHS, ResultType, "sub.ptr.rhs.cast");
1062 Value *BytesBetween = Builder.CreateSub(LHS, RHS, "sub.ptr.sub");
1063
Chris Lattner6d2e3492009-02-11 07:21:43 +00001064 // Optimize out the shift for element size of 1.
1065 if (ElementSize == 1)
1066 return BytesBetween;
1067
Daniel Dunbar5d7d0382008-08-06 02:00:38 +00001068 // HACK: LLVM doesn't have an divide instruction that 'knows' there is no
1069 // remainder. As such, we handle common power-of-two cases here to generate
1070 // better code. See PR2247.
1071 if (llvm::isPowerOf2_64(ElementSize)) {
1072 Value *ShAmt =
1073 llvm::ConstantInt::get(ResultType, llvm::Log2_64(ElementSize));
1074 return Builder.CreateAShr(BytesBetween, ShAmt, "sub.ptr.shr");
1075 }
1076
1077 // Otherwise, do a full sdiv.
1078 Value *BytesPerElt = llvm::ConstantInt::get(ResultType, ElementSize);
1079 return Builder.CreateSDiv(BytesBetween, BytesPerElt, "sub.ptr.div");
Chris Lattner9fba49a2007-08-24 05:35:26 +00001080 }
Chris Lattner9fba49a2007-08-24 05:35:26 +00001081}
1082
1083Value *ScalarExprEmitter::EmitShl(const BinOpInfo &Ops) {
1084 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
1085 // RHS to the same size as the LHS.
1086 Value *RHS = Ops.RHS;
1087 if (Ops.LHS->getType() != RHS->getType())
1088 RHS = Builder.CreateIntCast(RHS, Ops.LHS->getType(), false, "sh_prom");
1089
1090 return Builder.CreateShl(Ops.LHS, RHS, "shl");
1091}
1092
1093Value *ScalarExprEmitter::EmitShr(const BinOpInfo &Ops) {
1094 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
1095 // RHS to the same size as the LHS.
1096 Value *RHS = Ops.RHS;
1097 if (Ops.LHS->getType() != RHS->getType())
1098 RHS = Builder.CreateIntCast(RHS, Ops.LHS->getType(), false, "sh_prom");
1099
Chris Lattner660e31d2007-08-24 21:00:35 +00001100 if (Ops.Ty->isUnsignedIntegerType())
Chris Lattner9fba49a2007-08-24 05:35:26 +00001101 return Builder.CreateLShr(Ops.LHS, RHS, "shr");
1102 return Builder.CreateAShr(Ops.LHS, RHS, "shr");
1103}
1104
1105Value *ScalarExprEmitter::EmitCompare(const BinaryOperator *E,unsigned UICmpOpc,
1106 unsigned SICmpOpc, unsigned FCmpOpc) {
Chris Lattnerfb182ee2007-08-26 16:34:22 +00001107 Value *Result;
Chris Lattner9fba49a2007-08-24 05:35:26 +00001108 QualType LHSTy = E->getLHS()->getType();
Nate Begeman1591bc52008-07-25 20:16:05 +00001109 if (!LHSTy->isAnyComplexType() && !LHSTy->isVectorType()) {
Chris Lattner9fba49a2007-08-24 05:35:26 +00001110 Value *LHS = Visit(E->getLHS());
1111 Value *RHS = Visit(E->getRHS());
1112
1113 if (LHS->getType()->isFloatingPoint()) {
Nate Begeman1591bc52008-07-25 20:16:05 +00001114 Result = Builder.CreateFCmp((llvm::CmpInst::Predicate)FCmpOpc,
Chris Lattner9fba49a2007-08-24 05:35:26 +00001115 LHS, RHS, "cmp");
Eli Friedman850ea372008-05-29 15:09:15 +00001116 } else if (LHSTy->isSignedIntegerType()) {
1117 Result = Builder.CreateICmp((llvm::ICmpInst::Predicate)SICmpOpc,
Chris Lattner9fba49a2007-08-24 05:35:26 +00001118 LHS, RHS, "cmp");
1119 } else {
Eli Friedman850ea372008-05-29 15:09:15 +00001120 // Unsigned integers and pointers.
1121 Result = Builder.CreateICmp((llvm::ICmpInst::Predicate)UICmpOpc,
Chris Lattner9fba49a2007-08-24 05:35:26 +00001122 LHS, RHS, "cmp");
1123 }
Nate Begeman1591bc52008-07-25 20:16:05 +00001124 } else if (LHSTy->isVectorType()) {
1125 Value *LHS = Visit(E->getLHS());
1126 Value *RHS = Visit(E->getRHS());
1127
1128 if (LHS->getType()->isFPOrFPVector()) {
1129 Result = Builder.CreateVFCmp((llvm::CmpInst::Predicate)FCmpOpc,
1130 LHS, RHS, "cmp");
1131 } else if (LHSTy->isUnsignedIntegerType()) {
1132 Result = Builder.CreateVICmp((llvm::CmpInst::Predicate)UICmpOpc,
1133 LHS, RHS, "cmp");
1134 } else {
1135 // Signed integers and pointers.
1136 Result = Builder.CreateVICmp((llvm::CmpInst::Predicate)SICmpOpc,
1137 LHS, RHS, "cmp");
1138 }
1139 return Result;
Chris Lattner9fba49a2007-08-24 05:35:26 +00001140 } else {
1141 // Complex Comparison: can only be an equality comparison.
1142 CodeGenFunction::ComplexPairTy LHS = CGF.EmitComplexExpr(E->getLHS());
1143 CodeGenFunction::ComplexPairTy RHS = CGF.EmitComplexExpr(E->getRHS());
1144
Chris Lattnerc154ac12008-07-26 22:37:01 +00001145 QualType CETy = LHSTy->getAsComplexType()->getElementType();
Chris Lattner9fba49a2007-08-24 05:35:26 +00001146
Chris Lattnerfb182ee2007-08-26 16:34:22 +00001147 Value *ResultR, *ResultI;
Chris Lattner9fba49a2007-08-24 05:35:26 +00001148 if (CETy->isRealFloatingType()) {
1149 ResultR = Builder.CreateFCmp((llvm::FCmpInst::Predicate)FCmpOpc,
1150 LHS.first, RHS.first, "cmp.r");
1151 ResultI = Builder.CreateFCmp((llvm::FCmpInst::Predicate)FCmpOpc,
1152 LHS.second, RHS.second, "cmp.i");
1153 } else {
1154 // Complex comparisons can only be equality comparisons. As such, signed
1155 // and unsigned opcodes are the same.
1156 ResultR = Builder.CreateICmp((llvm::ICmpInst::Predicate)UICmpOpc,
1157 LHS.first, RHS.first, "cmp.r");
1158 ResultI = Builder.CreateICmp((llvm::ICmpInst::Predicate)UICmpOpc,
1159 LHS.second, RHS.second, "cmp.i");
1160 }
1161
1162 if (E->getOpcode() == BinaryOperator::EQ) {
1163 Result = Builder.CreateAnd(ResultR, ResultI, "and.ri");
1164 } else {
1165 assert(E->getOpcode() == BinaryOperator::NE &&
1166 "Complex comparison other than == or != ?");
1167 Result = Builder.CreateOr(ResultR, ResultI, "or.ri");
1168 }
1169 }
Nuno Lopes92577002009-01-11 23:22:37 +00001170
1171 return EmitScalarConversion(Result, CGF.getContext().BoolTy, E->getType());
Chris Lattner9fba49a2007-08-24 05:35:26 +00001172}
1173
1174Value *ScalarExprEmitter::VisitBinAssign(const BinaryOperator *E) {
1175 LValue LHS = EmitLValue(E->getLHS());
1176 Value *RHS = Visit(E->getRHS());
1177
Daniel Dunbar2668dd12008-11-19 09:36:46 +00001178 // Store the value into the LHS. Bit-fields are handled specially
Daniel Dunbar2710fc92008-11-19 11:54:05 +00001179 // because the result is altered by the store, i.e., [C99 6.5.16p1]
1180 // 'An assignment expression has the value of the left operand after
Eli Friedman4a0073b2009-03-28 02:45:41 +00001181 // the assignment...'.
Eli Friedmanf9b930c2008-05-25 14:13:57 +00001182 if (LHS.isBitfield())
Daniel Dunbar2668dd12008-11-19 09:36:46 +00001183 CGF.EmitStoreThroughBitfieldLValue(RValue::get(RHS), LHS, E->getType(),
1184 &RHS);
1185 else
1186 CGF.EmitStoreThroughLValue(RValue::get(RHS), LHS, E->getType());
Daniel Dunbare6c31752008-08-29 08:11:39 +00001187
Chris Lattner9fba49a2007-08-24 05:35:26 +00001188 // Return the RHS.
1189 return RHS;
1190}
1191
1192Value *ScalarExprEmitter::VisitBinLAnd(const BinaryOperator *E) {
Chris Lattner715c2a72008-11-12 08:26:50 +00001193 // If we have 0 && RHS, see if we can elide RHS, if so, just return 0.
1194 // If we have 1 && X, just emit X without inserting the control flow.
1195 if (int Cond = CGF.ConstantFoldsToSimpleInteger(E->getLHS())) {
1196 if (Cond == 1) { // If we have 1 && X, just emit X.
Chris Lattner3f73d0d2008-11-11 07:41:27 +00001197 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1198 // ZExt result to int.
1199 return Builder.CreateZExt(RHSCond, CGF.LLVMIntTy, "land.ext");
1200 }
Chris Lattner715c2a72008-11-12 08:26:50 +00001201
1202 // 0 && RHS: If it is safe, just elide the RHS, and return 0.
1203 if (!CGF.ContainsLabel(E->getRHS()))
1204 return llvm::Constant::getNullValue(CGF.LLVMIntTy);
Chris Lattner3f73d0d2008-11-11 07:41:27 +00001205 }
1206
Daniel Dunbar6e3a10c2008-11-13 01:38:36 +00001207 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("land.end");
1208 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("land.rhs");
Chris Lattner715c2a72008-11-12 08:26:50 +00001209
Chris Lattner7f80bb32008-11-12 08:38:24 +00001210 // Branch on the LHS first. If it is false, go to the failure (cont) block.
1211 CGF.EmitBranchOnBoolExpr(E->getLHS(), RHSBlock, ContBlock);
1212
1213 // Any edges into the ContBlock are now from an (indeterminate number of)
1214 // edges from this first condition. All of these values will be false. Start
1215 // setting up the PHI node in the Cont Block for this.
1216 llvm::PHINode *PN = llvm::PHINode::Create(llvm::Type::Int1Ty, "", ContBlock);
1217 PN->reserveOperandSpace(2); // Normal case, two inputs.
1218 for (llvm::pred_iterator PI = pred_begin(ContBlock), PE = pred_end(ContBlock);
1219 PI != PE; ++PI)
1220 PN->addIncoming(llvm::ConstantInt::getFalse(), *PI);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001221
1222 CGF.EmitBlock(RHSBlock);
1223 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1224
1225 // Reaquire the RHS block, as there may be subblocks inserted.
1226 RHSBlock = Builder.GetInsertBlock();
Chris Lattner7f80bb32008-11-12 08:38:24 +00001227
1228 // Emit an unconditional branch from this block to ContBlock. Insert an entry
1229 // into the phi node for the edge with the value of RHSCond.
Chris Lattner9fba49a2007-08-24 05:35:26 +00001230 CGF.EmitBlock(ContBlock);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001231 PN->addIncoming(RHSCond, RHSBlock);
1232
1233 // ZExt result to int.
1234 return Builder.CreateZExt(PN, CGF.LLVMIntTy, "land.ext");
1235}
1236
1237Value *ScalarExprEmitter::VisitBinLOr(const BinaryOperator *E) {
Chris Lattner715c2a72008-11-12 08:26:50 +00001238 // If we have 1 || RHS, see if we can elide RHS, if so, just return 1.
1239 // If we have 0 || X, just emit X without inserting the control flow.
1240 if (int Cond = CGF.ConstantFoldsToSimpleInteger(E->getLHS())) {
1241 if (Cond == -1) { // If we have 0 || X, just emit X.
Chris Lattner3f73d0d2008-11-11 07:41:27 +00001242 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1243 // ZExt result to int.
1244 return Builder.CreateZExt(RHSCond, CGF.LLVMIntTy, "lor.ext");
1245 }
Chris Lattner715c2a72008-11-12 08:26:50 +00001246
Eli Friedmanea137cd2008-12-02 16:02:46 +00001247 // 1 || RHS: If it is safe, just elide the RHS, and return 1.
Chris Lattner715c2a72008-11-12 08:26:50 +00001248 if (!CGF.ContainsLabel(E->getRHS()))
Eli Friedmanea137cd2008-12-02 16:02:46 +00001249 return llvm::ConstantInt::get(CGF.LLVMIntTy, 1);
Chris Lattner3f73d0d2008-11-11 07:41:27 +00001250 }
1251
Daniel Dunbar6e3a10c2008-11-13 01:38:36 +00001252 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("lor.end");
1253 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("lor.rhs");
Chris Lattner9fba49a2007-08-24 05:35:26 +00001254
Chris Lattner7f80bb32008-11-12 08:38:24 +00001255 // Branch on the LHS first. If it is true, go to the success (cont) block.
1256 CGF.EmitBranchOnBoolExpr(E->getLHS(), ContBlock, RHSBlock);
1257
1258 // Any edges into the ContBlock are now from an (indeterminate number of)
1259 // edges from this first condition. All of these values will be true. Start
1260 // setting up the PHI node in the Cont Block for this.
1261 llvm::PHINode *PN = llvm::PHINode::Create(llvm::Type::Int1Ty, "", ContBlock);
1262 PN->reserveOperandSpace(2); // Normal case, two inputs.
1263 for (llvm::pred_iterator PI = pred_begin(ContBlock), PE = pred_end(ContBlock);
1264 PI != PE; ++PI)
1265 PN->addIncoming(llvm::ConstantInt::getTrue(), *PI);
1266
1267 // Emit the RHS condition as a bool value.
Chris Lattner9fba49a2007-08-24 05:35:26 +00001268 CGF.EmitBlock(RHSBlock);
1269 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1270
1271 // Reaquire the RHS block, as there may be subblocks inserted.
1272 RHSBlock = Builder.GetInsertBlock();
Chris Lattner9fba49a2007-08-24 05:35:26 +00001273
Chris Lattner7f80bb32008-11-12 08:38:24 +00001274 // Emit an unconditional branch from this block to ContBlock. Insert an entry
1275 // into the phi node for the edge with the value of RHSCond.
1276 CGF.EmitBlock(ContBlock);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001277 PN->addIncoming(RHSCond, RHSBlock);
1278
1279 // ZExt result to int.
1280 return Builder.CreateZExt(PN, CGF.LLVMIntTy, "lor.ext");
1281}
1282
1283Value *ScalarExprEmitter::VisitBinComma(const BinaryOperator *E) {
1284 CGF.EmitStmt(E->getLHS());
Daniel Dunbar5aa22bc2008-11-11 23:11:34 +00001285 CGF.EnsureInsertPoint();
Chris Lattner9fba49a2007-08-24 05:35:26 +00001286 return Visit(E->getRHS());
1287}
1288
1289//===----------------------------------------------------------------------===//
1290// Other Operators
1291//===----------------------------------------------------------------------===//
1292
Chris Lattner504a5282008-11-12 08:55:54 +00001293/// isCheapEnoughToEvaluateUnconditionally - Return true if the specified
1294/// expression is cheap enough and side-effect-free enough to evaluate
1295/// unconditionally instead of conditionally. This is used to convert control
1296/// flow into selects in some cases.
1297static bool isCheapEnoughToEvaluateUnconditionally(const Expr *E) {
1298 if (const ParenExpr *PE = dyn_cast<ParenExpr>(E))
1299 return isCheapEnoughToEvaluateUnconditionally(PE->getSubExpr());
1300
1301 // TODO: Allow anything we can constant fold to an integer or fp constant.
1302 if (isa<IntegerLiteral>(E) || isa<CharacterLiteral>(E) ||
1303 isa<FloatingLiteral>(E))
1304 return true;
1305
1306 // Non-volatile automatic variables too, to get "cond ? X : Y" where
1307 // X and Y are local variables.
1308 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
1309 if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
1310 if (VD->hasLocalStorage() && !VD->getType().isVolatileQualified())
1311 return true;
1312
1313 return false;
1314}
1315
1316
Chris Lattner9fba49a2007-08-24 05:35:26 +00001317Value *ScalarExprEmitter::
1318VisitConditionalOperator(const ConditionalOperator *E) {
Chris Lattner3d6606b2008-11-12 08:04:58 +00001319 // If the condition constant folds and can be elided, try to avoid emitting
1320 // the condition and the dead arm.
1321 if (int Cond = CGF.ConstantFoldsToSimpleInteger(E->getCond())){
Chris Lattner044bffc2008-11-11 18:56:45 +00001322 Expr *Live = E->getLHS(), *Dead = E->getRHS();
Chris Lattner3d6606b2008-11-12 08:04:58 +00001323 if (Cond == -1)
Chris Lattner044bffc2008-11-11 18:56:45 +00001324 std::swap(Live, Dead);
Chris Lattner3d6606b2008-11-12 08:04:58 +00001325
1326 // If the dead side doesn't have labels we need, and if the Live side isn't
1327 // the gnu missing ?: extension (which we could handle, but don't bother
1328 // to), just emit the Live part.
1329 if ((!Dead || !CGF.ContainsLabel(Dead)) && // No labels in dead part
1330 Live) // Live part isn't missing.
1331 return Visit(Live);
Chris Lattner044bffc2008-11-11 18:56:45 +00001332 }
1333
Chris Lattner504a5282008-11-12 08:55:54 +00001334
1335 // If this is a really simple expression (like x ? 4 : 5), emit this as a
1336 // select instead of as control flow. We can only do this if it is cheap and
Chris Lattner1f11af22008-11-16 06:16:27 +00001337 // safe to evaluate the LHS and RHS unconditionally.
Chris Lattner504a5282008-11-12 08:55:54 +00001338 if (E->getLHS() && isCheapEnoughToEvaluateUnconditionally(E->getLHS()) &&
1339 isCheapEnoughToEvaluateUnconditionally(E->getRHS())) {
1340 llvm::Value *CondV = CGF.EvaluateExprAsBool(E->getCond());
1341 llvm::Value *LHS = Visit(E->getLHS());
1342 llvm::Value *RHS = Visit(E->getRHS());
1343 return Builder.CreateSelect(CondV, LHS, RHS, "cond");
1344 }
1345
1346
Daniel Dunbarb23e9922008-11-12 10:13:37 +00001347 llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true");
1348 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false");
Daniel Dunbar6e3a10c2008-11-13 01:38:36 +00001349 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end");
Chris Lattner67e22462008-11-12 08:08:13 +00001350 Value *CondVal = 0;
Chris Lattner3d6606b2008-11-12 08:04:58 +00001351
Chris Lattner86031712009-02-13 23:35:32 +00001352 // If we don't have the GNU missing condition extension, emit a branch on
1353 // bool the normal way.
1354 if (E->getLHS()) {
1355 // Otherwise, just use EmitBranchOnBoolExpr to get small and simple code for
1356 // the branch on bool.
1357 CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock);
1358 } else {
1359 // Otherwise, for the ?: extension, evaluate the conditional and then
1360 // convert it to bool the hard way. We do this explicitly because we need
1361 // the unconverted value for the missing middle value of the ?:.
Chris Lattner67e22462008-11-12 08:08:13 +00001362 CondVal = CGF.EmitScalarExpr(E->getCond());
Chris Lattner86031712009-02-13 23:35:32 +00001363
1364 // In some cases, EmitScalarConversion will delete the "CondVal" expression
1365 // if there are no extra uses (an optimization). Inhibit this by making an
1366 // extra dead use, because we're going to add a use of CondVal later. We
1367 // don't use the builder for this, because we don't want it to get optimized
1368 // away. This leaves dead code, but the ?: extension isn't common.
1369 new llvm::BitCastInst(CondVal, CondVal->getType(), "dummy?:holder",
1370 Builder.GetInsertBlock());
1371
Chris Lattner67e22462008-11-12 08:08:13 +00001372 Value *CondBoolVal =
1373 CGF.EmitScalarConversion(CondVal, E->getCond()->getType(),
1374 CGF.getContext().BoolTy);
1375 Builder.CreateCondBr(CondBoolVal, LHSBlock, RHSBlock);
Chris Lattner67e22462008-11-12 08:08:13 +00001376 }
Chris Lattner9fba49a2007-08-24 05:35:26 +00001377
1378 CGF.EmitBlock(LHSBlock);
1379
1380 // Handle the GNU extension for missing LHS.
Chris Lattner98a425c2007-11-26 01:40:58 +00001381 Value *LHS;
1382 if (E->getLHS())
Eli Friedmance8d7032008-05-16 20:38:39 +00001383 LHS = Visit(E->getLHS());
Chris Lattner98a425c2007-11-26 01:40:58 +00001384 else // Perform promotions, to handle cases like "short ?: int"
1385 LHS = EmitScalarConversion(CondVal, E->getCond()->getType(), E->getType());
1386
Chris Lattner9fba49a2007-08-24 05:35:26 +00001387 LHSBlock = Builder.GetInsertBlock();
Daniel Dunbar5276caa2008-11-11 09:41:28 +00001388 CGF.EmitBranch(ContBlock);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001389
1390 CGF.EmitBlock(RHSBlock);
1391
Eli Friedmance8d7032008-05-16 20:38:39 +00001392 Value *RHS = Visit(E->getRHS());
Chris Lattner9fba49a2007-08-24 05:35:26 +00001393 RHSBlock = Builder.GetInsertBlock();
Daniel Dunbar5276caa2008-11-11 09:41:28 +00001394 CGF.EmitBranch(ContBlock);
Chris Lattner9fba49a2007-08-24 05:35:26 +00001395
1396 CGF.EmitBlock(ContBlock);
1397
Nuno Lopesb62ff242008-06-04 19:15:45 +00001398 if (!LHS || !RHS) {
Chris Lattner307da022007-11-30 17:56:23 +00001399 assert(E->getType()->isVoidType() && "Non-void value should have a value");
1400 return 0;
1401 }
1402
Chris Lattner9fba49a2007-08-24 05:35:26 +00001403 // Create a PHI node for the real part.
1404 llvm::PHINode *PN = Builder.CreatePHI(LHS->getType(), "cond");
1405 PN->reserveOperandSpace(2);
1406 PN->addIncoming(LHS, LHSBlock);
1407 PN->addIncoming(RHS, RHSBlock);
1408 return PN;
1409}
1410
1411Value *ScalarExprEmitter::VisitChooseExpr(ChooseExpr *E) {
Eli Friedmand540c112009-03-04 05:52:32 +00001412 return Visit(E->getChosenSubExpr(CGF.getContext()));
Chris Lattner9fba49a2007-08-24 05:35:26 +00001413}
1414
Chris Lattner307da022007-11-30 17:56:23 +00001415Value *ScalarExprEmitter::VisitVAArgExpr(VAArgExpr *VE) {
Eli Friedman8f5e8782009-01-20 17:46:04 +00001416 llvm::Value *ArgValue = CGF.EmitVAListRef(VE->getSubExpr());
Anders Carlsson285611e2008-11-04 05:30:00 +00001417 llvm::Value *ArgPtr = CGF.EmitVAArg(ArgValue, VE->getType());
1418
1419 // If EmitVAArg fails, we fall back to the LLVM instruction.
1420 if (!ArgPtr)
1421 return Builder.CreateVAArg(ArgValue, ConvertType(VE->getType()));
1422
Anders Carlsson285611e2008-11-04 05:30:00 +00001423 return Builder.CreateLoad(ArgPtr);
Anders Carlsson36760332007-10-15 20:28:48 +00001424}
1425
Mike Stump4eb81dc2009-02-12 18:29:15 +00001426Value *ScalarExprEmitter::VisitBlockExpr(const BlockExpr *BE) {
Mike Stump1fa52fe2009-03-07 02:35:30 +00001427 return CGF.BuildBlockLiteralTmp(BE);
Mike Stump4eb81dc2009-02-12 18:29:15 +00001428}
1429
Chris Lattner9fba49a2007-08-24 05:35:26 +00001430//===----------------------------------------------------------------------===//
1431// Entry Point into this File
1432//===----------------------------------------------------------------------===//
1433
1434/// EmitComplexExpr - Emit the computation of the specified expression of
1435/// complex type, ignoring the result.
1436Value *CodeGenFunction::EmitScalarExpr(const Expr *E) {
1437 assert(E && !hasAggregateLLVMType(E->getType()) &&
1438 "Invalid scalar expression to emit");
1439
1440 return ScalarExprEmitter(*this).Visit(const_cast<Expr*>(E));
1441}
Chris Lattner4e05d1e2007-08-26 06:48:56 +00001442
1443/// EmitScalarConversion - Emit a conversion from the specified type to the
1444/// specified destination type, both of which are LLVM scalar types.
Chris Lattnerfb182ee2007-08-26 16:34:22 +00001445Value *CodeGenFunction::EmitScalarConversion(Value *Src, QualType SrcTy,
1446 QualType DstTy) {
Chris Lattner4e05d1e2007-08-26 06:48:56 +00001447 assert(!hasAggregateLLVMType(SrcTy) && !hasAggregateLLVMType(DstTy) &&
1448 "Invalid scalar expression to emit");
1449 return ScalarExprEmitter(*this).EmitScalarConversion(Src, SrcTy, DstTy);
1450}
Chris Lattnerfb182ee2007-08-26 16:34:22 +00001451
1452/// EmitComplexToScalarConversion - Emit a conversion from the specified
1453/// complex type to the specified destination type, where the destination
1454/// type is an LLVM scalar type.
1455Value *CodeGenFunction::EmitComplexToScalarConversion(ComplexPairTy Src,
1456 QualType SrcTy,
1457 QualType DstTy) {
Chris Lattnerde0908b2008-04-04 16:54:41 +00001458 assert(SrcTy->isAnyComplexType() && !hasAggregateLLVMType(DstTy) &&
Chris Lattnerfb182ee2007-08-26 16:34:22 +00001459 "Invalid complex -> scalar conversion");
1460 return ScalarExprEmitter(*this).EmitComplexToScalarConversion(Src, SrcTy,
1461 DstTy);
1462}
Anders Carlssona9234fe2007-12-10 19:35:18 +00001463
1464Value *CodeGenFunction::EmitShuffleVector(Value* V1, Value *V2, ...) {
1465 assert(V1->getType() == V2->getType() &&
1466 "Vector operands must be of the same type");
Anders Carlssona9234fe2007-12-10 19:35:18 +00001467 unsigned NumElements =
1468 cast<llvm::VectorType>(V1->getType())->getNumElements();
1469
1470 va_list va;
1471 va_start(va, V2);
1472
1473 llvm::SmallVector<llvm::Constant*, 16> Args;
Anders Carlssona9234fe2007-12-10 19:35:18 +00001474 for (unsigned i = 0; i < NumElements; i++) {
1475 int n = va_arg(va, int);
Anders Carlssona9234fe2007-12-10 19:35:18 +00001476 assert(n >= 0 && n < (int)NumElements * 2 &&
1477 "Vector shuffle index out of bounds!");
Anders Carlssona9234fe2007-12-10 19:35:18 +00001478 Args.push_back(llvm::ConstantInt::get(llvm::Type::Int32Ty, n));
1479 }
1480
1481 const char *Name = va_arg(va, const char *);
1482 va_end(va);
1483
1484 llvm::Constant *Mask = llvm::ConstantVector::get(&Args[0], NumElements);
1485
1486 return Builder.CreateShuffleVector(V1, V2, Mask, Name);
1487}
1488
Anders Carlsson68b8be92007-12-15 21:23:30 +00001489llvm::Value *CodeGenFunction::EmitVector(llvm::Value * const *Vals,
Chris Lattnera23eb7b2008-07-26 20:15:14 +00001490 unsigned NumVals, bool isSplat) {
Anders Carlsson68b8be92007-12-15 21:23:30 +00001491 llvm::Value *Vec
Chris Lattnera23eb7b2008-07-26 20:15:14 +00001492 = llvm::UndefValue::get(llvm::VectorType::get(Vals[0]->getType(), NumVals));
Anders Carlsson68b8be92007-12-15 21:23:30 +00001493
Chris Lattnera23eb7b2008-07-26 20:15:14 +00001494 for (unsigned i = 0, e = NumVals; i != e; ++i) {
Nate Begemanec2d1062007-12-30 02:59:45 +00001495 llvm::Value *Val = isSplat ? Vals[0] : Vals[i];
Anders Carlsson68b8be92007-12-15 21:23:30 +00001496 llvm::Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, i);
Nate Begemanec2d1062007-12-30 02:59:45 +00001497 Vec = Builder.CreateInsertElement(Vec, Val, Idx, "tmp");
Anders Carlsson68b8be92007-12-15 21:23:30 +00001498 }
1499
1500 return Vec;
1501}