diff options
author | yugui <yugui@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2008-11-03 04:13:04 +0000 |
---|---|---|
committer | yugui <yugui@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2008-11-03 04:13:04 +0000 |
commit | c2ded646534f362b9e409b5de58c5d869bc0fdf4 (patch) | |
tree | 71ed7be9714b0245309bc31823cf22f2e8ee2816 /array.c | |
parent | d5c6b265b977bc617c2863cf1e27dd0d16beaccc (diff) |
* array.c (rb_ary_take_first_or_last): renamed rb_ary_partial more
appropriately.
(ary_take_pos_flags): new enum. makes the fourth argument of
rb_ary_take_first_or_last more descriptive.
(rb_ary_partial): renamed rb_ary_partial0 more appropriately.
takes a new parameter klass for what class to instanciate.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@20096 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'array.c')
-rw-r--r-- | array.c | 26 |
1 files changed, 16 insertions, 10 deletions
@@ -603,20 +603,20 @@ rb_ary_store(VALUE ary, long idx, VALUE val) } static VALUE -ary_make_partial0(VALUE ary, long offset, long len) +ary_make_partial(VALUE ary, VALUE klass, long offset, long len) { assert(offset >= 0); assert(len >= 0); assert(offset+len <= RARRAY_LEN(ary)); if (len <= RARRAY_EMBED_LEN_MAX) { - VALUE result = ary_alloc(rb_obj_class(ary)); + VALUE result = ary_alloc(klass); MEMCPY(ARY_EMBED_PTR(result), RARRAY_PTR(ary) + offset, VALUE, len); ARY_SET_EMBED_LEN(result, len); return result; } else { - VALUE shared, result = ary_alloc(rb_obj_class(ary)); + VALUE shared, result = ary_alloc(klass); FL_UNSET_EMBED(result); shared = ary_make_shared(ary); @@ -630,8 +630,14 @@ ary_make_partial0(VALUE ary, long offset, long len) } } +enum ary_take_pos_flags +{ + ARY_TAKE_FIRST = 0, + ARY_TAKE_LAST = 1 +}; + static VALUE -ary_make_partial(int argc, VALUE *argv, VALUE ary, int last) +ary_take_first_or_last(int argc, VALUE *argv, VALUE ary, enum ary_take_pos_flags last) { VALUE nv; long n; @@ -648,7 +654,7 @@ ary_make_partial(int argc, VALUE *argv, VALUE ary, int last) if (last) { offset = RARRAY_LEN(ary) - n; } - return ary_make_partial0(ary, offset, n); + return ary_make_partial(ary, rb_cArray, offset, n); } /* @@ -737,7 +743,7 @@ rb_ary_pop_m(int argc, VALUE *argv, VALUE ary) } rb_ary_modify_check(ary); - result = ary_make_partial(argc, argv, ary, Qtrue); + result = ary_take_first_or_last(argc, argv, ary, Qtrue); ARY_INCREASE_LEN(ary, -RARRAY_LEN(result)); return result; } @@ -799,7 +805,7 @@ rb_ary_shift_m(int argc, VALUE *argv, VALUE ary) } rb_ary_modify_check(ary); - result = ary_make_partial(argc, argv, ary, Qfalse); + result = ary_take_first_or_last(argc, argv, ary, ARY_TAKE_FIRST); n = RARRAY_LEN(result); if (ARY_SHARED_P(ary)) { ARY_INCREASE_PTR(ary, n); @@ -883,7 +889,7 @@ rb_ary_subseq(VALUE ary, long beg, long len) klass = rb_obj_class(ary); if (len == 0) return ary_new(klass, 0); - return ary_make_partial0(ary, beg, len); + return ary_make_partial(ary, klass, beg, len); } /* @@ -993,7 +999,7 @@ rb_ary_first(int argc, VALUE *argv, VALUE ary) return RARRAY_PTR(ary)[0]; } else { - return ary_make_partial(argc, argv, ary, Qfalse); + return ary_take_first_or_last(argc, argv, ary, ARY_TAKE_FIRST); } } @@ -1018,7 +1024,7 @@ rb_ary_last(int argc, VALUE *argv, VALUE ary) return RARRAY_PTR(ary)[RARRAY_LEN(ary)-1]; } else { - return ary_make_partial(argc, argv, ary, Qtrue); + return ary_take_first_or_last(argc, argv, ary, ARY_TAKE_LAST); } } |