mirror of
https://github.com/mruby/mruby
synced 2026-06-08 16:11:16 +00:00
Paolo Bosetti
620 lines
12 KiB
C
620 lines
12 KiB
C
/*
|
|
** math.c - Math module
|
|
**
|
|
** See Copyright Notice in mruby.h
|
|
*/
|
|
|
|
#include "mruby.h"
|
|
#include <math.h>
|
|
|
|
#define domain_error(msg) \
|
|
mrb_raise(mrb, E_RANGE_ERROR, "Numerical argument is out of domain - " #msg);
|
|
|
|
/* math functions not provided under Microsoft Visual C++ */
|
|
#ifdef _MSC_VER
|
|
#define asinh(x) log(x + sqrt(pow(x,2.0) + 1))
|
|
#define acosh(x) log(x + sqrt(pow(x,2.0) - 1))
|
|
#define atanh(x) (log(1+x) - log(1-x))/2.0
|
|
#define cbrt(x) pow(x,1.0/3.0)
|
|
|
|
|
|
double erf(double x)
|
|
{
|
|
// constants
|
|
double a1 = 0.254829592;
|
|
double a2 = -0.284496736;
|
|
double a3 = 1.421413741;
|
|
double a4 = -1.453152027;
|
|
double a5 = 1.061405429;
|
|
double p = 0.3275911;
|
|
|
|
// Save the sign of x
|
|
int sign = 1;
|
|
if (x < 0)
|
|
sign = -1;
|
|
x = fabs(x);
|
|
|
|
// A&S formula 7.1.26
|
|
double t = 1.0/(1.0 + p*x);
|
|
double y = 1.0 - (((((a5*t + a4)*t) + a3)*t + a2)*t + a1)*t*exp(-x*x);
|
|
|
|
return sign*y;
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
mrb_value
|
|
mrb_assoc_new(mrb_state *mrb, mrb_value car, mrb_value cdr);
|
|
|
|
/*
|
|
TRIGONOMETRIC FUNCTIONS
|
|
*/
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.sin(x) -> float
|
|
*
|
|
* Computes the sine of <i>x</i> (expressed in radians). Returns
|
|
* -1..1.
|
|
*/
|
|
static mrb_value
|
|
math_sin(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = sin(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.cos(x) -> float
|
|
*
|
|
* Computes the cosine of <i>x</i> (expressed in radians). Returns
|
|
* -1..1.
|
|
*/
|
|
static mrb_value
|
|
math_cos(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = cos(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.tan(x) -> float
|
|
*
|
|
* Returns the tangent of <i>x</i> (expressed in radians).
|
|
*/
|
|
static mrb_value
|
|
math_tan(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = tan(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
INVERSE TRIGONOMETRIC FUNCTIONS
|
|
*/
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.asin(x) -> float
|
|
*
|
|
* Computes the arc sine of <i>x</i>. Returns -{PI/2} .. {PI/2}.
|
|
*/
|
|
static mrb_value
|
|
math_asin(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = asin(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.acos(x) -> float
|
|
*
|
|
* Computes the arc cosine of <i>x</i>. Returns 0..PI.
|
|
*/
|
|
static mrb_value
|
|
math_acos(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = acos(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.atan(x) -> float
|
|
*
|
|
* Computes the arc tangent of <i>x</i>. Returns -{PI/2} .. {PI/2}.
|
|
*/
|
|
static mrb_value
|
|
math_atan(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = atan(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.atan2(y, x) -> float
|
|
*
|
|
* Computes the arc tangent given <i>y</i> and <i>x</i>. Returns
|
|
* -PI..PI.
|
|
*
|
|
* Math.atan2(-0.0, -1.0) #=> -3.141592653589793
|
|
* Math.atan2(-1.0, -1.0) #=> -2.356194490192345
|
|
* Math.atan2(-1.0, 0.0) #=> -1.5707963267948966
|
|
* Math.atan2(-1.0, 1.0) #=> -0.7853981633974483
|
|
* Math.atan2(-0.0, 1.0) #=> -0.0
|
|
* Math.atan2(0.0, 1.0) #=> 0.0
|
|
* Math.atan2(1.0, 1.0) #=> 0.7853981633974483
|
|
* Math.atan2(1.0, 0.0) #=> 1.5707963267948966
|
|
* Math.atan2(1.0, -1.0) #=> 2.356194490192345
|
|
* Math.atan2(0.0, -1.0) #=> 3.141592653589793
|
|
*
|
|
*/
|
|
static mrb_value
|
|
math_atan2(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x, y;
|
|
|
|
mrb_get_args(mrb, "ff", &x, &y);
|
|
x = atan2(x, y);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
HYPERBOLIC TRIG FUNCTIONS
|
|
*/
|
|
/*
|
|
* call-seq:
|
|
* Math.sinh(x) -> float
|
|
*
|
|
* Computes the hyperbolic sine of <i>x</i> (expressed in
|
|
* radians).
|
|
*/
|
|
static mrb_value
|
|
math_sinh(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = sinh(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.cosh(x) -> float
|
|
*
|
|
* Computes the hyperbolic cosine of <i>x</i> (expressed in radians).
|
|
*/
|
|
static mrb_value
|
|
math_cosh(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = cosh(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.tanh() -> float
|
|
*
|
|
* Computes the hyperbolic tangent of <i>x</i> (expressed in
|
|
* radians).
|
|
*/
|
|
static mrb_value
|
|
math_tanh(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = tanh(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
|
|
/*
|
|
INVERSE HYPERBOLIC TRIG FUNCTIONS
|
|
*/
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.asinh(x) -> float
|
|
*
|
|
* Computes the inverse hyperbolic sine of <i>x</i>.
|
|
*/
|
|
static mrb_value
|
|
math_asinh(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
|
|
x = asinh(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.acosh(x) -> float
|
|
*
|
|
* Computes the inverse hyperbolic cosine of <i>x</i>.
|
|
*/
|
|
static mrb_value
|
|
math_acosh(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = acosh(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.atanh(x) -> float
|
|
*
|
|
* Computes the inverse hyperbolic tangent of <i>x</i>.
|
|
*/
|
|
static mrb_value
|
|
math_atanh(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = atanh(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
EXPONENTIALS AND LOGARITHMS
|
|
*/
|
|
#if defined __CYGWIN__
|
|
# include <cygwin/version.h>
|
|
# if CYGWIN_VERSION_DLL_MAJOR < 1005
|
|
# define nan(x) nan()
|
|
# endif
|
|
# define log(x) ((x) < 0.0 ? nan("") : log(x))
|
|
# define log10(x) ((x) < 0.0 ? nan("") : log10(x))
|
|
#endif
|
|
|
|
#ifndef log2
|
|
#ifndef HAVE_LOG2
|
|
double
|
|
log2(double x)
|
|
{
|
|
return log10(x)/log10(2.0);
|
|
}
|
|
#else
|
|
extern double log2(double);
|
|
#endif
|
|
#endif
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.exp(x) -> float
|
|
*
|
|
* Returns e**x.
|
|
*
|
|
* Math.exp(0) #=> 1.0
|
|
* Math.exp(1) #=> 2.718281828459045
|
|
* Math.exp(1.5) #=> 4.4816890703380645
|
|
*
|
|
*/
|
|
static mrb_value
|
|
math_exp(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = exp(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.log(numeric) -> float
|
|
* Math.log(num,base) -> float
|
|
*
|
|
* Returns the natural logarithm of <i>numeric</i>.
|
|
* If additional second argument is given, it will be the base
|
|
* of logarithm.
|
|
*
|
|
* Math.log(1) #=> 0.0
|
|
* Math.log(Math::E) #=> 1.0
|
|
* Math.log(Math::E**3) #=> 3.0
|
|
* Math.log(12,3) #=> 2.2618595071429146
|
|
*
|
|
*/
|
|
static mrb_value
|
|
math_log(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = log(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.log2(numeric) -> float
|
|
*
|
|
* Returns the base 2 logarithm of <i>numeric</i>.
|
|
*
|
|
* Math.log2(1) #=> 0.0
|
|
* Math.log2(2) #=> 1.0
|
|
* Math.log2(32768) #=> 15.0
|
|
* Math.log2(65536) #=> 16.0
|
|
*
|
|
*/
|
|
static mrb_value
|
|
math_log2(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = log2(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.log10(numeric) -> float
|
|
*
|
|
* Returns the base 10 logarithm of <i>numeric</i>.
|
|
*
|
|
* Math.log10(1) #=> 0.0
|
|
* Math.log10(10) #=> 1.0
|
|
* Math.log10(10**100) #=> 100.0
|
|
*
|
|
*/
|
|
static mrb_value
|
|
math_log10(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = log10(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.cbrt(numeric) -> float
|
|
*
|
|
* Returns the cube root of <i>numeric</i>.
|
|
*
|
|
* -9.upto(9) {|x|
|
|
* p [x, Math.cbrt(x), Math.cbrt(x)**3]
|
|
* }
|
|
* #=>
|
|
* [-9, -2.0800838230519, -9.0]
|
|
* [-8, -2.0, -8.0]
|
|
* [-7, -1.91293118277239, -7.0]
|
|
* [-6, -1.81712059283214, -6.0]
|
|
* [-5, -1.7099759466767, -5.0]
|
|
* [-4, -1.5874010519682, -4.0]
|
|
* [-3, -1.44224957030741, -3.0]
|
|
* [-2, -1.25992104989487, -2.0]
|
|
* [-1, -1.0, -1.0]
|
|
* [0, 0.0, 0.0]
|
|
* [1, 1.0, 1.0]
|
|
* [2, 1.25992104989487, 2.0]
|
|
* [3, 1.44224957030741, 3.0]
|
|
* [4, 1.5874010519682, 4.0]
|
|
* [5, 1.7099759466767, 5.0]
|
|
* [6, 1.81712059283214, 6.0]
|
|
* [7, 1.91293118277239, 7.0]
|
|
* [8, 2.0, 8.0]
|
|
* [9, 2.0800838230519, 9.0]
|
|
*
|
|
*/
|
|
static mrb_value
|
|
math_cbrt(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = cbrt(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.frexp(numeric) -> [ fraction, exponent ]
|
|
*
|
|
* Returns a two-element array containing the normalized fraction (a
|
|
* <code>Float</code>) and exponent (a <code>Fixnum</code>) of
|
|
* <i>numeric</i>.
|
|
*
|
|
* fraction, exponent = Math.frexp(1234) #=> [0.6025390625, 11]
|
|
* fraction * 2**exponent #=> 1234.0
|
|
*/
|
|
static mrb_value
|
|
math_frexp(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
int exp;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = frexp(x, &exp);
|
|
|
|
return mrb_assoc_new(mrb, mrb_float_value(x), mrb_fixnum_value(exp));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.ldexp(flt, int) -> float
|
|
*
|
|
* Returns the value of <i>flt</i>*(2**<i>int</i>).
|
|
*
|
|
* fraction, exponent = Math.frexp(1234)
|
|
* Math.ldexp(fraction, exponent) #=> 1234.0
|
|
*/
|
|
static mrb_value
|
|
math_ldexp(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
mrb_int i;
|
|
|
|
mrb_get_args(mrb, "fi", &x, &i);
|
|
x = ldexp(x, i);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.hypot(x, y) -> float
|
|
*
|
|
* Returns sqrt(x**2 + y**2), the hypotenuse of a right-angled triangle
|
|
* with sides <i>x</i> and <i>y</i>.
|
|
*
|
|
* Math.hypot(3, 4) #=> 5.0
|
|
*/
|
|
static mrb_value
|
|
math_hypot(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x, y;
|
|
|
|
mrb_get_args(mrb, "ff", &x, &y);
|
|
x = hypot(x, y);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* Math.erf(x) -> float
|
|
*
|
|
* Calculates the error function of x.
|
|
*/
|
|
static mrb_value
|
|
math_erf(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = erf(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
|
|
|
|
#ifndef _MSC_VER
|
|
/*
|
|
* call-seq:
|
|
* Math.erfc(x) -> float
|
|
*
|
|
* Calculates the complementary error function of x.
|
|
*/
|
|
static mrb_value
|
|
math_erfc(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_float x;
|
|
|
|
mrb_get_args(mrb, "f", &x);
|
|
x = erfc(x);
|
|
|
|
return mrb_float_value(x);
|
|
}
|
|
#endif
|
|
|
|
/* ------------------------------------------------------------------------*/
|
|
void
|
|
mrb_init_math(mrb_state *mrb)
|
|
{
|
|
struct RClass *mrb_math;
|
|
mrb_math = mrb_define_module(mrb, "Math");
|
|
|
|
#ifdef M_PI
|
|
mrb_define_const(mrb, mrb_math, "PI", mrb_float_value(M_PI));
|
|
#else
|
|
mrb_define_const(mrb, mrb_math, "PI", mrb_float_value(atan(1.0)*4.0));
|
|
#endif
|
|
|
|
#ifdef M_E
|
|
mrb_define_const(mrb, mrb_math, "E", mrb_float_value(M_E));
|
|
#else
|
|
mrb_define_const(mrb, mrb_math, "E", mrb_float_value(exp(1.0)));
|
|
#endif
|
|
|
|
mrb_define_class_method(mrb, mrb_math, "sin", math_sin, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "cos", math_cos, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "tan", math_tan, 1);
|
|
|
|
mrb_define_class_method(mrb, mrb_math, "asin", math_asin, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "acos", math_acos, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "atan", math_atan, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "atan2", math_atan2, 2);
|
|
|
|
mrb_define_class_method(mrb, mrb_math, "sinh", math_sinh, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "cosh", math_cosh, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "tanh", math_tanh, 1);
|
|
|
|
mrb_define_class_method(mrb, mrb_math, "asinh", math_asinh, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "acosh", math_acosh, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "atanh", math_atanh, 1);
|
|
|
|
mrb_define_class_method(mrb, mrb_math, "exp", math_exp, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "log", math_log, -1);
|
|
mrb_define_class_method(mrb, mrb_math, "log2", math_log2, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "log10", math_log10, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "cbrt", math_cbrt, 1);
|
|
|
|
mrb_define_class_method(mrb, mrb_math, "frexp", math_frexp, 1);
|
|
mrb_define_class_method(mrb, mrb_math, "ldexp", math_ldexp, 2);
|
|
|
|
mrb_define_class_method(mrb, mrb_math, "hypot", math_hypot, 2);
|
|
|
|
mrb_define_class_method(mrb, mrb_math, "erf", math_erf, 1);
|
|
#ifndef _MSC_VER
|
|
mrb_define_class_method(mrb, mrb_math, "erfc", math_erfc, 1);
|
|
#endif
|
|
}
|