diff options
-rw-r--r-- | ChangeLog | 5 | ||||
-rw-r--r-- | array.c | 28 |
2 files changed, 33 insertions, 0 deletions
@@ -39,6 +39,11 @@ Mon Feb 2 14:53:35 2009 Tanaka Akira <akr@fsij.org> * ext/socket/socket.c (sock_initialize): make 3rd argument, protocol, optional. +Mon Feb 2 14:47:53 2009 Yukihiro Matsumoto <matz@ruby-lang.org> + + * array.c (rb_ary_sort_by_bang): new method. requested in + [ruby-core:21709] + Mon Feb 2 14:22:56 2009 NAKAMURA Usaku <usa@ruby-lang.org> * ext/socket/constants.c (cmsg_type_arg): INET6 check. @@ -1843,6 +1843,33 @@ rb_ary_sort(VALUE ary) } +static VALUE +sort_by_i(VALUE i) +{ + return rb_yield(i); +} + +/* + * call-seq: + * enum.sort_by! {| obj | block } -> array + * + * Sorts <i>enum</i> in place using a set of keys generated by mapping the + * values in <i>enum</i> through the given block. + */ + +static VALUE +rb_ary_sort_by_bang(VALUE ary) +{ + VALUE sorted; + + RETURN_ENUMERATOR(ary, 0, 0); + rb_ary_modify(ary); + sorted = rb_block_call(ary, rb_intern("sort_by"), 0, 0, sort_by_i, 0); + rb_ary_replace(ary, sorted); + return ary; +} + + /* * call-seq: * array.collect {|item| block } -> an_array @@ -3830,6 +3857,7 @@ Init_Array(void) rb_define_method(rb_cArray, "reverse!", rb_ary_reverse_bang, 0); rb_define_method(rb_cArray, "sort", rb_ary_sort, 0); rb_define_method(rb_cArray, "sort!", rb_ary_sort_bang, 0); + rb_define_method(rb_cArray, "sort_by!", rb_ary_sort_by_bang, 0); rb_define_method(rb_cArray, "collect", rb_ary_collect, 0); rb_define_method(rb_cArray, "collect!", rb_ary_collect_bang, 0); rb_define_method(rb_cArray, "map", rb_ary_collect, 0); |