scipy 1.15.3__cp312-cp312-musllinux_1_2_aarch64.whl → 1.16.0rc2__cp312-cp312-musllinux_1_2_aarch64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (641) hide show
  1. scipy/__config__.py +10 -10
  2. scipy/__init__.py +3 -6
  3. scipy/_cyutility.cpython-312-aarch64-linux-musl.so +0 -0
  4. scipy/_lib/_array_api.py +486 -161
  5. scipy/_lib/_array_api_compat_vendor.py +9 -0
  6. scipy/_lib/_bunch.py +4 -0
  7. scipy/_lib/_ccallback_c.cpython-312-aarch64-linux-musl.so +0 -0
  8. scipy/_lib/_docscrape.py +1 -1
  9. scipy/_lib/_elementwise_iterative_method.py +15 -26
  10. scipy/_lib/_fpumode.cpython-312-aarch64-linux-musl.so +0 -0
  11. scipy/_lib/_sparse.py +41 -0
  12. scipy/_lib/_test_ccallback.cpython-312-aarch64-linux-musl.so +0 -0
  13. scipy/_lib/_test_deprecation_call.cpython-312-aarch64-linux-musl.so +0 -0
  14. scipy/_lib/_test_deprecation_def.cpython-312-aarch64-linux-musl.so +0 -0
  15. scipy/_lib/_testutils.py +6 -2
  16. scipy/_lib/_uarray/_uarray.cpython-312-aarch64-linux-musl.so +0 -0
  17. scipy/_lib/_util.py +222 -125
  18. scipy/_lib/array_api_compat/__init__.py +4 -4
  19. scipy/_lib/array_api_compat/_internal.py +19 -6
  20. scipy/_lib/array_api_compat/common/__init__.py +1 -1
  21. scipy/_lib/array_api_compat/common/_aliases.py +365 -193
  22. scipy/_lib/array_api_compat/common/_fft.py +94 -64
  23. scipy/_lib/array_api_compat/common/_helpers.py +413 -180
  24. scipy/_lib/array_api_compat/common/_linalg.py +116 -40
  25. scipy/_lib/array_api_compat/common/_typing.py +179 -10
  26. scipy/_lib/array_api_compat/cupy/__init__.py +1 -4
  27. scipy/_lib/array_api_compat/cupy/_aliases.py +61 -41
  28. scipy/_lib/array_api_compat/cupy/_info.py +16 -6
  29. scipy/_lib/array_api_compat/cupy/_typing.py +24 -39
  30. scipy/_lib/array_api_compat/dask/array/__init__.py +6 -3
  31. scipy/_lib/array_api_compat/dask/array/_aliases.py +267 -108
  32. scipy/_lib/array_api_compat/dask/array/_info.py +105 -34
  33. scipy/_lib/array_api_compat/dask/array/fft.py +5 -8
  34. scipy/_lib/array_api_compat/dask/array/linalg.py +21 -22
  35. scipy/_lib/array_api_compat/numpy/__init__.py +13 -15
  36. scipy/_lib/array_api_compat/numpy/_aliases.py +98 -49
  37. scipy/_lib/array_api_compat/numpy/_info.py +36 -16
  38. scipy/_lib/array_api_compat/numpy/_typing.py +27 -43
  39. scipy/_lib/array_api_compat/numpy/fft.py +11 -5
  40. scipy/_lib/array_api_compat/numpy/linalg.py +75 -22
  41. scipy/_lib/array_api_compat/torch/__init__.py +3 -5
  42. scipy/_lib/array_api_compat/torch/_aliases.py +262 -159
  43. scipy/_lib/array_api_compat/torch/_info.py +27 -16
  44. scipy/_lib/array_api_compat/torch/_typing.py +3 -0
  45. scipy/_lib/array_api_compat/torch/fft.py +17 -18
  46. scipy/_lib/array_api_compat/torch/linalg.py +16 -16
  47. scipy/_lib/array_api_extra/__init__.py +26 -3
  48. scipy/_lib/array_api_extra/_delegation.py +171 -0
  49. scipy/_lib/array_api_extra/_lib/__init__.py +1 -0
  50. scipy/_lib/array_api_extra/_lib/_at.py +463 -0
  51. scipy/_lib/array_api_extra/_lib/_backends.py +46 -0
  52. scipy/_lib/array_api_extra/_lib/_funcs.py +937 -0
  53. scipy/_lib/array_api_extra/_lib/_lazy.py +357 -0
  54. scipy/_lib/array_api_extra/_lib/_testing.py +278 -0
  55. scipy/_lib/array_api_extra/_lib/_utils/__init__.py +1 -0
  56. scipy/_lib/array_api_extra/_lib/_utils/_compat.py +74 -0
  57. scipy/_lib/array_api_extra/_lib/_utils/_compat.pyi +45 -0
  58. scipy/_lib/array_api_extra/_lib/_utils/_helpers.py +559 -0
  59. scipy/_lib/array_api_extra/_lib/_utils/_typing.py +10 -0
  60. scipy/_lib/array_api_extra/_lib/_utils/_typing.pyi +105 -0
  61. scipy/_lib/array_api_extra/testing.py +359 -0
  62. scipy/_lib/decorator.py +2 -2
  63. scipy/_lib/doccer.py +1 -7
  64. scipy/_lib/messagestream.cpython-312-aarch64-linux-musl.so +0 -0
  65. scipy/_lib/pyprima/__init__.py +212 -0
  66. scipy/_lib/pyprima/cobyla/__init__.py +0 -0
  67. scipy/_lib/pyprima/cobyla/cobyla.py +559 -0
  68. scipy/_lib/pyprima/cobyla/cobylb.py +714 -0
  69. scipy/_lib/pyprima/cobyla/geometry.py +226 -0
  70. scipy/_lib/pyprima/cobyla/initialize.py +215 -0
  71. scipy/_lib/pyprima/cobyla/trustregion.py +492 -0
  72. scipy/_lib/pyprima/cobyla/update.py +289 -0
  73. scipy/_lib/pyprima/common/__init__.py +0 -0
  74. scipy/_lib/pyprima/common/_bounds.py +34 -0
  75. scipy/_lib/pyprima/common/_linear_constraints.py +46 -0
  76. scipy/_lib/pyprima/common/_nonlinear_constraints.py +54 -0
  77. scipy/_lib/pyprima/common/_project.py +173 -0
  78. scipy/_lib/pyprima/common/checkbreak.py +93 -0
  79. scipy/_lib/pyprima/common/consts.py +47 -0
  80. scipy/_lib/pyprima/common/evaluate.py +99 -0
  81. scipy/_lib/pyprima/common/history.py +38 -0
  82. scipy/_lib/pyprima/common/infos.py +30 -0
  83. scipy/_lib/pyprima/common/linalg.py +435 -0
  84. scipy/_lib/pyprima/common/message.py +290 -0
  85. scipy/_lib/pyprima/common/powalg.py +131 -0
  86. scipy/_lib/pyprima/common/preproc.py +277 -0
  87. scipy/_lib/pyprima/common/present.py +5 -0
  88. scipy/_lib/pyprima/common/ratio.py +54 -0
  89. scipy/_lib/pyprima/common/redrho.py +47 -0
  90. scipy/_lib/pyprima/common/selectx.py +296 -0
  91. scipy/_lib/tests/test__util.py +105 -121
  92. scipy/_lib/tests/test_array_api.py +166 -35
  93. scipy/_lib/tests/test_bunch.py +7 -0
  94. scipy/_lib/tests/test_ccallback.py +2 -10
  95. scipy/_lib/tests/test_public_api.py +13 -0
  96. scipy/cluster/_hierarchy.cpython-312-aarch64-linux-musl.so +0 -0
  97. scipy/cluster/_optimal_leaf_ordering.cpython-312-aarch64-linux-musl.so +0 -0
  98. scipy/cluster/_vq.cpython-312-aarch64-linux-musl.so +0 -0
  99. scipy/cluster/hierarchy.py +393 -223
  100. scipy/cluster/tests/test_hierarchy.py +273 -335
  101. scipy/cluster/tests/test_vq.py +45 -61
  102. scipy/cluster/vq.py +39 -35
  103. scipy/conftest.py +263 -157
  104. scipy/constants/_constants.py +4 -1
  105. scipy/constants/tests/test_codata.py +2 -2
  106. scipy/constants/tests/test_constants.py +11 -18
  107. scipy/datasets/_download_all.py +15 -1
  108. scipy/datasets/_fetchers.py +7 -1
  109. scipy/datasets/_utils.py +1 -1
  110. scipy/differentiate/_differentiate.py +25 -25
  111. scipy/differentiate/tests/test_differentiate.py +24 -25
  112. scipy/fft/_basic.py +20 -0
  113. scipy/fft/_helper.py +3 -34
  114. scipy/fft/_pocketfft/helper.py +29 -1
  115. scipy/fft/_pocketfft/pypocketfft.cpython-312-aarch64-linux-musl.so +0 -0
  116. scipy/fft/_pocketfft/tests/test_basic.py +2 -4
  117. scipy/fft/_pocketfft/tests/test_real_transforms.py +4 -4
  118. scipy/fft/_realtransforms.py +13 -0
  119. scipy/fft/tests/test_basic.py +27 -25
  120. scipy/fft/tests/test_fftlog.py +16 -7
  121. scipy/fft/tests/test_helper.py +18 -34
  122. scipy/fft/tests/test_real_transforms.py +8 -10
  123. scipy/fftpack/convolve.cpython-312-aarch64-linux-musl.so +0 -0
  124. scipy/fftpack/tests/test_basic.py +2 -4
  125. scipy/fftpack/tests/test_real_transforms.py +8 -9
  126. scipy/integrate/_bvp.py +9 -3
  127. scipy/integrate/_cubature.py +3 -2
  128. scipy/integrate/_dop.cpython-312-aarch64-linux-musl.so +0 -0
  129. scipy/integrate/_lsoda.cpython-312-aarch64-linux-musl.so +0 -0
  130. scipy/integrate/_ode.py +9 -2
  131. scipy/integrate/_odepack.cpython-312-aarch64-linux-musl.so +0 -0
  132. scipy/integrate/_quad_vec.py +21 -29
  133. scipy/integrate/_quadpack.cpython-312-aarch64-linux-musl.so +0 -0
  134. scipy/integrate/_quadpack_py.py +11 -7
  135. scipy/integrate/_quadrature.py +3 -3
  136. scipy/integrate/_rules/_base.py +2 -2
  137. scipy/integrate/_tanhsinh.py +48 -47
  138. scipy/integrate/_test_multivariate.cpython-312-aarch64-linux-musl.so +0 -0
  139. scipy/integrate/_test_odeint_banded.cpython-312-aarch64-linux-musl.so +0 -0
  140. scipy/integrate/_vode.cpython-312-aarch64-linux-musl.so +0 -0
  141. scipy/integrate/tests/test__quad_vec.py +0 -6
  142. scipy/integrate/tests/test_banded_ode_solvers.py +85 -0
  143. scipy/integrate/tests/test_cubature.py +21 -35
  144. scipy/integrate/tests/test_quadrature.py +6 -8
  145. scipy/integrate/tests/test_tanhsinh.py +56 -48
  146. scipy/interpolate/__init__.py +70 -58
  147. scipy/interpolate/_bary_rational.py +22 -22
  148. scipy/interpolate/_bsplines.py +119 -66
  149. scipy/interpolate/_cubic.py +65 -50
  150. scipy/interpolate/_dfitpack.cpython-312-aarch64-linux-musl.so +0 -0
  151. scipy/interpolate/_dierckx.cpython-312-aarch64-linux-musl.so +0 -0
  152. scipy/interpolate/_fitpack.cpython-312-aarch64-linux-musl.so +0 -0
  153. scipy/interpolate/_fitpack2.py +9 -6
  154. scipy/interpolate/_fitpack_impl.py +32 -26
  155. scipy/interpolate/_fitpack_repro.py +23 -19
  156. scipy/interpolate/_interpnd.cpython-312-aarch64-linux-musl.so +0 -0
  157. scipy/interpolate/_interpolate.py +30 -12
  158. scipy/interpolate/_ndbspline.py +13 -18
  159. scipy/interpolate/_ndgriddata.py +5 -8
  160. scipy/interpolate/_polyint.py +95 -31
  161. scipy/interpolate/_ppoly.cpython-312-aarch64-linux-musl.so +0 -0
  162. scipy/interpolate/_rbf.py +2 -2
  163. scipy/interpolate/_rbfinterp.py +1 -1
  164. scipy/interpolate/_rbfinterp_pythran.cpython-312-aarch64-linux-musl.so +0 -0
  165. scipy/interpolate/_rgi.py +31 -26
  166. scipy/interpolate/_rgi_cython.cpython-312-aarch64-linux-musl.so +0 -0
  167. scipy/interpolate/dfitpack.py +0 -20
  168. scipy/interpolate/interpnd.py +1 -2
  169. scipy/interpolate/tests/test_bary_rational.py +2 -2
  170. scipy/interpolate/tests/test_bsplines.py +97 -1
  171. scipy/interpolate/tests/test_fitpack2.py +39 -1
  172. scipy/interpolate/tests/test_interpnd.py +32 -20
  173. scipy/interpolate/tests/test_interpolate.py +48 -4
  174. scipy/interpolate/tests/test_rgi.py +2 -1
  175. scipy/io/_fast_matrix_market/__init__.py +2 -0
  176. scipy/io/_fast_matrix_market/_fmm_core.cpython-312-aarch64-linux-musl.so +0 -0
  177. scipy/io/_harwell_boeing/_fortran_format_parser.py +19 -16
  178. scipy/io/_harwell_boeing/hb.py +7 -11
  179. scipy/io/_idl.py +5 -7
  180. scipy/io/_netcdf.py +15 -5
  181. scipy/io/_test_fortran.cpython-312-aarch64-linux-musl.so +0 -0
  182. scipy/io/arff/tests/test_arffread.py +3 -3
  183. scipy/io/matlab/__init__.py +5 -3
  184. scipy/io/matlab/_mio.py +4 -1
  185. scipy/io/matlab/_mio5.py +19 -13
  186. scipy/io/matlab/_mio5_utils.cpython-312-aarch64-linux-musl.so +0 -0
  187. scipy/io/matlab/_mio_utils.cpython-312-aarch64-linux-musl.so +0 -0
  188. scipy/io/matlab/_miobase.py +4 -1
  189. scipy/io/matlab/_streams.cpython-312-aarch64-linux-musl.so +0 -0
  190. scipy/io/matlab/tests/test_mio.py +46 -18
  191. scipy/io/matlab/tests/test_mio_funcs.py +1 -1
  192. scipy/io/tests/test_mmio.py +7 -1
  193. scipy/io/tests/test_wavfile.py +41 -0
  194. scipy/io/wavfile.py +57 -10
  195. scipy/linalg/_basic.py +113 -86
  196. scipy/linalg/_cythonized_array_utils.cpython-312-aarch64-linux-musl.so +0 -0
  197. scipy/linalg/_decomp.py +22 -9
  198. scipy/linalg/_decomp_cholesky.py +28 -13
  199. scipy/linalg/_decomp_cossin.py +45 -30
  200. scipy/linalg/_decomp_interpolative.cpython-312-aarch64-linux-musl.so +0 -0
  201. scipy/linalg/_decomp_ldl.py +4 -1
  202. scipy/linalg/_decomp_lu.py +18 -6
  203. scipy/linalg/_decomp_lu_cython.cpython-312-aarch64-linux-musl.so +0 -0
  204. scipy/linalg/_decomp_polar.py +2 -0
  205. scipy/linalg/_decomp_qr.py +6 -2
  206. scipy/linalg/_decomp_qz.py +3 -0
  207. scipy/linalg/_decomp_schur.py +3 -1
  208. scipy/linalg/_decomp_svd.py +13 -2
  209. scipy/linalg/_decomp_update.cpython-312-aarch64-linux-musl.so +0 -0
  210. scipy/linalg/_expm_frechet.py +4 -0
  211. scipy/linalg/_fblas.cpython-312-aarch64-linux-musl.so +0 -0
  212. scipy/linalg/_flapack.cpython-312-aarch64-linux-musl.so +0 -0
  213. scipy/linalg/_linalg_pythran.cpython-312-aarch64-linux-musl.so +0 -0
  214. scipy/linalg/_matfuncs.py +187 -4
  215. scipy/linalg/_matfuncs_expm.cpython-312-aarch64-linux-musl.so +0 -0
  216. scipy/linalg/_matfuncs_schur_sqrtm.cpython-312-aarch64-linux-musl.so +0 -0
  217. scipy/linalg/_matfuncs_sqrtm.py +1 -99
  218. scipy/linalg/_matfuncs_sqrtm_triu.cpython-312-aarch64-linux-musl.so +0 -0
  219. scipy/linalg/_procrustes.py +2 -0
  220. scipy/linalg/_sketches.py +17 -6
  221. scipy/linalg/_solve_toeplitz.cpython-312-aarch64-linux-musl.so +0 -0
  222. scipy/linalg/_solvers.py +7 -2
  223. scipy/linalg/_special_matrices.py +26 -36
  224. scipy/linalg/cython_blas.cpython-312-aarch64-linux-musl.so +0 -0
  225. scipy/linalg/cython_lapack.cpython-312-aarch64-linux-musl.so +0 -0
  226. scipy/linalg/lapack.py +22 -2
  227. scipy/linalg/tests/_cython_examples/meson.build +7 -0
  228. scipy/linalg/tests/test_basic.py +31 -16
  229. scipy/linalg/tests/test_batch.py +588 -0
  230. scipy/linalg/tests/test_cythonized_array_utils.py +0 -2
  231. scipy/linalg/tests/test_decomp.py +40 -3
  232. scipy/linalg/tests/test_decomp_cossin.py +14 -0
  233. scipy/linalg/tests/test_decomp_ldl.py +1 -1
  234. scipy/linalg/tests/test_lapack.py +115 -7
  235. scipy/linalg/tests/test_matfuncs.py +157 -102
  236. scipy/linalg/tests/test_procrustes.py +0 -7
  237. scipy/linalg/tests/test_solve_toeplitz.py +1 -1
  238. scipy/linalg/tests/test_special_matrices.py +1 -5
  239. scipy/ndimage/__init__.py +1 -0
  240. scipy/ndimage/_ctest.cpython-312-aarch64-linux-musl.so +0 -0
  241. scipy/ndimage/_cytest.cpython-312-aarch64-linux-musl.so +0 -0
  242. scipy/ndimage/_delegators.py +8 -2
  243. scipy/ndimage/_filters.py +453 -5
  244. scipy/ndimage/_interpolation.py +36 -6
  245. scipy/ndimage/_measurements.py +4 -2
  246. scipy/ndimage/_morphology.py +5 -0
  247. scipy/ndimage/_nd_image.cpython-312-aarch64-linux-musl.so +0 -0
  248. scipy/ndimage/_ni_docstrings.py +5 -1
  249. scipy/ndimage/_ni_label.cpython-312-aarch64-linux-musl.so +0 -0
  250. scipy/ndimage/_ni_support.py +1 -5
  251. scipy/ndimage/_rank_filter_1d.cpython-312-aarch64-linux-musl.so +0 -0
  252. scipy/ndimage/_support_alternative_backends.py +18 -6
  253. scipy/ndimage/tests/test_filters.py +370 -259
  254. scipy/ndimage/tests/test_fourier.py +7 -9
  255. scipy/ndimage/tests/test_interpolation.py +68 -61
  256. scipy/ndimage/tests/test_measurements.py +18 -35
  257. scipy/ndimage/tests/test_morphology.py +143 -131
  258. scipy/ndimage/tests/test_splines.py +1 -3
  259. scipy/odr/__odrpack.cpython-312-aarch64-linux-musl.so +0 -0
  260. scipy/optimize/_basinhopping.py +13 -7
  261. scipy/optimize/_bglu_dense.cpython-312-aarch64-linux-musl.so +0 -0
  262. scipy/optimize/_bracket.py +17 -24
  263. scipy/optimize/_chandrupatla.py +9 -10
  264. scipy/optimize/_cobyla_py.py +104 -123
  265. scipy/optimize/_constraints.py +14 -10
  266. scipy/optimize/_differentiable_functions.py +371 -230
  267. scipy/optimize/_differentialevolution.py +4 -3
  268. scipy/optimize/_direct.cpython-312-aarch64-linux-musl.so +0 -0
  269. scipy/optimize/_dual_annealing.py +1 -1
  270. scipy/optimize/_elementwise.py +1 -4
  271. scipy/optimize/_group_columns.cpython-312-aarch64-linux-musl.so +0 -0
  272. scipy/optimize/_highspy/_core.cpython-312-aarch64-linux-musl.so +0 -0
  273. scipy/optimize/_highspy/_highs_options.cpython-312-aarch64-linux-musl.so +0 -0
  274. scipy/optimize/_lbfgsb.cpython-312-aarch64-linux-musl.so +0 -0
  275. scipy/optimize/_lbfgsb_py.py +57 -16
  276. scipy/optimize/_linprog_doc.py +2 -2
  277. scipy/optimize/_linprog_highs.py +2 -2
  278. scipy/optimize/_linprog_ip.py +25 -10
  279. scipy/optimize/_linprog_util.py +14 -16
  280. scipy/optimize/_lsap.cpython-312-aarch64-linux-musl.so +0 -0
  281. scipy/optimize/_lsq/common.py +3 -3
  282. scipy/optimize/_lsq/dogbox.py +16 -2
  283. scipy/optimize/_lsq/givens_elimination.cpython-312-aarch64-linux-musl.so +0 -0
  284. scipy/optimize/_lsq/least_squares.py +198 -126
  285. scipy/optimize/_lsq/lsq_linear.py +6 -6
  286. scipy/optimize/_lsq/trf.py +35 -8
  287. scipy/optimize/_milp.py +3 -1
  288. scipy/optimize/_minimize.py +105 -36
  289. scipy/optimize/_minpack.cpython-312-aarch64-linux-musl.so +0 -0
  290. scipy/optimize/_minpack_py.py +21 -14
  291. scipy/optimize/_moduleTNC.cpython-312-aarch64-linux-musl.so +0 -0
  292. scipy/optimize/_nnls.py +20 -21
  293. scipy/optimize/_nonlin.py +34 -3
  294. scipy/optimize/_numdiff.py +288 -110
  295. scipy/optimize/_optimize.py +86 -48
  296. scipy/optimize/_pava_pybind.cpython-312-aarch64-linux-musl.so +0 -0
  297. scipy/optimize/_remove_redundancy.py +5 -5
  298. scipy/optimize/_root_scalar.py +1 -1
  299. scipy/optimize/_shgo.py +6 -0
  300. scipy/optimize/_shgo_lib/_complex.py +1 -1
  301. scipy/optimize/_slsqp_py.py +216 -124
  302. scipy/optimize/_slsqplib.cpython-312-aarch64-linux-musl.so +0 -0
  303. scipy/optimize/_spectral.py +1 -1
  304. scipy/optimize/_tnc.py +8 -1
  305. scipy/optimize/_trlib/_trlib.cpython-312-aarch64-linux-musl.so +0 -0
  306. scipy/optimize/_trustregion.py +20 -6
  307. scipy/optimize/_trustregion_constr/canonical_constraint.py +7 -7
  308. scipy/optimize/_trustregion_constr/equality_constrained_sqp.py +1 -1
  309. scipy/optimize/_trustregion_constr/minimize_trustregion_constr.py +11 -3
  310. scipy/optimize/_trustregion_constr/projections.py +12 -8
  311. scipy/optimize/_trustregion_constr/qp_subproblem.py +9 -9
  312. scipy/optimize/_trustregion_constr/tests/test_projections.py +7 -7
  313. scipy/optimize/_trustregion_constr/tests/test_qp_subproblem.py +77 -77
  314. scipy/optimize/_trustregion_constr/tr_interior_point.py +5 -5
  315. scipy/optimize/_trustregion_exact.py +0 -1
  316. scipy/optimize/_zeros.cpython-312-aarch64-linux-musl.so +0 -0
  317. scipy/optimize/_zeros_py.py +97 -17
  318. scipy/optimize/cython_optimize/_zeros.cpython-312-aarch64-linux-musl.so +0 -0
  319. scipy/optimize/slsqp.py +0 -1
  320. scipy/optimize/tests/test__basinhopping.py +1 -1
  321. scipy/optimize/tests/test__differential_evolution.py +4 -4
  322. scipy/optimize/tests/test__linprog_clean_inputs.py +5 -3
  323. scipy/optimize/tests/test__numdiff.py +66 -22
  324. scipy/optimize/tests/test__remove_redundancy.py +2 -2
  325. scipy/optimize/tests/test__shgo.py +9 -1
  326. scipy/optimize/tests/test_bracket.py +36 -46
  327. scipy/optimize/tests/test_chandrupatla.py +133 -135
  328. scipy/optimize/tests/test_cobyla.py +74 -45
  329. scipy/optimize/tests/test_constraints.py +1 -1
  330. scipy/optimize/tests/test_differentiable_functions.py +226 -6
  331. scipy/optimize/tests/test_lbfgsb_hessinv.py +22 -0
  332. scipy/optimize/tests/test_least_squares.py +125 -13
  333. scipy/optimize/tests/test_linear_assignment.py +3 -3
  334. scipy/optimize/tests/test_linprog.py +3 -3
  335. scipy/optimize/tests/test_lsq_linear.py +6 -6
  336. scipy/optimize/tests/test_minimize_constrained.py +2 -2
  337. scipy/optimize/tests/test_minpack.py +4 -4
  338. scipy/optimize/tests/test_nnls.py +43 -3
  339. scipy/optimize/tests/test_nonlin.py +36 -0
  340. scipy/optimize/tests/test_optimize.py +95 -17
  341. scipy/optimize/tests/test_slsqp.py +36 -4
  342. scipy/optimize/tests/test_zeros.py +34 -1
  343. scipy/signal/__init__.py +12 -23
  344. scipy/signal/_delegators.py +568 -0
  345. scipy/signal/_filter_design.py +459 -241
  346. scipy/signal/_fir_filter_design.py +262 -90
  347. scipy/signal/_lti_conversion.py +3 -2
  348. scipy/signal/_ltisys.py +118 -91
  349. scipy/signal/_max_len_seq_inner.cpython-312-aarch64-linux-musl.so +0 -0
  350. scipy/signal/_peak_finding_utils.cpython-312-aarch64-linux-musl.so +0 -0
  351. scipy/signal/_polyutils.py +172 -0
  352. scipy/signal/_short_time_fft.py +519 -70
  353. scipy/signal/_signal_api.py +30 -0
  354. scipy/signal/_signaltools.py +719 -399
  355. scipy/signal/_sigtools.cpython-312-aarch64-linux-musl.so +0 -0
  356. scipy/signal/_sosfilt.cpython-312-aarch64-linux-musl.so +0 -0
  357. scipy/signal/_spectral_py.py +230 -50
  358. scipy/signal/_spline.cpython-312-aarch64-linux-musl.so +0 -0
  359. scipy/signal/_spline_filters.py +108 -68
  360. scipy/signal/_support_alternative_backends.py +73 -0
  361. scipy/signal/_upfirdn.py +4 -1
  362. scipy/signal/_upfirdn_apply.cpython-312-aarch64-linux-musl.so +0 -0
  363. scipy/signal/_waveforms.py +2 -11
  364. scipy/signal/_wavelets.py +1 -1
  365. scipy/signal/fir_filter_design.py +1 -0
  366. scipy/signal/spline.py +4 -11
  367. scipy/signal/tests/_scipy_spectral_test_shim.py +2 -171
  368. scipy/signal/tests/test_bsplines.py +114 -79
  369. scipy/signal/tests/test_cont2discrete.py +9 -2
  370. scipy/signal/tests/test_filter_design.py +721 -481
  371. scipy/signal/tests/test_fir_filter_design.py +332 -140
  372. scipy/signal/tests/test_savitzky_golay.py +4 -3
  373. scipy/signal/tests/test_short_time_fft.py +221 -3
  374. scipy/signal/tests/test_signaltools.py +2144 -1348
  375. scipy/signal/tests/test_spectral.py +50 -6
  376. scipy/signal/tests/test_splines.py +161 -96
  377. scipy/signal/tests/test_upfirdn.py +84 -50
  378. scipy/signal/tests/test_waveforms.py +20 -0
  379. scipy/signal/tests/test_windows.py +607 -466
  380. scipy/signal/windows/_windows.py +287 -148
  381. scipy/sparse/__init__.py +23 -4
  382. scipy/sparse/_base.py +270 -108
  383. scipy/sparse/_bsr.py +7 -4
  384. scipy/sparse/_compressed.py +59 -231
  385. scipy/sparse/_construct.py +90 -38
  386. scipy/sparse/_coo.py +115 -181
  387. scipy/sparse/_csc.py +4 -4
  388. scipy/sparse/_csparsetools.cpython-312-aarch64-linux-musl.so +0 -0
  389. scipy/sparse/_csr.py +2 -2
  390. scipy/sparse/_data.py +48 -48
  391. scipy/sparse/_dia.py +105 -18
  392. scipy/sparse/_dok.py +0 -23
  393. scipy/sparse/_index.py +4 -4
  394. scipy/sparse/_matrix.py +23 -0
  395. scipy/sparse/_sparsetools.cpython-312-aarch64-linux-musl.so +0 -0
  396. scipy/sparse/_sputils.py +37 -22
  397. scipy/sparse/base.py +0 -9
  398. scipy/sparse/bsr.py +0 -14
  399. scipy/sparse/compressed.py +0 -23
  400. scipy/sparse/construct.py +0 -6
  401. scipy/sparse/coo.py +0 -14
  402. scipy/sparse/csc.py +0 -3
  403. scipy/sparse/csgraph/_flow.cpython-312-aarch64-linux-musl.so +0 -0
  404. scipy/sparse/csgraph/_matching.cpython-312-aarch64-linux-musl.so +0 -0
  405. scipy/sparse/csgraph/_min_spanning_tree.cpython-312-aarch64-linux-musl.so +0 -0
  406. scipy/sparse/csgraph/_reordering.cpython-312-aarch64-linux-musl.so +0 -0
  407. scipy/sparse/csgraph/_shortest_path.cpython-312-aarch64-linux-musl.so +0 -0
  408. scipy/sparse/csgraph/_tools.cpython-312-aarch64-linux-musl.so +0 -0
  409. scipy/sparse/csgraph/_traversal.cpython-312-aarch64-linux-musl.so +0 -0
  410. scipy/sparse/csgraph/tests/test_matching.py +14 -2
  411. scipy/sparse/csgraph/tests/test_pydata_sparse.py +4 -1
  412. scipy/sparse/csgraph/tests/test_shortest_path.py +83 -27
  413. scipy/sparse/csr.py +0 -5
  414. scipy/sparse/data.py +1 -6
  415. scipy/sparse/dia.py +0 -7
  416. scipy/sparse/dok.py +0 -10
  417. scipy/sparse/linalg/_dsolve/_superlu.cpython-312-aarch64-linux-musl.so +0 -0
  418. scipy/sparse/linalg/_dsolve/linsolve.py +9 -0
  419. scipy/sparse/linalg/_dsolve/tests/test_linsolve.py +35 -28
  420. scipy/sparse/linalg/_eigen/arpack/_arpack.cpython-312-aarch64-linux-musl.so +0 -0
  421. scipy/sparse/linalg/_eigen/arpack/arpack.py +23 -17
  422. scipy/sparse/linalg/_eigen/lobpcg/lobpcg.py +6 -6
  423. scipy/sparse/linalg/_interface.py +17 -18
  424. scipy/sparse/linalg/_isolve/_gcrotmk.py +4 -4
  425. scipy/sparse/linalg/_isolve/iterative.py +51 -45
  426. scipy/sparse/linalg/_isolve/lgmres.py +6 -6
  427. scipy/sparse/linalg/_isolve/minres.py +5 -5
  428. scipy/sparse/linalg/_isolve/tfqmr.py +7 -7
  429. scipy/sparse/linalg/_isolve/utils.py +2 -8
  430. scipy/sparse/linalg/_matfuncs.py +1 -1
  431. scipy/sparse/linalg/_norm.py +1 -1
  432. scipy/sparse/linalg/_propack/_cpropack.cpython-312-aarch64-linux-musl.so +0 -0
  433. scipy/sparse/linalg/_propack/_dpropack.cpython-312-aarch64-linux-musl.so +0 -0
  434. scipy/sparse/linalg/_propack/_spropack.cpython-312-aarch64-linux-musl.so +0 -0
  435. scipy/sparse/linalg/_propack/_zpropack.cpython-312-aarch64-linux-musl.so +0 -0
  436. scipy/sparse/linalg/_special_sparse_arrays.py +39 -38
  437. scipy/sparse/linalg/tests/test_pydata_sparse.py +14 -0
  438. scipy/sparse/tests/test_arithmetic1d.py +5 -2
  439. scipy/sparse/tests/test_base.py +214 -42
  440. scipy/sparse/tests/test_common1d.py +7 -7
  441. scipy/sparse/tests/test_construct.py +1 -1
  442. scipy/sparse/tests/test_coo.py +272 -4
  443. scipy/sparse/tests/test_sparsetools.py +5 -0
  444. scipy/sparse/tests/test_sputils.py +36 -7
  445. scipy/spatial/_ckdtree.cpython-312-aarch64-linux-musl.so +0 -0
  446. scipy/spatial/_distance_pybind.cpython-312-aarch64-linux-musl.so +0 -0
  447. scipy/spatial/_distance_wrap.cpython-312-aarch64-linux-musl.so +0 -0
  448. scipy/spatial/_hausdorff.cpython-312-aarch64-linux-musl.so +0 -0
  449. scipy/spatial/_qhull.cpython-312-aarch64-linux-musl.so +0 -0
  450. scipy/spatial/_voronoi.cpython-312-aarch64-linux-musl.so +0 -0
  451. scipy/spatial/distance.py +49 -42
  452. scipy/spatial/tests/test_distance.py +15 -1
  453. scipy/spatial/tests/test_kdtree.py +1 -0
  454. scipy/spatial/tests/test_qhull.py +7 -2
  455. scipy/spatial/transform/__init__.py +5 -3
  456. scipy/spatial/transform/_rigid_transform.cpython-312-aarch64-linux-musl.so +0 -0
  457. scipy/spatial/transform/_rotation.cpython-312-aarch64-linux-musl.so +0 -0
  458. scipy/spatial/transform/tests/test_rigid_transform.py +1221 -0
  459. scipy/spatial/transform/tests/test_rotation.py +1213 -832
  460. scipy/spatial/transform/tests/test_rotation_groups.py +3 -3
  461. scipy/spatial/transform/tests/test_rotation_spline.py +29 -8
  462. scipy/special/__init__.py +1 -47
  463. scipy/special/_add_newdocs.py +34 -772
  464. scipy/special/_basic.py +22 -25
  465. scipy/special/_comb.cpython-312-aarch64-linux-musl.so +0 -0
  466. scipy/special/_ellip_harm_2.cpython-312-aarch64-linux-musl.so +0 -0
  467. scipy/special/_gufuncs.cpython-312-aarch64-linux-musl.so +0 -0
  468. scipy/special/_logsumexp.py +67 -58
  469. scipy/special/_orthogonal.pyi +1 -1
  470. scipy/special/_specfun.cpython-312-aarch64-linux-musl.so +0 -0
  471. scipy/special/_special_ufuncs.cpython-312-aarch64-linux-musl.so +0 -0
  472. scipy/special/_spherical_bessel.py +4 -4
  473. scipy/special/_support_alternative_backends.py +212 -119
  474. scipy/special/_test_internal.cpython-312-aarch64-linux-musl.so +0 -0
  475. scipy/special/_testutils.py +4 -4
  476. scipy/special/_ufuncs.cpython-312-aarch64-linux-musl.so +0 -0
  477. scipy/special/_ufuncs.pyi +1 -0
  478. scipy/special/_ufuncs.pyx +215 -1400
  479. scipy/special/_ufuncs_cxx.cpython-312-aarch64-linux-musl.so +0 -0
  480. scipy/special/_ufuncs_cxx.pxd +2 -15
  481. scipy/special/_ufuncs_cxx.pyx +5 -44
  482. scipy/special/_ufuncs_cxx_defs.h +2 -16
  483. scipy/special/_ufuncs_defs.h +0 -8
  484. scipy/special/cython_special.cpython-312-aarch64-linux-musl.so +0 -0
  485. scipy/special/cython_special.pxd +1 -1
  486. scipy/special/tests/_cython_examples/meson.build +10 -1
  487. scipy/special/tests/test_basic.py +153 -20
  488. scipy/special/tests/test_boost_ufuncs.py +3 -0
  489. scipy/special/tests/test_cdflib.py +35 -11
  490. scipy/special/tests/test_gammainc.py +16 -0
  491. scipy/special/tests/test_hyp2f1.py +2 -2
  492. scipy/special/tests/test_log1mexp.py +85 -0
  493. scipy/special/tests/test_logsumexp.py +206 -64
  494. scipy/special/tests/test_mpmath.py +1 -0
  495. scipy/special/tests/test_nan_inputs.py +1 -1
  496. scipy/special/tests/test_orthogonal.py +17 -18
  497. scipy/special/tests/test_sf_error.py +3 -2
  498. scipy/special/tests/test_sph_harm.py +6 -7
  499. scipy/special/tests/test_support_alternative_backends.py +211 -76
  500. scipy/stats/__init__.py +4 -1
  501. scipy/stats/_ansari_swilk_statistics.cpython-312-aarch64-linux-musl.so +0 -0
  502. scipy/stats/_axis_nan_policy.py +5 -12
  503. scipy/stats/_biasedurn.cpython-312-aarch64-linux-musl.so +0 -0
  504. scipy/stats/_continued_fraction.py +387 -0
  505. scipy/stats/_continuous_distns.py +277 -310
  506. scipy/stats/_correlation.py +1 -1
  507. scipy/stats/_covariance.py +6 -3
  508. scipy/stats/_discrete_distns.py +39 -32
  509. scipy/stats/_distn_infrastructure.py +39 -12
  510. scipy/stats/_distribution_infrastructure.py +900 -238
  511. scipy/stats/_entropy.py +9 -10
  512. scipy/{_lib → stats}/_finite_differences.py +1 -1
  513. scipy/stats/_hypotests.py +83 -50
  514. scipy/stats/_kde.py +53 -49
  515. scipy/stats/_ksstats.py +1 -1
  516. scipy/stats/_levy_stable/__init__.py +7 -15
  517. scipy/stats/_levy_stable/levyst.cpython-312-aarch64-linux-musl.so +0 -0
  518. scipy/stats/_morestats.py +118 -73
  519. scipy/stats/_mstats_basic.py +13 -17
  520. scipy/stats/_mstats_extras.py +8 -8
  521. scipy/stats/_multivariate.py +89 -113
  522. scipy/stats/_new_distributions.py +97 -20
  523. scipy/stats/_page_trend_test.py +12 -5
  524. scipy/stats/_probability_distribution.py +265 -43
  525. scipy/stats/_qmc.py +14 -9
  526. scipy/stats/_qmc_cy.cpython-312-aarch64-linux-musl.so +0 -0
  527. scipy/stats/_qmvnt.py +16 -95
  528. scipy/stats/_qmvnt_cy.cpython-312-aarch64-linux-musl.so +0 -0
  529. scipy/stats/_quantile.py +335 -0
  530. scipy/stats/_rcont/rcont.cpython-312-aarch64-linux-musl.so +0 -0
  531. scipy/stats/_resampling.py +4 -29
  532. scipy/stats/_sampling.py +1 -1
  533. scipy/stats/_sobol.cpython-312-aarch64-linux-musl.so +0 -0
  534. scipy/stats/_stats.cpython-312-aarch64-linux-musl.so +0 -0
  535. scipy/stats/_stats_mstats_common.py +21 -2
  536. scipy/stats/_stats_py.py +550 -476
  537. scipy/stats/_stats_pythran.cpython-312-aarch64-linux-musl.so +0 -0
  538. scipy/stats/_unuran/unuran_wrapper.cpython-312-aarch64-linux-musl.so +0 -0
  539. scipy/stats/_unuran/unuran_wrapper.pyi +2 -1
  540. scipy/stats/_variation.py +6 -8
  541. scipy/stats/_wilcoxon.py +13 -7
  542. scipy/stats/tests/common_tests.py +6 -4
  543. scipy/stats/tests/test_axis_nan_policy.py +62 -24
  544. scipy/stats/tests/test_continued_fraction.py +173 -0
  545. scipy/stats/tests/test_continuous.py +379 -60
  546. scipy/stats/tests/test_continuous_basic.py +18 -12
  547. scipy/stats/tests/test_discrete_basic.py +14 -8
  548. scipy/stats/tests/test_discrete_distns.py +16 -16
  549. scipy/stats/tests/test_distributions.py +95 -75
  550. scipy/stats/tests/test_entropy.py +40 -48
  551. scipy/stats/tests/test_fit.py +4 -3
  552. scipy/stats/tests/test_hypotests.py +153 -24
  553. scipy/stats/tests/test_kdeoth.py +109 -41
  554. scipy/stats/tests/test_marray.py +289 -0
  555. scipy/stats/tests/test_morestats.py +79 -47
  556. scipy/stats/tests/test_mstats_basic.py +3 -3
  557. scipy/stats/tests/test_multivariate.py +434 -83
  558. scipy/stats/tests/test_qmc.py +13 -10
  559. scipy/stats/tests/test_quantile.py +199 -0
  560. scipy/stats/tests/test_rank.py +119 -112
  561. scipy/stats/tests/test_resampling.py +47 -56
  562. scipy/stats/tests/test_sampling.py +9 -4
  563. scipy/stats/tests/test_stats.py +799 -939
  564. scipy/stats/tests/test_variation.py +8 -6
  565. scipy/version.py +2 -2
  566. {scipy-1.15.3.dist-info → scipy-1.16.0rc2.dist-info}/LICENSE.txt +4 -4
  567. {scipy-1.15.3.dist-info → scipy-1.16.0rc2.dist-info}/METADATA +11 -11
  568. {scipy-1.15.3.dist-info → scipy-1.16.0rc2.dist-info}/RECORD +1262 -1269
  569. scipy.libs/libgcc_s-69c45f16.so.1 +0 -0
  570. scipy.libs/libgfortran-db0b6589.so.5.0.0 +0 -0
  571. scipy.libs/{libstdc++-1b614e01.so.6.0.32 → libstdc++-1f1a71be.so.6.0.33} +0 -0
  572. scipy/_lib/array_api_extra/_funcs.py +0 -484
  573. scipy/_lib/array_api_extra/_typing.py +0 -8
  574. scipy/interpolate/_bspl.cpython-312-aarch64-linux-musl.so +0 -0
  575. scipy/optimize/_cobyla.cpython-312-aarch64-linux-musl.so +0 -0
  576. scipy/optimize/_cython_nnls.cpython-312-aarch64-linux-musl.so +0 -0
  577. scipy/optimize/_slsqp.cpython-312-aarch64-linux-musl.so +0 -0
  578. scipy/spatial/qhull_src/COPYING.txt +0 -38
  579. scipy/special/libsf_error_state.so +0 -0
  580. scipy/special/tests/test_log_softmax.py +0 -109
  581. scipy/special/tests/test_xsf_cuda.py +0 -114
  582. scipy/special/xsf/binom.h +0 -89
  583. scipy/special/xsf/cdflib.h +0 -100
  584. scipy/special/xsf/cephes/airy.h +0 -307
  585. scipy/special/xsf/cephes/besselpoly.h +0 -51
  586. scipy/special/xsf/cephes/beta.h +0 -257
  587. scipy/special/xsf/cephes/cbrt.h +0 -131
  588. scipy/special/xsf/cephes/chbevl.h +0 -85
  589. scipy/special/xsf/cephes/chdtr.h +0 -193
  590. scipy/special/xsf/cephes/const.h +0 -87
  591. scipy/special/xsf/cephes/ellie.h +0 -293
  592. scipy/special/xsf/cephes/ellik.h +0 -251
  593. scipy/special/xsf/cephes/ellpe.h +0 -107
  594. scipy/special/xsf/cephes/ellpk.h +0 -117
  595. scipy/special/xsf/cephes/expn.h +0 -260
  596. scipy/special/xsf/cephes/gamma.h +0 -398
  597. scipy/special/xsf/cephes/hyp2f1.h +0 -596
  598. scipy/special/xsf/cephes/hyperg.h +0 -361
  599. scipy/special/xsf/cephes/i0.h +0 -149
  600. scipy/special/xsf/cephes/i1.h +0 -158
  601. scipy/special/xsf/cephes/igam.h +0 -421
  602. scipy/special/xsf/cephes/igam_asymp_coeff.h +0 -195
  603. scipy/special/xsf/cephes/igami.h +0 -313
  604. scipy/special/xsf/cephes/j0.h +0 -225
  605. scipy/special/xsf/cephes/j1.h +0 -198
  606. scipy/special/xsf/cephes/jv.h +0 -715
  607. scipy/special/xsf/cephes/k0.h +0 -164
  608. scipy/special/xsf/cephes/k1.h +0 -163
  609. scipy/special/xsf/cephes/kn.h +0 -243
  610. scipy/special/xsf/cephes/lanczos.h +0 -112
  611. scipy/special/xsf/cephes/ndtr.h +0 -275
  612. scipy/special/xsf/cephes/poch.h +0 -85
  613. scipy/special/xsf/cephes/polevl.h +0 -167
  614. scipy/special/xsf/cephes/psi.h +0 -194
  615. scipy/special/xsf/cephes/rgamma.h +0 -111
  616. scipy/special/xsf/cephes/scipy_iv.h +0 -811
  617. scipy/special/xsf/cephes/shichi.h +0 -248
  618. scipy/special/xsf/cephes/sici.h +0 -224
  619. scipy/special/xsf/cephes/sindg.h +0 -221
  620. scipy/special/xsf/cephes/tandg.h +0 -139
  621. scipy/special/xsf/cephes/trig.h +0 -58
  622. scipy/special/xsf/cephes/unity.h +0 -186
  623. scipy/special/xsf/cephes/zeta.h +0 -172
  624. scipy/special/xsf/config.h +0 -304
  625. scipy/special/xsf/digamma.h +0 -205
  626. scipy/special/xsf/error.h +0 -57
  627. scipy/special/xsf/evalpoly.h +0 -47
  628. scipy/special/xsf/expint.h +0 -266
  629. scipy/special/xsf/hyp2f1.h +0 -694
  630. scipy/special/xsf/iv_ratio.h +0 -173
  631. scipy/special/xsf/lambertw.h +0 -150
  632. scipy/special/xsf/loggamma.h +0 -163
  633. scipy/special/xsf/sici.h +0 -200
  634. scipy/special/xsf/tools.h +0 -427
  635. scipy/special/xsf/trig.h +0 -164
  636. scipy/special/xsf/wright_bessel.h +0 -843
  637. scipy/special/xsf/zlog1.h +0 -35
  638. scipy/stats/_mvn.cpython-312-aarch64-linux-musl.so +0 -0
  639. scipy.libs/libgcc_s-7393e603.so.1 +0 -0
  640. scipy.libs/libgfortran-eb933d8e.so.5.0.0 +0 -0
  641. {scipy-1.15.3.dist-info → scipy-1.16.0rc2.dist-info}/WHEEL +0 -0
@@ -1,173 +0,0 @@
1
- // Numerically stable computation of iv(v+1, x) / iv(v, x)
2
-
3
- #pragma once
4
-
5
- #include "config.h"
6
- #include "tools.h"
7
- #include "error.h"
8
- #include "cephes/dd_real.h"
9
-
10
- namespace xsf {
11
-
12
- /* Generates the "tail" of Perron's continued fraction for iv(v,x)/iv(v-1,x).
13
- *
14
- * The Perron continued fraction is studied in [1]. It is given by
15
- *
16
- * iv(v, x) x -(2v+1)x -(2v+3)x -(2v+5)x
17
- * R := --------- = ------ ---------- ---------- ---------- ...
18
- * iv(v-1,x) x+2v + 2(v+x)+1 + 2(v+x)+2 + 2(v+x)+3 +
19
- *
20
- * Given a suitable constant c, the continued fraction may be rearranged
21
- * into the following form to avoid premature floating point overflow:
22
- *
23
- * xc -(2vc+c)(xc) -(2vc+3c)(xc) -(2vc+5c)(xc)
24
- * R = -----, fc = 2vc + ------------ ------------- ------------- ...
25
- * xc+fc 2(vc+xc)+c + 2(vc+xc)+2c + 2(vc+xc)+3c +
26
- *
27
- * This class generates the fractions of fc after 2vc.
28
- *
29
- * [1] Gautschi, W. and Slavik, J. (1978). "On the computation of modified
30
- * Bessel function ratios." Mathematics of Computation, 32(143):865-875.
31
- */
32
- template <class T>
33
- struct IvRatioCFTailGenerator {
34
-
35
- XSF_HOST_DEVICE IvRatioCFTailGenerator(T vc, T xc, T c) noexcept {
36
- a0_ = -(2*vc-c)*xc;
37
- as_ = -2*c*xc;
38
- b0_ = 2*(vc+xc);
39
- bs_ = c;
40
- k_ = 0;
41
- }
42
-
43
- XSF_HOST_DEVICE std::pair<T, T> operator()() noexcept {
44
- using std::fma;
45
- ++k_;
46
- return {fma(static_cast<T>(k_), as_, a0_),
47
- fma(static_cast<T>(k_), bs_, b0_)};
48
- }
49
-
50
- private:
51
- T a0_, as_; // a[k] == a0 + as*k, k >= 1
52
- T b0_, bs_; // b[k] == b0 + bs*k, k >= 1
53
- std::uint64_t k_; // current index
54
- };
55
-
56
- // Computes f(v, x) using Perron's continued fraction.
57
- //
58
- // T specifies the working type. This allows the function to perform
59
- // calculations in a higher precision, such as double-double, even if
60
- // the return type is hardcoded to be double.
61
- template <class T>
62
- XSF_HOST_DEVICE inline std::pair<double, std::uint64_t>
63
- _iv_ratio_cf(double v, double x, bool complement) {
64
-
65
- int e;
66
- std::frexp(std::fmax(v, x), &e);
67
- T c = T(std::ldexp(1, 2-e)); // rescaling multiplier
68
- T vc = v * c;
69
- T xc = x * c;
70
-
71
- IvRatioCFTailGenerator<T> cf(vc, xc, c);
72
- auto [fc, terms] = detail::series_eval_kahan(
73
- detail::continued_fraction_series(cf),
74
- T(std::numeric_limits<double>::epsilon()),
75
- 1000,
76
- 2*vc);
77
-
78
- T ret = (complement ? fc : xc) / (xc + fc);
79
- return {static_cast<double>(ret), terms};
80
- }
81
-
82
- XSF_HOST_DEVICE inline double iv_ratio(double v, double x) {
83
-
84
- if (std::isnan(v) || std::isnan(x)) {
85
- return std::numeric_limits<double>::quiet_NaN();
86
- }
87
- if (v < 0.5 || x < 0) {
88
- set_error("iv_ratio", SF_ERROR_DOMAIN, NULL);
89
- return std::numeric_limits<double>::quiet_NaN();
90
- }
91
- if (std::isinf(v) && std::isinf(x)) {
92
- // There is not a unique limit as both v and x tends to infinity.
93
- set_error("iv_ratio", SF_ERROR_DOMAIN, NULL);
94
- return std::numeric_limits<double>::quiet_NaN();
95
- }
96
- if (x == 0.0) {
97
- return x; // keep sign of x because iv_ratio is an odd function
98
- }
99
- if (std::isinf(v)) {
100
- return 0.0;
101
- }
102
- if (std::isinf(x)) {
103
- return 1.0;
104
- }
105
-
106
- auto [ret, terms] = _iv_ratio_cf<double>(v, x, false);
107
- if (terms == 0) { // failed to converge; should not happen
108
- set_error("iv_ratio", SF_ERROR_NO_RESULT, NULL);
109
- return std::numeric_limits<double>::quiet_NaN();
110
- }
111
- return ret;
112
- }
113
-
114
- XSF_HOST_DEVICE inline float iv_ratio(float v, float x) {
115
- return iv_ratio(static_cast<double>(v), static_cast<double>(x));
116
- }
117
-
118
- XSF_HOST_DEVICE inline double iv_ratio_c(double v, double x) {
119
-
120
- if (std::isnan(v) || std::isnan(x)) {
121
- return std::numeric_limits<double>::quiet_NaN();
122
- }
123
- if (v < 0.5 || x < 0) {
124
- set_error("iv_ratio_c", SF_ERROR_DOMAIN, NULL);
125
- return std::numeric_limits<double>::quiet_NaN();
126
- }
127
- if (std::isinf(v) && std::isinf(x)) {
128
- // There is not a unique limit as both v and x tends to infinity.
129
- set_error("iv_ratio_c", SF_ERROR_DOMAIN, NULL);
130
- return std::numeric_limits<double>::quiet_NaN();
131
- }
132
- if (x == 0.0) {
133
- return 1.0;
134
- }
135
- if (std::isinf(v)) {
136
- return 1.0;
137
- }
138
- if (std::isinf(x)) {
139
- return 0.0;
140
- }
141
-
142
- if (v >= 1) {
143
- // Numerical experiments show that evaluating the Perron c.f.
144
- // in double precision is sufficiently accurate if v >= 1.
145
- auto [ret, terms] = _iv_ratio_cf<double>(v, x, true);
146
- if (terms == 0) { // failed to converge; should not happen
147
- set_error("iv_ratio_c", SF_ERROR_NO_RESULT, NULL);
148
- return std::numeric_limits<double>::quiet_NaN();
149
- }
150
- return ret;
151
- } else if (v > 0.5) {
152
- // double-double arithmetic is needed for 0.5 < v < 1 to
153
- // achieve relative error on the scale of machine precision.
154
- using cephes::detail::double_double;
155
- auto [ret, terms] = _iv_ratio_cf<double_double>(v, x, true);
156
- if (terms == 0) { // failed to converge; should not happen
157
- set_error("iv_ratio_c", SF_ERROR_NO_RESULT, NULL);
158
- return std::numeric_limits<double>::quiet_NaN();
159
- }
160
- return ret;
161
- } else {
162
- // The previous branch (v > 0.5) also works for v == 0.5, but
163
- // the closed-form formula "1 - tanh(x)" is more efficient.
164
- double t = std::exp(-2*x);
165
- return (2 * t) / (1 + t);
166
- }
167
- }
168
-
169
- XSF_HOST_DEVICE inline float iv_ratio_c(float v, float x) {
170
- return iv_ratio_c(static_cast<double>(v), static_cast<double>(x));
171
- }
172
-
173
- } // namespace xsf
@@ -1,150 +0,0 @@
1
- /* Translated from Cython into C++ by SciPy developers in 2023.
2
- * Original header with Copyright information appears below.
3
- */
4
-
5
- /* Implementation of the Lambert W function [1]. Based on MPMath
6
- * Implementation [2], and documentation [3].
7
- *
8
- * Copyright: Yosef Meller, 2009
9
- * Author email: mellerf@netvision.net.il
10
- *
11
- * Distributed under the same license as SciPy
12
- *
13
- *
14
- * References:
15
- * [1] On the Lambert W function, Adv. Comp. Math. 5 (1996) 329-359,
16
- * available online: https://web.archive.org/web/20230123211413/https://cs.uwaterloo.ca/research/tr/1993/03/W.pdf
17
- * [2] mpmath source code,
18
- https://github.com/mpmath/mpmath/blob/c5939823669e1bcce151d89261b802fe0d8978b4/mpmath/functions/functions.py#L435-L461
19
- * [3]
20
- https://web.archive.org/web/20230504171447/https://mpmath.org/doc/current/functions/powers.html#lambert-w-function
21
- *
22
-
23
- * TODO: use a series expansion when extremely close to the branch point
24
- * at `-1/e` and make sure that the proper branch is chosen there.
25
- */
26
-
27
- #pragma once
28
-
29
- #include "config.h"
30
- #include "error.h"
31
- #include "evalpoly.h"
32
-
33
- namespace xsf {
34
- constexpr double EXPN1 = 0.36787944117144232159553; // exp(-1)
35
- constexpr double OMEGA = 0.56714329040978387299997; // W(1, 0)
36
-
37
- namespace detail {
38
- XSF_HOST_DEVICE inline std::complex<double> lambertw_branchpt(std::complex<double> z) {
39
- // Series for W(z, 0) around the branch point; see 4.22 in [1].
40
- double coeffs[] = {-1.0 / 3.0, 1.0, -1.0};
41
- std::complex<double> p = std::sqrt(2.0 * (M_E * z + 1.0));
42
-
43
- return cevalpoly(coeffs, 2, p);
44
- }
45
-
46
- XSF_HOST_DEVICE inline std::complex<double> lambertw_pade0(std::complex<double> z) {
47
- // (3, 2) Pade approximation for W(z, 0) around 0.
48
- double num[] = {12.85106382978723404255, 12.34042553191489361902, 1.0};
49
- double denom[] = {32.53191489361702127660, 14.34042553191489361702, 1.0};
50
-
51
- /* This only gets evaluated close to 0, so we don't need a more
52
- * careful algorithm that avoids overflow in the numerator for
53
- * large z. */
54
- return z * cevalpoly(num, 2, z) / cevalpoly(denom, 2, z);
55
- }
56
-
57
- XSF_HOST_DEVICE inline std::complex<double> lambertw_asy(std::complex<double> z, long k) {
58
- /* Compute the W function using the first two terms of the
59
- * asymptotic series. See 4.20 in [1].
60
- */
61
- std::complex<double> w = std::log(z) + 2.0 * M_PI * k * std::complex<double>(0, 1);
62
- return w - std::log(w);
63
- }
64
-
65
- } // namespace detail
66
-
67
- XSF_HOST_DEVICE inline std::complex<double> lambertw(std::complex<double> z, long k, double tol) {
68
- double absz;
69
- std::complex<double> w;
70
- std::complex<double> ew, wew, wewz, wn;
71
-
72
- if (std::isnan(z.real()) || std::isnan(z.imag())) {
73
- return z;
74
- }
75
- if (z.real() == std::numeric_limits<double>::infinity()) {
76
- return z + 2.0 * M_PI * k * std::complex<double>(0, 1);
77
- }
78
- if (z.real() == -std::numeric_limits<double>::infinity()) {
79
- return -z + (2.0 * M_PI * k + M_PI) * std::complex<double>(0, 1);
80
- }
81
- if (z == 0.0) {
82
- if (k == 0) {
83
- return z;
84
- }
85
- set_error("lambertw", SF_ERROR_SINGULAR, NULL);
86
- return -std::numeric_limits<double>::infinity();
87
- }
88
- if (z == 1.0 && k == 0) {
89
- // Split out this case because the asymptotic series blows up
90
- return OMEGA;
91
- }
92
-
93
- absz = std::abs(z);
94
- // Get an initial guess for Halley's method
95
- if (k == 0) {
96
- if (std::abs(z + EXPN1) < 0.3) {
97
- w = detail::lambertw_branchpt(z);
98
- } else if (-1.0 < z.real() && z.real() < 1.5 && std::abs(z.imag()) < 1.0 &&
99
- -2.5 * std::abs(z.imag()) - 0.2 < z.real()) {
100
- /* Empirically determined decision boundary where the Pade
101
- * approximation is more accurate. */
102
- w = detail::lambertw_pade0(z);
103
- } else {
104
- w = detail::lambertw_asy(z, k);
105
- }
106
- } else if (k == -1) {
107
- if (absz <= EXPN1 && z.imag() == 0.0 && z.real() < 0.0) {
108
- w = std::log(-z.real());
109
- } else {
110
- w = detail::lambertw_asy(z, k);
111
- }
112
- } else {
113
- w = detail::lambertw_asy(z, k);
114
- }
115
-
116
- // Halley's method; see 5.9 in [1]
117
- if (w.real() >= 0) {
118
- // Rearrange the formula to avoid overflow in exp
119
- for (int i = 0; i < 100; i++) {
120
- ew = std::exp(-w);
121
- wewz = w - z * ew;
122
- wn = w - wewz / (w + 1.0 - (w + 2.0) * wewz / (2.0 * w + 2.0));
123
- if (std::abs(wn - w) <= tol * std::abs(wn)) {
124
- return wn;
125
- }
126
- w = wn;
127
- }
128
- } else {
129
- for (int i = 0; i < 100; i++) {
130
- ew = std::exp(w);
131
- wew = w * ew;
132
- wewz = wew - z;
133
- wn = w - wewz / (wew + ew - (w + 2.0) * wewz / (2.0 * w + 2.0));
134
- if (std::abs(wn - w) <= tol * std::abs(wn)) {
135
- return wn;
136
- }
137
- w = wn;
138
- }
139
- }
140
-
141
- set_error("lambertw", SF_ERROR_SLOW, "iteration failed to converge: %g + %gj", z.real(), z.imag());
142
- return {std::numeric_limits<double>::quiet_NaN(), std::numeric_limits<double>::quiet_NaN()};
143
- }
144
-
145
- XSF_HOST_DEVICE inline std::complex<float> lambertw(std::complex<float> z, long k, float tol) {
146
- return static_cast<std::complex<float>>(
147
- lambertw(static_cast<std::complex<double>>(z), k, static_cast<double>(tol)));
148
- }
149
-
150
- } // namespace xsf
@@ -1,163 +0,0 @@
1
- /* Translated from Cython into C++ by SciPy developers in 2024.
2
- * Original header comment appears below.
3
- */
4
-
5
- /* An implementation of the principal branch of the logarithm of
6
- * Gamma. Also contains implementations of Gamma and 1/Gamma which are
7
- * easily computed from log-Gamma.
8
- *
9
- * Author: Josh Wilson
10
- *
11
- * Distributed under the same license as Scipy.
12
- *
13
- * References
14
- * ----------
15
- * [1] Hare, "Computing the Principal Branch of log-Gamma",
16
- * Journal of Algorithms, 1997.
17
- *
18
- * [2] Julia,
19
- * https://github.com/JuliaLang/julia/blob/master/base/special/gamma.jl
20
- */
21
-
22
- #pragma once
23
-
24
- #include "cephes/gamma.h"
25
- #include "cephes/rgamma.h"
26
- #include "config.h"
27
- #include "error.h"
28
- #include "evalpoly.h"
29
- #include "trig.h"
30
- #include "zlog1.h"
31
-
32
- namespace xsf {
33
-
34
- namespace detail {
35
- constexpr double loggamma_SMALLX = 7;
36
- constexpr double loggamma_SMALLY = 7;
37
- constexpr double loggamma_HLOG2PI = 0.918938533204672742; // log(2*pi)/2
38
- constexpr double loggamma_LOGPI = 1.1447298858494001741434262; // log(pi)
39
- constexpr double loggamma_TAYLOR_RADIUS = 0.2;
40
-
41
- XSF_HOST_DEVICE std::complex<double> loggamma_stirling(std::complex<double> z) {
42
- /* Stirling series for log-Gamma
43
- *
44
- * The coefficients are B[2*n]/(2*n*(2*n - 1)) where B[2*n] is the
45
- * (2*n)th Bernoulli number. See (1.1) in [1].
46
- */
47
- double coeffs[] = {-2.955065359477124183E-2, 6.4102564102564102564E-3, -1.9175269175269175269E-3,
48
- 8.4175084175084175084E-4, -5.952380952380952381E-4, 7.9365079365079365079E-4,
49
- -2.7777777777777777778E-3, 8.3333333333333333333E-2};
50
- std::complex<double> rz = 1.0 / z;
51
- std::complex<double> rzz = rz / z;
52
-
53
- return (z - 0.5) * std::log(z) - z + loggamma_HLOG2PI + rz * cevalpoly(coeffs, 7, rzz);
54
- }
55
-
56
- XSF_HOST_DEVICE std::complex<double> loggamma_recurrence(std::complex<double> z) {
57
- /* Backward recurrence relation.
58
- *
59
- * See Proposition 2.2 in [1] and the Julia implementation [2].
60
- *
61
- */
62
- int signflips = 0;
63
- int sb = 0;
64
- std::complex<double> shiftprod = z;
65
-
66
- z += 1.0;
67
- int nsb;
68
- while (z.real() <= loggamma_SMALLX) {
69
- shiftprod *= z;
70
- nsb = std::signbit(shiftprod.imag());
71
- signflips += nsb != 0 && sb == 0 ? 1 : 0;
72
- sb = nsb;
73
- z += 1.0;
74
- }
75
- return loggamma_stirling(z) - std::log(shiftprod) - signflips * 2 * M_PI * std::complex<double>(0, 1);
76
- }
77
-
78
- XSF_HOST_DEVICE std::complex<double> loggamma_taylor(std::complex<double> z) {
79
- /* Taylor series for log-Gamma around z = 1.
80
- *
81
- * It is
82
- *
83
- * loggamma(z + 1) = -gamma*z + zeta(2)*z**2/2 - zeta(3)*z**3/3 ...
84
- *
85
- * where gamma is the Euler-Mascheroni constant.
86
- */
87
-
88
- double coeffs[] = {
89
- -4.3478266053040259361E-2, 4.5454556293204669442E-2, -4.7619070330142227991E-2, 5.000004769810169364E-2,
90
- -5.2631679379616660734E-2, 5.5555767627403611102E-2, -5.8823978658684582339E-2, 6.2500955141213040742E-2,
91
- -6.6668705882420468033E-2, 7.1432946295361336059E-2, -7.6932516411352191473E-2, 8.3353840546109004025E-2,
92
- -9.0954017145829042233E-2, 1.0009945751278180853E-1, -1.1133426586956469049E-1, 1.2550966952474304242E-1,
93
- -1.4404989676884611812E-1, 1.6955717699740818995E-1, -2.0738555102867398527E-1, 2.7058080842778454788E-1,
94
- -4.0068563438653142847E-1, 8.2246703342411321824E-1, -5.7721566490153286061E-1};
95
-
96
- z -= 1.0;
97
- return z * cevalpoly(coeffs, 22, z);
98
- }
99
- } // namespace detail
100
-
101
- XSF_HOST_DEVICE inline double loggamma(double x) {
102
- if (x < 0.0) {
103
- return std::numeric_limits<double>::quiet_NaN();
104
- }
105
- return cephes::lgam(x);
106
- }
107
-
108
- XSF_HOST_DEVICE inline float loggamma(float x) { return loggamma(static_cast<double>(x)); }
109
-
110
- XSF_HOST_DEVICE inline std::complex<double> loggamma(std::complex<double> z) {
111
- // Compute the principal branch of log-Gamma
112
-
113
- if (std::isnan(z.real()) || std::isnan(z.imag())) {
114
- return {std::numeric_limits<double>::quiet_NaN(), std::numeric_limits<double>::quiet_NaN()};
115
- }
116
- if (z.real() <= 0 and z == std::floor(z.real())) {
117
- set_error("loggamma", SF_ERROR_SINGULAR, NULL);
118
- return {std::numeric_limits<double>::quiet_NaN(), std::numeric_limits<double>::quiet_NaN()};
119
- }
120
- if (z.real() > detail::loggamma_SMALLX || std::abs(z.imag()) > detail::loggamma_SMALLY) {
121
- return detail::loggamma_stirling(z);
122
- }
123
- if (std::abs(z - 1.0) < detail::loggamma_TAYLOR_RADIUS) {
124
- return detail::loggamma_taylor(z);
125
- }
126
- if (std::abs(z - 2.0) < detail::loggamma_TAYLOR_RADIUS) {
127
- // Recurrence relation and the Taylor series around 1.
128
- return detail::zlog1(z - 1.0) + detail::loggamma_taylor(z - 1.0);
129
- }
130
- if (z.real() < 0.1) {
131
- // Reflection formula; see Proposition 3.1 in [1]
132
- double tmp = std::copysign(2 * M_PI, z.imag()) * std::floor(0.5 * z.real() + 0.25);
133
- return std::complex<double>(detail::loggamma_LOGPI, tmp) - std::log(sinpi(z)) - loggamma(1.0 - z);
134
- }
135
- if (std::signbit(z.imag()) == 0) {
136
- // z.imag() >= 0 but is not -0.0
137
- return detail::loggamma_recurrence(z);
138
- }
139
- return std::conj(detail::loggamma_recurrence(std::conj(z)));
140
- }
141
-
142
- XSF_HOST_DEVICE inline std::complex<float> loggamma(std::complex<float> z) {
143
- return static_cast<std::complex<float>>(loggamma(static_cast<std::complex<double>>(z)));
144
- }
145
-
146
- XSF_HOST_DEVICE inline double rgamma(double z) { return cephes::rgamma(z); }
147
-
148
- XSF_HOST_DEVICE inline float rgamma(float z) { return rgamma(static_cast<double>(z)); }
149
-
150
- XSF_HOST_DEVICE inline std::complex<double> rgamma(std::complex<double> z) {
151
- // Compute 1/Gamma(z) using loggamma.
152
- if (z.real() <= 0 && z == std::floor(z.real())) {
153
- // Zeros at 0, -1, -2, ...
154
- return 0.0;
155
- }
156
- return std::exp(-loggamma(z));
157
- }
158
-
159
- XSF_HOST_DEVICE inline std::complex<float> rgamma(std::complex<float> z) {
160
- return static_cast<std::complex<float>>(rgamma(static_cast<std::complex<double>>(z)));
161
- }
162
-
163
- } // namespace xsf
scipy/special/xsf/sici.h DELETED
@@ -1,200 +0,0 @@
1
- /* Translated from Cython into C++ by SciPy developers in 2024.
2
- * Original header with Copyright information appears below.
3
- */
4
-
5
- /* Implementation of sin/cos/sinh/cosh integrals for complex arguments
6
- *
7
- * Sources
8
- * [1] Fredrik Johansson and others. mpmath: a Python library for
9
- * arbitrary-precision floating-point arithmetic (version 0.19),
10
- * December 2013. http://mpmath.org/.
11
- * [2] NIST, "Digital Library of Mathematical Functions",
12
- * https://dlmf.nist.gov/
13
- */
14
-
15
- #pragma once
16
-
17
- #include "config.h"
18
- #include "error.h"
19
-
20
- #include "expint.h"
21
- #include "cephes/const.h"
22
- #include "cephes/sici.h"
23
- #include "cephes/shichi.h"
24
-
25
- namespace xsf {
26
- namespace detail {
27
-
28
- XSF_HOST_DEVICE inline void sici_power_series(int sgn, std::complex<double> z,
29
- std::complex<double> *s, std::complex<double> *c) {
30
- /* DLMF 6.6.5 and 6.6.6. If sgn = -1 computes si/ci, and if sgn = 1
31
- * computes shi/chi.
32
- */
33
- std::complex<double> fac = z;
34
- *s = fac;
35
- *c = 0;
36
- std::complex<double> term1, term2;
37
- for (int n = 1; n < 100; n++) {
38
- fac *= static_cast<double>(sgn)*z/(2.0*n);
39
- term2 = fac/(2.0*n);
40
- *c += term2;
41
- fac *= z/(2.0*n + 1.0);
42
- term1 = fac/(2.0*n + 1.0);
43
- *s += term1;
44
- constexpr double tol = std::numeric_limits<double>::epsilon();
45
- if (std::abs(term1) < tol*std::abs(*s) && std::abs(term2) < tol*std::abs(*c)) {
46
- break;
47
- }
48
- }
49
- }
50
-
51
- }
52
-
53
-
54
- XSF_HOST_DEVICE inline int sici(std::complex<double> z,
55
- std::complex<double> *si, std::complex<double> *ci) {
56
- /* Compute sin/cos integrals at complex arguments. The algorithm
57
- * largely follows that of [1].
58
- */
59
-
60
- constexpr double EULER = xsf::cephes::detail::SCIPY_EULER;
61
-
62
- if (z == std::numeric_limits<double>::infinity()) {
63
- *si = M_PI_2;
64
- *ci = 0;
65
- return 0;
66
- }
67
- if (z == -std::numeric_limits<double>::infinity()) {
68
- *si = -M_PI_2;
69
- *ci = {0.0, M_PI};
70
- return 0;
71
- }
72
-
73
- if (std::abs(z) < 0.8) {
74
- // Use the series to avoid cancellation in si
75
- detail::sici_power_series(-1, z, si, ci);
76
-
77
- if (z == 0.0) {
78
- set_error("sici", SF_ERROR_DOMAIN, NULL);
79
- *ci = {-std::numeric_limits<double>::infinity(), std::numeric_limits<double>::quiet_NaN()};
80
- } else {
81
- *ci += EULER + std::log(z);
82
- }
83
- return 0;
84
- }
85
-
86
- // DLMF 6.5.5/6.5.6 plus DLMF 6.4.4/6.4.6/6.4.7
87
- std::complex<double> jz = std::complex<double>(0.0, 1.0) * z;
88
- std::complex<double> term1 = expi(jz);
89
- std::complex<double> term2 = expi(-jz);
90
- *si = std::complex<double>(0.0, -0.5)*(term1 - term2);
91
- *ci = 0.5*(term1 + term2);
92
- if (z.real() == 0) {
93
- if (z.imag() > 0) {
94
- *ci += std::complex<double>(0.0, M_PI_2);
95
- } else if (z.imag() < 0) {
96
- *ci -= std::complex<double>(0.0, M_PI_2);
97
- }
98
- } else if (z.real() > 0) {
99
- *si -= M_PI_2;
100
- } else {
101
- *si += M_PI_2;
102
- if (z.imag() >= 0) {
103
- *ci += std::complex<double>(0.0, M_PI);
104
- } else {
105
- *ci -= std::complex<double>(0.0, M_PI);
106
- }
107
- }
108
- return 0;
109
- }
110
-
111
- XSF_HOST_DEVICE inline int sici(std::complex<float> z,
112
- std::complex<float> *si_f, std::complex<float> *ci_f) {
113
- std::complex<double> si;
114
- std::complex<double> ci;
115
- int res = sici(z, &si, &ci);
116
- *si_f = si;
117
- *ci_f = ci;
118
- return res;
119
- }
120
-
121
- XSF_HOST_DEVICE inline int shichi(std::complex<double> z,
122
- std::complex<double> *shi, std::complex<double> *chi) {
123
- /* Compute sinh/cosh integrals at complex arguments. The algorithm
124
- * largely follows that of [1].
125
- */
126
- constexpr double EULER = xsf::cephes::detail::SCIPY_EULER;
127
- if (z == std::numeric_limits<double>::infinity()) {
128
- *shi = std::numeric_limits<double>::infinity();
129
- *chi = std::numeric_limits<double>::infinity();
130
- return 0;
131
- }
132
- if (z == -std::numeric_limits<double>::infinity()) {
133
- *shi = -std::numeric_limits<double>::infinity();
134
- *chi = std::numeric_limits<double>::infinity();
135
- return 0;
136
- }
137
- if (std::abs(z) < 0.8) {
138
- // Use the series to avoid cancellation in shi
139
- detail::sici_power_series(1, z, shi, chi);
140
- if (z == 0.0) {
141
- set_error("shichi", SF_ERROR_DOMAIN, NULL);
142
- *chi = {-std::numeric_limits<double>::infinity(), std::numeric_limits<double>::quiet_NaN()};
143
- } else {
144
- *chi += EULER + std::log(z);
145
- }
146
- return 0;
147
- }
148
-
149
- std::complex<double> term1 = expi(z);
150
- std::complex<double> term2 = expi(-z);
151
- *shi = 0.5*(term1 - term2);
152
- *chi = 0.5*(term1 + term2);
153
- if (z.imag() > 0) {
154
- *shi -= std::complex<double>(0.0, 0.5*M_PI);
155
- *chi += std::complex<double>(0.0, 0.5*M_PI);
156
- } else if (z.imag() < 0) {
157
- *shi += std::complex<double>(0.0, 0.5*M_PI);
158
- *chi -= std::complex<double>(0.0, 0.5*M_PI);
159
- } else if (z.real() < 0) {
160
- *chi += std::complex<double>(0.0, M_PI);
161
- }
162
- return 0;
163
- }
164
-
165
- XSF_HOST_DEVICE inline int shichi(std::complex<float> z,
166
- std::complex<float> *shi_f, std::complex<float> *chi_f) {
167
- std::complex<double> shi;
168
- std::complex<double> chi;
169
- int res = shichi(z, &shi, &chi);
170
- *shi_f = shi;
171
- *chi_f = chi;
172
- return res;
173
- }
174
-
175
- XSF_HOST_DEVICE inline int sici(double x, double *si, double *ci) {
176
- return cephes::sici(x, si, ci);
177
- }
178
-
179
- XSF_HOST_DEVICE inline int shichi(double x, double *shi, double *chi) {
180
- return cephes::shichi(x, shi, chi);
181
- }
182
-
183
- XSF_HOST_DEVICE inline int sici(float x, float *si_f, float *ci_f) {
184
- double si;
185
- double ci;
186
- int res = cephes::sici(x, &si, &ci);
187
- *si_f = si;
188
- *ci_f = ci;
189
- return res;
190
- }
191
-
192
- XSF_HOST_DEVICE inline int shichi(float x, float *shi_f, float *chi_f) {
193
- double shi;
194
- double chi;
195
- int res = cephes::shichi(x, &shi, &chi);
196
- *shi_f = shi;
197
- *chi_f = chi;
198
- return res;
199
- }
200
- }