summaryrefslogtreecommitdiff
path: root/spec/ruby/optional/capi/ext/data_spec.c
blob: ac6b51b454fe03f17c7d3ea45db7210731604efe (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include "ruby.h"
#include "rubyspec.h"

#include <string.h>

#ifdef __cplusplus
extern "C" {
#endif

#if defined(HAVE_RDATA) && defined(HAVE_DATA_WRAP_STRUCT)
struct sample_wrapped_struct {
    int foo;
};

void sample_wrapped_struct_free(void* st) {
  free(st);
}

void sample_wrapped_struct_mark(void* st) {
}

VALUE sdaf_alloc_func(VALUE klass) {
  struct sample_wrapped_struct* bar = malloc(sizeof(struct sample_wrapped_struct));
  bar->foo = 42;
  return Data_Wrap_Struct(klass, &sample_wrapped_struct_mark, &sample_wrapped_struct_free, bar);
}

VALUE sdaf_get_struct(VALUE self) {
  struct sample_wrapped_struct* bar;
  Data_Get_Struct(self, struct sample_wrapped_struct, bar);

  return INT2FIX((*bar).foo);
}

VALUE sws_wrap_struct(VALUE self, VALUE val) {
  struct sample_wrapped_struct* bar = malloc(sizeof(struct sample_wrapped_struct));
  bar->foo = FIX2INT(val);
  return Data_Wrap_Struct(rb_cObject, &sample_wrapped_struct_mark, &sample_wrapped_struct_free, bar);
}

VALUE sws_get_struct(VALUE self, VALUE obj) {
  struct sample_wrapped_struct* bar;
  Data_Get_Struct(obj, struct sample_wrapped_struct, bar);

  return INT2FIX((*bar).foo);
}

VALUE sws_get_struct_rdata(VALUE self, VALUE obj) {
  struct sample_wrapped_struct* bar;
  bar = (struct sample_wrapped_struct*) RDATA(obj)->data;
  return INT2FIX(bar->foo);
}

VALUE sws_get_struct_data_ptr(VALUE self, VALUE obj) {
  struct sample_wrapped_struct* bar;
  bar = (struct sample_wrapped_struct*) DATA_PTR(obj);
  return INT2FIX(bar->foo);
}

VALUE sws_change_struct(VALUE self, VALUE obj, VALUE new_val) {
  struct sample_wrapped_struct *old_struct, *new_struct;
  new_struct = malloc(sizeof(struct sample_wrapped_struct));
  new_struct->foo = FIX2INT(new_val);
  old_struct = RDATA(obj)->data;
  free(old_struct);
  RDATA(obj)->data = new_struct;
  return Qnil;
}
#endif

void Init_data_spec(void) {
  VALUE cls;
  cls = rb_define_class("CApiAllocSpecs", rb_cObject);

#if defined(HAVE_RDATA) && defined(HAVE_DATA_WRAP_STRUCT)
  rb_define_alloc_func(cls, sdaf_alloc_func);
  rb_define_method(cls, "wrapped_data", sdaf_get_struct, 0);

  cls = rb_define_class("CApiWrappedStructSpecs", rb_cObject);
  rb_define_method(cls, "wrap_struct", sws_wrap_struct, 1);
  rb_define_method(cls, "get_struct", sws_get_struct, 1);
  rb_define_method(cls, "get_struct_rdata", sws_get_struct_rdata, 1);
  rb_define_method(cls, "get_struct_data_ptr", sws_get_struct_data_ptr, 1);
  rb_define_method(cls, "change_struct", sws_change_struct, 2);
#endif
}

#ifdef __cplusplus
}
#endif