[cfe-commits] r151603 - in /cfe/trunk/lib/CodeGen: CGExprScalar.cpp CGObjC.cpp CodeGenFunction.h

[Eli Friedman]

Author: efriedma
Date: Mon Feb 27 19:08:45 2012
New Revision: 151603

URL: http://llvm.org/viewvc/llvm-project?rev=151603&view=rev
Log:
Implement IRGen for the retain-autorelease in the lambda conversion-to-block-pointer outside of ARC.  Testcases coming up soon.

Modified:
    cfe/trunk/lib/CodeGen/CGExprScalar.cpp
    cfe/trunk/lib/CodeGen/CGObjC.cpp
    cfe/trunk/lib/CodeGen/CodeGenFunction.h

Modified: cfe/trunk/lib/CodeGen/CGExprScalar.cpp
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/lib/CodeGen/CGExprScalar.cpp?rev=151603&r1=151602&r2=151603&view=diff
==============================================================================
--- cfe/trunk/lib/CodeGen/CGExprScalar.cpp (original)
+++ cfe/trunk/lib/CodeGen/CGExprScalar.cpp Mon Feb 27 19:08:45 2012
@@ -1149,8 +1149,7 @@
     return CGF.EmitARCExtendBlockObject(E);
 
   case CK_CopyAndAutoreleaseBlockObject:
-    CGF.ErrorUnsupported(E, "copy/autorelease block object");
-    return 0;
+    return CGF.EmitBlockCopyAndAutorelease(Visit(E), E->getType());
       
   case CK_FloatingRealToComplex:
   case CK_FloatingComplexCast:

Modified: cfe/trunk/lib/CodeGen/CGObjC.cpp
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/lib/CodeGen/CGObjC.cpp?rev=151603&r1=151602&r2=151603&view=diff
==============================================================================
--- cfe/trunk/lib/CodeGen/CGObjC.cpp (original)
+++ cfe/trunk/lib/CodeGen/CGObjC.cpp Mon Feb 27 19:08:45 2012
@@ -2774,5 +2774,30 @@
   return HelperFn;
 }
 
+llvm::Value *
+CodeGenFunction::EmitBlockCopyAndAutorelease(llvm::Value *Block, QualType Ty) {
+  // Get selectors for retain/autorelease.
+  IdentifierInfo *RetainID = &getContext().Idents.get("retain");
+  Selector RetainSelector =
+      getContext().Selectors.getNullarySelector(RetainID);
+  IdentifierInfo *AutoreleaseID = &getContext().Idents.get("autorelease");
+  Selector AutoreleaseSelector =
+      getContext().Selectors.getNullarySelector(AutoreleaseID);
+
+  // Emit calls to retain/autorelease.
+  CGObjCRuntime &Runtime = CGM.getObjCRuntime();
+  llvm::Value *Val = Block;
+  RValue Result;
+  Result = Runtime.GenerateMessageSend(*this, ReturnValueSlot(),
+                                       Ty, RetainSelector,
+                                       Val, CallArgList(), 0, 0);
+  Val = Result.getScalarVal();
+  Result = Runtime.GenerateMessageSend(*this, ReturnValueSlot(),
+                                       Ty, AutoreleaseSelector,
+                                       Val, CallArgList(), 0, 0);
+  Val = Result.getScalarVal();
+  return Val;
+}
+
 
 CGObjCRuntime::~CGObjCRuntime() {}

Modified: cfe/trunk/lib/CodeGen/CodeGenFunction.h
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/lib/CodeGen/CodeGenFunction.h?rev=151603&r1=151602&r2=151603&view=diff
==============================================================================
--- cfe/trunk/lib/CodeGen/CodeGenFunction.h (original)
+++ cfe/trunk/lib/CodeGen/CodeGenFunction.h Mon Feb 27 19:08:45 2012
@@ -1345,6 +1345,7 @@
                                              const ObjCPropertyImplDecl *PID);
   llvm::Constant *GenerateObjCAtomicGetterCopyHelperFunction(
                                              const ObjCPropertyImplDecl *PID);
+  llvm::Value *EmitBlockCopyAndAutorelease(llvm::Value *Block, QualType Ty);
 
   void BuildBlockRelease(llvm::Value *DeclPtr, BlockFieldFlags flags);