[331] | 1 | #include <mruby.h>
|
---|
| 2 | #include <mruby/class.h>
|
---|
| 3 | #include <mruby/string.h>
|
---|
| 4 | #include <mruby/istruct.h>
|
---|
| 5 |
|
---|
| 6 | static mrb_value
|
---|
| 7 | istruct_test_initialize(mrb_state *mrb, mrb_value self)
|
---|
| 8 | {
|
---|
| 9 | char *string = (char*)mrb_istruct_ptr(self);
|
---|
| 10 | mrb_int size = mrb_istruct_size();
|
---|
| 11 | mrb_value object;
|
---|
| 12 | mrb_get_args(mrb, "o", &object);
|
---|
| 13 |
|
---|
| 14 | if (mrb_float_p(object))
|
---|
| 15 | {
|
---|
| 16 | snprintf(string, size, "float(%.3f)", mrb_float(object));
|
---|
| 17 | }
|
---|
| 18 | else if (mrb_fixnum_p(object))
|
---|
| 19 | {
|
---|
| 20 | snprintf(string, size, "fixnum(%" MRB_PRId ")", mrb_fixnum(object));
|
---|
| 21 | }
|
---|
| 22 | else if (mrb_string_p(object))
|
---|
| 23 | {
|
---|
| 24 | snprintf(string, size, "string(%s)", mrb_string_value_cstr(mrb, &object));
|
---|
| 25 | }
|
---|
| 26 |
|
---|
| 27 | string[size - 1] = 0; // force NULL at the end
|
---|
| 28 | return self;
|
---|
| 29 | }
|
---|
| 30 |
|
---|
| 31 | static mrb_value
|
---|
| 32 | istruct_test_to_s(mrb_state *mrb, mrb_value self)
|
---|
| 33 | {
|
---|
| 34 | return mrb_str_new_cstr(mrb, (const char*)mrb_istruct_ptr(self));
|
---|
| 35 | }
|
---|
| 36 |
|
---|
| 37 | static mrb_value
|
---|
| 38 | istruct_test_length(mrb_state *mrb, mrb_value self)
|
---|
| 39 | {
|
---|
| 40 | return mrb_fixnum_value(mrb_istruct_size());
|
---|
| 41 | }
|
---|
| 42 |
|
---|
| 43 | static mrb_value
|
---|
| 44 | istruct_test_test_receive(mrb_state *mrb, mrb_value self)
|
---|
| 45 | {
|
---|
| 46 | mrb_value object;
|
---|
| 47 | mrb_get_args(mrb, "o", &object);
|
---|
| 48 | if (mrb_obj_class(mrb, object) != mrb_class_get(mrb, "InlineStructTest"))
|
---|
| 49 | {
|
---|
| 50 | mrb_raisef(mrb, E_TYPE_ERROR, "Expected InlineStructTest");
|
---|
| 51 | }
|
---|
| 52 | return mrb_bool_value(((char*)mrb_istruct_ptr(object))[0] == 's');
|
---|
| 53 | }
|
---|
| 54 |
|
---|
| 55 | static mrb_value
|
---|
| 56 | istruct_test_test_receive_direct(mrb_state *mrb, mrb_value self)
|
---|
| 57 | {
|
---|
| 58 | char *ptr;
|
---|
| 59 | mrb_get_args(mrb, "I", &ptr);
|
---|
| 60 | return mrb_bool_value(ptr[0] == 's');
|
---|
| 61 | }
|
---|
| 62 |
|
---|
| 63 | static mrb_value
|
---|
| 64 | istruct_test_mutate(mrb_state *mrb, mrb_value self)
|
---|
| 65 | {
|
---|
| 66 | char *ptr = (char*)mrb_istruct_ptr(self);
|
---|
| 67 | memcpy(ptr, "mutate", 6);
|
---|
| 68 | return mrb_nil_value();
|
---|
| 69 | }
|
---|
| 70 |
|
---|
| 71 | void mrb_mruby_inline_struct_gem_test(mrb_state *mrb)
|
---|
| 72 | {
|
---|
| 73 | struct RClass *cls;
|
---|
| 74 |
|
---|
| 75 | cls = mrb_define_class(mrb, "InlineStructTest", mrb->object_class);
|
---|
| 76 | MRB_SET_INSTANCE_TT(cls, MRB_TT_ISTRUCT);
|
---|
| 77 | mrb_define_method(mrb, cls, "initialize", istruct_test_initialize, MRB_ARGS_REQ(1));
|
---|
| 78 | mrb_define_method(mrb, cls, "to_s", istruct_test_to_s, MRB_ARGS_NONE());
|
---|
| 79 | mrb_define_method(mrb, cls, "mutate", istruct_test_mutate, MRB_ARGS_NONE());
|
---|
| 80 | mrb_define_class_method(mrb, cls, "length", istruct_test_length, MRB_ARGS_NONE());
|
---|
| 81 | mrb_define_class_method(mrb, cls, "test_receive", istruct_test_test_receive, MRB_ARGS_REQ(1));
|
---|
| 82 | mrb_define_class_method(mrb, cls, "test_receive_direct", istruct_test_test_receive_direct, MRB_ARGS_REQ(1));
|
---|
| 83 | }
|
---|