kotovalexarian-likes-github/ruby--ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
Code Releases Activity

* enumerator.c (lazy_flat_map): add Enumerable::Lazy#flat_map.

Browse source 
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@34956 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
shugo 2012-03-09 05:34:41 +00:00
parent 5452b18f99
commit a21d0f72c2
3 changed files with 40 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ChangeLog
View file

@ -1,3 +1,7 @@
Fri Mar 9 14:29:32 2012 Shugo Maeda <shugo@ruby-lang.org>
* enumerator.c (lazy_flat_map): add Enumerable::Lazy#flat_map.
Fri Mar 9 06:29:22 2012 Aaron Patterson <aaron@tenderlovemaking.com> Fri Mar 9 06:29:22 2012 Aaron Patterson <aaron@tenderlovemaking.com>
* ext/psych/lib/psych.rb (load, parse): stop parsing or loading after * ext/psych/lib/psych.rb (load, parse): stop parsing or loading after

28
enumerator.c
View file

@ -1272,6 +1272,32 @@ lazy_map(VALUE obj)
return rb_block_call(rb_cLazy, id_new, 1, &obj, lazy_map_func, 0); return rb_block_call(rb_cLazy, id_new, 1, &obj, lazy_map_func, 0);
} }
static VALUE
lazy_flat_map_func(VALUE val, VALUE m, int argc, VALUE *argv)
{
VALUE result = rb_yield_values2(argc - 1, &argv[1]);
VALUE ary = rb_check_array_type(result);
if (NIL_P(ary)) {
return rb_funcall(argv[0], id_yield, 1, result);
}
else {
int i;
for (i = 0; i < RARRAY_LEN(ary); i++) {
rb_funcall(argv[0], id_yield, 1, RARRAY_PTR(ary)[i]);
}
return Qnil;
}
}
static VALUE
lazy_flat_map(VALUE obj)
{
if (!rb_block_given_p()) {
rb_raise(rb_eArgError, "tried to call lazy flat_map without a block");
}
return rb_block_call(rb_cLazy, id_new, 1, &obj, lazy_flat_map_func, 0);
}
static VALUE static VALUE
lazy_select_func(VALUE val, VALUE m, int argc, VALUE *argv) lazy_select_func(VALUE val, VALUE m, int argc, VALUE *argv)
@ -1377,11 +1403,13 @@ Init_Enumerator(void)
rb_define_method(rb_mEnumerable, "lazy", enumerable_lazy, 0); rb_define_method(rb_mEnumerable, "lazy", enumerable_lazy, 0);
rb_define_method(rb_cLazy, "initialize", lazy_initialize, 1); rb_define_method(rb_cLazy, "initialize", lazy_initialize, 1);
rb_define_method(rb_cLazy, "map", lazy_map, 0); rb_define_method(rb_cLazy, "map", lazy_map, 0);
rb_define_method(rb_cLazy, "flat_map", lazy_flat_map, 0);
rb_define_method(rb_cLazy, "select", lazy_select, 0); rb_define_method(rb_cLazy, "select", lazy_select, 0);
rb_define_method(rb_cLazy, "reject", lazy_reject, 0); rb_define_method(rb_cLazy, "reject", lazy_reject, 0);
rb_define_method(rb_cLazy, "grep", lazy_grep, 1); rb_define_method(rb_cLazy, "grep", lazy_grep, 1);
rb_define_alias(rb_cLazy, "collect", "map"); rb_define_alias(rb_cLazy, "collect", "map");
rb_define_alias(rb_cLazy, "collect_concat", "flat_map");
rb_define_alias(rb_cLazy, "find_all", "select"); rb_define_alias(rb_cLazy, "find_all", "select");
rb_eStopIteration = rb_define_class("StopIteration", rb_eIndexError); rb_eStopIteration = rb_define_class("StopIteration", rb_eIndexError);

8
test/ruby/test_lazy_enumerator.rb
View file

@ -80,6 +80,14 @@ class TestLazyEnumerator < Test::Unit::TestCase
assert_equal(1, a.current) assert_equal(1, a.current)
end end
def test_flat_map
a = Step.new(1..3)
assert_equal(2, a.flat_map {|x| [x * 2]}.first)
assert_equal(3, a.current)
assert_equal(2, a.lazy.flat_map {|x| [x * 2]}.first)
assert_equal(1, a.current)
end
def test_reject def test_reject
a = Step.new(1..6) a = Step.new(1..6)
assert_equal(4, a.reject {|x| x < 4}.first) assert_equal(4, a.reject {|x| x < 4}.first)