admin/mruby-mruby
1
0
Fork 0
mirror of https://github.com/mruby/mruby synced 2026-06-08 16:11:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

range.c (mrb_get_values_at): support bigint too

Browse Source
This commit is contained in:
Yukihiro "Matz" Matsumoto
2024-08-05 16:35:41 +09:00
parent 6da2e60edc
commit aad2aacf25
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/range.c
+11 -4
View File
@@ -420,10 +420,17 @@ mrb_get_values_at(mrb_state *mrb, mrb_value obj, mrb_int olen, mrb_int argc, con
result = mrb_ary_new(mrb);
for (i = 0; i < argc; i++) {
if (mrb_integer_p(argv[i])) {
mrb_ary_push(mrb, result, func(mrb, obj, mrb_integer(argv[i])));
mrb_value v = argv[i];
if (mrb_integer_p(v)
#ifdef MRB_USE_BIGINT
|| mrb_bigint_p(v)
#endif
) {
mrb_int i = mrb_as_int(mrb, v);
mrb_ary_push(mrb, result, func(mrb, obj, i));
}
else if (mrb_range_beg_len(mrb, argv[i], &beg, &len, olen, FALSE) == MRB_RANGE_OK) {
else if (mrb_range_beg_len(mrb, v, &beg, &len, olen, FALSE) == MRB_RANGE_OK) {
mrb_int const end = olen < beg + len ? olen : beg + len;
for (j = beg; j < end; j++) {
mrb_ary_push(mrb, result, func(mrb, obj, j));
@@ -434,7 +441,7 @@ mrb_get_values_at(mrb_state *mrb, mrb_value obj, mrb_int olen, mrb_int argc, con
}
}
else {
mrb_raisef(mrb, E_TYPE_ERROR, "invalid values selector: %v", argv[i]);
mrb_raisef(mrb, E_TYPE_ERROR, "invalid values selector: %v", v);
}
}