From 90d30f306e640df790a1f9a1ba96c933c94c02c7 Mon Sep 17 00:00:00 2001 From: take_cheeze Date: Wed, 30 Apr 2014 22:55:10 +0900 Subject: Implement Struct#to_h . --- mrbgems/mruby-struct/src/struct.c | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) (limited to 'mrbgems/mruby-struct/src') diff --git a/mrbgems/mruby-struct/src/struct.c b/mrbgems/mruby-struct/src/struct.c index 121226717..a4d70ae1a 100644 --- a/mrbgems/mruby-struct/src/struct.c +++ b/mrbgems/mruby-struct/src/struct.c @@ -11,6 +11,7 @@ #include "mruby/string.h" #include "mruby/class.h" #include "mruby/variable.h" +#include "mruby/hash.h" #define RSTRUCT_LEN(st) RARRAY_LEN(st) #define RSTRUCT_PTR(st) RARRAY_PTR(st) @@ -805,6 +806,28 @@ mrb_struct_to_a(mrb_state *mrb, mrb_value self) return mrb_ary_new_from_values(mrb, RSTRUCT_LEN(self), RSTRUCT_PTR(self)); } +/* + * call-seq: + * struct.to_h -> hash + * + * Create a hash from member names and struct values. + */ +static mrb_value +mrb_struct_to_h(mrb_state *mrb, mrb_value self) +{ + mrb_value members, ret; + mrb_int i; + + members = mrb_struct_s_members(mrb, mrb_obj_value(mrb_class(mrb, self))); + ret = mrb_hash_new_capa(mrb, RARRAY_LEN(members)); + + for (i = 0; i < RARRAY_LEN(members); ++i) { + mrb_hash_set(mrb, ret, RARRAY_PTR(members)[i], RSTRUCT_PTR(self)[i]); + } + + return ret; +} + /* * A Struct is a convenient way to bundle a number of * attributes together, using accessor methods, without having to write @@ -842,6 +865,7 @@ mrb_mruby_struct_gem_init(mrb_state* mrb) mrb_define_method(mrb, st, "length", mrb_struct_len, MRB_ARGS_NONE()); mrb_define_method(mrb, st, "to_a", mrb_struct_to_a, MRB_ARGS_NONE()); mrb_define_method(mrb, st, "values", mrb_struct_to_a, MRB_ARGS_NONE()); + mrb_define_method(mrb, st, "to_h", mrb_struct_to_h, MRB_ARGS_NONE()); } void -- cgit v1.2.3