[ruby-dev:13873] Re: massign to Hash

なかだです。

At Sun, 15 Jul 2001 23:15:37 +0900,
Nobuyoshi-Nakada wrote:
>   そっちは忘れてましたが、Hash#indexesと同じような気がするんで
> すが。

  失礼。引数が0と1のときが違いますね。2個以上のときは
rb_hash_indexes()を使うというのでどうでしょうか。

  それとArgumentErrorのときのメッセージにargcを渡してるのに書式
には入ってませんでしたが、これはコピー元から抜けてたので追加し
ました。

  rb_hash_mset()は最初他からも使えるかと思ったんですが、あんま
り使いそうもないのでstaticにして、関数名も揃えました。


Index: hash.c
===================================================================
RCS file: /cvs/ruby/src/ruby/hash.c,v
retrieving revision 1.49
diff -u -2 -p -r1.49 hash.c
--- hash.c	2001/06/22 09:12:19	1.49
+++ hash.c	2001/07/15 14:37:44
@@ -318,5 +318,5 @@ rb_hash_fetch(argc, argv, hash)
 	if (rb_block_given_p()) {
 	    if (argc > 1) {
-		rb_raise(rb_eArgError, "wrong # of arguments", argc);
+		rb_raise(rb_eArgError, "wrong # of arguments (%d for %d)", argc, 1);
 	    }
 	    return rb_yield(key);
@@ -390,4 +390,20 @@ rb_hash_indexes(argc, argv, hash)
 
 static VALUE
+rb_hash_aref_m(argc, argv, hash)
+    int argc;
+    VALUE *argv;
+    VALUE hash;
+{
+    switch (argc) {
+      case 0:
+	rb_raise(rb_eArgError, "wrong # of arguments (%d for %d)", argc, 1);
+      case 1:
+	return rb_hash_aref(hash, argv[0]);
+      default:
+	return rb_hash_indexes(argc, argv, hash);
+    }
+}
+
+static VALUE
 rb_hash_delete(hash, key)
     VALUE hash, key;
@@ -510,4 +526,44 @@ rb_hash_aset(hash, key, val)
 }
 
+static VALUE
+rb_hash_aset_m(argc, argv, hash)
+    int argc;
+    VALUE *argv;
+    VALUE hash;
+{
+    VALUE values, *ptr = NULL;
+    long len;
+    int i, isary;
+
+    if (argc < 2) {
+	rb_raise(rb_eArgError, "wrong # of arguments (%d for %d)", argc, 2);
+    }
+    else if (argc == 2) {
+	return rb_hash_aset(hash, argv[0], argv[1]);
+    }
+    rb_hash_modify(hash);
+    values = argv[--argc];
+    isary = TYPE(values) == T_ARRAY;
+    if (!isary) {
+	len = 1;
+	ptr = &values;
+    }
+    for (i = 0; i < argc; ++i) {
+	VALUE key = argv[i], val = Qnil;
+	if (isary) {
+	    len = RARRAY(values)->len;
+	    ptr = RARRAY(values)->ptr;
+	}
+	if (i < len) val = ptr[i];
+	if (TYPE(key) != T_STRING || st_lookup(RHASH(hash)->tbl, key, 0)) {
+	    st_insert(RHASH(hash)->tbl, key, val);
+	}
+	else {
+	    st_add_direct(RHASH(hash)->tbl, rb_str_new4(key), val);
+	}
+    }
+    return values;
+}
+
 static int
 replace_i(key, val, hash)
@@ -934,5 +990,5 @@ env_fetch(argc, argv)
 	if (rb_block_given_p()) {
 	    if (argc > 1) {
-		rb_raise(rb_eArgError, "wrong # of arguments", argc);
+		rb_raise(rb_eArgError, "wrong # of arguments (%d for %d)", argc, 1);
 	    }
 	    return rb_yield(key);
@@ -1150,4 +1206,38 @@ rb_f_setenv(obj, nm, val)
 
 static VALUE
+rb_f_setenv_m(argc, argv, obj)
+    int argc;
+    VALUE *argv;
+    VALUE obj;
+{
+    VALUE values, *ptr;
+    long len;
+    int i, isary;
+
+    if (argc < 2) {
+	rb_raise(rb_eArgError, "wrong # of arguments (%d for %d)", argc, 2);
+    }
+    if (argc == 2) {
+	return rb_f_setenv(obj, argv[0], argv[1]);
+    }
+    values = argv[--argc];
+    isary = TYPE(values) == T_ARRAY;
+    if (!isary) {
+	len = 1;
+	ptr = &values;
+    }
+    for (i = 0; i < argc; ++i) {
+	VALUE key = argv[i], val = Qnil;
+	if (isary) {
+	    len = RARRAY(values)->len;
+	    ptr = RARRAY(values)->ptr;
+	}
+	if (i < len) val = ptr[i];
+	rb_f_setenv(obj, key, val);
+    }
+    return values;
+}
+
+static VALUE
 env_keys()
 {
@@ -1422,4 +1512,22 @@ env_indexes(argc, argv)
 
 static VALUE
+rb_f_getenv_m(argc, argv, obj)
+    int argc;
+    VALUE *argv;
+    VALUE obj;
+{
+    VALUE values;
+
+    switch (argc) {
+      case 0:
+	rb_raise(rb_eArgError, "wrong # of arguments (%d for %d)", argc, 1);
+      case 1:
+	return rb_f_getenv(obj, argv[0]);
+      default:
+	return env_indexes(argc, argv);
+    }
+}
+
+static VALUE
 env_to_hash()
 {
@@ -1467,7 +1575,7 @@ Init_Hash()
 
     rb_define_method(rb_cHash,"==", rb_hash_equal, 1);
-    rb_define_method(rb_cHash,"[]", rb_hash_aref, 1);
+    rb_define_method(rb_cHash,"[]", rb_hash_aref_m, -1);
     rb_define_method(rb_cHash,"fetch", rb_hash_fetch, -1);
-    rb_define_method(rb_cHash,"[]=", rb_hash_aset, 2);
+    rb_define_method(rb_cHash,"[]=", rb_hash_aset_m, -1);
     rb_define_method(rb_cHash,"store", rb_hash_aset, 2);
     rb_define_method(rb_cHash,"default", rb_hash_default, 0);
@@ -1511,7 +1619,7 @@ Init_Hash()
     rb_extend_object(envtbl, rb_mEnumerable);
 
-    rb_define_singleton_method(envtbl,"[]", rb_f_getenv, 1);
+    rb_define_singleton_method(envtbl,"[]", rb_f_getenv_m, -1);
     rb_define_singleton_method(envtbl,"fetch", env_fetch, -1);
-    rb_define_singleton_method(envtbl,"[]=", rb_f_setenv, 2);
+    rb_define_singleton_method(envtbl,"[]=", rb_f_setenv_m, -1);
     rb_define_singleton_method(envtbl,"store", rb_f_setenv, 2);
     rb_define_singleton_method(envtbl,"each", env_each, 0);


-- 
--- 僕の前にBugはない。
--- 僕の後ろにBugはできる。
    中田 伸悦