diff --git a/ChangeLog b/ChangeLog index 51eb158b86..55a2a97726 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,10 @@ +Fri Apr 11 16:44:43 2008 Akinori MUSHA + + * enum.c (enum_find_index): Add support for find_index(obj); + [ruby-dev:34313]. + + * array.c (rb_ary_index): Define find_index as an alias to index. + Fri Apr 11 16:42:33 2008 Yukihiro Matsumoto * lib/yaml/store.rb (YAML::load): modified to support empty diff --git a/array.c b/array.c index 5852347584..577953a389 100644 --- a/array.c +++ b/array.c @@ -877,6 +877,8 @@ rb_ary_fetch(int argc, VALUE *argv, VALUE ary) * a.index("b") #=> 1 * a.index("z") #=> nil * a.index{|x|x=="b"} #=> 1 + * + * This is an alias of #find_index. */ static VALUE @@ -3378,6 +3380,7 @@ Init_Array(void) rb_define_method(rb_cArray, "length", rb_ary_length, 0); rb_define_alias(rb_cArray, "size", "length"); rb_define_method(rb_cArray, "empty?", rb_ary_empty_p, 0); + rb_define_method(rb_cArray, "find_index", rb_ary_index, -1); rb_define_method(rb_cArray, "index", rb_ary_index, -1); rb_define_method(rb_cArray, "rindex", rb_ary_rindex, -1); rb_define_method(rb_cArray, "join", rb_ary_join_m, -1); diff --git a/enum.c b/enum.c index b09e82623f..d4de80d7b5 100644 --- a/enum.c +++ b/enum.c @@ -186,7 +186,18 @@ enum_find(int argc, VALUE *argv, VALUE obj) } static VALUE -find_index_i(VALUE i, VALUE *memo, int argc, VALUE *argv) +find_index_i(VALUE i, VALUE *memo) +{ + if (rb_equal(i, memo[2])) { + memo[0] = UINT2NUM(memo[1]); + rb_iter_break(); + } + memo[1]++; + return Qnil; +} + +static VALUE +find_index_iter_i(VALUE i, VALUE *memo, int argc, VALUE *argv) { if (RTEST(enum_yield(argc, argv))) { memo[0] = UINT2NUM(memo[1]); @@ -198,30 +209,48 @@ find_index_i(VALUE i, VALUE *memo, int argc, VALUE *argv) /* * call-seq: - * enum.find_index() {| obj | block } => int + * enum.find_index(value) => int or nil + * enum.find_index {| obj | block } => int or nil * - * Passes each entry in enum to block. Returns the - * index for the first for which block is not false. - * If no object matches, returns nil + * Compares each entry in enum with value or passes + * to block. Returns the index for the first for which the + * evaluated value is non-false. If no object matches, returns + * nil * * (1..10).find_index {|i| i % 5 == 0 and i % 7 == 0 } #=> nil * (1..100).find_index {|i| i % 5 == 0 and i % 7 == 0 } #=> 34 + * (1..100).find_index(50) #=> 49 * */ static VALUE -enum_find_index(VALUE obj) +enum_find_index(int argc, VALUE *argv, VALUE obj) { - VALUE memo[2]; + VALUE memo[3]; /* [return value, current index, condition value] */ + rb_block_call_func *func; - RETURN_ENUMERATOR(obj, 0, 0); - memo[0] = Qundef; - memo[1] = 0; - rb_block_call(obj, id_each, 0, 0, find_index_i, (VALUE)memo); - if (memo[0] != Qundef) { - return memo[0]; + if (argc == 0) { + RETURN_ENUMERATOR(obj, 0, 0); + memo[0] = Qnil; + memo[1] = 0; + memo[2] = Qundef; + func = find_index_iter_i; } - return Qnil; + else { + VALUE item; + + rb_scan_args(argc, argv, "1", &item); + if (rb_block_given_p()) { + rb_warn("given block not used"); + } + memo[0] = Qnil; + memo[1] = 0; + memo[2] = item; + func = find_index_i; + } + + rb_block_call(obj, id_each, 0, 0, func, (VALUE)memo); + return memo[0]; } static VALUE @@ -1688,7 +1717,7 @@ Init_Enumerable(void) rb_define_method(rb_mEnumerable,"count", enum_count, -1); rb_define_method(rb_mEnumerable,"find", enum_find, -1); rb_define_method(rb_mEnumerable,"detect", enum_find, -1); - rb_define_method(rb_mEnumerable,"find_index", enum_find_index, 0); + rb_define_method(rb_mEnumerable,"find_index", enum_find_index, -1); rb_define_method(rb_mEnumerable,"find_all", enum_find_all, 0); rb_define_method(rb_mEnumerable,"select", enum_find_all, 0); rb_define_method(rb_mEnumerable,"reject", enum_reject, 0); diff --git a/test/ruby/test_enum.rb b/test/ruby/test_enum.rb index 74f6e4e466..97b4ec468b 100644 --- a/test/ruby/test_enum.rb +++ b/test/ruby/test_enum.rb @@ -43,8 +43,10 @@ class TestEnumerable < Test::Unit::TestCase end def test_find_index + assert_equal(1, @obj.find_index(2)) assert_equal(1, @obj.find_index {|x| x % 2 == 0 }) assert_equal(nil, @obj.find_index {|x| false }) + assert_raise(ArgumentError) { @obj.find_index(0, 1) } end def test_find_all