mirror of
https://github.com/mruby/mruby
synced 2026-06-08 16:11:16 +00:00
Yukihiro "Matz" Matsumoto
6aa5d53867
`MRB_USE_COMPLEX` is exclusive with `MRB_NO_FLOAT`, so we only need to check the former.
113 lines
2.9 KiB
C
113 lines
2.9 KiB
C
/*
|
|
** numeric.c - Numeric, Integer, Float class
|
|
**
|
|
** See Copyright Notice in mruby.h
|
|
*/
|
|
|
|
#include <mruby.h>
|
|
#include <mruby/numeric.h>
|
|
#include <mruby/presym.h>
|
|
|
|
mrb_value mrb_int_add(mrb_state *mrb, mrb_value x, mrb_value y);
|
|
mrb_value mrb_int_sub(mrb_state *mrb, mrb_value x, mrb_value y);
|
|
mrb_value mrb_int_mul(mrb_state *mrb, mrb_value x, mrb_value y);
|
|
|
|
#ifdef MRB_USE_COMPLEX
|
|
mrb_value mrb_complex_new(mrb_state *mrb, mrb_float x, mrb_float y);
|
|
mrb_value mrb_complex_add(mrb_state *mrb, mrb_value x, mrb_value y);
|
|
mrb_value mrb_complex_sub(mrb_state *mrb, mrb_value x, mrb_value y);
|
|
mrb_value mrb_complex_mul(mrb_state *mrb, mrb_value x, mrb_value y);
|
|
#endif
|
|
|
|
mrb_value mrb_rational_new(mrb_state *mrb, mrb_int x, mrb_int y);
|
|
mrb_value mrb_rational_add(mrb_state *mrb, mrb_value x, mrb_value y);
|
|
mrb_value mrb_rational_sub(mrb_state *mrb, mrb_value x, mrb_value y);
|
|
mrb_value mrb_rational_mul(mrb_state *mrb, mrb_value x, mrb_value y);
|
|
|
|
MRB_API mrb_value
|
|
mrb_num_add(mrb_state *mrb, mrb_value x, mrb_value y)
|
|
{
|
|
if (mrb_integer_p(x)) {
|
|
return mrb_int_add(mrb, x, y);
|
|
}
|
|
#ifndef MRB_NO_FLOAT
|
|
if (mrb_float_p(x)) {
|
|
return mrb_float_value(mrb, mrb_float(x) + mrb_as_float(mrb, y));
|
|
}
|
|
#endif
|
|
#if defined(MRB_USE_RATIONAL) || defined(MRB_USE_COMPLEX)
|
|
switch (mrb_type(x)) {
|
|
#if defined(MRB_USE_RATIONAL)
|
|
case MRB_TT_RATIONAL:
|
|
return mrb_rational_add(mrb, x, y);
|
|
#endif
|
|
#if defined(MRB_USE_COMPLEX)
|
|
case MRB_TT_COMPLEX:
|
|
return mrb_complex_add(mrb, x, y);
|
|
#endif
|
|
default:
|
|
break;
|
|
}
|
|
#endif
|
|
mrb_raise(mrb, E_TYPE_ERROR, "no number addition");
|
|
return mrb_nil_value(); /* not reached */
|
|
}
|
|
|
|
MRB_API mrb_value
|
|
mrb_num_sub(mrb_state *mrb, mrb_value x, mrb_value y)
|
|
{
|
|
if (mrb_integer_p(x)) {
|
|
return mrb_int_sub(mrb, x, y);
|
|
}
|
|
#ifndef MRB_NO_FLOAT
|
|
if (mrb_float_p(x)) {
|
|
return mrb_float_value(mrb, mrb_float(x) - mrb_as_float(mrb, y));
|
|
}
|
|
#endif
|
|
#if defined(MRB_USE_RATIONAL) || defined(MRB_USE_COMPLEX)
|
|
switch (mrb_type(x)) {
|
|
#if defined(MRB_USE_RATIONAL)
|
|
case MRB_TT_RATIONAL:
|
|
return mrb_rational_sub(mrb, x, y);
|
|
#endif
|
|
#if defined(MRB_USE_COMPLEX)
|
|
case MRB_TT_COMPLEX:
|
|
return mrb_complex_sub(mrb, x, y);
|
|
#endif
|
|
default:
|
|
break;
|
|
}
|
|
#endif
|
|
mrb_raise(mrb, E_TYPE_ERROR, "no number subtraction");
|
|
return mrb_nil_value(); /* not reached */
|
|
}
|
|
|
|
MRB_API mrb_value
|
|
mrb_num_mul(mrb_state *mrb, mrb_value x, mrb_value y)
|
|
{
|
|
if (mrb_integer_p(x)) {
|
|
return mrb_int_mul(mrb, x, y);
|
|
}
|
|
#ifndef MRB_NO_FLOAT
|
|
if (mrb_float_p(x)) {
|
|
return mrb_float_value(mrb, mrb_float(x) * mrb_as_float(mrb, y));
|
|
}
|
|
#endif
|
|
#if defined(MRB_USE_RATIONAL) || defined(MRB_USE_COMPLEX)
|
|
switch (mrb_type(x)) {
|
|
#if defined(MRB_USE_RATIONAL)
|
|
case MRB_TT_RATIONAL:
|
|
return mrb_rational_mul(mrb, x, y);
|
|
#endif
|
|
#if defined(MRB_USE_COMPLEX)
|
|
case MRB_TT_COMPLEX:
|
|
return mrb_complex_mul(mrb, x, y);
|
|
#endif
|
|
default:
|
|
break;
|
|
}
|
|
#endif
|
|
mrb_raise(mrb, E_TYPE_ERROR, "no number multiply");
|
|
return mrb_nil_value(); /* not reached */
|
|
}
|