[ruby-dev:32389] Re: enumerator with single array and multiple arguments.

　ささだです．

Tanaka Akira wrote:
> enumerator を通すとひとつの配列と複数の引数が区別できません。

　こんな感じでどうでしょうか．rb_iterate で積む C で書いたブロック呼
び出し関数で，1.8 だと self を渡していたところを，引数の数を渡すよう
にしてみました．誰も self 使ってないからいいかなぁ，と．


Index: include/ruby/ruby.h
===================================================================
--- include/ruby/ruby.h	(リビジョン 14039)
+++ include/ruby/ruby.h	(作業コピー)
@@ -772,6 +772,7 @@
 VALUE rb_each(VALUE);
 VALUE rb_yield(VALUE);
 VALUE rb_yield_values(int n, ...);
+VALUE rb_yield_values2(int n, VALUE *argv);
 VALUE rb_yield_splat(VALUE);
 int rb_block_given_p(void);
 void rb_need_block(void);
Index: enumerator.c
===================================================================
--- enumerator.c	(リビジョン 14039)
+++ enumerator.c	(作業コピー)
@@ -225,6 +225,17 @@
 }

 static VALUE
+enumerator_each_i(VALUE v, VALUE enum_obj, int argc)
+{
+    if (argc == 1) {
+	return rb_yield(v);
+    }
+    else {
+	return rb_yield_values2(argc, RARRAY_PTR(v));
+    }
+}
+
+static VALUE
 enumerator_init(VALUE enum_obj, VALUE obj, VALUE meth, int argc,
VALUE *argv)
 {
     struct enumerator *ptr = enumerator_ptr(enum_obj);
@@ -235,7 +246,7 @@
 	ptr->iter = enumerator_iter_i;
     }
     else {
-	ptr->iter = (enum_iter *)rb_yield;
+	ptr->iter = (enum_iter *)enumerator_each_i;
     }
     if (argc) ptr->args = rb_ary_new4(argc, argv);
     ptr->fib = 0;
Index: eval.c
===================================================================
--- eval.c	(リビジョン 14039)
+++ eval.c	(作業コピー)
@@ -953,6 +953,12 @@
 }

 VALUE
+rb_yield_values2(int argc, VALUE *argv)
+{
+    return rb_yield_0(argc, argv);
+}
+
+VALUE
 rb_yield_splat(VALUE values)
 {
     VALUE tmp = rb_check_array_type(values);
Index: insnhelper.ci
===================================================================
--- insnhelper.ci	(リビジョン 14039)
+++ insnhelper.ci	(作業コピー)
@@ -661,7 +661,7 @@
 		  self, (VALUE)block->dfp,
 		  0, th->cfp->sp, block->lfp, 1);

-    val = (*ifunc->nd_cfnc) (arg, ifunc->nd_tval, Qnil);
+    val = (*ifunc->nd_cfnc) (arg, ifunc->nd_tval, argc);

     th->cfp++;
     return val;


-- 
// SASADA Koichi at atdot dot net