scipy 1.15.3__cp311-cp311-musllinux_1_2_aarch64.whl → 1.16.0__cp311-cp311-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 (642) hide show
  1. scipy/__config__.py +11 -11
  2. scipy/__init__.py +3 -6
  3. scipy/_cyutility.cpython-311-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-311-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-311-aarch64-linux-musl.so +0 -0
  11. scipy/_lib/_sparse.py +41 -0
  12. scipy/_lib/_test_ccallback.cpython-311-aarch64-linux-musl.so +0 -0
  13. scipy/_lib/_test_deprecation_call.cpython-311-aarch64-linux-musl.so +0 -0
  14. scipy/_lib/_test_deprecation_def.cpython-311-aarch64-linux-musl.so +0 -0
  15. scipy/_lib/_testutils.py +6 -2
  16. scipy/_lib/_uarray/_uarray.cpython-311-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-311-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-311-aarch64-linux-musl.so +0 -0
  97. scipy/cluster/_optimal_leaf_ordering.cpython-311-aarch64-linux-musl.so +0 -0
  98. scipy/cluster/_vq.cpython-311-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 +282 -151
  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-311-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-311-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-311-aarch64-linux-musl.so +0 -0
  129. scipy/integrate/_lsoda.cpython-311-aarch64-linux-musl.so +0 -0
  130. scipy/integrate/_ode.py +9 -2
  131. scipy/integrate/_odepack.cpython-311-aarch64-linux-musl.so +0 -0
  132. scipy/integrate/_quad_vec.py +21 -29
  133. scipy/integrate/_quadpack.cpython-311-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-311-aarch64-linux-musl.so +0 -0
  139. scipy/integrate/_test_odeint_banded.cpython-311-aarch64-linux-musl.so +0 -0
  140. scipy/integrate/_vode.cpython-311-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-311-aarch64-linux-musl.so +0 -0
  151. scipy/interpolate/_dierckx.cpython-311-aarch64-linux-musl.so +0 -0
  152. scipy/interpolate/_fitpack.cpython-311-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-311-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-311-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-311-aarch64-linux-musl.so +0 -0
  165. scipy/interpolate/_rgi.py +31 -26
  166. scipy/interpolate/_rgi_cython.cpython-311-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-311-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-311-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-311-aarch64-linux-musl.so +0 -0
  187. scipy/io/matlab/_mio_utils.cpython-311-aarch64-linux-musl.so +0 -0
  188. scipy/io/matlab/_miobase.py +4 -1
  189. scipy/io/matlab/_streams.cpython-311-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-311-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-311-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-311-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-311-aarch64-linux-musl.so +0 -0
  210. scipy/linalg/_expm_frechet.py +4 -0
  211. scipy/linalg/_fblas.cpython-311-aarch64-linux-musl.so +0 -0
  212. scipy/linalg/_flapack.cpython-311-aarch64-linux-musl.so +0 -0
  213. scipy/linalg/_linalg_pythran.cpython-311-aarch64-linux-musl.so +0 -0
  214. scipy/linalg/_matfuncs.py +187 -4
  215. scipy/linalg/_matfuncs_expm.cpython-311-aarch64-linux-musl.so +0 -0
  216. scipy/linalg/_matfuncs_schur_sqrtm.cpython-311-aarch64-linux-musl.so +0 -0
  217. scipy/linalg/_matfuncs_sqrtm.py +1 -99
  218. scipy/linalg/_matfuncs_sqrtm_triu.cpython-311-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-311-aarch64-linux-musl.so +0 -0
  222. scipy/linalg/_solvers.py +7 -2
  223. scipy/linalg/_special_matrices.py +26 -36
  224. scipy/linalg/blas.py +35 -24
  225. scipy/linalg/cython_blas.cpython-311-aarch64-linux-musl.so +0 -0
  226. scipy/linalg/cython_lapack.cpython-311-aarch64-linux-musl.so +0 -0
  227. scipy/linalg/lapack.py +22 -2
  228. scipy/linalg/tests/_cython_examples/meson.build +7 -0
  229. scipy/linalg/tests/test_basic.py +31 -16
  230. scipy/linalg/tests/test_batch.py +588 -0
  231. scipy/linalg/tests/test_cythonized_array_utils.py +0 -2
  232. scipy/linalg/tests/test_decomp.py +40 -3
  233. scipy/linalg/tests/test_decomp_cossin.py +14 -0
  234. scipy/linalg/tests/test_decomp_ldl.py +1 -1
  235. scipy/linalg/tests/test_lapack.py +115 -7
  236. scipy/linalg/tests/test_matfuncs.py +157 -102
  237. scipy/linalg/tests/test_procrustes.py +0 -7
  238. scipy/linalg/tests/test_solve_toeplitz.py +1 -1
  239. scipy/linalg/tests/test_special_matrices.py +1 -5
  240. scipy/ndimage/__init__.py +1 -0
  241. scipy/ndimage/_ctest.cpython-311-aarch64-linux-musl.so +0 -0
  242. scipy/ndimage/_cytest.cpython-311-aarch64-linux-musl.so +0 -0
  243. scipy/ndimage/_delegators.py +8 -2
  244. scipy/ndimage/_filters.py +453 -5
  245. scipy/ndimage/_interpolation.py +36 -6
  246. scipy/ndimage/_measurements.py +4 -2
  247. scipy/ndimage/_morphology.py +5 -0
  248. scipy/ndimage/_nd_image.cpython-311-aarch64-linux-musl.so +0 -0
  249. scipy/ndimage/_ni_docstrings.py +5 -1
  250. scipy/ndimage/_ni_label.cpython-311-aarch64-linux-musl.so +0 -0
  251. scipy/ndimage/_ni_support.py +1 -5
  252. scipy/ndimage/_rank_filter_1d.cpython-311-aarch64-linux-musl.so +0 -0
  253. scipy/ndimage/_support_alternative_backends.py +18 -6
  254. scipy/ndimage/tests/test_filters.py +370 -259
  255. scipy/ndimage/tests/test_fourier.py +7 -9
  256. scipy/ndimage/tests/test_interpolation.py +68 -61
  257. scipy/ndimage/tests/test_measurements.py +18 -35
  258. scipy/ndimage/tests/test_morphology.py +143 -131
  259. scipy/ndimage/tests/test_splines.py +1 -3
  260. scipy/odr/__odrpack.cpython-311-aarch64-linux-musl.so +0 -0
  261. scipy/optimize/_basinhopping.py +13 -7
  262. scipy/optimize/_bglu_dense.cpython-311-aarch64-linux-musl.so +0 -0
  263. scipy/optimize/_bracket.py +17 -24
  264. scipy/optimize/_chandrupatla.py +9 -10
  265. scipy/optimize/_cobyla_py.py +104 -123
  266. scipy/optimize/_constraints.py +14 -10
  267. scipy/optimize/_differentiable_functions.py +371 -230
  268. scipy/optimize/_differentialevolution.py +4 -3
  269. scipy/optimize/_direct.cpython-311-aarch64-linux-musl.so +0 -0
  270. scipy/optimize/_dual_annealing.py +1 -1
  271. scipy/optimize/_elementwise.py +1 -4
  272. scipy/optimize/_group_columns.cpython-311-aarch64-linux-musl.so +0 -0
  273. scipy/optimize/_highspy/_core.cpython-311-aarch64-linux-musl.so +0 -0
  274. scipy/optimize/_highspy/_highs_options.cpython-311-aarch64-linux-musl.so +0 -0
  275. scipy/optimize/_lbfgsb.cpython-311-aarch64-linux-musl.so +0 -0
  276. scipy/optimize/_lbfgsb_py.py +80 -24
  277. scipy/optimize/_linprog_doc.py +2 -2
  278. scipy/optimize/_linprog_highs.py +2 -2
  279. scipy/optimize/_linprog_ip.py +25 -10
  280. scipy/optimize/_linprog_util.py +14 -16
  281. scipy/optimize/_lsap.cpython-311-aarch64-linux-musl.so +0 -0
  282. scipy/optimize/_lsq/common.py +3 -3
  283. scipy/optimize/_lsq/dogbox.py +16 -2
  284. scipy/optimize/_lsq/givens_elimination.cpython-311-aarch64-linux-musl.so +0 -0
  285. scipy/optimize/_lsq/least_squares.py +198 -126
  286. scipy/optimize/_lsq/lsq_linear.py +6 -6
  287. scipy/optimize/_lsq/trf.py +35 -8
  288. scipy/optimize/_milp.py +3 -1
  289. scipy/optimize/_minimize.py +105 -36
  290. scipy/optimize/_minpack.cpython-311-aarch64-linux-musl.so +0 -0
  291. scipy/optimize/_minpack_py.py +21 -14
  292. scipy/optimize/_moduleTNC.cpython-311-aarch64-linux-musl.so +0 -0
  293. scipy/optimize/_nnls.py +20 -21
  294. scipy/optimize/_nonlin.py +34 -3
  295. scipy/optimize/_numdiff.py +288 -110
  296. scipy/optimize/_optimize.py +86 -48
  297. scipy/optimize/_pava_pybind.cpython-311-aarch64-linux-musl.so +0 -0
  298. scipy/optimize/_remove_redundancy.py +5 -5
  299. scipy/optimize/_root_scalar.py +1 -1
  300. scipy/optimize/_shgo.py +6 -0
  301. scipy/optimize/_shgo_lib/_complex.py +1 -1
  302. scipy/optimize/_slsqp_py.py +216 -124
  303. scipy/optimize/_slsqplib.cpython-311-aarch64-linux-musl.so +0 -0
  304. scipy/optimize/_spectral.py +1 -1
  305. scipy/optimize/_tnc.py +8 -1
  306. scipy/optimize/_trlib/_trlib.cpython-311-aarch64-linux-musl.so +0 -0
  307. scipy/optimize/_trustregion.py +20 -6
  308. scipy/optimize/_trustregion_constr/canonical_constraint.py +7 -7
  309. scipy/optimize/_trustregion_constr/equality_constrained_sqp.py +1 -1
  310. scipy/optimize/_trustregion_constr/minimize_trustregion_constr.py +11 -3
  311. scipy/optimize/_trustregion_constr/projections.py +12 -8
  312. scipy/optimize/_trustregion_constr/qp_subproblem.py +9 -9
  313. scipy/optimize/_trustregion_constr/tests/test_projections.py +7 -7
  314. scipy/optimize/_trustregion_constr/tests/test_qp_subproblem.py +77 -77
  315. scipy/optimize/_trustregion_constr/tr_interior_point.py +5 -5
  316. scipy/optimize/_trustregion_exact.py +0 -1
  317. scipy/optimize/_zeros.cpython-311-aarch64-linux-musl.so +0 -0
  318. scipy/optimize/_zeros_py.py +97 -17
  319. scipy/optimize/cython_optimize/_zeros.cpython-311-aarch64-linux-musl.so +0 -0
  320. scipy/optimize/slsqp.py +0 -1
  321. scipy/optimize/tests/test__basinhopping.py +1 -1
  322. scipy/optimize/tests/test__differential_evolution.py +4 -4
  323. scipy/optimize/tests/test__linprog_clean_inputs.py +5 -3
  324. scipy/optimize/tests/test__numdiff.py +66 -22
  325. scipy/optimize/tests/test__remove_redundancy.py +2 -2
  326. scipy/optimize/tests/test__shgo.py +9 -1
  327. scipy/optimize/tests/test_bracket.py +36 -46
  328. scipy/optimize/tests/test_chandrupatla.py +133 -135
  329. scipy/optimize/tests/test_cobyla.py +74 -45
  330. scipy/optimize/tests/test_constraints.py +1 -1
  331. scipy/optimize/tests/test_differentiable_functions.py +226 -6
  332. scipy/optimize/tests/test_lbfgsb_hessinv.py +22 -0
  333. scipy/optimize/tests/test_least_squares.py +125 -13
  334. scipy/optimize/tests/test_linear_assignment.py +3 -3
  335. scipy/optimize/tests/test_linprog.py +3 -3
  336. scipy/optimize/tests/test_lsq_linear.py +6 -6
  337. scipy/optimize/tests/test_minimize_constrained.py +2 -2
  338. scipy/optimize/tests/test_minpack.py +4 -4
  339. scipy/optimize/tests/test_nnls.py +43 -3
  340. scipy/optimize/tests/test_nonlin.py +36 -0
  341. scipy/optimize/tests/test_optimize.py +98 -20
  342. scipy/optimize/tests/test_slsqp.py +36 -4
  343. scipy/optimize/tests/test_zeros.py +34 -1
  344. scipy/signal/__init__.py +12 -23
  345. scipy/signal/_delegators.py +568 -0
  346. scipy/signal/_filter_design.py +459 -241
  347. scipy/signal/_fir_filter_design.py +262 -90
  348. scipy/signal/_lti_conversion.py +3 -2
  349. scipy/signal/_ltisys.py +118 -91
  350. scipy/signal/_max_len_seq_inner.cpython-311-aarch64-linux-musl.so +0 -0
  351. scipy/signal/_peak_finding_utils.cpython-311-aarch64-linux-musl.so +0 -0
  352. scipy/signal/_polyutils.py +172 -0
  353. scipy/signal/_short_time_fft.py +519 -70
  354. scipy/signal/_signal_api.py +30 -0
  355. scipy/signal/_signaltools.py +719 -399
  356. scipy/signal/_sigtools.cpython-311-aarch64-linux-musl.so +0 -0
  357. scipy/signal/_sosfilt.cpython-311-aarch64-linux-musl.so +0 -0
  358. scipy/signal/_spectral_py.py +230 -50
  359. scipy/signal/_spline.cpython-311-aarch64-linux-musl.so +0 -0
  360. scipy/signal/_spline_filters.py +108 -68
  361. scipy/signal/_support_alternative_backends.py +73 -0
  362. scipy/signal/_upfirdn.py +4 -1
  363. scipy/signal/_upfirdn_apply.cpython-311-aarch64-linux-musl.so +0 -0
  364. scipy/signal/_waveforms.py +2 -11
  365. scipy/signal/_wavelets.py +1 -1
  366. scipy/signal/fir_filter_design.py +1 -0
  367. scipy/signal/spline.py +4 -11
  368. scipy/signal/tests/_scipy_spectral_test_shim.py +2 -171
  369. scipy/signal/tests/test_bsplines.py +114 -79
  370. scipy/signal/tests/test_cont2discrete.py +9 -2
  371. scipy/signal/tests/test_filter_design.py +721 -481
  372. scipy/signal/tests/test_fir_filter_design.py +332 -140
  373. scipy/signal/tests/test_savitzky_golay.py +4 -3
  374. scipy/signal/tests/test_short_time_fft.py +221 -3
  375. scipy/signal/tests/test_signaltools.py +2145 -1349
  376. scipy/signal/tests/test_spectral.py +50 -6
  377. scipy/signal/tests/test_splines.py +161 -96
  378. scipy/signal/tests/test_upfirdn.py +84 -50
  379. scipy/signal/tests/test_waveforms.py +20 -0
  380. scipy/signal/tests/test_windows.py +607 -466
  381. scipy/signal/windows/_windows.py +287 -148
  382. scipy/sparse/__init__.py +23 -4
  383. scipy/sparse/_base.py +270 -108
  384. scipy/sparse/_bsr.py +7 -4
  385. scipy/sparse/_compressed.py +59 -231
  386. scipy/sparse/_construct.py +90 -38
  387. scipy/sparse/_coo.py +115 -181
  388. scipy/sparse/_csc.py +4 -4
  389. scipy/sparse/_csparsetools.cpython-311-aarch64-linux-musl.so +0 -0
  390. scipy/sparse/_csr.py +2 -2
  391. scipy/sparse/_data.py +48 -48
  392. scipy/sparse/_dia.py +105 -18
  393. scipy/sparse/_dok.py +0 -23
  394. scipy/sparse/_index.py +4 -4
  395. scipy/sparse/_matrix.py +23 -0
  396. scipy/sparse/_sparsetools.cpython-311-aarch64-linux-musl.so +0 -0
  397. scipy/sparse/_sputils.py +37 -22
  398. scipy/sparse/base.py +0 -9
  399. scipy/sparse/bsr.py +0 -14
  400. scipy/sparse/compressed.py +0 -23
  401. scipy/sparse/construct.py +0 -6
  402. scipy/sparse/coo.py +0 -14
  403. scipy/sparse/csc.py +0 -3
  404. scipy/sparse/csgraph/_flow.cpython-311-aarch64-linux-musl.so +0 -0
  405. scipy/sparse/csgraph/_matching.cpython-311-aarch64-linux-musl.so +0 -0
  406. scipy/sparse/csgraph/_min_spanning_tree.cpython-311-aarch64-linux-musl.so +0 -0
  407. scipy/sparse/csgraph/_reordering.cpython-311-aarch64-linux-musl.so +0 -0
  408. scipy/sparse/csgraph/_shortest_path.cpython-311-aarch64-linux-musl.so +0 -0
  409. scipy/sparse/csgraph/_tools.cpython-311-aarch64-linux-musl.so +0 -0
  410. scipy/sparse/csgraph/_traversal.cpython-311-aarch64-linux-musl.so +0 -0
  411. scipy/sparse/csgraph/tests/test_matching.py +14 -2
  412. scipy/sparse/csgraph/tests/test_pydata_sparse.py +4 -1
  413. scipy/sparse/csgraph/tests/test_shortest_path.py +83 -27
  414. scipy/sparse/csr.py +0 -5
  415. scipy/sparse/data.py +1 -6
  416. scipy/sparse/dia.py +0 -7
  417. scipy/sparse/dok.py +0 -10
  418. scipy/sparse/linalg/_dsolve/_superlu.cpython-311-aarch64-linux-musl.so +0 -0
  419. scipy/sparse/linalg/_dsolve/linsolve.py +9 -0
  420. scipy/sparse/linalg/_dsolve/tests/test_linsolve.py +35 -28
  421. scipy/sparse/linalg/_eigen/arpack/_arpack.cpython-311-aarch64-linux-musl.so +0 -0
  422. scipy/sparse/linalg/_eigen/arpack/arpack.py +23 -17
  423. scipy/sparse/linalg/_eigen/lobpcg/lobpcg.py +6 -6
  424. scipy/sparse/linalg/_interface.py +17 -18
  425. scipy/sparse/linalg/_isolve/_gcrotmk.py +4 -4
  426. scipy/sparse/linalg/_isolve/iterative.py +51 -45
  427. scipy/sparse/linalg/_isolve/lgmres.py +6 -6
  428. scipy/sparse/linalg/_isolve/minres.py +5 -5
  429. scipy/sparse/linalg/_isolve/tfqmr.py +7 -7
  430. scipy/sparse/linalg/_isolve/utils.py +2 -8
  431. scipy/sparse/linalg/_matfuncs.py +1 -1
  432. scipy/sparse/linalg/_norm.py +1 -1
  433. scipy/sparse/linalg/_propack/_cpropack.cpython-311-aarch64-linux-musl.so +0 -0
  434. scipy/sparse/linalg/_propack/_dpropack.cpython-311-aarch64-linux-musl.so +0 -0
  435. scipy/sparse/linalg/_propack/_spropack.cpython-311-aarch64-linux-musl.so +0 -0
  436. scipy/sparse/linalg/_propack/_zpropack.cpython-311-aarch64-linux-musl.so +0 -0
  437. scipy/sparse/linalg/_special_sparse_arrays.py +39 -38
  438. scipy/sparse/linalg/tests/test_pydata_sparse.py +14 -0
  439. scipy/sparse/tests/test_arithmetic1d.py +5 -2
  440. scipy/sparse/tests/test_base.py +214 -42
  441. scipy/sparse/tests/test_common1d.py +7 -7
  442. scipy/sparse/tests/test_construct.py +1 -1
  443. scipy/sparse/tests/test_coo.py +272 -4
  444. scipy/sparse/tests/test_sparsetools.py +5 -0
  445. scipy/sparse/tests/test_sputils.py +36 -7
  446. scipy/spatial/_ckdtree.cpython-311-aarch64-linux-musl.so +0 -0
  447. scipy/spatial/_distance_pybind.cpython-311-aarch64-linux-musl.so +0 -0
  448. scipy/spatial/_distance_wrap.cpython-311-aarch64-linux-musl.so +0 -0
  449. scipy/spatial/_hausdorff.cpython-311-aarch64-linux-musl.so +0 -0
  450. scipy/spatial/_qhull.cpython-311-aarch64-linux-musl.so +0 -0
  451. scipy/spatial/_voronoi.cpython-311-aarch64-linux-musl.so +0 -0
  452. scipy/spatial/distance.py +49 -42
  453. scipy/spatial/tests/test_distance.py +15 -1
  454. scipy/spatial/tests/test_kdtree.py +1 -0
  455. scipy/spatial/tests/test_qhull.py +7 -2
  456. scipy/spatial/transform/__init__.py +5 -3
  457. scipy/spatial/transform/_rigid_transform.cpython-311-aarch64-linux-musl.so +0 -0
  458. scipy/spatial/transform/_rotation.cpython-311-aarch64-linux-musl.so +0 -0
  459. scipy/spatial/transform/tests/test_rigid_transform.py +1221 -0
  460. scipy/spatial/transform/tests/test_rotation.py +1213 -832
  461. scipy/spatial/transform/tests/test_rotation_groups.py +3 -3
  462. scipy/spatial/transform/tests/test_rotation_spline.py +29 -8
  463. scipy/special/__init__.py +1 -47
  464. scipy/special/_add_newdocs.py +34 -772
  465. scipy/special/_basic.py +22 -25
  466. scipy/special/_comb.cpython-311-aarch64-linux-musl.so +0 -0
  467. scipy/special/_ellip_harm_2.cpython-311-aarch64-linux-musl.so +0 -0
  468. scipy/special/_gufuncs.cpython-311-aarch64-linux-musl.so +0 -0
  469. scipy/special/_logsumexp.py +67 -58
  470. scipy/special/_orthogonal.pyi +1 -1
  471. scipy/special/_specfun.cpython-311-aarch64-linux-musl.so +0 -0
  472. scipy/special/_special_ufuncs.cpython-311-aarch64-linux-musl.so +0 -0
  473. scipy/special/_spherical_bessel.py +4 -4
  474. scipy/special/_support_alternative_backends.py +212 -119
  475. scipy/special/_test_internal.cpython-311-aarch64-linux-musl.so +0 -0
  476. scipy/special/_testutils.py +4 -4
  477. scipy/special/_ufuncs.cpython-311-aarch64-linux-musl.so +0 -0
  478. scipy/special/_ufuncs.pyi +1 -0
  479. scipy/special/_ufuncs.pyx +215 -1400
  480. scipy/special/_ufuncs_cxx.cpython-311-aarch64-linux-musl.so +0 -0
  481. scipy/special/_ufuncs_cxx.pxd +2 -15
  482. scipy/special/_ufuncs_cxx.pyx +5 -44
  483. scipy/special/_ufuncs_cxx_defs.h +2 -16
  484. scipy/special/_ufuncs_defs.h +0 -8
  485. scipy/special/cython_special.cpython-311-aarch64-linux-musl.so +0 -0
  486. scipy/special/cython_special.pxd +1 -1
  487. scipy/special/tests/_cython_examples/meson.build +10 -1
  488. scipy/special/tests/test_basic.py +153 -20
  489. scipy/special/tests/test_boost_ufuncs.py +3 -0
  490. scipy/special/tests/test_cdflib.py +35 -11
  491. scipy/special/tests/test_gammainc.py +16 -0
  492. scipy/special/tests/test_hyp2f1.py +2 -2
  493. scipy/special/tests/test_log1mexp.py +85 -0
  494. scipy/special/tests/test_logsumexp.py +206 -64
  495. scipy/special/tests/test_mpmath.py +1 -0
  496. scipy/special/tests/test_nan_inputs.py +1 -1
  497. scipy/special/tests/test_orthogonal.py +17 -18
  498. scipy/special/tests/test_sf_error.py +3 -2
  499. scipy/special/tests/test_sph_harm.py +6 -7
  500. scipy/special/tests/test_support_alternative_backends.py +211 -76
  501. scipy/stats/__init__.py +4 -1
  502. scipy/stats/_ansari_swilk_statistics.cpython-311-aarch64-linux-musl.so +0 -0
  503. scipy/stats/_axis_nan_policy.py +5 -12
  504. scipy/stats/_biasedurn.cpython-311-aarch64-linux-musl.so +0 -0
  505. scipy/stats/_continued_fraction.py +387 -0
  506. scipy/stats/_continuous_distns.py +277 -310
  507. scipy/stats/_correlation.py +1 -1
  508. scipy/stats/_covariance.py +6 -3
  509. scipy/stats/_discrete_distns.py +39 -32
  510. scipy/stats/_distn_infrastructure.py +39 -12
  511. scipy/stats/_distribution_infrastructure.py +920 -238
  512. scipy/stats/_entropy.py +9 -10
  513. scipy/{_lib → stats}/_finite_differences.py +1 -1
  514. scipy/stats/_hypotests.py +83 -50
  515. scipy/stats/_kde.py +53 -49
  516. scipy/stats/_ksstats.py +1 -1
  517. scipy/stats/_levy_stable/__init__.py +7 -15
  518. scipy/stats/_levy_stable/levyst.cpython-311-aarch64-linux-musl.so +0 -0
  519. scipy/stats/_morestats.py +118 -73
  520. scipy/stats/_mstats_basic.py +13 -17
  521. scipy/stats/_mstats_extras.py +8 -8
  522. scipy/stats/_multivariate.py +89 -113
  523. scipy/stats/_new_distributions.py +97 -20
  524. scipy/stats/_page_trend_test.py +12 -5
  525. scipy/stats/_probability_distribution.py +265 -43
  526. scipy/stats/_qmc.py +14 -9
  527. scipy/stats/_qmc_cy.cpython-311-aarch64-linux-musl.so +0 -0
  528. scipy/stats/_qmvnt.py +16 -95
  529. scipy/stats/_qmvnt_cy.cpython-311-aarch64-linux-musl.so +0 -0
  530. scipy/stats/_quantile.py +335 -0
  531. scipy/stats/_rcont/rcont.cpython-311-aarch64-linux-musl.so +0 -0
  532. scipy/stats/_resampling.py +5 -30
  533. scipy/stats/_sampling.py +1 -1
  534. scipy/stats/_sobol.cpython-311-aarch64-linux-musl.so +0 -0
  535. scipy/stats/_stats.cpython-311-aarch64-linux-musl.so +0 -0
  536. scipy/stats/_stats_mstats_common.py +21 -2
  537. scipy/stats/_stats_py.py +551 -477
  538. scipy/stats/_stats_pythran.cpython-311-aarch64-linux-musl.so +0 -0
  539. scipy/stats/_unuran/unuran_wrapper.cpython-311-aarch64-linux-musl.so +0 -0
  540. scipy/stats/_unuran/unuran_wrapper.pyi +2 -1
  541. scipy/stats/_variation.py +6 -8
  542. scipy/stats/_wilcoxon.py +13 -7
  543. scipy/stats/tests/common_tests.py +6 -4
  544. scipy/stats/tests/test_axis_nan_policy.py +62 -24
  545. scipy/stats/tests/test_continued_fraction.py +173 -0
  546. scipy/stats/tests/test_continuous.py +379 -60
  547. scipy/stats/tests/test_continuous_basic.py +18 -12
  548. scipy/stats/tests/test_discrete_basic.py +14 -8
  549. scipy/stats/tests/test_discrete_distns.py +16 -16
  550. scipy/stats/tests/test_distributions.py +95 -75
  551. scipy/stats/tests/test_entropy.py +40 -48
  552. scipy/stats/tests/test_fit.py +4 -3
  553. scipy/stats/tests/test_hypotests.py +153 -24
  554. scipy/stats/tests/test_kdeoth.py +109 -41
  555. scipy/stats/tests/test_marray.py +289 -0
  556. scipy/stats/tests/test_morestats.py +81 -49
  557. scipy/stats/tests/test_mstats_basic.py +3 -3
  558. scipy/stats/tests/test_multivariate.py +434 -83
  559. scipy/stats/tests/test_qmc.py +13 -10
  560. scipy/stats/tests/test_quantile.py +199 -0
  561. scipy/stats/tests/test_rank.py +119 -112
  562. scipy/stats/tests/test_resampling.py +47 -56
  563. scipy/stats/tests/test_sampling.py +9 -4
  564. scipy/stats/tests/test_stats.py +799 -939
  565. scipy/stats/tests/test_variation.py +8 -6
  566. scipy/version.py +2 -2
  567. {scipy-1.15.3.dist-info → scipy-1.16.0.dist-info}/LICENSE.txt +4 -4
  568. {scipy-1.15.3.dist-info → scipy-1.16.0.dist-info}/METADATA +11 -11
  569. {scipy-1.15.3.dist-info → scipy-1.16.0.dist-info}/RECORD +1284 -1291
  570. scipy.libs/libgcc_s-2d945d6c.so.1 +0 -0
  571. scipy.libs/libgfortran-67378ab2.so.5.0.0 +0 -0
  572. scipy.libs/{libstdc++-1b614e01.so.6.0.32 → libstdc++-85f2cd6d.so.6.0.33} +0 -0
  573. scipy/_lib/array_api_extra/_funcs.py +0 -484
  574. scipy/_lib/array_api_extra/_typing.py +0 -8
  575. scipy/interpolate/_bspl.cpython-311-aarch64-linux-musl.so +0 -0
  576. scipy/optimize/_cobyla.cpython-311-aarch64-linux-musl.so +0 -0
  577. scipy/optimize/_cython_nnls.cpython-311-aarch64-linux-musl.so +0 -0
  578. scipy/optimize/_slsqp.cpython-311-aarch64-linux-musl.so +0 -0
  579. scipy/spatial/qhull_src/COPYING.txt +0 -38
  580. scipy/special/libsf_error_state.so +0 -0
  581. scipy/special/tests/test_log_softmax.py +0 -109
  582. scipy/special/tests/test_xsf_cuda.py +0 -114
  583. scipy/special/xsf/binom.h +0 -89
  584. scipy/special/xsf/cdflib.h +0 -100
  585. scipy/special/xsf/cephes/airy.h +0 -307
  586. scipy/special/xsf/cephes/besselpoly.h +0 -51
  587. scipy/special/xsf/cephes/beta.h +0 -257
  588. scipy/special/xsf/cephes/cbrt.h +0 -131
  589. scipy/special/xsf/cephes/chbevl.h +0 -85
  590. scipy/special/xsf/cephes/chdtr.h +0 -193
  591. scipy/special/xsf/cephes/const.h +0 -87
  592. scipy/special/xsf/cephes/ellie.h +0 -293
  593. scipy/special/xsf/cephes/ellik.h +0 -251
  594. scipy/special/xsf/cephes/ellpe.h +0 -107
  595. scipy/special/xsf/cephes/ellpk.h +0 -117
  596. scipy/special/xsf/cephes/expn.h +0 -260
  597. scipy/special/xsf/cephes/gamma.h +0 -398
  598. scipy/special/xsf/cephes/hyp2f1.h +0 -596
  599. scipy/special/xsf/cephes/hyperg.h +0 -361
  600. scipy/special/xsf/cephes/i0.h +0 -149
  601. scipy/special/xsf/cephes/i1.h +0 -158
  602. scipy/special/xsf/cephes/igam.h +0 -421
  603. scipy/special/xsf/cephes/igam_asymp_coeff.h +0 -195
  604. scipy/special/xsf/cephes/igami.h +0 -313
  605. scipy/special/xsf/cephes/j0.h +0 -225
  606. scipy/special/xsf/cephes/j1.h +0 -198
  607. scipy/special/xsf/cephes/jv.h +0 -715
  608. scipy/special/xsf/cephes/k0.h +0 -164
  609. scipy/special/xsf/cephes/k1.h +0 -163
  610. scipy/special/xsf/cephes/kn.h +0 -243
  611. scipy/special/xsf/cephes/lanczos.h +0 -112
  612. scipy/special/xsf/cephes/ndtr.h +0 -275
  613. scipy/special/xsf/cephes/poch.h +0 -85
  614. scipy/special/xsf/cephes/polevl.h +0 -167
  615. scipy/special/xsf/cephes/psi.h +0 -194
  616. scipy/special/xsf/cephes/rgamma.h +0 -111
  617. scipy/special/xsf/cephes/scipy_iv.h +0 -811
  618. scipy/special/xsf/cephes/shichi.h +0 -248
  619. scipy/special/xsf/cephes/sici.h +0 -224
  620. scipy/special/xsf/cephes/sindg.h +0 -221
  621. scipy/special/xsf/cephes/tandg.h +0 -139
  622. scipy/special/xsf/cephes/trig.h +0 -58
  623. scipy/special/xsf/cephes/unity.h +0 -186
  624. scipy/special/xsf/cephes/zeta.h +0 -172
  625. scipy/special/xsf/config.h +0 -304
  626. scipy/special/xsf/digamma.h +0 -205
  627. scipy/special/xsf/error.h +0 -57
  628. scipy/special/xsf/evalpoly.h +0 -47
  629. scipy/special/xsf/expint.h +0 -266
  630. scipy/special/xsf/hyp2f1.h +0 -694
  631. scipy/special/xsf/iv_ratio.h +0 -173
  632. scipy/special/xsf/lambertw.h +0 -150
  633. scipy/special/xsf/loggamma.h +0 -163
  634. scipy/special/xsf/sici.h +0 -200
  635. scipy/special/xsf/tools.h +0 -427
  636. scipy/special/xsf/trig.h +0 -164
  637. scipy/special/xsf/wright_bessel.h +0 -843
  638. scipy/special/xsf/zlog1.h +0 -35
  639. scipy/stats/_mvn.cpython-311-aarch64-linux-musl.so +0 -0
  640. scipy.libs/libgcc_s-7393e603.so.1 +0 -0
  641. scipy.libs/libgfortran-eb933d8e.so.5.0.0 +0 -0
  642. {scipy-1.15.3.dist-info → scipy-1.16.0.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
- }