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/ADT/SmallString.h" 19139f7f9bSDimitry Andric #include "llvm/IR/Constants.h" 20139f7f9bSDimitry Andric #include "llvm/IR/Function.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 30139f7f9bSDimitry Andric /// Return the complex type that we are meant to emit. 31139f7f9bSDimitry Andric static const ComplexType *getComplexType(QualType type) { 32139f7f9bSDimitry Andric type = type.getCanonicalType(); 33139f7f9bSDimitry Andric if (const ComplexType *comp = dyn_cast<ComplexType>(type)) { 34139f7f9bSDimitry Andric return comp; 35139f7f9bSDimitry Andric } else { 36139f7f9bSDimitry Andric return cast<ComplexType>(cast<AtomicType>(type)->getValueType()); 37139f7f9bSDimitry Andric } 38139f7f9bSDimitry Andric } 39139f7f9bSDimitry Andric 40f22ef01cSRoman Divacky namespace { 41f22ef01cSRoman Divacky class ComplexExprEmitter 42f22ef01cSRoman Divacky : public StmtVisitor<ComplexExprEmitter, ComplexPairTy> { 43f22ef01cSRoman Divacky CodeGenFunction &CGF; 44f22ef01cSRoman Divacky CGBuilderTy &Builder; 45f22ef01cSRoman Divacky bool IgnoreReal; 46f22ef01cSRoman Divacky bool IgnoreImag; 47f22ef01cSRoman Divacky public: 482754fe60SDimitry Andric ComplexExprEmitter(CodeGenFunction &cgf, bool ir=false, bool ii=false) 492754fe60SDimitry Andric : CGF(cgf), Builder(CGF.Builder), IgnoreReal(ir), IgnoreImag(ii) { 50f22ef01cSRoman Divacky } 51f22ef01cSRoman Divacky 52f22ef01cSRoman Divacky 53f22ef01cSRoman Divacky //===--------------------------------------------------------------------===// 54f22ef01cSRoman Divacky // Utilities 55f22ef01cSRoman Divacky //===--------------------------------------------------------------------===// 56f22ef01cSRoman Divacky 57f22ef01cSRoman Divacky bool TestAndClearIgnoreReal() { 58f22ef01cSRoman Divacky bool I = IgnoreReal; 59f22ef01cSRoman Divacky IgnoreReal = false; 60f22ef01cSRoman Divacky return I; 61f22ef01cSRoman Divacky } 62f22ef01cSRoman Divacky bool TestAndClearIgnoreImag() { 63f22ef01cSRoman Divacky bool I = IgnoreImag; 64f22ef01cSRoman Divacky IgnoreImag = false; 65f22ef01cSRoman Divacky return I; 66f22ef01cSRoman Divacky } 67f22ef01cSRoman Divacky 68f22ef01cSRoman Divacky /// EmitLoadOfLValue - Given an expression with complex type that represents a 69f22ef01cSRoman Divacky /// value l-value, this method emits the address of the l-value, then loads 70f22ef01cSRoman Divacky /// and returns the result. 71f22ef01cSRoman Divacky ComplexPairTy EmitLoadOfLValue(const Expr *E) { 722754fe60SDimitry Andric return EmitLoadOfLValue(CGF.EmitLValue(E)); 732754fe60SDimitry Andric } 742754fe60SDimitry Andric 75139f7f9bSDimitry Andric ComplexPairTy EmitLoadOfLValue(LValue LV); 762754fe60SDimitry Andric 77f22ef01cSRoman Divacky /// EmitStoreOfComplex - Store the specified real/imag parts into the 78f22ef01cSRoman Divacky /// specified value pointer. 79139f7f9bSDimitry Andric void EmitStoreOfComplex(ComplexPairTy Val, LValue LV, bool isInit); 80f22ef01cSRoman Divacky 81f22ef01cSRoman Divacky /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType. 82f22ef01cSRoman Divacky ComplexPairTy EmitComplexToComplexCast(ComplexPairTy Val, QualType SrcType, 83f22ef01cSRoman Divacky QualType DestType); 84f22ef01cSRoman Divacky 85f22ef01cSRoman Divacky //===--------------------------------------------------------------------===// 86f22ef01cSRoman Divacky // Visitor Methods 87f22ef01cSRoman Divacky //===--------------------------------------------------------------------===// 88f22ef01cSRoman Divacky 892754fe60SDimitry Andric ComplexPairTy Visit(Expr *E) { 902754fe60SDimitry Andric return StmtVisitor<ComplexExprEmitter, ComplexPairTy>::Visit(E); 912754fe60SDimitry Andric } 922754fe60SDimitry Andric 93f22ef01cSRoman Divacky ComplexPairTy VisitStmt(Stmt *S) { 94f22ef01cSRoman Divacky S->dump(CGF.getContext().getSourceManager()); 956122f3e6SDimitry Andric llvm_unreachable("Stmt can't have complex result type!"); 96f22ef01cSRoman Divacky } 97f22ef01cSRoman Divacky ComplexPairTy VisitExpr(Expr *S); 98f22ef01cSRoman Divacky ComplexPairTy VisitParenExpr(ParenExpr *PE) { return Visit(PE->getSubExpr());} 993b0f4066SDimitry Andric ComplexPairTy VisitGenericSelectionExpr(GenericSelectionExpr *GE) { 1003b0f4066SDimitry Andric return Visit(GE->getResultExpr()); 1013b0f4066SDimitry Andric } 102f22ef01cSRoman Divacky ComplexPairTy VisitImaginaryLiteral(const ImaginaryLiteral *IL); 10317a519f9SDimitry Andric ComplexPairTy 10417a519f9SDimitry Andric VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *PE) { 10517a519f9SDimitry Andric return Visit(PE->getReplacement()); 10617a519f9SDimitry Andric } 107f22ef01cSRoman Divacky 108f22ef01cSRoman Divacky // l-values. 109dff0c46cSDimitry Andric ComplexPairTy VisitDeclRefExpr(DeclRefExpr *E) { 110dff0c46cSDimitry Andric if (CodeGenFunction::ConstantEmission result = CGF.tryEmitAsConstant(E)) { 111dff0c46cSDimitry Andric if (result.isReference()) 112dff0c46cSDimitry Andric return EmitLoadOfLValue(result.getReferenceLValue(CGF, E)); 113dff0c46cSDimitry Andric 114dff0c46cSDimitry Andric llvm::ConstantStruct *pair = 115dff0c46cSDimitry Andric cast<llvm::ConstantStruct>(result.getValue()); 116dff0c46cSDimitry Andric return ComplexPairTy(pair->getOperand(0), pair->getOperand(1)); 117dff0c46cSDimitry Andric } 118f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 119f22ef01cSRoman Divacky } 120dff0c46cSDimitry Andric ComplexPairTy VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) { 121f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 122f22ef01cSRoman Divacky } 123f22ef01cSRoman Divacky ComplexPairTy VisitObjCMessageExpr(ObjCMessageExpr *E) { 124f22ef01cSRoman Divacky return CGF.EmitObjCMessageExpr(E).getComplexVal(); 125f22ef01cSRoman Divacky } 126f22ef01cSRoman Divacky ComplexPairTy VisitArraySubscriptExpr(Expr *E) { return EmitLoadOfLValue(E); } 127f22ef01cSRoman Divacky ComplexPairTy VisitMemberExpr(const Expr *E) { return EmitLoadOfLValue(E); } 1282754fe60SDimitry Andric ComplexPairTy VisitOpaqueValueExpr(OpaqueValueExpr *E) { 1292754fe60SDimitry Andric if (E->isGLValue()) 1302754fe60SDimitry Andric return EmitLoadOfLValue(CGF.getOpaqueLValueMapping(E)); 1312754fe60SDimitry Andric return CGF.getOpaqueRValueMapping(E).getComplexVal(); 1322754fe60SDimitry Andric } 133f22ef01cSRoman Divacky 134dff0c46cSDimitry Andric ComplexPairTy VisitPseudoObjectExpr(PseudoObjectExpr *E) { 135dff0c46cSDimitry Andric return CGF.EmitPseudoObjectRValue(E).getComplexVal(); 136dff0c46cSDimitry Andric } 137dff0c46cSDimitry Andric 138f22ef01cSRoman Divacky // FIXME: CompoundLiteralExpr 139f22ef01cSRoman Divacky 140ffd1746dSEd Schouten ComplexPairTy EmitCast(CastExpr::CastKind CK, Expr *Op, QualType DestTy); 141f22ef01cSRoman Divacky ComplexPairTy VisitImplicitCastExpr(ImplicitCastExpr *E) { 142f22ef01cSRoman Divacky // Unlike for scalars, we don't have to worry about function->ptr demotion 143f22ef01cSRoman Divacky // here. 144ffd1746dSEd Schouten return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType()); 145f22ef01cSRoman Divacky } 146f22ef01cSRoman Divacky ComplexPairTy VisitCastExpr(CastExpr *E) { 147ffd1746dSEd Schouten return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType()); 148f22ef01cSRoman Divacky } 149f22ef01cSRoman Divacky ComplexPairTy VisitCallExpr(const CallExpr *E); 150f22ef01cSRoman Divacky ComplexPairTy VisitStmtExpr(const StmtExpr *E); 151f22ef01cSRoman Divacky 152f22ef01cSRoman Divacky // Operators. 153f22ef01cSRoman Divacky ComplexPairTy VisitPrePostIncDec(const UnaryOperator *E, 154f22ef01cSRoman Divacky bool isInc, bool isPre) { 155f22ef01cSRoman Divacky LValue LV = CGF.EmitLValue(E->getSubExpr()); 156f22ef01cSRoman Divacky return CGF.EmitComplexPrePostIncDec(E, LV, isInc, isPre); 157f22ef01cSRoman Divacky } 158f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPostDec(const UnaryOperator *E) { 159f22ef01cSRoman Divacky return VisitPrePostIncDec(E, false, false); 160f22ef01cSRoman Divacky } 161f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPostInc(const UnaryOperator *E) { 162f22ef01cSRoman Divacky return VisitPrePostIncDec(E, true, false); 163f22ef01cSRoman Divacky } 164f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPreDec(const UnaryOperator *E) { 165f22ef01cSRoman Divacky return VisitPrePostIncDec(E, false, true); 166f22ef01cSRoman Divacky } 167f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPreInc(const UnaryOperator *E) { 168f22ef01cSRoman Divacky return VisitPrePostIncDec(E, true, true); 169f22ef01cSRoman Divacky } 170f22ef01cSRoman Divacky ComplexPairTy VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); } 171f22ef01cSRoman Divacky ComplexPairTy VisitUnaryPlus (const UnaryOperator *E) { 172f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 173f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 174f22ef01cSRoman Divacky return Visit(E->getSubExpr()); 175f22ef01cSRoman Divacky } 176f22ef01cSRoman Divacky ComplexPairTy VisitUnaryMinus (const UnaryOperator *E); 177f22ef01cSRoman Divacky ComplexPairTy VisitUnaryNot (const UnaryOperator *E); 178f22ef01cSRoman Divacky // LNot,Real,Imag never return complex. 179f22ef01cSRoman Divacky ComplexPairTy VisitUnaryExtension(const UnaryOperator *E) { 180f22ef01cSRoman Divacky return Visit(E->getSubExpr()); 181f22ef01cSRoman Divacky } 182f22ef01cSRoman Divacky ComplexPairTy VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) { 183f22ef01cSRoman Divacky return Visit(DAE->getExpr()); 184f22ef01cSRoman Divacky } 185284c1978SDimitry Andric ComplexPairTy VisitCXXDefaultInitExpr(CXXDefaultInitExpr *DIE) { 186284c1978SDimitry Andric CodeGenFunction::CXXDefaultInitExprScope Scope(CGF); 187284c1978SDimitry Andric return Visit(DIE->getExpr()); 188284c1978SDimitry Andric } 1892754fe60SDimitry Andric ComplexPairTy VisitExprWithCleanups(ExprWithCleanups *E) { 190dff0c46cSDimitry Andric CGF.enterFullExpression(E); 191dff0c46cSDimitry Andric CodeGenFunction::RunCleanupsScope Scope(CGF); 192dff0c46cSDimitry Andric return Visit(E->getSubExpr()); 193f22ef01cSRoman Divacky } 194ffd1746dSEd Schouten ComplexPairTy VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) { 195f22ef01cSRoman Divacky assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 196139f7f9bSDimitry Andric QualType Elem = E->getType()->castAs<ComplexType>()->getElementType(); 197f22ef01cSRoman Divacky llvm::Constant *Null = llvm::Constant::getNullValue(CGF.ConvertType(Elem)); 198f22ef01cSRoman Divacky return ComplexPairTy(Null, Null); 199f22ef01cSRoman Divacky } 200f22ef01cSRoman Divacky ComplexPairTy VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) { 201f22ef01cSRoman Divacky assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 202139f7f9bSDimitry Andric QualType Elem = E->getType()->castAs<ComplexType>()->getElementType(); 203f22ef01cSRoman Divacky llvm::Constant *Null = 204f22ef01cSRoman Divacky llvm::Constant::getNullValue(CGF.ConvertType(Elem)); 205f22ef01cSRoman Divacky return ComplexPairTy(Null, Null); 206f22ef01cSRoman Divacky } 207f22ef01cSRoman Divacky 208f22ef01cSRoman Divacky struct BinOpInfo { 209f22ef01cSRoman Divacky ComplexPairTy LHS; 210f22ef01cSRoman Divacky ComplexPairTy RHS; 211f22ef01cSRoman Divacky QualType Ty; // Computation Type. 212f22ef01cSRoman Divacky }; 213f22ef01cSRoman Divacky 214f22ef01cSRoman Divacky BinOpInfo EmitBinOps(const BinaryOperator *E); 2152754fe60SDimitry Andric LValue EmitCompoundAssignLValue(const CompoundAssignOperator *E, 2162754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func) 2172754fe60SDimitry Andric (const BinOpInfo &), 2182754fe60SDimitry Andric ComplexPairTy &Val); 219f22ef01cSRoman Divacky ComplexPairTy EmitCompoundAssign(const CompoundAssignOperator *E, 220f22ef01cSRoman Divacky ComplexPairTy (ComplexExprEmitter::*Func) 221f22ef01cSRoman Divacky (const BinOpInfo &)); 222f22ef01cSRoman Divacky 223f22ef01cSRoman Divacky ComplexPairTy EmitBinAdd(const BinOpInfo &Op); 224f22ef01cSRoman Divacky ComplexPairTy EmitBinSub(const BinOpInfo &Op); 225f22ef01cSRoman Divacky ComplexPairTy EmitBinMul(const BinOpInfo &Op); 226f22ef01cSRoman Divacky ComplexPairTy EmitBinDiv(const BinOpInfo &Op); 227f22ef01cSRoman Divacky 228f22ef01cSRoman Divacky ComplexPairTy VisitBinAdd(const BinaryOperator *E) { 229f22ef01cSRoman Divacky return EmitBinAdd(EmitBinOps(E)); 230f22ef01cSRoman Divacky } 231f22ef01cSRoman Divacky ComplexPairTy VisitBinSub(const BinaryOperator *E) { 232f22ef01cSRoman Divacky return EmitBinSub(EmitBinOps(E)); 233f22ef01cSRoman Divacky } 2342754fe60SDimitry Andric ComplexPairTy VisitBinMul(const BinaryOperator *E) { 2352754fe60SDimitry Andric return EmitBinMul(EmitBinOps(E)); 2362754fe60SDimitry Andric } 237f22ef01cSRoman Divacky ComplexPairTy VisitBinDiv(const BinaryOperator *E) { 238f22ef01cSRoman Divacky return EmitBinDiv(EmitBinOps(E)); 239f22ef01cSRoman Divacky } 240f22ef01cSRoman Divacky 241f22ef01cSRoman Divacky // Compound assignments. 242f22ef01cSRoman Divacky ComplexPairTy VisitBinAddAssign(const CompoundAssignOperator *E) { 243f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinAdd); 244f22ef01cSRoman Divacky } 245f22ef01cSRoman Divacky ComplexPairTy VisitBinSubAssign(const CompoundAssignOperator *E) { 246f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinSub); 247f22ef01cSRoman Divacky } 248f22ef01cSRoman Divacky ComplexPairTy VisitBinMulAssign(const CompoundAssignOperator *E) { 249f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinMul); 250f22ef01cSRoman Divacky } 251f22ef01cSRoman Divacky ComplexPairTy VisitBinDivAssign(const CompoundAssignOperator *E) { 252f22ef01cSRoman Divacky return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinDiv); 253f22ef01cSRoman Divacky } 254f22ef01cSRoman Divacky 255f22ef01cSRoman Divacky // GCC rejects rem/and/or/xor for integer complex. 256f22ef01cSRoman Divacky // Logical and/or always return int, never complex. 257f22ef01cSRoman Divacky 258f22ef01cSRoman Divacky // No comparisons produce a complex result. 2592754fe60SDimitry Andric 2602754fe60SDimitry Andric LValue EmitBinAssignLValue(const BinaryOperator *E, 2612754fe60SDimitry Andric ComplexPairTy &Val); 262f22ef01cSRoman Divacky ComplexPairTy VisitBinAssign (const BinaryOperator *E); 263f22ef01cSRoman Divacky ComplexPairTy VisitBinComma (const BinaryOperator *E); 264f22ef01cSRoman Divacky 265f22ef01cSRoman Divacky 2662754fe60SDimitry Andric ComplexPairTy 2672754fe60SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *CO); 268f22ef01cSRoman Divacky ComplexPairTy VisitChooseExpr(ChooseExpr *CE); 269f22ef01cSRoman Divacky 270f22ef01cSRoman Divacky ComplexPairTy VisitInitListExpr(InitListExpr *E); 271f22ef01cSRoman Divacky 272dff0c46cSDimitry Andric ComplexPairTy VisitCompoundLiteralExpr(CompoundLiteralExpr *E) { 273dff0c46cSDimitry Andric return EmitLoadOfLValue(E); 274dff0c46cSDimitry Andric } 275dff0c46cSDimitry Andric 276f22ef01cSRoman Divacky ComplexPairTy VisitVAArgExpr(VAArgExpr *E); 2776122f3e6SDimitry Andric 2786122f3e6SDimitry Andric ComplexPairTy VisitAtomicExpr(AtomicExpr *E) { 2796122f3e6SDimitry Andric return CGF.EmitAtomicExpr(E).getComplexVal(); 2806122f3e6SDimitry Andric } 281f22ef01cSRoman Divacky }; 282f22ef01cSRoman Divacky } // end anonymous namespace. 283f22ef01cSRoman Divacky 284f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 285f22ef01cSRoman Divacky // Utilities 286f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 287f22ef01cSRoman Divacky 288139f7f9bSDimitry Andric /// EmitLoadOfLValue - Given an RValue reference for a complex, emit code to 289f22ef01cSRoman Divacky /// load the real and imaginary pieces, returning them as Real/Imag. 290139f7f9bSDimitry Andric ComplexPairTy ComplexExprEmitter::EmitLoadOfLValue(LValue lvalue) { 291139f7f9bSDimitry Andric assert(lvalue.isSimple() && "non-simple complex l-value?"); 292139f7f9bSDimitry Andric if (lvalue.getType()->isAtomicType()) 293139f7f9bSDimitry Andric return CGF.EmitAtomicLoad(lvalue).getComplexVal(); 294139f7f9bSDimitry Andric 295139f7f9bSDimitry Andric llvm::Value *SrcPtr = lvalue.getAddress(); 296139f7f9bSDimitry Andric bool isVolatile = lvalue.isVolatileQualified(); 297139f7f9bSDimitry Andric 298f22ef01cSRoman Divacky llvm::Value *Real=0, *Imag=0; 299f22ef01cSRoman Divacky 3002754fe60SDimitry Andric if (!IgnoreReal || isVolatile) { 301f22ef01cSRoman Divacky llvm::Value *RealP = Builder.CreateStructGEP(SrcPtr, 0, 302f22ef01cSRoman Divacky SrcPtr->getName() + ".realp"); 303f22ef01cSRoman Divacky Real = Builder.CreateLoad(RealP, isVolatile, SrcPtr->getName() + ".real"); 304f22ef01cSRoman Divacky } 305f22ef01cSRoman Divacky 3062754fe60SDimitry Andric if (!IgnoreImag || isVolatile) { 307f22ef01cSRoman Divacky llvm::Value *ImagP = Builder.CreateStructGEP(SrcPtr, 1, 308f22ef01cSRoman Divacky SrcPtr->getName() + ".imagp"); 309f22ef01cSRoman Divacky Imag = Builder.CreateLoad(ImagP, isVolatile, SrcPtr->getName() + ".imag"); 310f22ef01cSRoman Divacky } 311f22ef01cSRoman Divacky return ComplexPairTy(Real, Imag); 312f22ef01cSRoman Divacky } 313f22ef01cSRoman Divacky 314f22ef01cSRoman Divacky /// EmitStoreOfComplex - Store the specified real/imag parts into the 315f22ef01cSRoman Divacky /// specified value pointer. 316139f7f9bSDimitry Andric void ComplexExprEmitter::EmitStoreOfComplex(ComplexPairTy Val, 317139f7f9bSDimitry Andric LValue lvalue, 318139f7f9bSDimitry Andric bool isInit) { 319139f7f9bSDimitry Andric if (lvalue.getType()->isAtomicType()) 320139f7f9bSDimitry Andric return CGF.EmitAtomicStore(RValue::getComplex(Val), lvalue, isInit); 321139f7f9bSDimitry Andric 322139f7f9bSDimitry Andric llvm::Value *Ptr = lvalue.getAddress(); 323f22ef01cSRoman Divacky llvm::Value *RealPtr = Builder.CreateStructGEP(Ptr, 0, "real"); 324f22ef01cSRoman Divacky llvm::Value *ImagPtr = Builder.CreateStructGEP(Ptr, 1, "imag"); 325f22ef01cSRoman Divacky 326139f7f9bSDimitry Andric // TODO: alignment 327139f7f9bSDimitry Andric Builder.CreateStore(Val.first, RealPtr, lvalue.isVolatileQualified()); 328139f7f9bSDimitry Andric Builder.CreateStore(Val.second, ImagPtr, lvalue.isVolatileQualified()); 329f22ef01cSRoman Divacky } 330f22ef01cSRoman Divacky 331f22ef01cSRoman Divacky 332f22ef01cSRoman Divacky 333f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 334f22ef01cSRoman Divacky // Visitor Methods 335f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 336f22ef01cSRoman Divacky 337f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitExpr(Expr *E) { 338f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex expression"); 3396122f3e6SDimitry Andric llvm::Type *EltTy = 340139f7f9bSDimitry Andric CGF.ConvertType(getComplexType(E->getType())->getElementType()); 341f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy); 342f22ef01cSRoman Divacky return ComplexPairTy(U, U); 343f22ef01cSRoman Divacky } 344f22ef01cSRoman Divacky 345f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter:: 346f22ef01cSRoman Divacky VisitImaginaryLiteral(const ImaginaryLiteral *IL) { 347f22ef01cSRoman Divacky llvm::Value *Imag = CGF.EmitScalarExpr(IL->getSubExpr()); 3482754fe60SDimitry Andric return ComplexPairTy(llvm::Constant::getNullValue(Imag->getType()), Imag); 349f22ef01cSRoman Divacky } 350f22ef01cSRoman Divacky 351f22ef01cSRoman Divacky 352f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitCallExpr(const CallExpr *E) { 353f22ef01cSRoman Divacky if (E->getCallReturnType()->isReferenceType()) 354f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 355f22ef01cSRoman Divacky 356f22ef01cSRoman Divacky return CGF.EmitCallExpr(E).getComplexVal(); 357f22ef01cSRoman Divacky } 358f22ef01cSRoman Divacky 359f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitStmtExpr(const StmtExpr *E) { 3602754fe60SDimitry Andric CodeGenFunction::StmtExprEvaluation eval(CGF); 361f22ef01cSRoman Divacky return CGF.EmitCompoundStmt(*E->getSubStmt(), true).getComplexVal(); 362f22ef01cSRoman Divacky } 363f22ef01cSRoman Divacky 364f22ef01cSRoman Divacky /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType. 365f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitComplexToComplexCast(ComplexPairTy Val, 366f22ef01cSRoman Divacky QualType SrcType, 367f22ef01cSRoman Divacky QualType DestType) { 368f22ef01cSRoman Divacky // Get the src/dest element type. 369139f7f9bSDimitry Andric SrcType = SrcType->castAs<ComplexType>()->getElementType(); 370139f7f9bSDimitry Andric DestType = DestType->castAs<ComplexType>()->getElementType(); 371f22ef01cSRoman Divacky 372f22ef01cSRoman Divacky // C99 6.3.1.6: When a value of complex type is converted to another 373f22ef01cSRoman Divacky // complex type, both the real and imaginary parts follow the conversion 374f22ef01cSRoman Divacky // rules for the corresponding real types. 375f22ef01cSRoman Divacky Val.first = CGF.EmitScalarConversion(Val.first, SrcType, DestType); 376f22ef01cSRoman Divacky Val.second = CGF.EmitScalarConversion(Val.second, SrcType, DestType); 377f22ef01cSRoman Divacky return Val; 378f22ef01cSRoman Divacky } 379f22ef01cSRoman Divacky 380ffd1746dSEd Schouten ComplexPairTy ComplexExprEmitter::EmitCast(CastExpr::CastKind CK, Expr *Op, 381ffd1746dSEd Schouten QualType DestTy) { 3822754fe60SDimitry Andric switch (CK) { 38317a519f9SDimitry Andric case CK_Dependent: llvm_unreachable("dependent cast kind in IR gen!"); 38417a519f9SDimitry Andric 385dff0c46cSDimitry Andric // Atomic to non-atomic casts may be more than a no-op for some platforms and 386dff0c46cSDimitry Andric // for some types. 387dff0c46cSDimitry Andric case CK_AtomicToNonAtomic: 388dff0c46cSDimitry Andric case CK_NonAtomicToAtomic: 3892754fe60SDimitry Andric case CK_NoOp: 3902754fe60SDimitry Andric case CK_LValueToRValue: 39117a519f9SDimitry Andric case CK_UserDefinedConversion: 3922754fe60SDimitry Andric return Visit(Op); 3932754fe60SDimitry Andric 39417a519f9SDimitry Andric case CK_LValueBitCast: { 395139f7f9bSDimitry Andric LValue origLV = CGF.EmitLValue(Op); 396139f7f9bSDimitry Andric llvm::Value *V = origLV.getAddress(); 397ffd1746dSEd Schouten V = Builder.CreateBitCast(V, 398ffd1746dSEd Schouten CGF.ConvertType(CGF.getContext().getPointerType(DestTy))); 399139f7f9bSDimitry Andric return EmitLoadOfLValue(CGF.MakeAddrLValue(V, DestTy, 400139f7f9bSDimitry Andric origLV.getAlignment())); 401ffd1746dSEd Schouten } 402ffd1746dSEd Schouten 40317a519f9SDimitry Andric case CK_BitCast: 40417a519f9SDimitry Andric case CK_BaseToDerived: 40517a519f9SDimitry Andric case CK_DerivedToBase: 40617a519f9SDimitry Andric case CK_UncheckedDerivedToBase: 40717a519f9SDimitry Andric case CK_Dynamic: 40817a519f9SDimitry Andric case CK_ToUnion: 40917a519f9SDimitry Andric case CK_ArrayToPointerDecay: 41017a519f9SDimitry Andric case CK_FunctionToPointerDecay: 41117a519f9SDimitry Andric case CK_NullToPointer: 41217a519f9SDimitry Andric case CK_NullToMemberPointer: 41317a519f9SDimitry Andric case CK_BaseToDerivedMemberPointer: 41417a519f9SDimitry Andric case CK_DerivedToBaseMemberPointer: 41517a519f9SDimitry Andric case CK_MemberPointerToBoolean: 416dff0c46cSDimitry Andric case CK_ReinterpretMemberPointer: 41717a519f9SDimitry Andric case CK_ConstructorConversion: 41817a519f9SDimitry Andric case CK_IntegralToPointer: 41917a519f9SDimitry Andric case CK_PointerToIntegral: 42017a519f9SDimitry Andric case CK_PointerToBoolean: 42117a519f9SDimitry Andric case CK_ToVoid: 42217a519f9SDimitry Andric case CK_VectorSplat: 42317a519f9SDimitry Andric case CK_IntegralCast: 42417a519f9SDimitry Andric case CK_IntegralToBoolean: 42517a519f9SDimitry Andric case CK_IntegralToFloating: 42617a519f9SDimitry Andric case CK_FloatingToIntegral: 42717a519f9SDimitry Andric case CK_FloatingToBoolean: 42817a519f9SDimitry Andric case CK_FloatingCast: 4296122f3e6SDimitry Andric case CK_CPointerToObjCPointerCast: 4306122f3e6SDimitry Andric case CK_BlockPointerToObjCPointerCast: 43117a519f9SDimitry Andric case CK_AnyPointerToBlockPointerCast: 43217a519f9SDimitry Andric case CK_ObjCObjectLValueCast: 43317a519f9SDimitry Andric case CK_FloatingComplexToReal: 43417a519f9SDimitry Andric case CK_FloatingComplexToBoolean: 43517a519f9SDimitry Andric case CK_IntegralComplexToReal: 43617a519f9SDimitry Andric case CK_IntegralComplexToBoolean: 4376122f3e6SDimitry Andric case CK_ARCProduceObject: 4386122f3e6SDimitry Andric case CK_ARCConsumeObject: 4396122f3e6SDimitry Andric case CK_ARCReclaimReturnedObject: 4406122f3e6SDimitry Andric case CK_ARCExtendBlockObject: 441dff0c46cSDimitry Andric case CK_CopyAndAutoreleaseBlockObject: 4423861d79fSDimitry Andric case CK_BuiltinFnToFnPtr: 443139f7f9bSDimitry Andric case CK_ZeroToOCLEvent: 44417a519f9SDimitry Andric llvm_unreachable("invalid cast kind for complex value"); 44517a519f9SDimitry Andric 44617a519f9SDimitry Andric case CK_FloatingRealToComplex: 44717a519f9SDimitry Andric case CK_IntegralRealToComplex: { 448f22ef01cSRoman Divacky llvm::Value *Elt = CGF.EmitScalarExpr(Op); 449f22ef01cSRoman Divacky 450f22ef01cSRoman Divacky // Convert the input element to the element type of the complex. 451139f7f9bSDimitry Andric DestTy = DestTy->castAs<ComplexType>()->getElementType(); 452f22ef01cSRoman Divacky Elt = CGF.EmitScalarConversion(Elt, Op->getType(), DestTy); 453f22ef01cSRoman Divacky 454f22ef01cSRoman Divacky // Return (realval, 0). 455f22ef01cSRoman Divacky return ComplexPairTy(Elt, llvm::Constant::getNullValue(Elt->getType())); 456f22ef01cSRoman Divacky } 457f22ef01cSRoman Divacky 45817a519f9SDimitry Andric case CK_FloatingComplexCast: 45917a519f9SDimitry Andric case CK_FloatingComplexToIntegralComplex: 46017a519f9SDimitry Andric case CK_IntegralComplexCast: 46117a519f9SDimitry Andric case CK_IntegralComplexToFloatingComplex: 46217a519f9SDimitry Andric return EmitComplexToComplexCast(Visit(Op), Op->getType(), DestTy); 46317a519f9SDimitry Andric } 46417a519f9SDimitry Andric 46517a519f9SDimitry Andric llvm_unreachable("unknown cast resulting in complex value"); 46617a519f9SDimitry Andric } 46717a519f9SDimitry Andric 468f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryMinus(const UnaryOperator *E) { 469f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 470f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 471f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr()); 472f22ef01cSRoman Divacky 473f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 474f22ef01cSRoman Divacky if (Op.first->getType()->isFloatingPointTy()) { 475f22ef01cSRoman Divacky ResR = Builder.CreateFNeg(Op.first, "neg.r"); 476f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "neg.i"); 477f22ef01cSRoman Divacky } else { 478f22ef01cSRoman Divacky ResR = Builder.CreateNeg(Op.first, "neg.r"); 479f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "neg.i"); 480f22ef01cSRoman Divacky } 481f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 482f22ef01cSRoman Divacky } 483f22ef01cSRoman Divacky 484f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryNot(const UnaryOperator *E) { 485f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 486f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 487f22ef01cSRoman Divacky // ~(a+ib) = a + i*-b 488f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr()); 489f22ef01cSRoman Divacky llvm::Value *ResI; 490f22ef01cSRoman Divacky if (Op.second->getType()->isFloatingPointTy()) 491f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "conj.i"); 492f22ef01cSRoman Divacky else 493f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "conj.i"); 494f22ef01cSRoman Divacky 495f22ef01cSRoman Divacky return ComplexPairTy(Op.first, ResI); 496f22ef01cSRoman Divacky } 497f22ef01cSRoman Divacky 498f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinAdd(const BinOpInfo &Op) { 499f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 500f22ef01cSRoman Divacky 501f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 502f22ef01cSRoman Divacky ResR = Builder.CreateFAdd(Op.LHS.first, Op.RHS.first, "add.r"); 503f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(Op.LHS.second, Op.RHS.second, "add.i"); 504f22ef01cSRoman Divacky } else { 505f22ef01cSRoman Divacky ResR = Builder.CreateAdd(Op.LHS.first, Op.RHS.first, "add.r"); 506f22ef01cSRoman Divacky ResI = Builder.CreateAdd(Op.LHS.second, Op.RHS.second, "add.i"); 507f22ef01cSRoman Divacky } 508f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 509f22ef01cSRoman Divacky } 510f22ef01cSRoman Divacky 511f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinSub(const BinOpInfo &Op) { 512f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 513f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 514f22ef01cSRoman Divacky ResR = Builder.CreateFSub(Op.LHS.first, Op.RHS.first, "sub.r"); 515f22ef01cSRoman Divacky ResI = Builder.CreateFSub(Op.LHS.second, Op.RHS.second, "sub.i"); 516f22ef01cSRoman Divacky } else { 517f22ef01cSRoman Divacky ResR = Builder.CreateSub(Op.LHS.first, Op.RHS.first, "sub.r"); 518f22ef01cSRoman Divacky ResI = Builder.CreateSub(Op.LHS.second, Op.RHS.second, "sub.i"); 519f22ef01cSRoman Divacky } 520f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 521f22ef01cSRoman Divacky } 522f22ef01cSRoman Divacky 523f22ef01cSRoman Divacky 524f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinMul(const BinOpInfo &Op) { 525f22ef01cSRoman Divacky using llvm::Value; 526f22ef01cSRoman Divacky Value *ResR, *ResI; 527f22ef01cSRoman Divacky 528f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 529f22ef01cSRoman Divacky Value *ResRl = Builder.CreateFMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 530f22ef01cSRoman Divacky Value *ResRr = Builder.CreateFMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 531f22ef01cSRoman Divacky ResR = Builder.CreateFSub(ResRl, ResRr, "mul.r"); 532f22ef01cSRoman Divacky 533f22ef01cSRoman Divacky Value *ResIl = Builder.CreateFMul(Op.LHS.second, Op.RHS.first, "mul.il"); 534f22ef01cSRoman Divacky Value *ResIr = Builder.CreateFMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 535f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(ResIl, ResIr, "mul.i"); 536f22ef01cSRoman Divacky } else { 537f22ef01cSRoman Divacky Value *ResRl = Builder.CreateMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 538f22ef01cSRoman Divacky Value *ResRr = Builder.CreateMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 539f22ef01cSRoman Divacky ResR = Builder.CreateSub(ResRl, ResRr, "mul.r"); 540f22ef01cSRoman Divacky 541f22ef01cSRoman Divacky Value *ResIl = Builder.CreateMul(Op.LHS.second, Op.RHS.first, "mul.il"); 542f22ef01cSRoman Divacky Value *ResIr = Builder.CreateMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 543f22ef01cSRoman Divacky ResI = Builder.CreateAdd(ResIl, ResIr, "mul.i"); 544f22ef01cSRoman Divacky } 545f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 546f22ef01cSRoman Divacky } 547f22ef01cSRoman Divacky 548f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinDiv(const BinOpInfo &Op) { 549f22ef01cSRoman Divacky llvm::Value *LHSr = Op.LHS.first, *LHSi = Op.LHS.second; 550f22ef01cSRoman Divacky llvm::Value *RHSr = Op.RHS.first, *RHSi = Op.RHS.second; 551f22ef01cSRoman Divacky 552f22ef01cSRoman Divacky 553f22ef01cSRoman Divacky llvm::Value *DSTr, *DSTi; 554f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 555f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 5566122f3e6SDimitry Andric llvm::Value *Tmp1 = Builder.CreateFMul(LHSr, RHSr); // a*c 5576122f3e6SDimitry Andric llvm::Value *Tmp2 = Builder.CreateFMul(LHSi, RHSi); // b*d 5586122f3e6SDimitry Andric llvm::Value *Tmp3 = Builder.CreateFAdd(Tmp1, Tmp2); // ac+bd 559f22ef01cSRoman Divacky 5606122f3e6SDimitry Andric llvm::Value *Tmp4 = Builder.CreateFMul(RHSr, RHSr); // c*c 5616122f3e6SDimitry Andric llvm::Value *Tmp5 = Builder.CreateFMul(RHSi, RHSi); // d*d 5626122f3e6SDimitry Andric llvm::Value *Tmp6 = Builder.CreateFAdd(Tmp4, Tmp5); // cc+dd 563f22ef01cSRoman Divacky 5646122f3e6SDimitry Andric llvm::Value *Tmp7 = Builder.CreateFMul(LHSi, RHSr); // b*c 5656122f3e6SDimitry Andric llvm::Value *Tmp8 = Builder.CreateFMul(LHSr, RHSi); // a*d 5666122f3e6SDimitry Andric llvm::Value *Tmp9 = Builder.CreateFSub(Tmp7, Tmp8); // bc-ad 567f22ef01cSRoman Divacky 5686122f3e6SDimitry Andric DSTr = Builder.CreateFDiv(Tmp3, Tmp6); 5696122f3e6SDimitry Andric DSTi = Builder.CreateFDiv(Tmp9, Tmp6); 570f22ef01cSRoman Divacky } else { 571f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 5726122f3e6SDimitry Andric llvm::Value *Tmp1 = Builder.CreateMul(LHSr, RHSr); // a*c 5736122f3e6SDimitry Andric llvm::Value *Tmp2 = Builder.CreateMul(LHSi, RHSi); // b*d 5746122f3e6SDimitry Andric llvm::Value *Tmp3 = Builder.CreateAdd(Tmp1, Tmp2); // ac+bd 575f22ef01cSRoman Divacky 5766122f3e6SDimitry Andric llvm::Value *Tmp4 = Builder.CreateMul(RHSr, RHSr); // c*c 5776122f3e6SDimitry Andric llvm::Value *Tmp5 = Builder.CreateMul(RHSi, RHSi); // d*d 5786122f3e6SDimitry Andric llvm::Value *Tmp6 = Builder.CreateAdd(Tmp4, Tmp5); // cc+dd 579f22ef01cSRoman Divacky 5806122f3e6SDimitry Andric llvm::Value *Tmp7 = Builder.CreateMul(LHSi, RHSr); // b*c 5816122f3e6SDimitry Andric llvm::Value *Tmp8 = Builder.CreateMul(LHSr, RHSi); // a*d 5826122f3e6SDimitry Andric llvm::Value *Tmp9 = Builder.CreateSub(Tmp7, Tmp8); // bc-ad 583f22ef01cSRoman Divacky 584139f7f9bSDimitry Andric if (Op.Ty->castAs<ComplexType>()->getElementType()->isUnsignedIntegerType()) { 5856122f3e6SDimitry Andric DSTr = Builder.CreateUDiv(Tmp3, Tmp6); 5866122f3e6SDimitry Andric DSTi = Builder.CreateUDiv(Tmp9, Tmp6); 587f22ef01cSRoman Divacky } else { 5886122f3e6SDimitry Andric DSTr = Builder.CreateSDiv(Tmp3, Tmp6); 5896122f3e6SDimitry Andric DSTi = Builder.CreateSDiv(Tmp9, Tmp6); 590f22ef01cSRoman Divacky } 591f22ef01cSRoman Divacky } 592f22ef01cSRoman Divacky 593f22ef01cSRoman Divacky return ComplexPairTy(DSTr, DSTi); 594f22ef01cSRoman Divacky } 595f22ef01cSRoman Divacky 596f22ef01cSRoman Divacky ComplexExprEmitter::BinOpInfo 597f22ef01cSRoman Divacky ComplexExprEmitter::EmitBinOps(const BinaryOperator *E) { 598f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 599f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 600f22ef01cSRoman Divacky BinOpInfo Ops; 601f22ef01cSRoman Divacky Ops.LHS = Visit(E->getLHS()); 602f22ef01cSRoman Divacky Ops.RHS = Visit(E->getRHS()); 603f22ef01cSRoman Divacky Ops.Ty = E->getType(); 604f22ef01cSRoman Divacky return Ops; 605f22ef01cSRoman Divacky } 606f22ef01cSRoman Divacky 607f22ef01cSRoman Divacky 6082754fe60SDimitry Andric LValue ComplexExprEmitter:: 6092754fe60SDimitry Andric EmitCompoundAssignLValue(const CompoundAssignOperator *E, 6102754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&), 6112754fe60SDimitry Andric ComplexPairTy &Val) { 612f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 613f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 6142754fe60SDimitry Andric QualType LHSTy = E->getLHS()->getType(); 615f22ef01cSRoman Divacky 616f22ef01cSRoman Divacky BinOpInfo OpInfo; 617f22ef01cSRoman Divacky 618f22ef01cSRoman Divacky // Load the RHS and LHS operands. 619f22ef01cSRoman Divacky // __block variables need to have the rhs evaluated first, plus this should 6202754fe60SDimitry Andric // improve codegen a little. 621f22ef01cSRoman Divacky OpInfo.Ty = E->getComputationResultType(); 6222754fe60SDimitry Andric 6232754fe60SDimitry Andric // The RHS should have been converted to the computation type. 6242754fe60SDimitry Andric assert(OpInfo.Ty->isAnyComplexType()); 6252754fe60SDimitry Andric assert(CGF.getContext().hasSameUnqualifiedType(OpInfo.Ty, 6262754fe60SDimitry Andric E->getRHS()->getType())); 6272754fe60SDimitry Andric OpInfo.RHS = Visit(E->getRHS()); 628f22ef01cSRoman Divacky 629ffd1746dSEd Schouten LValue LHS = CGF.EmitLValue(E->getLHS()); 6302754fe60SDimitry Andric 6312754fe60SDimitry Andric // Load from the l-value. 6322754fe60SDimitry Andric ComplexPairTy LHSComplexPair = EmitLoadOfLValue(LHS); 633f22ef01cSRoman Divacky 634f22ef01cSRoman Divacky OpInfo.LHS = EmitComplexToComplexCast(LHSComplexPair, LHSTy, OpInfo.Ty); 635f22ef01cSRoman Divacky 636f22ef01cSRoman Divacky // Expand the binary operator. 637f22ef01cSRoman Divacky ComplexPairTy Result = (this->*Func)(OpInfo); 638f22ef01cSRoman Divacky 639f22ef01cSRoman Divacky // Truncate the result back to the LHS type. 640f22ef01cSRoman Divacky Result = EmitComplexToComplexCast(Result, OpInfo.Ty, LHSTy); 6412754fe60SDimitry Andric Val = Result; 642f22ef01cSRoman Divacky 643f22ef01cSRoman Divacky // Store the result value into the LHS lvalue. 644139f7f9bSDimitry Andric EmitStoreOfComplex(Result, LHS, /*isInit*/ false); 645ffd1746dSEd Schouten 6462754fe60SDimitry Andric return LHS; 647f22ef01cSRoman Divacky } 648f22ef01cSRoman Divacky 6492754fe60SDimitry Andric // Compound assignments. 6502754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter:: 6512754fe60SDimitry Andric EmitCompoundAssign(const CompoundAssignOperator *E, 6522754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&)){ 6532754fe60SDimitry Andric ComplexPairTy Val; 6542754fe60SDimitry Andric LValue LV = EmitCompoundAssignLValue(E, Func, Val); 6552754fe60SDimitry Andric 6562754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value. 6573861d79fSDimitry Andric if (!CGF.getLangOpts().CPlusPlus) 6582754fe60SDimitry Andric return Val; 6592754fe60SDimitry Andric 6602754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 6612754fe60SDimitry Andric if (!LV.isVolatileQualified()) 6622754fe60SDimitry Andric return Val; 6632754fe60SDimitry Andric 664139f7f9bSDimitry Andric return EmitLoadOfLValue(LV); 6652754fe60SDimitry Andric } 6662754fe60SDimitry Andric 6672754fe60SDimitry Andric LValue ComplexExprEmitter::EmitBinAssignLValue(const BinaryOperator *E, 6682754fe60SDimitry Andric ComplexPairTy &Val) { 669ffd1746dSEd Schouten assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(), 670ffd1746dSEd Schouten E->getRHS()->getType()) && 671f22ef01cSRoman Divacky "Invalid assignment"); 6722754fe60SDimitry Andric TestAndClearIgnoreReal(); 6732754fe60SDimitry Andric TestAndClearIgnoreImag(); 6742754fe60SDimitry Andric 6752754fe60SDimitry Andric // Emit the RHS. __block variables need the RHS evaluated first. 6762754fe60SDimitry Andric Val = Visit(E->getRHS()); 677f22ef01cSRoman Divacky 678f22ef01cSRoman Divacky // Compute the address to store into. 679f22ef01cSRoman Divacky LValue LHS = CGF.EmitLValue(E->getLHS()); 680f22ef01cSRoman Divacky 681ffd1746dSEd Schouten // Store the result value into the LHS lvalue. 682139f7f9bSDimitry Andric EmitStoreOfComplex(Val, LHS, /*isInit*/ false); 683f22ef01cSRoman Divacky 6842754fe60SDimitry Andric return LHS; 6852754fe60SDimitry Andric } 686ffd1746dSEd Schouten 6872754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitBinAssign(const BinaryOperator *E) { 6882754fe60SDimitry Andric ComplexPairTy Val; 6892754fe60SDimitry Andric LValue LV = EmitBinAssignLValue(E, Val); 6902754fe60SDimitry Andric 6912754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value. 6923861d79fSDimitry Andric if (!CGF.getLangOpts().CPlusPlus) 6932754fe60SDimitry Andric return Val; 6942754fe60SDimitry Andric 6952754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 6962754fe60SDimitry Andric if (!LV.isVolatileQualified()) 6972754fe60SDimitry Andric return Val; 6982754fe60SDimitry Andric 699139f7f9bSDimitry Andric return EmitLoadOfLValue(LV); 700f22ef01cSRoman Divacky } 701f22ef01cSRoman Divacky 702f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitBinComma(const BinaryOperator *E) { 7032754fe60SDimitry Andric CGF.EmitIgnoredExpr(E->getLHS()); 704f22ef01cSRoman Divacky return Visit(E->getRHS()); 705f22ef01cSRoman Divacky } 706f22ef01cSRoman Divacky 707f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter:: 7082754fe60SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) { 709f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 710f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 711f22ef01cSRoman Divacky llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true"); 712f22ef01cSRoman Divacky llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false"); 713f22ef01cSRoman Divacky llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end"); 714f22ef01cSRoman Divacky 7152754fe60SDimitry Andric // Bind the common expression if necessary. 7162754fe60SDimitry Andric CodeGenFunction::OpaqueValueMapping binding(CGF, E); 7172754fe60SDimitry Andric 7182754fe60SDimitry Andric CodeGenFunction::ConditionalEvaluation eval(CGF); 719f22ef01cSRoman Divacky CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock); 720f22ef01cSRoman Divacky 7212754fe60SDimitry Andric eval.begin(CGF); 722f22ef01cSRoman Divacky CGF.EmitBlock(LHSBlock); 7232754fe60SDimitry Andric ComplexPairTy LHS = Visit(E->getTrueExpr()); 724f22ef01cSRoman Divacky LHSBlock = Builder.GetInsertBlock(); 725f22ef01cSRoman Divacky CGF.EmitBranch(ContBlock); 7262754fe60SDimitry Andric eval.end(CGF); 727f22ef01cSRoman Divacky 7282754fe60SDimitry Andric eval.begin(CGF); 729f22ef01cSRoman Divacky CGF.EmitBlock(RHSBlock); 7302754fe60SDimitry Andric ComplexPairTy RHS = Visit(E->getFalseExpr()); 731f22ef01cSRoman Divacky RHSBlock = Builder.GetInsertBlock(); 732f22ef01cSRoman Divacky CGF.EmitBlock(ContBlock); 7332754fe60SDimitry Andric eval.end(CGF); 734f22ef01cSRoman Divacky 735f22ef01cSRoman Divacky // Create a PHI node for the real part. 7363b0f4066SDimitry Andric llvm::PHINode *RealPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.r"); 737f22ef01cSRoman Divacky RealPN->addIncoming(LHS.first, LHSBlock); 738f22ef01cSRoman Divacky RealPN->addIncoming(RHS.first, RHSBlock); 739f22ef01cSRoman Divacky 740f22ef01cSRoman Divacky // Create a PHI node for the imaginary part. 7413b0f4066SDimitry Andric llvm::PHINode *ImagPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.i"); 742f22ef01cSRoman Divacky ImagPN->addIncoming(LHS.second, LHSBlock); 743f22ef01cSRoman Divacky ImagPN->addIncoming(RHS.second, RHSBlock); 744f22ef01cSRoman Divacky 745f22ef01cSRoman Divacky return ComplexPairTy(RealPN, ImagPN); 746f22ef01cSRoman Divacky } 747f22ef01cSRoman Divacky 748f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitChooseExpr(ChooseExpr *E) { 749f22ef01cSRoman Divacky return Visit(E->getChosenSubExpr(CGF.getContext())); 750f22ef01cSRoman Divacky } 751f22ef01cSRoman Divacky 752f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitInitListExpr(InitListExpr *E) { 753f22ef01cSRoman Divacky bool Ignore = TestAndClearIgnoreReal(); 754f22ef01cSRoman Divacky (void)Ignore; 755f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored"); 756f22ef01cSRoman Divacky Ignore = TestAndClearIgnoreImag(); 757f22ef01cSRoman Divacky (void)Ignore; 758f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored"); 7596122f3e6SDimitry Andric 7606122f3e6SDimitry Andric if (E->getNumInits() == 2) { 7616122f3e6SDimitry Andric llvm::Value *Real = CGF.EmitScalarExpr(E->getInit(0)); 7626122f3e6SDimitry Andric llvm::Value *Imag = CGF.EmitScalarExpr(E->getInit(1)); 7636122f3e6SDimitry Andric return ComplexPairTy(Real, Imag); 7646122f3e6SDimitry Andric } else if (E->getNumInits() == 1) { 765f22ef01cSRoman Divacky return Visit(E->getInit(0)); 7666122f3e6SDimitry Andric } 767f22ef01cSRoman Divacky 768f22ef01cSRoman Divacky // Empty init list intializes to null 7696122f3e6SDimitry Andric assert(E->getNumInits() == 0 && "Unexpected number of inits"); 770139f7f9bSDimitry Andric QualType Ty = E->getType()->castAs<ComplexType>()->getElementType(); 7716122f3e6SDimitry Andric llvm::Type* LTy = CGF.ConvertType(Ty); 772f22ef01cSRoman Divacky llvm::Value* zeroConstant = llvm::Constant::getNullValue(LTy); 773f22ef01cSRoman Divacky return ComplexPairTy(zeroConstant, zeroConstant); 774f22ef01cSRoman Divacky } 775f22ef01cSRoman Divacky 776f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitVAArgExpr(VAArgExpr *E) { 777f22ef01cSRoman Divacky llvm::Value *ArgValue = CGF.EmitVAListRef(E->getSubExpr()); 778f22ef01cSRoman Divacky llvm::Value *ArgPtr = CGF.EmitVAArg(ArgValue, E->getType()); 779f22ef01cSRoman Divacky 780f22ef01cSRoman Divacky if (!ArgPtr) { 781f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex va_arg expression"); 7826122f3e6SDimitry Andric llvm::Type *EltTy = 783139f7f9bSDimitry Andric CGF.ConvertType(E->getType()->castAs<ComplexType>()->getElementType()); 784f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy); 785f22ef01cSRoman Divacky return ComplexPairTy(U, U); 786f22ef01cSRoman Divacky } 787f22ef01cSRoman Divacky 788139f7f9bSDimitry Andric return EmitLoadOfLValue( 789139f7f9bSDimitry Andric CGF.MakeNaturalAlignAddrLValue(ArgPtr, E->getType())); 790f22ef01cSRoman Divacky } 791f22ef01cSRoman Divacky 792f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 793f22ef01cSRoman Divacky // Entry Point into this File 794f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 795f22ef01cSRoman Divacky 796f22ef01cSRoman Divacky /// EmitComplexExpr - Emit the computation of the specified expression of 797f22ef01cSRoman Divacky /// complex type, ignoring the result. 798f22ef01cSRoman Divacky ComplexPairTy CodeGenFunction::EmitComplexExpr(const Expr *E, bool IgnoreReal, 7992754fe60SDimitry Andric bool IgnoreImag) { 800139f7f9bSDimitry Andric assert(E && getComplexType(E->getType()) && 801f22ef01cSRoman Divacky "Invalid complex expression to emit"); 802f22ef01cSRoman Divacky 8032754fe60SDimitry Andric return ComplexExprEmitter(*this, IgnoreReal, IgnoreImag) 804f22ef01cSRoman Divacky .Visit(const_cast<Expr*>(E)); 805f22ef01cSRoman Divacky } 806f22ef01cSRoman Divacky 807139f7f9bSDimitry Andric void CodeGenFunction::EmitComplexExprIntoLValue(const Expr *E, LValue dest, 808139f7f9bSDimitry Andric bool isInit) { 809139f7f9bSDimitry Andric assert(E && getComplexType(E->getType()) && 810f22ef01cSRoman Divacky "Invalid complex expression to emit"); 811f22ef01cSRoman Divacky ComplexExprEmitter Emitter(*this); 812f22ef01cSRoman Divacky ComplexPairTy Val = Emitter.Visit(const_cast<Expr*>(E)); 813139f7f9bSDimitry Andric Emitter.EmitStoreOfComplex(Val, dest, isInit); 814f22ef01cSRoman Divacky } 815f22ef01cSRoman Divacky 816139f7f9bSDimitry Andric /// EmitStoreOfComplex - Store a complex number into the specified l-value. 817139f7f9bSDimitry Andric void CodeGenFunction::EmitStoreOfComplex(ComplexPairTy V, LValue dest, 818139f7f9bSDimitry Andric bool isInit) { 819139f7f9bSDimitry Andric ComplexExprEmitter(*this).EmitStoreOfComplex(V, dest, isInit); 820f22ef01cSRoman Divacky } 821f22ef01cSRoman Divacky 822139f7f9bSDimitry Andric /// EmitLoadOfComplex - Load a complex number from the specified address. 823139f7f9bSDimitry Andric ComplexPairTy CodeGenFunction::EmitLoadOfComplex(LValue src) { 824139f7f9bSDimitry Andric return ComplexExprEmitter(*this).EmitLoadOfLValue(src); 825f22ef01cSRoman Divacky } 8262754fe60SDimitry Andric 8272754fe60SDimitry Andric LValue CodeGenFunction::EmitComplexAssignmentLValue(const BinaryOperator *E) { 8282754fe60SDimitry Andric assert(E->getOpcode() == BO_Assign); 8292754fe60SDimitry Andric ComplexPairTy Val; // ignored 8302754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitBinAssignLValue(E, Val); 8312754fe60SDimitry Andric } 8322754fe60SDimitry Andric 8332754fe60SDimitry Andric LValue CodeGenFunction:: 8342754fe60SDimitry Andric EmitComplexCompoundAssignmentLValue(const CompoundAssignOperator *E) { 8352754fe60SDimitry Andric ComplexPairTy(ComplexExprEmitter::*Op)(const ComplexExprEmitter::BinOpInfo &); 8362754fe60SDimitry Andric switch (E->getOpcode()) { 8372754fe60SDimitry Andric case BO_MulAssign: Op = &ComplexExprEmitter::EmitBinMul; break; 8382754fe60SDimitry Andric case BO_DivAssign: Op = &ComplexExprEmitter::EmitBinDiv; break; 8392754fe60SDimitry Andric case BO_SubAssign: Op = &ComplexExprEmitter::EmitBinSub; break; 8402754fe60SDimitry Andric case BO_AddAssign: Op = &ComplexExprEmitter::EmitBinAdd; break; 8412754fe60SDimitry Andric 8422754fe60SDimitry Andric default: 8432754fe60SDimitry Andric llvm_unreachable("unexpected complex compound assignment"); 8442754fe60SDimitry Andric } 8452754fe60SDimitry Andric 8462754fe60SDimitry Andric ComplexPairTy Val; // ignored 8472754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val); 8482754fe60SDimitry Andric } 849