From df6ef39d1c588a945588eb80ced16a8e27d777cf Mon Sep 17 00:00:00 2001 From: "Yukihiro \"Matz\" Matsumoto" Date: Tue, 3 Aug 2021 13:13:42 +0900 Subject: numeric_ext.c: define `modulo` and `remainder` methods for `Integer`. --- mrbgems/mruby-numeric-ext/src/numeric_ext.c | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'mrbgems/mruby-numeric-ext') diff --git a/mrbgems/mruby-numeric-ext/src/numeric_ext.c b/mrbgems/mruby-numeric-ext/src/numeric_ext.c index 8389f913d..d5d031a9a 100644 --- a/mrbgems/mruby-numeric-ext/src/numeric_ext.c +++ b/mrbgems/mruby-numeric-ext/src/numeric_ext.c @@ -50,6 +50,24 @@ int_nobits(mrb_state *mrb, mrb_value self) return mrb_bool_value((n & m) == 0); } +/* + * call-seq: + * num.remainder(numeric) -> real + * + * x.remainder(y) means x-y*(x/y).truncate. + * + * See Numeric#divmod. + */ +static mrb_value +int_remainder(mrb_state *mrb, mrb_value self) +{ + mrb_int n, m; + + mrb_get_args(mrb, "i", &m); + n = mrb_integer(self); + return mrb_int_value(mrb, n % m); +} + void mrb_mruby_numeric_ext_gem_init(mrb_state* mrb) { @@ -59,6 +77,9 @@ mrb_mruby_numeric_ext_gem_init(mrb_state* mrb) mrb_define_method(mrb, i, "anybits?", int_anybits, MRB_ARGS_REQ(1)); mrb_define_method(mrb, i, "nobits?", int_nobits, MRB_ARGS_REQ(1)); + mrb_define_alias(mrb, i, "modulo", "%"); + mrb_define_method(mrb, i, "remainder", int_remainder, MRB_ARGS_REQ(1)); + #ifndef MRB_NO_FLOAT mrb_define_const_id(mrb, mrb->float_class, MRB_SYM(RADIX), mrb_fixnum_value(MRB_FLT_RADIX)); mrb_define_const_id(mrb, mrb->float_class, MRB_SYM(MANT_DIG), mrb_fixnum_value(MRB_FLT_MANT_DIG)); -- cgit v1.2.3