sabme ua / mruby-mbed

mbed I/F binding for mruby

Dependents:   mruby_mbed_web mirb_mbed

mbed-mruby

How to use

Class

mrbgems/mruby-struct/struct.c@0:158c61bb030f, 2015-03-25 (annotated)

Committer:
mzta
Date:
Wed Mar 25 17:36:16 2015 +0000
Revision:
0:158c61bb030f
mirb_mbed initial commit;

Who changed what in which revision?

UserRevisionLine numberNew contents of line
mzta 0:158c61bb030f 1 /*
mzta 0:158c61bb030f 2 ** struct.c - Struct class
mzta 0:158c61bb030f 3 **
mzta 0:158c61bb030f 4 ** See Copyright Notice in mruby.h
mzta 0:158c61bb030f 5 */
mzta 0:158c61bb030f 6
mzta 0:158c61bb030f 7 #include <ctype.h>
mzta 0:158c61bb030f 8 #include <string.h>
mzta 0:158c61bb030f 9 #include "mruby.h"
mzta 0:158c61bb030f 10 #include "mruby/array.h"
mzta 0:158c61bb030f 11 #include "mruby/string.h"
mzta 0:158c61bb030f 12 #include "mruby/class.h"
mzta 0:158c61bb030f 13 #include "mruby/variable.h"
mzta 0:158c61bb030f 14 #include "mruby/hash.h"
mzta 0:158c61bb030f 15 #include "mruby/range.h"
mzta 0:158c61bb030f 16
mzta 0:158c61bb030f 17 #define RSTRUCT_LEN(st) mrb_ary_ptr(st)->len
mzta 0:158c61bb030f 18 #define RSTRUCT_PTR(st) mrb_ary_ptr(st)->ptr
mzta 0:158c61bb030f 19
mzta 0:158c61bb030f 20 static struct RClass *
mzta 0:158c61bb030f 21 struct_class(mrb_state *mrb)
mzta 0:158c61bb030f 22 {
mzta 0:158c61bb030f 23 return mrb_class_get(mrb, "Struct");
mzta 0:158c61bb030f 24 }
mzta 0:158c61bb030f 25
mzta 0:158c61bb030f 26 static inline mrb_value
mzta 0:158c61bb030f 27 struct_ivar_get(mrb_state *mrb, mrb_value c, mrb_sym id)
mzta 0:158c61bb030f 28 {
mzta 0:158c61bb030f 29 struct RClass* kclass;
mzta 0:158c61bb030f 30 struct RClass* sclass = struct_class(mrb);
mzta 0:158c61bb030f 31 mrb_value ans;
mzta 0:158c61bb030f 32
mzta 0:158c61bb030f 33 for (;;) {
mzta 0:158c61bb030f 34 ans = mrb_iv_get(mrb, c, id);
mzta 0:158c61bb030f 35 if (!mrb_nil_p(ans)) return ans;
mzta 0:158c61bb030f 36 kclass = RCLASS_SUPER(c);
mzta 0:158c61bb030f 37 if (kclass == 0 || kclass == sclass)
mzta 0:158c61bb030f 38 return mrb_nil_value();
mzta 0:158c61bb030f 39 c = mrb_obj_value(kclass);
mzta 0:158c61bb030f 40 }
mzta 0:158c61bb030f 41 }
mzta 0:158c61bb030f 42
mzta 0:158c61bb030f 43 static mrb_value
mzta 0:158c61bb030f 44 mrb_struct_s_members(mrb_state *mrb, mrb_value klass)
mzta 0:158c61bb030f 45 {
mzta 0:158c61bb030f 46 mrb_value members = struct_ivar_get(mrb, klass, mrb_intern_lit(mrb, "__members__"));
mzta 0:158c61bb030f 47
mzta 0:158c61bb030f 48 if (mrb_nil_p(members)) {
mzta 0:158c61bb030f 49 mrb_raise(mrb, E_TYPE_ERROR, "uninitialized struct");
mzta 0:158c61bb030f 50 }
mzta 0:158c61bb030f 51 if (!mrb_array_p(members)) {
mzta 0:158c61bb030f 52 mrb_raise(mrb, E_TYPE_ERROR, "corrupted struct");
mzta 0:158c61bb030f 53 }
mzta 0:158c61bb030f 54 return members;
mzta 0:158c61bb030f 55 }
mzta 0:158c61bb030f 56
mzta 0:158c61bb030f 57 static mrb_value
mzta 0:158c61bb030f 58 mrb_struct_members(mrb_state *mrb, mrb_value s)
mzta 0:158c61bb030f 59 {
mzta 0:158c61bb030f 60 mrb_value members = mrb_struct_s_members(mrb, mrb_obj_value(mrb_obj_class(mrb, s)));
mzta 0:158c61bb030f 61 if (!strcmp(mrb_class_name(mrb, mrb_obj_class(mrb, s)), "Struct")) {
mzta 0:158c61bb030f 62 if (RSTRUCT_LEN(s) != RARRAY_LEN(members)) {
mzta 0:158c61bb030f 63 mrb_raisef(mrb, E_TYPE_ERROR,
mzta 0:158c61bb030f 64 "struct size differs (%S required %S given)",
mzta 0:158c61bb030f 65 mrb_fixnum_value(RARRAY_LEN(members)), mrb_fixnum_value(RSTRUCT_LEN(s)));
mzta 0:158c61bb030f 66 }
mzta 0:158c61bb030f 67 }
mzta 0:158c61bb030f 68 return members;
mzta 0:158c61bb030f 69 }
mzta 0:158c61bb030f 70
mzta 0:158c61bb030f 71 static mrb_value
mzta 0:158c61bb030f 72 mrb_struct_s_members_m(mrb_state *mrb, mrb_value klass)
mzta 0:158c61bb030f 73 {
mzta 0:158c61bb030f 74 mrb_value members, ary;
mzta 0:158c61bb030f 75
mzta 0:158c61bb030f 76 members = mrb_struct_s_members(mrb, klass);
mzta 0:158c61bb030f 77 ary = mrb_ary_new_capa(mrb, RARRAY_LEN(members));
mzta 0:158c61bb030f 78 mrb_ary_replace(mrb, ary, members);
mzta 0:158c61bb030f 79 return ary;
mzta 0:158c61bb030f 80 }
mzta 0:158c61bb030f 81
mzta 0:158c61bb030f 82 /* 15.2.18.4.6 */
mzta 0:158c61bb030f 83 /*
mzta 0:158c61bb030f 84 * call-seq:
mzta 0:158c61bb030f 85 * struct.members -> array
mzta 0:158c61bb030f 86 *
mzta 0:158c61bb030f 87 * Returns an array of strings representing the names of the instance
mzta 0:158c61bb030f 88 * variables.
mzta 0:158c61bb030f 89 *
mzta 0:158c61bb030f 90 * Customer = Struct.new(:name, :address, :zip)
mzta 0:158c61bb030f 91 * joe = Customer.new("Joe Smith", "123 Maple, Anytown NC", 12345)
mzta 0:158c61bb030f 92 * joe.members #=> [:name, :address, :zip]
mzta 0:158c61bb030f 93 */
mzta 0:158c61bb030f 94
mzta 0:158c61bb030f 95 static mrb_value
mzta 0:158c61bb030f 96 mrb_struct_members_m(mrb_state *mrb, mrb_value obj)
mzta 0:158c61bb030f 97 {
mzta 0:158c61bb030f 98 return mrb_struct_s_members_m(mrb, mrb_obj_value(mrb_obj_class(mrb, obj)));
mzta 0:158c61bb030f 99 }
mzta 0:158c61bb030f 100
mzta 0:158c61bb030f 101 static mrb_value
mzta 0:158c61bb030f 102 mrb_struct_getmember(mrb_state *mrb, mrb_value obj, mrb_sym id)
mzta 0:158c61bb030f 103 {
mzta 0:158c61bb030f 104 mrb_value members, slot, *ptr;
mzta 0:158c61bb030f 105 const mrb_value *ptr_members;
mzta 0:158c61bb030f 106 mrb_int i, len;
mzta 0:158c61bb030f 107
mzta 0:158c61bb030f 108 ptr = RSTRUCT_PTR(obj);
mzta 0:158c61bb030f 109 members = mrb_struct_members(mrb, obj);
mzta 0:158c61bb030f 110 ptr_members = RARRAY_PTR(members);
mzta 0:158c61bb030f 111 slot = mrb_symbol_value(id);
mzta 0:158c61bb030f 112 len = RARRAY_LEN(members);
mzta 0:158c61bb030f 113 for (i=0; i<len; i++) {
mzta 0:158c61bb030f 114 if (mrb_obj_equal(mrb, ptr_members[i], slot)) {
mzta 0:158c61bb030f 115 return ptr[i];
mzta 0:158c61bb030f 116 }
mzta 0:158c61bb030f 117 }
mzta 0:158c61bb030f 118 mrb_raisef(mrb, E_INDEX_ERROR, "`%S' is not a struct member", mrb_sym2str(mrb, id));
mzta 0:158c61bb030f 119 return mrb_nil_value(); /* not reached */
mzta 0:158c61bb030f 120 }
mzta 0:158c61bb030f 121
mzta 0:158c61bb030f 122 static mrb_value
mzta 0:158c61bb030f 123 mrb_struct_ref(mrb_state *mrb, mrb_value obj)
mzta 0:158c61bb030f 124 {
mzta 0:158c61bb030f 125 return mrb_struct_getmember(mrb, obj, mrb->c->ci->mid);
mzta 0:158c61bb030f 126 }
mzta 0:158c61bb030f 127
mzta 0:158c61bb030f 128 static mrb_value mrb_struct_ref0(mrb_state* mrb, mrb_value obj) {return RSTRUCT_PTR(obj)[0];}
mzta 0:158c61bb030f 129 static mrb_value mrb_struct_ref1(mrb_state* mrb, mrb_value obj) {return RSTRUCT_PTR(obj)[1];}
mzta 0:158c61bb030f 130 static mrb_value mrb_struct_ref2(mrb_state* mrb, mrb_value obj) {return RSTRUCT_PTR(obj)[2];}
mzta 0:158c61bb030f 131 static mrb_value mrb_struct_ref3(mrb_state* mrb, mrb_value obj) {return RSTRUCT_PTR(obj)[3];}
mzta 0:158c61bb030f 132 static mrb_value mrb_struct_ref4(mrb_state* mrb, mrb_value obj) {return RSTRUCT_PTR(obj)[4];}
mzta 0:158c61bb030f 133 static mrb_value mrb_struct_ref5(mrb_state* mrb, mrb_value obj) {return RSTRUCT_PTR(obj)[5];}
mzta 0:158c61bb030f 134 static mrb_value mrb_struct_ref6(mrb_state* mrb, mrb_value obj) {return RSTRUCT_PTR(obj)[6];}
mzta 0:158c61bb030f 135 static mrb_value mrb_struct_ref7(mrb_state* mrb, mrb_value obj) {return RSTRUCT_PTR(obj)[7];}
mzta 0:158c61bb030f 136 static mrb_value mrb_struct_ref8(mrb_state* mrb, mrb_value obj) {return RSTRUCT_PTR(obj)[8];}
mzta 0:158c61bb030f 137 static mrb_value mrb_struct_ref9(mrb_state* mrb, mrb_value obj) {return RSTRUCT_PTR(obj)[9];}
mzta 0:158c61bb030f 138
mzta 0:158c61bb030f 139 #define numberof(array) (int)(sizeof(array) / sizeof((array)[0]))
mzta 0:158c61bb030f 140 #define N_REF_FUNC numberof(ref_func)
mzta 0:158c61bb030f 141
mzta 0:158c61bb030f 142 static const mrb_func_t ref_func[] = {
mzta 0:158c61bb030f 143 mrb_struct_ref0,
mzta 0:158c61bb030f 144 mrb_struct_ref1,
mzta 0:158c61bb030f 145 mrb_struct_ref2,
mzta 0:158c61bb030f 146 mrb_struct_ref3,
mzta 0:158c61bb030f 147 mrb_struct_ref4,
mzta 0:158c61bb030f 148 mrb_struct_ref5,
mzta 0:158c61bb030f 149 mrb_struct_ref6,
mzta 0:158c61bb030f 150 mrb_struct_ref7,
mzta 0:158c61bb030f 151 mrb_struct_ref8,
mzta 0:158c61bb030f 152 mrb_struct_ref9,
mzta 0:158c61bb030f 153 };
mzta 0:158c61bb030f 154
mzta 0:158c61bb030f 155 static mrb_sym
mzta 0:158c61bb030f 156 mrb_id_attrset(mrb_state *mrb, mrb_sym id)
mzta 0:158c61bb030f 157 {
mzta 0:158c61bb030f 158 const char *name;
mzta 0:158c61bb030f 159 char *buf;
mzta 0:158c61bb030f 160 mrb_int len;
mzta 0:158c61bb030f 161 mrb_sym mid;
mzta 0:158c61bb030f 162
mzta 0:158c61bb030f 163 name = mrb_sym2name_len(mrb, id, &len);
mzta 0:158c61bb030f 164 buf = (char *)mrb_malloc(mrb, (size_t)len+2);
mzta 0:158c61bb030f 165 memcpy(buf, name, (size_t)len);
mzta 0:158c61bb030f 166 buf[len] = '=';
mzta 0:158c61bb030f 167 buf[len+1] = '\0';
mzta 0:158c61bb030f 168
mzta 0:158c61bb030f 169 mid = mrb_intern(mrb, buf, len+1);
mzta 0:158c61bb030f 170 mrb_free(mrb, buf);
mzta 0:158c61bb030f 171 return mid;
mzta 0:158c61bb030f 172 }
mzta 0:158c61bb030f 173
mzta 0:158c61bb030f 174 static mrb_value
mzta 0:158c61bb030f 175 mrb_struct_set(mrb_state *mrb, mrb_value obj, mrb_value val)
mzta 0:158c61bb030f 176 {
mzta 0:158c61bb030f 177 const char *name;
mzta 0:158c61bb030f 178 mrb_int i, len, slen;
mzta 0:158c61bb030f 179 mrb_sym mid;
mzta 0:158c61bb030f 180 mrb_value members, slot, *ptr;
mzta 0:158c61bb030f 181 const mrb_value *ptr_members;
mzta 0:158c61bb030f 182
mzta 0:158c61bb030f 183 /* get base id */
mzta 0:158c61bb030f 184 name = mrb_sym2name_len(mrb, mrb->c->ci->mid, &slen);
mzta 0:158c61bb030f 185 mid = mrb_intern(mrb, name, slen-1); /* omit last "=" */
mzta 0:158c61bb030f 186
mzta 0:158c61bb030f 187 members = mrb_struct_members(mrb, obj);
mzta 0:158c61bb030f 188 ptr_members = RARRAY_PTR(members);
mzta 0:158c61bb030f 189 len = RARRAY_LEN(members);
mzta 0:158c61bb030f 190 ptr = RSTRUCT_PTR(obj);
mzta 0:158c61bb030f 191 for (i=0; i<len; i++) {
mzta 0:158c61bb030f 192 slot = ptr_members[i];
mzta 0:158c61bb030f 193 if (mrb_symbol(slot) == mid) {
mzta 0:158c61bb030f 194 return ptr[i] = val;
mzta 0:158c61bb030f 195 }
mzta 0:158c61bb030f 196 }
mzta 0:158c61bb030f 197 mrb_raisef(mrb, E_INDEX_ERROR, "`%S' is not a struct member", mrb_sym2str(mrb, mid));
mzta 0:158c61bb030f 198 return mrb_nil_value(); /* not reached */
mzta 0:158c61bb030f 199 }
mzta 0:158c61bb030f 200
mzta 0:158c61bb030f 201 static mrb_value
mzta 0:158c61bb030f 202 mrb_struct_set_m(mrb_state *mrb, mrb_value obj)
mzta 0:158c61bb030f 203 {
mzta 0:158c61bb030f 204 mrb_value val;
mzta 0:158c61bb030f 205
mzta 0:158c61bb030f 206 mrb_get_args(mrb, "o", &val);
mzta 0:158c61bb030f 207 return mrb_struct_set(mrb, obj, val);
mzta 0:158c61bb030f 208 }
mzta 0:158c61bb030f 209
mzta 0:158c61bb030f 210 static mrb_bool
mzta 0:158c61bb030f 211 is_local_id(mrb_state *mrb, const char *name)
mzta 0:158c61bb030f 212 {
mzta 0:158c61bb030f 213 if (!name) return FALSE;
mzta 0:158c61bb030f 214 return !ISUPPER(name[0]);
mzta 0:158c61bb030f 215 }
mzta 0:158c61bb030f 216
mzta 0:158c61bb030f 217 static mrb_bool
mzta 0:158c61bb030f 218 is_const_id(mrb_state *mrb, const char *name)
mzta 0:158c61bb030f 219 {
mzta 0:158c61bb030f 220 if (!name) return FALSE;
mzta 0:158c61bb030f 221 return ISUPPER(name[0]);
mzta 0:158c61bb030f 222 }
mzta 0:158c61bb030f 223
mzta 0:158c61bb030f 224 static void
mzta 0:158c61bb030f 225 make_struct_define_accessors(mrb_state *mrb, mrb_value members, struct RClass *c)
mzta 0:158c61bb030f 226 {
mzta 0:158c61bb030f 227 const mrb_value *ptr_members = RARRAY_PTR(members);
mzta 0:158c61bb030f 228 mrb_int i;
mzta 0:158c61bb030f 229 mrb_int len = RARRAY_LEN(members);
mzta 0:158c61bb030f 230 int ai = mrb_gc_arena_save(mrb);
mzta 0:158c61bb030f 231
mzta 0:158c61bb030f 232 for (i=0; i<len; i++) {
mzta 0:158c61bb030f 233 mrb_sym id = mrb_symbol(ptr_members[i]);
mzta 0:158c61bb030f 234 const char *name = mrb_sym2name_len(mrb, id, NULL);
mzta 0:158c61bb030f 235
mzta 0:158c61bb030f 236 if (is_local_id(mrb, name) || is_const_id(mrb, name)) {
mzta 0:158c61bb030f 237 if (i < N_REF_FUNC) {
mzta 0:158c61bb030f 238 mrb_define_method_id(mrb, c, id, ref_func[i], MRB_ARGS_NONE());
mzta 0:158c61bb030f 239 }
mzta 0:158c61bb030f 240 else {
mzta 0:158c61bb030f 241 mrb_define_method_id(mrb, c, id, mrb_struct_ref, MRB_ARGS_NONE());
mzta 0:158c61bb030f 242 }
mzta 0:158c61bb030f 243 mrb_define_method_id(mrb, c, mrb_id_attrset(mrb, id), mrb_struct_set_m, MRB_ARGS_REQ(1));
mzta 0:158c61bb030f 244 mrb_gc_arena_restore(mrb, ai);
mzta 0:158c61bb030f 245 }
mzta 0:158c61bb030f 246 }
mzta 0:158c61bb030f 247 }
mzta 0:158c61bb030f 248
mzta 0:158c61bb030f 249 static mrb_value
mzta 0:158c61bb030f 250 make_struct(mrb_state *mrb, mrb_value name, mrb_value members, struct RClass * klass)
mzta 0:158c61bb030f 251 {
mzta 0:158c61bb030f 252 mrb_value nstr;
mzta 0:158c61bb030f 253 mrb_sym id;
mzta 0:158c61bb030f 254 struct RClass *c;
mzta 0:158c61bb030f 255
mzta 0:158c61bb030f 256 if (mrb_nil_p(name)) {
mzta 0:158c61bb030f 257 c = mrb_class_new(mrb, klass);
mzta 0:158c61bb030f 258 }
mzta 0:158c61bb030f 259 else {
mzta 0:158c61bb030f 260 /* old style: should we warn? */
mzta 0:158c61bb030f 261 name = mrb_str_to_str(mrb, name);
mzta 0:158c61bb030f 262 id = mrb_obj_to_sym(mrb, name);
mzta 0:158c61bb030f 263 if (!is_const_id(mrb, mrb_sym2name_len(mrb, id, NULL))) {
mzta 0:158c61bb030f 264 mrb_name_error(mrb, id, "identifier %S needs to be constant", name);
mzta 0:158c61bb030f 265 }
mzta 0:158c61bb030f 266 if (mrb_const_defined_at(mrb, mrb_obj_value(klass), id)) {
mzta 0:158c61bb030f 267 mrb_warn(mrb, "redefining constant Struct::%S", name);
mzta 0:158c61bb030f 268 /* ?rb_mod_remove_const(klass, mrb_sym2name(mrb, id)); */
mzta 0:158c61bb030f 269 }
mzta 0:158c61bb030f 270 c = mrb_define_class_under(mrb, klass, RSTRING_PTR(name), klass);
mzta 0:158c61bb030f 271 }
mzta 0:158c61bb030f 272 MRB_SET_INSTANCE_TT(c, MRB_TT_ARRAY);
mzta 0:158c61bb030f 273 nstr = mrb_obj_value(c);
mzta 0:158c61bb030f 274 mrb_iv_set(mrb, nstr, mrb_intern_lit(mrb, "__members__"), members);
mzta 0:158c61bb030f 275
mzta 0:158c61bb030f 276 mrb_define_class_method(mrb, c, "new", mrb_instance_new, MRB_ARGS_ANY());
mzta 0:158c61bb030f 277 mrb_define_class_method(mrb, c, "[]", mrb_instance_new, MRB_ARGS_ANY());
mzta 0:158c61bb030f 278 mrb_define_class_method(mrb, c, "members", mrb_struct_s_members_m, MRB_ARGS_NONE());
mzta 0:158c61bb030f 279 /* RSTRUCT(nstr)->basic.c->super = c->c; */
mzta 0:158c61bb030f 280 make_struct_define_accessors(mrb, members, c);
mzta 0:158c61bb030f 281 return nstr;
mzta 0:158c61bb030f 282 }
mzta 0:158c61bb030f 283
mzta 0:158c61bb030f 284 /* 15.2.18.3.1 */
mzta 0:158c61bb030f 285 /*
mzta 0:158c61bb030f 286 * call-seq:
mzta 0:158c61bb030f 287 * Struct.new( [aString] [, aSym]+> ) -> StructClass
mzta 0:158c61bb030f 288 * StructClass.new(arg, ...) -> obj
mzta 0:158c61bb030f 289 * StructClass[arg, ...] -> obj
mzta 0:158c61bb030f 290 *
mzta 0:158c61bb030f 291 * Creates a new class, named by <i>aString</i>, containing accessor
mzta 0:158c61bb030f 292 * methods for the given symbols. If the name <i>aString</i> is
mzta 0:158c61bb030f 293 * omitted, an anonymous structure class will be created. Otherwise,
mzta 0:158c61bb030f 294 * the name of this struct will appear as a constant in class
mzta 0:158c61bb030f 295 * <code>Struct</code>, so it must be unique for all
mzta 0:158c61bb030f 296 * <code>Struct</code>s in the system and should start with a capital
mzta 0:158c61bb030f 297 * letter. Assigning a structure class to a constant effectively gives
mzta 0:158c61bb030f 298 * the class the name of the constant.
mzta 0:158c61bb030f 299 *
mzta 0:158c61bb030f 300 * <code>Struct::new</code> returns a new <code>Class</code> object,
mzta 0:158c61bb030f 301 * which can then be used to create specific instances of the new
mzta 0:158c61bb030f 302 * structure. The number of actual parameters must be
mzta 0:158c61bb030f 303 * less than or equal to the number of attributes defined for this
mzta 0:158c61bb030f 304 * class; unset parameters default to <code>nil</code>. Passing too many
mzta 0:158c61bb030f 305 * parameters will raise an <code>ArgumentError</code>.
mzta 0:158c61bb030f 306 *
mzta 0:158c61bb030f 307 * The remaining methods listed in this section (class and instance)
mzta 0:158c61bb030f 308 * are defined for this generated class.
mzta 0:158c61bb030f 309 *
mzta 0:158c61bb030f 310 * # Create a structure with a name in Struct
mzta 0:158c61bb030f 311 * Struct.new("Customer", :name, :address) #=> Struct::Customer
mzta 0:158c61bb030f 312 * Struct::Customer.new("Dave", "123 Main") #=> #<struct Struct::Customer name="Dave", address="123 Main">
mzta 0:158c61bb030f 313 *
mzta 0:158c61bb030f 314 * # Create a structure named by its constant
mzta 0:158c61bb030f 315 * Customer = Struct.new(:name, :address) #=> Customer
mzta 0:158c61bb030f 316 * Customer.new("Dave", "123 Main") #=> #<struct Customer name="Dave", address="123 Main">
mzta 0:158c61bb030f 317 */
mzta 0:158c61bb030f 318 static mrb_value
mzta 0:158c61bb030f 319 mrb_struct_s_def(mrb_state *mrb, mrb_value klass)
mzta 0:158c61bb030f 320 {
mzta 0:158c61bb030f 321 mrb_value name, rest;
mzta 0:158c61bb030f 322 mrb_value *pargv;
mzta 0:158c61bb030f 323 mrb_int argcnt;
mzta 0:158c61bb030f 324 mrb_int i;
mzta 0:158c61bb030f 325 mrb_value b, st;
mzta 0:158c61bb030f 326 mrb_sym id;
mzta 0:158c61bb030f 327 mrb_value *argv;
mzta 0:158c61bb030f 328 mrb_int argc;
mzta 0:158c61bb030f 329
mzta 0:158c61bb030f 330 name = mrb_nil_value();
mzta 0:158c61bb030f 331 rest = mrb_nil_value();
mzta 0:158c61bb030f 332 mrb_get_args(mrb, "*&", &argv, &argc, &b);
mzta 0:158c61bb030f 333 if (argc == 0) { /* special case to avoid crash */
mzta 0:158c61bb030f 334 rest = mrb_ary_new(mrb);
mzta 0:158c61bb030f 335 }
mzta 0:158c61bb030f 336 else {
mzta 0:158c61bb030f 337 if (argc > 0) name = argv[0];
mzta 0:158c61bb030f 338 if (argc > 1) rest = argv[1];
mzta 0:158c61bb030f 339 if (mrb_array_p(rest)) {
mzta 0:158c61bb030f 340 if (!mrb_nil_p(name) && mrb_symbol_p(name)) {
mzta 0:158c61bb030f 341 /* 1stArgument:symbol -> name=nil rest=argv[0]-[n] */
mzta 0:158c61bb030f 342 mrb_ary_unshift(mrb, rest, name);
mzta 0:158c61bb030f 343 name = mrb_nil_value();
mzta 0:158c61bb030f 344 }
mzta 0:158c61bb030f 345 }
mzta 0:158c61bb030f 346 else {
mzta 0:158c61bb030f 347 pargv = &argv[1];
mzta 0:158c61bb030f 348 argcnt = argc-1;
mzta 0:158c61bb030f 349 if (!mrb_nil_p(name) && mrb_symbol_p(name)) {
mzta 0:158c61bb030f 350 /* 1stArgument:symbol -> name=nil rest=argv[0]-[n] */
mzta 0:158c61bb030f 351 name = mrb_nil_value();
mzta 0:158c61bb030f 352 pargv = &argv[0];
mzta 0:158c61bb030f 353 argcnt++;
mzta 0:158c61bb030f 354 }
mzta 0:158c61bb030f 355 rest = mrb_ary_new_from_values(mrb, argcnt, pargv);
mzta 0:158c61bb030f 356 }
mzta 0:158c61bb030f 357 for (i=0; i<RARRAY_LEN(rest); i++) {
mzta 0:158c61bb030f 358 id = mrb_obj_to_sym(mrb, RARRAY_PTR(rest)[i]);
mzta 0:158c61bb030f 359 mrb_ary_set(mrb, rest, i, mrb_symbol_value(id));
mzta 0:158c61bb030f 360 }
mzta 0:158c61bb030f 361 }
mzta 0:158c61bb030f 362 st = make_struct(mrb, name, rest, struct_class(mrb));
mzta 0:158c61bb030f 363 if (!mrb_nil_p(b)) {
mzta 0:158c61bb030f 364 mrb_yield_with_class(mrb, b, 1, &st, st, mrb_class_ptr(klass));
mzta 0:158c61bb030f 365 }
mzta 0:158c61bb030f 366
mzta 0:158c61bb030f 367 return st;
mzta 0:158c61bb030f 368 }
mzta 0:158c61bb030f 369
mzta 0:158c61bb030f 370 static mrb_int
mzta 0:158c61bb030f 371 num_members(mrb_state *mrb, struct RClass *klass)
mzta 0:158c61bb030f 372 {
mzta 0:158c61bb030f 373 mrb_value members;
mzta 0:158c61bb030f 374
mzta 0:158c61bb030f 375 members = struct_ivar_get(mrb, mrb_obj_value(klass), mrb_intern_lit(mrb, "__members__"));
mzta 0:158c61bb030f 376 if (!mrb_array_p(members)) {
mzta 0:158c61bb030f 377 mrb_raise(mrb, E_TYPE_ERROR, "broken members");
mzta 0:158c61bb030f 378 }
mzta 0:158c61bb030f 379 return RARRAY_LEN(members);
mzta 0:158c61bb030f 380 }
mzta 0:158c61bb030f 381
mzta 0:158c61bb030f 382 /* 15.2.18.4.8 */
mzta 0:158c61bb030f 383 /*
mzta 0:158c61bb030f 384 */
mzta 0:158c61bb030f 385 static mrb_value
mzta 0:158c61bb030f 386 mrb_struct_initialize_withArg(mrb_state *mrb, mrb_int argc, mrb_value *argv, mrb_value self)
mzta 0:158c61bb030f 387 {
mzta 0:158c61bb030f 388 struct RClass *klass = mrb_obj_class(mrb, self);
mzta 0:158c61bb030f 389 mrb_int i, n;
mzta 0:158c61bb030f 390
mzta 0:158c61bb030f 391 n = num_members(mrb, klass);
mzta 0:158c61bb030f 392 if (n < argc) {
mzta 0:158c61bb030f 393 mrb_raise(mrb, E_ARGUMENT_ERROR, "struct size differs");
mzta 0:158c61bb030f 394 }
mzta 0:158c61bb030f 395
mzta 0:158c61bb030f 396 for (i = 0; i < argc; i++) {
mzta 0:158c61bb030f 397 mrb_ary_set(mrb, self, i, argv[i]);
mzta 0:158c61bb030f 398 }
mzta 0:158c61bb030f 399 for (i = argc; i < n; i++) {
mzta 0:158c61bb030f 400 mrb_ary_set(mrb, self, i, mrb_nil_value());
mzta 0:158c61bb030f 401 }
mzta 0:158c61bb030f 402 return self;
mzta 0:158c61bb030f 403 }
mzta 0:158c61bb030f 404
mzta 0:158c61bb030f 405 static mrb_value
mzta 0:158c61bb030f 406 mrb_struct_initialize_m(mrb_state *mrb, /*int argc, mrb_value *argv,*/ mrb_value self)
mzta 0:158c61bb030f 407 {
mzta 0:158c61bb030f 408 mrb_value *argv;
mzta 0:158c61bb030f 409 mrb_int argc;
mzta 0:158c61bb030f 410
mzta 0:158c61bb030f 411 mrb_get_args(mrb, "*", &argv, &argc);
mzta 0:158c61bb030f 412 return mrb_struct_initialize_withArg(mrb, argc, argv, self);
mzta 0:158c61bb030f 413 }
mzta 0:158c61bb030f 414
mzta 0:158c61bb030f 415 static mrb_value
mzta 0:158c61bb030f 416 inspect_struct(mrb_state *mrb, mrb_value s, mrb_bool recur)
mzta 0:158c61bb030f 417 {
mzta 0:158c61bb030f 418 const char *cn = mrb_class_name(mrb, mrb_obj_class(mrb, s));
mzta 0:158c61bb030f 419 mrb_value members, str = mrb_str_new_lit(mrb, "#<struct ");
mzta 0:158c61bb030f 420 mrb_value *ptr;
mzta 0:158c61bb030f 421 const mrb_value *ptr_members;
mzta 0:158c61bb030f 422 mrb_int i, len;
mzta 0:158c61bb030f 423
mzta 0:158c61bb030f 424 if (cn) {
mzta 0:158c61bb030f 425 mrb_str_append(mrb, str, mrb_str_new_cstr(mrb, cn));
mzta 0:158c61bb030f 426 }
mzta 0:158c61bb030f 427 if (recur) {
mzta 0:158c61bb030f 428 return mrb_str_cat_lit(mrb, str, ":...>");
mzta 0:158c61bb030f 429 }
mzta 0:158c61bb030f 430
mzta 0:158c61bb030f 431 members = mrb_struct_members(mrb, s);
mzta 0:158c61bb030f 432 ptr_members = RARRAY_PTR(members);
mzta 0:158c61bb030f 433 ptr = RSTRUCT_PTR(s);
mzta 0:158c61bb030f 434 len = RSTRUCT_LEN(s);
mzta 0:158c61bb030f 435 for (i=0; i<len; i++) {
mzta 0:158c61bb030f 436 mrb_value slot;
mzta 0:158c61bb030f 437 mrb_sym id;
mzta 0:158c61bb030f 438 const char *name;
mzta 0:158c61bb030f 439 mrb_int namelen;
mzta 0:158c61bb030f 440
mzta 0:158c61bb030f 441 if (i > 0) {
mzta 0:158c61bb030f 442 mrb_str_cat_lit(mrb, str, ", ");
mzta 0:158c61bb030f 443 }
mzta 0:158c61bb030f 444 else if (cn) {
mzta 0:158c61bb030f 445 mrb_str_cat_lit(mrb, str, " ");
mzta 0:158c61bb030f 446 }
mzta 0:158c61bb030f 447 slot = ptr_members[i];
mzta 0:158c61bb030f 448 id = mrb_symbol(slot);
mzta 0:158c61bb030f 449 name = mrb_sym2name_len(mrb, id, &namelen);
mzta 0:158c61bb030f 450 if (is_local_id(mrb, name) || is_const_id(mrb, name)) {
mzta 0:158c61bb030f 451 mrb_str_append(mrb, str, mrb_str_new(mrb, name, namelen));
mzta 0:158c61bb030f 452 }
mzta 0:158c61bb030f 453 else {
mzta 0:158c61bb030f 454 mrb_str_append(mrb, str, mrb_inspect(mrb, slot));
mzta 0:158c61bb030f 455 }
mzta 0:158c61bb030f 456 mrb_str_cat_lit(mrb, str, "=");
mzta 0:158c61bb030f 457 mrb_str_append(mrb, str, mrb_inspect(mrb, ptr[i]));
mzta 0:158c61bb030f 458 }
mzta 0:158c61bb030f 459 mrb_str_cat_lit(mrb, str, ">");
mzta 0:158c61bb030f 460
mzta 0:158c61bb030f 461 return str;
mzta 0:158c61bb030f 462 }
mzta 0:158c61bb030f 463
mzta 0:158c61bb030f 464 /*
mzta 0:158c61bb030f 465 * call-seq:
mzta 0:158c61bb030f 466 * struct.to_s -> string
mzta 0:158c61bb030f 467 * struct.inspect -> string
mzta 0:158c61bb030f 468 *
mzta 0:158c61bb030f 469 * Describe the contents of this struct in a string.
mzta 0:158c61bb030f 470 */
mzta 0:158c61bb030f 471 static mrb_value
mzta 0:158c61bb030f 472 mrb_struct_inspect(mrb_state *mrb, mrb_value s)
mzta 0:158c61bb030f 473 {
mzta 0:158c61bb030f 474 return inspect_struct(mrb, s, FALSE);
mzta 0:158c61bb030f 475 }
mzta 0:158c61bb030f 476
mzta 0:158c61bb030f 477 /* 15.2.18.4.9 */
mzta 0:158c61bb030f 478 /* :nodoc: */
mzta 0:158c61bb030f 479 static mrb_value
mzta 0:158c61bb030f 480 mrb_struct_init_copy(mrb_state *mrb, mrb_value copy)
mzta 0:158c61bb030f 481 {
mzta 0:158c61bb030f 482 mrb_value s;
mzta 0:158c61bb030f 483 mrb_int i, len;
mzta 0:158c61bb030f 484
mzta 0:158c61bb030f 485 mrb_get_args(mrb, "o", &s);
mzta 0:158c61bb030f 486
mzta 0:158c61bb030f 487 if (mrb_obj_equal(mrb, copy, s)) return copy;
mzta 0:158c61bb030f 488 if (!mrb_obj_is_instance_of(mrb, s, mrb_obj_class(mrb, copy))) {
mzta 0:158c61bb030f 489 mrb_raise(mrb, E_TYPE_ERROR, "wrong argument class");
mzta 0:158c61bb030f 490 }
mzta 0:158c61bb030f 491 if (!mrb_array_p(s)) {
mzta 0:158c61bb030f 492 mrb_raise(mrb, E_TYPE_ERROR, "corrupted struct");
mzta 0:158c61bb030f 493 }
mzta 0:158c61bb030f 494 if (RSTRUCT_LEN(copy) != RSTRUCT_LEN(s)) {
mzta 0:158c61bb030f 495 mrb_raise(mrb, E_TYPE_ERROR, "struct size mismatch");
mzta 0:158c61bb030f 496 }
mzta 0:158c61bb030f 497 len = RSTRUCT_LEN(copy);
mzta 0:158c61bb030f 498 for (i = 0; i < len; i++) {
mzta 0:158c61bb030f 499 mrb_ary_set(mrb, copy, i, RSTRUCT_PTR(s)[i]);
mzta 0:158c61bb030f 500 }
mzta 0:158c61bb030f 501 return copy;
mzta 0:158c61bb030f 502 }
mzta 0:158c61bb030f 503
mzta 0:158c61bb030f 504 static mrb_value
mzta 0:158c61bb030f 505 struct_aref_sym(mrb_state *mrb, mrb_value s, mrb_sym id)
mzta 0:158c61bb030f 506 {
mzta 0:158c61bb030f 507 mrb_value *ptr, members;
mzta 0:158c61bb030f 508 const mrb_value *ptr_members;
mzta 0:158c61bb030f 509 mrb_int i, len;
mzta 0:158c61bb030f 510
mzta 0:158c61bb030f 511 ptr = RSTRUCT_PTR(s);
mzta 0:158c61bb030f 512 members = mrb_struct_members(mrb, s);
mzta 0:158c61bb030f 513 ptr_members = RARRAY_PTR(members);
mzta 0:158c61bb030f 514 len = RARRAY_LEN(members);
mzta 0:158c61bb030f 515 for (i=0; i<len; i++) {
mzta 0:158c61bb030f 516 if (mrb_symbol(ptr_members[i]) == id) {
mzta 0:158c61bb030f 517 return ptr[i];
mzta 0:158c61bb030f 518 }
mzta 0:158c61bb030f 519 }
mzta 0:158c61bb030f 520 mrb_raisef(mrb, E_INDEX_ERROR, "no member '%S' in struct", mrb_sym2str(mrb, id));
mzta 0:158c61bb030f 521 return mrb_nil_value(); /* not reached */
mzta 0:158c61bb030f 522 }
mzta 0:158c61bb030f 523
mzta 0:158c61bb030f 524 static mrb_value
mzta 0:158c61bb030f 525 struct_aref_int(mrb_state *mrb, mrb_value s, mrb_int i)
mzta 0:158c61bb030f 526 {
mzta 0:158c61bb030f 527 if (i < 0) i = RSTRUCT_LEN(s) + i;
mzta 0:158c61bb030f 528 if (i < 0)
mzta 0:158c61bb030f 529 mrb_raisef(mrb, E_INDEX_ERROR,
mzta 0:158c61bb030f 530 "offset %S too small for struct(size:%S)",
mzta 0:158c61bb030f 531 mrb_fixnum_value(i), mrb_fixnum_value(RSTRUCT_LEN(s)));
mzta 0:158c61bb030f 532 if (RSTRUCT_LEN(s) <= i)
mzta 0:158c61bb030f 533 mrb_raisef(mrb, E_INDEX_ERROR,
mzta 0:158c61bb030f 534 "offset %S too large for struct(size:%S)",
mzta 0:158c61bb030f 535 mrb_fixnum_value(i), mrb_fixnum_value(RSTRUCT_LEN(s)));
mzta 0:158c61bb030f 536 return RSTRUCT_PTR(s)[i];
mzta 0:158c61bb030f 537 }
mzta 0:158c61bb030f 538
mzta 0:158c61bb030f 539 /* 15.2.18.4.2 */
mzta 0:158c61bb030f 540 /*
mzta 0:158c61bb030f 541 * call-seq:
mzta 0:158c61bb030f 542 * struct[symbol] -> anObject
mzta 0:158c61bb030f 543 * struct[fixnum] -> anObject
mzta 0:158c61bb030f 544 *
mzta 0:158c61bb030f 545 * Attribute Reference---Returns the value of the instance variable
mzta 0:158c61bb030f 546 * named by <i>symbol</i>, or indexed (0..length-1) by
mzta 0:158c61bb030f 547 * <i>fixnum</i>. Will raise <code>NameError</code> if the named
mzta 0:158c61bb030f 548 * variable does not exist, or <code>IndexError</code> if the index is
mzta 0:158c61bb030f 549 * out of range.
mzta 0:158c61bb030f 550 *
mzta 0:158c61bb030f 551 * Customer = Struct.new(:name, :address, :zip)
mzta 0:158c61bb030f 552 * joe = Customer.new("Joe Smith", "123 Maple, Anytown NC", 12345)
mzta 0:158c61bb030f 553 *
mzta 0:158c61bb030f 554 * joe["name"] #=> "Joe Smith"
mzta 0:158c61bb030f 555 * joe[:name] #=> "Joe Smith"
mzta 0:158c61bb030f 556 * joe[0] #=> "Joe Smith"
mzta 0:158c61bb030f 557 */
mzta 0:158c61bb030f 558 static mrb_value
mzta 0:158c61bb030f 559 mrb_struct_aref(mrb_state *mrb, mrb_value s)
mzta 0:158c61bb030f 560 {
mzta 0:158c61bb030f 561 mrb_value idx;
mzta 0:158c61bb030f 562
mzta 0:158c61bb030f 563 mrb_get_args(mrb, "o", &idx);
mzta 0:158c61bb030f 564 if (mrb_string_p(idx)) {
mzta 0:158c61bb030f 565 mrb_value sym = mrb_check_intern_str(mrb, idx);
mzta 0:158c61bb030f 566
mzta 0:158c61bb030f 567 if (mrb_nil_p(sym)) {
mzta 0:158c61bb030f 568 mrb_raisef(mrb, E_INDEX_ERROR, "no member '%S' in struct", idx);
mzta 0:158c61bb030f 569 }
mzta 0:158c61bb030f 570 idx = sym;
mzta 0:158c61bb030f 571 }
mzta 0:158c61bb030f 572 if (mrb_symbol_p(idx)) {
mzta 0:158c61bb030f 573 return struct_aref_sym(mrb, s, mrb_symbol(idx));
mzta 0:158c61bb030f 574 }
mzta 0:158c61bb030f 575 return struct_aref_int(mrb, s, mrb_int(mrb, idx));
mzta 0:158c61bb030f 576 }
mzta 0:158c61bb030f 577
mzta 0:158c61bb030f 578 static mrb_value
mzta 0:158c61bb030f 579 mrb_struct_aset_sym(mrb_state *mrb, mrb_value s, mrb_sym id, mrb_value val)
mzta 0:158c61bb030f 580 {
mzta 0:158c61bb030f 581 mrb_value members, *ptr;
mzta 0:158c61bb030f 582 const mrb_value *ptr_members;
mzta 0:158c61bb030f 583 mrb_int i, len;
mzta 0:158c61bb030f 584
mzta 0:158c61bb030f 585 members = mrb_struct_members(mrb, s);
mzta 0:158c61bb030f 586 len = RARRAY_LEN(members);
mzta 0:158c61bb030f 587 if (RSTRUCT_LEN(s) != len) {
mzta 0:158c61bb030f 588 mrb_raisef(mrb, E_TYPE_ERROR,
mzta 0:158c61bb030f 589 "struct size differs (%S required %S given)",
mzta 0:158c61bb030f 590 mrb_fixnum_value(len), mrb_fixnum_value(RSTRUCT_LEN(s)));
mzta 0:158c61bb030f 591 }
mzta 0:158c61bb030f 592 ptr = RSTRUCT_PTR(s);
mzta 0:158c61bb030f 593 ptr_members = RARRAY_PTR(members);
mzta 0:158c61bb030f 594 for (i=0; i<len; i++) {
mzta 0:158c61bb030f 595 if (mrb_symbol(ptr_members[i]) == id) {
mzta 0:158c61bb030f 596 ptr[i] = val;
mzta 0:158c61bb030f 597 return val;
mzta 0:158c61bb030f 598 }
mzta 0:158c61bb030f 599 }
mzta 0:158c61bb030f 600 mrb_raisef(mrb, E_INDEX_ERROR, "no member '%S' in struct", mrb_sym2str(mrb, id));
mzta 0:158c61bb030f 601 return val; /* not reach */
mzta 0:158c61bb030f 602 }
mzta 0:158c61bb030f 603
mzta 0:158c61bb030f 604 /* 15.2.18.4.3 */
mzta 0:158c61bb030f 605 /*
mzta 0:158c61bb030f 606 * call-seq:
mzta 0:158c61bb030f 607 * struct[symbol] = obj -> obj
mzta 0:158c61bb030f 608 * struct[fixnum] = obj -> obj
mzta 0:158c61bb030f 609 *
mzta 0:158c61bb030f 610 * Attribute Assignment---Assigns to the instance variable named by
mzta 0:158c61bb030f 611 * <i>symbol</i> or <i>fixnum</i> the value <i>obj</i> and
mzta 0:158c61bb030f 612 * returns it. Will raise a <code>NameError</code> if the named
mzta 0:158c61bb030f 613 * variable does not exist, or an <code>IndexError</code> if the index
mzta 0:158c61bb030f 614 * is out of range.
mzta 0:158c61bb030f 615 *
mzta 0:158c61bb030f 616 * Customer = Struct.new(:name, :address, :zip)
mzta 0:158c61bb030f 617 * joe = Customer.new("Joe Smith", "123 Maple, Anytown NC", 12345)
mzta 0:158c61bb030f 618 *
mzta 0:158c61bb030f 619 * joe["name"] = "Luke"
mzta 0:158c61bb030f 620 * joe[:zip] = "90210"
mzta 0:158c61bb030f 621 *
mzta 0:158c61bb030f 622 * joe.name #=> "Luke"
mzta 0:158c61bb030f 623 * joe.zip #=> "90210"
mzta 0:158c61bb030f 624 */
mzta 0:158c61bb030f 625
mzta 0:158c61bb030f 626 static mrb_value
mzta 0:158c61bb030f 627 mrb_struct_aset(mrb_state *mrb, mrb_value s)
mzta 0:158c61bb030f 628 {
mzta 0:158c61bb030f 629 mrb_int i;
mzta 0:158c61bb030f 630 mrb_value idx;
mzta 0:158c61bb030f 631 mrb_value val;
mzta 0:158c61bb030f 632
mzta 0:158c61bb030f 633 mrb_get_args(mrb, "oo", &idx, &val);
mzta 0:158c61bb030f 634
mzta 0:158c61bb030f 635 if (mrb_string_p(idx)) {
mzta 0:158c61bb030f 636 mrb_value sym = mrb_check_intern_str(mrb, idx);
mzta 0:158c61bb030f 637
mzta 0:158c61bb030f 638 if (mrb_nil_p(sym)) {
mzta 0:158c61bb030f 639 mrb_raisef(mrb, E_INDEX_ERROR, "no member '%S' in struct", idx);
mzta 0:158c61bb030f 640 }
mzta 0:158c61bb030f 641 idx = sym;
mzta 0:158c61bb030f 642 }
mzta 0:158c61bb030f 643 if (mrb_symbol_p(idx)) {
mzta 0:158c61bb030f 644 return mrb_struct_aset_sym(mrb, s, mrb_symbol(idx), val);
mzta 0:158c61bb030f 645 }
mzta 0:158c61bb030f 646
mzta 0:158c61bb030f 647 i = mrb_int(mrb, idx);
mzta 0:158c61bb030f 648 if (i < 0) i = RSTRUCT_LEN(s) + i;
mzta 0:158c61bb030f 649 if (i < 0) {
mzta 0:158c61bb030f 650 mrb_raisef(mrb, E_INDEX_ERROR,
mzta 0:158c61bb030f 651 "offset %S too small for struct(size:%S)",
mzta 0:158c61bb030f 652 mrb_fixnum_value(i), mrb_fixnum_value(RSTRUCT_LEN(s)));
mzta 0:158c61bb030f 653 }
mzta 0:158c61bb030f 654 if (RSTRUCT_LEN(s) <= i) {
mzta 0:158c61bb030f 655 mrb_raisef(mrb, E_INDEX_ERROR,
mzta 0:158c61bb030f 656 "offset %S too large for struct(size:%S)",
mzta 0:158c61bb030f 657 mrb_fixnum_value(i), mrb_fixnum_value(RSTRUCT_LEN(s)));
mzta 0:158c61bb030f 658 }
mzta 0:158c61bb030f 659 return RSTRUCT_PTR(s)[i] = val;
mzta 0:158c61bb030f 660 }
mzta 0:158c61bb030f 661
mzta 0:158c61bb030f 662 /* 15.2.18.4.1 */
mzta 0:158c61bb030f 663 /*
mzta 0:158c61bb030f 664 * call-seq:
mzta 0:158c61bb030f 665 * struct == other_struct -> true or false
mzta 0:158c61bb030f 666 *
mzta 0:158c61bb030f 667 * Equality---Returns <code>true</code> if <i>other_struct</i> is
mzta 0:158c61bb030f 668 * equal to this one: they must be of the same class as generated by
mzta 0:158c61bb030f 669 * <code>Struct::new</code>, and the values of all instance variables
mzta 0:158c61bb030f 670 * must be equal (according to <code>Object#==</code>).
mzta 0:158c61bb030f 671 *
mzta 0:158c61bb030f 672 * Customer = Struct.new(:name, :address, :zip)
mzta 0:158c61bb030f 673 * joe = Customer.new("Joe Smith", "123 Maple, Anytown NC", 12345)
mzta 0:158c61bb030f 674 * joejr = Customer.new("Joe Smith", "123 Maple, Anytown NC", 12345)
mzta 0:158c61bb030f 675 * jane = Customer.new("Jane Doe", "456 Elm, Anytown NC", 12345)
mzta 0:158c61bb030f 676 * joe == joejr #=> true
mzta 0:158c61bb030f 677 * joe == jane #=> false
mzta 0:158c61bb030f 678 */
mzta 0:158c61bb030f 679
mzta 0:158c61bb030f 680 static mrb_value
mzta 0:158c61bb030f 681 mrb_struct_equal(mrb_state *mrb, mrb_value s)
mzta 0:158c61bb030f 682 {
mzta 0:158c61bb030f 683 mrb_value s2;
mzta 0:158c61bb030f 684 mrb_value *ptr, *ptr2;
mzta 0:158c61bb030f 685 mrb_int i, len;
mzta 0:158c61bb030f 686 mrb_bool equal_p;
mzta 0:158c61bb030f 687
mzta 0:158c61bb030f 688 mrb_get_args(mrb, "o", &s2);
mzta 0:158c61bb030f 689 if (mrb_obj_equal(mrb, s, s2)) {
mzta 0:158c61bb030f 690 equal_p = 1;
mzta 0:158c61bb030f 691 }
mzta 0:158c61bb030f 692 else if (!strcmp(mrb_class_name(mrb, mrb_obj_class(mrb, s)), "Struct") ||
mzta 0:158c61bb030f 693 mrb_obj_class(mrb, s) != mrb_obj_class(mrb, s2)) {
mzta 0:158c61bb030f 694 equal_p = 0;
mzta 0:158c61bb030f 695 }
mzta 0:158c61bb030f 696 else if (RSTRUCT_LEN(s) != RSTRUCT_LEN(s2)) {
mzta 0:158c61bb030f 697 mrb_bug(mrb, "inconsistent struct"); /* should never happen */
mzta 0:158c61bb030f 698 equal_p = 0; /* This substuture is just to suppress warnings. never called. */
mzta 0:158c61bb030f 699 }
mzta 0:158c61bb030f 700 else {
mzta 0:158c61bb030f 701 ptr = RSTRUCT_PTR(s);
mzta 0:158c61bb030f 702 ptr2 = RSTRUCT_PTR(s2);
mzta 0:158c61bb030f 703 len = RSTRUCT_LEN(s);
mzta 0:158c61bb030f 704 equal_p = 1;
mzta 0:158c61bb030f 705 for (i=0; i<len; i++) {
mzta 0:158c61bb030f 706 if (!mrb_equal(mrb, ptr[i], ptr2[i])) {
mzta 0:158c61bb030f 707 equal_p = 0;
mzta 0:158c61bb030f 708 break;
mzta 0:158c61bb030f 709 }
mzta 0:158c61bb030f 710 }
mzta 0:158c61bb030f 711 }
mzta 0:158c61bb030f 712
mzta 0:158c61bb030f 713 return mrb_bool_value(equal_p);
mzta 0:158c61bb030f 714 }
mzta 0:158c61bb030f 715
mzta 0:158c61bb030f 716 /* 15.2.18.4.12(x) */
mzta 0:158c61bb030f 717 /*
mzta 0:158c61bb030f 718 * code-seq:
mzta 0:158c61bb030f 719 * struct.eql?(other) -> true or false
mzta 0:158c61bb030f 720 *
mzta 0:158c61bb030f 721 * Two structures are equal if they are the same object, or if all their
mzta 0:158c61bb030f 722 * fields are equal (using <code>eql?</code>).
mzta 0:158c61bb030f 723 */
mzta 0:158c61bb030f 724 static mrb_value
mzta 0:158c61bb030f 725 mrb_struct_eql(mrb_state *mrb, mrb_value s)
mzta 0:158c61bb030f 726 {
mzta 0:158c61bb030f 727 mrb_value s2;
mzta 0:158c61bb030f 728 mrb_value *ptr, *ptr2;
mzta 0:158c61bb030f 729 mrb_int i, len;
mzta 0:158c61bb030f 730 mrb_bool eql_p;
mzta 0:158c61bb030f 731
mzta 0:158c61bb030f 732 mrb_get_args(mrb, "o", &s2);
mzta 0:158c61bb030f 733 if (mrb_obj_equal(mrb, s, s2)) {
mzta 0:158c61bb030f 734 eql_p = 1;
mzta 0:158c61bb030f 735 }
mzta 0:158c61bb030f 736 else if (strcmp(mrb_class_name(mrb, mrb_obj_class(mrb, s2)), "Struct") ||
mzta 0:158c61bb030f 737 mrb_obj_class(mrb, s) != mrb_obj_class(mrb, s2)) {
mzta 0:158c61bb030f 738 eql_p = 0;
mzta 0:158c61bb030f 739 }
mzta 0:158c61bb030f 740 else if (RSTRUCT_LEN(s) != RSTRUCT_LEN(s2)) {
mzta 0:158c61bb030f 741 mrb_bug(mrb, "inconsistent struct"); /* should never happen */
mzta 0:158c61bb030f 742 eql_p = 0; /* This substuture is just to suppress warnings. never called. */
mzta 0:158c61bb030f 743 }
mzta 0:158c61bb030f 744 else {
mzta 0:158c61bb030f 745 ptr = RSTRUCT_PTR(s);
mzta 0:158c61bb030f 746 ptr2 = RSTRUCT_PTR(s2);
mzta 0:158c61bb030f 747 len = RSTRUCT_LEN(s);
mzta 0:158c61bb030f 748 eql_p = 1;
mzta 0:158c61bb030f 749 for (i=0; i<len; i++) {
mzta 0:158c61bb030f 750 if (!mrb_eql(mrb, ptr[i], ptr2[i])) {
mzta 0:158c61bb030f 751 eql_p = 0;
mzta 0:158c61bb030f 752 break;
mzta 0:158c61bb030f 753 }
mzta 0:158c61bb030f 754 }
mzta 0:158c61bb030f 755 }
mzta 0:158c61bb030f 756
mzta 0:158c61bb030f 757 return mrb_bool_value(eql_p);
mzta 0:158c61bb030f 758 }
mzta 0:158c61bb030f 759
mzta 0:158c61bb030f 760 /*
mzta 0:158c61bb030f 761 * call-seq:
mzta 0:158c61bb030f 762 * struct.length -> Fixnum
mzta 0:158c61bb030f 763 * struct.size -> Fixnum
mzta 0:158c61bb030f 764 *
mzta 0:158c61bb030f 765 * Returns number of struct members.
mzta 0:158c61bb030f 766 */
mzta 0:158c61bb030f 767 static mrb_value
mzta 0:158c61bb030f 768 mrb_struct_len(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 769 {
mzta 0:158c61bb030f 770 return mrb_fixnum_value(RSTRUCT_LEN(self));
mzta 0:158c61bb030f 771 }
mzta 0:158c61bb030f 772
mzta 0:158c61bb030f 773 /*
mzta 0:158c61bb030f 774 * call-seq:
mzta 0:158c61bb030f 775 * struct.to_a -> array
mzta 0:158c61bb030f 776 * struct.values -> array
mzta 0:158c61bb030f 777 *
mzta 0:158c61bb030f 778 * Create an array from struct values.
mzta 0:158c61bb030f 779 */
mzta 0:158c61bb030f 780 static mrb_value
mzta 0:158c61bb030f 781 mrb_struct_to_a(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 782 {
mzta 0:158c61bb030f 783 return mrb_ary_new_from_values(mrb, RSTRUCT_LEN(self), RSTRUCT_PTR(self));
mzta 0:158c61bb030f 784 }
mzta 0:158c61bb030f 785
mzta 0:158c61bb030f 786 /*
mzta 0:158c61bb030f 787 * call-seq:
mzta 0:158c61bb030f 788 * struct.to_h -> hash
mzta 0:158c61bb030f 789 *
mzta 0:158c61bb030f 790 * Create a hash from member names and struct values.
mzta 0:158c61bb030f 791 */
mzta 0:158c61bb030f 792 static mrb_value
mzta 0:158c61bb030f 793 mrb_struct_to_h(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 794 {
mzta 0:158c61bb030f 795 mrb_value members, ret;
mzta 0:158c61bb030f 796 mrb_int i;
mzta 0:158c61bb030f 797
mzta 0:158c61bb030f 798 members = mrb_struct_s_members(mrb, mrb_obj_value(mrb_class(mrb, self)));
mzta 0:158c61bb030f 799 ret = mrb_hash_new_capa(mrb, RARRAY_LEN(members));
mzta 0:158c61bb030f 800
mzta 0:158c61bb030f 801 for (i = 0; i < RARRAY_LEN(members); ++i) {
mzta 0:158c61bb030f 802 mrb_hash_set(mrb, ret, RARRAY_PTR(members)[i], RSTRUCT_PTR(self)[i]);
mzta 0:158c61bb030f 803 }
mzta 0:158c61bb030f 804
mzta 0:158c61bb030f 805 return ret;
mzta 0:158c61bb030f 806 }
mzta 0:158c61bb030f 807
mzta 0:158c61bb030f 808 static mrb_value
mzta 0:158c61bb030f 809 mrb_struct_values_at(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 810 {
mzta 0:158c61bb030f 811 mrb_int argc;
mzta 0:158c61bb030f 812 mrb_value *argv;
mzta 0:158c61bb030f 813
mzta 0:158c61bb030f 814 mrb_get_args(mrb, "*", &argv, &argc);
mzta 0:158c61bb030f 815
mzta 0:158c61bb030f 816 return mrb_get_values_at(mrb, self, RSTRUCT_LEN(self), argc, argv, struct_aref_int);
mzta 0:158c61bb030f 817 }
mzta 0:158c61bb030f 818
mzta 0:158c61bb030f 819 /*
mzta 0:158c61bb030f 820 * A <code>Struct</code> is a convenient way to bundle a number of
mzta 0:158c61bb030f 821 * attributes together, using accessor methods, without having to write
mzta 0:158c61bb030f 822 * an explicit class.
mzta 0:158c61bb030f 823 *
mzta 0:158c61bb030f 824 * The <code>Struct</code> class is a generator of specific classes,
mzta 0:158c61bb030f 825 * each one of which is defined to hold a set of variables and their
mzta 0:158c61bb030f 826 * accessors. In these examples, we'll call the generated class
mzta 0:158c61bb030f 827 * ``<i>Customer</i>Class,'' and we'll show an example instance of that
mzta 0:158c61bb030f 828 * class as ``<i>Customer</i>Inst.''
mzta 0:158c61bb030f 829 *
mzta 0:158c61bb030f 830 * In the descriptions that follow, the parameter <i>symbol</i> refers
mzta 0:158c61bb030f 831 * to a symbol, which is either a quoted string or a
mzta 0:158c61bb030f 832 * <code>Symbol</code> (such as <code>:name</code>).
mzta 0:158c61bb030f 833 */
mzta 0:158c61bb030f 834 void
mzta 0:158c61bb030f 835 mrb_mruby_struct_gem_init(mrb_state* mrb)
mzta 0:158c61bb030f 836 {
mzta 0:158c61bb030f 837 struct RClass *st;
mzta 0:158c61bb030f 838 st = mrb_define_class(mrb, "Struct", mrb->object_class);
mzta 0:158c61bb030f 839
mzta 0:158c61bb030f 840 mrb_define_class_method(mrb, st, "new", mrb_struct_s_def, MRB_ARGS_ANY()); /* 15.2.18.3.1 */
mzta 0:158c61bb030f 841
mzta 0:158c61bb030f 842 mrb_define_method(mrb, st, "==", mrb_struct_equal, MRB_ARGS_REQ(1)); /* 15.2.18.4.1 */
mzta 0:158c61bb030f 843 mrb_define_method(mrb, st, "[]", mrb_struct_aref, MRB_ARGS_REQ(1)); /* 15.2.18.4.2 */
mzta 0:158c61bb030f 844 mrb_define_method(mrb, st, "[]=", mrb_struct_aset, MRB_ARGS_REQ(2)); /* 15.2.18.4.3 */
mzta 0:158c61bb030f 845 mrb_define_method(mrb, st, "members", mrb_struct_members_m, MRB_ARGS_NONE()); /* 15.2.18.4.6 */
mzta 0:158c61bb030f 846 mrb_define_method(mrb, st, "initialize", mrb_struct_initialize_m,MRB_ARGS_ANY()); /* 15.2.18.4.8 */
mzta 0:158c61bb030f 847 mrb_define_method(mrb, st, "initialize_copy", mrb_struct_init_copy, MRB_ARGS_REQ(1)); /* 15.2.18.4.9 */
mzta 0:158c61bb030f 848 mrb_define_method(mrb, st, "inspect", mrb_struct_inspect, MRB_ARGS_NONE()); /* 15.2.18.4.10(x) */
mzta 0:158c61bb030f 849 mrb_define_alias(mrb, st, "to_s", "inspect"); /* 15.2.18.4.11(x) */
mzta 0:158c61bb030f 850 mrb_define_method(mrb, st, "eql?", mrb_struct_eql, MRB_ARGS_REQ(1)); /* 15.2.18.4.12(x) */
mzta 0:158c61bb030f 851
mzta 0:158c61bb030f 852 mrb_define_method(mrb, st, "size", mrb_struct_len, MRB_ARGS_NONE());
mzta 0:158c61bb030f 853 mrb_define_method(mrb, st, "length", mrb_struct_len, MRB_ARGS_NONE());
mzta 0:158c61bb030f 854 mrb_define_method(mrb, st, "to_a", mrb_struct_to_a, MRB_ARGS_NONE());
mzta 0:158c61bb030f 855 mrb_define_method(mrb, st, "values", mrb_struct_to_a, MRB_ARGS_NONE());
mzta 0:158c61bb030f 856 mrb_define_method(mrb, st, "to_h", mrb_struct_to_h, MRB_ARGS_NONE());
mzta 0:158c61bb030f 857 mrb_define_method(mrb, st, "values_at", mrb_struct_values_at, MRB_ARGS_NONE());
mzta 0:158c61bb030f 858 }
mzta 0:158c61bb030f 859
mzta 0:158c61bb030f 860 void
mzta 0:158c61bb030f 861 mrb_mruby_struct_gem_final(mrb_state* mrb)
mzta 0:158c61bb030f 862 {
mzta 0:158c61bb030f 863 }
mzta 0:158c61bb030f 864