1berry/ruby
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Get rid of flatten_memo_data_type

Browse Source 
We can use an hidden Hash instead, it's simpler, triggers
write barriers, handle compaction, etc.
This commit is contained in:
Jean Boussier 2023-11-22 16:01:37 +01:00 committed by Jean Boussier
parent 6968b289e7
commit 9691532428
1 changed files with 12 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
array.c
View File

@ -6295,16 +6295,9 @@ rb_ary_count(int argc, VALUE *argv, VALUE ary)
static VALUE
flatten(VALUE ary, int level)
{
static const rb_data_type_t flatten_memo_data_type = {
.wrap_struct_name = "array_flatten_memo_data_type",
.function = { NULL, (RUBY_DATA_FUNC)st_free_table },
NULL, NULL, RUBY_TYPED_FREE_IMMEDIATELY | RUBY_TYPED_WB_PROTECTED
};
long i;
VALUE stack, result, tmp = 0, elt, vmemo;
st_table *memo = 0;
st_data_t id;
VALUE stack, result, tmp = 0, elt;
VALUE memo = Qfalse;
for (i = 0; i < RARRAY_LEN(ary); i++) {
elt = RARRAY_AREF(ary, i);
@ -6326,10 +6319,9 @@ flatten(VALUE ary, int level)
rb_ary_push(stack, LONG2NUM(i + 1));
if (level < 0) {
memo = st_init_numtable();
vmemo = TypedData_Wrap_Struct(0, &flatten_memo_data_type, memo);
st_insert(memo, (st_data_t)ary, (st_data_t)Qtrue);
st_insert(memo, (st_data_t)tmp, (st_data_t)Qtrue);
memo = rb_obj_hide(rb_ident_hash_new());
rb_hash_aset(memo, ary, Qtrue);
rb_hash_aset(memo, tmp, Qtrue);
}
ary = tmp;
@ -6344,9 +6336,8 @@ flatten(VALUE ary, int level)
}
tmp = rb_check_array_type(elt);
if (RBASIC(result)->klass) {
if (memo) {
RB_GC_GUARD(vmemo);
st_clear(memo);
if (RTEST(memo)) {
rb_hash_clear(memo);
}
rb_raise(rb_eRuntimeError, "flatten reentered");
}
@ -6355,12 +6346,11 @@ flatten(VALUE ary, int level)
}
else {
if (memo) {
id = (st_data_t)tmp;
if (st_is_member(memo, id)) {
st_clear(memo);
if (rb_hash_aref(memo, tmp) == Qtrue) {
rb_hash_clear(memo);
rb_raise(rb_eArgError, "tried to flatten recursive array");
}
st_insert(memo, id, (st_data_t)Qtrue);
rb_hash_aset(memo, tmp, Qtrue);
}
rb_ary_push(stack, ary);
rb_ary_push(stack, LONG2NUM(i));
@ -6372,8 +6362,7 @@ flatten(VALUE ary, int level)
break;
}
if (memo) {
id = (st_data_t)ary;
st_delete(memo, &id, 0);
rb_hash_delete(memo, ary);
}
tmp = rb_ary_pop(stack);
i = NUM2LONG(tmp);
@ -6381,7 +6370,7 @@ flatten(VALUE ary, int level)
}
if (memo) {
st_clear(memo);
rb_hash_clear(memo);
}
RBASIC_SET_CLASS(result, rb_cArray);