1//===--- CGExprComplex.cpp - Emit LLVM Code for Complex Exprs -------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This contains code to emit Expr nodes with complex types as LLVM code.
10//
11//===----------------------------------------------------------------------===//
12
13#include "CGDebugInfo.h"
14#include "CGOpenMPRuntime.h"
15#include "CodeGenFunction.h"
16#include "CodeGenModule.h"
17#include "ConstantEmitter.h"
18#include "clang/AST/StmtVisitor.h"
19#include "llvm/IR/Constants.h"
20#include "llvm/IR/Instructions.h"
21#include "llvm/IR/MDBuilder.h"
22#include "llvm/IR/Metadata.h"
23using namespace clang;
24using namespace CodeGen;
25
26//===----------------------------------------------------------------------===//
27// Complex Expression Emitter
28//===----------------------------------------------------------------------===//
29
30typedef CodeGenFunction::ComplexPairTy ComplexPairTy;
31
32/// Return the complex type that we are meant to emit.
33static const ComplexType *getComplexType(QualType type) {
34 type = type.getCanonicalType();
35 if (const ComplexType *comp = dyn_cast<ComplexType>(Val&: type)) {
36 return comp;
37 } else {
38 return cast<ComplexType>(Val: cast<AtomicType>(Val&: type)->getValueType());
39 }
40}
41
42namespace {
43class ComplexExprEmitter
44 : public StmtVisitor<ComplexExprEmitter, ComplexPairTy> {
45 CodeGenFunction &CGF;
46 CGBuilderTy &Builder;
47 bool IgnoreReal;
48 bool IgnoreImag;
49 bool FPHasBeenPromoted;
50
51public:
52 ComplexExprEmitter(CodeGenFunction &cgf, bool ir = false, bool ii = false)
53 : CGF(cgf), Builder(CGF.Builder), IgnoreReal(ir), IgnoreImag(ii),
54 FPHasBeenPromoted(false) {}
55
56 //===--------------------------------------------------------------------===//
57 // Utilities
58 //===--------------------------------------------------------------------===//
59
60 bool TestAndClearIgnoreReal() {
61 bool I = IgnoreReal;
62 IgnoreReal = false;
63 return I;
64 }
65 bool TestAndClearIgnoreImag() {
66 bool I = IgnoreImag;
67 IgnoreImag = false;
68 return I;
69 }
70
71 /// EmitLoadOfLValue - Given an expression with complex type that represents a
72 /// value l-value, this method emits the address of the l-value, then loads
73 /// and returns the result.
74 ComplexPairTy EmitLoadOfLValue(const Expr *E) {
75 return EmitLoadOfLValue(LV: CGF.EmitLValue(E), Loc: E->getExprLoc());
76 }
77
78 ComplexPairTy EmitLoadOfLValue(LValue LV, SourceLocation Loc);
79
80 /// EmitStoreOfComplex - Store the specified real/imag parts into the
81 /// specified value pointer.
82 void EmitStoreOfComplex(ComplexPairTy Val, LValue LV, bool isInit);
83
84 /// Emit a cast from complex value Val to DestType.
85 ComplexPairTy EmitComplexToComplexCast(ComplexPairTy Val, QualType SrcType,
86 QualType DestType, SourceLocation Loc);
87 /// Emit a cast from scalar value Val to DestType.
88 ComplexPairTy EmitScalarToComplexCast(llvm::Value *Val, QualType SrcType,
89 QualType DestType, SourceLocation Loc);
90
91 //===--------------------------------------------------------------------===//
92 // Visitor Methods
93 //===--------------------------------------------------------------------===//
94
95 ComplexPairTy Visit(Expr *E) {
96 ApplyDebugLocation DL(CGF, E);
97 return StmtVisitor<ComplexExprEmitter, ComplexPairTy>::Visit(S: E);
98 }
99
100 ComplexPairTy VisitStmt(Stmt *S) {
101 S->dump(OS&: llvm::errs(), Context: CGF.getContext());
102 llvm_unreachable("Stmt can't have complex result type!");
103 }
104 ComplexPairTy VisitExpr(Expr *S);
105 ComplexPairTy VisitConstantExpr(ConstantExpr *E) {
106 if (llvm::Constant *Result = ConstantEmitter(CGF).tryEmitConstantExpr(CE: E))
107 return ComplexPairTy(Result->getAggregateElement(Elt: 0U),
108 Result->getAggregateElement(Elt: 1U));
109 return Visit(E: E->getSubExpr());
110 }
111 ComplexPairTy VisitParenExpr(ParenExpr *PE) {
112 return Visit(E: PE->getSubExpr());
113 }
114 ComplexPairTy VisitGenericSelectionExpr(GenericSelectionExpr *GE) {
115 return Visit(E: GE->getResultExpr());
116 }
117 ComplexPairTy VisitImaginaryLiteral(const ImaginaryLiteral *IL);
118 ComplexPairTy
119 VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *PE) {
120 return Visit(E: PE->getReplacement());
121 }
122 ComplexPairTy VisitCoawaitExpr(CoawaitExpr *S) {
123 return CGF.EmitCoawaitExpr(E: *S).getComplexVal();
124 }
125 ComplexPairTy VisitCoyieldExpr(CoyieldExpr *S) {
126 return CGF.EmitCoyieldExpr(E: *S).getComplexVal();
127 }
128 ComplexPairTy VisitUnaryCoawait(const UnaryOperator *E) {
129 return Visit(E: E->getSubExpr());
130 }
131
132 ComplexPairTy emitConstant(const CodeGenFunction::ConstantEmission &Constant,
133 Expr *E) {
134 assert(Constant && "not a constant");
135 if (Constant.isReference())
136 return EmitLoadOfLValue(LV: Constant.getReferenceLValue(CGF, RefExpr: E),
137 Loc: E->getExprLoc());
138
139 llvm::Constant *pair = Constant.getValue();
140 return ComplexPairTy(pair->getAggregateElement(Elt: 0U),
141 pair->getAggregateElement(Elt: 1U));
142 }
143
144 // l-values.
145 ComplexPairTy VisitDeclRefExpr(DeclRefExpr *E) {
146 if (CodeGenFunction::ConstantEmission Constant = CGF.tryEmitAsConstant(RefExpr: E))
147 return emitConstant(Constant, E);
148 return EmitLoadOfLValue(E);
149 }
150 ComplexPairTy VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
151 return EmitLoadOfLValue(E);
152 }
153 ComplexPairTy VisitObjCMessageExpr(ObjCMessageExpr *E) {
154 return CGF.EmitObjCMessageExpr(E).getComplexVal();
155 }
156 ComplexPairTy VisitArraySubscriptExpr(Expr *E) { return EmitLoadOfLValue(E); }
157 ComplexPairTy VisitMemberExpr(MemberExpr *ME) {
158 if (CodeGenFunction::ConstantEmission Constant =
159 CGF.tryEmitAsConstant(ME)) {
160 CGF.EmitIgnoredExpr(E: ME->getBase());
161 return emitConstant(Constant, E: ME);
162 }
163 return EmitLoadOfLValue(E: ME);
164 }
165 ComplexPairTy VisitOpaqueValueExpr(OpaqueValueExpr *E) {
166 if (E->isGLValue())
167 return EmitLoadOfLValue(LV: CGF.getOrCreateOpaqueLValueMapping(e: E),
168 Loc: E->getExprLoc());
169 return CGF.getOrCreateOpaqueRValueMapping(e: E).getComplexVal();
170 }
171
172 ComplexPairTy VisitPseudoObjectExpr(PseudoObjectExpr *E) {
173 return CGF.EmitPseudoObjectRValue(e: E).getComplexVal();
174 }
175
176 // FIXME: CompoundLiteralExpr
177
178 ComplexPairTy EmitCast(CastKind CK, Expr *Op, QualType DestTy);
179 ComplexPairTy VisitImplicitCastExpr(ImplicitCastExpr *E) {
180 // Unlike for scalars, we don't have to worry about function->ptr demotion
181 // here.
182 if (E->changesVolatileQualification())
183 return EmitLoadOfLValue(E);
184 return EmitCast(CK: E->getCastKind(), Op: E->getSubExpr(), DestTy: E->getType());
185 }
186 ComplexPairTy VisitCastExpr(CastExpr *E) {
187 if (const auto *ECE = dyn_cast<ExplicitCastExpr>(Val: E))
188 CGF.CGM.EmitExplicitCastExprType(E: ECE, CGF: &CGF);
189 if (E->changesVolatileQualification())
190 return EmitLoadOfLValue(E);
191 return EmitCast(CK: E->getCastKind(), Op: E->getSubExpr(), DestTy: E->getType());
192 }
193 ComplexPairTy VisitCallExpr(const CallExpr *E);
194 ComplexPairTy VisitStmtExpr(const StmtExpr *E);
195
196 // Operators.
197 ComplexPairTy VisitPrePostIncDec(const UnaryOperator *E, bool isInc,
198 bool isPre) {
199 LValue LV = CGF.EmitLValue(E: E->getSubExpr());
200 return CGF.EmitComplexPrePostIncDec(E, LV, isInc, isPre);
201 }
202 ComplexPairTy VisitUnaryPostDec(const UnaryOperator *E) {
203 return VisitPrePostIncDec(E, isInc: false, isPre: false);
204 }
205 ComplexPairTy VisitUnaryPostInc(const UnaryOperator *E) {
206 return VisitPrePostIncDec(E, isInc: true, isPre: false);
207 }
208 ComplexPairTy VisitUnaryPreDec(const UnaryOperator *E) {
209 return VisitPrePostIncDec(E, isInc: false, isPre: true);
210 }
211 ComplexPairTy VisitUnaryPreInc(const UnaryOperator *E) {
212 return VisitPrePostIncDec(E, isInc: true, isPre: true);
213 }
214 ComplexPairTy VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); }
215
216 ComplexPairTy VisitUnaryPlus(const UnaryOperator *E,
217 QualType PromotionType = QualType());
218 ComplexPairTy VisitPlus(const UnaryOperator *E, QualType PromotionType);
219 ComplexPairTy VisitUnaryMinus(const UnaryOperator *E,
220 QualType PromotionType = QualType());
221 ComplexPairTy VisitMinus(const UnaryOperator *E, QualType PromotionType);
222 ComplexPairTy VisitUnaryNot(const UnaryOperator *E);
223 // LNot,Real,Imag never return complex.
224 ComplexPairTy VisitUnaryExtension(const UnaryOperator *E) {
225 return Visit(E: E->getSubExpr());
226 }
227 ComplexPairTy VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
228 CodeGenFunction::CXXDefaultArgExprScope Scope(CGF, DAE);
229 return Visit(E: DAE->getExpr());
230 }
231 ComplexPairTy VisitCXXDefaultInitExpr(CXXDefaultInitExpr *DIE) {
232 CodeGenFunction::CXXDefaultInitExprScope Scope(CGF, DIE);
233 return Visit(E: DIE->getExpr());
234 }
235 ComplexPairTy VisitExprWithCleanups(ExprWithCleanups *E) {
236 CodeGenFunction::RunCleanupsScope Scope(CGF);
237 ComplexPairTy Vals = Visit(E: E->getSubExpr());
238 // Defend against dominance problems caused by jumps out of expression
239 // evaluation through the shared cleanup block.
240 Scope.ForceCleanup(ValuesToReload: {&Vals.first, &Vals.second});
241 return Vals;
242 }
243 ComplexPairTy VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
244 assert(E->getType()->isAnyComplexType() && "Expected complex type!");
245 QualType Elem = E->getType()->castAs<ComplexType>()->getElementType();
246 llvm::Constant *Null = llvm::Constant::getNullValue(Ty: CGF.ConvertType(T: Elem));
247 return ComplexPairTy(Null, Null);
248 }
249 ComplexPairTy VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
250 assert(E->getType()->isAnyComplexType() && "Expected complex type!");
251 QualType Elem = E->getType()->castAs<ComplexType>()->getElementType();
252 llvm::Constant *Null = llvm::Constant::getNullValue(Ty: CGF.ConvertType(T: Elem));
253 return ComplexPairTy(Null, Null);
254 }
255
256 struct BinOpInfo {
257 ComplexPairTy LHS;
258 ComplexPairTy RHS;
259 QualType Ty; // Computation Type.
260 FPOptions FPFeatures;
261 };
262
263 BinOpInfo EmitBinOps(const BinaryOperator *E,
264 QualType PromotionTy = QualType());
265 ComplexPairTy EmitPromoted(const Expr *E, QualType PromotionTy);
266 ComplexPairTy EmitPromotedComplexOperand(const Expr *E, QualType PromotionTy);
267 LValue EmitCompoundAssignLValue(
268 const CompoundAssignOperator *E,
269 ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo &),
270 RValue &Val);
271 ComplexPairTy EmitCompoundAssign(
272 const CompoundAssignOperator *E,
273 ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo &));
274
275 ComplexPairTy EmitBinAdd(const BinOpInfo &Op);
276 ComplexPairTy EmitBinSub(const BinOpInfo &Op);
277 ComplexPairTy EmitBinMul(const BinOpInfo &Op);
278 ComplexPairTy EmitBinDiv(const BinOpInfo &Op);
279 ComplexPairTy EmitAlgebraicDiv(llvm::Value *A, llvm::Value *B, llvm::Value *C,
280 llvm::Value *D);
281 ComplexPairTy EmitRangeReductionDiv(llvm::Value *A, llvm::Value *B,
282 llvm::Value *C, llvm::Value *D);
283
284 ComplexPairTy EmitComplexBinOpLibCall(StringRef LibCallName,
285 const BinOpInfo &Op);
286
287 QualType HigherPrecisionTypeForComplexArithmetic(QualType ElementType) {
288 ASTContext &Ctx = CGF.getContext();
289 const QualType HigherElementType =
290 Ctx.GetHigherPrecisionFPType(ElementType);
291 const llvm::fltSemantics &ElementTypeSemantics =
292 Ctx.getFloatTypeSemantics(T: ElementType);
293 const llvm::fltSemantics &HigherElementTypeSemantics =
294 Ctx.getFloatTypeSemantics(T: HigherElementType);
295 // Check that the promoted type can handle the intermediate values without
296 // overflowing. This can be interpreted as:
297 // (SmallerType.LargestFiniteVal * SmallerType.LargestFiniteVal) * 2 <=
298 // LargerType.LargestFiniteVal.
299 // In terms of exponent it gives this formula:
300 // (SmallerType.LargestFiniteVal * SmallerType.LargestFiniteVal
301 // doubles the exponent of SmallerType.LargestFiniteVal)
302 if (llvm::APFloat::semanticsMaxExponent(ElementTypeSemantics) * 2 + 1 <=
303 llvm::APFloat::semanticsMaxExponent(HigherElementTypeSemantics)) {
304 if (!Ctx.getTargetInfo().hasLongDoubleType() &&
305 HigherElementType.getCanonicalType().getUnqualifiedType() ==
306 Ctx.LongDoubleTy)
307 return QualType();
308 FPHasBeenPromoted = true;
309 return Ctx.getComplexType(T: HigherElementType);
310 } else {
311 // The intermediate values can't be represented in the promoted type
312 // without overflowing.
313 return QualType();
314 }
315 }
316
317 QualType getPromotionType(FPOptionsOverride Features, QualType Ty,
318 bool IsComplexDivisor) {
319 if (auto *CT = Ty->getAs<ComplexType>()) {
320 QualType ElementType = CT->getElementType().getCanonicalType();
321 bool IsFloatingType = ElementType->isFloatingType();
322 bool IsComplexRangePromoted = CGF.getLangOpts().getComplexRange() ==
323 LangOptions::ComplexRangeKind::CX_Promoted;
324 bool HasNoComplexRangeOverride = !Features.hasComplexRangeOverride();
325 bool HasMatchingComplexRange = Features.hasComplexRangeOverride() &&
326 Features.getComplexRangeOverride() ==
327 CGF.getLangOpts().getComplexRange();
328
329 if (IsComplexDivisor && IsFloatingType && IsComplexRangePromoted &&
330 (HasNoComplexRangeOverride || HasMatchingComplexRange))
331 return HigherPrecisionTypeForComplexArithmetic(ElementType);
332 if (ElementType.UseExcessPrecision(Ctx: CGF.getContext()))
333 return CGF.getContext().getComplexType(T: CGF.getContext().FloatTy);
334 }
335 if (Ty.UseExcessPrecision(Ctx: CGF.getContext()))
336 return CGF.getContext().FloatTy;
337 return QualType();
338 }
339
340#define HANDLEBINOP(OP) \
341 ComplexPairTy VisitBin##OP(const BinaryOperator *E) { \
342 QualType promotionTy = \
343 getPromotionType(E->getStoredFPFeaturesOrDefault(), E->getType(), \
344 (E->getOpcode() == BinaryOperatorKind::BO_Div && \
345 E->getRHS()->getType()->isAnyComplexType())); \
346 ComplexPairTy result = EmitBin##OP(EmitBinOps(E, promotionTy)); \
347 if (!promotionTy.isNull()) \
348 result = CGF.EmitUnPromotedValue(result, E->getType()); \
349 return result; \
350 }
351
352 HANDLEBINOP(Mul)
353 HANDLEBINOP(Div)
354 HANDLEBINOP(Add)
355 HANDLEBINOP(Sub)
356#undef HANDLEBINOP
357
358 ComplexPairTy VisitCXXRewrittenBinaryOperator(CXXRewrittenBinaryOperator *E) {
359 return Visit(E: E->getSemanticForm());
360 }
361
362 // Compound assignments.
363 ComplexPairTy VisitBinAddAssign(const CompoundAssignOperator *E) {
364 ApplyAtomGroup Grp(CGF.getDebugInfo());
365 return EmitCompoundAssign(E, Func: &ComplexExprEmitter::EmitBinAdd);
366 }
367 ComplexPairTy VisitBinSubAssign(const CompoundAssignOperator *E) {
368 ApplyAtomGroup Grp(CGF.getDebugInfo());
369 return EmitCompoundAssign(E, Func: &ComplexExprEmitter::EmitBinSub);
370 }
371 ComplexPairTy VisitBinMulAssign(const CompoundAssignOperator *E) {
372 ApplyAtomGroup Grp(CGF.getDebugInfo());
373 return EmitCompoundAssign(E, Func: &ComplexExprEmitter::EmitBinMul);
374 }
375 ComplexPairTy VisitBinDivAssign(const CompoundAssignOperator *E) {
376 ApplyAtomGroup Grp(CGF.getDebugInfo());
377 return EmitCompoundAssign(E, Func: &ComplexExprEmitter::EmitBinDiv);
378 }
379
380 // GCC rejects rem/and/or/xor for integer complex.
381 // Logical and/or always return int, never complex.
382
383 // No comparisons produce a complex result.
384
385 LValue EmitBinAssignLValue(const BinaryOperator *E, ComplexPairTy &Val);
386 ComplexPairTy VisitBinAssign(const BinaryOperator *E);
387 ComplexPairTy VisitBinComma(const BinaryOperator *E);
388
389 ComplexPairTy
390 VisitAbstractConditionalOperator(const AbstractConditionalOperator *CO);
391 ComplexPairTy VisitChooseExpr(ChooseExpr *CE);
392
393 ComplexPairTy VisitInitListExpr(InitListExpr *E);
394
395 ComplexPairTy VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
396 return EmitLoadOfLValue(E);
397 }
398
399 ComplexPairTy VisitVAArgExpr(VAArgExpr *E);
400
401 ComplexPairTy VisitAtomicExpr(AtomicExpr *E) {
402 return CGF.EmitAtomicExpr(E).getComplexVal();
403 }
404
405 ComplexPairTy VisitPackIndexingExpr(PackIndexingExpr *E) {
406 return Visit(E: E->getSelectedExpr());
407 }
408};
409} // end anonymous namespace.
410
411//===----------------------------------------------------------------------===//
412// Utilities
413//===----------------------------------------------------------------------===//
414
415Address CodeGenFunction::emitAddrOfRealComponent(Address addr,
416 QualType complexType) {
417 return Builder.CreateStructGEP(Addr: addr, Index: 0, Name: addr.getName() + ".realp");
418}
419
420Address CodeGenFunction::emitAddrOfImagComponent(Address addr,
421 QualType complexType) {
422 return Builder.CreateStructGEP(Addr: addr, Index: 1, Name: addr.getName() + ".imagp");
423}
424
425/// EmitLoadOfLValue - Given an RValue reference for a complex, emit code to
426/// load the real and imaginary pieces, returning them as Real/Imag.
427ComplexPairTy ComplexExprEmitter::EmitLoadOfLValue(LValue lvalue,
428 SourceLocation loc) {
429 assert(lvalue.isSimple() && "non-simple complex l-value?");
430 if (lvalue.getType()->isAtomicType())
431 return CGF.EmitAtomicLoad(LV: lvalue, SL: loc).getComplexVal();
432
433 Address SrcPtr = lvalue.getAddress();
434 bool isVolatile = lvalue.isVolatileQualified();
435
436 llvm::Value *Real = nullptr, *Imag = nullptr;
437
438 if (!IgnoreReal || isVolatile) {
439 Address RealP = CGF.emitAddrOfRealComponent(addr: SrcPtr, complexType: lvalue.getType());
440 Real = Builder.CreateLoad(Addr: RealP, IsVolatile: isVolatile, Name: SrcPtr.getName() + ".real");
441 }
442
443 if (!IgnoreImag || isVolatile) {
444 Address ImagP = CGF.emitAddrOfImagComponent(addr: SrcPtr, complexType: lvalue.getType());
445 Imag = Builder.CreateLoad(Addr: ImagP, IsVolatile: isVolatile, Name: SrcPtr.getName() + ".imag");
446 }
447
448 return ComplexPairTy(Real, Imag);
449}
450
451/// EmitStoreOfComplex - Store the specified real/imag parts into the
452/// specified value pointer.
453void ComplexExprEmitter::EmitStoreOfComplex(ComplexPairTy Val, LValue lvalue,
454 bool isInit) {
455 if (lvalue.getType()->isAtomicType() ||
456 (!isInit && CGF.LValueIsSuitableForInlineAtomic(Src: lvalue)))
457 return CGF.EmitAtomicStore(rvalue: RValue::getComplex(C: Val), lvalue, isInit);
458
459 Address Ptr = lvalue.getAddress();
460 Address RealPtr = CGF.emitAddrOfRealComponent(addr: Ptr, complexType: lvalue.getType());
461 Address ImagPtr = CGF.emitAddrOfImagComponent(addr: Ptr, complexType: lvalue.getType());
462
463 auto *R =
464 Builder.CreateStore(Val: Val.first, Addr: RealPtr, IsVolatile: lvalue.isVolatileQualified());
465 CGF.addInstToCurrentSourceAtom(KeyInstruction: R, Backup: Val.first);
466 auto *I =
467 Builder.CreateStore(Val: Val.second, Addr: ImagPtr, IsVolatile: lvalue.isVolatileQualified());
468 CGF.addInstToCurrentSourceAtom(KeyInstruction: I, Backup: Val.second);
469}
470
471//===----------------------------------------------------------------------===//
472// Visitor Methods
473//===----------------------------------------------------------------------===//
474
475ComplexPairTy ComplexExprEmitter::VisitExpr(Expr *E) {
476 CGF.ErrorUnsupported(S: E, Type: "complex expression");
477 llvm::Type *EltTy =
478 CGF.ConvertType(T: getComplexType(type: E->getType())->getElementType());
479 llvm::Value *U = llvm::PoisonValue::get(T: EltTy);
480 return ComplexPairTy(U, U);
481}
482
483ComplexPairTy
484ComplexExprEmitter::VisitImaginaryLiteral(const ImaginaryLiteral *IL) {
485 llvm::Value *Imag = CGF.EmitScalarExpr(E: IL->getSubExpr());
486 return ComplexPairTy(llvm::Constant::getNullValue(Ty: Imag->getType()), Imag);
487}
488
489ComplexPairTy ComplexExprEmitter::VisitCallExpr(const CallExpr *E) {
490 if (E->getCallReturnType(Ctx: CGF.getContext())->isReferenceType())
491 return EmitLoadOfLValue(E);
492
493 return CGF.EmitCallExpr(E).getComplexVal();
494}
495
496ComplexPairTy ComplexExprEmitter::VisitStmtExpr(const StmtExpr *E) {
497 CodeGenFunction::StmtExprEvaluation eval(CGF);
498 Address RetAlloca = CGF.EmitCompoundStmt(S: *E->getSubStmt(), GetLast: true);
499 assert(RetAlloca.isValid() && "Expected complex return value");
500 return EmitLoadOfLValue(lvalue: CGF.MakeAddrLValue(Addr: RetAlloca, T: E->getType()),
501 loc: E->getExprLoc());
502}
503
504/// Emit a cast from complex value Val to DestType.
505ComplexPairTy ComplexExprEmitter::EmitComplexToComplexCast(ComplexPairTy Val,
506 QualType SrcType,
507 QualType DestType,
508 SourceLocation Loc) {
509 // Get the src/dest element type.
510 SrcType = SrcType.getAtomicUnqualifiedType()
511 ->castAs<ComplexType>()
512 ->getElementType();
513 DestType = DestType.getAtomicUnqualifiedType()
514 ->castAs<ComplexType>()
515 ->getElementType();
516
517 // C99 6.3.1.6: When a value of complex type is converted to another
518 // complex type, both the real and imaginary parts follow the conversion
519 // rules for the corresponding real types.
520 if (Val.first)
521 Val.first = CGF.EmitScalarConversion(Src: Val.first, SrcTy: SrcType, DstTy: DestType, Loc);
522 if (Val.second)
523 Val.second = CGF.EmitScalarConversion(Src: Val.second, SrcTy: SrcType, DstTy: DestType, Loc);
524 return Val;
525}
526
527ComplexPairTy ComplexExprEmitter::EmitScalarToComplexCast(llvm::Value *Val,
528 QualType SrcType,
529 QualType DestType,
530 SourceLocation Loc) {
531 // Convert the input element to the element type of the complex.
532 DestType = DestType->castAs<ComplexType>()->getElementType();
533 Val = CGF.EmitScalarConversion(Src: Val, SrcTy: SrcType, DstTy: DestType, Loc);
534
535 // Return (realval, 0).
536 return ComplexPairTy(Val, llvm::Constant::getNullValue(Ty: Val->getType()));
537}
538
539ComplexPairTy ComplexExprEmitter::EmitCast(CastKind CK, Expr *Op,
540 QualType DestTy) {
541 DestTy = DestTy.getAtomicUnqualifiedType();
542 switch (CK) {
543 case CK_Dependent:
544 llvm_unreachable("dependent cast kind in IR gen!");
545
546 // Atomic to non-atomic casts may be more than a no-op for some platforms and
547 // for some types.
548 case CK_AtomicToNonAtomic:
549 case CK_NonAtomicToAtomic:
550 case CK_NoOp:
551 case CK_LValueToRValue:
552 case CK_UserDefinedConversion:
553 return Visit(E: Op);
554
555 case CK_LValueBitCast: {
556 LValue origLV = CGF.EmitLValue(E: Op);
557 Address V = origLV.getAddress().withElementType(ElemTy: CGF.ConvertType(T: DestTy));
558 return EmitLoadOfLValue(lvalue: CGF.MakeAddrLValue(Addr: V, T: DestTy), loc: Op->getExprLoc());
559 }
560
561 case CK_LValueToRValueBitCast: {
562 LValue SourceLVal = CGF.EmitLValue(E: Op);
563 Address Addr =
564 SourceLVal.getAddress().withElementType(ElemTy: CGF.ConvertTypeForMem(T: DestTy));
565 LValue DestLV = CGF.MakeAddrLValue(Addr, T: DestTy);
566 DestLV.setTBAAInfo(TBAAAccessInfo::getMayAliasInfo());
567 return EmitLoadOfLValue(lvalue: DestLV, loc: Op->getExprLoc());
568 }
569
570 case CK_BitCast:
571 case CK_BaseToDerived:
572 case CK_DerivedToBase:
573 case CK_UncheckedDerivedToBase:
574 case CK_Dynamic:
575 case CK_ToUnion:
576 case CK_ArrayToPointerDecay:
577 case CK_FunctionToPointerDecay:
578 case CK_NullToPointer:
579 case CK_NullToMemberPointer:
580 case CK_BaseToDerivedMemberPointer:
581 case CK_DerivedToBaseMemberPointer:
582 case CK_MemberPointerToBoolean:
583 case CK_ReinterpretMemberPointer:
584 case CK_ConstructorConversion:
585 case CK_IntegralToPointer:
586 case CK_PointerToIntegral:
587 case CK_PointerToBoolean:
588 case CK_ToVoid:
589 case CK_VectorSplat:
590 case CK_IntegralCast:
591 case CK_BooleanToSignedIntegral:
592 case CK_IntegralToBoolean:
593 case CK_IntegralToFloating:
594 case CK_FloatingToIntegral:
595 case CK_FloatingToBoolean:
596 case CK_FloatingCast:
597 case CK_CPointerToObjCPointerCast:
598 case CK_BlockPointerToObjCPointerCast:
599 case CK_AnyPointerToBlockPointerCast:
600 case CK_ObjCObjectLValueCast:
601 case CK_FloatingComplexToReal:
602 case CK_FloatingComplexToBoolean:
603 case CK_IntegralComplexToReal:
604 case CK_IntegralComplexToBoolean:
605 case CK_ARCProduceObject:
606 case CK_ARCConsumeObject:
607 case CK_ARCReclaimReturnedObject:
608 case CK_ARCExtendBlockObject:
609 case CK_CopyAndAutoreleaseBlockObject:
610 case CK_BuiltinFnToFnPtr:
611 case CK_ZeroToOCLOpaqueType:
612 case CK_AddressSpaceConversion:
613 case CK_IntToOCLSampler:
614 case CK_FloatingToFixedPoint:
615 case CK_FixedPointToFloating:
616 case CK_FixedPointCast:
617 case CK_FixedPointToBoolean:
618 case CK_FixedPointToIntegral:
619 case CK_IntegralToFixedPoint:
620 case CK_MatrixCast:
621 case CK_HLSLVectorTruncation:
622 case CK_HLSLMatrixTruncation:
623 case CK_HLSLArrayRValue:
624 case CK_HLSLElementwiseCast:
625 case CK_HLSLAggregateSplatCast:
626 llvm_unreachable("invalid cast kind for complex value");
627
628 case CK_FloatingRealToComplex:
629 case CK_IntegralRealToComplex: {
630 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, Op);
631 return EmitScalarToComplexCast(Val: CGF.EmitScalarExpr(E: Op), SrcType: Op->getType(),
632 DestType: DestTy, Loc: Op->getExprLoc());
633 }
634
635 case CK_FloatingComplexCast:
636 case CK_FloatingComplexToIntegralComplex:
637 case CK_IntegralComplexCast:
638 case CK_IntegralComplexToFloatingComplex: {
639 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, Op);
640 return EmitComplexToComplexCast(Val: Visit(E: Op), SrcType: Op->getType(), DestType: DestTy,
641 Loc: Op->getExprLoc());
642 }
643 }
644
645 llvm_unreachable("unknown cast resulting in complex value");
646}
647
648ComplexPairTy ComplexExprEmitter::VisitUnaryPlus(const UnaryOperator *E,
649 QualType PromotionType) {
650 QualType promotionTy =
651 PromotionType.isNull()
652 ? getPromotionType(Features: E->getStoredFPFeaturesOrDefault(),
653 Ty: E->getSubExpr()->getType(),
654 /*IsComplexDivisor=*/false)
655 : PromotionType;
656 ComplexPairTy result = VisitPlus(E, PromotionType: promotionTy);
657 if (!promotionTy.isNull())
658 return CGF.EmitUnPromotedValue(result, PromotionType: E->getSubExpr()->getType());
659 return result;
660}
661
662ComplexPairTy ComplexExprEmitter::VisitPlus(const UnaryOperator *E,
663 QualType PromotionType) {
664 TestAndClearIgnoreReal();
665 TestAndClearIgnoreImag();
666 if (!PromotionType.isNull())
667 return CGF.EmitPromotedComplexExpr(E: E->getSubExpr(), PromotionType);
668 return Visit(E: E->getSubExpr());
669}
670
671ComplexPairTy ComplexExprEmitter::VisitUnaryMinus(const UnaryOperator *E,
672 QualType PromotionType) {
673 QualType promotionTy =
674 PromotionType.isNull()
675 ? getPromotionType(Features: E->getStoredFPFeaturesOrDefault(),
676 Ty: E->getSubExpr()->getType(),
677 /*IsComplexDivisor=*/false)
678 : PromotionType;
679 ComplexPairTy result = VisitMinus(E, PromotionType: promotionTy);
680 if (!promotionTy.isNull())
681 return CGF.EmitUnPromotedValue(result, PromotionType: E->getSubExpr()->getType());
682 return result;
683}
684ComplexPairTy ComplexExprEmitter::VisitMinus(const UnaryOperator *E,
685 QualType PromotionType) {
686 TestAndClearIgnoreReal();
687 TestAndClearIgnoreImag();
688 ComplexPairTy Op;
689 if (!PromotionType.isNull())
690 Op = CGF.EmitPromotedComplexExpr(E: E->getSubExpr(), PromotionType);
691 else
692 Op = Visit(E: E->getSubExpr());
693
694 llvm::Value *ResR, *ResI;
695 if (Op.first->getType()->isFloatingPointTy()) {
696 ResR = Builder.CreateFNeg(V: Op.first, Name: "neg.r");
697 ResI = Builder.CreateFNeg(V: Op.second, Name: "neg.i");
698 } else {
699 ResR = Builder.CreateNeg(V: Op.first, Name: "neg.r");
700 ResI = Builder.CreateNeg(V: Op.second, Name: "neg.i");
701 }
702 return ComplexPairTy(ResR, ResI);
703}
704
705ComplexPairTy ComplexExprEmitter::VisitUnaryNot(const UnaryOperator *E) {
706 TestAndClearIgnoreReal();
707 TestAndClearIgnoreImag();
708 // ~(a+ib) = a + i*-b
709 ComplexPairTy Op = Visit(E: E->getSubExpr());
710 llvm::Value *ResI;
711 if (Op.second->getType()->isFloatingPointTy())
712 ResI = Builder.CreateFNeg(V: Op.second, Name: "conj.i");
713 else
714 ResI = Builder.CreateNeg(V: Op.second, Name: "conj.i");
715
716 return ComplexPairTy(Op.first, ResI);
717}
718
719ComplexPairTy ComplexExprEmitter::EmitBinAdd(const BinOpInfo &Op) {
720 llvm::Value *ResR, *ResI;
721
722 if (Op.LHS.first->getType()->isFloatingPointTy()) {
723 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, Op.FPFeatures);
724 ResR = Builder.CreateFAdd(L: Op.LHS.first, R: Op.RHS.first, Name: "add.r");
725 if (Op.LHS.second && Op.RHS.second)
726 ResI = Builder.CreateFAdd(L: Op.LHS.second, R: Op.RHS.second, Name: "add.i");
727 else
728 ResI = Op.LHS.second ? Op.LHS.second : Op.RHS.second;
729 assert(ResI && "Only one operand may be real!");
730 } else {
731 ResR = Builder.CreateAdd(LHS: Op.LHS.first, RHS: Op.RHS.first, Name: "add.r");
732 assert(Op.LHS.second && Op.RHS.second &&
733 "Both operands of integer complex operators must be complex!");
734 ResI = Builder.CreateAdd(LHS: Op.LHS.second, RHS: Op.RHS.second, Name: "add.i");
735 }
736 return ComplexPairTy(ResR, ResI);
737}
738
739ComplexPairTy ComplexExprEmitter::EmitBinSub(const BinOpInfo &Op) {
740 llvm::Value *ResR, *ResI;
741 if (Op.LHS.first->getType()->isFloatingPointTy()) {
742 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, Op.FPFeatures);
743 ResR = Builder.CreateFSub(L: Op.LHS.first, R: Op.RHS.first, Name: "sub.r");
744 if (Op.LHS.second && Op.RHS.second)
745 ResI = Builder.CreateFSub(L: Op.LHS.second, R: Op.RHS.second, Name: "sub.i");
746 else
747 ResI = Op.LHS.second ? Op.LHS.second
748 : Builder.CreateFNeg(V: Op.RHS.second, Name: "sub.i");
749 assert(ResI && "Only one operand may be real!");
750 } else {
751 ResR = Builder.CreateSub(LHS: Op.LHS.first, RHS: Op.RHS.first, Name: "sub.r");
752 assert(Op.LHS.second && Op.RHS.second &&
753 "Both operands of integer complex operators must be complex!");
754 ResI = Builder.CreateSub(LHS: Op.LHS.second, RHS: Op.RHS.second, Name: "sub.i");
755 }
756 return ComplexPairTy(ResR, ResI);
757}
758
759/// Emit a libcall for a binary operation on complex types.
760ComplexPairTy ComplexExprEmitter::EmitComplexBinOpLibCall(StringRef LibCallName,
761 const BinOpInfo &Op) {
762 CallArgList Args;
763 Args.add(rvalue: RValue::get(V: Op.LHS.first),
764 type: Op.Ty->castAs<ComplexType>()->getElementType());
765 Args.add(rvalue: RValue::get(V: Op.LHS.second),
766 type: Op.Ty->castAs<ComplexType>()->getElementType());
767 Args.add(rvalue: RValue::get(V: Op.RHS.first),
768 type: Op.Ty->castAs<ComplexType>()->getElementType());
769 Args.add(rvalue: RValue::get(V: Op.RHS.second),
770 type: Op.Ty->castAs<ComplexType>()->getElementType());
771
772 // We *must* use the full CG function call building logic here because the
773 // complex type has special ABI handling. We also should not forget about
774 // special calling convention which may be used for compiler builtins.
775
776 // We create a function qualified type to state that this call does not have
777 // any exceptions.
778 FunctionProtoType::ExtProtoInfo EPI;
779 EPI = EPI.withExceptionSpec(
780 ESI: FunctionProtoType::ExceptionSpecInfo(EST_BasicNoexcept));
781 SmallVector<QualType, 4> ArgsQTys(
782 4, Op.Ty->castAs<ComplexType>()->getElementType());
783 QualType FQTy = CGF.getContext().getFunctionType(ResultTy: Op.Ty, Args: ArgsQTys, EPI);
784 const CGFunctionInfo &FuncInfo = CGF.CGM.getTypes().arrangeFreeFunctionCall(
785 Args, Ty: cast<FunctionType>(Val: FQTy.getTypePtr()), ChainCall: false,
786 ABIInfoFD: CGF.getCurrentFunctionDecl());
787
788 llvm::FunctionType *FTy = CGF.CGM.getTypes().GetFunctionType(Info: FuncInfo);
789 llvm::FunctionCallee Func = CGF.CGM.CreateRuntimeFunction(
790 Ty: FTy, Name: LibCallName, ExtraAttrs: llvm::AttributeList(), Local: true);
791 CGCallee Callee = CGCallee::forDirect(functionPtr: Func, abstractInfo: FQTy->getAs<FunctionProtoType>());
792
793 llvm::CallBase *Call;
794 RValue Res = CGF.EmitCall(CallInfo: FuncInfo, Callee, ReturnValue: ReturnValueSlot(), Args, CallOrInvoke: &Call);
795 Call->setCallingConv(CGF.CGM.getRuntimeCC());
796 return Res.getComplexVal();
797}
798
799/// Lookup the libcall name for a given floating point type complex
800/// multiply.
801static StringRef getComplexMultiplyLibCallName(const llvm::Triple &T,
802 llvm::Type *Ty) {
803 switch (Ty->getTypeID()) {
804 default:
805 llvm_unreachable("Unsupported floating point type!");
806 case llvm::Type::HalfTyID:
807 return "__mulhc3";
808 case llvm::Type::FloatTyID:
809 return "__mulsc3";
810 case llvm::Type::DoubleTyID:
811 return "__muldc3";
812 case llvm::Type::PPC_FP128TyID:
813 return "__multc3";
814 case llvm::Type::X86_FP80TyID:
815 return "__mulxc3";
816 case llvm::Type::FP128TyID:
817 return T.isPPC() ? "__mulkc3" : "__multc3";
818 }
819}
820
821// See C11 Annex G.5.1 for the semantics of multiplicative operators on complex
822// typed values.
823ComplexPairTy ComplexExprEmitter::EmitBinMul(const BinOpInfo &Op) {
824 using llvm::Value;
825 Value *ResR, *ResI;
826 llvm::MDBuilder MDHelper(CGF.getLLVMContext());
827
828 if (Op.LHS.first->getType()->isFloatingPointTy()) {
829 // The general formulation is:
830 // (a + ib) * (c + id) = (a * c - b * d) + i(a * d + b * c)
831 //
832 // But we can fold away components which would be zero due to a real
833 // operand according to C11 Annex G.5.1p2.
834
835 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, Op.FPFeatures);
836 if (Op.LHS.second && Op.RHS.second) {
837 // If both operands are complex, emit the core math directly, and then
838 // test for NaNs. If we find NaNs in the result, we delegate to a libcall
839 // to carefully re-compute the correct infinity representation if
840 // possible. The expectation is that the presence of NaNs here is
841 // *extremely* rare, and so the cost of the libcall is almost irrelevant.
842 // This is good, because the libcall re-computes the core multiplication
843 // exactly the same as we do here and re-tests for NaNs in order to be
844 // a generic complex*complex libcall.
845
846 // First compute the four products.
847 Value *AC = Builder.CreateFMul(L: Op.LHS.first, R: Op.RHS.first, Name: "mul_ac");
848 Value *BD = Builder.CreateFMul(L: Op.LHS.second, R: Op.RHS.second, Name: "mul_bd");
849 Value *AD = Builder.CreateFMul(L: Op.LHS.first, R: Op.RHS.second, Name: "mul_ad");
850 Value *BC = Builder.CreateFMul(L: Op.LHS.second, R: Op.RHS.first, Name: "mul_bc");
851
852 // The real part is the difference of the first two, the imaginary part is
853 // the sum of the second.
854 ResR = Builder.CreateFSub(L: AC, R: BD, Name: "mul_r");
855 ResI = Builder.CreateFAdd(L: AD, R: BC, Name: "mul_i");
856
857 if (Op.FPFeatures.getComplexRange() == LangOptions::CX_Basic ||
858 Op.FPFeatures.getComplexRange() == LangOptions::CX_Improved ||
859 Op.FPFeatures.getComplexRange() == LangOptions::CX_Promoted)
860 return ComplexPairTy(ResR, ResI);
861
862 // Emit the test for the real part becoming NaN and create a branch to
863 // handle it. We test for NaN by comparing the number to itself.
864 Value *IsRNaN = Builder.CreateFCmpUNO(LHS: ResR, RHS: ResR, Name: "isnan_cmp");
865 llvm::BasicBlock *ContBB = CGF.createBasicBlock(name: "complex_mul_cont");
866 llvm::BasicBlock *INaNBB = CGF.createBasicBlock(name: "complex_mul_imag_nan");
867 llvm::Instruction *Branch = Builder.CreateCondBr(Cond: IsRNaN, True: INaNBB, False: ContBB);
868 llvm::BasicBlock *OrigBB = Branch->getParent();
869
870 // Give hint that we very much don't expect to see NaNs.
871 llvm::MDNode *BrWeight = MDHelper.createUnlikelyBranchWeights();
872 Branch->setMetadata(KindID: llvm::LLVMContext::MD_prof, Node: BrWeight);
873
874 // Now test the imaginary part and create its branch.
875 CGF.EmitBlock(BB: INaNBB);
876 Value *IsINaN = Builder.CreateFCmpUNO(LHS: ResI, RHS: ResI, Name: "isnan_cmp");
877 llvm::BasicBlock *LibCallBB = CGF.createBasicBlock(name: "complex_mul_libcall");
878 Branch = Builder.CreateCondBr(Cond: IsINaN, True: LibCallBB, False: ContBB);
879 Branch->setMetadata(KindID: llvm::LLVMContext::MD_prof, Node: BrWeight);
880
881 // Now emit the libcall on this slowest of the slow paths.
882 CGF.EmitBlock(BB: LibCallBB);
883 Value *LibCallR, *LibCallI;
884 llvm::Triple Triple = CGF.getTarget().getTriple();
885 std::tie(args&: LibCallR, args&: LibCallI) = EmitComplexBinOpLibCall(
886 LibCallName: getComplexMultiplyLibCallName(T: Triple, Ty: Op.LHS.first->getType()), Op);
887 Builder.CreateBr(Dest: ContBB);
888
889 // Finally continue execution by phi-ing together the different
890 // computation paths.
891 CGF.EmitBlock(BB: ContBB);
892 llvm::PHINode *RealPHI =
893 Builder.CreatePHI(Ty: ResR->getType(), NumReservedValues: 3, Name: "real_mul_phi");
894 RealPHI->addIncoming(V: ResR, BB: OrigBB);
895 RealPHI->addIncoming(V: ResR, BB: INaNBB);
896 RealPHI->addIncoming(V: LibCallR, BB: LibCallBB);
897 llvm::PHINode *ImagPHI =
898 Builder.CreatePHI(Ty: ResI->getType(), NumReservedValues: 3, Name: "imag_mul_phi");
899 ImagPHI->addIncoming(V: ResI, BB: OrigBB);
900 ImagPHI->addIncoming(V: ResI, BB: INaNBB);
901 ImagPHI->addIncoming(V: LibCallI, BB: LibCallBB);
902 return ComplexPairTy(RealPHI, ImagPHI);
903 }
904 assert((Op.LHS.second || Op.RHS.second) &&
905 "At least one operand must be complex!");
906
907 // If either of the operands is a real rather than a complex, the
908 // imaginary component is ignored when computing the real component of the
909 // result.
910 ResR = Builder.CreateFMul(L: Op.LHS.first, R: Op.RHS.first, Name: "mul.rl");
911
912 ResI = Op.LHS.second
913 ? Builder.CreateFMul(L: Op.LHS.second, R: Op.RHS.first, Name: "mul.il")
914 : Builder.CreateFMul(L: Op.LHS.first, R: Op.RHS.second, Name: "mul.ir");
915 } else {
916 assert(Op.LHS.second && Op.RHS.second &&
917 "Both operands of integer complex operators must be complex!");
918 Value *ResRl = Builder.CreateMul(LHS: Op.LHS.first, RHS: Op.RHS.first, Name: "mul.rl");
919 Value *ResRr = Builder.CreateMul(LHS: Op.LHS.second, RHS: Op.RHS.second, Name: "mul.rr");
920 ResR = Builder.CreateSub(LHS: ResRl, RHS: ResRr, Name: "mul.r");
921
922 Value *ResIl = Builder.CreateMul(LHS: Op.LHS.second, RHS: Op.RHS.first, Name: "mul.il");
923 Value *ResIr = Builder.CreateMul(LHS: Op.LHS.first, RHS: Op.RHS.second, Name: "mul.ir");
924 ResI = Builder.CreateAdd(LHS: ResIl, RHS: ResIr, Name: "mul.i");
925 }
926 return ComplexPairTy(ResR, ResI);
927}
928
929ComplexPairTy ComplexExprEmitter::EmitAlgebraicDiv(llvm::Value *LHSr,
930 llvm::Value *LHSi,
931 llvm::Value *RHSr,
932 llvm::Value *RHSi) {
933 // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd))
934 llvm::Value *DSTr, *DSTi;
935
936 llvm::Value *AC = Builder.CreateFMul(L: LHSr, R: RHSr); // a*c
937 llvm::Value *BD = Builder.CreateFMul(L: LHSi, R: RHSi); // b*d
938 llvm::Value *ACpBD = Builder.CreateFAdd(L: AC, R: BD); // ac+bd
939
940 llvm::Value *CC = Builder.CreateFMul(L: RHSr, R: RHSr); // c*c
941 llvm::Value *DD = Builder.CreateFMul(L: RHSi, R: RHSi); // d*d
942 llvm::Value *CCpDD = Builder.CreateFAdd(L: CC, R: DD); // cc+dd
943
944 llvm::Value *BC = Builder.CreateFMul(L: LHSi, R: RHSr); // b*c
945 llvm::Value *AD = Builder.CreateFMul(L: LHSr, R: RHSi); // a*d
946 llvm::Value *BCmAD = Builder.CreateFSub(L: BC, R: AD); // bc-ad
947
948 DSTr = Builder.CreateFDiv(L: ACpBD, R: CCpDD);
949 DSTi = Builder.CreateFDiv(L: BCmAD, R: CCpDD);
950 return ComplexPairTy(DSTr, DSTi);
951}
952
953// EmitFAbs - Emit a call to @llvm.fabs.
954static llvm::Value *EmitllvmFAbs(CodeGenFunction &CGF, llvm::Value *Value) {
955 return CGF.Builder.CreateFAbs(V: Value);
956}
957
958// EmitRangeReductionDiv - Implements Smith's algorithm for complex division.
959// SMITH, R. L. Algorithm 116: Complex division. Commun. ACM 5, 8 (1962).
960ComplexPairTy ComplexExprEmitter::EmitRangeReductionDiv(llvm::Value *LHSr,
961 llvm::Value *LHSi,
962 llvm::Value *RHSr,
963 llvm::Value *RHSi) {
964 // FIXME: This could eventually be replaced by an LLVM intrinsic to
965 // avoid this long IR sequence.
966
967 // (a + ib) / (c + id) = (e + if)
968 llvm::Value *FAbsRHSr = EmitllvmFAbs(CGF, Value: RHSr); // |c|
969 llvm::Value *FAbsRHSi = EmitllvmFAbs(CGF, Value: RHSi); // |d|
970 // |c| >= |d|
971 llvm::Value *IsR = Builder.CreateFCmpUGT(LHS: FAbsRHSr, RHS: FAbsRHSi, Name: "abs_cmp");
972
973 llvm::BasicBlock *TrueBB =
974 CGF.createBasicBlock(name: "abs_rhsr_greater_or_equal_abs_rhsi");
975 llvm::BasicBlock *FalseBB =
976 CGF.createBasicBlock(name: "abs_rhsr_less_than_abs_rhsi");
977 llvm::BasicBlock *ContBB = CGF.createBasicBlock(name: "complex_div");
978 Builder.CreateCondBr(Cond: IsR, True: TrueBB, False: FalseBB);
979
980 CGF.EmitBlock(BB: TrueBB);
981 // abs(c) >= abs(d)
982 // r = d/c
983 // tmp = c + rd
984 // e = (a + br)/tmp
985 // f = (b - ar)/tmp
986 llvm::Value *DdC = Builder.CreateFDiv(L: RHSi, R: RHSr); // r=d/c
987
988 llvm::Value *RD = Builder.CreateFMul(L: DdC, R: RHSi); // rd
989 llvm::Value *CpRD = Builder.CreateFAdd(L: RHSr, R: RD); // tmp=c+rd
990
991 llvm::Value *T3 = Builder.CreateFMul(L: LHSi, R: DdC); // br
992 llvm::Value *T4 = Builder.CreateFAdd(L: LHSr, R: T3); // a+br
993 llvm::Value *DSTTr = Builder.CreateFDiv(L: T4, R: CpRD); // (a+br)/tmp
994
995 llvm::Value *T5 = Builder.CreateFMul(L: LHSr, R: DdC); // ar
996 llvm::Value *T6 = Builder.CreateFSub(L: LHSi, R: T5); // b-ar
997 llvm::Value *DSTTi = Builder.CreateFDiv(L: T6, R: CpRD); // (b-ar)/tmp
998 Builder.CreateBr(Dest: ContBB);
999
1000 CGF.EmitBlock(BB: FalseBB);
1001 // abs(c) < abs(d)
1002 // r = c/d
1003 // tmp = d + rc
1004 // e = (ar + b)/tmp
1005 // f = (br - a)/tmp
1006 llvm::Value *CdD = Builder.CreateFDiv(L: RHSr, R: RHSi); // r=c/d
1007
1008 llvm::Value *RC = Builder.CreateFMul(L: CdD, R: RHSr); // rc
1009 llvm::Value *DpRC = Builder.CreateFAdd(L: RHSi, R: RC); // tmp=d+rc
1010
1011 llvm::Value *T7 = Builder.CreateFMul(L: LHSr, R: CdD); // ar
1012 llvm::Value *T8 = Builder.CreateFAdd(L: T7, R: LHSi); // ar+b
1013 llvm::Value *DSTFr = Builder.CreateFDiv(L: T8, R: DpRC); // (ar+b)/tmp
1014
1015 llvm::Value *T9 = Builder.CreateFMul(L: LHSi, R: CdD); // br
1016 llvm::Value *T10 = Builder.CreateFSub(L: T9, R: LHSr); // br-a
1017 llvm::Value *DSTFi = Builder.CreateFDiv(L: T10, R: DpRC); // (br-a)/tmp
1018 Builder.CreateBr(Dest: ContBB);
1019
1020 // Phi together the computation paths.
1021 CGF.EmitBlock(BB: ContBB);
1022 llvm::PHINode *VALr = Builder.CreatePHI(Ty: DSTTr->getType(), NumReservedValues: 2);
1023 VALr->addIncoming(V: DSTTr, BB: TrueBB);
1024 VALr->addIncoming(V: DSTFr, BB: FalseBB);
1025 llvm::PHINode *VALi = Builder.CreatePHI(Ty: DSTTi->getType(), NumReservedValues: 2);
1026 VALi->addIncoming(V: DSTTi, BB: TrueBB);
1027 VALi->addIncoming(V: DSTFi, BB: FalseBB);
1028 return ComplexPairTy(VALr, VALi);
1029}
1030
1031// See C11 Annex G.5.1 for the semantics of multiplicative operators on complex
1032// typed values.
1033ComplexPairTy ComplexExprEmitter::EmitBinDiv(const BinOpInfo &Op) {
1034 llvm::Value *LHSr = Op.LHS.first, *LHSi = Op.LHS.second;
1035 llvm::Value *RHSr = Op.RHS.first, *RHSi = Op.RHS.second;
1036 llvm::Value *DSTr, *DSTi;
1037 if (LHSr->getType()->isFloatingPointTy()) {
1038 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, Op.FPFeatures);
1039 if (!RHSi) {
1040 assert(LHSi && "Can have at most one non-complex operand!");
1041
1042 DSTr = Builder.CreateFDiv(L: LHSr, R: RHSr);
1043 DSTi = Builder.CreateFDiv(L: LHSi, R: RHSr);
1044 return ComplexPairTy(DSTr, DSTi);
1045 }
1046 llvm::Value *OrigLHSi = LHSi;
1047 if (!LHSi)
1048 LHSi = llvm::Constant::getNullValue(Ty: RHSi->getType());
1049 if (Op.FPFeatures.getComplexRange() == LangOptions::CX_Improved ||
1050 (Op.FPFeatures.getComplexRange() == LangOptions::CX_Promoted &&
1051 !FPHasBeenPromoted))
1052 return EmitRangeReductionDiv(LHSr, LHSi, RHSr, RHSi);
1053 else if (Op.FPFeatures.getComplexRange() == LangOptions::CX_Basic ||
1054 Op.FPFeatures.getComplexRange() == LangOptions::CX_Promoted)
1055 return EmitAlgebraicDiv(LHSr, LHSi, RHSr, RHSi);
1056 // '-ffast-math' is used in the command line but followed by an
1057 // '-fno-cx-limited-range' or '-fcomplex-arithmetic=full'.
1058 else if (Op.FPFeatures.getComplexRange() == LangOptions::CX_Full) {
1059 LHSi = OrigLHSi;
1060 // If we have a complex operand on the RHS and FastMath is not allowed, we
1061 // delegate to a libcall to handle all of the complexities and minimize
1062 // underflow/overflow cases. When FastMath is allowed we construct the
1063 // divide inline using the same algorithm as for integer operands.
1064 BinOpInfo LibCallOp = Op;
1065 // If LHS was a real, supply a null imaginary part.
1066 if (!LHSi)
1067 LibCallOp.LHS.second = llvm::Constant::getNullValue(Ty: LHSr->getType());
1068
1069 switch (LHSr->getType()->getTypeID()) {
1070 default:
1071 llvm_unreachable("Unsupported floating point type!");
1072 case llvm::Type::HalfTyID:
1073 return EmitComplexBinOpLibCall(LibCallName: "__divhc3", Op: LibCallOp);
1074 case llvm::Type::FloatTyID:
1075 return EmitComplexBinOpLibCall(LibCallName: "__divsc3", Op: LibCallOp);
1076 case llvm::Type::DoubleTyID:
1077 return EmitComplexBinOpLibCall(LibCallName: "__divdc3", Op: LibCallOp);
1078 case llvm::Type::PPC_FP128TyID:
1079 return EmitComplexBinOpLibCall(LibCallName: "__divtc3", Op: LibCallOp);
1080 case llvm::Type::X86_FP80TyID:
1081 return EmitComplexBinOpLibCall(LibCallName: "__divxc3", Op: LibCallOp);
1082 case llvm::Type::FP128TyID:
1083 return EmitComplexBinOpLibCall(
1084 LibCallName: CGF.getTarget().getTriple().isPPC() ? "__divkc3" : "__divtc3",
1085 Op: LibCallOp);
1086 }
1087 } else {
1088 return EmitAlgebraicDiv(LHSr, LHSi, RHSr, RHSi);
1089 }
1090 } else {
1091 assert(Op.LHS.second && Op.RHS.second &&
1092 "Both operands of integer complex operators must be complex!");
1093 // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd))
1094 llvm::Value *Tmp1 = Builder.CreateMul(LHS: LHSr, RHS: RHSr); // a*c
1095 llvm::Value *Tmp2 = Builder.CreateMul(LHS: LHSi, RHS: RHSi); // b*d
1096 llvm::Value *Tmp3 = Builder.CreateAdd(LHS: Tmp1, RHS: Tmp2); // ac+bd
1097
1098 llvm::Value *Tmp4 = Builder.CreateMul(LHS: RHSr, RHS: RHSr); // c*c
1099 llvm::Value *Tmp5 = Builder.CreateMul(LHS: RHSi, RHS: RHSi); // d*d
1100 llvm::Value *Tmp6 = Builder.CreateAdd(LHS: Tmp4, RHS: Tmp5); // cc+dd
1101
1102 llvm::Value *Tmp7 = Builder.CreateMul(LHS: LHSi, RHS: RHSr); // b*c
1103 llvm::Value *Tmp8 = Builder.CreateMul(LHS: LHSr, RHS: RHSi); // a*d
1104 llvm::Value *Tmp9 = Builder.CreateSub(LHS: Tmp7, RHS: Tmp8); // bc-ad
1105
1106 if (Op.Ty->castAs<ComplexType>()
1107 ->getElementType()
1108 ->isUnsignedIntegerType()) {
1109 DSTr = Builder.CreateUDiv(LHS: Tmp3, RHS: Tmp6);
1110 DSTi = Builder.CreateUDiv(LHS: Tmp9, RHS: Tmp6);
1111 } else {
1112 DSTr = Builder.CreateSDiv(LHS: Tmp3, RHS: Tmp6);
1113 DSTi = Builder.CreateSDiv(LHS: Tmp9, RHS: Tmp6);
1114 }
1115 }
1116
1117 return ComplexPairTy(DSTr, DSTi);
1118}
1119
1120ComplexPairTy CodeGenFunction::EmitUnPromotedValue(ComplexPairTy result,
1121 QualType UnPromotionType) {
1122 llvm::Type *ComplexElementTy =
1123 ConvertType(T: UnPromotionType->castAs<ComplexType>()->getElementType());
1124 if (result.first)
1125 result.first =
1126 Builder.CreateFPTrunc(V: result.first, DestTy: ComplexElementTy, Name: "unpromotion");
1127 if (result.second)
1128 result.second =
1129 Builder.CreateFPTrunc(V: result.second, DestTy: ComplexElementTy, Name: "unpromotion");
1130 return result;
1131}
1132
1133ComplexPairTy CodeGenFunction::EmitPromotedValue(ComplexPairTy result,
1134 QualType PromotionType) {
1135 llvm::Type *ComplexElementTy =
1136 ConvertType(T: PromotionType->castAs<ComplexType>()->getElementType());
1137 if (result.first)
1138 result.first = Builder.CreateFPExt(V: result.first, DestTy: ComplexElementTy, Name: "ext");
1139 if (result.second)
1140 result.second = Builder.CreateFPExt(V: result.second, DestTy: ComplexElementTy, Name: "ext");
1141
1142 return result;
1143}
1144
1145ComplexPairTy ComplexExprEmitter::EmitPromoted(const Expr *E,
1146 QualType PromotionType) {
1147 E = E->IgnoreParens();
1148 if (auto BO = dyn_cast<BinaryOperator>(Val: E)) {
1149 switch (BO->getOpcode()) {
1150#define HANDLE_BINOP(OP) \
1151 case BO_##OP: \
1152 return EmitBin##OP(EmitBinOps(BO, PromotionType));
1153 HANDLE_BINOP(Add)
1154 HANDLE_BINOP(Sub)
1155 HANDLE_BINOP(Mul)
1156 HANDLE_BINOP(Div)
1157#undef HANDLE_BINOP
1158 default:
1159 break;
1160 }
1161 } else if (auto UO = dyn_cast<UnaryOperator>(Val: E)) {
1162 switch (UO->getOpcode()) {
1163 case UO_Minus:
1164 return VisitMinus(E: UO, PromotionType);
1165 case UO_Plus:
1166 return VisitPlus(E: UO, PromotionType);
1167 default:
1168 break;
1169 }
1170 }
1171 auto result = Visit(E: const_cast<Expr *>(E));
1172 if (!PromotionType.isNull())
1173 return CGF.EmitPromotedValue(result, PromotionType);
1174 else
1175 return result;
1176}
1177
1178ComplexPairTy CodeGenFunction::EmitPromotedComplexExpr(const Expr *E,
1179 QualType DstTy) {
1180 return ComplexExprEmitter(*this).EmitPromoted(E, PromotionType: DstTy);
1181}
1182
1183ComplexPairTy
1184ComplexExprEmitter::EmitPromotedComplexOperand(const Expr *E,
1185 QualType OverallPromotionType) {
1186 if (E->getType()->isAnyComplexType()) {
1187 if (!OverallPromotionType.isNull())
1188 return CGF.EmitPromotedComplexExpr(E, DstTy: OverallPromotionType);
1189 else
1190 return Visit(E: const_cast<Expr *>(E));
1191 } else {
1192 if (!OverallPromotionType.isNull()) {
1193 QualType ComplexElementTy =
1194 OverallPromotionType->castAs<ComplexType>()->getElementType();
1195 return ComplexPairTy(CGF.EmitPromotedScalarExpr(E, PromotionType: ComplexElementTy),
1196 nullptr);
1197 } else {
1198 return ComplexPairTy(CGF.EmitScalarExpr(E), nullptr);
1199 }
1200 }
1201}
1202
1203ComplexExprEmitter::BinOpInfo
1204ComplexExprEmitter::EmitBinOps(const BinaryOperator *E,
1205 QualType PromotionType) {
1206 TestAndClearIgnoreReal();
1207 TestAndClearIgnoreImag();
1208 BinOpInfo Ops;
1209
1210 Ops.LHS = EmitPromotedComplexOperand(E: E->getLHS(), OverallPromotionType: PromotionType);
1211 Ops.RHS = EmitPromotedComplexOperand(E: E->getRHS(), OverallPromotionType: PromotionType);
1212 if (!PromotionType.isNull())
1213 Ops.Ty = PromotionType;
1214 else
1215 Ops.Ty = E->getType();
1216 Ops.FPFeatures = E->getFPFeaturesInEffect(LO: CGF.getLangOpts());
1217 return Ops;
1218}
1219
1220LValue ComplexExprEmitter::EmitCompoundAssignLValue(
1221 const CompoundAssignOperator *E,
1222 ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo &), RValue &Val) {
1223 TestAndClearIgnoreReal();
1224 TestAndClearIgnoreImag();
1225 QualType LHSTy = E->getLHS()->getType().getAtomicUnqualifiedType();
1226
1227 BinOpInfo OpInfo;
1228 OpInfo.FPFeatures = E->getFPFeaturesInEffect(LO: CGF.getLangOpts());
1229 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, OpInfo.FPFeatures);
1230
1231 const bool IsComplexDivisor = E->getOpcode() == BO_DivAssign &&
1232 E->getRHS()->getType()->isAnyComplexType();
1233
1234 // Load the RHS and LHS operands.
1235 // __block variables need to have the rhs evaluated first, plus this should
1236 // improve codegen a little.
1237 QualType PromotionTypeCR;
1238 PromotionTypeCR =
1239 getPromotionType(Features: E->getStoredFPFeaturesOrDefault(),
1240 Ty: E->getComputationResultType(), IsComplexDivisor);
1241 if (PromotionTypeCR.isNull())
1242 PromotionTypeCR = E->getComputationResultType();
1243 OpInfo.Ty = PromotionTypeCR;
1244 QualType ComplexElementTy =
1245 OpInfo.Ty->castAs<ComplexType>()->getElementType();
1246 QualType PromotionTypeRHS =
1247 getPromotionType(Features: E->getStoredFPFeaturesOrDefault(),
1248 Ty: E->getRHS()->getType(), IsComplexDivisor);
1249
1250 // The RHS should have been converted to the computation type.
1251 if (E->getRHS()->getType()->isRealFloatingType()) {
1252 if (!PromotionTypeRHS.isNull())
1253 OpInfo.RHS = ComplexPairTy(
1254 CGF.EmitPromotedScalarExpr(E: E->getRHS(), PromotionType: PromotionTypeRHS), nullptr);
1255 else {
1256 assert(CGF.getContext().hasSameUnqualifiedType(ComplexElementTy,
1257 E->getRHS()->getType()));
1258
1259 OpInfo.RHS = ComplexPairTy(CGF.EmitScalarExpr(E: E->getRHS()), nullptr);
1260 }
1261 } else {
1262 if (!PromotionTypeRHS.isNull()) {
1263 OpInfo.RHS = ComplexPairTy(
1264 CGF.EmitPromotedComplexExpr(E: E->getRHS(), DstTy: PromotionTypeRHS));
1265 } else {
1266 assert(CGF.getContext().hasSameUnqualifiedType(OpInfo.Ty,
1267 E->getRHS()->getType()));
1268 OpInfo.RHS = Visit(E: E->getRHS());
1269 }
1270 }
1271
1272 LValue LHS = CGF.EmitLValue(E: E->getLHS());
1273
1274 // Load from the l-value and convert it.
1275 SourceLocation Loc = E->getExprLoc();
1276 QualType PromotionTypeLHS =
1277 getPromotionType(Features: E->getStoredFPFeaturesOrDefault(),
1278 Ty: E->getComputationLHSType(), IsComplexDivisor);
1279 if (LHSTy->isAnyComplexType()) {
1280 ComplexPairTy LHSVal = EmitLoadOfLValue(lvalue: LHS, loc: Loc);
1281 if (!PromotionTypeLHS.isNull())
1282 OpInfo.LHS =
1283 EmitComplexToComplexCast(Val: LHSVal, SrcType: LHSTy, DestType: PromotionTypeLHS, Loc);
1284 else
1285 OpInfo.LHS = EmitComplexToComplexCast(Val: LHSVal, SrcType: LHSTy, DestType: OpInfo.Ty, Loc);
1286 } else {
1287 llvm::Value *LHSVal = CGF.EmitLoadOfLValue(V: LHS, Loc).getScalarVal();
1288 // For floating point real operands we can directly pass the scalar form
1289 // to the binary operator emission and potentially get more efficient code.
1290 if (LHSTy->isRealFloatingType()) {
1291 QualType PromotedComplexElementTy;
1292 if (!PromotionTypeLHS.isNull()) {
1293 PromotedComplexElementTy =
1294 cast<ComplexType>(Val&: PromotionTypeLHS)->getElementType();
1295 if (!CGF.getContext().hasSameUnqualifiedType(T1: PromotedComplexElementTy,
1296 T2: PromotionTypeLHS))
1297 LHSVal = CGF.EmitScalarConversion(Src: LHSVal, SrcTy: LHSTy,
1298 DstTy: PromotedComplexElementTy, Loc);
1299 } else {
1300 if (!CGF.getContext().hasSameUnqualifiedType(T1: ComplexElementTy, T2: LHSTy))
1301 LHSVal =
1302 CGF.EmitScalarConversion(Src: LHSVal, SrcTy: LHSTy, DstTy: ComplexElementTy, Loc);
1303 }
1304 OpInfo.LHS = ComplexPairTy(LHSVal, nullptr);
1305 } else {
1306 OpInfo.LHS = EmitScalarToComplexCast(Val: LHSVal, SrcType: LHSTy, DestType: OpInfo.Ty, Loc);
1307 }
1308 }
1309
1310 // Expand the binary operator.
1311 ComplexPairTy Result = (this->*Func)(OpInfo);
1312
1313 // Truncate the result and store it into the LHS lvalue.
1314 if (LHSTy->isAnyComplexType()) {
1315 ComplexPairTy ResVal =
1316 EmitComplexToComplexCast(Val: Result, SrcType: OpInfo.Ty, DestType: LHSTy, Loc);
1317 EmitStoreOfComplex(Val: ResVal, lvalue: LHS, /*isInit*/ false);
1318 Val = RValue::getComplex(C: ResVal);
1319 } else {
1320 llvm::Value *ResVal =
1321 CGF.EmitComplexToScalarConversion(Src: Result, SrcTy: OpInfo.Ty, DstTy: LHSTy, Loc);
1322 CGF.EmitStoreThroughLValue(Src: RValue::get(V: ResVal), Dst: LHS, /*isInit*/ false);
1323 Val = RValue::get(V: ResVal);
1324 }
1325
1326 return LHS;
1327}
1328
1329// Compound assignments.
1330ComplexPairTy ComplexExprEmitter::EmitCompoundAssign(
1331 const CompoundAssignOperator *E,
1332 ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo &)) {
1333 RValue Val;
1334 LValue LV = EmitCompoundAssignLValue(E, Func, Val);
1335
1336 // The result of an assignment in C is the assigned r-value.
1337 if (!CGF.getLangOpts().CPlusPlus)
1338 return Val.getComplexVal();
1339
1340 // If the lvalue is non-volatile, return the computed value of the assignment.
1341 if (!LV.isVolatileQualified())
1342 return Val.getComplexVal();
1343
1344 return EmitLoadOfLValue(lvalue: LV, loc: E->getExprLoc());
1345}
1346
1347LValue ComplexExprEmitter::EmitBinAssignLValue(const BinaryOperator *E,
1348 ComplexPairTy &Val) {
1349 assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(),
1350 E->getRHS()->getType()) &&
1351 "Invalid assignment");
1352 TestAndClearIgnoreReal();
1353 TestAndClearIgnoreImag();
1354
1355 // Emit the RHS. __block variables need the RHS evaluated first.
1356 Val = Visit(E: E->getRHS());
1357
1358 // Compute the address to store into.
1359 LValue LHS = CGF.EmitLValue(E: E->getLHS());
1360
1361 // Store the result value into the LHS lvalue.
1362 EmitStoreOfComplex(Val, lvalue: LHS, /*isInit*/ false);
1363
1364 return LHS;
1365}
1366
1367ComplexPairTy ComplexExprEmitter::VisitBinAssign(const BinaryOperator *E) {
1368 ComplexPairTy Val;
1369 ApplyAtomGroup Grp(CGF.getDebugInfo());
1370 LValue LV = EmitBinAssignLValue(E, Val);
1371
1372 // The result of an assignment in C is the assigned r-value.
1373 if (!CGF.getLangOpts().CPlusPlus)
1374 return Val;
1375
1376 // If the lvalue is non-volatile, return the computed value of the assignment.
1377 if (!LV.isVolatileQualified())
1378 return Val;
1379
1380 return EmitLoadOfLValue(lvalue: LV, loc: E->getExprLoc());
1381}
1382
1383ComplexPairTy ComplexExprEmitter::VisitBinComma(const BinaryOperator *E) {
1384 CGF.EmitIgnoredExpr(E: E->getLHS());
1385 return Visit(E: E->getRHS());
1386}
1387
1388ComplexPairTy ComplexExprEmitter::VisitAbstractConditionalOperator(
1389 const AbstractConditionalOperator *E) {
1390 TestAndClearIgnoreReal();
1391 TestAndClearIgnoreImag();
1392 llvm::BasicBlock *LHSBlock = CGF.createBasicBlock(name: "cond.true");
1393 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock(name: "cond.false");
1394 llvm::BasicBlock *ContBlock = CGF.createBasicBlock(name: "cond.end");
1395
1396 // Bind the common expression if necessary.
1397 CodeGenFunction::OpaqueValueMapping binding(CGF, E);
1398
1399 CodeGenFunction::ConditionalEvaluation eval(CGF);
1400 CGF.EmitBranchOnBoolExpr(Cond: E->getCond(), TrueBlock: LHSBlock, FalseBlock: RHSBlock,
1401 TrueCount: CGF.getProfileCount(S: E));
1402
1403 eval.begin(CGF);
1404 CGF.EmitBlock(BB: LHSBlock);
1405 CGF.incrementProfileCounter(ExecSkip: CGF.UseExecPath, S: E);
1406 ComplexPairTy LHS = Visit(E: E->getTrueExpr());
1407 LHSBlock = Builder.GetInsertBlock();
1408 CGF.EmitBranch(Block: ContBlock);
1409 eval.end(CGF);
1410
1411 eval.begin(CGF);
1412 CGF.EmitBlock(BB: RHSBlock);
1413 CGF.incrementProfileCounter(ExecSkip: CGF.UseSkipPath, S: E);
1414 ComplexPairTy RHS = Visit(E: E->getFalseExpr());
1415 RHSBlock = Builder.GetInsertBlock();
1416 CGF.EmitBlock(BB: ContBlock);
1417 eval.end(CGF);
1418
1419 // Create a PHI node for the real part.
1420 llvm::PHINode *RealPN = Builder.CreatePHI(Ty: LHS.first->getType(), NumReservedValues: 2, Name: "cond.r");
1421 RealPN->addIncoming(V: LHS.first, BB: LHSBlock);
1422 RealPN->addIncoming(V: RHS.first, BB: RHSBlock);
1423
1424 // Create a PHI node for the imaginary part.
1425 llvm::PHINode *ImagPN = Builder.CreatePHI(Ty: LHS.first->getType(), NumReservedValues: 2, Name: "cond.i");
1426 ImagPN->addIncoming(V: LHS.second, BB: LHSBlock);
1427 ImagPN->addIncoming(V: RHS.second, BB: RHSBlock);
1428
1429 return ComplexPairTy(RealPN, ImagPN);
1430}
1431
1432ComplexPairTy ComplexExprEmitter::VisitChooseExpr(ChooseExpr *E) {
1433 return Visit(E: E->getChosenSubExpr());
1434}
1435
1436ComplexPairTy ComplexExprEmitter::VisitInitListExpr(InitListExpr *E) {
1437 bool Ignore = TestAndClearIgnoreReal();
1438 (void)Ignore;
1439 assert(Ignore == false && "init list ignored");
1440 Ignore = TestAndClearIgnoreImag();
1441 (void)Ignore;
1442 assert(Ignore == false && "init list ignored");
1443
1444 if (E->getNumInits() == 2) {
1445 llvm::Value *Real = CGF.EmitScalarExpr(E: E->getInit(Init: 0));
1446 llvm::Value *Imag = CGF.EmitScalarExpr(E: E->getInit(Init: 1));
1447 return ComplexPairTy(Real, Imag);
1448 } else if (E->getNumInits() == 1) {
1449 return Visit(E: E->getInit(Init: 0));
1450 }
1451
1452 // Empty init list initializes to null
1453 assert(E->getNumInits() == 0 && "Unexpected number of inits");
1454 QualType Ty = E->getType()->castAs<ComplexType>()->getElementType();
1455 llvm::Type *LTy = CGF.ConvertType(T: Ty);
1456 llvm::Value *zeroConstant = llvm::Constant::getNullValue(Ty: LTy);
1457 return ComplexPairTy(zeroConstant, zeroConstant);
1458}
1459
1460ComplexPairTy ComplexExprEmitter::VisitVAArgExpr(VAArgExpr *E) {
1461 Address ArgValue = Address::invalid();
1462 RValue RV = CGF.EmitVAArg(VE: E, VAListAddr&: ArgValue);
1463
1464 if (!ArgValue.isValid()) {
1465 CGF.ErrorUnsupported(S: E, Type: "complex va_arg expression");
1466 llvm::Type *EltTy =
1467 CGF.ConvertType(T: E->getType()->castAs<ComplexType>()->getElementType());
1468 llvm::Value *U = llvm::PoisonValue::get(T: EltTy);
1469 return ComplexPairTy(U, U);
1470 }
1471
1472 return RV.getComplexVal();
1473}
1474
1475//===----------------------------------------------------------------------===//
1476// Entry Point into this File
1477//===----------------------------------------------------------------------===//
1478
1479/// EmitComplexExpr - Emit the computation of the specified expression of
1480/// complex type, ignoring the result.
1481ComplexPairTy CodeGenFunction::EmitComplexExpr(const Expr *E, bool IgnoreReal,
1482 bool IgnoreImag) {
1483 assert(E && getComplexType(E->getType()) &&
1484 "Invalid complex expression to emit");
1485
1486 return ComplexExprEmitter(*this, IgnoreReal, IgnoreImag)
1487 .Visit(E: const_cast<Expr *>(E));
1488}
1489
1490void CodeGenFunction::EmitComplexExprIntoLValue(const Expr *E, LValue dest,
1491 bool isInit) {
1492 assert(E && getComplexType(E->getType()) &&
1493 "Invalid complex expression to emit");
1494 ComplexExprEmitter Emitter(*this);
1495 ComplexPairTy Val = Emitter.Visit(E: const_cast<Expr *>(E));
1496 Emitter.EmitStoreOfComplex(Val, lvalue: dest, isInit);
1497}
1498
1499/// EmitStoreOfComplex - Store a complex number into the specified l-value.
1500void CodeGenFunction::EmitStoreOfComplex(ComplexPairTy V, LValue dest,
1501 bool isInit) {
1502 ComplexExprEmitter(*this).EmitStoreOfComplex(Val: V, lvalue: dest, isInit);
1503}
1504
1505/// EmitLoadOfComplex - Load a complex number from the specified address.
1506ComplexPairTy CodeGenFunction::EmitLoadOfComplex(LValue src,
1507 SourceLocation loc) {
1508 return ComplexExprEmitter(*this).EmitLoadOfLValue(lvalue: src, loc);
1509}
1510
1511LValue CodeGenFunction::EmitComplexAssignmentLValue(const BinaryOperator *E) {
1512 assert(E->getOpcode() == BO_Assign);
1513 ComplexPairTy Val; // ignored
1514 LValue LVal = ComplexExprEmitter(*this).EmitBinAssignLValue(E, Val);
1515 if (getLangOpts().OpenMP)
1516 CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF&: *this,
1517 LHS: E->getLHS());
1518 return LVal;
1519}
1520
1521typedef ComplexPairTy (ComplexExprEmitter::*CompoundFunc)(
1522 const ComplexExprEmitter::BinOpInfo &);
1523
1524static CompoundFunc getComplexOp(BinaryOperatorKind Op) {
1525 switch (Op) {
1526 case BO_MulAssign:
1527 return &ComplexExprEmitter::EmitBinMul;
1528 case BO_DivAssign:
1529 return &ComplexExprEmitter::EmitBinDiv;
1530 case BO_SubAssign:
1531 return &ComplexExprEmitter::EmitBinSub;
1532 case BO_AddAssign:
1533 return &ComplexExprEmitter::EmitBinAdd;
1534 default:
1535 llvm_unreachable("unexpected complex compound assignment");
1536 }
1537}
1538
1539LValue CodeGenFunction::EmitComplexCompoundAssignmentLValue(
1540 const CompoundAssignOperator *E) {
1541 ApplyAtomGroup Grp(getDebugInfo());
1542 CompoundFunc Op = getComplexOp(Op: E->getOpcode());
1543 RValue Val;
1544 return ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Func: Op, Val);
1545}
1546
1547LValue CodeGenFunction::EmitScalarCompoundAssignWithComplex(
1548 const CompoundAssignOperator *E, llvm::Value *&Result) {
1549 // Key Instructions: Don't need to create an atom group here; one will already
1550 // be active through scalar handling code.
1551 CompoundFunc Op = getComplexOp(Op: E->getOpcode());
1552 RValue Val;
1553 LValue Ret = ComplexExprEmitter(*this).EmitCompoundAssignLValue(E, Func: Op, Val);
1554 Result = Val.getScalarVal();
1555 return Ret;
1556}
1557