gsl 1.12.108

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (158) hide show
  1. data/README.rdoc +29 -0
  2. data/Rakefile +54 -0
  3. data/VERSION +2 -0
  4. data/ext/MANIFEST +119 -0
  5. data/ext/alf.c +206 -0
  6. data/ext/array.c +666 -0
  7. data/ext/array_complex.c +247 -0
  8. data/ext/blas.c +29 -0
  9. data/ext/blas1.c +731 -0
  10. data/ext/blas2.c +1093 -0
  11. data/ext/blas3.c +881 -0
  12. data/ext/block.c +44 -0
  13. data/ext/block_source.c +886 -0
  14. data/ext/bspline.c +130 -0
  15. data/ext/bundle.c +3 -0
  16. data/ext/cdf.c +754 -0
  17. data/ext/cheb.c +542 -0
  18. data/ext/combination.c +283 -0
  19. data/ext/common.c +325 -0
  20. data/ext/complex.c +1004 -0
  21. data/ext/const.c +673 -0
  22. data/ext/const_additional.c +120 -0
  23. data/ext/cqp.c +283 -0
  24. data/ext/deriv.c +195 -0
  25. data/ext/dht.c +361 -0
  26. data/ext/diff.c +166 -0
  27. data/ext/dirac.c +395 -0
  28. data/ext/eigen.c +2373 -0
  29. data/ext/error.c +194 -0
  30. data/ext/extconf.rb +281 -0
  31. data/ext/fcmp.c +66 -0
  32. data/ext/fft.c +1092 -0
  33. data/ext/fit.c +205 -0
  34. data/ext/fresnel.c +312 -0
  35. data/ext/function.c +524 -0
  36. data/ext/geometry.c +139 -0
  37. data/ext/graph.c +1638 -0
  38. data/ext/gsl.c +271 -0
  39. data/ext/gsl_narray.c +653 -0
  40. data/ext/histogram.c +1995 -0
  41. data/ext/histogram2d.c +1068 -0
  42. data/ext/histogram3d.c +884 -0
  43. data/ext/histogram3d_source.c +750 -0
  44. data/ext/histogram_find.c +101 -0
  45. data/ext/histogram_oper.c +159 -0
  46. data/ext/ieee.c +98 -0
  47. data/ext/integration.c +1138 -0
  48. data/ext/interp.c +512 -0
  49. data/ext/jacobi.c +739 -0
  50. data/ext/linalg.c +4047 -0
  51. data/ext/linalg_complex.c +741 -0
  52. data/ext/math.c +725 -0
  53. data/ext/matrix.c +39 -0
  54. data/ext/matrix_complex.c +1732 -0
  55. data/ext/matrix_double.c +560 -0
  56. data/ext/matrix_int.c +256 -0
  57. data/ext/matrix_source.c +2733 -0
  58. data/ext/min.c +250 -0
  59. data/ext/monte.c +992 -0
  60. data/ext/multifit.c +1879 -0
  61. data/ext/multimin.c +808 -0
  62. data/ext/multimin_fsdf.c +156 -0
  63. data/ext/multiroots.c +955 -0
  64. data/ext/ndlinear.c +321 -0
  65. data/ext/nmf.c +167 -0
  66. data/ext/nmf_wrap.c +72 -0
  67. data/ext/ntuple.c +469 -0
  68. data/ext/odeiv.c +959 -0
  69. data/ext/ool.c +879 -0
  70. data/ext/oper_complex_source.c +253 -0
  71. data/ext/permutation.c +596 -0
  72. data/ext/poly.c +42 -0
  73. data/ext/poly2.c +265 -0
  74. data/ext/poly_source.c +1885 -0
  75. data/ext/qrng.c +171 -0
  76. data/ext/randist.c +1873 -0
  77. data/ext/rational.c +480 -0
  78. data/ext/rng.c +612 -0
  79. data/ext/root.c +408 -0
  80. data/ext/sf.c +1494 -0
  81. data/ext/sf_airy.c +200 -0
  82. data/ext/sf_bessel.c +867 -0
  83. data/ext/sf_clausen.c +28 -0
  84. data/ext/sf_coulomb.c +206 -0
  85. data/ext/sf_coupling.c +118 -0
  86. data/ext/sf_dawson.c +29 -0
  87. data/ext/sf_debye.c +157 -0
  88. data/ext/sf_dilog.c +42 -0
  89. data/ext/sf_elementary.c +44 -0
  90. data/ext/sf_ellint.c +206 -0
  91. data/ext/sf_elljac.c +29 -0
  92. data/ext/sf_erfc.c +93 -0
  93. data/ext/sf_exp.c +164 -0
  94. data/ext/sf_expint.c +211 -0
  95. data/ext/sf_fermi_dirac.c +148 -0
  96. data/ext/sf_gamma.c +344 -0
  97. data/ext/sf_gegenbauer.c +96 -0
  98. data/ext/sf_hyperg.c +197 -0
  99. data/ext/sf_laguerre.c +112 -0
  100. data/ext/sf_lambert.c +47 -0
  101. data/ext/sf_legendre.c +367 -0
  102. data/ext/sf_log.c +104 -0
  103. data/ext/sf_mathieu.c +238 -0
  104. data/ext/sf_power.c +46 -0
  105. data/ext/sf_psi.c +98 -0
  106. data/ext/sf_synchrotron.c +48 -0
  107. data/ext/sf_transport.c +76 -0
  108. data/ext/sf_trigonometric.c +207 -0
  109. data/ext/sf_zeta.c +119 -0
  110. data/ext/signal.c +310 -0
  111. data/ext/siman.c +718 -0
  112. data/ext/sort.c +208 -0
  113. data/ext/spline.c +395 -0
  114. data/ext/stats.c +799 -0
  115. data/ext/sum.c +168 -0
  116. data/ext/tamu_anova.c +56 -0
  117. data/ext/tensor.c +38 -0
  118. data/ext/tensor_source.c +1123 -0
  119. data/ext/vector.c +38 -0
  120. data/ext/vector_complex.c +2236 -0
  121. data/ext/vector_double.c +1433 -0
  122. data/ext/vector_int.c +204 -0
  123. data/ext/vector_source.c +3329 -0
  124. data/ext/wavelet.c +937 -0
  125. data/include/rb_gsl.h +151 -0
  126. data/include/rb_gsl_array.h +238 -0
  127. data/include/rb_gsl_cheb.h +21 -0
  128. data/include/rb_gsl_common.h +343 -0
  129. data/include/rb_gsl_complex.h +25 -0
  130. data/include/rb_gsl_const.h +29 -0
  131. data/include/rb_gsl_dirac.h +13 -0
  132. data/include/rb_gsl_eigen.h +17 -0
  133. data/include/rb_gsl_fft.h +62 -0
  134. data/include/rb_gsl_fit.h +25 -0
  135. data/include/rb_gsl_function.h +27 -0
  136. data/include/rb_gsl_graph.h +70 -0
  137. data/include/rb_gsl_histogram.h +63 -0
  138. data/include/rb_gsl_histogram3d.h +97 -0
  139. data/include/rb_gsl_integration.h +17 -0
  140. data/include/rb_gsl_interp.h +46 -0
  141. data/include/rb_gsl_linalg.h +25 -0
  142. data/include/rb_gsl_math.h +26 -0
  143. data/include/rb_gsl_odeiv.h +21 -0
  144. data/include/rb_gsl_poly.h +71 -0
  145. data/include/rb_gsl_rational.h +37 -0
  146. data/include/rb_gsl_rng.h +21 -0
  147. data/include/rb_gsl_root.h +22 -0
  148. data/include/rb_gsl_sf.h +119 -0
  149. data/include/rb_gsl_statistics.h +17 -0
  150. data/include/rb_gsl_tensor.h +45 -0
  151. data/include/rb_gsl_with_narray.h +22 -0
  152. data/include/templates_off.h +87 -0
  153. data/include/templates_on.h +241 -0
  154. data/lib/gsl/gnuplot.rb +41 -0
  155. data/lib/gsl/oper.rb +68 -0
  156. data/lib/ool.rb +22 -0
  157. data/lib/ool/conmin.rb +30 -0
  158. metadata +224 -0
@@ -0,0 +1,560 @@
1
+ /*
2
+ matrix_double.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.
10
+ */
11
+ #include "rb_gsl_config.h"
12
+ #include "rb_gsl_array.h"
13
+ #include "rb_gsl_complex.h"
14
+ #ifdef HAVE_NARRAY_H
15
+ #include "rb_gsl_with_narray.h"
16
+ #endif
17
+
18
+ enum {
19
+ GSL_MATRIX_ADD,
20
+ GSL_MATRIX_SUB,
21
+ GSL_MATRIX_MUL,
22
+ GSL_MATRIX_DIV,
23
+ };
24
+
25
+ static VALUE rb_gsl_matrix_arithmetics(int flag, VALUE obj, VALUE bb);
26
+ VALUE rb_gsl_linalg_LU_solve(int argc, VALUE *argv, VALUE obj);
27
+
28
+ static VALUE rb_gsl_matrix_arithmetics(int flag, VALUE obj, VALUE bb)
29
+ {
30
+ gsl_matrix *m = NULL, *mb = NULL, *mnew = NULL;
31
+ gsl_matrix_complex *cm = NULL, *cmb = NULL, *cmnew = NULL;
32
+ gsl_complex *c = NULL;
33
+ gsl_vector *v = NULL, *vnew = NULL;
34
+ gsl_vector_complex *cv = NULL, *cvnew = NULL;
35
+ Data_Get_Struct(obj, gsl_matrix, m);
36
+ switch (TYPE(bb)) {
37
+ case T_FLOAT:
38
+ case T_FIXNUM:
39
+ switch (flag) {
40
+ case GSL_MATRIX_ADD:
41
+ mnew = make_matrix_clone(m);
42
+ gsl_matrix_add_constant(mnew, NUM2DBL(bb));
43
+ break;
44
+ case GSL_MATRIX_SUB:
45
+ mnew = make_matrix_clone(m);
46
+ gsl_matrix_add_constant(mnew, -NUM2DBL(bb));
47
+ break;
48
+ case GSL_MATRIX_MUL:
49
+ mnew = make_matrix_clone(m);
50
+ gsl_matrix_scale(mnew, NUM2DBL(bb));
51
+ break;
52
+ case GSL_MATRIX_DIV:
53
+ mnew = make_matrix_clone(m);
54
+ gsl_matrix_scale(mnew, 1.0/NUM2DBL(bb));
55
+ break;
56
+ default:
57
+ rb_raise(rb_eRuntimeError, "operation not defined");
58
+ break;
59
+ }
60
+ return Data_Wrap_Struct(cgsl_matrix, 0, gsl_matrix_free, mnew);
61
+ break;
62
+ default:
63
+ if (MATRIX_INT_P(bb)) bb = rb_gsl_matrix_int_to_f(bb);
64
+ if (VECTOR_INT_P(bb)) bb = rb_gsl_vector_int_to_f(bb);
65
+
66
+ if (rb_obj_is_kind_of(bb, cgsl_matrix)) {
67
+ Data_Get_Struct(bb, gsl_matrix, mb);
68
+ switch (flag) {
69
+ case GSL_MATRIX_ADD:
70
+ mnew = make_matrix_clone(m);
71
+ gsl_matrix_add(mnew, mb);
72
+ break;
73
+ case GSL_MATRIX_SUB:
74
+ mnew = make_matrix_clone(m);
75
+ gsl_matrix_sub(mnew,mb);
76
+ break;
77
+ case GSL_MATRIX_MUL:
78
+ mnew = make_matrix_clone(m);
79
+ gsl_matrix_mul_elements(mnew, mb);
80
+ break;
81
+ case GSL_MATRIX_DIV:
82
+ mnew = make_matrix_clone(m);
83
+ gsl_matrix_div_elements(mnew, mb);
84
+ break;
85
+ default:
86
+ rb_raise(rb_eRuntimeError, "operation not defined");
87
+ break;
88
+ }
89
+ return Data_Wrap_Struct(cgsl_matrix, 0, gsl_matrix_free, mnew);
90
+ } else if (rb_obj_is_kind_of(bb, cgsl_matrix_complex)) {
91
+ Data_Get_Struct(bb, gsl_matrix_complex, cmb);
92
+ cmnew = matrix_to_complex(m);
93
+ switch (flag) {
94
+ case GSL_MATRIX_ADD:
95
+ gsl_matrix_complex_add(cmnew, cmb);
96
+ break;
97
+ case GSL_MATRIX_SUB:
98
+ gsl_matrix_complex_sub(cmnew,cmb);
99
+ break;
100
+ case GSL_MATRIX_MUL:
101
+ gsl_matrix_complex_mul_elements(cmnew, cmb);
102
+ break;
103
+ case GSL_MATRIX_DIV:
104
+ gsl_matrix_complex_div_elements(cmnew, cmb);
105
+ break;
106
+ default:
107
+ rb_raise(rb_eRuntimeError, "operation not defined");
108
+ break;
109
+ }
110
+ return Data_Wrap_Struct(cgsl_matrix_complex, 0, gsl_matrix_complex_free, cmnew);
111
+ } else if (rb_obj_is_kind_of(bb, cgsl_complex)) {
112
+ Data_Get_Struct(bb, gsl_complex, c);
113
+ cmnew = matrix_to_complex(m);
114
+ switch (flag) {
115
+ case GSL_MATRIX_ADD:
116
+ gsl_matrix_complex_add_constant(cmnew, *c);
117
+ break;
118
+ case GSL_MATRIX_SUB:
119
+ gsl_matrix_complex_add_constant(cmnew, gsl_complex_negative(*c));
120
+ break;
121
+ case GSL_MATRIX_MUL:
122
+ gsl_matrix_complex_scale(cmnew, *c);
123
+ break;
124
+ case GSL_MATRIX_DIV:
125
+ gsl_matrix_complex_scale(cmnew, gsl_complex_inverse(*c));
126
+ break;
127
+ default:
128
+ rb_raise(rb_eRuntimeError, "operation not defined");
129
+ break;
130
+ }
131
+ return Data_Wrap_Struct(cgsl_matrix_complex, 0, gsl_matrix_complex_free, cmnew);
132
+ } else if (rb_obj_is_kind_of(bb, cgsl_vector)) {
133
+ if (!VECTOR_COL_P(bb))
134
+ rb_raise(rb_eTypeError,
135
+ "Operation with %s is not defined (GSL::Vector::Col expected)",
136
+ rb_class2name(CLASS_OF(bb)));
137
+ Data_Get_Struct(bb, gsl_vector, v);
138
+ switch (flag) {
139
+ case GSL_MATRIX_MUL:
140
+ // vnew = gsl_vector_alloc(v->size);
141
+ vnew = gsl_vector_alloc(m->size1);
142
+ if (vnew == NULL) rb_raise(rb_eNoMemError, "gsl_vector_alloc failed");
143
+ gsl_matrix_mul_vector(vnew, m, v);
144
+ return Data_Wrap_Struct(cgsl_vector_col, 0, gsl_vector_free, vnew);
145
+ break;
146
+ case GSL_MATRIX_DIV:
147
+ return rb_gsl_linalg_LU_solve(1, &bb, obj);
148
+ break;
149
+ default:
150
+ rb_raise(rb_eRuntimeError, "operation is not defined %s and Matrix",
151
+ rb_class2name(CLASS_OF(bb)));
152
+ break;
153
+ }
154
+ } else if (rb_obj_is_kind_of(bb, cgsl_vector_complex)) {
155
+ Data_Get_Struct(bb, gsl_vector_complex, cv);
156
+ switch (flag) {
157
+ case GSL_MATRIX_MUL:
158
+ cm = matrix_to_complex(m);
159
+ // cvnew = gsl_vector_complex_alloc(cv->size);
160
+ cvnew = gsl_vector_complex_alloc(m->size1);
161
+ if (cvnew == NULL) rb_raise(rb_eNoMemError, "gsl_vector_complex_alloc failed");
162
+ gsl_matrix_complex_mul_vector(cvnew, cm, cv);
163
+ gsl_matrix_complex_free(cm);
164
+ return Data_Wrap_Struct(cgsl_vector_complex, 0, gsl_vector_complex_free, cvnew);
165
+ break;
166
+ default:
167
+ rb_raise(rb_eRuntimeError, "operation is not defined %s and Matrix",
168
+ rb_class2name(CLASS_OF(bb)));
169
+ break;
170
+ }
171
+ } else {
172
+ rb_raise(rb_eTypeError, "wrong argument type %s", rb_class2name(CLASS_OF(bb)));
173
+ }
174
+ }
175
+ /* never reach here */
176
+ return Qnil;
177
+ }
178
+
179
+ static VALUE rb_gsl_matrix_add(VALUE obj, VALUE bb)
180
+ {
181
+ return rb_gsl_matrix_arithmetics(GSL_MATRIX_ADD, obj, bb);
182
+ }
183
+
184
+ static VALUE rb_gsl_matrix_sub(VALUE obj, VALUE bb)
185
+ {
186
+ return rb_gsl_matrix_arithmetics(GSL_MATRIX_SUB, obj, bb);
187
+ }
188
+
189
+ VALUE rb_gsl_matrix_mul_elements(VALUE obj, VALUE bb)
190
+ {
191
+ return rb_gsl_matrix_arithmetics(GSL_MATRIX_MUL, obj, bb);
192
+ }
193
+
194
+ /* matrix multiplication */
195
+ static VALUE rb_gsl_matrix_mul(VALUE obj, VALUE bb)
196
+ {
197
+ gsl_matrix *m = NULL, *b = NULL, *mnew = NULL;
198
+ gsl_vector *v = NULL, *vnew = NULL;
199
+ gsl_matrix_complex *mc, *mcb, *mcnew;
200
+ gsl_vector_complex *vc, *vcnew;
201
+ gsl_complex za, zb;
202
+ Data_Get_Struct(obj, gsl_matrix, m);
203
+ if (VECTOR_INT_P(bb)) bb = rb_gsl_vector_int_to_f(bb);
204
+ if (MATRIX_P(bb)) {
205
+ Data_Get_Struct(bb, gsl_matrix, b);
206
+ mnew = gsl_matrix_alloc(m->size1, b->size2);
207
+ gsl_linalg_matmult(m, b, mnew);
208
+ return Data_Wrap_Struct(cgsl_matrix, 0, gsl_matrix_free, mnew);
209
+ } else if (VECTOR_P(bb)) {
210
+ Data_Get_Struct(bb, gsl_vector, v);
211
+ // vnew = gsl_vector_alloc(v->size);
212
+ // printf("%d %d\n", m->size1, m->size2);
213
+ vnew = gsl_vector_alloc(m->size1);
214
+ if (vnew == NULL) rb_raise(rb_eNoMemError, "gsl_vector_alloc failed");
215
+ gsl_matrix_mul_vector(vnew, m, v);
216
+ return Data_Wrap_Struct(VECTOR_ROW_COL(bb), 0, gsl_vector_free, vnew);
217
+ } else if (MATRIX_COMPLEX_P(bb)) {
218
+ Data_Get_Struct(bb, gsl_matrix_complex, mcb);
219
+ mc = matrix_to_complex(m);
220
+ mcnew = gsl_matrix_complex_alloc(m->size1, m->size2);
221
+ gsl_matrix_complex_mul(mcnew, mc, mcb);
222
+ gsl_matrix_complex_free(mc);
223
+ return Data_Wrap_Struct(cgsl_matrix_complex, 0, gsl_matrix_complex_free, mcnew);
224
+ } else if (VECTOR_COMPLEX_P(bb)) {
225
+ Data_Get_Struct(bb, gsl_vector_complex, vc);
226
+ vcnew = gsl_vector_complex_calloc(m->size1);
227
+ mc = matrix_to_complex(m);
228
+ za.dat[0] = 1.0; za.dat[1] = 0.0;
229
+ zb.dat[0] = 0.0; zb.dat[1] = 0.0;
230
+ gsl_blas_zgemv(CblasNoTrans, za, mc, vc, zb, vcnew);
231
+ gsl_matrix_complex_free(mc);
232
+ return Data_Wrap_Struct(cgsl_vector_complex, 0, gsl_vector_complex_free, vcnew);
233
+ } else {
234
+ switch (TYPE(bb)) {
235
+ case T_FIXNUM: case T_BIGNUM: case T_FLOAT:
236
+ return rb_gsl_matrix_mul_elements(obj, bb);
237
+ break;
238
+ default:
239
+ rb_raise(rb_eTypeError,
240
+ "wrong argument type %s", rb_class2name(CLASS_OF(bb)));
241
+ break;
242
+ }
243
+ }
244
+ return Qnil;
245
+ }
246
+
247
+ static VALUE rb_gsl_matrix_mul_bang(VALUE obj, VALUE bb)
248
+ {
249
+ gsl_matrix *m = NULL, *b = NULL, *mtmp = NULL;
250
+ gsl_vector *v = NULL, *vnew = NULL;
251
+ Data_Get_Struct(obj, gsl_matrix, m);
252
+ if (MATRIX_P(bb)) {
253
+ Data_Get_Struct(bb, gsl_matrix, b);
254
+ mtmp = gsl_matrix_alloc(m->size1, b->size2);
255
+ gsl_linalg_matmult(m, b, mtmp);
256
+ gsl_matrix_memcpy(m, mtmp);
257
+ gsl_matrix_free(mtmp);
258
+ return obj;
259
+ } else if (VECTOR_P(bb)) {
260
+ Data_Get_Struct(bb, gsl_vector, v);
261
+ vnew = gsl_vector_alloc(v->size);
262
+ if (vnew == NULL) rb_raise(rb_eNoMemError, "gsl_vector_alloc failed");
263
+ gsl_matrix_mul_vector(vnew, m, v);
264
+ return Data_Wrap_Struct(VECTOR_ROW_COL(bb), 0, gsl_vector_free, vnew);
265
+ } else {
266
+ switch (TYPE(bb)) {
267
+ case T_FIXNUM:
268
+ case T_BIGNUM:
269
+ case T_FLOAT:
270
+ gsl_matrix_scale(m, NUM2DBL(bb));
271
+ return obj;
272
+ break;
273
+ default:
274
+ rb_raise(rb_eTypeError,
275
+ "wrong argument type %s", rb_class2name(CLASS_OF(bb)));
276
+ break;
277
+ }
278
+ }
279
+ }
280
+
281
+ static VALUE rb_gsl_matrix_div_elements(VALUE obj, VALUE bb)
282
+ {
283
+ return rb_gsl_matrix_arithmetics(GSL_MATRIX_DIV, obj, bb);
284
+ }
285
+
286
+ static VALUE rb_gsl_matrix_to_complex(VALUE obj)
287
+ {
288
+ gsl_matrix *m = NULL;
289
+ gsl_matrix_complex *cm = NULL;
290
+ Data_Get_Struct(obj, gsl_matrix, m);
291
+ cm = matrix_to_complex(m);
292
+ return Data_Wrap_Struct(cgsl_matrix_complex, 0, gsl_matrix_complex_free, cm);
293
+ }
294
+
295
+ gsl_matrix_view* gsl_matrix_view_alloc()
296
+ {
297
+ gsl_matrix_view *mv = NULL;
298
+ mv = ALLOC(gsl_matrix_view);
299
+ if (mv == NULL) rb_raise(rb_eRuntimeError, "malloc failed");
300
+ return mv;
301
+ }
302
+
303
+ void gsl_matrix_view_free(gsl_matrix_view * mv);
304
+ void gsl_matrix_view_free(gsl_matrix_view * mv)
305
+ {
306
+ free((gsl_matrix_view *) mv);
307
+ }
308
+
309
+ static VALUE rb_gsl_matrix_coerce(VALUE obj, VALUE other)
310
+ {
311
+ gsl_matrix *m = NULL, *mnew = NULL;
312
+ gsl_matrix_complex *cm = NULL;
313
+ double x;
314
+ gsl_complex *z = NULL;
315
+ VALUE vcm;
316
+ Data_Get_Struct(obj, gsl_matrix, m);
317
+ switch (TYPE(other)) {
318
+ case T_FLOAT:
319
+ case T_FIXNUM:
320
+ mnew = gsl_matrix_alloc(m->size1, m->size2);
321
+ if (mnew == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_alloc failed");
322
+ x = NUM2DBL(other);
323
+ gsl_matrix_set_all(mnew, x);
324
+ return rb_ary_new3(2, Data_Wrap_Struct(cgsl_matrix, 0, gsl_matrix_free, mnew), obj);
325
+ break;
326
+ default:
327
+ if (rb_obj_is_kind_of(other, cgsl_complex)) {
328
+ Data_Get_Struct(other, gsl_complex, z);
329
+ cm = gsl_matrix_complex_alloc(m->size1, m->size2);
330
+ if (cm == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_complex_alloc failed");
331
+ gsl_matrix_complex_set_all(cm, *z);
332
+ vcm = Data_Wrap_Struct(cgsl_matrix_complex, 0, gsl_matrix_complex_free, cm);
333
+ return rb_ary_new3(2, vcm, rb_gsl_matrix_to_complex(obj));
334
+ } else if (rb_obj_is_kind_of(other, cgsl_matrix_complex)) {
335
+ cm = matrix_to_complex(m);
336
+ vcm = Data_Wrap_Struct(cgsl_matrix_complex, 0, gsl_matrix_complex_free, cm);
337
+ return rb_ary_new3(2, other, vcm);
338
+ } else {
339
+ rb_raise(rb_eTypeError, "cannot coerce %s to Matrix",
340
+ rb_class2name(CLASS_OF(other)));
341
+ }
342
+ break;
343
+ }
344
+ }
345
+
346
+ static VALUE rb_gsl_matrix_to_f(VALUE obj)
347
+ {
348
+ return obj;
349
+ }
350
+
351
+ VALUE rb_gsl_matrix_to_i(VALUE obj)
352
+ {
353
+ gsl_matrix *m = NULL;
354
+ gsl_matrix_int *mi = NULL;
355
+ size_t i, j;
356
+ Data_Get_Struct(obj, gsl_matrix, m);
357
+ mi = gsl_matrix_int_alloc(m->size1, m->size2);
358
+ for (i = 0; i < m->size1; i++) {
359
+ for (j = 0; j < m->size2; j++) {
360
+ gsl_matrix_int_set(mi, i, j, (int) gsl_matrix_get(m, i, j));
361
+ }
362
+ }
363
+ return Data_Wrap_Struct(cgsl_matrix_int, 0, gsl_matrix_int_free, mi);
364
+ }
365
+
366
+ static VALUE rb_gsl_matrix_clean(int argc, VALUE *argv, VALUE obj)
367
+ {
368
+ gsl_matrix *m = NULL, *mnew = NULL;
369
+ double eps = 1e-10;
370
+ size_t n, i;
371
+ switch (argc) {
372
+ case 0:
373
+ /* do nothing */
374
+ break;
375
+ case 1:
376
+ Need_Float(argv[0]);
377
+ eps = NUM2DBL(argv[0]);
378
+ break;
379
+ default:
380
+ rb_raise(rb_eArgError, "too many arguments (%d for 0 or 1)", argc);
381
+ break;
382
+ }
383
+ Data_Get_Struct(obj, gsl_matrix, m);
384
+ mnew = make_matrix_clone(m);
385
+ n = m->size1*m->size2;
386
+ for (i = 0; i < n; i++) if (fabs(mnew->data[i]) < eps) mnew->data[i] = 0.0;
387
+ return Data_Wrap_Struct(cgsl_matrix, 0, gsl_matrix_free, mnew);;
388
+ }
389
+
390
+ static VALUE rb_gsl_matrix_clean_bang(int argc, VALUE *argv, VALUE obj)
391
+ {
392
+ gsl_matrix *m = NULL;
393
+ double eps = 1e-10;
394
+ size_t n, i;
395
+ switch (argc) {
396
+ case 0:
397
+ /* do nothing */
398
+ break;
399
+ case 1:
400
+ Need_Float(argv[0]);
401
+ eps = NUM2DBL(argv[0]);
402
+ break;
403
+ default:
404
+ rb_raise(rb_eArgError, "too many arguments (%d for 0 or 1)", argc);
405
+ break;
406
+ }
407
+ Data_Get_Struct(obj, gsl_matrix, m);
408
+ n = m->size1*m->size2;
409
+ for (i = 0; i < n; i++) if (fabs(m->data[i]) < eps) m->data[i] = 0.0;
410
+ return obj;
411
+ }
412
+
413
+
414
+ static VALUE rb_gsl_matrix_op_inplace(VALUE mm1, VALUE mm2,
415
+ int (*f)(gsl_matrix*, const gsl_matrix*))
416
+ {
417
+ gsl_matrix *m1, *m2;
418
+ Data_Get_Struct(mm1, gsl_matrix, m1);
419
+ Data_Get_Struct(mm2, gsl_matrix, m2);
420
+ (*f)(m1, m2);
421
+ return mm1;
422
+ }
423
+
424
+ static VALUE rb_gsl_matrix_add_inplace(VALUE mm1, VALUE mm2)
425
+ {
426
+ return rb_gsl_matrix_op_inplace(mm1, mm2, gsl_matrix_add);
427
+ }
428
+
429
+ static VALUE rb_gsl_matrix_sub_inplace(VALUE mm1, VALUE mm2)
430
+ {
431
+ return rb_gsl_matrix_op_inplace(mm1, mm2, gsl_matrix_sub);
432
+ }
433
+
434
+ VALUE rb_gsl_sf_eval1(double (*func)(double), VALUE argv);
435
+
436
+ static VALUE rb_gsl_matrix_sin(VALUE obj)
437
+ {
438
+ return rb_gsl_sf_eval1(sin, obj);
439
+ }
440
+
441
+ static VALUE rb_gsl_matrix_cos(VALUE obj)
442
+ {
443
+ return rb_gsl_sf_eval1(cos, obj);
444
+ }
445
+
446
+ static VALUE rb_gsl_matrix_tan(VALUE obj)
447
+ {
448
+ return rb_gsl_sf_eval1(tan, obj);
449
+ }
450
+
451
+ static VALUE rb_gsl_matrix_exp(VALUE obj)
452
+ {
453
+ return rb_gsl_sf_eval1(exp, obj);
454
+ }
455
+
456
+ static VALUE rb_gsl_matrix_log(VALUE obj)
457
+ {
458
+ return rb_gsl_sf_eval1(log, obj);
459
+ }
460
+
461
+ static VALUE rb_gsl_matrix_log10(VALUE obj)
462
+ {
463
+ return rb_gsl_sf_eval1(log10, obj);
464
+ }
465
+
466
+ #include <gsl/gsl_rng.h>
467
+ #include <gsl/gsl_randist.h>
468
+ #include "rb_gsl_rng.h"
469
+ static VALUE rb_gsl_matrix_randx(int argc, VALUE *argv, VALUE klass,
470
+ double (*f)(const gsl_rng*));
471
+ static VALUE rb_gsl_matrix_rand(int argc, VALUE *argv, VALUE klass)
472
+ {
473
+ return rb_gsl_matrix_randx(argc, argv, klass, gsl_rng_uniform);
474
+ }
475
+
476
+ static VALUE rb_gsl_matrix_randn(int argc, VALUE *argv, VALUE klass)
477
+ {
478
+ return rb_gsl_matrix_randx(argc, argv, klass, gsl_ran_ugaussian);
479
+ }
480
+
481
+ static VALUE rb_gsl_matrix_randx(int argc, VALUE *argv, VALUE klass,
482
+ double (*f)(const gsl_rng*))
483
+ {
484
+ gsl_matrix *m;
485
+ gsl_rng *rng;
486
+ size_t size1, size2;
487
+ size_t i, j;
488
+ switch (argc) {
489
+ case 3:
490
+ if (!rb_obj_is_kind_of(argv[2], cgsl_rng)) {
491
+ rb_raise(rb_eTypeError,
492
+ "Wrong argument type (GSL::Rng expected)");
493
+ }
494
+ Data_Get_Struct(argv[2], gsl_rng, rng);
495
+ size1 = FIX2INT(argv[0]);
496
+ size2 = FIX2INT(argv[1]);
497
+ break;
498
+ case 2:
499
+ size1 = FIX2INT(argv[0]);
500
+ size2 = FIX2INT(argv[1]);
501
+ rng = gsl_rng_alloc(gsl_rng_default);
502
+ break;
503
+ default:
504
+ rb_raise(rb_eArgError, "Wrong number of arguments (%d for 2 or 3)", argc);
505
+ }
506
+ m = gsl_matrix_alloc(size1, size2);
507
+ for (i = 0; i < size1; i++) {
508
+ for (j = 0; j < size2; j++) {
509
+ gsl_matrix_set(m, i, j, (*f)(rng));
510
+ }
511
+ }
512
+ if (argc == 2) gsl_rng_free(rng);
513
+ return Data_Wrap_Struct(klass, 0, gsl_matrix_free, m);
514
+ }
515
+
516
+ void Init_gsl_matrix_init(VALUE module);
517
+ void Init_gsl_matrix(VALUE module)
518
+ {
519
+ Init_gsl_matrix_init(module);
520
+
521
+ rb_define_method(cgsl_matrix, "add", rb_gsl_matrix_add, 1);
522
+ rb_define_alias(cgsl_matrix, "+", "add");
523
+ rb_define_method(cgsl_matrix, "sub", rb_gsl_matrix_sub, 1);
524
+ rb_define_alias(cgsl_matrix, "-", "sub");
525
+ rb_define_method(cgsl_matrix, "mul_elements", rb_gsl_matrix_mul_elements, 1);
526
+ rb_define_method(cgsl_matrix, "div_elements", rb_gsl_matrix_div_elements, 1);
527
+ rb_define_alias(cgsl_matrix, "/", "div_elements");
528
+
529
+ rb_define_method(cgsl_matrix, "mul", rb_gsl_matrix_mul, 1);
530
+ rb_define_alias(cgsl_matrix, "*", "mul");
531
+ rb_define_method(cgsl_matrix, "mul!", rb_gsl_matrix_mul_bang, 1);
532
+ /***/
533
+
534
+ rb_define_method(cgsl_matrix, "to_complex", rb_gsl_matrix_to_complex, 0);
535
+
536
+ rb_define_method(cgsl_matrix, "coerce", rb_gsl_matrix_coerce, 1);
537
+
538
+ rb_undef_method(cgsl_matrix_view_ro, "set");
539
+
540
+ rb_define_method(cgsl_matrix, "to_i", rb_gsl_matrix_to_i, 0);
541
+ rb_define_method(cgsl_matrix, "to_f", rb_gsl_matrix_to_f, 0);
542
+
543
+ rb_define_method(cgsl_matrix, "clean", rb_gsl_matrix_clean, -1);
544
+ rb_define_method(cgsl_matrix, "clean!", rb_gsl_matrix_clean_bang, -1);
545
+
546
+ /*****/
547
+ rb_define_method(cgsl_matrix, "add!", rb_gsl_matrix_add_inplace, 1);
548
+ rb_define_method(cgsl_matrix, "sub!", rb_gsl_matrix_sub_inplace, 1);
549
+
550
+ rb_define_method(cgsl_matrix, "sin", rb_gsl_matrix_sin, 0);
551
+ rb_define_method(cgsl_matrix, "cos", rb_gsl_matrix_cos, 0);
552
+ rb_define_method(cgsl_matrix, "tan", rb_gsl_matrix_tan, 0);
553
+ rb_define_method(cgsl_matrix, "exp", rb_gsl_matrix_exp, 0);
554
+ rb_define_method(cgsl_matrix, "log", rb_gsl_matrix_log, 0);
555
+ rb_define_method(cgsl_matrix, "log10", rb_gsl_matrix_log10, 0);
556
+
557
+ rb_define_singleton_method(cgsl_matrix, "rand", rb_gsl_matrix_rand, -1);
558
+ rb_define_singleton_method(cgsl_matrix, "randn", rb_gsl_matrix_randn, -1);
559
+
560
+ }