diff options
author | shyouhei <shyouhei@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2008-07-07 07:38:25 +0000 |
---|---|---|
committer | shyouhei <shyouhei@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2008-07-07 07:38:25 +0000 |
commit | 9ff1e787f915539b1980654e3d3d2013ff5c81d2 (patch) | |
tree | 8d0fc9ca5b4dbfa9885dc56862292d55091bcaac /ruby_1_8_6/ext/enumerator/enumerator.c | |
parent | 441546edcfbb1b346c87b69c5f578d1a0e522e06 (diff) |
wrong commit; sorryv1_8_6_269
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/tags/v1_8_6_269@17938 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'ruby_1_8_6/ext/enumerator/enumerator.c')
-rw-r--r-- | ruby_1_8_6/ext/enumerator/enumerator.c | 298 |
1 files changed, 0 insertions, 298 deletions
diff --git a/ruby_1_8_6/ext/enumerator/enumerator.c b/ruby_1_8_6/ext/enumerator/enumerator.c deleted file mode 100644 index 1c6e1d1ace..0000000000 --- a/ruby_1_8_6/ext/enumerator/enumerator.c +++ /dev/null @@ -1,298 +0,0 @@ -/************************************************ - - enumerator.c - provides Enumerator class - - $Author$ - - Copyright (C) 2001-2003 Akinori MUSHA - - $Idaemons: /home/cvs/rb/enumerator/enumerator.c,v 1.1.1.1 2001/07/15 10:12:48 knu Exp $ - $RoughId: enumerator.c,v 1.6 2003/07/27 11:03:24 nobu Exp $ - $Id$ - -************************************************/ - -#include "ruby.h" -#include "node.h" - -/* - * Document-class: Enumerable::Enumerator - * - * A class which provides a method `each' to be used as an Enumerable - * object. - */ -static VALUE rb_cEnumerator; -static ID sym_each, sym_each_with_index, sym_each_slice, sym_each_cons; -static ID id_new, id_enum_obj, id_enum_method, id_enum_args; - -/* - * call-seq: - * obj.to_enum(method = :each, *args) - * obj.enum_for(method = :each, *args) - * - * Returns Enumerable::Enumerator.new(self, method, *args). - * - * e.g.: - * str = "xyz" - * - * enum = str.enum_for(:each_byte) - * a = enum.map {|b| '%02x' % b } #=> ["78", "79", "7a"] - * - * # protects an array from being modified - * a = [1, 2, 3] - * some_method(a.to_enum) - * - */ -static VALUE -obj_to_enum(obj, enum_args) - VALUE obj, enum_args; -{ - rb_ary_unshift(enum_args, obj); - - return rb_apply(rb_cEnumerator, id_new, enum_args); -} - -/* - * call-seq: - * enum_with_index - * - * Returns Enumerable::Enumerator.new(self, :each_with_index). - * - */ -static VALUE -enumerator_enum_with_index(obj) - VALUE obj; -{ - return rb_funcall(rb_cEnumerator, id_new, 2, obj, sym_each_with_index); -} - -static VALUE -each_slice_i(val, memo) - VALUE val; - NODE *memo; -{ - VALUE ary = memo->u1.value; - long size = memo->u3.cnt; - - rb_ary_push(ary, val); - - if (RARRAY(ary)->len == size) { - rb_yield(ary); - memo->u1.value = rb_ary_new2(size); - } - - return Qnil; -} - -/* - * call-seq: - * e.each_slice(n) {...} - * - * Iterates the given block for each slice of <n> elements. - * - * e.g.: - * (1..10).each_slice(3) {|a| p a} - * # outputs below - * [1, 2, 3] - * [4, 5, 6] - * [7, 8, 9] - * [10] - * - */ -static VALUE -enum_each_slice(obj, n) - VALUE obj, n; -{ - long size = NUM2LONG(n); - NODE *memo; - VALUE ary; - - if (size <= 0) rb_raise(rb_eArgError, "invalid slice size"); - - memo = rb_node_newnode(NODE_MEMO, rb_ary_new2(size), 0, size); - - rb_iterate(rb_each, obj, each_slice_i, (VALUE)memo); - - ary = memo->u1.value; - if (RARRAY(ary)->len > 0) rb_yield(ary); - - return Qnil; -} - -/* - * call-seq: - * e.enum_slice(n) - * - * Returns Enumerable::Enumerator.new(self, :each_slice, n). - * - */ -static VALUE -enumerator_enum_slice(obj, n) - VALUE obj, n; -{ - return rb_funcall(rb_cEnumerator, id_new, 3, obj, sym_each_slice, n); -} - -static VALUE -each_cons_i(val, memo) - VALUE val; - NODE *memo; -{ - VALUE ary = memo->u1.value; - long size = memo->u3.cnt; - - if (RARRAY(ary)->len == size) { - rb_ary_shift(ary); - } - rb_ary_push(ary, val); - if (RARRAY(ary)->len == size) { - rb_yield(rb_ary_dup(ary)); - } - return Qnil; -} - -/* - * call-seq: - * each_cons(n) {...} - * - * Iterates the given block for each array of consecutive <n> - * elements. - * - * e.g.: - * (1..10).each_cons(3) {|a| p a} - * # outputs below - * [1, 2, 3] - * [2, 3, 4] - * [3, 4, 5] - * [4, 5, 6] - * [5, 6, 7] - * [6, 7, 8] - * [7, 8, 9] - * [8, 9, 10] - * - */ -static VALUE -enum_each_cons(obj, n) - VALUE obj, n; -{ - long size = NUM2LONG(n); - NODE *memo; - - if (size <= 0) rb_raise(rb_eArgError, "invalid size"); - memo = rb_node_newnode(NODE_MEMO, rb_ary_new2(size), 0, size); - - rb_iterate(rb_each, obj, each_cons_i, (VALUE)memo); - - return Qnil; -} - -/* - * call-seq: - * e.enum_cons(n) - * - * Returns Enumerable::Enumerator.new(self, :each_cons, n). - * - */ -static VALUE -enumerator_enum_cons(obj, n) - VALUE obj, n; -{ - return rb_funcall(rb_cEnumerator, id_new, 3, obj, sym_each_cons, n); -} - -/* - * call-seq: - * Enumerable::Enumerator.new(obj, method = :each, *args) - * - * Creates a new Enumerable::Enumerator object, which is to be - * used as an Enumerable object using the given object's given - * method with the given arguments. - * - * e.g.: - * str = "xyz" - * - * enum = Enumerable::Enumerator.new(str, :each_byte) - * a = enum.map {|b| '%02x' % b } #=> ["78", "79", "7a"] - * - */ -static VALUE -enumerator_initialize(argc, argv, obj) - int argc; - VALUE *argv; - VALUE obj; -{ - VALUE enum_obj, enum_method, enum_args; - - rb_scan_args(argc, argv, "11*", &enum_obj, &enum_method, &enum_args); - - if (enum_method == Qnil) - enum_method = sym_each; - - rb_ivar_set(obj, id_enum_obj, enum_obj); - rb_ivar_set(obj, id_enum_method, enum_method); - rb_ivar_set(obj, id_enum_args, enum_args); - - return Qnil; -} - -static VALUE -enumerator_iter(memo) - NODE *memo; -{ - return rb_apply(memo->u1.value, memo->u2.id, memo->u3.value); -} - -/* - * call-seq: - * enum.each {...} - * - * Iterates the given block using the object and the method specified - * in the first place. - * - */ -static VALUE -enumerator_each(obj) - VALUE obj; -{ - VALUE val; - - obj = (VALUE)rb_node_newnode(NODE_MEMO, - rb_ivar_get(obj, id_enum_obj), - rb_to_id(rb_ivar_get(obj, id_enum_method)), - rb_ivar_get(obj, id_enum_args)); - val = rb_iterate((VALUE (*)_((VALUE)))enumerator_iter, obj, rb_yield, 0); - return val; -} - -void -Init_enumerator() -{ - VALUE rb_mEnumerable; - - rb_define_method(rb_mKernel, "to_enum", obj_to_enum, -2); - rb_define_method(rb_mKernel, "enum_for", obj_to_enum, -2); - - rb_mEnumerable = rb_path2class("Enumerable"); - - rb_define_method(rb_mEnumerable, "enum_with_index", enumerator_enum_with_index, 0); - rb_define_method(rb_mEnumerable, "each_slice", enum_each_slice, 1); - rb_define_method(rb_mEnumerable, "enum_slice", enumerator_enum_slice, 1); - rb_define_method(rb_mEnumerable, "each_cons", enum_each_cons, 1); - rb_define_method(rb_mEnumerable, "enum_cons", enumerator_enum_cons, 1); - - rb_cEnumerator = rb_define_class_under(rb_mEnumerable, "Enumerator", rb_cObject); - rb_include_module(rb_cEnumerator, rb_mEnumerable); - - rb_define_method(rb_cEnumerator, "initialize", enumerator_initialize, -1); - rb_define_method(rb_cEnumerator, "each", enumerator_each, 0); - - sym_each = ID2SYM(rb_intern("each")); - sym_each_with_index = ID2SYM(rb_intern("each_with_index")); - sym_each_slice = ID2SYM(rb_intern("each_slice")); - sym_each_cons = ID2SYM(rb_intern("each_cons")); - - id_new = rb_intern("new"); - id_enum_obj = rb_intern("enum_obj"); - id_enum_method = rb_intern("enum_method"); - id_enum_args = rb_intern("enum_args"); -} |