* 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

1 files changed, 117 insertions, 92 deletions

diff --git a/array.c b/array.c
index 6a419d800e..0401dd3598 100644
--- a/array.c
+++ b/array.c
@@ -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("<=>");
 }