source: EcnlProtoTool/trunk/mruby-1.2.0/src/hash.c@ 270

Last change on this file since 270 was 270, checked in by coas-nagasima, 7 years ago

mruby版ECNLプロトタイピング・ツールを追加

  • Property svn:eol-style set to native
  • Property svn:keywords set to Id
  • Property svn:mime-type set to text/x-csrc
File size: 21.4 KB
Line 
1/*
2** hash.c - Hash class
3**
4** See Copyright Notice in mruby.h
5*/
6
7#include "mruby.h"
8#include "mruby/array.h"
9#include "mruby/class.h"
10#include "mruby/hash.h"
11#include "mruby/khash.h"
12#include "mruby/string.h"
13#include "mruby/variable.h"
14
15/* a function to get hash value of a float number */
16mrb_int mrb_float_id(mrb_float f);
17
18static inline khint_t
19mrb_hash_ht_hash_func(mrb_state *mrb, mrb_value key)
20{
21 enum mrb_vtype t = mrb_type(key);
22 mrb_value hv;
23 const char *p;
24 mrb_int i, len;
25 khint_t h;
26
27 switch (t) {
28 case MRB_TT_STRING:
29 p = RSTRING_PTR(key);
30 len = RSTRING_LEN(key);
31 h = 0;
32 for (i=0; i<len; i++) {
33 h = (h << 5) - h + *p++;
34 }
35 return h;
36
37 case MRB_TT_SYMBOL:
38 h = (khint_t)mrb_symbol(key);
39 return kh_int_hash_func(mrb, h);
40
41 case MRB_TT_FIXNUM:
42 h = (khint_t)mrb_float_id((mrb_float)mrb_fixnum(key));
43 return kh_int_hash_func(mrb, h);
44
45 case MRB_TT_FLOAT:
46 h = (khint_t)mrb_float_id(mrb_float(key));
47 return kh_int_hash_func(mrb, h);
48
49 default:
50 hv = mrb_funcall(mrb, key, "hash", 0);
51 h = (khint_t)t ^ mrb_fixnum(hv);
52 return kh_int_hash_func(mrb, h);
53 }
54}
55
56static inline khint_t
57mrb_hash_ht_hash_equal(mrb_state *mrb, mrb_value a, mrb_value b)
58{
59 enum mrb_vtype t = mrb_type(a);
60
61 switch (t) {
62 case MRB_TT_STRING:
63 return mrb_str_equal(mrb, a, b);
64
65 case MRB_TT_SYMBOL:
66 if (mrb_type(b) != MRB_TT_SYMBOL) return FALSE;
67 return mrb_symbol(a) == mrb_symbol(b);
68
69 case MRB_TT_FIXNUM:
70 switch (mrb_type(b)) {
71 case MRB_TT_FIXNUM:
72 return mrb_fixnum(a) == mrb_fixnum(b);
73 case MRB_TT_FLOAT:
74 return (mrb_float)mrb_fixnum(a) == mrb_float(b);
75 default:
76 return FALSE;
77 }
78
79 case MRB_TT_FLOAT:
80 switch (mrb_type(b)) {
81 case MRB_TT_FIXNUM:
82 return mrb_float(a) == (mrb_float)mrb_fixnum(b);
83 case MRB_TT_FLOAT:
84 return mrb_float(a) == mrb_float(b);
85 default:
86 return FALSE;
87 }
88
89 default:
90 return mrb_eql(mrb, a, b);
91 }
92}
93
94typedef struct {
95 mrb_value v;
96 mrb_int n;
97} mrb_hash_value;
98
99KHASH_DECLARE(ht, mrb_value, mrb_hash_value, TRUE)
100KHASH_DEFINE (ht, mrb_value, mrb_hash_value, TRUE, mrb_hash_ht_hash_func, mrb_hash_ht_hash_equal)
101
102static void mrb_hash_modify(mrb_state *mrb, mrb_value hash);
103
104static inline mrb_value
105mrb_hash_ht_key(mrb_state *mrb, mrb_value key)
106{
107 if (mrb_string_p(key) && !RSTR_FROZEN_P(mrb_str_ptr(key))) {
108 key = mrb_str_dup(mrb, key);
109 RSTR_SET_FROZEN_FLAG(mrb_str_ptr(key));
110 }
111 return key;
112}
113
114#define KEY(key) mrb_hash_ht_key(mrb, key)
115
116void
117mrb_gc_mark_hash(mrb_state *mrb, struct RHash *hash)
118{
119 khiter_t k;
120 khash_t(ht) *h = hash->ht;
121
122 if (!h) return;
123 for (k = kh_begin(h); k != kh_end(h); k++) {
124 if (kh_exist(h, k)) {
125 mrb_value key = kh_key(h, k);
126 mrb_value val = kh_value(h, k).v;
127
128 mrb_gc_mark_value(mrb, key);
129 mrb_gc_mark_value(mrb, val);
130 }
131 }
132}
133
134size_t
135mrb_gc_mark_hash_size(mrb_state *mrb, struct RHash *hash)
136{
137 if (!hash->ht) return 0;
138 return kh_size(hash->ht)*2;
139}
140
141void
142mrb_gc_free_hash(mrb_state *mrb, struct RHash *hash)
143{
144 if (hash->ht) kh_destroy(ht, mrb, hash->ht);
145}
146
147
148MRB_API mrb_value
149mrb_hash_new_capa(mrb_state *mrb, int capa)
150{
151 struct RHash *h;
152
153 h = (struct RHash*)mrb_obj_alloc(mrb, MRB_TT_HASH, mrb->hash_class);
154 h->ht = kh_init(ht, mrb);
155 if (capa > 0) {
156 kh_resize(ht, mrb, h->ht, capa);
157 }
158 h->iv = 0;
159 return mrb_obj_value(h);
160}
161
162MRB_API mrb_value
163mrb_hash_new(mrb_state *mrb)
164{
165 return mrb_hash_new_capa(mrb, 0);
166}
167
168MRB_API mrb_value
169mrb_hash_get(mrb_state *mrb, mrb_value hash, mrb_value key)
170{
171 khash_t(ht) *h = RHASH_TBL(hash);
172 khiter_t k;
173
174 if (h) {
175 k = kh_get(ht, mrb, h, key);
176 if (k != kh_end(h))
177 return kh_value(h, k).v;
178 }
179
180 /* not found */
181 if (MRB_RHASH_PROCDEFAULT_P(hash)) {
182 return mrb_funcall(mrb, RHASH_PROCDEFAULT(hash), "call", 2, hash, key);
183 }
184 return RHASH_IFNONE(hash);
185}
186
187MRB_API mrb_value
188mrb_hash_fetch(mrb_state *mrb, mrb_value hash, mrb_value key, mrb_value def)
189{
190 khash_t(ht) *h = RHASH_TBL(hash);
191 khiter_t k;
192
193 if (h) {
194 k = kh_get(ht, mrb, h, key);
195 if (k != kh_end(h))
196 return kh_value(h, k).v;
197 }
198
199 /* not found */
200 return def;
201}
202
203MRB_API void
204mrb_hash_set(mrb_state *mrb, mrb_value hash, mrb_value key, mrb_value val)
205{
206 khash_t(ht) *h;
207 khiter_t k;
208 int r;
209
210 mrb_hash_modify(mrb, hash);
211 h = RHASH_TBL(hash);
212
213 if (!h) h = RHASH_TBL(hash) = kh_init(ht, mrb);
214 k = kh_put2(ht, mrb, h, key, &r);
215 kh_value(h, k).v = val;
216
217 if (r != 0) {
218 /* expand */
219 int ai = mrb_gc_arena_save(mrb);
220 key = kh_key(h, k) = KEY(key);
221 mrb_gc_arena_restore(mrb, ai);
222 kh_value(h, k).n = kh_size(h)-1;
223 }
224
225 mrb_field_write_barrier_value(mrb, (struct RBasic*)RHASH(hash), key);
226 mrb_field_write_barrier_value(mrb, (struct RBasic*)RHASH(hash), val);
227 return;
228}
229
230static mrb_value
231mrb_hash_dup(mrb_state *mrb, mrb_value hash)
232{
233 struct RHash* ret;
234 khash_t(ht) *h, *ret_h;
235 khiter_t k, ret_k;
236
237 h = RHASH_TBL(hash);
238 ret = (struct RHash*)mrb_obj_alloc(mrb, MRB_TT_HASH, mrb->hash_class);
239 ret->ht = kh_init(ht, mrb);
240
241 if (kh_size(h) > 0) {
242 ret_h = ret->ht;
243
244 for (k = kh_begin(h); k != kh_end(h); k++) {
245 if (kh_exist(h, k)) {
246 int ai = mrb_gc_arena_save(mrb);
247 ret_k = kh_put(ht, mrb, ret_h, KEY(kh_key(h, k)));
248 mrb_gc_arena_restore(mrb, ai);
249 kh_val(ret_h, ret_k) = kh_val(h, k);
250 }
251 }
252 }
253
254 return mrb_obj_value(ret);
255}
256
257MRB_API mrb_value
258mrb_check_hash_type(mrb_state *mrb, mrb_value hash)
259{
260 return mrb_check_convert_type(mrb, hash, MRB_TT_HASH, "Hash", "to_hash");
261}
262
263MRB_API khash_t(ht)*
264mrb_hash_tbl(mrb_state *mrb, mrb_value hash)
265{
266 khash_t(ht) *h = RHASH_TBL(hash);
267
268 if (!h) {
269 return RHASH_TBL(hash) = kh_init(ht, mrb);
270 }
271 return h;
272}
273
274static void
275mrb_hash_modify(mrb_state *mrb, mrb_value hash)
276{
277 mrb_hash_tbl(mrb, hash);
278}
279
280/* 15.2.13.4.16 */
281/*
282 * call-seq:
283 * Hash.new -> new_hash
284 * Hash.new(obj) -> new_hash
285 * Hash.new {|hash, key| block } -> new_hash
286 *
287 * Returns a new, empty hash. If this hash is subsequently accessed by
288 * a key that doesn't correspond to a hash entry, the value returned
289 * depends on the style of <code>new</code> used to create the hash. In
290 * the first form, the access returns <code>nil</code>. If
291 * <i>obj</i> is specified, this single object will be used for
292 * all <em>default values</em>. If a block is specified, it will be
293 * called with the hash object and the key, and should return the
294 * default value. It is the block's responsibility to store the value
295 * in the hash if required.
296 *
297 * h = Hash.new("Go Fish")
298 * h["a"] = 100
299 * h["b"] = 200
300 * h["a"] #=> 100
301 * h["c"] #=> "Go Fish"
302 * # The following alters the single default object
303 * h["c"].upcase! #=> "GO FISH"
304 * h["d"] #=> "GO FISH"
305 * h.keys #=> ["a", "b"]
306 *
307 * # While this creates a new default object each time
308 * h = Hash.new { |hash, key| hash[key] = "Go Fish: #{key}" }
309 * h["c"] #=> "Go Fish: c"
310 * h["c"].upcase! #=> "GO FISH: C"
311 * h["d"] #=> "Go Fish: d"
312 * h.keys #=> ["c", "d"]
313 *
314 */
315
316static mrb_value
317mrb_hash_init(mrb_state *mrb, mrb_value hash)
318{
319 mrb_value block, ifnone;
320 mrb_bool ifnone_p;
321
322 ifnone = mrb_nil_value();
323 mrb_get_args(mrb, "&|o?", &block, &ifnone, &ifnone_p);
324 mrb_hash_modify(mrb, hash);
325 if (!mrb_nil_p(block)) {
326 if (ifnone_p) {
327 mrb_raise(mrb, E_ARGUMENT_ERROR, "wrong number of arguments");
328 }
329 RHASH(hash)->flags |= MRB_HASH_PROC_DEFAULT;
330 ifnone = block;
331 }
332 mrb_iv_set(mrb, hash, mrb_intern_lit(mrb, "ifnone"), ifnone);
333 return hash;
334}
335
336/* 15.2.13.4.2 */
337/*
338 * call-seq:
339 * hsh[key] -> value
340 *
341 * Element Reference---Retrieves the <i>value</i> object corresponding
342 * to the <i>key</i> object. If not found, returns the default value (see
343 * <code>Hash::new</code> for details).
344 *
345 * h = { "a" => 100, "b" => 200 }
346 * h["a"] #=> 100
347 * h["c"] #=> nil
348 *
349 */
350static mrb_value
351mrb_hash_aget(mrb_state *mrb, mrb_value self)
352{
353 mrb_value key;
354
355 mrb_get_args(mrb, "o", &key);
356 return mrb_hash_get(mrb, self, key);
357}
358
359/* 15.2.13.4.5 */
360/*
361 * call-seq:
362 * hsh.default(key=nil) -> obj
363 *
364 * Returns the default value, the value that would be returned by
365 * <i>hsh</i>[<i>key</i>] if <i>key</i> did not exist in <i>hsh</i>.
366 * See also <code>Hash::new</code> and <code>Hash#default=</code>.
367 *
368 * h = Hash.new #=> {}
369 * h.default #=> nil
370 * h.default(2) #=> nil
371 *
372 * h = Hash.new("cat") #=> {}
373 * h.default #=> "cat"
374 * h.default(2) #=> "cat"
375 *
376 * h = Hash.new {|h,k| h[k] = k.to_i*10} #=> {}
377 * h.default #=> nil
378 * h.default(2) #=> 20
379 */
380
381static mrb_value
382mrb_hash_default(mrb_state *mrb, mrb_value hash)
383{
384 mrb_value key;
385 mrb_bool given;
386
387 mrb_get_args(mrb, "|o?", &key, &given);
388 if (MRB_RHASH_PROCDEFAULT_P(hash)) {
389 if (!given) return mrb_nil_value();
390 return mrb_funcall(mrb, RHASH_PROCDEFAULT(hash), "call", 2, hash, key);
391 }
392 else {
393 return RHASH_IFNONE(hash);
394 }
395}
396
397/* 15.2.13.4.6 */
398/*
399 * call-seq:
400 * hsh.default = obj -> obj
401 *
402 * Sets the default value, the value returned for a key that does not
403 * exist in the hash. It is not possible to set the default to a
404 * <code>Proc</code> that will be executed on each key lookup.
405 *
406 * h = { "a" => 100, "b" => 200 }
407 * h.default = "Go fish"
408 * h["a"] #=> 100
409 * h["z"] #=> "Go fish"
410 * # This doesn't do what you might hope...
411 * h.default = proc do |hash, key|
412 * hash[key] = key + key
413 * end
414 * h[2] #=> #<Proc:0x401b3948@-:6>
415 * h["cat"] #=> #<Proc:0x401b3948@-:6>
416 */
417
418static mrb_value
419mrb_hash_set_default(mrb_state *mrb, mrb_value hash)
420{
421 mrb_value ifnone;
422
423 mrb_get_args(mrb, "o", &ifnone);
424 mrb_hash_modify(mrb, hash);
425 mrb_iv_set(mrb, hash, mrb_intern_lit(mrb, "ifnone"), ifnone);
426 RHASH(hash)->flags &= ~(MRB_HASH_PROC_DEFAULT);
427
428 return ifnone;
429}
430
431/* 15.2.13.4.7 */
432/*
433 * call-seq:
434 * hsh.default_proc -> anObject
435 *
436 * If <code>Hash::new</code> was invoked with a block, return that
437 * block, otherwise return <code>nil</code>.
438 *
439 * h = Hash.new {|h,k| h[k] = k*k } #=> {}
440 * p = h.default_proc #=> #<Proc:0x401b3d08@-:1>
441 * a = [] #=> []
442 * p.call(a, 2)
443 * a #=> [nil, nil, 4]
444 */
445
446
447static mrb_value
448mrb_hash_default_proc(mrb_state *mrb, mrb_value hash)
449{
450 if (MRB_RHASH_PROCDEFAULT_P(hash)) {
451 return RHASH_PROCDEFAULT(hash);
452 }
453 return mrb_nil_value();
454}
455
456/*
457 * call-seq:
458 * hsh.default_proc = proc_obj -> proc_obj
459 *
460 * Sets the default proc to be executed on each key lookup.
461 *
462 * h.default_proc = proc do |hash, key|
463 * hash[key] = key + key
464 * end
465 * h[2] #=> 4
466 * h["cat"] #=> "catcat"
467 */
468
469static mrb_value
470mrb_hash_set_default_proc(mrb_state *mrb, mrb_value hash)
471{
472 mrb_value ifnone;
473
474 mrb_get_args(mrb, "o", &ifnone);
475 mrb_hash_modify(mrb, hash);
476 mrb_iv_set(mrb, hash, mrb_intern_lit(mrb, "ifnone"), ifnone);
477 RHASH(hash)->flags |= MRB_HASH_PROC_DEFAULT;
478
479 return ifnone;
480}
481
482MRB_API mrb_value
483mrb_hash_delete_key(mrb_state *mrb, mrb_value hash, mrb_value key)
484{
485 khash_t(ht) *h = RHASH_TBL(hash);
486 khiter_t k;
487 mrb_value delVal;
488 mrb_int n;
489
490 if (h) {
491 k = kh_get(ht, mrb, h, key);
492 if (k != kh_end(h)) {
493 delVal = kh_value(h, k).v;
494 n = kh_value(h, k).n;
495 kh_del(ht, mrb, h, k);
496 for (k = kh_begin(h); k != kh_end(h); k++) {
497 if (!kh_exist(h, k)) continue;
498 if (kh_value(h, k).n > n) kh_value(h, k).n--;
499 }
500 return delVal;
501 }
502 }
503
504 /* not found */
505 return mrb_nil_value();
506}
507
508/* 15.2.13.4.8 */
509/*
510 * call-seq:
511 * hsh.delete(key) -> value
512 * hsh.delete(key) {| key | block } -> value
513 *
514 * Deletes and returns a key-value pair from <i>hsh</i> whose key is
515 * equal to <i>key</i>. If the key is not found, returns the
516 * <em>default value</em>. If the optional code block is given and the
517 * key is not found, pass in the key and return the result of
518 * <i>block</i>.
519 *
520 * h = { "a" => 100, "b" => 200 }
521 * h.delete("a") #=> 100
522 * h.delete("z") #=> nil
523 * h.delete("z") { |el| "#{el} not found" } #=> "z not found"
524 *
525 */
526static mrb_value
527mrb_hash_delete(mrb_state *mrb, mrb_value self)
528{
529 mrb_value key;
530
531 mrb_get_args(mrb, "o", &key);
532 return mrb_hash_delete_key(mrb, self, key);
533}
534
535/* 15.2.13.4.24 */
536/*
537 * call-seq:
538 * hsh.shift -> anArray or obj
539 *
540 * Removes a key-value pair from <i>hsh</i> and returns it as the
541 * two-item array <code>[</code> <i>key, value</i> <code>]</code>, or
542 * the hash's default value if the hash is empty.
543 *
544 * h = { 1 => "a", 2 => "b", 3 => "c" }
545 * h.shift #=> [1, "a"]
546 * h #=> {2=>"b", 3=>"c"}
547 */
548
549static mrb_value
550mrb_hash_shift(mrb_state *mrb, mrb_value hash)
551{
552 khash_t(ht) *h = RHASH_TBL(hash);
553 khiter_t k;
554 mrb_value delKey, delVal;
555
556 mrb_hash_modify(mrb, hash);
557 if (h && kh_size(h) > 0) {
558 for (k = kh_begin(h); k != kh_end(h); k++) {
559 if (!kh_exist(h, k)) continue;
560
561 delKey = kh_key(h, k);
562 mrb_gc_protect(mrb, delKey);
563 delVal = mrb_hash_delete_key(mrb, hash, delKey);
564 mrb_gc_protect(mrb, delVal);
565
566 return mrb_assoc_new(mrb, delKey, delVal);
567 }
568 }
569
570 if (MRB_RHASH_PROCDEFAULT_P(hash)) {
571 return mrb_funcall(mrb, RHASH_PROCDEFAULT(hash), "call", 2, hash, mrb_nil_value());
572 }
573 else {
574 return RHASH_IFNONE(hash);
575 }
576}
577
578/* 15.2.13.4.4 */
579/*
580 * call-seq:
581 * hsh.clear -> hsh
582 *
583 * Removes all key-value pairs from `hsh`.
584 *
585 * h = { "a" => 100, "b" => 200 } #=> {"a"=>100, "b"=>200}
586 * h.clear #=> {}
587 *
588 */
589
590MRB_API mrb_value
591mrb_hash_clear(mrb_state *mrb, mrb_value hash)
592{
593 khash_t(ht) *h = RHASH_TBL(hash);
594
595 if (h) kh_clear(ht, mrb, h);
596 return hash;
597}
598
599/* 15.2.13.4.3 */
600/* 15.2.13.4.26 */
601/*
602 * call-seq:
603 * hsh[key] = value -> value
604 * hsh.store(key, value) -> value
605 *
606 * Element Assignment---Associates the value given by
607 * <i>value</i> with the key given by <i>key</i>.
608 * <i>key</i> should not have its value changed while it is in
609 * use as a key (a <code>String</code> passed as a key will be
610 * duplicated and frozen).
611 *
612 * h = { "a" => 100, "b" => 200 }
613 * h["a"] = 9
614 * h["c"] = 4
615 * h #=> {"a"=>9, "b"=>200, "c"=>4}
616 *
617 */
618static mrb_value
619mrb_hash_aset(mrb_state *mrb, mrb_value self)
620{
621 mrb_value key, val;
622
623 mrb_get_args(mrb, "oo", &key, &val);
624 mrb_hash_set(mrb, self, key, val);
625 return val;
626}
627
628/* 15.2.13.4.20 */
629/* 15.2.13.4.25 */
630/*
631 * call-seq:
632 * hsh.length -> fixnum
633 * hsh.size -> fixnum
634 *
635 * Returns the number of key-value pairs in the hash.
636 *
637 * h = { "d" => 100, "a" => 200, "v" => 300, "e" => 400 }
638 * h.length #=> 4
639 * h.delete("a") #=> 200
640 * h.length #=> 3
641 */
642static mrb_value
643mrb_hash_size_m(mrb_state *mrb, mrb_value self)
644{
645 khash_t(ht) *h = RHASH_TBL(self);
646
647 if (!h) return mrb_fixnum_value(0);
648 return mrb_fixnum_value(kh_size(h));
649}
650
651/* 15.2.13.4.12 */
652/*
653 * call-seq:
654 * hsh.empty? -> true or false
655 *
656 * Returns <code>true</code> if <i>hsh</i> contains no key-value pairs.
657 *
658 * {}.empty? #=> true
659 *
660 */
661MRB_API mrb_value
662mrb_hash_empty_p(mrb_state *mrb, mrb_value self)
663{
664 khash_t(ht) *h = RHASH_TBL(self);
665
666 if (h) return mrb_bool_value(kh_size(h) == 0);
667 return mrb_true_value();
668}
669
670/* 15.2.13.4.29 (x)*/
671/*
672 * call-seq:
673 * hsh.to_hash => hsh
674 *
675 * Returns +self+.
676 */
677
678static mrb_value
679mrb_hash_to_hash(mrb_state *mrb, mrb_value hash)
680{
681 return hash;
682}
683
684/* 15.2.13.4.19 */
685/*
686 * call-seq:
687 * hsh.keys -> array
688 *
689 * Returns a new array populated with the keys from this hash. See also
690 * <code>Hash#values</code>.
691 *
692 * h = { "a" => 100, "b" => 200, "c" => 300, "d" => 400 }
693 * h.keys #=> ["a", "b", "c", "d"]
694 *
695 */
696
697MRB_API mrb_value
698mrb_hash_keys(mrb_state *mrb, mrb_value hash)
699{
700 khash_t(ht) *h = RHASH_TBL(hash);
701 khiter_t k;
702 mrb_value ary;
703 mrb_value *p;
704
705 if (!h || kh_size(h) == 0) return mrb_ary_new(mrb);
706 ary = mrb_ary_new_capa(mrb, kh_size(h));
707 mrb_ary_set(mrb, ary, kh_size(h)-1, mrb_nil_value());
708 p = mrb_ary_ptr(ary)->ptr;
709 for (k = kh_begin(h); k != kh_end(h); k++) {
710 if (kh_exist(h, k)) {
711 mrb_value kv = kh_key(h, k);
712 mrb_hash_value hv = kh_value(h, k);
713
714 p[hv.n] = kv;
715 }
716 }
717 return ary;
718}
719
720/* 15.2.13.4.28 */
721/*
722 * call-seq:
723 * hsh.values -> array
724 *
725 * Returns a new array populated with the values from <i>hsh</i>. See
726 * also <code>Hash#keys</code>.
727 *
728 * h = { "a" => 100, "b" => 200, "c" => 300 }
729 * h.values #=> [100, 200, 300]
730 *
731 */
732
733static mrb_value
734mrb_hash_values(mrb_state *mrb, mrb_value hash)
735{
736 khash_t(ht) *h = RHASH_TBL(hash);
737 khiter_t k;
738 mrb_value ary;
739
740 if (!h) return mrb_ary_new(mrb);
741 ary = mrb_ary_new_capa(mrb, kh_size(h));
742 for (k = kh_begin(h); k != kh_end(h); k++) {
743 if (kh_exist(h, k)) {
744 mrb_hash_value hv = kh_value(h, k);
745
746 mrb_ary_set(mrb, ary, hv.n, hv.v);
747 }
748 }
749 return ary;
750}
751
752/* 15.2.13.4.13 */
753/* 15.2.13.4.15 */
754/* 15.2.13.4.18 */
755/* 15.2.13.4.21 */
756/*
757 * call-seq:
758 * hsh.has_key?(key) -> true or false
759 * hsh.include?(key) -> true or false
760 * hsh.key?(key) -> true or false
761 * hsh.member?(key) -> true or false
762 *
763 * Returns <code>true</code> if the given key is present in <i>hsh</i>.
764 *
765 * h = { "a" => 100, "b" => 200 }
766 * h.has_key?("a") #=> true
767 * h.has_key?("z") #=> false
768 *
769 */
770
771static mrb_value
772mrb_hash_has_key(mrb_state *mrb, mrb_value hash)
773{
774 mrb_value key;
775 khash_t(ht) *h;
776 khiter_t k;
777
778 mrb_get_args(mrb, "o", &key);
779
780 h = RHASH_TBL(hash);
781 if (h) {
782 k = kh_get(ht, mrb, h, key);
783 return mrb_bool_value(k != kh_end(h));
784 }
785 return mrb_false_value();
786}
787
788/* 15.2.13.4.14 */
789/* 15.2.13.4.27 */
790/*
791 * call-seq:
792 * hsh.has_value?(value) -> true or false
793 * hsh.value?(value) -> true or false
794 *
795 * Returns <code>true</code> if the given value is present for some key
796 * in <i>hsh</i>.
797 *
798 * h = { "a" => 100, "b" => 200 }
799 * h.has_value?(100) #=> true
800 * h.has_value?(999) #=> false
801 */
802
803static mrb_value
804mrb_hash_has_value(mrb_state *mrb, mrb_value hash)
805{
806 mrb_value val;
807 khash_t(ht) *h;
808 khiter_t k;
809
810 mrb_get_args(mrb, "o", &val);
811 h = RHASH_TBL(hash);
812
813 if (h) {
814 for (k = kh_begin(h); k != kh_end(h); k++) {
815 if (!kh_exist(h, k)) continue;
816
817 if (mrb_equal(mrb, kh_value(h, k).v, val)) {
818 return mrb_true_value();
819 }
820 }
821 }
822 return mrb_false_value();
823}
824
825void
826mrb_init_hash(mrb_state *mrb)
827{
828 struct RClass *h;
829
830 mrb->hash_class = h = mrb_define_class(mrb, "Hash", mrb->object_class); /* 15.2.13 */
831 MRB_SET_INSTANCE_TT(h, MRB_TT_HASH);
832
833 mrb_define_method(mrb, h, "[]", mrb_hash_aget, MRB_ARGS_REQ(1)); /* 15.2.13.4.2 */
834 mrb_define_method(mrb, h, "[]=", mrb_hash_aset, MRB_ARGS_REQ(2)); /* 15.2.13.4.3 */
835 mrb_define_method(mrb, h, "clear", mrb_hash_clear, MRB_ARGS_NONE()); /* 15.2.13.4.4 */
836 mrb_define_method(mrb, h, "default", mrb_hash_default, MRB_ARGS_ANY()); /* 15.2.13.4.5 */
837 mrb_define_method(mrb, h, "default=", mrb_hash_set_default, MRB_ARGS_REQ(1)); /* 15.2.13.4.6 */
838 mrb_define_method(mrb, h, "default_proc", mrb_hash_default_proc,MRB_ARGS_NONE()); /* 15.2.13.4.7 */
839 mrb_define_method(mrb, h, "default_proc=", mrb_hash_set_default_proc,MRB_ARGS_REQ(1)); /* 15.2.13.4.7 */
840 mrb_define_method(mrb, h, "__delete", mrb_hash_delete, MRB_ARGS_REQ(1)); /* core of 15.2.13.4.8 */
841 mrb_define_method(mrb, h, "empty?", mrb_hash_empty_p, MRB_ARGS_NONE()); /* 15.2.13.4.12 */
842 mrb_define_method(mrb, h, "has_key?", mrb_hash_has_key, MRB_ARGS_REQ(1)); /* 15.2.13.4.13 */
843 mrb_define_method(mrb, h, "has_value?", mrb_hash_has_value, MRB_ARGS_REQ(1)); /* 15.2.13.4.14 */
844 mrb_define_method(mrb, h, "include?", mrb_hash_has_key, MRB_ARGS_REQ(1)); /* 15.2.13.4.15 */
845 mrb_define_method(mrb, h, "initialize", mrb_hash_init, MRB_ARGS_OPT(1)); /* 15.2.13.4.16 */
846 mrb_define_method(mrb, h, "key?", mrb_hash_has_key, MRB_ARGS_REQ(1)); /* 15.2.13.4.18 */
847 mrb_define_method(mrb, h, "keys", mrb_hash_keys, MRB_ARGS_NONE()); /* 15.2.13.4.19 */
848 mrb_define_method(mrb, h, "length", mrb_hash_size_m, MRB_ARGS_NONE()); /* 15.2.13.4.20 */
849 mrb_define_method(mrb, h, "member?", mrb_hash_has_key, MRB_ARGS_REQ(1)); /* 15.2.13.4.21 */
850 mrb_define_method(mrb, h, "shift", mrb_hash_shift, MRB_ARGS_NONE()); /* 15.2.13.4.24 */
851 mrb_define_method(mrb, h, "dup", mrb_hash_dup, MRB_ARGS_NONE());
852 mrb_define_method(mrb, h, "size", mrb_hash_size_m, MRB_ARGS_NONE()); /* 15.2.13.4.25 */
853 mrb_define_method(mrb, h, "store", mrb_hash_aset, MRB_ARGS_REQ(2)); /* 15.2.13.4.26 */
854 mrb_define_method(mrb, h, "value?", mrb_hash_has_value, MRB_ARGS_REQ(1)); /* 15.2.13.4.27 */
855 mrb_define_method(mrb, h, "values", mrb_hash_values, MRB_ARGS_NONE()); /* 15.2.13.4.28 */
856
857 mrb_define_method(mrb, h, "to_hash", mrb_hash_to_hash, MRB_ARGS_NONE()); /* 15.2.13.4.29 (x)*/
858}
Note: See TracBrowser for help on using the repository browser.