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()); 1066122f3e6SDimitry Andric llvm_unreachable("Stmt can't have complex result type!"); 107f22ef01cSRoman Divacky } 108f22ef01cSRoman Divacky ComplexPairTy VisitExpr(Expr *S); 109f22ef01cSRoman Divacky ComplexPairTy VisitParenExpr(ParenExpr *PE) { return Visit(PE->getSubExpr());} 1103b0f4066SDimitry Andric ComplexPairTy VisitGenericSelectionExpr(GenericSelectionExpr *GE) { 1113b0f4066SDimitry Andric return Visit(GE->getResultExpr()); 1123b0f4066SDimitry Andric } 113f22ef01cSRoman Divacky ComplexPairTy VisitImaginaryLiteral(const ImaginaryLiteral *IL); 11417a519f9SDimitry Andric ComplexPairTy 11517a519f9SDimitry Andric VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *PE) { 11617a519f9SDimitry Andric return Visit(PE->getReplacement()); 11717a519f9SDimitry Andric } 118f22ef01cSRoman Divacky 119f22ef01cSRoman Divacky // l-values. 120f22ef01cSRoman Divacky ComplexPairTy VisitDeclRefExpr(const Expr *E) { return EmitLoadOfLValue(E); } 1216122f3e6SDimitry Andric ComplexPairTy VisitBlockDeclRefExpr(const Expr *E) { return EmitLoadOfLValue(E); } 122f22ef01cSRoman Divacky ComplexPairTy VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) { 123f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 124f22ef01cSRoman Divacky } 125f22ef01cSRoman Divacky ComplexPairTy VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) { 1262754fe60SDimitry Andric assert(E->getObjectKind() == OK_Ordinary); 127f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 128f22ef01cSRoman Divacky } 129f22ef01cSRoman Divacky ComplexPairTy VisitObjCMessageExpr(ObjCMessageExpr *E) { 130f22ef01cSRoman Divacky return CGF.EmitObjCMessageExpr(E).getComplexVal(); 131f22ef01cSRoman Divacky } 132f22ef01cSRoman Divacky ComplexPairTy VisitArraySubscriptExpr(Expr *E) { return EmitLoadOfLValue(E); } 133f22ef01cSRoman Divacky ComplexPairTy VisitMemberExpr(const Expr *E) { return EmitLoadOfLValue(E); } 1342754fe60SDimitry Andric ComplexPairTy VisitOpaqueValueExpr(OpaqueValueExpr *E) { 1352754fe60SDimitry Andric if (E->isGLValue()) 1362754fe60SDimitry Andric return EmitLoadOfLValue(CGF.getOpaqueLValueMapping(E)); 1372754fe60SDimitry Andric return CGF.getOpaqueRValueMapping(E).getComplexVal(); 1382754fe60SDimitry Andric } 139f22ef01cSRoman Divacky 140f22ef01cSRoman Divacky // FIXME: CompoundLiteralExpr 141f22ef01cSRoman Divacky 142ffd1746dSEd Schouten ComplexPairTy EmitCast(CastExpr::CastKind CK, Expr *Op, QualType DestTy); 143f22ef01cSRoman Divacky ComplexPairTy VisitImplicitCastExpr(ImplicitCastExpr *E) { 144f22ef01cSRoman Divacky // Unlike for scalars, we don't have to worry about function->ptr demotion 145f22ef01cSRoman Divacky // here. 146ffd1746dSEd Schouten return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType()); 147f22ef01cSRoman Divacky } 148f22ef01cSRoman Divacky ComplexPairTy VisitCastExpr(CastExpr *E) { 149ffd1746dSEd Schouten return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType()); 150f22ef01cSRoman Divacky } 151f22ef01cSRoman Divacky ComplexPairTy VisitCallExpr(const CallExpr *E); 152f22ef01cSRoman Divacky ComplexPairTy VisitStmtExpr(const StmtExpr *E); 153f22ef01cSRoman Divacky 154f22ef01cSRoman Divacky // Operators. 155f22ef01cSRoman Divacky ComplexPairTy VisitPrePostIncDec(const UnaryOperator *E, 156f22ef01cSRoman Divacky bool isInc, bool isPre) { 157f22ef01cSRoman Divacky LValue LV = CGF.EmitLValue(E->getSubExpr()); 158f22ef01cSRoman Divacky return CGF.EmitComplexPrePostIncDec(E, LV, isInc, isPre); 159f22ef01cSRoman Divacky } 160f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPostDec(const UnaryOperator *E) { 161f22ef01cSRoman Divacky return VisitPrePostIncDec(E, false, false); 162f22ef01cSRoman Divacky } 163f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPostInc(const UnaryOperator *E) { 164f22ef01cSRoman Divacky return VisitPrePostIncDec(E, true, false); 165f22ef01cSRoman Divacky } 166f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPreDec(const UnaryOperator *E) { 167f22ef01cSRoman Divacky return VisitPrePostIncDec(E, false, true); 168f22ef01cSRoman Divacky } 169f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPreInc(const UnaryOperator *E) { 170f22ef01cSRoman Divacky return VisitPrePostIncDec(E, true, true); 171f22ef01cSRoman Divacky } 172f22ef01cSRoman Divacky ComplexPairTy VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); } 173f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPlus (const UnaryOperator *E) { 174f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 175f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 176f22ef01cSRoman Divacky return Visit(E->getSubExpr()); 177f22ef01cSRoman Divacky } 178f22ef01cSRoman Divacky ComplexPairTy VisitUnaryMinus (const UnaryOperator *E); 179f22ef01cSRoman Divacky ComplexPairTy VisitUnaryNot (const UnaryOperator *E); 180f22ef01cSRoman Divacky // LNot,Real,Imag never return complex. 181f22ef01cSRoman Divacky ComplexPairTy VisitUnaryExtension(const UnaryOperator *E) { 182f22ef01cSRoman Divacky return Visit(E->getSubExpr()); 183f22ef01cSRoman Divacky } 184f22ef01cSRoman Divacky ComplexPairTy VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) { 185f22ef01cSRoman Divacky return Visit(DAE->getExpr()); 186f22ef01cSRoman Divacky } 1872754fe60SDimitry Andric ComplexPairTy VisitExprWithCleanups(ExprWithCleanups *E) { 1882754fe60SDimitry Andric return CGF.EmitExprWithCleanups(E).getComplexVal(); 189f22ef01cSRoman Divacky } 190ffd1746dSEd Schouten ComplexPairTy VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) { 191f22ef01cSRoman Divacky assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 192f22ef01cSRoman Divacky QualType Elem = E->getType()->getAs<ComplexType>()->getElementType(); 193f22ef01cSRoman Divacky llvm::Constant *Null = llvm::Constant::getNullValue(CGF.ConvertType(Elem)); 194f22ef01cSRoman Divacky return ComplexPairTy(Null, Null); 195f22ef01cSRoman Divacky } 196f22ef01cSRoman Divacky ComplexPairTy VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) { 197f22ef01cSRoman Divacky assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 198f22ef01cSRoman Divacky QualType Elem = E->getType()->getAs<ComplexType>()->getElementType(); 199f22ef01cSRoman Divacky llvm::Constant *Null = 200f22ef01cSRoman Divacky llvm::Constant::getNullValue(CGF.ConvertType(Elem)); 201f22ef01cSRoman Divacky return ComplexPairTy(Null, Null); 202f22ef01cSRoman Divacky } 203f22ef01cSRoman Divacky 204f22ef01cSRoman Divacky struct BinOpInfo { 205f22ef01cSRoman Divacky ComplexPairTy LHS; 206f22ef01cSRoman Divacky ComplexPairTy RHS; 207f22ef01cSRoman Divacky QualType Ty; // Computation Type. 208f22ef01cSRoman Divacky }; 209f22ef01cSRoman Divacky 210f22ef01cSRoman Divacky BinOpInfo EmitBinOps(const BinaryOperator *E); 2112754fe60SDimitry Andric LValue EmitCompoundAssignLValue(const CompoundAssignOperator *E, 2122754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func) 2132754fe60SDimitry Andric (const BinOpInfo &), 2142754fe60SDimitry Andric ComplexPairTy &Val); 215f22ef01cSRoman Divacky ComplexPairTy EmitCompoundAssign(const CompoundAssignOperator *E, 216f22ef01cSRoman Divacky ComplexPairTy (ComplexExprEmitter::*Func) 217f22ef01cSRoman Divacky (const BinOpInfo &)); 218f22ef01cSRoman Divacky 219f22ef01cSRoman Divacky ComplexPairTy EmitBinAdd(const BinOpInfo &Op); 220f22ef01cSRoman Divacky ComplexPairTy EmitBinSub(const BinOpInfo &Op); 221f22ef01cSRoman Divacky ComplexPairTy EmitBinMul(const BinOpInfo &Op); 222f22ef01cSRoman Divacky ComplexPairTy EmitBinDiv(const BinOpInfo &Op); 223f22ef01cSRoman Divacky 224f22ef01cSRoman Divacky ComplexPairTy VisitBinAdd(const BinaryOperator *E) { 225f22ef01cSRoman Divacky return EmitBinAdd(EmitBinOps(E)); 226f22ef01cSRoman Divacky } 227f22ef01cSRoman Divacky ComplexPairTy VisitBinSub(const BinaryOperator *E) { 228f22ef01cSRoman Divacky return EmitBinSub(EmitBinOps(E)); 229f22ef01cSRoman Divacky } 2302754fe60SDimitry Andric ComplexPairTy VisitBinMul(const BinaryOperator *E) { 2312754fe60SDimitry Andric return EmitBinMul(EmitBinOps(E)); 2322754fe60SDimitry Andric } 233f22ef01cSRoman Divacky ComplexPairTy VisitBinDiv(const BinaryOperator *E) { 234f22ef01cSRoman Divacky return EmitBinDiv(EmitBinOps(E)); 235f22ef01cSRoman Divacky } 236f22ef01cSRoman Divacky 237f22ef01cSRoman Divacky // Compound assignments. 238f22ef01cSRoman Divacky ComplexPairTy VisitBinAddAssign(const CompoundAssignOperator *E) { 239f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinAdd); 240f22ef01cSRoman Divacky } 241f22ef01cSRoman Divacky ComplexPairTy VisitBinSubAssign(const CompoundAssignOperator *E) { 242f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinSub); 243f22ef01cSRoman Divacky } 244f22ef01cSRoman Divacky ComplexPairTy VisitBinMulAssign(const CompoundAssignOperator *E) { 245f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinMul); 246f22ef01cSRoman Divacky } 247f22ef01cSRoman Divacky ComplexPairTy VisitBinDivAssign(const CompoundAssignOperator *E) { 248f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinDiv); 249f22ef01cSRoman Divacky } 250f22ef01cSRoman Divacky 251f22ef01cSRoman Divacky // GCC rejects rem/and/or/xor for integer complex. 252f22ef01cSRoman Divacky // Logical and/or always return int, never complex. 253f22ef01cSRoman Divacky 254f22ef01cSRoman Divacky // No comparisons produce a complex result. 2552754fe60SDimitry Andric 2562754fe60SDimitry Andric LValue EmitBinAssignLValue(const BinaryOperator *E, 2572754fe60SDimitry Andric ComplexPairTy &Val); 258f22ef01cSRoman Divacky ComplexPairTy VisitBinAssign (const BinaryOperator *E); 259f22ef01cSRoman Divacky ComplexPairTy VisitBinComma (const BinaryOperator *E); 260f22ef01cSRoman Divacky 261f22ef01cSRoman Divacky 2622754fe60SDimitry Andric ComplexPairTy 2632754fe60SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *CO); 264f22ef01cSRoman Divacky ComplexPairTy VisitChooseExpr(ChooseExpr *CE); 265f22ef01cSRoman Divacky 266f22ef01cSRoman Divacky ComplexPairTy VisitInitListExpr(InitListExpr *E); 267f22ef01cSRoman Divacky 268f22ef01cSRoman Divacky ComplexPairTy VisitVAArgExpr(VAArgExpr *E); 2696122f3e6SDimitry Andric 2706122f3e6SDimitry Andric ComplexPairTy VisitAtomicExpr(AtomicExpr *E) { 2716122f3e6SDimitry Andric return CGF.EmitAtomicExpr(E).getComplexVal(); 2726122f3e6SDimitry Andric } 273f22ef01cSRoman Divacky }; 274f22ef01cSRoman Divacky } // end anonymous namespace. 275f22ef01cSRoman Divacky 276f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 277f22ef01cSRoman Divacky // Utilities 278f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 279f22ef01cSRoman Divacky 280f22ef01cSRoman Divacky /// EmitLoadOfComplex - Given an RValue reference for a complex, emit code to 281f22ef01cSRoman Divacky /// load the real and imaginary pieces, returning them as Real/Imag. 282f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitLoadOfComplex(llvm::Value *SrcPtr, 283f22ef01cSRoman Divacky bool isVolatile) { 284f22ef01cSRoman Divacky llvm::Value *Real=0, *Imag=0; 285f22ef01cSRoman Divacky 2862754fe60SDimitry Andric if (!IgnoreReal || isVolatile) { 287f22ef01cSRoman Divacky llvm::Value *RealP = Builder.CreateStructGEP(SrcPtr, 0, 288f22ef01cSRoman Divacky SrcPtr->getName() + ".realp"); 289f22ef01cSRoman Divacky Real = Builder.CreateLoad(RealP, isVolatile, SrcPtr->getName() + ".real"); 290f22ef01cSRoman Divacky } 291f22ef01cSRoman Divacky 2922754fe60SDimitry Andric if (!IgnoreImag || isVolatile) { 293f22ef01cSRoman Divacky llvm::Value *ImagP = Builder.CreateStructGEP(SrcPtr, 1, 294f22ef01cSRoman Divacky SrcPtr->getName() + ".imagp"); 295f22ef01cSRoman Divacky Imag = Builder.CreateLoad(ImagP, isVolatile, SrcPtr->getName() + ".imag"); 296f22ef01cSRoman Divacky } 297f22ef01cSRoman Divacky return ComplexPairTy(Real, Imag); 298f22ef01cSRoman Divacky } 299f22ef01cSRoman Divacky 300f22ef01cSRoman Divacky /// EmitStoreOfComplex - Store the specified real/imag parts into the 301f22ef01cSRoman Divacky /// specified value pointer. 302f22ef01cSRoman Divacky void ComplexExprEmitter::EmitStoreOfComplex(ComplexPairTy Val, llvm::Value *Ptr, 303f22ef01cSRoman Divacky bool isVolatile) { 304f22ef01cSRoman Divacky llvm::Value *RealPtr = Builder.CreateStructGEP(Ptr, 0, "real"); 305f22ef01cSRoman Divacky llvm::Value *ImagPtr = Builder.CreateStructGEP(Ptr, 1, "imag"); 306f22ef01cSRoman Divacky 307f22ef01cSRoman Divacky Builder.CreateStore(Val.first, RealPtr, isVolatile); 308f22ef01cSRoman Divacky Builder.CreateStore(Val.second, ImagPtr, isVolatile); 309f22ef01cSRoman Divacky } 310f22ef01cSRoman Divacky 311f22ef01cSRoman Divacky 312f22ef01cSRoman Divacky 313f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 314f22ef01cSRoman Divacky // Visitor Methods 315f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 316f22ef01cSRoman Divacky 317f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitExpr(Expr *E) { 318f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex expression"); 3196122f3e6SDimitry Andric llvm::Type *EltTy = 320f22ef01cSRoman Divacky CGF.ConvertType(E->getType()->getAs<ComplexType>()->getElementType()); 321f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy); 322f22ef01cSRoman Divacky return ComplexPairTy(U, U); 323f22ef01cSRoman Divacky } 324f22ef01cSRoman Divacky 325f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter:: 326f22ef01cSRoman Divacky VisitImaginaryLiteral(const ImaginaryLiteral *IL) { 327f22ef01cSRoman Divacky llvm::Value *Imag = CGF.EmitScalarExpr(IL->getSubExpr()); 3282754fe60SDimitry Andric return ComplexPairTy(llvm::Constant::getNullValue(Imag->getType()), Imag); 329f22ef01cSRoman Divacky } 330f22ef01cSRoman Divacky 331f22ef01cSRoman Divacky 332f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitCallExpr(const CallExpr *E) { 333f22ef01cSRoman Divacky if (E->getCallReturnType()->isReferenceType()) 334f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 335f22ef01cSRoman Divacky 336f22ef01cSRoman Divacky return CGF.EmitCallExpr(E).getComplexVal(); 337f22ef01cSRoman Divacky } 338f22ef01cSRoman Divacky 339f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitStmtExpr(const StmtExpr *E) { 3402754fe60SDimitry Andric CodeGenFunction::StmtExprEvaluation eval(CGF); 341f22ef01cSRoman Divacky return CGF.EmitCompoundStmt(*E->getSubStmt(), true).getComplexVal(); 342f22ef01cSRoman Divacky } 343f22ef01cSRoman Divacky 344f22ef01cSRoman Divacky /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType. 345f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitComplexToComplexCast(ComplexPairTy Val, 346f22ef01cSRoman Divacky QualType SrcType, 347f22ef01cSRoman Divacky QualType DestType) { 348f22ef01cSRoman Divacky // Get the src/dest element type. 349f22ef01cSRoman Divacky SrcType = SrcType->getAs<ComplexType>()->getElementType(); 350f22ef01cSRoman Divacky DestType = DestType->getAs<ComplexType>()->getElementType(); 351f22ef01cSRoman Divacky 352f22ef01cSRoman Divacky // C99 6.3.1.6: When a value of complex type is converted to another 353f22ef01cSRoman Divacky // complex type, both the real and imaginary parts follow the conversion 354f22ef01cSRoman Divacky // rules for the corresponding real types. 355f22ef01cSRoman Divacky Val.first = CGF.EmitScalarConversion(Val.first, SrcType, DestType); 356f22ef01cSRoman Divacky Val.second = CGF.EmitScalarConversion(Val.second, SrcType, DestType); 357f22ef01cSRoman Divacky return Val; 358f22ef01cSRoman Divacky } 359f22ef01cSRoman Divacky 360ffd1746dSEd Schouten ComplexPairTy ComplexExprEmitter::EmitCast(CastExpr::CastKind CK, Expr *Op, 361ffd1746dSEd Schouten QualType DestTy) { 3622754fe60SDimitry Andric switch (CK) { 36317a519f9SDimitry Andric case CK_Dependent: llvm_unreachable("dependent cast kind in IR gen!"); 36417a519f9SDimitry Andric 3652754fe60SDimitry Andric case CK_GetObjCProperty: { 3662754fe60SDimitry Andric LValue LV = CGF.EmitLValue(Op); 3672754fe60SDimitry Andric assert(LV.isPropertyRef() && "Unknown LValue type!"); 3682754fe60SDimitry Andric return CGF.EmitLoadOfPropertyRefLValue(LV).getComplexVal(); 3692754fe60SDimitry Andric } 3702754fe60SDimitry Andric 3712754fe60SDimitry Andric case CK_NoOp: 3722754fe60SDimitry Andric case CK_LValueToRValue: 37317a519f9SDimitry Andric case CK_UserDefinedConversion: 3742754fe60SDimitry Andric return Visit(Op); 3752754fe60SDimitry Andric 37617a519f9SDimitry Andric case CK_LValueBitCast: { 377ffd1746dSEd Schouten llvm::Value *V = CGF.EmitLValue(Op).getAddress(); 378ffd1746dSEd Schouten V = Builder.CreateBitCast(V, 379ffd1746dSEd Schouten CGF.ConvertType(CGF.getContext().getPointerType(DestTy))); 380ffd1746dSEd Schouten // FIXME: Are the qualifiers correct here? 381ffd1746dSEd Schouten return EmitLoadOfComplex(V, DestTy.isVolatileQualified()); 382ffd1746dSEd Schouten } 383ffd1746dSEd Schouten 38417a519f9SDimitry Andric case CK_BitCast: 38517a519f9SDimitry Andric case CK_BaseToDerived: 38617a519f9SDimitry Andric case CK_DerivedToBase: 38717a519f9SDimitry Andric case CK_UncheckedDerivedToBase: 38817a519f9SDimitry Andric case CK_Dynamic: 38917a519f9SDimitry Andric case CK_ToUnion: 39017a519f9SDimitry Andric case CK_ArrayToPointerDecay: 39117a519f9SDimitry Andric case CK_FunctionToPointerDecay: 39217a519f9SDimitry Andric case CK_NullToPointer: 39317a519f9SDimitry Andric case CK_NullToMemberPointer: 39417a519f9SDimitry Andric case CK_BaseToDerivedMemberPointer: 39517a519f9SDimitry Andric case CK_DerivedToBaseMemberPointer: 39617a519f9SDimitry Andric case CK_MemberPointerToBoolean: 39717a519f9SDimitry Andric case CK_ConstructorConversion: 39817a519f9SDimitry Andric case CK_IntegralToPointer: 39917a519f9SDimitry Andric case CK_PointerToIntegral: 40017a519f9SDimitry Andric case CK_PointerToBoolean: 40117a519f9SDimitry Andric case CK_ToVoid: 40217a519f9SDimitry Andric case CK_VectorSplat: 40317a519f9SDimitry Andric case CK_IntegralCast: 40417a519f9SDimitry Andric case CK_IntegralToBoolean: 40517a519f9SDimitry Andric case CK_IntegralToFloating: 40617a519f9SDimitry Andric case CK_FloatingToIntegral: 40717a519f9SDimitry Andric case CK_FloatingToBoolean: 40817a519f9SDimitry Andric case CK_FloatingCast: 4096122f3e6SDimitry Andric case CK_CPointerToObjCPointerCast: 4106122f3e6SDimitry Andric case CK_BlockPointerToObjCPointerCast: 41117a519f9SDimitry Andric case CK_AnyPointerToBlockPointerCast: 41217a519f9SDimitry Andric case CK_ObjCObjectLValueCast: 41317a519f9SDimitry Andric case CK_FloatingComplexToReal: 41417a519f9SDimitry Andric case CK_FloatingComplexToBoolean: 41517a519f9SDimitry Andric case CK_IntegralComplexToReal: 41617a519f9SDimitry Andric case CK_IntegralComplexToBoolean: 4176122f3e6SDimitry Andric case CK_ARCProduceObject: 4186122f3e6SDimitry Andric case CK_ARCConsumeObject: 4196122f3e6SDimitry Andric case CK_ARCReclaimReturnedObject: 4206122f3e6SDimitry Andric case CK_ARCExtendBlockObject: 42117a519f9SDimitry Andric llvm_unreachable("invalid cast kind for complex value"); 42217a519f9SDimitry Andric 42317a519f9SDimitry Andric case CK_FloatingRealToComplex: 42417a519f9SDimitry Andric case CK_IntegralRealToComplex: { 425f22ef01cSRoman Divacky llvm::Value *Elt = CGF.EmitScalarExpr(Op); 426f22ef01cSRoman Divacky 427f22ef01cSRoman Divacky // Convert the input element to the element type of the complex. 428f22ef01cSRoman Divacky DestTy = DestTy->getAs<ComplexType>()->getElementType(); 429f22ef01cSRoman Divacky Elt = CGF.EmitScalarConversion(Elt, Op->getType(), DestTy); 430f22ef01cSRoman Divacky 431f22ef01cSRoman Divacky // Return (realval, 0). 432f22ef01cSRoman Divacky return ComplexPairTy(Elt, llvm::Constant::getNullValue(Elt->getType())); 433f22ef01cSRoman Divacky } 434f22ef01cSRoman Divacky 43517a519f9SDimitry Andric case CK_FloatingComplexCast: 43617a519f9SDimitry Andric case CK_FloatingComplexToIntegralComplex: 43717a519f9SDimitry Andric case CK_IntegralComplexCast: 43817a519f9SDimitry Andric case CK_IntegralComplexToFloatingComplex: 43917a519f9SDimitry Andric return EmitComplexToComplexCast(Visit(Op), Op->getType(), DestTy); 44017a519f9SDimitry Andric } 44117a519f9SDimitry Andric 44217a519f9SDimitry Andric llvm_unreachable("unknown cast resulting in complex value"); 44317a519f9SDimitry Andric } 44417a519f9SDimitry Andric 445f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryMinus(const UnaryOperator *E) { 446f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 447f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 448f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr()); 449f22ef01cSRoman Divacky 450f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 451f22ef01cSRoman Divacky if (Op.first->getType()->isFloatingPointTy()) { 452f22ef01cSRoman Divacky ResR = Builder.CreateFNeg(Op.first, "neg.r"); 453f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "neg.i"); 454f22ef01cSRoman Divacky } else { 455f22ef01cSRoman Divacky ResR = Builder.CreateNeg(Op.first, "neg.r"); 456f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "neg.i"); 457f22ef01cSRoman Divacky } 458f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 459f22ef01cSRoman Divacky } 460f22ef01cSRoman Divacky 461f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryNot(const UnaryOperator *E) { 462f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 463f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 464f22ef01cSRoman Divacky // ~(a+ib) = a + i*-b 465f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr()); 466f22ef01cSRoman Divacky llvm::Value *ResI; 467f22ef01cSRoman Divacky if (Op.second->getType()->isFloatingPointTy()) 468f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "conj.i"); 469f22ef01cSRoman Divacky else 470f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "conj.i"); 471f22ef01cSRoman Divacky 472f22ef01cSRoman Divacky return ComplexPairTy(Op.first, ResI); 473f22ef01cSRoman Divacky } 474f22ef01cSRoman Divacky 475f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinAdd(const BinOpInfo &Op) { 476f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 477f22ef01cSRoman Divacky 478f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 479f22ef01cSRoman Divacky ResR = Builder.CreateFAdd(Op.LHS.first, Op.RHS.first, "add.r"); 480f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(Op.LHS.second, Op.RHS.second, "add.i"); 481f22ef01cSRoman Divacky } else { 482f22ef01cSRoman Divacky ResR = Builder.CreateAdd(Op.LHS.first, Op.RHS.first, "add.r"); 483f22ef01cSRoman Divacky ResI = Builder.CreateAdd(Op.LHS.second, Op.RHS.second, "add.i"); 484f22ef01cSRoman Divacky } 485f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 486f22ef01cSRoman Divacky } 487f22ef01cSRoman Divacky 488f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinSub(const BinOpInfo &Op) { 489f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 490f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 491f22ef01cSRoman Divacky ResR = Builder.CreateFSub(Op.LHS.first, Op.RHS.first, "sub.r"); 492f22ef01cSRoman Divacky ResI = Builder.CreateFSub(Op.LHS.second, Op.RHS.second, "sub.i"); 493f22ef01cSRoman Divacky } else { 494f22ef01cSRoman Divacky ResR = Builder.CreateSub(Op.LHS.first, Op.RHS.first, "sub.r"); 495f22ef01cSRoman Divacky ResI = Builder.CreateSub(Op.LHS.second, Op.RHS.second, "sub.i"); 496f22ef01cSRoman Divacky } 497f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 498f22ef01cSRoman Divacky } 499f22ef01cSRoman Divacky 500f22ef01cSRoman Divacky 501f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinMul(const BinOpInfo &Op) { 502f22ef01cSRoman Divacky using llvm::Value; 503f22ef01cSRoman Divacky Value *ResR, *ResI; 504f22ef01cSRoman Divacky 505f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 506f22ef01cSRoman Divacky Value *ResRl = Builder.CreateFMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 507f22ef01cSRoman Divacky Value *ResRr = Builder.CreateFMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 508f22ef01cSRoman Divacky ResR = Builder.CreateFSub(ResRl, ResRr, "mul.r"); 509f22ef01cSRoman Divacky 510f22ef01cSRoman Divacky Value *ResIl = Builder.CreateFMul(Op.LHS.second, Op.RHS.first, "mul.il"); 511f22ef01cSRoman Divacky Value *ResIr = Builder.CreateFMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 512f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(ResIl, ResIr, "mul.i"); 513f22ef01cSRoman Divacky } else { 514f22ef01cSRoman Divacky Value *ResRl = Builder.CreateMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 515f22ef01cSRoman Divacky Value *ResRr = Builder.CreateMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 516f22ef01cSRoman Divacky ResR = Builder.CreateSub(ResRl, ResRr, "mul.r"); 517f22ef01cSRoman Divacky 518f22ef01cSRoman Divacky Value *ResIl = Builder.CreateMul(Op.LHS.second, Op.RHS.first, "mul.il"); 519f22ef01cSRoman Divacky Value *ResIr = Builder.CreateMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 520f22ef01cSRoman Divacky ResI = Builder.CreateAdd(ResIl, ResIr, "mul.i"); 521f22ef01cSRoman Divacky } 522f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 523f22ef01cSRoman Divacky } 524f22ef01cSRoman Divacky 525f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinDiv(const BinOpInfo &Op) { 526f22ef01cSRoman Divacky llvm::Value *LHSr = Op.LHS.first, *LHSi = Op.LHS.second; 527f22ef01cSRoman Divacky llvm::Value *RHSr = Op.RHS.first, *RHSi = Op.RHS.second; 528f22ef01cSRoman Divacky 529f22ef01cSRoman Divacky 530f22ef01cSRoman Divacky llvm::Value *DSTr, *DSTi; 531f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 532f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 5336122f3e6SDimitry Andric llvm::Value *Tmp1 = Builder.CreateFMul(LHSr, RHSr); // a*c 5346122f3e6SDimitry Andric llvm::Value *Tmp2 = Builder.CreateFMul(LHSi, RHSi); // b*d 5356122f3e6SDimitry Andric llvm::Value *Tmp3 = Builder.CreateFAdd(Tmp1, Tmp2); // ac+bd 536f22ef01cSRoman Divacky 5376122f3e6SDimitry Andric llvm::Value *Tmp4 = Builder.CreateFMul(RHSr, RHSr); // c*c 5386122f3e6SDimitry Andric llvm::Value *Tmp5 = Builder.CreateFMul(RHSi, RHSi); // d*d 5396122f3e6SDimitry Andric llvm::Value *Tmp6 = Builder.CreateFAdd(Tmp4, Tmp5); // cc+dd 540f22ef01cSRoman Divacky 5416122f3e6SDimitry Andric llvm::Value *Tmp7 = Builder.CreateFMul(LHSi, RHSr); // b*c 5426122f3e6SDimitry Andric llvm::Value *Tmp8 = Builder.CreateFMul(LHSr, RHSi); // a*d 5436122f3e6SDimitry Andric llvm::Value *Tmp9 = Builder.CreateFSub(Tmp7, Tmp8); // bc-ad 544f22ef01cSRoman Divacky 5456122f3e6SDimitry Andric DSTr = Builder.CreateFDiv(Tmp3, Tmp6); 5466122f3e6SDimitry Andric DSTi = Builder.CreateFDiv(Tmp9, Tmp6); 547f22ef01cSRoman Divacky } else { 548f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 5496122f3e6SDimitry Andric llvm::Value *Tmp1 = Builder.CreateMul(LHSr, RHSr); // a*c 5506122f3e6SDimitry Andric llvm::Value *Tmp2 = Builder.CreateMul(LHSi, RHSi); // b*d 5516122f3e6SDimitry Andric llvm::Value *Tmp3 = Builder.CreateAdd(Tmp1, Tmp2); // ac+bd 552f22ef01cSRoman Divacky 5536122f3e6SDimitry Andric llvm::Value *Tmp4 = Builder.CreateMul(RHSr, RHSr); // c*c 5546122f3e6SDimitry Andric llvm::Value *Tmp5 = Builder.CreateMul(RHSi, RHSi); // d*d 5556122f3e6SDimitry Andric llvm::Value *Tmp6 = Builder.CreateAdd(Tmp4, Tmp5); // cc+dd 556f22ef01cSRoman Divacky 5576122f3e6SDimitry Andric llvm::Value *Tmp7 = Builder.CreateMul(LHSi, RHSr); // b*c 5586122f3e6SDimitry Andric llvm::Value *Tmp8 = Builder.CreateMul(LHSr, RHSi); // a*d 5596122f3e6SDimitry Andric llvm::Value *Tmp9 = Builder.CreateSub(Tmp7, Tmp8); // bc-ad 560f22ef01cSRoman Divacky 561f22ef01cSRoman Divacky if (Op.Ty->getAs<ComplexType>()->getElementType()->isUnsignedIntegerType()) { 5626122f3e6SDimitry Andric DSTr = Builder.CreateUDiv(Tmp3, Tmp6); 5636122f3e6SDimitry Andric DSTi = Builder.CreateUDiv(Tmp9, Tmp6); 564f22ef01cSRoman Divacky } else { 5656122f3e6SDimitry Andric DSTr = Builder.CreateSDiv(Tmp3, Tmp6); 5666122f3e6SDimitry Andric DSTi = Builder.CreateSDiv(Tmp9, Tmp6); 567f22ef01cSRoman Divacky } 568f22ef01cSRoman Divacky } 569f22ef01cSRoman Divacky 570f22ef01cSRoman Divacky return ComplexPairTy(DSTr, DSTi); 571f22ef01cSRoman Divacky } 572f22ef01cSRoman Divacky 573f22ef01cSRoman Divacky ComplexExprEmitter::BinOpInfo 574f22ef01cSRoman Divacky ComplexExprEmitter::EmitBinOps(const BinaryOperator *E) { 575f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 576f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 577f22ef01cSRoman Divacky BinOpInfo Ops; 578f22ef01cSRoman Divacky Ops.LHS = Visit(E->getLHS()); 579f22ef01cSRoman Divacky Ops.RHS = Visit(E->getRHS()); 580f22ef01cSRoman Divacky Ops.Ty = E->getType(); 581f22ef01cSRoman Divacky return Ops; 582f22ef01cSRoman Divacky } 583f22ef01cSRoman Divacky 584f22ef01cSRoman Divacky 5852754fe60SDimitry Andric LValue ComplexExprEmitter:: 5862754fe60SDimitry Andric EmitCompoundAssignLValue(const CompoundAssignOperator *E, 5872754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&), 5882754fe60SDimitry Andric ComplexPairTy &Val) { 589f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 590f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 5912754fe60SDimitry Andric QualType LHSTy = E->getLHS()->getType(); 592f22ef01cSRoman Divacky 593f22ef01cSRoman Divacky BinOpInfo OpInfo; 594f22ef01cSRoman Divacky 595f22ef01cSRoman Divacky // Load the RHS and LHS operands. 596f22ef01cSRoman Divacky // __block variables need to have the rhs evaluated first, plus this should 5972754fe60SDimitry Andric // improve codegen a little. 598f22ef01cSRoman Divacky OpInfo.Ty = E->getComputationResultType(); 5992754fe60SDimitry Andric 6002754fe60SDimitry Andric // The RHS should have been converted to the computation type. 6012754fe60SDimitry Andric assert(OpInfo.Ty->isAnyComplexType()); 6022754fe60SDimitry Andric assert(CGF.getContext().hasSameUnqualifiedType(OpInfo.Ty, 6032754fe60SDimitry Andric E->getRHS()->getType())); 6042754fe60SDimitry Andric OpInfo.RHS = Visit(E->getRHS()); 605f22ef01cSRoman Divacky 606ffd1746dSEd Schouten LValue LHS = CGF.EmitLValue(E->getLHS()); 6072754fe60SDimitry Andric 6082754fe60SDimitry Andric // Load from the l-value. 6092754fe60SDimitry Andric ComplexPairTy LHSComplexPair = EmitLoadOfLValue(LHS); 610f22ef01cSRoman Divacky 611f22ef01cSRoman Divacky OpInfo.LHS = EmitComplexToComplexCast(LHSComplexPair, LHSTy, OpInfo.Ty); 612f22ef01cSRoman Divacky 613f22ef01cSRoman Divacky // Expand the binary operator. 614f22ef01cSRoman Divacky ComplexPairTy Result = (this->*Func)(OpInfo); 615f22ef01cSRoman Divacky 616f22ef01cSRoman Divacky // Truncate the result back to the LHS type. 617f22ef01cSRoman Divacky Result = EmitComplexToComplexCast(Result, OpInfo.Ty, LHSTy); 6182754fe60SDimitry Andric Val = Result; 619f22ef01cSRoman Divacky 620f22ef01cSRoman Divacky // Store the result value into the LHS lvalue. 6212754fe60SDimitry Andric EmitStoreThroughLValue(Result, LHS); 622ffd1746dSEd Schouten 6232754fe60SDimitry Andric return LHS; 624f22ef01cSRoman Divacky } 625f22ef01cSRoman Divacky 6262754fe60SDimitry Andric // Compound assignments. 6272754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter:: 6282754fe60SDimitry Andric EmitCompoundAssign(const CompoundAssignOperator *E, 6292754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&)){ 6302754fe60SDimitry Andric ComplexPairTy Val; 6312754fe60SDimitry Andric LValue LV = EmitCompoundAssignLValue(E, Func, Val); 6322754fe60SDimitry Andric 6332754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value. 6342754fe60SDimitry Andric if (!CGF.getContext().getLangOptions().CPlusPlus) 6352754fe60SDimitry Andric return Val; 6362754fe60SDimitry Andric 6372754fe60SDimitry Andric // Objective-C property assignment never reloads the value following a store. 6382754fe60SDimitry Andric if (LV.isPropertyRef()) 6392754fe60SDimitry Andric return Val; 6402754fe60SDimitry Andric 6412754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 6422754fe60SDimitry Andric if (!LV.isVolatileQualified()) 6432754fe60SDimitry Andric return Val; 6442754fe60SDimitry Andric 6452754fe60SDimitry Andric return EmitLoadOfComplex(LV.getAddress(), LV.isVolatileQualified()); 6462754fe60SDimitry Andric } 6472754fe60SDimitry Andric 6482754fe60SDimitry Andric LValue ComplexExprEmitter::EmitBinAssignLValue(const BinaryOperator *E, 6492754fe60SDimitry Andric ComplexPairTy &Val) { 650ffd1746dSEd Schouten assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(), 651ffd1746dSEd Schouten E->getRHS()->getType()) && 652f22ef01cSRoman Divacky "Invalid assignment"); 6532754fe60SDimitry Andric TestAndClearIgnoreReal(); 6542754fe60SDimitry Andric TestAndClearIgnoreImag(); 6552754fe60SDimitry Andric 6562754fe60SDimitry Andric // Emit the RHS. __block variables need the RHS evaluated first. 6572754fe60SDimitry Andric Val = Visit(E->getRHS()); 658f22ef01cSRoman Divacky 659f22ef01cSRoman Divacky // Compute the address to store into. 660f22ef01cSRoman Divacky LValue LHS = CGF.EmitLValue(E->getLHS()); 661f22ef01cSRoman Divacky 662ffd1746dSEd Schouten // Store the result value into the LHS lvalue. 6632754fe60SDimitry Andric EmitStoreThroughLValue(Val, LHS); 664f22ef01cSRoman Divacky 6652754fe60SDimitry Andric return LHS; 6662754fe60SDimitry Andric } 667ffd1746dSEd Schouten 6682754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitBinAssign(const BinaryOperator *E) { 6692754fe60SDimitry Andric ComplexPairTy Val; 6702754fe60SDimitry Andric LValue LV = EmitBinAssignLValue(E, Val); 6712754fe60SDimitry Andric 6722754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value. 6732754fe60SDimitry Andric if (!CGF.getContext().getLangOptions().CPlusPlus) 674f22ef01cSRoman Divacky return Val; 675ffd1746dSEd Schouten 6762754fe60SDimitry Andric // Objective-C property assignment never reloads the value following a store. 6772754fe60SDimitry Andric if (LV.isPropertyRef()) 6782754fe60SDimitry Andric return Val; 6792754fe60SDimitry Andric 6802754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 6812754fe60SDimitry Andric if (!LV.isVolatileQualified()) 6822754fe60SDimitry Andric return Val; 6832754fe60SDimitry Andric 6842754fe60SDimitry Andric return EmitLoadOfComplex(LV.getAddress(), LV.isVolatileQualified()); 685f22ef01cSRoman Divacky } 686f22ef01cSRoman Divacky 687f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitBinComma(const BinaryOperator *E) { 6882754fe60SDimitry Andric CGF.EmitIgnoredExpr(E->getLHS()); 689f22ef01cSRoman Divacky return Visit(E->getRHS()); 690f22ef01cSRoman Divacky } 691f22ef01cSRoman Divacky 692f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter:: 6932754fe60SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) { 694f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 695f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 696f22ef01cSRoman Divacky llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true"); 697f22ef01cSRoman Divacky llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false"); 698f22ef01cSRoman Divacky llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end"); 699f22ef01cSRoman Divacky 7002754fe60SDimitry Andric // Bind the common expression if necessary. 7012754fe60SDimitry Andric CodeGenFunction::OpaqueValueMapping binding(CGF, E); 7022754fe60SDimitry Andric 7032754fe60SDimitry Andric CodeGenFunction::ConditionalEvaluation eval(CGF); 704f22ef01cSRoman Divacky CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock); 705f22ef01cSRoman Divacky 7062754fe60SDimitry Andric eval.begin(CGF); 707f22ef01cSRoman Divacky CGF.EmitBlock(LHSBlock); 7082754fe60SDimitry Andric ComplexPairTy LHS = Visit(E->getTrueExpr()); 709f22ef01cSRoman Divacky LHSBlock = Builder.GetInsertBlock(); 710f22ef01cSRoman Divacky CGF.EmitBranch(ContBlock); 7112754fe60SDimitry Andric eval.end(CGF); 712f22ef01cSRoman Divacky 7132754fe60SDimitry Andric eval.begin(CGF); 714f22ef01cSRoman Divacky CGF.EmitBlock(RHSBlock); 7152754fe60SDimitry Andric ComplexPairTy RHS = Visit(E->getFalseExpr()); 716f22ef01cSRoman Divacky RHSBlock = Builder.GetInsertBlock(); 717f22ef01cSRoman Divacky CGF.EmitBlock(ContBlock); 7182754fe60SDimitry Andric eval.end(CGF); 719f22ef01cSRoman Divacky 720f22ef01cSRoman Divacky // Create a PHI node for the real part. 7213b0f4066SDimitry Andric llvm::PHINode *RealPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.r"); 722f22ef01cSRoman Divacky RealPN->addIncoming(LHS.first, LHSBlock); 723f22ef01cSRoman Divacky RealPN->addIncoming(RHS.first, RHSBlock); 724f22ef01cSRoman Divacky 725f22ef01cSRoman Divacky // Create a PHI node for the imaginary part. 7263b0f4066SDimitry Andric llvm::PHINode *ImagPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.i"); 727f22ef01cSRoman Divacky ImagPN->addIncoming(LHS.second, LHSBlock); 728f22ef01cSRoman Divacky ImagPN->addIncoming(RHS.second, RHSBlock); 729f22ef01cSRoman Divacky 730f22ef01cSRoman Divacky return ComplexPairTy(RealPN, ImagPN); 731f22ef01cSRoman Divacky } 732f22ef01cSRoman Divacky 733f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitChooseExpr(ChooseExpr *E) { 734f22ef01cSRoman Divacky return Visit(E->getChosenSubExpr(CGF.getContext())); 735f22ef01cSRoman Divacky } 736f22ef01cSRoman Divacky 737f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitInitListExpr(InitListExpr *E) { 738f22ef01cSRoman Divacky bool Ignore = TestAndClearIgnoreReal(); 739f22ef01cSRoman Divacky (void)Ignore; 740f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored"); 741f22ef01cSRoman Divacky Ignore = TestAndClearIgnoreImag(); 742f22ef01cSRoman Divacky (void)Ignore; 743f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored"); 7446122f3e6SDimitry Andric 7456122f3e6SDimitry Andric if (E->getNumInits() == 2) { 7466122f3e6SDimitry Andric llvm::Value *Real = CGF.EmitScalarExpr(E->getInit(0)); 7476122f3e6SDimitry Andric llvm::Value *Imag = CGF.EmitScalarExpr(E->getInit(1)); 7486122f3e6SDimitry Andric return ComplexPairTy(Real, Imag); 7496122f3e6SDimitry Andric } else if (E->getNumInits() == 1) { 750f22ef01cSRoman Divacky return Visit(E->getInit(0)); 7516122f3e6SDimitry Andric } 752f22ef01cSRoman Divacky 753f22ef01cSRoman Divacky // Empty init list intializes to null 7546122f3e6SDimitry Andric assert(E->getNumInits() == 0 && "Unexpected number of inits"); 755f22ef01cSRoman Divacky QualType Ty = E->getType()->getAs<ComplexType>()->getElementType(); 7566122f3e6SDimitry Andric llvm::Type* LTy = CGF.ConvertType(Ty); 757f22ef01cSRoman Divacky llvm::Value* zeroConstant = llvm::Constant::getNullValue(LTy); 758f22ef01cSRoman Divacky return ComplexPairTy(zeroConstant, zeroConstant); 759f22ef01cSRoman Divacky } 760f22ef01cSRoman Divacky 761f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitVAArgExpr(VAArgExpr *E) { 762f22ef01cSRoman Divacky llvm::Value *ArgValue = CGF.EmitVAListRef(E->getSubExpr()); 763f22ef01cSRoman Divacky llvm::Value *ArgPtr = CGF.EmitVAArg(ArgValue, E->getType()); 764f22ef01cSRoman Divacky 765f22ef01cSRoman Divacky if (!ArgPtr) { 766f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex va_arg expression"); 7676122f3e6SDimitry Andric llvm::Type *EltTy = 768f22ef01cSRoman Divacky CGF.ConvertType(E->getType()->getAs<ComplexType>()->getElementType()); 769f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy); 770f22ef01cSRoman Divacky return ComplexPairTy(U, U); 771f22ef01cSRoman Divacky } 772f22ef01cSRoman Divacky 773f22ef01cSRoman Divacky // FIXME Volatility. 774f22ef01cSRoman Divacky return EmitLoadOfComplex(ArgPtr, false); 775f22ef01cSRoman Divacky } 776f22ef01cSRoman Divacky 777f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 778f22ef01cSRoman Divacky // Entry Point into this File 779f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 780f22ef01cSRoman Divacky 781f22ef01cSRoman Divacky /// EmitComplexExpr - Emit the computation of the specified expression of 782f22ef01cSRoman Divacky /// complex type, ignoring the result. 783f22ef01cSRoman Divacky ComplexPairTy CodeGenFunction::EmitComplexExpr(const Expr *E, bool IgnoreReal, 7842754fe60SDimitry Andric bool IgnoreImag) { 785f22ef01cSRoman Divacky assert(E && E->getType()->isAnyComplexType() && 786f22ef01cSRoman Divacky "Invalid complex expression to emit"); 787f22ef01cSRoman Divacky 7882754fe60SDimitry Andric return ComplexExprEmitter(*this, IgnoreReal, IgnoreImag) 789f22ef01cSRoman Divacky .Visit(const_cast<Expr*>(E)); 790f22ef01cSRoman Divacky } 791f22ef01cSRoman Divacky 792f22ef01cSRoman Divacky /// EmitComplexExprIntoAddr - Emit the computation of the specified expression 793f22ef01cSRoman Divacky /// of complex type, storing into the specified Value*. 794f22ef01cSRoman Divacky void CodeGenFunction::EmitComplexExprIntoAddr(const Expr *E, 795f22ef01cSRoman Divacky llvm::Value *DestAddr, 796f22ef01cSRoman Divacky bool DestIsVolatile) { 797f22ef01cSRoman Divacky assert(E && E->getType()->isAnyComplexType() && 798f22ef01cSRoman Divacky "Invalid complex expression to emit"); 799f22ef01cSRoman Divacky ComplexExprEmitter Emitter(*this); 800f22ef01cSRoman Divacky ComplexPairTy Val = Emitter.Visit(const_cast<Expr*>(E)); 801f22ef01cSRoman Divacky Emitter.EmitStoreOfComplex(Val, DestAddr, DestIsVolatile); 802f22ef01cSRoman Divacky } 803f22ef01cSRoman Divacky 804f22ef01cSRoman Divacky /// StoreComplexToAddr - Store a complex number into the specified address. 805f22ef01cSRoman Divacky void CodeGenFunction::StoreComplexToAddr(ComplexPairTy V, 806f22ef01cSRoman Divacky llvm::Value *DestAddr, 807f22ef01cSRoman Divacky bool DestIsVolatile) { 808f22ef01cSRoman Divacky ComplexExprEmitter(*this).EmitStoreOfComplex(V, DestAddr, DestIsVolatile); 809f22ef01cSRoman Divacky } 810f22ef01cSRoman Divacky 811f22ef01cSRoman Divacky /// LoadComplexFromAddr - Load a complex number from the specified address. 812f22ef01cSRoman Divacky ComplexPairTy CodeGenFunction::LoadComplexFromAddr(llvm::Value *SrcAddr, 813f22ef01cSRoman Divacky bool SrcIsVolatile) { 814f22ef01cSRoman Divacky return ComplexExprEmitter(*this).EmitLoadOfComplex(SrcAddr, SrcIsVolatile); 815f22ef01cSRoman Divacky } 8162754fe60SDimitry Andric 8172754fe60SDimitry Andric LValue CodeGenFunction::EmitComplexAssignmentLValue(const BinaryOperator *E) { 8182754fe60SDimitry Andric assert(E->getOpcode() == BO_Assign); 8192754fe60SDimitry Andric ComplexPairTy Val; // ignored 8202754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitBinAssignLValue(E, Val); 8212754fe60SDimitry Andric } 8222754fe60SDimitry Andric 8232754fe60SDimitry Andric LValue CodeGenFunction:: 8242754fe60SDimitry Andric EmitComplexCompoundAssignmentLValue(const CompoundAssignOperator *E) { 8252754fe60SDimitry Andric ComplexPairTy(ComplexExprEmitter::*Op)(const ComplexExprEmitter::BinOpInfo &); 8262754fe60SDimitry Andric switch (E->getOpcode()) { 8272754fe60SDimitry Andric case BO_MulAssign: Op = &ComplexExprEmitter::EmitBinMul; break; 8282754fe60SDimitry Andric case BO_DivAssign: Op = &ComplexExprEmitter::EmitBinDiv; break; 8292754fe60SDimitry Andric case BO_SubAssign: Op = &ComplexExprEmitter::EmitBinSub; break; 8302754fe60SDimitry Andric case BO_AddAssign: Op = &ComplexExprEmitter::EmitBinAdd; break; 8312754fe60SDimitry Andric 8322754fe60SDimitry Andric default: 8332754fe60SDimitry Andric llvm_unreachable("unexpected complex compound assignment"); 8342754fe60SDimitry Andric Op = 0; 8352754fe60SDimitry Andric } 8362754fe60SDimitry Andric 8372754fe60SDimitry Andric ComplexPairTy Val; // ignored 8382754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val); 8392754fe60SDimitry Andric } 840