1 //===--- CGExprComplex.cpp - Emit LLVM Code for Complex Exprs -------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // This contains code to emit Expr nodes with complex types as LLVM code. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #include "CodeGenFunction.h" 15 #include "CodeGenModule.h" 16 #include "clang/AST/ASTContext.h" 17 #include "clang/AST/StmtVisitor.h" 18 #include "llvm/Constants.h" 19 #include "llvm/Function.h" 20 #include "llvm/ADT/SmallString.h" 21 #include "llvm/Support/Compiler.h" 22 using namespace clang; 23 using namespace CodeGen; 24 25 //===----------------------------------------------------------------------===// 26 // Complex Expression Emitter 27 //===----------------------------------------------------------------------===// 28 29 typedef CodeGenFunction::ComplexPairTy ComplexPairTy; 30 31 namespace { 32 class VISIBILITY_HIDDEN ComplexExprEmitter 33 : public StmtVisitor<ComplexExprEmitter, ComplexPairTy> { 34 CodeGenFunction &CGF; 35 CGBuilderTy &Builder; 36 public: 37 ComplexExprEmitter(CodeGenFunction &cgf) : CGF(cgf), Builder(CGF.Builder) { 38 } 39 40 41 //===--------------------------------------------------------------------===// 42 // Utilities 43 //===--------------------------------------------------------------------===// 44 45 /// EmitLoadOfLValue - Given an expression with complex type that represents a 46 /// value l-value, this method emits the address of the l-value, then loads 47 /// and returns the result. 48 ComplexPairTy EmitLoadOfLValue(const Expr *E) { 49 LValue LV = CGF.EmitLValue(E); 50 // FIXME: Volatile 51 return EmitLoadOfComplex(LV.getAddress(), false); 52 } 53 54 /// EmitLoadOfComplex - Given a pointer to a complex value, emit code to load 55 /// the real and imaginary pieces. 56 ComplexPairTy EmitLoadOfComplex(llvm::Value *SrcPtr, bool isVolatile); 57 58 /// EmitStoreOfComplex - Store the specified real/imag parts into the 59 /// specified value pointer. 60 void EmitStoreOfComplex(ComplexPairTy Val, llvm::Value *ResPtr, bool isVol); 61 62 /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType. 63 ComplexPairTy EmitComplexToComplexCast(ComplexPairTy Val, QualType SrcType, 64 QualType DestType); 65 66 //===--------------------------------------------------------------------===// 67 // Visitor Methods 68 //===--------------------------------------------------------------------===// 69 70 ComplexPairTy VisitStmt(Stmt *S) { 71 S->dump(CGF.getContext().getSourceManager()); 72 assert(0 && "Stmt can't have complex result type!"); 73 return ComplexPairTy(); 74 } 75 ComplexPairTy VisitExpr(Expr *S); 76 ComplexPairTy VisitParenExpr(ParenExpr *PE) { return Visit(PE->getSubExpr());} 77 ComplexPairTy VisitImaginaryLiteral(const ImaginaryLiteral *IL); 78 79 // l-values. 80 ComplexPairTy VisitDeclRefExpr(const Expr *E) { return EmitLoadOfLValue(E); } 81 ComplexPairTy VisitArraySubscriptExpr(Expr *E) { return EmitLoadOfLValue(E); } 82 ComplexPairTy VisitMemberExpr(const Expr *E) { return EmitLoadOfLValue(E); } 83 84 // FIXME: CompoundLiteralExpr 85 86 ComplexPairTy EmitCast(Expr *Op, QualType DestTy); 87 ComplexPairTy VisitImplicitCastExpr(ImplicitCastExpr *E) { 88 // Unlike for scalars, we don't have to worry about function->ptr demotion 89 // here. 90 return EmitCast(E->getSubExpr(), E->getType()); 91 } 92 ComplexPairTy VisitCastExpr(CastExpr *E) { 93 return EmitCast(E->getSubExpr(), E->getType()); 94 } 95 ComplexPairTy VisitCallExpr(const CallExpr *E); 96 ComplexPairTy VisitStmtExpr(const StmtExpr *E); 97 ComplexPairTy VisitOverloadExpr(const OverloadExpr *OE); 98 99 // Operators. 100 ComplexPairTy VisitPrePostIncDec(const UnaryOperator *E, 101 bool isInc, bool isPre); 102 ComplexPairTy VisitUnaryPostDec(const UnaryOperator *E) { 103 return VisitPrePostIncDec(E, false, false); 104 } 105 ComplexPairTy VisitUnaryPostInc(const UnaryOperator *E) { 106 return VisitPrePostIncDec(E, true, false); 107 } 108 ComplexPairTy VisitUnaryPreDec(const UnaryOperator *E) { 109 return VisitPrePostIncDec(E, false, true); 110 } 111 ComplexPairTy VisitUnaryPreInc(const UnaryOperator *E) { 112 return VisitPrePostIncDec(E, true, true); 113 } 114 ComplexPairTy VisitUnaryDeref(const Expr *E) { return EmitLoadOfLValue(E); } 115 ComplexPairTy VisitUnaryPlus (const UnaryOperator *E) { 116 return Visit(E->getSubExpr()); 117 } 118 ComplexPairTy VisitUnaryMinus (const UnaryOperator *E); 119 ComplexPairTy VisitUnaryNot (const UnaryOperator *E); 120 // LNot,Real,Imag never return complex. 121 ComplexPairTy VisitUnaryExtension(const UnaryOperator *E) { 122 return Visit(E->getSubExpr()); 123 } 124 ComplexPairTy VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) { 125 return Visit(DAE->getExpr()); 126 } 127 ComplexPairTy VisitCXXZeroInitValueExpr(CXXZeroInitValueExpr *E) { 128 assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 129 QualType Elem = E->getType()->getAsComplexType()->getElementType(); 130 llvm::Constant *Null = llvm::Constant::getNullValue(CGF.ConvertType(Elem)); 131 return ComplexPairTy(Null, Null); 132 } 133 ComplexPairTy VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) { 134 assert(E->getType()->isAnyComplexType() && "Expected complex type!"); 135 QualType Elem = E->getType()->getAsComplexType()->getElementType(); 136 llvm::Constant *Null = llvm::Constant::getNullValue(CGF.ConvertType(Elem)); 137 return ComplexPairTy(Null, Null); 138 } 139 140 struct BinOpInfo { 141 ComplexPairTy LHS; 142 ComplexPairTy RHS; 143 QualType Ty; // Computation Type. 144 }; 145 146 BinOpInfo EmitBinOps(const BinaryOperator *E); 147 ComplexPairTy EmitCompoundAssign(const CompoundAssignOperator *E, 148 ComplexPairTy (ComplexExprEmitter::*Func) 149 (const BinOpInfo &)); 150 151 ComplexPairTy EmitBinAdd(const BinOpInfo &Op); 152 ComplexPairTy EmitBinSub(const BinOpInfo &Op); 153 ComplexPairTy EmitBinMul(const BinOpInfo &Op); 154 ComplexPairTy EmitBinDiv(const BinOpInfo &Op); 155 156 ComplexPairTy VisitBinMul(const BinaryOperator *E) { 157 return EmitBinMul(EmitBinOps(E)); 158 } 159 ComplexPairTy VisitBinAdd(const BinaryOperator *E) { 160 return EmitBinAdd(EmitBinOps(E)); 161 } 162 ComplexPairTy VisitBinSub(const BinaryOperator *E) { 163 return EmitBinSub(EmitBinOps(E)); 164 } 165 ComplexPairTy VisitBinDiv(const BinaryOperator *E) { 166 return EmitBinDiv(EmitBinOps(E)); 167 } 168 169 // Compound assignments. 170 ComplexPairTy VisitBinAddAssign(const CompoundAssignOperator *E) { 171 return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinAdd); 172 } 173 ComplexPairTy VisitBinSubAssign(const CompoundAssignOperator *E) { 174 return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinSub); 175 } 176 ComplexPairTy VisitBinMulAssign(const CompoundAssignOperator *E) { 177 return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinMul); 178 } 179 ComplexPairTy VisitBinDivAssign(const CompoundAssignOperator *E) { 180 return EmitCompoundAssign(E, &ComplexExprEmitter::EmitBinDiv); 181 } 182 183 // GCC rejects rem/and/or/xor for integer complex. 184 // Logical and/or always return int, never complex. 185 186 // No comparisons produce a complex result. 187 ComplexPairTy VisitBinAssign (const BinaryOperator *E); 188 ComplexPairTy VisitBinComma (const BinaryOperator *E); 189 190 191 ComplexPairTy VisitConditionalOperator(const ConditionalOperator *CO); 192 ComplexPairTy VisitChooseExpr(ChooseExpr *CE); 193 194 ComplexPairTy VisitInitListExpr(InitListExpr *E); 195 }; 196 } // end anonymous namespace. 197 198 //===----------------------------------------------------------------------===// 199 // Utilities 200 //===----------------------------------------------------------------------===// 201 202 /// EmitLoadOfComplex - Given an RValue reference for a complex, emit code to 203 /// load the real and imaginary pieces, returning them as Real/Imag. 204 ComplexPairTy ComplexExprEmitter::EmitLoadOfComplex(llvm::Value *SrcPtr, 205 bool isVolatile) { 206 llvm::SmallString<64> Name(SrcPtr->getNameStart(), 207 SrcPtr->getNameStart()+SrcPtr->getNameLen()); 208 209 Name += ".realp"; 210 llvm::Value *RealPtr = Builder.CreateStructGEP(SrcPtr, 0, Name.c_str()); 211 212 Name.pop_back(); // .realp -> .real 213 llvm::Value *Real = Builder.CreateLoad(RealPtr, isVolatile, Name.c_str()); 214 215 Name.resize(Name.size()-4); // .real -> .imagp 216 Name += "imagp"; 217 218 llvm::Value *ImagPtr = Builder.CreateStructGEP(SrcPtr, 1, Name.c_str()); 219 220 Name.pop_back(); // .imagp -> .imag 221 llvm::Value *Imag = Builder.CreateLoad(ImagPtr, isVolatile, Name.c_str()); 222 return ComplexPairTy(Real, Imag); 223 } 224 225 /// EmitStoreOfComplex - Store the specified real/imag parts into the 226 /// specified value pointer. 227 void ComplexExprEmitter::EmitStoreOfComplex(ComplexPairTy Val, llvm::Value *Ptr, 228 bool isVolatile) { 229 llvm::Value *RealPtr = Builder.CreateStructGEP(Ptr, 0, "real"); 230 llvm::Value *ImagPtr = Builder.CreateStructGEP(Ptr, 1, "imag"); 231 232 Builder.CreateStore(Val.first, RealPtr, isVolatile); 233 Builder.CreateStore(Val.second, ImagPtr, isVolatile); 234 } 235 236 237 238 //===----------------------------------------------------------------------===// 239 // Visitor Methods 240 //===----------------------------------------------------------------------===// 241 242 ComplexPairTy ComplexExprEmitter::VisitExpr(Expr *E) { 243 CGF.ErrorUnsupported(E, "complex expression"); 244 const llvm::Type *EltTy = 245 CGF.ConvertType(E->getType()->getAsComplexType()->getElementType()); 246 llvm::Value *U = llvm::UndefValue::get(EltTy); 247 return ComplexPairTy(U, U); 248 } 249 250 ComplexPairTy ComplexExprEmitter:: 251 VisitImaginaryLiteral(const ImaginaryLiteral *IL) { 252 llvm::Value *Imag = CGF.EmitScalarExpr(IL->getSubExpr()); 253 return ComplexPairTy(llvm::Constant::getNullValue(Imag->getType()), Imag); 254 } 255 256 257 ComplexPairTy ComplexExprEmitter::VisitCallExpr(const CallExpr *E) { 258 return CGF.EmitCallExpr(E).getComplexVal(); 259 } 260 261 ComplexPairTy ComplexExprEmitter::VisitOverloadExpr(const OverloadExpr *E) { 262 return CGF.EmitCallExpr(E->getFn(), E->arg_begin(), 263 E->arg_end(CGF.getContext())).getComplexVal(); 264 } 265 266 ComplexPairTy ComplexExprEmitter::VisitStmtExpr(const StmtExpr *E) { 267 return CGF.EmitCompoundStmt(*E->getSubStmt(), true).getComplexVal(); 268 } 269 270 /// EmitComplexToComplexCast - Emit a cast from complex value Val to DestType. 271 ComplexPairTy ComplexExprEmitter::EmitComplexToComplexCast(ComplexPairTy Val, 272 QualType SrcType, 273 QualType DestType) { 274 // Get the src/dest element type. 275 SrcType = SrcType->getAsComplexType()->getElementType(); 276 DestType = DestType->getAsComplexType()->getElementType(); 277 278 // C99 6.3.1.6: When a value of complex type is converted to another 279 // complex type, both the real and imaginary parts follow the conversion 280 // rules for the corresponding real types. 281 Val.first = CGF.EmitScalarConversion(Val.first, SrcType, DestType); 282 Val.second = CGF.EmitScalarConversion(Val.second, SrcType, DestType); 283 return Val; 284 } 285 286 ComplexPairTy ComplexExprEmitter::EmitCast(Expr *Op, QualType DestTy) { 287 // Two cases here: cast from (complex to complex) and (scalar to complex). 288 if (Op->getType()->isAnyComplexType()) 289 return EmitComplexToComplexCast(Visit(Op), Op->getType(), DestTy); 290 291 // C99 6.3.1.7: When a value of real type is converted to a complex type, the 292 // real part of the complex result value is determined by the rules of 293 // conversion to the corresponding real type and the imaginary part of the 294 // complex result value is a positive zero or an unsigned zero. 295 llvm::Value *Elt = CGF.EmitScalarExpr(Op); 296 297 // Convert the input element to the element type of the complex. 298 DestTy = DestTy->getAsComplexType()->getElementType(); 299 Elt = CGF.EmitScalarConversion(Elt, Op->getType(), DestTy); 300 301 // Return (realval, 0). 302 return ComplexPairTy(Elt, llvm::Constant::getNullValue(Elt->getType())); 303 } 304 305 ComplexPairTy ComplexExprEmitter::VisitPrePostIncDec(const UnaryOperator *E, 306 bool isInc, bool isPre) { 307 LValue LV = CGF.EmitLValue(E->getSubExpr()); 308 // FIXME: Handle volatile! 309 ComplexPairTy InVal = EmitLoadOfComplex(LV.getAddress(), false); 310 311 uint64_t AmountVal = isInc ? 1 : -1; 312 313 llvm::Value *NextVal; 314 if (isa<llvm::IntegerType>(InVal.first->getType())) 315 NextVal = llvm::ConstantInt::get(InVal.first->getType(), AmountVal); 316 else if (InVal.first->getType() == llvm::Type::FloatTy) 317 // FIXME: Handle long double. 318 NextVal = 319 llvm::ConstantFP::get(llvm::APFloat(static_cast<float>(AmountVal))); 320 else { 321 // FIXME: Handle long double. 322 assert(InVal.first->getType() == llvm::Type::DoubleTy); 323 NextVal = 324 llvm::ConstantFP::get(llvm::APFloat(static_cast<double>(AmountVal))); 325 } 326 327 // Add the inc/dec to the real part. 328 NextVal = Builder.CreateAdd(InVal.first, NextVal, isInc ? "inc" : "dec"); 329 330 ComplexPairTy IncVal(NextVal, InVal.second); 331 332 // Store the updated result through the lvalue. 333 EmitStoreOfComplex(IncVal, LV.getAddress(), false); /* FIXME: Volatile */ 334 335 // If this is a postinc, return the value read from memory, otherwise use the 336 // updated value. 337 return isPre ? IncVal : InVal; 338 } 339 340 ComplexPairTy ComplexExprEmitter::VisitUnaryMinus(const UnaryOperator *E) { 341 ComplexPairTy Op = Visit(E->getSubExpr()); 342 llvm::Value *ResR = Builder.CreateNeg(Op.first, "neg.r"); 343 llvm::Value *ResI = Builder.CreateNeg(Op.second, "neg.i"); 344 return ComplexPairTy(ResR, ResI); 345 } 346 347 ComplexPairTy ComplexExprEmitter::VisitUnaryNot(const UnaryOperator *E) { 348 // ~(a+ib) = a + i*-b 349 ComplexPairTy Op = Visit(E->getSubExpr()); 350 llvm::Value *ResI = Builder.CreateNeg(Op.second, "conj.i"); 351 return ComplexPairTy(Op.first, ResI); 352 } 353 354 ComplexPairTy ComplexExprEmitter::EmitBinAdd(const BinOpInfo &Op) { 355 llvm::Value *ResR = Builder.CreateAdd(Op.LHS.first, Op.RHS.first, "add.r"); 356 llvm::Value *ResI = Builder.CreateAdd(Op.LHS.second, Op.RHS.second, "add.i"); 357 return ComplexPairTy(ResR, ResI); 358 } 359 360 ComplexPairTy ComplexExprEmitter::EmitBinSub(const BinOpInfo &Op) { 361 llvm::Value *ResR = Builder.CreateSub(Op.LHS.first, Op.RHS.first, "sub.r"); 362 llvm::Value *ResI = Builder.CreateSub(Op.LHS.second, Op.RHS.second, "sub.i"); 363 return ComplexPairTy(ResR, ResI); 364 } 365 366 367 ComplexPairTy ComplexExprEmitter::EmitBinMul(const BinOpInfo &Op) { 368 llvm::Value *ResRl = Builder.CreateMul(Op.LHS.first, Op.RHS.first, "mul.rl"); 369 llvm::Value *ResRr = Builder.CreateMul(Op.LHS.second, Op.RHS.second,"mul.rr"); 370 llvm::Value *ResR = Builder.CreateSub(ResRl, ResRr, "mul.r"); 371 372 llvm::Value *ResIl = Builder.CreateMul(Op.LHS.second, Op.RHS.first, "mul.il"); 373 llvm::Value *ResIr = Builder.CreateMul(Op.LHS.first, Op.RHS.second, "mul.ir"); 374 llvm::Value *ResI = Builder.CreateAdd(ResIl, ResIr, "mul.i"); 375 return ComplexPairTy(ResR, ResI); 376 } 377 378 ComplexPairTy ComplexExprEmitter::EmitBinDiv(const BinOpInfo &Op) { 379 llvm::Value *LHSr = Op.LHS.first, *LHSi = Op.LHS.second; 380 llvm::Value *RHSr = Op.RHS.first, *RHSi = Op.RHS.second; 381 382 // (a+ib) / (c+id) = ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd)) 383 llvm::Value *Tmp1 = Builder.CreateMul(LHSr, RHSr, "tmp"); // a*c 384 llvm::Value *Tmp2 = Builder.CreateMul(LHSi, RHSi, "tmp"); // b*d 385 llvm::Value *Tmp3 = Builder.CreateAdd(Tmp1, Tmp2, "tmp"); // ac+bd 386 387 llvm::Value *Tmp4 = Builder.CreateMul(RHSr, RHSr, "tmp"); // c*c 388 llvm::Value *Tmp5 = Builder.CreateMul(RHSi, RHSi, "tmp"); // d*d 389 llvm::Value *Tmp6 = Builder.CreateAdd(Tmp4, Tmp5, "tmp"); // cc+dd 390 391 llvm::Value *Tmp7 = Builder.CreateMul(LHSi, RHSr, "tmp"); // b*c 392 llvm::Value *Tmp8 = Builder.CreateMul(LHSr, RHSi, "tmp"); // a*d 393 llvm::Value *Tmp9 = Builder.CreateSub(Tmp7, Tmp8, "tmp"); // bc-ad 394 395 llvm::Value *DSTr, *DSTi; 396 if (Tmp3->getType()->isFloatingPoint()) { 397 DSTr = Builder.CreateFDiv(Tmp3, Tmp6, "tmp"); 398 DSTi = Builder.CreateFDiv(Tmp9, Tmp6, "tmp"); 399 } else { 400 if (Op.Ty->getAsComplexType()->getElementType()->isUnsignedIntegerType()) { 401 DSTr = Builder.CreateUDiv(Tmp3, Tmp6, "tmp"); 402 DSTi = Builder.CreateUDiv(Tmp9, Tmp6, "tmp"); 403 } else { 404 DSTr = Builder.CreateSDiv(Tmp3, Tmp6, "tmp"); 405 DSTi = Builder.CreateSDiv(Tmp9, Tmp6, "tmp"); 406 } 407 } 408 409 return ComplexPairTy(DSTr, DSTi); 410 } 411 412 ComplexExprEmitter::BinOpInfo 413 ComplexExprEmitter::EmitBinOps(const BinaryOperator *E) { 414 BinOpInfo Ops; 415 Ops.LHS = Visit(E->getLHS()); 416 Ops.RHS = Visit(E->getRHS()); 417 Ops.Ty = E->getType(); 418 return Ops; 419 } 420 421 422 // Compound assignments. 423 ComplexPairTy ComplexExprEmitter:: 424 EmitCompoundAssign(const CompoundAssignOperator *E, 425 ComplexPairTy (ComplexExprEmitter::*Func)(const BinOpInfo&)){ 426 QualType LHSTy = E->getLHS()->getType(), RHSTy = E->getRHS()->getType(); 427 428 // Load the LHS and RHS operands. 429 LValue LHSLV = CGF.EmitLValue(E->getLHS()); 430 431 BinOpInfo OpInfo; 432 OpInfo.Ty = E->getComputationType(); 433 434 // We know the LHS is a complex lvalue. 435 OpInfo.LHS = EmitLoadOfComplex(LHSLV.getAddress(), false);// FIXME: Volatile. 436 OpInfo.LHS = EmitComplexToComplexCast(OpInfo.LHS, LHSTy, OpInfo.Ty); 437 438 // It is possible for the RHS to be complex or scalar. 439 OpInfo.RHS = EmitCast(E->getRHS(), OpInfo.Ty); 440 441 // Expand the binary operator. 442 ComplexPairTy Result = (this->*Func)(OpInfo); 443 444 // Truncate the result back to the LHS type. 445 Result = EmitComplexToComplexCast(Result, OpInfo.Ty, LHSTy); 446 447 // Store the result value into the LHS lvalue. 448 EmitStoreOfComplex(Result, LHSLV.getAddress(), false); // FIXME: VOLATILE 449 return Result; 450 } 451 452 ComplexPairTy ComplexExprEmitter::VisitBinAssign(const BinaryOperator *E) { 453 assert(CGF.getContext().getCanonicalType(E->getLHS()->getType()) == 454 CGF.getContext().getCanonicalType(E->getRHS()->getType()) && 455 "Invalid assignment"); 456 // Emit the RHS. 457 ComplexPairTy Val = Visit(E->getRHS()); 458 459 // Compute the address to store into. 460 LValue LHS = CGF.EmitLValue(E->getLHS()); 461 462 // Store into it. 463 // FIXME: Volatility! 464 EmitStoreOfComplex(Val, LHS.getAddress(), false); 465 return Val; 466 } 467 468 ComplexPairTy ComplexExprEmitter::VisitBinComma(const BinaryOperator *E) { 469 CGF.EmitStmt(E->getLHS()); 470 CGF.EnsureInsertPoint(); 471 return Visit(E->getRHS()); 472 } 473 474 ComplexPairTy ComplexExprEmitter:: 475 VisitConditionalOperator(const ConditionalOperator *E) { 476 llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true"); 477 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false"); 478 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end"); 479 480 llvm::Value *Cond = CGF.EvaluateExprAsBool(E->getCond()); 481 Builder.CreateCondBr(Cond, LHSBlock, RHSBlock); 482 483 CGF.EmitBlock(LHSBlock); 484 485 // Handle the GNU extension for missing LHS. 486 assert(E->getLHS() && "Must have LHS for complex value"); 487 488 ComplexPairTy LHS = Visit(E->getLHS()); 489 LHSBlock = Builder.GetInsertBlock(); 490 CGF.EmitBranch(ContBlock); 491 492 CGF.EmitBlock(RHSBlock); 493 494 ComplexPairTy RHS = Visit(E->getRHS()); 495 RHSBlock = Builder.GetInsertBlock(); 496 CGF.EmitBranch(ContBlock); 497 498 CGF.EmitBlock(ContBlock); 499 500 // Create a PHI node for the real part. 501 llvm::PHINode *RealPN = Builder.CreatePHI(LHS.first->getType(), "cond.r"); 502 RealPN->reserveOperandSpace(2); 503 RealPN->addIncoming(LHS.first, LHSBlock); 504 RealPN->addIncoming(RHS.first, RHSBlock); 505 506 // Create a PHI node for the imaginary part. 507 llvm::PHINode *ImagPN = Builder.CreatePHI(LHS.first->getType(), "cond.i"); 508 ImagPN->reserveOperandSpace(2); 509 ImagPN->addIncoming(LHS.second, LHSBlock); 510 ImagPN->addIncoming(RHS.second, RHSBlock); 511 512 return ComplexPairTy(RealPN, ImagPN); 513 } 514 515 ComplexPairTy ComplexExprEmitter::VisitChooseExpr(ChooseExpr *E) { 516 // Emit the LHS or RHS as appropriate. 517 return Visit(E->isConditionTrue(CGF.getContext()) ? E->getLHS() :E->getRHS()); 518 } 519 520 ComplexPairTy ComplexExprEmitter::VisitInitListExpr(InitListExpr *E) { 521 if (E->getNumInits()) 522 return Visit(E->getInit(0)); 523 524 // Empty init list intializes to null 525 QualType Ty = E->getType()->getAsComplexType()->getElementType(); 526 const llvm::Type* LTy = CGF.ConvertType(Ty); 527 llvm::Value* zeroConstant = llvm::Constant::getNullValue(LTy); 528 return ComplexPairTy(zeroConstant, zeroConstant); 529 } 530 531 //===----------------------------------------------------------------------===// 532 // Entry Point into this File 533 //===----------------------------------------------------------------------===// 534 535 /// EmitComplexExpr - Emit the computation of the specified expression of 536 /// complex type, ignoring the result. 537 ComplexPairTy CodeGenFunction::EmitComplexExpr(const Expr *E) { 538 assert(E && E->getType()->isAnyComplexType() && 539 "Invalid complex expression to emit"); 540 541 return ComplexExprEmitter(*this).Visit(const_cast<Expr*>(E)); 542 } 543 544 /// EmitComplexExprIntoAddr - Emit the computation of the specified expression 545 /// of complex type, storing into the specified Value*. 546 void CodeGenFunction::EmitComplexExprIntoAddr(const Expr *E, 547 llvm::Value *DestAddr, 548 bool DestIsVolatile) { 549 assert(E && E->getType()->isAnyComplexType() && 550 "Invalid complex expression to emit"); 551 ComplexExprEmitter Emitter(*this); 552 ComplexPairTy Val = Emitter.Visit(const_cast<Expr*>(E)); 553 Emitter.EmitStoreOfComplex(Val, DestAddr, DestIsVolatile); 554 } 555 556 /// StoreComplexToAddr - Store a complex number into the specified address. 557 void CodeGenFunction::StoreComplexToAddr(ComplexPairTy V, 558 llvm::Value *DestAddr, 559 bool DestIsVolatile) { 560 ComplexExprEmitter(*this).EmitStoreOfComplex(V, DestAddr, DestIsVolatile); 561 } 562 563 /// LoadComplexFromAddr - Load a complex number from the specified address. 564 ComplexPairTy CodeGenFunction::LoadComplexFromAddr(llvm::Value *SrcAddr, 565 bool SrcIsVolatile) { 566 return ComplexExprEmitter(*this).EmitLoadOfComplex(SrcAddr, SrcIsVolatile); 567 } 568