blob: 7f417b374aa1669720f31a618c317bb8dd2f25de [file] [log] [blame]
Chris Lattner7f02f722007-08-24 05:35:26 +00001//===--- CGExprScalar.cpp - Emit LLVM Code for Scalar Exprs ---------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner0bc735f2007-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 Lattner7f02f722007-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 Dunbarde7fb842008-08-11 05:00:27 +000016#include "clang/AST/ASTContext.h"
Daniel Dunbar98c5ead2008-08-12 05:08:18 +000017#include "clang/AST/DeclObjC.h"
Daniel Dunbarde7fb842008-08-11 05:00:27 +000018#include "clang/AST/StmtVisitor.h"
Chris Lattner25ddea72008-04-20 00:50:39 +000019#include "clang/Basic/TargetInfo.h"
Chris Lattner7f02f722007-08-24 05:35:26 +000020#include "llvm/Constants.h"
21#include "llvm/Function.h"
Anders Carlsson85f9bce2007-10-29 05:01:08 +000022#include "llvm/GlobalVariable.h"
Anders Carlsson7c50aca2007-10-15 20:28:48 +000023#include "llvm/Intrinsics.h"
Chris Lattner7f02f722007-08-24 05:35:26 +000024#include "llvm/Support/Compiler.h"
Chris Lattnerf7b5ea92008-11-12 08:38:24 +000025#include "llvm/Support/CFG.h"
Chris Lattnerc89bf692008-01-03 07:05:49 +000026#include <cstdarg>
Ted Kremenek6aad91a2007-12-10 23:44:32 +000027
Chris Lattner7f02f722007-08-24 05:35:26 +000028using namespace clang;
29using namespace CodeGen;
30using llvm::Value;
31
32//===----------------------------------------------------------------------===//
33// Scalar Expression Emitter
34//===----------------------------------------------------------------------===//
35
36struct BinOpInfo {
37 Value *LHS;
38 Value *RHS;
Chris Lattner1f1ded92007-08-24 21:00:35 +000039 QualType Ty; // Computation Type.
Chris Lattner7f02f722007-08-24 05:35:26 +000040 const BinaryOperator *E;
41};
42
43namespace {
44class VISIBILITY_HIDDEN ScalarExprEmitter
45 : public StmtVisitor<ScalarExprEmitter, Value*> {
46 CodeGenFunction &CGF;
Daniel Dunbar45d196b2008-11-01 01:53:16 +000047 CGBuilderTy &Builder;
Chris Lattner2b94fe32008-03-01 08:45:05 +000048
Chris Lattner7f02f722007-08-24 05:35:26 +000049public:
50
Chris Lattner2b94fe32008-03-01 08:45:05 +000051 ScalarExprEmitter(CodeGenFunction &cgf) : CGF(cgf),
Daniel Dunbared7c6182008-08-20 00:28:19 +000052 Builder(CGF.Builder) {
Chris Lattner7f02f722007-08-24 05:35:26 +000053 }
Chris Lattner7f02f722007-08-24 05:35:26 +000054
55 //===--------------------------------------------------------------------===//
56 // Utilities
57 //===--------------------------------------------------------------------===//
58
59 const llvm::Type *ConvertType(QualType T) { return CGF.ConvertType(T); }
60 LValue EmitLValue(const Expr *E) { return CGF.EmitLValue(E); }
61
62 Value *EmitLoadOfLValue(LValue LV, QualType T) {
Chris Lattner9b655512007-08-31 22:49:20 +000063 return CGF.EmitLoadOfLValue(LV, T).getScalarVal();
Chris Lattner7f02f722007-08-24 05:35:26 +000064 }
65
66 /// EmitLoadOfLValue - Given an expression with complex type that represents a
67 /// value l-value, this method emits the address of the l-value, then loads
68 /// and returns the result.
69 Value *EmitLoadOfLValue(const Expr *E) {
70 // FIXME: Volatile
71 return EmitLoadOfLValue(EmitLValue(E), E->getType());
72 }
73
Chris Lattner9abc84e2007-08-26 16:42:57 +000074 /// EmitConversionToBool - Convert the specified expression value to a
Chris Lattner3420d0d2007-08-26 17:25:57 +000075 /// boolean (i1) truth value. This is equivalent to "Val != 0".
Chris Lattner9abc84e2007-08-26 16:42:57 +000076 Value *EmitConversionToBool(Value *Src, QualType DstTy);
77
Chris Lattner3707b252007-08-26 06:48:56 +000078 /// EmitScalarConversion - Emit a conversion from the specified type to the
79 /// specified destination type, both of which are LLVM scalar types.
Chris Lattner4f1a7b32007-08-26 16:34:22 +000080 Value *EmitScalarConversion(Value *Src, QualType SrcTy, QualType DstTy);
81
82 /// EmitComplexToScalarConversion - Emit a conversion from the specified
83 /// complex type to the specified destination type, where the destination
84 /// type is an LLVM scalar type.
85 Value *EmitComplexToScalarConversion(CodeGenFunction::ComplexPairTy Src,
86 QualType SrcTy, QualType DstTy);
Chris Lattner3707b252007-08-26 06:48:56 +000087
Chris Lattner7f02f722007-08-24 05:35:26 +000088 //===--------------------------------------------------------------------===//
89 // Visitor Methods
90 //===--------------------------------------------------------------------===//
91
92 Value *VisitStmt(Stmt *S) {
Ted Kremenek7a9d49f2007-12-11 21:27:55 +000093 S->dump(CGF.getContext().getSourceManager());
Chris Lattner7f02f722007-08-24 05:35:26 +000094 assert(0 && "Stmt can't have complex result type!");
95 return 0;
96 }
97 Value *VisitExpr(Expr *S);
98 Value *VisitParenExpr(ParenExpr *PE) { return Visit(PE->getSubExpr()); }
99
100 // Leaves.
101 Value *VisitIntegerLiteral(const IntegerLiteral *E) {
102 return llvm::ConstantInt::get(E->getValue());
103 }
104 Value *VisitFloatingLiteral(const FloatingLiteral *E) {
Chris Lattner59138ba2008-04-20 00:45:53 +0000105 return llvm::ConstantFP::get(E->getValue());
Chris Lattner7f02f722007-08-24 05:35:26 +0000106 }
107 Value *VisitCharacterLiteral(const CharacterLiteral *E) {
108 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
109 }
Nate Begemane7579b52007-11-15 05:40:03 +0000110 Value *VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
111 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
112 }
Argyrios Kyrtzidis7267f782008-08-23 19:35:47 +0000113 Value *VisitCXXZeroInitValueExpr(const CXXZeroInitValueExpr *E) {
114 return llvm::Constant::getNullValue(ConvertType(E->getType()));
115 }
Chris Lattner7f02f722007-08-24 05:35:26 +0000116 Value *VisitTypesCompatibleExpr(const TypesCompatibleExpr *E) {
117 return llvm::ConstantInt::get(ConvertType(E->getType()),
Steve Naroffec0550f2007-10-15 20:41:53 +0000118 CGF.getContext().typesAreCompatible(
119 E->getArgType1(), E->getArgType2()));
Chris Lattner7f02f722007-08-24 05:35:26 +0000120 }
Sebastian Redl05189992008-11-11 17:56:53 +0000121 Value *VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E);
Daniel Dunbar0ffb1252008-08-04 16:51:22 +0000122 Value *VisitAddrLabelExpr(const AddrLabelExpr *E) {
Daniel Dunbar54d19092008-08-16 01:41:47 +0000123 llvm::Value *V =
124 llvm::ConstantInt::get(llvm::Type::Int32Ty,
125 CGF.GetIDForAddrOfLabel(E->getLabel()));
126
127 return Builder.CreateIntToPtr(V, ConvertType(E->getType()));
Daniel Dunbar0ffb1252008-08-04 16:51:22 +0000128 }
Chris Lattner7f02f722007-08-24 05:35:26 +0000129
130 // l-values.
131 Value *VisitDeclRefExpr(DeclRefExpr *E) {
132 if (const EnumConstantDecl *EC = dyn_cast<EnumConstantDecl>(E->getDecl()))
133 return llvm::ConstantInt::get(EC->getInitVal());
134 return EmitLoadOfLValue(E);
135 }
Daniel Dunbar9c3fc702008-08-27 06:57:25 +0000136 Value *VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
137 return CGF.EmitObjCSelectorExpr(E);
138 }
139 Value *VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
140 return CGF.EmitObjCProtocolExpr(E);
141 }
142 Value *VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
143 return EmitLoadOfLValue(E);
144 }
Daniel Dunbar0a04d772008-08-23 10:51:21 +0000145 Value *VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
Daniel Dunbar85c59ed2008-08-29 08:11:39 +0000146 return EmitLoadOfLValue(E);
Daniel Dunbar9c3fc702008-08-27 06:57:25 +0000147 }
Fariborz Jahanian43f44702008-11-22 22:30:21 +0000148 Value *VisitObjCKVCRefExpr(ObjCKVCRefExpr *E) {
149 return EmitLoadOfLValue(E);
150 }
Daniel Dunbar9c3fc702008-08-27 06:57:25 +0000151 Value *VisitObjCMessageExpr(ObjCMessageExpr *E) {
152 return CGF.EmitObjCMessageExpr(E).getScalarVal();
Daniel Dunbar0a04d772008-08-23 10:51:21 +0000153 }
154
Chris Lattner7f02f722007-08-24 05:35:26 +0000155 Value *VisitArraySubscriptExpr(ArraySubscriptExpr *E);
Eli Friedmand38617c2008-05-14 19:38:39 +0000156 Value *VisitShuffleVectorExpr(ShuffleVectorExpr *E);
Chris Lattner7f02f722007-08-24 05:35:26 +0000157 Value *VisitMemberExpr(Expr *E) { return EmitLoadOfLValue(E); }
Nate Begeman213541a2008-04-18 23:10:10 +0000158 Value *VisitExtVectorElementExpr(Expr *E) { return EmitLoadOfLValue(E); }
Chris Lattnerbe20bb52008-10-26 23:53:12 +0000159 Value *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
160 return EmitLoadOfLValue(E);
161 }
Chris Lattner7f02f722007-08-24 05:35:26 +0000162 Value *VisitStringLiteral(Expr *E) { return EmitLValue(E).getAddress(); }
Chris Lattnerd9f69102008-08-10 01:53:14 +0000163 Value *VisitPredefinedExpr(Expr *E) { return EmitLValue(E).getAddress(); }
Devang Patel35634f52007-10-24 17:18:43 +0000164
165 Value *VisitInitListExpr(InitListExpr *E) {
Anders Carlsson7019a9e2007-12-05 07:36:10 +0000166 unsigned NumInitElements = E->getNumInits();
167
Anders Carlsson7019a9e2007-12-05 07:36:10 +0000168 const llvm::VectorType *VType =
Anders Carlssonf6884ac2008-01-29 01:15:48 +0000169 dyn_cast<llvm::VectorType>(ConvertType(E->getType()));
170
171 // We have a scalar in braces. Just use the first element.
172 if (!VType)
173 return Visit(E->getInit(0));
Anders Carlsson7019a9e2007-12-05 07:36:10 +0000174
Chris Lattnerbe20bb52008-10-26 23:53:12 +0000175 if (E->hadDesignators()) {
176 CGF.ErrorUnsupported(E, "initializer list with designators");
177 return llvm::UndefValue::get(CGF.ConvertType(E->getType()));
178 }
179
Anders Carlsson7019a9e2007-12-05 07:36:10 +0000180 unsigned NumVectorElements = VType->getNumElements();
181 const llvm::Type *ElementType = VType->getElementType();
Anders Carlsson7019a9e2007-12-05 07:36:10 +0000182
183 // Emit individual vector element stores.
184 llvm::Value *V = llvm::UndefValue::get(VType);
185
Anders Carlsson222d2c82007-12-18 02:45:33 +0000186 // Emit initializers
187 unsigned i;
188 for (i = 0; i < NumInitElements; ++i) {
Devang Patela83cc332007-10-24 18:05:48 +0000189 Value *NewV = Visit(E->getInit(i));
190 Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, i);
191 V = Builder.CreateInsertElement(V, NewV, Idx);
Devang Patel35634f52007-10-24 17:18:43 +0000192 }
Anders Carlsson7019a9e2007-12-05 07:36:10 +0000193
194 // Emit remaining default initializers
195 for (/* Do not initialize i*/; i < NumVectorElements; ++i) {
196 Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, i);
197 llvm::Value *NewV = llvm::Constant::getNullValue(ElementType);
198 V = Builder.CreateInsertElement(V, NewV, Idx);
199 }
200
Devang Patela83cc332007-10-24 18:05:48 +0000201 return V;
Devang Patel35634f52007-10-24 17:18:43 +0000202 }
Chris Lattner04421082008-04-08 04:40:51 +0000203
Chris Lattner7f02f722007-08-24 05:35:26 +0000204 Value *VisitImplicitCastExpr(const ImplicitCastExpr *E);
205 Value *VisitCastExpr(const CastExpr *E) {
206 return EmitCastExpr(E->getSubExpr(), E->getType());
207 }
208 Value *EmitCastExpr(const Expr *E, QualType T);
209
210 Value *VisitCallExpr(const CallExpr *E) {
Chris Lattner9b655512007-08-31 22:49:20 +0000211 return CGF.EmitCallExpr(E).getScalarVal();
Chris Lattner7f02f722007-08-24 05:35:26 +0000212 }
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000213
Chris Lattner33793202007-08-31 22:09:40 +0000214 Value *VisitStmtExpr(const StmtExpr *E);
215
Chris Lattner7f02f722007-08-24 05:35:26 +0000216 // Unary Operators.
217 Value *VisitPrePostIncDec(const UnaryOperator *E, bool isInc, bool isPre);
218 Value *VisitUnaryPostDec(const UnaryOperator *E) {
219 return VisitPrePostIncDec(E, false, false);
220 }
221 Value *VisitUnaryPostInc(const UnaryOperator *E) {
222 return VisitPrePostIncDec(E, true, false);
223 }
224 Value *VisitUnaryPreDec(const UnaryOperator *E) {
225 return VisitPrePostIncDec(E, false, true);
226 }
227 Value *VisitUnaryPreInc(const UnaryOperator *E) {
228 return VisitPrePostIncDec(E, true, true);
229 }
230 Value *VisitUnaryAddrOf(const UnaryOperator *E) {
231 return EmitLValue(E->getSubExpr()).getAddress();
232 }
233 Value *VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); }
234 Value *VisitUnaryPlus(const UnaryOperator *E) {
235 return Visit(E->getSubExpr());
236 }
237 Value *VisitUnaryMinus (const UnaryOperator *E);
238 Value *VisitUnaryNot (const UnaryOperator *E);
239 Value *VisitUnaryLNot (const UnaryOperator *E);
Chris Lattner46f93d02007-08-24 21:20:17 +0000240 Value *VisitUnaryReal (const UnaryOperator *E);
241 Value *VisitUnaryImag (const UnaryOperator *E);
Chris Lattner7f02f722007-08-24 05:35:26 +0000242 Value *VisitUnaryExtension(const UnaryOperator *E) {
243 return Visit(E->getSubExpr());
244 }
Anders Carlsson5a1deb82008-01-29 15:56:48 +0000245 Value *VisitUnaryOffsetOf(const UnaryOperator *E);
Chris Lattner04421082008-04-08 04:40:51 +0000246 Value *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
247 return Visit(DAE->getExpr());
248 }
Anders Carlsson5a1deb82008-01-29 15:56:48 +0000249
Chris Lattner7f02f722007-08-24 05:35:26 +0000250 // Binary Operators.
Chris Lattner7f02f722007-08-24 05:35:26 +0000251 Value *EmitMul(const BinOpInfo &Ops) {
252 return Builder.CreateMul(Ops.LHS, Ops.RHS, "mul");
253 }
254 Value *EmitDiv(const BinOpInfo &Ops);
255 Value *EmitRem(const BinOpInfo &Ops);
256 Value *EmitAdd(const BinOpInfo &Ops);
257 Value *EmitSub(const BinOpInfo &Ops);
258 Value *EmitShl(const BinOpInfo &Ops);
259 Value *EmitShr(const BinOpInfo &Ops);
260 Value *EmitAnd(const BinOpInfo &Ops) {
261 return Builder.CreateAnd(Ops.LHS, Ops.RHS, "and");
262 }
263 Value *EmitXor(const BinOpInfo &Ops) {
264 return Builder.CreateXor(Ops.LHS, Ops.RHS, "xor");
265 }
266 Value *EmitOr (const BinOpInfo &Ops) {
267 return Builder.CreateOr(Ops.LHS, Ops.RHS, "or");
268 }
269
Chris Lattner1f1ded92007-08-24 21:00:35 +0000270 BinOpInfo EmitBinOps(const BinaryOperator *E);
Chris Lattner3ccf7742007-08-26 21:41:21 +0000271 Value *EmitCompoundAssign(const CompoundAssignOperator *E,
Chris Lattner1f1ded92007-08-24 21:00:35 +0000272 Value *(ScalarExprEmitter::*F)(const BinOpInfo &));
273
274 // Binary operators and binary compound assignment operators.
275#define HANDLEBINOP(OP) \
Chris Lattner3ccf7742007-08-26 21:41:21 +0000276 Value *VisitBin ## OP(const BinaryOperator *E) { \
277 return Emit ## OP(EmitBinOps(E)); \
278 } \
279 Value *VisitBin ## OP ## Assign(const CompoundAssignOperator *E) { \
280 return EmitCompoundAssign(E, &ScalarExprEmitter::Emit ## OP); \
Chris Lattner1f1ded92007-08-24 21:00:35 +0000281 }
282 HANDLEBINOP(Mul);
283 HANDLEBINOP(Div);
284 HANDLEBINOP(Rem);
285 HANDLEBINOP(Add);
Daniel Dunbar8c6f57c2008-08-06 02:00:38 +0000286 HANDLEBINOP(Sub);
Chris Lattner1f1ded92007-08-24 21:00:35 +0000287 HANDLEBINOP(Shl);
288 HANDLEBINOP(Shr);
289 HANDLEBINOP(And);
290 HANDLEBINOP(Xor);
291 HANDLEBINOP(Or);
292#undef HANDLEBINOP
Daniel Dunbar8c6f57c2008-08-06 02:00:38 +0000293
Chris Lattner7f02f722007-08-24 05:35:26 +0000294 // Comparisons.
295 Value *EmitCompare(const BinaryOperator *E, unsigned UICmpOpc,
296 unsigned SICmpOpc, unsigned FCmpOpc);
297#define VISITCOMP(CODE, UI, SI, FP) \
298 Value *VisitBin##CODE(const BinaryOperator *E) { \
299 return EmitCompare(E, llvm::ICmpInst::UI, llvm::ICmpInst::SI, \
300 llvm::FCmpInst::FP); }
301 VISITCOMP(LT, ICMP_ULT, ICMP_SLT, FCMP_OLT);
302 VISITCOMP(GT, ICMP_UGT, ICMP_SGT, FCMP_OGT);
303 VISITCOMP(LE, ICMP_ULE, ICMP_SLE, FCMP_OLE);
304 VISITCOMP(GE, ICMP_UGE, ICMP_SGE, FCMP_OGE);
305 VISITCOMP(EQ, ICMP_EQ , ICMP_EQ , FCMP_OEQ);
306 VISITCOMP(NE, ICMP_NE , ICMP_NE , FCMP_UNE);
307#undef VISITCOMP
308
309 Value *VisitBinAssign (const BinaryOperator *E);
310
311 Value *VisitBinLAnd (const BinaryOperator *E);
312 Value *VisitBinLOr (const BinaryOperator *E);
Chris Lattner7f02f722007-08-24 05:35:26 +0000313 Value *VisitBinComma (const BinaryOperator *E);
314
315 // Other Operators.
316 Value *VisitConditionalOperator(const ConditionalOperator *CO);
317 Value *VisitChooseExpr(ChooseExpr *CE);
Nate Begemane2ce1d92008-01-17 17:46:27 +0000318 Value *VisitOverloadExpr(OverloadExpr *OE);
Anders Carlsson7c50aca2007-10-15 20:28:48 +0000319 Value *VisitVAArgExpr(VAArgExpr *VE);
Chris Lattner7f02f722007-08-24 05:35:26 +0000320 Value *VisitObjCStringLiteral(const ObjCStringLiteral *E) {
321 return CGF.EmitObjCStringLiteral(E);
322 }
Anders Carlsson85f9bce2007-10-29 05:01:08 +0000323 Value *VisitObjCEncodeExpr(const ObjCEncodeExpr *E);
Chris Lattner7f02f722007-08-24 05:35:26 +0000324};
325} // end anonymous namespace.
326
327//===----------------------------------------------------------------------===//
328// Utilities
329//===----------------------------------------------------------------------===//
330
Chris Lattner9abc84e2007-08-26 16:42:57 +0000331/// EmitConversionToBool - Convert the specified expression value to a
Chris Lattner3420d0d2007-08-26 17:25:57 +0000332/// boolean (i1) truth value. This is equivalent to "Val != 0".
Chris Lattner9abc84e2007-08-26 16:42:57 +0000333Value *ScalarExprEmitter::EmitConversionToBool(Value *Src, QualType SrcType) {
334 assert(SrcType->isCanonical() && "EmitScalarConversion strips typedefs");
335
336 if (SrcType->isRealFloatingType()) {
337 // Compare against 0.0 for fp scalars.
338 llvm::Value *Zero = llvm::Constant::getNullValue(Src->getType());
Chris Lattner9abc84e2007-08-26 16:42:57 +0000339 return Builder.CreateFCmpUNE(Src, Zero, "tobool");
340 }
341
Daniel Dunbard1d66bc2008-08-25 10:38:11 +0000342 assert((SrcType->isIntegerType() || isa<llvm::PointerType>(Src->getType())) &&
Chris Lattner9abc84e2007-08-26 16:42:57 +0000343 "Unknown scalar type to convert");
344
345 // Because of the type rules of C, we often end up computing a logical value,
346 // then zero extending it to int, then wanting it as a logical value again.
347 // Optimize this common case.
348 if (llvm::ZExtInst *ZI = dyn_cast<llvm::ZExtInst>(Src)) {
349 if (ZI->getOperand(0)->getType() == llvm::Type::Int1Ty) {
350 Value *Result = ZI->getOperand(0);
Eli Friedman356916e2008-01-29 18:13:51 +0000351 // If there aren't any more uses, zap the instruction to save space.
352 // Note that there can be more uses, for example if this
353 // is the result of an assignment.
354 if (ZI->use_empty())
355 ZI->eraseFromParent();
Chris Lattner9abc84e2007-08-26 16:42:57 +0000356 return Result;
357 }
358 }
359
360 // Compare against an integer or pointer null.
361 llvm::Value *Zero = llvm::Constant::getNullValue(Src->getType());
362 return Builder.CreateICmpNE(Src, Zero, "tobool");
363}
364
Chris Lattner3707b252007-08-26 06:48:56 +0000365/// EmitScalarConversion - Emit a conversion from the specified type to the
366/// specified destination type, both of which are LLVM scalar types.
Chris Lattner4f1a7b32007-08-26 16:34:22 +0000367Value *ScalarExprEmitter::EmitScalarConversion(Value *Src, QualType SrcType,
368 QualType DstType) {
Chris Lattner96196622008-07-26 22:37:01 +0000369 SrcType = CGF.getContext().getCanonicalType(SrcType);
370 DstType = CGF.getContext().getCanonicalType(DstType);
Chris Lattner3707b252007-08-26 06:48:56 +0000371 if (SrcType == DstType) return Src;
Chris Lattnercf289082007-08-26 07:21:11 +0000372
373 if (DstType->isVoidType()) return 0;
Chris Lattner3707b252007-08-26 06:48:56 +0000374
375 // Handle conversions to bool first, they are special: comparisons against 0.
Chris Lattnered70f0a2007-08-26 16:52:28 +0000376 if (DstType->isBooleanType())
377 return EmitConversionToBool(Src, SrcType);
Chris Lattner3707b252007-08-26 06:48:56 +0000378
379 const llvm::Type *DstTy = ConvertType(DstType);
380
381 // Ignore conversions like int -> uint.
382 if (Src->getType() == DstTy)
383 return Src;
384
Daniel Dunbar270cc662008-08-25 09:51:32 +0000385 // Handle pointer conversions next: pointers can only be converted
386 // to/from other pointers and integers. Check for pointer types in
387 // terms of LLVM, as some native types (like Obj-C id) may map to a
388 // pointer type.
389 if (isa<llvm::PointerType>(DstTy)) {
Chris Lattner3707b252007-08-26 06:48:56 +0000390 // The source value may be an integer, or a pointer.
391 if (isa<llvm::PointerType>(Src->getType()))
392 return Builder.CreateBitCast(Src, DstTy, "conv");
393 assert(SrcType->isIntegerType() && "Not ptr->ptr or int->ptr conversion?");
394 return Builder.CreateIntToPtr(Src, DstTy, "conv");
395 }
396
Daniel Dunbar270cc662008-08-25 09:51:32 +0000397 if (isa<llvm::PointerType>(Src->getType())) {
Chris Lattner3707b252007-08-26 06:48:56 +0000398 // Must be an ptr to int cast.
399 assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
Anders Carlsson50b5a302007-10-31 23:18:02 +0000400 return Builder.CreatePtrToInt(Src, DstTy, "conv");
Chris Lattner3707b252007-08-26 06:48:56 +0000401 }
402
Nate Begeman213541a2008-04-18 23:10:10 +0000403 // A scalar can be splatted to an extended vector of the same element type
404 if (DstType->isExtVectorType() && !isa<VectorType>(SrcType) &&
Chris Lattner3b1ae002008-02-02 04:51:41 +0000405 cast<llvm::VectorType>(DstTy)->getElementType() == Src->getType())
Nate Begeman4119d1a2007-12-30 02:59:45 +0000406 return CGF.EmitVector(&Src, DstType->getAsVectorType()->getNumElements(),
407 true);
Nate Begeman4119d1a2007-12-30 02:59:45 +0000408
Chris Lattner3b1ae002008-02-02 04:51:41 +0000409 // Allow bitcast from vector to integer/fp of the same size.
Anders Carlsson7019a9e2007-12-05 07:36:10 +0000410 if (isa<llvm::VectorType>(Src->getType()) ||
Chris Lattner3b1ae002008-02-02 04:51:41 +0000411 isa<llvm::VectorType>(DstTy))
Anders Carlsson7019a9e2007-12-05 07:36:10 +0000412 return Builder.CreateBitCast(Src, DstTy, "conv");
Anders Carlsson7019a9e2007-12-05 07:36:10 +0000413
Chris Lattner3707b252007-08-26 06:48:56 +0000414 // Finally, we have the arithmetic types: real int/float.
415 if (isa<llvm::IntegerType>(Src->getType())) {
416 bool InputSigned = SrcType->isSignedIntegerType();
Anders Carlssonb5ce0972007-12-26 18:20:19 +0000417 if (isa<llvm::IntegerType>(DstTy))
418 return Builder.CreateIntCast(Src, DstTy, InputSigned, "conv");
419 else if (InputSigned)
420 return Builder.CreateSIToFP(Src, DstTy, "conv");
421 else
422 return Builder.CreateUIToFP(Src, DstTy, "conv");
Chris Lattner3707b252007-08-26 06:48:56 +0000423 }
424
425 assert(Src->getType()->isFloatingPoint() && "Unknown real conversion");
426 if (isa<llvm::IntegerType>(DstTy)) {
Anders Carlssonb5ce0972007-12-26 18:20:19 +0000427 if (DstType->isSignedIntegerType())
428 return Builder.CreateFPToSI(Src, DstTy, "conv");
429 else
430 return Builder.CreateFPToUI(Src, DstTy, "conv");
Chris Lattner3707b252007-08-26 06:48:56 +0000431 }
432
433 assert(DstTy->isFloatingPoint() && "Unknown real conversion");
Anders Carlssonb5ce0972007-12-26 18:20:19 +0000434 if (DstTy->getTypeID() < Src->getType()->getTypeID())
435 return Builder.CreateFPTrunc(Src, DstTy, "conv");
436 else
437 return Builder.CreateFPExt(Src, DstTy, "conv");
Chris Lattner3707b252007-08-26 06:48:56 +0000438}
439
Chris Lattner4f1a7b32007-08-26 16:34:22 +0000440/// EmitComplexToScalarConversion - Emit a conversion from the specified
441/// complex type to the specified destination type, where the destination
442/// type is an LLVM scalar type.
443Value *ScalarExprEmitter::
444EmitComplexToScalarConversion(CodeGenFunction::ComplexPairTy Src,
445 QualType SrcTy, QualType DstTy) {
Chris Lattnered70f0a2007-08-26 16:52:28 +0000446 // Get the source element type.
Chris Lattner96196622008-07-26 22:37:01 +0000447 SrcTy = SrcTy->getAsComplexType()->getElementType();
Chris Lattnered70f0a2007-08-26 16:52:28 +0000448
449 // Handle conversions to bool first, they are special: comparisons against 0.
450 if (DstTy->isBooleanType()) {
451 // Complex != 0 -> (Real != 0) | (Imag != 0)
452 Src.first = EmitScalarConversion(Src.first, SrcTy, DstTy);
453 Src.second = EmitScalarConversion(Src.second, SrcTy, DstTy);
454 return Builder.CreateOr(Src.first, Src.second, "tobool");
455 }
456
Chris Lattner4f1a7b32007-08-26 16:34:22 +0000457 // C99 6.3.1.7p2: "When a value of complex type is converted to a real type,
458 // the imaginary part of the complex value is discarded and the value of the
459 // real part is converted according to the conversion rules for the
460 // corresponding real type.
Chris Lattner4f1a7b32007-08-26 16:34:22 +0000461 return EmitScalarConversion(Src.first, SrcTy, DstTy);
462}
463
464
Chris Lattner7f02f722007-08-24 05:35:26 +0000465//===----------------------------------------------------------------------===//
466// Visitor Methods
467//===----------------------------------------------------------------------===//
468
469Value *ScalarExprEmitter::VisitExpr(Expr *E) {
Daniel Dunbar488e9932008-08-16 00:56:44 +0000470 CGF.ErrorUnsupported(E, "scalar expression");
Chris Lattner7f02f722007-08-24 05:35:26 +0000471 if (E->getType()->isVoidType())
472 return 0;
473 return llvm::UndefValue::get(CGF.ConvertType(E->getType()));
474}
475
Eli Friedmand38617c2008-05-14 19:38:39 +0000476Value *ScalarExprEmitter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
477 llvm::SmallVector<llvm::Constant*, 32> indices;
478 for (unsigned i = 2; i < E->getNumSubExprs(); i++) {
479 indices.push_back(cast<llvm::Constant>(CGF.EmitScalarExpr(E->getExpr(i))));
480 }
481 Value* V1 = CGF.EmitScalarExpr(E->getExpr(0));
482 Value* V2 = CGF.EmitScalarExpr(E->getExpr(1));
483 Value* SV = llvm::ConstantVector::get(indices.begin(), indices.size());
484 return Builder.CreateShuffleVector(V1, V2, SV, "shuffle");
485}
486
Chris Lattner7f02f722007-08-24 05:35:26 +0000487Value *ScalarExprEmitter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
488 // Emit subscript expressions in rvalue context's. For most cases, this just
489 // loads the lvalue formed by the subscript expr. However, we have to be
490 // careful, because the base of a vector subscript is occasionally an rvalue,
491 // so we can't get it as an lvalue.
492 if (!E->getBase()->getType()->isVectorType())
493 return EmitLoadOfLValue(E);
494
495 // Handle the vector case. The base must be a vector, the index must be an
496 // integer value.
497 Value *Base = Visit(E->getBase());
498 Value *Idx = Visit(E->getIdx());
499
500 // FIXME: Convert Idx to i32 type.
501 return Builder.CreateExtractElement(Base, Idx, "vecext");
502}
503
504/// VisitImplicitCastExpr - Implicit casts are the same as normal casts, but
505/// also handle things like function to pointer-to-function decay, and array to
506/// pointer decay.
507Value *ScalarExprEmitter::VisitImplicitCastExpr(const ImplicitCastExpr *E) {
508 const Expr *Op = E->getSubExpr();
509
510 // If this is due to array->pointer conversion, emit the array expression as
511 // an l-value.
512 if (Op->getType()->isArrayType()) {
513 // FIXME: For now we assume that all source arrays map to LLVM arrays. This
514 // will not true when we add support for VLAs.
Chris Lattner4f1a7b32007-08-26 16:34:22 +0000515 Value *V = EmitLValue(Op).getAddress(); // Bitfields can't be arrays.
Chris Lattner7f02f722007-08-24 05:35:26 +0000516
Daniel Dunbar662174c82008-08-29 17:28:43 +0000517 if (!(isa<llvm::PointerType>(V->getType()) &&
518 isa<llvm::ArrayType>(cast<llvm::PointerType>(V->getType())
519 ->getElementType()))) {
Daniel Dunbar90df4b62008-09-04 03:43:08 +0000520 CGF.ErrorUnsupported(E, "variable-length array cast", true);
Daniel Dunbar662174c82008-08-29 17:28:43 +0000521 if (E->getType()->isVoidType())
522 return 0;
523 return llvm::UndefValue::get(CGF.ConvertType(E->getType()));
524 }
Chris Lattner36b6a0a2008-03-19 05:19:41 +0000525 V = Builder.CreateStructGEP(V, 0, "arraydecay");
Chris Lattnera9e63722007-12-12 04:13:20 +0000526
527 // The resultant pointer type can be implicitly casted to other pointer
Chris Lattnerf31627f2008-07-23 06:31:27 +0000528 // types as well (e.g. void*) and can be implicitly converted to integer.
529 const llvm::Type *DestTy = ConvertType(E->getType());
530 if (V->getType() != DestTy) {
531 if (isa<llvm::PointerType>(DestTy))
532 V = Builder.CreateBitCast(V, DestTy, "ptrconv");
533 else {
534 assert(isa<llvm::IntegerType>(DestTy) && "Unknown array decay");
535 V = Builder.CreatePtrToInt(V, DestTy, "ptrconv");
536 }
537 }
Chris Lattnera9e63722007-12-12 04:13:20 +0000538 return V;
539
Anders Carlsson793680e2007-10-12 23:56:29 +0000540 } else if (E->getType()->isReferenceType()) {
Anders Carlsson793680e2007-10-12 23:56:29 +0000541 return EmitLValue(Op).getAddress();
Chris Lattner7f02f722007-08-24 05:35:26 +0000542 }
543
544 return EmitCastExpr(Op, E->getType());
545}
546
547
548// VisitCastExpr - Emit code for an explicit or implicit cast. Implicit casts
549// have to handle a more broad range of conversions than explicit casts, as they
550// handle things like function to ptr-to-function decay etc.
551Value *ScalarExprEmitter::EmitCastExpr(const Expr *E, QualType DestTy) {
Chris Lattner58a2e942007-08-26 07:26:12 +0000552 // Handle cases where the source is an non-complex type.
Chris Lattner19a1d7c2008-02-16 23:55:16 +0000553
554 if (!CGF.hasAggregateLLVMType(E->getType())) {
Chris Lattner3707b252007-08-26 06:48:56 +0000555 Value *Src = Visit(const_cast<Expr*>(E));
556
Chris Lattner3707b252007-08-26 06:48:56 +0000557 // Use EmitScalarConversion to perform the conversion.
558 return EmitScalarConversion(Src, E->getType(), DestTy);
559 }
Chris Lattner19a1d7c2008-02-16 23:55:16 +0000560
Chris Lattner9b2dc282008-04-04 16:54:41 +0000561 if (E->getType()->isAnyComplexType()) {
Chris Lattner19a1d7c2008-02-16 23:55:16 +0000562 // Handle cases where the source is a complex type.
563 return EmitComplexToScalarConversion(CGF.EmitComplexExpr(E), E->getType(),
564 DestTy);
565 }
Chris Lattner10b00cf2007-08-26 07:16:41 +0000566
Chris Lattner19a1d7c2008-02-16 23:55:16 +0000567 // Okay, this is a cast from an aggregate. It must be a cast to void. Just
568 // evaluate the result and return.
569 CGF.EmitAggExpr(E, 0, false);
570 return 0;
Chris Lattner7f02f722007-08-24 05:35:26 +0000571}
572
Chris Lattner33793202007-08-31 22:09:40 +0000573Value *ScalarExprEmitter::VisitStmtExpr(const StmtExpr *E) {
Chris Lattner91d723d2008-07-26 20:23:23 +0000574 return CGF.EmitCompoundStmt(*E->getSubStmt(),
575 !E->getType()->isVoidType()).getScalarVal();
Chris Lattner33793202007-08-31 22:09:40 +0000576}
577
578
Chris Lattner7f02f722007-08-24 05:35:26 +0000579//===----------------------------------------------------------------------===//
580// Unary Operators
581//===----------------------------------------------------------------------===//
582
583Value *ScalarExprEmitter::VisitPrePostIncDec(const UnaryOperator *E,
Chris Lattnerdfce2a52007-08-24 16:24:49 +0000584 bool isInc, bool isPre) {
Chris Lattner7f02f722007-08-24 05:35:26 +0000585 LValue LV = EmitLValue(E->getSubExpr());
586 // FIXME: Handle volatile!
Chris Lattnere936cc82007-08-26 05:10:16 +0000587 Value *InVal = CGF.EmitLoadOfLValue(LV, // false
Chris Lattner9b655512007-08-31 22:49:20 +0000588 E->getSubExpr()->getType()).getScalarVal();
Chris Lattner7f02f722007-08-24 05:35:26 +0000589
590 int AmountVal = isInc ? 1 : -1;
591
592 Value *NextVal;
Chris Lattnere936cc82007-08-26 05:10:16 +0000593 if (isa<llvm::PointerType>(InVal->getType())) {
594 // FIXME: This isn't right for VLAs.
595 NextVal = llvm::ConstantInt::get(llvm::Type::Int32Ty, AmountVal);
Chris Lattner36b6a0a2008-03-19 05:19:41 +0000596 NextVal = Builder.CreateGEP(InVal, NextVal, "ptrincdec");
Chris Lattnere936cc82007-08-26 05:10:16 +0000597 } else {
598 // Add the inc/dec to the real part.
599 if (isa<llvm::IntegerType>(InVal->getType()))
600 NextVal = llvm::ConstantInt::get(InVal->getType(), AmountVal);
Chris Lattnerca2617c2007-09-13 06:19:18 +0000601 else if (InVal->getType() == llvm::Type::FloatTy)
Devang Patele9b8c0a2007-10-30 20:59:40 +0000602 NextVal =
Chris Lattner59138ba2008-04-20 00:45:53 +0000603 llvm::ConstantFP::get(llvm::APFloat(static_cast<float>(AmountVal)));
Chris Lattner25ddea72008-04-20 00:50:39 +0000604 else if (InVal->getType() == llvm::Type::DoubleTy)
Devang Patele9b8c0a2007-10-30 20:59:40 +0000605 NextVal =
Chris Lattner59138ba2008-04-20 00:45:53 +0000606 llvm::ConstantFP::get(llvm::APFloat(static_cast<double>(AmountVal)));
Chris Lattner25ddea72008-04-20 00:50:39 +0000607 else {
608 llvm::APFloat F(static_cast<float>(AmountVal));
Dale Johannesenee5a7002008-10-09 23:02:32 +0000609 bool ignored;
610 F.convert(CGF.Target.getLongDoubleFormat(), llvm::APFloat::rmTowardZero,
611 &ignored);
Chris Lattner25ddea72008-04-20 00:50:39 +0000612 NextVal = llvm::ConstantFP::get(F);
Chris Lattnerca2617c2007-09-13 06:19:18 +0000613 }
Chris Lattnere936cc82007-08-26 05:10:16 +0000614 NextVal = Builder.CreateAdd(InVal, NextVal, isInc ? "inc" : "dec");
615 }
Chris Lattner7f02f722007-08-24 05:35:26 +0000616
617 // Store the updated result through the lvalue.
618 CGF.EmitStoreThroughLValue(RValue::get(NextVal), LV,
619 E->getSubExpr()->getType());
620
621 // If this is a postinc, return the value read from memory, otherwise use the
622 // updated value.
623 return isPre ? NextVal : InVal;
624}
625
626
627Value *ScalarExprEmitter::VisitUnaryMinus(const UnaryOperator *E) {
628 Value *Op = Visit(E->getSubExpr());
629 return Builder.CreateNeg(Op, "neg");
630}
631
632Value *ScalarExprEmitter::VisitUnaryNot(const UnaryOperator *E) {
633 Value *Op = Visit(E->getSubExpr());
634 return Builder.CreateNot(Op, "neg");
635}
636
637Value *ScalarExprEmitter::VisitUnaryLNot(const UnaryOperator *E) {
638 // Compare operand to zero.
639 Value *BoolVal = CGF.EvaluateExprAsBool(E->getSubExpr());
640
641 // Invert value.
642 // TODO: Could dynamically modify easy computations here. For example, if
643 // the operand is an icmp ne, turn into icmp eq.
644 BoolVal = Builder.CreateNot(BoolVal, "lnot");
645
646 // ZExt result to int.
647 return Builder.CreateZExt(BoolVal, CGF.LLVMIntTy, "lnot.ext");
648}
649
Sebastian Redl05189992008-11-11 17:56:53 +0000650/// VisitSizeOfAlignOfExpr - Return the size or alignment of the type of
651/// argument of the sizeof expression as an integer.
652Value *
653ScalarExprEmitter::VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E) {
654 QualType RetType = E->getType();
Chris Lattnera269ebf2008-02-21 05:45:29 +0000655 assert(RetType->isIntegerType() && "Result type must be an integer!");
656 uint32_t ResultWidth =
Chris Lattner98be4942008-03-05 18:54:05 +0000657 static_cast<uint32_t>(CGF.getContext().getTypeSize(RetType));
Chris Lattnera269ebf2008-02-21 05:45:29 +0000658
Sebastian Redl05189992008-11-11 17:56:53 +0000659 QualType TypeToSize = E->getTypeOfArgument();
Daniel Dunbar91408452008-07-22 01:35:47 +0000660 // sizeof(void) and __alignof__(void) = 1 as a gcc extension. Also
661 // for function types.
Daniel Dunbar8ee6a632008-07-22 19:44:18 +0000662 // FIXME: what is alignof a function type in gcc?
Daniel Dunbar91408452008-07-22 01:35:47 +0000663 if (TypeToSize->isVoidType() || TypeToSize->isFunctionType())
Chris Lattnera269ebf2008-02-21 05:45:29 +0000664 return llvm::ConstantInt::get(llvm::APInt(ResultWidth, 1));
665
Chris Lattner7f02f722007-08-24 05:35:26 +0000666 /// FIXME: This doesn't handle VLAs yet!
Chris Lattner98be4942008-03-05 18:54:05 +0000667 std::pair<uint64_t, unsigned> Info = CGF.getContext().getTypeInfo(TypeToSize);
Chris Lattner7f02f722007-08-24 05:35:26 +0000668
Sebastian Redl05189992008-11-11 17:56:53 +0000669 uint64_t Val = E->isSizeOf() ? Info.first : Info.second;
Chris Lattner7f02f722007-08-24 05:35:26 +0000670 Val /= 8; // Return size in bytes, not bits.
671
Chris Lattner7f02f722007-08-24 05:35:26 +0000672 return llvm::ConstantInt::get(llvm::APInt(ResultWidth, Val));
673}
674
Chris Lattner46f93d02007-08-24 21:20:17 +0000675Value *ScalarExprEmitter::VisitUnaryReal(const UnaryOperator *E) {
676 Expr *Op = E->getSubExpr();
Chris Lattner9b2dc282008-04-04 16:54:41 +0000677 if (Op->getType()->isAnyComplexType())
Chris Lattner46f93d02007-08-24 21:20:17 +0000678 return CGF.EmitComplexExpr(Op).first;
679 return Visit(Op);
680}
681Value *ScalarExprEmitter::VisitUnaryImag(const UnaryOperator *E) {
682 Expr *Op = E->getSubExpr();
Chris Lattner9b2dc282008-04-04 16:54:41 +0000683 if (Op->getType()->isAnyComplexType())
Chris Lattner46f93d02007-08-24 21:20:17 +0000684 return CGF.EmitComplexExpr(Op).second;
Chris Lattner36f84062007-08-26 05:29:21 +0000685
686 // __imag on a scalar returns zero. Emit it the subexpr to ensure side
687 // effects are evaluated.
688 CGF.EmitScalarExpr(Op);
689 return llvm::Constant::getNullValue(ConvertType(E->getType()));
Chris Lattner46f93d02007-08-24 21:20:17 +0000690}
691
Anders Carlsson5a1deb82008-01-29 15:56:48 +0000692Value *ScalarExprEmitter::VisitUnaryOffsetOf(const UnaryOperator *E)
693{
694 int64_t Val = E->evaluateOffsetOf(CGF.getContext());
695
696 assert(E->getType()->isIntegerType() && "Result type must be an integer!");
697
Chris Lattner98be4942008-03-05 18:54:05 +0000698 uint32_t ResultWidth =
699 static_cast<uint32_t>(CGF.getContext().getTypeSize(E->getType()));
Anders Carlsson5a1deb82008-01-29 15:56:48 +0000700 return llvm::ConstantInt::get(llvm::APInt(ResultWidth, Val));
701}
Chris Lattner46f93d02007-08-24 21:20:17 +0000702
Chris Lattner7f02f722007-08-24 05:35:26 +0000703//===----------------------------------------------------------------------===//
704// Binary Operators
705//===----------------------------------------------------------------------===//
706
707BinOpInfo ScalarExprEmitter::EmitBinOps(const BinaryOperator *E) {
708 BinOpInfo Result;
709 Result.LHS = Visit(E->getLHS());
710 Result.RHS = Visit(E->getRHS());
Chris Lattner1f1ded92007-08-24 21:00:35 +0000711 Result.Ty = E->getType();
Chris Lattner7f02f722007-08-24 05:35:26 +0000712 Result.E = E;
713 return Result;
714}
715
Chris Lattner3ccf7742007-08-26 21:41:21 +0000716Value *ScalarExprEmitter::EmitCompoundAssign(const CompoundAssignOperator *E,
Chris Lattner1f1ded92007-08-24 21:00:35 +0000717 Value *(ScalarExprEmitter::*Func)(const BinOpInfo &)) {
718 QualType LHSTy = E->getLHS()->getType(), RHSTy = E->getRHS()->getType();
719
720 BinOpInfo OpInfo;
721
722 // Load the LHS and RHS operands.
723 LValue LHSLV = EmitLValue(E->getLHS());
724 OpInfo.LHS = EmitLoadOfLValue(LHSLV, LHSTy);
Chris Lattner04dc7642007-08-26 22:37:40 +0000725
726 // Determine the computation type. If the RHS is complex, then this is one of
727 // the add/sub/mul/div operators. All of these operators can be computed in
728 // with just their real component even though the computation domain really is
729 // complex.
Chris Lattner3ccf7742007-08-26 21:41:21 +0000730 QualType ComputeType = E->getComputationType();
Chris Lattner1f1ded92007-08-24 21:00:35 +0000731
Chris Lattner04dc7642007-08-26 22:37:40 +0000732 // If the computation type is complex, then the RHS is complex. Emit the RHS.
733 if (const ComplexType *CT = ComputeType->getAsComplexType()) {
734 ComputeType = CT->getElementType();
735
736 // Emit the RHS, only keeping the real component.
737 OpInfo.RHS = CGF.EmitComplexExpr(E->getRHS()).first;
738 RHSTy = RHSTy->getAsComplexType()->getElementType();
739 } else {
740 // Otherwise the RHS is a simple scalar value.
741 OpInfo.RHS = Visit(E->getRHS());
742 }
743
Daniel Dunbar8c6f57c2008-08-06 02:00:38 +0000744 QualType LComputeTy, RComputeTy, ResultTy;
745
746 // Compound assignment does not contain enough information about all
747 // the types involved for pointer arithmetic cases. Figure it out
748 // here for now.
749 if (E->getLHS()->getType()->isPointerType()) {
750 // Pointer arithmetic cases: ptr +=,-= int and ptr -= ptr,
751 assert((E->getOpcode() == BinaryOperator::AddAssign ||
752 E->getOpcode() == BinaryOperator::SubAssign) &&
753 "Invalid compound assignment operator on pointer type.");
754 LComputeTy = E->getLHS()->getType();
755
756 if (E->getRHS()->getType()->isPointerType()) {
757 // Degenerate case of (ptr -= ptr) allowed by GCC implicit cast
758 // extension, the conversion from the pointer difference back to
759 // the LHS type is handled at the end.
760 assert(E->getOpcode() == BinaryOperator::SubAssign &&
761 "Invalid compound assignment operator on pointer type.");
762 RComputeTy = E->getLHS()->getType();
763 ResultTy = CGF.getContext().getPointerDiffType();
764 } else {
765 RComputeTy = E->getRHS()->getType();
766 ResultTy = LComputeTy;
767 }
768 } else if (E->getRHS()->getType()->isPointerType()) {
769 // Degenerate case of (int += ptr) allowed by GCC implicit cast
770 // extension.
771 assert(E->getOpcode() == BinaryOperator::AddAssign &&
772 "Invalid compound assignment operator on pointer type.");
773 LComputeTy = E->getLHS()->getType();
774 RComputeTy = E->getRHS()->getType();
775 ResultTy = RComputeTy;
776 } else {
777 LComputeTy = RComputeTy = ResultTy = ComputeType;
Chris Lattner1f1ded92007-08-24 21:00:35 +0000778 }
Daniel Dunbar8c6f57c2008-08-06 02:00:38 +0000779
780 // Convert the LHS/RHS values to the computation type.
781 OpInfo.LHS = EmitScalarConversion(OpInfo.LHS, LHSTy, LComputeTy);
782 OpInfo.RHS = EmitScalarConversion(OpInfo.RHS, RHSTy, RComputeTy);
783 OpInfo.Ty = ResultTy;
Chris Lattner1f1ded92007-08-24 21:00:35 +0000784 OpInfo.E = E;
785
786 // Expand the binary operator.
787 Value *Result = (this->*Func)(OpInfo);
788
Daniel Dunbar8c6f57c2008-08-06 02:00:38 +0000789 // Convert the result back to the LHS type.
790 Result = EmitScalarConversion(Result, ResultTy, LHSTy);
Chris Lattner1f1ded92007-08-24 21:00:35 +0000791
Daniel Dunbared3849b2008-11-19 09:36:46 +0000792 // Store the result value into the LHS lvalue. Bit-fields are
Daniel Dunbar371d16f2008-11-19 11:54:05 +0000793 // handled specially because the result is altered by the store,
794 // i.e., [C99 6.5.16p1] 'An assignment expression has the value of
795 // the left operand after the assignment...'.
Eli Friedman18491282008-05-25 14:13:57 +0000796 if (LHSLV.isBitfield())
Daniel Dunbared3849b2008-11-19 09:36:46 +0000797 CGF.EmitStoreThroughBitfieldLValue(RValue::get(Result), LHSLV, LHSTy,
798 &Result);
799 else
800 CGF.EmitStoreThroughLValue(RValue::get(Result), LHSLV, LHSTy);
801
Chris Lattner1f1ded92007-08-24 21:00:35 +0000802 return Result;
803}
804
805
Chris Lattner7f02f722007-08-24 05:35:26 +0000806Value *ScalarExprEmitter::EmitDiv(const BinOpInfo &Ops) {
Nate Begemanb3ab8dc2007-12-30 01:28:16 +0000807 if (Ops.LHS->getType()->isFPOrFPVector())
Chris Lattner7f02f722007-08-24 05:35:26 +0000808 return Builder.CreateFDiv(Ops.LHS, Ops.RHS, "div");
Chris Lattner1f1ded92007-08-24 21:00:35 +0000809 else if (Ops.Ty->isUnsignedIntegerType())
Chris Lattner7f02f722007-08-24 05:35:26 +0000810 return Builder.CreateUDiv(Ops.LHS, Ops.RHS, "div");
811 else
812 return Builder.CreateSDiv(Ops.LHS, Ops.RHS, "div");
813}
814
815Value *ScalarExprEmitter::EmitRem(const BinOpInfo &Ops) {
816 // Rem in C can't be a floating point type: C99 6.5.5p2.
Chris Lattner1f1ded92007-08-24 21:00:35 +0000817 if (Ops.Ty->isUnsignedIntegerType())
Chris Lattner7f02f722007-08-24 05:35:26 +0000818 return Builder.CreateURem(Ops.LHS, Ops.RHS, "rem");
819 else
820 return Builder.CreateSRem(Ops.LHS, Ops.RHS, "rem");
821}
822
823
824Value *ScalarExprEmitter::EmitAdd(const BinOpInfo &Ops) {
Chris Lattner1f1ded92007-08-24 21:00:35 +0000825 if (!Ops.Ty->isPointerType())
Chris Lattner7f02f722007-08-24 05:35:26 +0000826 return Builder.CreateAdd(Ops.LHS, Ops.RHS, "add");
Chris Lattner1f1ded92007-08-24 21:00:35 +0000827
828 // FIXME: What about a pointer to a VLA?
Chris Lattner8f925282008-01-03 06:36:51 +0000829 Value *Ptr, *Idx;
830 Expr *IdxExp;
831 if (isa<llvm::PointerType>(Ops.LHS->getType())) { // pointer + int
832 Ptr = Ops.LHS;
833 Idx = Ops.RHS;
834 IdxExp = Ops.E->getRHS();
835 } else { // int + pointer
836 Ptr = Ops.RHS;
837 Idx = Ops.LHS;
838 IdxExp = Ops.E->getLHS();
839 }
840
841 unsigned Width = cast<llvm::IntegerType>(Idx->getType())->getBitWidth();
842 if (Width < CGF.LLVMPointerWidth) {
843 // Zero or sign extend the pointer value based on whether the index is
844 // signed or not.
845 const llvm::Type *IdxType = llvm::IntegerType::get(CGF.LLVMPointerWidth);
Chris Lattner96196622008-07-26 22:37:01 +0000846 if (IdxExp->getType()->isSignedIntegerType())
Chris Lattner8f925282008-01-03 06:36:51 +0000847 Idx = Builder.CreateSExt(Idx, IdxType, "idx.ext");
848 else
849 Idx = Builder.CreateZExt(Idx, IdxType, "idx.ext");
850 }
851
852 return Builder.CreateGEP(Ptr, Idx, "add.ptr");
Chris Lattner7f02f722007-08-24 05:35:26 +0000853}
854
855Value *ScalarExprEmitter::EmitSub(const BinOpInfo &Ops) {
856 if (!isa<llvm::PointerType>(Ops.LHS->getType()))
857 return Builder.CreateSub(Ops.LHS, Ops.RHS, "sub");
Chris Lattner1f1ded92007-08-24 21:00:35 +0000858
Daniel Dunbar8c6f57c2008-08-06 02:00:38 +0000859 if (!isa<llvm::PointerType>(Ops.RHS->getType())) {
860 // pointer - int
861 Value *Idx = Ops.RHS;
862 unsigned Width = cast<llvm::IntegerType>(Idx->getType())->getBitWidth();
863 if (Width < CGF.LLVMPointerWidth) {
864 // Zero or sign extend the pointer value based on whether the index is
865 // signed or not.
866 const llvm::Type *IdxType = llvm::IntegerType::get(CGF.LLVMPointerWidth);
867 if (Ops.E->getRHS()->getType()->isSignedIntegerType())
868 Idx = Builder.CreateSExt(Idx, IdxType, "idx.ext");
869 else
870 Idx = Builder.CreateZExt(Idx, IdxType, "idx.ext");
871 }
872 Idx = Builder.CreateNeg(Idx, "sub.ptr.neg");
873
874 // FIXME: The pointer could point to a VLA.
875 // The GNU void* - int case is automatically handled here because
876 // our LLVM type for void* is i8*.
877 return Builder.CreateGEP(Ops.LHS, Idx, "sub.ptr");
Daniel Dunbar820b0332008-08-05 00:47:03 +0000878 } else {
Daniel Dunbar8c6f57c2008-08-06 02:00:38 +0000879 // pointer - pointer
880 Value *LHS = Ops.LHS;
881 Value *RHS = Ops.RHS;
Chris Lattner1f1ded92007-08-24 21:00:35 +0000882
Daniel Dunbar8c6f57c2008-08-06 02:00:38 +0000883 const QualType LHSType = Ops.E->getLHS()->getType();
884 const QualType LHSElementType = LHSType->getAsPointerType()->getPointeeType();
885 uint64_t ElementSize;
Daniel Dunbar820b0332008-08-05 00:47:03 +0000886
Daniel Dunbar8c6f57c2008-08-06 02:00:38 +0000887 // Handle GCC extension for pointer arithmetic on void* types.
888 if (LHSElementType->isVoidType()) {
889 ElementSize = 1;
890 } else {
891 ElementSize = CGF.getContext().getTypeSize(LHSElementType) / 8;
892 }
893
894 const llvm::Type *ResultType = ConvertType(Ops.Ty);
895 LHS = Builder.CreatePtrToInt(LHS, ResultType, "sub.ptr.lhs.cast");
896 RHS = Builder.CreatePtrToInt(RHS, ResultType, "sub.ptr.rhs.cast");
897 Value *BytesBetween = Builder.CreateSub(LHS, RHS, "sub.ptr.sub");
898
899 // HACK: LLVM doesn't have an divide instruction that 'knows' there is no
900 // remainder. As such, we handle common power-of-two cases here to generate
901 // better code. See PR2247.
902 if (llvm::isPowerOf2_64(ElementSize)) {
903 Value *ShAmt =
904 llvm::ConstantInt::get(ResultType, llvm::Log2_64(ElementSize));
905 return Builder.CreateAShr(BytesBetween, ShAmt, "sub.ptr.shr");
906 }
907
908 // Otherwise, do a full sdiv.
909 Value *BytesPerElt = llvm::ConstantInt::get(ResultType, ElementSize);
910 return Builder.CreateSDiv(BytesBetween, BytesPerElt, "sub.ptr.div");
Chris Lattner7f02f722007-08-24 05:35:26 +0000911 }
Chris Lattner7f02f722007-08-24 05:35:26 +0000912}
913
914Value *ScalarExprEmitter::EmitShl(const BinOpInfo &Ops) {
915 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
916 // RHS to the same size as the LHS.
917 Value *RHS = Ops.RHS;
918 if (Ops.LHS->getType() != RHS->getType())
919 RHS = Builder.CreateIntCast(RHS, Ops.LHS->getType(), false, "sh_prom");
920
921 return Builder.CreateShl(Ops.LHS, RHS, "shl");
922}
923
924Value *ScalarExprEmitter::EmitShr(const BinOpInfo &Ops) {
925 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
926 // RHS to the same size as the LHS.
927 Value *RHS = Ops.RHS;
928 if (Ops.LHS->getType() != RHS->getType())
929 RHS = Builder.CreateIntCast(RHS, Ops.LHS->getType(), false, "sh_prom");
930
Chris Lattner1f1ded92007-08-24 21:00:35 +0000931 if (Ops.Ty->isUnsignedIntegerType())
Chris Lattner7f02f722007-08-24 05:35:26 +0000932 return Builder.CreateLShr(Ops.LHS, RHS, "shr");
933 return Builder.CreateAShr(Ops.LHS, RHS, "shr");
934}
935
936Value *ScalarExprEmitter::EmitCompare(const BinaryOperator *E,unsigned UICmpOpc,
937 unsigned SICmpOpc, unsigned FCmpOpc) {
Chris Lattner4f1a7b32007-08-26 16:34:22 +0000938 Value *Result;
Chris Lattner7f02f722007-08-24 05:35:26 +0000939 QualType LHSTy = E->getLHS()->getType();
Nate Begeman7a66d7b2008-07-25 20:16:05 +0000940 if (!LHSTy->isAnyComplexType() && !LHSTy->isVectorType()) {
Chris Lattner7f02f722007-08-24 05:35:26 +0000941 Value *LHS = Visit(E->getLHS());
942 Value *RHS = Visit(E->getRHS());
943
944 if (LHS->getType()->isFloatingPoint()) {
Nate Begeman7a66d7b2008-07-25 20:16:05 +0000945 Result = Builder.CreateFCmp((llvm::CmpInst::Predicate)FCmpOpc,
Chris Lattner7f02f722007-08-24 05:35:26 +0000946 LHS, RHS, "cmp");
Eli Friedmanec2c1262008-05-29 15:09:15 +0000947 } else if (LHSTy->isSignedIntegerType()) {
948 Result = Builder.CreateICmp((llvm::ICmpInst::Predicate)SICmpOpc,
Chris Lattner7f02f722007-08-24 05:35:26 +0000949 LHS, RHS, "cmp");
950 } else {
Eli Friedmanec2c1262008-05-29 15:09:15 +0000951 // Unsigned integers and pointers.
952 Result = Builder.CreateICmp((llvm::ICmpInst::Predicate)UICmpOpc,
Chris Lattner7f02f722007-08-24 05:35:26 +0000953 LHS, RHS, "cmp");
954 }
Nate Begeman7a66d7b2008-07-25 20:16:05 +0000955 } else if (LHSTy->isVectorType()) {
956 Value *LHS = Visit(E->getLHS());
957 Value *RHS = Visit(E->getRHS());
958
959 if (LHS->getType()->isFPOrFPVector()) {
960 Result = Builder.CreateVFCmp((llvm::CmpInst::Predicate)FCmpOpc,
961 LHS, RHS, "cmp");
962 } else if (LHSTy->isUnsignedIntegerType()) {
963 Result = Builder.CreateVICmp((llvm::CmpInst::Predicate)UICmpOpc,
964 LHS, RHS, "cmp");
965 } else {
966 // Signed integers and pointers.
967 Result = Builder.CreateVICmp((llvm::CmpInst::Predicate)SICmpOpc,
968 LHS, RHS, "cmp");
969 }
970 return Result;
Chris Lattner7f02f722007-08-24 05:35:26 +0000971 } else {
972 // Complex Comparison: can only be an equality comparison.
973 CodeGenFunction::ComplexPairTy LHS = CGF.EmitComplexExpr(E->getLHS());
974 CodeGenFunction::ComplexPairTy RHS = CGF.EmitComplexExpr(E->getRHS());
975
Chris Lattner96196622008-07-26 22:37:01 +0000976 QualType CETy = LHSTy->getAsComplexType()->getElementType();
Chris Lattner7f02f722007-08-24 05:35:26 +0000977
Chris Lattner4f1a7b32007-08-26 16:34:22 +0000978 Value *ResultR, *ResultI;
Chris Lattner7f02f722007-08-24 05:35:26 +0000979 if (CETy->isRealFloatingType()) {
980 ResultR = Builder.CreateFCmp((llvm::FCmpInst::Predicate)FCmpOpc,
981 LHS.first, RHS.first, "cmp.r");
982 ResultI = Builder.CreateFCmp((llvm::FCmpInst::Predicate)FCmpOpc,
983 LHS.second, RHS.second, "cmp.i");
984 } else {
985 // Complex comparisons can only be equality comparisons. As such, signed
986 // and unsigned opcodes are the same.
987 ResultR = Builder.CreateICmp((llvm::ICmpInst::Predicate)UICmpOpc,
988 LHS.first, RHS.first, "cmp.r");
989 ResultI = Builder.CreateICmp((llvm::ICmpInst::Predicate)UICmpOpc,
990 LHS.second, RHS.second, "cmp.i");
991 }
992
993 if (E->getOpcode() == BinaryOperator::EQ) {
994 Result = Builder.CreateAnd(ResultR, ResultI, "and.ri");
995 } else {
996 assert(E->getOpcode() == BinaryOperator::NE &&
997 "Complex comparison other than == or != ?");
998 Result = Builder.CreateOr(ResultR, ResultI, "or.ri");
999 }
1000 }
1001
1002 // ZExt result to int.
1003 return Builder.CreateZExt(Result, CGF.LLVMIntTy, "cmp.ext");
1004}
1005
1006Value *ScalarExprEmitter::VisitBinAssign(const BinaryOperator *E) {
1007 LValue LHS = EmitLValue(E->getLHS());
1008 Value *RHS = Visit(E->getRHS());
1009
Daniel Dunbared3849b2008-11-19 09:36:46 +00001010 // Store the value into the LHS. Bit-fields are handled specially
Daniel Dunbar371d16f2008-11-19 11:54:05 +00001011 // because the result is altered by the store, i.e., [C99 6.5.16p1]
1012 // 'An assignment expression has the value of the left operand after
1013 // the assignment...'.
Chris Lattner7f02f722007-08-24 05:35:26 +00001014 // FIXME: Volatility!
Eli Friedman18491282008-05-25 14:13:57 +00001015 if (LHS.isBitfield())
Daniel Dunbared3849b2008-11-19 09:36:46 +00001016 CGF.EmitStoreThroughBitfieldLValue(RValue::get(RHS), LHS, E->getType(),
1017 &RHS);
1018 else
1019 CGF.EmitStoreThroughLValue(RValue::get(RHS), LHS, E->getType());
Daniel Dunbar85c59ed2008-08-29 08:11:39 +00001020
Chris Lattner7f02f722007-08-24 05:35:26 +00001021 // Return the RHS.
1022 return RHS;
1023}
1024
1025Value *ScalarExprEmitter::VisitBinLAnd(const BinaryOperator *E) {
Chris Lattner20eb09d2008-11-12 08:26:50 +00001026 // If we have 0 && RHS, see if we can elide RHS, if so, just return 0.
1027 // If we have 1 && X, just emit X without inserting the control flow.
1028 if (int Cond = CGF.ConstantFoldsToSimpleInteger(E->getLHS())) {
1029 if (Cond == 1) { // If we have 1 && X, just emit X.
Chris Lattner0946ccd2008-11-11 07:41:27 +00001030 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1031 // ZExt result to int.
1032 return Builder.CreateZExt(RHSCond, CGF.LLVMIntTy, "land.ext");
1033 }
Chris Lattner20eb09d2008-11-12 08:26:50 +00001034
1035 // 0 && RHS: If it is safe, just elide the RHS, and return 0.
1036 if (!CGF.ContainsLabel(E->getRHS()))
1037 return llvm::Constant::getNullValue(CGF.LLVMIntTy);
Chris Lattner0946ccd2008-11-11 07:41:27 +00001038 }
1039
Daniel Dunbar9615ecb2008-11-13 01:38:36 +00001040 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("land.end");
1041 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("land.rhs");
Chris Lattner20eb09d2008-11-12 08:26:50 +00001042
Chris Lattnerf7b5ea92008-11-12 08:38:24 +00001043 // Branch on the LHS first. If it is false, go to the failure (cont) block.
1044 CGF.EmitBranchOnBoolExpr(E->getLHS(), RHSBlock, ContBlock);
1045
1046 // Any edges into the ContBlock are now from an (indeterminate number of)
1047 // edges from this first condition. All of these values will be false. Start
1048 // setting up the PHI node in the Cont Block for this.
1049 llvm::PHINode *PN = llvm::PHINode::Create(llvm::Type::Int1Ty, "", ContBlock);
1050 PN->reserveOperandSpace(2); // Normal case, two inputs.
1051 for (llvm::pred_iterator PI = pred_begin(ContBlock), PE = pred_end(ContBlock);
1052 PI != PE; ++PI)
1053 PN->addIncoming(llvm::ConstantInt::getFalse(), *PI);
Chris Lattner7f02f722007-08-24 05:35:26 +00001054
1055 CGF.EmitBlock(RHSBlock);
1056 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1057
1058 // Reaquire the RHS block, as there may be subblocks inserted.
1059 RHSBlock = Builder.GetInsertBlock();
Chris Lattnerf7b5ea92008-11-12 08:38:24 +00001060
1061 // Emit an unconditional branch from this block to ContBlock. Insert an entry
1062 // into the phi node for the edge with the value of RHSCond.
Chris Lattner7f02f722007-08-24 05:35:26 +00001063 CGF.EmitBlock(ContBlock);
Chris Lattner7f02f722007-08-24 05:35:26 +00001064 PN->addIncoming(RHSCond, RHSBlock);
1065
1066 // ZExt result to int.
1067 return Builder.CreateZExt(PN, CGF.LLVMIntTy, "land.ext");
1068}
1069
1070Value *ScalarExprEmitter::VisitBinLOr(const BinaryOperator *E) {
Chris Lattner20eb09d2008-11-12 08:26:50 +00001071 // If we have 1 || RHS, see if we can elide RHS, if so, just return 1.
1072 // If we have 0 || X, just emit X without inserting the control flow.
1073 if (int Cond = CGF.ConstantFoldsToSimpleInteger(E->getLHS())) {
1074 if (Cond == -1) { // If we have 0 || X, just emit X.
Chris Lattner0946ccd2008-11-11 07:41:27 +00001075 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1076 // ZExt result to int.
1077 return Builder.CreateZExt(RHSCond, CGF.LLVMIntTy, "lor.ext");
1078 }
Chris Lattner20eb09d2008-11-12 08:26:50 +00001079
Eli Friedman8de8d1d2008-12-02 16:02:46 +00001080 // 1 || RHS: If it is safe, just elide the RHS, and return 1.
Chris Lattner20eb09d2008-11-12 08:26:50 +00001081 if (!CGF.ContainsLabel(E->getRHS()))
Eli Friedman8de8d1d2008-12-02 16:02:46 +00001082 return llvm::ConstantInt::get(CGF.LLVMIntTy, 1);
Chris Lattner0946ccd2008-11-11 07:41:27 +00001083 }
1084
Daniel Dunbar9615ecb2008-11-13 01:38:36 +00001085 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("lor.end");
1086 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("lor.rhs");
Chris Lattner7f02f722007-08-24 05:35:26 +00001087
Chris Lattnerf7b5ea92008-11-12 08:38:24 +00001088 // Branch on the LHS first. If it is true, go to the success (cont) block.
1089 CGF.EmitBranchOnBoolExpr(E->getLHS(), ContBlock, RHSBlock);
1090
1091 // Any edges into the ContBlock are now from an (indeterminate number of)
1092 // edges from this first condition. All of these values will be true. Start
1093 // setting up the PHI node in the Cont Block for this.
1094 llvm::PHINode *PN = llvm::PHINode::Create(llvm::Type::Int1Ty, "", ContBlock);
1095 PN->reserveOperandSpace(2); // Normal case, two inputs.
1096 for (llvm::pred_iterator PI = pred_begin(ContBlock), PE = pred_end(ContBlock);
1097 PI != PE; ++PI)
1098 PN->addIncoming(llvm::ConstantInt::getTrue(), *PI);
1099
1100 // Emit the RHS condition as a bool value.
Chris Lattner7f02f722007-08-24 05:35:26 +00001101 CGF.EmitBlock(RHSBlock);
1102 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
1103
1104 // Reaquire the RHS block, as there may be subblocks inserted.
1105 RHSBlock = Builder.GetInsertBlock();
Chris Lattner7f02f722007-08-24 05:35:26 +00001106
Chris Lattnerf7b5ea92008-11-12 08:38:24 +00001107 // Emit an unconditional branch from this block to ContBlock. Insert an entry
1108 // into the phi node for the edge with the value of RHSCond.
1109 CGF.EmitBlock(ContBlock);
Chris Lattner7f02f722007-08-24 05:35:26 +00001110 PN->addIncoming(RHSCond, RHSBlock);
1111
1112 // ZExt result to int.
1113 return Builder.CreateZExt(PN, CGF.LLVMIntTy, "lor.ext");
1114}
1115
1116Value *ScalarExprEmitter::VisitBinComma(const BinaryOperator *E) {
1117 CGF.EmitStmt(E->getLHS());
Daniel Dunbara448fb22008-11-11 23:11:34 +00001118 CGF.EnsureInsertPoint();
Chris Lattner7f02f722007-08-24 05:35:26 +00001119 return Visit(E->getRHS());
1120}
1121
1122//===----------------------------------------------------------------------===//
1123// Other Operators
1124//===----------------------------------------------------------------------===//
1125
Chris Lattner9802a512008-11-12 08:55:54 +00001126/// isCheapEnoughToEvaluateUnconditionally - Return true if the specified
1127/// expression is cheap enough and side-effect-free enough to evaluate
1128/// unconditionally instead of conditionally. This is used to convert control
1129/// flow into selects in some cases.
1130static bool isCheapEnoughToEvaluateUnconditionally(const Expr *E) {
1131 if (const ParenExpr *PE = dyn_cast<ParenExpr>(E))
1132 return isCheapEnoughToEvaluateUnconditionally(PE->getSubExpr());
1133
1134 // TODO: Allow anything we can constant fold to an integer or fp constant.
1135 if (isa<IntegerLiteral>(E) || isa<CharacterLiteral>(E) ||
1136 isa<FloatingLiteral>(E))
1137 return true;
1138
1139 // Non-volatile automatic variables too, to get "cond ? X : Y" where
1140 // X and Y are local variables.
1141 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
1142 if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
1143 if (VD->hasLocalStorage() && !VD->getType().isVolatileQualified())
1144 return true;
1145
1146 return false;
1147}
1148
1149
Chris Lattner7f02f722007-08-24 05:35:26 +00001150Value *ScalarExprEmitter::
1151VisitConditionalOperator(const ConditionalOperator *E) {
Chris Lattner31a09842008-11-12 08:04:58 +00001152 // If the condition constant folds and can be elided, try to avoid emitting
1153 // the condition and the dead arm.
1154 if (int Cond = CGF.ConstantFoldsToSimpleInteger(E->getCond())){
Chris Lattnerc657e922008-11-11 18:56:45 +00001155 Expr *Live = E->getLHS(), *Dead = E->getRHS();
Chris Lattner31a09842008-11-12 08:04:58 +00001156 if (Cond == -1)
Chris Lattnerc657e922008-11-11 18:56:45 +00001157 std::swap(Live, Dead);
Chris Lattner31a09842008-11-12 08:04:58 +00001158
1159 // If the dead side doesn't have labels we need, and if the Live side isn't
1160 // the gnu missing ?: extension (which we could handle, but don't bother
1161 // to), just emit the Live part.
1162 if ((!Dead || !CGF.ContainsLabel(Dead)) && // No labels in dead part
1163 Live) // Live part isn't missing.
1164 return Visit(Live);
Chris Lattnerc657e922008-11-11 18:56:45 +00001165 }
1166
Chris Lattner9802a512008-11-12 08:55:54 +00001167
1168 // If this is a really simple expression (like x ? 4 : 5), emit this as a
1169 // select instead of as control flow. We can only do this if it is cheap and
Chris Lattner531a5502008-11-16 06:16:27 +00001170 // safe to evaluate the LHS and RHS unconditionally.
Chris Lattner9802a512008-11-12 08:55:54 +00001171 if (E->getLHS() && isCheapEnoughToEvaluateUnconditionally(E->getLHS()) &&
1172 isCheapEnoughToEvaluateUnconditionally(E->getRHS())) {
1173 llvm::Value *CondV = CGF.EvaluateExprAsBool(E->getCond());
1174 llvm::Value *LHS = Visit(E->getLHS());
1175 llvm::Value *RHS = Visit(E->getRHS());
1176 return Builder.CreateSelect(CondV, LHS, RHS, "cond");
1177 }
1178
1179
Daniel Dunbarbe65abc2008-11-12 10:13:37 +00001180 llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true");
1181 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false");
Daniel Dunbar9615ecb2008-11-13 01:38:36 +00001182 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end");
Chris Lattner035cf422008-11-12 08:08:13 +00001183 Value *CondVal = 0;
Chris Lattner31a09842008-11-12 08:04:58 +00001184
Chris Lattner035cf422008-11-12 08:08:13 +00001185 // If we have the GNU missing condition extension, evaluate the conditional
1186 // and then convert it to bool the hard way. We do this explicitly
1187 // because we need the unconverted value for the missing middle value of
1188 // the ?:.
1189 if (E->getLHS() == 0) {
1190 CondVal = CGF.EmitScalarExpr(E->getCond());
1191 Value *CondBoolVal =
1192 CGF.EmitScalarConversion(CondVal, E->getCond()->getType(),
1193 CGF.getContext().BoolTy);
1194 Builder.CreateCondBr(CondBoolVal, LHSBlock, RHSBlock);
1195 } else {
1196 // Otherwise, just use EmitBranchOnBoolExpr to get small and simple code for
1197 // the branch on bool.
1198 CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock);
1199 }
Chris Lattner7f02f722007-08-24 05:35:26 +00001200
1201 CGF.EmitBlock(LHSBlock);
1202
1203 // Handle the GNU extension for missing LHS.
Chris Lattnera21ddb32007-11-26 01:40:58 +00001204 Value *LHS;
1205 if (E->getLHS())
Eli Friedman856226c2008-05-16 20:38:39 +00001206 LHS = Visit(E->getLHS());
Chris Lattnera21ddb32007-11-26 01:40:58 +00001207 else // Perform promotions, to handle cases like "short ?: int"
1208 LHS = EmitScalarConversion(CondVal, E->getCond()->getType(), E->getType());
1209
Chris Lattner7f02f722007-08-24 05:35:26 +00001210 LHSBlock = Builder.GetInsertBlock();
Daniel Dunbard57a8712008-11-11 09:41:28 +00001211 CGF.EmitBranch(ContBlock);
Chris Lattner7f02f722007-08-24 05:35:26 +00001212
1213 CGF.EmitBlock(RHSBlock);
1214
Eli Friedman856226c2008-05-16 20:38:39 +00001215 Value *RHS = Visit(E->getRHS());
Chris Lattner7f02f722007-08-24 05:35:26 +00001216 RHSBlock = Builder.GetInsertBlock();
Daniel Dunbard57a8712008-11-11 09:41:28 +00001217 CGF.EmitBranch(ContBlock);
Chris Lattner7f02f722007-08-24 05:35:26 +00001218
1219 CGF.EmitBlock(ContBlock);
1220
Nuno Lopes108f55d2008-06-04 19:15:45 +00001221 if (!LHS || !RHS) {
Chris Lattner2202bce2007-11-30 17:56:23 +00001222 assert(E->getType()->isVoidType() && "Non-void value should have a value");
1223 return 0;
1224 }
1225
Chris Lattner7f02f722007-08-24 05:35:26 +00001226 // Create a PHI node for the real part.
1227 llvm::PHINode *PN = Builder.CreatePHI(LHS->getType(), "cond");
1228 PN->reserveOperandSpace(2);
1229 PN->addIncoming(LHS, LHSBlock);
1230 PN->addIncoming(RHS, RHSBlock);
1231 return PN;
1232}
1233
1234Value *ScalarExprEmitter::VisitChooseExpr(ChooseExpr *E) {
Chris Lattner7f02f722007-08-24 05:35:26 +00001235 // Emit the LHS or RHS as appropriate.
Devang Patele9b8c0a2007-10-30 20:59:40 +00001236 return
1237 Visit(E->isConditionTrue(CGF.getContext()) ? E->getLHS() : E->getRHS());
Chris Lattner7f02f722007-08-24 05:35:26 +00001238}
1239
Nate Begemane2ce1d92008-01-17 17:46:27 +00001240Value *ScalarExprEmitter::VisitOverloadExpr(OverloadExpr *E) {
Nate Begeman67295d02008-01-30 20:50:20 +00001241 return CGF.EmitCallExpr(E->getFn(), E->arg_begin(),
Ted Kremenek55499762008-06-17 02:43:46 +00001242 E->arg_end(CGF.getContext())).getScalarVal();
Nate Begemane2ce1d92008-01-17 17:46:27 +00001243}
1244
Chris Lattner2202bce2007-11-30 17:56:23 +00001245Value *ScalarExprEmitter::VisitVAArgExpr(VAArgExpr *VE) {
Anders Carlsson7c50aca2007-10-15 20:28:48 +00001246 llvm::Value *ArgValue = EmitLValue(VE->getSubExpr()).getAddress();
1247
Anders Carlssonddf7cac2008-11-04 05:30:00 +00001248 llvm::Value *ArgPtr = CGF.EmitVAArg(ArgValue, VE->getType());
1249
1250 // If EmitVAArg fails, we fall back to the LLVM instruction.
1251 if (!ArgPtr)
1252 return Builder.CreateVAArg(ArgValue, ConvertType(VE->getType()));
1253
1254 // FIXME: volatile?
1255 return Builder.CreateLoad(ArgPtr);
Anders Carlsson7c50aca2007-10-15 20:28:48 +00001256}
1257
Chris Lattner2202bce2007-11-30 17:56:23 +00001258Value *ScalarExprEmitter::VisitObjCEncodeExpr(const ObjCEncodeExpr *E) {
Anders Carlsson85f9bce2007-10-29 05:01:08 +00001259 std::string str;
Daniel Dunbar0d504c12008-10-17 20:21:44 +00001260 CGF.getContext().getObjCEncodingForType(E->getEncodedType(), str);
Anders Carlsson85f9bce2007-10-29 05:01:08 +00001261
1262 llvm::Constant *C = llvm::ConstantArray::get(str);
1263 C = new llvm::GlobalVariable(C->getType(), true,
1264 llvm::GlobalValue::InternalLinkage,
1265 C, ".str", &CGF.CGM.getModule());
1266 llvm::Constant *Zero = llvm::Constant::getNullValue(llvm::Type::Int32Ty);
1267 llvm::Constant *Zeros[] = { Zero, Zero };
1268 C = llvm::ConstantExpr::getGetElementPtr(C, Zeros, 2);
1269
1270 return C;
1271}
1272
Chris Lattner7f02f722007-08-24 05:35:26 +00001273//===----------------------------------------------------------------------===//
1274// Entry Point into this File
1275//===----------------------------------------------------------------------===//
1276
1277/// EmitComplexExpr - Emit the computation of the specified expression of
1278/// complex type, ignoring the result.
1279Value *CodeGenFunction::EmitScalarExpr(const Expr *E) {
1280 assert(E && !hasAggregateLLVMType(E->getType()) &&
1281 "Invalid scalar expression to emit");
1282
1283 return ScalarExprEmitter(*this).Visit(const_cast<Expr*>(E));
1284}
Chris Lattner3707b252007-08-26 06:48:56 +00001285
1286/// EmitScalarConversion - Emit a conversion from the specified type to the
1287/// specified destination type, both of which are LLVM scalar types.
Chris Lattner4f1a7b32007-08-26 16:34:22 +00001288Value *CodeGenFunction::EmitScalarConversion(Value *Src, QualType SrcTy,
1289 QualType DstTy) {
Chris Lattner3707b252007-08-26 06:48:56 +00001290 assert(!hasAggregateLLVMType(SrcTy) && !hasAggregateLLVMType(DstTy) &&
1291 "Invalid scalar expression to emit");
1292 return ScalarExprEmitter(*this).EmitScalarConversion(Src, SrcTy, DstTy);
1293}
Chris Lattner4f1a7b32007-08-26 16:34:22 +00001294
1295/// EmitComplexToScalarConversion - Emit a conversion from the specified
1296/// complex type to the specified destination type, where the destination
1297/// type is an LLVM scalar type.
1298Value *CodeGenFunction::EmitComplexToScalarConversion(ComplexPairTy Src,
1299 QualType SrcTy,
1300 QualType DstTy) {
Chris Lattner9b2dc282008-04-04 16:54:41 +00001301 assert(SrcTy->isAnyComplexType() && !hasAggregateLLVMType(DstTy) &&
Chris Lattner4f1a7b32007-08-26 16:34:22 +00001302 "Invalid complex -> scalar conversion");
1303 return ScalarExprEmitter(*this).EmitComplexToScalarConversion(Src, SrcTy,
1304 DstTy);
1305}
Anders Carlssoncc23aca2007-12-10 19:35:18 +00001306
1307Value *CodeGenFunction::EmitShuffleVector(Value* V1, Value *V2, ...) {
1308 assert(V1->getType() == V2->getType() &&
1309 "Vector operands must be of the same type");
Anders Carlssoncc23aca2007-12-10 19:35:18 +00001310 unsigned NumElements =
1311 cast<llvm::VectorType>(V1->getType())->getNumElements();
1312
1313 va_list va;
1314 va_start(va, V2);
1315
1316 llvm::SmallVector<llvm::Constant*, 16> Args;
Anders Carlssoncc23aca2007-12-10 19:35:18 +00001317 for (unsigned i = 0; i < NumElements; i++) {
1318 int n = va_arg(va, int);
Anders Carlssoncc23aca2007-12-10 19:35:18 +00001319 assert(n >= 0 && n < (int)NumElements * 2 &&
1320 "Vector shuffle index out of bounds!");
Anders Carlssoncc23aca2007-12-10 19:35:18 +00001321 Args.push_back(llvm::ConstantInt::get(llvm::Type::Int32Ty, n));
1322 }
1323
1324 const char *Name = va_arg(va, const char *);
1325 va_end(va);
1326
1327 llvm::Constant *Mask = llvm::ConstantVector::get(&Args[0], NumElements);
1328
1329 return Builder.CreateShuffleVector(V1, V2, Mask, Name);
1330}
1331
Anders Carlsson6086bbd2007-12-15 21:23:30 +00001332llvm::Value *CodeGenFunction::EmitVector(llvm::Value * const *Vals,
Chris Lattner345f7202008-07-26 20:15:14 +00001333 unsigned NumVals, bool isSplat) {
Anders Carlsson6086bbd2007-12-15 21:23:30 +00001334 llvm::Value *Vec
Chris Lattner345f7202008-07-26 20:15:14 +00001335 = llvm::UndefValue::get(llvm::VectorType::get(Vals[0]->getType(), NumVals));
Anders Carlsson6086bbd2007-12-15 21:23:30 +00001336
Chris Lattner345f7202008-07-26 20:15:14 +00001337 for (unsigned i = 0, e = NumVals; i != e; ++i) {
Nate Begeman4119d1a2007-12-30 02:59:45 +00001338 llvm::Value *Val = isSplat ? Vals[0] : Vals[i];
Anders Carlsson6086bbd2007-12-15 21:23:30 +00001339 llvm::Value *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty, i);
Nate Begeman4119d1a2007-12-30 02:59:45 +00001340 Vec = Builder.CreateInsertElement(Vec, Val, Idx, "tmp");
Anders Carlsson6086bbd2007-12-15 21:23:30 +00001341 }
1342
1343 return Vec;
1344}