rb-gsl 1.15.3.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (660) hide show
  1. checksums.yaml +7 -0
  2. data/AUTHORS +6 -0
  3. data/COPYING +339 -0
  4. data/ChangeLog +567 -0
  5. data/README +32 -0
  6. data/Rakefile +99 -0
  7. data/THANKS +17 -0
  8. data/VERSION +1 -0
  9. data/examples/alf/alf.gp +15 -0
  10. data/examples/alf/alf.rb +32 -0
  11. data/examples/blas/blas.rb +13 -0
  12. data/examples/blas/dnrm2.rb +16 -0
  13. data/examples/blas/level1.rb +81 -0
  14. data/examples/blas/level2.rb +11 -0
  15. data/examples/blas/level3.rb +12 -0
  16. data/examples/bspline.rb +57 -0
  17. data/examples/cdf.rb +16 -0
  18. data/examples/cheb.rb +21 -0
  19. data/examples/combination.rb +23 -0
  20. data/examples/complex/RC-lpf.rb +47 -0
  21. data/examples/complex/add.rb +36 -0
  22. data/examples/complex/coerce.rb +14 -0
  23. data/examples/complex/complex.rb +25 -0
  24. data/examples/complex/fpmi.rb +70 -0
  25. data/examples/complex/functions.rb +77 -0
  26. data/examples/complex/michelson.rb +36 -0
  27. data/examples/complex/mul.rb +28 -0
  28. data/examples/complex/oscillator.rb +17 -0
  29. data/examples/complex/set.rb +37 -0
  30. data/examples/const/physconst.rb +151 -0
  31. data/examples/const/travel.rb +45 -0
  32. data/examples/deriv/demo.rb +13 -0
  33. data/examples/deriv/deriv.rb +36 -0
  34. data/examples/deriv/diff.rb +35 -0
  35. data/examples/dht.rb +42 -0
  36. data/examples/dirac.rb +56 -0
  37. data/examples/eigen/eigen.rb +34 -0
  38. data/examples/eigen/herm.rb +22 -0
  39. data/examples/eigen/narray.rb +9 -0
  40. data/examples/eigen/nonsymm.rb +37 -0
  41. data/examples/eigen/nonsymmv.rb +43 -0
  42. data/examples/eigen/qhoscillator.gp +35 -0
  43. data/examples/eigen/qhoscillator.rb +90 -0
  44. data/examples/eigen/vander.rb +41 -0
  45. data/examples/fft/fft.rb +17 -0
  46. data/examples/fft/fft2.rb +17 -0
  47. data/examples/fft/forward.rb +25 -0
  48. data/examples/fft/forward2.rb +26 -0
  49. data/examples/fft/radix2.rb +18 -0
  50. data/examples/fft/real-halfcomplex.rb +33 -0
  51. data/examples/fft/real-halfcomplex2.rb +30 -0
  52. data/examples/fft/realradix2.rb +19 -0
  53. data/examples/fft/sunspot.dat +256 -0
  54. data/examples/fft/sunspot.rb +16 -0
  55. data/examples/fit/expdata.dat +20 -0
  56. data/examples/fit/expfit.rb +31 -0
  57. data/examples/fit/gaussfit.rb +29 -0
  58. data/examples/fit/gaussian_2peaks.rb +34 -0
  59. data/examples/fit/hillfit.rb +40 -0
  60. data/examples/fit/lognormal.rb +26 -0
  61. data/examples/fit/lorentzfit.rb +22 -0
  62. data/examples/fit/multifit.rb +72 -0
  63. data/examples/fit/ndlinear.rb +133 -0
  64. data/examples/fit/nonlinearfit.rb +89 -0
  65. data/examples/fit/plot.gp +36 -0
  66. data/examples/fit/polyfit.rb +9 -0
  67. data/examples/fit/powerfit.rb +21 -0
  68. data/examples/fit/sigmoidfit.rb +40 -0
  69. data/examples/fit/sinfit.rb +22 -0
  70. data/examples/fit/wlinear.rb +46 -0
  71. data/examples/fresnel.rb +11 -0
  72. data/examples/function/function.rb +36 -0
  73. data/examples/function/log.rb +7 -0
  74. data/examples/function/min.rb +33 -0
  75. data/examples/function/sin.rb +10 -0
  76. data/examples/function/synchrotron.rb +18 -0
  77. data/examples/gallery/butterfly.rb +7 -0
  78. data/examples/gallery/cayley.rb +12 -0
  79. data/examples/gallery/cornu.rb +23 -0
  80. data/examples/gallery/eight.rb +11 -0
  81. data/examples/gallery/koch.rb +40 -0
  82. data/examples/gallery/lemniscate.rb +11 -0
  83. data/examples/gallery/polar.rb +11 -0
  84. data/examples/gallery/rgplot/cossin.rb +35 -0
  85. data/examples/gallery/rgplot/rgplot.replaced +0 -0
  86. data/examples/gallery/rgplot/roesller.rb +55 -0
  87. data/examples/gallery/roesller.rb +39 -0
  88. data/examples/gallery/scarabaeus.rb +14 -0
  89. data/examples/histogram/cauchy.rb +27 -0
  90. data/examples/histogram/cauchy.sh +2 -0
  91. data/examples/histogram/exponential.rb +19 -0
  92. data/examples/histogram/gauss.rb +16 -0
  93. data/examples/histogram/gsl-histogram.rb +40 -0
  94. data/examples/histogram/histo2d.rb +31 -0
  95. data/examples/histogram/histo3d.rb +34 -0
  96. data/examples/histogram/histogram-pdf.rb +27 -0
  97. data/examples/histogram/histogram.rb +26 -0
  98. data/examples/histogram/integral.rb +28 -0
  99. data/examples/histogram/poisson.rb +27 -0
  100. data/examples/histogram/power.rb +25 -0
  101. data/examples/histogram/rebin.rb +17 -0
  102. data/examples/histogram/smp.dat +5 -0
  103. data/examples/histogram/xexp.rb +21 -0
  104. data/examples/integration/ahmed.rb +21 -0
  105. data/examples/integration/cosmology.rb +75 -0
  106. data/examples/integration/friedmann.gp +16 -0
  107. data/examples/integration/friedmann.rb +35 -0
  108. data/examples/integration/gamma-zeta.rb +35 -0
  109. data/examples/integration/integration.rb +22 -0
  110. data/examples/integration/qag.rb +8 -0
  111. data/examples/integration/qag2.rb +14 -0
  112. data/examples/integration/qag3.rb +8 -0
  113. data/examples/integration/qagi.rb +28 -0
  114. data/examples/integration/qagi2.rb +49 -0
  115. data/examples/integration/qagiu.rb +29 -0
  116. data/examples/integration/qagp.rb +20 -0
  117. data/examples/integration/qags.rb +14 -0
  118. data/examples/integration/qawc.rb +18 -0
  119. data/examples/integration/qawf.rb +41 -0
  120. data/examples/integration/qawo.rb +29 -0
  121. data/examples/integration/qaws.rb +30 -0
  122. data/examples/integration/qng.rb +17 -0
  123. data/examples/interp/demo.gp +20 -0
  124. data/examples/interp/demo.rb +45 -0
  125. data/examples/interp/interp.rb +37 -0
  126. data/examples/interp/points +10 -0
  127. data/examples/interp/spline.rb +20 -0
  128. data/examples/jacobi/deriv.rb +40 -0
  129. data/examples/jacobi/integrate.rb +34 -0
  130. data/examples/jacobi/interp.rb +43 -0
  131. data/examples/jacobi/jacobi.rb +11 -0
  132. data/examples/linalg/HH.rb +15 -0
  133. data/examples/linalg/HH_narray.rb +13 -0
  134. data/examples/linalg/LQ_solve.rb +73 -0
  135. data/examples/linalg/LU.rb +84 -0
  136. data/examples/linalg/LU2.rb +31 -0
  137. data/examples/linalg/LU_narray.rb +24 -0
  138. data/examples/linalg/PTLQ.rb +47 -0
  139. data/examples/linalg/QR.rb +18 -0
  140. data/examples/linalg/QRPT.rb +47 -0
  141. data/examples/linalg/QR_solve.rb +78 -0
  142. data/examples/linalg/QR_solve_narray.rb +13 -0
  143. data/examples/linalg/SV.rb +16 -0
  144. data/examples/linalg/SV_narray.rb +12 -0
  145. data/examples/linalg/SV_solve.rb +49 -0
  146. data/examples/linalg/chol.rb +29 -0
  147. data/examples/linalg/chol_narray.rb +15 -0
  148. data/examples/linalg/complex.rb +57 -0
  149. data/examples/linalg/invert_narray.rb +10 -0
  150. data/examples/math/const.rb +67 -0
  151. data/examples/math/elementary.rb +35 -0
  152. data/examples/math/functions.rb +41 -0
  153. data/examples/math/inf_nan.rb +34 -0
  154. data/examples/math/minmax.rb +22 -0
  155. data/examples/math/power.rb +18 -0
  156. data/examples/math/test.rb +31 -0
  157. data/examples/matrix/a.dat +0 -0
  158. data/examples/matrix/add.rb +45 -0
  159. data/examples/matrix/b.dat +4 -0
  160. data/examples/matrix/cat.rb +31 -0
  161. data/examples/matrix/colvectors.rb +24 -0
  162. data/examples/matrix/complex.rb +41 -0
  163. data/examples/matrix/det.rb +29 -0
  164. data/examples/matrix/diagonal.rb +23 -0
  165. data/examples/matrix/get_all.rb +159 -0
  166. data/examples/matrix/hilbert.rb +31 -0
  167. data/examples/matrix/iterator.rb +19 -0
  168. data/examples/matrix/matrix.rb +57 -0
  169. data/examples/matrix/minmax.rb +53 -0
  170. data/examples/matrix/mul.rb +39 -0
  171. data/examples/matrix/rand.rb +20 -0
  172. data/examples/matrix/read.rb +29 -0
  173. data/examples/matrix/rowcol.rb +47 -0
  174. data/examples/matrix/set.rb +41 -0
  175. data/examples/matrix/set_all.rb +100 -0
  176. data/examples/matrix/view.rb +32 -0
  177. data/examples/matrix/view_all.rb +148 -0
  178. data/examples/matrix/write.rb +23 -0
  179. data/examples/min.rb +29 -0
  180. data/examples/monte/miser.rb +47 -0
  181. data/examples/monte/monte.rb +47 -0
  182. data/examples/monte/plain.rb +47 -0
  183. data/examples/monte/vegas.rb +46 -0
  184. data/examples/multimin/bundle.rb +66 -0
  185. data/examples/multimin/cqp.rb +109 -0
  186. data/examples/multimin/fdfminimizer.rb +40 -0
  187. data/examples/multimin/fminimizer.rb +41 -0
  188. data/examples/multiroot/demo.rb +36 -0
  189. data/examples/multiroot/fdfsolver.rb +50 -0
  190. data/examples/multiroot/fsolver.rb +33 -0
  191. data/examples/multiroot/fsolver2.rb +32 -0
  192. data/examples/multiroot/fsolver3.rb +26 -0
  193. data/examples/narray/histogram.rb +14 -0
  194. data/examples/narray/mandel.rb +27 -0
  195. data/examples/narray/narray.rb +28 -0
  196. data/examples/narray/narray2.rb +44 -0
  197. data/examples/narray/sf.rb +26 -0
  198. data/examples/ntuple/create.rb +17 -0
  199. data/examples/ntuple/project.rb +31 -0
  200. data/examples/odeiv/binarysystem.gp +23 -0
  201. data/examples/odeiv/binarysystem.rb +104 -0
  202. data/examples/odeiv/demo.gp +24 -0
  203. data/examples/odeiv/demo.rb +69 -0
  204. data/examples/odeiv/demo2.gp +26 -0
  205. data/examples/odeiv/duffing.rb +45 -0
  206. data/examples/odeiv/frei1.rb +109 -0
  207. data/examples/odeiv/frei2.rb +76 -0
  208. data/examples/odeiv/legendre.rb +52 -0
  209. data/examples/odeiv/odeiv.rb +32 -0
  210. data/examples/odeiv/odeiv2.rb +45 -0
  211. data/examples/odeiv/oscillator.rb +42 -0
  212. data/examples/odeiv/sedov.rb +97 -0
  213. data/examples/odeiv/whitedwarf.gp +40 -0
  214. data/examples/odeiv/whitedwarf.rb +158 -0
  215. data/examples/ool/conmin.rb +100 -0
  216. data/examples/ool/gencan.rb +99 -0
  217. data/examples/ool/pgrad.rb +100 -0
  218. data/examples/ool/spg.rb +100 -0
  219. data/examples/pdf/bernoulli.rb +5 -0
  220. data/examples/pdf/beta.rb +7 -0
  221. data/examples/pdf/binomiral.rb +10 -0
  222. data/examples/pdf/cauchy.rb +6 -0
  223. data/examples/pdf/chisq.rb +8 -0
  224. data/examples/pdf/exponential.rb +7 -0
  225. data/examples/pdf/exppow.rb +6 -0
  226. data/examples/pdf/fdist.rb +7 -0
  227. data/examples/pdf/flat.rb +7 -0
  228. data/examples/pdf/gamma.rb +8 -0
  229. data/examples/pdf/gauss-tail.rb +5 -0
  230. data/examples/pdf/gauss.rb +6 -0
  231. data/examples/pdf/geometric.rb +5 -0
  232. data/examples/pdf/gumbel.rb +6 -0
  233. data/examples/pdf/hypergeometric.rb +11 -0
  234. data/examples/pdf/landau.rb +5 -0
  235. data/examples/pdf/laplace.rb +7 -0
  236. data/examples/pdf/logarithmic.rb +5 -0
  237. data/examples/pdf/logistic.rb +6 -0
  238. data/examples/pdf/lognormal.rb +6 -0
  239. data/examples/pdf/neg-binomiral.rb +10 -0
  240. data/examples/pdf/pareto.rb +7 -0
  241. data/examples/pdf/pascal.rb +10 -0
  242. data/examples/pdf/poisson.rb +5 -0
  243. data/examples/pdf/rayleigh-tail.rb +6 -0
  244. data/examples/pdf/rayleigh.rb +6 -0
  245. data/examples/pdf/tdist.rb +6 -0
  246. data/examples/pdf/weibull.rb +8 -0
  247. data/examples/permutation/ex1.rb +22 -0
  248. data/examples/permutation/permutation.rb +16 -0
  249. data/examples/poly/bell.rb +6 -0
  250. data/examples/poly/bessel.rb +6 -0
  251. data/examples/poly/cheb.rb +6 -0
  252. data/examples/poly/cheb_II.rb +6 -0
  253. data/examples/poly/cubic.rb +9 -0
  254. data/examples/poly/demo.rb +20 -0
  255. data/examples/poly/eval.rb +28 -0
  256. data/examples/poly/eval_derivs.rb +14 -0
  257. data/examples/poly/fit.rb +21 -0
  258. data/examples/poly/hermite.rb +6 -0
  259. data/examples/poly/poly.rb +13 -0
  260. data/examples/poly/quadratic.rb +25 -0
  261. data/examples/random/diffusion.rb +34 -0
  262. data/examples/random/gaussian.rb +9 -0
  263. data/examples/random/generator.rb +27 -0
  264. data/examples/random/hdsobol.rb +21 -0
  265. data/examples/random/poisson.rb +9 -0
  266. data/examples/random/qrng.rb +19 -0
  267. data/examples/random/randomwalk.rb +37 -0
  268. data/examples/random/randomwalk2d.rb +19 -0
  269. data/examples/random/rayleigh.rb +36 -0
  270. data/examples/random/rng.rb +33 -0
  271. data/examples/random/rngextra.rb +14 -0
  272. data/examples/roots/bisection.rb +25 -0
  273. data/examples/roots/brent.rb +43 -0
  274. data/examples/roots/demo.rb +30 -0
  275. data/examples/roots/newton.rb +46 -0
  276. data/examples/roots/recombination.gp +12 -0
  277. data/examples/roots/recombination.rb +61 -0
  278. data/examples/roots/steffenson.rb +48 -0
  279. data/examples/sf/ShiChi.rb +6 -0
  280. data/examples/sf/SiCi.rb +6 -0
  281. data/examples/sf/airy_Ai.rb +8 -0
  282. data/examples/sf/airy_Bi.rb +8 -0
  283. data/examples/sf/bessel_IK.rb +12 -0
  284. data/examples/sf/bessel_JY.rb +13 -0
  285. data/examples/sf/beta_inc.rb +9 -0
  286. data/examples/sf/clausen.rb +6 -0
  287. data/examples/sf/dawson.rb +5 -0
  288. data/examples/sf/debye.rb +9 -0
  289. data/examples/sf/dilog.rb +6 -0
  290. data/examples/sf/ellint.rb +6 -0
  291. data/examples/sf/expint.rb +8 -0
  292. data/examples/sf/fermi.rb +10 -0
  293. data/examples/sf/gamma_inc_P.rb +9 -0
  294. data/examples/sf/gegenbauer.rb +8 -0
  295. data/examples/sf/hyperg.rb +7 -0
  296. data/examples/sf/laguerre.rb +19 -0
  297. data/examples/sf/lambertW.rb +5 -0
  298. data/examples/sf/legendre_P.rb +10 -0
  299. data/examples/sf/lngamma.rb +5 -0
  300. data/examples/sf/psi.rb +54 -0
  301. data/examples/sf/sphbessel.gp +27 -0
  302. data/examples/sf/sphbessel.rb +30 -0
  303. data/examples/sf/synchrotron.rb +5 -0
  304. data/examples/sf/transport.rb +10 -0
  305. data/examples/sf/zetam1.rb +5 -0
  306. data/examples/siman.rb +44 -0
  307. data/examples/sort/heapsort.rb +23 -0
  308. data/examples/sort/heapsort_vector_complex.rb +21 -0
  309. data/examples/sort/sort.rb +23 -0
  310. data/examples/sort/sort2.rb +16 -0
  311. data/examples/stats/mean.rb +17 -0
  312. data/examples/stats/statistics.rb +18 -0
  313. data/examples/stats/test.rb +9 -0
  314. data/examples/sum.rb +34 -0
  315. data/examples/tamu_anova.rb +18 -0
  316. data/examples/vector/a.dat +0 -0
  317. data/examples/vector/add.rb +56 -0
  318. data/examples/vector/b.dat +4 -0
  319. data/examples/vector/c.dat +3 -0
  320. data/examples/vector/collect.rb +26 -0
  321. data/examples/vector/compare.rb +28 -0
  322. data/examples/vector/complex.rb +51 -0
  323. data/examples/vector/complex_get_all.rb +85 -0
  324. data/examples/vector/complex_set_all.rb +131 -0
  325. data/examples/vector/complex_view_all.rb +77 -0
  326. data/examples/vector/connect.rb +22 -0
  327. data/examples/vector/decimate.rb +38 -0
  328. data/examples/vector/diff.rb +31 -0
  329. data/examples/vector/filescan.rb +17 -0
  330. data/examples/vector/floor.rb +23 -0
  331. data/examples/vector/get_all.rb +82 -0
  332. data/examples/vector/gnuplot.rb +38 -0
  333. data/examples/vector/graph.rb +28 -0
  334. data/examples/vector/histogram.rb +22 -0
  335. data/examples/vector/linspace.rb +24 -0
  336. data/examples/vector/log.rb +17 -0
  337. data/examples/vector/logic.rb +33 -0
  338. data/examples/vector/logspace.rb +25 -0
  339. data/examples/vector/minmax.rb +47 -0
  340. data/examples/vector/mul.rb +49 -0
  341. data/examples/vector/narray.rb +46 -0
  342. data/examples/vector/read.rb +29 -0
  343. data/examples/vector/set.rb +35 -0
  344. data/examples/vector/set_all.rb +121 -0
  345. data/examples/vector/smpv.dat +15 -0
  346. data/examples/vector/test.rb +43 -0
  347. data/examples/vector/test_gslblock.rb +58 -0
  348. data/examples/vector/vector.rb +110 -0
  349. data/examples/vector/view.rb +35 -0
  350. data/examples/vector/view_all.rb +73 -0
  351. data/examples/vector/where.rb +29 -0
  352. data/examples/vector/write.rb +24 -0
  353. data/examples/vector/zip.rb +34 -0
  354. data/examples/wavelet/ecg.dat +256 -0
  355. data/examples/wavelet/wavelet1.rb +50 -0
  356. data/ext/alf.c +206 -0
  357. data/ext/array.c +642 -0
  358. data/ext/array_complex.c +248 -0
  359. data/ext/blas.c +29 -0
  360. data/ext/blas1.c +734 -0
  361. data/ext/blas2.c +1093 -0
  362. data/ext/blas3.c +881 -0
  363. data/ext/block.c +44 -0
  364. data/ext/block_source.c +887 -0
  365. data/ext/bspline.c +130 -0
  366. data/ext/bundle.c +3 -0
  367. data/ext/cdf.c +754 -0
  368. data/ext/cheb.c +542 -0
  369. data/ext/combination.c +283 -0
  370. data/ext/common.c +325 -0
  371. data/ext/complex.c +1004 -0
  372. data/ext/const.c +673 -0
  373. data/ext/const_additional.c +120 -0
  374. data/ext/cqp.c +283 -0
  375. data/ext/deriv.c +195 -0
  376. data/ext/dht.c +361 -0
  377. data/ext/diff.c +166 -0
  378. data/ext/dirac.c +389 -0
  379. data/ext/eigen.c +2373 -0
  380. data/ext/error.c +193 -0
  381. data/ext/extconf.rb +303 -0
  382. data/ext/fcmp.c +66 -0
  383. data/ext/fft.c +1102 -0
  384. data/ext/fit.c +205 -0
  385. data/ext/fresnel.c +312 -0
  386. data/ext/function.c +524 -0
  387. data/ext/geometry.c +139 -0
  388. data/ext/graph.c +1640 -0
  389. data/ext/gsl.c +280 -0
  390. data/ext/gsl_narray.c +804 -0
  391. data/ext/histogram.c +1991 -0
  392. data/ext/histogram2d.c +1068 -0
  393. data/ext/histogram3d.c +884 -0
  394. data/ext/histogram3d_source.c +750 -0
  395. data/ext/histogram_find.c +101 -0
  396. data/ext/histogram_oper.c +159 -0
  397. data/ext/ieee.c +100 -0
  398. data/ext/integration.c +1179 -0
  399. data/ext/interp.c +512 -0
  400. data/ext/jacobi.c +739 -0
  401. data/ext/linalg.c +4042 -0
  402. data/ext/linalg_complex.c +739 -0
  403. data/ext/math.c +725 -0
  404. data/ext/matrix.c +39 -0
  405. data/ext/matrix_complex.c +1737 -0
  406. data/ext/matrix_double.c +560 -0
  407. data/ext/matrix_int.c +256 -0
  408. data/ext/matrix_source.c +2734 -0
  409. data/ext/min.c +250 -0
  410. data/ext/monte.c +992 -0
  411. data/ext/multifit.c +1871 -0
  412. data/ext/multimin.c +806 -0
  413. data/ext/multimin_fsdf.c +156 -0
  414. data/ext/multiroots.c +955 -0
  415. data/ext/multiset.c +214 -0
  416. data/ext/ndlinear.c +321 -0
  417. data/ext/nmf.c +171 -0
  418. data/ext/nmf_wrap.c +75 -0
  419. data/ext/ntuple.c +469 -0
  420. data/ext/odeiv.c +962 -0
  421. data/ext/ool.c +879 -0
  422. data/ext/oper_complex_source.c +253 -0
  423. data/ext/permutation.c +596 -0
  424. data/ext/poly.c +42 -0
  425. data/ext/poly2.c +265 -0
  426. data/ext/poly_source.c +1915 -0
  427. data/ext/qrng.c +171 -0
  428. data/ext/randist.c +1873 -0
  429. data/ext/rational.c +480 -0
  430. data/ext/rng.c +612 -0
  431. data/ext/root.c +408 -0
  432. data/ext/sf.c +1494 -0
  433. data/ext/sf_airy.c +200 -0
  434. data/ext/sf_bessel.c +871 -0
  435. data/ext/sf_clausen.c +28 -0
  436. data/ext/sf_coulomb.c +206 -0
  437. data/ext/sf_coupling.c +121 -0
  438. data/ext/sf_dawson.c +29 -0
  439. data/ext/sf_debye.c +157 -0
  440. data/ext/sf_dilog.c +43 -0
  441. data/ext/sf_elementary.c +46 -0
  442. data/ext/sf_ellint.c +206 -0
  443. data/ext/sf_elljac.c +30 -0
  444. data/ext/sf_erfc.c +93 -0
  445. data/ext/sf_exp.c +169 -0
  446. data/ext/sf_expint.c +211 -0
  447. data/ext/sf_fermi_dirac.c +148 -0
  448. data/ext/sf_gamma.c +347 -0
  449. data/ext/sf_gegenbauer.c +97 -0
  450. data/ext/sf_hyperg.c +203 -0
  451. data/ext/sf_laguerre.c +113 -0
  452. data/ext/sf_lambert.c +47 -0
  453. data/ext/sf_legendre.c +368 -0
  454. data/ext/sf_log.c +105 -0
  455. data/ext/sf_mathieu.c +238 -0
  456. data/ext/sf_power.c +47 -0
  457. data/ext/sf_psi.c +98 -0
  458. data/ext/sf_synchrotron.c +48 -0
  459. data/ext/sf_transport.c +76 -0
  460. data/ext/sf_trigonometric.c +210 -0
  461. data/ext/sf_zeta.c +119 -0
  462. data/ext/signal.c +308 -0
  463. data/ext/siman.c +717 -0
  464. data/ext/sort.c +208 -0
  465. data/ext/spline.c +395 -0
  466. data/ext/stats.c +800 -0
  467. data/ext/sum.c +168 -0
  468. data/ext/tamu_anova.c +56 -0
  469. data/ext/tensor.c +38 -0
  470. data/ext/tensor_source.c +1123 -0
  471. data/ext/vector.c +38 -0
  472. data/ext/vector_complex.c +2246 -0
  473. data/ext/vector_double.c +1438 -0
  474. data/ext/vector_int.c +204 -0
  475. data/ext/vector_source.c +3336 -0
  476. data/ext/wavelet.c +945 -0
  477. data/include/rb_gsl.h +158 -0
  478. data/include/rb_gsl_array.h +238 -0
  479. data/include/rb_gsl_cheb.h +21 -0
  480. data/include/rb_gsl_common.h +348 -0
  481. data/include/rb_gsl_complex.h +25 -0
  482. data/include/rb_gsl_config.h +62 -0
  483. data/include/rb_gsl_const.h +29 -0
  484. data/include/rb_gsl_dirac.h +13 -0
  485. data/include/rb_gsl_eigen.h +17 -0
  486. data/include/rb_gsl_fft.h +62 -0
  487. data/include/rb_gsl_fit.h +25 -0
  488. data/include/rb_gsl_function.h +27 -0
  489. data/include/rb_gsl_graph.h +70 -0
  490. data/include/rb_gsl_histogram.h +63 -0
  491. data/include/rb_gsl_histogram3d.h +97 -0
  492. data/include/rb_gsl_integration.h +17 -0
  493. data/include/rb_gsl_interp.h +46 -0
  494. data/include/rb_gsl_linalg.h +25 -0
  495. data/include/rb_gsl_math.h +26 -0
  496. data/include/rb_gsl_odeiv.h +21 -0
  497. data/include/rb_gsl_poly.h +71 -0
  498. data/include/rb_gsl_rational.h +37 -0
  499. data/include/rb_gsl_rng.h +21 -0
  500. data/include/rb_gsl_root.h +22 -0
  501. data/include/rb_gsl_sf.h +119 -0
  502. data/include/rb_gsl_statistics.h +17 -0
  503. data/include/rb_gsl_tensor.h +45 -0
  504. data/include/rb_gsl_with_narray.h +29 -0
  505. data/include/templates_off.h +87 -0
  506. data/include/templates_on.h +241 -0
  507. data/lib/gsl.rb +3 -0
  508. data/lib/gsl/gnuplot.rb +41 -0
  509. data/lib/gsl/oper.rb +68 -0
  510. data/lib/ool.rb +22 -0
  511. data/lib/ool/conmin.rb +30 -0
  512. data/lib/rbgsl.rb +3 -0
  513. data/rdoc/alf.rdoc +77 -0
  514. data/rdoc/blas.rdoc +269 -0
  515. data/rdoc/bspline.rdoc +42 -0
  516. data/rdoc/changes.rdoc +164 -0
  517. data/rdoc/cheb.rdoc +99 -0
  518. data/rdoc/cholesky_complex.rdoc +46 -0
  519. data/rdoc/combi.rdoc +125 -0
  520. data/rdoc/complex.rdoc +210 -0
  521. data/rdoc/const.rdoc +546 -0
  522. data/rdoc/dht.rdoc +122 -0
  523. data/rdoc/diff.rdoc +133 -0
  524. data/rdoc/ehandling.rdoc +50 -0
  525. data/rdoc/eigen.rdoc +401 -0
  526. data/rdoc/fft.rdoc +535 -0
  527. data/rdoc/fit.rdoc +284 -0
  528. data/rdoc/function.rdoc +94 -0
  529. data/rdoc/graph.rdoc +137 -0
  530. data/rdoc/hist.rdoc +409 -0
  531. data/rdoc/hist2d.rdoc +279 -0
  532. data/rdoc/hist3d.rdoc +112 -0
  533. data/rdoc/index.rdoc +62 -0
  534. data/rdoc/integration.rdoc +398 -0
  535. data/rdoc/interp.rdoc +231 -0
  536. data/rdoc/intro.rdoc +27 -0
  537. data/rdoc/linalg.rdoc +681 -0
  538. data/rdoc/linalg_complex.rdoc +88 -0
  539. data/rdoc/math.rdoc +276 -0
  540. data/rdoc/matrix.rdoc +1093 -0
  541. data/rdoc/min.rdoc +189 -0
  542. data/rdoc/monte.rdoc +234 -0
  543. data/rdoc/multimin.rdoc +312 -0
  544. data/rdoc/multiroot.rdoc +293 -0
  545. data/rdoc/narray.rdoc +177 -0
  546. data/rdoc/ndlinear.rdoc +250 -0
  547. data/rdoc/nonlinearfit.rdoc +348 -0
  548. data/rdoc/ntuple.rdoc +88 -0
  549. data/rdoc/odeiv.rdoc +378 -0
  550. data/rdoc/perm.rdoc +221 -0
  551. data/rdoc/poly.rdoc +335 -0
  552. data/rdoc/qrng.rdoc +90 -0
  553. data/rdoc/randist.rdoc +233 -0
  554. data/rdoc/ref.rdoc +93 -0
  555. data/rdoc/rng.rdoc +203 -0
  556. data/rdoc/rngextra.rdoc +11 -0
  557. data/rdoc/roots.rdoc +305 -0
  558. data/rdoc/screenshot.rdoc +40 -0
  559. data/rdoc/sf.rdoc +1622 -0
  560. data/rdoc/siman.rdoc +89 -0
  561. data/rdoc/sort.rdoc +94 -0
  562. data/rdoc/start.rdoc +16 -0
  563. data/rdoc/stats.rdoc +219 -0
  564. data/rdoc/sum.rdoc +65 -0
  565. data/rdoc/tensor.rdoc +251 -0
  566. data/rdoc/tut.rdoc +5 -0
  567. data/rdoc/use.rdoc +177 -0
  568. data/rdoc/vector.rdoc +1243 -0
  569. data/rdoc/vector_complex.rdoc +347 -0
  570. data/rdoc/wavelet.rdoc +218 -0
  571. data/setup.rb +1585 -0
  572. data/tests/blas/amax.rb +14 -0
  573. data/tests/blas/asum.rb +16 -0
  574. data/tests/blas/axpy.rb +25 -0
  575. data/tests/blas/copy.rb +23 -0
  576. data/tests/blas/dot.rb +23 -0
  577. data/tests/bspline.rb +53 -0
  578. data/tests/cdf.rb +1388 -0
  579. data/tests/cheb.rb +112 -0
  580. data/tests/combination.rb +123 -0
  581. data/tests/complex.rb +17 -0
  582. data/tests/const.rb +24 -0
  583. data/tests/deriv.rb +85 -0
  584. data/tests/dht/dht1.rb +17 -0
  585. data/tests/dht/dht2.rb +23 -0
  586. data/tests/dht/dht3.rb +23 -0
  587. data/tests/dht/dht4.rb +23 -0
  588. data/tests/diff.rb +78 -0
  589. data/tests/eigen/eigen.rb +220 -0
  590. data/tests/eigen/gen.rb +105 -0
  591. data/tests/eigen/genherm.rb +66 -0
  592. data/tests/eigen/gensymm.rb +68 -0
  593. data/tests/eigen/nonsymm.rb +53 -0
  594. data/tests/eigen/nonsymmv.rb +53 -0
  595. data/tests/eigen/symm-herm.rb +74 -0
  596. data/tests/err.rb +58 -0
  597. data/tests/fit.rb +124 -0
  598. data/tests/gsl_test.rb +118 -0
  599. data/tests/gsl_test2.rb +110 -0
  600. data/tests/histo.rb +12 -0
  601. data/tests/integration/integration1.rb +72 -0
  602. data/tests/integration/integration2.rb +71 -0
  603. data/tests/integration/integration3.rb +71 -0
  604. data/tests/integration/integration4.rb +71 -0
  605. data/tests/interp.rb +45 -0
  606. data/tests/linalg/HH.rb +64 -0
  607. data/tests/linalg/LU.rb +47 -0
  608. data/tests/linalg/QR.rb +77 -0
  609. data/tests/linalg/SV.rb +24 -0
  610. data/tests/linalg/TDN.rb +116 -0
  611. data/tests/linalg/TDS.rb +122 -0
  612. data/tests/linalg/bidiag.rb +73 -0
  613. data/tests/linalg/cholesky.rb +20 -0
  614. data/tests/linalg/linalg.rb +158 -0
  615. data/tests/matrix/matrix_complex_test.rb +36 -0
  616. data/tests/matrix/matrix_nmf_test.rb +39 -0
  617. data/tests/matrix/matrix_test.rb +48 -0
  618. data/tests/min.rb +99 -0
  619. data/tests/monte/miser.rb +31 -0
  620. data/tests/monte/vegas.rb +45 -0
  621. data/tests/multifit/test_2dgauss.rb +112 -0
  622. data/tests/multifit/test_brown.rb +90 -0
  623. data/tests/multifit/test_enso.rb +246 -0
  624. data/tests/multifit/test_filip.rb +155 -0
  625. data/tests/multifit/test_gauss.rb +97 -0
  626. data/tests/multifit/test_longley.rb +110 -0
  627. data/tests/multifit/test_multifit.rb +52 -0
  628. data/tests/multimin.rb +139 -0
  629. data/tests/multiroot.rb +131 -0
  630. data/tests/multiset.rb +52 -0
  631. data/tests/narray/blas_dnrm2.rb +20 -0
  632. data/tests/odeiv.rb +353 -0
  633. data/tests/poly/poly.rb +290 -0
  634. data/tests/poly/special.rb +65 -0
  635. data/tests/qrng.rb +131 -0
  636. data/tests/quartic.rb +29 -0
  637. data/tests/randist.rb +134 -0
  638. data/tests/rng.rb +305 -0
  639. data/tests/roots.rb +76 -0
  640. data/tests/run-test.sh +17 -0
  641. data/tests/sf/gsl_test_sf.rb +249 -0
  642. data/tests/sf/test_airy.rb +83 -0
  643. data/tests/sf/test_bessel.rb +306 -0
  644. data/tests/sf/test_coulomb.rb +17 -0
  645. data/tests/sf/test_dilog.rb +25 -0
  646. data/tests/sf/test_gamma.rb +209 -0
  647. data/tests/sf/test_hyperg.rb +356 -0
  648. data/tests/sf/test_legendre.rb +227 -0
  649. data/tests/sf/test_mathieu.rb +59 -0
  650. data/tests/sf/test_mode.rb +19 -0
  651. data/tests/sf/test_sf.rb +839 -0
  652. data/tests/stats.rb +174 -0
  653. data/tests/stats_mt.rb +16 -0
  654. data/tests/sum.rb +98 -0
  655. data/tests/sys.rb +323 -0
  656. data/tests/tensor.rb +419 -0
  657. data/tests/vector/vector_complex_test.rb +101 -0
  658. data/tests/vector/vector_test.rb +141 -0
  659. data/tests/wavelet.rb +142 -0
  660. metadata +789 -0
@@ -0,0 +1,66 @@
1
+ /* sys/gsl_compare.c
2
+ *
3
+ * Copyright (C) 2002 Gert Van den Eynde
4
+ *
5
+ * This program is free software; you can redistribute it and/or modify
6
+ * it under the terms of the GNU General Public License as published by
7
+ * the Free Software Foundation; either version 2 of the License, or (at
8
+ * your option) any later version.
9
+ *
10
+ * This program is distributed in the hope that it will be useful, but
11
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
12
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13
+ * General Public License for more details.
14
+ *
15
+ * You should have received a copy of the GNU General Public License
16
+ * along with this program; if not, write to the Free Software
17
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18
+ *
19
+ * Based on fcmp 1.2.2 Copyright (c) 1998-2000 Theodore C. Belding
20
+ * University of Michigan Center for the Study of Complex Systems
21
+ * Ted.Belding@umich.edu
22
+ *
23
+ */
24
+
25
+ #include "rb_gsl_config.h"
26
+
27
+ #ifndef GSL_1_3_LATER
28
+
29
+ #include <config.h>
30
+ #include <gsl/gsl_sys.h>
31
+ #include <math.h>
32
+
33
+ int gsl_fcmp(const double x1, const double x2, const double epsilon)
34
+ {
35
+ int exponent;
36
+ double delta, difference;
37
+
38
+ /* Find exponent of largest absolute value */
39
+
40
+ {
41
+ double max = (fabs (x1) > fabs (x2)) ? x1 : x2;
42
+
43
+ frexp (max, &exponent);
44
+ }
45
+
46
+ /* Form a neighborhood of size 2 * delta */
47
+
48
+ delta = ldexp (epsilon, exponent);
49
+
50
+ difference = x1 - x2;
51
+
52
+ if (difference > delta) /* x1 > x2 */
53
+ {
54
+ return 1;
55
+ }
56
+ else if (difference < -delta) /* x1 < x2 */
57
+ {
58
+ return -1;
59
+ }
60
+ else /* -delta <= difference <= delta */
61
+ {
62
+ return 0; /* x1 ~=~ x2 */
63
+ }
64
+ }
65
+
66
+ #endif
@@ -0,0 +1,1102 @@
1
+ /*
2
+ fft.c
3
+ Ruby/GSL: Ruby extension library for GSL (GNU Scientific Library)
4
+ (C) Copyright 2004 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
+ #include "rb_gsl_config.h"
13
+ #include "rb_gsl_fft.h"
14
+
15
+ VALUE mgsl_fft;
16
+ VALUE cgsl_fft_wavetable;
17
+ VALUE cgsl_fft_complex_wavetable, cgsl_fft_complex_workspace;
18
+ VALUE cgsl_fft_real_wavetable, cgsl_fft_halfcomplex_wavetable;
19
+ VALUE cgsl_fft_real_workspace;
20
+
21
+ extern VALUE cgsl_vector_int;
22
+ extern VALUE cgsl_vector_complex;
23
+
24
+ static void GSL_FFT_Workspace_free(GSL_FFT_Workspace *space);
25
+
26
+ static VALUE rb_gsl_fft_complex_wavetable_new(VALUE klass, VALUE n)
27
+ {
28
+ CHECK_FIXNUM(n);
29
+ return Data_Wrap_Struct(cgsl_fft_complex_wavetable, 0,
30
+ gsl_fft_complex_wavetable_free,
31
+ gsl_fft_complex_wavetable_alloc(FIX2INT(n)));
32
+ }
33
+
34
+ static VALUE rb_gsl_fft_real_wavetable_new(VALUE klass, VALUE n)
35
+ {
36
+ CHECK_FIXNUM(n);
37
+ return Data_Wrap_Struct(klass, 0, gsl_fft_real_wavetable_free,
38
+ gsl_fft_real_wavetable_alloc(FIX2INT(n)));
39
+ }
40
+
41
+ static VALUE rb_gsl_fft_halfcomplex_wavetable_new(VALUE klass, VALUE n)
42
+ {
43
+ CHECK_FIXNUM(n);
44
+ return Data_Wrap_Struct(klass, 0, gsl_fft_halfcomplex_wavetable_free,
45
+ gsl_fft_halfcomplex_wavetable_alloc(FIX2INT(n)));
46
+
47
+ }
48
+
49
+ static void GSL_FFT_Wavetable_free(GSL_FFT_Wavetable *table)
50
+ {
51
+ gsl_fft_complex_wavetable_free((gsl_fft_complex_wavetable *) table);
52
+ }
53
+
54
+ static VALUE rb_gsl_fft_complex_workspace_new(VALUE klass, VALUE n)
55
+ {
56
+ CHECK_FIXNUM(n);
57
+ return Data_Wrap_Struct(klass, 0, gsl_fft_complex_workspace_free,
58
+ gsl_fft_complex_workspace_alloc(FIX2INT(n)));
59
+ }
60
+
61
+ static VALUE rb_gsl_fft_real_workspace_new(VALUE klass, VALUE n)
62
+ {
63
+ CHECK_FIXNUM(n);
64
+ return Data_Wrap_Struct(klass, 0, gsl_fft_real_workspace_free,
65
+ gsl_fft_real_workspace_alloc(FIX2INT(n)));
66
+ }
67
+
68
+ static void GSL_FFT_Workspace_free(GSL_FFT_Workspace *space)
69
+ {
70
+ gsl_fft_complex_workspace_free((gsl_fft_complex_workspace *) space);
71
+ }
72
+
73
+ // The FFT methods used to allow passing stride and n values as optional
74
+ // parameters to control which elements get transformed. This created problems
75
+ // for Views which can have their own stride, so support for stride and n
76
+ // parameters to the transform methods is being dropped. This method used to
77
+ // be called to determine the stride and n values to use based on the
78
+ // parameters and/or the vector itself (depending on how many parameters were
79
+ // passed). Now this function is somewhat unneceesary, but to simplify the code
80
+ // refactoring, it has been left in place for the time being. Eventually it
81
+ // can be refactored away completely.
82
+ static VALUE get_complex_stride_n(VALUE obj,
83
+ gsl_vector_complex **vin,
84
+ gsl_complex_packed_array *data, size_t *stride, size_t *n)
85
+ {
86
+ gsl_vector_complex *v = NULL;
87
+
88
+ // obj must be a GSL::Vector::Complex
89
+ CHECK_VECTOR_COMPLEX(obj);
90
+ Data_Get_Struct(obj, gsl_vector_complex, v);
91
+
92
+ if(vin) *vin = v;
93
+ *data = (gsl_complex_packed_array) v->data;
94
+ *stride = v->stride;
95
+ *n = v->size;
96
+ return obj;
97
+ }
98
+
99
+ static VALUE rb_fft_complex_radix2(VALUE obj,
100
+ int (*trans)(gsl_complex_packed_array,
101
+ size_t, size_t), int flag)
102
+ {
103
+ size_t stride, n;
104
+ gsl_complex_packed_array data;
105
+ gsl_vector_complex *vin, *vout;
106
+ VALUE ary;
107
+ ary = get_complex_stride_n(obj, &vin, &data, &stride, &n);
108
+ if (flag == RB_GSL_FFT_COPY) {
109
+ vout = gsl_vector_complex_alloc(n);
110
+ gsl_vector_complex_memcpy(vout, vin);
111
+ (*trans)(vout->data, vout->stride /*1*/, vout->size /*n*/);
112
+ return Data_Wrap_Struct(cgsl_vector_complex, 0, gsl_vector_complex_free, vout);
113
+ } else { /* in-place */
114
+ (*trans)(data, stride, n);
115
+ return ary;
116
+ }
117
+ }
118
+
119
+ static VALUE rb_gsl_fft_complex_radix2_forward(VALUE obj)
120
+ {
121
+ return rb_fft_complex_radix2(obj, gsl_fft_complex_radix2_forward,
122
+ RB_GSL_FFT_COPY);
123
+ }
124
+
125
+ static VALUE rb_gsl_fft_complex_radix2_forward2(VALUE obj)
126
+ {
127
+ return rb_fft_complex_radix2(obj, gsl_fft_complex_radix2_forward,
128
+ RB_GSL_FFT_INPLACE);
129
+ }
130
+
131
+ static VALUE rb_gsl_fft_complex_radix2_transform(VALUE obj, VALUE val_sign)
132
+ {
133
+ size_t stride, n;
134
+ gsl_complex_packed_array data;
135
+ gsl_fft_direction sign;
136
+ gsl_vector_complex *vin, *vout;
137
+ sign = NUM2INT(val_sign);
138
+ get_complex_stride_n(obj, &vin, &data, &stride, &n);
139
+ vout = gsl_vector_complex_alloc(n);
140
+ gsl_vector_complex_memcpy(vout, vin);
141
+ gsl_fft_complex_radix2_transform(vout->data, vout->stride /*1*/, vout->size /*n*/, sign);
142
+ return Data_Wrap_Struct(cgsl_vector_complex, 0, gsl_vector_complex_free, vout);
143
+ }
144
+
145
+ static VALUE rb_gsl_fft_complex_radix2_transform2(VALUE obj, VALUE val_sign)
146
+ {
147
+ size_t stride, n;
148
+ gsl_complex_packed_array data;
149
+ gsl_fft_direction sign;
150
+ VALUE ary;
151
+ sign = NUM2INT(val_sign);
152
+ ary = get_complex_stride_n(obj, NULL, &data, &stride, &n);
153
+ gsl_fft_complex_radix2_transform(data, stride, n, sign);
154
+ return ary;
155
+ }
156
+
157
+ static VALUE rb_gsl_fft_complex_radix2_backward(VALUE obj)
158
+ {
159
+ return rb_fft_complex_radix2(obj, gsl_fft_complex_radix2_backward,
160
+ RB_GSL_FFT_COPY);
161
+ }
162
+
163
+ static VALUE rb_gsl_fft_complex_radix2_inverse(VALUE obj)
164
+ {
165
+ return rb_fft_complex_radix2(obj, gsl_fft_complex_radix2_inverse,
166
+ RB_GSL_FFT_COPY);
167
+ }
168
+
169
+ static VALUE rb_gsl_fft_complex_radix2_dif_forward(VALUE obj)
170
+ {
171
+ return rb_fft_complex_radix2(obj,
172
+ gsl_fft_complex_radix2_dif_forward,
173
+ RB_GSL_FFT_COPY);
174
+ }
175
+
176
+ static VALUE rb_gsl_fft_complex_radix2_backward2(VALUE obj)
177
+ {
178
+ return rb_fft_complex_radix2(obj, gsl_fft_complex_radix2_backward,
179
+ RB_GSL_FFT_INPLACE);
180
+ }
181
+
182
+ static VALUE rb_gsl_fft_complex_radix2_inverse2(VALUE obj)
183
+ {
184
+ return rb_fft_complex_radix2(obj, gsl_fft_complex_radix2_inverse,
185
+ RB_GSL_FFT_INPLACE);
186
+ }
187
+
188
+
189
+ static VALUE rb_gsl_fft_complex_radix2_dif_forward2(VALUE obj)
190
+ {
191
+ return rb_fft_complex_radix2(obj,
192
+ gsl_fft_complex_radix2_dif_forward,
193
+ RB_GSL_FFT_INPLACE);
194
+ }
195
+
196
+ static VALUE rb_gsl_fft_complex_radix2_dif_transform(VALUE obj, VALUE val_sign)
197
+ {
198
+ size_t stride, n;
199
+ gsl_complex_packed_array data;
200
+ gsl_vector_complex *vin, *vout;
201
+ gsl_fft_direction sign;
202
+ sign = NUM2INT(val_sign);
203
+ get_complex_stride_n(obj, &vin, &data, &stride, &n);
204
+ vout = gsl_vector_complex_alloc(n);
205
+ gsl_vector_complex_memcpy(vout, vin);
206
+ gsl_fft_complex_radix2_dif_transform(vout->data, vout->stride /*1*/, vout->size /*n*/, sign);
207
+ return Data_Wrap_Struct(cgsl_vector_complex, 0, gsl_vector_complex_free, vout);
208
+ }
209
+
210
+ /* in-place */
211
+ static VALUE rb_gsl_fft_complex_radix2_dif_transform2(VALUE obj, VALUE val_sign)
212
+ {
213
+ size_t stride, n;
214
+ gsl_complex_packed_array data;
215
+ gsl_fft_direction sign;
216
+ VALUE ary;
217
+ sign = NUM2INT(val_sign);
218
+ ary = get_complex_stride_n(obj, NULL, &data, &stride, &n);
219
+ gsl_fft_complex_radix2_dif_transform(data, stride, n, sign);
220
+ return ary;
221
+ }
222
+
223
+ static VALUE rb_gsl_fft_complex_radix2_dif_backward(VALUE obj)
224
+ {
225
+ return rb_fft_complex_radix2(obj,
226
+ gsl_fft_complex_radix2_dif_backward,
227
+ RB_GSL_FFT_COPY);
228
+ }
229
+ static VALUE rb_gsl_fft_complex_radix2_dif_inverse(VALUE obj)
230
+ {
231
+ return rb_fft_complex_radix2(obj,
232
+ gsl_fft_complex_radix2_dif_inverse,
233
+ RB_GSL_FFT_COPY);
234
+ }
235
+
236
+ static VALUE rb_gsl_fft_complex_radix2_dif_backward2(VALUE obj)
237
+ {
238
+ return rb_fft_complex_radix2(obj,
239
+ gsl_fft_complex_radix2_dif_backward,
240
+ RB_GSL_FFT_INPLACE);
241
+ }
242
+ static VALUE rb_gsl_fft_complex_radix2_dif_inverse2(VALUE obj)
243
+ {
244
+ return rb_fft_complex_radix2(obj,
245
+ gsl_fft_complex_radix2_dif_inverse,
246
+ RB_GSL_FFT_INPLACE);
247
+ }
248
+
249
+ static VALUE rb_GSL_FFT_Wavetable_n(VALUE obj)
250
+ {
251
+ GSL_FFT_Wavetable *table;
252
+ Data_Get_Struct(obj, GSL_FFT_Wavetable, table);
253
+ return INT2FIX(table->n);
254
+ }
255
+
256
+ static VALUE rb_GSL_FFT_Wavetable_nf(VALUE obj)
257
+ {
258
+ GSL_FFT_Wavetable *table;
259
+ Data_Get_Struct(obj, GSL_FFT_Wavetable, table);
260
+ return INT2FIX(table->nf);
261
+ }
262
+
263
+ static VALUE rb_GSL_FFT_Wavetable_factor(VALUE obj)
264
+ {
265
+ GSL_FFT_Wavetable *table;
266
+ gsl_vector_int *v;
267
+ size_t i;
268
+ Data_Get_Struct(obj, GSL_FFT_Wavetable, table);
269
+ v = gsl_vector_int_alloc(table->nf);
270
+ for (i = 0; i < table->nf; i++) gsl_vector_int_set(v, i, table->factor[i]);
271
+ return Data_Wrap_Struct(cgsl_vector_int, 0, gsl_vector_int_free, v);
272
+ }
273
+
274
+ enum {
275
+ NONE_OF_TWO = 0,
276
+ ALLOC_SPACE = 1,
277
+ ALLOC_TABLE = 2,
278
+ BOTH_OF_TWO = 3,
279
+ } FFTComplexStructAllocFlag;
280
+
281
+ static void gsl_fft_free(int flag, GSL_FFT_Wavetable *table,
282
+ GSL_FFT_Workspace *space);
283
+
284
+ // Parse argc, argv. obj must be GSL::Vector::Complex.
285
+ // This can be simplified at some point.
286
+ // See comments preceding get_complex_stride_n()
287
+ static int gsl_fft_get_argv_complex(int argc, VALUE *argv, VALUE obj,
288
+ gsl_vector_complex ** vin,
289
+ gsl_complex_packed_array *data, size_t *stride,
290
+ size_t *n, gsl_fft_complex_wavetable **table,
291
+ gsl_fft_complex_workspace **space)
292
+ {
293
+ int flag = NONE_OF_TWO, flagtmp, i, itmp = argc, itmp2 = 0, ccc;
294
+ int flagw = 0;
295
+
296
+ CHECK_VECTOR_COMPLEX(obj);
297
+
298
+ ccc = argc;
299
+ flagtmp = 0;
300
+ flagw = 0;
301
+ for (i = argc-1; i >= itmp2; i--) {
302
+ if (rb_obj_is_kind_of(argv[i], cgsl_fft_complex_workspace)) {
303
+ Data_Get_Struct(argv[i], gsl_fft_complex_workspace, *space);
304
+ flagtmp = 1;
305
+ flagw = 1;
306
+ itmp = i;
307
+ ccc--;
308
+ break;
309
+ }
310
+ }
311
+ flagtmp = 0;
312
+ for (i = itmp-1; i >= itmp2; i--) {
313
+ if (rb_obj_is_kind_of(argv[i], cgsl_fft_complex_wavetable)) {
314
+ Data_Get_Struct(argv[i], gsl_fft_complex_wavetable, *table);
315
+ flagtmp = 1;
316
+ ccc--;
317
+ break;
318
+ }
319
+ }
320
+ get_complex_stride_n(obj, vin, data, stride, n);
321
+ if (flagw == 0) {
322
+ *space = gsl_fft_complex_workspace_alloc(*n);
323
+ flag += ALLOC_SPACE;
324
+ }
325
+ if (flagtmp == 0) {
326
+ *table = gsl_fft_complex_wavetable_alloc(*n);
327
+ flag += ALLOC_TABLE;
328
+ }
329
+ if (*table == NULL) {
330
+ rb_raise(rb_eRuntimeError, "something wrong with wavetable");
331
+ }
332
+ if (*space == NULL) {
333
+ rb_raise(rb_eRuntimeError, "something wrong with workspace");
334
+ }
335
+ return flag;
336
+ }
337
+
338
+ // Parse argc, argv. obj must be GSL::Vector of real data
339
+ static int gsl_fft_get_argv_real(int argc, VALUE *argv, VALUE obj,
340
+ double **ptr, size_t *stride,
341
+ size_t *n, gsl_fft_real_wavetable **table,
342
+ gsl_fft_real_workspace **space, int *naflag)
343
+ {
344
+ int flag = NONE_OF_TWO, flagtmp, i, itmp = argc, itmp2 = 0, ccc;
345
+ int flagw = 0;
346
+ *naflag = 0;
347
+
348
+ *ptr = get_ptr_double3(obj, n, stride, naflag);
349
+
350
+ ccc = argc;
351
+ flagtmp = 0;
352
+ flagw = 0;
353
+ for (i = argc-1; i >= itmp2; i--) {
354
+ if (rb_obj_is_kind_of(argv[i], cgsl_fft_real_workspace)) {
355
+ Data_Get_Struct(argv[i], gsl_fft_real_workspace, *space);
356
+ flagtmp = 1;
357
+ flagw = 1;
358
+ itmp = i;
359
+ ccc--;
360
+ break;
361
+ }
362
+ }
363
+ flagtmp = 0;
364
+ for (i = itmp-1; i >= itmp2; i--) {
365
+ if (rb_obj_is_kind_of(argv[i], cgsl_fft_real_wavetable)) {
366
+ Data_Get_Struct(argv[i], gsl_fft_real_wavetable, *table);
367
+ flagtmp = 1;
368
+ ccc--;
369
+ break;
370
+ }
371
+ }
372
+ if (flagw == 0) {
373
+ *space = gsl_fft_real_workspace_alloc(*n);
374
+ flag += ALLOC_SPACE;
375
+ }
376
+ if (flagtmp == 0) {
377
+ *table = gsl_fft_real_wavetable_alloc(*n);
378
+ flag += ALLOC_TABLE;
379
+ }
380
+ if (*table == NULL) {
381
+ rb_raise(rb_eRuntimeError, "something wrong with wavetable");
382
+ }
383
+ if (*space == NULL) {
384
+ rb_raise(rb_eRuntimeError, "something wrong with workspace");
385
+ }
386
+ return flag;
387
+ }
388
+
389
+ // Parse argc, argv. obj must be GSL::Vector of halfcomplex data
390
+ static int gsl_fft_get_argv_halfcomplex(int argc, VALUE *argv, VALUE obj,
391
+ double **ptr, size_t *stride,
392
+ size_t *n, gsl_fft_halfcomplex_wavetable **table,
393
+ gsl_fft_real_workspace **space, int *naflag)
394
+ {
395
+ int flag = NONE_OF_TWO, flagtmp, i, itmp = argc, itmp2 = 0, ccc;
396
+ int flagw = 0;
397
+
398
+ *ptr = get_ptr_double3(obj, n, stride, naflag);
399
+
400
+ ccc = argc;
401
+ flagtmp = 0;
402
+ flagw = 0;
403
+ for (i = argc-1; i >= itmp2; i--) {
404
+ if (rb_obj_is_kind_of(argv[i], cgsl_fft_real_workspace)) {
405
+ Data_Get_Struct(argv[i], gsl_fft_real_workspace, *space);
406
+ flagtmp = 1;
407
+ flagw = 1;
408
+ itmp = i;
409
+ ccc--;
410
+ break;
411
+ }
412
+ }
413
+ flagtmp = 0;
414
+ for (i = itmp-1; i >= itmp2; i--) {
415
+ if (rb_obj_is_kind_of(argv[i], cgsl_fft_halfcomplex_wavetable)) {
416
+ Data_Get_Struct(argv[i], gsl_fft_halfcomplex_wavetable, *table);
417
+ flagtmp = 1;
418
+ ccc--;
419
+ break;
420
+ }
421
+ }
422
+ if (flagw == 0) {
423
+ *space = gsl_fft_real_workspace_alloc(*n);
424
+ flag += ALLOC_SPACE;
425
+ }
426
+ if (flagtmp == 0) {
427
+ *table = gsl_fft_halfcomplex_wavetable_alloc(*n);
428
+ flag += ALLOC_TABLE;
429
+ }
430
+ if (*table == NULL) {
431
+ rb_raise(rb_eRuntimeError, "something wrong with wavetable");
432
+ }
433
+ if (*space == NULL) {
434
+ rb_raise(rb_eRuntimeError, "something wrong with workspace");
435
+ }
436
+ return flag;
437
+ }
438
+
439
+ static void gsl_fft_free(int flag, GSL_FFT_Wavetable *table,
440
+ GSL_FFT_Workspace *space)
441
+ {
442
+ switch (flag) {
443
+ case ALLOC_TABLE:
444
+ GSL_FFT_Wavetable_free(table);
445
+ break;
446
+ case ALLOC_SPACE:
447
+ GSL_FFT_Workspace_free(space);
448
+ break;
449
+ case BOTH_OF_TWO:
450
+ GSL_FFT_Wavetable_free(table);
451
+ GSL_FFT_Workspace_free(space);
452
+ break;
453
+ default:
454
+ /* never happens */
455
+ break;
456
+ }
457
+ }
458
+
459
+ static VALUE rb_fft_complex_trans(int argc, VALUE *argv, VALUE obj,
460
+ int (*transform)(gsl_complex_packed_array,
461
+ size_t, size_t,
462
+ const gsl_fft_complex_wavetable *,
463
+ gsl_fft_complex_workspace *),
464
+ int sss)
465
+ {
466
+ int flag = 0;
467
+ // local variable "status" was defined and set, but never used
468
+ //int status;
469
+ size_t stride, n;
470
+ gsl_complex_packed_array data;
471
+ gsl_vector_complex *vin, *vout;
472
+ gsl_fft_complex_wavetable *table = NULL;
473
+ gsl_fft_complex_workspace *space = NULL;
474
+ flag = gsl_fft_get_argv_complex(argc, argv, obj, &vin, &data, &stride, &n, &table, &space);
475
+ if (sss == RB_GSL_FFT_COPY) {
476
+ vout = gsl_vector_complex_alloc(n);
477
+ gsl_vector_complex_memcpy(vout, vin);
478
+ /*status =*/ (*transform)(vout->data, vout->stride /*1*/, vout->size /*n*/, table, space);
479
+ gsl_fft_free(flag, (GSL_FFT_Wavetable *) table, (GSL_FFT_Workspace *) space);
480
+ return Data_Wrap_Struct(cgsl_vector_complex, 0, gsl_vector_complex_free, vout);
481
+ } else { /* in-place */
482
+ /*status =*/ (*transform)(data, stride, n, table, space);
483
+ gsl_fft_free(flag, (GSL_FFT_Wavetable *) table, (GSL_FFT_Workspace *) space);
484
+ return obj;
485
+ }
486
+ }
487
+
488
+ static VALUE rb_gsl_fft_complex_forward(int argc, VALUE *argv, VALUE obj)
489
+ {
490
+ return rb_fft_complex_trans(argc, argv, obj, gsl_fft_complex_forward,
491
+ RB_GSL_FFT_COPY);
492
+ }
493
+
494
+ static VALUE rb_gsl_fft_complex_forward2(int argc, VALUE *argv, VALUE obj)
495
+ {
496
+ return rb_fft_complex_trans(argc, argv, obj, gsl_fft_complex_forward,
497
+ RB_GSL_FFT_INPLACE);
498
+ }
499
+
500
+ static VALUE rb_gsl_fft_complex_transform(int argc, VALUE *argv, VALUE obj)
501
+ {
502
+ int flag = 0;
503
+ // local variable "status" was defined and set, but never used
504
+ //int status;
505
+ size_t stride, n;
506
+ gsl_vector_complex *vin, *vout;
507
+ gsl_fft_direction sign;
508
+ gsl_complex_packed_array data;
509
+ gsl_fft_complex_wavetable *table = NULL;
510
+ gsl_fft_complex_workspace *space = NULL;
511
+ CHECK_FIXNUM(argv[argc-1]);
512
+ sign = FIX2INT(argv[argc-1]);
513
+ flag = gsl_fft_get_argv_complex(argc-1, argv, obj, &vin, &data, &stride, &n, &table, &space);
514
+ vout = gsl_vector_complex_alloc(n);
515
+ gsl_vector_complex_memcpy(vout, vin);
516
+ /*status =*/ gsl_fft_complex_transform(vout->data, stride, n, table, space, sign);
517
+ gsl_fft_free(flag, (GSL_FFT_Wavetable *) table, (GSL_FFT_Workspace *) space);
518
+ return Data_Wrap_Struct(cgsl_vector_complex, 0, gsl_vector_complex_free, vout);
519
+ }
520
+
521
+ /* in-place */
522
+ static VALUE rb_gsl_fft_complex_transform2(int argc, VALUE *argv, VALUE obj)
523
+ {
524
+ int flag = 0;
525
+ // local variable "status" was defined and set, but never used
526
+ //int status;
527
+ size_t stride, n;
528
+ gsl_fft_direction sign;
529
+ gsl_complex_packed_array data;
530
+ gsl_fft_complex_wavetable *table = NULL;
531
+ gsl_fft_complex_workspace *space = NULL;
532
+ CHECK_FIXNUM(argv[argc-1]);
533
+ sign = FIX2INT(argv[argc-1]);
534
+ flag = gsl_fft_get_argv_complex(argc-1, argv, obj, NULL, &data, &stride, &n, &table, &space);
535
+ /*status =*/ gsl_fft_complex_transform(data, stride, n, table, space, sign);
536
+ gsl_fft_free(flag, (GSL_FFT_Wavetable *) table, (GSL_FFT_Workspace *) space);
537
+ return obj;
538
+ }
539
+
540
+ static VALUE rb_gsl_fft_complex_backward(int argc, VALUE *argv, VALUE obj)
541
+ {
542
+ return rb_fft_complex_trans(argc, argv, obj, gsl_fft_complex_backward,
543
+ RB_GSL_FFT_COPY);
544
+ }
545
+
546
+ static VALUE rb_gsl_fft_complex_backward2(int argc, VALUE *argv, VALUE obj)
547
+ {
548
+ return rb_fft_complex_trans(argc, argv, obj, gsl_fft_complex_backward,
549
+ RB_GSL_FFT_INPLACE);
550
+ }
551
+
552
+ static VALUE rb_gsl_fft_complex_inverse(int argc, VALUE *argv, VALUE obj)
553
+ {
554
+ return rb_fft_complex_trans(argc, argv, obj, gsl_fft_complex_inverse,
555
+ RB_GSL_FFT_COPY);
556
+ }
557
+
558
+ static VALUE rb_gsl_fft_complex_inverse2(int argc, VALUE *argv, VALUE obj)
559
+ {
560
+ return rb_fft_complex_trans(argc, argv, obj, gsl_fft_complex_inverse,
561
+ RB_GSL_FFT_INPLACE);
562
+ }
563
+
564
+ // The FFT methods used to allow passing stride and n values as optional
565
+ // parameters to control which elements get transformed. This created problems
566
+ // for Views which can have their own stride, so support for stride and n
567
+ // parameters to the transform methods is being dropped. This method used to
568
+ // be called to determine the stride and n values to use based on the
569
+ // parameters and/or the vector itself (depending on how many parameters were
570
+ // passed). Now this function is somewhat unneceesary, but to simplify the code
571
+ // refactoring, it has been left in place for the time being. Eventually it
572
+ // can be refactored away completely.
573
+ //
574
+ // obj must be GSL::Vector of real or halfcomplex data
575
+ static VALUE get_ptr_stride_n(VALUE obj,
576
+ double **ptr, size_t *stride, size_t *n, int *flag)
577
+ {
578
+ *flag = 0;
579
+ *ptr = get_ptr_double3(obj, n, stride, flag);
580
+ return obj;
581
+ }
582
+
583
+ static VALUE rb_fft_radix2(VALUE obj,
584
+ int (*trans)(double [], size_t, size_t),
585
+ int sss)
586
+ {
587
+ size_t stride, n;
588
+ gsl_vector *vnew;
589
+ gsl_vector_view vv;
590
+ double *ptr1, *ptr2;
591
+ int flag;
592
+ #ifdef HAVE_NARRAY_H
593
+ int shape[1];
594
+ #endif
595
+ VALUE ary;
596
+ get_ptr_stride_n(obj, &ptr1, &stride, &n, &flag);
597
+ if (flag == 0) {
598
+ if (sss == RB_GSL_FFT_COPY) {
599
+ vnew = gsl_vector_alloc(n);
600
+ vv.vector.data = ptr1;
601
+ vv.vector.stride = stride;
602
+ vv.vector.size = n;
603
+ gsl_vector_memcpy(vnew, &vv.vector);
604
+ ptr2 = vnew->data;
605
+ stride = 1;
606
+ ary = Data_Wrap_Struct(cgsl_vector, 0, gsl_vector_free, vnew);
607
+ } else {
608
+ ary = obj;
609
+ ptr2 = ptr1;
610
+ }
611
+ #ifdef HAVE_NARRAY_H
612
+ } else if (flag == 1) {
613
+ if (sss == RB_GSL_FFT_COPY) {
614
+ shape[0] = n;
615
+ ary = na_make_object(NA_DFLOAT, 1, shape, cNArray);
616
+ ptr2 = NA_PTR_TYPE(ary, double*);
617
+ memcpy(ptr2, ptr1, sizeof(double)*n);
618
+ stride = 1;
619
+ } else {
620
+ ary = obj;
621
+ ptr2 = NA_PTR_TYPE(ary, double*);
622
+ }
623
+ #endif
624
+ } else {
625
+ rb_raise(rb_eRuntimeError, "something wrong");
626
+ }
627
+ (*trans)(ptr2, stride, n);
628
+ return ary;
629
+ }
630
+
631
+ static VALUE rb_gsl_fft_real_radix2_transform(VALUE obj)
632
+ {
633
+ return rb_fft_radix2(obj, gsl_fft_real_radix2_transform,
634
+ RB_GSL_FFT_COPY);
635
+ }
636
+
637
+ static VALUE rb_gsl_fft_real_radix2_transform2(VALUE obj)
638
+ {
639
+ return rb_fft_radix2(obj, gsl_fft_real_radix2_transform,
640
+ RB_GSL_FFT_INPLACE);
641
+ }
642
+
643
+ static VALUE rb_gsl_fft_halfcomplex_radix2_inverse(VALUE obj)
644
+ {
645
+ return rb_fft_radix2(obj, gsl_fft_halfcomplex_radix2_inverse,
646
+ RB_GSL_FFT_COPY);
647
+ }
648
+
649
+ static VALUE rb_gsl_fft_halfcomplex_radix2_inverse2(VALUE obj)
650
+ {
651
+ return rb_fft_radix2(obj, gsl_fft_halfcomplex_radix2_inverse,
652
+ RB_GSL_FFT_INPLACE);
653
+ }
654
+
655
+ static VALUE rb_gsl_fft_halfcomplex_radix2_backward(VALUE obj)
656
+ {
657
+ return rb_fft_radix2(obj, gsl_fft_halfcomplex_radix2_backward,
658
+ RB_GSL_FFT_COPY);
659
+ }
660
+
661
+ static VALUE rb_gsl_fft_halfcomplex_radix2_backward2(VALUE obj)
662
+ {
663
+ return rb_fft_radix2(obj, gsl_fft_halfcomplex_radix2_backward,
664
+ RB_GSL_FFT_INPLACE);
665
+ }
666
+
667
+ /*****/
668
+
669
+ static VALUE rb_fft_real_trans(int argc, VALUE *argv, VALUE obj,
670
+ int (*trans)(double [], size_t, size_t,
671
+ const gsl_fft_real_wavetable *,
672
+ gsl_fft_real_workspace *),
673
+ int sss)
674
+ {
675
+ int flag = 0, naflag = 0;
676
+ // local variable "status" was defined and set, but never used
677
+ //int status;
678
+ size_t stride, n;
679
+ gsl_vector *vnew;
680
+ gsl_vector_view vv;
681
+ double *ptr1, *ptr2;
682
+ #ifdef HAVE_NARRAY_H
683
+ int shape[1];
684
+ #endif
685
+ gsl_fft_real_wavetable *table = NULL;
686
+ gsl_fft_real_workspace *space = NULL;
687
+ VALUE ary;
688
+ flag = gsl_fft_get_argv_real(argc, argv, obj, &ptr1, &stride, &n, &table, &space, &naflag);
689
+ if (naflag == 0) {
690
+ if (sss == RB_GSL_FFT_COPY) {
691
+ vnew = gsl_vector_alloc(n);
692
+ vv.vector.data = ptr1;
693
+ vv.vector.stride = stride;
694
+ vv.vector.size = n;
695
+ gsl_vector_memcpy(vnew, &vv.vector);
696
+ ptr2 = vnew->data;
697
+ stride = 1;
698
+ ary = Data_Wrap_Struct(cgsl_vector, 0, gsl_vector_free, vnew);
699
+ } else {
700
+ ptr2 = ptr1;
701
+ ary = obj;
702
+ }
703
+ #ifdef HAVE_NARRAY_H
704
+ } else if (naflag == 1) {
705
+ if (sss == RB_GSL_FFT_COPY) {
706
+ shape[0] = n;
707
+ ary = na_make_object(NA_DFLOAT, 1, shape, cNArray);
708
+ ptr2 = NA_PTR_TYPE(ary, double*);
709
+ memcpy(ptr2, ptr1, sizeof(double)*n);
710
+ stride = 1;
711
+ } else {
712
+ ptr2 = ptr1;
713
+ ary = obj;
714
+ }
715
+ #endif
716
+ } else {
717
+ rb_raise(rb_eRuntimeError, "something wrong");
718
+ }
719
+ /*status =*/ (*trans)(ptr2, stride, n, table, space);
720
+ gsl_fft_free(flag, (GSL_FFT_Wavetable *) table, (GSL_FFT_Workspace *) space);
721
+ return ary;
722
+ }
723
+
724
+ static VALUE rb_gsl_fft_real_transform(int argc, VALUE *argv, VALUE obj)
725
+ {
726
+ return rb_fft_real_trans(argc, argv, obj, gsl_fft_real_transform,
727
+ RB_GSL_FFT_COPY);
728
+ }
729
+
730
+ static VALUE rb_gsl_fft_real_transform2(int argc, VALUE *argv, VALUE obj)
731
+ {
732
+ return rb_fft_real_trans(argc, argv, obj, gsl_fft_real_transform,
733
+ RB_GSL_FFT_INPLACE);
734
+ }
735
+
736
+ static VALUE rb_fft_halfcomplex_trans(int argc, VALUE *argv, VALUE obj,
737
+ int (*trans)(double [], size_t, size_t,
738
+ const gsl_fft_halfcomplex_wavetable *, gsl_fft_real_workspace *),
739
+ int sss)
740
+ {
741
+ int flag = 0, naflag = 0;
742
+ // local variable "status" was defined and set, but never used
743
+ //int status;
744
+ size_t stride, n;
745
+ gsl_vector *vnew;
746
+ gsl_vector_view vv;
747
+ double *ptr1, *ptr2;
748
+ #ifdef HAVE_NARRAY_H
749
+ int shape[1];
750
+ #endif
751
+ gsl_fft_halfcomplex_wavetable *table = NULL;
752
+ gsl_fft_real_workspace *space = NULL;
753
+ VALUE ary;
754
+ flag = gsl_fft_get_argv_halfcomplex(argc, argv, obj, &ptr1, &stride, &n,
755
+ &table, &space, &naflag);
756
+ if (naflag == 0) {
757
+ if (sss == RB_GSL_FFT_COPY) {
758
+ vnew = gsl_vector_alloc(n);
759
+ vv.vector.data = ptr1;
760
+ vv.vector.stride = stride;
761
+ vv.vector.size = n;
762
+ gsl_vector_memcpy(vnew, &vv.vector);
763
+ ptr2 = vnew->data;
764
+ stride = 1;
765
+ ary = Data_Wrap_Struct(cgsl_vector, 0, gsl_vector_free, vnew);
766
+ } else {
767
+ ptr2 = ptr1;
768
+ ary = obj;
769
+ }
770
+ #ifdef HAVE_NARRAY_H
771
+ } else if (naflag == 1) {
772
+ if (sss == RB_GSL_FFT_COPY) {
773
+ shape[0] = n;
774
+ ary = na_make_object(NA_DFLOAT, 1, shape, cNArray);
775
+ ptr2 = NA_PTR_TYPE(ary, double*);
776
+ memcpy(ptr2, ptr1, sizeof(double)*n);
777
+ stride = 1;
778
+ } else {
779
+ ptr2 = ptr1;
780
+ ary = obj;
781
+ }
782
+ #endif
783
+ } else {
784
+ rb_raise(rb_eRuntimeError, "something wrong");
785
+ }
786
+ /*status =*/ (*trans)(ptr2, stride, n, table, space);
787
+ gsl_fft_free(flag, (GSL_FFT_Wavetable *) table, (GSL_FFT_Workspace *) space);
788
+ return ary;
789
+ }
790
+
791
+ static VALUE rb_gsl_fft_halfcomplex_transform(int argc, VALUE *argv, VALUE obj)
792
+ {
793
+ return rb_fft_halfcomplex_trans(argc, argv, obj,
794
+ gsl_fft_halfcomplex_transform,
795
+ RB_GSL_FFT_COPY);
796
+ }
797
+
798
+ static VALUE rb_gsl_fft_halfcomplex_transform2(int argc, VALUE *argv, VALUE obj)
799
+ {
800
+ return rb_fft_halfcomplex_trans(argc, argv, obj,
801
+ gsl_fft_halfcomplex_transform,
802
+ RB_GSL_FFT_INPLACE);
803
+ }
804
+
805
+ static VALUE rb_gsl_fft_halfcomplex_backward(int argc, VALUE *argv, VALUE obj)
806
+ {
807
+ return rb_fft_halfcomplex_trans(argc, argv, obj,
808
+ gsl_fft_halfcomplex_backward,
809
+ RB_GSL_FFT_COPY);
810
+ }
811
+
812
+ static VALUE rb_gsl_fft_halfcomplex_backward2(int argc, VALUE *argv, VALUE obj)
813
+ {
814
+ return rb_fft_halfcomplex_trans(argc, argv, obj,
815
+ gsl_fft_halfcomplex_backward,
816
+ RB_GSL_FFT_INPLACE);
817
+ }
818
+
819
+ static VALUE rb_gsl_fft_halfcomplex_inverse(int argc, VALUE *argv, VALUE obj)
820
+ {
821
+ return rb_fft_halfcomplex_trans(argc, argv, obj,
822
+ gsl_fft_halfcomplex_inverse,
823
+ RB_GSL_FFT_COPY);
824
+ }
825
+
826
+ static VALUE rb_gsl_fft_halfcomplex_inverse2(int argc, VALUE *argv, VALUE obj)
827
+ {
828
+ return rb_fft_halfcomplex_trans(argc, argv, obj,
829
+ gsl_fft_halfcomplex_inverse,
830
+ RB_GSL_FFT_INPLACE);
831
+ }
832
+
833
+ static VALUE rb_gsl_fft_real_unpack(VALUE obj)
834
+ {
835
+ gsl_vector *v;
836
+ gsl_vector_complex *vout;
837
+
838
+ CHECK_VECTOR(obj);
839
+ Data_Get_Struct(obj, gsl_vector, v);
840
+
841
+ vout = gsl_vector_complex_alloc(v->size);
842
+ gsl_fft_real_unpack(v->data, (gsl_complex_packed_array) vout->data, v->stride, v->size);
843
+ return Data_Wrap_Struct(cgsl_vector_complex, 0, gsl_vector_complex_free, vout);
844
+ }
845
+
846
+ static VALUE rb_gsl_fft_halfcomplex_unpack(VALUE obj)
847
+ {
848
+ gsl_vector *v;
849
+ gsl_vector_complex *vout;
850
+
851
+ CHECK_VECTOR(obj);
852
+ Data_Get_Struct(obj, gsl_vector, v);
853
+
854
+ vout = gsl_vector_complex_alloc(v->size);
855
+ gsl_fft_halfcomplex_unpack(v->data, (gsl_complex_packed_array) vout->data, v->stride, v->size);
856
+ return Data_Wrap_Struct(cgsl_vector_complex, 0, gsl_vector_complex_free, vout);
857
+ }
858
+
859
+ /* Convert a halfcomplex data to Numerical Recipes style */
860
+ static VALUE rb_gsl_fft_halfcomplex_to_nrc(VALUE obj)
861
+ {
862
+ gsl_vector *v, *vnew;
863
+ size_t i, k;
864
+
865
+ CHECK_VECTOR(obj);
866
+ Data_Get_Struct(obj, gsl_vector, v);
867
+
868
+ vnew = gsl_vector_alloc(v->size);
869
+ gsl_vector_set(vnew, 0, gsl_vector_get(v, 0)); /* DC */
870
+ gsl_vector_set(vnew, 1, gsl_vector_get(v, v->size/2)); /* Nyquist freq */
871
+ for (i = 2, k = 1; i < vnew->size; i+=2, k++) {
872
+ gsl_vector_set(vnew, i, gsl_vector_get(v, k));
873
+ gsl_vector_set(vnew, i+1, -gsl_vector_get(v, v->size-k));
874
+ }
875
+ return Data_Wrap_Struct(cgsl_vector, 0, gsl_vector_free, vnew);
876
+ }
877
+
878
+ static VALUE rb_gsl_fft_halfcomplex_amp_phase(VALUE obj)
879
+ {
880
+ gsl_vector *v;
881
+ gsl_vector *amp, *phase;
882
+ double re, im;
883
+ VALUE vamp, vphase;
884
+ size_t i;
885
+ CHECK_VECTOR(obj);
886
+ Data_Get_Struct(obj, gsl_vector, v);
887
+ amp = gsl_vector_alloc(v->size/2);
888
+ phase = gsl_vector_alloc(v->size/2);
889
+ gsl_vector_set(amp, 0, gsl_vector_get(v, 0));
890
+ gsl_vector_set(phase, 0, 0);
891
+ gsl_vector_set(amp, amp->size-1, gsl_vector_get(v, v->size-1));
892
+ gsl_vector_set(phase, phase->size-1, 0);
893
+ for (i = 1; i < v->size-1; i+=2) {
894
+ re = gsl_vector_get(v, i);
895
+ im = gsl_vector_get(v, i+1);
896
+ gsl_vector_set(amp, i/2+1, sqrt(re*re + im*im));
897
+ gsl_vector_set(phase, i/2+1, atan2(im, re));
898
+ }
899
+ vamp = Data_Wrap_Struct(VECTOR_ROW_COL(obj), 0, gsl_vector_free, amp);
900
+ vphase = Data_Wrap_Struct(VECTOR_ROW_COL(obj), 0, gsl_vector_free, phase);
901
+ return rb_ary_new3(2, vamp, vphase);
902
+ }
903
+
904
+ void Init_gsl_fft(VALUE module)
905
+ {
906
+ mgsl_fft = rb_define_module_under(module, "FFT");
907
+
908
+ /*****/
909
+
910
+ rb_define_const(mgsl_fft, "Forward", INT2FIX(forward));
911
+ rb_define_const(mgsl_fft, "FORWARD", INT2FIX(forward));
912
+ rb_define_const(mgsl_fft, "Backward", INT2FIX(backward));
913
+ rb_define_const(mgsl_fft, "BACKWARD", INT2FIX(backward));
914
+
915
+ /* Transforms for complex vectors */
916
+ rb_define_method(cgsl_vector_complex, "radix2_forward",
917
+ rb_gsl_fft_complex_radix2_forward, 0);
918
+ rb_define_method(cgsl_vector_complex, "radix2_transform",
919
+ rb_gsl_fft_complex_radix2_transform, 1);
920
+ rb_define_method(cgsl_vector_complex, "radix2_backward",
921
+ rb_gsl_fft_complex_radix2_backward, 0);
922
+ rb_define_method(cgsl_vector_complex, "radix2_inverse",
923
+ rb_gsl_fft_complex_radix2_inverse, 0);
924
+ rb_define_method(cgsl_vector_complex, "radix2_dif_forward",
925
+ rb_gsl_fft_complex_radix2_dif_forward, 0);
926
+ rb_define_method(cgsl_vector_complex, "radix2_dif_transform",
927
+ rb_gsl_fft_complex_radix2_dif_transform, 1);
928
+ rb_define_method(cgsl_vector_complex, "radix2_dif_backward",
929
+ rb_gsl_fft_complex_radix2_dif_backward, 0);
930
+ rb_define_method(cgsl_vector_complex, "radix2_dif_inverse",
931
+ rb_gsl_fft_complex_radix2_dif_inverse, 0);
932
+
933
+ /* In-place radix-2 transforms for complex vectors */
934
+ rb_define_method(cgsl_vector_complex, "radix2_forward!",
935
+ rb_gsl_fft_complex_radix2_forward2, 0);
936
+ rb_define_method(cgsl_vector_complex, "radix2_transform!",
937
+ rb_gsl_fft_complex_radix2_transform2, 1);
938
+ rb_define_method(cgsl_vector_complex, "radix2_backward!",
939
+ rb_gsl_fft_complex_radix2_backward2, 0);
940
+ rb_define_method(cgsl_vector_complex, "radix2_inverse!",
941
+ rb_gsl_fft_complex_radix2_inverse2, 0);
942
+ rb_define_method(cgsl_vector_complex, "radix2_dif_forward!",
943
+ rb_gsl_fft_complex_radix2_dif_forward2, 0);
944
+ rb_define_method(cgsl_vector_complex, "radix2_dif_transform!",
945
+ rb_gsl_fft_complex_radix2_dif_transform2, 1);
946
+ rb_define_method(cgsl_vector_complex, "radix2_dif_backward!",
947
+ rb_gsl_fft_complex_radix2_dif_backward2, 0);
948
+ rb_define_method(cgsl_vector_complex, "radix2_dif_inverse!",
949
+ rb_gsl_fft_complex_radix2_dif_inverse2, 0);
950
+
951
+ // class GSL::FFT::Wavetable < GSL::Object
952
+ //
953
+ // Useful since some functionality is shared among
954
+ // GSL::FFT::Complex::Wavetable
955
+ // GSL::FFT::Real::Wavetable
956
+ // GSL::FFT::HalfComplex::Wavetable
957
+ cgsl_fft_wavetable = rb_define_class_under(mgsl_fft, "Wavetable", cGSL_Object);
958
+ // No alloc
959
+ // TODO Make GSL::FFT::Wavetable#initialize private?
960
+ rb_define_method(cgsl_fft_wavetable, "n",
961
+ rb_GSL_FFT_Wavetable_n, 0);
962
+ rb_define_method(cgsl_fft_wavetable, "nf",
963
+ rb_GSL_FFT_Wavetable_nf, 0);
964
+ rb_define_method(cgsl_fft_wavetable, "factor",
965
+ rb_GSL_FFT_Wavetable_factor, 0);
966
+
967
+ // class GSL::FFT::ComplexWavetable < GSL::FFT::Wavetable
968
+ cgsl_fft_complex_wavetable = rb_define_class_under(mgsl_fft, "ComplexWavetable",
969
+ cgsl_fft_wavetable);
970
+ rb_define_singleton_method(cgsl_fft_complex_wavetable, "alloc",
971
+ rb_gsl_fft_complex_wavetable_new, 1);
972
+
973
+ // class GSL::FFT::ComplexWorkspace < GSL::Object
974
+ cgsl_fft_complex_workspace = rb_define_class_under(mgsl_fft, "ComplexWorkspace",
975
+ cGSL_Object);
976
+ rb_define_singleton_method(cgsl_fft_complex_workspace, "alloc",
977
+ rb_gsl_fft_complex_workspace_new, 1);
978
+
979
+ rb_define_method(cgsl_vector_complex, "forward", rb_gsl_fft_complex_forward, -1);
980
+ rb_define_method(cgsl_vector_complex, "transform", rb_gsl_fft_complex_transform, -1);
981
+ rb_define_method(cgsl_vector_complex, "backward", rb_gsl_fft_complex_backward, -1);
982
+ rb_define_method(cgsl_vector_complex, "inverse", rb_gsl_fft_complex_inverse, -1);
983
+
984
+ rb_define_method(cgsl_vector_complex, "forward!", rb_gsl_fft_complex_forward2, -1);
985
+ rb_define_method(cgsl_vector_complex, "transform!", rb_gsl_fft_complex_transform2, -1);
986
+ rb_define_method(cgsl_vector_complex, "backward!", rb_gsl_fft_complex_backward2, -1);
987
+ rb_define_method(cgsl_vector_complex, "inverse!", rb_gsl_fft_complex_inverse2, -1);
988
+
989
+ /*****/
990
+
991
+ // TODO Do these method names need the "real_" and "halfcomplex_" prefixes?
992
+ rb_define_method(cgsl_vector, "real_radix2_transform",
993
+ rb_gsl_fft_real_radix2_transform, 0);
994
+ rb_define_alias(cgsl_vector, "radix2_transform", "real_radix2_transform");
995
+ rb_define_alias(cgsl_vector, "radix2_forward", "real_radix2_transform");
996
+ rb_define_method(cgsl_vector, "real_radix2_inverse",
997
+ rb_gsl_fft_halfcomplex_radix2_inverse, 0);
998
+ rb_define_alias(cgsl_vector, "radix2_inverse", "real_radix2_inverse");
999
+ rb_define_alias(cgsl_vector, "halfcomplex_radix2_inverse",
1000
+ "real_radix2_inverse");
1001
+ rb_define_method(cgsl_vector, "real_radix2_backward",
1002
+ rb_gsl_fft_halfcomplex_radix2_backward, 0);
1003
+ rb_define_alias(cgsl_vector, "radix2_backward", "real_radix2_backward");
1004
+ rb_define_alias(cgsl_vector, "halfcomplex_radix2_backward",
1005
+ "real_radix2_backward");
1006
+
1007
+ // TODO Do these method names need the "real_" and "halfcomplex_" prefixes?
1008
+ rb_define_method(cgsl_vector, "real_radix2_transform!",
1009
+ rb_gsl_fft_real_radix2_transform2, 0);
1010
+ rb_define_alias(cgsl_vector, "radix2_transform!", "real_radix2_transform!");
1011
+ rb_define_alias(cgsl_vector, "radix2_forward!", "real_radix2_transform!");
1012
+ rb_define_method(cgsl_vector, "real_radix2_inverse!",
1013
+ rb_gsl_fft_halfcomplex_radix2_inverse2, 0);
1014
+ rb_define_alias(cgsl_vector, "radix2_inverse!", "real_radix2_inverse!");
1015
+ rb_define_alias(cgsl_vector, "halfcomplex_radix2_inverse!",
1016
+ "real_radix2_inverse!");
1017
+ rb_define_method(cgsl_vector, "real_radix2_backward!",
1018
+ rb_gsl_fft_halfcomplex_radix2_backward2, 0);
1019
+ rb_define_alias(cgsl_vector, "radix2_backward!", "real_radix2_backward!");
1020
+ rb_define_alias(cgsl_vector, "halfcomplex_radix2_backward!",
1021
+ "real_radix2_backward!");
1022
+
1023
+ /*****/
1024
+
1025
+ // class GSL::FFT::RealWavetable < GSL::FFT::Wavetable
1026
+ cgsl_fft_real_wavetable = rb_define_class_under(mgsl_fft, "RealWavetable",
1027
+ cgsl_fft_wavetable);
1028
+ rb_define_singleton_method(cgsl_fft_real_wavetable, "alloc",
1029
+ rb_gsl_fft_real_wavetable_new, 1);
1030
+
1031
+ // class GSL::FFT::HalfComplexWavetable < GSL::FFT::Wavetable
1032
+ cgsl_fft_halfcomplex_wavetable = rb_define_class_under(mgsl_fft,
1033
+ "HalfComplexWavetable", cgsl_fft_wavetable);
1034
+ rb_define_singleton_method(cgsl_fft_halfcomplex_wavetable, "alloc",
1035
+ rb_gsl_fft_halfcomplex_wavetable_new, 1);
1036
+
1037
+ /*****/
1038
+
1039
+ // class GSL::FFT::RealWorkspace < GSL::Object
1040
+ cgsl_fft_real_workspace = rb_define_class_under(mgsl_fft, "RealWorkspace",
1041
+ cGSL_Object);
1042
+ rb_define_singleton_method(cgsl_fft_real_workspace, "alloc",
1043
+ rb_gsl_fft_real_workspace_new, 1);
1044
+
1045
+ /*****/
1046
+
1047
+ // TODO Do these method names need the "real_" and "halfcomplex_" prefixes?
1048
+ rb_define_method(cgsl_vector, "real_transform", rb_gsl_fft_real_transform, -1);
1049
+ rb_define_alias(cgsl_vector, "transform", "real_transform");
1050
+ rb_define_alias(cgsl_vector, "forward", "real_transform");
1051
+ rb_define_alias(cgsl_vector, "fft_forward", "real_transform");
1052
+ rb_define_alias(cgsl_vector, "fft", "real_transform");
1053
+ rb_define_method(cgsl_vector, "halfcomplex_transform",
1054
+ rb_gsl_fft_halfcomplex_transform, -1);
1055
+ rb_define_method(cgsl_vector, "halfcomplex_backward",
1056
+ rb_gsl_fft_halfcomplex_backward, -1);
1057
+ rb_define_alias(cgsl_vector, "backward", "halfcomplex_backward");
1058
+ rb_define_alias(cgsl_vector, "fft_backward", "halfcomplex_backward");
1059
+ rb_define_method(cgsl_vector, "halfcomplex_inverse",
1060
+ rb_gsl_fft_halfcomplex_inverse, -1);
1061
+ rb_define_alias(cgsl_vector, "fft_inverse", "halfcomplex_inverse");
1062
+ rb_define_alias(cgsl_vector, "ifft", "halfcomplex_inverse");
1063
+ rb_define_alias(cgsl_vector, "inverse", "halfcomplex_inverse");
1064
+
1065
+ rb_define_method(cgsl_vector, "real_transform!", rb_gsl_fft_real_transform2, -1);
1066
+ rb_define_alias(cgsl_vector, "transform!", "real_transform!");
1067
+ rb_define_alias(cgsl_vector, "forward!", "real_transform!");
1068
+ rb_define_alias(cgsl_vector, "fft_forward!", "real_transform!");
1069
+ rb_define_alias(cgsl_vector, "fft!", "real_transform!");
1070
+ rb_define_method(cgsl_vector, "halfcomplex_transform!",
1071
+ rb_gsl_fft_halfcomplex_transform2, -1);
1072
+ rb_define_method(cgsl_vector, "halfcomplex_backward!",
1073
+ rb_gsl_fft_halfcomplex_backward2, -1);
1074
+ rb_define_alias(cgsl_vector, "backward!", "halfcomplex_backward!");
1075
+ rb_define_alias(cgsl_vector, "fft_backward!", "halfcomplex_backward!");
1076
+ rb_define_method(cgsl_vector, "halfcomplex_inverse!",
1077
+ rb_gsl_fft_halfcomplex_inverse2, -1);
1078
+ rb_define_alias(cgsl_vector, "fft_inverse!", "halfcomplex_inverse!");
1079
+ rb_define_alias(cgsl_vector, "ifft!", "halfcomplex_inverse!");
1080
+ rb_define_alias(cgsl_vector, "inverse!", "halfcomplex_inverse!");
1081
+
1082
+ /***/
1083
+ rb_define_method(cgsl_vector, "fft_real_unpack", rb_gsl_fft_real_unpack, 0);
1084
+ rb_define_alias(cgsl_vector, "real_unpack", "fft_real_unpack");
1085
+ rb_define_alias(cgsl_vector, "real_to_complex", "fft_real_unpack");
1086
+ rb_define_alias(cgsl_vector, "r_to_c", "fft_real_unpack");
1087
+
1088
+ rb_define_method(cgsl_vector, "fft_halfcomplex_unpack",
1089
+ rb_gsl_fft_halfcomplex_unpack, 0);
1090
+ rb_define_alias(cgsl_vector, "halfcomplex_unpack", "fft_halfcomplex_unpack");
1091
+ rb_define_alias(cgsl_vector, "halfcomplex_to_complex", "fft_halfcomplex_unpack");
1092
+ rb_define_alias(cgsl_vector, "hc_to_c", "fft_halfcomplex_unpack");
1093
+
1094
+ /*****/
1095
+
1096
+ rb_define_method(cgsl_vector, "to_nrc_order",
1097
+ rb_gsl_fft_halfcomplex_to_nrc, 0);
1098
+
1099
+ rb_define_method(cgsl_vector, "halfcomplex_amp_phase",
1100
+ rb_gsl_fft_halfcomplex_amp_phase, 0);
1101
+ rb_define_alias(cgsl_vector, "hc_amp_phase", "halfcomplex_amp_phase");
1102
+ }