From 74433fd300ff90e61086681fd1152d2da0d28171 Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Tue, 30 Aug 2005 14:49:51 +0000
Subject: * array.c, dir.c, enum.c, hash.c, io.c, range.c, string.c, struct.c: 
  let enumerable methods return Enumerator.  [ruby-dev:26924]

* intern.h (RETURN_ENUMERATOR): utility macro for enumerable methods.


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@9053 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 io.c | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'io.c')

diff --git a/io.c b/io.c
index 0a66f4bb43..c1081a7464 100644
--- a/io.c
+++ b/io.c
@@ -1851,6 +1851,7 @@ rb_io_each_line(argc, argv, io)
     VALUE str;
     VALUE rs;
 
+    RETURN_ENUMERATOR(io, argc, argv);
     if (argc == 0) {
 	rs = rb_rs;
     }
@@ -1885,6 +1886,7 @@ rb_io_each_byte(io)
     OpenFile *fptr;
     int c;
 
+    RETURN_ENUMERATOR(io, 0, 0);
     GetOpenFile(io, fptr);
 
     for (;;) {
@@ -5153,13 +5155,15 @@ io_s_foreach(arg)
  */
 
 static VALUE
-rb_io_s_foreach(argc, argv)
+rb_io_s_foreach(argc, argv, self)
     int argc;
     VALUE *argv;
+    VALUE self;
 {
     VALUE fname;
     struct foreach_arg arg;
 
+    RETURN_ENUMERATOR(self, argc, argv);
     rb_scan_args(argc, argv, "11", &fname, &arg.sep);
     FilePathValue(fname);
     if (argc == 1) {
@@ -5453,12 +5457,14 @@ argf_readchar()
 }
 
 static VALUE
-argf_each_line(argc, argv)
+argf_each_line(argc, argv, self)
     int argc;
     VALUE *argv;
+    VALUE self;
 {
     VALUE str;
 
+    RETURN_ENUMERATOR(self, argc, argv);
     if (!next_argv()) return Qnil;
     if (TYPE(current_file) != T_FILE) {
 	for (;;) {
@@ -5474,10 +5480,12 @@ argf_each_line(argc, argv)
 }
 
 static VALUE
-argf_each_byte()
+argf_each_byte(self)
+    VALUE self;
 {
     VALUE byte;
 
+    RETURN_ENUMERATOR(self, 0, 0);
     while (!NIL_P(byte = argf_getc())) {
 	rb_yield(byte);
     }
-- 
cgit v1.2.3