1f22ef01cSRoman Divacky //===--- CGExprComplex.cpp - Emit LLVM Code for Complex Exprs -------------===//
2f22ef01cSRoman Divacky //
3f22ef01cSRoman Divacky // The LLVM Compiler Infrastructure
4f22ef01cSRoman Divacky //
5f22ef01cSRoman Divacky // This file is distributed under the University of Illinois Open Source
6f22ef01cSRoman Divacky // License. See LICENSE.TXT for details.
7f22ef01cSRoman Divacky //
8f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
9f22ef01cSRoman Divacky //
10f22ef01cSRoman Divacky // This contains code to emit Expr nodes with complex types as LLVM code.
11f22ef01cSRoman Divacky //
12f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
13f22ef01cSRoman Divacky
14f22ef01cSRoman Divacky #include "CodeGenFunction.h"
15f22ef01cSRoman Divacky #include "CodeGenModule.h"
16f22ef01cSRoman Divacky #include "clang/AST/StmtVisitor.h"
1739d628a0SDimitry Andric #include "llvm/ADT/STLExtras.h"
18139f7f9bSDimitry Andric #include "llvm/IR/Constants.h"
1939d628a0SDimitry Andric #include "llvm/IR/Instructions.h"
2039d628a0SDimitry Andric #include "llvm/IR/MDBuilder.h"
2139d628a0SDimitry Andric #include "llvm/IR/Metadata.h"
22f785676fSDimitry Andric #include <algorithm>
23f22ef01cSRoman Divacky using namespace clang;
24f22ef01cSRoman Divacky using namespace CodeGen;
25f22ef01cSRoman Divacky
26f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
27f22ef01cSRoman Divacky // Complex Expression Emitter
28f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
29f22ef01cSRoman Divacky
30f22ef01cSRoman Divacky typedef CodeGenFunction::ComplexPairTy ComplexPairTy;
31f22ef01cSRoman Divacky
32139f7f9bSDimitry Andric /// Return the complex type that we are meant to emit.
getComplexType(QualType type)33139f7f9bSDimitry Andric static const ComplexType *getComplexType(QualType type) {
34139f7f9bSDimitry Andric type = type.getCanonicalType();
35139f7f9bSDimitry Andric if (const ComplexType *comp = dyn_cast<ComplexType>(type)) {
36139f7f9bSDimitry Andric return comp;
37139f7f9bSDimitry Andric } else {
38139f7f9bSDimitry Andric return cast<ComplexType>(cast<AtomicType>(type)->getValueType());
39139f7f9bSDimitry Andric }
40139f7f9bSDimitry Andric }
41139f7f9bSDimitry Andric
42f22ef01cSRoman Divacky namespace {
43f22ef01cSRoman Divacky class ComplexExprEmitter
44f22ef01cSRoman Divacky : public StmtVisitor<ComplexExprEmitter, ComplexPairTy> {
45f22ef01cSRoman Divacky CodeGenFunction &CGF;
46f22ef01cSRoman Divacky CGBuilderTy &Builder;
47f22ef01cSRoman Divacky bool IgnoreReal;
48f22ef01cSRoman Divacky bool IgnoreImag;
49f22ef01cSRoman Divacky public:
ComplexExprEmitter(CodeGenFunction & cgf,bool ir=false,bool ii=false)502754fe60SDimitry Andric ComplexExprEmitter(CodeGenFunction &cgf, bool ir=false, bool ii=false)
512754fe60SDimitry Andric : CGF(cgf), Builder(CGF.Builder), IgnoreReal(ir), IgnoreImag(ii) {
52f22ef01cSRoman Divacky }
53f22ef01cSRoman Divacky
54f22ef01cSRoman Divacky
55f22ef01cSRoman Divacky //===--------------------------------------------------------------------===//
56f22ef01cSRoman Divacky // Utilities
57f22ef01cSRoman Divacky //===--------------------------------------------------------------------===//
58f22ef01cSRoman Divacky
TestAndClearIgnoreReal()59f22ef01cSRoman Divacky bool TestAndClearIgnoreReal() {
60f22ef01cSRoman Divacky bool I = IgnoreReal;
61f22ef01cSRoman Divacky IgnoreReal = false;
62f22ef01cSRoman Divacky return I;
63f22ef01cSRoman Divacky }
TestAndClearIgnoreImag()64f22ef01cSRoman Divacky bool TestAndClearIgnoreImag() {
65f22ef01cSRoman Divacky bool I = IgnoreImag;
66f22ef01cSRoman Divacky IgnoreImag = false;
67f22ef01cSRoman Divacky return I;
68f22ef01cSRoman Divacky }
69f22ef01cSRoman Divacky
70f22ef01cSRoman Divacky /// EmitLoadOfLValue - Given an expression with complex type that represents a
71f22ef01cSRoman Divacky /// value l-value, this method emits the address of the l-value, then loads
72f22ef01cSRoman Divacky /// and returns the result.
EmitLoadOfLValue(const Expr * E)73f22ef01cSRoman Divacky ComplexPairTy EmitLoadOfLValue(const Expr *E) {
74f785676fSDimitry Andric return EmitLoadOfLValue(CGF.EmitLValue(E), E->getExprLoc());
752754fe60SDimitry Andric }
762754fe60SDimitry Andric
77f785676fSDimitry Andric ComplexPairTy EmitLoadOfLValue(LValue LV, SourceLocation Loc);
782754fe60SDimitry Andric
79f22ef01cSRoman Divacky /// EmitStoreOfComplex - Store the specified real/imag parts into the
80f22ef01cSRoman Divacky /// specified value pointer.
81139f7f9bSDimitry Andric void EmitStoreOfComplex(ComplexPairTy Val, LValue LV, bool isInit);
82f22ef01cSRoman Divacky
830623d748SDimitry Andric /// Emit a cast from complex value Val to DestType.
84f22ef01cSRoman Divacky ComplexPairTy EmitComplexToComplexCast(ComplexPairTy Val, QualType SrcType,
850623d748SDimitry Andric QualType DestType, SourceLocation Loc);
860623d748SDimitry Andric /// Emit a cast from scalar value Val to DestType.
87f785676fSDimitry Andric ComplexPairTy EmitScalarToComplexCast(llvm::Value *Val, QualType SrcType,
880623d748SDimitry Andric QualType DestType, SourceLocation Loc);
89f22ef01cSRoman Divacky
90f22ef01cSRoman Divacky //===--------------------------------------------------------------------===//
91f22ef01cSRoman Divacky // Visitor Methods
92f22ef01cSRoman Divacky //===--------------------------------------------------------------------===//
93f22ef01cSRoman Divacky
Visit(Expr * E)942754fe60SDimitry Andric ComplexPairTy Visit(Expr *E) {
9533956c43SDimitry Andric ApplyDebugLocation DL(CGF, E);
962754fe60SDimitry Andric return StmtVisitor<ComplexExprEmitter, ComplexPairTy>::Visit(E);
972754fe60SDimitry Andric }
982754fe60SDimitry Andric
VisitStmt(Stmt * S)99f22ef01cSRoman Divacky ComplexPairTy VisitStmt(Stmt *S) {
100f22ef01cSRoman Divacky S->dump(CGF.getContext().getSourceManager());
1016122f3e6SDimitry Andric llvm_unreachable("Stmt can't have complex result type!");
102f22ef01cSRoman Divacky }
103f22ef01cSRoman Divacky ComplexPairTy VisitExpr(Expr *S);
VisitConstantExpr(ConstantExpr * E)104*b5893f02SDimitry Andric ComplexPairTy VisitConstantExpr(ConstantExpr *E) {
105*b5893f02SDimitry Andric return Visit(E->getSubExpr());
106*b5893f02SDimitry Andric }
VisitParenExpr(ParenExpr * PE)107f22ef01cSRoman Divacky ComplexPairTy VisitParenExpr(ParenExpr *PE) { return Visit(PE->getSubExpr());}
VisitGenericSelectionExpr(GenericSelectionExpr * GE)1083b0f4066SDimitry Andric ComplexPairTy VisitGenericSelectionExpr(GenericSelectionExpr *GE) {
1093b0f4066SDimitry Andric return Visit(GE->getResultExpr());
1103b0f4066SDimitry Andric }
111f22ef01cSRoman Divacky ComplexPairTy VisitImaginaryLiteral(const ImaginaryLiteral *IL);
11217a519f9SDimitry Andric ComplexPairTy
VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr * PE)11317a519f9SDimitry Andric VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *PE) {
11417a519f9SDimitry Andric return Visit(PE->getReplacement());
11517a519f9SDimitry Andric }
VisitCoawaitExpr(CoawaitExpr * S)11620e90f04SDimitry Andric ComplexPairTy VisitCoawaitExpr(CoawaitExpr *S) {
11720e90f04SDimitry Andric return CGF.EmitCoawaitExpr(*S).getComplexVal();
11820e90f04SDimitry Andric }
VisitCoyieldExpr(CoyieldExpr * S)11920e90f04SDimitry Andric ComplexPairTy VisitCoyieldExpr(CoyieldExpr *S) {
12020e90f04SDimitry Andric return CGF.EmitCoyieldExpr(*S).getComplexVal();
12120e90f04SDimitry Andric }
VisitUnaryCoawait(const UnaryOperator * E)12220e90f04SDimitry Andric ComplexPairTy VisitUnaryCoawait(const UnaryOperator *E) {
12320e90f04SDimitry Andric return Visit(E->getSubExpr());
12420e90f04SDimitry Andric }
12520e90f04SDimitry Andric
emitConstant(const CodeGenFunction::ConstantEmission & Constant,Expr * E)1269a199699SDimitry Andric ComplexPairTy emitConstant(const CodeGenFunction::ConstantEmission &Constant,
1279a199699SDimitry Andric Expr *E) {
1289a199699SDimitry Andric assert(Constant && "not a constant");
1299a199699SDimitry Andric if (Constant.isReference())
1309a199699SDimitry Andric return EmitLoadOfLValue(Constant.getReferenceLValue(CGF, E),
131f785676fSDimitry Andric E->getExprLoc());
132dff0c46cSDimitry Andric
1339a199699SDimitry Andric llvm::Constant *pair = Constant.getValue();
134f785676fSDimitry Andric return ComplexPairTy(pair->getAggregateElement(0U),
135f785676fSDimitry Andric pair->getAggregateElement(1U));
136dff0c46cSDimitry Andric }
1379a199699SDimitry Andric
1389a199699SDimitry Andric // l-values.
VisitDeclRefExpr(DeclRefExpr * E)1399a199699SDimitry Andric ComplexPairTy VisitDeclRefExpr(DeclRefExpr *E) {
1409a199699SDimitry Andric if (CodeGenFunction::ConstantEmission Constant = CGF.tryEmitAsConstant(E))
1419a199699SDimitry Andric return emitConstant(Constant, E);
142f22ef01cSRoman Divacky return EmitLoadOfLValue(E);
143f22ef01cSRoman Divacky }
VisitObjCIvarRefExpr(ObjCIvarRefExpr * E)144dff0c46cSDimitry Andric ComplexPairTy VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
145f22ef01cSRoman Divacky return EmitLoadOfLValue(E);
146f22ef01cSRoman Divacky }
VisitObjCMessageExpr(ObjCMessageExpr * E)147f22ef01cSRoman Divacky ComplexPairTy VisitObjCMessageExpr(ObjCMessageExpr *E) {
148f22ef01cSRoman Divacky return CGF.EmitObjCMessageExpr(E).getComplexVal();
149f22ef01cSRoman Divacky }
VisitArraySubscriptExpr(Expr * E)150f22ef01cSRoman Divacky ComplexPairTy VisitArraySubscriptExpr(Expr *E) { return EmitLoadOfLValue(E); }
VisitMemberExpr(MemberExpr * ME)1519a199699SDimitry Andric ComplexPairTy VisitMemberExpr(MemberExpr *ME) {
1529a199699SDimitry Andric if (CodeGenFunction::ConstantEmission Constant =
1539a199699SDimitry Andric CGF.tryEmitAsConstant(ME)) {
1549a199699SDimitry Andric CGF.EmitIgnoredExpr(ME->getBase());
1559a199699SDimitry Andric return emitConstant(Constant, ME);
1569a199699SDimitry Andric }
1579a199699SDimitry Andric return EmitLoadOfLValue(ME);
1589a199699SDimitry Andric }
VisitOpaqueValueExpr(OpaqueValueExpr * E)1592754fe60SDimitry Andric ComplexPairTy VisitOpaqueValueExpr(OpaqueValueExpr *E) {
1602754fe60SDimitry Andric if (E->isGLValue())
1614ba319b5SDimitry Andric return EmitLoadOfLValue(CGF.getOrCreateOpaqueLValueMapping(E),
1624ba319b5SDimitry Andric E->getExprLoc());
1634ba319b5SDimitry Andric return CGF.getOrCreateOpaqueRValueMapping(E).getComplexVal();
1642754fe60SDimitry Andric }
165f22ef01cSRoman Divacky
VisitPseudoObjectExpr(PseudoObjectExpr * E)166dff0c46cSDimitry Andric ComplexPairTy VisitPseudoObjectExpr(PseudoObjectExpr *E) {
167dff0c46cSDimitry Andric return CGF.EmitPseudoObjectRValue(E).getComplexVal();
168dff0c46cSDimitry Andric }
169dff0c46cSDimitry Andric
170f22ef01cSRoman Divacky // FIXME: CompoundLiteralExpr
171f22ef01cSRoman Divacky
17239d628a0SDimitry Andric ComplexPairTy EmitCast(CastKind CK, Expr *Op, QualType DestTy);
VisitImplicitCastExpr(ImplicitCastExpr * E)173f22ef01cSRoman Divacky ComplexPairTy VisitImplicitCastExpr(ImplicitCastExpr *E) {
174f22ef01cSRoman Divacky // Unlike for scalars, we don't have to worry about function->ptr demotion
175f22ef01cSRoman Divacky // here.
176ffd1746dSEd Schouten return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType());
177f22ef01cSRoman Divacky }
VisitCastExpr(CastExpr * E)178f22ef01cSRoman Divacky ComplexPairTy VisitCastExpr(CastExpr *E) {
1790623d748SDimitry Andric if (const auto *ECE = dyn_cast<ExplicitCastExpr>(E))
1800623d748SDimitry Andric CGF.CGM.EmitExplicitCastExprType(ECE, &CGF);
181ffd1746dSEd Schouten return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType());
182f22ef01cSRoman Divacky }
183f22ef01cSRoman Divacky ComplexPairTy VisitCallExpr(const CallExpr *E);
184f22ef01cSRoman Divacky ComplexPairTy VisitStmtExpr(const StmtExpr *E);
185f22ef01cSRoman Divacky
186f22ef01cSRoman Divacky // Operators.
VisitPrePostIncDec(const UnaryOperator * E,bool isInc,bool isPre)187f22ef01cSRoman Divacky ComplexPairTy VisitPrePostIncDec(const UnaryOperator *E,
188f22ef01cSRoman Divacky bool isInc, bool isPre) {
189f22ef01cSRoman Divacky LValue LV = CGF.EmitLValue(E->getSubExpr());
190f22ef01cSRoman Divacky return CGF.EmitComplexPrePostIncDec(E, LV, isInc, isPre);
191f22ef01cSRoman Divacky }
VisitUnaryPostDec(const UnaryOperator * E)192f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPostDec(const UnaryOperator *E) {
193f22ef01cSRoman Divacky return VisitPrePostIncDec(E, false, false);
194f22ef01cSRoman Divacky }
VisitUnaryPostInc(const UnaryOperator * E)195f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPostInc(const UnaryOperator *E) {
196f22ef01cSRoman Divacky return VisitPrePostIncDec(E, true, false);
197f22ef01cSRoman Divacky }
VisitUnaryPreDec(const UnaryOperator * E)198f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPreDec(const UnaryOperator *E) {
199f22ef01cSRoman Divacky return VisitPrePostIncDec(E, false, true);
200f22ef01cSRoman Divacky }
VisitUnaryPreInc(const UnaryOperator * E)201f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPreInc(const UnaryOperator *E) {
202f22ef01cSRoman Divacky return VisitPrePostIncDec(E, true, true);
203f22ef01cSRoman Divacky }
VisitUnaryDeref(const Expr * E)204f22ef01cSRoman Divacky ComplexPairTy VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); }
VisitUnaryPlus(const UnaryOperator * E)205f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPlus (const UnaryOperator *E) {
206f22ef01cSRoman Divacky TestAndClearIgnoreReal();
207f22ef01cSRoman Divacky TestAndClearIgnoreImag();
208f22ef01cSRoman Divacky return Visit(E->getSubExpr());
209f22ef01cSRoman Divacky }
210f22ef01cSRoman Divacky ComplexPairTy VisitUnaryMinus (const UnaryOperator *E);
211f22ef01cSRoman Divacky ComplexPairTy VisitUnaryNot (const UnaryOperator *E);
212f22ef01cSRoman Divacky // LNot,Real,Imag never return complex.
VisitUnaryExtension(const UnaryOperator * E)213f22ef01cSRoman Divacky ComplexPairTy VisitUnaryExtension(const UnaryOperator *E) {
214f22ef01cSRoman Divacky return Visit(E->getSubExpr());
215f22ef01cSRoman Divacky }
VisitCXXDefaultArgExpr(CXXDefaultArgExpr * DAE)216f22ef01cSRoman Divacky ComplexPairTy VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
217f22ef01cSRoman Divacky return Visit(DAE->getExpr());
218f22ef01cSRoman Divacky }
VisitCXXDefaultInitExpr(CXXDefaultInitExpr * DIE)219284c1978SDimitry Andric ComplexPairTy VisitCXXDefaultInitExpr(CXXDefaultInitExpr *DIE) {
220284c1978SDimitry Andric CodeGenFunction::CXXDefaultInitExprScope Scope(CGF);
221284c1978SDimitry Andric return Visit(DIE->getExpr());
222284c1978SDimitry Andric }
VisitExprWithCleanups(ExprWithCleanups * E)2232754fe60SDimitry Andric ComplexPairTy VisitExprWithCleanups(ExprWithCleanups *E) {
224dff0c46cSDimitry Andric CGF.enterFullExpression(E);
225dff0c46cSDimitry Andric CodeGenFunction::RunCleanupsScope Scope(CGF);
22620e90f04SDimitry Andric ComplexPairTy Vals = Visit(E->getSubExpr());
22720e90f04SDimitry Andric // Defend against dominance problems caused by jumps out of expression
22820e90f04SDimitry Andric // evaluation through the shared cleanup block.
22920e90f04SDimitry Andric Scope.ForceCleanup({&Vals.first, &Vals.second});
23020e90f04SDimitry Andric return Vals;
231f22ef01cSRoman Divacky }
VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr * E)232ffd1746dSEd Schouten ComplexPairTy VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
233f22ef01cSRoman Divacky assert(E->getType()->isAnyComplexType() && "Expected complex type!");
234139f7f9bSDimitry Andric QualType Elem = E->getType()->castAs<ComplexType>()->getElementType();
235f22ef01cSRoman Divacky llvm::Constant *Null = llvm::Constant::getNullValue(CGF.ConvertType(Elem));
236f22ef01cSRoman Divacky return ComplexPairTy(Null, Null);
237f22ef01cSRoman Divacky }
VisitImplicitValueInitExpr(ImplicitValueInitExpr * E)238f22ef01cSRoman Divacky ComplexPairTy VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
239f22ef01cSRoman Divacky assert(E->getType()->isAnyComplexType() && "Expected complex type!");
240139f7f9bSDimitry Andric QualType Elem = E->getType()->castAs<ComplexType>()->getElementType();
241f22ef01cSRoman Divacky llvm::Constant *Null =
242f22ef01cSRoman Divacky llvm::Constant::getNullValue(CGF.ConvertType(Elem));
243f22ef01cSRoman Divacky return ComplexPairTy(Null, Null);
244f22ef01cSRoman Divacky }
245f22ef01cSRoman Divacky
246f22ef01cSRoman Divacky struct BinOpInfo {
247f22ef01cSRoman Divacky ComplexPairTy LHS;
248f22ef01cSRoman Divacky ComplexPairTy RHS;
249f22ef01cSRoman Divacky QualType Ty; // Computation Type.
250f22ef01cSRoman Divacky };
251f22ef01cSRoman Divacky
252f22ef01cSRoman Divacky BinOpInfo EmitBinOps(const BinaryOperator *E);
2532754fe60SDimitry Andric LValue EmitCompoundAssignLValue(const CompoundAssignOperator *E,
2542754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)
2552754fe60SDimitry Andric (const BinOpInfo &),
256f785676fSDimitry Andric RValue &Val);
257f22ef01cSRoman Divacky ComplexPairTy EmitCompoundAssign(const CompoundAssignOperator *E,
258f22ef01cSRoman Divacky ComplexPairTy (ComplexExprEmitter::*Func)
259f22ef01cSRoman Divacky (const BinOpInfo &));
260f22ef01cSRoman Divacky
261f22ef01cSRoman Divacky ComplexPairTy EmitBinAdd(const BinOpInfo &Op);
262f22ef01cSRoman Divacky ComplexPairTy EmitBinSub(const BinOpInfo &Op);
263f22ef01cSRoman Divacky ComplexPairTy EmitBinMul(const BinOpInfo &Op);
264f22ef01cSRoman Divacky ComplexPairTy EmitBinDiv(const BinOpInfo &Op);
265f22ef01cSRoman Divacky
26639d628a0SDimitry Andric ComplexPairTy EmitComplexBinOpLibCall(StringRef LibCallName,
26739d628a0SDimitry Andric const BinOpInfo &Op);
26839d628a0SDimitry Andric
VisitBinAdd(const BinaryOperator * E)269f22ef01cSRoman Divacky ComplexPairTy VisitBinAdd(const BinaryOperator *E) {
270f22ef01cSRoman Divacky return EmitBinAdd(EmitBinOps(E));
271f22ef01cSRoman Divacky }
VisitBinSub(const BinaryOperator * E)272f22ef01cSRoman Divacky ComplexPairTy VisitBinSub(const BinaryOperator *E) {
273f22ef01cSRoman Divacky return EmitBinSub(EmitBinOps(E));
274f22ef01cSRoman Divacky }
VisitBinMul(const BinaryOperator * E)2752754fe60SDimitry Andric ComplexPairTy VisitBinMul(const BinaryOperator *E) {
2762754fe60SDimitry Andric return EmitBinMul(EmitBinOps(E));
2772754fe60SDimitry Andric }
VisitBinDiv(const BinaryOperator * E)278f22ef01cSRoman Divacky ComplexPairTy VisitBinDiv(const BinaryOperator *E) {
279f22ef01cSRoman Divacky return EmitBinDiv(EmitBinOps(E));
280f22ef01cSRoman Divacky }
281f22ef01cSRoman Divacky
282f22ef01cSRoman Divacky // Compound assignments.
VisitBinAddAssign(const CompoundAssignOperator * E)283f22ef01cSRoman Divacky ComplexPairTy VisitBinAddAssign(const CompoundAssignOperator *E) {
284f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinAdd);
285f22ef01cSRoman Divacky }
VisitBinSubAssign(const CompoundAssignOperator * E)286f22ef01cSRoman Divacky ComplexPairTy VisitBinSubAssign(const CompoundAssignOperator *E) {
287f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinSub);
288f22ef01cSRoman Divacky }
VisitBinMulAssign(const CompoundAssignOperator * E)289f22ef01cSRoman Divacky ComplexPairTy VisitBinMulAssign(const CompoundAssignOperator *E) {
290f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinMul);
291f22ef01cSRoman Divacky }
VisitBinDivAssign(const CompoundAssignOperator * E)292f22ef01cSRoman Divacky ComplexPairTy VisitBinDivAssign(const CompoundAssignOperator *E) {
293f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinDiv);
294f22ef01cSRoman Divacky }
295f22ef01cSRoman Divacky
296f22ef01cSRoman Divacky // GCC rejects rem/and/or/xor for integer complex.
297f22ef01cSRoman Divacky // Logical and/or always return int, never complex.
298f22ef01cSRoman Divacky
299f22ef01cSRoman Divacky // No comparisons produce a complex result.
3002754fe60SDimitry Andric
3012754fe60SDimitry Andric LValue EmitBinAssignLValue(const BinaryOperator *E,
3022754fe60SDimitry Andric ComplexPairTy &Val);
303f22ef01cSRoman Divacky ComplexPairTy VisitBinAssign (const BinaryOperator *E);
304f22ef01cSRoman Divacky ComplexPairTy VisitBinComma (const BinaryOperator *E);
305f22ef01cSRoman Divacky
306f22ef01cSRoman Divacky
3072754fe60SDimitry Andric ComplexPairTy
3082754fe60SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *CO);
309f22ef01cSRoman Divacky ComplexPairTy VisitChooseExpr(ChooseExpr *CE);
310f22ef01cSRoman Divacky
311f22ef01cSRoman Divacky ComplexPairTy VisitInitListExpr(InitListExpr *E);
312f22ef01cSRoman Divacky
VisitCompoundLiteralExpr(CompoundLiteralExpr * E)313dff0c46cSDimitry Andric ComplexPairTy VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
314dff0c46cSDimitry Andric return EmitLoadOfLValue(E);
315dff0c46cSDimitry Andric }
316dff0c46cSDimitry Andric
317f22ef01cSRoman Divacky ComplexPairTy VisitVAArgExpr(VAArgExpr *E);
3186122f3e6SDimitry Andric
VisitAtomicExpr(AtomicExpr * E)3196122f3e6SDimitry Andric ComplexPairTy VisitAtomicExpr(AtomicExpr *E) {
3206122f3e6SDimitry Andric return CGF.EmitAtomicExpr(E).getComplexVal();
3216122f3e6SDimitry Andric }
322f22ef01cSRoman Divacky };
323f22ef01cSRoman Divacky } // end anonymous namespace.
324f22ef01cSRoman Divacky
325f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
326f22ef01cSRoman Divacky // Utilities
327f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
328f22ef01cSRoman Divacky
emitAddrOfRealComponent(Address addr,QualType complexType)3290623d748SDimitry Andric Address CodeGenFunction::emitAddrOfRealComponent(Address addr,
3300623d748SDimitry Andric QualType complexType) {
3310623d748SDimitry Andric CharUnits offset = CharUnits::Zero();
3320623d748SDimitry Andric return Builder.CreateStructGEP(addr, 0, offset, addr.getName() + ".realp");
3330623d748SDimitry Andric }
3340623d748SDimitry Andric
emitAddrOfImagComponent(Address addr,QualType complexType)3350623d748SDimitry Andric Address CodeGenFunction::emitAddrOfImagComponent(Address addr,
3360623d748SDimitry Andric QualType complexType) {
3370623d748SDimitry Andric QualType eltType = complexType->castAs<ComplexType>()->getElementType();
3380623d748SDimitry Andric CharUnits offset = getContext().getTypeSizeInChars(eltType);
3390623d748SDimitry Andric return Builder.CreateStructGEP(addr, 1, offset, addr.getName() + ".imagp");
3400623d748SDimitry Andric }
3410623d748SDimitry Andric
342139f7f9bSDimitry Andric /// EmitLoadOfLValue - Given an RValue reference for a complex, emit code to
343f22ef01cSRoman Divacky /// load the real and imaginary pieces, returning them as Real/Imag.
EmitLoadOfLValue(LValue lvalue,SourceLocation loc)344f785676fSDimitry Andric ComplexPairTy ComplexExprEmitter::EmitLoadOfLValue(LValue lvalue,
345f785676fSDimitry Andric SourceLocation loc) {
346139f7f9bSDimitry Andric assert(lvalue.isSimple() && "non-simple complex l-value?");
347139f7f9bSDimitry Andric if (lvalue.getType()->isAtomicType())
348f785676fSDimitry Andric return CGF.EmitAtomicLoad(lvalue, loc).getComplexVal();
349139f7f9bSDimitry Andric
3500623d748SDimitry Andric Address SrcPtr = lvalue.getAddress();
351139f7f9bSDimitry Andric bool isVolatile = lvalue.isVolatileQualified();
352139f7f9bSDimitry Andric
35359d1ed5bSDimitry Andric llvm::Value *Real = nullptr, *Imag = nullptr;
354f22ef01cSRoman Divacky
3552754fe60SDimitry Andric if (!IgnoreReal || isVolatile) {
3560623d748SDimitry Andric Address RealP = CGF.emitAddrOfRealComponent(SrcPtr, lvalue.getType());
3570623d748SDimitry Andric Real = Builder.CreateLoad(RealP, isVolatile, SrcPtr.getName() + ".real");
358f22ef01cSRoman Divacky }
359f22ef01cSRoman Divacky
3602754fe60SDimitry Andric if (!IgnoreImag || isVolatile) {
3610623d748SDimitry Andric Address ImagP = CGF.emitAddrOfImagComponent(SrcPtr, lvalue.getType());
3620623d748SDimitry Andric Imag = Builder.CreateLoad(ImagP, isVolatile, SrcPtr.getName() + ".imag");
363f22ef01cSRoman Divacky }
3640623d748SDimitry Andric
365f22ef01cSRoman Divacky return ComplexPairTy(Real, Imag);
366f22ef01cSRoman Divacky }
367f22ef01cSRoman Divacky
368f22ef01cSRoman Divacky /// EmitStoreOfComplex - Store the specified real/imag parts into the
369f22ef01cSRoman Divacky /// specified value pointer.
EmitStoreOfComplex(ComplexPairTy Val,LValue lvalue,bool isInit)37039d628a0SDimitry Andric void ComplexExprEmitter::EmitStoreOfComplex(ComplexPairTy Val, LValue lvalue,
371139f7f9bSDimitry Andric bool isInit) {
37233956c43SDimitry Andric if (lvalue.getType()->isAtomicType() ||
37333956c43SDimitry Andric (!isInit && CGF.LValueIsSuitableForInlineAtomic(lvalue)))
374139f7f9bSDimitry Andric return CGF.EmitAtomicStore(RValue::getComplex(Val), lvalue, isInit);
375139f7f9bSDimitry Andric
3760623d748SDimitry Andric Address Ptr = lvalue.getAddress();
3770623d748SDimitry Andric Address RealPtr = CGF.emitAddrOfRealComponent(Ptr, lvalue.getType());
3780623d748SDimitry Andric Address ImagPtr = CGF.emitAddrOfImagComponent(Ptr, lvalue.getType());
379f22ef01cSRoman Divacky
3800623d748SDimitry Andric Builder.CreateStore(Val.first, RealPtr, lvalue.isVolatileQualified());
3810623d748SDimitry Andric Builder.CreateStore(Val.second, ImagPtr, lvalue.isVolatileQualified());
382f22ef01cSRoman Divacky }
383f22ef01cSRoman Divacky
384f22ef01cSRoman Divacky
385f22ef01cSRoman Divacky
386f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
387f22ef01cSRoman Divacky // Visitor Methods
388f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
389f22ef01cSRoman Divacky
VisitExpr(Expr * E)390f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitExpr(Expr *E) {
391f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex expression");
3926122f3e6SDimitry Andric llvm::Type *EltTy =
393139f7f9bSDimitry Andric CGF.ConvertType(getComplexType(E->getType())->getElementType());
394f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy);
395f22ef01cSRoman Divacky return ComplexPairTy(U, U);
396f22ef01cSRoman Divacky }
397f22ef01cSRoman Divacky
398f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::
VisitImaginaryLiteral(const ImaginaryLiteral * IL)399f22ef01cSRoman Divacky VisitImaginaryLiteral(const ImaginaryLiteral *IL) {
400f22ef01cSRoman Divacky llvm::Value *Imag = CGF.EmitScalarExpr(IL->getSubExpr());
4012754fe60SDimitry Andric return ComplexPairTy(llvm::Constant::getNullValue(Imag->getType()), Imag);
402f22ef01cSRoman Divacky }
403f22ef01cSRoman Divacky
404f22ef01cSRoman Divacky
VisitCallExpr(const CallExpr * E)405f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitCallExpr(const CallExpr *E) {
40633956c43SDimitry Andric if (E->getCallReturnType(CGF.getContext())->isReferenceType())
407f22ef01cSRoman Divacky return EmitLoadOfLValue(E);
408f22ef01cSRoman Divacky
409f22ef01cSRoman Divacky return CGF.EmitCallExpr(E).getComplexVal();
410f22ef01cSRoman Divacky }
411f22ef01cSRoman Divacky
VisitStmtExpr(const StmtExpr * E)412f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitStmtExpr(const StmtExpr *E) {
4132754fe60SDimitry Andric CodeGenFunction::StmtExprEvaluation eval(CGF);
4140623d748SDimitry Andric Address RetAlloca = CGF.EmitCompoundStmt(*E->getSubStmt(), true);
4150623d748SDimitry Andric assert(RetAlloca.isValid() && "Expected complex return value");
416f785676fSDimitry Andric return EmitLoadOfLValue(CGF.MakeAddrLValue(RetAlloca, E->getType()),
417f785676fSDimitry Andric E->getExprLoc());
418f22ef01cSRoman Divacky }
419f22ef01cSRoman Divacky
4200623d748SDimitry Andric /// Emit a cast from complex value Val to DestType.
EmitComplexToComplexCast(ComplexPairTy Val,QualType SrcType,QualType DestType,SourceLocation Loc)421f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitComplexToComplexCast(ComplexPairTy Val,
422f22ef01cSRoman Divacky QualType SrcType,
4230623d748SDimitry Andric QualType DestType,
4240623d748SDimitry Andric SourceLocation Loc) {
425f22ef01cSRoman Divacky // Get the src/dest element type.
426139f7f9bSDimitry Andric SrcType = SrcType->castAs<ComplexType>()->getElementType();
427139f7f9bSDimitry Andric DestType = DestType->castAs<ComplexType>()->getElementType();
428f22ef01cSRoman Divacky
429f22ef01cSRoman Divacky // C99 6.3.1.6: When a value of complex type is converted to another
430f22ef01cSRoman Divacky // complex type, both the real and imaginary parts follow the conversion
431f22ef01cSRoman Divacky // rules for the corresponding real types.
4320623d748SDimitry Andric Val.first = CGF.EmitScalarConversion(Val.first, SrcType, DestType, Loc);
4330623d748SDimitry Andric Val.second = CGF.EmitScalarConversion(Val.second, SrcType, DestType, Loc);
434f22ef01cSRoman Divacky return Val;
435f22ef01cSRoman Divacky }
436f22ef01cSRoman Divacky
EmitScalarToComplexCast(llvm::Value * Val,QualType SrcType,QualType DestType,SourceLocation Loc)437f785676fSDimitry Andric ComplexPairTy ComplexExprEmitter::EmitScalarToComplexCast(llvm::Value *Val,
438f785676fSDimitry Andric QualType SrcType,
4390623d748SDimitry Andric QualType DestType,
4400623d748SDimitry Andric SourceLocation Loc) {
441f785676fSDimitry Andric // Convert the input element to the element type of the complex.
442f785676fSDimitry Andric DestType = DestType->castAs<ComplexType>()->getElementType();
4430623d748SDimitry Andric Val = CGF.EmitScalarConversion(Val, SrcType, DestType, Loc);
444f785676fSDimitry Andric
445f785676fSDimitry Andric // Return (realval, 0).
446f785676fSDimitry Andric return ComplexPairTy(Val, llvm::Constant::getNullValue(Val->getType()));
447f785676fSDimitry Andric }
448f785676fSDimitry Andric
EmitCast(CastKind CK,Expr * Op,QualType DestTy)44939d628a0SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitCast(CastKind CK, Expr *Op,
450ffd1746dSEd Schouten QualType DestTy) {
4512754fe60SDimitry Andric switch (CK) {
45217a519f9SDimitry Andric case CK_Dependent: llvm_unreachable("dependent cast kind in IR gen!");
45317a519f9SDimitry Andric
454dff0c46cSDimitry Andric // Atomic to non-atomic casts may be more than a no-op for some platforms and
455dff0c46cSDimitry Andric // for some types.
456dff0c46cSDimitry Andric case CK_AtomicToNonAtomic:
457dff0c46cSDimitry Andric case CK_NonAtomicToAtomic:
4582754fe60SDimitry Andric case CK_NoOp:
4592754fe60SDimitry Andric case CK_LValueToRValue:
46017a519f9SDimitry Andric case CK_UserDefinedConversion:
4612754fe60SDimitry Andric return Visit(Op);
4622754fe60SDimitry Andric
46317a519f9SDimitry Andric case CK_LValueBitCast: {
464139f7f9bSDimitry Andric LValue origLV = CGF.EmitLValue(Op);
4650623d748SDimitry Andric Address V = origLV.getAddress();
4660623d748SDimitry Andric V = Builder.CreateElementBitCast(V, CGF.ConvertType(DestTy));
4670623d748SDimitry Andric return EmitLoadOfLValue(CGF.MakeAddrLValue(V, DestTy), Op->getExprLoc());
468ffd1746dSEd Schouten }
469ffd1746dSEd Schouten
47017a519f9SDimitry Andric case CK_BitCast:
47117a519f9SDimitry Andric case CK_BaseToDerived:
47217a519f9SDimitry Andric case CK_DerivedToBase:
47317a519f9SDimitry Andric case CK_UncheckedDerivedToBase:
47417a519f9SDimitry Andric case CK_Dynamic:
47517a519f9SDimitry Andric case CK_ToUnion:
47617a519f9SDimitry Andric case CK_ArrayToPointerDecay:
47717a519f9SDimitry Andric case CK_FunctionToPointerDecay:
47817a519f9SDimitry Andric case CK_NullToPointer:
47917a519f9SDimitry Andric case CK_NullToMemberPointer:
48017a519f9SDimitry Andric case CK_BaseToDerivedMemberPointer:
48117a519f9SDimitry Andric case CK_DerivedToBaseMemberPointer:
48217a519f9SDimitry Andric case CK_MemberPointerToBoolean:
483dff0c46cSDimitry Andric case CK_ReinterpretMemberPointer:
48417a519f9SDimitry Andric case CK_ConstructorConversion:
48517a519f9SDimitry Andric case CK_IntegralToPointer:
48617a519f9SDimitry Andric case CK_PointerToIntegral:
48717a519f9SDimitry Andric case CK_PointerToBoolean:
48817a519f9SDimitry Andric case CK_ToVoid:
48917a519f9SDimitry Andric case CK_VectorSplat:
49017a519f9SDimitry Andric case CK_IntegralCast:
491444ed5c5SDimitry Andric case CK_BooleanToSignedIntegral:
49217a519f9SDimitry Andric case CK_IntegralToBoolean:
49317a519f9SDimitry Andric case CK_IntegralToFloating:
49417a519f9SDimitry Andric case CK_FloatingToIntegral:
49517a519f9SDimitry Andric case CK_FloatingToBoolean:
49617a519f9SDimitry Andric case CK_FloatingCast:
4976122f3e6SDimitry Andric case CK_CPointerToObjCPointerCast:
4986122f3e6SDimitry Andric case CK_BlockPointerToObjCPointerCast:
49917a519f9SDimitry Andric case CK_AnyPointerToBlockPointerCast:
50017a519f9SDimitry Andric case CK_ObjCObjectLValueCast:
50117a519f9SDimitry Andric case CK_FloatingComplexToReal:
50217a519f9SDimitry Andric case CK_FloatingComplexToBoolean:
50317a519f9SDimitry Andric case CK_IntegralComplexToReal:
50417a519f9SDimitry Andric case CK_IntegralComplexToBoolean:
5056122f3e6SDimitry Andric case CK_ARCProduceObject:
5066122f3e6SDimitry Andric case CK_ARCConsumeObject:
5076122f3e6SDimitry Andric case CK_ARCReclaimReturnedObject:
5086122f3e6SDimitry Andric case CK_ARCExtendBlockObject:
509dff0c46cSDimitry Andric case CK_CopyAndAutoreleaseBlockObject:
5103861d79fSDimitry Andric case CK_BuiltinFnToFnPtr:
511*b5893f02SDimitry Andric case CK_ZeroToOCLOpaqueType:
51259d1ed5bSDimitry Andric case CK_AddressSpaceConversion:
51344290647SDimitry Andric case CK_IntToOCLSampler:
514*b5893f02SDimitry Andric case CK_FixedPointCast:
515*b5893f02SDimitry Andric case CK_FixedPointToBoolean:
51617a519f9SDimitry Andric llvm_unreachable("invalid cast kind for complex value");
51717a519f9SDimitry Andric
51817a519f9SDimitry Andric case CK_FloatingRealToComplex:
519f785676fSDimitry Andric case CK_IntegralRealToComplex:
5200623d748SDimitry Andric return EmitScalarToComplexCast(CGF.EmitScalarExpr(Op), Op->getType(),
5210623d748SDimitry Andric DestTy, Op->getExprLoc());
522f22ef01cSRoman Divacky
52317a519f9SDimitry Andric case CK_FloatingComplexCast:
52417a519f9SDimitry Andric case CK_FloatingComplexToIntegralComplex:
52517a519f9SDimitry Andric case CK_IntegralComplexCast:
52617a519f9SDimitry Andric case CK_IntegralComplexToFloatingComplex:
5270623d748SDimitry Andric return EmitComplexToComplexCast(Visit(Op), Op->getType(), DestTy,
5280623d748SDimitry Andric Op->getExprLoc());
52917a519f9SDimitry Andric }
53017a519f9SDimitry Andric
53117a519f9SDimitry Andric llvm_unreachable("unknown cast resulting in complex value");
53217a519f9SDimitry Andric }
53317a519f9SDimitry Andric
VisitUnaryMinus(const UnaryOperator * E)534f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryMinus(const UnaryOperator *E) {
535f22ef01cSRoman Divacky TestAndClearIgnoreReal();
536f22ef01cSRoman Divacky TestAndClearIgnoreImag();
537f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr());
538f22ef01cSRoman Divacky
539f22ef01cSRoman Divacky llvm::Value *ResR, *ResI;
540f22ef01cSRoman Divacky if (Op.first->getType()->isFloatingPointTy()) {
541f22ef01cSRoman Divacky ResR = Builder.CreateFNeg(Op.first, "neg.r");
542f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "neg.i");
543f22ef01cSRoman Divacky } else {
544f22ef01cSRoman Divacky ResR = Builder.CreateNeg(Op.first, "neg.r");
545f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "neg.i");
546f22ef01cSRoman Divacky }
547f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI);
548f22ef01cSRoman Divacky }
549f22ef01cSRoman Divacky
VisitUnaryNot(const UnaryOperator * E)550f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryNot(const UnaryOperator *E) {
551f22ef01cSRoman Divacky TestAndClearIgnoreReal();
552f22ef01cSRoman Divacky TestAndClearIgnoreImag();
553f22ef01cSRoman Divacky // ~(a+ib) = a + i*-b
554f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr());
555f22ef01cSRoman Divacky llvm::Value *ResI;
556f22ef01cSRoman Divacky if (Op.second->getType()->isFloatingPointTy())
557f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "conj.i");
558f22ef01cSRoman Divacky else
559f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "conj.i");
560f22ef01cSRoman Divacky
561f22ef01cSRoman Divacky return ComplexPairTy(Op.first, ResI);
562f22ef01cSRoman Divacky }
563f22ef01cSRoman Divacky
EmitBinAdd(const BinOpInfo & Op)564f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinAdd(const BinOpInfo &Op) {
565f22ef01cSRoman Divacky llvm::Value *ResR, *ResI;
566f22ef01cSRoman Divacky
567f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) {
568f22ef01cSRoman Divacky ResR = Builder.CreateFAdd(Op.LHS.first, Op.RHS.first, "add.r");
56939d628a0SDimitry Andric if (Op.LHS.second && Op.RHS.second)
570f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(Op.LHS.second, Op.RHS.second, "add.i");
57139d628a0SDimitry Andric else
57239d628a0SDimitry Andric ResI = Op.LHS.second ? Op.LHS.second : Op.RHS.second;
57339d628a0SDimitry Andric assert(ResI && "Only one operand may be real!");
574f22ef01cSRoman Divacky } else {
575f22ef01cSRoman Divacky ResR = Builder.CreateAdd(Op.LHS.first, Op.RHS.first, "add.r");
57639d628a0SDimitry Andric assert(Op.LHS.second && Op.RHS.second &&
57739d628a0SDimitry Andric "Both operands of integer complex operators must be complex!");
578f22ef01cSRoman Divacky ResI = Builder.CreateAdd(Op.LHS.second, Op.RHS.second, "add.i");
579f22ef01cSRoman Divacky }
580f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI);
581f22ef01cSRoman Divacky }
582f22ef01cSRoman Divacky
EmitBinSub(const BinOpInfo & Op)583f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinSub(const BinOpInfo &Op) {
584f22ef01cSRoman Divacky llvm::Value *ResR, *ResI;
585f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) {
586f22ef01cSRoman Divacky ResR = Builder.CreateFSub(Op.LHS.first, Op.RHS.first, "sub.r");
58739d628a0SDimitry Andric if (Op.LHS.second && Op.RHS.second)
588f22ef01cSRoman Divacky ResI = Builder.CreateFSub(Op.LHS.second, Op.RHS.second, "sub.i");
58939d628a0SDimitry Andric else
59039d628a0SDimitry Andric ResI = Op.LHS.second ? Op.LHS.second
59139d628a0SDimitry Andric : Builder.CreateFNeg(Op.RHS.second, "sub.i");
59239d628a0SDimitry Andric assert(ResI && "Only one operand may be real!");
593f22ef01cSRoman Divacky } else {
594f22ef01cSRoman Divacky ResR = Builder.CreateSub(Op.LHS.first, Op.RHS.first, "sub.r");
59539d628a0SDimitry Andric assert(Op.LHS.second && Op.RHS.second &&
59639d628a0SDimitry Andric "Both operands of integer complex operators must be complex!");
597f22ef01cSRoman Divacky ResI = Builder.CreateSub(Op.LHS.second, Op.RHS.second, "sub.i");
598f22ef01cSRoman Divacky }
599f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI);
600f22ef01cSRoman Divacky }
601f22ef01cSRoman Divacky
6024ba319b5SDimitry Andric /// Emit a libcall for a binary operation on complex types.
EmitComplexBinOpLibCall(StringRef LibCallName,const BinOpInfo & Op)60339d628a0SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitComplexBinOpLibCall(StringRef LibCallName,
60439d628a0SDimitry Andric const BinOpInfo &Op) {
60539d628a0SDimitry Andric CallArgList Args;
60639d628a0SDimitry Andric Args.add(RValue::get(Op.LHS.first),
60739d628a0SDimitry Andric Op.Ty->castAs<ComplexType>()->getElementType());
60839d628a0SDimitry Andric Args.add(RValue::get(Op.LHS.second),
60939d628a0SDimitry Andric Op.Ty->castAs<ComplexType>()->getElementType());
61039d628a0SDimitry Andric Args.add(RValue::get(Op.RHS.first),
61139d628a0SDimitry Andric Op.Ty->castAs<ComplexType>()->getElementType());
61239d628a0SDimitry Andric Args.add(RValue::get(Op.RHS.second),
61339d628a0SDimitry Andric Op.Ty->castAs<ComplexType>()->getElementType());
614f22ef01cSRoman Divacky
61539d628a0SDimitry Andric // We *must* use the full CG function call building logic here because the
61639d628a0SDimitry Andric // complex type has special ABI handling. We also should not forget about
61739d628a0SDimitry Andric // special calling convention which may be used for compiler builtins.
6180623d748SDimitry Andric
6190623d748SDimitry Andric // We create a function qualified type to state that this call does not have
6200623d748SDimitry Andric // any exceptions.
6210623d748SDimitry Andric FunctionProtoType::ExtProtoInfo EPI;
6220623d748SDimitry Andric EPI = EPI.withExceptionSpec(
6230623d748SDimitry Andric FunctionProtoType::ExceptionSpecInfo(EST_BasicNoexcept));
6240623d748SDimitry Andric SmallVector<QualType, 4> ArgsQTys(
6250623d748SDimitry Andric 4, Op.Ty->castAs<ComplexType>()->getElementType());
6260623d748SDimitry Andric QualType FQTy = CGF.getContext().getFunctionType(Op.Ty, ArgsQTys, EPI);
6270623d748SDimitry Andric const CGFunctionInfo &FuncInfo = CGF.CGM.getTypes().arrangeFreeFunctionCall(
6280623d748SDimitry Andric Args, cast<FunctionType>(FQTy.getTypePtr()), false);
6290623d748SDimitry Andric
63039d628a0SDimitry Andric llvm::FunctionType *FTy = CGF.CGM.getTypes().GetFunctionType(FuncInfo);
63139d628a0SDimitry Andric llvm::Constant *Func = CGF.CGM.CreateBuiltinFunction(FTy, LibCallName);
63244290647SDimitry Andric CGCallee Callee = CGCallee::forDirect(Func, FQTy->getAs<FunctionProtoType>());
63339d628a0SDimitry Andric
63444290647SDimitry Andric llvm::Instruction *Call;
63544290647SDimitry Andric RValue Res = CGF.EmitCall(FuncInfo, Callee, ReturnValueSlot(), Args, &Call);
6364ba319b5SDimitry Andric cast<llvm::CallInst>(Call)->setCallingConv(CGF.CGM.getRuntimeCC());
63739d628a0SDimitry Andric return Res.getComplexVal();
63839d628a0SDimitry Andric }
63939d628a0SDimitry Andric
6404ba319b5SDimitry Andric /// Lookup the libcall name for a given floating point type complex
64139d628a0SDimitry Andric /// multiply.
getComplexMultiplyLibCallName(llvm::Type * Ty)64239d628a0SDimitry Andric static StringRef getComplexMultiplyLibCallName(llvm::Type *Ty) {
64339d628a0SDimitry Andric switch (Ty->getTypeID()) {
64439d628a0SDimitry Andric default:
64539d628a0SDimitry Andric llvm_unreachable("Unsupported floating point type!");
64639d628a0SDimitry Andric case llvm::Type::HalfTyID:
64739d628a0SDimitry Andric return "__mulhc3";
64839d628a0SDimitry Andric case llvm::Type::FloatTyID:
64939d628a0SDimitry Andric return "__mulsc3";
65039d628a0SDimitry Andric case llvm::Type::DoubleTyID:
65139d628a0SDimitry Andric return "__muldc3";
65239d628a0SDimitry Andric case llvm::Type::PPC_FP128TyID:
65339d628a0SDimitry Andric return "__multc3";
65439d628a0SDimitry Andric case llvm::Type::X86_FP80TyID:
65539d628a0SDimitry Andric return "__mulxc3";
65639d628a0SDimitry Andric case llvm::Type::FP128TyID:
65739d628a0SDimitry Andric return "__multc3";
65839d628a0SDimitry Andric }
65939d628a0SDimitry Andric }
66039d628a0SDimitry Andric
66139d628a0SDimitry Andric // See C11 Annex G.5.1 for the semantics of multiplicative operators on complex
66239d628a0SDimitry Andric // typed values.
EmitBinMul(const BinOpInfo & Op)663f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinMul(const BinOpInfo &Op) {
664f22ef01cSRoman Divacky using llvm::Value;
665f22ef01cSRoman Divacky Value *ResR, *ResI;
66639d628a0SDimitry Andric llvm::MDBuilder MDHelper(CGF.getLLVMContext());
667f22ef01cSRoman Divacky
668f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) {
66939d628a0SDimitry Andric // The general formulation is:
67039d628a0SDimitry Andric // (a + ib) * (c + id) = (a * c - b * d) + i(a * d + b * c)
67139d628a0SDimitry Andric //
67239d628a0SDimitry Andric // But we can fold away components which would be zero due to a real
67339d628a0SDimitry Andric // operand according to C11 Annex G.5.1p2.
67439d628a0SDimitry Andric // FIXME: C11 also provides for imaginary types which would allow folding
67539d628a0SDimitry Andric // still more of this within the type system.
676f22ef01cSRoman Divacky
67739d628a0SDimitry Andric if (Op.LHS.second && Op.RHS.second) {
67839d628a0SDimitry Andric // If both operands are complex, emit the core math directly, and then
67939d628a0SDimitry Andric // test for NaNs. If we find NaNs in the result, we delegate to a libcall
68039d628a0SDimitry Andric // to carefully re-compute the correct infinity representation if
68139d628a0SDimitry Andric // possible. The expectation is that the presence of NaNs here is
68239d628a0SDimitry Andric // *extremely* rare, and so the cost of the libcall is almost irrelevant.
68339d628a0SDimitry Andric // This is good, because the libcall re-computes the core multiplication
68439d628a0SDimitry Andric // exactly the same as we do here and re-tests for NaNs in order to be
68539d628a0SDimitry Andric // a generic complex*complex libcall.
68639d628a0SDimitry Andric
68739d628a0SDimitry Andric // First compute the four products.
68839d628a0SDimitry Andric Value *AC = Builder.CreateFMul(Op.LHS.first, Op.RHS.first, "mul_ac");
68939d628a0SDimitry Andric Value *BD = Builder.CreateFMul(Op.LHS.second, Op.RHS.second, "mul_bd");
69039d628a0SDimitry Andric Value *AD = Builder.CreateFMul(Op.LHS.first, Op.RHS.second, "mul_ad");
69139d628a0SDimitry Andric Value *BC = Builder.CreateFMul(Op.LHS.second, Op.RHS.first, "mul_bc");
69239d628a0SDimitry Andric
69339d628a0SDimitry Andric // The real part is the difference of the first two, the imaginary part is
69439d628a0SDimitry Andric // the sum of the second.
69539d628a0SDimitry Andric ResR = Builder.CreateFSub(AC, BD, "mul_r");
69639d628a0SDimitry Andric ResI = Builder.CreateFAdd(AD, BC, "mul_i");
69739d628a0SDimitry Andric
69839d628a0SDimitry Andric // Emit the test for the real part becoming NaN and create a branch to
69939d628a0SDimitry Andric // handle it. We test for NaN by comparing the number to itself.
70039d628a0SDimitry Andric Value *IsRNaN = Builder.CreateFCmpUNO(ResR, ResR, "isnan_cmp");
70139d628a0SDimitry Andric llvm::BasicBlock *ContBB = CGF.createBasicBlock("complex_mul_cont");
70239d628a0SDimitry Andric llvm::BasicBlock *INaNBB = CGF.createBasicBlock("complex_mul_imag_nan");
70339d628a0SDimitry Andric llvm::Instruction *Branch = Builder.CreateCondBr(IsRNaN, INaNBB, ContBB);
70439d628a0SDimitry Andric llvm::BasicBlock *OrigBB = Branch->getParent();
70539d628a0SDimitry Andric
70639d628a0SDimitry Andric // Give hint that we very much don't expect to see NaNs.
70739d628a0SDimitry Andric // Value chosen to match UR_NONTAKEN_WEIGHT, see BranchProbabilityInfo.cpp
70839d628a0SDimitry Andric llvm::MDNode *BrWeight = MDHelper.createBranchWeights(1, (1U << 20) - 1);
70939d628a0SDimitry Andric Branch->setMetadata(llvm::LLVMContext::MD_prof, BrWeight);
71039d628a0SDimitry Andric
71139d628a0SDimitry Andric // Now test the imaginary part and create its branch.
71239d628a0SDimitry Andric CGF.EmitBlock(INaNBB);
71339d628a0SDimitry Andric Value *IsINaN = Builder.CreateFCmpUNO(ResI, ResI, "isnan_cmp");
71439d628a0SDimitry Andric llvm::BasicBlock *LibCallBB = CGF.createBasicBlock("complex_mul_libcall");
71539d628a0SDimitry Andric Branch = Builder.CreateCondBr(IsINaN, LibCallBB, ContBB);
71639d628a0SDimitry Andric Branch->setMetadata(llvm::LLVMContext::MD_prof, BrWeight);
71739d628a0SDimitry Andric
71839d628a0SDimitry Andric // Now emit the libcall on this slowest of the slow paths.
71939d628a0SDimitry Andric CGF.EmitBlock(LibCallBB);
72039d628a0SDimitry Andric Value *LibCallR, *LibCallI;
72139d628a0SDimitry Andric std::tie(LibCallR, LibCallI) = EmitComplexBinOpLibCall(
72239d628a0SDimitry Andric getComplexMultiplyLibCallName(Op.LHS.first->getType()), Op);
72339d628a0SDimitry Andric Builder.CreateBr(ContBB);
72439d628a0SDimitry Andric
72539d628a0SDimitry Andric // Finally continue execution by phi-ing together the different
72639d628a0SDimitry Andric // computation paths.
72739d628a0SDimitry Andric CGF.EmitBlock(ContBB);
72839d628a0SDimitry Andric llvm::PHINode *RealPHI = Builder.CreatePHI(ResR->getType(), 3, "real_mul_phi");
72939d628a0SDimitry Andric RealPHI->addIncoming(ResR, OrigBB);
73039d628a0SDimitry Andric RealPHI->addIncoming(ResR, INaNBB);
73139d628a0SDimitry Andric RealPHI->addIncoming(LibCallR, LibCallBB);
73239d628a0SDimitry Andric llvm::PHINode *ImagPHI = Builder.CreatePHI(ResI->getType(), 3, "imag_mul_phi");
73339d628a0SDimitry Andric ImagPHI->addIncoming(ResI, OrigBB);
73439d628a0SDimitry Andric ImagPHI->addIncoming(ResI, INaNBB);
73539d628a0SDimitry Andric ImagPHI->addIncoming(LibCallI, LibCallBB);
73639d628a0SDimitry Andric return ComplexPairTy(RealPHI, ImagPHI);
73739d628a0SDimitry Andric }
73839d628a0SDimitry Andric assert((Op.LHS.second || Op.RHS.second) &&
73939d628a0SDimitry Andric "At least one operand must be complex!");
74039d628a0SDimitry Andric
74139d628a0SDimitry Andric // If either of the operands is a real rather than a complex, the
74239d628a0SDimitry Andric // imaginary component is ignored when computing the real component of the
74339d628a0SDimitry Andric // result.
74439d628a0SDimitry Andric ResR = Builder.CreateFMul(Op.LHS.first, Op.RHS.first, "mul.rl");
74539d628a0SDimitry Andric
74639d628a0SDimitry Andric ResI = Op.LHS.second
74739d628a0SDimitry Andric ? Builder.CreateFMul(Op.LHS.second, Op.RHS.first, "mul.il")
74839d628a0SDimitry Andric : Builder.CreateFMul(Op.LHS.first, Op.RHS.second, "mul.ir");
749f22ef01cSRoman Divacky } else {
75039d628a0SDimitry Andric assert(Op.LHS.second && Op.RHS.second &&
75139d628a0SDimitry Andric "Both operands of integer complex operators must be complex!");
752f22ef01cSRoman Divacky Value *ResRl = Builder.CreateMul(Op.LHS.first, Op.RHS.first, "mul.rl");
753f22ef01cSRoman Divacky Value *ResRr = Builder.CreateMul(Op.LHS.second, Op.RHS.second, "mul.rr");
754f22ef01cSRoman Divacky ResR = Builder.CreateSub(ResRl, ResRr, "mul.r");
755f22ef01cSRoman Divacky
756f22ef01cSRoman Divacky Value *ResIl = Builder.CreateMul(Op.LHS.second, Op.RHS.first, "mul.il");
757f22ef01cSRoman Divacky Value *ResIr = Builder.CreateMul(Op.LHS.first, Op.RHS.second, "mul.ir");
758f22ef01cSRoman Divacky ResI = Builder.CreateAdd(ResIl, ResIr, "mul.i");
759f22ef01cSRoman Divacky }
760f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI);
761f22ef01cSRoman Divacky }
762f22ef01cSRoman Divacky
76339d628a0SDimitry Andric // See C11 Annex G.5.1 for the semantics of multiplicative operators on complex
76439d628a0SDimitry Andric // typed values.
EmitBinDiv(const BinOpInfo & Op)765f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinDiv(const BinOpInfo &Op) {
766f22ef01cSRoman Divacky llvm::Value *LHSr = Op.LHS.first, *LHSi = Op.LHS.second;
767f22ef01cSRoman Divacky llvm::Value *RHSr = Op.RHS.first, *RHSi = Op.RHS.second;
768f22ef01cSRoman Divacky
769f22ef01cSRoman Divacky llvm::Value *DSTr, *DSTi;
77039d628a0SDimitry Andric if (LHSr->getType()->isFloatingPointTy()) {
77113ddaa84SDimitry Andric // If we have a complex operand on the RHS and FastMath is not allowed, we
77213ddaa84SDimitry Andric // delegate to a libcall to handle all of the complexities and minimize
77313ddaa84SDimitry Andric // underflow/overflow cases. When FastMath is allowed we construct the
77413ddaa84SDimitry Andric // divide inline using the same algorithm as for integer operands.
77539d628a0SDimitry Andric //
77639d628a0SDimitry Andric // FIXME: We would be able to avoid the libcall in many places if we
77739d628a0SDimitry Andric // supported imaginary types in addition to complex types.
77813ddaa84SDimitry Andric if (RHSi && !CGF.getLangOpts().FastMath) {
77939d628a0SDimitry Andric BinOpInfo LibCallOp = Op;
78039d628a0SDimitry Andric // If LHS was a real, supply a null imaginary part.
78139d628a0SDimitry Andric if (!LHSi)
78239d628a0SDimitry Andric LibCallOp.LHS.second = llvm::Constant::getNullValue(LHSr->getType());
783f22ef01cSRoman Divacky
78439d628a0SDimitry Andric switch (LHSr->getType()->getTypeID()) {
78539d628a0SDimitry Andric default:
78639d628a0SDimitry Andric llvm_unreachable("Unsupported floating point type!");
78739d628a0SDimitry Andric case llvm::Type::HalfTyID:
78839d628a0SDimitry Andric return EmitComplexBinOpLibCall("__divhc3", LibCallOp);
78939d628a0SDimitry Andric case llvm::Type::FloatTyID:
79039d628a0SDimitry Andric return EmitComplexBinOpLibCall("__divsc3", LibCallOp);
79139d628a0SDimitry Andric case llvm::Type::DoubleTyID:
79239d628a0SDimitry Andric return EmitComplexBinOpLibCall("__divdc3", LibCallOp);
79339d628a0SDimitry Andric case llvm::Type::PPC_FP128TyID:
79439d628a0SDimitry Andric return EmitComplexBinOpLibCall("__divtc3", LibCallOp);
79539d628a0SDimitry Andric case llvm::Type::X86_FP80TyID:
79639d628a0SDimitry Andric return EmitComplexBinOpLibCall("__divxc3", LibCallOp);
79739d628a0SDimitry Andric case llvm::Type::FP128TyID:
79839d628a0SDimitry Andric return EmitComplexBinOpLibCall("__divtc3", LibCallOp);
79939d628a0SDimitry Andric }
80013ddaa84SDimitry Andric } else if (RHSi) {
80113ddaa84SDimitry Andric if (!LHSi)
80213ddaa84SDimitry Andric LHSi = llvm::Constant::getNullValue(RHSi->getType());
80313ddaa84SDimitry Andric
80413ddaa84SDimitry Andric // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd))
80513ddaa84SDimitry Andric llvm::Value *AC = Builder.CreateFMul(LHSr, RHSr); // a*c
80613ddaa84SDimitry Andric llvm::Value *BD = Builder.CreateFMul(LHSi, RHSi); // b*d
80713ddaa84SDimitry Andric llvm::Value *ACpBD = Builder.CreateFAdd(AC, BD); // ac+bd
80813ddaa84SDimitry Andric
80913ddaa84SDimitry Andric llvm::Value *CC = Builder.CreateFMul(RHSr, RHSr); // c*c
81013ddaa84SDimitry Andric llvm::Value *DD = Builder.CreateFMul(RHSi, RHSi); // d*d
81113ddaa84SDimitry Andric llvm::Value *CCpDD = Builder.CreateFAdd(CC, DD); // cc+dd
81213ddaa84SDimitry Andric
81313ddaa84SDimitry Andric llvm::Value *BC = Builder.CreateFMul(LHSi, RHSr); // b*c
81413ddaa84SDimitry Andric llvm::Value *AD = Builder.CreateFMul(LHSr, RHSi); // a*d
81513ddaa84SDimitry Andric llvm::Value *BCmAD = Builder.CreateFSub(BC, AD); // bc-ad
81613ddaa84SDimitry Andric
81713ddaa84SDimitry Andric DSTr = Builder.CreateFDiv(ACpBD, CCpDD);
81813ddaa84SDimitry Andric DSTi = Builder.CreateFDiv(BCmAD, CCpDD);
81913ddaa84SDimitry Andric } else {
82039d628a0SDimitry Andric assert(LHSi && "Can have at most one non-complex operand!");
821f22ef01cSRoman Divacky
82239d628a0SDimitry Andric DSTr = Builder.CreateFDiv(LHSr, RHSr);
82339d628a0SDimitry Andric DSTi = Builder.CreateFDiv(LHSi, RHSr);
82413ddaa84SDimitry Andric }
825f22ef01cSRoman Divacky } else {
82639d628a0SDimitry Andric assert(Op.LHS.second && Op.RHS.second &&
82739d628a0SDimitry Andric "Both operands of integer complex operators must be complex!");
828f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd))
8296122f3e6SDimitry Andric llvm::Value *Tmp1 = Builder.CreateMul(LHSr, RHSr); // a*c
8306122f3e6SDimitry Andric llvm::Value *Tmp2 = Builder.CreateMul(LHSi, RHSi); // b*d
8316122f3e6SDimitry Andric llvm::Value *Tmp3 = Builder.CreateAdd(Tmp1, Tmp2); // ac+bd
832f22ef01cSRoman Divacky
8336122f3e6SDimitry Andric llvm::Value *Tmp4 = Builder.CreateMul(RHSr, RHSr); // c*c
8346122f3e6SDimitry Andric llvm::Value *Tmp5 = Builder.CreateMul(RHSi, RHSi); // d*d
8356122f3e6SDimitry Andric llvm::Value *Tmp6 = Builder.CreateAdd(Tmp4, Tmp5); // cc+dd
836f22ef01cSRoman Divacky
8376122f3e6SDimitry Andric llvm::Value *Tmp7 = Builder.CreateMul(LHSi, RHSr); // b*c
8386122f3e6SDimitry Andric llvm::Value *Tmp8 = Builder.CreateMul(LHSr, RHSi); // a*d
8396122f3e6SDimitry Andric llvm::Value *Tmp9 = Builder.CreateSub(Tmp7, Tmp8); // bc-ad
840f22ef01cSRoman Divacky
841139f7f9bSDimitry Andric if (Op.Ty->castAs<ComplexType>()->getElementType()->isUnsignedIntegerType()) {
8426122f3e6SDimitry Andric DSTr = Builder.CreateUDiv(Tmp3, Tmp6);
8436122f3e6SDimitry Andric DSTi = Builder.CreateUDiv(Tmp9, Tmp6);
844f22ef01cSRoman Divacky } else {
8456122f3e6SDimitry Andric DSTr = Builder.CreateSDiv(Tmp3, Tmp6);
8466122f3e6SDimitry Andric DSTi = Builder.CreateSDiv(Tmp9, Tmp6);
847f22ef01cSRoman Divacky }
848f22ef01cSRoman Divacky }
849f22ef01cSRoman Divacky
850f22ef01cSRoman Divacky return ComplexPairTy(DSTr, DSTi);
851f22ef01cSRoman Divacky }
852f22ef01cSRoman Divacky
853f22ef01cSRoman Divacky ComplexExprEmitter::BinOpInfo
EmitBinOps(const BinaryOperator * E)854f22ef01cSRoman Divacky ComplexExprEmitter::EmitBinOps(const BinaryOperator *E) {
855f22ef01cSRoman Divacky TestAndClearIgnoreReal();
856f22ef01cSRoman Divacky TestAndClearIgnoreImag();
857f22ef01cSRoman Divacky BinOpInfo Ops;
85839d628a0SDimitry Andric if (E->getLHS()->getType()->isRealFloatingType())
85939d628a0SDimitry Andric Ops.LHS = ComplexPairTy(CGF.EmitScalarExpr(E->getLHS()), nullptr);
86039d628a0SDimitry Andric else
861f22ef01cSRoman Divacky Ops.LHS = Visit(E->getLHS());
86239d628a0SDimitry Andric if (E->getRHS()->getType()->isRealFloatingType())
86339d628a0SDimitry Andric Ops.RHS = ComplexPairTy(CGF.EmitScalarExpr(E->getRHS()), nullptr);
86439d628a0SDimitry Andric else
865f22ef01cSRoman Divacky Ops.RHS = Visit(E->getRHS());
86639d628a0SDimitry Andric
867f22ef01cSRoman Divacky Ops.Ty = E->getType();
868f22ef01cSRoman Divacky return Ops;
869f22ef01cSRoman Divacky }
870f22ef01cSRoman Divacky
871f22ef01cSRoman Divacky
8722754fe60SDimitry Andric LValue ComplexExprEmitter::
EmitCompoundAssignLValue(const CompoundAssignOperator * E,ComplexPairTy (ComplexExprEmitter::* Func)(const BinOpInfo &),RValue & Val)8732754fe60SDimitry Andric EmitCompoundAssignLValue(const CompoundAssignOperator *E,
8742754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&),
875f785676fSDimitry Andric RValue &Val) {
876f22ef01cSRoman Divacky TestAndClearIgnoreReal();
877f22ef01cSRoman Divacky TestAndClearIgnoreImag();
8782754fe60SDimitry Andric QualType LHSTy = E->getLHS()->getType();
87933956c43SDimitry Andric if (const AtomicType *AT = LHSTy->getAs<AtomicType>())
88033956c43SDimitry Andric LHSTy = AT->getValueType();
881f22ef01cSRoman Divacky
882f22ef01cSRoman Divacky BinOpInfo OpInfo;
883f22ef01cSRoman Divacky
884f22ef01cSRoman Divacky // Load the RHS and LHS operands.
885f22ef01cSRoman Divacky // __block variables need to have the rhs evaluated first, plus this should
8862754fe60SDimitry Andric // improve codegen a little.
887f22ef01cSRoman Divacky OpInfo.Ty = E->getComputationResultType();
88839d628a0SDimitry Andric QualType ComplexElementTy = cast<ComplexType>(OpInfo.Ty)->getElementType();
8892754fe60SDimitry Andric
8902754fe60SDimitry Andric // The RHS should have been converted to the computation type.
89139d628a0SDimitry Andric if (E->getRHS()->getType()->isRealFloatingType()) {
89239d628a0SDimitry Andric assert(
89339d628a0SDimitry Andric CGF.getContext()
89439d628a0SDimitry Andric .hasSameUnqualifiedType(ComplexElementTy, E->getRHS()->getType()));
89539d628a0SDimitry Andric OpInfo.RHS = ComplexPairTy(CGF.EmitScalarExpr(E->getRHS()), nullptr);
89639d628a0SDimitry Andric } else {
89739d628a0SDimitry Andric assert(CGF.getContext()
89839d628a0SDimitry Andric .hasSameUnqualifiedType(OpInfo.Ty, E->getRHS()->getType()));
8992754fe60SDimitry Andric OpInfo.RHS = Visit(E->getRHS());
90039d628a0SDimitry Andric }
901f22ef01cSRoman Divacky
902ffd1746dSEd Schouten LValue LHS = CGF.EmitLValue(E->getLHS());
9032754fe60SDimitry Andric
904f785676fSDimitry Andric // Load from the l-value and convert it.
9050623d748SDimitry Andric SourceLocation Loc = E->getExprLoc();
906f785676fSDimitry Andric if (LHSTy->isAnyComplexType()) {
9070623d748SDimitry Andric ComplexPairTy LHSVal = EmitLoadOfLValue(LHS, Loc);
9080623d748SDimitry Andric OpInfo.LHS = EmitComplexToComplexCast(LHSVal, LHSTy, OpInfo.Ty, Loc);
909f785676fSDimitry Andric } else {
9100623d748SDimitry Andric llvm::Value *LHSVal = CGF.EmitLoadOfScalar(LHS, Loc);
91139d628a0SDimitry Andric // For floating point real operands we can directly pass the scalar form
91239d628a0SDimitry Andric // to the binary operator emission and potentially get more efficient code.
91339d628a0SDimitry Andric if (LHSTy->isRealFloatingType()) {
91439d628a0SDimitry Andric if (!CGF.getContext().hasSameUnqualifiedType(ComplexElementTy, LHSTy))
9150623d748SDimitry Andric LHSVal = CGF.EmitScalarConversion(LHSVal, LHSTy, ComplexElementTy, Loc);
91639d628a0SDimitry Andric OpInfo.LHS = ComplexPairTy(LHSVal, nullptr);
91739d628a0SDimitry Andric } else {
9180623d748SDimitry Andric OpInfo.LHS = EmitScalarToComplexCast(LHSVal, LHSTy, OpInfo.Ty, Loc);
919f785676fSDimitry Andric }
92039d628a0SDimitry Andric }
921f22ef01cSRoman Divacky
922f22ef01cSRoman Divacky // Expand the binary operator.
923f22ef01cSRoman Divacky ComplexPairTy Result = (this->*Func)(OpInfo);
924f22ef01cSRoman Divacky
925f785676fSDimitry Andric // Truncate the result and store it into the LHS lvalue.
926f785676fSDimitry Andric if (LHSTy->isAnyComplexType()) {
9270623d748SDimitry Andric ComplexPairTy ResVal =
9280623d748SDimitry Andric EmitComplexToComplexCast(Result, OpInfo.Ty, LHSTy, Loc);
929f785676fSDimitry Andric EmitStoreOfComplex(ResVal, LHS, /*isInit*/ false);
930f785676fSDimitry Andric Val = RValue::getComplex(ResVal);
931f785676fSDimitry Andric } else {
932f785676fSDimitry Andric llvm::Value *ResVal =
9330623d748SDimitry Andric CGF.EmitComplexToScalarConversion(Result, OpInfo.Ty, LHSTy, Loc);
934f785676fSDimitry Andric CGF.EmitStoreOfScalar(ResVal, LHS, /*isInit*/ false);
935f785676fSDimitry Andric Val = RValue::get(ResVal);
936f785676fSDimitry Andric }
937ffd1746dSEd Schouten
9382754fe60SDimitry Andric return LHS;
939f22ef01cSRoman Divacky }
940f22ef01cSRoman Divacky
9412754fe60SDimitry Andric // Compound assignments.
9422754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter::
EmitCompoundAssign(const CompoundAssignOperator * E,ComplexPairTy (ComplexExprEmitter::* Func)(const BinOpInfo &))9432754fe60SDimitry Andric EmitCompoundAssign(const CompoundAssignOperator *E,
9442754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&)){
945f785676fSDimitry Andric RValue Val;
9462754fe60SDimitry Andric LValue LV = EmitCompoundAssignLValue(E, Func, Val);
9472754fe60SDimitry Andric
9482754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value.
9493861d79fSDimitry Andric if (!CGF.getLangOpts().CPlusPlus)
950f785676fSDimitry Andric return Val.getComplexVal();
9512754fe60SDimitry Andric
9522754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment.
9532754fe60SDimitry Andric if (!LV.isVolatileQualified())
954f785676fSDimitry Andric return Val.getComplexVal();
9552754fe60SDimitry Andric
956f785676fSDimitry Andric return EmitLoadOfLValue(LV, E->getExprLoc());
9572754fe60SDimitry Andric }
9582754fe60SDimitry Andric
EmitBinAssignLValue(const BinaryOperator * E,ComplexPairTy & Val)9592754fe60SDimitry Andric LValue ComplexExprEmitter::EmitBinAssignLValue(const BinaryOperator *E,
9602754fe60SDimitry Andric ComplexPairTy &Val) {
961ffd1746dSEd Schouten assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(),
962ffd1746dSEd Schouten E->getRHS()->getType()) &&
963f22ef01cSRoman Divacky "Invalid assignment");
9642754fe60SDimitry Andric TestAndClearIgnoreReal();
9652754fe60SDimitry Andric TestAndClearIgnoreImag();
9662754fe60SDimitry Andric
9672754fe60SDimitry Andric // Emit the RHS. __block variables need the RHS evaluated first.
9682754fe60SDimitry Andric Val = Visit(E->getRHS());
969f22ef01cSRoman Divacky
970f22ef01cSRoman Divacky // Compute the address to store into.
971f22ef01cSRoman Divacky LValue LHS = CGF.EmitLValue(E->getLHS());
972f22ef01cSRoman Divacky
973ffd1746dSEd Schouten // Store the result value into the LHS lvalue.
974139f7f9bSDimitry Andric EmitStoreOfComplex(Val, LHS, /*isInit*/ false);
975f22ef01cSRoman Divacky
9762754fe60SDimitry Andric return LHS;
9772754fe60SDimitry Andric }
978ffd1746dSEd Schouten
VisitBinAssign(const BinaryOperator * E)9792754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitBinAssign(const BinaryOperator *E) {
9802754fe60SDimitry Andric ComplexPairTy Val;
9812754fe60SDimitry Andric LValue LV = EmitBinAssignLValue(E, Val);
9822754fe60SDimitry Andric
9832754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value.
9843861d79fSDimitry Andric if (!CGF.getLangOpts().CPlusPlus)
9852754fe60SDimitry Andric return Val;
9862754fe60SDimitry Andric
9872754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment.
9882754fe60SDimitry Andric if (!LV.isVolatileQualified())
9892754fe60SDimitry Andric return Val;
9902754fe60SDimitry Andric
991f785676fSDimitry Andric return EmitLoadOfLValue(LV, E->getExprLoc());
992f22ef01cSRoman Divacky }
993f22ef01cSRoman Divacky
VisitBinComma(const BinaryOperator * E)994f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitBinComma(const BinaryOperator *E) {
9952754fe60SDimitry Andric CGF.EmitIgnoredExpr(E->getLHS());
996f22ef01cSRoman Divacky return Visit(E->getRHS());
997f22ef01cSRoman Divacky }
998f22ef01cSRoman Divacky
999f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::
VisitAbstractConditionalOperator(const AbstractConditionalOperator * E)10002754fe60SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) {
1001f22ef01cSRoman Divacky TestAndClearIgnoreReal();
1002f22ef01cSRoman Divacky TestAndClearIgnoreImag();
1003f22ef01cSRoman Divacky llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true");
1004f22ef01cSRoman Divacky llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false");
1005f22ef01cSRoman Divacky llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end");
1006f22ef01cSRoman Divacky
10072754fe60SDimitry Andric // Bind the common expression if necessary.
10082754fe60SDimitry Andric CodeGenFunction::OpaqueValueMapping binding(CGF, E);
10092754fe60SDimitry Andric
101033956c43SDimitry Andric
10112754fe60SDimitry Andric CodeGenFunction::ConditionalEvaluation eval(CGF);
101233956c43SDimitry Andric CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock,
101333956c43SDimitry Andric CGF.getProfileCount(E));
1014f22ef01cSRoman Divacky
10152754fe60SDimitry Andric eval.begin(CGF);
1016f22ef01cSRoman Divacky CGF.EmitBlock(LHSBlock);
101733956c43SDimitry Andric CGF.incrementProfileCounter(E);
10182754fe60SDimitry Andric ComplexPairTy LHS = Visit(E->getTrueExpr());
1019f22ef01cSRoman Divacky LHSBlock = Builder.GetInsertBlock();
1020f22ef01cSRoman Divacky CGF.EmitBranch(ContBlock);
10212754fe60SDimitry Andric eval.end(CGF);
1022f22ef01cSRoman Divacky
10232754fe60SDimitry Andric eval.begin(CGF);
1024f22ef01cSRoman Divacky CGF.EmitBlock(RHSBlock);
10252754fe60SDimitry Andric ComplexPairTy RHS = Visit(E->getFalseExpr());
1026f22ef01cSRoman Divacky RHSBlock = Builder.GetInsertBlock();
1027f22ef01cSRoman Divacky CGF.EmitBlock(ContBlock);
10282754fe60SDimitry Andric eval.end(CGF);
1029f22ef01cSRoman Divacky
1030f22ef01cSRoman Divacky // Create a PHI node for the real part.
10313b0f4066SDimitry Andric llvm::PHINode *RealPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.r");
1032f22ef01cSRoman Divacky RealPN->addIncoming(LHS.first, LHSBlock);
1033f22ef01cSRoman Divacky RealPN->addIncoming(RHS.first, RHSBlock);
1034f22ef01cSRoman Divacky
1035f22ef01cSRoman Divacky // Create a PHI node for the imaginary part.
10363b0f4066SDimitry Andric llvm::PHINode *ImagPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.i");
1037f22ef01cSRoman Divacky ImagPN->addIncoming(LHS.second, LHSBlock);
1038f22ef01cSRoman Divacky ImagPN->addIncoming(RHS.second, RHSBlock);
1039f22ef01cSRoman Divacky
1040f22ef01cSRoman Divacky return ComplexPairTy(RealPN, ImagPN);
1041f22ef01cSRoman Divacky }
1042f22ef01cSRoman Divacky
VisitChooseExpr(ChooseExpr * E)1043f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitChooseExpr(ChooseExpr *E) {
1044f785676fSDimitry Andric return Visit(E->getChosenSubExpr());
1045f22ef01cSRoman Divacky }
1046f22ef01cSRoman Divacky
VisitInitListExpr(InitListExpr * E)1047f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitInitListExpr(InitListExpr *E) {
1048f22ef01cSRoman Divacky bool Ignore = TestAndClearIgnoreReal();
1049f22ef01cSRoman Divacky (void)Ignore;
1050f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored");
1051f22ef01cSRoman Divacky Ignore = TestAndClearIgnoreImag();
1052f22ef01cSRoman Divacky (void)Ignore;
1053f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored");
10546122f3e6SDimitry Andric
10556122f3e6SDimitry Andric if (E->getNumInits() == 2) {
10566122f3e6SDimitry Andric llvm::Value *Real = CGF.EmitScalarExpr(E->getInit(0));
10576122f3e6SDimitry Andric llvm::Value *Imag = CGF.EmitScalarExpr(E->getInit(1));
10586122f3e6SDimitry Andric return ComplexPairTy(Real, Imag);
10596122f3e6SDimitry Andric } else if (E->getNumInits() == 1) {
1060f22ef01cSRoman Divacky return Visit(E->getInit(0));
10616122f3e6SDimitry Andric }
1062f22ef01cSRoman Divacky
10634ba319b5SDimitry Andric // Empty init list initializes to null
10646122f3e6SDimitry Andric assert(E->getNumInits() == 0 && "Unexpected number of inits");
1065139f7f9bSDimitry Andric QualType Ty = E->getType()->castAs<ComplexType>()->getElementType();
10666122f3e6SDimitry Andric llvm::Type* LTy = CGF.ConvertType(Ty);
1067f22ef01cSRoman Divacky llvm::Value* zeroConstant = llvm::Constant::getNullValue(LTy);
1068f22ef01cSRoman Divacky return ComplexPairTy(zeroConstant, zeroConstant);
1069f22ef01cSRoman Divacky }
1070f22ef01cSRoman Divacky
VisitVAArgExpr(VAArgExpr * E)1071f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitVAArgExpr(VAArgExpr *E) {
10720623d748SDimitry Andric Address ArgValue = Address::invalid();
10730623d748SDimitry Andric Address ArgPtr = CGF.EmitVAArg(E, ArgValue);
1074f22ef01cSRoman Divacky
10750623d748SDimitry Andric if (!ArgPtr.isValid()) {
1076f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex va_arg expression");
10776122f3e6SDimitry Andric llvm::Type *EltTy =
1078139f7f9bSDimitry Andric CGF.ConvertType(E->getType()->castAs<ComplexType>()->getElementType());
1079f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy);
1080f22ef01cSRoman Divacky return ComplexPairTy(U, U);
1081f22ef01cSRoman Divacky }
1082f22ef01cSRoman Divacky
10830623d748SDimitry Andric return EmitLoadOfLValue(CGF.MakeAddrLValue(ArgPtr, E->getType()),
1084f785676fSDimitry Andric E->getExprLoc());
1085f22ef01cSRoman Divacky }
1086f22ef01cSRoman Divacky
1087f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
1088f22ef01cSRoman Divacky // Entry Point into this File
1089f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
1090f22ef01cSRoman Divacky
1091f22ef01cSRoman Divacky /// EmitComplexExpr - Emit the computation of the specified expression of
1092f22ef01cSRoman Divacky /// complex type, ignoring the result.
EmitComplexExpr(const Expr * E,bool IgnoreReal,bool IgnoreImag)1093f22ef01cSRoman Divacky ComplexPairTy CodeGenFunction::EmitComplexExpr(const Expr *E, bool IgnoreReal,
10942754fe60SDimitry Andric bool IgnoreImag) {
1095139f7f9bSDimitry Andric assert(E && getComplexType(E->getType()) &&
1096f22ef01cSRoman Divacky "Invalid complex expression to emit");
1097f22ef01cSRoman Divacky
10982754fe60SDimitry Andric return ComplexExprEmitter(*this, IgnoreReal, IgnoreImag)
1099f22ef01cSRoman Divacky .Visit(const_cast<Expr *>(E));
1100f22ef01cSRoman Divacky }
1101f22ef01cSRoman Divacky
EmitComplexExprIntoLValue(const Expr * E,LValue dest,bool isInit)1102139f7f9bSDimitry Andric void CodeGenFunction::EmitComplexExprIntoLValue(const Expr *E, LValue dest,
1103139f7f9bSDimitry Andric bool isInit) {
1104139f7f9bSDimitry Andric assert(E && getComplexType(E->getType()) &&
1105f22ef01cSRoman Divacky "Invalid complex expression to emit");
1106f22ef01cSRoman Divacky ComplexExprEmitter Emitter(*this);
1107f22ef01cSRoman Divacky ComplexPairTy Val = Emitter.Visit(const_cast<Expr*>(E));
1108139f7f9bSDimitry Andric Emitter.EmitStoreOfComplex(Val, dest, isInit);
1109f22ef01cSRoman Divacky }
1110f22ef01cSRoman Divacky
1111139f7f9bSDimitry Andric /// EmitStoreOfComplex - Store a complex number into the specified l-value.
EmitStoreOfComplex(ComplexPairTy V,LValue dest,bool isInit)1112139f7f9bSDimitry Andric void CodeGenFunction::EmitStoreOfComplex(ComplexPairTy V, LValue dest,
1113139f7f9bSDimitry Andric bool isInit) {
1114139f7f9bSDimitry Andric ComplexExprEmitter(*this).EmitStoreOfComplex(V, dest, isInit);
1115f22ef01cSRoman Divacky }
1116f22ef01cSRoman Divacky
1117139f7f9bSDimitry Andric /// EmitLoadOfComplex - Load a complex number from the specified address.
EmitLoadOfComplex(LValue src,SourceLocation loc)1118f785676fSDimitry Andric ComplexPairTy CodeGenFunction::EmitLoadOfComplex(LValue src,
1119f785676fSDimitry Andric SourceLocation loc) {
1120f785676fSDimitry Andric return ComplexExprEmitter(*this).EmitLoadOfLValue(src, loc);
1121f22ef01cSRoman Divacky }
11222754fe60SDimitry Andric
EmitComplexAssignmentLValue(const BinaryOperator * E)11232754fe60SDimitry Andric LValue CodeGenFunction::EmitComplexAssignmentLValue(const BinaryOperator *E) {
11242754fe60SDimitry Andric assert(E->getOpcode() == BO_Assign);
11252754fe60SDimitry Andric ComplexPairTy Val; // ignored
11262754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitBinAssignLValue(E, Val);
11272754fe60SDimitry Andric }
11282754fe60SDimitry Andric
1129f785676fSDimitry Andric typedef ComplexPairTy (ComplexExprEmitter::*CompoundFunc)(
1130f785676fSDimitry Andric const ComplexExprEmitter::BinOpInfo &);
11312754fe60SDimitry Andric
getComplexOp(BinaryOperatorKind Op)1132f785676fSDimitry Andric static CompoundFunc getComplexOp(BinaryOperatorKind Op) {
1133f785676fSDimitry Andric switch (Op) {
1134f785676fSDimitry Andric case BO_MulAssign: return &ComplexExprEmitter::EmitBinMul;
1135f785676fSDimitry Andric case BO_DivAssign: return &ComplexExprEmitter::EmitBinDiv;
1136f785676fSDimitry Andric case BO_SubAssign: return &ComplexExprEmitter::EmitBinSub;
1137f785676fSDimitry Andric case BO_AddAssign: return &ComplexExprEmitter::EmitBinAdd;
11382754fe60SDimitry Andric default:
11392754fe60SDimitry Andric llvm_unreachable("unexpected complex compound assignment");
11402754fe60SDimitry Andric }
1141f785676fSDimitry Andric }
11422754fe60SDimitry Andric
1143f785676fSDimitry Andric LValue CodeGenFunction::
EmitComplexCompoundAssignmentLValue(const CompoundAssignOperator * E)1144f785676fSDimitry Andric EmitComplexCompoundAssignmentLValue(const CompoundAssignOperator *E) {
1145f785676fSDimitry Andric CompoundFunc Op = getComplexOp(E->getOpcode());
1146f785676fSDimitry Andric RValue Val;
11472754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val);
11482754fe60SDimitry Andric }
1149f785676fSDimitry Andric
1150f785676fSDimitry Andric LValue CodeGenFunction::
EmitScalarCompoundAssignWithComplex(const CompoundAssignOperator * E,llvm::Value * & Result)115133956c43SDimitry Andric EmitScalarCompoundAssignWithComplex(const CompoundAssignOperator *E,
1152f785676fSDimitry Andric llvm::Value *&Result) {
1153f785676fSDimitry Andric CompoundFunc Op = getComplexOp(E->getOpcode());
1154f785676fSDimitry Andric RValue Val;
1155f785676fSDimitry Andric LValue Ret = ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val);
1156f785676fSDimitry Andric Result = Val.getScalarVal();
1157f785676fSDimitry Andric return Ret;
1158f785676fSDimitry Andric }
1159