Implement Array#__svalue in C.

This commit is contained in:
Yukihiro "Matz" Matsumoto
2018-04-17 11:32:31 +09:00
parent b51b21fc63
commit d88191e7e7
2 changed files with 16 additions and 6 deletions
-6
View File
@@ -184,12 +184,6 @@ class Array
return block.call if ret.nil? && block
ret
end
# internal method to convert multi-value to single value
def __svalue
return self.first if self.size < 2
self
end
end
##
+16
View File
@@ -1207,6 +1207,21 @@ mrb_ary_cmp(mrb_state *mrb, mrb_value ary1)
return ary2;
}
/* internal method to convert multi-value to single value */
static mrb_value
mrb_ary_svalue(mrb_state *mrb, mrb_value ary)
{
mrb_get_args(mrb, "");
switch (RARRAY_LEN(ary)) {
case 0:
return mrb_nil_value();
case 1:
return RARRAY_PTR(ary)[0];
default:
return ary;
}
}
void
mrb_init_array(mrb_state *mrb)
{
@@ -1248,4 +1263,5 @@ mrb_init_array(mrb_state *mrb)
mrb_define_method(mrb, a, "__ary_eq", mrb_ary_eq, MRB_ARGS_REQ(1));
mrb_define_method(mrb, a, "__ary_cmp", mrb_ary_cmp, MRB_ARGS_REQ(1));
mrb_define_method(mrb, a, "__ary_index", mrb_ary_index_m, MRB_ARGS_REQ(1)); /* kept for mruby-array-ext */
mrb_define_method(mrb, a, "__svalue", mrb_ary_svalue, MRB_ARGS_NONE());
}