From 391d69112374bb3635175f6b0031e3c59e39c54c Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Wed, 13 Feb 2013 08:51:27 +0000
Subject: io.c: FOREACH_ARGF

* io.c (FOREACH_ARGF): loop for ARGF.

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

(limited to 'io.c')

diff --git a/io.c b/io.c
index 9d2ffbd5f7..6107fd3e30 100644
--- a/io.c
+++ b/io.c
@@ -10904,6 +10904,8 @@ argf_readbyte(VALUE argf)
     return c;
 }
 
+#define FOREACH_ARGF() for (; next_argv(); ARGF.next_p = 1)
+
 /*
  *  call-seq:
  *     ARGF.each(sep=$/)            {|line| block }  -> ARGF
@@ -10940,11 +10942,10 @@ static VALUE
 argf_each_line(int argc, VALUE *argv, VALUE argf)
 {
     RETURN_ENUMERATOR(argf, argc, argv);
-    for (;;) {
-	if (!next_argv()) return argf;
+    FOREACH_ARGF() {
 	rb_block_call(ARGF.current_file, rb_intern("each_line"), argc, argv, 0, 0);
-	ARGF.next_p = 1;
     }
+    return argf;
 }
 
 /*
@@ -10988,11 +10989,10 @@ static VALUE
 argf_each_byte(VALUE argf)
 {
     RETURN_ENUMERATOR(argf, 0, 0);
-    for (;;) {
-	if (!next_argv()) return argf;
+    FOREACH_ARGF() {
 	rb_block_call(ARGF.current_file, rb_intern("each_byte"), 0, 0, 0, 0);
-	ARGF.next_p = 1;
     }
+    return argf;
 }
 
 /*
@@ -11028,11 +11028,10 @@ static VALUE
 argf_each_char(VALUE argf)
 {
     RETURN_ENUMERATOR(argf, 0, 0);
-    for (;;) {
-	if (!next_argv()) return argf;
+    FOREACH_ARGF() {
 	rb_block_call(ARGF.current_file, rb_intern("each_char"), 0, 0, 0, 0);
-	ARGF.next_p = 1;
     }
+    return argf;
 }
 
 /*
@@ -11068,11 +11067,10 @@ static VALUE
 argf_each_codepoint(VALUE argf)
 {
     RETURN_ENUMERATOR(argf, 0, 0);
-    for (;;) {
-	if (!next_argv()) return argf;
+    FOREACH_ARGF() {
 	rb_block_call(ARGF.current_file, rb_intern("each_codepoint"), 0, 0, 0, 0);
-	ARGF.next_p = 1;
     }
+    return argf;
 }
 
 /*
-- 
cgit v1.2.3