17a51313dSChris Lattner //===--- CGExprComplex.cpp - Emit LLVM Code for Complex Exprs -------------===//
27a51313dSChris Lattner //
37a51313dSChris Lattner //                     The LLVM Compiler Infrastructure
47a51313dSChris Lattner //
57a51313dSChris Lattner // This file is distributed under the University of Illinois Open Source
67a51313dSChris Lattner // License. See LICENSE.TXT for details.
77a51313dSChris Lattner //
87a51313dSChris Lattner //===----------------------------------------------------------------------===//
97a51313dSChris Lattner //
107a51313dSChris Lattner // This contains code to emit Expr nodes with complex types as LLVM code.
117a51313dSChris Lattner //
127a51313dSChris Lattner //===----------------------------------------------------------------------===//
137a51313dSChris Lattner 
147a51313dSChris Lattner #include "CodeGenFunction.h"
157a51313dSChris Lattner #include "CodeGenModule.h"
16ad319a73SDaniel Dunbar #include "clang/AST/ASTContext.h"
17ad319a73SDaniel Dunbar #include "clang/AST/StmtVisitor.h"
183a02247dSChandler Carruth #include "llvm/ADT/SmallString.h"
19ffd5551bSChandler Carruth #include "llvm/IR/Constants.h"
20ffd5551bSChandler Carruth #include "llvm/IR/Function.h"
2127dcbb24SJF Bastien #include <algorithm>
227a51313dSChris Lattner using namespace clang;
237a51313dSChris Lattner using namespace CodeGen;
247a51313dSChris Lattner 
257a51313dSChris Lattner //===----------------------------------------------------------------------===//
267a51313dSChris Lattner //                        Complex Expression Emitter
277a51313dSChris Lattner //===----------------------------------------------------------------------===//
287a51313dSChris Lattner 
297a51313dSChris Lattner typedef CodeGenFunction::ComplexPairTy ComplexPairTy;
307a51313dSChris Lattner 
3147fb9508SJohn McCall /// Return the complex type that we are meant to emit.
3247fb9508SJohn McCall static const ComplexType *getComplexType(QualType type) {
3347fb9508SJohn McCall   type = type.getCanonicalType();
3447fb9508SJohn McCall   if (const ComplexType *comp = dyn_cast<ComplexType>(type)) {
3547fb9508SJohn McCall     return comp;
3647fb9508SJohn McCall   } else {
3747fb9508SJohn McCall     return cast<ComplexType>(cast<AtomicType>(type)->getValueType());
3847fb9508SJohn McCall   }
3947fb9508SJohn McCall }
4047fb9508SJohn McCall 
417a51313dSChris Lattner namespace  {
42337e3a5fSBenjamin Kramer class ComplexExprEmitter
437a51313dSChris Lattner   : public StmtVisitor<ComplexExprEmitter, ComplexPairTy> {
447a51313dSChris Lattner   CodeGenFunction &CGF;
45cb463859SDaniel Dunbar   CGBuilderTy &Builder;
46df0fe27bSMike Stump   bool IgnoreReal;
47df0fe27bSMike Stump   bool IgnoreImag;
487a51313dSChris Lattner public:
4907bb1966SJohn McCall   ComplexExprEmitter(CodeGenFunction &cgf, bool ir=false, bool ii=false)
5007bb1966SJohn McCall     : CGF(cgf), Builder(CGF.Builder), IgnoreReal(ir), IgnoreImag(ii) {
517a51313dSChris Lattner   }
527a51313dSChris Lattner 
537a51313dSChris Lattner 
547a51313dSChris Lattner   //===--------------------------------------------------------------------===//
557a51313dSChris Lattner   //                               Utilities
567a51313dSChris Lattner   //===--------------------------------------------------------------------===//
577a51313dSChris Lattner 
58df0fe27bSMike Stump   bool TestAndClearIgnoreReal() {
59df0fe27bSMike Stump     bool I = IgnoreReal;
60df0fe27bSMike Stump     IgnoreReal = false;
61df0fe27bSMike Stump     return I;
62df0fe27bSMike Stump   }
63df0fe27bSMike Stump   bool TestAndClearIgnoreImag() {
64df0fe27bSMike Stump     bool I = IgnoreImag;
65df0fe27bSMike Stump     IgnoreImag = false;
66df0fe27bSMike Stump     return I;
67df0fe27bSMike Stump   }
68df0fe27bSMike Stump 
697a51313dSChris Lattner   /// EmitLoadOfLValue - Given an expression with complex type that represents a
707a51313dSChris Lattner   /// value l-value, this method emits the address of the l-value, then loads
717a51313dSChris Lattner   /// and returns the result.
727a51313dSChris Lattner   ComplexPairTy EmitLoadOfLValue(const Expr *E) {
732d84e842SNick Lewycky     return EmitLoadOfLValue(CGF.EmitLValue(E), E->getExprLoc());
74e26a872bSJohn McCall   }
75e26a872bSJohn McCall 
762d84e842SNick Lewycky   ComplexPairTy EmitLoadOfLValue(LValue LV, SourceLocation Loc);
77e26a872bSJohn McCall 
787a51313dSChris Lattner   /// EmitStoreOfComplex - Store the specified real/imag parts into the
797a51313dSChris Lattner   /// specified value pointer.
8047fb9508SJohn McCall   void EmitStoreOfComplex(ComplexPairTy Val, LValue LV, bool isInit);
817a51313dSChris Lattner 
827a51313dSChris Lattner   /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType.
837a51313dSChris Lattner   ComplexPairTy EmitComplexToComplexCast(ComplexPairTy Val, QualType SrcType,
847a51313dSChris Lattner                                          QualType DestType);
85f045007fSEli Friedman   /// EmitComplexToComplexCast - Emit a cast from scalar value Val to DestType.
86f045007fSEli Friedman   ComplexPairTy EmitScalarToComplexCast(llvm::Value *Val, QualType SrcType,
87f045007fSEli Friedman                                         QualType DestType);
887a51313dSChris Lattner 
897a51313dSChris Lattner   //===--------------------------------------------------------------------===//
907a51313dSChris Lattner   //                            Visitor Methods
917a51313dSChris Lattner   //===--------------------------------------------------------------------===//
927a51313dSChris Lattner 
938162d4adSFariborz Jahanian   ComplexPairTy Visit(Expr *E) {
948162d4adSFariborz Jahanian     return StmtVisitor<ComplexExprEmitter, ComplexPairTy>::Visit(E);
958162d4adSFariborz Jahanian   }
968162d4adSFariborz Jahanian 
977a51313dSChris Lattner   ComplexPairTy VisitStmt(Stmt *S) {
987a51313dSChris Lattner     S->dump(CGF.getContext().getSourceManager());
9983d382b1SDavid Blaikie     llvm_unreachable("Stmt can't have complex result type!");
1007a51313dSChris Lattner   }
1017a51313dSChris Lattner   ComplexPairTy VisitExpr(Expr *S);
1027a51313dSChris Lattner   ComplexPairTy VisitParenExpr(ParenExpr *PE) { return Visit(PE->getSubExpr());}
10391147596SPeter Collingbourne   ComplexPairTy VisitGenericSelectionExpr(GenericSelectionExpr *GE) {
10491147596SPeter Collingbourne     return Visit(GE->getResultExpr());
10591147596SPeter Collingbourne   }
1067a51313dSChris Lattner   ComplexPairTy VisitImaginaryLiteral(const ImaginaryLiteral *IL);
1077c454bb8SJohn McCall   ComplexPairTy
1087c454bb8SJohn McCall   VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *PE) {
1097c454bb8SJohn McCall     return Visit(PE->getReplacement());
1107c454bb8SJohn McCall   }
1117a51313dSChris Lattner 
1127a51313dSChris Lattner   // l-values.
113113bee05SJohn McCall   ComplexPairTy VisitDeclRefExpr(DeclRefExpr *E) {
114113bee05SJohn McCall     if (CodeGenFunction::ConstantEmission result = CGF.tryEmitAsConstant(E)) {
11571335059SJohn McCall       if (result.isReference())
1162d84e842SNick Lewycky         return EmitLoadOfLValue(result.getReferenceLValue(CGF, E),
1172d84e842SNick Lewycky                                 E->getExprLoc());
11871335059SJohn McCall 
11964f23918SEli Friedman       llvm::Constant *pair = result.getValue();
12064f23918SEli Friedman       return ComplexPairTy(pair->getAggregateElement(0U),
12164f23918SEli Friedman                            pair->getAggregateElement(1U));
12271335059SJohn McCall     }
123113bee05SJohn McCall     return EmitLoadOfLValue(E);
12471335059SJohn McCall   }
12576d864c7SDaniel Dunbar   ComplexPairTy VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
12676d864c7SDaniel Dunbar     return EmitLoadOfLValue(E);
12776d864c7SDaniel Dunbar   }
12876d864c7SDaniel Dunbar   ComplexPairTy VisitObjCMessageExpr(ObjCMessageExpr *E) {
12976d864c7SDaniel Dunbar     return CGF.EmitObjCMessageExpr(E).getComplexVal();
13076d864c7SDaniel Dunbar   }
1317a51313dSChris Lattner   ComplexPairTy VisitArraySubscriptExpr(Expr *E) { return EmitLoadOfLValue(E); }
1327a51313dSChris Lattner   ComplexPairTy VisitMemberExpr(const Expr *E) { return EmitLoadOfLValue(E); }
1331bf5846aSJohn McCall   ComplexPairTy VisitOpaqueValueExpr(OpaqueValueExpr *E) {
134c07a0c7eSJohn McCall     if (E->isGLValue())
1352d84e842SNick Lewycky       return EmitLoadOfLValue(CGF.getOpaqueLValueMapping(E), E->getExprLoc());
136c07a0c7eSJohn McCall     return CGF.getOpaqueRValueMapping(E).getComplexVal();
1371bf5846aSJohn McCall   }
1387a51313dSChris Lattner 
139fe96e0b6SJohn McCall   ComplexPairTy VisitPseudoObjectExpr(PseudoObjectExpr *E) {
140fe96e0b6SJohn McCall     return CGF.EmitPseudoObjectRValue(E).getComplexVal();
141fe96e0b6SJohn McCall   }
142fe96e0b6SJohn McCall 
1437a51313dSChris Lattner   // FIXME: CompoundLiteralExpr
1447a51313dSChris Lattner 
145c357f412SDouglas Gregor   ComplexPairTy EmitCast(CastExpr::CastKind CK, Expr *Op, QualType DestTy);
1467a51313dSChris Lattner   ComplexPairTy VisitImplicitCastExpr(ImplicitCastExpr *E) {
1477a51313dSChris Lattner     // Unlike for scalars, we don't have to worry about function->ptr demotion
1487a51313dSChris Lattner     // here.
149c357f412SDouglas Gregor     return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType());
1507a51313dSChris Lattner   }
1517a51313dSChris Lattner   ComplexPairTy VisitCastExpr(CastExpr *E) {
152c357f412SDouglas Gregor     return EmitCast(E->getCastKind(), E->getSubExpr(), E->getType());
1537a51313dSChris Lattner   }
1547a51313dSChris Lattner   ComplexPairTy VisitCallExpr(const CallExpr *E);
1557a51313dSChris Lattner   ComplexPairTy VisitStmtExpr(const StmtExpr *E);
1567a51313dSChris Lattner 
1577a51313dSChris Lattner   // Operators.
1587a51313dSChris Lattner   ComplexPairTy VisitPrePostIncDec(const UnaryOperator *E,
159116ce8f1SChris Lattner                                    bool isInc, bool isPre) {
160116ce8f1SChris Lattner     LValue LV = CGF.EmitLValue(E->getSubExpr());
161116ce8f1SChris Lattner     return CGF.EmitComplexPrePostIncDec(E, LV, isInc, isPre);
162116ce8f1SChris Lattner   }
1637a51313dSChris Lattner   ComplexPairTy VisitUnaryPostDec(const UnaryOperator *E) {
1647a51313dSChris Lattner     return VisitPrePostIncDec(E, false, false);
1657a51313dSChris Lattner   }
1667a51313dSChris Lattner   ComplexPairTy VisitUnaryPostInc(const UnaryOperator *E) {
1677a51313dSChris Lattner     return VisitPrePostIncDec(E, true, false);
1687a51313dSChris Lattner   }
1697a51313dSChris Lattner   ComplexPairTy VisitUnaryPreDec(const UnaryOperator *E) {
1707a51313dSChris Lattner     return VisitPrePostIncDec(E, false, true);
1717a51313dSChris Lattner   }
1727a51313dSChris Lattner   ComplexPairTy VisitUnaryPreInc(const UnaryOperator *E) {
1737a51313dSChris Lattner     return VisitPrePostIncDec(E, true, true);
1747a51313dSChris Lattner   }
1757a51313dSChris Lattner   ComplexPairTy VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); }
1767a51313dSChris Lattner   ComplexPairTy VisitUnaryPlus     (const UnaryOperator *E) {
177df0fe27bSMike Stump     TestAndClearIgnoreReal();
178df0fe27bSMike Stump     TestAndClearIgnoreImag();
1797a51313dSChris Lattner     return Visit(E->getSubExpr());
1807a51313dSChris Lattner   }
1817a51313dSChris Lattner   ComplexPairTy VisitUnaryMinus    (const UnaryOperator *E);
1827a51313dSChris Lattner   ComplexPairTy VisitUnaryNot      (const UnaryOperator *E);
1836f28289aSSebastian Redl   // LNot,Real,Imag never return complex.
1847a51313dSChris Lattner   ComplexPairTy VisitUnaryExtension(const UnaryOperator *E) {
1857a51313dSChris Lattner     return Visit(E->getSubExpr());
1867a51313dSChris Lattner   }
187aa9c7aedSChris Lattner   ComplexPairTy VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
188aa9c7aedSChris Lattner     return Visit(DAE->getExpr());
189aa9c7aedSChris Lattner   }
190852c9db7SRichard Smith   ComplexPairTy VisitCXXDefaultInitExpr(CXXDefaultInitExpr *DIE) {
191852c9db7SRichard Smith     CodeGenFunction::CXXDefaultInitExprScope Scope(CGF);
192852c9db7SRichard Smith     return Visit(DIE->getExpr());
193852c9db7SRichard Smith   }
1945d413781SJohn McCall   ComplexPairTy VisitExprWithCleanups(ExprWithCleanups *E) {
19508ef4660SJohn McCall     CGF.enterFullExpression(E);
19608ef4660SJohn McCall     CodeGenFunction::RunCleanupsScope Scope(CGF);
19708ef4660SJohn McCall     return Visit(E->getSubExpr());
198c0092ad3SAnders Carlsson   }
199747eb784SDouglas Gregor   ComplexPairTy VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
200ce4528f0SArgyrios Kyrtzidis     assert(E->getType()->isAnyComplexType() && "Expected complex type!");
20147fb9508SJohn McCall     QualType Elem = E->getType()->castAs<ComplexType>()->getElementType();
2024a3999feSMike Stump     llvm::Constant *Null = llvm::Constant::getNullValue(CGF.ConvertType(Elem));
203ce4528f0SArgyrios Kyrtzidis     return ComplexPairTy(Null, Null);
204ce4528f0SArgyrios Kyrtzidis   }
2050202cb40SDouglas Gregor   ComplexPairTy VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
2060202cb40SDouglas Gregor     assert(E->getType()->isAnyComplexType() && "Expected complex type!");
20747fb9508SJohn McCall     QualType Elem = E->getType()->castAs<ComplexType>()->getElementType();
208ae86c19eSOwen Anderson     llvm::Constant *Null =
2090b75f23bSOwen Anderson                        llvm::Constant::getNullValue(CGF.ConvertType(Elem));
2100202cb40SDouglas Gregor     return ComplexPairTy(Null, Null);
2110202cb40SDouglas Gregor   }
2127a51313dSChris Lattner 
2137a51313dSChris Lattner   struct BinOpInfo {
2147a51313dSChris Lattner     ComplexPairTy LHS;
2157a51313dSChris Lattner     ComplexPairTy RHS;
2167a51313dSChris Lattner     QualType Ty;  // Computation Type.
2177a51313dSChris Lattner   };
2187a51313dSChris Lattner 
2197a51313dSChris Lattner   BinOpInfo EmitBinOps(const BinaryOperator *E);
22007bb1966SJohn McCall   LValue EmitCompoundAssignLValue(const CompoundAssignOperator *E,
22107bb1966SJohn McCall                                   ComplexPairTy (ComplexExprEmitter::*Func)
22207bb1966SJohn McCall                                   (const BinOpInfo &),
223f045007fSEli Friedman                                   RValue &Val);
2247a51313dSChris Lattner   ComplexPairTy EmitCompoundAssign(const CompoundAssignOperator *E,
2257a51313dSChris Lattner                                    ComplexPairTy (ComplexExprEmitter::*Func)
2267a51313dSChris Lattner                                    (const BinOpInfo &));
2277a51313dSChris Lattner 
2287a51313dSChris Lattner   ComplexPairTy EmitBinAdd(const BinOpInfo &Op);
2297a51313dSChris Lattner   ComplexPairTy EmitBinSub(const BinOpInfo &Op);
2307a51313dSChris Lattner   ComplexPairTy EmitBinMul(const BinOpInfo &Op);
2317a51313dSChris Lattner   ComplexPairTy EmitBinDiv(const BinOpInfo &Op);
2327a51313dSChris Lattner 
2337a51313dSChris Lattner   ComplexPairTy VisitBinAdd(const BinaryOperator *E) {
2347a51313dSChris Lattner     return EmitBinAdd(EmitBinOps(E));
2357a51313dSChris Lattner   }
2367a51313dSChris Lattner   ComplexPairTy VisitBinSub(const BinaryOperator *E) {
2377a51313dSChris Lattner     return EmitBinSub(EmitBinOps(E));
2387a51313dSChris Lattner   }
23907bb1966SJohn McCall   ComplexPairTy VisitBinMul(const BinaryOperator *E) {
24007bb1966SJohn McCall     return EmitBinMul(EmitBinOps(E));
24107bb1966SJohn McCall   }
2427a51313dSChris Lattner   ComplexPairTy VisitBinDiv(const BinaryOperator *E) {
2437a51313dSChris Lattner     return EmitBinDiv(EmitBinOps(E));
2447a51313dSChris Lattner   }
2457a51313dSChris Lattner 
2467a51313dSChris Lattner   // Compound assignments.
2477a51313dSChris Lattner   ComplexPairTy VisitBinAddAssign(const CompoundAssignOperator *E) {
2487a51313dSChris Lattner     return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinAdd);
2497a51313dSChris Lattner   }
2507a51313dSChris Lattner   ComplexPairTy VisitBinSubAssign(const CompoundAssignOperator *E) {
2517a51313dSChris Lattner     return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinSub);
2527a51313dSChris Lattner   }
2537a51313dSChris Lattner   ComplexPairTy VisitBinMulAssign(const CompoundAssignOperator *E) {
2547a51313dSChris Lattner     return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinMul);
2557a51313dSChris Lattner   }
2567a51313dSChris Lattner   ComplexPairTy VisitBinDivAssign(const CompoundAssignOperator *E) {
2577a51313dSChris Lattner     return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinDiv);
2587a51313dSChris Lattner   }
2597a51313dSChris Lattner 
2607a51313dSChris Lattner   // GCC rejects rem/and/or/xor for integer complex.
2617a51313dSChris Lattner   // Logical and/or always return int, never complex.
2627a51313dSChris Lattner 
2637a51313dSChris Lattner   // No comparisons produce a complex result.
26407bb1966SJohn McCall 
26507bb1966SJohn McCall   LValue EmitBinAssignLValue(const BinaryOperator *E,
26607bb1966SJohn McCall                              ComplexPairTy &Val);
2677a51313dSChris Lattner   ComplexPairTy VisitBinAssign     (const BinaryOperator *E);
2687a51313dSChris Lattner   ComplexPairTy VisitBinComma      (const BinaryOperator *E);
2697a51313dSChris Lattner 
2707a51313dSChris Lattner 
271c07a0c7eSJohn McCall   ComplexPairTy
272c07a0c7eSJohn McCall   VisitAbstractConditionalOperator(const AbstractConditionalOperator *CO);
2737a51313dSChris Lattner   ComplexPairTy VisitChooseExpr(ChooseExpr *CE);
274dd7406e6SEli Friedman 
275dd7406e6SEli Friedman   ComplexPairTy VisitInitListExpr(InitListExpr *E);
27600079612SDaniel Dunbar 
277afa84ae8SEli Friedman   ComplexPairTy VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
278afa84ae8SEli Friedman     return EmitLoadOfLValue(E);
279afa84ae8SEli Friedman   }
280afa84ae8SEli Friedman 
28100079612SDaniel Dunbar   ComplexPairTy VisitVAArgExpr(VAArgExpr *E);
282df14b3a8SEli Friedman 
283df14b3a8SEli Friedman   ComplexPairTy VisitAtomicExpr(AtomicExpr *E) {
284df14b3a8SEli Friedman     return CGF.EmitAtomicExpr(E).getComplexVal();
285df14b3a8SEli Friedman   }
2867a51313dSChris Lattner };
2877a51313dSChris Lattner }  // end anonymous namespace.
2887a51313dSChris Lattner 
2897a51313dSChris Lattner //===----------------------------------------------------------------------===//
2907a51313dSChris Lattner //                                Utilities
2917a51313dSChris Lattner //===----------------------------------------------------------------------===//
2927a51313dSChris Lattner 
29347fb9508SJohn McCall /// EmitLoadOfLValue - Given an RValue reference for a complex, emit code to
2947a51313dSChris Lattner /// load the real and imaginary pieces, returning them as Real/Imag.
2952d84e842SNick Lewycky ComplexPairTy ComplexExprEmitter::EmitLoadOfLValue(LValue lvalue,
2962d84e842SNick Lewycky                                                    SourceLocation loc) {
29747fb9508SJohn McCall   assert(lvalue.isSimple() && "non-simple complex l-value?");
298a8ec7eb9SJohn McCall   if (lvalue.getType()->isAtomicType())
2992d84e842SNick Lewycky     return CGF.EmitAtomicLoad(lvalue, loc).getComplexVal();
300a8ec7eb9SJohn McCall 
30147fb9508SJohn McCall   llvm::Value *SrcPtr = lvalue.getAddress();
30247fb9508SJohn McCall   bool isVolatile = lvalue.isVolatileQualified();
30327dcbb24SJF Bastien   unsigned AlignR = lvalue.getAlignment().getQuantity();
30427dcbb24SJF Bastien   ASTContext &C = CGF.getContext();
30527dcbb24SJF Bastien   QualType ComplexTy = lvalue.getType();
30627dcbb24SJF Bastien   unsigned ComplexAlign = C.getTypeAlignInChars(ComplexTy).getQuantity();
30727dcbb24SJF Bastien   unsigned AlignI = std::min(AlignR, ComplexAlign);
30847fb9508SJohn McCall 
309*8a13c418SCraig Topper   llvm::Value *Real=nullptr, *Imag=nullptr;
310df0fe27bSMike Stump 
31183fe49d1SJohn McCall   if (!IgnoreReal || isVolatile) {
312ba9fd986SBenjamin Kramer     llvm::Value *RealP = Builder.CreateStructGEP(SrcPtr, 0,
313ba9fd986SBenjamin Kramer                                                  SrcPtr->getName() + ".realp");
31427dcbb24SJF Bastien     Real = Builder.CreateAlignedLoad(RealP, AlignR, isVolatile,
31527dcbb24SJF Bastien                                      SrcPtr->getName() + ".real");
316df0fe27bSMike Stump   }
317df0fe27bSMike Stump 
31883fe49d1SJohn McCall   if (!IgnoreImag || isVolatile) {
319ba9fd986SBenjamin Kramer     llvm::Value *ImagP = Builder.CreateStructGEP(SrcPtr, 1,
320ba9fd986SBenjamin Kramer                                                  SrcPtr->getName() + ".imagp");
32127dcbb24SJF Bastien     Imag = Builder.CreateAlignedLoad(ImagP, AlignI, isVolatile,
32227dcbb24SJF Bastien                                      SrcPtr->getName() + ".imag");
323df0fe27bSMike Stump   }
3247a51313dSChris Lattner   return ComplexPairTy(Real, Imag);
3257a51313dSChris Lattner }
3267a51313dSChris Lattner 
3277a51313dSChris Lattner /// EmitStoreOfComplex - Store the specified real/imag parts into the
3287a51313dSChris Lattner /// specified value pointer.
32947fb9508SJohn McCall void ComplexExprEmitter::EmitStoreOfComplex(ComplexPairTy Val,
33047fb9508SJohn McCall                                             LValue lvalue,
33147fb9508SJohn McCall                                             bool isInit) {
332a8ec7eb9SJohn McCall   if (lvalue.getType()->isAtomicType())
333a8ec7eb9SJohn McCall     return CGF.EmitAtomicStore(RValue::getComplex(Val), lvalue, isInit);
334a8ec7eb9SJohn McCall 
33547fb9508SJohn McCall   llvm::Value *Ptr = lvalue.getAddress();
3363e593cdbSChris Lattner   llvm::Value *RealPtr = Builder.CreateStructGEP(Ptr, 0, "real");
3373e593cdbSChris Lattner   llvm::Value *ImagPtr = Builder.CreateStructGEP(Ptr, 1, "imag");
33827dcbb24SJF Bastien   unsigned AlignR = lvalue.getAlignment().getQuantity();
33927dcbb24SJF Bastien   ASTContext &C = CGF.getContext();
34027dcbb24SJF Bastien   QualType ComplexTy = lvalue.getType();
34127dcbb24SJF Bastien   unsigned ComplexAlign = C.getTypeAlignInChars(ComplexTy).getQuantity();
34227dcbb24SJF Bastien   unsigned AlignI = std::min(AlignR, ComplexAlign);
3437a51313dSChris Lattner 
34427dcbb24SJF Bastien   Builder.CreateAlignedStore(Val.first, RealPtr, AlignR,
34527dcbb24SJF Bastien                              lvalue.isVolatileQualified());
34627dcbb24SJF Bastien   Builder.CreateAlignedStore(Val.second, ImagPtr, AlignI,
34727dcbb24SJF Bastien                              lvalue.isVolatileQualified());
3487a51313dSChris Lattner }
3497a51313dSChris Lattner 
3507a51313dSChris Lattner 
3517a51313dSChris Lattner 
3527a51313dSChris Lattner //===----------------------------------------------------------------------===//
3537a51313dSChris Lattner //                            Visitor Methods
3547a51313dSChris Lattner //===----------------------------------------------------------------------===//
3557a51313dSChris Lattner 
3567a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::VisitExpr(Expr *E) {
357b74711dfSFariborz Jahanian   CGF.ErrorUnsupported(E, "complex expression");
358b74711dfSFariborz Jahanian   llvm::Type *EltTy =
35947fb9508SJohn McCall     CGF.ConvertType(getComplexType(E->getType())->getElementType());
360b74711dfSFariborz Jahanian   llvm::Value *U = llvm::UndefValue::get(EltTy);
361b74711dfSFariborz Jahanian   return ComplexPairTy(U, U);
3627a51313dSChris Lattner }
3637a51313dSChris Lattner 
3647a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::
3657a51313dSChris Lattner VisitImaginaryLiteral(const ImaginaryLiteral *IL) {
3667a51313dSChris Lattner   llvm::Value *Imag = CGF.EmitScalarExpr(IL->getSubExpr());
367294c2db4SJohn McCall   return ComplexPairTy(llvm::Constant::getNullValue(Imag->getType()), Imag);
3687a51313dSChris Lattner }
3697a51313dSChris Lattner 
3707a51313dSChris Lattner 
3717a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::VisitCallExpr(const CallExpr *E) {
372d8b7ae20SAnders Carlsson   if (E->getCallReturnType()->isReferenceType())
373d8b7ae20SAnders Carlsson     return EmitLoadOfLValue(E);
374d8b7ae20SAnders Carlsson 
3757a51313dSChris Lattner   return CGF.EmitCallExpr(E).getComplexVal();
3767a51313dSChris Lattner }
3777a51313dSChris Lattner 
3787a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::VisitStmtExpr(const StmtExpr *E) {
379ce1de617SJohn McCall   CodeGenFunction::StmtExprEvaluation eval(CGF);
3804871a46cSEli Friedman   llvm::Value *RetAlloca = CGF.EmitCompoundStmt(*E->getSubStmt(), true);
3814871a46cSEli Friedman   assert(RetAlloca && "Expected complex return value");
3822d84e842SNick Lewycky   return EmitLoadOfLValue(CGF.MakeAddrLValue(RetAlloca, E->getType()),
3832d84e842SNick Lewycky                           E->getExprLoc());
3847a51313dSChris Lattner }
3857a51313dSChris Lattner 
3867a51313dSChris Lattner /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType.
3877a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::EmitComplexToComplexCast(ComplexPairTy Val,
3887a51313dSChris Lattner                                                            QualType SrcType,
3897a51313dSChris Lattner                                                            QualType DestType) {
3907a51313dSChris Lattner   // Get the src/dest element type.
39147fb9508SJohn McCall   SrcType = SrcType->castAs<ComplexType>()->getElementType();
39247fb9508SJohn McCall   DestType = DestType->castAs<ComplexType>()->getElementType();
3937a51313dSChris Lattner 
3947a51313dSChris Lattner   // C99 6.3.1.6: When a value of complex type is converted to another
3957a51313dSChris Lattner   // complex type, both the real and imaginary parts follow the conversion
3967a51313dSChris Lattner   // rules for the corresponding real types.
3977a51313dSChris Lattner   Val.first = CGF.EmitScalarConversion(Val.first, SrcType, DestType);
3987a51313dSChris Lattner   Val.second = CGF.EmitScalarConversion(Val.second, SrcType, DestType);
3997a51313dSChris Lattner   return Val;
4007a51313dSChris Lattner }
4017a51313dSChris Lattner 
402f045007fSEli Friedman ComplexPairTy ComplexExprEmitter::EmitScalarToComplexCast(llvm::Value *Val,
403f045007fSEli Friedman                                                           QualType SrcType,
404f045007fSEli Friedman                                                           QualType DestType) {
405f045007fSEli Friedman   // Convert the input element to the element type of the complex.
406f045007fSEli Friedman   DestType = DestType->castAs<ComplexType>()->getElementType();
407f045007fSEli Friedman   Val = CGF.EmitScalarConversion(Val, SrcType, DestType);
408f045007fSEli Friedman 
409f045007fSEli Friedman   // Return (realval, 0).
410f045007fSEli Friedman   return ComplexPairTy(Val, llvm::Constant::getNullValue(Val->getType()));
411f045007fSEli Friedman }
412f045007fSEli Friedman 
413c357f412SDouglas Gregor ComplexPairTy ComplexExprEmitter::EmitCast(CastExpr::CastKind CK, Expr *Op,
414c357f412SDouglas Gregor                                            QualType DestTy) {
41534376a68SJohn McCall   switch (CK) {
4165836852eSEli Friedman   case CK_Dependent: llvm_unreachable("dependent cast kind in IR gen!");
41734376a68SJohn McCall 
418fa35df62SDavid Chisnall   // Atomic to non-atomic casts may be more than a no-op for some platforms and
419fa35df62SDavid Chisnall   // for some types.
420fa35df62SDavid Chisnall   case CK_AtomicToNonAtomic:
421fa35df62SDavid Chisnall   case CK_NonAtomicToAtomic:
42234376a68SJohn McCall   case CK_NoOp:
42334376a68SJohn McCall   case CK_LValueToRValue:
4245836852eSEli Friedman   case CK_UserDefinedConversion:
42534376a68SJohn McCall     return Visit(Op);
42634376a68SJohn McCall 
4275836852eSEli Friedman   case CK_LValueBitCast: {
42847fb9508SJohn McCall     LValue origLV = CGF.EmitLValue(Op);
42947fb9508SJohn McCall     llvm::Value *V = origLV.getAddress();
430c357f412SDouglas Gregor     V = Builder.CreateBitCast(V,
431c357f412SDouglas Gregor                     CGF.ConvertType(CGF.getContext().getPointerType(DestTy)));
43247fb9508SJohn McCall     return EmitLoadOfLValue(CGF.MakeAddrLValue(V, DestTy,
4332d84e842SNick Lewycky                                                origLV.getAlignment()),
4342d84e842SNick Lewycky                             Op->getExprLoc());
435c357f412SDouglas Gregor   }
436c357f412SDouglas Gregor 
4375836852eSEli Friedman   case CK_BitCast:
4385836852eSEli Friedman   case CK_BaseToDerived:
4395836852eSEli Friedman   case CK_DerivedToBase:
4405836852eSEli Friedman   case CK_UncheckedDerivedToBase:
4415836852eSEli Friedman   case CK_Dynamic:
4425836852eSEli Friedman   case CK_ToUnion:
4435836852eSEli Friedman   case CK_ArrayToPointerDecay:
4445836852eSEli Friedman   case CK_FunctionToPointerDecay:
4455836852eSEli Friedman   case CK_NullToPointer:
4465836852eSEli Friedman   case CK_NullToMemberPointer:
4475836852eSEli Friedman   case CK_BaseToDerivedMemberPointer:
4485836852eSEli Friedman   case CK_DerivedToBaseMemberPointer:
4495836852eSEli Friedman   case CK_MemberPointerToBoolean:
450c62bb391SJohn McCall   case CK_ReinterpretMemberPointer:
4515836852eSEli Friedman   case CK_ConstructorConversion:
4525836852eSEli Friedman   case CK_IntegralToPointer:
4535836852eSEli Friedman   case CK_PointerToIntegral:
4545836852eSEli Friedman   case CK_PointerToBoolean:
4555836852eSEli Friedman   case CK_ToVoid:
4565836852eSEli Friedman   case CK_VectorSplat:
4575836852eSEli Friedman   case CK_IntegralCast:
4585836852eSEli Friedman   case CK_IntegralToBoolean:
4595836852eSEli Friedman   case CK_IntegralToFloating:
4605836852eSEli Friedman   case CK_FloatingToIntegral:
4615836852eSEli Friedman   case CK_FloatingToBoolean:
4625836852eSEli Friedman   case CK_FloatingCast:
4639320b87cSJohn McCall   case CK_CPointerToObjCPointerCast:
4649320b87cSJohn McCall   case CK_BlockPointerToObjCPointerCast:
4655836852eSEli Friedman   case CK_AnyPointerToBlockPointerCast:
4665836852eSEli Friedman   case CK_ObjCObjectLValueCast:
4675836852eSEli Friedman   case CK_FloatingComplexToReal:
4685836852eSEli Friedman   case CK_FloatingComplexToBoolean:
4695836852eSEli Friedman   case CK_IntegralComplexToReal:
4705836852eSEli Friedman   case CK_IntegralComplexToBoolean:
4712d637d2eSJohn McCall   case CK_ARCProduceObject:
4722d637d2eSJohn McCall   case CK_ARCConsumeObject:
4732d637d2eSJohn McCall   case CK_ARCReclaimReturnedObject:
4742d637d2eSJohn McCall   case CK_ARCExtendBlockObject:
475ed90df38SDouglas Gregor   case CK_CopyAndAutoreleaseBlockObject:
47634866c77SEli Friedman   case CK_BuiltinFnToFnPtr:
4771b4fb3e0SGuy Benyei   case CK_ZeroToOCLEvent:
478e1468322SDavid Tweed   case CK_AddressSpaceConversion:
4795836852eSEli Friedman     llvm_unreachable("invalid cast kind for complex value");
4805836852eSEli Friedman 
4815836852eSEli Friedman   case CK_FloatingRealToComplex:
482f045007fSEli Friedman   case CK_IntegralRealToComplex:
483f045007fSEli Friedman     return EmitScalarToComplexCast(CGF.EmitScalarExpr(Op),
484f045007fSEli Friedman                                    Op->getType(), DestTy);
4857a51313dSChris Lattner 
4865836852eSEli Friedman   case CK_FloatingComplexCast:
4875836852eSEli Friedman   case CK_FloatingComplexToIntegralComplex:
4885836852eSEli Friedman   case CK_IntegralComplexCast:
4895836852eSEli Friedman   case CK_IntegralComplexToFloatingComplex:
4905836852eSEli Friedman     return EmitComplexToComplexCast(Visit(Op), Op->getType(), DestTy);
4915836852eSEli Friedman   }
4925836852eSEli Friedman 
4935836852eSEli Friedman   llvm_unreachable("unknown cast resulting in complex value");
4945836852eSEli Friedman }
4955836852eSEli Friedman 
4967a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::VisitUnaryMinus(const UnaryOperator *E) {
497df0fe27bSMike Stump   TestAndClearIgnoreReal();
498df0fe27bSMike Stump   TestAndClearIgnoreImag();
4997a51313dSChris Lattner   ComplexPairTy Op = Visit(E->getSubExpr());
50094dfae24SChris Lattner 
50194dfae24SChris Lattner   llvm::Value *ResR, *ResI;
502998f9d97SDuncan Sands   if (Op.first->getType()->isFloatingPointTy()) {
50394dfae24SChris Lattner     ResR = Builder.CreateFNeg(Op.first,  "neg.r");
50494dfae24SChris Lattner     ResI = Builder.CreateFNeg(Op.second, "neg.i");
50594dfae24SChris Lattner   } else {
50694dfae24SChris Lattner     ResR = Builder.CreateNeg(Op.first,  "neg.r");
50794dfae24SChris Lattner     ResI = Builder.CreateNeg(Op.second, "neg.i");
50894dfae24SChris Lattner   }
5097a51313dSChris Lattner   return ComplexPairTy(ResR, ResI);
5107a51313dSChris Lattner }
5117a51313dSChris Lattner 
5127a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::VisitUnaryNot(const UnaryOperator *E) {
513df0fe27bSMike Stump   TestAndClearIgnoreReal();
514df0fe27bSMike Stump   TestAndClearIgnoreImag();
5157a51313dSChris Lattner   // ~(a+ib) = a + i*-b
5167a51313dSChris Lattner   ComplexPairTy Op = Visit(E->getSubExpr());
51794dfae24SChris Lattner   llvm::Value *ResI;
518998f9d97SDuncan Sands   if (Op.second->getType()->isFloatingPointTy())
51994dfae24SChris Lattner     ResI = Builder.CreateFNeg(Op.second, "conj.i");
52094dfae24SChris Lattner   else
52194dfae24SChris Lattner     ResI = Builder.CreateNeg(Op.second, "conj.i");
52294dfae24SChris Lattner 
5237a51313dSChris Lattner   return ComplexPairTy(Op.first, ResI);
5247a51313dSChris Lattner }
5257a51313dSChris Lattner 
5267a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::EmitBinAdd(const BinOpInfo &Op) {
52794dfae24SChris Lattner   llvm::Value *ResR, *ResI;
52894dfae24SChris Lattner 
529998f9d97SDuncan Sands   if (Op.LHS.first->getType()->isFloatingPointTy()) {
53094dfae24SChris Lattner     ResR = Builder.CreateFAdd(Op.LHS.first,  Op.RHS.first,  "add.r");
53194dfae24SChris Lattner     ResI = Builder.CreateFAdd(Op.LHS.second, Op.RHS.second, "add.i");
53294dfae24SChris Lattner   } else {
53394dfae24SChris Lattner     ResR = Builder.CreateAdd(Op.LHS.first,  Op.RHS.first,  "add.r");
53494dfae24SChris Lattner     ResI = Builder.CreateAdd(Op.LHS.second, Op.RHS.second, "add.i");
53594dfae24SChris Lattner   }
5367a51313dSChris Lattner   return ComplexPairTy(ResR, ResI);
5377a51313dSChris Lattner }
5387a51313dSChris Lattner 
5397a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::EmitBinSub(const BinOpInfo &Op) {
54094dfae24SChris Lattner   llvm::Value *ResR, *ResI;
541998f9d97SDuncan Sands   if (Op.LHS.first->getType()->isFloatingPointTy()) {
54294dfae24SChris Lattner     ResR = Builder.CreateFSub(Op.LHS.first,  Op.RHS.first,  "sub.r");
54394dfae24SChris Lattner     ResI = Builder.CreateFSub(Op.LHS.second, Op.RHS.second, "sub.i");
54494dfae24SChris Lattner   } else {
54594dfae24SChris Lattner     ResR = Builder.CreateSub(Op.LHS.first,  Op.RHS.first,  "sub.r");
54694dfae24SChris Lattner     ResI = Builder.CreateSub(Op.LHS.second, Op.RHS.second, "sub.i");
54794dfae24SChris Lattner   }
5487a51313dSChris Lattner   return ComplexPairTy(ResR, ResI);
5497a51313dSChris Lattner }
5507a51313dSChris Lattner 
5517a51313dSChris Lattner 
5527a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::EmitBinMul(const BinOpInfo &Op) {
55394dfae24SChris Lattner   using llvm::Value;
55494dfae24SChris Lattner   Value *ResR, *ResI;
5557a51313dSChris Lattner 
556998f9d97SDuncan Sands   if (Op.LHS.first->getType()->isFloatingPointTy()) {
55794dfae24SChris Lattner     Value *ResRl = Builder.CreateFMul(Op.LHS.first, Op.RHS.first, "mul.rl");
55894dfae24SChris Lattner     Value *ResRr = Builder.CreateFMul(Op.LHS.second, Op.RHS.second,"mul.rr");
55994dfae24SChris Lattner     ResR  = Builder.CreateFSub(ResRl, ResRr, "mul.r");
56094dfae24SChris Lattner 
56194dfae24SChris Lattner     Value *ResIl = Builder.CreateFMul(Op.LHS.second, Op.RHS.first, "mul.il");
56294dfae24SChris Lattner     Value *ResIr = Builder.CreateFMul(Op.LHS.first, Op.RHS.second, "mul.ir");
56394dfae24SChris Lattner     ResI  = Builder.CreateFAdd(ResIl, ResIr, "mul.i");
56494dfae24SChris Lattner   } else {
56594dfae24SChris Lattner     Value *ResRl = Builder.CreateMul(Op.LHS.first, Op.RHS.first, "mul.rl");
56694dfae24SChris Lattner     Value *ResRr = Builder.CreateMul(Op.LHS.second, Op.RHS.second,"mul.rr");
56794dfae24SChris Lattner     ResR  = Builder.CreateSub(ResRl, ResRr, "mul.r");
56894dfae24SChris Lattner 
56994dfae24SChris Lattner     Value *ResIl = Builder.CreateMul(Op.LHS.second, Op.RHS.first, "mul.il");
57094dfae24SChris Lattner     Value *ResIr = Builder.CreateMul(Op.LHS.first, Op.RHS.second, "mul.ir");
57194dfae24SChris Lattner     ResI  = Builder.CreateAdd(ResIl, ResIr, "mul.i");
57294dfae24SChris Lattner   }
5737a51313dSChris Lattner   return ComplexPairTy(ResR, ResI);
5747a51313dSChris Lattner }
5757a51313dSChris Lattner 
5767a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::EmitBinDiv(const BinOpInfo &Op) {
5777a51313dSChris Lattner   llvm::Value *LHSr = Op.LHS.first, *LHSi = Op.LHS.second;
5787a51313dSChris Lattner   llvm::Value *RHSr = Op.RHS.first, *RHSi = Op.RHS.second;
5797a51313dSChris Lattner 
58094dfae24SChris Lattner 
58194dfae24SChris Lattner   llvm::Value *DSTr, *DSTi;
582998f9d97SDuncan Sands   if (Op.LHS.first->getType()->isFloatingPointTy()) {
58394dfae24SChris Lattner     // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd))
58476399eb2SBenjamin Kramer     llvm::Value *Tmp1 = Builder.CreateFMul(LHSr, RHSr); // a*c
58576399eb2SBenjamin Kramer     llvm::Value *Tmp2 = Builder.CreateFMul(LHSi, RHSi); // b*d
58676399eb2SBenjamin Kramer     llvm::Value *Tmp3 = Builder.CreateFAdd(Tmp1, Tmp2); // ac+bd
58794dfae24SChris Lattner 
58876399eb2SBenjamin Kramer     llvm::Value *Tmp4 = Builder.CreateFMul(RHSr, RHSr); // c*c
58976399eb2SBenjamin Kramer     llvm::Value *Tmp5 = Builder.CreateFMul(RHSi, RHSi); // d*d
59076399eb2SBenjamin Kramer     llvm::Value *Tmp6 = Builder.CreateFAdd(Tmp4, Tmp5); // cc+dd
59194dfae24SChris Lattner 
59276399eb2SBenjamin Kramer     llvm::Value *Tmp7 = Builder.CreateFMul(LHSi, RHSr); // b*c
59376399eb2SBenjamin Kramer     llvm::Value *Tmp8 = Builder.CreateFMul(LHSr, RHSi); // a*d
59476399eb2SBenjamin Kramer     llvm::Value *Tmp9 = Builder.CreateFSub(Tmp7, Tmp8); // bc-ad
59594dfae24SChris Lattner 
59676399eb2SBenjamin Kramer     DSTr = Builder.CreateFDiv(Tmp3, Tmp6);
59776399eb2SBenjamin Kramer     DSTi = Builder.CreateFDiv(Tmp9, Tmp6);
59894dfae24SChris Lattner   } else {
5997a51313dSChris Lattner     // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd))
60076399eb2SBenjamin Kramer     llvm::Value *Tmp1 = Builder.CreateMul(LHSr, RHSr); // a*c
60176399eb2SBenjamin Kramer     llvm::Value *Tmp2 = Builder.CreateMul(LHSi, RHSi); // b*d
60276399eb2SBenjamin Kramer     llvm::Value *Tmp3 = Builder.CreateAdd(Tmp1, Tmp2); // ac+bd
6037a51313dSChris Lattner 
60476399eb2SBenjamin Kramer     llvm::Value *Tmp4 = Builder.CreateMul(RHSr, RHSr); // c*c
60576399eb2SBenjamin Kramer     llvm::Value *Tmp5 = Builder.CreateMul(RHSi, RHSi); // d*d
60676399eb2SBenjamin Kramer     llvm::Value *Tmp6 = Builder.CreateAdd(Tmp4, Tmp5); // cc+dd
6077a51313dSChris Lattner 
60876399eb2SBenjamin Kramer     llvm::Value *Tmp7 = Builder.CreateMul(LHSi, RHSr); // b*c
60976399eb2SBenjamin Kramer     llvm::Value *Tmp8 = Builder.CreateMul(LHSr, RHSi); // a*d
61076399eb2SBenjamin Kramer     llvm::Value *Tmp9 = Builder.CreateSub(Tmp7, Tmp8); // bc-ad
6117a51313dSChris Lattner 
61247fb9508SJohn McCall     if (Op.Ty->castAs<ComplexType>()->getElementType()->isUnsignedIntegerType()) {
61376399eb2SBenjamin Kramer       DSTr = Builder.CreateUDiv(Tmp3, Tmp6);
61476399eb2SBenjamin Kramer       DSTi = Builder.CreateUDiv(Tmp9, Tmp6);
6157a51313dSChris Lattner     } else {
61676399eb2SBenjamin Kramer       DSTr = Builder.CreateSDiv(Tmp3, Tmp6);
61776399eb2SBenjamin Kramer       DSTi = Builder.CreateSDiv(Tmp9, Tmp6);
6187a51313dSChris Lattner     }
6197a51313dSChris Lattner   }
6207a51313dSChris Lattner 
6217a51313dSChris Lattner   return ComplexPairTy(DSTr, DSTi);
6227a51313dSChris Lattner }
6237a51313dSChris Lattner 
6247a51313dSChris Lattner ComplexExprEmitter::BinOpInfo
6257a51313dSChris Lattner ComplexExprEmitter::EmitBinOps(const BinaryOperator *E) {
626df0fe27bSMike Stump   TestAndClearIgnoreReal();
627df0fe27bSMike Stump   TestAndClearIgnoreImag();
6287a51313dSChris Lattner   BinOpInfo Ops;
6297a51313dSChris Lattner   Ops.LHS = Visit(E->getLHS());
6307a51313dSChris Lattner   Ops.RHS = Visit(E->getRHS());
6317a51313dSChris Lattner   Ops.Ty = E->getType();
6327a51313dSChris Lattner   return Ops;
6337a51313dSChris Lattner }
6347a51313dSChris Lattner 
6357a51313dSChris Lattner 
63607bb1966SJohn McCall LValue ComplexExprEmitter::
63707bb1966SJohn McCall EmitCompoundAssignLValue(const CompoundAssignOperator *E,
63807bb1966SJohn McCall           ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&),
639f045007fSEli Friedman                          RValue &Val) {
640df0fe27bSMike Stump   TestAndClearIgnoreReal();
641df0fe27bSMike Stump   TestAndClearIgnoreImag();
6428dfa5f17SJeffrey Yasskin   QualType LHSTy = E->getLHS()->getType();
6437a51313dSChris Lattner 
644df0fe27bSMike Stump   BinOpInfo OpInfo;
645df0fe27bSMike Stump 
646df0fe27bSMike Stump   // Load the RHS and LHS operands.
647df0fe27bSMike Stump   // __block variables need to have the rhs evaluated first, plus this should
648fa8edb11SJohn McCall   // improve codegen a little.
649df0fe27bSMike Stump   OpInfo.Ty = E->getComputationResultType();
650fa8edb11SJohn McCall 
651fa8edb11SJohn McCall   // The RHS should have been converted to the computation type.
652fa8edb11SJohn McCall   assert(OpInfo.Ty->isAnyComplexType());
653fa8edb11SJohn McCall   assert(CGF.getContext().hasSameUnqualifiedType(OpInfo.Ty,
654fa8edb11SJohn McCall                                                  E->getRHS()->getType()));
655fa8edb11SJohn McCall   OpInfo.RHS = Visit(E->getRHS());
656df0fe27bSMike Stump 
6578c94ffe7SDaniel Dunbar   LValue LHS = CGF.EmitLValue(E->getLHS());
658e26a872bSJohn McCall 
659f045007fSEli Friedman   // Load from the l-value and convert it.
660f045007fSEli Friedman   if (LHSTy->isAnyComplexType()) {
6612d84e842SNick Lewycky     ComplexPairTy LHSVal = EmitLoadOfLValue(LHS, E->getExprLoc());
662f045007fSEli Friedman     OpInfo.LHS = EmitComplexToComplexCast(LHSVal, LHSTy, OpInfo.Ty);
663f045007fSEli Friedman   } else {
6642d84e842SNick Lewycky     llvm::Value *LHSVal = CGF.EmitLoadOfScalar(LHS, E->getExprLoc());
665f045007fSEli Friedman     OpInfo.LHS = EmitScalarToComplexCast(LHSVal, LHSTy, OpInfo.Ty);
666f045007fSEli Friedman   }
6677a51313dSChris Lattner 
6687a51313dSChris Lattner   // Expand the binary operator.
6697a51313dSChris Lattner   ComplexPairTy Result = (this->*Func)(OpInfo);
6707a51313dSChris Lattner 
671f045007fSEli Friedman   // Truncate the result and store it into the LHS lvalue.
672f045007fSEli Friedman   if (LHSTy->isAnyComplexType()) {
673f045007fSEli Friedman     ComplexPairTy ResVal = EmitComplexToComplexCast(Result, OpInfo.Ty, LHSTy);
674f045007fSEli Friedman     EmitStoreOfComplex(ResVal, LHS, /*isInit*/ false);
675f045007fSEli Friedman     Val = RValue::getComplex(ResVal);
676f045007fSEli Friedman   } else {
677f045007fSEli Friedman     llvm::Value *ResVal =
678f045007fSEli Friedman         CGF.EmitComplexToScalarConversion(Result, OpInfo.Ty, LHSTy);
679f045007fSEli Friedman     CGF.EmitStoreOfScalar(ResVal, LHS, /*isInit*/ false);
680f045007fSEli Friedman     Val = RValue::get(ResVal);
681f045007fSEli Friedman   }
6828c94ffe7SDaniel Dunbar 
68307bb1966SJohn McCall   return LHS;
6847a51313dSChris Lattner }
6857a51313dSChris Lattner 
68607bb1966SJohn McCall // Compound assignments.
68707bb1966SJohn McCall ComplexPairTy ComplexExprEmitter::
68807bb1966SJohn McCall EmitCompoundAssign(const CompoundAssignOperator *E,
68907bb1966SJohn McCall                    ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&)){
690f045007fSEli Friedman   RValue Val;
69107bb1966SJohn McCall   LValue LV = EmitCompoundAssignLValue(E, Func, Val);
69207bb1966SJohn McCall 
69307bb1966SJohn McCall   // The result of an assignment in C is the assigned r-value.
6949c6890a7SRichard Smith   if (!CGF.getLangOpts().CPlusPlus)
695f045007fSEli Friedman     return Val.getComplexVal();
69607bb1966SJohn McCall 
69707bb1966SJohn McCall   // If the lvalue is non-volatile, return the computed value of the assignment.
69807bb1966SJohn McCall   if (!LV.isVolatileQualified())
699f045007fSEli Friedman     return Val.getComplexVal();
70007bb1966SJohn McCall 
7012d84e842SNick Lewycky   return EmitLoadOfLValue(LV, E->getExprLoc());
70207bb1966SJohn McCall }
70307bb1966SJohn McCall 
70407bb1966SJohn McCall LValue ComplexExprEmitter::EmitBinAssignLValue(const BinaryOperator *E,
70507bb1966SJohn McCall                                                ComplexPairTy &Val) {
706a8a089bfSDouglas Gregor   assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(),
707a8a089bfSDouglas Gregor                                                  E->getRHS()->getType()) &&
7080f398c44SChris Lattner          "Invalid assignment");
70907bb1966SJohn McCall   TestAndClearIgnoreReal();
71007bb1966SJohn McCall   TestAndClearIgnoreImag();
71107bb1966SJohn McCall 
712d0a30016SJohn McCall   // Emit the RHS.  __block variables need the RHS evaluated first.
71307bb1966SJohn McCall   Val = Visit(E->getRHS());
7147a51313dSChris Lattner 
7157a51313dSChris Lattner   // Compute the address to store into.
7167a51313dSChris Lattner   LValue LHS = CGF.EmitLValue(E->getLHS());
7177a51313dSChris Lattner 
7188c94ffe7SDaniel Dunbar   // Store the result value into the LHS lvalue.
71947fb9508SJohn McCall   EmitStoreOfComplex(Val, LHS, /*isInit*/ false);
72076d864c7SDaniel Dunbar 
72107bb1966SJohn McCall   return LHS;
72207bb1966SJohn McCall }
7238c94ffe7SDaniel Dunbar 
72407bb1966SJohn McCall ComplexPairTy ComplexExprEmitter::VisitBinAssign(const BinaryOperator *E) {
72507bb1966SJohn McCall   ComplexPairTy Val;
72607bb1966SJohn McCall   LValue LV = EmitBinAssignLValue(E, Val);
72707bb1966SJohn McCall 
72807bb1966SJohn McCall   // The result of an assignment in C is the assigned r-value.
7299c6890a7SRichard Smith   if (!CGF.getLangOpts().CPlusPlus)
73076d864c7SDaniel Dunbar     return Val;
73107bb1966SJohn McCall 
73207bb1966SJohn McCall   // If the lvalue is non-volatile, return the computed value of the assignment.
73307bb1966SJohn McCall   if (!LV.isVolatileQualified())
73407bb1966SJohn McCall     return Val;
73507bb1966SJohn McCall 
7362d84e842SNick Lewycky   return EmitLoadOfLValue(LV, E->getExprLoc());
73776d864c7SDaniel Dunbar }
73876d864c7SDaniel Dunbar 
7397a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::VisitBinComma(const BinaryOperator *E) {
740a2342eb8SJohn McCall   CGF.EmitIgnoredExpr(E->getLHS());
7417a51313dSChris Lattner   return Visit(E->getRHS());
7427a51313dSChris Lattner }
7437a51313dSChris Lattner 
7447a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::
745c07a0c7eSJohn McCall VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) {
746df0fe27bSMike Stump   TestAndClearIgnoreReal();
747df0fe27bSMike Stump   TestAndClearIgnoreImag();
748a612e79bSDaniel Dunbar   llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true");
749a612e79bSDaniel Dunbar   llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false");
750a612e79bSDaniel Dunbar   llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end");
7517a51313dSChris Lattner 
752c07a0c7eSJohn McCall   // Bind the common expression if necessary.
75348fd89adSEli Friedman   CodeGenFunction::OpaqueValueMapping binding(CGF, E);
754ce1de617SJohn McCall 
755ef512b99SJustin Bogner   RegionCounter Cnt = CGF.getPGORegionCounter(E);
756c07a0c7eSJohn McCall   CodeGenFunction::ConditionalEvaluation eval(CGF);
757ef512b99SJustin Bogner   CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock, Cnt.getCount());
7587a51313dSChris Lattner 
759ce1de617SJohn McCall   eval.begin(CGF);
7607a51313dSChris Lattner   CGF.EmitBlock(LHSBlock);
761ef512b99SJustin Bogner   Cnt.beginRegion(Builder);
7628162d4adSFariborz Jahanian   ComplexPairTy LHS = Visit(E->getTrueExpr());
7637a51313dSChris Lattner   LHSBlock = Builder.GetInsertBlock();
764c56e6764SDaniel Dunbar   CGF.EmitBranch(ContBlock);
765ce1de617SJohn McCall   eval.end(CGF);
7667a51313dSChris Lattner 
767ce1de617SJohn McCall   eval.begin(CGF);
7687a51313dSChris Lattner   CGF.EmitBlock(RHSBlock);
769c07a0c7eSJohn McCall   ComplexPairTy RHS = Visit(E->getFalseExpr());
7707a51313dSChris Lattner   RHSBlock = Builder.GetInsertBlock();
7717a51313dSChris Lattner   CGF.EmitBlock(ContBlock);
772ce1de617SJohn McCall   eval.end(CGF);
7737a51313dSChris Lattner 
7747a51313dSChris Lattner   // Create a PHI node for the real part.
77520c0f02cSJay Foad   llvm::PHINode *RealPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.r");
7767a51313dSChris Lattner   RealPN->addIncoming(LHS.first, LHSBlock);
7777a51313dSChris Lattner   RealPN->addIncoming(RHS.first, RHSBlock);
7787a51313dSChris Lattner 
7797a51313dSChris Lattner   // Create a PHI node for the imaginary part.
78020c0f02cSJay Foad   llvm::PHINode *ImagPN = Builder.CreatePHI(LHS.first->getType(), 2, "cond.i");
7817a51313dSChris Lattner   ImagPN->addIncoming(LHS.second, LHSBlock);
7827a51313dSChris Lattner   ImagPN->addIncoming(RHS.second, RHSBlock);
7837a51313dSChris Lattner 
7847a51313dSChris Lattner   return ComplexPairTy(RealPN, ImagPN);
7857a51313dSChris Lattner }
7867a51313dSChris Lattner 
7877a51313dSChris Lattner ComplexPairTy ComplexExprEmitter::VisitChooseExpr(ChooseExpr *E) {
78875807f23SEli Friedman   return Visit(E->getChosenSubExpr());
7897a51313dSChris Lattner }
7907a51313dSChris Lattner 
791dd7406e6SEli Friedman ComplexPairTy ComplexExprEmitter::VisitInitListExpr(InitListExpr *E) {
792df0fe27bSMike Stump     bool Ignore = TestAndClearIgnoreReal();
793df0fe27bSMike Stump     (void)Ignore;
794df0fe27bSMike Stump     assert (Ignore == false && "init list ignored");
795df0fe27bSMike Stump     Ignore = TestAndClearIgnoreImag();
796df0fe27bSMike Stump     (void)Ignore;
797df0fe27bSMike Stump     assert (Ignore == false && "init list ignored");
7986b9c41eaSEli Friedman 
7996b9c41eaSEli Friedman   if (E->getNumInits() == 2) {
8006b9c41eaSEli Friedman     llvm::Value *Real = CGF.EmitScalarExpr(E->getInit(0));
8016b9c41eaSEli Friedman     llvm::Value *Imag = CGF.EmitScalarExpr(E->getInit(1));
8026b9c41eaSEli Friedman     return ComplexPairTy(Real, Imag);
8036b9c41eaSEli Friedman   } else if (E->getNumInits() == 1) {
804dd7406e6SEli Friedman     return Visit(E->getInit(0));
8056b9c41eaSEli Friedman   }
806dd7406e6SEli Friedman 
807dd7406e6SEli Friedman   // Empty init list intializes to null
8086b9c41eaSEli Friedman   assert(E->getNumInits() == 0 && "Unexpected number of inits");
80947fb9508SJohn McCall   QualType Ty = E->getType()->castAs<ComplexType>()->getElementType();
8102192fe50SChris Lattner   llvm::Type* LTy = CGF.ConvertType(Ty);
8110b75f23bSOwen Anderson   llvm::Value* zeroConstant = llvm::Constant::getNullValue(LTy);
812dd7406e6SEli Friedman   return ComplexPairTy(zeroConstant, zeroConstant);
813dd7406e6SEli Friedman }
814dd7406e6SEli Friedman 
81500079612SDaniel Dunbar ComplexPairTy ComplexExprEmitter::VisitVAArgExpr(VAArgExpr *E) {
816e9fcadd2SDaniel Dunbar   llvm::Value *ArgValue = CGF.EmitVAListRef(E->getSubExpr());
81700079612SDaniel Dunbar   llvm::Value *ArgPtr = CGF.EmitVAArg(ArgValue, E->getType());
81800079612SDaniel Dunbar 
81900079612SDaniel Dunbar   if (!ArgPtr) {
82000079612SDaniel Dunbar     CGF.ErrorUnsupported(E, "complex va_arg expression");
8212192fe50SChris Lattner     llvm::Type *EltTy =
82247fb9508SJohn McCall       CGF.ConvertType(E->getType()->castAs<ComplexType>()->getElementType());
82300079612SDaniel Dunbar     llvm::Value *U = llvm::UndefValue::get(EltTy);
82400079612SDaniel Dunbar     return ComplexPairTy(U, U);
82500079612SDaniel Dunbar   }
82600079612SDaniel Dunbar 
8272d84e842SNick Lewycky   return EmitLoadOfLValue(CGF.MakeNaturalAlignAddrLValue(ArgPtr, E->getType()),
8282d84e842SNick Lewycky                           E->getExprLoc());
82900079612SDaniel Dunbar }
83000079612SDaniel Dunbar 
8317a51313dSChris Lattner //===----------------------------------------------------------------------===//
8327a51313dSChris Lattner //                         Entry Point into this File
8337a51313dSChris Lattner //===----------------------------------------------------------------------===//
8347a51313dSChris Lattner 
8357a51313dSChris Lattner /// EmitComplexExpr - Emit the computation of the specified expression of
8367a51313dSChris Lattner /// complex type, ignoring the result.
837df0fe27bSMike Stump ComplexPairTy CodeGenFunction::EmitComplexExpr(const Expr *E, bool IgnoreReal,
83807bb1966SJohn McCall                                                bool IgnoreImag) {
83947fb9508SJohn McCall   assert(E && getComplexType(E->getType()) &&
8407a51313dSChris Lattner          "Invalid complex expression to emit");
8417a51313dSChris Lattner 
84207bb1966SJohn McCall   return ComplexExprEmitter(*this, IgnoreReal, IgnoreImag)
843df0fe27bSMike Stump     .Visit(const_cast<Expr*>(E));
8447a51313dSChris Lattner }
8457a51313dSChris Lattner 
84647fb9508SJohn McCall void CodeGenFunction::EmitComplexExprIntoLValue(const Expr *E, LValue dest,
84747fb9508SJohn McCall                                                 bool isInit) {
84847fb9508SJohn McCall   assert(E && getComplexType(E->getType()) &&
8497a51313dSChris Lattner          "Invalid complex expression to emit");
8507a51313dSChris Lattner   ComplexExprEmitter Emitter(*this);
8517a51313dSChris Lattner   ComplexPairTy Val = Emitter.Visit(const_cast<Expr*>(E));
85247fb9508SJohn McCall   Emitter.EmitStoreOfComplex(Val, dest, isInit);
8537a51313dSChris Lattner }
8547a51313dSChris Lattner 
85547fb9508SJohn McCall /// EmitStoreOfComplex - Store a complex number into the specified l-value.
85647fb9508SJohn McCall void CodeGenFunction::EmitStoreOfComplex(ComplexPairTy V, LValue dest,
85747fb9508SJohn McCall                                          bool isInit) {
85847fb9508SJohn McCall   ComplexExprEmitter(*this).EmitStoreOfComplex(V, dest, isInit);
8594b8c6db9SDaniel Dunbar }
8604b8c6db9SDaniel Dunbar 
86147fb9508SJohn McCall /// EmitLoadOfComplex - Load a complex number from the specified address.
8622d84e842SNick Lewycky ComplexPairTy CodeGenFunction::EmitLoadOfComplex(LValue src,
8632d84e842SNick Lewycky                                                  SourceLocation loc) {
8642d84e842SNick Lewycky   return ComplexExprEmitter(*this).EmitLoadOfLValue(src, loc);
8657a51313dSChris Lattner }
8664f29b49dSJohn McCall 
8674f29b49dSJohn McCall LValue CodeGenFunction::EmitComplexAssignmentLValue(const BinaryOperator *E) {
868a2342eb8SJohn McCall   assert(E->getOpcode() == BO_Assign);
8694f29b49dSJohn McCall   ComplexPairTy Val; // ignored
8704f29b49dSJohn McCall   return ComplexExprEmitter(*this).EmitBinAssignLValue(E, Val);
871a2342eb8SJohn McCall }
8724f29b49dSJohn McCall 
873f045007fSEli Friedman typedef ComplexPairTy (ComplexExprEmitter::*CompoundFunc)(
874f045007fSEli Friedman     const ComplexExprEmitter::BinOpInfo &);
8754f29b49dSJohn McCall 
876f045007fSEli Friedman static CompoundFunc getComplexOp(BinaryOperatorKind Op) {
877f045007fSEli Friedman   switch (Op) {
878f045007fSEli Friedman   case BO_MulAssign: return &ComplexExprEmitter::EmitBinMul;
879f045007fSEli Friedman   case BO_DivAssign: return &ComplexExprEmitter::EmitBinDiv;
880f045007fSEli Friedman   case BO_SubAssign: return &ComplexExprEmitter::EmitBinSub;
881f045007fSEli Friedman   case BO_AddAssign: return &ComplexExprEmitter::EmitBinAdd;
8824f29b49dSJohn McCall   default:
8834f29b49dSJohn McCall     llvm_unreachable("unexpected complex compound assignment");
8844f29b49dSJohn McCall   }
885f045007fSEli Friedman }
8864f29b49dSJohn McCall 
887f045007fSEli Friedman LValue CodeGenFunction::
888f045007fSEli Friedman EmitComplexCompoundAssignmentLValue(const CompoundAssignOperator *E) {
889f045007fSEli Friedman   CompoundFunc Op = getComplexOp(E->getOpcode());
890f045007fSEli Friedman   RValue Val;
891a2342eb8SJohn McCall   return ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val);
8924f29b49dSJohn McCall }
893f045007fSEli Friedman 
894f045007fSEli Friedman LValue CodeGenFunction::
895f045007fSEli Friedman EmitScalarCompooundAssignWithComplex(const CompoundAssignOperator *E,
896f045007fSEli Friedman                                      llvm::Value *&Result) {
897f045007fSEli Friedman   CompoundFunc Op = getComplexOp(E->getOpcode());
898f045007fSEli Friedman   RValue Val;
899f045007fSEli Friedman   LValue Ret = ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Op, Val);
900f045007fSEli Friedman   Result = Val.getScalarVal();
901f045007fSEli Friedman   return Ret;
902f045007fSEli Friedman }
903