romanbsd-gsl 1.11.2.2

Sign up to get free protection for your applications and to get access to all the features.
Files changed (156) hide show
  1. data/README.rdoc +29 -0
  2. data/Rakefile +54 -0
  3. data/VERSION +1 -0
  4. data/ext/MANIFEST +116 -0
  5. data/ext/array.c +665 -0
  6. data/ext/array_complex.c +247 -0
  7. data/ext/blas.c +29 -0
  8. data/ext/blas1.c +729 -0
  9. data/ext/blas2.c +1093 -0
  10. data/ext/blas3.c +881 -0
  11. data/ext/block.c +44 -0
  12. data/ext/block_source.c +885 -0
  13. data/ext/bspline.c +116 -0
  14. data/ext/bundle.c +3 -0
  15. data/ext/cdf.c +754 -0
  16. data/ext/cheb.c +538 -0
  17. data/ext/combination.c +283 -0
  18. data/ext/common.c +310 -0
  19. data/ext/complex.c +1005 -0
  20. data/ext/const.c +668 -0
  21. data/ext/const_additional.c +120 -0
  22. data/ext/cqp.c +283 -0
  23. data/ext/deriv.c +194 -0
  24. data/ext/dht.c +360 -0
  25. data/ext/diff.c +165 -0
  26. data/ext/dirac.c +395 -0
  27. data/ext/eigen.c +2373 -0
  28. data/ext/error.c +194 -0
  29. data/ext/extconf.rb +272 -0
  30. data/ext/fcmp.c +66 -0
  31. data/ext/fft.c +1092 -0
  32. data/ext/fit.c +205 -0
  33. data/ext/fresnel.c +312 -0
  34. data/ext/function.c +522 -0
  35. data/ext/graph.c +1634 -0
  36. data/ext/gsl.c +265 -0
  37. data/ext/gsl_narray.c +652 -0
  38. data/ext/histogram.c +1717 -0
  39. data/ext/histogram2d.c +1067 -0
  40. data/ext/histogram3d.c +883 -0
  41. data/ext/histogram3d_source.c +750 -0
  42. data/ext/histogram_find.c +101 -0
  43. data/ext/histogram_oper.c +159 -0
  44. data/ext/ieee.c +98 -0
  45. data/ext/integration.c +1138 -0
  46. data/ext/interp.c +511 -0
  47. data/ext/jacobi.c +737 -0
  48. data/ext/linalg.c +4045 -0
  49. data/ext/linalg_complex.c +743 -0
  50. data/ext/math.c +724 -0
  51. data/ext/matrix.c +39 -0
  52. data/ext/matrix_complex.c +1731 -0
  53. data/ext/matrix_double.c +560 -0
  54. data/ext/matrix_int.c +256 -0
  55. data/ext/matrix_source.c +2678 -0
  56. data/ext/min.c +234 -0
  57. data/ext/monte.c +759 -0
  58. data/ext/multifit.c +1810 -0
  59. data/ext/multimin.c +793 -0
  60. data/ext/multimin_fsdf.c +156 -0
  61. data/ext/multiroots.c +952 -0
  62. data/ext/ndlinear.c +320 -0
  63. data/ext/nmf.c +159 -0
  64. data/ext/nmf_wrap.c +63 -0
  65. data/ext/ntuple.c +469 -0
  66. data/ext/odeiv.c +958 -0
  67. data/ext/ool.c +879 -0
  68. data/ext/oper_complex_source.c +253 -0
  69. data/ext/permutation.c +596 -0
  70. data/ext/poly.c +42 -0
  71. data/ext/poly2.c +265 -0
  72. data/ext/poly_source.c +1799 -0
  73. data/ext/qrng.c +171 -0
  74. data/ext/randist.c +1869 -0
  75. data/ext/rational.c +480 -0
  76. data/ext/rng.c +612 -0
  77. data/ext/root.c +408 -0
  78. data/ext/sf.c +1494 -0
  79. data/ext/sf_airy.c +200 -0
  80. data/ext/sf_bessel.c +867 -0
  81. data/ext/sf_clausen.c +28 -0
  82. data/ext/sf_coulomb.c +206 -0
  83. data/ext/sf_coupling.c +118 -0
  84. data/ext/sf_dawson.c +29 -0
  85. data/ext/sf_debye.c +157 -0
  86. data/ext/sf_dilog.c +42 -0
  87. data/ext/sf_elementary.c +44 -0
  88. data/ext/sf_ellint.c +206 -0
  89. data/ext/sf_elljac.c +29 -0
  90. data/ext/sf_erfc.c +93 -0
  91. data/ext/sf_exp.c +164 -0
  92. data/ext/sf_expint.c +211 -0
  93. data/ext/sf_fermi_dirac.c +148 -0
  94. data/ext/sf_gamma.c +344 -0
  95. data/ext/sf_gegenbauer.c +96 -0
  96. data/ext/sf_hyperg.c +197 -0
  97. data/ext/sf_laguerre.c +112 -0
  98. data/ext/sf_lambert.c +47 -0
  99. data/ext/sf_legendre.c +367 -0
  100. data/ext/sf_log.c +104 -0
  101. data/ext/sf_mathieu.c +238 -0
  102. data/ext/sf_power.c +46 -0
  103. data/ext/sf_psi.c +98 -0
  104. data/ext/sf_synchrotron.c +48 -0
  105. data/ext/sf_transport.c +76 -0
  106. data/ext/sf_trigonometric.c +207 -0
  107. data/ext/sf_zeta.c +119 -0
  108. data/ext/signal.c +310 -0
  109. data/ext/siman.c +718 -0
  110. data/ext/sort.c +208 -0
  111. data/ext/spline.c +393 -0
  112. data/ext/stats.c +799 -0
  113. data/ext/sum.c +168 -0
  114. data/ext/tamu_anova.c +56 -0
  115. data/ext/tensor.c +38 -0
  116. data/ext/tensor_source.c +1121 -0
  117. data/ext/vector.c +38 -0
  118. data/ext/vector_complex.c +2139 -0
  119. data/ext/vector_double.c +1445 -0
  120. data/ext/vector_int.c +204 -0
  121. data/ext/vector_source.c +3325 -0
  122. data/ext/wavelet.c +937 -0
  123. data/include/rb_gsl.h +140 -0
  124. data/include/rb_gsl_array.h +230 -0
  125. data/include/rb_gsl_cheb.h +21 -0
  126. data/include/rb_gsl_common.h +343 -0
  127. data/include/rb_gsl_complex.h +25 -0
  128. data/include/rb_gsl_const.h +29 -0
  129. data/include/rb_gsl_dirac.h +15 -0
  130. data/include/rb_gsl_eigen.h +17 -0
  131. data/include/rb_gsl_fft.h +62 -0
  132. data/include/rb_gsl_fit.h +25 -0
  133. data/include/rb_gsl_function.h +27 -0
  134. data/include/rb_gsl_graph.h +70 -0
  135. data/include/rb_gsl_histogram.h +65 -0
  136. data/include/rb_gsl_histogram3d.h +97 -0
  137. data/include/rb_gsl_integration.h +17 -0
  138. data/include/rb_gsl_interp.h +46 -0
  139. data/include/rb_gsl_linalg.h +25 -0
  140. data/include/rb_gsl_math.h +26 -0
  141. data/include/rb_gsl_odeiv.h +21 -0
  142. data/include/rb_gsl_poly.h +71 -0
  143. data/include/rb_gsl_rational.h +37 -0
  144. data/include/rb_gsl_rng.h +21 -0
  145. data/include/rb_gsl_root.h +22 -0
  146. data/include/rb_gsl_sf.h +119 -0
  147. data/include/rb_gsl_statistics.h +17 -0
  148. data/include/rb_gsl_tensor.h +45 -0
  149. data/include/rb_gsl_with_narray.h +22 -0
  150. data/include/templates_off.h +87 -0
  151. data/include/templates_on.h +241 -0
  152. data/lib/gsl/gnuplot.rb +41 -0
  153. data/lib/gsl/oper.rb +68 -0
  154. data/lib/ool.rb +22 -0
  155. data/lib/ool/conmin.rb +30 -0
  156. metadata +221 -0
data/ext/math.c ADDED
@@ -0,0 +1,724 @@
1
+ /*
2
+ math.c
3
+ Ruby/GSL: Ruby extension library for GSL (GNU Scientific Library)
4
+ (C) Copyright 2001-2006 by Yoshiki Tsunesada
5
+
6
+ Ruby/GSL is free software: you can redistribute it and/or modify it
7
+ under the terms of the GNU General Public License.
8
+ This library is distributed in the hope that it will be useful, but
9
+ WITHOUT ANY WARRANTY; without even the implied warranty of
10
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
11
+ */
12
+
13
+ #include "rb_gsl_config.h"
14
+ #include "rb_gsl_math.h"
15
+ #include "rb_gsl_complex.h"
16
+ #ifdef HAVE_NARRAY_H
17
+ #include "narray.h"
18
+ #endif
19
+
20
+ static void rb_gsl_define_const(VALUE module);
21
+
22
+ static void rb_gsl_define_const(VALUE module)
23
+ {
24
+ rb_define_const(module, "M_E", rb_float_new(M_E));
25
+ rb_define_const(module, "M_LOG2E", rb_float_new(M_LOG2E));
26
+ rb_define_const(module, "M_LOG10E", rb_float_new(M_LOG10E));
27
+ rb_define_const(module, "M_SQRT2", rb_float_new(M_SQRT2));
28
+ rb_define_const(module, "M_SQRT1_2", rb_float_new(M_SQRT1_2));
29
+ rb_define_const(module, "M_SQRT3", rb_float_new(M_SQRT3));
30
+ rb_define_const(module, "M_PI", rb_float_new(M_PI));
31
+ rb_define_const(module, "M_PI_2", rb_float_new(M_PI_2));
32
+ rb_define_const(module, "M_PI_4", rb_float_new(M_PI_4));
33
+ rb_define_const(module, "M_SQRTPI", rb_float_new(M_SQRTPI));
34
+ rb_define_const(module, "M_2_SQRTPI", rb_float_new(M_2_SQRTPI));
35
+ rb_define_const(module, "M_1_PI", rb_float_new(M_1_PI));
36
+ rb_define_const(module, "M_2_PI", rb_float_new(M_2_PI));
37
+ rb_define_const(module, "M_LN10", rb_float_new(M_LN10));
38
+ rb_define_const(module, "M_LN2", rb_float_new(M_LN2));
39
+ rb_define_const(module, "M_LNPI", rb_float_new(M_LNPI));
40
+ rb_define_const(module, "M_EULER", rb_float_new(M_EULER));
41
+ rb_define_const(module, "POSINF", rb_float_new(GSL_POSINF));
42
+ rb_define_const(module, "NEGINF", rb_float_new(GSL_NEGINF));
43
+ rb_define_const(module, "NAN", rb_float_new(GSL_NAN));
44
+ }
45
+
46
+ static VALUE rb_GSL_POSINF(VALUE obj)
47
+ {
48
+ return rb_float_new(GSL_POSINF);
49
+ }
50
+
51
+ static VALUE rb_GSL_NEGINF(VALUE obj)
52
+ {
53
+ return rb_float_new(GSL_NEGINF);
54
+ }
55
+
56
+ static VALUE rb_GSL_NAN(VALUE obj)
57
+ {
58
+ return rb_float_new(GSL_NAN);
59
+ }
60
+
61
+ static VALUE rb_gsl_isnan(VALUE obj, VALUE x)
62
+ {
63
+ Need_Float(x);
64
+ return INT2FIX(gsl_isnan(NUM2DBL(x)));
65
+ }
66
+
67
+ static VALUE rb_gsl_isnan2(VALUE obj, VALUE x)
68
+ {
69
+ Need_Float(x);
70
+ if (gsl_isnan(NUM2DBL(x))) return Qtrue;
71
+ else return Qfalse;
72
+ }
73
+
74
+ static VALUE rb_gsl_isinf(VALUE obj, VALUE x)
75
+ {
76
+ Need_Float(x);
77
+ return INT2FIX(gsl_isinf(NUM2DBL(x)));
78
+ }
79
+
80
+ static VALUE rb_gsl_isinf2(VALUE obj, VALUE x)
81
+ {
82
+ // Need_Float(x);
83
+ if (gsl_isinf(NUM2DBL(x))) return Qtrue;
84
+ else return Qfalse;
85
+ }
86
+
87
+ static VALUE rb_gsl_finite(VALUE obj, VALUE x)
88
+ {
89
+ Need_Float(x);
90
+ return INT2FIX(gsl_finite(NUM2DBL(x)));
91
+ }
92
+
93
+ static VALUE rb_gsl_finite2(VALUE obj, VALUE x)
94
+ {
95
+ Need_Float(x);
96
+ if (gsl_finite(NUM2DBL(x))) return Qtrue;
97
+ else return Qfalse;
98
+ }
99
+
100
+ /*****/
101
+
102
+ static VALUE rb_gsl_math_eval(double (*func)(const double), VALUE xx);
103
+
104
+ VALUE rb_gsl_math_complex_eval(gsl_complex (*func)(gsl_complex), VALUE obj)
105
+ {
106
+ gsl_complex *z, *znew;
107
+ gsl_vector_complex *v, *vnew;
108
+ gsl_matrix_complex *m, *mnew;
109
+ size_t i, j;
110
+ if (COMPLEX_P(obj)) {
111
+ Data_Get_Struct(obj, gsl_complex, z);
112
+ znew = xmalloc(sizeof(gsl_complex));
113
+ *znew = (*func)(*z);
114
+ return Data_Wrap_Struct(cgsl_complex, 0, free, znew);
115
+ } else if (VECTOR_COMPLEX_P(obj)) {
116
+ Data_Get_Struct(obj, gsl_vector_complex, v);
117
+ vnew = gsl_vector_complex_alloc(v->size);
118
+ for (i = 0; i < v->size; i++) {
119
+ z = GSL_COMPLEX_AT(v, i);
120
+ gsl_vector_complex_set(vnew, i, (*func)(*z));
121
+ }
122
+ return Data_Wrap_Struct(cgsl_vector_complex, 0, gsl_vector_complex_free, vnew);
123
+ } else if (MATRIX_COMPLEX_P(obj)) {
124
+ Data_Get_Struct(obj, gsl_matrix_complex, m);
125
+ mnew = gsl_matrix_complex_alloc(m->size1, m->size2);
126
+ for (i = 0; i < m->size1; i++) {
127
+ for (j = 0; j < m->size2; j++) {
128
+ gsl_matrix_complex_set(mnew, i, j, (*func)(gsl_matrix_complex_get(m, i, j)));
129
+ }
130
+ }
131
+ return Data_Wrap_Struct(cgsl_matrix_complex, 0, gsl_matrix_complex_free, mnew);
132
+ } else {
133
+ rb_raise(rb_eTypeError,
134
+ "wrong argument type %s "
135
+ " (GSL::Complex or GSL::Vector::Complex expected)",
136
+ rb_class2name(CLASS_OF(obj)));
137
+ }
138
+ }
139
+
140
+ static VALUE rb_gsl_math_eval(double (*func)(const double), VALUE xx)
141
+ {
142
+ VALUE x, ary;
143
+ size_t i, size;
144
+ #ifdef HAVE_NARRAY_H
145
+ struct NARRAY *na;
146
+ double *ptr1, *ptr2;
147
+ #endif
148
+ if (CLASS_OF(xx) == rb_cRange) xx = rb_gsl_range2ary(xx);
149
+ switch (TYPE(xx)) {
150
+ case T_FIXNUM:
151
+ case T_BIGNUM:
152
+ case T_FLOAT:
153
+ return rb_float_new((*func)(NUM2DBL(xx)));
154
+ break;
155
+ case T_ARRAY:
156
+ size = RARRAY(xx)->len;
157
+ ary = rb_ary_new2(size);
158
+ for (i = 0; i < size; i++) {
159
+ x = rb_ary_entry(xx, i);
160
+ Need_Float(x);
161
+ // rb_ary_store(ary, i, rb_float_new((*func)(RFLOAT(x)->value)));
162
+ rb_ary_store(ary, i, rb_float_new((*func)(NUM2DBL(x))));
163
+ }
164
+ return ary;
165
+ break;
166
+ default:
167
+ #ifdef HAVE_NARRAY_H
168
+ if (NA_IsNArray(xx)) {
169
+ GetNArray(xx, na);
170
+ ptr1 = (double*) na->ptr;
171
+ size = na->total;
172
+ ary = na_make_object(NA_DFLOAT, na->rank, na->shape, CLASS_OF(xx));
173
+ ptr2 = NA_PTR_TYPE(ary, double*);
174
+ for (i = 0; i < size; i++) ptr2[i] = (*func)(ptr1[i]);
175
+ return ary;
176
+ }
177
+ #endif
178
+ if (VECTOR_P(xx)) {
179
+ return vector_eval_create(xx, func);
180
+ } else if (MATRIX_P(xx)) {
181
+ return matrix_eval_create(xx, func);
182
+ } else {
183
+ rb_raise(rb_eTypeError, "wrong argument type %s (Array or Vector or Matrix expected)", rb_class2name(CLASS_OF(xx)));
184
+ }
185
+ break;
186
+ }
187
+ /* never reach here */
188
+ return Qnil;
189
+ }
190
+
191
+ static VALUE rb_gsl_math_eval2(double (*func)(const double, const double), VALUE xx,
192
+ VALUE yy);
193
+ static VALUE rb_gsl_math_eval2(double (*func)(const double, const double), VALUE xx,
194
+ VALUE yy)
195
+ {
196
+ VALUE x, y, ary;
197
+ size_t i, j, size;
198
+ gsl_vector *v = NULL, *v2 = NULL, *vnew = NULL;
199
+ gsl_matrix *m = NULL, *m2 = NULL, *mnew = NULL;
200
+ #ifdef HAVE_NARRAY_H
201
+ struct NARRAY *nax, *nay;
202
+ double *ptr1, *ptr2, *ptr3;
203
+ #endif
204
+ if (CLASS_OF(xx) == rb_cRange) xx = rb_gsl_range2ary(xx);
205
+ switch (TYPE(xx)) {
206
+ case T_FIXNUM:
207
+ case T_BIGNUM:
208
+ case T_FLOAT:
209
+ Need_Float(yy);
210
+ return rb_float_new((*func)(NUM2DBL(xx), NUM2DBL(yy)));
211
+ break;
212
+ case T_ARRAY:
213
+ Check_Type(yy, T_ARRAY);
214
+ size = RARRAY(xx)->len;
215
+ if (size != RARRAY(yy)->len) rb_raise(rb_eRuntimeError, "array sizes are different.");
216
+ ary = rb_ary_new2(size);
217
+ for (i = 0; i < size; i++) {
218
+ x = rb_ary_entry(xx, i);
219
+ y = rb_ary_entry(yy, i);
220
+ Need_Float(x); Need_Float(y);
221
+ // rb_ary_store(ary, i, rb_float_new((*func)(RFLOAT(x)->value, RFLOAT(y)->value)));
222
+ rb_ary_store(ary, i, rb_float_new((*func)(NUM2DBL(x), NUM2DBL(y))));
223
+ }
224
+ return ary;
225
+ break;
226
+ default:
227
+ #ifdef HAVE_NARRAY_H
228
+ if (NA_IsNArray(xx)) {
229
+ GetNArray(xx, nax);
230
+ GetNArray(yy, nay);
231
+ ptr1 = (double*) nax->ptr;
232
+ ptr2 = (double*) nay->ptr;
233
+ size = nax->total;
234
+ ary = na_make_object(NA_DFLOAT, nax->rank, nax->shape, CLASS_OF(xx));
235
+ ptr3 = NA_PTR_TYPE(ary, double*);
236
+ for (i = 0; i < size; i++) ptr3[i] = (*func)(ptr1[i], ptr2[i]);
237
+ return ary;
238
+ }
239
+ #endif
240
+ if (VECTOR_P(xx)) {
241
+ CHECK_VECTOR(yy);
242
+ Data_Get_Struct(xx, gsl_vector, v);
243
+ Data_Get_Struct(yy, gsl_vector, v2);
244
+ vnew = gsl_vector_alloc(v->size);
245
+ for (i = 0; i < v->size; i++) {
246
+ gsl_vector_set(vnew, i, (*func)(gsl_vector_get(v, i), gsl_vector_get(v2, i)));
247
+ }
248
+ return Data_Wrap_Struct(cgsl_vector, 0, gsl_vector_free, vnew);
249
+ } else if (MATRIX_P(xx)) {
250
+ CHECK_MATRIX(yy);
251
+ Data_Get_Struct(xx, gsl_matrix, m);
252
+ Data_Get_Struct(yy, gsl_matrix, m2);
253
+ mnew = gsl_matrix_alloc(m->size1, m->size2);
254
+ for (i = 0; i < m->size1; i++) {
255
+ for (j = 0; j < m->size2; j++) {
256
+ gsl_matrix_set(mnew, i, j, (*func)(gsl_matrix_get(m, i, j), gsl_matrix_get(m2, i, j)));
257
+ }
258
+ }
259
+ return Data_Wrap_Struct(cgsl_matrix, 0, gsl_matrix_free, mnew);
260
+ } else {
261
+ rb_raise(rb_eTypeError,
262
+ "wrong argument type %s "
263
+ "(Array or Vector or Matrix expected)", rb_class2name(CLASS_OF(xx)));
264
+ }
265
+ break;
266
+ }
267
+ /* never reach here */
268
+ return Qnil;
269
+ }
270
+
271
+ static VALUE rb_gsl_sqrt(VALUE obj, VALUE x)
272
+ {
273
+ return rb_gsl_math_eval(sqrt, x);
274
+ }
275
+
276
+ static VALUE rb_gsl_log1p(VALUE obj, VALUE x)
277
+ {
278
+ return rb_gsl_math_eval(gsl_log1p, x);
279
+ }
280
+
281
+ static VALUE rb_gsl_expm1(VALUE obj, VALUE x)
282
+ {
283
+ return rb_gsl_math_eval(gsl_expm1, x);
284
+ }
285
+
286
+ static VALUE rb_gsl_hypot(VALUE obj, VALUE x, VALUE y)
287
+ {
288
+ return rb_gsl_math_eval2(gsl_hypot, x, y);
289
+ }
290
+ #ifdef GSL_1_10_LATER
291
+ static VALUE rb_gsl_hypot3(VALUE obj, VALUE x, VALUE y, VALUE z)
292
+ {
293
+ Need_Float(x);
294
+ Need_Float(y);
295
+ Need_Float(z);
296
+ return rb_float_new(gsl_hypot3(NUM2DBL(x), NUM2DBL(y), NUM2DBL(z)));
297
+ }
298
+ #endif
299
+ static VALUE rb_gsl_acosh(VALUE obj, VALUE x)
300
+ {
301
+ if (COMPLEX_P(x) || VECTOR_COMPLEX_P(x) || MATRIX_COMPLEX_P(x))
302
+ return rb_gsl_math_complex_eval(gsl_complex_arccosh, x);
303
+ return rb_gsl_math_eval(gsl_acosh, x);
304
+ }
305
+
306
+ static VALUE rb_gsl_asinh(VALUE obj, VALUE x)
307
+ {
308
+ if (COMPLEX_P(x) || VECTOR_COMPLEX_P(x) || MATRIX_COMPLEX_P(x))
309
+ return rb_gsl_math_complex_eval(gsl_complex_arcsinh, x);
310
+ return rb_gsl_math_eval(gsl_asinh, x);
311
+ }
312
+
313
+ static VALUE rb_gsl_atanh(VALUE obj, VALUE x)
314
+ {
315
+ if (COMPLEX_P(x) || VECTOR_COMPLEX_P(x) || MATRIX_COMPLEX_P(x))
316
+ return rb_gsl_math_complex_eval(gsl_complex_arctanh, x);
317
+ return rb_gsl_math_eval(gsl_atanh, x);
318
+ }
319
+
320
+ #include <math.h>
321
+ /* xx: Numeric, Complex, Vector, Matrix
322
+ nn: Numeric, Complex
323
+ */
324
+ VALUE rb_gsl_pow(VALUE obj, VALUE xx, VALUE nn)
325
+ {
326
+ VALUE x, ary, argv[2];
327
+ size_t i, j, size;
328
+ double n;
329
+ gsl_vector *v = NULL, *vnew = NULL;
330
+ gsl_matrix *m = NULL, *mnew = NULL;
331
+ #ifdef HAVE_NARRAY_H
332
+ struct NARRAY *na;
333
+ double *ptr1, *ptr2;
334
+ #endif
335
+ if (CLASS_OF(xx) == rb_cRange) xx = rb_gsl_range2ary(xx);
336
+ switch (TYPE(xx)) {
337
+ case T_FIXNUM:
338
+ case T_BIGNUM:
339
+ case T_FLOAT:
340
+ return rb_float_new(pow(NUM2DBL(xx), NUM2DBL(nn)));
341
+ break;
342
+ case T_ARRAY:
343
+ n = NUM2DBL(nn);
344
+ size = RARRAY(xx)->len;
345
+ ary = rb_ary_new2(size);
346
+ for (i = 0; i < size; i++) {
347
+ x = rb_ary_entry(xx, i);
348
+ Need_Float(x);
349
+ rb_ary_store(ary, i, rb_float_new(pow(NUM2DBL(x), n)));
350
+ }
351
+ return ary;
352
+ break;
353
+ default:
354
+ #ifdef HAVE_NARRAY_H
355
+ if (NA_IsNArray(xx)) {
356
+ n = NUM2DBL(nn);
357
+ GetNArray(xx, na);
358
+ ptr1 = (double*) na->ptr;
359
+ size = na->total;
360
+ ary = na_make_object(NA_DFLOAT, na->rank, na->shape, CLASS_OF(xx));
361
+ ptr2 = NA_PTR_TYPE(ary, double*);
362
+ for (i = 0; i < size; i++) ptr2[i] = pow(ptr1[i], n);
363
+ return ary;
364
+ }
365
+ #endif
366
+ if (VECTOR_P(xx)) {
367
+ n = NUM2DBL(nn);
368
+ Data_Get_Struct(xx, gsl_vector, v);
369
+ vnew = gsl_vector_alloc(v->size);
370
+ for (i = 0; i < v->size; i++) {
371
+ gsl_vector_set(vnew, i, pow(gsl_vector_get(v, i), n));
372
+ }
373
+ return Data_Wrap_Struct(cgsl_vector, 0, gsl_vector_free, vnew);
374
+ }
375
+ if (MATRIX_P(xx)) {
376
+ n = NUM2DBL(nn);
377
+ Data_Get_Struct(xx, gsl_matrix, m);
378
+ mnew = gsl_matrix_alloc(m->size1, m->size2);
379
+ for (i = 0; i < m->size1; i++) {
380
+ for (j = 0; j < m->size2; j++) {
381
+ gsl_matrix_set(mnew, i, j, pow(gsl_matrix_get(m, i, j), n));
382
+ }
383
+ }
384
+ return Data_Wrap_Struct(cgsl_matrix, 0, gsl_matrix_free, mnew);
385
+ }
386
+ if (COMPLEX_P(xx) || VECTOR_COMPLEX_P(xx) || MATRIX_COMPLEX_P(xx)) {
387
+ argv[0] = xx;
388
+ argv[1] = nn;
389
+ return rb_gsl_complex_pow(2, argv, obj);
390
+ }
391
+ rb_raise(rb_eTypeError, "wrong argument type %s (Array or Vector or Matrix expected)", rb_class2name(CLASS_OF(xx)));
392
+ break;
393
+ }
394
+ /* never reach here */
395
+ return Qnil;
396
+ }
397
+
398
+ static VALUE rb_gsl_pow_int(VALUE obj, VALUE xx, VALUE nn)
399
+ {
400
+ VALUE x, ary, argv[2];
401
+ size_t i, j, size;
402
+ int n;
403
+ gsl_vector *v = NULL, *vnew = NULL;
404
+ gsl_matrix *m = NULL, *mnew = NULL;
405
+ #ifdef HAVE_NARRAY_H
406
+ struct NARRAY *na;
407
+ double *ptr1, *ptr2;
408
+ #endif
409
+
410
+ if (CLASS_OF(xx) == rb_cRange) xx = rb_gsl_range2ary(xx);
411
+ switch (TYPE(xx)) {
412
+ case T_FIXNUM:
413
+ case T_BIGNUM:
414
+ case T_FLOAT:
415
+ return rb_float_new(gsl_pow_int(NUM2DBL(xx), FIX2INT(nn)));
416
+ break;
417
+ case T_ARRAY:
418
+ CHECK_FIXNUM(nn);
419
+ n = FIX2INT(nn);
420
+ size = RARRAY(xx)->len;
421
+ ary = rb_ary_new2(size);
422
+ for (i = 0; i < size; i++) {
423
+ x = rb_ary_entry(xx, i);
424
+ Need_Float(x);
425
+ // rb_ary_store(ary, i, rb_float_new(gsl_pow_int(RFLOAT(x)->value, n)));
426
+ rb_ary_store(ary, i, rb_float_new(gsl_pow_int(NUM2DBL(x), n)));
427
+ }
428
+ return ary;
429
+ break;
430
+ default:
431
+ #ifdef HAVE_NARRAY_H
432
+ if (NA_IsNArray(xx)) {
433
+ CHECK_FIXNUM(nn);
434
+ n = FIX2INT(nn);
435
+ GetNArray(xx, na);
436
+ ptr1 = (double*) na->ptr;
437
+ size = na->total;
438
+ ary = na_make_object(NA_DFLOAT, na->rank, na->shape, CLASS_OF(xx));
439
+ ptr2 = NA_PTR_TYPE(ary, double*);
440
+ for (i = 0; i < size; i++) ptr2[i] = gsl_pow_int(ptr1[i], n);
441
+ return ary;
442
+ }
443
+ #endif
444
+ if (VECTOR_P(xx)) {
445
+ CHECK_FIXNUM(nn);
446
+ n = FIX2INT(nn);
447
+ Data_Get_Struct(xx, gsl_vector, v);
448
+ vnew = gsl_vector_alloc(v->size);
449
+ for (i = 0; i < v->size; i++) {
450
+ gsl_vector_set(vnew, i, gsl_pow_int(gsl_vector_get(v, i), n));
451
+ }
452
+ return Data_Wrap_Struct(cgsl_vector, 0, gsl_vector_free, vnew);
453
+ } else if (MATRIX_P(xx)) {
454
+ CHECK_FIXNUM(nn);
455
+ n = FIX2INT(nn);
456
+ Data_Get_Struct(xx, gsl_matrix, m);
457
+ mnew = gsl_matrix_alloc(m->size1, m->size2);
458
+ for (i = 0; i < m->size1; i++) {
459
+ for (j = 0; j < m->size2; j++) {
460
+ gsl_matrix_set(mnew, i, j, gsl_pow_int(gsl_matrix_get(m, i, j), n));
461
+ }
462
+ }
463
+ return Data_Wrap_Struct(cgsl_matrix, 0, gsl_matrix_free, mnew);
464
+ } else if (COMPLEX_P(xx) || VECTOR_COMPLEX_P(xx) || MATRIX_COMPLEX_P(xx)) {
465
+ argv[0] = xx;
466
+ argv[1] = nn;
467
+ return rb_gsl_complex_pow_real(2, argv, obj);
468
+ } else {
469
+ rb_raise(rb_eTypeError, "wrong argument type %s (Array or Vector or Matrix expected)", rb_class2name(CLASS_OF(xx)));
470
+ }
471
+ break;
472
+ }
473
+ /* never reach here */
474
+ return Qnil;
475
+ }
476
+
477
+ static VALUE rb_gsl_pow_2(VALUE obj, VALUE x)
478
+ {
479
+ return rb_gsl_math_eval(gsl_pow_2, x);
480
+ }
481
+
482
+ static VALUE rb_gsl_pow_3(VALUE obj, VALUE x)
483
+ {
484
+ return rb_gsl_math_eval(gsl_pow_3, x);
485
+ }
486
+
487
+ static VALUE rb_gsl_pow_4(VALUE obj, VALUE x)
488
+ {
489
+ return rb_gsl_math_eval(gsl_pow_4, x);
490
+ }
491
+
492
+ static VALUE rb_gsl_pow_5(VALUE obj, VALUE x)
493
+ {
494
+ return rb_gsl_math_eval(gsl_pow_5, x);
495
+ }
496
+
497
+ static VALUE rb_gsl_pow_6(VALUE obj, VALUE x)
498
+ {
499
+ return rb_gsl_math_eval(gsl_pow_6, x);
500
+ }
501
+
502
+ static VALUE rb_gsl_pow_7(VALUE obj, VALUE x)
503
+ {
504
+ return rb_gsl_math_eval(gsl_pow_7, x);
505
+ }
506
+
507
+ static VALUE rb_gsl_pow_8(VALUE obj, VALUE x)
508
+ {
509
+ return rb_gsl_math_eval(gsl_pow_8, x);
510
+ }
511
+
512
+ static VALUE rb_gsl_pow_9(VALUE obj, VALUE x)
513
+ {
514
+ return rb_gsl_math_eval(gsl_pow_9, x);
515
+ }
516
+
517
+ /*****/
518
+ static VALUE rb_GSL_SIGN(VALUE obj, VALUE x)
519
+ {
520
+ return INT2FIX(GSL_SIGN(NUM2DBL(x)));
521
+ }
522
+
523
+ static VALUE rb_GSL_IS_ODD(VALUE obj, VALUE n)
524
+ {
525
+ CHECK_FIXNUM(n);
526
+ return INT2FIX(GSL_IS_ODD(FIX2INT(n)));
527
+ }
528
+
529
+ static VALUE rb_GSL_IS_ODD2(VALUE obj, VALUE n)
530
+ {
531
+ CHECK_FIXNUM(n);
532
+ if (GSL_IS_ODD(FIX2INT(n))) return Qtrue;
533
+ else return Qfalse;
534
+ }
535
+
536
+ static VALUE rb_GSL_IS_EVEN(VALUE obj, VALUE n)
537
+ {
538
+ CHECK_FIXNUM(n);
539
+ return INT2FIX(GSL_IS_EVEN(FIX2INT(n)));
540
+ }
541
+
542
+ static VALUE rb_GSL_IS_EVEN2(VALUE obj, VALUE n)
543
+ {
544
+ CHECK_FIXNUM(n);
545
+ if (GSL_IS_EVEN(FIX2INT(n))) return Qtrue;
546
+ else return Qfalse;
547
+ }
548
+
549
+ static VALUE rb_GSL_MAX(VALUE obj, VALUE aa, VALUE bb)
550
+ {
551
+ double a, b;
552
+ double max;
553
+ /* Need_Float(aa); Need_Float(bb);*/
554
+ a = NUM2DBL(aa);
555
+ b = NUM2DBL(bb);
556
+ max = GSL_MAX_DBL(a, b);
557
+ if (gsl_fcmp(max, a, 1.0e-10) == 0) return aa;
558
+ else return bb;
559
+
560
+ }
561
+
562
+ static VALUE rb_GSL_MIN(VALUE obj, VALUE aa, VALUE bb)
563
+ {
564
+ double a, b;
565
+ double min;
566
+ /* Need_Float(aa); Need_Float(bb);*/
567
+ a = NUM2DBL(aa);
568
+ b = NUM2DBL(bb);
569
+ min = GSL_MIN_DBL(a, b);
570
+ if (gsl_fcmp(min, a, 1.0e-10) == 0) return aa;
571
+ else return bb;
572
+ }
573
+
574
+ static VALUE rb_GSL_MAX_DBL(VALUE obj, VALUE aa, VALUE bb)
575
+ {
576
+ Need_Float(aa); Need_Float(bb);
577
+ return rb_float_new(GSL_MAX_DBL(NUM2DBL(aa), NUM2DBL(bb)));
578
+ }
579
+
580
+ static VALUE rb_GSL_MIN_DBL(VALUE obj, VALUE aa, VALUE bb)
581
+ {
582
+ Need_Float(aa); Need_Float(bb);
583
+ return rb_float_new(GSL_MAX_DBL(NUM2DBL(aa), NUM2DBL(bb)));
584
+ }
585
+
586
+ static VALUE rb_GSL_MAX_INT(VALUE obj, VALUE aa, VALUE bb)
587
+ {
588
+ if (TYPE(aa) != T_FIXNUM || TYPE(bb) != T_FIXNUM)
589
+ return rb_GSL_MAX(obj, aa, bb);
590
+ else
591
+ return INT2FIX(GSL_MAX_INT(FIX2INT(aa), FIX2INT(bb)));
592
+ }
593
+
594
+ static VALUE rb_GSL_MIN_INT(VALUE obj, VALUE aa, VALUE bb)
595
+ {
596
+ if (TYPE(aa) != T_FIXNUM || TYPE(bb) != T_FIXNUM)
597
+ return rb_GSL_MIN(obj, aa, bb);
598
+ return
599
+ INT2FIX(GSL_MIN_INT(FIX2INT(aa), FIX2INT(bb)));
600
+ }
601
+
602
+ #ifdef GSL_1_3_LATER
603
+ static VALUE rb_gsl_ldexp(VALUE obj, VALUE x, VALUE e)
604
+ {
605
+ return rb_float_new(gsl_ldexp(NUM2DBL(x), FIX2INT(e)));
606
+ }
607
+
608
+ static VALUE rb_gsl_frexp(VALUE obj, VALUE x)
609
+ {
610
+ int e;
611
+ double val;
612
+ Need_Float(x);
613
+ val = gsl_frexp(NUM2DBL(x), &e);
614
+ return rb_ary_new3(2, rb_float_new(val), INT2FIX(e));
615
+ }
616
+ #endif
617
+
618
+ static VALUE rb_gsl_fcmp(int argc, VALUE *argv, VALUE obj)
619
+ {
620
+ double a, b, epsilon = 1e-10;
621
+ switch (argc) {
622
+ case 3:
623
+ epsilon = NUM2DBL(argv[2]);
624
+ /* no break, do next */
625
+ case 2:
626
+ a = NUM2DBL(argv[0]);
627
+ b = NUM2DBL(argv[1]);
628
+ break;
629
+ default:
630
+ rb_raise(rb_eArgError, "wrong number of arguments");
631
+ break;
632
+ }
633
+ return INT2FIX(gsl_fcmp(a, b, epsilon));
634
+ }
635
+
636
+ static VALUE rb_gsl_equal(int argc, VALUE *argv, VALUE obj)
637
+ {
638
+ double a, b, epsilon = 1e-10;
639
+ int retval;
640
+
641
+ switch (argc) {
642
+ case 3:
643
+ epsilon = NUM2DBL(argv[2]);
644
+ /* no break, do next */
645
+ case 2:
646
+ a = NUM2DBL(argv[0]);
647
+ b = NUM2DBL(argv[1]);
648
+ break;
649
+ default:
650
+ rb_raise(rb_eArgError, "wrong number of arguments");
651
+ break;
652
+ }
653
+ retval = gsl_fcmp(a, b, epsilon);
654
+ if (retval == 0) return Qtrue;
655
+ else return Qfalse;
656
+ }
657
+
658
+ void Init_gsl_math(VALUE module)
659
+ {
660
+ rb_gsl_define_const(module);
661
+
662
+ rb_define_module_function(module, "posinf", rb_GSL_POSINF, 0);
663
+ rb_define_module_function(module, "neginf", rb_GSL_NEGINF, 0);
664
+ rb_define_module_function(module, "nan", rb_GSL_NAN, 0);
665
+ rb_define_module_function(module, "isnan", rb_gsl_isnan, 1);
666
+ rb_define_module_function(module, "isnan?", rb_gsl_isnan2, 1);
667
+ rb_define_module_function(module, "isinf", rb_gsl_isinf, 1);
668
+ rb_define_module_function(module, "isinf?", rb_gsl_isinf2, 1);
669
+ rb_define_module_function(module, "finite", rb_gsl_finite, 1);
670
+ rb_define_module_function(module, "finite?", rb_gsl_finite2, 1);
671
+
672
+ rb_define_module_function(module, "sqrt", rb_gsl_sqrt, 1);
673
+ rb_define_module_function(module, "log1p", rb_gsl_log1p, 1);
674
+ rb_define_module_function(module, "expm1", rb_gsl_expm1, 1);
675
+ rb_define_module_function(module, "hypot", rb_gsl_hypot, 2);
676
+ #ifdef GSL_1_10_LATER
677
+ rb_define_module_function(module, "hypot3", rb_gsl_hypot3, 3);
678
+ #endif
679
+ rb_define_module_function(module, "acosh", rb_gsl_acosh, 1);
680
+ rb_define_module_function(module, "asinh", rb_gsl_asinh, 1);
681
+ rb_define_module_function(module, "atanh", rb_gsl_atanh, 1);
682
+ rb_define_module_function(module, "pow", rb_gsl_pow, 2);
683
+ rb_define_module_function(module, "pow_int", rb_gsl_pow_int, 2);
684
+ rb_define_module_function(module, "pow_2", rb_gsl_pow_2, 1);
685
+ rb_define_module_function(module, "pow_3", rb_gsl_pow_3, 1);
686
+ rb_define_module_function(module, "pow_4", rb_gsl_pow_4, 1);
687
+ rb_define_module_function(module, "pow_5", rb_gsl_pow_5, 1);
688
+ rb_define_module_function(module, "pow_6", rb_gsl_pow_6, 1);
689
+ rb_define_module_function(module, "pow_7", rb_gsl_pow_7, 1);
690
+ rb_define_module_function(module, "pow_8", rb_gsl_pow_8, 1);
691
+ rb_define_module_function(module, "pow_9", rb_gsl_pow_9, 1);
692
+ rb_define_module_function(module, "sign", rb_GSL_SIGN, 1);
693
+ rb_define_module_function(module, "SIGN", rb_GSL_SIGN, 1);
694
+
695
+ rb_define_module_function(module, "is_odd", rb_GSL_IS_ODD, 1);
696
+ rb_define_module_function(module, "IS_ODD", rb_GSL_IS_ODD, 1);
697
+ rb_define_module_function(module, "is_odd?", rb_GSL_IS_ODD2, 1);
698
+ rb_define_module_function(module, "IS_ODD?", rb_GSL_IS_ODD2, 1);
699
+ rb_define_module_function(module, "is_even", rb_GSL_IS_EVEN, 1);
700
+ rb_define_module_function(module, "IS_EVEN", rb_GSL_IS_EVEN, 1);
701
+ rb_define_module_function(module, "is_even?", rb_GSL_IS_EVEN2, 1);
702
+ rb_define_module_function(module, "IS_EVEN?", rb_GSL_IS_EVEN2, 1);
703
+ rb_define_module_function(module, "max", rb_GSL_MAX, 2);
704
+ rb_define_module_function(module, "MAX", rb_GSL_MAX, 2);
705
+ rb_define_module_function(module, "min", rb_GSL_MIN, 2);
706
+ rb_define_module_function(module, "MIN", rb_GSL_MIN, 2);
707
+
708
+ rb_define_module_function(module, "max_dbl", rb_GSL_MAX_DBL, 2);
709
+ rb_define_module_function(module, "MAX_DBL", rb_GSL_MAX_DBL, 2);
710
+ rb_define_module_function(module, "min_dbl", rb_GSL_MIN_DBL, 2);
711
+ rb_define_module_function(module, "MIN_DBL", rb_GSL_MIN_DBL, 2);
712
+
713
+ rb_define_module_function(module, "max_int", rb_GSL_MAX_INT, 2);
714
+ rb_define_module_function(module, "MAX_INT", rb_GSL_MAX_INT, 2);
715
+ rb_define_module_function(module, "min_int", rb_GSL_MIN_INT, 2);
716
+ rb_define_module_function(module, "MIN_INT", rb_GSL_MIN_INT, 2);
717
+
718
+ rb_define_module_function(module, "fcmp", rb_gsl_fcmp, -1);
719
+ rb_define_singleton_method(module, "equal?", rb_gsl_equal, -1);
720
+ #ifdef GSL_1_3_LATER
721
+ rb_define_module_function(module, "ldexp", rb_gsl_ldexp, 2);
722
+ rb_define_module_function(module, "frexp", rb_gsl_frexp, 1);
723
+ #endif
724
+ }