From 2bb8270e6f149cf56e2c5ea0c71cf7b9c02e7f74 Mon Sep 17 00:00:00 2001
From: Fariborz Jahanian <fjahanian@apple.com>
Date: Fri, 2 Sep 2011 21:33:44 +0000
Subject: [PATCH] blocks: Support capturing complex variable in block. //
 rdar://10033896

llvm-svn: 139041
---
 clang/lib/CodeGen/CGExprComplex.cpp           |  1 +
 .../CodeGen/capture-complex-expr-in-block.c   | 20 +++++++++++++++++++
 2 files changed, 21 insertions(+)
 create mode 100644 clang/test/CodeGen/capture-complex-expr-in-block.c

diff --git a/clang/lib/CodeGen/CGExprComplex.cpp b/clang/lib/CodeGen/CGExprComplex.cpp
index 1a27d1322bc6..c7b9a3071659 100644
--- a/clang/lib/CodeGen/CGExprComplex.cpp
+++ b/clang/lib/CodeGen/CGExprComplex.cpp
@@ -119,6 +119,7 @@ public:
 
   // l-values.
   ComplexPairTy VisitDeclRefExpr(const Expr *E) { return EmitLoadOfLValue(E); }
+  ComplexPairTy VisitBlockDeclRefExpr(const Expr *E) { return EmitLoadOfLValue(E); }
   ComplexPairTy VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
     return EmitLoadOfLValue(E);
   }
diff --git a/clang/test/CodeGen/capture-complex-expr-in-block.c b/clang/test/CodeGen/capture-complex-expr-in-block.c
new file mode 100644
index 000000000000..c38823707c28
--- /dev/null
+++ b/clang/test/CodeGen/capture-complex-expr-in-block.c
@@ -0,0 +1,20 @@
+// RUN: %clang_cc1 %s -emit-llvm -o - -fblocks -triple x86_64-apple-darwin10 | FileCheck %s
+// rdar://10033986
+
+typedef void (^BLOCK)(void);
+int main ()
+{
+    _Complex double c;
+    BLOCK b =  ^() {
+      _Complex double z;
+      z = z + c;
+    };
+    b();
+}
+
+// CHECK: define internal void @__main_block_invoke_0
+// CHECK:  [[C1:%.*]] = alloca { double, double }, align 8
+// CHECK:  [[C1]].realp = getelementptr inbounds { double, double }* [[C1]], i32 0, i32 0
+// CHECK-NEXT:  [[C1]].real = load double* [[C1]].realp
+// CHECK-NEXT:  [[C1]].imagp = getelementptr inbounds { double, double }* [[C1]], i32 0, i32 1
+// CHECK-NEXT:  [[C1]].imag = load double* [[C1]].imagp