summaryrefslogtreecommitdiff
path: root/math.c
blob: 0e427035f43280f349f87464f240f333c4b18a4b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
/************************************************

  math.c -

  $Author$
  $Date$
  created at: Tue Jan 25 14:12:56 JST 1994

  Copyright (C) 1993-1996 Yukihiro Matsumoto

************************************************/

#include "ruby.h"
#include <math.h>

VALUE mMath;

#define Need_Float(x) (x) = rb_Float(x)
#define Need_Float2(x,y) {\
    Need_Float(x);\
    Need_Float(y);\
}

static VALUE
math_atan2(obj, x, y)
    VALUE obj, x, y;
{
    Need_Float2(x, y);
    return float_new(atan2(RFLOAT(x)->value, RFLOAT(y)->value));
}

static VALUE
math_cos(obj, x)
    VALUE obj, x;
{
    Need_Float(x);

    return float_new(cos(RFLOAT(x)->value));
}

static VALUE
math_sin(obj, x)
    VALUE obj, x;
{
    Need_Float(x);

    return float_new(sin(RFLOAT(x)->value));
}

static VALUE
math_tan(obj, x)
    VALUE obj, x;
{
    Need_Float(x);

    return float_new(tan(RFLOAT(x)->value));
}

static VALUE
math_exp(obj, x)
    VALUE obj, x;
{
    Need_Float(x);
    return float_new(exp(RFLOAT(x)->value));
}

static VALUE
math_log(obj, x)
    VALUE obj, x;
{
    Need_Float(x);
    return float_new(log(RFLOAT(x)->value));
}

static VALUE
math_log10(obj, x)
    VALUE obj, x;
{
    Need_Float(x);
    return float_new(log10(RFLOAT(x)->value));
}

static VALUE
math_sqrt(obj, x)
    VALUE obj, x;
{
    Need_Float(x);

    if (RFLOAT(x)->value < 0.0) ArgError("square root for negative number");
    return float_new(sqrt(RFLOAT(x)->value));
}

void
Init_Math()
{
    mMath = rb_define_module("Math");

#ifdef M_PI
    rb_define_const(mMath, "PI", float_new(M_PI));
#else
    rb_define_const(mMath, "PI", float_new(atan(1.0)*4.0));
#endif

#ifdef M_E
    rb_define_const(mMath, "E", float_new(M_E));
#else
    rb_define_const(mMath, "E", float_new(exp(1.0)));
#endif

    rb_define_module_function(mMath, "atan2", math_atan2, 2);
    rb_define_module_function(mMath, "cos", math_cos, 1);
    rb_define_module_function(mMath, "sin", math_sin, 1);
    rb_define_module_function(mMath, "tan", math_tan, 1);

    rb_define_module_function(mMath, "exp", math_exp, 1);
    rb_define_module_function(mMath, "log", math_log, 1);
    rb_define_module_function(mMath, "log10", math_log10, 1);
    rb_define_module_function(mMath, "sqrt", math_sqrt, 1);
}