1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

internal.h: round macros

* internal.h (ROUND_FUNC, ROUND_CALL): macros wrapping round
  functions.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@56824 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
nobu 2016-11-18 06:29:51 +00:00
parent d9035a8fd4
commit 81ec47ea18
3 changed files with 26 additions and 16 deletions

View file

@ -1151,8 +1151,14 @@ enum ruby_num_rounding_mode {
RUBY_NUM_ROUND_HALF_EVEN,
RUBY_NUM_ROUND_DEFAULT = ROUND_DEFAULT
};
#define ROUND_TO(mode, up, even) \
((mode) == RUBY_NUM_ROUND_HALF_EVEN ? even : up)
#define ROUND_TO(mode, even, up) \
((mode) == RUBY_NUM_ROUND_HALF_EVEN ? even : \
up)
#define ROUND_FUNC(mode, name) \
ROUND_TO(mode, name##_half_even, name##_half_up)
#define ROUND_CALL(mode, name, args) \
ROUND_TO(mode, name##_half_even args, \
name##_half_up args)
int rb_num_to_uint(VALUE val, unsigned int *ret);
VALUE ruby_num_interval_step_size(VALUE from, VALUE to, VALUE step, int excl);

View file

@ -99,6 +99,7 @@ round_half_up(double x, double s)
#ifdef HAVE_ROUND
f = round(xs);
if (s == 1.0) return f;
#endif
if (x > 0) {
#ifndef HAVE_ROUND
@ -2033,6 +2034,18 @@ int_round_half_up(SIGNED_VALUE x, SIGNED_VALUE y)
return (x + y / 2) / y * y;
}
static int
int_half_p_half_even(VALUE num, VALUE n, VALUE f)
{
return (int)int_odd_p(rb_int_idiv(n, f));
}
static int
int_half_p_half_up(VALUE num, VALUE n, VALUE f)
{
return int_pos_p(num);
}
/*
* Assumes num is an Integer, ndigits <= 0
*/
@ -2050,9 +2063,7 @@ rb_int_round(VALUE num, int ndigits, enum ruby_num_rounding_mode mode)
SIGNED_VALUE x = FIX2LONG(num), y = FIX2LONG(f);
int neg = x < 0;
if (neg) x = -x;
x = ROUND_TO(mode,
int_round_half_up(x, y),
int_round_half_even(x, y));
x = ROUND_CALL(mode, int_round, (x, y));
if (neg) x = -x;
return LONG2NUM(x);
}
@ -2065,10 +2076,7 @@ rb_int_round(VALUE num, int ndigits, enum ruby_num_rounding_mode mode)
n = rb_int_minus(num, r);
r = rb_int_cmp(r, h);
if (FIXNUM_POSITIVE_P(r) ||
(FIXNUM_ZERO_P(r) &&
ROUND_TO(mode,
int_pos_p(num),
(SIGNED_VALUE) int_odd_p(rb_int_idiv(n, f))))) {
(FIXNUM_ZERO_P(r) && ROUND_CALL(mode, int_half_p, (num, n, f)))) {
n = rb_int_plus(n, f);
}
return n;
@ -2199,14 +2207,12 @@ flo_round(int argc, VALUE *argv, VALUE num)
}
number = RFLOAT_VALUE(num);
if (ndigits == 0) {
x = ROUND_TO(mode,
round(number), round_half_even(number, 1.0));
x = ROUND_CALL(mode, round, (number, 1.0));
return dbl2ival(x);
}
if (float_invariant_round(number, ndigits, &num)) return num;
f = pow(10, ndigits);
x = ROUND_TO(mode,
round_half_up(number, f), round_half_even(number, f));
x = ROUND_CALL(mode, round, (number, f));
return DBL2NUM(x / f);
}

View file

@ -1481,9 +1481,7 @@ nurat_round_n(int argc, VALUE *argv, VALUE self)
enum ruby_num_rounding_mode mode = (
argc = rb_scan_args(argc, argv, "*:", NULL, &opt),
rb_num_get_rounding_option(opt));
VALUE (*round_func)(VALUE) =
ROUND_TO(mode,
nurat_round_half_up, nurat_round_half_even);
VALUE (*round_func)(VALUE) = ROUND_FUNC(mode, nurat_round);
return f_round_common(argc, argv, self, round_func);
}