diff options
author | marcandre <marcandre@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2012-11-06 17:12:05 +0000 |
---|---|---|
committer | marcandre <marcandre@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2012-11-06 17:12:05 +0000 |
commit | b8b01ab995b963db825b9b493aa98500f0be83e0 (patch) | |
tree | 2a849367b23c9f58743dabdd7d8a88287f9ee2bb /array.c | |
parent | ba59365db9283b23b1a2dd9b91dac7303c798326 (diff) | |
download | ruby-b8b01ab995b963db825b9b493aa98500f0be83e0.tar.gz |
* array.c (rb_ary_cycle): Support for Array#cycle.size
[Feature #6636]
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@37505 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'array.c')
-rw-r--r-- | array.c | 16 |
1 files changed, 15 insertions, 1 deletions
@@ -4229,6 +4229,20 @@ rb_ary_sample(int argc, VALUE *argv, VALUE ary) return result; } +static VALUE +rb_ary_cycle_size(VALUE self, VALUE args) +{ + long mul; + VALUE n = Qnil; + if (args && (RARRAY_LEN(args) > 0)) { + n = RARRAY_PTR(args)[0]; + } + if (RARRAY_LEN(self) == 0) return INT2FIX(0); + if (n == Qnil) return DBL2NUM(INFINITY); + mul = NUM2LONG(n); + if (mul <= 0) return INT2FIX(0); + return rb_funcall(rb_ary_length(self), '*', 1, LONG2FIX(mul)); +} /* * call-seq: @@ -4258,7 +4272,7 @@ rb_ary_cycle(int argc, VALUE *argv, VALUE ary) rb_scan_args(argc, argv, "01", &nv); - RETURN_ENUMERATOR(ary, argc, argv); + RETURN_SIZED_ENUMERATOR(ary, argc, argv, rb_ary_cycle_size); if (NIL_P(nv)) { n = -1; } |