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/ASTContext.h" 17f22ef01cSRoman Divacky #include "clang/AST/StmtVisitor.h" 18f22ef01cSRoman Divacky #include "llvm/Constants.h" 19f22ef01cSRoman Divacky #include "llvm/Function.h" 20f22ef01cSRoman Divacky #include "llvm/ADT/SmallString.h" 21f22ef01cSRoman Divacky using namespace clang; 22f22ef01cSRoman Divacky using namespace CodeGen; 23f22ef01cSRoman Divacky 24f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 25f22ef01cSRoman Divacky // Complex Expression Emitter 26f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 27f22ef01cSRoman Divacky 28f22ef01cSRoman Divacky typedef CodeGenFunction::ComplexPairTy ComplexPairTy; 29f22ef01cSRoman Divacky 30f22ef01cSRoman Divacky namespace { 31f22ef01cSRoman Divacky class ComplexExprEmitter 32f22ef01cSRoman Divacky : public StmtVisitor<ComplexExprEmitter, ComplexPairTy> { 33f22ef01cSRoman Divacky CodeGenFunction &CGF; 34f22ef01cSRoman Divacky CGBuilderTy &Builder; 35f22ef01cSRoman Divacky // True is we should ignore the value of a 36f22ef01cSRoman Divacky bool IgnoreReal; 37f22ef01cSRoman Divacky bool IgnoreImag; 38f22ef01cSRoman Divacky public: 392754fe60SDimitry Andric ComplexExprEmitter(CodeGenFunction &cgf, bool ir=false, bool ii=false) 402754fe60SDimitry Andric : CGF(cgf), Builder(CGF.Builder), IgnoreReal(ir), IgnoreImag(ii) { 41f22ef01cSRoman Divacky } 42f22ef01cSRoman Divacky 43f22ef01cSRoman Divacky 44f22ef01cSRoman Divacky //===--------------------------------------------------------------------===// 45f22ef01cSRoman Divacky // Utilities 46f22ef01cSRoman Divacky //===--------------------------------------------------------------------===// 47f22ef01cSRoman Divacky 48f22ef01cSRoman Divacky bool TestAndClearIgnoreReal() { 49f22ef01cSRoman Divacky bool I = IgnoreReal; 50f22ef01cSRoman Divacky IgnoreReal = false; 51f22ef01cSRoman Divacky return I; 52f22ef01cSRoman Divacky } 53f22ef01cSRoman Divacky bool TestAndClearIgnoreImag() { 54f22ef01cSRoman Divacky bool I = IgnoreImag; 55f22ef01cSRoman Divacky IgnoreImag = false; 56f22ef01cSRoman Divacky return I; 57f22ef01cSRoman Divacky } 58f22ef01cSRoman Divacky 59f22ef01cSRoman Divacky /// EmitLoadOfLValue - Given an expression with complex type that represents a 60f22ef01cSRoman Divacky /// value l-value, this method emits the address of the l-value, then loads 61f22ef01cSRoman Divacky /// and returns the result. 62f22ef01cSRoman Divacky ComplexPairTy EmitLoadOfLValue(const Expr *E) { 632754fe60SDimitry Andric return EmitLoadOfLValue(CGF.EmitLValue(E)); 642754fe60SDimitry Andric } 652754fe60SDimitry Andric 662754fe60SDimitry Andric ComplexPairTy EmitLoadOfLValue(LValue LV) { 67f22ef01cSRoman Divacky if (LV.isSimple()) 68f22ef01cSRoman Divacky return EmitLoadOfComplex(LV.getAddress(), LV.isVolatileQualified()); 69f22ef01cSRoman Divacky 702754fe60SDimitry Andric assert(LV.isPropertyRef() && "Unknown LValue type!"); 712754fe60SDimitry Andric return CGF.EmitLoadOfPropertyRefLValue(LV).getComplexVal(); 72f22ef01cSRoman Divacky } 73f22ef01cSRoman Divacky 74f22ef01cSRoman Divacky /// EmitLoadOfComplex - Given a pointer to a complex value, emit code to load 75f22ef01cSRoman Divacky /// the real and imaginary pieces. 76f22ef01cSRoman Divacky ComplexPairTy EmitLoadOfComplex(llvm::Value *SrcPtr, bool isVolatile); 77f22ef01cSRoman Divacky 782754fe60SDimitry Andric /// EmitStoreThroughLValue - Given an l-value of complex type, store 792754fe60SDimitry Andric /// a complex number into it. 802754fe60SDimitry Andric void EmitStoreThroughLValue(ComplexPairTy Val, LValue LV) { 812754fe60SDimitry Andric if (LV.isSimple()) 822754fe60SDimitry Andric return EmitStoreOfComplex(Val, LV.getAddress(), LV.isVolatileQualified()); 832754fe60SDimitry Andric 842754fe60SDimitry Andric assert(LV.isPropertyRef() && "Unknown LValue type!"); 852754fe60SDimitry Andric CGF.EmitStoreThroughPropertyRefLValue(RValue::getComplex(Val), LV); 862754fe60SDimitry Andric } 872754fe60SDimitry Andric 88f22ef01cSRoman Divacky /// EmitStoreOfComplex - Store the specified real/imag parts into the 89f22ef01cSRoman Divacky /// specified value pointer. 90f22ef01cSRoman Divacky void EmitStoreOfComplex(ComplexPairTy Val, llvm::Value *ResPtr, bool isVol); 91f22ef01cSRoman Divacky 92f22ef01cSRoman Divacky /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType. 93f22ef01cSRoman Divacky ComplexPairTy EmitComplexToComplexCast(ComplexPairTy Val, QualType SrcType, 94f22ef01cSRoman Divacky QualType DestType); 95f22ef01cSRoman Divacky 96f22ef01cSRoman Divacky //===--------------------------------------------------------------------===// 97f22ef01cSRoman Divacky // Visitor Methods 98f22ef01cSRoman Divacky //===--------------------------------------------------------------------===// 99f22ef01cSRoman Divacky 1002754fe60SDimitry Andric ComplexPairTy Visit(Expr *E) { 1012754fe60SDimitry Andric return StmtVisitor<ComplexExprEmitter, ComplexPairTy>::Visit(E); 1022754fe60SDimitry Andric } 1032754fe60SDimitry Andric 104f22ef01cSRoman Divacky ComplexPairTy VisitStmt(Stmt *S) { 105f22ef01cSRoman Divacky S->dump(CGF.getContext().getSourceManager()); 106f22ef01cSRoman Divacky assert(0 && "Stmt can't have complex result type!"); 107f22ef01cSRoman Divacky return ComplexPairTy(); 108f22ef01cSRoman Divacky } 109f22ef01cSRoman Divacky ComplexPairTy VisitExpr(Expr *S); 110f22ef01cSRoman Divacky ComplexPairTy VisitParenExpr(ParenExpr *PE) { return Visit(PE->getSubExpr());} 111f22ef01cSRoman Divacky ComplexPairTy VisitImaginaryLiteral(const ImaginaryLiteral *IL); 112f22ef01cSRoman Divacky 113f22ef01cSRoman Divacky // l-values. 114f22ef01cSRoman Divacky ComplexPairTy VisitDeclRefExpr(const Expr *E) { return EmitLoadOfLValue(E); } 115f22ef01cSRoman Divacky ComplexPairTy VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) { 116f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 117f22ef01cSRoman Divacky } 118f22ef01cSRoman Divacky ComplexPairTy VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) { 1192754fe60SDimitry Andric assert(E->getObjectKind() == OK_Ordinary); 120f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 121f22ef01cSRoman Divacky } 122f22ef01cSRoman Divacky ComplexPairTy VisitObjCMessageExpr(ObjCMessageExpr *E) { 123f22ef01cSRoman Divacky return CGF.EmitObjCMessageExpr(E).getComplexVal(); 124f22ef01cSRoman Divacky } 125f22ef01cSRoman Divacky ComplexPairTy VisitArraySubscriptExpr(Expr *E) { return EmitLoadOfLValue(E); } 126f22ef01cSRoman Divacky ComplexPairTy VisitMemberExpr(const Expr *E) { return EmitLoadOfLValue(E); } 1272754fe60SDimitry Andric ComplexPairTy VisitOpaqueValueExpr(OpaqueValueExpr *E) { 1282754fe60SDimitry Andric if (E->isGLValue()) 1292754fe60SDimitry Andric return EmitLoadOfLValue(CGF.getOpaqueLValueMapping(E)); 1302754fe60SDimitry Andric return CGF.getOpaqueRValueMapping(E).getComplexVal(); 1312754fe60SDimitry Andric } 132f22ef01cSRoman Divacky 133f22ef01cSRoman Divacky // FIXME: CompoundLiteralExpr 134f22ef01cSRoman Divacky 135ffd1746dSEd Schouten ComplexPairTy EmitCast(CastExpr::CastKind CK, Expr *Op, QualType DestTy); 136f22ef01cSRoman Divacky ComplexPairTy VisitImplicitCastExpr(ImplicitCastExpr *E) { 137f22ef01cSRoman Divacky // Unlike for scalars, we don't have to worry about function->ptr demotion 138f22ef01cSRoman Divacky // here. 139ffd1746dSEd Schouten return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType()); 140f22ef01cSRoman Divacky } 141f22ef01cSRoman Divacky ComplexPairTy VisitCastExpr(CastExpr *E) { 142ffd1746dSEd Schouten return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType()); 143f22ef01cSRoman Divacky } 144f22ef01cSRoman Divacky ComplexPairTy VisitCallExpr(const CallExpr *E); 145f22ef01cSRoman Divacky ComplexPairTy VisitStmtExpr(const StmtExpr *E); 146f22ef01cSRoman Divacky 147f22ef01cSRoman Divacky // Operators. 148f22ef01cSRoman Divacky ComplexPairTy VisitPrePostIncDec(const UnaryOperator *E, 149f22ef01cSRoman Divacky bool isInc, bool isPre) { 150f22ef01cSRoman Divacky LValue LV = CGF.EmitLValue(E->getSubExpr()); 151f22ef01cSRoman Divacky return CGF.EmitComplexPrePostIncDec(E, LV, isInc, isPre); 152f22ef01cSRoman Divacky } 153f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPostDec(const UnaryOperator *E) { 154f22ef01cSRoman Divacky return VisitPrePostIncDec(E, false, false); 155f22ef01cSRoman Divacky } 156f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPostInc(const UnaryOperator *E) { 157f22ef01cSRoman Divacky return VisitPrePostIncDec(E, true, false); 158f22ef01cSRoman Divacky } 159f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPreDec(const UnaryOperator *E) { 160f22ef01cSRoman Divacky return VisitPrePostIncDec(E, false, true); 161f22ef01cSRoman Divacky } 162f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPreInc(const UnaryOperator *E) { 163f22ef01cSRoman Divacky return VisitPrePostIncDec(E, true, true); 164f22ef01cSRoman Divacky } 165f22ef01cSRoman Divacky ComplexPairTy VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); } 166f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPlus (const UnaryOperator *E) { 167f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 168f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 169f22ef01cSRoman Divacky return Visit(E->getSubExpr()); 170f22ef01cSRoman Divacky } 171f22ef01cSRoman Divacky ComplexPairTy VisitUnaryMinus (const UnaryOperator *E); 172f22ef01cSRoman Divacky ComplexPairTy VisitUnaryNot (const UnaryOperator *E); 173f22ef01cSRoman Divacky // LNot,Real,Imag never return complex. 174f22ef01cSRoman Divacky ComplexPairTy VisitUnaryExtension(const UnaryOperator *E) { 175f22ef01cSRoman Divacky return Visit(E->getSubExpr()); 176f22ef01cSRoman Divacky } 177f22ef01cSRoman Divacky ComplexPairTy VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) { 178f22ef01cSRoman Divacky return Visit(DAE->getExpr()); 179f22ef01cSRoman Divacky } 1802754fe60SDimitry Andric ComplexPairTy VisitExprWithCleanups(ExprWithCleanups *E) { 1812754fe60SDimitry Andric return CGF.EmitExprWithCleanups(E).getComplexVal(); 182f22ef01cSRoman Divacky } 183ffd1746dSEd Schouten ComplexPairTy VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) { 184f22ef01cSRoman Divacky assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 185f22ef01cSRoman Divacky QualType Elem = E->getType()->getAs<ComplexType>()->getElementType(); 186f22ef01cSRoman Divacky llvm::Constant *Null = llvm::Constant::getNullValue(CGF.ConvertType(Elem)); 187f22ef01cSRoman Divacky return ComplexPairTy(Null, Null); 188f22ef01cSRoman Divacky } 189f22ef01cSRoman Divacky ComplexPairTy VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) { 190f22ef01cSRoman Divacky assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 191f22ef01cSRoman Divacky QualType Elem = E->getType()->getAs<ComplexType>()->getElementType(); 192f22ef01cSRoman Divacky llvm::Constant *Null = 193f22ef01cSRoman Divacky llvm::Constant::getNullValue(CGF.ConvertType(Elem)); 194f22ef01cSRoman Divacky return ComplexPairTy(Null, Null); 195f22ef01cSRoman Divacky } 196f22ef01cSRoman Divacky 197f22ef01cSRoman Divacky struct BinOpInfo { 198f22ef01cSRoman Divacky ComplexPairTy LHS; 199f22ef01cSRoman Divacky ComplexPairTy RHS; 200f22ef01cSRoman Divacky QualType Ty; // Computation Type. 201f22ef01cSRoman Divacky }; 202f22ef01cSRoman Divacky 203f22ef01cSRoman Divacky BinOpInfo EmitBinOps(const BinaryOperator *E); 2042754fe60SDimitry Andric LValue EmitCompoundAssignLValue(const CompoundAssignOperator *E, 2052754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func) 2062754fe60SDimitry Andric (const BinOpInfo &), 2072754fe60SDimitry Andric ComplexPairTy &Val); 208f22ef01cSRoman Divacky ComplexPairTy EmitCompoundAssign(const CompoundAssignOperator *E, 209f22ef01cSRoman Divacky ComplexPairTy (ComplexExprEmitter::*Func) 210f22ef01cSRoman Divacky (const BinOpInfo &)); 211f22ef01cSRoman Divacky 212f22ef01cSRoman Divacky ComplexPairTy EmitBinAdd(const BinOpInfo &Op); 213f22ef01cSRoman Divacky ComplexPairTy EmitBinSub(const BinOpInfo &Op); 214f22ef01cSRoman Divacky ComplexPairTy EmitBinMul(const BinOpInfo &Op); 215f22ef01cSRoman Divacky ComplexPairTy EmitBinDiv(const BinOpInfo &Op); 216f22ef01cSRoman Divacky 217f22ef01cSRoman Divacky ComplexPairTy VisitBinAdd(const BinaryOperator *E) { 218f22ef01cSRoman Divacky return EmitBinAdd(EmitBinOps(E)); 219f22ef01cSRoman Divacky } 220f22ef01cSRoman Divacky ComplexPairTy VisitBinSub(const BinaryOperator *E) { 221f22ef01cSRoman Divacky return EmitBinSub(EmitBinOps(E)); 222f22ef01cSRoman Divacky } 2232754fe60SDimitry Andric ComplexPairTy VisitBinMul(const BinaryOperator *E) { 2242754fe60SDimitry Andric return EmitBinMul(EmitBinOps(E)); 2252754fe60SDimitry Andric } 226f22ef01cSRoman Divacky ComplexPairTy VisitBinDiv(const BinaryOperator *E) { 227f22ef01cSRoman Divacky return EmitBinDiv(EmitBinOps(E)); 228f22ef01cSRoman Divacky } 229f22ef01cSRoman Divacky 230f22ef01cSRoman Divacky // Compound assignments. 231f22ef01cSRoman Divacky ComplexPairTy VisitBinAddAssign(const CompoundAssignOperator *E) { 232f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinAdd); 233f22ef01cSRoman Divacky } 234f22ef01cSRoman Divacky ComplexPairTy VisitBinSubAssign(const CompoundAssignOperator *E) { 235f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinSub); 236f22ef01cSRoman Divacky } 237f22ef01cSRoman Divacky ComplexPairTy VisitBinMulAssign(const CompoundAssignOperator *E) { 238f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinMul); 239f22ef01cSRoman Divacky } 240f22ef01cSRoman Divacky ComplexPairTy VisitBinDivAssign(const CompoundAssignOperator *E) { 241f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinDiv); 242f22ef01cSRoman Divacky } 243f22ef01cSRoman Divacky 244f22ef01cSRoman Divacky // GCC rejects rem/and/or/xor for integer complex. 245f22ef01cSRoman Divacky // Logical and/or always return int, never complex. 246f22ef01cSRoman Divacky 247f22ef01cSRoman Divacky // No comparisons produce a complex result. 2482754fe60SDimitry Andric 2492754fe60SDimitry Andric LValue EmitBinAssignLValue(const BinaryOperator *E, 2502754fe60SDimitry Andric ComplexPairTy &Val); 251f22ef01cSRoman Divacky ComplexPairTy VisitBinAssign (const BinaryOperator *E); 252f22ef01cSRoman Divacky ComplexPairTy VisitBinComma (const BinaryOperator *E); 253f22ef01cSRoman Divacky 254f22ef01cSRoman Divacky 2552754fe60SDimitry Andric ComplexPairTy 2562754fe60SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *CO); 257f22ef01cSRoman Divacky ComplexPairTy VisitChooseExpr(ChooseExpr *CE); 258f22ef01cSRoman Divacky 259f22ef01cSRoman Divacky ComplexPairTy VisitInitListExpr(InitListExpr *E); 260f22ef01cSRoman Divacky 261f22ef01cSRoman Divacky ComplexPairTy VisitVAArgExpr(VAArgExpr *E); 262f22ef01cSRoman Divacky }; 263f22ef01cSRoman Divacky } // end anonymous namespace. 264f22ef01cSRoman Divacky 265f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 266f22ef01cSRoman Divacky // Utilities 267f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 268f22ef01cSRoman Divacky 269f22ef01cSRoman Divacky /// EmitLoadOfComplex - Given an RValue reference for a complex, emit code to 270f22ef01cSRoman Divacky /// load the real and imaginary pieces, returning them as Real/Imag. 271f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitLoadOfComplex(llvm::Value *SrcPtr, 272f22ef01cSRoman Divacky bool isVolatile) { 273f22ef01cSRoman Divacky llvm::Value *Real=0, *Imag=0; 274f22ef01cSRoman Divacky 2752754fe60SDimitry Andric if (!IgnoreReal || isVolatile) { 276f22ef01cSRoman Divacky llvm::Value *RealP = Builder.CreateStructGEP(SrcPtr, 0, 277f22ef01cSRoman Divacky SrcPtr->getName() + ".realp"); 278f22ef01cSRoman Divacky Real = Builder.CreateLoad(RealP, isVolatile, SrcPtr->getName() + ".real"); 279f22ef01cSRoman Divacky } 280f22ef01cSRoman Divacky 2812754fe60SDimitry Andric if (!IgnoreImag || isVolatile) { 282f22ef01cSRoman Divacky llvm::Value *ImagP = Builder.CreateStructGEP(SrcPtr, 1, 283f22ef01cSRoman Divacky SrcPtr->getName() + ".imagp"); 284f22ef01cSRoman Divacky Imag = Builder.CreateLoad(ImagP, isVolatile, SrcPtr->getName() + ".imag"); 285f22ef01cSRoman Divacky } 286f22ef01cSRoman Divacky return ComplexPairTy(Real, Imag); 287f22ef01cSRoman Divacky } 288f22ef01cSRoman Divacky 289f22ef01cSRoman Divacky /// EmitStoreOfComplex - Store the specified real/imag parts into the 290f22ef01cSRoman Divacky /// specified value pointer. 291f22ef01cSRoman Divacky void ComplexExprEmitter::EmitStoreOfComplex(ComplexPairTy Val, llvm::Value *Ptr, 292f22ef01cSRoman Divacky bool isVolatile) { 293f22ef01cSRoman Divacky llvm::Value *RealPtr = Builder.CreateStructGEP(Ptr, 0, "real"); 294f22ef01cSRoman Divacky llvm::Value *ImagPtr = Builder.CreateStructGEP(Ptr, 1, "imag"); 295f22ef01cSRoman Divacky 296f22ef01cSRoman Divacky Builder.CreateStore(Val.first, RealPtr, isVolatile); 297f22ef01cSRoman Divacky Builder.CreateStore(Val.second, ImagPtr, isVolatile); 298f22ef01cSRoman Divacky } 299f22ef01cSRoman Divacky 300f22ef01cSRoman Divacky 301f22ef01cSRoman Divacky 302f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 303f22ef01cSRoman Divacky // Visitor Methods 304f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 305f22ef01cSRoman Divacky 306f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitExpr(Expr *E) { 307f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex expression"); 308f22ef01cSRoman Divacky const llvm::Type *EltTy = 309f22ef01cSRoman Divacky CGF.ConvertType(E->getType()->getAs<ComplexType>()->getElementType()); 310f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy); 311f22ef01cSRoman Divacky return ComplexPairTy(U, U); 312f22ef01cSRoman Divacky } 313f22ef01cSRoman Divacky 314f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter:: 315f22ef01cSRoman Divacky VisitImaginaryLiteral(const ImaginaryLiteral *IL) { 316f22ef01cSRoman Divacky llvm::Value *Imag = CGF.EmitScalarExpr(IL->getSubExpr()); 3172754fe60SDimitry Andric return ComplexPairTy(llvm::Constant::getNullValue(Imag->getType()), Imag); 318f22ef01cSRoman Divacky } 319f22ef01cSRoman Divacky 320f22ef01cSRoman Divacky 321f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitCallExpr(const CallExpr *E) { 322f22ef01cSRoman Divacky if (E->getCallReturnType()->isReferenceType()) 323f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 324f22ef01cSRoman Divacky 325f22ef01cSRoman Divacky return CGF.EmitCallExpr(E).getComplexVal(); 326f22ef01cSRoman Divacky } 327f22ef01cSRoman Divacky 328f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitStmtExpr(const StmtExpr *E) { 3292754fe60SDimitry Andric CodeGenFunction::StmtExprEvaluation eval(CGF); 330f22ef01cSRoman Divacky return CGF.EmitCompoundStmt(*E->getSubStmt(), true).getComplexVal(); 331f22ef01cSRoman Divacky } 332f22ef01cSRoman Divacky 333f22ef01cSRoman Divacky /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType. 334f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitComplexToComplexCast(ComplexPairTy Val, 335f22ef01cSRoman Divacky QualType SrcType, 336f22ef01cSRoman Divacky QualType DestType) { 337f22ef01cSRoman Divacky // Get the src/dest element type. 338f22ef01cSRoman Divacky SrcType = SrcType->getAs<ComplexType>()->getElementType(); 339f22ef01cSRoman Divacky DestType = DestType->getAs<ComplexType>()->getElementType(); 340f22ef01cSRoman Divacky 341f22ef01cSRoman Divacky // C99 6.3.1.6: When a value of complex type is converted to another 342f22ef01cSRoman Divacky // complex type, both the real and imaginary parts follow the conversion 343f22ef01cSRoman Divacky // rules for the corresponding real types. 344f22ef01cSRoman Divacky Val.first = CGF.EmitScalarConversion(Val.first, SrcType, DestType); 345f22ef01cSRoman Divacky Val.second = CGF.EmitScalarConversion(Val.second, SrcType, DestType); 346f22ef01cSRoman Divacky return Val; 347f22ef01cSRoman Divacky } 348f22ef01cSRoman Divacky 349ffd1746dSEd Schouten ComplexPairTy ComplexExprEmitter::EmitCast(CastExpr::CastKind CK, Expr *Op, 350ffd1746dSEd Schouten QualType DestTy) { 3512754fe60SDimitry Andric switch (CK) { 3522754fe60SDimitry Andric case CK_GetObjCProperty: { 3532754fe60SDimitry Andric LValue LV = CGF.EmitLValue(Op); 3542754fe60SDimitry Andric assert(LV.isPropertyRef() && "Unknown LValue type!"); 3552754fe60SDimitry Andric return CGF.EmitLoadOfPropertyRefLValue(LV).getComplexVal(); 3562754fe60SDimitry Andric } 3572754fe60SDimitry Andric 3582754fe60SDimitry Andric case CK_NoOp: 3592754fe60SDimitry Andric case CK_LValueToRValue: 3602754fe60SDimitry Andric return Visit(Op); 3612754fe60SDimitry Andric 3622754fe60SDimitry Andric // TODO: do all of these 3632754fe60SDimitry Andric default: 3642754fe60SDimitry Andric break; 3652754fe60SDimitry Andric } 3662754fe60SDimitry Andric 367f22ef01cSRoman Divacky // Two cases here: cast from (complex to complex) and (scalar to complex). 368f22ef01cSRoman Divacky if (Op->getType()->isAnyComplexType()) 369f22ef01cSRoman Divacky return EmitComplexToComplexCast(Visit(Op), Op->getType(), DestTy); 370f22ef01cSRoman Divacky 371ffd1746dSEd Schouten // FIXME: We should be looking at all of the cast kinds here, not 372ffd1746dSEd Schouten // cherry-picking the ones we have test cases for. 373e580952dSDimitry Andric if (CK == CK_LValueBitCast) { 374ffd1746dSEd Schouten llvm::Value *V = CGF.EmitLValue(Op).getAddress(); 375ffd1746dSEd Schouten V = Builder.CreateBitCast(V, 376ffd1746dSEd Schouten CGF.ConvertType(CGF.getContext().getPointerType(DestTy))); 377ffd1746dSEd Schouten // FIXME: Are the qualifiers correct here? 378ffd1746dSEd Schouten return EmitLoadOfComplex(V, DestTy.isVolatileQualified()); 379ffd1746dSEd Schouten } 380ffd1746dSEd Schouten 381f22ef01cSRoman Divacky // C99 6.3.1.7: When a value of real type is converted to a complex type, the 382f22ef01cSRoman Divacky // real part of the complex result value is determined by the rules of 383f22ef01cSRoman Divacky // conversion to the corresponding real type and the imaginary part of the 384f22ef01cSRoman Divacky // complex result value is a positive zero or an unsigned zero. 385f22ef01cSRoman Divacky llvm::Value *Elt = CGF.EmitScalarExpr(Op); 386f22ef01cSRoman Divacky 387f22ef01cSRoman Divacky // Convert the input element to the element type of the complex. 388f22ef01cSRoman Divacky DestTy = DestTy->getAs<ComplexType>()->getElementType(); 389f22ef01cSRoman Divacky Elt = CGF.EmitScalarConversion(Elt, Op->getType(), DestTy); 390f22ef01cSRoman Divacky 391f22ef01cSRoman Divacky // Return (realval, 0). 392f22ef01cSRoman Divacky return ComplexPairTy(Elt, llvm::Constant::getNullValue(Elt->getType())); 393f22ef01cSRoman Divacky } 394f22ef01cSRoman Divacky 395f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryMinus(const UnaryOperator *E) { 396f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 397f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 398f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr()); 399f22ef01cSRoman Divacky 400f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 401f22ef01cSRoman Divacky if (Op.first->getType()->isFloatingPointTy()) { 402f22ef01cSRoman Divacky ResR = Builder.CreateFNeg(Op.first, "neg.r"); 403f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "neg.i"); 404f22ef01cSRoman Divacky } else { 405f22ef01cSRoman Divacky ResR = Builder.CreateNeg(Op.first, "neg.r"); 406f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "neg.i"); 407f22ef01cSRoman Divacky } 408f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 409f22ef01cSRoman Divacky } 410f22ef01cSRoman Divacky 411f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryNot(const UnaryOperator *E) { 412f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 413f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 414f22ef01cSRoman Divacky // ~(a+ib) = a + i*-b 415f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr()); 416f22ef01cSRoman Divacky llvm::Value *ResI; 417f22ef01cSRoman Divacky if (Op.second->getType()->isFloatingPointTy()) 418f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "conj.i"); 419f22ef01cSRoman Divacky else 420f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "conj.i"); 421f22ef01cSRoman Divacky 422f22ef01cSRoman Divacky return ComplexPairTy(Op.first, ResI); 423f22ef01cSRoman Divacky } 424f22ef01cSRoman Divacky 425f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinAdd(const BinOpInfo &Op) { 426f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 427f22ef01cSRoman Divacky 428f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 429f22ef01cSRoman Divacky ResR = Builder.CreateFAdd(Op.LHS.first, Op.RHS.first, "add.r"); 430f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(Op.LHS.second, Op.RHS.second, "add.i"); 431f22ef01cSRoman Divacky } else { 432f22ef01cSRoman Divacky ResR = Builder.CreateAdd(Op.LHS.first, Op.RHS.first, "add.r"); 433f22ef01cSRoman Divacky ResI = Builder.CreateAdd(Op.LHS.second, Op.RHS.second, "add.i"); 434f22ef01cSRoman Divacky } 435f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 436f22ef01cSRoman Divacky } 437f22ef01cSRoman Divacky 438f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinSub(const BinOpInfo &Op) { 439f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 440f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 441f22ef01cSRoman Divacky ResR = Builder.CreateFSub(Op.LHS.first, Op.RHS.first, "sub.r"); 442f22ef01cSRoman Divacky ResI = Builder.CreateFSub(Op.LHS.second, Op.RHS.second, "sub.i"); 443f22ef01cSRoman Divacky } else { 444f22ef01cSRoman Divacky ResR = Builder.CreateSub(Op.LHS.first, Op.RHS.first, "sub.r"); 445f22ef01cSRoman Divacky ResI = Builder.CreateSub(Op.LHS.second, Op.RHS.second, "sub.i"); 446f22ef01cSRoman Divacky } 447f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 448f22ef01cSRoman Divacky } 449f22ef01cSRoman Divacky 450f22ef01cSRoman Divacky 451f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinMul(const BinOpInfo &Op) { 452f22ef01cSRoman Divacky using llvm::Value; 453f22ef01cSRoman Divacky Value *ResR, *ResI; 454f22ef01cSRoman Divacky 455f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 456f22ef01cSRoman Divacky Value *ResRl = Builder.CreateFMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 457f22ef01cSRoman Divacky Value *ResRr = Builder.CreateFMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 458f22ef01cSRoman Divacky ResR = Builder.CreateFSub(ResRl, ResRr, "mul.r"); 459f22ef01cSRoman Divacky 460f22ef01cSRoman Divacky Value *ResIl = Builder.CreateFMul(Op.LHS.second, Op.RHS.first, "mul.il"); 461f22ef01cSRoman Divacky Value *ResIr = Builder.CreateFMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 462f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(ResIl, ResIr, "mul.i"); 463f22ef01cSRoman Divacky } else { 464f22ef01cSRoman Divacky Value *ResRl = Builder.CreateMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 465f22ef01cSRoman Divacky Value *ResRr = Builder.CreateMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 466f22ef01cSRoman Divacky ResR = Builder.CreateSub(ResRl, ResRr, "mul.r"); 467f22ef01cSRoman Divacky 468f22ef01cSRoman Divacky Value *ResIl = Builder.CreateMul(Op.LHS.second, Op.RHS.first, "mul.il"); 469f22ef01cSRoman Divacky Value *ResIr = Builder.CreateMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 470f22ef01cSRoman Divacky ResI = Builder.CreateAdd(ResIl, ResIr, "mul.i"); 471f22ef01cSRoman Divacky } 472f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 473f22ef01cSRoman Divacky } 474f22ef01cSRoman Divacky 475f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinDiv(const BinOpInfo &Op) { 476f22ef01cSRoman Divacky llvm::Value *LHSr = Op.LHS.first, *LHSi = Op.LHS.second; 477f22ef01cSRoman Divacky llvm::Value *RHSr = Op.RHS.first, *RHSi = Op.RHS.second; 478f22ef01cSRoman Divacky 479f22ef01cSRoman Divacky 480f22ef01cSRoman Divacky llvm::Value *DSTr, *DSTi; 481f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 482f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 483f22ef01cSRoman Divacky llvm::Value *Tmp1 = Builder.CreateFMul(LHSr, RHSr, "tmp"); // a*c 484f22ef01cSRoman Divacky llvm::Value *Tmp2 = Builder.CreateFMul(LHSi, RHSi, "tmp"); // b*d 485f22ef01cSRoman Divacky llvm::Value *Tmp3 = Builder.CreateFAdd(Tmp1, Tmp2, "tmp"); // ac+bd 486f22ef01cSRoman Divacky 487f22ef01cSRoman Divacky llvm::Value *Tmp4 = Builder.CreateFMul(RHSr, RHSr, "tmp"); // c*c 488f22ef01cSRoman Divacky llvm::Value *Tmp5 = Builder.CreateFMul(RHSi, RHSi, "tmp"); // d*d 489f22ef01cSRoman Divacky llvm::Value *Tmp6 = Builder.CreateFAdd(Tmp4, Tmp5, "tmp"); // cc+dd 490f22ef01cSRoman Divacky 491f22ef01cSRoman Divacky llvm::Value *Tmp7 = Builder.CreateFMul(LHSi, RHSr, "tmp"); // b*c 492f22ef01cSRoman Divacky llvm::Value *Tmp8 = Builder.CreateFMul(LHSr, RHSi, "tmp"); // a*d 493f22ef01cSRoman Divacky llvm::Value *Tmp9 = Builder.CreateFSub(Tmp7, Tmp8, "tmp"); // bc-ad 494f22ef01cSRoman Divacky 495f22ef01cSRoman Divacky DSTr = Builder.CreateFDiv(Tmp3, Tmp6, "tmp"); 496f22ef01cSRoman Divacky DSTi = Builder.CreateFDiv(Tmp9, Tmp6, "tmp"); 497f22ef01cSRoman Divacky } else { 498f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 499f22ef01cSRoman Divacky llvm::Value *Tmp1 = Builder.CreateMul(LHSr, RHSr, "tmp"); // a*c 500f22ef01cSRoman Divacky llvm::Value *Tmp2 = Builder.CreateMul(LHSi, RHSi, "tmp"); // b*d 501f22ef01cSRoman Divacky llvm::Value *Tmp3 = Builder.CreateAdd(Tmp1, Tmp2, "tmp"); // ac+bd 502f22ef01cSRoman Divacky 503f22ef01cSRoman Divacky llvm::Value *Tmp4 = Builder.CreateMul(RHSr, RHSr, "tmp"); // c*c 504f22ef01cSRoman Divacky llvm::Value *Tmp5 = Builder.CreateMul(RHSi, RHSi, "tmp"); // d*d 505f22ef01cSRoman Divacky llvm::Value *Tmp6 = Builder.CreateAdd(Tmp4, Tmp5, "tmp"); // cc+dd 506f22ef01cSRoman Divacky 507f22ef01cSRoman Divacky llvm::Value *Tmp7 = Builder.CreateMul(LHSi, RHSr, "tmp"); // b*c 508f22ef01cSRoman Divacky llvm::Value *Tmp8 = Builder.CreateMul(LHSr, RHSi, "tmp"); // a*d 509f22ef01cSRoman Divacky llvm::Value *Tmp9 = Builder.CreateSub(Tmp7, Tmp8, "tmp"); // bc-ad 510f22ef01cSRoman Divacky 511f22ef01cSRoman Divacky if (Op.Ty->getAs<ComplexType>()->getElementType()->isUnsignedIntegerType()) { 512f22ef01cSRoman Divacky DSTr = Builder.CreateUDiv(Tmp3, Tmp6, "tmp"); 513f22ef01cSRoman Divacky DSTi = Builder.CreateUDiv(Tmp9, Tmp6, "tmp"); 514f22ef01cSRoman Divacky } else { 515f22ef01cSRoman Divacky DSTr = Builder.CreateSDiv(Tmp3, Tmp6, "tmp"); 516f22ef01cSRoman Divacky DSTi = Builder.CreateSDiv(Tmp9, Tmp6, "tmp"); 517f22ef01cSRoman Divacky } 518f22ef01cSRoman Divacky } 519f22ef01cSRoman Divacky 520f22ef01cSRoman Divacky return ComplexPairTy(DSTr, DSTi); 521f22ef01cSRoman Divacky } 522f22ef01cSRoman Divacky 523f22ef01cSRoman Divacky ComplexExprEmitter::BinOpInfo 524f22ef01cSRoman Divacky ComplexExprEmitter::EmitBinOps(const BinaryOperator *E) { 525f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 526f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 527f22ef01cSRoman Divacky BinOpInfo Ops; 528f22ef01cSRoman Divacky Ops.LHS = Visit(E->getLHS()); 529f22ef01cSRoman Divacky Ops.RHS = Visit(E->getRHS()); 530f22ef01cSRoman Divacky Ops.Ty = E->getType(); 531f22ef01cSRoman Divacky return Ops; 532f22ef01cSRoman Divacky } 533f22ef01cSRoman Divacky 534f22ef01cSRoman Divacky 5352754fe60SDimitry Andric LValue ComplexExprEmitter:: 5362754fe60SDimitry Andric EmitCompoundAssignLValue(const CompoundAssignOperator *E, 5372754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&), 5382754fe60SDimitry Andric ComplexPairTy &Val) { 539f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 540f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 5412754fe60SDimitry Andric QualType LHSTy = E->getLHS()->getType(); 542f22ef01cSRoman Divacky 543f22ef01cSRoman Divacky BinOpInfo OpInfo; 544f22ef01cSRoman Divacky 545f22ef01cSRoman Divacky // Load the RHS and LHS operands. 546f22ef01cSRoman Divacky // __block variables need to have the rhs evaluated first, plus this should 5472754fe60SDimitry Andric // improve codegen a little. 548f22ef01cSRoman Divacky OpInfo.Ty = E->getComputationResultType(); 5492754fe60SDimitry Andric 5502754fe60SDimitry Andric // The RHS should have been converted to the computation type. 5512754fe60SDimitry Andric assert(OpInfo.Ty->isAnyComplexType()); 5522754fe60SDimitry Andric assert(CGF.getContext().hasSameUnqualifiedType(OpInfo.Ty, 5532754fe60SDimitry Andric E->getRHS()->getType())); 5542754fe60SDimitry Andric OpInfo.RHS = Visit(E->getRHS()); 555f22ef01cSRoman Divacky 556ffd1746dSEd Schouten LValue LHS = CGF.EmitLValue(E->getLHS()); 5572754fe60SDimitry Andric 5582754fe60SDimitry Andric // Load from the l-value. 5592754fe60SDimitry Andric ComplexPairTy LHSComplexPair = EmitLoadOfLValue(LHS); 560f22ef01cSRoman Divacky 561f22ef01cSRoman Divacky OpInfo.LHS = EmitComplexToComplexCast(LHSComplexPair, LHSTy, OpInfo.Ty); 562f22ef01cSRoman Divacky 563f22ef01cSRoman Divacky // Expand the binary operator. 564f22ef01cSRoman Divacky ComplexPairTy Result = (this->*Func)(OpInfo); 565f22ef01cSRoman Divacky 566f22ef01cSRoman Divacky // Truncate the result back to the LHS type. 567f22ef01cSRoman Divacky Result = EmitComplexToComplexCast(Result, OpInfo.Ty, LHSTy); 5682754fe60SDimitry Andric Val = Result; 569f22ef01cSRoman Divacky 570f22ef01cSRoman Divacky // Store the result value into the LHS lvalue. 5712754fe60SDimitry Andric EmitStoreThroughLValue(Result, LHS); 572ffd1746dSEd Schouten 5732754fe60SDimitry Andric return LHS; 574f22ef01cSRoman Divacky } 575f22ef01cSRoman Divacky 5762754fe60SDimitry Andric // Compound assignments. 5772754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter:: 5782754fe60SDimitry Andric EmitCompoundAssign(const CompoundAssignOperator *E, 5792754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&)){ 5802754fe60SDimitry Andric ComplexPairTy Val; 5812754fe60SDimitry Andric LValue LV = EmitCompoundAssignLValue(E, Func, Val); 5822754fe60SDimitry Andric 5832754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value. 5842754fe60SDimitry Andric if (!CGF.getContext().getLangOptions().CPlusPlus) 5852754fe60SDimitry Andric return Val; 5862754fe60SDimitry Andric 5872754fe60SDimitry Andric // Objective-C property assignment never reloads the value following a store. 5882754fe60SDimitry Andric if (LV.isPropertyRef()) 5892754fe60SDimitry Andric return Val; 5902754fe60SDimitry Andric 5912754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 5922754fe60SDimitry Andric if (!LV.isVolatileQualified()) 5932754fe60SDimitry Andric return Val; 5942754fe60SDimitry Andric 5952754fe60SDimitry Andric return EmitLoadOfComplex(LV.getAddress(), LV.isVolatileQualified()); 5962754fe60SDimitry Andric } 5972754fe60SDimitry Andric 5982754fe60SDimitry Andric LValue ComplexExprEmitter::EmitBinAssignLValue(const BinaryOperator *E, 5992754fe60SDimitry Andric ComplexPairTy &Val) { 600ffd1746dSEd Schouten assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(), 601ffd1746dSEd Schouten E->getRHS()->getType()) && 602f22ef01cSRoman Divacky "Invalid assignment"); 6032754fe60SDimitry Andric TestAndClearIgnoreReal(); 6042754fe60SDimitry Andric TestAndClearIgnoreImag(); 6052754fe60SDimitry Andric 6062754fe60SDimitry Andric // Emit the RHS. __block variables need the RHS evaluated first. 6072754fe60SDimitry Andric Val = Visit(E->getRHS()); 608f22ef01cSRoman Divacky 609f22ef01cSRoman Divacky // Compute the address to store into. 610f22ef01cSRoman Divacky LValue LHS = CGF.EmitLValue(E->getLHS()); 611f22ef01cSRoman Divacky 612ffd1746dSEd Schouten // Store the result value into the LHS lvalue. 6132754fe60SDimitry Andric EmitStoreThroughLValue(Val, LHS); 614f22ef01cSRoman Divacky 6152754fe60SDimitry Andric return LHS; 6162754fe60SDimitry Andric } 617ffd1746dSEd Schouten 6182754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitBinAssign(const BinaryOperator *E) { 6192754fe60SDimitry Andric ComplexPairTy Val; 6202754fe60SDimitry Andric LValue LV = EmitBinAssignLValue(E, Val); 6212754fe60SDimitry Andric 6222754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value. 6232754fe60SDimitry Andric if (!CGF.getContext().getLangOptions().CPlusPlus) 624f22ef01cSRoman Divacky return Val; 625ffd1746dSEd Schouten 6262754fe60SDimitry Andric // Objective-C property assignment never reloads the value following a store. 6272754fe60SDimitry Andric if (LV.isPropertyRef()) 6282754fe60SDimitry Andric return Val; 6292754fe60SDimitry Andric 6302754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 6312754fe60SDimitry Andric if (!LV.isVolatileQualified()) 6322754fe60SDimitry Andric return Val; 6332754fe60SDimitry Andric 6342754fe60SDimitry Andric return EmitLoadOfComplex(LV.getAddress(), LV.isVolatileQualified()); 635f22ef01cSRoman Divacky } 636f22ef01cSRoman Divacky 637f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitBinComma(const BinaryOperator *E) { 6382754fe60SDimitry Andric CGF.EmitIgnoredExpr(E->getLHS()); 639f22ef01cSRoman Divacky return Visit(E->getRHS()); 640f22ef01cSRoman Divacky } 641f22ef01cSRoman Divacky 642f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter:: 6432754fe60SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) { 644f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 645f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 646f22ef01cSRoman Divacky llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true"); 647f22ef01cSRoman Divacky llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false"); 648f22ef01cSRoman Divacky llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end"); 649f22ef01cSRoman Divacky 6502754fe60SDimitry Andric // Bind the common expression if necessary. 6512754fe60SDimitry Andric CodeGenFunction::OpaqueValueMapping binding(CGF, E); 6522754fe60SDimitry Andric 6532754fe60SDimitry Andric CodeGenFunction::ConditionalEvaluation eval(CGF); 654f22ef01cSRoman Divacky CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock); 655f22ef01cSRoman Divacky 6562754fe60SDimitry Andric eval.begin(CGF); 657f22ef01cSRoman Divacky CGF.EmitBlock(LHSBlock); 6582754fe60SDimitry Andric ComplexPairTy LHS = Visit(E->getTrueExpr()); 659f22ef01cSRoman Divacky LHSBlock = Builder.GetInsertBlock(); 660f22ef01cSRoman Divacky CGF.EmitBranch(ContBlock); 6612754fe60SDimitry Andric eval.end(CGF); 662f22ef01cSRoman Divacky 6632754fe60SDimitry Andric eval.begin(CGF); 664f22ef01cSRoman Divacky CGF.EmitBlock(RHSBlock); 6652754fe60SDimitry Andric ComplexPairTy RHS = Visit(E->getFalseExpr()); 666f22ef01cSRoman Divacky RHSBlock = Builder.GetInsertBlock(); 667f22ef01cSRoman Divacky CGF.EmitBlock(ContBlock); 6682754fe60SDimitry Andric eval.end(CGF); 669f22ef01cSRoman Divacky 670f22ef01cSRoman Divacky // Create a PHI node for the real part. 671f22ef01cSRoman Divacky llvm::PHINode *RealPN = Builder.CreatePHI(LHS.first->getType(), "cond.r"); 672f22ef01cSRoman Divacky RealPN->reserveOperandSpace(2); 673f22ef01cSRoman Divacky RealPN->addIncoming(LHS.first, LHSBlock); 674f22ef01cSRoman Divacky RealPN->addIncoming(RHS.first, RHSBlock); 675f22ef01cSRoman Divacky 676f22ef01cSRoman Divacky // Create a PHI node for the imaginary part. 677f22ef01cSRoman Divacky llvm::PHINode *ImagPN = Builder.CreatePHI(LHS.first->getType(), "cond.i"); 678f22ef01cSRoman Divacky ImagPN->reserveOperandSpace(2); 679f22ef01cSRoman Divacky ImagPN->addIncoming(LHS.second, LHSBlock); 680f22ef01cSRoman Divacky ImagPN->addIncoming(RHS.second, RHSBlock); 681f22ef01cSRoman Divacky 682f22ef01cSRoman Divacky return ComplexPairTy(RealPN, ImagPN); 683f22ef01cSRoman Divacky } 684f22ef01cSRoman Divacky 685f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitChooseExpr(ChooseExpr *E) { 686f22ef01cSRoman Divacky return Visit(E->getChosenSubExpr(CGF.getContext())); 687f22ef01cSRoman Divacky } 688f22ef01cSRoman Divacky 689f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitInitListExpr(InitListExpr *E) { 690f22ef01cSRoman Divacky bool Ignore = TestAndClearIgnoreReal(); 691f22ef01cSRoman Divacky (void)Ignore; 692f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored"); 693f22ef01cSRoman Divacky Ignore = TestAndClearIgnoreImag(); 694f22ef01cSRoman Divacky (void)Ignore; 695f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored"); 696f22ef01cSRoman Divacky if (E->getNumInits()) 697f22ef01cSRoman Divacky return Visit(E->getInit(0)); 698f22ef01cSRoman Divacky 699f22ef01cSRoman Divacky // Empty init list intializes to null 700f22ef01cSRoman Divacky QualType Ty = E->getType()->getAs<ComplexType>()->getElementType(); 701f22ef01cSRoman Divacky const llvm::Type* LTy = CGF.ConvertType(Ty); 702f22ef01cSRoman Divacky llvm::Value* zeroConstant = llvm::Constant::getNullValue(LTy); 703f22ef01cSRoman Divacky return ComplexPairTy(zeroConstant, zeroConstant); 704f22ef01cSRoman Divacky } 705f22ef01cSRoman Divacky 706f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitVAArgExpr(VAArgExpr *E) { 707f22ef01cSRoman Divacky llvm::Value *ArgValue = CGF.EmitVAListRef(E->getSubExpr()); 708f22ef01cSRoman Divacky llvm::Value *ArgPtr = CGF.EmitVAArg(ArgValue, E->getType()); 709f22ef01cSRoman Divacky 710f22ef01cSRoman Divacky if (!ArgPtr) { 711f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex va_arg expression"); 712f22ef01cSRoman Divacky const llvm::Type *EltTy = 713f22ef01cSRoman Divacky CGF.ConvertType(E->getType()->getAs<ComplexType>()->getElementType()); 714f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy); 715f22ef01cSRoman Divacky return ComplexPairTy(U, U); 716f22ef01cSRoman Divacky } 717f22ef01cSRoman Divacky 718f22ef01cSRoman Divacky // FIXME Volatility. 719f22ef01cSRoman Divacky return EmitLoadOfComplex(ArgPtr, false); 720f22ef01cSRoman Divacky } 721f22ef01cSRoman Divacky 722f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 723f22ef01cSRoman Divacky // Entry Point into this File 724f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 725f22ef01cSRoman Divacky 726f22ef01cSRoman Divacky /// EmitComplexExpr - Emit the computation of the specified expression of 727f22ef01cSRoman Divacky /// complex type, ignoring the result. 728f22ef01cSRoman Divacky ComplexPairTy CodeGenFunction::EmitComplexExpr(const Expr *E, bool IgnoreReal, 7292754fe60SDimitry Andric bool IgnoreImag) { 730f22ef01cSRoman Divacky assert(E && E->getType()->isAnyComplexType() && 731f22ef01cSRoman Divacky "Invalid complex expression to emit"); 732f22ef01cSRoman Divacky 7332754fe60SDimitry Andric return ComplexExprEmitter(*this, IgnoreReal, IgnoreImag) 734f22ef01cSRoman Divacky .Visit(const_cast<Expr*>(E)); 735f22ef01cSRoman Divacky } 736f22ef01cSRoman Divacky 737f22ef01cSRoman Divacky /// EmitComplexExprIntoAddr - Emit the computation of the specified expression 738f22ef01cSRoman Divacky /// of complex type, storing into the specified Value*. 739f22ef01cSRoman Divacky void CodeGenFunction::EmitComplexExprIntoAddr(const Expr *E, 740f22ef01cSRoman Divacky llvm::Value *DestAddr, 741f22ef01cSRoman Divacky bool DestIsVolatile) { 742f22ef01cSRoman Divacky assert(E && E->getType()->isAnyComplexType() && 743f22ef01cSRoman Divacky "Invalid complex expression to emit"); 744f22ef01cSRoman Divacky ComplexExprEmitter Emitter(*this); 745f22ef01cSRoman Divacky ComplexPairTy Val = Emitter.Visit(const_cast<Expr*>(E)); 746f22ef01cSRoman Divacky Emitter.EmitStoreOfComplex(Val, DestAddr, DestIsVolatile); 747f22ef01cSRoman Divacky } 748f22ef01cSRoman Divacky 749f22ef01cSRoman Divacky /// StoreComplexToAddr - Store a complex number into the specified address. 750f22ef01cSRoman Divacky void CodeGenFunction::StoreComplexToAddr(ComplexPairTy V, 751f22ef01cSRoman Divacky llvm::Value *DestAddr, 752f22ef01cSRoman Divacky bool DestIsVolatile) { 753f22ef01cSRoman Divacky ComplexExprEmitter(*this).EmitStoreOfComplex(V, DestAddr, DestIsVolatile); 754f22ef01cSRoman Divacky } 755f22ef01cSRoman Divacky 756f22ef01cSRoman Divacky /// LoadComplexFromAddr - Load a complex number from the specified address. 757f22ef01cSRoman Divacky ComplexPairTy CodeGenFunction::LoadComplexFromAddr(llvm::Value *SrcAddr, 758f22ef01cSRoman Divacky bool SrcIsVolatile) { 759f22ef01cSRoman Divacky return ComplexExprEmitter(*this).EmitLoadOfComplex(SrcAddr, SrcIsVolatile); 760f22ef01cSRoman Divacky } 7612754fe60SDimitry Andric 7622754fe60SDimitry Andric LValue CodeGenFunction::EmitComplexAssignmentLValue(const BinaryOperator *E) { 7632754fe60SDimitry Andric assert(E->getOpcode() == BO_Assign); 7642754fe60SDimitry Andric ComplexPairTy Val; // ignored 7652754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitBinAssignLValue(E, Val); 7662754fe60SDimitry Andric } 7672754fe60SDimitry Andric 7682754fe60SDimitry Andric LValue CodeGenFunction:: 7692754fe60SDimitry Andric EmitComplexCompoundAssignmentLValue(const CompoundAssignOperator *E) { 7702754fe60SDimitry Andric ComplexPairTy(ComplexExprEmitter::*Op)(const ComplexExprEmitter::BinOpInfo &); 7712754fe60SDimitry Andric switch (E->getOpcode()) { 7722754fe60SDimitry Andric case BO_MulAssign: Op = &ComplexExprEmitter::EmitBinMul; break; 7732754fe60SDimitry Andric case BO_DivAssign: Op = &ComplexExprEmitter::EmitBinDiv; break; 7742754fe60SDimitry Andric case BO_SubAssign: Op = &ComplexExprEmitter::EmitBinSub; break; 7752754fe60SDimitry Andric case BO_AddAssign: Op = &ComplexExprEmitter::EmitBinAdd; break; 7762754fe60SDimitry Andric 7772754fe60SDimitry Andric default: 7782754fe60SDimitry Andric llvm_unreachable("unexpected complex compound assignment"); 7792754fe60SDimitry Andric Op = 0; 7802754fe60SDimitry Andric } 7812754fe60SDimitry Andric 7822754fe60SDimitry Andric ComplexPairTy Val; // ignored 7832754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val); 7842754fe60SDimitry Andric } 785