ruby-changes:10743
From: nobu <ko1@a...>
Date: Sun, 15 Feb 2009 00:45:56 +0900 (JST)
Subject: [ruby-changes:10743] Ruby:r22307 (trunk): * array.c (rb_ary_uniq_bang, rb_ary_uniq): unique by the result of
nobu 2009-02-15 00:45:43 +0900 (Sun, 15 Feb 2009) New Revision: 22307 http://svn.ruby-lang.org/cgi-bin/viewvc.cgi?view=rev&revision=22307 Log: * array.c (rb_ary_uniq_bang, rb_ary_uniq): unique by the result of given block. [ruby-dev:37998] Modified files: trunk/ChangeLog trunk/array.c trunk/test/ruby/test_array.rb Index: array.c =================================================================== --- array.c (revision 22306) +++ array.c (revision 22307) @@ -2887,15 +2887,43 @@ return hash; } -static VALUE -ary_make_hash(VALUE ary) +static inline VALUE +ary_tmp_hash_new(void) { VALUE hash = rb_hash_new(); RBASIC(hash)->klass = 0; + return hash; +} + +static VALUE +ary_make_hash(VALUE ary) +{ + VALUE hash = ary_tmp_hash_new(); return ary_add_hash(hash, ary); } +static VALUE +ary_add_hash_by(VALUE hash, VALUE ary) +{ + long i; + + for (i = 0; i < RARRAY_LEN(ary); ++i) { + VALUE v = rb_ary_elt(ary, i), k = rb_yield(v); + if (rb_hash_lookup2(hash, k, Qundef) == Qundef) { + rb_hash_aset(hash, k, v); + } + } + return hash; +} + +static VALUE +ary_make_hash_by(VALUE ary) +{ + VALUE hash = ary_tmp_hash_new(); + return ary_add_hash_by(hash, ary); +} + static inline void ary_recycle_hash(VALUE hash) { @@ -3010,6 +3038,13 @@ return ary3; } +static int +push_value(st_data_t key, st_data_t val, st_data_t ary) +{ + rb_ary_push((VALUE)ary, (VALUE)val); + return ST_CONTINUE; +} + /* * call-seq: * array.uniq! -> array or nil @@ -3022,6 +3057,8 @@ * a.uniq! #=> ["a", "b", "c"] * b = [ "a", "b", "c" ] * b.uniq! #=> nil + * c = [ "a:def", "a:xyz", "b:abc", "b:xyz", "c:jkl" ] + * c.uniq! {|s| s[/^\w+/]} #=> [ "a:def", "b:abc", "c:jkl" ] */ static VALUE @@ -3030,18 +3067,28 @@ VALUE hash, v; long i, j; - hash = ary_make_hash(ary); - - if (RARRAY_LEN(ary) == RHASH_SIZE(hash)) { - return Qnil; + if (rb_block_given_p()) { + hash = ary_make_hash_by(ary); + if (RARRAY_LEN(ary) == (i = RHASH_SIZE(hash))) { + return Qnil; + } + ary_resize_capa(ary, i); + ARY_SET_LEN(ary, 0); + st_foreach(RHASH_TBL(hash), push_value, ary); } - for (i=j=0; i<RARRAY_LEN(ary); i++) { - st_data_t vv = (st_data_t)(v = rb_ary_elt(ary, i)); - if (st_delete(RHASH_TBL(hash), &vv, 0)) { - rb_ary_store(ary, j++, v); + else { + hash = ary_make_hash(ary); + if (RARRAY_LEN(ary) == RHASH_SIZE(hash)) { + return Qnil; } + for (i=j=0; i<RARRAY_LEN(ary); i++) { + st_data_t vv = (st_data_t)(v = rb_ary_elt(ary, i)); + if (st_delete(RHASH_TBL(hash), &vv, 0)) { + rb_ary_store(ary, j++, v); + } + } + ARY_SET_LEN(ary, j); } - ARY_SET_LEN(ary, j); ary_recycle_hash(hash); return ary; @@ -3055,19 +3102,29 @@ * * a = [ "a", "a", "b", "b", "c" ] * a.uniq #=> ["a", "b", "c"] + * c = [ "a:def", "a:xyz", "b:abc", "b:xyz", "c:jkl" ] + * c.uniq {|s| s[/^\w+/]} #=> [ "a:def", "b:abc", "c:jkl" ] */ static VALUE rb_ary_uniq(VALUE ary) { - VALUE hash = ary_make_hash(ary), v; - VALUE uniq = ary_new(rb_obj_class(ary), RHASH_SIZE(hash)); + VALUE hash, uniq, v; long i; - for (i=0; i<RARRAY_LEN(ary); i++) { - st_data_t vv = (st_data_t)(v = rb_ary_elt(ary, i)); - if (st_delete(RHASH_TBL(hash), &vv, 0)) { - rb_ary_push(uniq, v); + if (rb_block_given_p()) { + hash = ary_make_hash_by(ary); + uniq = ary_new(rb_obj_class(ary), RHASH_SIZE(hash)); + st_foreach(RHASH_TBL(hash), push_value, uniq); + } + else { + hash = ary_make_hash(ary); + uniq = ary_new(rb_obj_class(ary), RHASH_SIZE(hash)); + for (i=0; i<RARRAY_LEN(ary); i++) { + st_data_t vv = (st_data_t)(v = rb_ary_elt(ary, i)); + if (st_delete(RHASH_TBL(hash), &vv, 0)) { + rb_ary_push(uniq, v); + } } } ary_recycle_hash(hash); Index: ChangeLog =================================================================== --- ChangeLog (revision 22306) +++ ChangeLog (revision 22307) @@ -1,3 +1,8 @@ +Sun Feb 15 00:45:41 2009 Nobuyoshi Nakada <nobu@r...> + + * array.c (rb_ary_uniq_bang, rb_ary_uniq): unique by the result of + given block. [ruby-dev:37998] + Sun Feb 15 00:39:44 2009 Nobuyoshi Nakada <nobu@r...> * array.c (ary_resize_capa): should not overwrite outside embedded Index: test/ruby/test_array.rb =================================================================== --- test/ruby/test_array.rb (revision 22306) +++ test/ruby/test_array.rb (revision 22307) @@ -1240,6 +1240,11 @@ assert_equal(@cls[1, 2, 3, 4, nil], a.uniq) assert_equal(b, a) + c = @cls["a:def", "a:xyz", "b:abc", "b:xyz", "c:jkl"] + d = c.dup + assert_equal(@cls[ "a:def", "b:abc", "c:jkl" ], c.uniq {|s| s[/^\w+/]}) + assert_equal(d, c) + assert_equal(@cls[1, 2, 3], @cls[1, 2, 3].uniq) end @@ -1248,6 +1253,10 @@ assert_equal(@cls[1, 2, 3, 4, nil], a.uniq!) assert_equal(@cls[1, 2, 3, 4, nil], a) + c = @cls["a:def", "a:xyz", "b:abc", "b:xyz", "c:jkl"] + assert_equal(@cls[ "a:def", "b:abc", "c:jkl" ], c.uniq! {|s| s[/^\w+/]}) + assert_equal(@cls[ "a:def", "b:abc", "c:jkl" ], c) + assert_nil(@cls[1, 2, 3].uniq!) end -- ML: ruby-changes@q... Info: http://www.atdot.net/~ko1/quickml/