gsl 1.12.108

Sign up to get free protection for your applications and to get access to all the features.
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,750 @@
1
+ /*
2
+ histogram3d_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
+ /*
13
+ This source code is created modifying gsl-1.5/histogram/ *2d.c
14
+ */
15
+
16
+ #include "rb_gsl_histogram.h"
17
+
18
+ mygsl_histogram3d* mygsl_histogram3d_alloc(const size_t nx, const size_t ny,
19
+ const size_t nz)
20
+ {
21
+ mygsl_histogram3d *h = NULL;
22
+ if (nx == 0) GSL_ERROR_VAL ("histogram3d length nx must be positive integer",
23
+ GSL_EDOM, 0);
24
+ if (ny == 0) GSL_ERROR_VAL ("histogram3d length ny must be positive integer",
25
+ GSL_EDOM, 0);
26
+ if (nz == 0) GSL_ERROR_VAL ("histogram3d length nz must be positive integer",
27
+ GSL_EDOM, 0);
28
+ h = (mygsl_histogram3d *) malloc(sizeof(mygsl_histogram3d));
29
+ if (h == NULL) GSL_ERROR_VAL ("failed to allocate space for histogram3d struct",
30
+ GSL_ENOMEM, 0);
31
+ h->xrange = (double *) malloc ((nx + 1) * sizeof (double));
32
+ if (h->xrange == 0) {
33
+ free (h); /* exception in constructor, avoid memory leak */
34
+ GSL_ERROR_VAL ("failed to allocate space for histogram3d x ranges",
35
+ GSL_ENOMEM, 0);
36
+ }
37
+ h->yrange = (double *) malloc ((ny + 1) * sizeof (double));
38
+ if (h->yrange == 0) {
39
+ free (h->xrange);
40
+ free (h); /* exception in constructor, avoid memory leak */
41
+ GSL_ERROR_VAL ("failed to allocate space for histogram3d y ranges",
42
+ GSL_ENOMEM, 0);
43
+ }
44
+ h->zrange = (double *) malloc ((nz + 1) * sizeof (double));
45
+ if (h->zrange == 0) {
46
+ free (h->xrange);
47
+ free (h->yrange);
48
+ free (h); /* exception in constructor, avoid memory leak */
49
+ GSL_ERROR_VAL ("failed to allocate space for histogram3d z ranges",
50
+ GSL_ENOMEM, 0);
51
+ }
52
+ h->bin = (double *) malloc (nx*ny*nz*sizeof (double));
53
+ if (h->bin == 0) {
54
+ free (h->xrange);
55
+ free (h->yrange);
56
+ free (h->zrange);
57
+ free (h); /* exception in constructor, avoid memory leak */
58
+ GSL_ERROR_VAL ("failed to allocate space for histogram bins",
59
+ GSL_ENOMEM, 0);
60
+ }
61
+ h->nx = nx;
62
+ h->ny = ny;
63
+ h->nz = nz;
64
+ return h;
65
+ }
66
+
67
+ mygsl_histogram3d* mygsl_histogram3d_calloc_uniform(const size_t nx,
68
+ const size_t ny,
69
+ const size_t nz,
70
+ const double xmin,
71
+ const double xmax,
72
+ const double ymin,
73
+ const double ymax,
74
+ const double zmin,
75
+ const double zmax)
76
+ {
77
+ mygsl_histogram3d *h;
78
+ size_t i;
79
+ h = mygsl_histogram3d_alloc(nx, ny, nz);
80
+ for (i = 0; i < nx + 1; i++)
81
+ h->xrange[i] = xmin + ((double) i / (double) nx) * (xmax - xmin);
82
+ for (i = 0; i < ny + 1; i++)
83
+ h->yrange[i] = ymin + ((double) i / (double) ny) * (ymax - ymin);
84
+ for (i = 0; i < nz + 1; i++)
85
+ h->zrange[i] = zmin + ((double) i / (double) nz) * (zmax - zmin);
86
+ return h;
87
+ }
88
+
89
+ mygsl_histogram3d* mygsl_histogram3d_calloc(const size_t nx,
90
+ const size_t ny,
91
+ const size_t nz)
92
+ {
93
+ mygsl_histogram3d *h;
94
+ size_t i;
95
+ h = mygsl_histogram3d_alloc(nx, ny, nz);
96
+ for (i = 0; i < nx + 1; i++)
97
+ h->xrange[i] = i;
98
+ for (i = 0; i < ny + 1; i++)
99
+ h->yrange[i] = i;
100
+ for (i = 0; i < nz + 1; i++)
101
+ h->zrange[i] = i;
102
+ for (i = 0; i < nx*ny*nz; i++)
103
+ h->bin[i] = 0;
104
+ return h;
105
+ }
106
+
107
+ int mygsl_histogram3d_set_ranges_uniform (mygsl_histogram3d * h,
108
+ double xmin, double xmax,
109
+ double ymin, double ymax,
110
+ double zmin, double zmax)
111
+ {
112
+ size_t i;
113
+ const size_t nx = h->nx, ny = h->ny, nz = h->nz;
114
+
115
+ if (xmin >= xmax) GSL_ERROR_VAL ("xmin must be less than xmax", GSL_EINVAL, 0);
116
+ if (ymin >= ymax) GSL_ERROR_VAL ("ymin must be less than ymax", GSL_EINVAL, 0);
117
+ if (zmin >= zmax) GSL_ERROR_VAL ("zmin must be less than zmax", GSL_EINVAL, 0);
118
+
119
+ for (i = 0; i <= nx; i++)
120
+ h->xrange[i] = xmin + ((double) i / (double) nx) * (xmax - xmin);
121
+ for (i = 0; i <= ny; i++)
122
+ h->yrange[i] = ymin + ((double) i / (double) ny) * (ymax - ymin);
123
+ for (i = 0; i <= nz; i++)
124
+ h->zrange[i] = zmin + ((double) i / (double) nz) * (zmax - zmin);
125
+ for (i = 0; i < nx*ny*nz; i++) h->bin[i] = 0;
126
+ return GSL_SUCCESS;
127
+ }
128
+
129
+ int mygsl_histogram3d_set_ranges(mygsl_histogram3d * h,
130
+ const double xrange[], size_t xsize,
131
+ const double yrange[], size_t ysize,
132
+ const double zrange[], size_t zsize)
133
+ {
134
+ size_t i;
135
+ const size_t nx = h->nx, ny = h->ny, nz = h->nz;
136
+ if (xsize != (nx + 1))
137
+ GSL_ERROR_VAL ("size of xrange must match size of histogram",
138
+ GSL_EINVAL, 0);
139
+ if (ysize != (ny + 1))
140
+ GSL_ERROR_VAL ("size of yrange must match size of histogram",
141
+ GSL_EINVAL, 0);
142
+ if (zsize != (nz + 1))
143
+ GSL_ERROR_VAL ("size of yrange must match size of histogram",
144
+ GSL_EINVAL, 0);
145
+ memcpy(h->xrange, xrange, sizeof(double)*xsize);
146
+ memcpy(h->yrange, yrange, sizeof(double)*ysize);
147
+ memcpy(h->zrange, zrange, sizeof(double)*zsize);
148
+ for (i = 0; i < nx*ny*nz; i++) h->bin[i] = 0;
149
+ return GSL_SUCCESS;
150
+ }
151
+
152
+ void mygsl_histogram3d_free (mygsl_histogram3d * h)
153
+ {
154
+ free (h->xrange);
155
+ free (h->yrange);
156
+ free (h->zrange);
157
+ free (h->bin);
158
+ free (h);
159
+ }
160
+
161
+ /*****/
162
+
163
+ int mygsl_histogram3d_memcpy(mygsl_histogram3d * dest, const mygsl_histogram3d * src)
164
+ {
165
+ size_t nx = src->nx;
166
+ size_t ny = src->ny;
167
+ size_t nz = src->nz;
168
+ if (dest->nx != src->nx || dest->ny != src->ny || dest->nz != src->nz) {
169
+ GSL_ERROR ("histograms have different sizes, cannot copy",
170
+ GSL_EINVAL);
171
+ }
172
+ memcpy(dest->xrange, src->xrange, sizeof(double)*(nx+1));
173
+ memcpy(dest->yrange, src->yrange, sizeof(double)*(ny+1));
174
+ memcpy(dest->zrange, src->zrange, sizeof(double)*(nz+1));
175
+ memcpy(dest->bin, src->bin, sizeof(double)*nx*ny*nz);
176
+ return GSL_SUCCESS;
177
+ }
178
+
179
+ mygsl_histogram3d* mygsl_histogram3d_clone(const mygsl_histogram3d * src)
180
+ {
181
+ mygsl_histogram3d *h;
182
+ h = mygsl_histogram3d_alloc(src->nx, src->ny, src->nz);
183
+ mygsl_histogram3d_memcpy(h, src);
184
+ return h;
185
+ }
186
+
187
+ /*****/
188
+
189
+ int mygsl_histogram3d_fread(FILE * stream, mygsl_histogram3d * h)
190
+ {
191
+ int status = gsl_block_raw_fread (stream, h->xrange, h->nx + 1, 1);
192
+ if (status) return status;
193
+ status = gsl_block_raw_fread (stream, h->yrange, h->ny + 1, 1);
194
+ if (status) return status;
195
+ status = gsl_block_raw_fread (stream, h->zrange, h->nz + 1, 1);
196
+ if (status) return status;
197
+ status = gsl_block_raw_fread (stream, h->bin, h->nx * h->ny * h->nz, 1);
198
+ return status;
199
+ }
200
+
201
+ int mygsl_histogram3d_fwrite(FILE * stream, const mygsl_histogram3d * h)
202
+ {
203
+ int status = gsl_block_raw_fwrite (stream, h->xrange, h->nx + 1, 1);
204
+ if (status) return status;
205
+ status = gsl_block_raw_fwrite (stream, h->yrange, h->ny + 1, 1);
206
+ if (status) return status;
207
+ status = gsl_block_raw_fwrite (stream, h->zrange, h->nz + 1, 1);
208
+ if (status) return status;
209
+ status = gsl_block_raw_fwrite (stream, h->bin, h->nx * h->ny * h->nz, 1);
210
+ return status;
211
+ }
212
+
213
+ int mygsl_histogram3d_increment(mygsl_histogram3d * h, double x, double y, double z)
214
+ {
215
+ int status = mygsl_histogram3d_accumulate (h, x, y, z, 1.0);
216
+ return status;
217
+ }
218
+
219
+ int mygsl_find (const size_t n, const double range[], const double x, size_t * i);
220
+ int mygsl_find2d (const size_t nx, const double xrange[],
221
+ const size_t ny, const double yrange[],
222
+ const double x, const double y,
223
+ size_t * i, size_t * j);
224
+ int mygsl_find3d (const size_t nx, const double xrange[],
225
+ const size_t ny, const double yrange[],
226
+ const size_t nz, const double zrange[],
227
+ const double x, const double y, const double z,
228
+ size_t * i, size_t * j, size_t *k);
229
+
230
+ int mygsl_histogram3d_accumulate(mygsl_histogram3d * h,
231
+ double x, double y, double z, double weight)
232
+ {
233
+ const size_t nx = h->nx;
234
+ const size_t ny = h->ny;
235
+ const size_t nz = h->nz;
236
+ size_t i = 0, j = 0, k = 0;
237
+ int status = mygsl_find3d (h->nx, h->xrange, h->ny, h->yrange, h->nz, h->zrange,
238
+ x, y, z, &i, &j, &k);
239
+ if (status) return GSL_EDOM;
240
+ if (i >= nx) GSL_ERROR ("index lies outside valid range of 0 .. nx - 1",
241
+ GSL_ESANITY);
242
+ if (j >= ny) GSL_ERROR ("index lies outside valid range of 0 .. ny - 1",
243
+ GSL_ESANITY);
244
+ if (k >= nz) GSL_ERROR ("index lies outside valid range of 0 .. nz - 1",
245
+ GSL_ESANITY);
246
+ h->bin[i*ny*nz + j*nz + k] += weight;
247
+ return GSL_SUCCESS;
248
+ }
249
+
250
+ int mygsl_histogram3d_increment2(mygsl_histogram3d * h,
251
+ double x, double y, double z)
252
+ {
253
+ return mygsl_histogram3d_accumulate2(h, x, y, z, 1.0);
254
+ }
255
+
256
+ int mygsl_histogram3d_accumulate2(mygsl_histogram3d * h,
257
+ double x, double y, double z, double weight)
258
+ {
259
+ const size_t nx = h->nx;
260
+ const size_t ny = h->ny;
261
+ const size_t nz = h->nz;
262
+ size_t i = 0, j = 0, k = 0;
263
+ int status;
264
+ if (x < h->xrange[0]) x = h->xrange[0] + 4*GSL_DBL_EPSILON;
265
+ if (x > h->xrange[h->nx]) x = h->xrange[h->nx] - 4*GSL_DBL_EPSILON;
266
+ if (y < h->yrange[0]) y = h->yrange[0] + 4*GSL_DBL_EPSILON;
267
+ if (y > h->yrange[h->ny]) y = h->yrange[h->ny] - 4*GSL_DBL_EPSILON;
268
+ if (z < h->zrange[0]) z = h->zrange[0] + 4*GSL_DBL_EPSILON;
269
+ if (z > h->zrange[h->nz]) z = h->zrange[h->nz] - 4*GSL_DBL_EPSILON;
270
+ status = mygsl_find3d (h->nx, h->xrange, h->ny, h->yrange, h->nz, h->zrange,
271
+ x, y, z, &i, &j, &k);
272
+ if (status) return GSL_EDOM;
273
+ if (i >= nx) GSL_ERROR ("index lies outside valid range of 0 .. nx - 1",
274
+ GSL_ESANITY);
275
+ if (j >= ny) GSL_ERROR ("index lies outside valid range of 0 .. ny - 1",
276
+ GSL_ESANITY);
277
+ if (k >= nz) GSL_ERROR ("index lies outside valid range of 0 .. nz - 1",
278
+ GSL_ESANITY);
279
+ h->bin[i*ny*nz + j*nz + k] += weight;
280
+ return GSL_SUCCESS;
281
+ }
282
+
283
+ double mygsl_histogram3d_get(const mygsl_histogram3d * h, const size_t i,
284
+ const size_t j, const size_t k)
285
+ {
286
+ const size_t nx = h->nx;
287
+ const size_t ny = h->ny;
288
+ const size_t nz = h->nz;
289
+ if (i >= nx) GSL_ERROR_VAL ("index i lies outside valid range of 0 .. nx - 1",
290
+ GSL_EDOM, 0);
291
+ if (j >= ny) GSL_ERROR_VAL ("index j lies outside valid range of 0 .. ny - 1",
292
+ GSL_EDOM, 0);
293
+ if (k >= nz) GSL_ERROR_VAL ("index k lies outside valid range of 0 .. nz - 1",
294
+ GSL_EDOM, 0);
295
+ return h->bin[i*ny*nz + j*nz + k];
296
+ }
297
+
298
+ int mygsl_histogram3d_get_xrange(const mygsl_histogram3d * h, const size_t i,
299
+ double *xlower, double *xupper)
300
+ {
301
+ const size_t nx = h->nx;
302
+ if (i >= nx) GSL_ERROR ("index i lies outside valid range of 0 .. nx - 1",
303
+ GSL_EDOM);
304
+ *xlower = h->xrange[i];
305
+ *xupper = h->xrange[i + 1];
306
+ return GSL_SUCCESS;
307
+ }
308
+
309
+ int mygsl_histogram3d_get_yrange(const mygsl_histogram3d * h, const size_t j,
310
+ double *ylower, double *yupper)
311
+ {
312
+ const size_t ny = h->ny;
313
+ if (j >= ny) GSL_ERROR ("index j lies outside valid range of 0 .. ny - 1",
314
+ GSL_EDOM);
315
+ *ylower = h->yrange[j];
316
+ *yupper = h->yrange[j + 1];
317
+ return GSL_SUCCESS;
318
+ }
319
+
320
+ int mygsl_histogram3d_get_zrange(const mygsl_histogram3d * h, const size_t k,
321
+ double *zlower, double *zupper)
322
+ {
323
+ const size_t nz = h->nz;
324
+ if (k >= nz) GSL_ERROR ("index k lies outside valid range of 0 .. nz - 1",
325
+ GSL_EDOM);
326
+ *zlower = h->zrange[k];
327
+ *zupper = h->zrange[k + 1];
328
+ return GSL_SUCCESS;
329
+ }
330
+
331
+ int mygsl_histogram3d_find (const mygsl_histogram3d * h,
332
+ const double x, const double y, const double z,
333
+ size_t * i, size_t * j, size_t *k)
334
+ {
335
+ int status = mygsl_find(h->nx, h->xrange, x, i);
336
+ if (status) GSL_ERROR ("x not found in range of h", GSL_EDOM);
337
+ status = mygsl_find (h->ny, h->yrange, y, j);
338
+ if (status) GSL_ERROR ("y not found in range of h", GSL_EDOM);
339
+ status = mygsl_find (h->nz, h->zrange, z, k);
340
+ if (status) GSL_ERROR ("z not found in range of h", GSL_EDOM);
341
+ return GSL_SUCCESS;
342
+ }
343
+
344
+ gsl_histogram2d* mygsl_histogram3d_xyproject(const mygsl_histogram3d * h3,
345
+ size_t kstart, size_t kend)
346
+ {
347
+ gsl_histogram2d *h2;
348
+ double count;
349
+ size_t i, j, k;
350
+ h2 = gsl_histogram2d_calloc(h3->nx, h3->ny);
351
+ gsl_histogram2d_set_ranges(h2, h3->xrange, h3->nx+1, h3->yrange, h3->ny+1);
352
+ for (i = 0; i < h3->nx; i++) {
353
+ for (j = 0; j < h3->ny; j++) {
354
+ count = 0;
355
+ for (k = kstart; k <= kend; k++) {
356
+ if (k >= h3->nz) break;
357
+ count += mygsl_histogram3d_get(h3, i, j, k);
358
+ }
359
+ h2->bin[i*h2->ny + j] = count;
360
+ }
361
+ }
362
+ return h2;
363
+ }
364
+
365
+ gsl_histogram2d* mygsl_histogram3d_xzproject(const mygsl_histogram3d * h3,
366
+ size_t jstart, size_t jend)
367
+ {
368
+ gsl_histogram2d *h2;
369
+ double count;
370
+ size_t i, j, k;
371
+ h2 = gsl_histogram2d_calloc(h3->nx, h3->nz);
372
+ gsl_histogram2d_set_ranges(h2, h3->xrange, h3->nx+1, h3->zrange, h3->nz+1);
373
+ for (i = 0; i < h3->nx; i++) {
374
+ for (k = 0; k < h3->nz; k++) {
375
+ count = 0;
376
+ for (j = jstart; j <= jend; j++) {
377
+ if (j >= h3->ny) break;
378
+ count += mygsl_histogram3d_get(h3, i, j, k);
379
+ }
380
+ h2->bin[i*h2->ny + k] = count;
381
+ }
382
+ }
383
+ return h2;
384
+ }
385
+
386
+ gsl_histogram2d* mygsl_histogram3d_yzproject(const mygsl_histogram3d * h3,
387
+ size_t istart, size_t iend)
388
+ {
389
+ gsl_histogram2d *h2;
390
+ double count;
391
+ size_t i, j, k;
392
+ h2 = gsl_histogram2d_calloc(h3->ny, h3->nz);
393
+ gsl_histogram2d_set_ranges(h2, h3->yrange, h3->ny+1, h3->zrange, h3->nz+1);
394
+ for (j = 0; j < h3->ny; j++) {
395
+ for (k = 0; k < h3->nz; k++) {
396
+ count = 0;
397
+ for (i = istart; i <= iend; i++) {
398
+ if (i >= h3->nx) break;
399
+ count += mygsl_histogram3d_get(h3, i, j, k);
400
+ }
401
+ h2->bin[j*h2->ny + k] = count;
402
+ }
403
+ }
404
+ return h2;
405
+ }
406
+
407
+ int mygsl_histogram3d_scale(mygsl_histogram3d * h, double scale)
408
+ {
409
+ size_t i, n;
410
+ n = h->nx*h->ny*h->nz;
411
+ for (i = 0; i < n; i++) h->bin[i] *= scale;
412
+ return GSL_SUCCESS;
413
+ }
414
+
415
+ int mygsl_histogram3d_shift(mygsl_histogram3d * h, double shift)
416
+ {
417
+ size_t i, n;
418
+ n = h->nx*h->ny*h->nz;
419
+ for (i = 0; i < n; i++) h->bin[i] += shift;
420
+ return GSL_SUCCESS;
421
+ }
422
+
423
+ double mygsl_histogram3d_xmax(const mygsl_histogram3d * h)
424
+ {
425
+ const int nx = h->nx;
426
+ return h->xrange[nx];
427
+ }
428
+ double mygsl_histogram3d_xmin(const mygsl_histogram3d * h)
429
+ {
430
+ return h->xrange[0];
431
+ }
432
+
433
+ double mygsl_histogram3d_ymax(const mygsl_histogram3d * h)
434
+ {
435
+ const int ny = h->ny;
436
+ return h->yrange[ny];
437
+ }
438
+ double mygsl_histogram3d_ymin(const mygsl_histogram3d * h)
439
+ {
440
+ return h->yrange[0];
441
+ }
442
+
443
+ double mygsl_histogram3d_zmax(const mygsl_histogram3d * h)
444
+ {
445
+ const int nz = h->nz;
446
+ return h->zrange[nz];
447
+ }
448
+
449
+ double mygsl_histogram3d_zmin(const mygsl_histogram3d * h)
450
+ {
451
+ return h->zrange[0];
452
+ }
453
+
454
+ double mygsl_histogram3d_max_val(const mygsl_histogram3d * h)
455
+ {
456
+ const size_t nx = h->nx;
457
+ const size_t ny = h->ny;
458
+ const size_t nz = h->nz;
459
+ size_t i;
460
+ double max = h->bin[0];
461
+ for (i = 0; i < nx*ny*nz; i++) {
462
+ if (h->bin[i] > max) max = h->bin[i];
463
+ }
464
+ return max;
465
+ }
466
+
467
+ void mygsl_histogram3d_max_bin(const mygsl_histogram3d * h,
468
+ size_t *imax_out, size_t *jmax_out, size_t *kmax_out)
469
+ {
470
+ const size_t nx = h->nx;
471
+ const size_t ny = h->ny;
472
+ const size_t nz = h->nz;
473
+ size_t imax = 0, jmax = 0, kmax = 0, i, j, k;
474
+ double max = h->bin[0], x;
475
+ for (i = 0; i < nx; i++) {
476
+ for (j = 0; j < ny; j++) {
477
+ for (k = 0; k < nz; k++) {
478
+ x = h->bin[i * ny*nz + j*nz + k];
479
+ if (x > max) {
480
+ max = x;
481
+ imax = i;
482
+ jmax = j;
483
+ kmax = k;
484
+ }
485
+ }
486
+ }
487
+ }
488
+ *imax_out = imax;
489
+ *jmax_out = jmax;
490
+ *kmax_out = kmax;
491
+ }
492
+
493
+ double mygsl_histogram3d_min_val(const mygsl_histogram3d * h)
494
+ {
495
+ const size_t nx = h->nx;
496
+ const size_t ny = h->ny;
497
+ const size_t nz = h->nz;
498
+ size_t i;
499
+ double min = h->bin[0];
500
+ for (i = 0; i < nx*ny*nz; i++) {
501
+ if (h->bin[i] < min) min = h->bin[i];
502
+ }
503
+ return min;
504
+ }
505
+
506
+ void mygsl_histogram3d_min_bin(const mygsl_histogram3d * h,
507
+ size_t *imin_out, size_t *jmin_out, size_t *kmin_out)
508
+ {
509
+ const size_t nx = h->nx;
510
+ const size_t ny = h->ny;
511
+ const size_t nz = h->nz;
512
+ size_t imin = 0, jmin = 0, kmin = 0, i, j, k;
513
+ double min = h->bin[0], x;
514
+ for (i = 0; i < nx; i++) {
515
+ for (j = 0; j < ny; j++) {
516
+ for (k = 0; k < nz; k++) {
517
+ x = h->bin[i * ny*nz + j*nz + k];
518
+ if (x < min) {
519
+ min = x;
520
+ imin = i;
521
+ jmin = j;
522
+ kmin = k;
523
+ }
524
+ }
525
+ }
526
+ }
527
+ *imin_out = imin;
528
+ *jmin_out = jmin;
529
+ *kmin_out = kmin;
530
+ }
531
+
532
+ double mygsl_histogram3d_sum (const mygsl_histogram3d * h)
533
+ {
534
+ const size_t n = h->nx*h->ny*h->nz;
535
+ double sum = 0;
536
+ size_t i = 0;
537
+ while (i < n) sum += h->bin[i++];
538
+ return sum;
539
+ }
540
+
541
+ double mygsl_histogram3d_xmean (const mygsl_histogram3d * h)
542
+ {
543
+ const size_t nx = h->nx;
544
+ const size_t ny = h->ny;
545
+ const size_t nz = h->nz;
546
+ size_t i, j, k;
547
+ double wmean = 0;
548
+ double W = 0;
549
+ for (i = 0; i < nx; i++) {
550
+ double xi = (h->xrange[i + 1] + h->xrange[i]) / 2.0;
551
+ double wi = 0;
552
+ for (j = 0; j < ny; j++) {
553
+ for (k = 0; k < nz; k++) {
554
+ double wijk = h->bin[i * ny *nz + j * nz + k];
555
+ if (wijk > 0) wi += wijk;
556
+ }
557
+ }
558
+ if (wi > 0) {
559
+ W += wi;
560
+ wmean += (xi - wmean) * (wi / W);
561
+ }
562
+ }
563
+ return wmean;
564
+ }
565
+
566
+ double mygsl_histogram3d_ymean (const mygsl_histogram3d * h)
567
+ {
568
+ const size_t nx = h->nx;
569
+ const size_t ny = h->ny;
570
+ const size_t nz = h->nz;
571
+ size_t i, j, k;
572
+ double wmean = 0;
573
+ double W = 0;
574
+ for (j = 0; j < ny; j++) {
575
+ double yj = (h->yrange[j + 1] + h->yrange[j]) / 2.0;
576
+ double wj = 0;
577
+ for (i = 0; i < nx; i++) {
578
+ for (k = 0; k < nz; k++) {
579
+ double wijk = h->bin[i * ny *nz + j * nz + k];
580
+ if (wijk > 0) wj += wijk;
581
+ }
582
+ }
583
+ if (wj > 0) {
584
+ W += wj;
585
+ wmean += (yj - wmean) * (wj / W);
586
+ }
587
+ }
588
+ return wmean;
589
+ }
590
+
591
+ double mygsl_histogram3d_zmean (const mygsl_histogram3d * h)
592
+ {
593
+ const size_t nx = h->nx;
594
+ const size_t ny = h->ny;
595
+ const size_t nz = h->nz;
596
+ size_t i, j, k;
597
+ double wmean = 0;
598
+ double W = 0;
599
+ for (k = 0; k < nz; k++) {
600
+ double zk = (h->zrange[k + 1] + h->zrange[k]) / 2.0;
601
+ double wk = 0;
602
+ for (i = 0; i < nx; i++) {
603
+ for (j = 0; j < ny; j++) {
604
+ double wijk = h->bin[i * ny *nz + j * nz + k];
605
+ if (wijk > 0) wk += wijk;
606
+ }
607
+ }
608
+ if (wk > 0) {
609
+ W += wk;
610
+ wmean += (zk - wmean) * (wk / W);
611
+ }
612
+ }
613
+ return wmean;
614
+ }
615
+
616
+ double mygsl_histogram3d_xsigma(const mygsl_histogram3d * h)
617
+ {
618
+ const double xmean = mygsl_histogram3d_xmean(h);
619
+ const size_t nx = h->nx, ny = h->ny, nz = h->nz;
620
+ size_t i, j, k;
621
+ double wvariance = 0, W = 0;
622
+ for (i = 0; i < nx; i++) {
623
+ double xi = (h->xrange[i + 1] + h->xrange[i]) / 2 - xmean;
624
+ double wi = 0;
625
+ for (j = 0; j < ny; j++) {
626
+ for (k = 0; k < nz; k++) {
627
+ double wijk = h->bin[i * ny*nz + j*nz + k];
628
+ if (wijk > 0) wi += wijk;
629
+ }
630
+ }
631
+ if (wi > 0) {
632
+ W += wi;
633
+ wvariance += ((xi * xi) - wvariance) * (wi / W);
634
+ }
635
+ }
636
+ return sqrt(wvariance);
637
+ }
638
+
639
+ double mygsl_histogram3d_ysigma(const mygsl_histogram3d * h)
640
+ {
641
+ const double ymean = mygsl_histogram3d_ymean(h);
642
+ const size_t nx = h->nx, ny = h->ny, nz = h->nz;
643
+ size_t i, j, k;
644
+ double wvariance = 0, W = 0;
645
+ for (j = 0; j < ny; j++) {
646
+ double yj = (h->yrange[j + 1] + h->yrange[j]) / 2 - ymean;
647
+ double wj = 0;
648
+ for (i = 0; i < nx; i++) {
649
+ for (k = 0; k < nz; k++) {
650
+ double wjjk = h->bin[i * ny*nz + j*nz + k];
651
+ if (wjjk > 0) wj += wjjk;
652
+ }
653
+ }
654
+ if (wj > 0) {
655
+ W += wj;
656
+ wvariance += ((yj * yj) - wvariance) * (wj / W);
657
+ }
658
+ }
659
+ return sqrt(wvariance);
660
+ }
661
+
662
+ double mygsl_histogram3d_zsigma(const mygsl_histogram3d * h)
663
+ {
664
+ const double zmean = mygsl_histogram3d_zmean(h);
665
+ const size_t nx = h->nx, ny = h->ny, nz = h->nz;
666
+ size_t i, j, k;
667
+ double wvariance = 0, W = 0;
668
+ for (k = 0; k < nz; k++) {
669
+ double zk = (h->zrange[k + 1] + h->zrange[k]) / 2 - zmean;
670
+ double wk = 0;
671
+ for (i = 0; i < nx; i++) {
672
+ for (j = 0; j < ny; j++) {
673
+ double wijk = h->bin[i * ny*nz + j*nz + k];
674
+ if (wijk > 0) wk += wijk;
675
+ }
676
+ }
677
+ if (wk > 0) {
678
+ W += wk;
679
+ wvariance += ((zk * zk) - wvariance) * (wk / W);
680
+ }
681
+ }
682
+ return sqrt(wvariance);
683
+ }
684
+
685
+ void mygsl_histogram3d_reset(mygsl_histogram3d * h)
686
+ {
687
+ size_t i;
688
+ const size_t nx = h->nx;
689
+ const size_t ny = h->ny;
690
+ const size_t nz = h->nz;
691
+
692
+ for (i = 0; i < nx * ny * nz; i++)
693
+ {
694
+ h->bin[i] = 0;
695
+ }
696
+ }
697
+
698
+ int mygsl_histogram3d_equal_bins_p(const mygsl_histogram3d * h1,
699
+ const mygsl_histogram3d * h2)
700
+ {
701
+ size_t i;
702
+ if ((h1->nx != h2->nx) || (h1->ny != h2->ny) || (h1->nz != h2->nz)) return 0;
703
+ for (i = 0; i <= h1->nx; i++)
704
+ if (h1->xrange[i] != h2->xrange[i]) return 0;
705
+ for (i = 0; i <= h1->ny; i++)
706
+ if (h1->yrange[i] != h2->yrange[i]) return 0;
707
+ for (i = 0; i <= h1->nz; i++)
708
+ if (h1->zrange[i] != h2->zrange[i]) return 0;
709
+ return 1;
710
+ }
711
+
712
+ int mygsl_histogram3d_add(mygsl_histogram3d * h1, const mygsl_histogram3d * h2)
713
+ {
714
+ size_t i;
715
+ if (!mygsl_histogram3d_equal_bins_p(h1, h2))
716
+ GSL_ERROR ("histograms have different binning", GSL_EINVAL);
717
+ for (i = 0; i < (h1->nx) * (h1->ny) * (h1->nz); i++)
718
+ h1->bin[i] += h2->bin[i];
719
+ return GSL_SUCCESS;
720
+ }
721
+
722
+ int mygsl_histogram3d_sub(mygsl_histogram3d * h1, const mygsl_histogram3d * h2)
723
+ {
724
+ size_t i;
725
+ if (!mygsl_histogram3d_equal_bins_p(h1, h2))
726
+ GSL_ERROR ("histograms have different binning", GSL_EINVAL);
727
+ for (i = 0; i < (h1->nx) * (h1->ny) * (h1->nz); i++)
728
+ h1->bin[i] -= h2->bin[i];
729
+ return GSL_SUCCESS;
730
+ }
731
+
732
+ int mygsl_histogram3d_mul(mygsl_histogram3d * h1, const mygsl_histogram3d * h2)
733
+ {
734
+ size_t i;
735
+ if (!mygsl_histogram3d_equal_bins_p(h1, h2))
736
+ GSL_ERROR ("histograms have different binning", GSL_EINVAL);
737
+ for (i = 0; i < (h1->nx) * (h1->ny) * (h1->nz); i++)
738
+ h1->bin[i] *= h2->bin[i];
739
+ return GSL_SUCCESS;
740
+ }
741
+
742
+ int mygsl_histogram3d_div(mygsl_histogram3d * h1, const mygsl_histogram3d * h2)
743
+ {
744
+ size_t i;
745
+ if (!mygsl_histogram3d_equal_bins_p(h1, h2))
746
+ GSL_ERROR ("histograms have different binning", GSL_EINVAL);
747
+ for (i = 0; i < (h1->nx) * (h1->ny) * (h1->nz); i++)
748
+ h1->bin[i] /= h2->bin[i];
749
+ return GSL_SUCCESS;
750
+ }