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

At Sun, 15 Jul 2001 21:44:54 +0900,
[e-mail] wrote:
>   xlib/tools/Keysyms_lister.rbを見て思ったのですが、Hash#[]=も
> Array#[]=のように複数のインデックスを受け付けるようにする、とい
> うのはどんなもんでしょうか。

　いいですね。入れるのができるなら取り出す方も、ということで
Hash#[] もいじってみました。なかださんの版からの差分です。

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

p a[:a, :b]	# => [1, 2]

-- 
                     /
                    /__  __            Akinori.org / MUSHA.org
                   / )  )  ) )  /     FreeBSD.org / Ruby-lang.org
Akinori MUSHA aka / (_ /  ( (__(  @ iDaemons.org / and.or.jp

"Freeze this moment a little bit longer, make each impression
  a little bit stronger..  Experience slips away -- Time stand still"

--- hash.c.orig	Sun Jul 15 22:17:38 2001
+++ hash.c	Sun Jul 15 22:17:50 2001
@@ -304,6 +304,30 @@
 }
 
 static VALUE
+rb_hash_aref_m(argc, argv, obj)
+    int argc;
+    VALUE *argv;
+    VALUE obj;
+{
+    VALUE values;
+
+    if (argc == 0) {
+	rb_raise(rb_eArgError, "wrong # of arguments", argc);
+    }
+    if (argc == 1) {
+	return rb_hash_aref(obj, argv[0]);
+    }
+
+    values = rb_ary_new2(argc);
+
+    while (argc--) {
+	rb_ary_push(values, rb_hash_aref(obj, *argv++));
+    }
+
+    return values;
+}
+
+static VALUE
 rb_hash_fetch(argc, argv, hash)
     int argc;
     VALUE *argv;
@@ -956,6 +980,30 @@
 }
 
 static VALUE
+rb_f_getenv_m(argc, argv, obj)
+    int argc;
+    VALUE *argv;
+    VALUE obj;
+{
+    VALUE values;
+
+    if (argc == 0) {
+	rb_raise(rb_eArgError, "wrong # of arguments", argc);
+    }
+    if (argc == 1) {
+	return rb_f_getenv(obj, argv[0]);
+    }
+
+    values = rb_ary_new2(argc);
+
+    while (argc--) {
+	rb_ary_push(values, rb_f_getenv(obj, *argv++));
+    }
+
+    return values;
+}
+
+static VALUE
 env_fetch(argc, argv)
     int argc;
     VALUE *argv;
@@ -1540,7 +1588,7 @@
     rb_define_method(rb_cHash,"inspect", rb_hash_inspect, 0);
 
     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_mset, -1);
     rb_define_method(rb_cHash,"store", rb_hash_aset, 2);
@@ -1584,7 +1632,7 @@
     envtbl = rb_obj_alloc(rb_cObject);
     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_m, -1);
     rb_define_singleton_method(envtbl,"store", rb_f_setenv, 2);