1*0b57cec5SDimitry Andric //===--- CGExprComplex.cpp - Emit LLVM Code for Complex Exprs -------------===// 2*0b57cec5SDimitry Andric // 3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*0b57cec5SDimitry Andric // 7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 8*0b57cec5SDimitry Andric // 9*0b57cec5SDimitry Andric // This contains code to emit Expr nodes with complex types as LLVM code. 10*0b57cec5SDimitry Andric // 11*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 12*0b57cec5SDimitry Andric 13480093f4SDimitry Andric #include "CGOpenMPRuntime.h" 14*0b57cec5SDimitry Andric #include "CodeGenFunction.h" 15*0b57cec5SDimitry Andric #include "CodeGenModule.h" 16*0b57cec5SDimitry Andric #include "clang/AST/StmtVisitor.h" 17*0b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h" 18*0b57cec5SDimitry Andric #include "llvm/IR/Constants.h" 19*0b57cec5SDimitry Andric #include "llvm/IR/Instructions.h" 20*0b57cec5SDimitry Andric #include "llvm/IR/MDBuilder.h" 21*0b57cec5SDimitry Andric #include "llvm/IR/Metadata.h" 22*0b57cec5SDimitry Andric #include <algorithm> 23*0b57cec5SDimitry Andric using namespace clang; 24*0b57cec5SDimitry Andric using namespace CodeGen; 25*0b57cec5SDimitry Andric 26*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 27*0b57cec5SDimitry Andric // Complex Expression Emitter 28*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 29*0b57cec5SDimitry Andric 30*0b57cec5SDimitry Andric typedef CodeGenFunction::ComplexPairTy ComplexPairTy; 31*0b57cec5SDimitry Andric 32*0b57cec5SDimitry Andric /// Return the complex type that we are meant to emit. 33*0b57cec5SDimitry Andric static const ComplexType *getComplexType(QualType type) { 34*0b57cec5SDimitry Andric type = type.getCanonicalType(); 35*0b57cec5SDimitry Andric if (const ComplexType *comp = dyn_cast<ComplexType>(type)) { 36*0b57cec5SDimitry Andric return comp; 37*0b57cec5SDimitry Andric } else { 38*0b57cec5SDimitry Andric return cast<ComplexType>(cast<AtomicType>(type)->getValueType()); 39*0b57cec5SDimitry Andric } 40*0b57cec5SDimitry Andric } 41*0b57cec5SDimitry Andric 42*0b57cec5SDimitry Andric namespace { 43*0b57cec5SDimitry Andric class ComplexExprEmitter 44*0b57cec5SDimitry Andric : public StmtVisitor<ComplexExprEmitter, ComplexPairTy> { 45*0b57cec5SDimitry Andric CodeGenFunction &CGF; 46*0b57cec5SDimitry Andric CGBuilderTy &Builder; 47*0b57cec5SDimitry Andric bool IgnoreReal; 48*0b57cec5SDimitry Andric bool IgnoreImag; 49*0b57cec5SDimitry Andric public: 50*0b57cec5SDimitry Andric ComplexExprEmitter(CodeGenFunction &cgf, bool ir=false, bool ii=false) 51*0b57cec5SDimitry Andric : CGF(cgf), Builder(CGF.Builder), IgnoreReal(ir), IgnoreImag(ii) { 52*0b57cec5SDimitry Andric } 53*0b57cec5SDimitry Andric 54*0b57cec5SDimitry Andric 55*0b57cec5SDimitry Andric //===--------------------------------------------------------------------===// 56*0b57cec5SDimitry Andric // Utilities 57*0b57cec5SDimitry Andric //===--------------------------------------------------------------------===// 58*0b57cec5SDimitry Andric 59*0b57cec5SDimitry Andric bool TestAndClearIgnoreReal() { 60*0b57cec5SDimitry Andric bool I = IgnoreReal; 61*0b57cec5SDimitry Andric IgnoreReal = false; 62*0b57cec5SDimitry Andric return I; 63*0b57cec5SDimitry Andric } 64*0b57cec5SDimitry Andric bool TestAndClearIgnoreImag() { 65*0b57cec5SDimitry Andric bool I = IgnoreImag; 66*0b57cec5SDimitry Andric IgnoreImag = false; 67*0b57cec5SDimitry Andric return I; 68*0b57cec5SDimitry Andric } 69*0b57cec5SDimitry Andric 70*0b57cec5SDimitry Andric /// EmitLoadOfLValue - Given an expression with complex type that represents a 71*0b57cec5SDimitry Andric /// value l-value, this method emits the address of the l-value, then loads 72*0b57cec5SDimitry Andric /// and returns the result. 73*0b57cec5SDimitry Andric ComplexPairTy EmitLoadOfLValue(const Expr *E) { 74*0b57cec5SDimitry Andric return EmitLoadOfLValue(CGF.EmitLValue(E), E->getExprLoc()); 75*0b57cec5SDimitry Andric } 76*0b57cec5SDimitry Andric 77*0b57cec5SDimitry Andric ComplexPairTy EmitLoadOfLValue(LValue LV, SourceLocation Loc); 78*0b57cec5SDimitry Andric 79*0b57cec5SDimitry Andric /// EmitStoreOfComplex - Store the specified real/imag parts into the 80*0b57cec5SDimitry Andric /// specified value pointer. 81*0b57cec5SDimitry Andric void EmitStoreOfComplex(ComplexPairTy Val, LValue LV, bool isInit); 82*0b57cec5SDimitry Andric 83*0b57cec5SDimitry Andric /// Emit a cast from complex value Val to DestType. 84*0b57cec5SDimitry Andric ComplexPairTy EmitComplexToComplexCast(ComplexPairTy Val, QualType SrcType, 85*0b57cec5SDimitry Andric QualType DestType, SourceLocation Loc); 86*0b57cec5SDimitry Andric /// Emit a cast from scalar value Val to DestType. 87*0b57cec5SDimitry Andric ComplexPairTy EmitScalarToComplexCast(llvm::Value *Val, QualType SrcType, 88*0b57cec5SDimitry Andric QualType DestType, SourceLocation Loc); 89*0b57cec5SDimitry Andric 90*0b57cec5SDimitry Andric //===--------------------------------------------------------------------===// 91*0b57cec5SDimitry Andric // Visitor Methods 92*0b57cec5SDimitry Andric //===--------------------------------------------------------------------===// 93*0b57cec5SDimitry Andric 94*0b57cec5SDimitry Andric ComplexPairTy Visit(Expr *E) { 95*0b57cec5SDimitry Andric ApplyDebugLocation DL(CGF, E); 96*0b57cec5SDimitry Andric return StmtVisitor<ComplexExprEmitter, ComplexPairTy>::Visit(E); 97*0b57cec5SDimitry Andric } 98*0b57cec5SDimitry Andric 99*0b57cec5SDimitry Andric ComplexPairTy VisitStmt(Stmt *S) { 100*0b57cec5SDimitry Andric S->dump(CGF.getContext().getSourceManager()); 101*0b57cec5SDimitry Andric llvm_unreachable("Stmt can't have complex result type!"); 102*0b57cec5SDimitry Andric } 103*0b57cec5SDimitry Andric ComplexPairTy VisitExpr(Expr *S); 104*0b57cec5SDimitry Andric ComplexPairTy VisitConstantExpr(ConstantExpr *E) { 105*0b57cec5SDimitry Andric return Visit(E->getSubExpr()); 106*0b57cec5SDimitry Andric } 107*0b57cec5SDimitry Andric ComplexPairTy VisitParenExpr(ParenExpr *PE) { return Visit(PE->getSubExpr());} 108*0b57cec5SDimitry Andric ComplexPairTy VisitGenericSelectionExpr(GenericSelectionExpr *GE) { 109*0b57cec5SDimitry Andric return Visit(GE->getResultExpr()); 110*0b57cec5SDimitry Andric } 111*0b57cec5SDimitry Andric ComplexPairTy VisitImaginaryLiteral(const ImaginaryLiteral *IL); 112*0b57cec5SDimitry Andric ComplexPairTy 113*0b57cec5SDimitry Andric VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *PE) { 114*0b57cec5SDimitry Andric return Visit(PE->getReplacement()); 115*0b57cec5SDimitry Andric } 116*0b57cec5SDimitry Andric ComplexPairTy VisitCoawaitExpr(CoawaitExpr *S) { 117*0b57cec5SDimitry Andric return CGF.EmitCoawaitExpr(*S).getComplexVal(); 118*0b57cec5SDimitry Andric } 119*0b57cec5SDimitry Andric ComplexPairTy VisitCoyieldExpr(CoyieldExpr *S) { 120*0b57cec5SDimitry Andric return CGF.EmitCoyieldExpr(*S).getComplexVal(); 121*0b57cec5SDimitry Andric } 122*0b57cec5SDimitry Andric ComplexPairTy VisitUnaryCoawait(const UnaryOperator *E) { 123*0b57cec5SDimitry Andric return Visit(E->getSubExpr()); 124*0b57cec5SDimitry Andric } 125*0b57cec5SDimitry Andric 126*0b57cec5SDimitry Andric ComplexPairTy emitConstant(const CodeGenFunction::ConstantEmission &Constant, 127*0b57cec5SDimitry Andric Expr *E) { 128*0b57cec5SDimitry Andric assert(Constant && "not a constant"); 129*0b57cec5SDimitry Andric if (Constant.isReference()) 130*0b57cec5SDimitry Andric return EmitLoadOfLValue(Constant.getReferenceLValue(CGF, E), 131*0b57cec5SDimitry Andric E->getExprLoc()); 132*0b57cec5SDimitry Andric 133*0b57cec5SDimitry Andric llvm::Constant *pair = Constant.getValue(); 134*0b57cec5SDimitry Andric return ComplexPairTy(pair->getAggregateElement(0U), 135*0b57cec5SDimitry Andric pair->getAggregateElement(1U)); 136*0b57cec5SDimitry Andric } 137*0b57cec5SDimitry Andric 138*0b57cec5SDimitry Andric // l-values. 139*0b57cec5SDimitry Andric ComplexPairTy VisitDeclRefExpr(DeclRefExpr *E) { 140*0b57cec5SDimitry Andric if (CodeGenFunction::ConstantEmission Constant = CGF.tryEmitAsConstant(E)) 141*0b57cec5SDimitry Andric return emitConstant(Constant, E); 142*0b57cec5SDimitry Andric return EmitLoadOfLValue(E); 143*0b57cec5SDimitry Andric } 144*0b57cec5SDimitry Andric ComplexPairTy VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) { 145*0b57cec5SDimitry Andric return EmitLoadOfLValue(E); 146*0b57cec5SDimitry Andric } 147*0b57cec5SDimitry Andric ComplexPairTy VisitObjCMessageExpr(ObjCMessageExpr *E) { 148*0b57cec5SDimitry Andric return CGF.EmitObjCMessageExpr(E).getComplexVal(); 149*0b57cec5SDimitry Andric } 150*0b57cec5SDimitry Andric ComplexPairTy VisitArraySubscriptExpr(Expr *E) { return EmitLoadOfLValue(E); } 151*0b57cec5SDimitry Andric ComplexPairTy VisitMemberExpr(MemberExpr *ME) { 152*0b57cec5SDimitry Andric if (CodeGenFunction::ConstantEmission Constant = 153*0b57cec5SDimitry Andric CGF.tryEmitAsConstant(ME)) { 154*0b57cec5SDimitry Andric CGF.EmitIgnoredExpr(ME->getBase()); 155*0b57cec5SDimitry Andric return emitConstant(Constant, ME); 156*0b57cec5SDimitry Andric } 157*0b57cec5SDimitry Andric return EmitLoadOfLValue(ME); 158*0b57cec5SDimitry Andric } 159*0b57cec5SDimitry Andric ComplexPairTy VisitOpaqueValueExpr(OpaqueValueExpr *E) { 160*0b57cec5SDimitry Andric if (E->isGLValue()) 161*0b57cec5SDimitry Andric return EmitLoadOfLValue(CGF.getOrCreateOpaqueLValueMapping(E), 162*0b57cec5SDimitry Andric E->getExprLoc()); 163*0b57cec5SDimitry Andric return CGF.getOrCreateOpaqueRValueMapping(E).getComplexVal(); 164*0b57cec5SDimitry Andric } 165*0b57cec5SDimitry Andric 166*0b57cec5SDimitry Andric ComplexPairTy VisitPseudoObjectExpr(PseudoObjectExpr *E) { 167*0b57cec5SDimitry Andric return CGF.EmitPseudoObjectRValue(E).getComplexVal(); 168*0b57cec5SDimitry Andric } 169*0b57cec5SDimitry Andric 170*0b57cec5SDimitry Andric // FIXME: CompoundLiteralExpr 171*0b57cec5SDimitry Andric 172*0b57cec5SDimitry Andric ComplexPairTy EmitCast(CastKind CK, Expr *Op, QualType DestTy); 173*0b57cec5SDimitry Andric ComplexPairTy VisitImplicitCastExpr(ImplicitCastExpr *E) { 174*0b57cec5SDimitry Andric // Unlike for scalars, we don't have to worry about function->ptr demotion 175*0b57cec5SDimitry Andric // here. 176*0b57cec5SDimitry Andric return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType()); 177*0b57cec5SDimitry Andric } 178*0b57cec5SDimitry Andric ComplexPairTy VisitCastExpr(CastExpr *E) { 179*0b57cec5SDimitry Andric if (const auto *ECE = dyn_cast<ExplicitCastExpr>(E)) 180*0b57cec5SDimitry Andric CGF.CGM.EmitExplicitCastExprType(ECE, &CGF); 181*0b57cec5SDimitry Andric return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType()); 182*0b57cec5SDimitry Andric } 183*0b57cec5SDimitry Andric ComplexPairTy VisitCallExpr(const CallExpr *E); 184*0b57cec5SDimitry Andric ComplexPairTy VisitStmtExpr(const StmtExpr *E); 185*0b57cec5SDimitry Andric 186*0b57cec5SDimitry Andric // Operators. 187*0b57cec5SDimitry Andric ComplexPairTy VisitPrePostIncDec(const UnaryOperator *E, 188*0b57cec5SDimitry Andric bool isInc, bool isPre) { 189*0b57cec5SDimitry Andric LValue LV = CGF.EmitLValue(E->getSubExpr()); 190*0b57cec5SDimitry Andric return CGF.EmitComplexPrePostIncDec(E, LV, isInc, isPre); 191*0b57cec5SDimitry Andric } 192*0b57cec5SDimitry Andric ComplexPairTy VisitUnaryPostDec(const UnaryOperator *E) { 193*0b57cec5SDimitry Andric return VisitPrePostIncDec(E, false, false); 194*0b57cec5SDimitry Andric } 195*0b57cec5SDimitry Andric ComplexPairTy VisitUnaryPostInc(const UnaryOperator *E) { 196*0b57cec5SDimitry Andric return VisitPrePostIncDec(E, true, false); 197*0b57cec5SDimitry Andric } 198*0b57cec5SDimitry Andric ComplexPairTy VisitUnaryPreDec(const UnaryOperator *E) { 199*0b57cec5SDimitry Andric return VisitPrePostIncDec(E, false, true); 200*0b57cec5SDimitry Andric } 201*0b57cec5SDimitry Andric ComplexPairTy VisitUnaryPreInc(const UnaryOperator *E) { 202*0b57cec5SDimitry Andric return VisitPrePostIncDec(E, true, true); 203*0b57cec5SDimitry Andric } 204*0b57cec5SDimitry Andric ComplexPairTy VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); } 205*0b57cec5SDimitry Andric ComplexPairTy VisitUnaryPlus (const UnaryOperator *E) { 206*0b57cec5SDimitry Andric TestAndClearIgnoreReal(); 207*0b57cec5SDimitry Andric TestAndClearIgnoreImag(); 208*0b57cec5SDimitry Andric return Visit(E->getSubExpr()); 209*0b57cec5SDimitry Andric } 210*0b57cec5SDimitry Andric ComplexPairTy VisitUnaryMinus (const UnaryOperator *E); 211*0b57cec5SDimitry Andric ComplexPairTy VisitUnaryNot (const UnaryOperator *E); 212*0b57cec5SDimitry Andric // LNot,Real,Imag never return complex. 213*0b57cec5SDimitry Andric ComplexPairTy VisitUnaryExtension(const UnaryOperator *E) { 214*0b57cec5SDimitry Andric return Visit(E->getSubExpr()); 215*0b57cec5SDimitry Andric } 216*0b57cec5SDimitry Andric ComplexPairTy VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) { 217*0b57cec5SDimitry Andric CodeGenFunction::CXXDefaultArgExprScope Scope(CGF, DAE); 218*0b57cec5SDimitry Andric return Visit(DAE->getExpr()); 219*0b57cec5SDimitry Andric } 220*0b57cec5SDimitry Andric ComplexPairTy VisitCXXDefaultInitExpr(CXXDefaultInitExpr *DIE) { 221*0b57cec5SDimitry Andric CodeGenFunction::CXXDefaultInitExprScope Scope(CGF, DIE); 222*0b57cec5SDimitry Andric return Visit(DIE->getExpr()); 223*0b57cec5SDimitry Andric } 224*0b57cec5SDimitry Andric ComplexPairTy VisitExprWithCleanups(ExprWithCleanups *E) { 225*0b57cec5SDimitry Andric CGF.enterFullExpression(E); 226*0b57cec5SDimitry Andric CodeGenFunction::RunCleanupsScope Scope(CGF); 227*0b57cec5SDimitry Andric ComplexPairTy Vals = Visit(E->getSubExpr()); 228*0b57cec5SDimitry Andric // Defend against dominance problems caused by jumps out of expression 229*0b57cec5SDimitry Andric // evaluation through the shared cleanup block. 230*0b57cec5SDimitry Andric Scope.ForceCleanup({&Vals.first, &Vals.second}); 231*0b57cec5SDimitry Andric return Vals; 232*0b57cec5SDimitry Andric } 233*0b57cec5SDimitry Andric ComplexPairTy VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) { 234*0b57cec5SDimitry Andric assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 235*0b57cec5SDimitry Andric QualType Elem = E->getType()->castAs<ComplexType>()->getElementType(); 236*0b57cec5SDimitry Andric llvm::Constant *Null = llvm::Constant::getNullValue(CGF.ConvertType(Elem)); 237*0b57cec5SDimitry Andric return ComplexPairTy(Null, Null); 238*0b57cec5SDimitry Andric } 239*0b57cec5SDimitry Andric ComplexPairTy VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) { 240*0b57cec5SDimitry Andric assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 241*0b57cec5SDimitry Andric QualType Elem = E->getType()->castAs<ComplexType>()->getElementType(); 242*0b57cec5SDimitry Andric llvm::Constant *Null = 243*0b57cec5SDimitry Andric llvm::Constant::getNullValue(CGF.ConvertType(Elem)); 244*0b57cec5SDimitry Andric return ComplexPairTy(Null, Null); 245*0b57cec5SDimitry Andric } 246*0b57cec5SDimitry Andric 247*0b57cec5SDimitry Andric struct BinOpInfo { 248*0b57cec5SDimitry Andric ComplexPairTy LHS; 249*0b57cec5SDimitry Andric ComplexPairTy RHS; 250*0b57cec5SDimitry Andric QualType Ty; // Computation Type. 251*0b57cec5SDimitry Andric }; 252*0b57cec5SDimitry Andric 253*0b57cec5SDimitry Andric BinOpInfo EmitBinOps(const BinaryOperator *E); 254*0b57cec5SDimitry Andric LValue EmitCompoundAssignLValue(const CompoundAssignOperator *E, 255*0b57cec5SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func) 256*0b57cec5SDimitry Andric (const BinOpInfo &), 257*0b57cec5SDimitry Andric RValue &Val); 258*0b57cec5SDimitry Andric ComplexPairTy EmitCompoundAssign(const CompoundAssignOperator *E, 259*0b57cec5SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func) 260*0b57cec5SDimitry Andric (const BinOpInfo &)); 261*0b57cec5SDimitry Andric 262*0b57cec5SDimitry Andric ComplexPairTy EmitBinAdd(const BinOpInfo &Op); 263*0b57cec5SDimitry Andric ComplexPairTy EmitBinSub(const BinOpInfo &Op); 264*0b57cec5SDimitry Andric ComplexPairTy EmitBinMul(const BinOpInfo &Op); 265*0b57cec5SDimitry Andric ComplexPairTy EmitBinDiv(const BinOpInfo &Op); 266*0b57cec5SDimitry Andric 267*0b57cec5SDimitry Andric ComplexPairTy EmitComplexBinOpLibCall(StringRef LibCallName, 268*0b57cec5SDimitry Andric const BinOpInfo &Op); 269*0b57cec5SDimitry Andric 270*0b57cec5SDimitry Andric ComplexPairTy VisitBinAdd(const BinaryOperator *E) { 271*0b57cec5SDimitry Andric return EmitBinAdd(EmitBinOps(E)); 272*0b57cec5SDimitry Andric } 273*0b57cec5SDimitry Andric ComplexPairTy VisitBinSub(const BinaryOperator *E) { 274*0b57cec5SDimitry Andric return EmitBinSub(EmitBinOps(E)); 275*0b57cec5SDimitry Andric } 276*0b57cec5SDimitry Andric ComplexPairTy VisitBinMul(const BinaryOperator *E) { 277*0b57cec5SDimitry Andric return EmitBinMul(EmitBinOps(E)); 278*0b57cec5SDimitry Andric } 279*0b57cec5SDimitry Andric ComplexPairTy VisitBinDiv(const BinaryOperator *E) { 280*0b57cec5SDimitry Andric return EmitBinDiv(EmitBinOps(E)); 281*0b57cec5SDimitry Andric } 282*0b57cec5SDimitry Andric 283a7dea167SDimitry Andric ComplexPairTy VisitCXXRewrittenBinaryOperator(CXXRewrittenBinaryOperator *E) { 284a7dea167SDimitry Andric return Visit(E->getSemanticForm()); 285a7dea167SDimitry Andric } 286a7dea167SDimitry Andric 287*0b57cec5SDimitry Andric // Compound assignments. 288*0b57cec5SDimitry Andric ComplexPairTy VisitBinAddAssign(const CompoundAssignOperator *E) { 289*0b57cec5SDimitry Andric return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinAdd); 290*0b57cec5SDimitry Andric } 291*0b57cec5SDimitry Andric ComplexPairTy VisitBinSubAssign(const CompoundAssignOperator *E) { 292*0b57cec5SDimitry Andric return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinSub); 293*0b57cec5SDimitry Andric } 294*0b57cec5SDimitry Andric ComplexPairTy VisitBinMulAssign(const CompoundAssignOperator *E) { 295*0b57cec5SDimitry Andric return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinMul); 296*0b57cec5SDimitry Andric } 297*0b57cec5SDimitry Andric ComplexPairTy VisitBinDivAssign(const CompoundAssignOperator *E) { 298*0b57cec5SDimitry Andric return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinDiv); 299*0b57cec5SDimitry Andric } 300*0b57cec5SDimitry Andric 301*0b57cec5SDimitry Andric // GCC rejects rem/and/or/xor for integer complex. 302*0b57cec5SDimitry Andric // Logical and/or always return int, never complex. 303*0b57cec5SDimitry Andric 304*0b57cec5SDimitry Andric // No comparisons produce a complex result. 305*0b57cec5SDimitry Andric 306*0b57cec5SDimitry Andric LValue EmitBinAssignLValue(const BinaryOperator *E, 307*0b57cec5SDimitry Andric ComplexPairTy &Val); 308*0b57cec5SDimitry Andric ComplexPairTy VisitBinAssign (const BinaryOperator *E); 309*0b57cec5SDimitry Andric ComplexPairTy VisitBinComma (const BinaryOperator *E); 310*0b57cec5SDimitry Andric 311*0b57cec5SDimitry Andric 312*0b57cec5SDimitry Andric ComplexPairTy 313*0b57cec5SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *CO); 314*0b57cec5SDimitry Andric ComplexPairTy VisitChooseExpr(ChooseExpr *CE); 315*0b57cec5SDimitry Andric 316*0b57cec5SDimitry Andric ComplexPairTy VisitInitListExpr(InitListExpr *E); 317*0b57cec5SDimitry Andric 318*0b57cec5SDimitry Andric ComplexPairTy VisitCompoundLiteralExpr(CompoundLiteralExpr *E) { 319*0b57cec5SDimitry Andric return EmitLoadOfLValue(E); 320*0b57cec5SDimitry Andric } 321*0b57cec5SDimitry Andric 322*0b57cec5SDimitry Andric ComplexPairTy VisitVAArgExpr(VAArgExpr *E); 323*0b57cec5SDimitry Andric 324*0b57cec5SDimitry Andric ComplexPairTy VisitAtomicExpr(AtomicExpr *E) { 325*0b57cec5SDimitry Andric return CGF.EmitAtomicExpr(E).getComplexVal(); 326*0b57cec5SDimitry Andric } 327*0b57cec5SDimitry Andric }; 328*0b57cec5SDimitry Andric } // end anonymous namespace. 329*0b57cec5SDimitry Andric 330*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 331*0b57cec5SDimitry Andric // Utilities 332*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 333*0b57cec5SDimitry Andric 334*0b57cec5SDimitry Andric Address CodeGenFunction::emitAddrOfRealComponent(Address addr, 335*0b57cec5SDimitry Andric QualType complexType) { 336*0b57cec5SDimitry Andric return Builder.CreateStructGEP(addr, 0, addr.getName() + ".realp"); 337*0b57cec5SDimitry Andric } 338*0b57cec5SDimitry Andric 339*0b57cec5SDimitry Andric Address CodeGenFunction::emitAddrOfImagComponent(Address addr, 340*0b57cec5SDimitry Andric QualType complexType) { 341*0b57cec5SDimitry Andric return Builder.CreateStructGEP(addr, 1, addr.getName() + ".imagp"); 342*0b57cec5SDimitry Andric } 343*0b57cec5SDimitry Andric 344*0b57cec5SDimitry Andric /// EmitLoadOfLValue - Given an RValue reference for a complex, emit code to 345*0b57cec5SDimitry Andric /// load the real and imaginary pieces, returning them as Real/Imag. 346*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitLoadOfLValue(LValue lvalue, 347*0b57cec5SDimitry Andric SourceLocation loc) { 348*0b57cec5SDimitry Andric assert(lvalue.isSimple() && "non-simple complex l-value?"); 349*0b57cec5SDimitry Andric if (lvalue.getType()->isAtomicType()) 350*0b57cec5SDimitry Andric return CGF.EmitAtomicLoad(lvalue, loc).getComplexVal(); 351*0b57cec5SDimitry Andric 352480093f4SDimitry Andric Address SrcPtr = lvalue.getAddress(CGF); 353*0b57cec5SDimitry Andric bool isVolatile = lvalue.isVolatileQualified(); 354*0b57cec5SDimitry Andric 355*0b57cec5SDimitry Andric llvm::Value *Real = nullptr, *Imag = nullptr; 356*0b57cec5SDimitry Andric 357*0b57cec5SDimitry Andric if (!IgnoreReal || isVolatile) { 358*0b57cec5SDimitry Andric Address RealP = CGF.emitAddrOfRealComponent(SrcPtr, lvalue.getType()); 359*0b57cec5SDimitry Andric Real = Builder.CreateLoad(RealP, isVolatile, SrcPtr.getName() + ".real"); 360*0b57cec5SDimitry Andric } 361*0b57cec5SDimitry Andric 362*0b57cec5SDimitry Andric if (!IgnoreImag || isVolatile) { 363*0b57cec5SDimitry Andric Address ImagP = CGF.emitAddrOfImagComponent(SrcPtr, lvalue.getType()); 364*0b57cec5SDimitry Andric Imag = Builder.CreateLoad(ImagP, isVolatile, SrcPtr.getName() + ".imag"); 365*0b57cec5SDimitry Andric } 366*0b57cec5SDimitry Andric 367*0b57cec5SDimitry Andric return ComplexPairTy(Real, Imag); 368*0b57cec5SDimitry Andric } 369*0b57cec5SDimitry Andric 370*0b57cec5SDimitry Andric /// EmitStoreOfComplex - Store the specified real/imag parts into the 371*0b57cec5SDimitry Andric /// specified value pointer. 372*0b57cec5SDimitry Andric void ComplexExprEmitter::EmitStoreOfComplex(ComplexPairTy Val, LValue lvalue, 373*0b57cec5SDimitry Andric bool isInit) { 374*0b57cec5SDimitry Andric if (lvalue.getType()->isAtomicType() || 375*0b57cec5SDimitry Andric (!isInit && CGF.LValueIsSuitableForInlineAtomic(lvalue))) 376*0b57cec5SDimitry Andric return CGF.EmitAtomicStore(RValue::getComplex(Val), lvalue, isInit); 377*0b57cec5SDimitry Andric 378480093f4SDimitry Andric Address Ptr = lvalue.getAddress(CGF); 379*0b57cec5SDimitry Andric Address RealPtr = CGF.emitAddrOfRealComponent(Ptr, lvalue.getType()); 380*0b57cec5SDimitry Andric Address ImagPtr = CGF.emitAddrOfImagComponent(Ptr, lvalue.getType()); 381*0b57cec5SDimitry Andric 382*0b57cec5SDimitry Andric Builder.CreateStore(Val.first, RealPtr, lvalue.isVolatileQualified()); 383*0b57cec5SDimitry Andric Builder.CreateStore(Val.second, ImagPtr, lvalue.isVolatileQualified()); 384*0b57cec5SDimitry Andric } 385*0b57cec5SDimitry Andric 386*0b57cec5SDimitry Andric 387*0b57cec5SDimitry Andric 388*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 389*0b57cec5SDimitry Andric // Visitor Methods 390*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 391*0b57cec5SDimitry Andric 392*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitExpr(Expr *E) { 393*0b57cec5SDimitry Andric CGF.ErrorUnsupported(E, "complex expression"); 394*0b57cec5SDimitry Andric llvm::Type *EltTy = 395*0b57cec5SDimitry Andric CGF.ConvertType(getComplexType(E->getType())->getElementType()); 396*0b57cec5SDimitry Andric llvm::Value *U = llvm::UndefValue::get(EltTy); 397*0b57cec5SDimitry Andric return ComplexPairTy(U, U); 398*0b57cec5SDimitry Andric } 399*0b57cec5SDimitry Andric 400*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter:: 401*0b57cec5SDimitry Andric VisitImaginaryLiteral(const ImaginaryLiteral *IL) { 402*0b57cec5SDimitry Andric llvm::Value *Imag = CGF.EmitScalarExpr(IL->getSubExpr()); 403*0b57cec5SDimitry Andric return ComplexPairTy(llvm::Constant::getNullValue(Imag->getType()), Imag); 404*0b57cec5SDimitry Andric } 405*0b57cec5SDimitry Andric 406*0b57cec5SDimitry Andric 407*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitCallExpr(const CallExpr *E) { 408*0b57cec5SDimitry Andric if (E->getCallReturnType(CGF.getContext())->isReferenceType()) 409*0b57cec5SDimitry Andric return EmitLoadOfLValue(E); 410*0b57cec5SDimitry Andric 411*0b57cec5SDimitry Andric return CGF.EmitCallExpr(E).getComplexVal(); 412*0b57cec5SDimitry Andric } 413*0b57cec5SDimitry Andric 414*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitStmtExpr(const StmtExpr *E) { 415*0b57cec5SDimitry Andric CodeGenFunction::StmtExprEvaluation eval(CGF); 416*0b57cec5SDimitry Andric Address RetAlloca = CGF.EmitCompoundStmt(*E->getSubStmt(), true); 417*0b57cec5SDimitry Andric assert(RetAlloca.isValid() && "Expected complex return value"); 418*0b57cec5SDimitry Andric return EmitLoadOfLValue(CGF.MakeAddrLValue(RetAlloca, E->getType()), 419*0b57cec5SDimitry Andric E->getExprLoc()); 420*0b57cec5SDimitry Andric } 421*0b57cec5SDimitry Andric 422*0b57cec5SDimitry Andric /// Emit a cast from complex value Val to DestType. 423*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitComplexToComplexCast(ComplexPairTy Val, 424*0b57cec5SDimitry Andric QualType SrcType, 425*0b57cec5SDimitry Andric QualType DestType, 426*0b57cec5SDimitry Andric SourceLocation Loc) { 427*0b57cec5SDimitry Andric // Get the src/dest element type. 428*0b57cec5SDimitry Andric SrcType = SrcType->castAs<ComplexType>()->getElementType(); 429*0b57cec5SDimitry Andric DestType = DestType->castAs<ComplexType>()->getElementType(); 430*0b57cec5SDimitry Andric 431*0b57cec5SDimitry Andric // C99 6.3.1.6: When a value of complex type is converted to another 432*0b57cec5SDimitry Andric // complex type, both the real and imaginary parts follow the conversion 433*0b57cec5SDimitry Andric // rules for the corresponding real types. 434*0b57cec5SDimitry Andric Val.first = CGF.EmitScalarConversion(Val.first, SrcType, DestType, Loc); 435*0b57cec5SDimitry Andric Val.second = CGF.EmitScalarConversion(Val.second, SrcType, DestType, Loc); 436*0b57cec5SDimitry Andric return Val; 437*0b57cec5SDimitry Andric } 438*0b57cec5SDimitry Andric 439*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitScalarToComplexCast(llvm::Value *Val, 440*0b57cec5SDimitry Andric QualType SrcType, 441*0b57cec5SDimitry Andric QualType DestType, 442*0b57cec5SDimitry Andric SourceLocation Loc) { 443*0b57cec5SDimitry Andric // Convert the input element to the element type of the complex. 444*0b57cec5SDimitry Andric DestType = DestType->castAs<ComplexType>()->getElementType(); 445*0b57cec5SDimitry Andric Val = CGF.EmitScalarConversion(Val, SrcType, DestType, Loc); 446*0b57cec5SDimitry Andric 447*0b57cec5SDimitry Andric // Return (realval, 0). 448*0b57cec5SDimitry Andric return ComplexPairTy(Val, llvm::Constant::getNullValue(Val->getType())); 449*0b57cec5SDimitry Andric } 450*0b57cec5SDimitry Andric 451*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitCast(CastKind CK, Expr *Op, 452*0b57cec5SDimitry Andric QualType DestTy) { 453*0b57cec5SDimitry Andric switch (CK) { 454*0b57cec5SDimitry Andric case CK_Dependent: llvm_unreachable("dependent cast kind in IR gen!"); 455*0b57cec5SDimitry Andric 456*0b57cec5SDimitry Andric // Atomic to non-atomic casts may be more than a no-op for some platforms and 457*0b57cec5SDimitry Andric // for some types. 458*0b57cec5SDimitry Andric case CK_AtomicToNonAtomic: 459*0b57cec5SDimitry Andric case CK_NonAtomicToAtomic: 460*0b57cec5SDimitry Andric case CK_NoOp: 461*0b57cec5SDimitry Andric case CK_LValueToRValue: 462*0b57cec5SDimitry Andric case CK_UserDefinedConversion: 463*0b57cec5SDimitry Andric return Visit(Op); 464*0b57cec5SDimitry Andric 465*0b57cec5SDimitry Andric case CK_LValueBitCast: { 466*0b57cec5SDimitry Andric LValue origLV = CGF.EmitLValue(Op); 467480093f4SDimitry Andric Address V = origLV.getAddress(CGF); 468*0b57cec5SDimitry Andric V = Builder.CreateElementBitCast(V, CGF.ConvertType(DestTy)); 469*0b57cec5SDimitry Andric return EmitLoadOfLValue(CGF.MakeAddrLValue(V, DestTy), Op->getExprLoc()); 470*0b57cec5SDimitry Andric } 471*0b57cec5SDimitry Andric 472*0b57cec5SDimitry Andric case CK_LValueToRValueBitCast: { 473*0b57cec5SDimitry Andric LValue SourceLVal = CGF.EmitLValue(Op); 474480093f4SDimitry Andric Address Addr = Builder.CreateElementBitCast(SourceLVal.getAddress(CGF), 475*0b57cec5SDimitry Andric CGF.ConvertTypeForMem(DestTy)); 476*0b57cec5SDimitry Andric LValue DestLV = CGF.MakeAddrLValue(Addr, DestTy); 477*0b57cec5SDimitry Andric DestLV.setTBAAInfo(TBAAAccessInfo::getMayAliasInfo()); 478*0b57cec5SDimitry Andric return EmitLoadOfLValue(DestLV, Op->getExprLoc()); 479*0b57cec5SDimitry Andric } 480*0b57cec5SDimitry Andric 481*0b57cec5SDimitry Andric case CK_BitCast: 482*0b57cec5SDimitry Andric case CK_BaseToDerived: 483*0b57cec5SDimitry Andric case CK_DerivedToBase: 484*0b57cec5SDimitry Andric case CK_UncheckedDerivedToBase: 485*0b57cec5SDimitry Andric case CK_Dynamic: 486*0b57cec5SDimitry Andric case CK_ToUnion: 487*0b57cec5SDimitry Andric case CK_ArrayToPointerDecay: 488*0b57cec5SDimitry Andric case CK_FunctionToPointerDecay: 489*0b57cec5SDimitry Andric case CK_NullToPointer: 490*0b57cec5SDimitry Andric case CK_NullToMemberPointer: 491*0b57cec5SDimitry Andric case CK_BaseToDerivedMemberPointer: 492*0b57cec5SDimitry Andric case CK_DerivedToBaseMemberPointer: 493*0b57cec5SDimitry Andric case CK_MemberPointerToBoolean: 494*0b57cec5SDimitry Andric case CK_ReinterpretMemberPointer: 495*0b57cec5SDimitry Andric case CK_ConstructorConversion: 496*0b57cec5SDimitry Andric case CK_IntegralToPointer: 497*0b57cec5SDimitry Andric case CK_PointerToIntegral: 498*0b57cec5SDimitry Andric case CK_PointerToBoolean: 499*0b57cec5SDimitry Andric case CK_ToVoid: 500*0b57cec5SDimitry Andric case CK_VectorSplat: 501*0b57cec5SDimitry Andric case CK_IntegralCast: 502*0b57cec5SDimitry Andric case CK_BooleanToSignedIntegral: 503*0b57cec5SDimitry Andric case CK_IntegralToBoolean: 504*0b57cec5SDimitry Andric case CK_IntegralToFloating: 505*0b57cec5SDimitry Andric case CK_FloatingToIntegral: 506*0b57cec5SDimitry Andric case CK_FloatingToBoolean: 507*0b57cec5SDimitry Andric case CK_FloatingCast: 508*0b57cec5SDimitry Andric case CK_CPointerToObjCPointerCast: 509*0b57cec5SDimitry Andric case CK_BlockPointerToObjCPointerCast: 510*0b57cec5SDimitry Andric case CK_AnyPointerToBlockPointerCast: 511*0b57cec5SDimitry Andric case CK_ObjCObjectLValueCast: 512*0b57cec5SDimitry Andric case CK_FloatingComplexToReal: 513*0b57cec5SDimitry Andric case CK_FloatingComplexToBoolean: 514*0b57cec5SDimitry Andric case CK_IntegralComplexToReal: 515*0b57cec5SDimitry Andric case CK_IntegralComplexToBoolean: 516*0b57cec5SDimitry Andric case CK_ARCProduceObject: 517*0b57cec5SDimitry Andric case CK_ARCConsumeObject: 518*0b57cec5SDimitry Andric case CK_ARCReclaimReturnedObject: 519*0b57cec5SDimitry Andric case CK_ARCExtendBlockObject: 520*0b57cec5SDimitry Andric case CK_CopyAndAutoreleaseBlockObject: 521*0b57cec5SDimitry Andric case CK_BuiltinFnToFnPtr: 522*0b57cec5SDimitry Andric case CK_ZeroToOCLOpaqueType: 523*0b57cec5SDimitry Andric case CK_AddressSpaceConversion: 524*0b57cec5SDimitry Andric case CK_IntToOCLSampler: 525*0b57cec5SDimitry Andric case CK_FixedPointCast: 526*0b57cec5SDimitry Andric case CK_FixedPointToBoolean: 527*0b57cec5SDimitry Andric case CK_FixedPointToIntegral: 528*0b57cec5SDimitry Andric case CK_IntegralToFixedPoint: 529*0b57cec5SDimitry Andric llvm_unreachable("invalid cast kind for complex value"); 530*0b57cec5SDimitry Andric 531*0b57cec5SDimitry Andric case CK_FloatingRealToComplex: 532*0b57cec5SDimitry Andric case CK_IntegralRealToComplex: 533*0b57cec5SDimitry Andric return EmitScalarToComplexCast(CGF.EmitScalarExpr(Op), Op->getType(), 534*0b57cec5SDimitry Andric DestTy, Op->getExprLoc()); 535*0b57cec5SDimitry Andric 536*0b57cec5SDimitry Andric case CK_FloatingComplexCast: 537*0b57cec5SDimitry Andric case CK_FloatingComplexToIntegralComplex: 538*0b57cec5SDimitry Andric case CK_IntegralComplexCast: 539*0b57cec5SDimitry Andric case CK_IntegralComplexToFloatingComplex: 540*0b57cec5SDimitry Andric return EmitComplexToComplexCast(Visit(Op), Op->getType(), DestTy, 541*0b57cec5SDimitry Andric Op->getExprLoc()); 542*0b57cec5SDimitry Andric } 543*0b57cec5SDimitry Andric 544*0b57cec5SDimitry Andric llvm_unreachable("unknown cast resulting in complex value"); 545*0b57cec5SDimitry Andric } 546*0b57cec5SDimitry Andric 547*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitUnaryMinus(const UnaryOperator *E) { 548*0b57cec5SDimitry Andric TestAndClearIgnoreReal(); 549*0b57cec5SDimitry Andric TestAndClearIgnoreImag(); 550*0b57cec5SDimitry Andric ComplexPairTy Op = Visit(E->getSubExpr()); 551*0b57cec5SDimitry Andric 552*0b57cec5SDimitry Andric llvm::Value *ResR, *ResI; 553*0b57cec5SDimitry Andric if (Op.first->getType()->isFloatingPointTy()) { 554*0b57cec5SDimitry Andric ResR = Builder.CreateFNeg(Op.first, "neg.r"); 555*0b57cec5SDimitry Andric ResI = Builder.CreateFNeg(Op.second, "neg.i"); 556*0b57cec5SDimitry Andric } else { 557*0b57cec5SDimitry Andric ResR = Builder.CreateNeg(Op.first, "neg.r"); 558*0b57cec5SDimitry Andric ResI = Builder.CreateNeg(Op.second, "neg.i"); 559*0b57cec5SDimitry Andric } 560*0b57cec5SDimitry Andric return ComplexPairTy(ResR, ResI); 561*0b57cec5SDimitry Andric } 562*0b57cec5SDimitry Andric 563*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitUnaryNot(const UnaryOperator *E) { 564*0b57cec5SDimitry Andric TestAndClearIgnoreReal(); 565*0b57cec5SDimitry Andric TestAndClearIgnoreImag(); 566*0b57cec5SDimitry Andric // ~(a+ib) = a + i*-b 567*0b57cec5SDimitry Andric ComplexPairTy Op = Visit(E->getSubExpr()); 568*0b57cec5SDimitry Andric llvm::Value *ResI; 569*0b57cec5SDimitry Andric if (Op.second->getType()->isFloatingPointTy()) 570*0b57cec5SDimitry Andric ResI = Builder.CreateFNeg(Op.second, "conj.i"); 571*0b57cec5SDimitry Andric else 572*0b57cec5SDimitry Andric ResI = Builder.CreateNeg(Op.second, "conj.i"); 573*0b57cec5SDimitry Andric 574*0b57cec5SDimitry Andric return ComplexPairTy(Op.first, ResI); 575*0b57cec5SDimitry Andric } 576*0b57cec5SDimitry Andric 577*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitBinAdd(const BinOpInfo &Op) { 578*0b57cec5SDimitry Andric llvm::Value *ResR, *ResI; 579*0b57cec5SDimitry Andric 580*0b57cec5SDimitry Andric if (Op.LHS.first->getType()->isFloatingPointTy()) { 581*0b57cec5SDimitry Andric ResR = Builder.CreateFAdd(Op.LHS.first, Op.RHS.first, "add.r"); 582*0b57cec5SDimitry Andric if (Op.LHS.second && Op.RHS.second) 583*0b57cec5SDimitry Andric ResI = Builder.CreateFAdd(Op.LHS.second, Op.RHS.second, "add.i"); 584*0b57cec5SDimitry Andric else 585*0b57cec5SDimitry Andric ResI = Op.LHS.second ? Op.LHS.second : Op.RHS.second; 586*0b57cec5SDimitry Andric assert(ResI && "Only one operand may be real!"); 587*0b57cec5SDimitry Andric } else { 588*0b57cec5SDimitry Andric ResR = Builder.CreateAdd(Op.LHS.first, Op.RHS.first, "add.r"); 589*0b57cec5SDimitry Andric assert(Op.LHS.second && Op.RHS.second && 590*0b57cec5SDimitry Andric "Both operands of integer complex operators must be complex!"); 591*0b57cec5SDimitry Andric ResI = Builder.CreateAdd(Op.LHS.second, Op.RHS.second, "add.i"); 592*0b57cec5SDimitry Andric } 593*0b57cec5SDimitry Andric return ComplexPairTy(ResR, ResI); 594*0b57cec5SDimitry Andric } 595*0b57cec5SDimitry Andric 596*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitBinSub(const BinOpInfo &Op) { 597*0b57cec5SDimitry Andric llvm::Value *ResR, *ResI; 598*0b57cec5SDimitry Andric if (Op.LHS.first->getType()->isFloatingPointTy()) { 599*0b57cec5SDimitry Andric ResR = Builder.CreateFSub(Op.LHS.first, Op.RHS.first, "sub.r"); 600*0b57cec5SDimitry Andric if (Op.LHS.second && Op.RHS.second) 601*0b57cec5SDimitry Andric ResI = Builder.CreateFSub(Op.LHS.second, Op.RHS.second, "sub.i"); 602*0b57cec5SDimitry Andric else 603*0b57cec5SDimitry Andric ResI = Op.LHS.second ? Op.LHS.second 604*0b57cec5SDimitry Andric : Builder.CreateFNeg(Op.RHS.second, "sub.i"); 605*0b57cec5SDimitry Andric assert(ResI && "Only one operand may be real!"); 606*0b57cec5SDimitry Andric } else { 607*0b57cec5SDimitry Andric ResR = Builder.CreateSub(Op.LHS.first, Op.RHS.first, "sub.r"); 608*0b57cec5SDimitry Andric assert(Op.LHS.second && Op.RHS.second && 609*0b57cec5SDimitry Andric "Both operands of integer complex operators must be complex!"); 610*0b57cec5SDimitry Andric ResI = Builder.CreateSub(Op.LHS.second, Op.RHS.second, "sub.i"); 611*0b57cec5SDimitry Andric } 612*0b57cec5SDimitry Andric return ComplexPairTy(ResR, ResI); 613*0b57cec5SDimitry Andric } 614*0b57cec5SDimitry Andric 615*0b57cec5SDimitry Andric /// Emit a libcall for a binary operation on complex types. 616*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitComplexBinOpLibCall(StringRef LibCallName, 617*0b57cec5SDimitry Andric const BinOpInfo &Op) { 618*0b57cec5SDimitry Andric CallArgList Args; 619*0b57cec5SDimitry Andric Args.add(RValue::get(Op.LHS.first), 620*0b57cec5SDimitry Andric Op.Ty->castAs<ComplexType>()->getElementType()); 621*0b57cec5SDimitry Andric Args.add(RValue::get(Op.LHS.second), 622*0b57cec5SDimitry Andric Op.Ty->castAs<ComplexType>()->getElementType()); 623*0b57cec5SDimitry Andric Args.add(RValue::get(Op.RHS.first), 624*0b57cec5SDimitry Andric Op.Ty->castAs<ComplexType>()->getElementType()); 625*0b57cec5SDimitry Andric Args.add(RValue::get(Op.RHS.second), 626*0b57cec5SDimitry Andric Op.Ty->castAs<ComplexType>()->getElementType()); 627*0b57cec5SDimitry Andric 628*0b57cec5SDimitry Andric // We *must* use the full CG function call building logic here because the 629*0b57cec5SDimitry Andric // complex type has special ABI handling. We also should not forget about 630*0b57cec5SDimitry Andric // special calling convention which may be used for compiler builtins. 631*0b57cec5SDimitry Andric 632*0b57cec5SDimitry Andric // We create a function qualified type to state that this call does not have 633*0b57cec5SDimitry Andric // any exceptions. 634*0b57cec5SDimitry Andric FunctionProtoType::ExtProtoInfo EPI; 635*0b57cec5SDimitry Andric EPI = EPI.withExceptionSpec( 636*0b57cec5SDimitry Andric FunctionProtoType::ExceptionSpecInfo(EST_BasicNoexcept)); 637*0b57cec5SDimitry Andric SmallVector<QualType, 4> ArgsQTys( 638*0b57cec5SDimitry Andric 4, Op.Ty->castAs<ComplexType>()->getElementType()); 639*0b57cec5SDimitry Andric QualType FQTy = CGF.getContext().getFunctionType(Op.Ty, ArgsQTys, EPI); 640*0b57cec5SDimitry Andric const CGFunctionInfo &FuncInfo = CGF.CGM.getTypes().arrangeFreeFunctionCall( 641*0b57cec5SDimitry Andric Args, cast<FunctionType>(FQTy.getTypePtr()), false); 642*0b57cec5SDimitry Andric 643*0b57cec5SDimitry Andric llvm::FunctionType *FTy = CGF.CGM.getTypes().GetFunctionType(FuncInfo); 644*0b57cec5SDimitry Andric llvm::FunctionCallee Func = CGF.CGM.CreateRuntimeFunction( 645*0b57cec5SDimitry Andric FTy, LibCallName, llvm::AttributeList(), true); 646*0b57cec5SDimitry Andric CGCallee Callee = CGCallee::forDirect(Func, FQTy->getAs<FunctionProtoType>()); 647*0b57cec5SDimitry Andric 648*0b57cec5SDimitry Andric llvm::CallBase *Call; 649*0b57cec5SDimitry Andric RValue Res = CGF.EmitCall(FuncInfo, Callee, ReturnValueSlot(), Args, &Call); 650*0b57cec5SDimitry Andric Call->setCallingConv(CGF.CGM.getRuntimeCC()); 651*0b57cec5SDimitry Andric return Res.getComplexVal(); 652*0b57cec5SDimitry Andric } 653*0b57cec5SDimitry Andric 654*0b57cec5SDimitry Andric /// Lookup the libcall name for a given floating point type complex 655*0b57cec5SDimitry Andric /// multiply. 656*0b57cec5SDimitry Andric static StringRef getComplexMultiplyLibCallName(llvm::Type *Ty) { 657*0b57cec5SDimitry Andric switch (Ty->getTypeID()) { 658*0b57cec5SDimitry Andric default: 659*0b57cec5SDimitry Andric llvm_unreachable("Unsupported floating point type!"); 660*0b57cec5SDimitry Andric case llvm::Type::HalfTyID: 661*0b57cec5SDimitry Andric return "__mulhc3"; 662*0b57cec5SDimitry Andric case llvm::Type::FloatTyID: 663*0b57cec5SDimitry Andric return "__mulsc3"; 664*0b57cec5SDimitry Andric case llvm::Type::DoubleTyID: 665*0b57cec5SDimitry Andric return "__muldc3"; 666*0b57cec5SDimitry Andric case llvm::Type::PPC_FP128TyID: 667*0b57cec5SDimitry Andric return "__multc3"; 668*0b57cec5SDimitry Andric case llvm::Type::X86_FP80TyID: 669*0b57cec5SDimitry Andric return "__mulxc3"; 670*0b57cec5SDimitry Andric case llvm::Type::FP128TyID: 671*0b57cec5SDimitry Andric return "__multc3"; 672*0b57cec5SDimitry Andric } 673*0b57cec5SDimitry Andric } 674*0b57cec5SDimitry Andric 675*0b57cec5SDimitry Andric // See C11 Annex G.5.1 for the semantics of multiplicative operators on complex 676*0b57cec5SDimitry Andric // typed values. 677*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitBinMul(const BinOpInfo &Op) { 678*0b57cec5SDimitry Andric using llvm::Value; 679*0b57cec5SDimitry Andric Value *ResR, *ResI; 680*0b57cec5SDimitry Andric llvm::MDBuilder MDHelper(CGF.getLLVMContext()); 681*0b57cec5SDimitry Andric 682*0b57cec5SDimitry Andric if (Op.LHS.first->getType()->isFloatingPointTy()) { 683*0b57cec5SDimitry Andric // The general formulation is: 684*0b57cec5SDimitry Andric // (a + ib) * (c + id) = (a * c - b * d) + i(a * d + b * c) 685*0b57cec5SDimitry Andric // 686*0b57cec5SDimitry Andric // But we can fold away components which would be zero due to a real 687*0b57cec5SDimitry Andric // operand according to C11 Annex G.5.1p2. 688*0b57cec5SDimitry Andric // FIXME: C11 also provides for imaginary types which would allow folding 689*0b57cec5SDimitry Andric // still more of this within the type system. 690*0b57cec5SDimitry Andric 691*0b57cec5SDimitry Andric if (Op.LHS.second && Op.RHS.second) { 692*0b57cec5SDimitry Andric // If both operands are complex, emit the core math directly, and then 693*0b57cec5SDimitry Andric // test for NaNs. If we find NaNs in the result, we delegate to a libcall 694*0b57cec5SDimitry Andric // to carefully re-compute the correct infinity representation if 695*0b57cec5SDimitry Andric // possible. The expectation is that the presence of NaNs here is 696*0b57cec5SDimitry Andric // *extremely* rare, and so the cost of the libcall is almost irrelevant. 697*0b57cec5SDimitry Andric // This is good, because the libcall re-computes the core multiplication 698*0b57cec5SDimitry Andric // exactly the same as we do here and re-tests for NaNs in order to be 699*0b57cec5SDimitry Andric // a generic complex*complex libcall. 700*0b57cec5SDimitry Andric 701*0b57cec5SDimitry Andric // First compute the four products. 702*0b57cec5SDimitry Andric Value *AC = Builder.CreateFMul(Op.LHS.first, Op.RHS.first, "mul_ac"); 703*0b57cec5SDimitry Andric Value *BD = Builder.CreateFMul(Op.LHS.second, Op.RHS.second, "mul_bd"); 704*0b57cec5SDimitry Andric Value *AD = Builder.CreateFMul(Op.LHS.first, Op.RHS.second, "mul_ad"); 705*0b57cec5SDimitry Andric Value *BC = Builder.CreateFMul(Op.LHS.second, Op.RHS.first, "mul_bc"); 706*0b57cec5SDimitry Andric 707*0b57cec5SDimitry Andric // The real part is the difference of the first two, the imaginary part is 708*0b57cec5SDimitry Andric // the sum of the second. 709*0b57cec5SDimitry Andric ResR = Builder.CreateFSub(AC, BD, "mul_r"); 710*0b57cec5SDimitry Andric ResI = Builder.CreateFAdd(AD, BC, "mul_i"); 711*0b57cec5SDimitry Andric 712*0b57cec5SDimitry Andric // Emit the test for the real part becoming NaN and create a branch to 713*0b57cec5SDimitry Andric // handle it. We test for NaN by comparing the number to itself. 714*0b57cec5SDimitry Andric Value *IsRNaN = Builder.CreateFCmpUNO(ResR, ResR, "isnan_cmp"); 715*0b57cec5SDimitry Andric llvm::BasicBlock *ContBB = CGF.createBasicBlock("complex_mul_cont"); 716*0b57cec5SDimitry Andric llvm::BasicBlock *INaNBB = CGF.createBasicBlock("complex_mul_imag_nan"); 717*0b57cec5SDimitry Andric llvm::Instruction *Branch = Builder.CreateCondBr(IsRNaN, INaNBB, ContBB); 718*0b57cec5SDimitry Andric llvm::BasicBlock *OrigBB = Branch->getParent(); 719*0b57cec5SDimitry Andric 720*0b57cec5SDimitry Andric // Give hint that we very much don't expect to see NaNs. 721*0b57cec5SDimitry Andric // Value chosen to match UR_NONTAKEN_WEIGHT, see BranchProbabilityInfo.cpp 722*0b57cec5SDimitry Andric llvm::MDNode *BrWeight = MDHelper.createBranchWeights(1, (1U << 20) - 1); 723*0b57cec5SDimitry Andric Branch->setMetadata(llvm::LLVMContext::MD_prof, BrWeight); 724*0b57cec5SDimitry Andric 725*0b57cec5SDimitry Andric // Now test the imaginary part and create its branch. 726*0b57cec5SDimitry Andric CGF.EmitBlock(INaNBB); 727*0b57cec5SDimitry Andric Value *IsINaN = Builder.CreateFCmpUNO(ResI, ResI, "isnan_cmp"); 728*0b57cec5SDimitry Andric llvm::BasicBlock *LibCallBB = CGF.createBasicBlock("complex_mul_libcall"); 729*0b57cec5SDimitry Andric Branch = Builder.CreateCondBr(IsINaN, LibCallBB, ContBB); 730*0b57cec5SDimitry Andric Branch->setMetadata(llvm::LLVMContext::MD_prof, BrWeight); 731*0b57cec5SDimitry Andric 732*0b57cec5SDimitry Andric // Now emit the libcall on this slowest of the slow paths. 733*0b57cec5SDimitry Andric CGF.EmitBlock(LibCallBB); 734*0b57cec5SDimitry Andric Value *LibCallR, *LibCallI; 735*0b57cec5SDimitry Andric std::tie(LibCallR, LibCallI) = EmitComplexBinOpLibCall( 736*0b57cec5SDimitry Andric getComplexMultiplyLibCallName(Op.LHS.first->getType()), Op); 737*0b57cec5SDimitry Andric Builder.CreateBr(ContBB); 738*0b57cec5SDimitry Andric 739*0b57cec5SDimitry Andric // Finally continue execution by phi-ing together the different 740*0b57cec5SDimitry Andric // computation paths. 741*0b57cec5SDimitry Andric CGF.EmitBlock(ContBB); 742*0b57cec5SDimitry Andric llvm::PHINode *RealPHI = Builder.CreatePHI(ResR->getType(), 3, "real_mul_phi"); 743*0b57cec5SDimitry Andric RealPHI->addIncoming(ResR, OrigBB); 744*0b57cec5SDimitry Andric RealPHI->addIncoming(ResR, INaNBB); 745*0b57cec5SDimitry Andric RealPHI->addIncoming(LibCallR, LibCallBB); 746*0b57cec5SDimitry Andric llvm::PHINode *ImagPHI = Builder.CreatePHI(ResI->getType(), 3, "imag_mul_phi"); 747*0b57cec5SDimitry Andric ImagPHI->addIncoming(ResI, OrigBB); 748*0b57cec5SDimitry Andric ImagPHI->addIncoming(ResI, INaNBB); 749*0b57cec5SDimitry Andric ImagPHI->addIncoming(LibCallI, LibCallBB); 750*0b57cec5SDimitry Andric return ComplexPairTy(RealPHI, ImagPHI); 751*0b57cec5SDimitry Andric } 752*0b57cec5SDimitry Andric assert((Op.LHS.second || Op.RHS.second) && 753*0b57cec5SDimitry Andric "At least one operand must be complex!"); 754*0b57cec5SDimitry Andric 755*0b57cec5SDimitry Andric // If either of the operands is a real rather than a complex, the 756*0b57cec5SDimitry Andric // imaginary component is ignored when computing the real component of the 757*0b57cec5SDimitry Andric // result. 758*0b57cec5SDimitry Andric ResR = Builder.CreateFMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 759*0b57cec5SDimitry Andric 760*0b57cec5SDimitry Andric ResI = Op.LHS.second 761*0b57cec5SDimitry Andric ? Builder.CreateFMul(Op.LHS.second, Op.RHS.first, "mul.il") 762*0b57cec5SDimitry Andric : Builder.CreateFMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 763*0b57cec5SDimitry Andric } else { 764*0b57cec5SDimitry Andric assert(Op.LHS.second && Op.RHS.second && 765*0b57cec5SDimitry Andric "Both operands of integer complex operators must be complex!"); 766*0b57cec5SDimitry Andric Value *ResRl = Builder.CreateMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 767*0b57cec5SDimitry Andric Value *ResRr = Builder.CreateMul(Op.LHS.second, Op.RHS.second, "mul.rr"); 768*0b57cec5SDimitry Andric ResR = Builder.CreateSub(ResRl, ResRr, "mul.r"); 769*0b57cec5SDimitry Andric 770*0b57cec5SDimitry Andric Value *ResIl = Builder.CreateMul(Op.LHS.second, Op.RHS.first, "mul.il"); 771*0b57cec5SDimitry Andric Value *ResIr = Builder.CreateMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 772*0b57cec5SDimitry Andric ResI = Builder.CreateAdd(ResIl, ResIr, "mul.i"); 773*0b57cec5SDimitry Andric } 774*0b57cec5SDimitry Andric return ComplexPairTy(ResR, ResI); 775*0b57cec5SDimitry Andric } 776*0b57cec5SDimitry Andric 777*0b57cec5SDimitry Andric // See C11 Annex G.5.1 for the semantics of multiplicative operators on complex 778*0b57cec5SDimitry Andric // typed values. 779*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::EmitBinDiv(const BinOpInfo &Op) { 780*0b57cec5SDimitry Andric llvm::Value *LHSr = Op.LHS.first, *LHSi = Op.LHS.second; 781*0b57cec5SDimitry Andric llvm::Value *RHSr = Op.RHS.first, *RHSi = Op.RHS.second; 782*0b57cec5SDimitry Andric 783*0b57cec5SDimitry Andric llvm::Value *DSTr, *DSTi; 784*0b57cec5SDimitry Andric if (LHSr->getType()->isFloatingPointTy()) { 785*0b57cec5SDimitry Andric // If we have a complex operand on the RHS and FastMath is not allowed, we 786*0b57cec5SDimitry Andric // delegate to a libcall to handle all of the complexities and minimize 787*0b57cec5SDimitry Andric // underflow/overflow cases. When FastMath is allowed we construct the 788*0b57cec5SDimitry Andric // divide inline using the same algorithm as for integer operands. 789*0b57cec5SDimitry Andric // 790*0b57cec5SDimitry Andric // FIXME: We would be able to avoid the libcall in many places if we 791*0b57cec5SDimitry Andric // supported imaginary types in addition to complex types. 792*0b57cec5SDimitry Andric if (RHSi && !CGF.getLangOpts().FastMath) { 793*0b57cec5SDimitry Andric BinOpInfo LibCallOp = Op; 794*0b57cec5SDimitry Andric // If LHS was a real, supply a null imaginary part. 795*0b57cec5SDimitry Andric if (!LHSi) 796*0b57cec5SDimitry Andric LibCallOp.LHS.second = llvm::Constant::getNullValue(LHSr->getType()); 797*0b57cec5SDimitry Andric 798*0b57cec5SDimitry Andric switch (LHSr->getType()->getTypeID()) { 799*0b57cec5SDimitry Andric default: 800*0b57cec5SDimitry Andric llvm_unreachable("Unsupported floating point type!"); 801*0b57cec5SDimitry Andric case llvm::Type::HalfTyID: 802*0b57cec5SDimitry Andric return EmitComplexBinOpLibCall("__divhc3", LibCallOp); 803*0b57cec5SDimitry Andric case llvm::Type::FloatTyID: 804*0b57cec5SDimitry Andric return EmitComplexBinOpLibCall("__divsc3", LibCallOp); 805*0b57cec5SDimitry Andric case llvm::Type::DoubleTyID: 806*0b57cec5SDimitry Andric return EmitComplexBinOpLibCall("__divdc3", LibCallOp); 807*0b57cec5SDimitry Andric case llvm::Type::PPC_FP128TyID: 808*0b57cec5SDimitry Andric return EmitComplexBinOpLibCall("__divtc3", LibCallOp); 809*0b57cec5SDimitry Andric case llvm::Type::X86_FP80TyID: 810*0b57cec5SDimitry Andric return EmitComplexBinOpLibCall("__divxc3", LibCallOp); 811*0b57cec5SDimitry Andric case llvm::Type::FP128TyID: 812*0b57cec5SDimitry Andric return EmitComplexBinOpLibCall("__divtc3", LibCallOp); 813*0b57cec5SDimitry Andric } 814*0b57cec5SDimitry Andric } else if (RHSi) { 815*0b57cec5SDimitry Andric if (!LHSi) 816*0b57cec5SDimitry Andric LHSi = llvm::Constant::getNullValue(RHSi->getType()); 817*0b57cec5SDimitry Andric 818*0b57cec5SDimitry Andric // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 819*0b57cec5SDimitry Andric llvm::Value *AC = Builder.CreateFMul(LHSr, RHSr); // a*c 820*0b57cec5SDimitry Andric llvm::Value *BD = Builder.CreateFMul(LHSi, RHSi); // b*d 821*0b57cec5SDimitry Andric llvm::Value *ACpBD = Builder.CreateFAdd(AC, BD); // ac+bd 822*0b57cec5SDimitry Andric 823*0b57cec5SDimitry Andric llvm::Value *CC = Builder.CreateFMul(RHSr, RHSr); // c*c 824*0b57cec5SDimitry Andric llvm::Value *DD = Builder.CreateFMul(RHSi, RHSi); // d*d 825*0b57cec5SDimitry Andric llvm::Value *CCpDD = Builder.CreateFAdd(CC, DD); // cc+dd 826*0b57cec5SDimitry Andric 827*0b57cec5SDimitry Andric llvm::Value *BC = Builder.CreateFMul(LHSi, RHSr); // b*c 828*0b57cec5SDimitry Andric llvm::Value *AD = Builder.CreateFMul(LHSr, RHSi); // a*d 829*0b57cec5SDimitry Andric llvm::Value *BCmAD = Builder.CreateFSub(BC, AD); // bc-ad 830*0b57cec5SDimitry Andric 831*0b57cec5SDimitry Andric DSTr = Builder.CreateFDiv(ACpBD, CCpDD); 832*0b57cec5SDimitry Andric DSTi = Builder.CreateFDiv(BCmAD, CCpDD); 833*0b57cec5SDimitry Andric } else { 834*0b57cec5SDimitry Andric assert(LHSi && "Can have at most one non-complex operand!"); 835*0b57cec5SDimitry Andric 836*0b57cec5SDimitry Andric DSTr = Builder.CreateFDiv(LHSr, RHSr); 837*0b57cec5SDimitry Andric DSTi = Builder.CreateFDiv(LHSi, RHSr); 838*0b57cec5SDimitry Andric } 839*0b57cec5SDimitry Andric } else { 840*0b57cec5SDimitry Andric assert(Op.LHS.second && Op.RHS.second && 841*0b57cec5SDimitry Andric "Both operands of integer complex operators must be complex!"); 842*0b57cec5SDimitry Andric // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 843*0b57cec5SDimitry Andric llvm::Value *Tmp1 = Builder.CreateMul(LHSr, RHSr); // a*c 844*0b57cec5SDimitry Andric llvm::Value *Tmp2 = Builder.CreateMul(LHSi, RHSi); // b*d 845*0b57cec5SDimitry Andric llvm::Value *Tmp3 = Builder.CreateAdd(Tmp1, Tmp2); // ac+bd 846*0b57cec5SDimitry Andric 847*0b57cec5SDimitry Andric llvm::Value *Tmp4 = Builder.CreateMul(RHSr, RHSr); // c*c 848*0b57cec5SDimitry Andric llvm::Value *Tmp5 = Builder.CreateMul(RHSi, RHSi); // d*d 849*0b57cec5SDimitry Andric llvm::Value *Tmp6 = Builder.CreateAdd(Tmp4, Tmp5); // cc+dd 850*0b57cec5SDimitry Andric 851*0b57cec5SDimitry Andric llvm::Value *Tmp7 = Builder.CreateMul(LHSi, RHSr); // b*c 852*0b57cec5SDimitry Andric llvm::Value *Tmp8 = Builder.CreateMul(LHSr, RHSi); // a*d 853*0b57cec5SDimitry Andric llvm::Value *Tmp9 = Builder.CreateSub(Tmp7, Tmp8); // bc-ad 854*0b57cec5SDimitry Andric 855*0b57cec5SDimitry Andric if (Op.Ty->castAs<ComplexType>()->getElementType()->isUnsignedIntegerType()) { 856*0b57cec5SDimitry Andric DSTr = Builder.CreateUDiv(Tmp3, Tmp6); 857*0b57cec5SDimitry Andric DSTi = Builder.CreateUDiv(Tmp9, Tmp6); 858*0b57cec5SDimitry Andric } else { 859*0b57cec5SDimitry Andric DSTr = Builder.CreateSDiv(Tmp3, Tmp6); 860*0b57cec5SDimitry Andric DSTi = Builder.CreateSDiv(Tmp9, Tmp6); 861*0b57cec5SDimitry Andric } 862*0b57cec5SDimitry Andric } 863*0b57cec5SDimitry Andric 864*0b57cec5SDimitry Andric return ComplexPairTy(DSTr, DSTi); 865*0b57cec5SDimitry Andric } 866*0b57cec5SDimitry Andric 867*0b57cec5SDimitry Andric ComplexExprEmitter::BinOpInfo 868*0b57cec5SDimitry Andric ComplexExprEmitter::EmitBinOps(const BinaryOperator *E) { 869*0b57cec5SDimitry Andric TestAndClearIgnoreReal(); 870*0b57cec5SDimitry Andric TestAndClearIgnoreImag(); 871*0b57cec5SDimitry Andric BinOpInfo Ops; 872*0b57cec5SDimitry Andric if (E->getLHS()->getType()->isRealFloatingType()) 873*0b57cec5SDimitry Andric Ops.LHS = ComplexPairTy(CGF.EmitScalarExpr(E->getLHS()), nullptr); 874*0b57cec5SDimitry Andric else 875*0b57cec5SDimitry Andric Ops.LHS = Visit(E->getLHS()); 876*0b57cec5SDimitry Andric if (E->getRHS()->getType()->isRealFloatingType()) 877*0b57cec5SDimitry Andric Ops.RHS = ComplexPairTy(CGF.EmitScalarExpr(E->getRHS()), nullptr); 878*0b57cec5SDimitry Andric else 879*0b57cec5SDimitry Andric Ops.RHS = Visit(E->getRHS()); 880*0b57cec5SDimitry Andric 881*0b57cec5SDimitry Andric Ops.Ty = E->getType(); 882*0b57cec5SDimitry Andric return Ops; 883*0b57cec5SDimitry Andric } 884*0b57cec5SDimitry Andric 885*0b57cec5SDimitry Andric 886*0b57cec5SDimitry Andric LValue ComplexExprEmitter:: 887*0b57cec5SDimitry Andric EmitCompoundAssignLValue(const CompoundAssignOperator *E, 888*0b57cec5SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&), 889*0b57cec5SDimitry Andric RValue &Val) { 890*0b57cec5SDimitry Andric TestAndClearIgnoreReal(); 891*0b57cec5SDimitry Andric TestAndClearIgnoreImag(); 892*0b57cec5SDimitry Andric QualType LHSTy = E->getLHS()->getType(); 893*0b57cec5SDimitry Andric if (const AtomicType *AT = LHSTy->getAs<AtomicType>()) 894*0b57cec5SDimitry Andric LHSTy = AT->getValueType(); 895*0b57cec5SDimitry Andric 896*0b57cec5SDimitry Andric BinOpInfo OpInfo; 897*0b57cec5SDimitry Andric 898*0b57cec5SDimitry Andric // Load the RHS and LHS operands. 899*0b57cec5SDimitry Andric // __block variables need to have the rhs evaluated first, plus this should 900*0b57cec5SDimitry Andric // improve codegen a little. 901*0b57cec5SDimitry Andric OpInfo.Ty = E->getComputationResultType(); 902*0b57cec5SDimitry Andric QualType ComplexElementTy = cast<ComplexType>(OpInfo.Ty)->getElementType(); 903*0b57cec5SDimitry Andric 904*0b57cec5SDimitry Andric // The RHS should have been converted to the computation type. 905*0b57cec5SDimitry Andric if (E->getRHS()->getType()->isRealFloatingType()) { 906*0b57cec5SDimitry Andric assert( 907*0b57cec5SDimitry Andric CGF.getContext() 908*0b57cec5SDimitry Andric .hasSameUnqualifiedType(ComplexElementTy, E->getRHS()->getType())); 909*0b57cec5SDimitry Andric OpInfo.RHS = ComplexPairTy(CGF.EmitScalarExpr(E->getRHS()), nullptr); 910*0b57cec5SDimitry Andric } else { 911*0b57cec5SDimitry Andric assert(CGF.getContext() 912*0b57cec5SDimitry Andric .hasSameUnqualifiedType(OpInfo.Ty, E->getRHS()->getType())); 913*0b57cec5SDimitry Andric OpInfo.RHS = Visit(E->getRHS()); 914*0b57cec5SDimitry Andric } 915*0b57cec5SDimitry Andric 916*0b57cec5SDimitry Andric LValue LHS = CGF.EmitLValue(E->getLHS()); 917*0b57cec5SDimitry Andric 918*0b57cec5SDimitry Andric // Load from the l-value and convert it. 919*0b57cec5SDimitry Andric SourceLocation Loc = E->getExprLoc(); 920*0b57cec5SDimitry Andric if (LHSTy->isAnyComplexType()) { 921*0b57cec5SDimitry Andric ComplexPairTy LHSVal = EmitLoadOfLValue(LHS, Loc); 922*0b57cec5SDimitry Andric OpInfo.LHS = EmitComplexToComplexCast(LHSVal, LHSTy, OpInfo.Ty, Loc); 923*0b57cec5SDimitry Andric } else { 924*0b57cec5SDimitry Andric llvm::Value *LHSVal = CGF.EmitLoadOfScalar(LHS, Loc); 925*0b57cec5SDimitry Andric // For floating point real operands we can directly pass the scalar form 926*0b57cec5SDimitry Andric // to the binary operator emission and potentially get more efficient code. 927*0b57cec5SDimitry Andric if (LHSTy->isRealFloatingType()) { 928*0b57cec5SDimitry Andric if (!CGF.getContext().hasSameUnqualifiedType(ComplexElementTy, LHSTy)) 929*0b57cec5SDimitry Andric LHSVal = CGF.EmitScalarConversion(LHSVal, LHSTy, ComplexElementTy, Loc); 930*0b57cec5SDimitry Andric OpInfo.LHS = ComplexPairTy(LHSVal, nullptr); 931*0b57cec5SDimitry Andric } else { 932*0b57cec5SDimitry Andric OpInfo.LHS = EmitScalarToComplexCast(LHSVal, LHSTy, OpInfo.Ty, Loc); 933*0b57cec5SDimitry Andric } 934*0b57cec5SDimitry Andric } 935*0b57cec5SDimitry Andric 936*0b57cec5SDimitry Andric // Expand the binary operator. 937*0b57cec5SDimitry Andric ComplexPairTy Result = (this->*Func)(OpInfo); 938*0b57cec5SDimitry Andric 939*0b57cec5SDimitry Andric // Truncate the result and store it into the LHS lvalue. 940*0b57cec5SDimitry Andric if (LHSTy->isAnyComplexType()) { 941*0b57cec5SDimitry Andric ComplexPairTy ResVal = 942*0b57cec5SDimitry Andric EmitComplexToComplexCast(Result, OpInfo.Ty, LHSTy, Loc); 943*0b57cec5SDimitry Andric EmitStoreOfComplex(ResVal, LHS, /*isInit*/ false); 944*0b57cec5SDimitry Andric Val = RValue::getComplex(ResVal); 945*0b57cec5SDimitry Andric } else { 946*0b57cec5SDimitry Andric llvm::Value *ResVal = 947*0b57cec5SDimitry Andric CGF.EmitComplexToScalarConversion(Result, OpInfo.Ty, LHSTy, Loc); 948*0b57cec5SDimitry Andric CGF.EmitStoreOfScalar(ResVal, LHS, /*isInit*/ false); 949*0b57cec5SDimitry Andric Val = RValue::get(ResVal); 950*0b57cec5SDimitry Andric } 951*0b57cec5SDimitry Andric 952*0b57cec5SDimitry Andric return LHS; 953*0b57cec5SDimitry Andric } 954*0b57cec5SDimitry Andric 955*0b57cec5SDimitry Andric // Compound assignments. 956*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter:: 957*0b57cec5SDimitry Andric EmitCompoundAssign(const CompoundAssignOperator *E, 958*0b57cec5SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&)){ 959*0b57cec5SDimitry Andric RValue Val; 960*0b57cec5SDimitry Andric LValue LV = EmitCompoundAssignLValue(E, Func, Val); 961*0b57cec5SDimitry Andric 962*0b57cec5SDimitry Andric // The result of an assignment in C is the assigned r-value. 963*0b57cec5SDimitry Andric if (!CGF.getLangOpts().CPlusPlus) 964*0b57cec5SDimitry Andric return Val.getComplexVal(); 965*0b57cec5SDimitry Andric 966*0b57cec5SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 967*0b57cec5SDimitry Andric if (!LV.isVolatileQualified()) 968*0b57cec5SDimitry Andric return Val.getComplexVal(); 969*0b57cec5SDimitry Andric 970*0b57cec5SDimitry Andric return EmitLoadOfLValue(LV, E->getExprLoc()); 971*0b57cec5SDimitry Andric } 972*0b57cec5SDimitry Andric 973*0b57cec5SDimitry Andric LValue ComplexExprEmitter::EmitBinAssignLValue(const BinaryOperator *E, 974*0b57cec5SDimitry Andric ComplexPairTy &Val) { 975*0b57cec5SDimitry Andric assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(), 976*0b57cec5SDimitry Andric E->getRHS()->getType()) && 977*0b57cec5SDimitry Andric "Invalid assignment"); 978*0b57cec5SDimitry Andric TestAndClearIgnoreReal(); 979*0b57cec5SDimitry Andric TestAndClearIgnoreImag(); 980*0b57cec5SDimitry Andric 981*0b57cec5SDimitry Andric // Emit the RHS. __block variables need the RHS evaluated first. 982*0b57cec5SDimitry Andric Val = Visit(E->getRHS()); 983*0b57cec5SDimitry Andric 984*0b57cec5SDimitry Andric // Compute the address to store into. 985*0b57cec5SDimitry Andric LValue LHS = CGF.EmitLValue(E->getLHS()); 986*0b57cec5SDimitry Andric 987*0b57cec5SDimitry Andric // Store the result value into the LHS lvalue. 988*0b57cec5SDimitry Andric EmitStoreOfComplex(Val, LHS, /*isInit*/ false); 989*0b57cec5SDimitry Andric 990*0b57cec5SDimitry Andric return LHS; 991*0b57cec5SDimitry Andric } 992*0b57cec5SDimitry Andric 993*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitBinAssign(const BinaryOperator *E) { 994*0b57cec5SDimitry Andric ComplexPairTy Val; 995*0b57cec5SDimitry Andric LValue LV = EmitBinAssignLValue(E, Val); 996*0b57cec5SDimitry Andric 997*0b57cec5SDimitry Andric // The result of an assignment in C is the assigned r-value. 998*0b57cec5SDimitry Andric if (!CGF.getLangOpts().CPlusPlus) 999*0b57cec5SDimitry Andric return Val; 1000*0b57cec5SDimitry Andric 1001*0b57cec5SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 1002*0b57cec5SDimitry Andric if (!LV.isVolatileQualified()) 1003*0b57cec5SDimitry Andric return Val; 1004*0b57cec5SDimitry Andric 1005*0b57cec5SDimitry Andric return EmitLoadOfLValue(LV, E->getExprLoc()); 1006*0b57cec5SDimitry Andric } 1007*0b57cec5SDimitry Andric 1008*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitBinComma(const BinaryOperator *E) { 1009*0b57cec5SDimitry Andric CGF.EmitIgnoredExpr(E->getLHS()); 1010*0b57cec5SDimitry Andric return Visit(E->getRHS()); 1011*0b57cec5SDimitry Andric } 1012*0b57cec5SDimitry Andric 1013*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter:: 1014*0b57cec5SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) { 1015*0b57cec5SDimitry Andric TestAndClearIgnoreReal(); 1016*0b57cec5SDimitry Andric TestAndClearIgnoreImag(); 1017*0b57cec5SDimitry Andric llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true"); 1018*0b57cec5SDimitry Andric llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false"); 1019*0b57cec5SDimitry Andric llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end"); 1020*0b57cec5SDimitry Andric 1021*0b57cec5SDimitry Andric // Bind the common expression if necessary. 1022*0b57cec5SDimitry Andric CodeGenFunction::OpaqueValueMapping binding(CGF, E); 1023*0b57cec5SDimitry Andric 1024*0b57cec5SDimitry Andric 1025*0b57cec5SDimitry Andric CodeGenFunction::ConditionalEvaluation eval(CGF); 1026*0b57cec5SDimitry Andric CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock, 1027*0b57cec5SDimitry Andric CGF.getProfileCount(E)); 1028*0b57cec5SDimitry Andric 1029*0b57cec5SDimitry Andric eval.begin(CGF); 1030*0b57cec5SDimitry Andric CGF.EmitBlock(LHSBlock); 1031*0b57cec5SDimitry Andric CGF.incrementProfileCounter(E); 1032*0b57cec5SDimitry Andric ComplexPairTy LHS = Visit(E->getTrueExpr()); 1033*0b57cec5SDimitry Andric LHSBlock = Builder.GetInsertBlock(); 1034*0b57cec5SDimitry Andric CGF.EmitBranch(ContBlock); 1035*0b57cec5SDimitry Andric eval.end(CGF); 1036*0b57cec5SDimitry Andric 1037*0b57cec5SDimitry Andric eval.begin(CGF); 1038*0b57cec5SDimitry Andric CGF.EmitBlock(RHSBlock); 1039*0b57cec5SDimitry Andric ComplexPairTy RHS = Visit(E->getFalseExpr()); 1040*0b57cec5SDimitry Andric RHSBlock = Builder.GetInsertBlock(); 1041*0b57cec5SDimitry Andric CGF.EmitBlock(ContBlock); 1042*0b57cec5SDimitry Andric eval.end(CGF); 1043*0b57cec5SDimitry Andric 1044*0b57cec5SDimitry Andric // Create a PHI node for the real part. 1045*0b57cec5SDimitry Andric llvm::PHINode *RealPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.r"); 1046*0b57cec5SDimitry Andric RealPN->addIncoming(LHS.first, LHSBlock); 1047*0b57cec5SDimitry Andric RealPN->addIncoming(RHS.first, RHSBlock); 1048*0b57cec5SDimitry Andric 1049*0b57cec5SDimitry Andric // Create a PHI node for the imaginary part. 1050*0b57cec5SDimitry Andric llvm::PHINode *ImagPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.i"); 1051*0b57cec5SDimitry Andric ImagPN->addIncoming(LHS.second, LHSBlock); 1052*0b57cec5SDimitry Andric ImagPN->addIncoming(RHS.second, RHSBlock); 1053*0b57cec5SDimitry Andric 1054*0b57cec5SDimitry Andric return ComplexPairTy(RealPN, ImagPN); 1055*0b57cec5SDimitry Andric } 1056*0b57cec5SDimitry Andric 1057*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitChooseExpr(ChooseExpr *E) { 1058*0b57cec5SDimitry Andric return Visit(E->getChosenSubExpr()); 1059*0b57cec5SDimitry Andric } 1060*0b57cec5SDimitry Andric 1061*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitInitListExpr(InitListExpr *E) { 1062*0b57cec5SDimitry Andric bool Ignore = TestAndClearIgnoreReal(); 1063*0b57cec5SDimitry Andric (void)Ignore; 1064*0b57cec5SDimitry Andric assert (Ignore == false && "init list ignored"); 1065*0b57cec5SDimitry Andric Ignore = TestAndClearIgnoreImag(); 1066*0b57cec5SDimitry Andric (void)Ignore; 1067*0b57cec5SDimitry Andric assert (Ignore == false && "init list ignored"); 1068*0b57cec5SDimitry Andric 1069*0b57cec5SDimitry Andric if (E->getNumInits() == 2) { 1070*0b57cec5SDimitry Andric llvm::Value *Real = CGF.EmitScalarExpr(E->getInit(0)); 1071*0b57cec5SDimitry Andric llvm::Value *Imag = CGF.EmitScalarExpr(E->getInit(1)); 1072*0b57cec5SDimitry Andric return ComplexPairTy(Real, Imag); 1073*0b57cec5SDimitry Andric } else if (E->getNumInits() == 1) { 1074*0b57cec5SDimitry Andric return Visit(E->getInit(0)); 1075*0b57cec5SDimitry Andric } 1076*0b57cec5SDimitry Andric 1077*0b57cec5SDimitry Andric // Empty init list initializes to null 1078*0b57cec5SDimitry Andric assert(E->getNumInits() == 0 && "Unexpected number of inits"); 1079*0b57cec5SDimitry Andric QualType Ty = E->getType()->castAs<ComplexType>()->getElementType(); 1080*0b57cec5SDimitry Andric llvm::Type* LTy = CGF.ConvertType(Ty); 1081*0b57cec5SDimitry Andric llvm::Value* zeroConstant = llvm::Constant::getNullValue(LTy); 1082*0b57cec5SDimitry Andric return ComplexPairTy(zeroConstant, zeroConstant); 1083*0b57cec5SDimitry Andric } 1084*0b57cec5SDimitry Andric 1085*0b57cec5SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitVAArgExpr(VAArgExpr *E) { 1086*0b57cec5SDimitry Andric Address ArgValue = Address::invalid(); 1087*0b57cec5SDimitry Andric Address ArgPtr = CGF.EmitVAArg(E, ArgValue); 1088*0b57cec5SDimitry Andric 1089*0b57cec5SDimitry Andric if (!ArgPtr.isValid()) { 1090*0b57cec5SDimitry Andric CGF.ErrorUnsupported(E, "complex va_arg expression"); 1091*0b57cec5SDimitry Andric llvm::Type *EltTy = 1092*0b57cec5SDimitry Andric CGF.ConvertType(E->getType()->castAs<ComplexType>()->getElementType()); 1093*0b57cec5SDimitry Andric llvm::Value *U = llvm::UndefValue::get(EltTy); 1094*0b57cec5SDimitry Andric return ComplexPairTy(U, U); 1095*0b57cec5SDimitry Andric } 1096*0b57cec5SDimitry Andric 1097*0b57cec5SDimitry Andric return EmitLoadOfLValue(CGF.MakeAddrLValue(ArgPtr, E->getType()), 1098*0b57cec5SDimitry Andric E->getExprLoc()); 1099*0b57cec5SDimitry Andric } 1100*0b57cec5SDimitry Andric 1101*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 1102*0b57cec5SDimitry Andric // Entry Point into this File 1103*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 1104*0b57cec5SDimitry Andric 1105*0b57cec5SDimitry Andric /// EmitComplexExpr - Emit the computation of the specified expression of 1106*0b57cec5SDimitry Andric /// complex type, ignoring the result. 1107*0b57cec5SDimitry Andric ComplexPairTy CodeGenFunction::EmitComplexExpr(const Expr *E, bool IgnoreReal, 1108*0b57cec5SDimitry Andric bool IgnoreImag) { 1109*0b57cec5SDimitry Andric assert(E && getComplexType(E->getType()) && 1110*0b57cec5SDimitry Andric "Invalid complex expression to emit"); 1111*0b57cec5SDimitry Andric 1112*0b57cec5SDimitry Andric return ComplexExprEmitter(*this, IgnoreReal, IgnoreImag) 1113*0b57cec5SDimitry Andric .Visit(const_cast<Expr *>(E)); 1114*0b57cec5SDimitry Andric } 1115*0b57cec5SDimitry Andric 1116*0b57cec5SDimitry Andric void CodeGenFunction::EmitComplexExprIntoLValue(const Expr *E, LValue dest, 1117*0b57cec5SDimitry Andric bool isInit) { 1118*0b57cec5SDimitry Andric assert(E && getComplexType(E->getType()) && 1119*0b57cec5SDimitry Andric "Invalid complex expression to emit"); 1120*0b57cec5SDimitry Andric ComplexExprEmitter Emitter(*this); 1121*0b57cec5SDimitry Andric ComplexPairTy Val = Emitter.Visit(const_cast<Expr*>(E)); 1122*0b57cec5SDimitry Andric Emitter.EmitStoreOfComplex(Val, dest, isInit); 1123*0b57cec5SDimitry Andric } 1124*0b57cec5SDimitry Andric 1125*0b57cec5SDimitry Andric /// EmitStoreOfComplex - Store a complex number into the specified l-value. 1126*0b57cec5SDimitry Andric void CodeGenFunction::EmitStoreOfComplex(ComplexPairTy V, LValue dest, 1127*0b57cec5SDimitry Andric bool isInit) { 1128*0b57cec5SDimitry Andric ComplexExprEmitter(*this).EmitStoreOfComplex(V, dest, isInit); 1129*0b57cec5SDimitry Andric } 1130*0b57cec5SDimitry Andric 1131*0b57cec5SDimitry Andric /// EmitLoadOfComplex - Load a complex number from the specified address. 1132*0b57cec5SDimitry Andric ComplexPairTy CodeGenFunction::EmitLoadOfComplex(LValue src, 1133*0b57cec5SDimitry Andric SourceLocation loc) { 1134*0b57cec5SDimitry Andric return ComplexExprEmitter(*this).EmitLoadOfLValue(src, loc); 1135*0b57cec5SDimitry Andric } 1136*0b57cec5SDimitry Andric 1137*0b57cec5SDimitry Andric LValue CodeGenFunction::EmitComplexAssignmentLValue(const BinaryOperator *E) { 1138*0b57cec5SDimitry Andric assert(E->getOpcode() == BO_Assign); 1139*0b57cec5SDimitry Andric ComplexPairTy Val; // ignored 1140480093f4SDimitry Andric LValue LVal = ComplexExprEmitter(*this).EmitBinAssignLValue(E, Val); 1141480093f4SDimitry Andric if (getLangOpts().OpenMP) 1142480093f4SDimitry Andric CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(*this, 1143480093f4SDimitry Andric E->getLHS()); 1144480093f4SDimitry Andric return LVal; 1145*0b57cec5SDimitry Andric } 1146*0b57cec5SDimitry Andric 1147*0b57cec5SDimitry Andric typedef ComplexPairTy (ComplexExprEmitter::*CompoundFunc)( 1148*0b57cec5SDimitry Andric const ComplexExprEmitter::BinOpInfo &); 1149*0b57cec5SDimitry Andric 1150*0b57cec5SDimitry Andric static CompoundFunc getComplexOp(BinaryOperatorKind Op) { 1151*0b57cec5SDimitry Andric switch (Op) { 1152*0b57cec5SDimitry Andric case BO_MulAssign: return &ComplexExprEmitter::EmitBinMul; 1153*0b57cec5SDimitry Andric case BO_DivAssign: return &ComplexExprEmitter::EmitBinDiv; 1154*0b57cec5SDimitry Andric case BO_SubAssign: return &ComplexExprEmitter::EmitBinSub; 1155*0b57cec5SDimitry Andric case BO_AddAssign: return &ComplexExprEmitter::EmitBinAdd; 1156*0b57cec5SDimitry Andric default: 1157*0b57cec5SDimitry Andric llvm_unreachable("unexpected complex compound assignment"); 1158*0b57cec5SDimitry Andric } 1159*0b57cec5SDimitry Andric } 1160*0b57cec5SDimitry Andric 1161*0b57cec5SDimitry Andric LValue CodeGenFunction:: 1162*0b57cec5SDimitry Andric EmitComplexCompoundAssignmentLValue(const CompoundAssignOperator *E) { 1163*0b57cec5SDimitry Andric CompoundFunc Op = getComplexOp(E->getOpcode()); 1164*0b57cec5SDimitry Andric RValue Val; 1165*0b57cec5SDimitry Andric return ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val); 1166*0b57cec5SDimitry Andric } 1167*0b57cec5SDimitry Andric 1168*0b57cec5SDimitry Andric LValue CodeGenFunction:: 1169*0b57cec5SDimitry Andric EmitScalarCompoundAssignWithComplex(const CompoundAssignOperator *E, 1170*0b57cec5SDimitry Andric llvm::Value *&Result) { 1171*0b57cec5SDimitry Andric CompoundFunc Op = getComplexOp(E->getOpcode()); 1172*0b57cec5SDimitry Andric RValue Val; 1173*0b57cec5SDimitry Andric LValue Ret = ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val); 1174*0b57cec5SDimitry Andric Result = Val.getScalarVal(); 1175*0b57cec5SDimitry Andric return Ret; 1176*0b57cec5SDimitry Andric } 1177