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 } 1852754fe60SDimitry Andric ComplexPairTy VisitExprWithCleanups(ExprWithCleanups *E) { 186dff0c46cSDimitry Andric CGF.enterFullExpression(E); 187dff0c46cSDimitry Andric CodeGenFunction::RunCleanupsScope Scope(CGF); 188dff0c46cSDimitry Andric return Visit(E->getSubExpr()); 189f22ef01cSRoman Divacky } 190ffd1746dSEd Schouten ComplexPairTy VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) { 191f22ef01cSRoman Divacky assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 192139f7f9bSDimitry Andric QualType Elem = E->getType()->castAs<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!"); 198139f7f9bSDimitry Andric QualType Elem = E->getType()->castAs<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 268dff0c46cSDimitry Andric ComplexPairTy VisitCompoundLiteralExpr(CompoundLiteralExpr *E) { 269dff0c46cSDimitry Andric return EmitLoadOfLValue(E); 270dff0c46cSDimitry Andric } 271dff0c46cSDimitry Andric 272f22ef01cSRoman Divacky ComplexPairTy VisitVAArgExpr(VAArgExpr *E); 2736122f3e6SDimitry Andric 2746122f3e6SDimitry Andric ComplexPairTy VisitAtomicExpr(AtomicExpr *E) { 2756122f3e6SDimitry Andric return CGF.EmitAtomicExpr(E).getComplexVal(); 2766122f3e6SDimitry Andric } 277f22ef01cSRoman Divacky }; 278f22ef01cSRoman Divacky } // end anonymous namespace. 279f22ef01cSRoman Divacky 280f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 281f22ef01cSRoman Divacky // Utilities 282f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 283f22ef01cSRoman Divacky 284139f7f9bSDimitry Andric /// EmitLoadOfLValue - Given an RValue reference for a complex, emit code to 285f22ef01cSRoman Divacky /// load the real and imaginary pieces, returning them as Real/Imag. 286139f7f9bSDimitry Andric ComplexPairTy ComplexExprEmitter::EmitLoadOfLValue(LValue lvalue) { 287139f7f9bSDimitry Andric assert(lvalue.isSimple() && "non-simple complex l-value?"); 288139f7f9bSDimitry Andric if (lvalue.getType()->isAtomicType()) 289139f7f9bSDimitry Andric return CGF.EmitAtomicLoad(lvalue).getComplexVal(); 290139f7f9bSDimitry Andric 291139f7f9bSDimitry Andric llvm::Value *SrcPtr = lvalue.getAddress(); 292139f7f9bSDimitry Andric bool isVolatile = lvalue.isVolatileQualified(); 293139f7f9bSDimitry Andric 294f22ef01cSRoman Divacky llvm::Value *Real=0, *Imag=0; 295f22ef01cSRoman Divacky 2962754fe60SDimitry Andric if (!IgnoreReal || isVolatile) { 297f22ef01cSRoman Divacky llvm::Value *RealP = Builder.CreateStructGEP(SrcPtr, 0, 298f22ef01cSRoman Divacky SrcPtr->getName() + ".realp"); 299f22ef01cSRoman Divacky Real = Builder.CreateLoad(RealP, isVolatile, SrcPtr->getName() + ".real"); 300f22ef01cSRoman Divacky } 301f22ef01cSRoman Divacky 3022754fe60SDimitry Andric if (!IgnoreImag || isVolatile) { 303f22ef01cSRoman Divacky llvm::Value *ImagP = Builder.CreateStructGEP(SrcPtr, 1, 304f22ef01cSRoman Divacky SrcPtr->getName() + ".imagp"); 305f22ef01cSRoman Divacky Imag = Builder.CreateLoad(ImagP, isVolatile, SrcPtr->getName() + ".imag"); 306f22ef01cSRoman Divacky } 307f22ef01cSRoman Divacky return ComplexPairTy(Real, Imag); 308f22ef01cSRoman Divacky } 309f22ef01cSRoman Divacky 310f22ef01cSRoman Divacky /// EmitStoreOfComplex - Store the specified real/imag parts into the 311f22ef01cSRoman Divacky /// specified value pointer. 312139f7f9bSDimitry Andric void ComplexExprEmitter::EmitStoreOfComplex(ComplexPairTy Val, 313139f7f9bSDimitry Andric LValue lvalue, 314139f7f9bSDimitry Andric bool isInit) { 315139f7f9bSDimitry Andric if (lvalue.getType()->isAtomicType()) 316139f7f9bSDimitry Andric return CGF.EmitAtomicStore(RValue::getComplex(Val), lvalue, isInit); 317139f7f9bSDimitry Andric 318139f7f9bSDimitry Andric llvm::Value *Ptr = lvalue.getAddress(); 319f22ef01cSRoman Divacky llvm::Value *RealPtr = Builder.CreateStructGEP(Ptr, 0, "real"); 320f22ef01cSRoman Divacky llvm::Value *ImagPtr = Builder.CreateStructGEP(Ptr, 1, "imag"); 321f22ef01cSRoman Divacky 322139f7f9bSDimitry Andric // TODO: alignment 323139f7f9bSDimitry Andric Builder.CreateStore(Val.first, RealPtr, lvalue.isVolatileQualified()); 324139f7f9bSDimitry Andric Builder.CreateStore(Val.second, ImagPtr, lvalue.isVolatileQualified()); 325f22ef01cSRoman Divacky } 326f22ef01cSRoman Divacky 327f22ef01cSRoman Divacky 328f22ef01cSRoman Divacky 329f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 330f22ef01cSRoman Divacky // Visitor Methods 331f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 332f22ef01cSRoman Divacky 333f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitExpr(Expr *E) { 334f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex expression"); 3356122f3e6SDimitry Andric llvm::Type *EltTy = 336139f7f9bSDimitry Andric CGF.ConvertType(getComplexType(E->getType())->getElementType()); 337f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy); 338f22ef01cSRoman Divacky return ComplexPairTy(U, U); 339f22ef01cSRoman Divacky } 340f22ef01cSRoman Divacky 341f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter:: 342f22ef01cSRoman Divacky VisitImaginaryLiteral(const ImaginaryLiteral *IL) { 343f22ef01cSRoman Divacky llvm::Value *Imag = CGF.EmitScalarExpr(IL->getSubExpr()); 3442754fe60SDimitry Andric return ComplexPairTy(llvm::Constant::getNullValue(Imag->getType()), Imag); 345f22ef01cSRoman Divacky } 346f22ef01cSRoman Divacky 347f22ef01cSRoman Divacky 348f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitCallExpr(const CallExpr *E) { 349f22ef01cSRoman Divacky if (E->getCallReturnType()->isReferenceType()) 350f22ef01cSRoman Divacky return EmitLoadOfLValue(E); 351f22ef01cSRoman Divacky 352f22ef01cSRoman Divacky return CGF.EmitCallExpr(E).getComplexVal(); 353f22ef01cSRoman Divacky } 354f22ef01cSRoman Divacky 355f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitStmtExpr(const StmtExpr *E) { 3562754fe60SDimitry Andric CodeGenFunction::StmtExprEvaluation eval(CGF); 357f22ef01cSRoman Divacky return CGF.EmitCompoundStmt(*E->getSubStmt(), true).getComplexVal(); 358f22ef01cSRoman Divacky } 359f22ef01cSRoman Divacky 360f22ef01cSRoman Divacky /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType. 361f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitComplexToComplexCast(ComplexPairTy Val, 362f22ef01cSRoman Divacky QualType SrcType, 363f22ef01cSRoman Divacky QualType DestType) { 364f22ef01cSRoman Divacky // Get the src/dest element type. 365139f7f9bSDimitry Andric SrcType = SrcType->castAs<ComplexType>()->getElementType(); 366139f7f9bSDimitry Andric DestType = DestType->castAs<ComplexType>()->getElementType(); 367f22ef01cSRoman Divacky 368f22ef01cSRoman Divacky // C99 6.3.1.6: When a value of complex type is converted to another 369f22ef01cSRoman Divacky // complex type, both the real and imaginary parts follow the conversion 370f22ef01cSRoman Divacky // rules for the corresponding real types. 371f22ef01cSRoman Divacky Val.first = CGF.EmitScalarConversion(Val.first, SrcType, DestType); 372f22ef01cSRoman Divacky Val.second = CGF.EmitScalarConversion(Val.second, SrcType, DestType); 373f22ef01cSRoman Divacky return Val; 374f22ef01cSRoman Divacky } 375f22ef01cSRoman Divacky 376ffd1746dSEd Schouten ComplexPairTy ComplexExprEmitter::EmitCast(CastExpr::CastKind CK, Expr *Op, 377ffd1746dSEd Schouten QualType DestTy) { 3782754fe60SDimitry Andric switch (CK) { 37917a519f9SDimitry Andric case CK_Dependent: llvm_unreachable("dependent cast kind in IR gen!"); 38017a519f9SDimitry Andric 381dff0c46cSDimitry Andric // Atomic to non-atomic casts may be more than a no-op for some platforms and 382dff0c46cSDimitry Andric // for some types. 383dff0c46cSDimitry Andric case CK_AtomicToNonAtomic: 384dff0c46cSDimitry Andric case CK_NonAtomicToAtomic: 3852754fe60SDimitry Andric case CK_NoOp: 3862754fe60SDimitry Andric case CK_LValueToRValue: 38717a519f9SDimitry Andric case CK_UserDefinedConversion: 3882754fe60SDimitry Andric return Visit(Op); 3892754fe60SDimitry Andric 39017a519f9SDimitry Andric case CK_LValueBitCast: { 391139f7f9bSDimitry Andric LValue origLV = CGF.EmitLValue(Op); 392139f7f9bSDimitry Andric llvm::Value *V = origLV.getAddress(); 393ffd1746dSEd Schouten V = Builder.CreateBitCast(V, 394ffd1746dSEd Schouten CGF.ConvertType(CGF.getContext().getPointerType(DestTy))); 395139f7f9bSDimitry Andric return EmitLoadOfLValue(CGF.MakeAddrLValue(V, DestTy, 396139f7f9bSDimitry Andric origLV.getAlignment())); 397ffd1746dSEd Schouten } 398ffd1746dSEd Schouten 39917a519f9SDimitry Andric case CK_BitCast: 40017a519f9SDimitry Andric case CK_BaseToDerived: 40117a519f9SDimitry Andric case CK_DerivedToBase: 40217a519f9SDimitry Andric case CK_UncheckedDerivedToBase: 40317a519f9SDimitry Andric case CK_Dynamic: 40417a519f9SDimitry Andric case CK_ToUnion: 40517a519f9SDimitry Andric case CK_ArrayToPointerDecay: 40617a519f9SDimitry Andric case CK_FunctionToPointerDecay: 40717a519f9SDimitry Andric case CK_NullToPointer: 40817a519f9SDimitry Andric case CK_NullToMemberPointer: 40917a519f9SDimitry Andric case CK_BaseToDerivedMemberPointer: 41017a519f9SDimitry Andric case CK_DerivedToBaseMemberPointer: 41117a519f9SDimitry Andric case CK_MemberPointerToBoolean: 412dff0c46cSDimitry Andric case CK_ReinterpretMemberPointer: 41317a519f9SDimitry Andric case CK_ConstructorConversion: 41417a519f9SDimitry Andric case CK_IntegralToPointer: 41517a519f9SDimitry Andric case CK_PointerToIntegral: 41617a519f9SDimitry Andric case CK_PointerToBoolean: 41717a519f9SDimitry Andric case CK_ToVoid: 41817a519f9SDimitry Andric case CK_VectorSplat: 41917a519f9SDimitry Andric case CK_IntegralCast: 42017a519f9SDimitry Andric case CK_IntegralToBoolean: 42117a519f9SDimitry Andric case CK_IntegralToFloating: 42217a519f9SDimitry Andric case CK_FloatingToIntegral: 42317a519f9SDimitry Andric case CK_FloatingToBoolean: 42417a519f9SDimitry Andric case CK_FloatingCast: 4256122f3e6SDimitry Andric case CK_CPointerToObjCPointerCast: 4266122f3e6SDimitry Andric case CK_BlockPointerToObjCPointerCast: 42717a519f9SDimitry Andric case CK_AnyPointerToBlockPointerCast: 42817a519f9SDimitry Andric case CK_ObjCObjectLValueCast: 42917a519f9SDimitry Andric case CK_FloatingComplexToReal: 43017a519f9SDimitry Andric case CK_FloatingComplexToBoolean: 43117a519f9SDimitry Andric case CK_IntegralComplexToReal: 43217a519f9SDimitry Andric case CK_IntegralComplexToBoolean: 4336122f3e6SDimitry Andric case CK_ARCProduceObject: 4346122f3e6SDimitry Andric case CK_ARCConsumeObject: 4356122f3e6SDimitry Andric case CK_ARCReclaimReturnedObject: 4366122f3e6SDimitry Andric case CK_ARCExtendBlockObject: 437dff0c46cSDimitry Andric case CK_CopyAndAutoreleaseBlockObject: 4383861d79fSDimitry Andric case CK_BuiltinFnToFnPtr: 439139f7f9bSDimitry Andric case CK_ZeroToOCLEvent: 44017a519f9SDimitry Andric llvm_unreachable("invalid cast kind for complex value"); 44117a519f9SDimitry Andric 44217a519f9SDimitry Andric case CK_FloatingRealToComplex: 44317a519f9SDimitry Andric case CK_IntegralRealToComplex: { 444f22ef01cSRoman Divacky llvm::Value *Elt = CGF.EmitScalarExpr(Op); 445f22ef01cSRoman Divacky 446f22ef01cSRoman Divacky // Convert the input element to the element type of the complex. 447139f7f9bSDimitry Andric DestTy = DestTy->castAs<ComplexType>()->getElementType(); 448f22ef01cSRoman Divacky Elt = CGF.EmitScalarConversion(Elt, Op->getType(), DestTy); 449f22ef01cSRoman Divacky 450f22ef01cSRoman Divacky // Return (realval, 0). 451f22ef01cSRoman Divacky return ComplexPairTy(Elt, llvm::Constant::getNullValue(Elt->getType())); 452f22ef01cSRoman Divacky } 453f22ef01cSRoman Divacky 45417a519f9SDimitry Andric case CK_FloatingComplexCast: 45517a519f9SDimitry Andric case CK_FloatingComplexToIntegralComplex: 45617a519f9SDimitry Andric case CK_IntegralComplexCast: 45717a519f9SDimitry Andric case CK_IntegralComplexToFloatingComplex: 45817a519f9SDimitry Andric return EmitComplexToComplexCast(Visit(Op), Op->getType(), DestTy); 45917a519f9SDimitry Andric } 46017a519f9SDimitry Andric 46117a519f9SDimitry Andric llvm_unreachable("unknown cast resulting in complex value"); 46217a519f9SDimitry Andric } 46317a519f9SDimitry Andric 464f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryMinus(const UnaryOperator *E) { 465f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 466f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 467f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr()); 468f22ef01cSRoman Divacky 469f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 470f22ef01cSRoman Divacky if (Op.first->getType()->isFloatingPointTy()) { 471f22ef01cSRoman Divacky ResR = Builder.CreateFNeg(Op.first, "neg.r"); 472f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "neg.i"); 473f22ef01cSRoman Divacky } else { 474f22ef01cSRoman Divacky ResR = Builder.CreateNeg(Op.first, "neg.r"); 475f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "neg.i"); 476f22ef01cSRoman Divacky } 477f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 478f22ef01cSRoman Divacky } 479f22ef01cSRoman Divacky 480f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitUnaryNot(const UnaryOperator *E) { 481f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 482f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 483f22ef01cSRoman Divacky // ~(a+ib) = a + i*-b 484f22ef01cSRoman Divacky ComplexPairTy Op = Visit(E->getSubExpr()); 485f22ef01cSRoman Divacky llvm::Value *ResI; 486f22ef01cSRoman Divacky if (Op.second->getType()->isFloatingPointTy()) 487f22ef01cSRoman Divacky ResI = Builder.CreateFNeg(Op.second, "conj.i"); 488f22ef01cSRoman Divacky else 489f22ef01cSRoman Divacky ResI = Builder.CreateNeg(Op.second, "conj.i"); 490f22ef01cSRoman Divacky 491f22ef01cSRoman Divacky return ComplexPairTy(Op.first, ResI); 492f22ef01cSRoman Divacky } 493f22ef01cSRoman Divacky 494f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinAdd(const BinOpInfo &Op) { 495f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 496f22ef01cSRoman Divacky 497f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 498f22ef01cSRoman Divacky ResR = Builder.CreateFAdd(Op.LHS.first, Op.RHS.first, "add.r"); 499f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(Op.LHS.second, Op.RHS.second, "add.i"); 500f22ef01cSRoman Divacky } else { 501f22ef01cSRoman Divacky ResR = Builder.CreateAdd(Op.LHS.first, Op.RHS.first, "add.r"); 502f22ef01cSRoman Divacky ResI = Builder.CreateAdd(Op.LHS.second, Op.RHS.second, "add.i"); 503f22ef01cSRoman Divacky } 504f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 505f22ef01cSRoman Divacky } 506f22ef01cSRoman Divacky 507f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinSub(const BinOpInfo &Op) { 508f22ef01cSRoman Divacky llvm::Value *ResR, *ResI; 509f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 510f22ef01cSRoman Divacky ResR = Builder.CreateFSub(Op.LHS.first, Op.RHS.first, "sub.r"); 511f22ef01cSRoman Divacky ResI = Builder.CreateFSub(Op.LHS.second, Op.RHS.second, "sub.i"); 512f22ef01cSRoman Divacky } else { 513f22ef01cSRoman Divacky ResR = Builder.CreateSub(Op.LHS.first, Op.RHS.first, "sub.r"); 514f22ef01cSRoman Divacky ResI = Builder.CreateSub(Op.LHS.second, Op.RHS.second, "sub.i"); 515f22ef01cSRoman Divacky } 516f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 517f22ef01cSRoman Divacky } 518f22ef01cSRoman Divacky 519f22ef01cSRoman Divacky 520f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinMul(const BinOpInfo &Op) { 521f22ef01cSRoman Divacky using llvm::Value; 522f22ef01cSRoman Divacky Value *ResR, *ResI; 523f22ef01cSRoman Divacky 524f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 525f22ef01cSRoman Divacky Value *ResRl = Builder.CreateFMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 526f22ef01cSRoman Divacky Value *ResRr = Builder.CreateFMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 527f22ef01cSRoman Divacky ResR = Builder.CreateFSub(ResRl, ResRr, "mul.r"); 528f22ef01cSRoman Divacky 529f22ef01cSRoman Divacky Value *ResIl = Builder.CreateFMul(Op.LHS.second, Op.RHS.first, "mul.il"); 530f22ef01cSRoman Divacky Value *ResIr = Builder.CreateFMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 531f22ef01cSRoman Divacky ResI = Builder.CreateFAdd(ResIl, ResIr, "mul.i"); 532f22ef01cSRoman Divacky } else { 533f22ef01cSRoman Divacky Value *ResRl = Builder.CreateMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 534f22ef01cSRoman Divacky Value *ResRr = Builder.CreateMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 535f22ef01cSRoman Divacky ResR = Builder.CreateSub(ResRl, ResRr, "mul.r"); 536f22ef01cSRoman Divacky 537f22ef01cSRoman Divacky Value *ResIl = Builder.CreateMul(Op.LHS.second, Op.RHS.first, "mul.il"); 538f22ef01cSRoman Divacky Value *ResIr = Builder.CreateMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 539f22ef01cSRoman Divacky ResI = Builder.CreateAdd(ResIl, ResIr, "mul.i"); 540f22ef01cSRoman Divacky } 541f22ef01cSRoman Divacky return ComplexPairTy(ResR, ResI); 542f22ef01cSRoman Divacky } 543f22ef01cSRoman Divacky 544f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::EmitBinDiv(const BinOpInfo &Op) { 545f22ef01cSRoman Divacky llvm::Value *LHSr = Op.LHS.first, *LHSi = Op.LHS.second; 546f22ef01cSRoman Divacky llvm::Value *RHSr = Op.RHS.first, *RHSi = Op.RHS.second; 547f22ef01cSRoman Divacky 548f22ef01cSRoman Divacky 549f22ef01cSRoman Divacky llvm::Value *DSTr, *DSTi; 550f22ef01cSRoman Divacky if (Op.LHS.first->getType()->isFloatingPointTy()) { 551f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 5526122f3e6SDimitry Andric llvm::Value *Tmp1 = Builder.CreateFMul(LHSr, RHSr); // a*c 5536122f3e6SDimitry Andric llvm::Value *Tmp2 = Builder.CreateFMul(LHSi, RHSi); // b*d 5546122f3e6SDimitry Andric llvm::Value *Tmp3 = Builder.CreateFAdd(Tmp1, Tmp2); // ac+bd 555f22ef01cSRoman Divacky 5566122f3e6SDimitry Andric llvm::Value *Tmp4 = Builder.CreateFMul(RHSr, RHSr); // c*c 5576122f3e6SDimitry Andric llvm::Value *Tmp5 = Builder.CreateFMul(RHSi, RHSi); // d*d 5586122f3e6SDimitry Andric llvm::Value *Tmp6 = Builder.CreateFAdd(Tmp4, Tmp5); // cc+dd 559f22ef01cSRoman Divacky 5606122f3e6SDimitry Andric llvm::Value *Tmp7 = Builder.CreateFMul(LHSi, RHSr); // b*c 5616122f3e6SDimitry Andric llvm::Value *Tmp8 = Builder.CreateFMul(LHSr, RHSi); // a*d 5626122f3e6SDimitry Andric llvm::Value *Tmp9 = Builder.CreateFSub(Tmp7, Tmp8); // bc-ad 563f22ef01cSRoman Divacky 5646122f3e6SDimitry Andric DSTr = Builder.CreateFDiv(Tmp3, Tmp6); 5656122f3e6SDimitry Andric DSTi = Builder.CreateFDiv(Tmp9, Tmp6); 566f22ef01cSRoman Divacky } else { 567f22ef01cSRoman Divacky // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 5686122f3e6SDimitry Andric llvm::Value *Tmp1 = Builder.CreateMul(LHSr, RHSr); // a*c 5696122f3e6SDimitry Andric llvm::Value *Tmp2 = Builder.CreateMul(LHSi, RHSi); // b*d 5706122f3e6SDimitry Andric llvm::Value *Tmp3 = Builder.CreateAdd(Tmp1, Tmp2); // ac+bd 571f22ef01cSRoman Divacky 5726122f3e6SDimitry Andric llvm::Value *Tmp4 = Builder.CreateMul(RHSr, RHSr); // c*c 5736122f3e6SDimitry Andric llvm::Value *Tmp5 = Builder.CreateMul(RHSi, RHSi); // d*d 5746122f3e6SDimitry Andric llvm::Value *Tmp6 = Builder.CreateAdd(Tmp4, Tmp5); // cc+dd 575f22ef01cSRoman Divacky 5766122f3e6SDimitry Andric llvm::Value *Tmp7 = Builder.CreateMul(LHSi, RHSr); // b*c 5776122f3e6SDimitry Andric llvm::Value *Tmp8 = Builder.CreateMul(LHSr, RHSi); // a*d 5786122f3e6SDimitry Andric llvm::Value *Tmp9 = Builder.CreateSub(Tmp7, Tmp8); // bc-ad 579f22ef01cSRoman Divacky 580139f7f9bSDimitry Andric if (Op.Ty->castAs<ComplexType>()->getElementType()->isUnsignedIntegerType()) { 5816122f3e6SDimitry Andric DSTr = Builder.CreateUDiv(Tmp3, Tmp6); 5826122f3e6SDimitry Andric DSTi = Builder.CreateUDiv(Tmp9, Tmp6); 583f22ef01cSRoman Divacky } else { 5846122f3e6SDimitry Andric DSTr = Builder.CreateSDiv(Tmp3, Tmp6); 5856122f3e6SDimitry Andric DSTi = Builder.CreateSDiv(Tmp9, Tmp6); 586f22ef01cSRoman Divacky } 587f22ef01cSRoman Divacky } 588f22ef01cSRoman Divacky 589f22ef01cSRoman Divacky return ComplexPairTy(DSTr, DSTi); 590f22ef01cSRoman Divacky } 591f22ef01cSRoman Divacky 592f22ef01cSRoman Divacky ComplexExprEmitter::BinOpInfo 593f22ef01cSRoman Divacky ComplexExprEmitter::EmitBinOps(const BinaryOperator *E) { 594f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 595f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 596f22ef01cSRoman Divacky BinOpInfo Ops; 597f22ef01cSRoman Divacky Ops.LHS = Visit(E->getLHS()); 598f22ef01cSRoman Divacky Ops.RHS = Visit(E->getRHS()); 599f22ef01cSRoman Divacky Ops.Ty = E->getType(); 600f22ef01cSRoman Divacky return Ops; 601f22ef01cSRoman Divacky } 602f22ef01cSRoman Divacky 603f22ef01cSRoman Divacky 6042754fe60SDimitry Andric LValue ComplexExprEmitter:: 6052754fe60SDimitry Andric EmitCompoundAssignLValue(const CompoundAssignOperator *E, 6062754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&), 6072754fe60SDimitry Andric ComplexPairTy &Val) { 608f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 609f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 6102754fe60SDimitry Andric QualType LHSTy = E->getLHS()->getType(); 611f22ef01cSRoman Divacky 612f22ef01cSRoman Divacky BinOpInfo OpInfo; 613f22ef01cSRoman Divacky 614f22ef01cSRoman Divacky // Load the RHS and LHS operands. 615f22ef01cSRoman Divacky // __block variables need to have the rhs evaluated first, plus this should 6162754fe60SDimitry Andric // improve codegen a little. 617f22ef01cSRoman Divacky OpInfo.Ty = E->getComputationResultType(); 6182754fe60SDimitry Andric 6192754fe60SDimitry Andric // The RHS should have been converted to the computation type. 6202754fe60SDimitry Andric assert(OpInfo.Ty->isAnyComplexType()); 6212754fe60SDimitry Andric assert(CGF.getContext().hasSameUnqualifiedType(OpInfo.Ty, 6222754fe60SDimitry Andric E->getRHS()->getType())); 6232754fe60SDimitry Andric OpInfo.RHS = Visit(E->getRHS()); 624f22ef01cSRoman Divacky 625ffd1746dSEd Schouten LValue LHS = CGF.EmitLValue(E->getLHS()); 6262754fe60SDimitry Andric 6272754fe60SDimitry Andric // Load from the l-value. 6282754fe60SDimitry Andric ComplexPairTy LHSComplexPair = EmitLoadOfLValue(LHS); 629f22ef01cSRoman Divacky 630f22ef01cSRoman Divacky OpInfo.LHS = EmitComplexToComplexCast(LHSComplexPair, LHSTy, OpInfo.Ty); 631f22ef01cSRoman Divacky 632f22ef01cSRoman Divacky // Expand the binary operator. 633f22ef01cSRoman Divacky ComplexPairTy Result = (this->*Func)(OpInfo); 634f22ef01cSRoman Divacky 635f22ef01cSRoman Divacky // Truncate the result back to the LHS type. 636f22ef01cSRoman Divacky Result = EmitComplexToComplexCast(Result, OpInfo.Ty, LHSTy); 6372754fe60SDimitry Andric Val = Result; 638f22ef01cSRoman Divacky 639f22ef01cSRoman Divacky // Store the result value into the LHS lvalue. 640139f7f9bSDimitry Andric EmitStoreOfComplex(Result, LHS, /*isInit*/ false); 641ffd1746dSEd Schouten 6422754fe60SDimitry Andric return LHS; 643f22ef01cSRoman Divacky } 644f22ef01cSRoman Divacky 6452754fe60SDimitry Andric // Compound assignments. 6462754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter:: 6472754fe60SDimitry Andric EmitCompoundAssign(const CompoundAssignOperator *E, 6482754fe60SDimitry Andric ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&)){ 6492754fe60SDimitry Andric ComplexPairTy Val; 6502754fe60SDimitry Andric LValue LV = EmitCompoundAssignLValue(E, Func, Val); 6512754fe60SDimitry Andric 6522754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value. 6533861d79fSDimitry Andric if (!CGF.getLangOpts().CPlusPlus) 6542754fe60SDimitry Andric return Val; 6552754fe60SDimitry Andric 6562754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 6572754fe60SDimitry Andric if (!LV.isVolatileQualified()) 6582754fe60SDimitry Andric return Val; 6592754fe60SDimitry Andric 660139f7f9bSDimitry Andric return EmitLoadOfLValue(LV); 6612754fe60SDimitry Andric } 6622754fe60SDimitry Andric 6632754fe60SDimitry Andric LValue ComplexExprEmitter::EmitBinAssignLValue(const BinaryOperator *E, 6642754fe60SDimitry Andric ComplexPairTy &Val) { 665ffd1746dSEd Schouten assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(), 666ffd1746dSEd Schouten E->getRHS()->getType()) && 667f22ef01cSRoman Divacky "Invalid assignment"); 6682754fe60SDimitry Andric TestAndClearIgnoreReal(); 6692754fe60SDimitry Andric TestAndClearIgnoreImag(); 6702754fe60SDimitry Andric 6712754fe60SDimitry Andric // Emit the RHS. __block variables need the RHS evaluated first. 6722754fe60SDimitry Andric Val = Visit(E->getRHS()); 673f22ef01cSRoman Divacky 674f22ef01cSRoman Divacky // Compute the address to store into. 675f22ef01cSRoman Divacky LValue LHS = CGF.EmitLValue(E->getLHS()); 676f22ef01cSRoman Divacky 677ffd1746dSEd Schouten // Store the result value into the LHS lvalue. 678139f7f9bSDimitry Andric EmitStoreOfComplex(Val, LHS, /*isInit*/ false); 679f22ef01cSRoman Divacky 6802754fe60SDimitry Andric return LHS; 6812754fe60SDimitry Andric } 682ffd1746dSEd Schouten 6832754fe60SDimitry Andric ComplexPairTy ComplexExprEmitter::VisitBinAssign(const BinaryOperator *E) { 6842754fe60SDimitry Andric ComplexPairTy Val; 6852754fe60SDimitry Andric LValue LV = EmitBinAssignLValue(E, Val); 6862754fe60SDimitry Andric 6872754fe60SDimitry Andric // The result of an assignment in C is the assigned r-value. 6883861d79fSDimitry Andric if (!CGF.getLangOpts().CPlusPlus) 6892754fe60SDimitry Andric return Val; 6902754fe60SDimitry Andric 6912754fe60SDimitry Andric // If the lvalue is non-volatile, return the computed value of the assignment. 6922754fe60SDimitry Andric if (!LV.isVolatileQualified()) 6932754fe60SDimitry Andric return Val; 6942754fe60SDimitry Andric 695139f7f9bSDimitry Andric return EmitLoadOfLValue(LV); 696f22ef01cSRoman Divacky } 697f22ef01cSRoman Divacky 698f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitBinComma(const BinaryOperator *E) { 6992754fe60SDimitry Andric CGF.EmitIgnoredExpr(E->getLHS()); 700f22ef01cSRoman Divacky return Visit(E->getRHS()); 701f22ef01cSRoman Divacky } 702f22ef01cSRoman Divacky 703f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter:: 7042754fe60SDimitry Andric VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) { 705f22ef01cSRoman Divacky TestAndClearIgnoreReal(); 706f22ef01cSRoman Divacky TestAndClearIgnoreImag(); 707f22ef01cSRoman Divacky llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true"); 708f22ef01cSRoman Divacky llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false"); 709f22ef01cSRoman Divacky llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end"); 710f22ef01cSRoman Divacky 7112754fe60SDimitry Andric // Bind the common expression if necessary. 7122754fe60SDimitry Andric CodeGenFunction::OpaqueValueMapping binding(CGF, E); 7132754fe60SDimitry Andric 7142754fe60SDimitry Andric CodeGenFunction::ConditionalEvaluation eval(CGF); 715f22ef01cSRoman Divacky CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock); 716f22ef01cSRoman Divacky 7172754fe60SDimitry Andric eval.begin(CGF); 718f22ef01cSRoman Divacky CGF.EmitBlock(LHSBlock); 7192754fe60SDimitry Andric ComplexPairTy LHS = Visit(E->getTrueExpr()); 720f22ef01cSRoman Divacky LHSBlock = Builder.GetInsertBlock(); 721f22ef01cSRoman Divacky CGF.EmitBranch(ContBlock); 7222754fe60SDimitry Andric eval.end(CGF); 723f22ef01cSRoman Divacky 7242754fe60SDimitry Andric eval.begin(CGF); 725f22ef01cSRoman Divacky CGF.EmitBlock(RHSBlock); 7262754fe60SDimitry Andric ComplexPairTy RHS = Visit(E->getFalseExpr()); 727f22ef01cSRoman Divacky RHSBlock = Builder.GetInsertBlock(); 728f22ef01cSRoman Divacky CGF.EmitBlock(ContBlock); 7292754fe60SDimitry Andric eval.end(CGF); 730f22ef01cSRoman Divacky 731f22ef01cSRoman Divacky // Create a PHI node for the real part. 7323b0f4066SDimitry Andric llvm::PHINode *RealPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.r"); 733f22ef01cSRoman Divacky RealPN->addIncoming(LHS.first, LHSBlock); 734f22ef01cSRoman Divacky RealPN->addIncoming(RHS.first, RHSBlock); 735f22ef01cSRoman Divacky 736f22ef01cSRoman Divacky // Create a PHI node for the imaginary part. 7373b0f4066SDimitry Andric llvm::PHINode *ImagPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.i"); 738f22ef01cSRoman Divacky ImagPN->addIncoming(LHS.second, LHSBlock); 739f22ef01cSRoman Divacky ImagPN->addIncoming(RHS.second, RHSBlock); 740f22ef01cSRoman Divacky 741f22ef01cSRoman Divacky return ComplexPairTy(RealPN, ImagPN); 742f22ef01cSRoman Divacky } 743f22ef01cSRoman Divacky 744f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitChooseExpr(ChooseExpr *E) { 745f22ef01cSRoman Divacky return Visit(E->getChosenSubExpr(CGF.getContext())); 746f22ef01cSRoman Divacky } 747f22ef01cSRoman Divacky 748f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitInitListExpr(InitListExpr *E) { 749f22ef01cSRoman Divacky bool Ignore = TestAndClearIgnoreReal(); 750f22ef01cSRoman Divacky (void)Ignore; 751f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored"); 752f22ef01cSRoman Divacky Ignore = TestAndClearIgnoreImag(); 753f22ef01cSRoman Divacky (void)Ignore; 754f22ef01cSRoman Divacky assert (Ignore == false && "init list ignored"); 7556122f3e6SDimitry Andric 7566122f3e6SDimitry Andric if (E->getNumInits() == 2) { 7576122f3e6SDimitry Andric llvm::Value *Real = CGF.EmitScalarExpr(E->getInit(0)); 7586122f3e6SDimitry Andric llvm::Value *Imag = CGF.EmitScalarExpr(E->getInit(1)); 7596122f3e6SDimitry Andric return ComplexPairTy(Real, Imag); 7606122f3e6SDimitry Andric } else if (E->getNumInits() == 1) { 761f22ef01cSRoman Divacky return Visit(E->getInit(0)); 7626122f3e6SDimitry Andric } 763f22ef01cSRoman Divacky 764f22ef01cSRoman Divacky // Empty init list intializes to null 7656122f3e6SDimitry Andric assert(E->getNumInits() == 0 && "Unexpected number of inits"); 766139f7f9bSDimitry Andric QualType Ty = E->getType()->castAs<ComplexType>()->getElementType(); 7676122f3e6SDimitry Andric llvm::Type* LTy = CGF.ConvertType(Ty); 768f22ef01cSRoman Divacky llvm::Value* zeroConstant = llvm::Constant::getNullValue(LTy); 769f22ef01cSRoman Divacky return ComplexPairTy(zeroConstant, zeroConstant); 770f22ef01cSRoman Divacky } 771f22ef01cSRoman Divacky 772f22ef01cSRoman Divacky ComplexPairTy ComplexExprEmitter::VisitVAArgExpr(VAArgExpr *E) { 773f22ef01cSRoman Divacky llvm::Value *ArgValue = CGF.EmitVAListRef(E->getSubExpr()); 774f22ef01cSRoman Divacky llvm::Value *ArgPtr = CGF.EmitVAArg(ArgValue, E->getType()); 775f22ef01cSRoman Divacky 776f22ef01cSRoman Divacky if (!ArgPtr) { 777f22ef01cSRoman Divacky CGF.ErrorUnsupported(E, "complex va_arg expression"); 7786122f3e6SDimitry Andric llvm::Type *EltTy = 779139f7f9bSDimitry Andric CGF.ConvertType(E->getType()->castAs<ComplexType>()->getElementType()); 780f22ef01cSRoman Divacky llvm::Value *U = llvm::UndefValue::get(EltTy); 781f22ef01cSRoman Divacky return ComplexPairTy(U, U); 782f22ef01cSRoman Divacky } 783f22ef01cSRoman Divacky 784139f7f9bSDimitry Andric return EmitLoadOfLValue( 785139f7f9bSDimitry Andric CGF.MakeNaturalAlignAddrLValue(ArgPtr, E->getType())); 786f22ef01cSRoman Divacky } 787f22ef01cSRoman Divacky 788f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 789f22ef01cSRoman Divacky // Entry Point into this File 790f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 791f22ef01cSRoman Divacky 792f22ef01cSRoman Divacky /// EmitComplexExpr - Emit the computation of the specified expression of 793f22ef01cSRoman Divacky /// complex type, ignoring the result. 794f22ef01cSRoman Divacky ComplexPairTy CodeGenFunction::EmitComplexExpr(const Expr *E, bool IgnoreReal, 7952754fe60SDimitry Andric bool IgnoreImag) { 796139f7f9bSDimitry Andric assert(E && getComplexType(E->getType()) && 797f22ef01cSRoman Divacky "Invalid complex expression to emit"); 798f22ef01cSRoman Divacky 7992754fe60SDimitry Andric return ComplexExprEmitter(*this, IgnoreReal, IgnoreImag) 800f22ef01cSRoman Divacky .Visit(const_cast<Expr*>(E)); 801f22ef01cSRoman Divacky } 802f22ef01cSRoman Divacky 803139f7f9bSDimitry Andric void CodeGenFunction::EmitComplexExprIntoLValue(const Expr *E, LValue dest, 804139f7f9bSDimitry Andric bool isInit) { 805139f7f9bSDimitry Andric assert(E && getComplexType(E->getType()) && 806f22ef01cSRoman Divacky "Invalid complex expression to emit"); 807f22ef01cSRoman Divacky ComplexExprEmitter Emitter(*this); 808f22ef01cSRoman Divacky ComplexPairTy Val = Emitter.Visit(const_cast<Expr*>(E)); 809139f7f9bSDimitry Andric Emitter.EmitStoreOfComplex(Val, dest, isInit); 810f22ef01cSRoman Divacky } 811f22ef01cSRoman Divacky 812139f7f9bSDimitry Andric /// EmitStoreOfComplex - Store a complex number into the specified l-value. 813139f7f9bSDimitry Andric void CodeGenFunction::EmitStoreOfComplex(ComplexPairTy V, LValue dest, 814139f7f9bSDimitry Andric bool isInit) { 815139f7f9bSDimitry Andric ComplexExprEmitter(*this).EmitStoreOfComplex(V, dest, isInit); 816f22ef01cSRoman Divacky } 817f22ef01cSRoman Divacky 818139f7f9bSDimitry Andric /// EmitLoadOfComplex - Load a complex number from the specified address. 819139f7f9bSDimitry Andric ComplexPairTy CodeGenFunction::EmitLoadOfComplex(LValue src) { 820139f7f9bSDimitry Andric return ComplexExprEmitter(*this).EmitLoadOfLValue(src); 821f22ef01cSRoman Divacky } 8222754fe60SDimitry Andric 8232754fe60SDimitry Andric LValue CodeGenFunction::EmitComplexAssignmentLValue(const BinaryOperator *E) { 8242754fe60SDimitry Andric assert(E->getOpcode() == BO_Assign); 8252754fe60SDimitry Andric ComplexPairTy Val; // ignored 8262754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitBinAssignLValue(E, Val); 8272754fe60SDimitry Andric } 8282754fe60SDimitry Andric 8292754fe60SDimitry Andric LValue CodeGenFunction:: 8302754fe60SDimitry Andric EmitComplexCompoundAssignmentLValue(const CompoundAssignOperator *E) { 8312754fe60SDimitry Andric ComplexPairTy(ComplexExprEmitter::*Op)(const ComplexExprEmitter::BinOpInfo &); 8322754fe60SDimitry Andric switch (E->getOpcode()) { 8332754fe60SDimitry Andric case BO_MulAssign: Op = &ComplexExprEmitter::EmitBinMul; break; 8342754fe60SDimitry Andric case BO_DivAssign: Op = &ComplexExprEmitter::EmitBinDiv; break; 8352754fe60SDimitry Andric case BO_SubAssign: Op = &ComplexExprEmitter::EmitBinSub; break; 8362754fe60SDimitry Andric case BO_AddAssign: Op = &ComplexExprEmitter::EmitBinAdd; break; 8372754fe60SDimitry Andric 8382754fe60SDimitry Andric default: 8392754fe60SDimitry Andric llvm_unreachable("unexpected complex compound assignment"); 8402754fe60SDimitry Andric } 8412754fe60SDimitry Andric 8422754fe60SDimitry Andric ComplexPairTy Val; // ignored 8432754fe60SDimitry Andric return ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val); 8442754fe60SDimitry Andric } 845