From 765a6be7008d0eba3f374d9e9875aa9e65ea4112 Mon Sep 17 00:00:00 2001 From: matz Date: Mon, 2 Feb 2009 10:43:27 +0000 Subject: [PATCH] * array.c (rb_ary_sort_by_bang): new method. requested in [ruby-core:21709] git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@21967 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ChangeLog | 5 +++++ array.c | 28 ++++++++++++++++++++++++++++ 2 files changed, 33 insertions(+) diff --git a/ChangeLog b/ChangeLog index b023bb8b4f..b32115e63f 100644 --- a/ChangeLog +++ b/ChangeLog @@ -39,6 +39,11 @@ Mon Feb 2 14:53:35 2009 Tanaka Akira * ext/socket/socket.c (sock_initialize): make 3rd argument, protocol, optional. +Mon Feb 2 14:47:53 2009 Yukihiro Matsumoto + + * array.c (rb_ary_sort_by_bang): new method. requested in + [ruby-core:21709] + Mon Feb 2 14:22:56 2009 NAKAMURA Usaku * ext/socket/constants.c (cmsg_type_arg): INET6 check. diff --git a/array.c b/array.c index a3821a52b0..17f5e6119b 100644 --- a/array.c +++ b/array.c @@ -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 enum in place using a set of keys generated by mapping the + * values in enum 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);