archive/ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2025-08-15 13:39:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

* array.c (rb_ary_reject_bang, rb_ary_delete_if): rejected

Browse source 
elements should be removed.  fixed [Bug #2545]


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@32360 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
nobu 2011-07-01 22:17:46 +00:00
parent a943788fb7
commit e7047b8a37
3 changed files with 64 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

69
array.c
View file

@ -2563,6 +2563,48 @@ rb_ary_slice_bang(int argc, VALUE *argv, VALUE ary)
return rb_ary_delete_at(ary, NUM2LONG(arg1));
}
static VALUE
ary_reject(VALUE orig, VALUE result)
{
long i;
int rejected = 0;
for (i = 0; i < RARRAY_LEN(orig); i++) {
VALUE v = RARRAY_PTR(orig)[i];
if (!RTEST(rb_yield(v))) {
rb_ary_push_1(result, v);
}
else {
rejected = 1;
}
}
return rejected ? result : 0;
}
static VALUE
ary_protecting_reject(VALUE arg)
{
VALUE *args = (VALUE *)arg;
return ary_reject(args[0], args[1]);
}
static VALUE
ary_reject_bang(VALUE ary)
{
VALUE args[2];
int state = 0;
rb_ary_modify_check(ary);
args[0] = ary;
args[1] = rb_ary_new();
if (!rb_protect(ary_protecting_reject, (VALUE)args, &state)) {
return Qnil;
}
rb_ary_replace(ary, args[1]);
if (state) rb_jump_tag(state);
return ary;
}
/*
* call-seq:
* ary.reject! {|item| block } -> ary or nil
@ -2580,23 +2622,8 @@ rb_ary_slice_bang(int argc, VALUE *argv, VALUE ary)
static VALUE
rb_ary_reject_bang(VALUE ary)
{
long i1, i2;
RETURN_ENUMERATOR(ary, 0, 0);
rb_ary_modify(ary);
for (i1 = i2 = 0; i1 < RARRAY_LEN(ary); i1++) {
VALUE v = RARRAY_PTR(ary)[i1];
if (RTEST(rb_yield(v))) continue;
if (i1 != i2) {
rb_ary_store(ary, i2, v);
}
i2++;
}
if (RARRAY_LEN(ary) == i2) return Qnil;
if (i2 < RARRAY_LEN(ary))
ARY_SET_LEN(ary, i2);
return ary;
return ary_reject_bang(ary);
}
/*
@ -2615,10 +2642,12 @@ rb_ary_reject_bang(VALUE ary)
static VALUE
rb_ary_reject(VALUE ary)
{
VALUE rejected_ary;
RETURN_ENUMERATOR(ary, 0, 0);
ary = rb_ary_dup(ary);
rb_ary_reject_bang(ary);
return ary;
rejected_ary = rb_ary_new();
ary_reject(ary, rejected_ary);
return rejected_ary;
}
/*
@ -2640,7 +2669,7 @@ static VALUE
rb_ary_delete_if(VALUE ary)
{
RETURN_ENUMERATOR(ary, 0, 0);
rb_ary_reject_bang(ary);
ary_reject_bang(ary);
return ary;
}