From 032861ade7bdbe2c7e939b9b2995ed2ad1a1cdc6 Mon Sep 17 00:00:00 2001
From: shugo <shugo@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Thu, 15 Mar 2012 09:25:03 +0000
Subject: * enumerator.c (lazy_zip, lazy_cycle): Enumerator::Lazy#{zip,cycle}  
 should be eager when a block is given, to be consistent with  
 Enumerable#{zip,cycle}.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@35041 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 enumerator.c | 31 ++++++++-----------------------
 1 file changed, 8 insertions(+), 23 deletions(-)

(limited to 'enumerator.c')

diff --git a/enumerator.c b/enumerator.c
index 5f0cecb23a..ea4d4fcbe4 100644
--- a/enumerator.c
+++ b/enumerator.c
@@ -1373,24 +1373,6 @@ lazy_grep(VALUE obj, VALUE pattern)
     return rb_block_call(rb_cLazy, id_new, 1, &obj, lazy_grep_func, pattern);
 }
 
-static VALUE
-lazy_zip_func_i(VALUE val, VALUE arg, int argc, VALUE *argv)
-{
-    VALUE yielder, ary, v, result;
-    long i;
-
-    yielder = argv[0];
-    ary = rb_ary_new2(RARRAY_LEN(arg) + 1);
-    rb_ary_push(ary, argv[1]);
-    for (i = 0; i < RARRAY_LEN(arg); i++) {
-	v = rb_funcall(RARRAY_PTR(arg)[i], id_next, 0);
-	rb_ary_push(ary, v);
-    }
-    result = rb_yield(ary);
-    rb_funcall(yielder, id_yield, 1, result);
-    return Qnil;
-}
-
 static VALUE
 lazy_zip_func(VALUE val, VALUE arg, int argc, VALUE *argv)
 {
@@ -1414,14 +1396,15 @@ lazy_zip(int argc, VALUE *argv, VALUE obj)
     VALUE ary;
     int i;
 
+    if (rb_block_given_p()) {
+	return rb_call_super(argc, argv);
+    }
     ary = rb_ary_new2(argc);
     for (i = 0; i < argc; i++) {
 	rb_ary_push(ary, rb_funcall(argv[i], id_lazy, 0));
     }
 
-    return rb_block_call(rb_cLazy, id_new, 1, &obj,
-			 rb_block_given_p() ? lazy_zip_func_i :  lazy_zip_func,
-			 ary);
+    return rb_block_call(rb_cLazy, id_new, 1, &obj, lazy_zip_func, ary);
 }
 
 static VALUE
@@ -1528,6 +1511,9 @@ lazy_cycle(int argc, VALUE *argv, VALUE obj)
     VALUE args;
     int len = rb_long2int((long)argc + 2);
 
+    if (rb_block_given_p()) {
+	return rb_call_super(argc, argv);
+    }
     args = rb_ary_tmp_new(len);
     rb_ary_push(args, obj);
     rb_ary_push(args, sym_cycle);
@@ -1535,8 +1521,7 @@ lazy_cycle(int argc, VALUE *argv, VALUE obj)
 	rb_ary_cat(args, argv, argc);
     }
     return rb_block_call(rb_cLazy, id_new, len, RARRAY_PTR(args),
-			 rb_block_given_p() ? lazy_map_func : lazy_cycle_func,
-			 args /* prevent from GC */);
+			 lazy_cycle_func, args /* prevent from GC */);
 }
 
 static VALUE
-- 
cgit v1.2.3