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 | }
|
---|