diff options
author | matz <matz@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2001-12-10 07:18:16 +0000 |
---|---|---|
committer | matz <matz@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2001-12-10 07:18:16 +0000 |
commit | 2f8d3bdc21b37f0d6881b817c8fd9c5b4b351cd4 (patch) | |
tree | eb88114284c1e7a35677815487d122a68fdf9f57 /array.c | |
parent | 3493ea5c7acfeaf187873d4986181a5fda446aa3 (diff) | |
download | ruby-2f8d3bdc21b37f0d6881b817c8fd9c5b4b351cd4.tar.gz |
* array.c (rb_ary_modify): should copy the internal buffer if the
modifying buffer is shared.
* array.c (ary_make_shared): make an internal buffer of an array
to be shared.
* array.c (rb_ary_shift): avoid sliding an internal buffer by
using shared buffer.
* array.c (rb_ary_subseq): avoid copying the buffer.
* parse.y (gettable): should freeze __LINE__ string.
* io.c (rb_io_puts): old behavoir restored. rationale: a) if you
want to call to_s for arrays, you can just call print a, "\n".
b) to_s wastes memory if array (and sum of its contents) is
huge. c) now any object that has to_ary is treated as an array,
using rb_check_convert_type().
* hash.c (rb_hash_initialize): now accepts a block to calculate
the default value. [new]
* hash.c (rb_hash_aref): call "default" method to get the value
corrensponding to the non existing key.
* hash.c (rb_hash_default): get the default value based on the
block given to 'new'. Now it takes an optinal "key" argument.
"default" became the method to get the value for non existing
key. Users may override "default" method to change the hash
behavior.
* hash.c (rb_hash_set_default): clear the flag if a block is given
to 'new'
* object.c (Init_Object): undef Data.allocate, left Data.new.
* ext/curses/curses.c (window_scrollok): use RTEST().
* ext/curses/curses.c (window_idlok): ditto.
* ext/curses/curses.c (window_keypad): ditto.
* ext/curses/curses.c (window_idlok): idlok() may return void on
some platforms; so don't use return value.
* ext/curses/curses.c (window_scrollok): ditto for consistency.
* ext/curses/curses.c: replace FIX2INT() by typechecking NUM2INT().
* parse.y (str_extend): should not process immature #$x and
#@x interpolation, e.g #@#@ etc.
* enum.c (enum_sort_by): sort_by does not have to be stable always.
* enum.c (enum_sort_by): call qsort directly to gain performance.
* util.c (ruby_qsort): ruby_qsort(qs6) is now native thread safe.
* error.c (rb_sys_fail): it must be a bug if it's called when
errno == 0.
* regex.c (WC2MBC1ST): should not pass through > 0x80 number in UTF-8.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@1896 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'array.c')
-rw-r--r-- | array.c | 209 |
1 files changed, 117 insertions, 92 deletions
@@ -17,7 +17,7 @@ #include "st.h" VALUE rb_cArray; -static ID cmp; +static ID id_cmp; #define ARY_DEFAULT_SIZE 16 @@ -45,7 +45,7 @@ memfill(mem, size, val) #define ARY_TMPLOCK FL_USER1 static void -rb_ary_modify(ary) +rb_ary_modify_check(ary) VALUE ary; { if (OBJ_FROZEN(ary)) rb_error_frozen("array"); @@ -55,6 +55,21 @@ rb_ary_modify(ary) rb_raise(rb_eSecurityError, "Insecure: can't modify array"); } +static void +rb_ary_modify(ary) + VALUE ary; +{ + VALUE *ptr; + + rb_ary_modify_check(ary); + if (!FL_TEST(ary, ELTS_SHARED)) return; + ptr = ALLOC_N(VALUE, RARRAY(ary)->len); + FL_UNSET(ary, ELTS_SHARED); + RARRAY(ary)->aux.capa = RARRAY(ary)->len; + MEMCPY(ptr, RARRAY(ary)->ptr, VALUE, RARRAY(ary)->len); + RARRAY(ary)->ptr = ptr; +} + VALUE rb_ary_freeze(ary) VALUE ary; @@ -79,14 +94,14 @@ rb_ary_s_alloc(klass) OBJSETUP(ary, klass, T_ARRAY); ary->len = 0; - ary->capa = 0; ary->ptr = 0; + ary->aux.capa = 0; return (VALUE)ary; } -VALUE -rb_ary_new0(klass, len) +static VALUE +ary_new(klass, len) VALUE klass; long len; { @@ -100,7 +115,7 @@ rb_ary_new0(klass, len) } if (len == 0) len++; RARRAY(ary)->ptr = ALLOC_N(VALUE, len); - RARRAY(ary)->capa = len; + RARRAY(ary)->aux.capa = len; return ary; } @@ -109,7 +124,7 @@ VALUE rb_ary_new2(len) long len; { - return rb_ary_new0(rb_cArray, len); + return ary_new(rb_cArray, len); } @@ -228,9 +243,9 @@ rb_ary_initialize(argc, argv, ary) if (len > 0 && len*sizeof(VALUE) <= 0) { rb_raise(rb_eArgError, "array size too big"); } - if (len > RARRAY(ary)->capa) { - RARRAY(ary)->capa = len; - REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->capa); + if (len > RARRAY(ary)->aux.capa) { + RARRAY(ary)->aux.capa = len; + REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->aux.capa); } memfill(RARRAY(ary)->ptr, len, val); RARRAY(ary)->len = len; @@ -250,7 +265,7 @@ rb_ary_s_create(argc, argv, klass) RARRAY(ary)->ptr = ALLOC_N(VALUE, argc); MEMCPY(RARRAY(ary)->ptr, argv, VALUE, argc); } - RARRAY(ary)->len = RARRAY(ary)->capa = argc; + RARRAY(ary)->len = RARRAY(ary)->aux.capa = argc; return ary; } @@ -270,13 +285,13 @@ rb_ary_store(ary, idx, val) } } - if (idx >= RARRAY(ary)->capa) { - long capa_inc = RARRAY(ary)->capa / 2; + if (idx >= RARRAY(ary)->aux.capa) { + long capa_inc = RARRAY(ary)->aux.capa / 2; if (capa_inc < ARY_DEFAULT_SIZE) { capa_inc = ARY_DEFAULT_SIZE; } - RARRAY(ary)->capa = idx + capa_inc; - REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->capa); + RARRAY(ary)->aux.capa = idx + capa_inc; + REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->aux.capa); } if (idx > RARRAY(ary)->len) { rb_mem_clear(RARRAY(ary)->ptr+RARRAY(ary)->len, @@ -319,34 +334,45 @@ VALUE rb_ary_pop(ary) VALUE ary; { - rb_ary_modify(ary); + rb_ary_modify_check(ary); if (RARRAY(ary)->len == 0) return Qnil; - if (RARRAY(ary)->len * 10 < RARRAY(ary)->capa && RARRAY(ary)->capa > ARY_DEFAULT_SIZE) { - RARRAY(ary)->capa = RARRAY(ary)->len * 2; - REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->capa); + if (RARRAY(ary)->len * 10 < RARRAY(ary)->aux.capa && RARRAY(ary)->aux.capa > ARY_DEFAULT_SIZE) { + RARRAY(ary)->aux.capa = RARRAY(ary)->len * 2; + REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->aux.capa); } return RARRAY(ary)->ptr[--RARRAY(ary)->len]; } +static void +ary_make_shared(ary) + VALUE ary; +{ + if (FL_TEST(ary, ELTS_SHARED)) return; + else { + NEWOBJ(shared, struct RArray); + OBJSETUP(shared, rb_cArray, T_ARRAY); + + shared->len = RARRAY(ary)->len; + shared->ptr = RARRAY(ary)->ptr; + shared->aux.capa = RARRAY(ary)->aux.capa; + RARRAY(ary)->aux.shared = (VALUE)shared; + FL_SET(ary, ELTS_SHARED); + } +} + VALUE rb_ary_shift(ary) VALUE ary; { VALUE top; - rb_ary_modify(ary); + rb_ary_modify_check(ary); if (RARRAY(ary)->len == 0) return Qnil; - top = RARRAY(ary)->ptr[0]; + ary_make_shared(ary); + RARRAY(ary)->ptr++; /* shift ptr */ RARRAY(ary)->len--; - /* sliding items */ - MEMMOVE(RARRAY(ary)->ptr, RARRAY(ary)->ptr+1, VALUE, RARRAY(ary)->len); - if (RARRAY(ary)->len * 10 < RARRAY(ary)->capa && RARRAY(ary)->capa > ARY_DEFAULT_SIZE) { - RARRAY(ary)->capa = RARRAY(ary)->len * 2; - REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->capa); - } - return top; } @@ -355,13 +381,13 @@ rb_ary_unshift(ary, item) VALUE ary, item; { rb_ary_modify(ary); - if (RARRAY(ary)->len >= RARRAY(ary)->capa) { - long capa_inc = RARRAY(ary)->capa / 2; + if (RARRAY(ary)->len >= RARRAY(ary)->aux.capa) { + long capa_inc = RARRAY(ary)->aux.capa / 2; if (capa_inc < ARY_DEFAULT_SIZE) { capa_inc = ARY_DEFAULT_SIZE; } - RARRAY(ary)->capa+=capa_inc; - REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->capa); + RARRAY(ary)->aux.capa+=capa_inc; + REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->aux.capa); } /* sliding items */ @@ -429,11 +455,14 @@ rb_ary_subseq(ary, beg, len) len = 0; } klass = rb_obj_class(ary); - if (len == 0) return rb_ary_new0(klass,0); + if (len == 0) return ary_new(klass,0); - ary2 = rb_ary_new0(klass, len); - MEMCPY(RARRAY(ary2)->ptr, RARRAY(ary)->ptr+beg, VALUE, len); + ary_make_shared(ary); + ary2 = rb_obj_alloc(klass); + RARRAY(ary2)->ptr = RARRAY(ary)->ptr+beg; RARRAY(ary2)->len = len; + RARRAY(ary2)->aux.shared = RARRAY(ary)->aux.shared; + FL_SET(ary2, ELTS_SHARED); return ary2; } @@ -609,9 +638,9 @@ rb_ary_update(ary, beg, len, rpl) rb_ary_modify(ary); if (beg >= RARRAY(ary)->len) { len = beg + rlen; - if (len >= RARRAY(ary)->capa) { - RARRAY(ary)->capa=len; - REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->capa); + if (len >= RARRAY(ary)->aux.capa) { + RARRAY(ary)->aux.capa=len; + REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->aux.capa); } rb_mem_clear(RARRAY(ary)->ptr+RARRAY(ary)->len, beg-RARRAY(ary)->len); MEMCPY(RARRAY(ary)->ptr+beg, RARRAY(rpl)->ptr, VALUE, rlen); @@ -625,9 +654,9 @@ rb_ary_update(ary, beg, len, rpl) } alen = RARRAY(ary)->len + rlen - len; - if (alen >= RARRAY(ary)->capa) { - RARRAY(ary)->capa = alen; - REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->capa); + if (alen >= RARRAY(ary)->aux.capa) { + RARRAY(ary)->aux.capa = alen; + REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->aux.capa); } if (len != rlen) { @@ -749,32 +778,51 @@ rb_ary_empty_p(ary) } static VALUE -rb_ary_clone(ary) +ary_copy(ary, clone) VALUE ary; + int clone; { - VALUE clone = rb_ary_new2(RARRAY(ary)->len); + VALUE copy; + + ary_make_shared(ary); + copy = rb_obj_alloc(rb_cArray); + if (clone) CLONESETUP(copy, ary); + else DUPSETUP(copy, ary); + RARRAY(copy)->ptr = RARRAY(ary)->ptr; + RARRAY(copy)->len = RARRAY(ary)->len; + RARRAY(copy)->aux.shared = RARRAY(ary)->aux.shared; + FL_SET(copy, ELTS_SHARED); + + return copy; +} - CLONESETUP(clone, ary); - MEMCPY(RARRAY(clone)->ptr, RARRAY(ary)->ptr, VALUE, RARRAY(ary)->len); - RARRAY(clone)->len = RARRAY(ary)->len; - return clone; +static VALUE +rb_ary_clone(ary) + VALUE ary; +{ + return ary_copy(ary, Qtrue); } VALUE rb_ary_dup(ary) VALUE ary; { + return ary_copy(ary, Qfalse); +} + +static VALUE +ary_dup(ary) + VALUE ary; +{ VALUE dup = rb_ary_new2(RARRAY(ary)->len); - OBJSETUP(dup, rb_obj_class(ary), T_ARRAY); + DUPSETUP(dup, ary); MEMCPY(RARRAY(dup)->ptr, RARRAY(ary)->ptr, VALUE, RARRAY(ary)->len); RARRAY(dup)->len = RARRAY(ary)->len; - OBJ_INFECT(dup, ary); return dup; } extern VALUE rb_output_fs; -extern VALUE rb_default_rs; static VALUE inspect_join(ary, arg) @@ -861,9 +909,6 @@ rb_ary_to_s(ary) if (RARRAY(ary)->len == 0) return rb_str_new(0, 0); sep = rb_output_fs; -#if 1 - if (NIL_P(rb_output_fs)) sep = rb_default_rs; /* newline */ -#endif str = rb_ary_join(ary, sep); return str; } @@ -1006,7 +1051,7 @@ static VALUE rb_ary_reverse_m(ary) VALUE ary; { - return rb_ary_reverse(rb_ary_dup(ary)); + return rb_ary_reverse(ary_dup(ary)); } int @@ -1018,8 +1063,8 @@ rb_cmpint(cmp) if (RBIGNUM(cmp)->sign) return 1; return -1; } - if (rb_funcall(cmp, '>', 1, INT2FIX(0))) return 1; - if (rb_funcall(cmp, '<', 1, INT2FIX(0))) return -1; + if (rb_funcall(id_cmp, '>', 1, INT2FIX(0))) return 1; + if (rb_funcall(id_cmp, '<', 1, INT2FIX(0))) return -1; return 0; } @@ -1044,7 +1089,7 @@ sort_2(a, b) return rb_str_cmp(*a, *b); } - retval = rb_funcall(*a, cmp, 1, *b); + retval = rb_funcall(*a, id_cmp, 1, *b); return rb_cmpint(retval); } @@ -1081,32 +1126,12 @@ VALUE rb_ary_sort(ary) VALUE ary; { - ary = rb_ary_dup(ary); + ary = ary_dup(ary); rb_ary_sort_bang(ary); return ary; } static VALUE -sort_inplace(ary) - VALUE ary; -{ - qsort(RARRAY(ary)->ptr, RARRAY(ary)->len, sizeof(VALUE),sort_2); - return ary; -} - -VALUE -rb_ary_sort_inplace(ary) - VALUE ary; -{ - rb_ary_modify(ary); - if (RARRAY(ary)->len <= 1) return ary; - - FL_SET(ary, ARY_TMPLOCK); /* prohibit modification during sort */ - rb_ensure(sort_inplace, ary, sort_unlock, ary); - return ary; -} - -static VALUE rb_ary_collect(ary) VALUE ary; { @@ -1269,7 +1294,7 @@ static VALUE rb_ary_reject(ary) VALUE ary; { - ary = rb_ary_dup(ary); + ary = ary_dup(ary); rb_ary_reject_bang(ary); return ary; } @@ -1297,9 +1322,9 @@ rb_ary_clear(ary) { rb_ary_modify(ary); RARRAY(ary)->len = 0; - if (ARY_DEFAULT_SIZE*3 < RARRAY(ary)->capa) { - RARRAY(ary)->capa = ARY_DEFAULT_SIZE * 2; - REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->capa); + if (ARY_DEFAULT_SIZE*3 < RARRAY(ary)->aux.capa) { + RARRAY(ary)->aux.capa = ARY_DEFAULT_SIZE * 2; + REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->aux.capa); } return ary; } @@ -1337,9 +1362,9 @@ rb_ary_fill(argc, argv, ary) rb_ary_modify(ary); end = beg + len; if (end > RARRAY(ary)->len) { - if (end >= RARRAY(ary)->capa) { - RARRAY(ary)->capa = end; - REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->capa); + if (end >= RARRAY(ary)->aux.capa) { + RARRAY(ary)->aux.capa = end; + REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->aux.capa); } if (beg > RARRAY(ary)->len) { rb_mem_clear(RARRAY(ary)->ptr+RARRAY(ary)->len,end-RARRAY(ary)->len); @@ -1397,7 +1422,7 @@ rb_ary_times(ary, times) } len *= RARRAY(ary)->len; - ary2 = rb_ary_new0(rb_obj_class(ary), len); + ary2 = ary_new(rb_obj_class(ary), len); RARRAY(ary2)->len = len; for (i=0; i<len; i+=RARRAY(ary)->len) { @@ -1521,7 +1546,7 @@ rb_ary_cmp(ary, ary2) len = RARRAY(ary2)->len; } for (i=0; i<len; i++) { - VALUE v = rb_funcall(RARRAY(ary)->ptr[i],cmp,1,RARRAY(ary2)->ptr[i]); + VALUE v = rb_funcall(RARRAY(ary)->ptr[i],id_cmp,1,RARRAY(ary2)->ptr[i]); if (v != INT2FIX(0)) { return v; } @@ -1646,7 +1671,7 @@ static VALUE rb_ary_uniq(ary) VALUE ary; { - ary = rb_ary_dup(ary); + ary = ary_dup(ary); rb_ary_uniq_bang(ary); return ary; } @@ -1667,7 +1692,7 @@ rb_ary_compact_bang(ary) if (RARRAY(ary)->len == (p - RARRAY(ary)->ptr)) { return Qnil; } - RARRAY(ary)->len = RARRAY(ary)->capa = (p - RARRAY(ary)->ptr); + RARRAY(ary)->len = RARRAY(ary)->aux.capa = (p - RARRAY(ary)->ptr); REALLOC_N(RARRAY(ary)->ptr, VALUE, RARRAY(ary)->len); return ary; @@ -1677,7 +1702,7 @@ static VALUE rb_ary_compact(ary) VALUE ary; { - ary = rb_ary_dup(ary); + ary = ary_dup(ary); rb_ary_compact_bang(ary); return ary; } @@ -1755,7 +1780,7 @@ static VALUE rb_ary_flatten(ary) VALUE ary; { - ary = rb_ary_dup(ary); + ary = ary_dup(ary); rb_ary_flatten_bang(ary); return ary; } @@ -1847,5 +1872,5 @@ Init_Array() rb_define_method(rb_cArray, "flatten!", rb_ary_flatten_bang, 0); rb_define_method(rb_cArray, "nitems", rb_ary_nitems, 0); - cmp = rb_intern("<=>"); + id_cmp = rb_intern("<=>"); } |