diff options
Diffstat (limited to 'enum.c')
-rw-r--r-- | enum.c | 32 |
1 files changed, 28 insertions, 4 deletions
@@ -287,7 +287,7 @@ enum_sort_by(obj) } static VALUE -all_i(i, memo) +all_iter_i(i, memo) VALUE i; NODE *memo; { @@ -299,6 +299,18 @@ all_i(i, memo) } static VALUE +all_i(i, memo) + VALUE i; + NODE *memo; +{ + if (!RTEST(i)) { + memo->u1.value = Qfalse; + rb_iter_break(); + } + return Qnil; +} + +static VALUE enum_all(obj) VALUE obj; { @@ -306,14 +318,14 @@ enum_all(obj) NODE *memo = rb_node_newnode(NODE_MEMO, Qnil, 0, 0); memo->u1.value = Qtrue; - rb_iterate(rb_each, obj, all_i, (VALUE)memo); + rb_iterate(rb_each, obj, rb_block_given_p() ? all_iter_i : all_i, (VALUE)memo); result = memo->u1.value; rb_gc_force_recycle((VALUE)memo); return result; } static VALUE -any_i(i, memo) +any_iter_i(i, memo) VALUE i; NODE *memo; { @@ -325,6 +337,18 @@ any_i(i, memo) } static VALUE +any_i(i, memo) + VALUE i; + NODE *memo; +{ + if (RTEST(i)) { + memo->u1.value = Qtrue; + rb_iter_break(); + } + return Qnil; +} + +static VALUE enum_any(obj) VALUE obj; { @@ -332,7 +356,7 @@ enum_any(obj) NODE *memo = rb_node_newnode(NODE_MEMO, Qnil, 0, 0); memo->u1.value = Qfalse; - rb_iterate(rb_each, obj, any_i, (VALUE)memo); + rb_iterate(rb_each, obj, rb_block_given_p() ? any_iter_i : any_i, (VALUE)memo); result = memo->u1.value; rb_gc_force_recycle((VALUE)memo); return result; |