17a51313dSChris Lattner //===--- CGExprAgg.cpp - Emit LLVM Code from Aggregate Expressions --------===//
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 Aggregate Expr nodes as LLVM code.
117a51313dSChris Lattner //
127a51313dSChris Lattner //===----------------------------------------------------------------------===//
137a51313dSChris Lattner 
147a51313dSChris Lattner #include "CodeGenFunction.h"
157a51313dSChris Lattner #include "CodeGenModule.h"
165f21d2f6SFariborz Jahanian #include "CGObjCRuntime.h"
17ad319a73SDaniel Dunbar #include "clang/AST/ASTContext.h"
18b7f8f594SAnders Carlsson #include "clang/AST/DeclCXX.h"
19ad319a73SDaniel Dunbar #include "clang/AST/StmtVisitor.h"
207a51313dSChris Lattner #include "llvm/Constants.h"
217a51313dSChris Lattner #include "llvm/Function.h"
227a51313dSChris Lattner #include "llvm/GlobalVariable.h"
23579a05d7SChris Lattner #include "llvm/Intrinsics.h"
247a51313dSChris Lattner using namespace clang;
257a51313dSChris Lattner using namespace CodeGen;
267a51313dSChris Lattner 
277a51313dSChris Lattner //===----------------------------------------------------------------------===//
287a51313dSChris Lattner //                        Aggregate Expression Emitter
297a51313dSChris Lattner //===----------------------------------------------------------------------===//
307a51313dSChris Lattner 
317a51313dSChris Lattner namespace  {
32337e3a5fSBenjamin Kramer class AggExprEmitter : public StmtVisitor<AggExprEmitter> {
337a51313dSChris Lattner   CodeGenFunction &CGF;
34cb463859SDaniel Dunbar   CGBuilderTy &Builder;
357a626f63SJohn McCall   AggValueSlot Dest;
36ec3cbfe8SMike Stump   bool IgnoreResult;
3778a15113SJohn McCall 
38a5efa738SJohn McCall   /// We want to use 'dest' as the return slot except under two
39a5efa738SJohn McCall   /// conditions:
40a5efa738SJohn McCall   ///   - The destination slot requires garbage collection, so we
41a5efa738SJohn McCall   ///     need to use the GC API.
42a5efa738SJohn McCall   ///   - The destination slot is potentially aliased.
43a5efa738SJohn McCall   bool shouldUseDestForReturnSlot() const {
44a5efa738SJohn McCall     return !(Dest.requiresGCollection() || Dest.isPotentiallyAliased());
45a5efa738SJohn McCall   }
46a5efa738SJohn McCall 
4778a15113SJohn McCall   ReturnValueSlot getReturnValueSlot() const {
48a5efa738SJohn McCall     if (!shouldUseDestForReturnSlot())
49a5efa738SJohn McCall       return ReturnValueSlot();
50cc04e9f6SJohn McCall 
517a626f63SJohn McCall     return ReturnValueSlot(Dest.getAddr(), Dest.isVolatile());
527a626f63SJohn McCall   }
537a626f63SJohn McCall 
547a626f63SJohn McCall   AggValueSlot EnsureSlot(QualType T) {
557a626f63SJohn McCall     if (!Dest.isIgnored()) return Dest;
567a626f63SJohn McCall     return CGF.CreateAggTemp(T, "agg.tmp.ensured");
5778a15113SJohn McCall   }
58cc04e9f6SJohn McCall 
597a51313dSChris Lattner public:
607a626f63SJohn McCall   AggExprEmitter(CodeGenFunction &cgf, AggValueSlot Dest,
61b60e70f9SFariborz Jahanian                  bool ignore)
627a626f63SJohn McCall     : CGF(cgf), Builder(CGF.Builder), Dest(Dest),
63b60e70f9SFariborz Jahanian       IgnoreResult(ignore) {
647a51313dSChris Lattner   }
657a51313dSChris Lattner 
667a51313dSChris Lattner   //===--------------------------------------------------------------------===//
677a51313dSChris Lattner   //                               Utilities
687a51313dSChris Lattner   //===--------------------------------------------------------------------===//
697a51313dSChris Lattner 
707a51313dSChris Lattner   /// EmitAggLoadOfLValue - Given an expression with aggregate type that
717a51313dSChris Lattner   /// represents a value lvalue, this method emits the address of the lvalue,
727a51313dSChris Lattner   /// then loads the result into DestPtr.
737a51313dSChris Lattner   void EmitAggLoadOfLValue(const Expr *E);
747a51313dSChris Lattner 
75ca9fc09cSMike Stump   /// EmitFinalDestCopy - Perform the final copy to DestPtr, if desired.
76ec3cbfe8SMike Stump   void EmitFinalDestCopy(const Expr *E, LValue Src, bool Ignore = false);
77ec3cbfe8SMike Stump   void EmitFinalDestCopy(const Expr *E, RValue Src, bool Ignore = false);
78ca9fc09cSMike Stump 
79a5efa738SJohn McCall   void EmitMoveFromReturnSlot(const Expr *E, RValue Src);
80cc04e9f6SJohn McCall 
818d6fc958SJohn McCall   AggValueSlot::NeedsGCBarriers_t needsGC(QualType T) {
828d6fc958SJohn McCall     if (CGF.getLangOptions().getGCMode() && TypeRequiresGCollection(T))
838d6fc958SJohn McCall       return AggValueSlot::NeedsGCBarriers;
848d6fc958SJohn McCall     return AggValueSlot::DoesNotNeedGCBarriers;
858d6fc958SJohn McCall   }
868d6fc958SJohn McCall 
87cc04e9f6SJohn McCall   bool TypeRequiresGCollection(QualType T);
88cc04e9f6SJohn McCall 
897a51313dSChris Lattner   //===--------------------------------------------------------------------===//
907a51313dSChris Lattner   //                            Visitor Methods
917a51313dSChris Lattner   //===--------------------------------------------------------------------===//
927a51313dSChris Lattner 
937a51313dSChris Lattner   void VisitStmt(Stmt *S) {
94a7c8cf62SDaniel Dunbar     CGF.ErrorUnsupported(S, "aggregate expression");
957a51313dSChris Lattner   }
967a51313dSChris Lattner   void VisitParenExpr(ParenExpr *PE) { Visit(PE->getSubExpr()); }
9791147596SPeter Collingbourne   void VisitGenericSelectionExpr(GenericSelectionExpr *GE) {
9891147596SPeter Collingbourne     Visit(GE->getResultExpr());
9991147596SPeter Collingbourne   }
1003f66b84cSEli Friedman   void VisitUnaryExtension(UnaryOperator *E) { Visit(E->getSubExpr()); }
1017c454bb8SJohn McCall   void VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *E) {
1027c454bb8SJohn McCall     return Visit(E->getReplacement());
1037c454bb8SJohn McCall   }
1047a51313dSChris Lattner 
1057a51313dSChris Lattner   // l-values.
1067a51313dSChris Lattner   void VisitDeclRefExpr(DeclRefExpr *DRE) { EmitAggLoadOfLValue(DRE); }
1077a51313dSChris Lattner   void VisitMemberExpr(MemberExpr *ME) { EmitAggLoadOfLValue(ME); }
1087a51313dSChris Lattner   void VisitUnaryDeref(UnaryOperator *E) { EmitAggLoadOfLValue(E); }
109d443c0a0SDaniel Dunbar   void VisitStringLiteral(StringLiteral *E) { EmitAggLoadOfLValue(E); }
1109b71f0cfSDouglas Gregor   void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1117a51313dSChris Lattner   void VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
1127a51313dSChris Lattner     EmitAggLoadOfLValue(E);
1137a51313dSChris Lattner   }
1142f343dd5SChris Lattner   void VisitBlockDeclRefExpr(const BlockDeclRefExpr *E) {
1152f343dd5SChris Lattner     EmitAggLoadOfLValue(E);
1162f343dd5SChris Lattner   }
1172f343dd5SChris Lattner   void VisitPredefinedExpr(const PredefinedExpr *E) {
1182f343dd5SChris Lattner     EmitAggLoadOfLValue(E);
1192f343dd5SChris Lattner   }
120bc7d67ceSMike Stump 
1217a51313dSChris Lattner   // Operators.
122ec143777SAnders Carlsson   void VisitCastExpr(CastExpr *E);
1237a51313dSChris Lattner   void VisitCallExpr(const CallExpr *E);
1247a51313dSChris Lattner   void VisitStmtExpr(const StmtExpr *E);
1257a51313dSChris Lattner   void VisitBinaryOperator(const BinaryOperator *BO);
126ffba662dSFariborz Jahanian   void VisitPointerToDataMemberBinaryOperator(const BinaryOperator *BO);
1277a51313dSChris Lattner   void VisitBinAssign(const BinaryOperator *E);
1284b0e2a30SEli Friedman   void VisitBinComma(const BinaryOperator *E);
1297a51313dSChris Lattner 
130b1d329daSChris Lattner   void VisitObjCMessageExpr(ObjCMessageExpr *E);
131c8317a44SDaniel Dunbar   void VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
132c8317a44SDaniel Dunbar     EmitAggLoadOfLValue(E);
133c8317a44SDaniel Dunbar   }
13455310df7SDaniel Dunbar   void VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E);
1357a51313dSChris Lattner 
136c07a0c7eSJohn McCall   void VisitAbstractConditionalOperator(const AbstractConditionalOperator *CO);
1375b2095ceSAnders Carlsson   void VisitChooseExpr(const ChooseExpr *CE);
1387a51313dSChris Lattner   void VisitInitListExpr(InitListExpr *E);
13918ada985SAnders Carlsson   void VisitImplicitValueInitExpr(ImplicitValueInitExpr *E);
140aa9c7aedSChris Lattner   void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
141aa9c7aedSChris Lattner     Visit(DAE->getExpr());
142aa9c7aedSChris Lattner   }
1433be22e27SAnders Carlsson   void VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E);
1441619a504SAnders Carlsson   void VisitCXXConstructExpr(const CXXConstructExpr *E);
1455d413781SJohn McCall   void VisitExprWithCleanups(ExprWithCleanups *E);
146747eb784SDouglas Gregor   void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
1475bbbb137SMike Stump   void VisitCXXTypeidExpr(CXXTypeidExpr *E) { EmitAggLoadOfLValue(E); }
148fe31481fSDouglas Gregor   void VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E);
1491bf5846aSJohn McCall   void VisitOpaqueValueExpr(OpaqueValueExpr *E);
1501bf5846aSJohn McCall 
15121911e89SEli Friedman   void VisitVAArgExpr(VAArgExpr *E);
152579a05d7SChris Lattner 
1531553b190SJohn McCall   void EmitInitializationToLValue(Expr *E, LValue Address);
1541553b190SJohn McCall   void EmitNullInitializationToLValue(LValue Address);
1557a51313dSChris Lattner   //  case Expr::ChooseExprClass:
156f16b8c30SMike Stump   void VisitCXXThrowExpr(const CXXThrowExpr *E) { CGF.EmitCXXThrowExpr(E); }
1577a51313dSChris Lattner };
1587a51313dSChris Lattner }  // end anonymous namespace.
1597a51313dSChris Lattner 
1607a51313dSChris Lattner //===----------------------------------------------------------------------===//
1617a51313dSChris Lattner //                                Utilities
1627a51313dSChris Lattner //===----------------------------------------------------------------------===//
1637a51313dSChris Lattner 
1647a51313dSChris Lattner /// EmitAggLoadOfLValue - Given an expression with aggregate type that
1657a51313dSChris Lattner /// represents a value lvalue, this method emits the address of the lvalue,
1667a51313dSChris Lattner /// then loads the result into DestPtr.
1677a51313dSChris Lattner void AggExprEmitter::EmitAggLoadOfLValue(const Expr *E) {
1687a51313dSChris Lattner   LValue LV = CGF.EmitLValue(E);
169ca9fc09cSMike Stump   EmitFinalDestCopy(E, LV);
170ca9fc09cSMike Stump }
171ca9fc09cSMike Stump 
172cc04e9f6SJohn McCall /// \brief True if the given aggregate type requires special GC API calls.
173cc04e9f6SJohn McCall bool AggExprEmitter::TypeRequiresGCollection(QualType T) {
174cc04e9f6SJohn McCall   // Only record types have members that might require garbage collection.
175cc04e9f6SJohn McCall   const RecordType *RecordTy = T->getAs<RecordType>();
176cc04e9f6SJohn McCall   if (!RecordTy) return false;
177cc04e9f6SJohn McCall 
178cc04e9f6SJohn McCall   // Don't mess with non-trivial C++ types.
179cc04e9f6SJohn McCall   RecordDecl *Record = RecordTy->getDecl();
180cc04e9f6SJohn McCall   if (isa<CXXRecordDecl>(Record) &&
181cc04e9f6SJohn McCall       (!cast<CXXRecordDecl>(Record)->hasTrivialCopyConstructor() ||
182cc04e9f6SJohn McCall        !cast<CXXRecordDecl>(Record)->hasTrivialDestructor()))
183cc04e9f6SJohn McCall     return false;
184cc04e9f6SJohn McCall 
185cc04e9f6SJohn McCall   // Check whether the type has an object member.
186cc04e9f6SJohn McCall   return Record->hasObjectMember();
187cc04e9f6SJohn McCall }
188cc04e9f6SJohn McCall 
189a5efa738SJohn McCall /// \brief Perform the final move to DestPtr if for some reason
190a5efa738SJohn McCall /// getReturnValueSlot() didn't use it directly.
191cc04e9f6SJohn McCall ///
192cc04e9f6SJohn McCall /// The idea is that you do something like this:
193cc04e9f6SJohn McCall ///   RValue Result = EmitSomething(..., getReturnValueSlot());
194a5efa738SJohn McCall ///   EmitMoveFromReturnSlot(E, Result);
195a5efa738SJohn McCall ///
196a5efa738SJohn McCall /// If nothing interferes, this will cause the result to be emitted
197a5efa738SJohn McCall /// directly into the return value slot.  Otherwise, a final move
198a5efa738SJohn McCall /// will be performed.
199a5efa738SJohn McCall void AggExprEmitter::EmitMoveFromReturnSlot(const Expr *E, RValue Src) {
200a5efa738SJohn McCall   if (shouldUseDestForReturnSlot()) {
201a5efa738SJohn McCall     // Logically, Dest.getAddr() should equal Src.getAggregateAddr().
202a5efa738SJohn McCall     // The possibility of undef rvalues complicates that a lot,
203a5efa738SJohn McCall     // though, so we can't really assert.
204a5efa738SJohn McCall     return;
205021510e9SFariborz Jahanian   }
206a5efa738SJohn McCall 
207a5efa738SJohn McCall   // Otherwise, do a final copy,
208a5efa738SJohn McCall   assert(Dest.getAddr() != Src.getAggregateAddr());
209a5efa738SJohn McCall   EmitFinalDestCopy(E, Src, /*Ignore*/ true);
210cc04e9f6SJohn McCall }
211cc04e9f6SJohn McCall 
212ca9fc09cSMike Stump /// EmitFinalDestCopy - Perform the final copy to DestPtr, if desired.
213ec3cbfe8SMike Stump void AggExprEmitter::EmitFinalDestCopy(const Expr *E, RValue Src, bool Ignore) {
214ca9fc09cSMike Stump   assert(Src.isAggregate() && "value must be aggregate value!");
2157a51313dSChris Lattner 
2167a626f63SJohn McCall   // If Dest is ignored, then we're evaluating an aggregate expression
2178d752430SJohn McCall   // in a context (like an expression statement) that doesn't care
2188d752430SJohn McCall   // about the result.  C says that an lvalue-to-rvalue conversion is
2198d752430SJohn McCall   // performed in these cases; C++ says that it is not.  In either
2208d752430SJohn McCall   // case, we don't actually need to do anything unless the value is
2218d752430SJohn McCall   // volatile.
2227a626f63SJohn McCall   if (Dest.isIgnored()) {
2238d752430SJohn McCall     if (!Src.isVolatileQualified() ||
2248d752430SJohn McCall         CGF.CGM.getLangOptions().CPlusPlus ||
2258d752430SJohn McCall         (IgnoreResult && Ignore))
226ec3cbfe8SMike Stump       return;
227c123623dSFariborz Jahanian 
228332ec2ceSMike Stump     // If the source is volatile, we must read from it; to do that, we need
229332ec2ceSMike Stump     // some place to put it.
2307a626f63SJohn McCall     Dest = CGF.CreateAggTemp(E->getType(), "agg.tmp");
231332ec2ceSMike Stump   }
2327a51313dSChris Lattner 
23358649dc6SJohn McCall   if (Dest.requiresGCollection()) {
2343b4bd9a1SKen Dyck     CharUnits size = CGF.getContext().getTypeSizeInChars(E->getType());
2352192fe50SChris Lattner     llvm::Type *SizeTy = CGF.ConvertType(CGF.getContext().getSizeType());
2363b4bd9a1SKen Dyck     llvm::Value *SizeVal = llvm::ConstantInt::get(SizeTy, size.getQuantity());
237879d7266SFariborz Jahanian     CGF.CGM.getObjCRuntime().EmitGCMemmoveCollectable(CGF,
2387a626f63SJohn McCall                                                       Dest.getAddr(),
2397a626f63SJohn McCall                                                       Src.getAggregateAddr(),
240021510e9SFariborz Jahanian                                                       SizeVal);
241879d7266SFariborz Jahanian     return;
242879d7266SFariborz Jahanian   }
243ca9fc09cSMike Stump   // If the result of the assignment is used, copy the LHS there also.
244ca9fc09cSMike Stump   // FIXME: Pass VolatileDest as well.  I think we also need to merge volatile
245ca9fc09cSMike Stump   // from the source as well, as we can't eliminate it if either operand
246ca9fc09cSMike Stump   // is volatile, unless copy has volatile for both source and destination..
2477a626f63SJohn McCall   CGF.EmitAggregateCopy(Dest.getAddr(), Src.getAggregateAddr(), E->getType(),
2487a626f63SJohn McCall                         Dest.isVolatile()|Src.isVolatileQualified());
249ca9fc09cSMike Stump }
250ca9fc09cSMike Stump 
251ca9fc09cSMike Stump /// EmitFinalDestCopy - Perform the final copy to DestPtr, if desired.
252ec3cbfe8SMike Stump void AggExprEmitter::EmitFinalDestCopy(const Expr *E, LValue Src, bool Ignore) {
253ca9fc09cSMike Stump   assert(Src.isSimple() && "Can't have aggregate bitfield, vector, etc");
254ca9fc09cSMike Stump 
255ca9fc09cSMike Stump   EmitFinalDestCopy(E, RValue::getAggregate(Src.getAddress(),
256ec3cbfe8SMike Stump                                             Src.isVolatileQualified()),
257ec3cbfe8SMike Stump                     Ignore);
2587a51313dSChris Lattner }
2597a51313dSChris Lattner 
2607a51313dSChris Lattner //===----------------------------------------------------------------------===//
2617a51313dSChris Lattner //                            Visitor Methods
2627a51313dSChris Lattner //===----------------------------------------------------------------------===//
2637a51313dSChris Lattner 
264fe31481fSDouglas Gregor void AggExprEmitter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E){
265fe31481fSDouglas Gregor   Visit(E->GetTemporaryExpr());
266fe31481fSDouglas Gregor }
267fe31481fSDouglas Gregor 
2681bf5846aSJohn McCall void AggExprEmitter::VisitOpaqueValueExpr(OpaqueValueExpr *e) {
269c07a0c7eSJohn McCall   EmitFinalDestCopy(e, CGF.getOpaqueLValueMapping(e));
2701bf5846aSJohn McCall }
2711bf5846aSJohn McCall 
2729b71f0cfSDouglas Gregor void
2739b71f0cfSDouglas Gregor AggExprEmitter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
2746c9d31ebSDouglas Gregor   if (E->getType().isPODType(CGF.getContext())) {
2756c9d31ebSDouglas Gregor     // For a POD type, just emit a load of the lvalue + a copy, because our
2766c9d31ebSDouglas Gregor     // compound literal might alias the destination.
2776c9d31ebSDouglas Gregor     // FIXME: This is a band-aid; the real problem appears to be in our handling
2786c9d31ebSDouglas Gregor     // of assignments, where we store directly into the LHS without checking
2796c9d31ebSDouglas Gregor     // whether anything in the RHS aliases.
2806c9d31ebSDouglas Gregor     EmitAggLoadOfLValue(E);
2816c9d31ebSDouglas Gregor     return;
2826c9d31ebSDouglas Gregor   }
2836c9d31ebSDouglas Gregor 
2849b71f0cfSDouglas Gregor   AggValueSlot Slot = EnsureSlot(E->getType());
2859b71f0cfSDouglas Gregor   CGF.EmitAggExpr(E->getInitializer(), Slot);
2869b71f0cfSDouglas Gregor }
2879b71f0cfSDouglas Gregor 
2889b71f0cfSDouglas Gregor 
289ec143777SAnders Carlsson void AggExprEmitter::VisitCastExpr(CastExpr *E) {
2901fb7ae9eSAnders Carlsson   switch (E->getCastKind()) {
2918a01a751SAnders Carlsson   case CK_Dynamic: {
2921c073f47SDouglas Gregor     assert(isa<CXXDynamicCastExpr>(E) && "CK_Dynamic without a dynamic_cast?");
2931c073f47SDouglas Gregor     LValue LV = CGF.EmitCheckedLValue(E->getSubExpr());
2941c073f47SDouglas Gregor     // FIXME: Do we also need to handle property references here?
2951c073f47SDouglas Gregor     if (LV.isSimple())
2961c073f47SDouglas Gregor       CGF.EmitDynamicCast(LV.getAddress(), cast<CXXDynamicCastExpr>(E));
2971c073f47SDouglas Gregor     else
2981c073f47SDouglas Gregor       CGF.CGM.ErrorUnsupported(E, "non-simple lvalue dynamic_cast");
2991c073f47SDouglas Gregor 
3007a626f63SJohn McCall     if (!Dest.isIgnored())
3011c073f47SDouglas Gregor       CGF.CGM.ErrorUnsupported(E, "lvalue dynamic_cast with a destination");
3021c073f47SDouglas Gregor     break;
3031c073f47SDouglas Gregor   }
3041c073f47SDouglas Gregor 
305e302792bSJohn McCall   case CK_ToUnion: {
30658989b71SJohn McCall     if (Dest.isIgnored()) break;
30758989b71SJohn McCall 
3087ffcf93bSNuno Lopes     // GCC union extension
3092e442a00SDaniel Dunbar     QualType Ty = E->getSubExpr()->getType();
3102e442a00SDaniel Dunbar     QualType PtrTy = CGF.getContext().getPointerType(Ty);
3117a626f63SJohn McCall     llvm::Value *CastPtr = Builder.CreateBitCast(Dest.getAddr(),
312dd274848SEli Friedman                                                  CGF.ConvertType(PtrTy));
3131553b190SJohn McCall     EmitInitializationToLValue(E->getSubExpr(),
3141553b190SJohn McCall                                CGF.MakeAddrLValue(CastPtr, Ty));
3151fb7ae9eSAnders Carlsson     break;
3167ffcf93bSNuno Lopes   }
3177ffcf93bSNuno Lopes 
318e302792bSJohn McCall   case CK_DerivedToBase:
319e302792bSJohn McCall   case CK_BaseToDerived:
320e302792bSJohn McCall   case CK_UncheckedDerivedToBase: {
321aae38d66SDouglas Gregor     assert(0 && "cannot perform hierarchy conversion in EmitAggExpr: "
322aae38d66SDouglas Gregor                 "should have been unpacked before we got here");
323aae38d66SDouglas Gregor     break;
324aae38d66SDouglas Gregor   }
325aae38d66SDouglas Gregor 
32634376a68SJohn McCall   case CK_GetObjCProperty: {
32734376a68SJohn McCall     LValue LV = CGF.EmitLValue(E->getSubExpr());
32834376a68SJohn McCall     assert(LV.isPropertyRef());
32934376a68SJohn McCall     RValue RV = CGF.EmitLoadOfPropertyRefLValue(LV, getReturnValueSlot());
330a5efa738SJohn McCall     EmitMoveFromReturnSlot(E, RV);
33134376a68SJohn McCall     break;
33234376a68SJohn McCall   }
33334376a68SJohn McCall 
33434376a68SJohn McCall   case CK_LValueToRValue: // hope for downstream optimization
335e302792bSJohn McCall   case CK_NoOp:
336e302792bSJohn McCall   case CK_UserDefinedConversion:
337e302792bSJohn McCall   case CK_ConstructorConversion:
3382a69547fSEli Friedman     assert(CGF.getContext().hasSameUnqualifiedType(E->getSubExpr()->getType(),
3392a69547fSEli Friedman                                                    E->getType()) &&
3400f398c44SChris Lattner            "Implicit cast types must be compatible");
3417a51313dSChris Lattner     Visit(E->getSubExpr());
3421fb7ae9eSAnders Carlsson     break;
343b05a3e55SAnders Carlsson 
344e302792bSJohn McCall   case CK_LValueBitCast:
345f3735e01SJohn McCall     llvm_unreachable("should not be emitting lvalue bitcast as rvalue");
34651954276SDouglas Gregor     break;
34731996343SJohn McCall 
348f3735e01SJohn McCall   case CK_Dependent:
349f3735e01SJohn McCall   case CK_BitCast:
350f3735e01SJohn McCall   case CK_ArrayToPointerDecay:
351f3735e01SJohn McCall   case CK_FunctionToPointerDecay:
352f3735e01SJohn McCall   case CK_NullToPointer:
353f3735e01SJohn McCall   case CK_NullToMemberPointer:
354f3735e01SJohn McCall   case CK_BaseToDerivedMemberPointer:
355f3735e01SJohn McCall   case CK_DerivedToBaseMemberPointer:
356f3735e01SJohn McCall   case CK_MemberPointerToBoolean:
357f3735e01SJohn McCall   case CK_IntegralToPointer:
358f3735e01SJohn McCall   case CK_PointerToIntegral:
359f3735e01SJohn McCall   case CK_PointerToBoolean:
360f3735e01SJohn McCall   case CK_ToVoid:
361f3735e01SJohn McCall   case CK_VectorSplat:
362f3735e01SJohn McCall   case CK_IntegralCast:
363f3735e01SJohn McCall   case CK_IntegralToBoolean:
364f3735e01SJohn McCall   case CK_IntegralToFloating:
365f3735e01SJohn McCall   case CK_FloatingToIntegral:
366f3735e01SJohn McCall   case CK_FloatingToBoolean:
367f3735e01SJohn McCall   case CK_FloatingCast:
368f3735e01SJohn McCall   case CK_AnyPointerToObjCPointerCast:
369f3735e01SJohn McCall   case CK_AnyPointerToBlockPointerCast:
370f3735e01SJohn McCall   case CK_ObjCObjectLValueCast:
371f3735e01SJohn McCall   case CK_FloatingRealToComplex:
372f3735e01SJohn McCall   case CK_FloatingComplexToReal:
373f3735e01SJohn McCall   case CK_FloatingComplexToBoolean:
374f3735e01SJohn McCall   case CK_FloatingComplexCast:
375f3735e01SJohn McCall   case CK_FloatingComplexToIntegralComplex:
376f3735e01SJohn McCall   case CK_IntegralRealToComplex:
377f3735e01SJohn McCall   case CK_IntegralComplexToReal:
378f3735e01SJohn McCall   case CK_IntegralComplexToBoolean:
379f3735e01SJohn McCall   case CK_IntegralComplexCast:
380f3735e01SJohn McCall   case CK_IntegralComplexToFloatingComplex:
38131168b07SJohn McCall   case CK_ObjCProduceObject:
38231168b07SJohn McCall   case CK_ObjCConsumeObject:
3834db5c3c8SJohn McCall   case CK_ObjCReclaimReturnedObject:
384f3735e01SJohn McCall     llvm_unreachable("cast kind invalid for aggregate types");
3851fb7ae9eSAnders Carlsson   }
3867a51313dSChris Lattner }
3877a51313dSChris Lattner 
3880f398c44SChris Lattner void AggExprEmitter::VisitCallExpr(const CallExpr *E) {
389ddcbfe7bSAnders Carlsson   if (E->getCallReturnType()->isReferenceType()) {
390ddcbfe7bSAnders Carlsson     EmitAggLoadOfLValue(E);
391ddcbfe7bSAnders Carlsson     return;
392ddcbfe7bSAnders Carlsson   }
393ddcbfe7bSAnders Carlsson 
394cc04e9f6SJohn McCall   RValue RV = CGF.EmitCallExpr(E, getReturnValueSlot());
395a5efa738SJohn McCall   EmitMoveFromReturnSlot(E, RV);
3967a51313dSChris Lattner }
3970f398c44SChris Lattner 
3980f398c44SChris Lattner void AggExprEmitter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
399cc04e9f6SJohn McCall   RValue RV = CGF.EmitObjCMessageExpr(E, getReturnValueSlot());
400a5efa738SJohn McCall   EmitMoveFromReturnSlot(E, RV);
401b1d329daSChris Lattner }
4027a51313dSChris Lattner 
40355310df7SDaniel Dunbar void AggExprEmitter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
40434376a68SJohn McCall   llvm_unreachable("direct property access not surrounded by "
40534376a68SJohn McCall                    "lvalue-to-rvalue cast");
40655310df7SDaniel Dunbar }
40755310df7SDaniel Dunbar 
4080f398c44SChris Lattner void AggExprEmitter::VisitBinComma(const BinaryOperator *E) {
409a2342eb8SJohn McCall   CGF.EmitIgnoredExpr(E->getLHS());
4107a626f63SJohn McCall   Visit(E->getRHS());
4114b0e2a30SEli Friedman }
4124b0e2a30SEli Friedman 
4137a51313dSChris Lattner void AggExprEmitter::VisitStmtExpr(const StmtExpr *E) {
414ce1de617SJohn McCall   CodeGenFunction::StmtExprEvaluation eval(CGF);
4157a626f63SJohn McCall   CGF.EmitCompoundStmt(*E->getSubStmt(), true, Dest);
4167a51313dSChris Lattner }
4177a51313dSChris Lattner 
4187a51313dSChris Lattner void AggExprEmitter::VisitBinaryOperator(const BinaryOperator *E) {
419e302792bSJohn McCall   if (E->getOpcode() == BO_PtrMemD || E->getOpcode() == BO_PtrMemI)
420ffba662dSFariborz Jahanian     VisitPointerToDataMemberBinaryOperator(E);
421ffba662dSFariborz Jahanian   else
422a7c8cf62SDaniel Dunbar     CGF.ErrorUnsupported(E, "aggregate binary expression");
4237a51313dSChris Lattner }
4247a51313dSChris Lattner 
425ffba662dSFariborz Jahanian void AggExprEmitter::VisitPointerToDataMemberBinaryOperator(
426ffba662dSFariborz Jahanian                                                     const BinaryOperator *E) {
427ffba662dSFariborz Jahanian   LValue LV = CGF.EmitPointerToDataMemberBinaryExpr(E);
428ffba662dSFariborz Jahanian   EmitFinalDestCopy(E, LV);
429ffba662dSFariborz Jahanian }
430ffba662dSFariborz Jahanian 
4317a51313dSChris Lattner void AggExprEmitter::VisitBinAssign(const BinaryOperator *E) {
4327a51313dSChris Lattner   // For an assignment to work, the value on the right has
4337a51313dSChris Lattner   // to be compatible with the value on the left.
4342a69547fSEli Friedman   assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(),
4352a69547fSEli Friedman                                                  E->getRHS()->getType())
4367a51313dSChris Lattner          && "Invalid assignment");
437d0a30016SJohn McCall 
43899514b91SFariborz Jahanian   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E->getLHS()))
43952a8cca5SFariborz Jahanian     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
44099514b91SFariborz Jahanian       if (VD->hasAttr<BlocksAttr>() &&
44199514b91SFariborz Jahanian           E->getRHS()->HasSideEffects(CGF.getContext())) {
44299514b91SFariborz Jahanian         // When __block variable on LHS, the RHS must be evaluated first
44399514b91SFariborz Jahanian         // as it may change the 'forwarding' field via call to Block_copy.
44499514b91SFariborz Jahanian         LValue RHS = CGF.EmitLValue(E->getRHS());
44599514b91SFariborz Jahanian         LValue LHS = CGF.EmitLValue(E->getLHS());
4468d6fc958SJohn McCall         Dest = AggValueSlot::forLValue(LHS, AggValueSlot::IsDestructed,
44746759f4fSJohn McCall                                        needsGC(E->getLHS()->getType()),
44846759f4fSJohn McCall                                        AggValueSlot::IsAliased);
44999514b91SFariborz Jahanian         EmitFinalDestCopy(E, RHS, true);
45099514b91SFariborz Jahanian         return;
45199514b91SFariborz Jahanian       }
45299514b91SFariborz Jahanian 
4537a51313dSChris Lattner   LValue LHS = CGF.EmitLValue(E->getLHS());
4547a51313dSChris Lattner 
4554b8c6db9SDaniel Dunbar   // We have to special case property setters, otherwise we must have
4564b8c6db9SDaniel Dunbar   // a simple lvalue (no aggregates inside vectors, bitfields).
4574b8c6db9SDaniel Dunbar   if (LHS.isPropertyRef()) {
4587a26ba4dSFariborz Jahanian     const ObjCPropertyRefExpr *RE = LHS.getPropertyRefExpr();
4597a26ba4dSFariborz Jahanian     QualType ArgType = RE->getSetterArgType();
4607a26ba4dSFariborz Jahanian     RValue Src;
4617a26ba4dSFariborz Jahanian     if (ArgType->isReferenceType())
4627a26ba4dSFariborz Jahanian       Src = CGF.EmitReferenceBindingToExpr(E->getRHS(), 0);
4637a26ba4dSFariborz Jahanian     else {
4647a626f63SJohn McCall       AggValueSlot Slot = EnsureSlot(E->getRHS()->getType());
4657a626f63SJohn McCall       CGF.EmitAggExpr(E->getRHS(), Slot);
4667a26ba4dSFariborz Jahanian       Src = Slot.asRValue();
4677a26ba4dSFariborz Jahanian     }
4687a26ba4dSFariborz Jahanian     CGF.EmitStoreThroughPropertyRefLValue(Src, LHS);
4694b8c6db9SDaniel Dunbar   } else {
4707a51313dSChris Lattner     // Codegen the RHS so that it stores directly into the LHS.
4718d6fc958SJohn McCall     AggValueSlot LHSSlot =
4728d6fc958SJohn McCall       AggValueSlot::forLValue(LHS, AggValueSlot::IsDestructed,
47346759f4fSJohn McCall                               needsGC(E->getLHS()->getType()),
47446759f4fSJohn McCall                               AggValueSlot::IsAliased);
475b60e70f9SFariborz Jahanian     CGF.EmitAggExpr(E->getRHS(), LHSSlot, false);
476ec3cbfe8SMike Stump     EmitFinalDestCopy(E, LHS, true);
4777a51313dSChris Lattner   }
4784b8c6db9SDaniel Dunbar }
4797a51313dSChris Lattner 
480c07a0c7eSJohn McCall void AggExprEmitter::
481c07a0c7eSJohn McCall VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) {
482a612e79bSDaniel Dunbar   llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true");
483a612e79bSDaniel Dunbar   llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false");
484a612e79bSDaniel Dunbar   llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end");
4857a51313dSChris Lattner 
486c07a0c7eSJohn McCall   // Bind the common expression if necessary.
487c07a0c7eSJohn McCall   CodeGenFunction::OpaqueValueMapping binding(CGF, E);
488c07a0c7eSJohn McCall 
489ce1de617SJohn McCall   CodeGenFunction::ConditionalEvaluation eval(CGF);
490b8841af8SEli Friedman   CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock);
4917a51313dSChris Lattner 
4925b26f65bSJohn McCall   // Save whether the destination's lifetime is externally managed.
493cac93853SJohn McCall   bool isExternallyDestructed = Dest.isExternallyDestructed();
4947a51313dSChris Lattner 
495ce1de617SJohn McCall   eval.begin(CGF);
496ce1de617SJohn McCall   CGF.EmitBlock(LHSBlock);
497c07a0c7eSJohn McCall   Visit(E->getTrueExpr());
498ce1de617SJohn McCall   eval.end(CGF);
4997a51313dSChris Lattner 
500ce1de617SJohn McCall   assert(CGF.HaveInsertPoint() && "expression evaluation ended with no IP!");
501ce1de617SJohn McCall   CGF.Builder.CreateBr(ContBlock);
5027a51313dSChris Lattner 
5035b26f65bSJohn McCall   // If the result of an agg expression is unused, then the emission
5045b26f65bSJohn McCall   // of the LHS might need to create a destination slot.  That's fine
5055b26f65bSJohn McCall   // with us, and we can safely emit the RHS into the same slot, but
506cac93853SJohn McCall   // we shouldn't claim that it's already being destructed.
507cac93853SJohn McCall   Dest.setExternallyDestructed(isExternallyDestructed);
5085b26f65bSJohn McCall 
509ce1de617SJohn McCall   eval.begin(CGF);
510ce1de617SJohn McCall   CGF.EmitBlock(RHSBlock);
511c07a0c7eSJohn McCall   Visit(E->getFalseExpr());
512ce1de617SJohn McCall   eval.end(CGF);
5137a51313dSChris Lattner 
5147a51313dSChris Lattner   CGF.EmitBlock(ContBlock);
5157a51313dSChris Lattner }
5167a51313dSChris Lattner 
5175b2095ceSAnders Carlsson void AggExprEmitter::VisitChooseExpr(const ChooseExpr *CE) {
5185b2095ceSAnders Carlsson   Visit(CE->getChosenSubExpr(CGF.getContext()));
5195b2095ceSAnders Carlsson }
5205b2095ceSAnders Carlsson 
52121911e89SEli Friedman void AggExprEmitter::VisitVAArgExpr(VAArgExpr *VE) {
522e9fcadd2SDaniel Dunbar   llvm::Value *ArgValue = CGF.EmitVAListRef(VE->getSubExpr());
52313abd7e9SAnders Carlsson   llvm::Value *ArgPtr = CGF.EmitVAArg(ArgValue, VE->getType());
52413abd7e9SAnders Carlsson 
525020cddcfSSebastian Redl   if (!ArgPtr) {
52613abd7e9SAnders Carlsson     CGF.ErrorUnsupported(VE, "aggregate va_arg expression");
527020cddcfSSebastian Redl     return;
528020cddcfSSebastian Redl   }
52913abd7e9SAnders Carlsson 
5302e442a00SDaniel Dunbar   EmitFinalDestCopy(VE, CGF.MakeAddrLValue(ArgPtr, VE->getType()));
53121911e89SEli Friedman }
53221911e89SEli Friedman 
5333be22e27SAnders Carlsson void AggExprEmitter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
5347a626f63SJohn McCall   // Ensure that we have a slot, but if we already do, remember
535cac93853SJohn McCall   // whether it was externally destructed.
536cac93853SJohn McCall   bool wasExternallyDestructed = Dest.isExternallyDestructed();
5377a626f63SJohn McCall   Dest = EnsureSlot(E->getType());
538cac93853SJohn McCall 
539cac93853SJohn McCall   // We're going to push a destructor if there isn't already one.
540cac93853SJohn McCall   Dest.setExternallyDestructed();
5413be22e27SAnders Carlsson 
5423be22e27SAnders Carlsson   Visit(E->getSubExpr());
5433be22e27SAnders Carlsson 
544cac93853SJohn McCall   // Push that destructor we promised.
545cac93853SJohn McCall   if (!wasExternallyDestructed)
5467a626f63SJohn McCall     CGF.EmitCXXTemporary(E->getTemporary(), Dest.getAddr());
5473be22e27SAnders Carlsson }
5483be22e27SAnders Carlsson 
549b7f8f594SAnders Carlsson void
5501619a504SAnders Carlsson AggExprEmitter::VisitCXXConstructExpr(const CXXConstructExpr *E) {
5517a626f63SJohn McCall   AggValueSlot Slot = EnsureSlot(E->getType());
5527a626f63SJohn McCall   CGF.EmitCXXConstructExpr(E, Slot);
553c82b86dfSAnders Carlsson }
554c82b86dfSAnders Carlsson 
5555d413781SJohn McCall void AggExprEmitter::VisitExprWithCleanups(ExprWithCleanups *E) {
5565d413781SJohn McCall   CGF.EmitExprWithCleanups(E, Dest);
557b7f8f594SAnders Carlsson }
558b7f8f594SAnders Carlsson 
559747eb784SDouglas Gregor void AggExprEmitter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
5607a626f63SJohn McCall   QualType T = E->getType();
5617a626f63SJohn McCall   AggValueSlot Slot = EnsureSlot(T);
5621553b190SJohn McCall   EmitNullInitializationToLValue(CGF.MakeAddrLValue(Slot.getAddr(), T));
56318ada985SAnders Carlsson }
56418ada985SAnders Carlsson 
56518ada985SAnders Carlsson void AggExprEmitter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
5667a626f63SJohn McCall   QualType T = E->getType();
5677a626f63SJohn McCall   AggValueSlot Slot = EnsureSlot(T);
5681553b190SJohn McCall   EmitNullInitializationToLValue(CGF.MakeAddrLValue(Slot.getAddr(), T));
569ff3507b9SNuno Lopes }
570ff3507b9SNuno Lopes 
57127a3631bSChris Lattner /// isSimpleZero - If emitting this value will obviously just cause a store of
57227a3631bSChris Lattner /// zero to memory, return true.  This can return false if uncertain, so it just
57327a3631bSChris Lattner /// handles simple cases.
57427a3631bSChris Lattner static bool isSimpleZero(const Expr *E, CodeGenFunction &CGF) {
57591147596SPeter Collingbourne   E = E->IgnoreParens();
57691147596SPeter Collingbourne 
57727a3631bSChris Lattner   // 0
57827a3631bSChris Lattner   if (const IntegerLiteral *IL = dyn_cast<IntegerLiteral>(E))
57927a3631bSChris Lattner     return IL->getValue() == 0;
58027a3631bSChris Lattner   // +0.0
58127a3631bSChris Lattner   if (const FloatingLiteral *FL = dyn_cast<FloatingLiteral>(E))
58227a3631bSChris Lattner     return FL->getValue().isPosZero();
58327a3631bSChris Lattner   // int()
58427a3631bSChris Lattner   if ((isa<ImplicitValueInitExpr>(E) || isa<CXXScalarValueInitExpr>(E)) &&
58527a3631bSChris Lattner       CGF.getTypes().isZeroInitializable(E->getType()))
58627a3631bSChris Lattner     return true;
58727a3631bSChris Lattner   // (int*)0 - Null pointer expressions.
58827a3631bSChris Lattner   if (const CastExpr *ICE = dyn_cast<CastExpr>(E))
58927a3631bSChris Lattner     return ICE->getCastKind() == CK_NullToPointer;
59027a3631bSChris Lattner   // '\0'
59127a3631bSChris Lattner   if (const CharacterLiteral *CL = dyn_cast<CharacterLiteral>(E))
59227a3631bSChris Lattner     return CL->getValue() == 0;
59327a3631bSChris Lattner 
59427a3631bSChris Lattner   // Otherwise, hard case: conservatively return false.
59527a3631bSChris Lattner   return false;
59627a3631bSChris Lattner }
59727a3631bSChris Lattner 
59827a3631bSChris Lattner 
599b247350eSAnders Carlsson void
6001553b190SJohn McCall AggExprEmitter::EmitInitializationToLValue(Expr* E, LValue LV) {
6011553b190SJohn McCall   QualType type = LV.getType();
602df0fe27bSMike Stump   // FIXME: Ignore result?
603579a05d7SChris Lattner   // FIXME: Are initializers affected by volatile?
60427a3631bSChris Lattner   if (Dest.isZeroed() && isSimpleZero(E, CGF)) {
60527a3631bSChris Lattner     // Storing "i32 0" to a zero'd memory location is a noop.
60627a3631bSChris Lattner   } else if (isa<ImplicitValueInitExpr>(E)) {
6071553b190SJohn McCall     EmitNullInitializationToLValue(LV);
6081553b190SJohn McCall   } else if (type->isReferenceType()) {
60904775f84SAnders Carlsson     RValue RV = CGF.EmitReferenceBindingToExpr(E, /*InitializedDecl=*/0);
61055e1fbc8SJohn McCall     CGF.EmitStoreThroughLValue(RV, LV);
6111553b190SJohn McCall   } else if (type->isAnyComplexType()) {
6120202cb40SDouglas Gregor     CGF.EmitComplexExprIntoAddr(E, LV.getAddress(), false);
6131553b190SJohn McCall   } else if (CGF.hasAggregateLLVMType(type)) {
6148d6fc958SJohn McCall     CGF.EmitAggExpr(E, AggValueSlot::forLValue(LV,
6158d6fc958SJohn McCall                                                AggValueSlot::IsDestructed,
6168d6fc958SJohn McCall                                       AggValueSlot::DoesNotNeedGCBarriers,
617a5efa738SJohn McCall                                                AggValueSlot::IsNotAliased,
6181553b190SJohn McCall                                                Dest.isZeroed()));
61931168b07SJohn McCall   } else if (LV.isSimple()) {
6201553b190SJohn McCall     CGF.EmitScalarInit(E, /*D=*/0, LV, /*Captured=*/false);
6216e313210SEli Friedman   } else {
62255e1fbc8SJohn McCall     CGF.EmitStoreThroughLValue(RValue::get(CGF.EmitScalarExpr(E)), LV);
6237a51313dSChris Lattner   }
624579a05d7SChris Lattner }
625579a05d7SChris Lattner 
6261553b190SJohn McCall void AggExprEmitter::EmitNullInitializationToLValue(LValue lv) {
6271553b190SJohn McCall   QualType type = lv.getType();
6281553b190SJohn McCall 
62927a3631bSChris Lattner   // If the destination slot is already zeroed out before the aggregate is
63027a3631bSChris Lattner   // copied into it, we don't have to emit any zeros here.
6311553b190SJohn McCall   if (Dest.isZeroed() && CGF.getTypes().isZeroInitializable(type))
63227a3631bSChris Lattner     return;
63327a3631bSChris Lattner 
6341553b190SJohn McCall   if (!CGF.hasAggregateLLVMType(type)) {
635579a05d7SChris Lattner     // For non-aggregates, we can store zero
6361553b190SJohn McCall     llvm::Value *null = llvm::Constant::getNullValue(CGF.ConvertType(type));
63755e1fbc8SJohn McCall     CGF.EmitStoreThroughLValue(RValue::get(null), lv);
638579a05d7SChris Lattner   } else {
639579a05d7SChris Lattner     // There's a potential optimization opportunity in combining
640579a05d7SChris Lattner     // memsets; that would be easy for arrays, but relatively
641579a05d7SChris Lattner     // difficult for structures with the current code.
6421553b190SJohn McCall     CGF.EmitNullInitialization(lv.getAddress(), lv.getType());
643579a05d7SChris Lattner   }
644579a05d7SChris Lattner }
645579a05d7SChris Lattner 
646579a05d7SChris Lattner void AggExprEmitter::VisitInitListExpr(InitListExpr *E) {
647f5d08c9eSEli Friedman #if 0
6486d11ec8cSEli Friedman   // FIXME: Assess perf here?  Figure out what cases are worth optimizing here
6496d11ec8cSEli Friedman   // (Length of globals? Chunks of zeroed-out space?).
650f5d08c9eSEli Friedman   //
65118bb9284SMike Stump   // If we can, prefer a copy from a global; this is a lot less code for long
65218bb9284SMike Stump   // globals, and it's easier for the current optimizers to analyze.
6536d11ec8cSEli Friedman   if (llvm::Constant* C = CGF.CGM.EmitConstantExpr(E, E->getType(), &CGF)) {
654c59bb48eSEli Friedman     llvm::GlobalVariable* GV =
6556d11ec8cSEli Friedman     new llvm::GlobalVariable(CGF.CGM.getModule(), C->getType(), true,
6566d11ec8cSEli Friedman                              llvm::GlobalValue::InternalLinkage, C, "");
6572e442a00SDaniel Dunbar     EmitFinalDestCopy(E, CGF.MakeAddrLValue(GV, E->getType()));
658c59bb48eSEli Friedman     return;
659c59bb48eSEli Friedman   }
660f5d08c9eSEli Friedman #endif
661f53c0968SChris Lattner   if (E->hadArrayRangeDesignator())
662bf7207a1SDouglas Gregor     CGF.ErrorUnsupported(E, "GNU array range designator extension");
663bf7207a1SDouglas Gregor 
6647a626f63SJohn McCall   llvm::Value *DestPtr = Dest.getAddr();
6657a626f63SJohn McCall 
666579a05d7SChris Lattner   // Handle initialization of an array.
667579a05d7SChris Lattner   if (E->getType()->isArrayType()) {
6682192fe50SChris Lattner     llvm::PointerType *APType =
669579a05d7SChris Lattner       cast<llvm::PointerType>(DestPtr->getType());
6702192fe50SChris Lattner     llvm::ArrayType *AType =
671579a05d7SChris Lattner       cast<llvm::ArrayType>(APType->getElementType());
672579a05d7SChris Lattner 
673579a05d7SChris Lattner     uint64_t NumInitElements = E->getNumInits();
674f23b6fa4SEli Friedman 
6750f398c44SChris Lattner     if (E->getNumInits() > 0) {
6760f398c44SChris Lattner       QualType T1 = E->getType();
6770f398c44SChris Lattner       QualType T2 = E->getInit(0)->getType();
6782a69547fSEli Friedman       if (CGF.getContext().hasSameUnqualifiedType(T1, T2)) {
679f23b6fa4SEli Friedman         EmitAggLoadOfLValue(E->getInit(0));
680f23b6fa4SEli Friedman         return;
681f23b6fa4SEli Friedman       }
6820f398c44SChris Lattner     }
683f23b6fa4SEli Friedman 
684579a05d7SChris Lattner     uint64_t NumArrayElements = AType->getNumElements();
68582fe67bbSJohn McCall     assert(NumInitElements <= NumArrayElements);
686579a05d7SChris Lattner 
68782fe67bbSJohn McCall     QualType elementType = E->getType().getCanonicalType();
68882fe67bbSJohn McCall     elementType = CGF.getContext().getQualifiedType(
68982fe67bbSJohn McCall                     cast<ArrayType>(elementType)->getElementType(),
69082fe67bbSJohn McCall                     elementType.getQualifiers() + Dest.getQualifiers());
69182fe67bbSJohn McCall 
69282fe67bbSJohn McCall     // DestPtr is an array*.  Construct an elementType* by drilling
69382fe67bbSJohn McCall     // down a level.
69482fe67bbSJohn McCall     llvm::Value *zero = llvm::ConstantInt::get(CGF.SizeTy, 0);
69582fe67bbSJohn McCall     llvm::Value *indices[] = { zero, zero };
69682fe67bbSJohn McCall     llvm::Value *begin =
697040dd82fSJay Foad       Builder.CreateInBoundsGEP(DestPtr, indices, "arrayinit.begin");
69882fe67bbSJohn McCall 
69982fe67bbSJohn McCall     // Exception safety requires us to destroy all the
70082fe67bbSJohn McCall     // already-constructed members if an initializer throws.
70182fe67bbSJohn McCall     // For that, we'll need an EH cleanup.
70282fe67bbSJohn McCall     QualType::DestructionKind dtorKind = elementType.isDestructedType();
70382fe67bbSJohn McCall     llvm::AllocaInst *endOfInit = 0;
70482fe67bbSJohn McCall     EHScopeStack::stable_iterator cleanup;
70582fe67bbSJohn McCall     if (CGF.needsEHCleanup(dtorKind)) {
70682fe67bbSJohn McCall       // In principle we could tell the cleanup where we are more
70782fe67bbSJohn McCall       // directly, but the control flow can get so varied here that it
70882fe67bbSJohn McCall       // would actually be quite complex.  Therefore we go through an
70982fe67bbSJohn McCall       // alloca.
71082fe67bbSJohn McCall       endOfInit = CGF.CreateTempAlloca(begin->getType(),
71182fe67bbSJohn McCall                                        "arrayinit.endOfInit");
71282fe67bbSJohn McCall       Builder.CreateStore(begin, endOfInit);
713178360e1SJohn McCall       CGF.pushIrregularPartialArrayCleanup(begin, endOfInit, elementType,
714178360e1SJohn McCall                                            CGF.getDestroyer(dtorKind));
71582fe67bbSJohn McCall       cleanup = CGF.EHStack.stable_begin();
71682fe67bbSJohn McCall 
71782fe67bbSJohn McCall     // Otherwise, remember that we didn't need a cleanup.
71882fe67bbSJohn McCall     } else {
71982fe67bbSJohn McCall       dtorKind = QualType::DK_none;
720e07425a5SArgyrios Kyrtzidis     }
721e07425a5SArgyrios Kyrtzidis 
72282fe67bbSJohn McCall     llvm::Value *one = llvm::ConstantInt::get(CGF.SizeTy, 1);
72327a3631bSChris Lattner 
72482fe67bbSJohn McCall     // The 'current element to initialize'.  The invariants on this
72582fe67bbSJohn McCall     // variable are complicated.  Essentially, after each iteration of
72682fe67bbSJohn McCall     // the loop, it points to the last initialized element, except
72782fe67bbSJohn McCall     // that it points to the beginning of the array before any
72882fe67bbSJohn McCall     // elements have been initialized.
72982fe67bbSJohn McCall     llvm::Value *element = begin;
73027a3631bSChris Lattner 
73182fe67bbSJohn McCall     // Emit the explicit initializers.
73282fe67bbSJohn McCall     for (uint64_t i = 0; i != NumInitElements; ++i) {
73382fe67bbSJohn McCall       // Advance to the next element.
734178360e1SJohn McCall       if (i > 0) {
73582fe67bbSJohn McCall         element = Builder.CreateInBoundsGEP(element, one, "arrayinit.element");
73682fe67bbSJohn McCall 
737178360e1SJohn McCall         // Tell the cleanup that it needs to destroy up to this
738178360e1SJohn McCall         // element.  TODO: some of these stores can be trivially
739178360e1SJohn McCall         // observed to be unnecessary.
740178360e1SJohn McCall         if (endOfInit) Builder.CreateStore(element, endOfInit);
741178360e1SJohn McCall       }
742178360e1SJohn McCall 
74382fe67bbSJohn McCall       LValue elementLV = CGF.MakeAddrLValue(element, elementType);
74482fe67bbSJohn McCall       EmitInitializationToLValue(E->getInit(i), elementLV);
74582fe67bbSJohn McCall     }
74682fe67bbSJohn McCall 
74782fe67bbSJohn McCall     // Check whether there's a non-trivial array-fill expression.
74882fe67bbSJohn McCall     // Note that this will be a CXXConstructExpr even if the element
74982fe67bbSJohn McCall     // type is an array (or array of array, etc.) of class type.
75082fe67bbSJohn McCall     Expr *filler = E->getArrayFiller();
75182fe67bbSJohn McCall     bool hasTrivialFiller = true;
75282fe67bbSJohn McCall     if (CXXConstructExpr *cons = dyn_cast_or_null<CXXConstructExpr>(filler)) {
75382fe67bbSJohn McCall       assert(cons->getConstructor()->isDefaultConstructor());
75482fe67bbSJohn McCall       hasTrivialFiller = cons->getConstructor()->isTrivial();
75582fe67bbSJohn McCall     }
75682fe67bbSJohn McCall 
75782fe67bbSJohn McCall     // Any remaining elements need to be zero-initialized, possibly
75882fe67bbSJohn McCall     // using the filler expression.  We can skip this if the we're
75982fe67bbSJohn McCall     // emitting to zeroed memory.
76082fe67bbSJohn McCall     if (NumInitElements != NumArrayElements &&
76182fe67bbSJohn McCall         !(Dest.isZeroed() && hasTrivialFiller &&
76282fe67bbSJohn McCall           CGF.getTypes().isZeroInitializable(elementType))) {
76382fe67bbSJohn McCall 
76482fe67bbSJohn McCall       // Use an actual loop.  This is basically
76582fe67bbSJohn McCall       //   do { *array++ = filler; } while (array != end);
76682fe67bbSJohn McCall 
76782fe67bbSJohn McCall       // Advance to the start of the rest of the array.
768178360e1SJohn McCall       if (NumInitElements) {
76982fe67bbSJohn McCall         element = Builder.CreateInBoundsGEP(element, one, "arrayinit.start");
770178360e1SJohn McCall         if (endOfInit) Builder.CreateStore(element, endOfInit);
771178360e1SJohn McCall       }
77282fe67bbSJohn McCall 
77382fe67bbSJohn McCall       // Compute the end of the array.
77482fe67bbSJohn McCall       llvm::Value *end = Builder.CreateInBoundsGEP(begin,
77582fe67bbSJohn McCall                         llvm::ConstantInt::get(CGF.SizeTy, NumArrayElements),
77682fe67bbSJohn McCall                                                    "arrayinit.end");
77782fe67bbSJohn McCall 
77882fe67bbSJohn McCall       llvm::BasicBlock *entryBB = Builder.GetInsertBlock();
77982fe67bbSJohn McCall       llvm::BasicBlock *bodyBB = CGF.createBasicBlock("arrayinit.body");
78082fe67bbSJohn McCall 
78182fe67bbSJohn McCall       // Jump into the body.
78282fe67bbSJohn McCall       CGF.EmitBlock(bodyBB);
78382fe67bbSJohn McCall       llvm::PHINode *currentElement =
78482fe67bbSJohn McCall         Builder.CreatePHI(element->getType(), 2, "arrayinit.cur");
78582fe67bbSJohn McCall       currentElement->addIncoming(element, entryBB);
78682fe67bbSJohn McCall 
78782fe67bbSJohn McCall       // Emit the actual filler expression.
78882fe67bbSJohn McCall       LValue elementLV = CGF.MakeAddrLValue(currentElement, elementType);
78982fe67bbSJohn McCall       if (filler)
79082fe67bbSJohn McCall         EmitInitializationToLValue(filler, elementLV);
791579a05d7SChris Lattner       else
79282fe67bbSJohn McCall         EmitNullInitializationToLValue(elementLV);
79327a3631bSChris Lattner 
79482fe67bbSJohn McCall       // Move on to the next element.
79582fe67bbSJohn McCall       llvm::Value *nextElement =
79682fe67bbSJohn McCall         Builder.CreateInBoundsGEP(currentElement, one, "arrayinit.next");
79782fe67bbSJohn McCall 
798178360e1SJohn McCall       // Tell the EH cleanup that we finished with the last element.
799178360e1SJohn McCall       if (endOfInit) Builder.CreateStore(nextElement, endOfInit);
800178360e1SJohn McCall 
80182fe67bbSJohn McCall       // Leave the loop if we're done.
80282fe67bbSJohn McCall       llvm::Value *done = Builder.CreateICmpEQ(nextElement, end,
80382fe67bbSJohn McCall                                                "arrayinit.done");
80482fe67bbSJohn McCall       llvm::BasicBlock *endBB = CGF.createBasicBlock("arrayinit.end");
80582fe67bbSJohn McCall       Builder.CreateCondBr(done, endBB, bodyBB);
80682fe67bbSJohn McCall       currentElement->addIncoming(nextElement, Builder.GetInsertBlock());
80782fe67bbSJohn McCall 
80882fe67bbSJohn McCall       CGF.EmitBlock(endBB);
809579a05d7SChris Lattner     }
81082fe67bbSJohn McCall 
81182fe67bbSJohn McCall     // Leave the partial-array cleanup if we entered one.
81282fe67bbSJohn McCall     if (dtorKind) CGF.DeactivateCleanupBlock(cleanup);
81382fe67bbSJohn McCall 
814579a05d7SChris Lattner     return;
815579a05d7SChris Lattner   }
816579a05d7SChris Lattner 
817579a05d7SChris Lattner   assert(E->getType()->isRecordType() && "Only support structs/unions here!");
818579a05d7SChris Lattner 
819579a05d7SChris Lattner   // Do struct initialization; this code just sets each individual member
820579a05d7SChris Lattner   // to the approprate value.  This makes bitfield support automatic;
821579a05d7SChris Lattner   // the disadvantage is that the generated code is more difficult for
822579a05d7SChris Lattner   // the optimizer, especially with bitfields.
823579a05d7SChris Lattner   unsigned NumInitElements = E->getNumInits();
8243b935d33SJohn McCall   RecordDecl *record = E->getType()->castAs<RecordType>()->getDecl();
82552bcf963SChris Lattner 
8263b935d33SJohn McCall   if (record->isUnion()) {
8275169570eSDouglas Gregor     // Only initialize one field of a union. The field itself is
8285169570eSDouglas Gregor     // specified by the initializer list.
8295169570eSDouglas Gregor     if (!E->getInitializedFieldInUnion()) {
8305169570eSDouglas Gregor       // Empty union; we have nothing to do.
8315169570eSDouglas Gregor 
8325169570eSDouglas Gregor #ifndef NDEBUG
8335169570eSDouglas Gregor       // Make sure that it's really an empty and not a failure of
8345169570eSDouglas Gregor       // semantic analysis.
8353b935d33SJohn McCall       for (RecordDecl::field_iterator Field = record->field_begin(),
8363b935d33SJohn McCall                                    FieldEnd = record->field_end();
8375169570eSDouglas Gregor            Field != FieldEnd; ++Field)
8385169570eSDouglas Gregor         assert(Field->isUnnamedBitfield() && "Only unnamed bitfields allowed");
8395169570eSDouglas Gregor #endif
8405169570eSDouglas Gregor       return;
8415169570eSDouglas Gregor     }
8425169570eSDouglas Gregor 
8435169570eSDouglas Gregor     // FIXME: volatility
8445169570eSDouglas Gregor     FieldDecl *Field = E->getInitializedFieldInUnion();
8455169570eSDouglas Gregor 
84627a3631bSChris Lattner     LValue FieldLoc = CGF.EmitLValueForFieldInitialization(DestPtr, Field, 0);
8475169570eSDouglas Gregor     if (NumInitElements) {
8485169570eSDouglas Gregor       // Store the initializer into the field
8491553b190SJohn McCall       EmitInitializationToLValue(E->getInit(0), FieldLoc);
8505169570eSDouglas Gregor     } else {
85127a3631bSChris Lattner       // Default-initialize to null.
8521553b190SJohn McCall       EmitNullInitializationToLValue(FieldLoc);
8535169570eSDouglas Gregor     }
8545169570eSDouglas Gregor 
8555169570eSDouglas Gregor     return;
8565169570eSDouglas Gregor   }
857579a05d7SChris Lattner 
8583b935d33SJohn McCall   // We'll need to enter cleanup scopes in case any of the member
8593b935d33SJohn McCall   // initializers throw an exception.
8600e62c1ccSChris Lattner   SmallVector<EHScopeStack::stable_iterator, 16> cleanups;
8613b935d33SJohn McCall 
862579a05d7SChris Lattner   // Here we iterate over the fields; this makes it simpler to both
863579a05d7SChris Lattner   // default-initialize fields and skip over unnamed fields.
8643b935d33SJohn McCall   unsigned curInitIndex = 0;
8653b935d33SJohn McCall   for (RecordDecl::field_iterator field = record->field_begin(),
8663b935d33SJohn McCall                                fieldEnd = record->field_end();
8673b935d33SJohn McCall        field != fieldEnd; ++field) {
8683b935d33SJohn McCall     // We're done once we hit the flexible array member.
8693b935d33SJohn McCall     if (field->getType()->isIncompleteArrayType())
87091f84216SDouglas Gregor       break;
87191f84216SDouglas Gregor 
8723b935d33SJohn McCall     // Always skip anonymous bitfields.
8733b935d33SJohn McCall     if (field->isUnnamedBitfield())
874579a05d7SChris Lattner       continue;
87517bd094aSDouglas Gregor 
8763b935d33SJohn McCall     // We're done if we reach the end of the explicit initializers, we
8773b935d33SJohn McCall     // have a zeroed object, and the rest of the fields are
8783b935d33SJohn McCall     // zero-initializable.
8793b935d33SJohn McCall     if (curInitIndex == NumInitElements && Dest.isZeroed() &&
88027a3631bSChris Lattner         CGF.getTypes().isZeroInitializable(E->getType()))
88127a3631bSChris Lattner       break;
88227a3631bSChris Lattner 
883327944b3SEli Friedman     // FIXME: volatility
8843b935d33SJohn McCall     LValue LV = CGF.EmitLValueForFieldInitialization(DestPtr, *field, 0);
8857c1baf46SFariborz Jahanian     // We never generate write-barries for initialized fields.
8863b935d33SJohn McCall     LV.setNonGC(true);
88727a3631bSChris Lattner 
8883b935d33SJohn McCall     if (curInitIndex < NumInitElements) {
889e18aaf2cSChris Lattner       // Store the initializer into the field.
8903b935d33SJohn McCall       EmitInitializationToLValue(E->getInit(curInitIndex++), LV);
891579a05d7SChris Lattner     } else {
892579a05d7SChris Lattner       // We're out of initalizers; default-initialize to null
8933b935d33SJohn McCall       EmitNullInitializationToLValue(LV);
8943b935d33SJohn McCall     }
8953b935d33SJohn McCall 
8963b935d33SJohn McCall     // Push a destructor if necessary.
8973b935d33SJohn McCall     // FIXME: if we have an array of structures, all explicitly
8983b935d33SJohn McCall     // initialized, we can end up pushing a linear number of cleanups.
8993b935d33SJohn McCall     bool pushedCleanup = false;
9003b935d33SJohn McCall     if (QualType::DestructionKind dtorKind
9013b935d33SJohn McCall           = field->getType().isDestructedType()) {
9023b935d33SJohn McCall       assert(LV.isSimple());
9033b935d33SJohn McCall       if (CGF.needsEHCleanup(dtorKind)) {
9043b935d33SJohn McCall         CGF.pushDestroy(EHCleanup, LV.getAddress(), field->getType(),
9053b935d33SJohn McCall                         CGF.getDestroyer(dtorKind), false);
9063b935d33SJohn McCall         cleanups.push_back(CGF.EHStack.stable_begin());
9073b935d33SJohn McCall         pushedCleanup = true;
9083b935d33SJohn McCall       }
909579a05d7SChris Lattner     }
91027a3631bSChris Lattner 
91127a3631bSChris Lattner     // If the GEP didn't get used because of a dead zero init or something
91227a3631bSChris Lattner     // else, clean it up for -O0 builds and general tidiness.
9133b935d33SJohn McCall     if (!pushedCleanup && LV.isSimple())
91427a3631bSChris Lattner       if (llvm::GetElementPtrInst *GEP =
9153b935d33SJohn McCall             dyn_cast<llvm::GetElementPtrInst>(LV.getAddress()))
91627a3631bSChris Lattner         if (GEP->use_empty())
91727a3631bSChris Lattner           GEP->eraseFromParent();
9187a51313dSChris Lattner   }
9193b935d33SJohn McCall 
9203b935d33SJohn McCall   // Deactivate all the partial cleanups in reverse order, which
9213b935d33SJohn McCall   // generally means popping them.
9223b935d33SJohn McCall   for (unsigned i = cleanups.size(); i != 0; --i)
9233b935d33SJohn McCall     CGF.DeactivateCleanupBlock(cleanups[i-1]);
9247a51313dSChris Lattner }
9257a51313dSChris Lattner 
9267a51313dSChris Lattner //===----------------------------------------------------------------------===//
9277a51313dSChris Lattner //                        Entry Points into this File
9287a51313dSChris Lattner //===----------------------------------------------------------------------===//
9297a51313dSChris Lattner 
93027a3631bSChris Lattner /// GetNumNonZeroBytesInInit - Get an approximate count of the number of
93127a3631bSChris Lattner /// non-zero bytes that will be stored when outputting the initializer for the
93227a3631bSChris Lattner /// specified initializer expression.
933df94cb7dSKen Dyck static CharUnits GetNumNonZeroBytesInInit(const Expr *E, CodeGenFunction &CGF) {
93491147596SPeter Collingbourne   E = E->IgnoreParens();
93527a3631bSChris Lattner 
93627a3631bSChris Lattner   // 0 and 0.0 won't require any non-zero stores!
937df94cb7dSKen Dyck   if (isSimpleZero(E, CGF)) return CharUnits::Zero();
93827a3631bSChris Lattner 
93927a3631bSChris Lattner   // If this is an initlist expr, sum up the size of sizes of the (present)
94027a3631bSChris Lattner   // elements.  If this is something weird, assume the whole thing is non-zero.
94127a3631bSChris Lattner   const InitListExpr *ILE = dyn_cast<InitListExpr>(E);
94227a3631bSChris Lattner   if (ILE == 0 || !CGF.getTypes().isZeroInitializable(ILE->getType()))
943df94cb7dSKen Dyck     return CGF.getContext().getTypeSizeInChars(E->getType());
94427a3631bSChris Lattner 
945c5cc2fb9SChris Lattner   // InitListExprs for structs have to be handled carefully.  If there are
946c5cc2fb9SChris Lattner   // reference members, we need to consider the size of the reference, not the
947c5cc2fb9SChris Lattner   // referencee.  InitListExprs for unions and arrays can't have references.
9485cd84755SChris Lattner   if (const RecordType *RT = E->getType()->getAs<RecordType>()) {
9495cd84755SChris Lattner     if (!RT->isUnionType()) {
950c5cc2fb9SChris Lattner       RecordDecl *SD = E->getType()->getAs<RecordType>()->getDecl();
951df94cb7dSKen Dyck       CharUnits NumNonZeroBytes = CharUnits::Zero();
952c5cc2fb9SChris Lattner 
953c5cc2fb9SChris Lattner       unsigned ILEElement = 0;
954c5cc2fb9SChris Lattner       for (RecordDecl::field_iterator Field = SD->field_begin(),
955c5cc2fb9SChris Lattner            FieldEnd = SD->field_end(); Field != FieldEnd; ++Field) {
956c5cc2fb9SChris Lattner         // We're done once we hit the flexible array member or run out of
957c5cc2fb9SChris Lattner         // InitListExpr elements.
958c5cc2fb9SChris Lattner         if (Field->getType()->isIncompleteArrayType() ||
959c5cc2fb9SChris Lattner             ILEElement == ILE->getNumInits())
960c5cc2fb9SChris Lattner           break;
961c5cc2fb9SChris Lattner         if (Field->isUnnamedBitfield())
962c5cc2fb9SChris Lattner           continue;
963c5cc2fb9SChris Lattner 
964c5cc2fb9SChris Lattner         const Expr *E = ILE->getInit(ILEElement++);
965c5cc2fb9SChris Lattner 
966c5cc2fb9SChris Lattner         // Reference values are always non-null and have the width of a pointer.
9675cd84755SChris Lattner         if (Field->getType()->isReferenceType())
968df94cb7dSKen Dyck           NumNonZeroBytes += CGF.getContext().toCharUnitsFromBits(
969*e8bbc121SDouglas Gregor               CGF.getContext().getTargetInfo().getPointerWidth(0));
9705cd84755SChris Lattner         else
971c5cc2fb9SChris Lattner           NumNonZeroBytes += GetNumNonZeroBytesInInit(E, CGF);
972c5cc2fb9SChris Lattner       }
973c5cc2fb9SChris Lattner 
974c5cc2fb9SChris Lattner       return NumNonZeroBytes;
975c5cc2fb9SChris Lattner     }
9765cd84755SChris Lattner   }
977c5cc2fb9SChris Lattner 
978c5cc2fb9SChris Lattner 
979df94cb7dSKen Dyck   CharUnits NumNonZeroBytes = CharUnits::Zero();
98027a3631bSChris Lattner   for (unsigned i = 0, e = ILE->getNumInits(); i != e; ++i)
98127a3631bSChris Lattner     NumNonZeroBytes += GetNumNonZeroBytesInInit(ILE->getInit(i), CGF);
98227a3631bSChris Lattner   return NumNonZeroBytes;
98327a3631bSChris Lattner }
98427a3631bSChris Lattner 
98527a3631bSChris Lattner /// CheckAggExprForMemSetUse - If the initializer is large and has a lot of
98627a3631bSChris Lattner /// zeros in it, emit a memset and avoid storing the individual zeros.
98727a3631bSChris Lattner ///
98827a3631bSChris Lattner static void CheckAggExprForMemSetUse(AggValueSlot &Slot, const Expr *E,
98927a3631bSChris Lattner                                      CodeGenFunction &CGF) {
99027a3631bSChris Lattner   // If the slot is already known to be zeroed, nothing to do.  Don't mess with
99127a3631bSChris Lattner   // volatile stores.
99227a3631bSChris Lattner   if (Slot.isZeroed() || Slot.isVolatile() || Slot.getAddr() == 0) return;
99327a3631bSChris Lattner 
99403535265SArgyrios Kyrtzidis   // C++ objects with a user-declared constructor don't need zero'ing.
99503535265SArgyrios Kyrtzidis   if (CGF.getContext().getLangOptions().CPlusPlus)
99603535265SArgyrios Kyrtzidis     if (const RecordType *RT = CGF.getContext()
99703535265SArgyrios Kyrtzidis                        .getBaseElementType(E->getType())->getAs<RecordType>()) {
99803535265SArgyrios Kyrtzidis       const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
99903535265SArgyrios Kyrtzidis       if (RD->hasUserDeclaredConstructor())
100003535265SArgyrios Kyrtzidis         return;
100103535265SArgyrios Kyrtzidis     }
100203535265SArgyrios Kyrtzidis 
100327a3631bSChris Lattner   // If the type is 16-bytes or smaller, prefer individual stores over memset.
1004239a3357SKen Dyck   std::pair<CharUnits, CharUnits> TypeInfo =
1005239a3357SKen Dyck     CGF.getContext().getTypeInfoInChars(E->getType());
1006239a3357SKen Dyck   if (TypeInfo.first <= CharUnits::fromQuantity(16))
100727a3631bSChris Lattner     return;
100827a3631bSChris Lattner 
100927a3631bSChris Lattner   // Check to see if over 3/4 of the initializer are known to be zero.  If so,
101027a3631bSChris Lattner   // we prefer to emit memset + individual stores for the rest.
1011239a3357SKen Dyck   CharUnits NumNonZeroBytes = GetNumNonZeroBytesInInit(E, CGF);
1012239a3357SKen Dyck   if (NumNonZeroBytes*4 > TypeInfo.first)
101327a3631bSChris Lattner     return;
101427a3631bSChris Lattner 
101527a3631bSChris Lattner   // Okay, it seems like a good idea to use an initial memset, emit the call.
1016239a3357SKen Dyck   llvm::Constant *SizeVal = CGF.Builder.getInt64(TypeInfo.first.getQuantity());
1017239a3357SKen Dyck   CharUnits Align = TypeInfo.second;
101827a3631bSChris Lattner 
101927a3631bSChris Lattner   llvm::Value *Loc = Slot.getAddr();
10202192fe50SChris Lattner   llvm::Type *BP = llvm::Type::getInt8PtrTy(CGF.getLLVMContext());
102127a3631bSChris Lattner 
102227a3631bSChris Lattner   Loc = CGF.Builder.CreateBitCast(Loc, BP);
1023239a3357SKen Dyck   CGF.Builder.CreateMemSet(Loc, CGF.Builder.getInt8(0), SizeVal,
1024239a3357SKen Dyck                            Align.getQuantity(), false);
102527a3631bSChris Lattner 
102627a3631bSChris Lattner   // Tell the AggExprEmitter that the slot is known zero.
102727a3631bSChris Lattner   Slot.setZeroed();
102827a3631bSChris Lattner }
102927a3631bSChris Lattner 
103027a3631bSChris Lattner 
103127a3631bSChris Lattner 
103227a3631bSChris Lattner 
103325306cacSMike Stump /// EmitAggExpr - Emit the computation of the specified expression of aggregate
103425306cacSMike Stump /// type.  The result is computed into DestPtr.  Note that if DestPtr is null,
103525306cacSMike Stump /// the value of the aggregate expression is not needed.  If VolatileDest is
103625306cacSMike Stump /// true, DestPtr cannot be 0.
10377a626f63SJohn McCall ///
10387a626f63SJohn McCall /// \param IsInitializer - true if this evaluation is initializing an
10397a626f63SJohn McCall /// object whose lifetime is already being managed.
10407a626f63SJohn McCall void CodeGenFunction::EmitAggExpr(const Expr *E, AggValueSlot Slot,
1041b60e70f9SFariborz Jahanian                                   bool IgnoreResult) {
10427a51313dSChris Lattner   assert(E && hasAggregateLLVMType(E->getType()) &&
10437a51313dSChris Lattner          "Invalid aggregate expression to emit");
104427a3631bSChris Lattner   assert((Slot.getAddr() != 0 || Slot.isIgnored()) &&
104527a3631bSChris Lattner          "slot has bits but no address");
10467a51313dSChris Lattner 
104727a3631bSChris Lattner   // Optimize the slot if possible.
104827a3631bSChris Lattner   CheckAggExprForMemSetUse(Slot, E, *this);
104927a3631bSChris Lattner 
105027a3631bSChris Lattner   AggExprEmitter(*this, Slot, IgnoreResult).Visit(const_cast<Expr*>(E));
10517a51313dSChris Lattner }
10520bc8e86dSDaniel Dunbar 
1053d0bc7b9dSDaniel Dunbar LValue CodeGenFunction::EmitAggExprToLValue(const Expr *E) {
1054d0bc7b9dSDaniel Dunbar   assert(hasAggregateLLVMType(E->getType()) && "Invalid argument!");
1055a7566f16SDaniel Dunbar   llvm::Value *Temp = CreateMemTemp(E->getType());
10562e442a00SDaniel Dunbar   LValue LV = MakeAddrLValue(Temp, E->getType());
10578d6fc958SJohn McCall   EmitAggExpr(E, AggValueSlot::forLValue(LV, AggValueSlot::IsNotDestructed,
105846759f4fSJohn McCall                                          AggValueSlot::DoesNotNeedGCBarriers,
105946759f4fSJohn McCall                                          AggValueSlot::IsNotAliased));
10602e442a00SDaniel Dunbar   return LV;
1061d0bc7b9dSDaniel Dunbar }
1062d0bc7b9dSDaniel Dunbar 
10630bc8e86dSDaniel Dunbar void CodeGenFunction::EmitAggregateCopy(llvm::Value *DestPtr,
10645e9e61b8SMike Stump                                         llvm::Value *SrcPtr, QualType Ty,
10655e9e61b8SMike Stump                                         bool isVolatile) {
10660bc8e86dSDaniel Dunbar   assert(!Ty->isAnyComplexType() && "Shouldn't happen for complex");
10670bc8e86dSDaniel Dunbar 
106816e94af6SAnders Carlsson   if (getContext().getLangOptions().CPlusPlus) {
106916e94af6SAnders Carlsson     if (const RecordType *RT = Ty->getAs<RecordType>()) {
1070f22101a0SDouglas Gregor       CXXRecordDecl *Record = cast<CXXRecordDecl>(RT->getDecl());
1071f22101a0SDouglas Gregor       assert((Record->hasTrivialCopyConstructor() ||
10726855ba2cSFariborz Jahanian               Record->hasTrivialCopyAssignment()) &&
1073f22101a0SDouglas Gregor              "Trying to aggregate-copy a type without a trivial copy "
1074f22101a0SDouglas Gregor              "constructor or assignment operator");
1075265b8b8dSDouglas Gregor       // Ignore empty classes in C++.
1076f22101a0SDouglas Gregor       if (Record->isEmpty())
107716e94af6SAnders Carlsson         return;
107816e94af6SAnders Carlsson     }
107916e94af6SAnders Carlsson   }
108016e94af6SAnders Carlsson 
1081ca05dfefSChris Lattner   // Aggregate assignment turns into llvm.memcpy.  This is almost valid per
10823ef668c2SChris Lattner   // C99 6.5.16.1p3, which states "If the value being stored in an object is
10833ef668c2SChris Lattner   // read from another object that overlaps in anyway the storage of the first
10843ef668c2SChris Lattner   // object, then the overlap shall be exact and the two objects shall have
10853ef668c2SChris Lattner   // qualified or unqualified versions of a compatible type."
10863ef668c2SChris Lattner   //
1087ca05dfefSChris Lattner   // memcpy is not defined if the source and destination pointers are exactly
10883ef668c2SChris Lattner   // equal, but other compilers do this optimization, and almost every memcpy
10893ef668c2SChris Lattner   // implementation handles this case safely.  If there is a libc that does not
10903ef668c2SChris Lattner   // safely handle this, we can add a target hook.
10910bc8e86dSDaniel Dunbar 
10920bc8e86dSDaniel Dunbar   // Get size and alignment info for this aggregate.
1093bb2c2400SKen Dyck   std::pair<CharUnits, CharUnits> TypeInfo =
1094bb2c2400SKen Dyck     getContext().getTypeInfoInChars(Ty);
10950bc8e86dSDaniel Dunbar 
10960bc8e86dSDaniel Dunbar   // FIXME: Handle variable sized types.
10970bc8e86dSDaniel Dunbar 
109886736572SMike Stump   // FIXME: If we have a volatile struct, the optimizer can remove what might
109986736572SMike Stump   // appear to be `extra' memory ops:
110086736572SMike Stump   //
110186736572SMike Stump   // volatile struct { int i; } a, b;
110286736572SMike Stump   //
110386736572SMike Stump   // int main() {
110486736572SMike Stump   //   a = b;
110586736572SMike Stump   //   a = b;
110686736572SMike Stump   // }
110786736572SMike Stump   //
1108cc2ab0cdSMon P Wang   // we need to use a different call here.  We use isVolatile to indicate when
1109ec3cbfe8SMike Stump   // either the source or the destination is volatile.
1110cc2ab0cdSMon P Wang 
11112192fe50SChris Lattner   llvm::PointerType *DPT = cast<llvm::PointerType>(DestPtr->getType());
11122192fe50SChris Lattner   llvm::Type *DBP =
1113ad7c5c16SJohn McCall     llvm::Type::getInt8PtrTy(getLLVMContext(), DPT->getAddressSpace());
1114cc2ab0cdSMon P Wang   DestPtr = Builder.CreateBitCast(DestPtr, DBP, "tmp");
1115cc2ab0cdSMon P Wang 
11162192fe50SChris Lattner   llvm::PointerType *SPT = cast<llvm::PointerType>(SrcPtr->getType());
11172192fe50SChris Lattner   llvm::Type *SBP =
1118ad7c5c16SJohn McCall     llvm::Type::getInt8PtrTy(getLLVMContext(), SPT->getAddressSpace());
1119cc2ab0cdSMon P Wang   SrcPtr = Builder.CreateBitCast(SrcPtr, SBP, "tmp");
1120cc2ab0cdSMon P Wang 
112131168b07SJohn McCall   // Don't do any of the memmove_collectable tests if GC isn't set.
112231168b07SJohn McCall   if (CGM.getLangOptions().getGCMode() == LangOptions::NonGC) {
112331168b07SJohn McCall     // fall through
112431168b07SJohn McCall   } else if (const RecordType *RecordTy = Ty->getAs<RecordType>()) {
1125021510e9SFariborz Jahanian     RecordDecl *Record = RecordTy->getDecl();
1126021510e9SFariborz Jahanian     if (Record->hasObjectMember()) {
1127bb2c2400SKen Dyck       CharUnits size = TypeInfo.first;
11282192fe50SChris Lattner       llvm::Type *SizeTy = ConvertType(getContext().getSizeType());
1129bb2c2400SKen Dyck       llvm::Value *SizeVal = llvm::ConstantInt::get(SizeTy, size.getQuantity());
1130021510e9SFariborz Jahanian       CGM.getObjCRuntime().EmitGCMemmoveCollectable(*this, DestPtr, SrcPtr,
1131021510e9SFariborz Jahanian                                                     SizeVal);
1132021510e9SFariborz Jahanian       return;
1133021510e9SFariborz Jahanian     }
113431168b07SJohn McCall   } else if (Ty->isArrayType()) {
1135021510e9SFariborz Jahanian     QualType BaseType = getContext().getBaseElementType(Ty);
1136021510e9SFariborz Jahanian     if (const RecordType *RecordTy = BaseType->getAs<RecordType>()) {
1137021510e9SFariborz Jahanian       if (RecordTy->getDecl()->hasObjectMember()) {
1138bb2c2400SKen Dyck         CharUnits size = TypeInfo.first;
11392192fe50SChris Lattner         llvm::Type *SizeTy = ConvertType(getContext().getSizeType());
1140bb2c2400SKen Dyck         llvm::Value *SizeVal =
1141bb2c2400SKen Dyck           llvm::ConstantInt::get(SizeTy, size.getQuantity());
1142021510e9SFariborz Jahanian         CGM.getObjCRuntime().EmitGCMemmoveCollectable(*this, DestPtr, SrcPtr,
1143021510e9SFariborz Jahanian                                                       SizeVal);
1144021510e9SFariborz Jahanian         return;
1145021510e9SFariborz Jahanian       }
1146021510e9SFariborz Jahanian     }
1147021510e9SFariborz Jahanian   }
1148021510e9SFariborz Jahanian 
1149acc6b4e2SBenjamin Kramer   Builder.CreateMemCpy(DestPtr, SrcPtr,
1150bb2c2400SKen Dyck                        llvm::ConstantInt::get(IntPtrTy,
1151bb2c2400SKen Dyck                                               TypeInfo.first.getQuantity()),
1152bb2c2400SKen Dyck                        TypeInfo.second.getQuantity(), isVolatile);
11530bc8e86dSDaniel Dunbar }
1154