third-party/mruby/mrbgems/mruby-inline-struct/test/inline.c

   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_fixnum_p(object)) {
  15     strncpy(string, "fixnum", size-1);
  16   }
  17 #ifndef MRB_WITHOUT_FLOAT
  18   else if (mrb_float_p(object)) {
  19     strncpy(string, "float", size-1);
  20   }
  21 #endif
  22   else if (mrb_string_p(object)) {
  23     strncpy(string, "string", size-1);
  24   }
  25   else {
  26     strncpy(string, "anything", size-1);
  27   }
  28
  29   string[size - 1] = 0; // force NULL at the end
  30   return self;
  31 }
  32
  33 static mrb_value
  34 istruct_test_to_s(mrb_state *mrb, mrb_value self)
  35 {
  36   return mrb_str_new_cstr(mrb, (const char*)mrb_istruct_ptr(self));
  37 }
  38
  39 static mrb_value
  40 istruct_test_length(mrb_state *mrb, mrb_value self)
  41 {
  42   return mrb_fixnum_value(mrb_istruct_size());
  43 }
  44
  45 static mrb_value
  46 istruct_test_test_receive(mrb_state *mrb, mrb_value self)
  47 {
  48   mrb_value object;
  49   mrb_get_args(mrb, "o", &object);
  50   if (mrb_obj_class(mrb, object) != mrb_class_get(mrb, "InlineStructTest"))
  51   {
  52     mrb_raise(mrb, E_TYPE_ERROR, "Expected InlineStructTest");
  53   }
  54   return mrb_bool_value(((char*)mrb_istruct_ptr(object))[0] == 's');
  55 }
  56
  57 static mrb_value
  58 istruct_test_test_receive_direct(mrb_state *mrb, mrb_value self)
  59 {
  60   char *ptr;
  61   mrb_get_args(mrb, "I", &ptr);
  62   return mrb_bool_value(ptr[0] == 's');
  63 }
  64
  65 static mrb_value
  66 istruct_test_mutate(mrb_state *mrb, mrb_value self)
  67 {
  68   char *ptr = (char*)mrb_istruct_ptr(self);
  69   memcpy(ptr, "mutate", 6);
  70   return mrb_nil_value();
  71 }
  72
  73 void mrb_mruby_inline_struct_gem_test(mrb_state *mrb)
  74 {
  75   struct RClass *cls;
  76
  77   cls = mrb_define_class(mrb, "InlineStructTest", mrb->object_class);
  78   MRB_SET_INSTANCE_TT(cls, MRB_TT_ISTRUCT);
  79   mrb_define_method(mrb, cls, "initialize", istruct_test_initialize, MRB_ARGS_REQ(1));
  80   mrb_define_method(mrb, cls, "to_s", istruct_test_to_s, MRB_ARGS_NONE());
  81   mrb_define_method(mrb, cls, "mutate", istruct_test_mutate, MRB_ARGS_NONE());
  82   mrb_define_class_method(mrb, cls, "length", istruct_test_length, MRB_ARGS_NONE());
  83   mrb_define_class_method(mrb, cls, "test_receive", istruct_test_test_receive, MRB_ARGS_REQ(1));
  84   mrb_define_class_method(mrb, cls, "test_receive_direct", istruct_test_test_receive_direct, MRB_ARGS_REQ(1));
  85 }