summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ChangeLog17
-rw-r--r--array.c131
-rw-r--r--gc.c10
-rw-r--r--intern.h6
-rw-r--r--string.c8
5 files changed, 124 insertions, 48 deletions
diff --git a/ChangeLog b/ChangeLog
index 8331f8c7a8..2b8dcb767d 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -4,6 +4,23 @@ Wed Sep 27 01:04:49 2006 Nobuyoshi Nakada <nobu@ruby-lang.org>
* lib/mkmf.rb (have_type): simplified with typedef and sizeof.
+Wed Sep 27 00:08:12 2006 Yukihiro Matsumoto <matz@ruby-lang.org>
+
+ * array.c (rb_ary_shift): shift/unshift performance boost patch,
+ based on the patch from Eric Mahurin <eric_mahurin at yahoo.com>.
+ [ruby-core:05861]
+
+ * array.c (rb_ary_unshift_m): ditto.
+
+ * array.c (ary_make_shared): ditto.
+
+ * array.c (RESIZE_CAPA): ditto.
+
+ * array.c (rb_ary_free): new function to free memory. code moved
+ from gc.c.
+
+ * string.c (rb_str_free): ditto.
+
Tue Sep 26 23:57:03 2006 Nobuyoshi Nakada <nobu@ruby-lang.org>
* lib/optparse.rb (OptionParser#getopts): use strings as key.
diff --git a/array.c b/array.c
index 4896ed24fb..a60df5d432 100644
--- a/array.c
+++ b/array.c
@@ -64,6 +64,11 @@ memfill(register VALUE *mem, register long size, register VALUE val)
}\
} while (0)
+#define ARY_LFREE FL_USER6
+#define ARY_LFREE_P(ary) FL_TEST(ary, ARY_LFREE)
+#define LFREE_SIZE(ary) RARRAY(ary)->as.heap.ptr[-1]
+#define LFREE_CAPA(ary) (LFREE_SIZE(ary)+RARRAY(ary)->as.heap.aux.capa)
+
#define ARY_CAPA(ary) ((ARY_EMBED_P(ary)) ? RARRAY_EMBED_LEN_MAX : RARRAY(ary)->as.heap.aux.capa)
#define RESIZE_CAPA(ary,capacity) do {\
if (ARY_EMBED_P(ary)) {\
@@ -77,6 +82,21 @@ memfill(register VALUE *mem, register long size, register VALUE val)
RARRAY(ary)->as.heap.aux.capa = (capacity);\
}\
}\
+ else if (ARY_LFREE_P(ary)) {\
+ VALUE *ptr = RARRAY(ary)->as.heap.ptr - LFREE_SIZE(ary);\
+ if (LFREE_CAPA(ary) >= (capacity)) {\
+ RARRAY(ary)->as.heap.aux.capa = LFREE_CAPA(ary);\
+ MEMMOVE(ptr, RARRAY(ary)->as.heap.ptr, VALUE, RARRAY_LEN(ary));\
+ FL_UNSET(ary, ARY_LFREE);\
+ RARRAY(ary)->as.heap.ptr = ptr;\
+ }\
+ else {\
+ long offset = LFREE_SIZE(ary);\
+ REALLOC_N(ptr, VALUE, offset+(capacity));\
+ RARRAY(ary)->as.heap.aux.capa = (capacity);\
+ RARRAY(ary)->as.heap.ptr = ptr + offset;\
+ }\
+ }\
else {\
REALLOC_N(RARRAY(ary)->as.heap.ptr, VALUE, (capacity));\
RARRAY(ary)->as.heap.aux.capa = (capacity);\
@@ -223,6 +243,19 @@ rb_ary_new4(long n, const VALUE *elts)
return ary;
}
+void
+rb_ary_free(VALUE ary)
+{
+ if (!ARY_SHARED_P(ary) && !ARY_EMBED_P(ary)) {
+ if (ARY_LFREE_P(ary)) {
+ xfree(RARRAY(ary)->as.heap.ptr - RARRAY(ary)->as.heap.ptr[-1]);
+ }
+ else {
+ xfree(RARRAY(ary)->as.heap.ptr);
+ }
+ }
+}
+
static VALUE
ary_make_shared(VALUE ary)
{
@@ -238,6 +271,10 @@ ary_make_shared(VALUE ary)
shared->as.heap.len = RARRAY(ary)->as.heap.len;
shared->as.heap.ptr = RARRAY(ary)->as.heap.ptr;
shared->as.heap.aux.capa = RARRAY(ary)->as.heap.aux.capa;
+ if (ARY_LFREE_P(ary)) {
+ FL_SET(shared,ARY_LFREE);
+ FL_UNSET(ary,ARY_LFREE);
+ }
RARRAY(ary)->as.heap.aux.shared = (VALUE)shared;
FL_SET(ary, ELTS_SHARED);
OBJ_FREEZE(shared);
@@ -578,18 +615,23 @@ rb_ary_shift(VALUE ary)
rb_ary_modify_check(ary);
if (RARRAY_LEN(ary) == 0) return Qnil;
top = RARRAY_PTR(ary)[0];
- if (RARRAY_LEN(ary) < ARY_DEFAULT_SIZE && !FL_TEST(ary, ELTS_SHARED)) {
+ if (ARY_EMBED_P(ary)) {
MEMMOVE(RARRAY_PTR(ary), RARRAY_PTR(ary)+1, VALUE, RARRAY_LEN(ary));
- ARY_SET_LEN(ary, RARRAY_LEN(ary)-1);
+ ARY_SET_EMBED_LEN(ary, RARRAY_LEN(ary)-1);
+ return top;
}
- else {
- if (!FL_TEST(ary, ELTS_SHARED)) {
- RARRAY_PTR(ary)[0] = Qnil;
+ if (!ARY_SHARED_P(ary)) {
+ if (ARY_LFREE_P(ary)) {
+ RARRAY(ary)->as.heap.ptr[0] = RARRAY(ary)->as.heap.ptr[-1]+1;
+ }
+ else {
+ FL_SET(ary, ARY_LFREE);
+ RARRAY(ary)->as.heap.ptr[0] = 1;
}
- ary_make_shared(ary);
- RARRAY(ary)->as.heap.ptr++; /* shift ptr */
- RARRAY(ary)->as.heap.len--;
+ RARRAY(ary)->as.heap.aux.capa--;
}
+ RARRAY(ary)->as.heap.ptr++; /* shift ptr */
+ RARRAY(ary)->as.heap.len--;
return top;
}
@@ -640,26 +682,6 @@ rb_ary_shift_m(int argc, VALUE *argv, VALUE ary)
return result;
}
-VALUE
-rb_ary_unshift(VALUE ary, VALUE item)
-{
- rb_ary_modify(ary);
- if (RARRAY_LEN(ary) == ARY_CAPA(ary)) {
- long capa_inc = ARY_CAPA(ary) / 2;
- if (capa_inc < ARY_DEFAULT_SIZE) {
- capa_inc = ARY_DEFAULT_SIZE;
- }
- RESIZE_CAPA(ary, ARY_CAPA(ary)+capa_inc);
- }
-
- /* sliding items */
- MEMMOVE(RARRAY_PTR(ary) + 1, RARRAY_PTR(ary), VALUE, RARRAY_LEN(ary));
- ARY_SET_LEN(ary, RARRAY_LEN(ary)+1);
- RARRAY_PTR(ary)[0] = item;
-
- return ary;
-}
-
/*
* call-seq:
* array.unshift(obj, ...) -> array
@@ -675,20 +697,57 @@ rb_ary_unshift(VALUE ary, VALUE item)
static VALUE
rb_ary_unshift_m(int argc, VALUE *argv, VALUE ary)
{
- long len = RARRAY_LEN(ary);
-
+ long lfree = ARY_LFREE_P(ary) ? RARRAY_PTR(ary)[-1] : 0;
+
+ rb_ary_modify_check(ary);
if (argc == 0) return ary;
- /* make rooms by setting the last item */
- rb_ary_store(ary, len + argc - 1, Qnil);
-
- /* sliding items */
- MEMMOVE(RARRAY_PTR(ary) + argc, RARRAY_PTR(ary), VALUE, len);
+ if (lfree < argc) {
+ int shared = ARY_SHARED_P(ary);
+ long len = RARRAY_LEN(ary);
+ long rfree = shared ? 0 : (ARY_CAPA(ary)-len)/2;
+ long free2 = argc+(RARRAY_LEN(ary)+argc)/2;
+ VALUE *ptr;
+ if (free2 < ARY_DEFAULT_SIZE) {
+ free2 = ARY_DEFAULT_SIZE;
+ }
+ ptr = ALLOC_N(VALUE,len+free2)+(free2-rfree);
+ MEMCPY(ptr, RARRAY_PTR(ary), VALUE, len);
+ if (shared) {
+ FL_UNSET(ary, ELTS_SHARED);
+ }
+ else if (ARY_EMBED_P(ary)) {
+ ARY_SET_NOEMBED(ary);
+ }
+ else {
+ free(RARRAY_PTR(ary)-lfree);
+ }
+ RARRAY(ary)->as.heap.ptr = ptr;
+ RARRAY(ary)->as.heap.len = len;
+ RARRAY(ary)->as.heap.aux.capa = len+rfree;
+ lfree = free2-rfree;
+ FL_SET(ary, ARY_LFREE);
+ }
+ RARRAY(ary)->as.heap.ptr -= argc;
+ RARRAY(ary)->as.heap.len += argc;
+ RARRAY(ary)->as.heap.aux.capa += argc;
+ lfree -= argc;
+ if (lfree > 0) {
+ RARRAY(ary)->as.heap.ptr[-1] = lfree;
+ } else {
+ FL_UNSET(ary, ARY_LFREE);
+ }
MEMCPY(RARRAY_PTR(ary), argv, VALUE, argc);
-
+
return ary;
}
+VALUE
+rb_ary_unshift(VALUE ary, VALUE item)
+{
+ return rb_ary_unshift_m(1,&item,ary);
+}
+
/* faster version - use this if you don't need to treat negative offset */
static inline VALUE
rb_ary_elt(VALUE ary, long offset)
diff --git a/gc.c b/gc.c
index 6c83bd43b5..88de1d506a 100644
--- a/gc.c
+++ b/gc.c
@@ -1174,16 +1174,10 @@ obj_free(VALUE obj)
}
break;
case T_STRING:
- if (FL_TEST(obj, RSTRING_NOEMBED) &&
- RANY(obj)->as.string.as.heap.ptr && !FL_TEST(obj, ELTS_SHARED)) {
- RUBY_CRITICAL(free(RANY(obj)->as.string.as.heap.ptr));
- }
+ rb_str_free(obj);
break;
case T_ARRAY:
- if (FL_TEST(obj, RARRAY_NOEMBED) &&
- RANY(obj)->as.array.as.heap.ptr && !FL_TEST(obj, ELTS_SHARED)) {
- RUBY_CRITICAL(free(RANY(obj)->as.array.as.heap.ptr));
- }
+ rb_ary_free(obj);
break;
case T_HASH:
if (RANY(obj)->as.hash.tbl) {
diff --git a/intern.h b/intern.h
index 5b2ce7d73e..c1461171af 100644
--- a/intern.h
+++ b/intern.h
@@ -37,10 +37,7 @@ VALUE rb_ary_new(void);
VALUE rb_ary_new2(long);
VALUE rb_ary_new3(long,...);
VALUE rb_ary_new4(long, const VALUE *);
-VALUE rb_values_new(long,...);
-VALUE rb_values_new2(long, const VALUE *);
-VALUE rb_values_from_ary(VALUE);
-VALUE rb_ary_from_values(VALUE);
+void rb_ary_free(VALUE);
VALUE rb_ary_freeze(VALUE);
VALUE rb_ary_aref(int, VALUE*, VALUE);
void rb_ary_store(VALUE, long, VALUE);
@@ -488,6 +485,7 @@ VALUE rb_tainted_str_new(const char*, long);
VALUE rb_tainted_str_new2(const char*);
VALUE rb_str_buf_new(long);
VALUE rb_str_buf_new2(const char*);
+void rb_str_free(VALUE);
VALUE rb_str_buf_append(VALUE, VALUE);
VALUE rb_str_buf_cat(VALUE, const char*, long);
VALUE rb_str_buf_cat2(VALUE, const char*);
diff --git a/string.c b/string.c
index 8cd93a69b5..9a45d2f5ce 100644
--- a/string.c
+++ b/string.c
@@ -299,6 +299,14 @@ rb_str_buf_new2(const char *ptr)
return str;
}
+void
+rb_str_free(VALUE str)
+{
+ if (!STR_EMBED_P(str) && !STR_SHARED_P(str)) {
+ xfree(RSTRING(str)->as.heap.ptr);
+ }
+}
+
VALUE
rb_str_to_str(VALUE str)
{