diff options
author | nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2007-12-27 02:02:47 +0000 |
---|---|---|
committer | nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2007-12-27 02:02:47 +0000 |
commit | 59821028483d2f2dfaced428be558d319b07a05a (patch) | |
tree | e22af79fdc73e9f643a5fd4ea1fc179ec0ab32d4 /string.c | |
parent | 9e1da70e1478583dfaeac97652599fff1c56f493 (diff) |
* string.c (rb_str_lines, rb_str_bytes): ditto.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_1_8@14728 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'string.c')
-rw-r--r-- | string.c | 94 |
1 files changed, 94 insertions, 0 deletions
@@ -3762,6 +3762,97 @@ rb_str_each_byte(str) } +static VALUE str_enumerator _((VALUE, VALUE, int, VALUE *)); +static VALUE +str_enumerator(str, sym, argc, argv) + VALUE str, sym; + int argc; + VALUE *argv; +{ + static VALUE enumerator; + static ID new; + int nargc; + VALUE *nargv, result; + volatile VALUE args; + + if (!enumerator) { + rb_require("enumerator"); + enumerator = rb_path2class("Enumerable::Enumerator"); + new = rb_intern("new"); + } + args = rb_ary_new2(nargc = argc + 2); + RBASIC(args)->klass = 0; + nargv = RARRAY_PTR(args); + nargv[0] = str; + nargv[1] = sym; + MEMCPY(nargv + 2, argv, VALUE, argc); + result = rb_funcall2(enumerator, new, nargc, nargv); + rb_ary_clear(args); + return result; +} + + +/* + * Document-method: lines + * call-seq: + * str.lines(separator=$/) => anEnumerator + * str.lines(separator=$/) {|substr| block } => str + * + * Returns an enumerator that gives each line in the string. If a block is + * given, it iterates over each line in the string. + * + * "foo\nbar\n".lines.to_a #=> ["foo\n", "bar\n"] + * "foo\nb ar".lines.sort #=> ["b ar", "foo\n"] + */ + +static VALUE +rb_str_lines(argc, argv, str) + int argc; + VALUE *argv; + VALUE str; +{ + if (rb_block_given_p()) { + return rb_str_each_line(argc, argv, str); + } + else { + static VALUE each_line; + + if (!each_line) each_line = ID2SYM(rb_intern("each_line")); + return str_enumerator(str, each_line, argc, argv); + } +} + + +/* + * Document-method: bytes + * call-seq: + * str.bytes => anEnumerator + * str.bytes {|fixnum| block } => str + * + * Returns an enumerator that gives each byte in the string. If a block is + * given, it iterates over each byte in the string. + * + * "hello".bytes.to_a #=> [104, 101, 108, 108, 111] + */ + +static VALUE +rb_str_bytes(argc, argv, str) + int argc; + VALUE *argv; + VALUE str; +{ + if (rb_block_given_p()) { + return rb_str_each_byte(argc, argv, str); + } + else { + static VALUE each_byte; + + if (!each_byte) each_byte = ID2SYM(rb_intern("each_byte")); + return str_enumerator(str, each_byte, argc, argv); + } +} + + /* * call-seq: * str.chop! => str or nil @@ -4754,6 +4845,9 @@ Init_String() rb_define_method(rb_cString, "each", rb_str_each_line, -1); rb_define_method(rb_cString, "each_byte", rb_str_each_byte, 0); + rb_define_method(rb_cString, "lines", rb_str_lines, -1); + rb_define_method(rb_cString, "bytes", rb_str_bytes, -1); + rb_define_method(rb_cString, "sum", rb_str_sum, -1); rb_define_global_function("sub", rb_f_sub, -1); |