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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
|
module Tplt # Template
class << self
def base(gem_name, init_body, final_body)
%{
void
mrb_mruby_#{gem_name}_gem_init(mrb_state* mrb) {
#{init_body}
}
void
mrb_mruby_#{gem_name}_gem_final(mrb_state* mrb) {
#{final_body}
}
}
end
def non_struct_types
@non_struct_types ||= ['bool', 'int', 'float', 'double', 'float', 'char *', 'const char *', 'unsigned int', 'void']
end
def init_module(module_name)
"struct RClass *#{module_name.downcase}_module = mrb_define_module(mrb, \"#{module_name}\");"
end
def get_module(module_name)
"struct RClass *#{module_name.downcase}_mrb_module = mrb_module_get(mrb, \"#{module_name}\");\n"
end
def get_class(class_name, defined_under)
"struct RClass *#{class_name.downcase}_mrb_class = mrb_class_get_under(mrb, #{defined_under.downcase}_mrb_module, mrb_#{class_name}_struct.struct_name);\n"
end
def init_module_function(module_name, function_name, mrb_function_name, mrb_args)
%{
mrb_define_module_function(mrb, #{module_name}, "#{function_name}", mrb_#{mrb_function_name}, #{mrb_args});
}
end
# define under needs the C name, not the ruby name which may be confusing
def init_class(class_name, define_under, is_struct_wrapper = true)
%{
struct RClass *#{class_name.downcase}_class = mrb_define_class_under(mrb, #{define_under}, \"#{class_name}\", mrb->object_class);#{
if is_struct_wrapper
"\nMRB_SET_INSTANCE_TT(#{class_name.downcase}_class, MRB_TT_DATA);"
end
}
}
end
def function(function_name, body)
%{
static mrb_value
mrb_#{function_name}(mrb_state* mrb, mrb_value self) {
#{body}
}
}
end
def init_function(class_name, function_name, mrb_function_name, mrb_args)
%{mrb_define_method(mrb, #{class_name}, "#{function_name}", mrb_#{mrb_function_name}, #{mrb_args});
}
end
def get_kwargs(kwarg_num, init_var_body, init_array_body)
%{
#{init_var_body}
uint32_t kw_num = #{kwarg_num};
const mrb_sym kw_names[] = {
#{init_array_body}
};
mrb_value kw_values[kw_num];
const mrb_kwargs kwargs = { kw_num, 0, kw_names, kw_values, NULL };
mrb_get_args(mrb, "|:", &kwargs);
}
end
def unwrap_kwarg(kwarg_iter, body_if_defined, body_if_undefined = nil, no_argument_error_message = 'Missing Keyword Argument')
%{
if (mrb_undef_p(kw_values[#{kwarg_iter}])) {
#{body_if_undefined || "mrb_load_string(mrb, \"raise ArgumentError.new \\\"#{no_argument_error_message}\\\"\");"}
} else {
#{body_if_defined}
}
}
end
def unwrap_struct(var_name, target, mrb_type, type)
%{#{var_name} = DATA_GET_PTR(mrb, #{target}, &#{mrb_type}, #{type})}
end
def wrap_struct(var_name, target, mrb_type, type)
%{
#{var_name} = (#{type} *)DATA_PTR(#{target})
if(#{var_name}) #{'{'} mrb_free(mrb, #{var_name}); #{'}'}
mrb_data_init(#{target}, NULL, &#{mrb_type});
#{var_name} = (#{type} *)mrb_malloc(mrb, sizeof(#{type}));
}
end
def define_module(module_name)
%{struct RClass *#{module_name.downcase} = mrb_define_module(mrb, "#{module_name}");
}
end
# for converting mrb to C
def to_c(type, variable)
if (type == 'int') || (type == 'unsigned int') || (type == 'bool')
"mrb_as_int(mrb, #{variable})"
elsif (type == 'float') || (type == 'double')
"mrb_as_float(mrb, #{variable})"
elsif (type == 'const char *') || (type == 'char *')
"mrb_str_to_cstr(mrb, #{variable})"
end
end
# for converting C to mrb
def to_mrb(type, variable)
if (type == 'int') || (type == 'unsigned int')
"mrb_fixnum_value(#{variable})"
elsif (type == 'float') || (type == 'double')
"mrb_float_value(mrb, #{variable})"
elsif type == 'bool'
"mrb_bool_value(#{variable})"
elsif (type == 'const char *') || (type == 'char *')
"mrb_str_new_cstr(mrb, #{variable})"
elsif type == 'NULL'
'mrb_nil_value()'
end
end
# convert a C function name to be
# formatted like a Ruby method name
def rubify_func_name(function)
func = function.underscore
if func.start_with? 'is_'
func = func.delete_prefix('is_') + '?'
end
func.delete_prefix('get_')
end
# generate a return of a ruby bound C function
def return_format(function, params)
func_rpart = function.rpartition(' ')
func_datatype = func_rpart.first
func_name = func_rpart.last
result = ''
if func_datatype == 'void'
if params.first == 'void'
result = "#{func_name}();\nreturn mrb_nil_value();"
else
result = "#{func_name}("
result += params.first.rpartition(' ').last
params.drop(1).each do |param|
result += ", #{param.rpartition(' ').last}"
end
result += ");\nreturn mrb_nil_value();"
end
elsif params.first == 'void'
result = "return " + Tplt.to_mrb(func_datatype, "#{func_name}()") + ';'
else
temp_params = params.first.rpartition(' ').last
params.drop(1).each do |param|
temp_params += ", #{param.rpartition(' ').last}"
end
result = 'return ' + Tplt.to_mrb(func_datatype, "#{func_name}(#{temp_params})") + ';'
end
result
end
def return_format_struct(function)
func_rpart = function.rpartition(' ')
func_datatype = func_rpart.first.delete_suffix(' *')
func_name = func_rpart.last
"return mrb_obj_value(Data_Wrap_Struct(mrb, #{func_datatype.downcase}_mrb_class, &mrb_#{func_datatype}_struct, return_value));"
end
# wrapping an existing struct to be used by ruby
def init_struct_wrapper(struct, free_body = nil)
%{
#{"void mrb_helper_#{struct}_free(mrb_state*, void*);" if free_body}
static const struct mrb_data_type mrb_#{struct}_struct = {
"#{struct}",
#{
if free_body
"mrb_helper_#{struct}_free"
else
"mrb_free"
end
}
};
#{
if free_body
%{
void
mrb_helper_#{struct}_free(mrb_state* mrb, void*ptr) {
#{struct} *struct_data = (#{struct}*)ptr;
#{free_body}
mrb_free(mrb, ptr);
}
}
end
}
}
end
end
end
|
