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());} 1113b0f4066SDimitry Andric ComplexPairTy VisitGenericSelectionExpr(GenericSelectionExpr *GE) { 1123b0f4066SDimitry Andric return Visit(GE->getResultExpr()); 1133b0f4066SDimitry Andric } 114f22ef01cSRoman Divacky ComplexPairTy VisitImaginaryLiteral(const ImaginaryLiteral *IL); 11517a519f9SDimitry Andric ComplexPairTy 11617a519f9SDimitry Andric VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *PE) { 11717a519f9SDimitry Andric return Visit(PE->getReplacement()); 11817a519f9SDimitry Andric } 119f22ef01cSRoman Divacky 120f22ef01cSRoman Divacky // l-values. 121f22ef01cSRoman Divacky ComplexPairTy VisitDeclRefExpr(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); 269f22ef01cSRoman Divacky }; 270f22ef01cSRoman Divacky } // end anonymous namespace. 271f22ef01cSRoman Divacky 272f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 273f22ef01cSRoman Divacky // Utilities 274f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 275f22ef01cSRoman Divacky 276f22ef01cSRoman Divacky /// EmitLoadOfComplex - Given an RValue reference for a complex, emit code to 277f22ef01cSRoman Divacky /// load the real and imaginary pieces, returning them as Real/Imag. 278f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitLoadOfComplex(llvm::Value *SrcPtr, 279f22ef01cSRoman Divacky bool isVolatile) { 280f22ef01cSRoman Divacky llvm::Value *Real=0, *Imag=0; 281f22ef01cSRoman Divacky 2822754fe60SDimitry Andric if (!IgnoreReal || isVolatile) { 283f22ef01cSRoman Divacky llvm::Value *RealP = Builder.CreateStructGEP(SrcPtr, 0, 284f22ef01cSRoman Divacky SrcPtr->getName() + ".realp"); 285f22ef01cSRoman Divacky Real = Builder.CreateLoad(RealP, isVolatile, SrcPtr->getName() + ".real"); 286f22ef01cSRoman Divacky } 287f22ef01cSRoman Divacky 2882754fe60SDimitry Andric if (!IgnoreImag || isVolatile) { 289f22ef01cSRoman Divacky llvm::Value *ImagP = Builder.CreateStructGEP(SrcPtr, 1, 290f22ef01cSRoman Divacky SrcPtr->getName() + ".imagp"); 291f22ef01cSRoman Divacky Imag = Builder.CreateLoad(ImagP, isVolatile, SrcPtr->getName() + ".imag"); 292f22ef01cSRoman Divacky } 293f22ef01cSRoman Divacky return ComplexPairTy(Real, Imag); 294f22ef01cSRoman Divacky } 295f22ef01cSRoman Divacky 296f22ef01cSRoman Divacky /// EmitStoreOfComplex - Store the specified real/imag parts into the 297f22ef01cSRoman Divacky /// specified value pointer. 298f22ef01cSRoman Divacky void ComplexExprEmitter::EmitStoreOfComplex(ComplexPairTy Val, llvm::Value *Ptr, 299f22ef01cSRoman Divacky bool isVolatile) { 300f22ef01cSRoman Divacky llvm::Value *RealPtr = Builder.CreateStructGEP(Ptr, 0, "real"); 301f22ef01cSRoman Divacky llvm::Value *ImagPtr = Builder.CreateStructGEP(Ptr, 1, "imag"); 302f22ef01cSRoman Divacky 303f22ef01cSRoman Divacky Builder.CreateStore(Val.first, RealPtr, isVolatile); 304f22ef01cSRoman Divacky Builder.CreateStore(Val.second, ImagPtr, isVolatile); 305f22ef01cSRoman Divacky } 306f22ef01cSRoman Divacky 307f22ef01cSRoman Divacky 308f22ef01cSRoman Divacky 309f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 310f22ef01cSRoman Divacky // Visitor Methods 311f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 312f22ef01cSRoman Divacky 313f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitExpr(Expr *E) { 314f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex expression"); 315f22ef01cSRoman Divacky const llvm::Type *EltTy = 316f22ef01cSRoman Divacky CGF.ConvertType(E->getType()->getAs<ComplexType>()->getElementType()); 317f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy); 318f22ef01cSRoman Divacky return ComplexPairTy(U, U); 319f22ef01cSRoman Divacky } 320f22ef01cSRoman Divacky 321f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter:: 322f22ef01cSRoman Divacky VisitImaginaryLiteral(const ImaginaryLiteral *IL) { 323f22ef01cSRoman Divacky llvm::Value *Imag = CGF.EmitScalarExpr(IL->getSubExpr()); 3242754fe60SDimitry Andric return ComplexPairTy(llvm::Constant::getNullValue(Imag->getType()), Imag); 325f22ef01cSRoman Divacky } 326f22ef01cSRoman Divacky 327f22ef01cSRoman Divacky 328f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitCallExpr(const CallExpr *E) { 329f22ef01cSRoman Divacky if (E->getCallReturnType()->isReferenceType()) 330f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 331f22ef01cSRoman Divacky 332f22ef01cSRoman Divacky return CGF.EmitCallExpr(E).getComplexVal(); 333f22ef01cSRoman Divacky } 334f22ef01cSRoman Divacky 335f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitStmtExpr(const StmtExpr *E) { 3362754fe60SDimitry Andric CodeGenFunction::StmtExprEvaluation eval(CGF); 337f22ef01cSRoman Divacky return CGF.EmitCompoundStmt(*E->getSubStmt(), true).getComplexVal(); 338f22ef01cSRoman Divacky } 339f22ef01cSRoman Divacky 340f22ef01cSRoman Divacky /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType. 341f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitComplexToComplexCast(ComplexPairTy Val, 342f22ef01cSRoman Divacky QualType SrcType, 343f22ef01cSRoman Divacky QualType DestType) { 344f22ef01cSRoman Divacky // Get the src/dest element type. 345f22ef01cSRoman Divacky SrcType = SrcType->getAs<ComplexType>()->getElementType(); 346f22ef01cSRoman Divacky DestType = DestType->getAs<ComplexType>()->getElementType(); 347f22ef01cSRoman Divacky 348f22ef01cSRoman Divacky // C99 6.3.1.6: When a value of complex type is converted to another 349f22ef01cSRoman Divacky // complex type, both the real and imaginary parts follow the conversion 350f22ef01cSRoman Divacky // rules for the corresponding real types. 351f22ef01cSRoman Divacky Val.first = CGF.EmitScalarConversion(Val.first, SrcType, DestType); 352f22ef01cSRoman Divacky Val.second = CGF.EmitScalarConversion(Val.second, SrcType, DestType); 353f22ef01cSRoman Divacky return Val; 354f22ef01cSRoman Divacky } 355f22ef01cSRoman Divacky 356ffd1746dSEd Schouten ComplexPairTy ComplexExprEmitter::EmitCast(CastExpr::CastKind CK, Expr *Op, 357ffd1746dSEd Schouten QualType DestTy) { 3582754fe60SDimitry Andric switch (CK) { 35917a519f9SDimitry Andric case CK_Dependent: llvm_unreachable("dependent cast kind in IR gen!"); 36017a519f9SDimitry Andric 3612754fe60SDimitry Andric case CK_GetObjCProperty: { 3622754fe60SDimitry Andric LValue LV = CGF.EmitLValue(Op); 3632754fe60SDimitry Andric assert(LV.isPropertyRef() && "Unknown LValue type!"); 3642754fe60SDimitry Andric return CGF.EmitLoadOfPropertyRefLValue(LV).getComplexVal(); 3652754fe60SDimitry Andric } 3662754fe60SDimitry Andric 3672754fe60SDimitry Andric case CK_NoOp: 3682754fe60SDimitry Andric case CK_LValueToRValue: 36917a519f9SDimitry Andric case CK_UserDefinedConversion: 3702754fe60SDimitry Andric return Visit(Op); 3712754fe60SDimitry Andric 37217a519f9SDimitry Andric case CK_LValueBitCast: { 373ffd1746dSEd Schouten llvm::Value *V = CGF.EmitLValue(Op).getAddress(); 374ffd1746dSEd Schouten V = Builder.CreateBitCast(V, 375ffd1746dSEd Schouten CGF.ConvertType(CGF.getContext().getPointerType(DestTy))); 376ffd1746dSEd Schouten // FIXME: Are the qualifiers correct here? 377ffd1746dSEd Schouten return EmitLoadOfComplex(V, DestTy.isVolatileQualified()); 378ffd1746dSEd Schouten } 379ffd1746dSEd Schouten 38017a519f9SDimitry Andric case CK_BitCast: 38117a519f9SDimitry Andric case CK_BaseToDerived: 38217a519f9SDimitry Andric case CK_DerivedToBase: 38317a519f9SDimitry Andric case CK_UncheckedDerivedToBase: 38417a519f9SDimitry Andric case CK_Dynamic: 38517a519f9SDimitry Andric case CK_ToUnion: 38617a519f9SDimitry Andric case CK_ArrayToPointerDecay: 38717a519f9SDimitry Andric case CK_FunctionToPointerDecay: 38817a519f9SDimitry Andric case CK_NullToPointer: 38917a519f9SDimitry Andric case CK_NullToMemberPointer: 39017a519f9SDimitry Andric case CK_BaseToDerivedMemberPointer: 39117a519f9SDimitry Andric case CK_DerivedToBaseMemberPointer: 39217a519f9SDimitry Andric case CK_MemberPointerToBoolean: 39317a519f9SDimitry Andric case CK_ConstructorConversion: 39417a519f9SDimitry Andric case CK_IntegralToPointer: 39517a519f9SDimitry Andric case CK_PointerToIntegral: 39617a519f9SDimitry Andric case CK_PointerToBoolean: 39717a519f9SDimitry Andric case CK_ToVoid: 39817a519f9SDimitry Andric case CK_VectorSplat: 39917a519f9SDimitry Andric case CK_IntegralCast: 40017a519f9SDimitry Andric case CK_IntegralToBoolean: 40117a519f9SDimitry Andric case CK_IntegralToFloating: 40217a519f9SDimitry Andric case CK_FloatingToIntegral: 40317a519f9SDimitry Andric case CK_FloatingToBoolean: 40417a519f9SDimitry Andric case CK_FloatingCast: 40517a519f9SDimitry Andric case CK_AnyPointerToObjCPointerCast: 40617a519f9SDimitry Andric case CK_AnyPointerToBlockPointerCast: 40717a519f9SDimitry Andric case CK_ObjCObjectLValueCast: 40817a519f9SDimitry Andric case CK_FloatingComplexToReal: 40917a519f9SDimitry Andric case CK_FloatingComplexToBoolean: 41017a519f9SDimitry Andric case CK_IntegralComplexToReal: 41117a519f9SDimitry Andric case CK_IntegralComplexToBoolean: 41217a519f9SDimitry Andric case CK_ObjCProduceObject: 41317a519f9SDimitry Andric case CK_ObjCConsumeObject: 41417a519f9SDimitry Andric case CK_ObjCReclaimReturnedObject: 41517a519f9SDimitry Andric llvm_unreachable("invalid cast kind for complex value"); 41617a519f9SDimitry Andric 41717a519f9SDimitry Andric case CK_FloatingRealToComplex: 41817a519f9SDimitry Andric case CK_IntegralRealToComplex: { 419f22ef01cSRoman Divacky llvm::Value *Elt = CGF.EmitScalarExpr(Op); 420f22ef01cSRoman Divacky 421f22ef01cSRoman Divacky // Convert the input element to the element type of the complex. 422f22ef01cSRoman Divacky DestTy = DestTy->getAs<ComplexType>()->getElementType(); 423f22ef01cSRoman Divacky Elt = CGF.EmitScalarConversion(Elt, Op->getType(), DestTy); 424f22ef01cSRoman Divacky 425f22ef01cSRoman Divacky // Return (realval, 0). 426f22ef01cSRoman Divacky return ComplexPairTy(Elt, llvm::Constant::getNullValue(Elt->getType())); 427f22ef01cSRoman Divacky } 428f22ef01cSRoman Divacky 42917a519f9SDimitry Andric case CK_FloatingComplexCast: 43017a519f9SDimitry Andric case CK_FloatingComplexToIntegralComplex: 43117a519f9SDimitry Andric case CK_IntegralComplexCast: 43217a519f9SDimitry Andric case CK_IntegralComplexToFloatingComplex: 43317a519f9SDimitry Andric return EmitComplexToComplexCast(Visit(Op), Op->getType(), DestTy); 43417a519f9SDimitry Andric } 43517a519f9SDimitry Andric 43617a519f9SDimitry Andric llvm_unreachable("unknown cast resulting in complex value"); 43717a519f9SDimitry Andric } 43817a519f9SDimitry Andric 439f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryMinus(const UnaryOperator *E) { 440f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 441f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 442f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr()); 443f22ef01cSRoman Divacky 444f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 445f22ef01cSRoman Divacky if (Op.first->getType()->isFloatingPointTy()) { 446f22ef01cSRoman Divacky ResR = Builder.CreateFNeg(Op.first, "neg.r"); 447f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "neg.i"); 448f22ef01cSRoman Divacky } else { 449f22ef01cSRoman Divacky ResR = Builder.CreateNeg(Op.first, "neg.r"); 450f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "neg.i"); 451f22ef01cSRoman Divacky } 452f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 453f22ef01cSRoman Divacky } 454f22ef01cSRoman Divacky 455f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryNot(const UnaryOperator *E) { 456f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 457f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 458f22ef01cSRoman Divacky // ~(a+ib) = a + i*-b 459f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr()); 460f22ef01cSRoman Divacky llvm::Value *ResI; 461f22ef01cSRoman Divacky if (Op.second->getType()->isFloatingPointTy()) 462f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "conj.i"); 463f22ef01cSRoman Divacky else 464f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "conj.i"); 465f22ef01cSRoman Divacky 466f22ef01cSRoman Divacky return ComplexPairTy(Op.first, ResI); 467f22ef01cSRoman Divacky } 468f22ef01cSRoman Divacky 469f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinAdd(const BinOpInfo &Op) { 470f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 471f22ef01cSRoman Divacky 472f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 473f22ef01cSRoman Divacky ResR = Builder.CreateFAdd(Op.LHS.first, Op.RHS.first, "add.r"); 474f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(Op.LHS.second, Op.RHS.second, "add.i"); 475f22ef01cSRoman Divacky } else { 476f22ef01cSRoman Divacky ResR = Builder.CreateAdd(Op.LHS.first, Op.RHS.first, "add.r"); 477f22ef01cSRoman Divacky ResI = Builder.CreateAdd(Op.LHS.second, Op.RHS.second, "add.i"); 478f22ef01cSRoman Divacky } 479f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 480f22ef01cSRoman Divacky } 481f22ef01cSRoman Divacky 482f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinSub(const BinOpInfo &Op) { 483f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 484f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 485f22ef01cSRoman Divacky ResR = Builder.CreateFSub(Op.LHS.first, Op.RHS.first, "sub.r"); 486f22ef01cSRoman Divacky ResI = Builder.CreateFSub(Op.LHS.second, Op.RHS.second, "sub.i"); 487f22ef01cSRoman Divacky } else { 488f22ef01cSRoman Divacky ResR = Builder.CreateSub(Op.LHS.first, Op.RHS.first, "sub.r"); 489f22ef01cSRoman Divacky ResI = Builder.CreateSub(Op.LHS.second, Op.RHS.second, "sub.i"); 490f22ef01cSRoman Divacky } 491f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 492f22ef01cSRoman Divacky } 493f22ef01cSRoman Divacky 494f22ef01cSRoman Divacky 495f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinMul(const BinOpInfo &Op) { 496f22ef01cSRoman Divacky using llvm::Value; 497f22ef01cSRoman Divacky Value *ResR, *ResI; 498f22ef01cSRoman Divacky 499f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 500f22ef01cSRoman Divacky Value *ResRl = Builder.CreateFMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 501f22ef01cSRoman Divacky Value *ResRr = Builder.CreateFMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 502f22ef01cSRoman Divacky ResR = Builder.CreateFSub(ResRl, ResRr, "mul.r"); 503f22ef01cSRoman Divacky 504f22ef01cSRoman Divacky Value *ResIl = Builder.CreateFMul(Op.LHS.second, Op.RHS.first, "mul.il"); 505f22ef01cSRoman Divacky Value *ResIr = Builder.CreateFMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 506f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(ResIl, ResIr, "mul.i"); 507f22ef01cSRoman Divacky } else { 508f22ef01cSRoman Divacky Value *ResRl = Builder.CreateMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 509f22ef01cSRoman Divacky Value *ResRr = Builder.CreateMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 510f22ef01cSRoman Divacky ResR = Builder.CreateSub(ResRl, ResRr, "mul.r"); 511f22ef01cSRoman Divacky 512f22ef01cSRoman Divacky Value *ResIl = Builder.CreateMul(Op.LHS.second, Op.RHS.first, "mul.il"); 513f22ef01cSRoman Divacky Value *ResIr = Builder.CreateMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 514f22ef01cSRoman Divacky ResI = Builder.CreateAdd(ResIl, ResIr, "mul.i"); 515f22ef01cSRoman Divacky } 516f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 517f22ef01cSRoman Divacky } 518f22ef01cSRoman Divacky 519f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinDiv(const BinOpInfo &Op) { 520f22ef01cSRoman Divacky llvm::Value *LHSr = Op.LHS.first, *LHSi = Op.LHS.second; 521f22ef01cSRoman Divacky llvm::Value *RHSr = Op.RHS.first, *RHSi = Op.RHS.second; 522f22ef01cSRoman Divacky 523f22ef01cSRoman Divacky 524f22ef01cSRoman Divacky llvm::Value *DSTr, *DSTi; 525f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 526f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 527f22ef01cSRoman Divacky llvm::Value *Tmp1 = Builder.CreateFMul(LHSr, RHSr, "tmp"); // a*c 528f22ef01cSRoman Divacky llvm::Value *Tmp2 = Builder.CreateFMul(LHSi, RHSi, "tmp"); // b*d 529f22ef01cSRoman Divacky llvm::Value *Tmp3 = Builder.CreateFAdd(Tmp1, Tmp2, "tmp"); // ac+bd 530f22ef01cSRoman Divacky 531f22ef01cSRoman Divacky llvm::Value *Tmp4 = Builder.CreateFMul(RHSr, RHSr, "tmp"); // c*c 532f22ef01cSRoman Divacky llvm::Value *Tmp5 = Builder.CreateFMul(RHSi, RHSi, "tmp"); // d*d 533f22ef01cSRoman Divacky llvm::Value *Tmp6 = Builder.CreateFAdd(Tmp4, Tmp5, "tmp"); // cc+dd 534f22ef01cSRoman Divacky 535f22ef01cSRoman Divacky llvm::Value *Tmp7 = Builder.CreateFMul(LHSi, RHSr, "tmp"); // b*c 536f22ef01cSRoman Divacky llvm::Value *Tmp8 = Builder.CreateFMul(LHSr, RHSi, "tmp"); // a*d 537f22ef01cSRoman Divacky llvm::Value *Tmp9 = Builder.CreateFSub(Tmp7, Tmp8, "tmp"); // bc-ad 538f22ef01cSRoman Divacky 539f22ef01cSRoman Divacky DSTr = Builder.CreateFDiv(Tmp3, Tmp6, "tmp"); 540f22ef01cSRoman Divacky DSTi = Builder.CreateFDiv(Tmp9, Tmp6, "tmp"); 541f22ef01cSRoman Divacky } else { 542f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 543f22ef01cSRoman Divacky llvm::Value *Tmp1 = Builder.CreateMul(LHSr, RHSr, "tmp"); // a*c 544f22ef01cSRoman Divacky llvm::Value *Tmp2 = Builder.CreateMul(LHSi, RHSi, "tmp"); // b*d 545f22ef01cSRoman Divacky llvm::Value *Tmp3 = Builder.CreateAdd(Tmp1, Tmp2, "tmp"); // ac+bd 546f22ef01cSRoman Divacky 547f22ef01cSRoman Divacky llvm::Value *Tmp4 = Builder.CreateMul(RHSr, RHSr, "tmp"); // c*c 548f22ef01cSRoman Divacky llvm::Value *Tmp5 = Builder.CreateMul(RHSi, RHSi, "tmp"); // d*d 549f22ef01cSRoman Divacky llvm::Value *Tmp6 = Builder.CreateAdd(Tmp4, Tmp5, "tmp"); // cc+dd 550f22ef01cSRoman Divacky 551f22ef01cSRoman Divacky llvm::Value *Tmp7 = Builder.CreateMul(LHSi, RHSr, "tmp"); // b*c 552f22ef01cSRoman Divacky llvm::Value *Tmp8 = Builder.CreateMul(LHSr, RHSi, "tmp"); // a*d 553f22ef01cSRoman Divacky llvm::Value *Tmp9 = Builder.CreateSub(Tmp7, Tmp8, "tmp"); // bc-ad 554f22ef01cSRoman Divacky 555f22ef01cSRoman Divacky if (Op.Ty->getAs<ComplexType>()->getElementType()->isUnsignedIntegerType()) { 556f22ef01cSRoman Divacky DSTr = Builder.CreateUDiv(Tmp3, Tmp6, "tmp"); 557f22ef01cSRoman Divacky DSTi = Builder.CreateUDiv(Tmp9, Tmp6, "tmp"); 558f22ef01cSRoman Divacky } else { 559f22ef01cSRoman Divacky DSTr = Builder.CreateSDiv(Tmp3, Tmp6, "tmp"); 560f22ef01cSRoman Divacky DSTi = Builder.CreateSDiv(Tmp9, Tmp6, "tmp"); 561f22ef01cSRoman Divacky } 562f22ef01cSRoman Divacky } 563f22ef01cSRoman Divacky 564f22ef01cSRoman Divacky return ComplexPairTy(DSTr, DSTi); 565f22ef01cSRoman Divacky } 566f22ef01cSRoman Divacky 567f22ef01cSRoman Divacky ComplexExprEmitter::BinOpInfo 568f22ef01cSRoman Divacky ComplexExprEmitter::EmitBinOps(const BinaryOperator *E) { 569f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 570f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 571f22ef01cSRoman Divacky BinOpInfo Ops; 572f22ef01cSRoman Divacky Ops.LHS = Visit(E->getLHS()); 573f22ef01cSRoman Divacky Ops.RHS = Visit(E->getRHS()); 574f22ef01cSRoman Divacky Ops.Ty = E->getType(); 575f22ef01cSRoman Divacky return Ops; 576f22ef01cSRoman Divacky } 577f22ef01cSRoman Divacky 578f22ef01cSRoman Divacky 5792754fe60SDimitry Andric LValue ComplexExprEmitter:: 5802754fe60SDimitry Andric EmitCompoundAssignLValue(const CompoundAssignOperator *E, 5812754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&), 5822754fe60SDimitry Andric ComplexPairTy &Val) { 583f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 584f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 5852754fe60SDimitry Andric QualType LHSTy = E->getLHS()->getType(); 586f22ef01cSRoman Divacky 587f22ef01cSRoman Divacky BinOpInfo OpInfo; 588f22ef01cSRoman Divacky 589f22ef01cSRoman Divacky // Load the RHS and LHS operands. 590f22ef01cSRoman Divacky // __block variables need to have the rhs evaluated first, plus this should 5912754fe60SDimitry Andric // improve codegen a little. 592f22ef01cSRoman Divacky OpInfo.Ty = E->getComputationResultType(); 5932754fe60SDimitry Andric 5942754fe60SDimitry Andric // The RHS should have been converted to the computation type. 5952754fe60SDimitry Andric assert(OpInfo.Ty->isAnyComplexType()); 5962754fe60SDimitry Andric assert(CGF.getContext().hasSameUnqualifiedType(OpInfo.Ty, 5972754fe60SDimitry Andric E->getRHS()->getType())); 5982754fe60SDimitry Andric OpInfo.RHS = Visit(E->getRHS()); 599f22ef01cSRoman Divacky 600ffd1746dSEd Schouten LValue LHS = CGF.EmitLValue(E->getLHS()); 6012754fe60SDimitry Andric 6022754fe60SDimitry Andric // Load from the l-value. 6032754fe60SDimitry Andric ComplexPairTy LHSComplexPair = EmitLoadOfLValue(LHS); 604f22ef01cSRoman Divacky 605f22ef01cSRoman Divacky OpInfo.LHS = EmitComplexToComplexCast(LHSComplexPair, LHSTy, OpInfo.Ty); 606f22ef01cSRoman Divacky 607f22ef01cSRoman Divacky // Expand the binary operator. 608f22ef01cSRoman Divacky ComplexPairTy Result = (this->*Func)(OpInfo); 609f22ef01cSRoman Divacky 610f22ef01cSRoman Divacky // Truncate the result back to the LHS type. 611f22ef01cSRoman Divacky Result = EmitComplexToComplexCast(Result, OpInfo.Ty, LHSTy); 6122754fe60SDimitry Andric Val = Result; 613f22ef01cSRoman Divacky 614f22ef01cSRoman Divacky // Store the result value into the LHS lvalue. 6152754fe60SDimitry Andric EmitStoreThroughLValue(Result, LHS); 616ffd1746dSEd Schouten 6172754fe60SDimitry Andric return LHS; 618f22ef01cSRoman Divacky } 619f22ef01cSRoman Divacky 6202754fe60SDimitry Andric // Compound assignments. 6212754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter:: 6222754fe60SDimitry Andric EmitCompoundAssign(const CompoundAssignOperator *E, 6232754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&)){ 6242754fe60SDimitry Andric ComplexPairTy Val; 6252754fe60SDimitry Andric LValue LV = EmitCompoundAssignLValue(E, Func, Val); 6262754fe60SDimitry Andric 6272754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value. 6282754fe60SDimitry Andric if (!CGF.getContext().getLangOptions().CPlusPlus) 6292754fe60SDimitry Andric return Val; 6302754fe60SDimitry Andric 6312754fe60SDimitry Andric // Objective-C property assignment never reloads the value following a store. 6322754fe60SDimitry Andric if (LV.isPropertyRef()) 6332754fe60SDimitry Andric return Val; 6342754fe60SDimitry Andric 6352754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 6362754fe60SDimitry Andric if (!LV.isVolatileQualified()) 6372754fe60SDimitry Andric return Val; 6382754fe60SDimitry Andric 6392754fe60SDimitry Andric return EmitLoadOfComplex(LV.getAddress(), LV.isVolatileQualified()); 6402754fe60SDimitry Andric } 6412754fe60SDimitry Andric 6422754fe60SDimitry Andric LValue ComplexExprEmitter::EmitBinAssignLValue(const BinaryOperator *E, 6432754fe60SDimitry Andric ComplexPairTy &Val) { 644ffd1746dSEd Schouten assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(), 645ffd1746dSEd Schouten E->getRHS()->getType()) && 646f22ef01cSRoman Divacky "Invalid assignment"); 6472754fe60SDimitry Andric TestAndClearIgnoreReal(); 6482754fe60SDimitry Andric TestAndClearIgnoreImag(); 6492754fe60SDimitry Andric 6502754fe60SDimitry Andric // Emit the RHS. __block variables need the RHS evaluated first. 6512754fe60SDimitry Andric Val = Visit(E->getRHS()); 652f22ef01cSRoman Divacky 653f22ef01cSRoman Divacky // Compute the address to store into. 654f22ef01cSRoman Divacky LValue LHS = CGF.EmitLValue(E->getLHS()); 655f22ef01cSRoman Divacky 656ffd1746dSEd Schouten // Store the result value into the LHS lvalue. 6572754fe60SDimitry Andric EmitStoreThroughLValue(Val, LHS); 658f22ef01cSRoman Divacky 6592754fe60SDimitry Andric return LHS; 6602754fe60SDimitry Andric } 661ffd1746dSEd Schouten 6622754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitBinAssign(const BinaryOperator *E) { 6632754fe60SDimitry Andric ComplexPairTy Val; 6642754fe60SDimitry Andric LValue LV = EmitBinAssignLValue(E, Val); 6652754fe60SDimitry Andric 6662754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value. 6672754fe60SDimitry Andric if (!CGF.getContext().getLangOptions().CPlusPlus) 668f22ef01cSRoman Divacky return Val; 669ffd1746dSEd Schouten 6702754fe60SDimitry Andric // Objective-C property assignment never reloads the value following a store. 6712754fe60SDimitry Andric if (LV.isPropertyRef()) 6722754fe60SDimitry Andric return Val; 6732754fe60SDimitry Andric 6742754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 6752754fe60SDimitry Andric if (!LV.isVolatileQualified()) 6762754fe60SDimitry Andric return Val; 6772754fe60SDimitry Andric 6782754fe60SDimitry Andric return EmitLoadOfComplex(LV.getAddress(), LV.isVolatileQualified()); 679f22ef01cSRoman Divacky } 680f22ef01cSRoman Divacky 681f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitBinComma(const BinaryOperator *E) { 6822754fe60SDimitry Andric CGF.EmitIgnoredExpr(E->getLHS()); 683f22ef01cSRoman Divacky return Visit(E->getRHS()); 684f22ef01cSRoman Divacky } 685f22ef01cSRoman Divacky 686f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter:: 6872754fe60SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) { 688f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 689f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 690f22ef01cSRoman Divacky llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true"); 691f22ef01cSRoman Divacky llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false"); 692f22ef01cSRoman Divacky llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end"); 693f22ef01cSRoman Divacky 6942754fe60SDimitry Andric // Bind the common expression if necessary. 6952754fe60SDimitry Andric CodeGenFunction::OpaqueValueMapping binding(CGF, E); 6962754fe60SDimitry Andric 6972754fe60SDimitry Andric CodeGenFunction::ConditionalEvaluation eval(CGF); 698f22ef01cSRoman Divacky CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock); 699f22ef01cSRoman Divacky 7002754fe60SDimitry Andric eval.begin(CGF); 701f22ef01cSRoman Divacky CGF.EmitBlock(LHSBlock); 7022754fe60SDimitry Andric ComplexPairTy LHS = Visit(E->getTrueExpr()); 703f22ef01cSRoman Divacky LHSBlock = Builder.GetInsertBlock(); 704f22ef01cSRoman Divacky CGF.EmitBranch(ContBlock); 7052754fe60SDimitry Andric eval.end(CGF); 706f22ef01cSRoman Divacky 7072754fe60SDimitry Andric eval.begin(CGF); 708f22ef01cSRoman Divacky CGF.EmitBlock(RHSBlock); 7092754fe60SDimitry Andric ComplexPairTy RHS = Visit(E->getFalseExpr()); 710f22ef01cSRoman Divacky RHSBlock = Builder.GetInsertBlock(); 711f22ef01cSRoman Divacky CGF.EmitBlock(ContBlock); 7122754fe60SDimitry Andric eval.end(CGF); 713f22ef01cSRoman Divacky 714f22ef01cSRoman Divacky // Create a PHI node for the real part. 7153b0f4066SDimitry Andric llvm::PHINode *RealPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.r"); 716f22ef01cSRoman Divacky RealPN->addIncoming(LHS.first, LHSBlock); 717f22ef01cSRoman Divacky RealPN->addIncoming(RHS.first, RHSBlock); 718f22ef01cSRoman Divacky 719f22ef01cSRoman Divacky // Create a PHI node for the imaginary part. 7203b0f4066SDimitry Andric llvm::PHINode *ImagPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.i"); 721f22ef01cSRoman Divacky ImagPN->addIncoming(LHS.second, LHSBlock); 722f22ef01cSRoman Divacky ImagPN->addIncoming(RHS.second, RHSBlock); 723f22ef01cSRoman Divacky 724f22ef01cSRoman Divacky return ComplexPairTy(RealPN, ImagPN); 725f22ef01cSRoman Divacky } 726f22ef01cSRoman Divacky 727f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitChooseExpr(ChooseExpr *E) { 728f22ef01cSRoman Divacky return Visit(E->getChosenSubExpr(CGF.getContext())); 729f22ef01cSRoman Divacky } 730f22ef01cSRoman Divacky 731f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitInitListExpr(InitListExpr *E) { 732f22ef01cSRoman Divacky bool Ignore = TestAndClearIgnoreReal(); 733f22ef01cSRoman Divacky (void)Ignore; 734f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored"); 735f22ef01cSRoman Divacky Ignore = TestAndClearIgnoreImag(); 736f22ef01cSRoman Divacky (void)Ignore; 737f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored"); 738f22ef01cSRoman Divacky if (E->getNumInits()) 739f22ef01cSRoman Divacky return Visit(E->getInit(0)); 740f22ef01cSRoman Divacky 741f22ef01cSRoman Divacky // Empty init list intializes to null 742f22ef01cSRoman Divacky QualType Ty = E->getType()->getAs<ComplexType>()->getElementType(); 743f22ef01cSRoman Divacky const llvm::Type* LTy = CGF.ConvertType(Ty); 744f22ef01cSRoman Divacky llvm::Value* zeroConstant = llvm::Constant::getNullValue(LTy); 745f22ef01cSRoman Divacky return ComplexPairTy(zeroConstant, zeroConstant); 746f22ef01cSRoman Divacky } 747f22ef01cSRoman Divacky 748f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitVAArgExpr(VAArgExpr *E) { 749f22ef01cSRoman Divacky llvm::Value *ArgValue = CGF.EmitVAListRef(E->getSubExpr()); 750f22ef01cSRoman Divacky llvm::Value *ArgPtr = CGF.EmitVAArg(ArgValue, E->getType()); 751f22ef01cSRoman Divacky 752f22ef01cSRoman Divacky if (!ArgPtr) { 753f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex va_arg expression"); 754f22ef01cSRoman Divacky const llvm::Type *EltTy = 755f22ef01cSRoman Divacky CGF.ConvertType(E->getType()->getAs<ComplexType>()->getElementType()); 756f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy); 757f22ef01cSRoman Divacky return ComplexPairTy(U, U); 758f22ef01cSRoman Divacky } 759f22ef01cSRoman Divacky 760f22ef01cSRoman Divacky // FIXME Volatility. 761f22ef01cSRoman Divacky return EmitLoadOfComplex(ArgPtr, false); 762f22ef01cSRoman Divacky } 763f22ef01cSRoman Divacky 764f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 765f22ef01cSRoman Divacky // Entry Point into this File 766f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 767f22ef01cSRoman Divacky 768f22ef01cSRoman Divacky /// EmitComplexExpr - Emit the computation of the specified expression of 769f22ef01cSRoman Divacky /// complex type, ignoring the result. 770f22ef01cSRoman Divacky ComplexPairTy CodeGenFunction::EmitComplexExpr(const Expr *E, bool IgnoreReal, 7712754fe60SDimitry Andric bool IgnoreImag) { 772f22ef01cSRoman Divacky assert(E && E->getType()->isAnyComplexType() && 773f22ef01cSRoman Divacky "Invalid complex expression to emit"); 774f22ef01cSRoman Divacky 7752754fe60SDimitry Andric return ComplexExprEmitter(*this, IgnoreReal, IgnoreImag) 776f22ef01cSRoman Divacky .Visit(const_cast<Expr*>(E)); 777f22ef01cSRoman Divacky } 778f22ef01cSRoman Divacky 779f22ef01cSRoman Divacky /// EmitComplexExprIntoAddr - Emit the computation of the specified expression 780f22ef01cSRoman Divacky /// of complex type, storing into the specified Value*. 781f22ef01cSRoman Divacky void CodeGenFunction::EmitComplexExprIntoAddr(const Expr *E, 782f22ef01cSRoman Divacky llvm::Value *DestAddr, 783f22ef01cSRoman Divacky bool DestIsVolatile) { 784f22ef01cSRoman Divacky assert(E && E->getType()->isAnyComplexType() && 785f22ef01cSRoman Divacky "Invalid complex expression to emit"); 786f22ef01cSRoman Divacky ComplexExprEmitter Emitter(*this); 787f22ef01cSRoman Divacky ComplexPairTy Val = Emitter.Visit(const_cast<Expr*>(E)); 788f22ef01cSRoman Divacky Emitter.EmitStoreOfComplex(Val, DestAddr, DestIsVolatile); 789f22ef01cSRoman Divacky } 790f22ef01cSRoman Divacky 791f22ef01cSRoman Divacky /// StoreComplexToAddr - Store a complex number into the specified address. 792f22ef01cSRoman Divacky void CodeGenFunction::StoreComplexToAddr(ComplexPairTy V, 793f22ef01cSRoman Divacky llvm::Value *DestAddr, 794f22ef01cSRoman Divacky bool DestIsVolatile) { 795f22ef01cSRoman Divacky ComplexExprEmitter(*this).EmitStoreOfComplex(V, DestAddr, DestIsVolatile); 796f22ef01cSRoman Divacky } 797f22ef01cSRoman Divacky 798f22ef01cSRoman Divacky /// LoadComplexFromAddr - Load a complex number from the specified address. 799f22ef01cSRoman Divacky ComplexPairTy CodeGenFunction::LoadComplexFromAddr(llvm::Value *SrcAddr, 800f22ef01cSRoman Divacky bool SrcIsVolatile) { 801f22ef01cSRoman Divacky return ComplexExprEmitter(*this).EmitLoadOfComplex(SrcAddr, SrcIsVolatile); 802f22ef01cSRoman Divacky } 8032754fe60SDimitry Andric 8042754fe60SDimitry Andric LValue CodeGenFunction::EmitComplexAssignmentLValue(const BinaryOperator *E) { 8052754fe60SDimitry Andric assert(E->getOpcode() == BO_Assign); 8062754fe60SDimitry Andric ComplexPairTy Val; // ignored 8072754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitBinAssignLValue(E, Val); 8082754fe60SDimitry Andric } 8092754fe60SDimitry Andric 8102754fe60SDimitry Andric LValue CodeGenFunction:: 8112754fe60SDimitry Andric EmitComplexCompoundAssignmentLValue(const CompoundAssignOperator *E) { 8122754fe60SDimitry Andric ComplexPairTy(ComplexExprEmitter::*Op)(const ComplexExprEmitter::BinOpInfo &); 8132754fe60SDimitry Andric switch (E->getOpcode()) { 8142754fe60SDimitry Andric case BO_MulAssign: Op = &ComplexExprEmitter::EmitBinMul; break; 8152754fe60SDimitry Andric case BO_DivAssign: Op = &ComplexExprEmitter::EmitBinDiv; break; 8162754fe60SDimitry Andric case BO_SubAssign: Op = &ComplexExprEmitter::EmitBinSub; break; 8172754fe60SDimitry Andric case BO_AddAssign: Op = &ComplexExprEmitter::EmitBinAdd; break; 8182754fe60SDimitry Andric 8192754fe60SDimitry Andric default: 8202754fe60SDimitry Andric llvm_unreachable("unexpected complex compound assignment"); 8212754fe60SDimitry Andric Op = 0; 8222754fe60SDimitry Andric } 8232754fe60SDimitry Andric 8242754fe60SDimitry Andric ComplexPairTy Val; // ignored 8252754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val); 8262754fe60SDimitry Andric } 827