[ruby-dev:13870] massign to Hash

なかだです。

  xlib/tools/Keysyms_lister.rbを見て思ったのですが、Hash#[]=も
Array#[]=のように複数のインデックスを受け付けるようにする、とい
うのはどんなもんでしょうか。

a = {}
a[:a, :b] = *1..2
p a	# => {:b=>2, :a=>1}


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 09:36:52
@@ -510,4 +510,44 @@ rb_hash_aset(hash, key, val)
 }
 
+VALUE
+rb_hash_mset(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", argc);
+    }
+    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)
@@ -1150,4 +1190,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", argc);
+    }
+    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()
 {
@@ -1469,5 +1543,5 @@ Init_Hash()
     rb_define_method(rb_cHash,"[]", rb_hash_aref, 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_mset, -1);
     rb_define_method(rb_cHash,"store", rb_hash_aset, 2);
     rb_define_method(rb_cHash,"default", rb_hash_default, 0);
@@ -1513,5 +1587,5 @@ Init_Hash()
     rb_define_singleton_method(envtbl,"[]", rb_f_getenv, 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はできる。
    中田 伸悦