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,256 @@
1
+ /*
2
+ matrix_int.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
+ int gsl_linalg_matmult_int(const gsl_matrix_int *A,
19
+ const gsl_matrix_int *B, gsl_matrix_int *C);
20
+
21
+
22
+ VALUE rb_gsl_matrix_to_i(VALUE obj);
23
+
24
+ static VALUE rb_gsl_matrix_int_to_i(VALUE obj)
25
+ {
26
+ return obj;
27
+ }
28
+
29
+ VALUE rb_gsl_matrix_int_to_f(VALUE obj)
30
+ {
31
+ gsl_matrix_int *m;
32
+ gsl_matrix *mnew;
33
+ size_t i, j;
34
+ Data_Get_Struct(obj, gsl_matrix_int, m);
35
+ mnew = gsl_matrix_alloc(m->size1, m->size2);
36
+ for (i = 0; i < m->size1; i++) {
37
+ for (j = 0; j < m->size2; j++) {
38
+ gsl_matrix_set(mnew, i, j, (double) gsl_matrix_int_get(m, i, j));
39
+ }
40
+ }
41
+ return Data_Wrap_Struct(cgsl_matrix, 0, gsl_matrix_free, mnew);
42
+ }
43
+
44
+ static VALUE rb_gsl_matrix_int_to_complex(VALUE obj)
45
+ {
46
+ gsl_matrix_int *m;
47
+ gsl_matrix_complex *mnew;
48
+ gsl_complex z;
49
+ size_t i, j;
50
+ Data_Get_Struct(obj, gsl_matrix_int, m);
51
+ mnew = gsl_matrix_complex_alloc(m->size1, m->size2);
52
+ for (i = 0; i < m->size1; i++) {
53
+ for (j = 0; j < m->size2; j++) {
54
+ GSL_SET_REAL(&z, (double) gsl_matrix_int_get(m, i, j));
55
+ GSL_SET_IMAG(&z, 0.0);
56
+ gsl_matrix_complex_set(mnew, i, j, z);
57
+ }
58
+ }
59
+ return Data_Wrap_Struct(cgsl_matrix_complex, 0, gsl_matrix_complex_free, mnew);
60
+ }
61
+
62
+ static VALUE rb_gsl_matrix_int_coerce(VALUE obj, VALUE other)
63
+ {
64
+ return rb_ary_new3(2, other, rb_gsl_matrix_int_to_f(obj));
65
+ }
66
+
67
+ enum {
68
+ GSL_MATRIX_INT_ADD,
69
+ GSL_MATRIX_INT_SUB,
70
+ GSL_MATRIX_INT_MUL,
71
+ GSL_MATRIX_INT_DIV,
72
+ };
73
+
74
+ static VALUE rb_gsl_matrix_int_operation1(VALUE obj, VALUE other, int flag);
75
+ static VALUE rb_gsl_matrix_int_operation1(VALUE obj, VALUE other, int flag)
76
+ {
77
+ gsl_matrix_int *a, *anew, *b;
78
+ gsl_vector_int *vi, *vinew;
79
+ double bval;
80
+ int result;
81
+ Data_Get_Struct(obj, gsl_matrix_int, a);
82
+ switch (TYPE(other)) {
83
+ case T_FIXNUM:
84
+ case T_FLOAT:
85
+ bval = NUM2INT(other);
86
+ anew = make_matrix_int_clone(a);
87
+ switch (flag) {
88
+ case GSL_MATRIX_INT_ADD:
89
+ result = gsl_matrix_int_add_constant(anew, bval);
90
+ break;
91
+ case GSL_MATRIX_INT_SUB:
92
+ result = gsl_matrix_int_add_constant(anew, -bval);
93
+ break;
94
+ case GSL_MATRIX_INT_MUL:
95
+ result = gsl_matrix_int_scale(anew, bval);
96
+ break;
97
+ case GSL_MATRIX_INT_DIV:
98
+ result = gsl_matrix_int_scale(anew, 1.0/bval);
99
+ break;
100
+ default:
101
+ break;
102
+ }
103
+ break;
104
+ default:
105
+ if (MATRIX_P(other)) other = rb_gsl_matrix_to_i(other);
106
+ if (VECTOR_P(other)) other = rb_gsl_vector_to_i(other);
107
+ if (MATRIX_INT_P(other)) {
108
+ anew = make_matrix_int_clone(a);
109
+ Data_Get_Struct(other, gsl_matrix_int, b);
110
+ switch (flag) {
111
+ case GSL_MATRIX_INT_ADD:
112
+ result = gsl_matrix_int_add(anew, b);
113
+ break;
114
+ case GSL_MATRIX_INT_SUB:
115
+ result = gsl_matrix_int_sub(anew, b);
116
+ break;
117
+ case GSL_MATRIX_INT_MUL:
118
+ result = gsl_matrix_int_mul_elements(anew, b);
119
+ break;
120
+ case GSL_MATRIX_INT_DIV:
121
+ result = gsl_matrix_int_div_elements(anew, b);
122
+ break;
123
+ default:
124
+ break;
125
+ }
126
+ } else if (VECTOR_INT_COL_P(other)) {
127
+ switch (flag) {
128
+ case GSL_MATRIX_INT_MUL:
129
+ Data_Get_Struct(other, gsl_vector_int, vi);
130
+ vinew = gsl_vector_int_alloc(vi->size);
131
+ gsl_matrix_int_mul_vector(vinew, a, vi);
132
+ return Data_Wrap_Struct(cgsl_vector_int_col, 0, gsl_vector_int_free, vinew);
133
+ break;
134
+ default:
135
+ rb_raise(rb_eRuntimeError, "Operation not defined");
136
+ }
137
+ } else {
138
+ rb_raise(rb_eTypeError, "Operation not defined with %s",
139
+ rb_class2name(CLASS_OF(other)));
140
+ }
141
+ break;
142
+ }
143
+ return Data_Wrap_Struct(cgsl_matrix_int, 0, gsl_matrix_int_free, anew);
144
+ }
145
+
146
+ static VALUE rb_gsl_matrix_int_add(VALUE obj, VALUE other)
147
+ {
148
+ return rb_gsl_matrix_int_operation1(obj, other, GSL_MATRIX_INT_ADD);
149
+ }
150
+
151
+ static VALUE rb_gsl_matrix_int_sub(VALUE obj, VALUE other)
152
+ {
153
+ return rb_gsl_matrix_int_operation1(obj, other, GSL_MATRIX_INT_SUB);
154
+ }
155
+
156
+ static VALUE rb_gsl_matrix_int_mul(VALUE obj, VALUE other)
157
+ {
158
+ return rb_gsl_matrix_int_operation1(obj, other, GSL_MATRIX_INT_MUL);
159
+ }
160
+
161
+ static VALUE rb_gsl_matrix_int_div(VALUE obj, VALUE other)
162
+ {
163
+ return rb_gsl_matrix_int_operation1(obj, other, GSL_MATRIX_INT_DIV);
164
+ }
165
+
166
+ static VALUE rb_gsl_matrix_int_matrix_mul(VALUE obj, VALUE bb)
167
+ {
168
+ gsl_matrix_int *m = NULL, *b = NULL, *mnew = NULL;
169
+ gsl_vector_int *vi, *vinew;
170
+ Data_Get_Struct(obj, gsl_matrix_int, m);
171
+ if (MATRIX_INT_P(bb)) {
172
+ Data_Get_Struct(bb, gsl_matrix_int, b);
173
+ mnew = gsl_matrix_int_alloc(m->size1, b->size2);
174
+ gsl_linalg_matmult_int(m, b, mnew);
175
+ return Data_Wrap_Struct(cgsl_matrix_int, 0, gsl_matrix_int_free, mnew);
176
+ } else {
177
+ if (VECTOR_INT_COL_P(bb)) {
178
+ Data_Get_Struct(bb, gsl_vector_int, vi);
179
+ vinew = gsl_vector_int_alloc(vi->size);
180
+ gsl_matrix_int_mul_vector(vinew, m, vi);
181
+ return Data_Wrap_Struct(cgsl_vector_int_col, 0, gsl_vector_int_free, vinew);
182
+ }
183
+ switch (TYPE(bb)) {
184
+ case T_FIXNUM:
185
+ return rb_gsl_matrix_int_mul(obj, bb);
186
+ /* return rb_gsl_matrix_int_power(obj, bb);*/
187
+ break;
188
+ default:
189
+ rb_raise(rb_eTypeError, "wrong argument type %s (Matrix::Int, Vector::Int::Col or Fixnum expected)",
190
+ rb_class2name(CLASS_OF(bb)));
191
+ break;
192
+ }
193
+ }
194
+ }
195
+
196
+ int gsl_linalg_matmult_int(const gsl_matrix_int *A,
197
+ const gsl_matrix_int *B, gsl_matrix_int *C)
198
+ {
199
+ if (A->size2 != B->size1 || A->size1 != C->size1 || B->size2 != C->size2)
200
+ {
201
+ GSL_ERROR ("matrix sizes are not conformant", GSL_EBADLEN);
202
+ }
203
+ else
204
+ {
205
+ int a, b;
206
+ int temp;
207
+ size_t i, j, k;
208
+
209
+ for (i = 0; i < C->size1; i++)
210
+ {
211
+ for (j = 0; j < C->size2; j++)
212
+ {
213
+ a = gsl_matrix_int_get(A, i, 0);
214
+ b = gsl_matrix_int_get(B, 0, j);
215
+ temp = a * b;
216
+ for (k = 1; k < A->size2; k++)
217
+ {
218
+ a = gsl_matrix_int_get(A, i, k);
219
+ b = gsl_matrix_int_get(B, k, j);
220
+ temp += a * b;
221
+ }
222
+ gsl_matrix_int_set(C, i, j, temp);
223
+ }
224
+ }
225
+
226
+ return GSL_SUCCESS;
227
+ }
228
+ }
229
+
230
+ void Init_gsl_matrix_int_init(VALUE module);
231
+ void Init_gsl_matrix_int(VALUE module)
232
+ {
233
+ Init_gsl_matrix_int_init(module);
234
+ /*****/
235
+
236
+ rb_define_method(cgsl_matrix_int, "to_f", rb_gsl_matrix_int_to_f, 0);
237
+ rb_define_method(cgsl_matrix_int, "to_i", rb_gsl_matrix_int_to_i, 0);
238
+ rb_define_method(cgsl_matrix_int, "to_complex", rb_gsl_matrix_int_to_complex, 0);
239
+
240
+ /*****/
241
+ rb_define_method(cgsl_matrix_int, "coerce", rb_gsl_matrix_int_coerce, 1);
242
+ /*****/
243
+ rb_define_method(cgsl_matrix_int, "add", rb_gsl_matrix_int_add, 1);
244
+ rb_define_alias(cgsl_matrix_int, "+", "add");
245
+ rb_define_method(cgsl_matrix_int, "sub", rb_gsl_matrix_int_sub, 1);
246
+ rb_define_alias(cgsl_matrix_int, "-", "sub");
247
+ rb_define_method(cgsl_matrix_int, "mul", rb_gsl_matrix_int_mul, 1);
248
+ /* rb_define_alias(cgsl_matrix_int, "*", "mul");*/
249
+ rb_define_method(cgsl_matrix_int, "div", rb_gsl_matrix_int_div, 1);
250
+ rb_define_alias(cgsl_matrix_int, "/", "div");
251
+
252
+ rb_define_method(cgsl_matrix_int, "matrix_mul", rb_gsl_matrix_int_matrix_mul, 1);
253
+ rb_define_alias(cgsl_matrix_int, "*", "matrix_mul");
254
+ /*****/
255
+
256
+ }
@@ -0,0 +1,2733 @@
1
+ /*
2
+ matrix_source.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
+
12
+ #ifdef BASE_DOUBLE
13
+ #define NUMCONV(x) NUM2DBL(x)
14
+ #define NUMCONV2(x) NUM2DBL(x)
15
+ #define PRINTF_FORMAT "%4.3e "
16
+ #define MAT_ROW_COL MATRIX_ROW_COL
17
+ #define MAT_P MATRIX_P
18
+ #define MAT_ROW_P MATRIX_ROW_P
19
+ #define MAT_COL_P MATRIX_COL_P
20
+ #define C_TO_VALUE rb_float_new
21
+ #define C_TO_VALUE2 rb_float_new
22
+ #define CHECK_MAT CHECK_MATRIX
23
+ #define MAT_VIEW_P MATRIX_VIEW_P
24
+ #define VEC_ROW_COL VECTOR_ROW_COL
25
+ #define VEC_P VECTOR_P
26
+ #define VEC_ROW_P VECTOR_ROW_P
27
+ #define VEC_COL_P VECTOR_COL_P
28
+ #define CHECK_VEC CHECK_VECTOR
29
+ #define VEC_VIEW_P VECTOR_VIEW_P
30
+ #elif defined(BASE_INT)
31
+ #define NUMCONV(x) FIX2INT(x)
32
+ #define NUMCONV2(x) NUM2INT(x)
33
+ #define PRINTF_FORMAT "%d "
34
+ #define MAT_ROW_COL MATRIX_INT_ROW_COL
35
+ #define MAT_P MATRIX_INT_P
36
+ #define C_TO_VALUE INT2FIX
37
+ #define C_TO_VALUE2 INT2NUM
38
+ #define MAT_ROW_P MATRIX_INT_ROW_P
39
+ #define MAT_COL_P MATRIX_INT_COL_P
40
+ #define CHECK_MAT CHECK_MATRIX_INT
41
+ #define MAT_VIEW_P MATRIX_INT_VIEW_P
42
+ #define VEC_ROW_COL VECTOR_INT_ROW_COL
43
+ #define VEC_P VECTOR_INT_P
44
+ #define VEC_ROW_P VECTOR_INT_ROW_P
45
+ #define VEC_COL_P VECTOR_INT_COL_P
46
+ #define CHECK_VEC CHECK_VECTOR_INT
47
+ #define VEC_VIEW_P VECTOR_INT_VIEW_P
48
+ #endif
49
+
50
+ // From ext/vector_source.c
51
+ void FUNCTION(get_range,beg_en_n)(VALUE range, BASE *beg, BASE *en, size_t *n, int *step);
52
+
53
+ // From ext/vector_source.c
54
+ void get_range_beg_en_n_for_size(VALUE range,
55
+ int *beg, int *en, size_t *n, int *step, size_t size);
56
+
57
+ void parse_submatrix_args(int argc, VALUE *argv, size_t size1, size_t size2,
58
+ size_t *i, size_t *j, size_t *n1, size_t *n2);
59
+
60
+ #ifdef BASE_DOUBLE
61
+ void parse_submatrix_args(int argc, VALUE *argv, size_t size1, size_t size2,
62
+ size_t *i, size_t *j, size_t *n1, size_t *n2)
63
+ {
64
+ int ii, ij, in1, in2, end, step;
65
+
66
+ switch (argc) {
67
+ // no args -> same as submatrix(0, 0, size1, size2)
68
+ case 0:
69
+ *i = 0; *j = 0;
70
+ *n1 = size1; *n2 = size2;
71
+ break;
72
+
73
+ // Fixnum -> Same as submatrix(i/size2, i%size2, 1, 1)
74
+ case 1:
75
+ CHECK_FIXNUM(argv[0]);
76
+ ii = FIX2INT(argv[0]);
77
+ *n1 = size1 * size2;
78
+ if(ii < 0) ii += *n1;
79
+ // TODO Bounds check?
80
+ *i = ii / size2; *j = ii % size2;
81
+ *n1 = 1; *n2 = 1;
82
+ break;
83
+
84
+
85
+ // nil, nil -> all rows, all cols (Matrix::View)
86
+ // nil, Range -> all rows, Range cols (Matrix::View)
87
+ // nil, Fixnum -> all rows, single col (Vector::Col::View)
88
+ // Range, nil -> Range rows, all cols (Matrix::View)
89
+ // Range, Range -> Range rows, Range cols (Matrix::View)
90
+ // Range, Fixnum -> Range rows, single col (Vector::Col::View)
91
+ // Fixnum, nil -> single row, all cols (Vector::View)
92
+ // Fixnum, Range -> single row, Range cols (Vector::View)
93
+ // Fixnum, Fixnum -> single row, single col (Matrix::View)
94
+ case 2:
95
+ // nil, ...
96
+ if(NIL_P(argv[0])) {
97
+ // Parse second arg
98
+ if(NIL_P(argv[1])) {
99
+ // nil, nil -> all rows, all cols (Matrix::View)
100
+ *i = 0; *j = 0;
101
+ *n1 = size1; *n2 = size2;
102
+ } else if(rb_obj_is_kind_of(argv[1], rb_cRange)) {
103
+ // nil, Range -> all rows, Range cols (Matrix::View)
104
+ *i = 0; *n1 = size1;
105
+ get_range_beg_en_n_for_size(argv[1], &ij, &end, n2, &step, size2);
106
+ if(step < 0 || *n2 <=0) {
107
+ rb_raise(rb_eRangeError, "begin > end");
108
+ }
109
+ *j = (size_t)ij;
110
+ } else {
111
+ // nil, Fixnum -> all rows, single col (Vector::Col::View)
112
+ ij = NUM2INT(argv[1]);
113
+ if(ij < 0) ij += size2;
114
+ *i = 0; *j = (size_t)ij;
115
+ *n1 = size1; *n2 = 0; // *n2 == 0 tells #submatrix to return Vector::Col::View
116
+ }
117
+ // Range, ...
118
+ } else if(rb_obj_is_kind_of(argv[0], rb_cRange)) {
119
+ get_range_beg_en_n_for_size(argv[0], &ii, &end, n1, &step, size1);
120
+ if(step < 0 || *n1 <= 0) {
121
+ rb_raise(rb_eRangeError, "arg0: begin > end");
122
+ }
123
+ *i = (size_t)ii;
124
+ // Parse second arg
125
+ if(NIL_P(argv[1])) {
126
+ // Range, nil -> Range rows, all cols (Matrix::View)
127
+ *j = 0; *n2 = size2;
128
+ } else if(rb_obj_is_kind_of(argv[1], rb_cRange)) {
129
+ // Range, Range -> Range rows, Range cols (Matrix::View)
130
+ get_range_beg_en_n_for_size(argv[1], &ij, &end, n2, &step, size2);
131
+ if(step < 0 || *n2 <= 0) {
132
+ rb_raise(rb_eRangeError, "arg1: begin > end");
133
+ }
134
+ *j = (size_t)ij;
135
+ } else {
136
+ // Range, Fixnum -> Range rows, single col (Vector::Col::View)
137
+ ij = NUM2INT(argv[1]);
138
+ if(ij < 0) ij += size2;
139
+ *j = (size_t) ij;
140
+ *n2 = 0; // *n2 == 0 tells #submatrix to return Vector::Col::View
141
+ }
142
+ // Fixnum, ...
143
+ } else {
144
+ ii = NUM2INT(argv[0]);
145
+ if(ii < 0) ii += size1;
146
+ if(NIL_P(argv[1])) {
147
+ // Fixnum, nil -> single row, all cols (Vector::View)
148
+ *i = (size_t)ii; *j = 0;
149
+ *n1 = 0; *n2 = size2; // *n1 == 0 tells #submatrix to return Vector::View
150
+ } else if(rb_obj_is_kind_of(argv[1], rb_cRange)) {
151
+ // Fixnum, Range -> single row, Range cols (Vector::View)
152
+ get_range_beg_en_n_for_size(argv[1], &ij, &end, n2, &step, size2);
153
+ if(step < 0 || *n2 <= 0) {
154
+ rb_raise(rb_eRangeError, "arg1: begin > end");
155
+ }
156
+ *i = (size_t)ii;
157
+ *j = (size_t)ij;
158
+ *n1 = 0; // *n1 == 0 tells #submatrix to return Vector::View
159
+ } else {
160
+ // Fixnum, Fixnum -> single row, single col (Matrix::View)
161
+ ij = NUM2INT(argv[1]);
162
+ if(ij < 0) ij += size2;
163
+ *i = (size_t)ii; *j = (size_t)ij;
164
+ *n1 = 1; *n2 = 1;
165
+ }
166
+ }
167
+ break;
168
+
169
+ // nil, Fixnum, Fixnum -> All rows, some cols
170
+ // Range, Fixnum, Fixnum -> Range rows, some cols
171
+ // Fixnum, Fixnum, nil -> Some rows, all cols
172
+ // Fixnum, Fixnum, Range -> Some rows, Range cols
173
+ case 3:
174
+ // nil, Fixnum, Fixnum
175
+ if(NIL_P(argv[0])) {
176
+ // nil, Fixnum, Fixnum -> All rows, some cols
177
+ CHECK_FIXNUM(argv[1]); CHECK_FIXNUM(argv[2]);
178
+ *i = 0;
179
+ ij = FIX2INT(argv[1]);
180
+ *n1 = size1;
181
+ in2 = FIX2INT(argv[2]);
182
+ if(ij < 0) ij += size2;
183
+ *j = (size_t) ij;
184
+ *n2 = (size_t) in2;
185
+ // Range, Fixnum, Fixnum
186
+ } else if(rb_obj_is_kind_of(argv[0], rb_cRange)) {
187
+ // Range, Fixnum, Fixnum -> Range rows, some cols
188
+ CHECK_FIXNUM(argv[1]); CHECK_FIXNUM(argv[2]);
189
+ get_range_beg_en_n_for_size(argv[0], &ii, &end, n1, &step, size1);
190
+ if(step < 0 || *n1 <= 0) {
191
+ rb_raise(rb_eRangeError, "arg0: begin > end");
192
+ }
193
+ ij = FIX2INT(argv[1]);
194
+ in2 = FIX2INT(argv[2]);
195
+ if(ij < 0) ij += size2;
196
+ *i = (size_t)ii;
197
+ *j = (size_t) ij;
198
+ *n2 = (size_t) in2;
199
+ // Fixnum, Fixnum, ...
200
+ } else {
201
+ CHECK_FIXNUM(argv[0]); CHECK_FIXNUM(argv[1]);
202
+ ii = FIX2INT(argv[0]);
203
+ if(ii < 0) ii += size1;
204
+ in1 = FIX2INT(argv[1]);
205
+ *i = (size_t)ii;
206
+ *n1 = (size_t)in1;
207
+ // Parse arg2
208
+ if(NIL_P(argv[2])) {
209
+ // Fixnum, Fixnum, nil -> Some rows, all cols
210
+ *j = 0;
211
+ *n2 = size2;
212
+ } else if(rb_obj_is_kind_of(argv[2], rb_cRange)) {
213
+ // Fixnum, Fixnum, Range -> Some rows, Range cols
214
+ get_range_beg_en_n_for_size(argv[2], &ij, &end, n2, &step, size2);
215
+ if(step < 0 || *n2 <= 0) {
216
+ rb_raise(rb_eRangeError, "arg2: begin > end");
217
+ }
218
+ *j = (size_t)ij;
219
+ } else {
220
+ rb_raise(rb_eArgError, "expected third argument to be nil or Range, not %s",
221
+ rb_class2name(CLASS_OF(argv[2])));
222
+ }
223
+ }
224
+ break;
225
+
226
+ case 4:
227
+ CHECK_FIXNUM(argv[0]); CHECK_FIXNUM(argv[1]);
228
+ CHECK_FIXNUM(argv[2]); CHECK_FIXNUM(argv[3]);
229
+ ii = FIX2INT(argv[0]); ij = FIX2INT(argv[1]);
230
+ in1 = FIX2INT(argv[2]); in2 = FIX2INT(argv[3]);
231
+ if(ii < 0) ii += size1;
232
+ if(ij < 0) ij += size2;
233
+ // TODO Bounds check?
234
+ *i = (size_t)ii; *j = (size_t)ij;
235
+ *n1 = (size_t)in1; *n2 = (size_t)in2;
236
+ break;
237
+ default:
238
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 0 to 4)", argc);
239
+ break;
240
+ }
241
+ }
242
+ #endif
243
+
244
+ VALUE FUNCTION(rb_gsl_matrix,do_something)(VALUE obj, void (*f)(GSL_TYPE(gsl_matrix) *))
245
+ {
246
+ GSL_TYPE(gsl_matrix) *m = NULL;
247
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
248
+ (*f)(m);
249
+ return obj;
250
+ }
251
+
252
+ static VALUE FUNCTION(create_matrix,from_range_shape)(VALUE range, VALUE nn1, VALUE nn2);
253
+ static VALUE FUNCTION(create_matrix,from_ranges)(int argc, VALUE *argv);
254
+ GSL_TYPE(gsl_matrix)* FUNCTION(gsl_matrix,alloc_from_colvectors)(int argc, VALUE *argv);
255
+
256
+ static VALUE FUNCTION(rb_gsl_matrix,alloc)(int argc, VALUE *argv, VALUE klass)
257
+ {
258
+ GSL_TYPE(gsl_matrix) *m = NULL;
259
+ size_t n1, n2;
260
+ #ifdef HAVE_NARRAY_H
261
+ size_t n;
262
+ VALUE ary;
263
+ struct NARRAY *na;
264
+ #endif
265
+ if (argc < 1) rb_raise(rb_eArgError,
266
+ "wrong number of arguments (%d for >= 1)", argc);
267
+
268
+ #ifdef HAVE_NARRAY_H
269
+ if (NA_IsNArray(argv[0])) {
270
+ GetNArray(argv[0], na);
271
+ n = na->shape[0]*na->shape[1];
272
+ m = FUNCTION(gsl_matrix,alloc)(na->shape[1], na->shape[0]);
273
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_alloc failed");
274
+ #ifdef BASE_DOUBLE
275
+ ary = na_change_type(argv[0], NA_DFLOAT);
276
+ #else
277
+ ary = na_change_type(argv[0], NA_LINT);
278
+ #endif
279
+ memcpy(m->data, NA_PTR_TYPE(ary,BASE*), n*sizeof(BASE));
280
+ return Data_Wrap_Struct(klass, 0, FUNCTION(gsl_matrix,free), m);
281
+ }
282
+ #endif
283
+ switch (TYPE(argv[0])) {
284
+ case T_FIXNUM:
285
+ if (argc != 2) rb_raise(rb_eArgError, "wrong number of arguments (%d for 2)",
286
+ argc);
287
+ CHECK_FIXNUM(argv[1]);
288
+ n1 = FIX2INT(argv[0]); n2 = FIX2INT(argv[1]);
289
+ m = FUNCTION(gsl_matrix,calloc)(n1, n2);
290
+ break;
291
+ case T_ARRAY:
292
+ if (argc == 1) {
293
+ m = FUNCTION(gsl_matrix,alloc_from_arrays)(argc, argv);
294
+ break;
295
+ }
296
+ if (CLASS_OF(argv[1]) == rb_cRange) argv[1] = rb_gsl_range2ary(argv[1]);
297
+ switch (TYPE(argv[1])) {
298
+ case T_ARRAY:
299
+ m = FUNCTION(gsl_matrix,alloc_from_arrays)(argc, argv);
300
+ break;
301
+ case T_FIXNUM:
302
+ if (argc != 3) rb_raise(rb_eArgError, "wrong number of arguments (%d for 3)",
303
+ argc);
304
+ CHECK_FIXNUM(argv[2]);
305
+ m = FUNCTION(gsl_matrix,alloc_from_array_sizes)(argv[0], argv[1], argv[2]);
306
+ break;
307
+ default:
308
+ rb_raise(rb_eTypeError,
309
+ "wrong argument type %s\nUsage: new(n1, n2), "
310
+ "new([], [], [], ...), new([], n1, n2)",
311
+ rb_class2name(CLASS_OF(argv[1])));
312
+ break;
313
+ }
314
+ break;
315
+ default:
316
+ if (CLASS_OF(argv[0]) == rb_cRange) {
317
+ if (argc==3 && TYPE(argv[1]) == T_FIXNUM && TYPE(argv[2]) == T_FIXNUM)
318
+ return FUNCTION(create_matrix,from_range_shape)(argv[0], argv[1], argv[2]);
319
+ else
320
+ return FUNCTION(create_matrix,from_ranges)(argc, argv);
321
+ } else if (VEC_P(argv[0])) {
322
+ if (argc == 3 && FIXNUM_P(argv[1]) && FIXNUM_P(argv[2])) {
323
+ m = FUNCTION(gsl_matrix,alloc_from_vector_sizes)(argv[0], argv[1], argv[2]);
324
+ } else {
325
+ if (VEC_COL_P(argv[0])) {
326
+ m = FUNCTION(gsl_matrix,alloc_from_colvectors)(argc, argv);
327
+ } else {
328
+ m = FUNCTION(gsl_matrix,alloc_from_vectors)(argc, argv);
329
+ }
330
+ }
331
+ } else {
332
+ rb_raise(rb_eTypeError, "wrong argument type %s\n"
333
+ "Usage: new(n1, n2), new([], [], [], ...), new([], n1, n2)",
334
+ rb_class2name(CLASS_OF(argv[0])));
335
+ }
336
+ break;
337
+ }
338
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), m);
339
+ }
340
+
341
+ void FUNCTION(set_ptr_data,by_range)(BASE *ptr, size_t n, VALUE range);
342
+
343
+ static GSL_TYPE(gsl_matrix)* FUNCTION(cr_matrix,from_ranges)(int argc, VALUE *argv)
344
+ {
345
+ GSL_TYPE(gsl_matrix) *m;
346
+ BASE beg, en;
347
+ size_t i, n;
348
+ int step;
349
+ FUNCTION(get_range,beg_en_n)(argv[0], &beg, &en, &n, &step);
350
+ m = FUNCTION(gsl_matrix,calloc)(argc, n);
351
+ FUNCTION(set_ptr_data,by_range)(m->data, n, argv[0]);
352
+ for (i = 1; i < argc; i++) {
353
+ if (CLASS_OF(argv[i]) != rb_cRange)
354
+ rb_raise(rb_eTypeError, "wrong argument type %s (Range expected)",
355
+ rb_class2name(CLASS_OF(argv[i])));
356
+ FUNCTION(set_ptr_data,by_range)(m->data+i*n, n, argv[i]);
357
+ }
358
+ return m;
359
+ }
360
+
361
+ static VALUE FUNCTION(create_matrix,from_ranges)(int argc, VALUE *argv)
362
+ {
363
+ GSL_TYPE(gsl_matrix) *m;
364
+ m = FUNCTION(cr_matrix,from_ranges)(argc, argv);
365
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), m);
366
+ }
367
+
368
+ static GSL_TYPE(gsl_matrix)* FUNCTION(cr_matrix,from_range_shape)(VALUE range, VALUE nn1, VALUE nn2)
369
+ {
370
+ size_t n1, n2;
371
+ GSL_TYPE(gsl_matrix) *m;
372
+ n1 = FIX2INT(nn1); n2 = FIX2INT(nn2);
373
+ m = FUNCTION(gsl_matrix,alloc)(n1, n2);
374
+ FUNCTION(set_ptr_data,by_range)(m->data, n1*n2, range);
375
+ return m;
376
+ }
377
+
378
+ static VALUE FUNCTION(create_matrix,from_range_shape)(VALUE range, VALUE nn1, VALUE nn2)
379
+ {
380
+ GSL_TYPE(gsl_matrix) *m;
381
+ m = FUNCTION(cr_matrix,from_range_shape)(range, nn1, nn2);
382
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), m);
383
+ }
384
+
385
+ /* create a matrix from arrays */
386
+ GSL_TYPE(gsl_matrix)* FUNCTION(gsl_matrix,alloc_from_arrays)(int argc, VALUE *argv)
387
+ {
388
+ size_t n, i, j;
389
+ GSL_TYPE(gsl_matrix) *m = NULL;
390
+ if (CLASS_OF(argv[0]) == rb_cRange) argv[0] = rb_gsl_range2ary(argv[0]);
391
+ else Check_Type(argv[0], T_ARRAY);
392
+ n = RARRAY_LEN(argv[0]);
393
+ m = FUNCTION(gsl_matrix,alloc)(argc, n);
394
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_alloc failed");
395
+ for (i = 0; i < argc; i++) {
396
+ if (CLASS_OF(argv[i]) == rb_cRange) argv[i] = rb_gsl_range2ary(argv[i]);
397
+ else Check_Type(argv[i], T_ARRAY);
398
+ for (j = 0; j < n; j++) {
399
+ if (j >= RARRAY_LEN(argv[i])) FUNCTION(gsl_matrix,set)(m, i, j, 0);
400
+ else FUNCTION(gsl_matrix,set)(m, i, j, NUMCONV2(rb_ary_entry(argv[i], j)));
401
+ }
402
+ }
403
+ return m;
404
+ }
405
+
406
+ GSL_TYPE(gsl_matrix)* FUNCTION(gsl_matrix,alloc_from_vectors)(int argc, VALUE *argv)
407
+ {
408
+ size_t i;
409
+ GSL_TYPE(gsl_matrix) *m = NULL;
410
+ GSL_TYPE(gsl_vector) *v = NULL;
411
+ if (argc < 1) rb_raise(rb_eArgError, "too few arguments");
412
+ CHECK_VEC(argv[0]);
413
+ Data_Get_Struct(argv[0], GSL_TYPE(gsl_vector), v);
414
+ m = FUNCTION(gsl_matrix,alloc)(argc, v->size);
415
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_alloc failed");
416
+ for (i = 0; i < argc; i++) {
417
+ CHECK_VEC(argv[i]);
418
+ Data_Get_Struct(argv[i], GSL_TYPE(gsl_vector), v);
419
+ FUNCTION(gsl_matrix,set_row)(m, i, v);
420
+ }
421
+ return m;
422
+ }
423
+
424
+ GSL_TYPE(gsl_matrix)* FUNCTION(gsl_matrix,alloc_from_colvectors)(int argc, VALUE *argv)
425
+ {
426
+ size_t i;
427
+ GSL_TYPE(gsl_matrix) *m = NULL;
428
+ GSL_TYPE(gsl_vector) *v = NULL;
429
+ if (argc < 1) rb_raise(rb_eArgError, "too few arguments");
430
+ CHECK_VEC(argv[0]);
431
+ Data_Get_Struct(argv[0], GSL_TYPE(gsl_vector), v);
432
+ // m = FUNCTION(gsl_matrix,alloc)(argc, v->size);
433
+ m = FUNCTION(gsl_matrix,alloc)(v->size, argc);
434
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_alloc failed");
435
+ for (i = 0; i < argc; i++) {
436
+ CHECK_VEC(argv[i]);
437
+ Data_Get_Struct(argv[i], GSL_TYPE(gsl_vector), v);
438
+ FUNCTION(gsl_matrix,set_col)(m, i, v);
439
+ }
440
+ return m;
441
+ }
442
+
443
+ /* create a matrix from two sizes and an array */
444
+ GSL_TYPE(gsl_matrix)* FUNCTION(gsl_matrix,alloc_from_array_sizes)(VALUE ary,
445
+ VALUE nn1, VALUE nn2)
446
+ {
447
+ size_t n1, n2, len;
448
+ GSL_TYPE(gsl_matrix) *m = NULL;
449
+ size_t i, j, k;
450
+ CHECK_FIXNUM(nn1); CHECK_FIXNUM(nn2);
451
+ Check_Type(ary, T_ARRAY);
452
+ n1 = FIX2INT(nn1);
453
+ n2 = FIX2INT(nn2);
454
+ m = FUNCTION(gsl_matrix,alloc)(n1, n2);
455
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_alloc failed");
456
+ k = 0;
457
+ len = RARRAY_LEN(ary);
458
+ for (i = 0; i < n1; i++) {
459
+ for (j = 0; j < n2; j++, k++) {
460
+ if (k >= len)
461
+ FUNCTION(gsl_matrix,set)(m, i, j, (BASE) 0);
462
+ else
463
+ FUNCTION(gsl_matrix,set)(m, i, j, (BASE) NUMCONV2(rb_ary_entry(ary, k)));
464
+
465
+ }
466
+ }
467
+ return m;
468
+ }
469
+
470
+ GSL_TYPE(gsl_matrix)* FUNCTION(gsl_matrix,alloc_from_vector_sizes)(VALUE ary,
471
+ VALUE nn1, VALUE nn2)
472
+ {
473
+ size_t n1, n2;
474
+ GSL_TYPE(gsl_matrix) *m = NULL;
475
+ GSL_TYPE(gsl_vector) *v = NULL;
476
+ size_t i, j, k;
477
+ CHECK_VEC(ary);
478
+ CHECK_FIXNUM(nn1); CHECK_FIXNUM(nn2);
479
+ Data_Get_Struct(ary, GSL_TYPE(gsl_vector), v);
480
+ n1 = FIX2INT(nn1); n2 = FIX2INT(nn2);
481
+ m = FUNCTION(gsl_matrix,alloc)(n1, n2);
482
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_alloc failed");
483
+ k = 0;
484
+ for (i = 0; i < n1; i++) {
485
+ for (j = 0; j < n2; j++, k++) {
486
+ if (k >= v->size)
487
+ FUNCTION(gsl_matrix,set)(m, i, j, (BASE) 0);
488
+ else
489
+ FUNCTION(gsl_matrix,set)(m, i, j, FUNCTION(gsl_vector,get)(v, k));
490
+ }
491
+ }
492
+ return m;
493
+ }
494
+
495
+ static VALUE FUNCTION(rb_gsl_matrix,calloc)(VALUE klass, VALUE nn1, VALUE nn2)
496
+ {
497
+ GSL_TYPE(gsl_matrix) *m = NULL;
498
+ CHECK_FIXNUM(nn1); CHECK_FIXNUM(nn2);
499
+ m = FUNCTION(gsl_matrix,calloc)(FIX2INT(nn1), FIX2INT(nn2));
500
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_calloc failed");
501
+ return Data_Wrap_Struct(klass, 0, FUNCTION(gsl_matrix,free), m);
502
+ }
503
+
504
+ static VALUE FUNCTION(rb_gsl_matrix,diagonal_singleton)(int argc, VALUE *argv, VALUE klass)
505
+ {
506
+ GSL_TYPE(gsl_matrix) *m = NULL;
507
+ GSL_TYPE(gsl_vector) *v = NULL;
508
+ VALUE ary, tmp;
509
+ size_t len, i;
510
+ switch (argc) {
511
+ case 1:
512
+ switch (TYPE(argv[0])) {
513
+ case T_FIXNUM:
514
+ case T_FLOAT:
515
+ len = FIX2INT(argv[0]);
516
+ m = FUNCTION(gsl_matrix,alloc)(len, len);
517
+ for (i = 0; i < len; i++)
518
+ FUNCTION(gsl_matrix,set)(m, i, i, 1);
519
+ return Data_Wrap_Struct(klass, 0, FUNCTION(gsl_matrix,free), m);
520
+ break;
521
+ default:
522
+ /* do next */
523
+ break;
524
+ }
525
+ if (rb_obj_is_kind_of(argv[0], rb_cRange))
526
+ ary = rb_gsl_range2ary(argv[0]);
527
+ else ary = argv[0];
528
+ switch (TYPE(ary)) {
529
+ case T_ARRAY:
530
+ len = RARRAY_LEN(ary);
531
+ m = FUNCTION(gsl_matrix,calloc)(len, len);
532
+ for (i = 0; i < len; i++) {
533
+ tmp = rb_ary_entry(ary, i);
534
+ FUNCTION(gsl_matrix,set)(m, i, i, NUMCONV2(tmp));
535
+ }
536
+ break;
537
+ default:
538
+ CHECK_VEC(ary);
539
+ Data_Get_Struct(ary, GSL_TYPE(gsl_vector), v);
540
+ len = v->size;
541
+ m = FUNCTION(gsl_matrix,calloc)(len, len);
542
+ for (i = 0; i < len; i++) {
543
+ FUNCTION(gsl_matrix,set)(m, i, i, FUNCTION(gsl_vector,get)(v, i));
544
+ }
545
+ break;
546
+ }
547
+ break;
548
+ default:
549
+ m = FUNCTION(gsl_matrix,calloc)(argc, argc);
550
+ for (i = 0; i < argc; i++) {
551
+ FUNCTION(gsl_matrix,set)(m, i, i, NUMCONV2(argv[i]));
552
+ }
553
+ break;
554
+ }
555
+ return Data_Wrap_Struct(klass, 0, FUNCTION(gsl_matrix,free), m);
556
+ }
557
+
558
+ static VALUE FUNCTION(rb_gsl_matrix,eye)(int argc, VALUE *argv, VALUE klass)
559
+ {
560
+ GSL_TYPE(gsl_matrix) *m = NULL;
561
+ size_t n1, n2, n, i;
562
+ switch (argc) {
563
+ case 1:
564
+ CHECK_FIXNUM(argv[0]);
565
+ n = FIX2INT(argv[0]);
566
+ n1 = n2 = n;
567
+ break;
568
+ case 2:
569
+ CHECK_FIXNUM(argv[0]); CHECK_FIXNUM(argv[1]);
570
+ n1 = FIX2INT(argv[0]); n2 = FIX2INT(argv[1]);
571
+ n = GSL_MIN_INT(n1, n2);
572
+ break;
573
+ default:
574
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 1 or 2)", argc);
575
+ break;
576
+ }
577
+ m = FUNCTION(gsl_matrix,calloc)(n1, n2);
578
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_calloc failed");
579
+ for (i = 0; i < n; i++) {
580
+ FUNCTION(gsl_matrix,set)(m, i, i, 1);
581
+ }
582
+ return Data_Wrap_Struct(klass, 0, FUNCTION(gsl_matrix,free), m);
583
+ }
584
+
585
+ static VALUE FUNCTION(rb_gsl_matrix,ones)(int argc, VALUE *argv, VALUE klass)
586
+ {
587
+ GSL_TYPE(gsl_matrix) *m = NULL;
588
+ size_t n1, n2, n, i, j;
589
+ switch (argc) {
590
+ case 1:
591
+ CHECK_FIXNUM(argv[0]);
592
+ n = FIX2INT(argv[0]);
593
+ n1 = n2 = n;
594
+ break;
595
+ case 2:
596
+ CHECK_FIXNUM(argv[0]); CHECK_FIXNUM(argv[1]);
597
+ n1 = FIX2INT(argv[0]); n2 = FIX2INT(argv[1]);
598
+ n = GSL_MIN_INT(n1, n2);
599
+ break;
600
+ default:
601
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 1 or 2)", argc);
602
+ break;
603
+ }
604
+ m = FUNCTION(gsl_matrix,calloc)(n1, n2);
605
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_calloc failed");
606
+ for (i = 0; i < n1; i++) {
607
+ for (j = 0; j < n2; j++) {
608
+ FUNCTION(gsl_matrix,set)(m, i, j, (BASE) 1);
609
+ }
610
+ }
611
+ return Data_Wrap_Struct(klass, 0, FUNCTION(gsl_matrix,free), m);
612
+ }
613
+
614
+ static VALUE FUNCTION(rb_gsl_matrix,zeros)(int argc, VALUE *argv, VALUE klass)
615
+ {
616
+ GSL_TYPE(gsl_matrix) *m = NULL;
617
+ size_t n1, n2, n, i, j;
618
+ switch (argc) {
619
+ case 1:
620
+ CHECK_FIXNUM(argv[0]);
621
+ n = FIX2INT(argv[0]);
622
+ n1 = n2 = n;
623
+ break;
624
+ case 2:
625
+ CHECK_FIXNUM(argv[0]); CHECK_FIXNUM(argv[1]);
626
+ n1 = FIX2INT(argv[0]); n2 = FIX2INT(argv[1]);
627
+ n = GSL_MIN_INT(n1, n2);
628
+ break;
629
+ default:
630
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 1 or 2)", argc);
631
+ break;
632
+ }
633
+ m = FUNCTION(gsl_matrix,calloc)(n1, n2);
634
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_calloc failed");
635
+ for (i = 0; i < n1; i++) {
636
+ for (j = 0; j < n2; j++) {
637
+ FUNCTION(gsl_matrix,set)(m, i, j, (BASE) 0);
638
+ }
639
+ }
640
+ return Data_Wrap_Struct(klass, 0, FUNCTION(gsl_matrix,free), m);
641
+ }
642
+
643
+ static VALUE FUNCTION(rb_gsl_matrix,identity)(VALUE klass, VALUE nn)
644
+ {
645
+ GSL_TYPE(gsl_matrix) *m = NULL;
646
+ size_t n, i;
647
+ CHECK_FIXNUM(nn);
648
+ n = FIX2INT(nn);
649
+ m = FUNCTION(gsl_matrix,calloc)(n, n);
650
+ if (m == NULL) rb_raise(rb_eNoMemError, "gsl_matrix_calloc failed");
651
+ for (i = 0; i < n; i++) {
652
+ FUNCTION(gsl_matrix,set)(m, i, i, 1);
653
+ }
654
+ return Data_Wrap_Struct(klass, 0, FUNCTION(gsl_matrix,free), m);
655
+ }
656
+
657
+ static VALUE FUNCTION(rb_gsl_matrix,size1)(VALUE obj)
658
+ {
659
+ GSL_TYPE(gsl_matrix) *m = NULL;
660
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
661
+ return INT2FIX(m->size1);
662
+ }
663
+
664
+ static VALUE FUNCTION(rb_gsl_matrix,size2)(VALUE obj)
665
+ {
666
+ GSL_TYPE(gsl_matrix) *m = NULL;
667
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
668
+ return INT2FIX(m->size2);
669
+ }
670
+
671
+ static VALUE FUNCTION(rb_gsl_matrix,shape)(VALUE obj)
672
+ {
673
+ GSL_TYPE(gsl_matrix) *m = NULL;
674
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
675
+ return rb_ary_new3(2, INT2FIX(m->size1), INT2FIX(m->size2));
676
+ }
677
+
678
+ static VALUE FUNCTION(rb_gsl_matrix,submatrix)(int argc, VALUE *argv, VALUE obj);
679
+ static VALUE FUNCTION(rb_gsl_matrix,get)(int argc, VALUE *argv, VALUE obj)
680
+ {
681
+ GSL_TYPE(gsl_matrix) *m = NULL;
682
+ VALUE retval;
683
+ int ii, ij;
684
+
685
+ if(argc == 2 && TYPE(argv[0]) == T_FIXNUM && TYPE(argv[1]) == T_FIXNUM) {
686
+ // m[i,j]
687
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
688
+ ii = FIX2INT(argv[0]);
689
+ ij = FIX2INT(argv[1]);
690
+ if(ii < 0) ii += m->size1;
691
+ if(ij < 0) ij += m->size2;
692
+ retval = C_TO_VALUE2(FUNCTION(gsl_matrix,get)(m, (size_t)ii, (size_t)ij));
693
+ } else if(argc == 1 && TYPE(argv[0]) == T_FIXNUM) {
694
+ // m[i]
695
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
696
+ ii = FIX2INT(argv[0]);
697
+ if(ii < 0) ii += m->size1 * m->size2;
698
+ retval = C_TO_VALUE2(FUNCTION(gsl_matrix,get)(m, (size_t)(ii / m->size2), (size_t)(ii % m->size2)));
699
+ } else if(argc == 1 && TYPE(argv[0]) == T_ARRAY) {
700
+ // m[[i,j]], to support m[m.max_index]
701
+ if(RARRAY_LEN(argv[0]) == 2) {
702
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
703
+ ii = FIX2INT(RARRAY_PTR(argv[0])[0]);
704
+ ij = FIX2INT(RARRAY_PTR(argv[0])[1]);
705
+ if(ii < 0) ii += m->size1;
706
+ if(ij < 0) ij += m->size2;
707
+ retval = C_TO_VALUE2(FUNCTION(gsl_matrix,get)(m, (size_t)ii, (size_t)ij));
708
+ } else {
709
+ rb_raise(rb_eArgError, "Array index must have length 2, not %d", (int) RARRAY_LEN(argv[0]));
710
+ }
711
+ } else {
712
+ retval = FUNCTION(rb_gsl_matrix,submatrix)(argc, argv, obj);
713
+ }
714
+
715
+ return retval;
716
+ }
717
+
718
+ void FUNCTION(rb_gsl_vector,set_subvector)(int argc, VALUE *argv, GSL_TYPE(gsl_vector) *v, VALUE other);
719
+ static VALUE FUNCTION(rb_gsl_matrix,set)(int argc, VALUE *argv, VALUE obj)
720
+ {
721
+ GSL_TYPE(gsl_matrix) *m, *mother;
722
+ QUALIFIED_VIEW(gsl_matrix,view) mv;
723
+ QUALIFIED_VIEW(gsl_vector,view) vv;
724
+ VALUE other, row, row_set_argv[2];
725
+ int ii, ij, step;
726
+ size_t i, j, k, n1, n2, nother;
727
+ BASE beg, end;
728
+
729
+ if(argc < 1 || argc > 5) {
730
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 1-5)", argc);
731
+ }
732
+
733
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
734
+ other = argv[argc-1];
735
+
736
+ if(argc == 1 && TYPE(argv[0]) == T_ARRAY) {
737
+ // m.set([row0,row1,...])
738
+ n1 = RARRAY_LEN(argv[0]);
739
+ if(n1 > m->size1) n1 = m->size1;
740
+ row_set_argv[0] = INT2FIX(0);
741
+ // Each given row must have as manay elements as m has columns.
742
+ // The bounds check happens inside rb_gsl_vector*set_subvector().
743
+ row_set_argv[1] = INT2FIX(m->size2);
744
+ for(k = 0; k < n1 && k < m->size1; k++) {
745
+ vv = FUNCTION(gsl_matrix,row)(m, k);
746
+ FUNCTION(rb_gsl_vector,set_subvector)(2, row_set_argv, &vv.vector, rb_ary_entry(argv[0], k));
747
+ }
748
+ } else if(argc == 1) {
749
+ // m[] = x
750
+ FUNCTION(gsl_matrix,set_all)(m, NUMCONV2(other));
751
+ } else if(argc==2 && TYPE(argv[0]) == T_ARRAY && TYPE(argv[1]) != T_ARRAY) {
752
+ // m.set([i, j], x) or m[[i,j]] = x
753
+ ii = FIX2INT(rb_ary_entry(argv[0], 0));
754
+ ij = FIX2INT(rb_ary_entry(argv[0], 1));
755
+ if(ii < 0) ii += m->size1;
756
+ if(ij < 0) ij += m->size2;
757
+ FUNCTION(gsl_matrix,set)(m, (size_t)ii, (size_t)ij, NUMCONV2(argv[1]));
758
+ } else if(argc == 3 && TYPE(argv[0]) == T_FIXNUM && TYPE(argv[1]) == T_FIXNUM) {
759
+ // m[i,j] = x
760
+ ii = FIX2INT(argv[0]);
761
+ ij = FIX2INT(argv[1]);
762
+ if(ii < 0) ii += m->size1;
763
+ if(ij < 0) ij += m->size2;
764
+ FUNCTION(gsl_matrix,set)(m, (size_t)ii, (size_t)ij, NUMCONV2(other));
765
+ } else if(TYPE(argv[0]) == T_ARRAY) {
766
+ // m.set(row0,row1,...)
767
+ n1 = argc;
768
+ if(n1 > m->size1) n1 = m->size1;
769
+ row_set_argv[0] = INT2FIX(0);
770
+ row_set_argv[1] = INT2FIX(m->size2);
771
+ for(k = 0; k < n1 && k < m->size1; k++) {
772
+ vv = FUNCTION(gsl_matrix,row)(m, k);
773
+ FUNCTION(rb_gsl_vector,set_subvector)(2, row_set_argv, &vv.vector, argv[k]);
774
+ }
775
+ } else {
776
+ // x -> assignment to m.submatrix(i...)
777
+ parse_submatrix_args(argc-1, argv, m->size1, m->size2, &i, &j, &n1, &n2);
778
+ if(n1 == 0) n1 = 1;
779
+ if(n2 == 0) n2 = 1;
780
+ mv = FUNCTION(gsl_matrix,submatrix)(m, i, j, n1, n2);
781
+ if(rb_obj_is_kind_of(other, GSL_TYPE(cgsl_matrix))) {
782
+ // m[...] = m_other
783
+ Data_Get_Struct(other, GSL_TYPE(gsl_matrix), mother);
784
+ if(n1 * n2 != mother->size1 * mother->size2) {
785
+ rb_raise(rb_eRangeError, "sizes do not match (%d x %d != %d x %d)",
786
+ (int) n1, (int) n2, (int) mother->size1, (int) mother->size2);
787
+ }
788
+ // TODO Change to gsl_matrix_memmove if/when GSL has such a function
789
+ // because gsl_matrix_memcpy does not handle overlapping regions (e.g.
790
+ // Views) well.
791
+ FUNCTION(gsl_matrix,memcpy)(&mv.matrix, mother);
792
+ } else if(rb_obj_is_kind_of(other, rb_cArray)) {
793
+ row_set_argv[0] = INT2FIX(0);
794
+ row_set_argv[1] = INT2FIX(n2);
795
+
796
+ if(n1 == 1) {
797
+ // m[...] = [col0, ...] # single row
798
+ vv = FUNCTION(gsl_matrix,row)(&mv.matrix, 0);
799
+ FUNCTION(rb_gsl_vector,set_subvector)(2, row_set_argv, &vv.vector, other);
800
+ } else {
801
+ // m[...] = [[row0], [row1], ...] # multiple rows
802
+ if(n1 != RARRAY_LEN(other)) {
803
+ rb_raise(rb_eRangeError, "row counts do not match (%d != %d)",
804
+ (int) n1, (int) RARRAY_LEN(other));
805
+ }
806
+ for(k = 0; k < n1; k++) {
807
+ vv = FUNCTION(gsl_matrix,row)(&mv.matrix, k);
808
+ row = rb_ary_entry(other, k);
809
+ FUNCTION(rb_gsl_vector,set_subvector)(2, row_set_argv, &vv.vector, row);
810
+ }
811
+ }
812
+ } else if(rb_obj_is_kind_of(other, rb_cRange)) {
813
+ // m[...] = beg..end
814
+ FUNCTION(get_range,beg_en_n)(other, &beg, &end, &nother, &step);
815
+ if(n1 * n2 != nother) {
816
+ rb_raise(rb_eRangeError, "sizes do not match (%d x %d != %d)", (int) n1, (int) n2, (int) nother);
817
+ }
818
+ for(k = 0; k < nother; k++) {
819
+ FUNCTION(gsl_matrix,set)(&mv.matrix, k / n2, k % n2, beg);
820
+ beg += step;
821
+ }
822
+ } else {
823
+ // m[...] = x
824
+ FUNCTION(gsl_matrix,set_all)(&mv.matrix, NUMCONV2(other));
825
+ }
826
+ }
827
+
828
+ return obj;
829
+ }
830
+
831
+ static VALUE FUNCTION(rb_gsl_matrix,set_all)(VALUE obj, VALUE x)
832
+ {
833
+ GSL_TYPE(gsl_matrix) *m = NULL;
834
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
835
+ FUNCTION(gsl_matrix,set_all)(m, NUMCONV2(x));
836
+ return obj;
837
+ }
838
+
839
+ static VALUE FUNCTION(rb_gsl_matrix,set_zero)(VALUE obj)
840
+ {
841
+ return FUNCTION(rb_gsl_matrix,do_something)(obj, FUNCTION(gsl_matrix,set_zero));
842
+ }
843
+
844
+ static VALUE FUNCTION(rb_gsl_matrix,set_identity)(VALUE obj)
845
+ {
846
+ return FUNCTION(rb_gsl_matrix,do_something)(obj, FUNCTION(gsl_matrix,set_identity));
847
+ }
848
+
849
+ static VALUE FUNCTION(rb_gsl_matrix,print)(VALUE obj)
850
+ {
851
+ GSL_TYPE(gsl_matrix) *m = NULL;
852
+ size_t i, j;
853
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
854
+ printf("[ ");
855
+ for (i = 0; i < m->size1; i++) {
856
+ if (i != 0) printf(" ");
857
+ for (j = 0; j < m->size2; j++) {
858
+ printf(PRINTF_FORMAT, FUNCTION(gsl_matrix,get)(m, i, j));
859
+ }
860
+ if (i == m->size1 - 1) printf("]\n");
861
+ else printf("\n");
862
+ }
863
+ return Qnil;
864
+ }
865
+
866
+ #ifdef BASE_DOUBLE
867
+ #define SHOW_ELM 6
868
+ #else
869
+ #define SHOW_ELM 12
870
+ #endif
871
+
872
+ static VALUE FUNCTION(rb_gsl_matrix,to_s)(VALUE obj)
873
+ {
874
+ GSL_TYPE(gsl_matrix) *m = NULL;
875
+ char buf[32], format[32], format2[32];
876
+ size_t i, j;
877
+ VALUE str;
878
+ BASE x, min;
879
+ int dig = 8;
880
+ #ifdef BASE_INT
881
+ BASE max;
882
+ #endif
883
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
884
+ min = FUNCTION(gsl_matrix,min)(m);
885
+ #ifdef BASE_INT
886
+ max = gsl_matrix_int_max(m);
887
+ dig = (int) GSL_MAX(fabs(max),fabs(min));
888
+ if (dig > 0) dig = ceil(log10(dig+1e-10));
889
+ else dig = 1;
890
+ if (min < 0) dig += 1;
891
+ sprintf(format, "%%%dd ", (int) dig);
892
+ strcpy(format2, format);
893
+ #else
894
+ strcpy(format, PRINTF_FORMAT);
895
+ strcpy(format2, " "PRINTF_FORMAT);
896
+ #endif
897
+ str = rb_str_new2("[ ");
898
+ for (i = 0; i < m->size1; i++) {
899
+ if (i != 0) {
900
+ strcpy(buf, " ");
901
+ rb_str_cat(str, buf, strlen(buf));
902
+ }
903
+ for (j = 0; j < m->size2; j++) {
904
+ x = FUNCTION(gsl_matrix,get)(m, i, j);
905
+ if (x < 0)
906
+ sprintf(buf, format, x);
907
+ else
908
+ sprintf(buf, format2, x);
909
+ rb_str_cat(str, buf, strlen(buf));
910
+ if (j >= (55/dig)) {
911
+ strcpy(buf, "... ");
912
+ rb_str_cat(str, buf, strlen(buf));
913
+ break;
914
+ }
915
+ }
916
+ if (i >= 20) {
917
+ strcpy(buf, "\n ... ]");
918
+ rb_str_cat(str, buf, strlen(buf));
919
+ break;
920
+ }
921
+ if (i == m->size1 - 1) {
922
+ strcpy(buf, "]");
923
+ rb_str_cat(str, buf, strlen(buf));
924
+ } else {
925
+ strcpy(buf, "\n");
926
+ rb_str_cat(str, buf, strlen(buf));
927
+ }
928
+ }
929
+ return str;
930
+ }
931
+ #undef SHOW_ELM
932
+
933
+ static VALUE FUNCTION(rb_gsl_matrix,inspect)(VALUE obj)
934
+ {
935
+ VALUE str;
936
+ char buf[64];
937
+ sprintf(buf, "%s\n", rb_class2name(CLASS_OF(obj)));
938
+ str = rb_str_new2(buf);
939
+ return rb_str_concat(str, FUNCTION(rb_gsl_matrix,to_s)(obj));
940
+ }
941
+
942
+ #ifdef BASE_DOUBLE
943
+ #define PRINTF_FORMAT2 "%g"
944
+ #elif defined(BASE_INT)
945
+ #define PRINTF_FORMAT2 "%d"
946
+ #endif
947
+ static VALUE FUNCTION(rb_gsl_matrix,fprintf)(int argc, VALUE *argv, VALUE obj)
948
+ {
949
+ GSL_TYPE(gsl_matrix) *h = NULL;
950
+ FILE *fp = NULL;
951
+ int status, flag = 0;
952
+ if (argc != 1 && argc != 2) {
953
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 1 or 2)", argc);
954
+ }
955
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), h);
956
+ fp = rb_gsl_open_writefile(argv[0], &flag);
957
+ if (argc == 2) {
958
+ Check_Type(argv[1], T_STRING);
959
+ status = FUNCTION(gsl_matrix,fprintf)(fp, h, STR2CSTR(argv[1]));
960
+ } else {
961
+ status = FUNCTION(gsl_matrix,fprintf)(fp, h, PRINTF_FORMAT2);
962
+ }
963
+ if (flag == 1) fclose(fp);
964
+ return INT2FIX(status);
965
+ }
966
+
967
+ static VALUE FUNCTION(rb_gsl_matrix,printf)(int argc, VALUE *argv, VALUE obj)
968
+ {
969
+ GSL_TYPE(gsl_matrix) *h = NULL;
970
+ int status;
971
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), h);
972
+ if (argc == 1) {
973
+ Check_Type(argv[0], T_STRING);
974
+ status = FUNCTION(gsl_matrix,fprintf)(stdout, h, STR2CSTR(argv[0]));
975
+ } else {
976
+ status = FUNCTION(gsl_matrix,fprintf)(stdout, h, PRINTF_FORMAT2);
977
+ }
978
+ return INT2FIX(status);
979
+ }
980
+
981
+ #undef PRINTF_FORMAT2
982
+ static VALUE FUNCTION(rb_gsl_matrix,fscanf)(VALUE obj, VALUE io)
983
+ {
984
+ GSL_TYPE(gsl_matrix) *h = NULL;
985
+ FILE *fp = NULL;
986
+ int status, flag = 0;
987
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), h);
988
+ fp = rb_gsl_open_readfile(io, &flag);
989
+ status = FUNCTION(gsl_matrix,fscanf)(fp, h);
990
+ if (flag == 1) fclose(fp);
991
+ return INT2FIX(status);
992
+ }
993
+
994
+ static VALUE FUNCTION(rb_gsl_matrix,set_diagonal)(VALUE obj, VALUE diag)
995
+ {
996
+ GSL_TYPE(gsl_matrix) *m = NULL;
997
+ GSL_TYPE(gsl_vector) *v;
998
+ size_t i, len;
999
+ BASE x;
1000
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1001
+ switch (TYPE(diag)) {
1002
+ case T_FIXNUM: case T_BIGNUM: case T_FLOAT:
1003
+ x = (BASE) NUMCONV2(diag);
1004
+ for (i = 0; i < m->size1; i++) FUNCTION(gsl_matrix,set)(m, i, i, x);
1005
+ break;
1006
+ case T_ARRAY:
1007
+ len = GSL_MIN_INT(m->size1, RARRAY_LEN(diag));
1008
+ for (i = 0; i < len; i++) {
1009
+ FUNCTION(gsl_matrix,set)(m, i, i, NUMCONV2(rb_ary_entry(diag, i)));
1010
+ }
1011
+ break;
1012
+ default:
1013
+ if (VEC_P(diag)) {
1014
+ Data_Get_Struct(diag, GSL_TYPE(gsl_vector), v);
1015
+ len = GSL_MIN_INT(m->size1, v->size);
1016
+ for (i = 0; i < len; i++) {
1017
+ FUNCTION(gsl_matrix,set)(m, i, i, FUNCTION(gsl_vector,get)(v, i));
1018
+ }
1019
+ } else {
1020
+ rb_raise(rb_eTypeError, "wrong argument type %s (GSL::Vector or Array expected)",
1021
+ rb_class2name(CLASS_OF(diag)));
1022
+ }
1023
+ break;
1024
+ }
1025
+ return obj;
1026
+ }
1027
+
1028
+ static VALUE FUNCTION(rb_gsl_matrix,get_row)(VALUE obj, VALUE i)
1029
+ {
1030
+ GSL_TYPE(gsl_matrix) *m = NULL;
1031
+ GSL_TYPE(gsl_vector) *v = NULL;
1032
+ CHECK_FIXNUM(i);
1033
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1034
+ v = FUNCTION(gsl_vector,alloc)(m->size1);
1035
+ if (v == NULL) rb_raise(rb_eNoMemError, "gsl_vector_alloc failed");
1036
+ FUNCTION(gsl_matrix,get_row)(v, m, FIX2INT(i));
1037
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_vector), 0, FUNCTION(gsl_vector,free), v);
1038
+ }
1039
+
1040
+ static VALUE FUNCTION(rb_gsl_matrix,get_col)(VALUE obj, VALUE i)
1041
+ {
1042
+ GSL_TYPE(gsl_matrix) *m = NULL;
1043
+ GSL_TYPE(gsl_vector) *v = NULL;
1044
+ CHECK_FIXNUM(i);
1045
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1046
+ v = FUNCTION(gsl_vector,alloc)(m->size2);
1047
+ if (v == NULL) rb_raise(rb_eNoMemError, "gsl_vector_alloc failed");
1048
+ FUNCTION(gsl_matrix,get_col)(v, m, FIX2INT(i));
1049
+ // TODO This is NOT returning a view! Is there a macro more appropriate than
1050
+ // QUALIFIED_VIEW?
1051
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_vector,col), 0, FUNCTION(gsl_vector,free), v);
1052
+ }
1053
+
1054
+ static VALUE FUNCTION(rb_gsl_matrix,set_row)(VALUE obj, VALUE i, VALUE vv)
1055
+ {
1056
+ GSL_TYPE(gsl_matrix) *m = NULL;
1057
+ GSL_TYPE(gsl_vector) *v = NULL;
1058
+ int flag = 0;
1059
+ size_t j;
1060
+ CHECK_FIXNUM(i);
1061
+ if (CLASS_OF(vv) == rb_cRange) vv = rb_gsl_range2ary(vv);
1062
+ if (TYPE(vv) == T_ARRAY) {
1063
+ v = FUNCTION(gsl_vector,alloc)(RARRAY_LEN(vv));
1064
+ for (j = 0; j < RARRAY_LEN(vv); j++) {
1065
+ FUNCTION(gsl_vector,set)(v, j, NUMCONV2(rb_ary_entry(vv, j)));
1066
+ }
1067
+ flag = 1;
1068
+ } else {
1069
+ CHECK_VEC(vv);
1070
+ Data_Get_Struct(vv, GSL_TYPE(gsl_vector), v);
1071
+ }
1072
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1073
+ FUNCTION(gsl_matrix,set_row)(m, FIX2INT(i), v);
1074
+ if (flag == 1) FUNCTION(gsl_vector,free)(v);
1075
+ return obj;
1076
+ }
1077
+
1078
+ static VALUE FUNCTION(rb_gsl_matrix,set_col)(VALUE obj, VALUE j, VALUE vv)
1079
+ {
1080
+ GSL_TYPE(gsl_matrix) *m = NULL;
1081
+ GSL_TYPE(gsl_vector) *v = NULL;
1082
+ int flag = 0;
1083
+ size_t i;
1084
+ CHECK_FIXNUM(j);
1085
+ if (CLASS_OF(vv) == rb_cRange) vv = rb_gsl_range2ary(vv);
1086
+ if (TYPE(vv) == T_ARRAY) {
1087
+ v = FUNCTION(gsl_vector,alloc)(RARRAY_LEN(vv));
1088
+ for (i = 0; i < RARRAY_LEN(vv); i++) {
1089
+ FUNCTION(gsl_vector,set)(v, i, NUMCONV2(rb_ary_entry(vv, i)));
1090
+ }
1091
+ flag = 1;
1092
+ } else {
1093
+ CHECK_VECTOR(vv);
1094
+ Data_Get_Struct(vv, GSL_TYPE(gsl_vector), v);
1095
+ }
1096
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1097
+ FUNCTION(gsl_matrix,set_col)(m, FIX2INT(j), v);
1098
+ if (flag == 1) FUNCTION(gsl_vector,free)(v);
1099
+ return obj;
1100
+ }
1101
+
1102
+ static VALUE FUNCTION(rb_gsl_matrix,clone)(VALUE obj)
1103
+ {
1104
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew = NULL;
1105
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1106
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
1107
+ FUNCTION(gsl_matrix,memcpy)(mnew, m);
1108
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1109
+ }
1110
+
1111
+ static VALUE FUNCTION(rb_gsl_matrix,memcpy)(VALUE obj, VALUE mm1, VALUE mm2)
1112
+ {
1113
+ GSL_TYPE(gsl_matrix) *m1 = NULL, *m2 = NULL;
1114
+ CHECK_MAT(mm1); CHECK_MAT(mm2);
1115
+ Data_Get_Struct(mm1, GSL_TYPE(gsl_matrix), m1);
1116
+ Data_Get_Struct(mm2, GSL_TYPE(gsl_matrix), m2);
1117
+ FUNCTION(gsl_matrix,memcpy)(m1, m2);
1118
+ return mm1;
1119
+ }
1120
+
1121
+ static VALUE FUNCTION(rb_gsl_matrix,isnull)(VALUE obj)
1122
+ {
1123
+ GSL_TYPE(gsl_matrix) *m = NULL;
1124
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1125
+ return INT2FIX(FUNCTION(gsl_matrix,isnull)(m));
1126
+ }
1127
+
1128
+ static VALUE FUNCTION(rb_gsl_matrix,isnull2)(VALUE obj)
1129
+ {
1130
+ GSL_TYPE(gsl_matrix) *m = NULL;
1131
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1132
+ if (FUNCTION(gsl_matrix,isnull)(m)) return Qtrue;
1133
+ else return Qfalse;
1134
+ }
1135
+
1136
+
1137
+ /* singleton */
1138
+ static VALUE FUNCTION(rb_gsl_matrix,swap)(VALUE obj, VALUE mm1, VALUE mm2)
1139
+ {
1140
+ GSL_TYPE(gsl_matrix) *m1 = NULL, *m2 = NULL;
1141
+ CHECK_MAT(mm1); CHECK_MAT(mm2);
1142
+ Data_Get_Struct(mm1, GSL_TYPE(gsl_matrix), m1);
1143
+ Data_Get_Struct(mm2, GSL_TYPE(gsl_matrix), m2);
1144
+ FUNCTION(gsl_matrix,swap)(m1, m2);
1145
+ return mm1;
1146
+ }
1147
+
1148
+ static VALUE FUNCTION(rb_gsl_matrix,swap_rows_bang)(VALUE obj, VALUE i, VALUE j)
1149
+ {
1150
+ GSL_TYPE(gsl_matrix) *m = NULL;
1151
+ CHECK_FIXNUM(i); CHECK_FIXNUM(j);
1152
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1153
+ FUNCTION(gsl_matrix,swap_rows)(m, FIX2INT(i), FIX2INT(j));
1154
+ return obj;
1155
+ }
1156
+
1157
+ static VALUE FUNCTION(rb_gsl_matrix,swap_rows)(VALUE obj, VALUE i, VALUE j)
1158
+ {
1159
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew;
1160
+ CHECK_FIXNUM(i); CHECK_FIXNUM(j);
1161
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1162
+ mnew = FUNCTION(make_matrix,clone)(m);
1163
+ FUNCTION(gsl_matrix,swap_rows)(mnew, FIX2INT(i), FIX2INT(j));
1164
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1165
+ }
1166
+
1167
+ static VALUE FUNCTION(rb_gsl_matrix,swap_columns_bang)(VALUE obj, VALUE i, VALUE j)
1168
+ {
1169
+ GSL_TYPE(gsl_matrix) *m = NULL;
1170
+ CHECK_FIXNUM(i); CHECK_FIXNUM(j);
1171
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1172
+ FUNCTION(gsl_matrix,swap_columns)(m, FIX2INT(i), FIX2INT(j));
1173
+ return obj;
1174
+ }
1175
+
1176
+ static VALUE FUNCTION(rb_gsl_matrix,swap_columns)(VALUE obj, VALUE i, VALUE j)
1177
+ {
1178
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew;
1179
+ CHECK_FIXNUM(i); CHECK_FIXNUM(j);
1180
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1181
+ mnew = FUNCTION(make_matrix,clone)(m);
1182
+ FUNCTION(gsl_matrix,swap_columns)(mnew, FIX2INT(i), FIX2INT(j));
1183
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1184
+ }
1185
+
1186
+ static VALUE FUNCTION(rb_gsl_matrix,swap_rowcol_bang)(VALUE obj, VALUE i, VALUE j)
1187
+ {
1188
+ GSL_TYPE(gsl_matrix) *m = NULL;
1189
+ CHECK_FIXNUM(i); CHECK_FIXNUM(j);
1190
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1191
+ FUNCTION(gsl_matrix,swap_rowcol)(m, FIX2INT(i), FIX2INT(j));
1192
+ return obj;
1193
+ }
1194
+
1195
+ static VALUE FUNCTION(rb_gsl_matrix,swap_rowcol)(VALUE obj, VALUE i, VALUE j)
1196
+ {
1197
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew = NULL;
1198
+ CHECK_FIXNUM(i); CHECK_FIXNUM(j);
1199
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1200
+ mnew = FUNCTION(make_matrix,clone)(m);
1201
+ FUNCTION(gsl_matrix,swap_rowcol)(mnew, FIX2INT(i), FIX2INT(j));
1202
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1203
+ }
1204
+
1205
+ static VALUE FUNCTION(rb_gsl_matrix,transpose_memcpy)(VALUE obj)
1206
+ {
1207
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew = NULL;
1208
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1209
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size2, m->size1);
1210
+ FUNCTION(gsl_matrix,transpose_memcpy)(mnew, m);
1211
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1212
+ }
1213
+
1214
+ static VALUE FUNCTION(rb_gsl_matrix,transpose_bang)(VALUE obj)
1215
+ {
1216
+ GSL_TYPE(gsl_matrix) *m = NULL;
1217
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1218
+ FUNCTION(gsl_matrix,transpose)(m);
1219
+ return obj;
1220
+ }
1221
+
1222
+ static VALUE FUNCTION(rb_gsl_matrix,max)(VALUE obj)
1223
+ {
1224
+ GSL_TYPE(gsl_matrix) *m = NULL;
1225
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1226
+ return C_TO_VALUE2(FUNCTION(gsl_matrix,max)(m));
1227
+ }
1228
+
1229
+ static VALUE FUNCTION(rb_gsl_matrix,min)(VALUE obj)
1230
+ {
1231
+ GSL_TYPE(gsl_matrix) *m = NULL;
1232
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1233
+ return C_TO_VALUE2(FUNCTION(gsl_matrix,min)(m));
1234
+ }
1235
+
1236
+ static VALUE FUNCTION(rb_gsl_matrix,minmax)(VALUE obj)
1237
+ {
1238
+ GSL_TYPE(gsl_matrix) *m = NULL;
1239
+ BASE min, max;
1240
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1241
+ FUNCTION(gsl_matrix,minmax)(m, &min, &max);
1242
+ return rb_ary_new3(2, C_TO_VALUE2(min), C_TO_VALUE2(max));
1243
+ }
1244
+
1245
+ static VALUE FUNCTION(rb_gsl_matrix,max_index)(VALUE obj)
1246
+ {
1247
+ GSL_TYPE(gsl_matrix) *m = NULL;
1248
+ size_t imax, jmax;
1249
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1250
+ FUNCTION(gsl_matrix,max_index)(m, &imax, &jmax);
1251
+ return rb_ary_new3(2, INT2FIX(imax), INT2FIX(jmax));
1252
+ }
1253
+
1254
+ static VALUE FUNCTION(rb_gsl_matrix,min_index)(VALUE obj)
1255
+ {
1256
+ GSL_TYPE(gsl_matrix) *m = NULL;
1257
+ size_t imin, jmin;
1258
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1259
+ FUNCTION(gsl_matrix,min_index)(m, &imin, &jmin);
1260
+ return rb_ary_new3(2, INT2FIX(imin), INT2FIX(jmin));
1261
+ }
1262
+
1263
+ static VALUE FUNCTION(rb_gsl_matrix,minmax_index)(VALUE obj)
1264
+ {
1265
+ GSL_TYPE(gsl_matrix) *m = NULL;
1266
+ size_t imin, jmin, imax, jmax;
1267
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1268
+ FUNCTION(gsl_matrix,minmax_index)(m, &imin, &jmin, &imax, &jmax);
1269
+ return rb_ary_new3(2, rb_ary_new3(2, INT2FIX(imin), INT2FIX(jmin)),
1270
+ rb_ary_new3(2, INT2FIX(imax), INT2FIX(jmax)));
1271
+ }
1272
+
1273
+ static VALUE FUNCTION(rb_gsl_matrix,fwrite)(VALUE obj, VALUE io)
1274
+ {
1275
+ GSL_TYPE(gsl_matrix) *h = NULL;
1276
+ FILE *f = NULL;
1277
+ int status, flag = 0;
1278
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), h);
1279
+ f = rb_gsl_open_writefile(io, &flag);
1280
+ status = FUNCTION(gsl_matrix,fwrite)(f, h);
1281
+ if (flag == 1) fclose(f);
1282
+ return INT2FIX(status);
1283
+ }
1284
+
1285
+ static VALUE FUNCTION(rb_gsl_matrix,fread)(VALUE obj, VALUE io)
1286
+ {
1287
+ GSL_TYPE(gsl_matrix) *h = NULL;
1288
+ FILE *f = NULL;
1289
+ int status, flag = 0;
1290
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), h);
1291
+ f = rb_gsl_open_readfile(io, &flag);
1292
+ status = FUNCTION(gsl_matrix,fread)(f, h);
1293
+ if (flag == 1) fclose(f);
1294
+ return INT2FIX(status);
1295
+ }
1296
+
1297
+ static VALUE FUNCTION(rb_gsl_matrix,trace)(VALUE obj)
1298
+ {
1299
+ GSL_TYPE(gsl_matrix) *m = NULL;
1300
+ size_t i;
1301
+ BASE trace = 0;
1302
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1303
+ for (i = 0; i < m->size1; i++) {
1304
+ trace += FUNCTION(gsl_matrix,get)(m, i, i);
1305
+ }
1306
+ return C_TO_VALUE2(trace);
1307
+ }
1308
+
1309
+
1310
+ static VALUE FUNCTION(rb_gsl_matrix,uplus)(VALUE obj)
1311
+ {
1312
+ return obj;
1313
+ }
1314
+
1315
+ static VALUE FUNCTION(rb_gsl_matrix,uminus)(VALUE obj)
1316
+ {
1317
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew = NULL;
1318
+ size_t i, j;
1319
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1320
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
1321
+ for (i = 0; i < m->size1; i++) {
1322
+ for (j = 0; j < m->size2; j++) {
1323
+ FUNCTION(gsl_matrix,set)(mnew, i, j, -FUNCTION(gsl_matrix,get)(m, i, j));
1324
+ }
1325
+ }
1326
+ return Data_Wrap_Struct(cgsl_matrix, 0, gsl_matrix_free, mnew);
1327
+ }
1328
+
1329
+ VALUE FUNCTION(rb_gsl_matrix,power)(VALUE obj, VALUE bb)
1330
+ {
1331
+ GSL_TYPE(gsl_matrix) *m = NULL, *mtmp = NULL, *mnew = NULL;
1332
+ size_t i, b;
1333
+ CHECK_FIXNUM(bb);
1334
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1335
+ b = FIX2INT(bb);
1336
+ mtmp = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
1337
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
1338
+ FUNCTION(gsl_matrix,memcpy)(mnew, m);
1339
+ for (i = 1; i < b; i++) {
1340
+ FUNCTION(gsl_matrix,memcpy)(mtmp, mnew);
1341
+ #ifdef BASE_DOUBLE
1342
+ gsl_linalg_matmult(mtmp, m, mnew);
1343
+ #else
1344
+ gsl_linalg_matmult_int(mtmp, m, mnew);
1345
+ #endif
1346
+ }
1347
+ FUNCTION(gsl_matrix,free)(mtmp);
1348
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1349
+ }
1350
+
1351
+ static VALUE FUNCTION(rb_gsl_matrix,submatrix)(int argc, VALUE *argv, VALUE obj)
1352
+ {
1353
+ GSL_TYPE(gsl_matrix) *m = NULL;
1354
+ QUALIFIED_VIEW(gsl_matrix,view) *mv = NULL;
1355
+ QUALIFIED_VIEW(gsl_vector,view) *vv = NULL;
1356
+ size_t i, j, n1, n2;
1357
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1358
+ parse_submatrix_args(argc, argv, m->size1, m->size2, &i, &j, &n1, &n2);
1359
+ if(n1 == 0) {
1360
+ vv = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1361
+ *vv = FUNCTION(gsl_matrix,subrow)(m, i, j, n2);
1362
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_vector,view), 0, free, vv);
1363
+ }
1364
+ else if(n2 == 0) {
1365
+ vv = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1366
+ *vv = FUNCTION(gsl_matrix,subcolumn)(m, j, i, n1);
1367
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_vector,col_view), 0, free, vv);
1368
+ } else {
1369
+ mv = ALLOC(QUALIFIED_VIEW(gsl_matrix,view));
1370
+ *mv = FUNCTION(gsl_matrix,submatrix)(m, i, j, n1, n2);
1371
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_matrix,view), 0, free, mv);
1372
+ }
1373
+ }
1374
+
1375
+ static VALUE FUNCTION(rb_gsl_matrix,return_vector_view)(VALUE obj, VALUE index,
1376
+ QUALIFIED_VIEW(gsl_vector,view) (*f)(GSL_TYPE(gsl_matrix)*,
1377
+ size_t))
1378
+ {
1379
+ GSL_TYPE(gsl_matrix) *m = NULL;
1380
+ QUALIFIED_VIEW(gsl_vector,view) *vv = NULL;
1381
+ CHECK_FIXNUM(index);
1382
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1383
+ vv = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1384
+ *vv = (*f)(m, FIX2INT(index));
1385
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_vector,view), 0, free, vv);
1386
+ }
1387
+
1388
+ static VALUE FUNCTION(rb_gsl_matrix,row)(VALUE obj, VALUE i)
1389
+ {
1390
+ return FUNCTION(rb_gsl_matrix,return_vector_view)(obj, i, FUNCTION(gsl_matrix,row));
1391
+ }
1392
+
1393
+ static VALUE FUNCTION(rb_gsl_matrix,column)(VALUE obj, VALUE j)
1394
+ {
1395
+ GSL_TYPE(gsl_matrix) *m = NULL;
1396
+ QUALIFIED_VIEW(gsl_vector,view) *vv = NULL;
1397
+ CHECK_FIXNUM(j);
1398
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1399
+ vv = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1400
+ *vv = FUNCTION(gsl_matrix,column)(m, FIX2INT(j));
1401
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_vector,col_view), 0, free, vv);
1402
+ }
1403
+
1404
+ #ifdef GSL_1_10_LATER
1405
+ static VALUE FUNCTION(rb_gsl_matrix,subrow)(VALUE obj, VALUE i, VALUE offset,
1406
+ VALUE n)
1407
+ {
1408
+ GSL_TYPE(gsl_matrix) *m = NULL;
1409
+ QUALIFIED_VIEW(gsl_vector,view) *vv = NULL;
1410
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1411
+ vv = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1412
+ *vv = FUNCTION(gsl_matrix,subrow)(m, FIX2INT(i), FIX2INT(offset), FIX2INT(n));
1413
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_vector,view), 0, free, vv);
1414
+ }
1415
+ static VALUE FUNCTION(rb_gsl_matrix,subcolumn)(VALUE obj, VALUE j, VALUE offset,
1416
+ VALUE n)
1417
+ {
1418
+ GSL_TYPE(gsl_matrix) *m = NULL;
1419
+ QUALIFIED_VIEW(gsl_vector,view) *vv = NULL;
1420
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1421
+ vv = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1422
+ *vv = FUNCTION(gsl_matrix,subcolumn)(m, FIX2INT(j), FIX2INT(offset), FIX2INT(n));
1423
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_vector,col_view), 0, free, vv);
1424
+ }
1425
+ #endif
1426
+
1427
+ static VALUE FUNCTION(rb_gsl_matrix,diagonal)(VALUE obj)
1428
+ {
1429
+ GSL_TYPE(gsl_matrix) *m = NULL;
1430
+ QUALIFIED_VIEW(gsl_vector,view) *vv = NULL;
1431
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1432
+ vv = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1433
+ *vv = FUNCTION(gsl_matrix,diagonal)(m);
1434
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_vector,view), 0, free, vv);
1435
+ }
1436
+
1437
+ static VALUE FUNCTION(rb_gsl_matrix,subdiagonal)(VALUE obj, VALUE k)
1438
+ {
1439
+ return rb_gsl_matrix_return_vector_view(obj, k, gsl_matrix_subdiagonal);
1440
+ }
1441
+
1442
+ static VALUE FUNCTION(rb_gsl_matrix,superdiagonal)(VALUE obj, VALUE k)
1443
+ {
1444
+ return rb_gsl_matrix_return_vector_view(obj, k, gsl_matrix_superdiagonal);
1445
+ }
1446
+
1447
+ static VALUE FUNCTION(rb_gsl_matrix,vector_view)(VALUE obj)
1448
+ {
1449
+ GSL_TYPE(gsl_matrix) *m = NULL;
1450
+ QUALIFIED_VIEW(gsl_vector,view) *vv;
1451
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1452
+ vv = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1453
+ vv->vector.size = m->size1*m->size2;
1454
+ vv->vector.owner = 0;
1455
+ vv->vector.stride = 1;
1456
+ vv->vector.data = m->data;
1457
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_vector,view), 0, free, vv);
1458
+ }
1459
+
1460
+
1461
+ static VALUE FUNCTION(rb_gsl_matrix,each_row)(VALUE obj)
1462
+ {
1463
+ GSL_TYPE(gsl_matrix) *m = NULL;
1464
+ QUALIFIED_VIEW(gsl_vector,view) *vv;
1465
+ size_t i;
1466
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1467
+ for (i = 0; i < m->size1; i++) {
1468
+ vv = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1469
+ *vv = FUNCTION(gsl_matrix,row)(m, i);
1470
+ #ifdef BASE_DOUBLE
1471
+ rb_yield(Data_Wrap_Struct(cgsl_vector_view, 0, free, vv));
1472
+ #else
1473
+ rb_yield(Data_Wrap_Struct(cgsl_vector_int_view, 0, free, vv));
1474
+ #endif
1475
+ }
1476
+ return obj;
1477
+ }
1478
+
1479
+ static VALUE FUNCTION(rb_gsl_matrix,each_col)(VALUE obj)
1480
+ {
1481
+ GSL_TYPE(gsl_matrix) *m = NULL;
1482
+ QUALIFIED_VIEW(gsl_vector,view) *vv;
1483
+ size_t i;
1484
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1485
+ for (i = 0; i < m->size2; i++) {
1486
+ vv = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1487
+ *vv = FUNCTION(gsl_matrix,column)(m, i);
1488
+ #ifdef BASE_DOUBLE
1489
+ rb_yield(Data_Wrap_Struct(cgsl_vector_col_view, 0, free, vv));
1490
+ #else
1491
+ rb_yield(Data_Wrap_Struct(cgsl_vector_int_col_view, 0, free, vv));
1492
+ #endif
1493
+ }
1494
+ return obj;
1495
+ }
1496
+
1497
+ static VALUE FUNCTION(rb_gsl_matrix,scale_bang)(VALUE obj, VALUE x)
1498
+ {
1499
+ GSL_TYPE(gsl_matrix) *m;
1500
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1501
+ FUNCTION(gsl_matrix,scale)(m, NUMCONV(x));
1502
+ return obj;
1503
+ }
1504
+
1505
+ static VALUE FUNCTION(rb_gsl_matrix,scale)(VALUE obj, VALUE b)
1506
+ {
1507
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew;
1508
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1509
+ mnew = FUNCTION(make_matrix,clone)(m);
1510
+ FUNCTION(gsl_matrix,scale)(mnew, NUMCONV(b));
1511
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1512
+ }
1513
+
1514
+ static VALUE FUNCTION(rb_gsl_matrix,add_constant_bang)(VALUE obj, VALUE x)
1515
+ {
1516
+ GSL_TYPE(gsl_matrix) *m;
1517
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1518
+ FUNCTION(gsl_matrix,add_constant)(m, NUMCONV(x));
1519
+ return obj;
1520
+ }
1521
+
1522
+ static VALUE FUNCTION(rb_gsl_matrix,add_constant)(VALUE obj, VALUE b)
1523
+ {
1524
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew;
1525
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1526
+ mnew = FUNCTION(make_matrix,clone)(m);
1527
+ FUNCTION(gsl_matrix,add_constant)(mnew, NUMCONV(b));
1528
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1529
+ }
1530
+
1531
+ static int FUNCTION(mygsl_matrix,equal)(GSL_TYPE(gsl_matrix) *a, GSL_TYPE(gsl_matrix) *b, double eps)
1532
+ {
1533
+ size_t i, j;
1534
+ BASE x, y;
1535
+ if (a->size1 != b->size1 || a->size2 != b->size2) return 0;
1536
+ for (i = 0; i < a->size1; i++) {
1537
+ for (j = 0; j < a->size2; j++) {
1538
+ x = FUNCTION(gsl_matrix,get)(a, i, j);
1539
+ y = FUNCTION(gsl_matrix,get)(b, i, j);
1540
+ if (fabs(x-y) > eps) return 0;
1541
+ }
1542
+ }
1543
+ return 1;
1544
+ }
1545
+
1546
+ #ifdef HAVE_TENSOR_TENSOR_H
1547
+ EXTERN VALUE cgsl_tensor, cgsl_tensor_int;
1548
+ VALUE rb_gsl_tensor_equal(int argc, VALUE *argv, VALUE obj);
1549
+ VALUE rb_gsl_tensor_int_equal(int argc, VALUE *argv, VALUE obj);
1550
+ #ifdef BASE_DOUBLE
1551
+ #define TEN_P(x) TENSOR_P(x)
1552
+ #else
1553
+ #define TEN_P(x) TENSOR_INT_P(x)
1554
+ #endif
1555
+ #endif
1556
+
1557
+ static VALUE FUNCTION(rb_gsl_matrix,equal)(int argc, VALUE *argv, VALUE obj)
1558
+ {
1559
+ GSL_TYPE(gsl_matrix) *a, *b;
1560
+ double eps = 1e-10;
1561
+ VALUE bb;
1562
+ switch (argc) {
1563
+ case 2:
1564
+ bb = argv[0];
1565
+ eps = NUM2DBL(argv[1]);
1566
+ break;
1567
+ case 1:
1568
+ bb = argv[0];
1569
+ break;
1570
+ default:
1571
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 1 or 2)", argc);
1572
+ break;
1573
+ }
1574
+ #ifdef HAVE_TENSOR_TENSOR_H
1575
+ if (TEN_P(bb)) {
1576
+ return FUNCTION(rb_gsl_tensor,equal)(argc, argv, obj);
1577
+ }
1578
+ #endif
1579
+ CHECK_MAT(bb);
1580
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), a);
1581
+ Data_Get_Struct(bb, GSL_TYPE(gsl_matrix), b);
1582
+ if (FUNCTION(mygsl_matrix,equal)(a, b, eps) == 1) return Qtrue;
1583
+ else return Qfalse;
1584
+ }
1585
+
1586
+ #ifdef HAVE_TENSOR_TENSOR_H
1587
+ #ifdef TEN_P
1588
+ #undef TEN_P
1589
+ #endif
1590
+ #endif
1591
+
1592
+ static VALUE FUNCTION(rb_gsl_matrix,equal_singleton)(int argc, VALUE *argv, VALUE obj)
1593
+ {
1594
+ GSL_TYPE(gsl_matrix) *a, *b;
1595
+ VALUE aa, bb;
1596
+ double eps = 1e-10;
1597
+ BASE x, y;
1598
+ size_t i, j;
1599
+ switch (argc) {
1600
+ case 3:
1601
+ aa = argv[0];
1602
+ bb = argv[1];
1603
+ eps = NUM2DBL(argv[2]);
1604
+ break;
1605
+ case 2:
1606
+ aa = argv[0];
1607
+ bb = argv[1];
1608
+ break;
1609
+ default:
1610
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 2 or 3)", argc);
1611
+ break;
1612
+ }
1613
+ CHECK_MAT(aa); CHECK_MAT(bb);
1614
+ Data_Get_Struct(aa, GSL_TYPE(gsl_matrix), a);
1615
+ Data_Get_Struct(bb, GSL_TYPE(gsl_matrix), b);
1616
+ if (a->size1 != b->size1 || a->size2 != b->size2) return Qfalse;
1617
+ for (i = 0; i < a->size1; i++) {
1618
+ for (j = 0; j < a->size2; j++) {
1619
+ x = FUNCTION(gsl_matrix,get)(a, i, j);
1620
+ y = FUNCTION(gsl_matrix,get)(b, i, j);
1621
+ if (fabs(x-y) > eps) return Qfalse;
1622
+ }
1623
+ }
1624
+ return Qtrue;
1625
+ }
1626
+
1627
+ #ifdef HAVE_TENSOR_TENSOR_H
1628
+ #include "rb_gsl_tensor.h"
1629
+ static VALUE FUNCTION(rb_gsl_matrix,to_tensor)(VALUE obj)
1630
+ {
1631
+ GSL_TYPE(gsl_matrix) *m;
1632
+ GSL_TYPE(rbgsl_tensor) *t;
1633
+ unsigned int rank;
1634
+ size_t dim;
1635
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1636
+ if (m->size1 != m->size2) rb_raise(rb_eRuntimeError, "matrix must have equal dimensions");
1637
+ rank = 2;
1638
+ dim = m->size1;
1639
+ t = FUNCTION(rbgsl_tensor,alloc)(rank, dim);
1640
+ memcpy(t->tensor->data, m->data, sizeof(BASE)*t->tensor->size);
1641
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_tensor), 0, FUNCTION(rbgsl_tensor,free), t);
1642
+ }
1643
+ #endif
1644
+
1645
+ static VALUE FUNCTION(rb_gsl_matrix,collect)(VALUE obj)
1646
+ {
1647
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew;
1648
+ size_t i, j;
1649
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1650
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
1651
+ for (i = 0; i < m->size1; i++) {
1652
+ for (j = 0; j < m->size2; j++) {
1653
+ FUNCTION(gsl_matrix,set)(mnew, i, j, NUMCONV(rb_yield(C_TO_VALUE(FUNCTION(gsl_matrix,get)(m, i, j)))));
1654
+ }
1655
+ }
1656
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1657
+ }
1658
+
1659
+ static VALUE FUNCTION(rb_gsl_matrix,collect_bang)(VALUE obj)
1660
+ {
1661
+ GSL_TYPE(gsl_matrix) *m = NULL;
1662
+ size_t i, j;
1663
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1664
+ for (i = 0; i < m->size1; i++) {
1665
+ for (j = 0; j < m->size2; j++) {
1666
+ FUNCTION(gsl_matrix,set)(m, i, j, NUMCONV(rb_yield(C_TO_VALUE(FUNCTION(gsl_matrix,get)(m, i, j)))));
1667
+ }
1668
+ }
1669
+ return obj;
1670
+ }
1671
+
1672
+ static VALUE FUNCTION(rb_gsl_matrix,upper)(VALUE obj)
1673
+ {
1674
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew;
1675
+ size_t i, j;
1676
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1677
+ mnew = FUNCTION(make_matrix,clone)(m);
1678
+ for (i = 0; i < m->size1; i++) {
1679
+ for (j = 0; j < i; j++) {
1680
+ FUNCTION(gsl_matrix,set)(mnew, i, j, 0);
1681
+ }
1682
+ }
1683
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1684
+ }
1685
+
1686
+ static VALUE FUNCTION(rb_gsl_matrix,lower)(VALUE obj)
1687
+ {
1688
+ GSL_TYPE(gsl_matrix) *m = NULL, *mnew;
1689
+ size_t i, j;
1690
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1691
+ mnew = FUNCTION(make_matrix,clone)(m);
1692
+ for (i = 0; i < m->size1; i++) {
1693
+ for (j = i+1; j < m->size2; j++) {
1694
+ FUNCTION(gsl_matrix,set)(mnew, i, j, 0);
1695
+ }
1696
+ }
1697
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1698
+ }
1699
+
1700
+ /* singleton method, creates Pascal matrix of dimension n */
1701
+ static VALUE FUNCTION(rb_gsl_matrix,pascal1)(VALUE obj, VALUE n)
1702
+ {
1703
+ GSL_TYPE(gsl_matrix) *m;
1704
+ BASE x;
1705
+ size_t i, j, dim;
1706
+ CHECK_FIXNUM(n);
1707
+ dim = (size_t) FIX2INT(n);
1708
+ m = FUNCTION(gsl_matrix,alloc)(dim, dim);
1709
+ for (j = 0; j < dim; j++) FUNCTION(gsl_matrix,set)(m, 0, j, (BASE) 1);
1710
+ for (i = 1; i < dim; i++) {
1711
+ FUNCTION(gsl_matrix,set)(m, i, 0, (BASE) 1);
1712
+ for (j = 1; j < dim; j++) {
1713
+ x=FUNCTION(gsl_matrix,get)(m,i-1,j)+FUNCTION(gsl_matrix,get)(m,i,j-1);
1714
+ FUNCTION(gsl_matrix,set)(m, i, j, x);
1715
+ }
1716
+ }
1717
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), m);
1718
+ }
1719
+
1720
+ #ifdef BASE_DOUBLE
1721
+ static VALUE FUNCTION(rb_gsl_matrix,hilbert)(VALUE obj, VALUE n)
1722
+ {
1723
+ GSL_TYPE(gsl_matrix) *m;
1724
+ double x;
1725
+ size_t i, j, dim;
1726
+ CHECK_FIXNUM(n);
1727
+ dim = (size_t) FIX2INT(n);
1728
+ m = FUNCTION(gsl_matrix,alloc)(dim, dim);
1729
+ for (i = 0; i < dim; i++) {
1730
+ for (j = 0; j < dim; j++) {
1731
+ x = 1.0/(i + j + 1);
1732
+ FUNCTION(gsl_matrix,set)(m, i, j, (BASE) x);
1733
+ }
1734
+ }
1735
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), m);
1736
+ }
1737
+
1738
+ double mygsl_binomial_coef(unsigned int n, unsigned int k);
1739
+ static VALUE FUNCTION(rb_gsl_matrix,invhilbert)(VALUE obj, VALUE n)
1740
+ {
1741
+ GSL_TYPE(gsl_matrix) *m;
1742
+ double x, y;
1743
+ size_t i, j, dim;
1744
+ CHECK_FIXNUM(n);
1745
+ dim = (size_t) FIX2INT(n);
1746
+ m = FUNCTION(gsl_matrix,alloc)(dim, dim);
1747
+ for (i = 0; i < dim; i++) {
1748
+ for (j = 0; j < dim; j++) {
1749
+ if ((i+j)%2 == 0) x = 1;
1750
+ else x = -1;
1751
+ x *= (i + j + 1);
1752
+ x *= mygsl_binomial_coef(dim + i, dim - j - 1);
1753
+ x *= mygsl_binomial_coef(dim + j, dim - i - 1);
1754
+ y = mygsl_binomial_coef(i + j, i);
1755
+ x *= y*y;
1756
+ FUNCTION(gsl_matrix,set)(m, i, j, (BASE) x);
1757
+ }
1758
+ }
1759
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), m);
1760
+ }
1761
+ #endif
1762
+
1763
+ static void FUNCTION(mygsl_matrix,vandermonde)(GSL_TYPE(gsl_matrix) *m, GSL_TYPE(gsl_vector) *v)
1764
+ {
1765
+ size_t i, j;
1766
+
1767
+ for (i = 0; i < v->size; i++) {
1768
+ for (j = 0; j < v->size; j++) {
1769
+ FUNCTION(gsl_matrix,set)(m, i, j, (BASE) gsl_pow_int(FUNCTION(gsl_vector,get)(v, i), v->size-j-1));
1770
+ }
1771
+ }
1772
+ }
1773
+
1774
+ /* singleton */
1775
+ static VALUE FUNCTION(rb_gsl_matrix,vandermonde)(VALUE obj, VALUE vv)
1776
+ {
1777
+ GSL_TYPE(gsl_vector) *v;
1778
+ GSL_TYPE(gsl_matrix) *m;
1779
+ int flag = 0;
1780
+ if (TYPE(vv) == T_ARRAY) {
1781
+ v = FUNCTION(make_cvector,from_rarray)(vv);
1782
+ flag = 1;
1783
+ } else if (VEC_P(vv)) {
1784
+ Data_Get_Struct(vv, GSL_TYPE(gsl_vector), v);
1785
+ } else {
1786
+ rb_raise(rb_eTypeError, "wrong argument type %s (Array or Vector expected)",
1787
+ rb_class2name(CLASS_OF(vv)));
1788
+ }
1789
+ m = FUNCTION(gsl_matrix,alloc)(v->size, v->size);
1790
+ FUNCTION(mygsl_matrix,vandermonde)(m, v);
1791
+ if (flag == 1) FUNCTION(gsl_vector,free)(v);
1792
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), m);
1793
+ }
1794
+
1795
+ static void FUNCTION(mygsl_matrix,toeplitz)(GSL_TYPE(gsl_matrix) *m, GSL_TYPE(gsl_vector) *v)
1796
+ {
1797
+ size_t i, j;
1798
+ for (i = 0; i < v->size; i++) {
1799
+ for (j = 0; j < v->size; j++) {
1800
+ if (j >= i)
1801
+ FUNCTION(gsl_matrix,set)(m, i, j, FUNCTION(gsl_vector,get)(v, j-i));
1802
+ else
1803
+ FUNCTION(gsl_matrix,set)(m, i, j, FUNCTION(gsl_vector,get)(v, i-j));
1804
+ }
1805
+ }
1806
+ }
1807
+
1808
+ static VALUE FUNCTION(rb_gsl_matrix,toeplitz)(VALUE obj, VALUE vv)
1809
+ {
1810
+ GSL_TYPE(gsl_vector) *v;
1811
+ GSL_TYPE(gsl_matrix) *m;
1812
+ int flag = 0;
1813
+ if (TYPE(vv) == T_ARRAY) {
1814
+ v = FUNCTION(make_cvector,from_rarray)(vv);
1815
+ flag = 1;
1816
+ } else if (VEC_P(vv)) {
1817
+ Data_Get_Struct(vv, GSL_TYPE(gsl_vector), v);
1818
+ } else {
1819
+ rb_raise(rb_eTypeError, "wrong argument type %s (Array or Vector expected)",
1820
+ rb_class2name(CLASS_OF(vv)));
1821
+ }
1822
+ m = FUNCTION(gsl_matrix,alloc)(v->size, v->size);
1823
+ FUNCTION(mygsl_matrix,toeplitz)(m, v);
1824
+ if (flag == 1) FUNCTION(gsl_vector,free)(v);
1825
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), m);
1826
+ }
1827
+
1828
+ void FUNCTION(mygsl_vector,to_m_circulant)(GSL_TYPE(gsl_matrix) *m, GSL_TYPE(gsl_vector) *v);
1829
+ static VALUE FUNCTION(rb_gsl_matrix,circulant)(VALUE obj, VALUE vv)
1830
+ {
1831
+ GSL_TYPE(gsl_vector) *v;
1832
+ GSL_TYPE(gsl_matrix) *m;
1833
+ int flag = 0;
1834
+ if (TYPE(vv) == T_ARRAY) {
1835
+ v = FUNCTION(make_cvector,from_rarray)(vv);
1836
+ flag = 1;
1837
+ } else if (VEC_P(vv)) {
1838
+ Data_Get_Struct(vv, GSL_TYPE(gsl_vector), v);
1839
+ } else {
1840
+ rb_raise(rb_eTypeError, "wrong argument type %s (Array or Vector expected)",
1841
+ rb_class2name(CLASS_OF(vv)));
1842
+ }
1843
+ m = FUNCTION(gsl_matrix,alloc)(v->size, v->size);
1844
+ FUNCTION(mygsl_vector,to_m_circulant)(m, v);
1845
+ if (flag == 1) FUNCTION(gsl_vector,free)(v);
1846
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), m);
1847
+ }
1848
+
1849
+ static void FUNCTION(mygsl_matrix,indgen)(GSL_TYPE(gsl_matrix) *m,
1850
+ BASE start, BASE step)
1851
+ {
1852
+ size_t i, j;
1853
+ BASE n;
1854
+ n = start;
1855
+ for (i = 0; i < m->size1; i++) {
1856
+ for (j = 0; j < m->size2; j++) {
1857
+ FUNCTION(gsl_matrix,set)(m, i, j, n);
1858
+ n += step;
1859
+ }
1860
+ }
1861
+ }
1862
+
1863
+ static VALUE FUNCTION(rb_gsl_matrix,indgen_singleton)(int argc, VALUE *argv, VALUE obj)
1864
+ {
1865
+ GSL_TYPE(gsl_matrix) *m;
1866
+ size_t n1, n2;
1867
+ BASE start = 0, step = 1;
1868
+ switch (argc) {
1869
+ case 4:
1870
+ step = NUMCONV2(argv[3]);
1871
+ /* no break */
1872
+ case 3:
1873
+ start = NUMCONV2(argv[2]);
1874
+ /* no break */
1875
+ case 2:
1876
+ n1 = NUM2INT(argv[0]);
1877
+ n2 = NUM2INT(argv[1]);
1878
+ break;
1879
+ default:
1880
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 2-4)", argc);
1881
+ break;
1882
+ }
1883
+ m = FUNCTION(gsl_matrix,alloc)(n1, n2);
1884
+ FUNCTION(mygsl_matrix,indgen)(m, start, step);
1885
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), m);
1886
+ }
1887
+
1888
+ static VALUE FUNCTION(rb_gsl_matrix,indgen)(int argc, VALUE *argv, VALUE obj)
1889
+ {
1890
+ GSL_TYPE(gsl_matrix) *m, *mnew;
1891
+ BASE start = 0, step = 1;
1892
+ switch (argc) {
1893
+ case 2:
1894
+ step = NUMCONV2(argv[1]);
1895
+ /* no break */
1896
+ case 1:
1897
+ start = NUMCONV2(argv[0]);
1898
+ break;
1899
+ case 0:
1900
+ break;
1901
+ default:
1902
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 0-2)", argc);
1903
+ break;
1904
+ }
1905
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1906
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
1907
+ FUNCTION(mygsl_matrix,indgen)(mnew, start, step);
1908
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
1909
+ }
1910
+
1911
+ static VALUE FUNCTION(rb_gsl_matrix,indgen_bang)(int argc, VALUE *argv, VALUE obj)
1912
+ {
1913
+ GSL_TYPE(gsl_matrix) *m;
1914
+ BASE start = 0, step = 1;
1915
+ switch (argc) {
1916
+ case 2:
1917
+ step = NUMCONV2(argv[1]);
1918
+ /* no break */
1919
+ case 1:
1920
+ start = NUMCONV2(argv[0]);
1921
+ break;
1922
+ case 0:
1923
+ break;
1924
+ default:
1925
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 0-2)", argc);
1926
+ break;
1927
+ }
1928
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1929
+ FUNCTION(mygsl_matrix,indgen)(m, start, step);
1930
+ return obj;
1931
+ }
1932
+
1933
+ static VALUE FUNCTION(rb_gsl_matrix,to_a)(VALUE obj)
1934
+ {
1935
+ GSL_TYPE(gsl_matrix) *m;
1936
+ VALUE ma, ra;
1937
+ size_t i, j;
1938
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1939
+ ma = rb_ary_new2(m->size1);
1940
+ for(i=0; i < m->size1; i++) {
1941
+ ra = rb_ary_new2(m->size2);
1942
+ rb_ary_store(ma, i, ra);
1943
+ for(j=0; j < m->size2; j++) {
1944
+ rb_ary_store(ra, j, C_TO_VALUE(FUNCTION(gsl_matrix,get)(m, i, j)));
1945
+ }
1946
+ }
1947
+ return ma;
1948
+ }
1949
+
1950
+ static VALUE FUNCTION(rb_gsl_matrix,to_v)(VALUE obj)
1951
+ {
1952
+ GSL_TYPE(gsl_matrix) *m;
1953
+ GSL_TYPE(gsl_vector) *v;
1954
+ size_t i, j, k;
1955
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1956
+ v = FUNCTION(gsl_vector,alloc)(m->size1*m->size2);
1957
+ // memcpy(v->data, m->data, sizeof(BASE)*v->size);
1958
+ for (i = 0, k = 0; i < m->size1; i++) {
1959
+ for (j = 0; j < m->size2; j++, k++) {
1960
+ FUNCTION(gsl_vector,set)(v, k, FUNCTION(gsl_matrix,get)(m, i, j));
1961
+ }
1962
+ }
1963
+
1964
+ if(m->size1 > 1 && m->size2 == 1) {
1965
+ return Data_Wrap_Struct(CONCAT2(GSL_TYPE(cgsl_vector),col), 0, FUNCTION(gsl_vector,free), v);
1966
+ } else {
1967
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_vector), 0, FUNCTION(gsl_vector,free), v);
1968
+ }
1969
+ }
1970
+
1971
+ static VALUE FUNCTION(rb_gsl_matrix,to_vview)(VALUE obj)
1972
+ {
1973
+ GSL_TYPE(gsl_matrix) *m;
1974
+ QUALIFIED_VIEW(gsl_vector,view) *v;
1975
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1976
+ v = ALLOC(QUALIFIED_VIEW(gsl_vector,view));
1977
+ v->vector.size = m->size1*m->size2;
1978
+ v->vector.stride = 1;
1979
+ v->vector.owner = 0;
1980
+ v->vector.data = m->data;
1981
+ return Data_Wrap_Struct(QUALIFIED_VIEW(cgsl_vector,view), 0, FUNCTION(gsl_vector,free), v);
1982
+ }
1983
+
1984
+ static VALUE FUNCTION(rb_gsl_matrix,norm)(VALUE obj)
1985
+ {
1986
+ GSL_TYPE(gsl_matrix) *m;
1987
+ size_t i, n;
1988
+ BASE x = 0;
1989
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
1990
+ n = m->size1*m->size2;
1991
+ for (i = 0; i < n; i++) x += m->data[i]*m->data[i];
1992
+ return rb_float_new(sqrt(x));
1993
+ }
1994
+
1995
+ static int FUNCTION(mygsl_matrix,reverse_columns)(GSL_TYPE(gsl_matrix) *dst,
1996
+ GSL_TYPE(gsl_matrix) *src)
1997
+ {
1998
+ size_t j;
1999
+ QUALIFIED_VIEW(gsl_vector,view) col;
2000
+ if (dst->size1 != src->size1 || dst->size2 != src->size2)
2001
+ rb_raise(rb_eRuntimeError, "matrix sizes are different.");
2002
+ for (j = 0; j < src->size2; j++) {
2003
+ col = FUNCTION(gsl_matrix,column)(src, j);
2004
+ FUNCTION(gsl_matrix,set_col)(dst, dst->size2-1-j, &col.vector);
2005
+ }
2006
+ return 0;
2007
+ }
2008
+
2009
+ static int FUNCTION(mygsl_matrix,reverse_rows)(GSL_TYPE(gsl_matrix) *dst,
2010
+ GSL_TYPE(gsl_matrix) *src)
2011
+ {
2012
+ size_t i;
2013
+ QUALIFIED_VIEW(gsl_vector,view) row;
2014
+ if (dst->size1 != src->size1 || dst->size2 != src->size2)
2015
+ rb_raise(rb_eRuntimeError, "matrix sizes are different.");
2016
+ for (i = 0; i < src->size1; i++) {
2017
+ row = FUNCTION(gsl_matrix,row)(src, i);
2018
+ FUNCTION(gsl_matrix,set_row)(dst, dst->size1-1-i, &row.vector);
2019
+ }
2020
+ return 0;
2021
+ }
2022
+
2023
+ static VALUE FUNCTION(rb_gsl_matrix,reverse_columns)(VALUE obj)
2024
+ {
2025
+ GSL_TYPE(gsl_matrix) *m, *mnew;
2026
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2027
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
2028
+ FUNCTION(mygsl_matrix,reverse_columns)(mnew, m);
2029
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
2030
+ }
2031
+
2032
+ static VALUE FUNCTION(rb_gsl_matrix,reverse_columns_bang)(VALUE obj)
2033
+ {
2034
+ GSL_TYPE(gsl_matrix) *m, *mnew;
2035
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2036
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
2037
+ FUNCTION(mygsl_matrix,reverse_columns)(mnew, m);
2038
+ FUNCTION(gsl_matrix,memcpy)(m, mnew);
2039
+ FUNCTION(gsl_matrix,free)(mnew);
2040
+ return obj;
2041
+ }
2042
+
2043
+ static VALUE FUNCTION(rb_gsl_matrix,reverse_rows)(VALUE obj)
2044
+ {
2045
+ GSL_TYPE(gsl_matrix) *m, *mnew;
2046
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2047
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
2048
+ FUNCTION(mygsl_matrix,reverse_rows)(mnew, m);
2049
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
2050
+ }
2051
+
2052
+ static VALUE FUNCTION(rb_gsl_matrix,reverse_rows_bang)(VALUE obj)
2053
+ {
2054
+ GSL_TYPE(gsl_matrix) *m, *mnew;
2055
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2056
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
2057
+ FUNCTION(mygsl_matrix,reverse_rows)(mnew, m);
2058
+ FUNCTION(gsl_matrix,memcpy)(m, mnew);
2059
+ FUNCTION(gsl_matrix,free)(mnew);
2060
+ return obj;
2061
+ }
2062
+
2063
+ static VALUE FUNCTION(rb_gsl_matrix,block)(VALUE obj)
2064
+ {
2065
+ GSL_TYPE(gsl_matrix) *m;
2066
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2067
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_block), 0, NULL, m->block);
2068
+ }
2069
+
2070
+ static VALUE FUNCTION(rb_gsl_matrix,info)(VALUE obj)
2071
+ {
2072
+ GSL_TYPE(gsl_matrix) *m;
2073
+ char buf[256];
2074
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2075
+ sprintf(buf, "Class: %s\n", rb_class2name(CLASS_OF(obj)));
2076
+ #ifdef RUBY_1_9_LATER
2077
+ sprintf(buf, "%sSuperClass: %s\n", buf, rb_class2name(RCLASS_SUPER(CLASS_OF(obj))));
2078
+ #else
2079
+ sprintf(buf, "%sSuperClass: %s\n", buf, rb_class2name(RCLASS(CLASS_OF(obj))->super));
2080
+ #endif
2081
+ sprintf(buf, "%sDimension: %dx%d\n", buf, (int) m->size1, (int) m->size2);
2082
+ sprintf(buf, "%sSize: %d\n", buf, (int) (m->size1*m->size2));
2083
+ return rb_str_new2(buf);
2084
+ }
2085
+
2086
+ static VALUE FUNCTION(rb_gsl_matrix,any)(VALUE obj)
2087
+ {
2088
+ GSL_TYPE(gsl_matrix) *m;
2089
+ QUALIFIED_VIEW(gsl_vector,view) vv;
2090
+ GSL_TYPE(gsl_vector) *v;
2091
+ gsl_vector_int *vnew;
2092
+ size_t j;
2093
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2094
+ vnew = gsl_vector_int_alloc(m->size2);
2095
+ for (j = 0; j < m->size2; j++) {
2096
+ vv = FUNCTION(gsl_matrix,column)(m, j);
2097
+ v = &vv.vector;
2098
+ if (FUNCTION(gsl_vector,isnull(v))) gsl_vector_int_set(vnew, j, 0);
2099
+ else gsl_vector_int_set(vnew, j, 1);
2100
+ }
2101
+ return Data_Wrap_Struct(cgsl_vector_int, 0, gsl_vector_int_free, vnew);
2102
+ }
2103
+
2104
+ static VALUE FUNCTION(rb_gsl_matrix,all)(VALUE obj)
2105
+ {
2106
+ GSL_TYPE(gsl_matrix) *m;
2107
+ QUALIFIED_VIEW(gsl_vector,view) vv;
2108
+ GSL_TYPE(gsl_vector) *v;
2109
+ gsl_vector_int *vnew;
2110
+ size_t i, j;
2111
+ int flag = 0;
2112
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2113
+ vnew = gsl_vector_int_alloc(m->size2);
2114
+ for (j = 0; j < m->size2; j++) {
2115
+ vv = FUNCTION(gsl_matrix,column)(m, j);
2116
+ v = &vv.vector;
2117
+ /* if (FUNCTION(gsl_vector,isnull(v))) gsl_vector_int_set(vnew, j, 0);
2118
+ else gsl_vector_int_set(vnew, j, 1);*/
2119
+ for (i = 0; i < v->size; i++) {
2120
+ if (FUNCTION(gsl_vector,get)(v, i) == (BASE) 0) {
2121
+ gsl_vector_int_set(vnew, j, 0);
2122
+ flag = 0;
2123
+ break;
2124
+ } else {
2125
+ flag = 1;
2126
+ }
2127
+ }
2128
+ if (flag == 1) gsl_vector_int_set(vnew, j, 1);
2129
+ }
2130
+ return Data_Wrap_Struct(cgsl_vector_int, 0, gsl_vector_int_free, vnew);
2131
+ }
2132
+
2133
+ static VALUE FUNCTION(rb_gsl_matrix,rot90)(int argc, VALUE *argv, VALUE obj)
2134
+ {
2135
+ GSL_TYPE(gsl_matrix) *m, *mnew, *mtmp;
2136
+ int p;
2137
+ switch (argc) {
2138
+ case 0:
2139
+ p = 1;
2140
+ break;
2141
+ case 1:
2142
+ p = FIX2INT(argv[0])%4;
2143
+ break;
2144
+ default:
2145
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 0 or 1)", argc);
2146
+ break;
2147
+ }
2148
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2149
+ switch (p) {
2150
+ case 0:
2151
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
2152
+ // FUNCTION(gsl_matrix,memcpy(mnew, m));
2153
+ FUNCTION(gsl_matrix,memcpy)(mnew, m);
2154
+ break;
2155
+ case 1:
2156
+ case -3:
2157
+ mtmp = FUNCTION(gsl_matrix,alloc)(m->size2, m->size1);
2158
+ FUNCTION(gsl_matrix,transpose_memcpy(mtmp, m));
2159
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size2, m->size1);
2160
+ FUNCTION(mygsl_matrix,reverse_rows)(mnew, mtmp);
2161
+ FUNCTION(gsl_matrix,free)(mtmp);
2162
+ break;
2163
+ case 2:
2164
+ case -2:
2165
+ mtmp = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
2166
+ FUNCTION(mygsl_matrix,reverse_rows)(mtmp,m);
2167
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
2168
+ FUNCTION(mygsl_matrix,reverse_columns)(mnew, mtmp);
2169
+ FUNCTION(gsl_matrix,free)(mtmp);
2170
+ break;
2171
+ case 3:
2172
+ case -1:
2173
+ mtmp = FUNCTION(gsl_matrix,alloc)(m->size2, m->size1);
2174
+ FUNCTION(gsl_matrix,transpose_memcpy(mtmp, m));
2175
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size2, m->size1);
2176
+ FUNCTION(mygsl_matrix,reverse_columns)(mnew, mtmp);
2177
+ FUNCTION(gsl_matrix,free)(mtmp);
2178
+ break;
2179
+ default:
2180
+ return Qnil;
2181
+ break;
2182
+ }
2183
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
2184
+ }
2185
+
2186
+ void FUNCTION(mygsl_vector,diff)(GSL_TYPE(gsl_vector) *vdst,
2187
+ GSL_TYPE(gsl_vector) *vsrc, size_t n);
2188
+
2189
+ static VALUE FUNCTION(rb_gsl_matrix,diff)(int argc, VALUE *argv, VALUE obj)
2190
+ {
2191
+ GSL_TYPE(gsl_matrix) *m, *mnew;
2192
+ QUALIFIED_VIEW(gsl_vector,view) v1, v2;
2193
+ size_t n, j;
2194
+ switch (argc) {
2195
+ case 0:
2196
+ n = 1;
2197
+ break;
2198
+ case 1:
2199
+ n = FIX2INT(argv[0]);
2200
+ break;
2201
+ default:
2202
+ rb_raise(rb_eArgError, "wrong number of arguments (%d for 0 or 1)", argc);
2203
+ break;
2204
+ }
2205
+ if (n <= 0) return obj;
2206
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2207
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1-n, m->size2);
2208
+ if (m->size1 <= n) return obj;
2209
+ for (j = 0; j < m->size2; j++) {
2210
+ v1 = FUNCTION(gsl_matrix,column)(m, j);
2211
+ v2 = FUNCTION(gsl_matrix,column)(mnew, j);
2212
+ FUNCTION(mygsl_vector,diff)(&v2.vector, &v1.vector, n);
2213
+ }
2214
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
2215
+ }
2216
+
2217
+ static VALUE FUNCTION(rb_gsl_matrix,test)(VALUE obj, int (*f)(const double))
2218
+ {
2219
+ GSL_TYPE(gsl_matrix) *m;
2220
+ gsl_matrix_int *mi;
2221
+ size_t i, j;
2222
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2223
+ mi = gsl_matrix_int_alloc(m->size1, m->size2);
2224
+ for (i = 0; i < m->size1; i++) {
2225
+ for (j = 0; j < m->size2; j++) {
2226
+ gsl_matrix_int_set(mi, i, j, (*f)(FUNCTION(gsl_matrix,get)(m, i, j)));
2227
+ }
2228
+ }
2229
+ return Data_Wrap_Struct(cgsl_matrix_int, 0, gsl_matrix_int_free, mi);
2230
+ }
2231
+
2232
+ static VALUE FUNCTION(rb_gsl_matrix,isnan)(VALUE obj)
2233
+ {
2234
+ return FUNCTION(rb_gsl_matrix,test)(obj, gsl_isnan);
2235
+ }
2236
+
2237
+ static VALUE FUNCTION(rb_gsl_matrix,isinf)(VALUE obj)
2238
+ {
2239
+ return FUNCTION(rb_gsl_matrix,test)(obj, gsl_isinf);
2240
+ }
2241
+
2242
+ static VALUE FUNCTION(rb_gsl_matrix,finite)(VALUE obj)
2243
+ {
2244
+ return FUNCTION(rb_gsl_matrix,test)(obj, gsl_finite);
2245
+ }
2246
+
2247
+ static VALUE FUNCTION(rb_gsl_matrix,sgn)(VALUE obj)
2248
+ {
2249
+ GSL_TYPE(gsl_matrix) *m, *mnew;
2250
+ BASE x;
2251
+ size_t i, j;
2252
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2253
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
2254
+ for (i = 0; i < m->size1; i++) {
2255
+ for (j = 0; j < m->size2; j++) {
2256
+ x = FUNCTION(gsl_matrix,get)(m, i, j);
2257
+ FUNCTION(gsl_matrix,set)(mnew, i, j, (BASE)(x>0 ? 1 : (x<0 ? -1 : 0)));
2258
+ }
2259
+ }
2260
+
2261
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
2262
+ }
2263
+
2264
+ static VALUE FUNCTION(rb_gsl_matrix,abs)(VALUE obj)
2265
+ {
2266
+ GSL_TYPE(gsl_matrix) *m, *mnew;
2267
+ size_t i, j;
2268
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2269
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
2270
+ for (i = 0; i < m->size1; i++) {
2271
+ for (j = 0; j < m->size2; j++) {
2272
+ FUNCTION(gsl_matrix,set)(mnew, i, j, (BASE) fabs(FUNCTION(gsl_matrix,get)(m, i, j)));
2273
+ }
2274
+ }
2275
+
2276
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
2277
+ }
2278
+
2279
+ static VALUE FUNCTION(rb_gsl_matrix,horzcat)(VALUE obj, VALUE mm2)
2280
+ {
2281
+ GSL_TYPE(gsl_matrix) *m, *m2, *mnew;
2282
+ QUALIFIED_VIEW(gsl_vector,view) v;
2283
+ size_t j, k;
2284
+ CHECK_MAT(mm2);
2285
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2286
+ Data_Get_Struct(mm2, GSL_TYPE(gsl_matrix), m2);
2287
+ if (m->size1 != m2->size1)
2288
+ rb_raise(rb_eRuntimeError, "Different number of rows (%d and %d).",
2289
+ (int) m->size1, (int) m2->size1);
2290
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2 + m2->size2);
2291
+ for (j = 0, k = 0; j < m->size2; j++, k++) {
2292
+ v = FUNCTION(gsl_matrix,column)(m, j);
2293
+ FUNCTION(gsl_matrix,set_col)(mnew, k, &v.vector);
2294
+ }
2295
+ for (j = 0; j < m2->size2; j++, k++) {
2296
+ v = FUNCTION(gsl_matrix,column)(m2, j);
2297
+ FUNCTION(gsl_matrix,set_col)(mnew, k, &v.vector);
2298
+ }
2299
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
2300
+ }
2301
+
2302
+ static VALUE FUNCTION(rb_gsl_matrix,horzcat_singleton)(VALUE klass, VALUE mm, VALUE mm2)
2303
+ {
2304
+ CHECK_MAT(mm);
2305
+ return FUNCTION(rb_gsl_matrix,horzcat)(mm, mm2);
2306
+ }
2307
+
2308
+
2309
+ static VALUE FUNCTION(rb_gsl_matrix,vertcat)(VALUE obj, VALUE mm2)
2310
+ {
2311
+ GSL_TYPE(gsl_matrix) *m, *m2, *mnew;
2312
+ QUALIFIED_VIEW(gsl_vector,view) v;
2313
+ size_t i, k;
2314
+ CHECK_MAT(mm2);
2315
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2316
+ Data_Get_Struct(mm2, GSL_TYPE(gsl_matrix), m2);
2317
+ if (m->size2 != m2->size2)
2318
+ rb_raise(rb_eRuntimeError, "Different number of columns (%d and %d).",
2319
+ (int) m->size2, (int) m2->size2);
2320
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1 + m2->size1, m->size2);
2321
+ for (i = 0, k = 0; i < m->size1; i++, k++) {
2322
+ v = FUNCTION(gsl_matrix,row)(m, i);
2323
+ FUNCTION(gsl_matrix,set_row)(mnew, k, &v.vector);
2324
+ }
2325
+ for (i = 0; i < m2->size1; i++, k++) {
2326
+ v = FUNCTION(gsl_matrix,row)(m2, i);
2327
+ FUNCTION(gsl_matrix,set_row)(mnew, k, &v.vector);
2328
+ }
2329
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
2330
+ }
2331
+
2332
+ static VALUE FUNCTION(rb_gsl_matrix,vertcat_singleton)(VALUE klass, VALUE mm, VALUE mm2)
2333
+ {
2334
+ CHECK_MAT(mm);
2335
+ return FUNCTION(rb_gsl_matrix,vertcat)(mm, mm2);
2336
+ }
2337
+
2338
+ #ifdef GSL_1_9_LATER
2339
+ static VALUE FUNCTION(rb_gsl_matrix,property)(VALUE obj,
2340
+ int (*f)(const GSL_TYPE(gsl_matrix)*)) {
2341
+ GSL_TYPE(gsl_matrix) *m;
2342
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2343
+ return INT2FIX((*f)(m));
2344
+ }
2345
+
2346
+ static VALUE FUNCTION(rb_gsl_matrix,property2)(VALUE obj,
2347
+ int (*f)(const GSL_TYPE(gsl_matrix) *)) {
2348
+ GSL_TYPE(gsl_matrix) *m;
2349
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2350
+ if ((*f)(m)) return Qtrue;
2351
+ else return Qfalse;
2352
+ }
2353
+ static VALUE FUNCTION(rb_gsl_matrix,ispos)(VALUE obj)
2354
+ {
2355
+ return FUNCTION(rb_gsl_matrix,property)(obj, FUNCTION(gsl_matrix,ispos));
2356
+ }
2357
+ static VALUE FUNCTION(rb_gsl_matrix,ispos2)(VALUE obj)
2358
+ {
2359
+ return FUNCTION(rb_gsl_matrix,property2)(obj, FUNCTION(gsl_matrix,ispos));
2360
+ }
2361
+ static VALUE FUNCTION(rb_gsl_matrix,isneg)(VALUE obj)
2362
+ {
2363
+ return FUNCTION(rb_gsl_matrix,property)(obj, FUNCTION(gsl_matrix,isneg));
2364
+ }
2365
+ static VALUE FUNCTION(rb_gsl_matrix,isneg2)(VALUE obj)
2366
+ {
2367
+ return FUNCTION(rb_gsl_matrix,property2)(obj, FUNCTION(gsl_matrix,isneg));
2368
+ }
2369
+ #endif
2370
+
2371
+ #ifdef GSL_1_10_LATER
2372
+ static VALUE FUNCTION(rb_gsl_matrix,isnonneg)(VALUE obj)
2373
+ {
2374
+ return FUNCTION(rb_gsl_matrix,property)(obj, FUNCTION(gsl_matrix,isnonneg));
2375
+ }
2376
+ static VALUE FUNCTION(rb_gsl_matrix,isnonneg2)(VALUE obj)
2377
+ {
2378
+ return FUNCTION(rb_gsl_matrix,property2)(obj, FUNCTION(gsl_matrix,isnonneg));
2379
+ }
2380
+ #endif
2381
+
2382
+ static VALUE FUNCTION(rb_gsl_matrix,symmetrize)(VALUE obj)
2383
+ {
2384
+ GSL_TYPE(gsl_matrix) *m, *mnew;
2385
+ size_t i, j;
2386
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2387
+ if (m->size1 != m->size2)
2388
+ rb_raise(rb_eRuntimeError, "symmetrize: not a square matrix.\n");
2389
+ mnew = FUNCTION(gsl_matrix,alloc)(m->size1, m->size2);
2390
+ for (i = 0; i < m->size1; i++) {
2391
+ for (j = i; j < m->size2; j++) {
2392
+ FUNCTION(gsl_matrix,set)(mnew, i, j, FUNCTION(gsl_matrix,get)(m, i, j));
2393
+ }
2394
+ for (j = 0; j < i; j++) {
2395
+ FUNCTION(gsl_matrix,set)(mnew, i, j, FUNCTION(gsl_matrix,get)(m, j, i));
2396
+ }
2397
+ }
2398
+ return Data_Wrap_Struct(GSL_TYPE(cgsl_matrix), 0, FUNCTION(gsl_matrix,free), mnew);
2399
+ }
2400
+
2401
+ static VALUE FUNCTION(rb_gsl_matrix,symmetrize_bang)(VALUE obj)
2402
+ {
2403
+ GSL_TYPE(gsl_matrix) *m;
2404
+ size_t i, j;
2405
+ Data_Get_Struct(obj, GSL_TYPE(gsl_matrix), m);
2406
+ if (m->size1 != m->size2)
2407
+ rb_raise(rb_eRuntimeError, "symmetrize: not a square matrix.\n");
2408
+ for (i = 0; i < m->size1; i++) {
2409
+ for (j = 0; j < i; j++) {
2410
+ FUNCTION(gsl_matrix,set)(m, i, j, FUNCTION(gsl_matrix,get)(m, j, i));
2411
+ }
2412
+ }
2413
+ return obj;
2414
+ }
2415
+
2416
+ void FUNCTION(Init_gsl_matrix,init)(VALUE module)
2417
+ {
2418
+ /* rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "new", FUNCTION(rb_gsl_matrix,alloc), -1);*/
2419
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "[]", FUNCTION(rb_gsl_matrix,alloc), -1);
2420
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "alloc", FUNCTION(rb_gsl_matrix,alloc), -1);
2421
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "calloc", FUNCTION(rb_gsl_matrix,calloc), 2);
2422
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "eye", FUNCTION(rb_gsl_matrix,eye), -1);
2423
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "ones", FUNCTION(rb_gsl_matrix,ones), -1);
2424
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "zeros", FUNCTION(rb_gsl_matrix,zeros), -1);
2425
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "diagonal",
2426
+ FUNCTION(rb_gsl_matrix,diagonal_singleton), -1);
2427
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "diag",
2428
+ FUNCTION(rb_gsl_matrix,diagonal_singleton), -1);
2429
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "identity",
2430
+ FUNCTION(rb_gsl_matrix,identity), 1);
2431
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "scalar",
2432
+ FUNCTION(rb_gsl_matrix,identity), 1);
2433
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "unit",
2434
+ FUNCTION(rb_gsl_matrix,identity), 1);
2435
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "I",
2436
+ FUNCTION(rb_gsl_matrix,identity), 1);
2437
+
2438
+ /*****/
2439
+ rb_define_method(GSL_TYPE(cgsl_matrix), "size1",
2440
+ FUNCTION(rb_gsl_matrix,size1), 0);
2441
+ rb_define_method(GSL_TYPE(cgsl_matrix), "size2",
2442
+ FUNCTION(rb_gsl_matrix,size2), 0);
2443
+ rb_define_method(GSL_TYPE(cgsl_matrix), "shape",
2444
+ FUNCTION(rb_gsl_matrix,shape), 0);
2445
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "size", "shape");
2446
+
2447
+ rb_define_method(GSL_TYPE(cgsl_matrix), "get", FUNCTION(rb_gsl_matrix,get), -1);
2448
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "[]", "get");
2449
+ rb_define_method(GSL_TYPE(cgsl_matrix), "set", FUNCTION(rb_gsl_matrix,set), -1);
2450
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "[]=", "set");
2451
+
2452
+ rb_define_method(GSL_TYPE(cgsl_matrix), "set_all",
2453
+ FUNCTION(rb_gsl_matrix,set_all), 1);
2454
+ rb_define_method(GSL_TYPE(cgsl_matrix), "set_zero",
2455
+ FUNCTION(rb_gsl_matrix,set_zero), 0);
2456
+ rb_define_method(GSL_TYPE(cgsl_matrix), "set_identity",
2457
+ FUNCTION(rb_gsl_matrix,set_identity), 0);
2458
+
2459
+ rb_define_method(GSL_TYPE(cgsl_matrix), "print",
2460
+ FUNCTION(rb_gsl_matrix,print), 0);
2461
+ rb_define_method(GSL_TYPE(cgsl_matrix), "inspect",
2462
+ FUNCTION(rb_gsl_matrix,inspect), 0);
2463
+ rb_define_method(GSL_TYPE(cgsl_matrix), "to_s",
2464
+ FUNCTION(rb_gsl_matrix,to_s), 0);
2465
+
2466
+ rb_define_method(GSL_TYPE(cgsl_matrix), "set_diagonal",
2467
+ FUNCTION(rb_gsl_matrix,set_diagonal), 1);
2468
+
2469
+ rb_define_method(GSL_TYPE(cgsl_matrix), "get_row",
2470
+ FUNCTION(rb_gsl_matrix,get_row), 1);
2471
+ rb_define_method(GSL_TYPE(cgsl_matrix), "get_column",
2472
+ FUNCTION(rb_gsl_matrix,get_col), 1);
2473
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "get_col", "get_column");
2474
+ rb_define_method(GSL_TYPE(cgsl_matrix), "set_column",
2475
+ FUNCTION(rb_gsl_matrix,set_col), 2);
2476
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "set_col", "set_column");
2477
+ rb_define_method(GSL_TYPE(cgsl_matrix), "set_row",
2478
+ FUNCTION(rb_gsl_matrix,set_row), 2);
2479
+
2480
+ rb_define_method(GSL_TYPE(cgsl_matrix), "clone",
2481
+ FUNCTION(rb_gsl_matrix,clone), 0);
2482
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "duplicate", "clone");
2483
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "dup", "clone");
2484
+ rb_define_method(GSL_TYPE(cgsl_matrix), "isnull",
2485
+ FUNCTION(rb_gsl_matrix,isnull), 0);
2486
+ rb_define_method(GSL_TYPE(cgsl_matrix), "isnull?",
2487
+ FUNCTION(rb_gsl_matrix,isnull2), 0);
2488
+
2489
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "memcpy",
2490
+ FUNCTION(rb_gsl_matrix,memcpy), 2);
2491
+
2492
+ rb_define_method(GSL_TYPE(cgsl_matrix), "swap_rows",
2493
+ FUNCTION(rb_gsl_matrix,swap_rows), 2);
2494
+ rb_define_method(GSL_TYPE(cgsl_matrix), "swap_rows!",
2495
+ FUNCTION(rb_gsl_matrix,swap_rows_bang), 2);
2496
+ rb_define_method(GSL_TYPE(cgsl_matrix), "swap_columns",
2497
+ FUNCTION(rb_gsl_matrix,swap_columns), 2);
2498
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "swap_cols", "swap_columns");
2499
+ rb_define_method(GSL_TYPE(cgsl_matrix), "swap_columns!",
2500
+ FUNCTION(rb_gsl_matrix,swap_columns_bang), 2);
2501
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "swap_cols!", "swap_columns!");
2502
+ rb_define_method(GSL_TYPE(cgsl_matrix), "swap_rowcol",
2503
+ FUNCTION(rb_gsl_matrix,swap_rowcol), 2);
2504
+ rb_define_method(GSL_TYPE(cgsl_matrix), "swap_rowcol!",
2505
+ FUNCTION(rb_gsl_matrix,swap_rowcol_bang), 2);
2506
+ rb_define_method(GSL_TYPE(cgsl_matrix), "transpose_memcpy",
2507
+ FUNCTION(rb_gsl_matrix,transpose_memcpy), 0);
2508
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "transpose", "transpose_memcpy");
2509
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "trans", "transpose_memcpy");
2510
+ rb_define_method(GSL_TYPE(cgsl_matrix), "transpose!",
2511
+ FUNCTION(rb_gsl_matrix,transpose_bang), 0);
2512
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "trans!", "transpose!");
2513
+ rb_define_method(GSL_TYPE(cgsl_matrix), "reverse_columns",
2514
+ FUNCTION(rb_gsl_matrix,reverse_columns), 0);
2515
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "fliplr", "reverse_columns");
2516
+ rb_define_method(GSL_TYPE(cgsl_matrix), "reverse_columns!",
2517
+ FUNCTION(rb_gsl_matrix,reverse_columns_bang), 0);
2518
+ rb_define_method(GSL_TYPE(cgsl_matrix), "reverse_rows",
2519
+ FUNCTION(rb_gsl_matrix,reverse_rows), 0);
2520
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "flipud", "reverse_rows");
2521
+ rb_define_method(GSL_TYPE(cgsl_matrix), "reverse_rows!",
2522
+ FUNCTION(rb_gsl_matrix,reverse_rows_bang), 0);
2523
+ /*****/
2524
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "swap",
2525
+ FUNCTION(rb_gsl_matrix,swap), 2);
2526
+
2527
+ rb_define_method(GSL_TYPE(cgsl_matrix), "max", FUNCTION(rb_gsl_matrix,max), 0);
2528
+ rb_define_method(GSL_TYPE(cgsl_matrix), "min", FUNCTION(rb_gsl_matrix,min), 0);
2529
+ rb_define_method(GSL_TYPE(cgsl_matrix), "minmax",
2530
+ FUNCTION(rb_gsl_matrix,minmax), 0);
2531
+ rb_define_method(GSL_TYPE(cgsl_matrix), "max_index",
2532
+ FUNCTION(rb_gsl_matrix,max_index), 0);
2533
+ rb_define_method(GSL_TYPE(cgsl_matrix), "min_index",
2534
+ FUNCTION(rb_gsl_matrix,min_index), 0);
2535
+ rb_define_method(GSL_TYPE(cgsl_matrix), "minmax_index",
2536
+ FUNCTION(rb_gsl_matrix,minmax_index), 0);
2537
+
2538
+ rb_define_method(GSL_TYPE(cgsl_matrix), "fwrite",
2539
+ FUNCTION(rb_gsl_matrix,fwrite), 1);
2540
+ rb_define_method(GSL_TYPE(cgsl_matrix), "fread",
2541
+ FUNCTION(rb_gsl_matrix,fread), 1);
2542
+ rb_define_method(GSL_TYPE(cgsl_matrix), "fprintf",
2543
+ FUNCTION(rb_gsl_matrix,fprintf), -1);
2544
+ rb_define_method(GSL_TYPE(cgsl_matrix), "printf",
2545
+ FUNCTION(rb_gsl_matrix,printf), -1);
2546
+ rb_define_method(GSL_TYPE(cgsl_matrix), "fscanf",
2547
+ FUNCTION(rb_gsl_matrix,fscanf), 1);
2548
+
2549
+ rb_define_method(GSL_TYPE(cgsl_matrix), "trace",
2550
+ FUNCTION(rb_gsl_matrix,trace), 0);
2551
+
2552
+ rb_define_method(GSL_TYPE(cgsl_matrix), "-@",
2553
+ FUNCTION(rb_gsl_matrix,uminus), 0);
2554
+ rb_define_method(GSL_TYPE(cgsl_matrix), "+@",
2555
+ FUNCTION(rb_gsl_matrix,uplus), 0);
2556
+
2557
+
2558
+ /*****/
2559
+ rb_define_method(GSL_TYPE(cgsl_matrix), "submatrix",
2560
+ FUNCTION(rb_gsl_matrix,submatrix), -1);
2561
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "view", "submatrix");
2562
+
2563
+ rb_define_method(GSL_TYPE(cgsl_matrix), "row", FUNCTION(rb_gsl_matrix,row), 1);
2564
+ /* rb_define_alias(GSL_TYPE(cgsl_matrix), "[]", "row");*/
2565
+ rb_define_method(GSL_TYPE(cgsl_matrix), "column",
2566
+ FUNCTION(rb_gsl_matrix,column), 1);
2567
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "col", "column");
2568
+
2569
+ #ifdef GSL_1_10_LATER
2570
+ rb_define_method(GSL_TYPE(cgsl_matrix), "subrow",
2571
+ FUNCTION(rb_gsl_matrix,subrow), 3);
2572
+ rb_define_method(GSL_TYPE(cgsl_matrix), "subcolumn",
2573
+ FUNCTION(rb_gsl_matrix,subcolumn), 3);
2574
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "subcol", "subcolumn");
2575
+ #endif
2576
+
2577
+ rb_define_method(GSL_TYPE(cgsl_matrix), "diagonal",
2578
+ FUNCTION(rb_gsl_matrix,diagonal), 0);
2579
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "diag", "diagonal");
2580
+
2581
+ rb_define_method(GSL_TYPE(cgsl_matrix), "subdiagonal",
2582
+ FUNCTION(rb_gsl_matrix,subdiagonal), 1);
2583
+ rb_define_method(GSL_TYPE(cgsl_matrix), "superdiagonal",
2584
+ FUNCTION(rb_gsl_matrix,superdiagonal), 1);
2585
+ rb_define_method(GSL_TYPE(cgsl_matrix), "vector_view",
2586
+ FUNCTION(rb_gsl_matrix,vector_view), 0);
2587
+
2588
+ rb_define_method(GSL_TYPE(cgsl_matrix), "each_row",
2589
+ FUNCTION(rb_gsl_matrix,each_row), 0);
2590
+ rb_define_method(GSL_TYPE(cgsl_matrix), "each_col",
2591
+ FUNCTION(rb_gsl_matrix,each_col), 0);
2592
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "each_column", "each_col");
2593
+
2594
+ rb_define_method(GSL_TYPE(cgsl_matrix), "scale",
2595
+ FUNCTION(rb_gsl_matrix,scale), 1);
2596
+ rb_define_method(GSL_TYPE(cgsl_matrix), "scale!",
2597
+ FUNCTION(rb_gsl_matrix,scale_bang), 1);
2598
+ rb_define_method(GSL_TYPE(cgsl_matrix), "add_constant",
2599
+ FUNCTION(rb_gsl_matrix,add_constant), 1);
2600
+ rb_define_method(GSL_TYPE(cgsl_matrix), "add_constant!",
2601
+ FUNCTION(rb_gsl_matrix,add_constant_bang), 1);
2602
+
2603
+ rb_define_method(GSL_TYPE(cgsl_matrix), "equal?",
2604
+ FUNCTION(rb_gsl_matrix,equal), -1);
2605
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "==", "equal?");
2606
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "equal?",
2607
+ FUNCTION(rb_gsl_matrix,equal_singleton), -1);
2608
+
2609
+ rb_define_method(GSL_TYPE(cgsl_matrix), "power",
2610
+ FUNCTION(rb_gsl_matrix,power), 1);
2611
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "**", "power");
2612
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "^", "power");
2613
+
2614
+ rb_define_method(GSL_TYPE(cgsl_matrix), "collect",
2615
+ FUNCTION(rb_gsl_matrix,collect), 0);
2616
+ rb_define_method(GSL_TYPE(cgsl_matrix), "collect!",
2617
+ FUNCTION(rb_gsl_matrix,collect_bang), 0);
2618
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "map", "collect");
2619
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "map!", "collect!");
2620
+ #ifdef HAVE_TENSOR_TENSOR_H
2621
+ rb_define_method(GSL_TYPE(cgsl_matrix), "to_tensor",
2622
+ FUNCTION(rb_gsl_matrix,to_tensor), 0);
2623
+ #endif
2624
+
2625
+ /*****/
2626
+
2627
+ rb_define_method(GSL_TYPE(cgsl_matrix), "to_a",
2628
+ FUNCTION(rb_gsl_matrix,to_a), 0);
2629
+ rb_define_method(GSL_TYPE(cgsl_matrix), "to_v",
2630
+ FUNCTION(rb_gsl_matrix,to_v), 0);
2631
+ rb_define_method(GSL_TYPE(cgsl_matrix), "to_vview",
2632
+ FUNCTION(rb_gsl_matrix,to_vview), 0);
2633
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "data", "to_vview");
2634
+ rb_define_method(GSL_TYPE(cgsl_matrix), "norm",
2635
+ FUNCTION(rb_gsl_matrix,norm), 0);
2636
+
2637
+ rb_define_method(GSL_TYPE(cgsl_matrix), "upper",
2638
+ FUNCTION(rb_gsl_matrix,upper), 0);
2639
+ rb_define_method(GSL_TYPE(cgsl_matrix), "lower",
2640
+ FUNCTION(rb_gsl_matrix,lower), 0);
2641
+ rb_define_method(GSL_TYPE(cgsl_matrix), "block",
2642
+ FUNCTION(rb_gsl_matrix,block), 0);
2643
+
2644
+ /***** Special matrices *****/
2645
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "pascal",
2646
+ FUNCTION(rb_gsl_matrix,pascal1), 1);
2647
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "vandermonde",
2648
+ FUNCTION(rb_gsl_matrix,vandermonde), 1);
2649
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "vander",
2650
+ FUNCTION(rb_gsl_matrix,vandermonde), 1);
2651
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "toeplitz",
2652
+ FUNCTION(rb_gsl_matrix,toeplitz), 1);
2653
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "circulant",
2654
+ FUNCTION(rb_gsl_matrix,circulant), 1);
2655
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "indgen",
2656
+ FUNCTION(rb_gsl_matrix,indgen_singleton), -1);
2657
+ rb_define_method(GSL_TYPE(cgsl_matrix), "indgen",
2658
+ FUNCTION(rb_gsl_matrix,indgen), -1);
2659
+ rb_define_method(GSL_TYPE(cgsl_matrix), "indgen!",
2660
+ FUNCTION(rb_gsl_matrix,indgen_bang), -1);
2661
+
2662
+ rb_define_method(GSL_TYPE(cgsl_matrix), "info",
2663
+ FUNCTION(rb_gsl_matrix,info), 0);
2664
+ #ifdef BASE_DOUBLE
2665
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "hilbert",
2666
+ FUNCTION(rb_gsl_matrix,hilbert), 1);
2667
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "hilb",
2668
+ FUNCTION(rb_gsl_matrix,hilbert), 1);
2669
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "invhilbert",
2670
+ FUNCTION(rb_gsl_matrix,invhilbert), 1);
2671
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "invhilb",
2672
+ FUNCTION(rb_gsl_matrix,invhilbert), 1);
2673
+ #endif
2674
+
2675
+ rb_define_method(GSL_TYPE(cgsl_matrix), "any",
2676
+ FUNCTION(rb_gsl_matrix,any), 0);
2677
+ rb_define_method(GSL_TYPE(cgsl_matrix), "all",
2678
+ FUNCTION(rb_gsl_matrix,all), 0);
2679
+ rb_define_method(GSL_TYPE(cgsl_matrix), "rot90",
2680
+ FUNCTION(rb_gsl_matrix,rot90), -1);
2681
+
2682
+ rb_define_method(GSL_TYPE(cgsl_matrix), "diff",
2683
+ FUNCTION(rb_gsl_matrix,diff), -1);
2684
+ rb_define_method(GSL_TYPE(cgsl_matrix), "isnan", FUNCTION(rb_gsl_matrix,isnan), 0);
2685
+ rb_define_method(GSL_TYPE(cgsl_matrix), "isinf", FUNCTION(rb_gsl_matrix,isinf), 0);
2686
+ rb_define_method(GSL_TYPE(cgsl_matrix), "finite", FUNCTION(rb_gsl_matrix,finite), 0);
2687
+
2688
+ rb_define_method(GSL_TYPE(cgsl_matrix), "sgn", FUNCTION(rb_gsl_matrix,sgn), 0);
2689
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "signum", "sgn");
2690
+ rb_define_method(GSL_TYPE(cgsl_matrix), "abs", FUNCTION(rb_gsl_matrix,abs), 0);
2691
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "fabs", "abs");
2692
+
2693
+ rb_define_method(GSL_TYPE(cgsl_matrix), "horzcat", FUNCTION(rb_gsl_matrix,horzcat), 1);
2694
+ rb_define_alias(GSL_TYPE(cgsl_matrix), "cat", "horzcat");
2695
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "horzcat", FUNCTION(rb_gsl_matrix,horzcat_singleton), 2);
2696
+
2697
+ rb_define_method(GSL_TYPE(cgsl_matrix), "vertcat", FUNCTION(rb_gsl_matrix,vertcat), 1);
2698
+ rb_define_singleton_method(GSL_TYPE(cgsl_matrix), "vertcat", FUNCTION(rb_gsl_matrix,vertcat_singleton), 2);
2699
+
2700
+ #ifdef GSL_1_9_LATER
2701
+ rb_define_method(GSL_TYPE(cgsl_matrix), "ispos", FUNCTION(rb_gsl_matrix,ispos), 0);
2702
+ rb_define_method(GSL_TYPE(cgsl_matrix), "ispos?", FUNCTION(rb_gsl_matrix,ispos2), 0);
2703
+ rb_define_method(GSL_TYPE(cgsl_matrix), "isneg", FUNCTION(rb_gsl_matrix,isneg), 0);
2704
+ rb_define_method(GSL_TYPE(cgsl_matrix), "isneg?", FUNCTION(rb_gsl_matrix,isneg2), 0);
2705
+ #endif
2706
+
2707
+ #ifdef GSL_1_10_LATER
2708
+ rb_define_method(GSL_TYPE(cgsl_matrix), "isnonneg", FUNCTION(rb_gsl_matrix,isnonneg), 0);
2709
+ rb_define_method(GSL_TYPE(cgsl_matrix), "isnonneg?", FUNCTION(rb_gsl_matrix,isnonneg2), 0);
2710
+ #endif
2711
+
2712
+ rb_define_method(GSL_TYPE(cgsl_matrix), "symmetrize", FUNCTION(rb_gsl_matrix,symmetrize), 0);
2713
+ rb_define_method(GSL_TYPE(cgsl_matrix), "symmetrize!", FUNCTION(rb_gsl_matrix,symmetrize_bang), 0);
2714
+ }
2715
+
2716
+ #undef NUMCONV
2717
+ #undef NUMCONV2
2718
+ #undef PRINTF_FORMAT
2719
+ #undef VEC_ROW_COL
2720
+ #undef VEC_P
2721
+ #undef C_TO_VALUE
2722
+ #undef C_TO_VALUE2
2723
+ #undef VEC_COL_P
2724
+ #undef VEC_ROW_P
2725
+ #undef CHECK_VEC
2726
+ #undef VEC_VIEW_P
2727
+
2728
+ #undef MAT_ROW_COL
2729
+ #undef MAT_P
2730
+ #undef MAT_COL_P
2731
+ #undef MAT_ROW_P
2732
+ #undef CHECK_MAT
2733
+ #undef MAT_VIEW_P