scipy 1.15.3__cp313-cp313-musllinux_1_2_aarch64.whl → 1.16.0rc1__cp313-cp313-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 (640) hide show
  1. scipy/__config__.py +9 -9
  2. scipy/__init__.py +3 -6
  3. scipy/_cyutility.cpython-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  11. scipy/_lib/_sparse.py +41 -0
  12. scipy/_lib/_test_ccallback.cpython-313-aarch64-linux-musl.so +0 -0
  13. scipy/_lib/_test_deprecation_call.cpython-313-aarch64-linux-musl.so +0 -0
  14. scipy/_lib/_test_deprecation_def.cpython-313-aarch64-linux-musl.so +0 -0
  15. scipy/_lib/_testutils.py +6 -2
  16. scipy/_lib/_uarray/_uarray.cpython-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  97. scipy/cluster/_optimal_leaf_ordering.cpython-313-aarch64-linux-musl.so +0 -0
  98. scipy/cluster/_vq.cpython-313-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-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  129. scipy/integrate/_lsoda.cpython-313-aarch64-linux-musl.so +0 -0
  130. scipy/integrate/_ode.py +9 -2
  131. scipy/integrate/_odepack.cpython-313-aarch64-linux-musl.so +0 -0
  132. scipy/integrate/_quad_vec.py +21 -29
  133. scipy/integrate/_quadpack.cpython-313-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-313-aarch64-linux-musl.so +0 -0
  139. scipy/integrate/_test_odeint_banded.cpython-313-aarch64-linux-musl.so +0 -0
  140. scipy/integrate/_vode.cpython-313-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-313-aarch64-linux-musl.so +0 -0
  151. scipy/interpolate/_dierckx.cpython-313-aarch64-linux-musl.so +0 -0
  152. scipy/interpolate/_fitpack.cpython-313-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-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  165. scipy/interpolate/_rgi.py +31 -26
  166. scipy/interpolate/_rgi_cython.cpython-313-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-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  187. scipy/io/matlab/_mio_utils.cpython-313-aarch64-linux-musl.so +0 -0
  188. scipy/io/matlab/_miobase.py +4 -1
  189. scipy/io/matlab/_streams.cpython-313-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-313-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-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  210. scipy/linalg/_expm_frechet.py +4 -0
  211. scipy/linalg/_fblas.cpython-313-aarch64-linux-musl.so +0 -0
  212. scipy/linalg/_flapack.cpython-313-aarch64-linux-musl.so +0 -0
  213. scipy/linalg/_linalg_pythran.cpython-313-aarch64-linux-musl.so +0 -0
  214. scipy/linalg/_matfuncs.py +187 -4
  215. scipy/linalg/_matfuncs_expm.cpython-313-aarch64-linux-musl.so +0 -0
  216. scipy/linalg/_matfuncs_schur_sqrtm.cpython-313-aarch64-linux-musl.so +0 -0
  217. scipy/linalg/_matfuncs_sqrtm.py +1 -99
  218. scipy/linalg/_matfuncs_sqrtm_triu.cpython-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  225. scipy/linalg/cython_lapack.cpython-313-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-313-aarch64-linux-musl.so +0 -0
  241. scipy/ndimage/_cytest.cpython-313-aarch64-linux-musl.so +0 -0
  242. scipy/ndimage/_delegators.py +8 -2
  243. scipy/ndimage/_filters.py +433 -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-313-aarch64-linux-musl.so +0 -0
  248. scipy/ndimage/_ni_docstrings.py +5 -1
  249. scipy/ndimage/_ni_label.cpython-313-aarch64-linux-musl.so +0 -0
  250. scipy/ndimage/_ni_support.py +1 -5
  251. scipy/ndimage/_rank_filter_1d.cpython-313-aarch64-linux-musl.so +0 -0
  252. scipy/ndimage/_support_alternative_backends.py +18 -6
  253. scipy/ndimage/tests/test_filters.py +337 -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-313-aarch64-linux-musl.so +0 -0
  260. scipy/optimize/_basinhopping.py +13 -7
  261. scipy/optimize/_bglu_dense.cpython-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  272. scipy/optimize/_highspy/_core.cpython-313-aarch64-linux-musl.so +0 -0
  273. scipy/optimize/_highspy/_highs_options.cpython-313-aarch64-linux-musl.so +0 -0
  274. scipy/optimize/_lbfgsb.cpython-313-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-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  290. scipy/optimize/_minpack_py.py +21 -14
  291. scipy/optimize/_moduleTNC.cpython-313-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-313-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-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  317. scipy/optimize/_zeros_py.py +97 -17
  318. scipy/optimize/cython_optimize/_zeros.cpython-313-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 +5 -5
  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-313-aarch64-linux-musl.so +0 -0
  350. scipy/signal/_peak_finding_utils.cpython-313-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-313-aarch64-linux-musl.so +0 -0
  356. scipy/signal/_sosfilt.cpython-313-aarch64-linux-musl.so +0 -0
  357. scipy/signal/_spectral_py.py +221 -50
  358. scipy/signal/_spline.cpython-313-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-313-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 +19 -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-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  404. scipy/sparse/csgraph/_matching.cpython-313-aarch64-linux-musl.so +0 -0
  405. scipy/sparse/csgraph/_min_spanning_tree.cpython-313-aarch64-linux-musl.so +0 -0
  406. scipy/sparse/csgraph/_reordering.cpython-313-aarch64-linux-musl.so +0 -0
  407. scipy/sparse/csgraph/_shortest_path.cpython-313-aarch64-linux-musl.so +0 -0
  408. scipy/sparse/csgraph/_tools.cpython-313-aarch64-linux-musl.so +0 -0
  409. scipy/sparse/csgraph/_traversal.cpython-313-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-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  433. scipy/sparse/linalg/_propack/_dpropack.cpython-313-aarch64-linux-musl.so +0 -0
  434. scipy/sparse/linalg/_propack/_spropack.cpython-313-aarch64-linux-musl.so +0 -0
  435. scipy/sparse/linalg/_propack/_zpropack.cpython-313-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 +207 -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-313-aarch64-linux-musl.so +0 -0
  446. scipy/spatial/_distance_pybind.cpython-313-aarch64-linux-musl.so +0 -0
  447. scipy/spatial/_distance_wrap.cpython-313-aarch64-linux-musl.so +0 -0
  448. scipy/spatial/_hausdorff.cpython-313-aarch64-linux-musl.so +0 -0
  449. scipy/spatial/_qhull.cpython-313-aarch64-linux-musl.so +0 -0
  450. scipy/spatial/_voronoi.cpython-313-aarch64-linux-musl.so +0 -0
  451. scipy/spatial/distance.py +49 -42
  452. scipy/spatial/tests/test_distance.py +3 -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-313-aarch64-linux-musl.so +0 -0
  457. scipy/spatial/transform/_rotation.cpython-313-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-313-aarch64-linux-musl.so +0 -0
  466. scipy/special/_ellip_harm_2.cpython-313-aarch64-linux-musl.so +0 -0
  467. scipy/special/_gufuncs.cpython-313-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-313-aarch64-linux-musl.so +0 -0
  471. scipy/special/_special_ufuncs.cpython-313-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-313-aarch64-linux-musl.so +0 -0
  475. scipy/special/_testutils.py +4 -4
  476. scipy/special/_ufuncs.cpython-313-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-313-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-313-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-313-aarch64-linux-musl.so +0 -0
  502. scipy/stats/_axis_nan_policy.py +4 -3
  503. scipy/stats/_biasedurn.cpython-313-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/_covariance.py +6 -3
  507. scipy/stats/_discrete_distns.py +39 -32
  508. scipy/stats/_distn_infrastructure.py +39 -12
  509. scipy/stats/_distribution_infrastructure.py +900 -238
  510. scipy/stats/_entropy.py +7 -8
  511. scipy/{_lib → stats}/_finite_differences.py +1 -1
  512. scipy/stats/_hypotests.py +82 -49
  513. scipy/stats/_kde.py +53 -49
  514. scipy/stats/_ksstats.py +1 -1
  515. scipy/stats/_levy_stable/__init__.py +7 -15
  516. scipy/stats/_levy_stable/levyst.cpython-313-aarch64-linux-musl.so +0 -0
  517. scipy/stats/_morestats.py +112 -67
  518. scipy/stats/_mstats_basic.py +13 -17
  519. scipy/stats/_mstats_extras.py +8 -8
  520. scipy/stats/_multivariate.py +89 -113
  521. scipy/stats/_new_distributions.py +97 -20
  522. scipy/stats/_page_trend_test.py +12 -5
  523. scipy/stats/_probability_distribution.py +265 -43
  524. scipy/stats/_qmc.py +14 -9
  525. scipy/stats/_qmc_cy.cpython-313-aarch64-linux-musl.so +0 -0
  526. scipy/stats/_qmvnt.py +16 -95
  527. scipy/stats/_qmvnt_cy.cpython-313-aarch64-linux-musl.so +0 -0
  528. scipy/stats/_quantile.py +335 -0
  529. scipy/stats/_rcont/rcont.cpython-313-aarch64-linux-musl.so +0 -0
  530. scipy/stats/_resampling.py +4 -29
  531. scipy/stats/_sampling.py +1 -1
  532. scipy/stats/_sobol.cpython-313-aarch64-linux-musl.so +0 -0
  533. scipy/stats/_stats.cpython-313-aarch64-linux-musl.so +0 -0
  534. scipy/stats/_stats_mstats_common.py +19 -2
  535. scipy/stats/_stats_py.py +534 -460
  536. scipy/stats/_stats_pythran.cpython-313-aarch64-linux-musl.so +0 -0
  537. scipy/stats/_unuran/unuran_wrapper.cpython-313-aarch64-linux-musl.so +0 -0
  538. scipy/stats/_unuran/unuran_wrapper.pyi +2 -1
  539. scipy/stats/_variation.py +5 -7
  540. scipy/stats/_wilcoxon.py +13 -7
  541. scipy/stats/tests/common_tests.py +6 -4
  542. scipy/stats/tests/test_axis_nan_policy.py +62 -24
  543. scipy/stats/tests/test_continued_fraction.py +173 -0
  544. scipy/stats/tests/test_continuous.py +379 -60
  545. scipy/stats/tests/test_continuous_basic.py +18 -12
  546. scipy/stats/tests/test_discrete_basic.py +14 -8
  547. scipy/stats/tests/test_discrete_distns.py +16 -16
  548. scipy/stats/tests/test_distributions.py +95 -75
  549. scipy/stats/tests/test_entropy.py +40 -48
  550. scipy/stats/tests/test_fit.py +4 -3
  551. scipy/stats/tests/test_hypotests.py +153 -24
  552. scipy/stats/tests/test_kdeoth.py +109 -41
  553. scipy/stats/tests/test_marray.py +289 -0
  554. scipy/stats/tests/test_morestats.py +79 -47
  555. scipy/stats/tests/test_mstats_basic.py +3 -3
  556. scipy/stats/tests/test_multivariate.py +434 -83
  557. scipy/stats/tests/test_qmc.py +13 -10
  558. scipy/stats/tests/test_quantile.py +199 -0
  559. scipy/stats/tests/test_rank.py +119 -112
  560. scipy/stats/tests/test_resampling.py +47 -56
  561. scipy/stats/tests/test_sampling.py +9 -4
  562. scipy/stats/tests/test_stats.py +799 -939
  563. scipy/stats/tests/test_variation.py +8 -6
  564. scipy/version.py +2 -2
  565. {scipy-1.15.3.dist-info → scipy-1.16.0rc1.dist-info}/LICENSE.txt +1 -1
  566. {scipy-1.15.3.dist-info → scipy-1.16.0rc1.dist-info}/METADATA +8 -8
  567. {scipy-1.15.3.dist-info → scipy-1.16.0rc1.dist-info}/RECORD +1262 -1269
  568. scipy.libs/libgcc_s-69c45f16.so.1 +0 -0
  569. scipy.libs/libgfortran-db0b6589.so.5.0.0 +0 -0
  570. scipy.libs/{libstdc++-1b614e01.so.6.0.32 → libstdc++-1f1a71be.so.6.0.33} +0 -0
  571. scipy/_lib/array_api_extra/_funcs.py +0 -484
  572. scipy/_lib/array_api_extra/_typing.py +0 -8
  573. scipy/interpolate/_bspl.cpython-313-aarch64-linux-musl.so +0 -0
  574. scipy/optimize/_cobyla.cpython-313-aarch64-linux-musl.so +0 -0
  575. scipy/optimize/_cython_nnls.cpython-313-aarch64-linux-musl.so +0 -0
  576. scipy/optimize/_slsqp.cpython-313-aarch64-linux-musl.so +0 -0
  577. scipy/spatial/qhull_src/COPYING.txt +0 -38
  578. scipy/special/libsf_error_state.so +0 -0
  579. scipy/special/tests/test_log_softmax.py +0 -109
  580. scipy/special/tests/test_xsf_cuda.py +0 -114
  581. scipy/special/xsf/binom.h +0 -89
  582. scipy/special/xsf/cdflib.h +0 -100
  583. scipy/special/xsf/cephes/airy.h +0 -307
  584. scipy/special/xsf/cephes/besselpoly.h +0 -51
  585. scipy/special/xsf/cephes/beta.h +0 -257
  586. scipy/special/xsf/cephes/cbrt.h +0 -131
  587. scipy/special/xsf/cephes/chbevl.h +0 -85
  588. scipy/special/xsf/cephes/chdtr.h +0 -193
  589. scipy/special/xsf/cephes/const.h +0 -87
  590. scipy/special/xsf/cephes/ellie.h +0 -293
  591. scipy/special/xsf/cephes/ellik.h +0 -251
  592. scipy/special/xsf/cephes/ellpe.h +0 -107
  593. scipy/special/xsf/cephes/ellpk.h +0 -117
  594. scipy/special/xsf/cephes/expn.h +0 -260
  595. scipy/special/xsf/cephes/gamma.h +0 -398
  596. scipy/special/xsf/cephes/hyp2f1.h +0 -596
  597. scipy/special/xsf/cephes/hyperg.h +0 -361
  598. scipy/special/xsf/cephes/i0.h +0 -149
  599. scipy/special/xsf/cephes/i1.h +0 -158
  600. scipy/special/xsf/cephes/igam.h +0 -421
  601. scipy/special/xsf/cephes/igam_asymp_coeff.h +0 -195
  602. scipy/special/xsf/cephes/igami.h +0 -313
  603. scipy/special/xsf/cephes/j0.h +0 -225
  604. scipy/special/xsf/cephes/j1.h +0 -198
  605. scipy/special/xsf/cephes/jv.h +0 -715
  606. scipy/special/xsf/cephes/k0.h +0 -164
  607. scipy/special/xsf/cephes/k1.h +0 -163
  608. scipy/special/xsf/cephes/kn.h +0 -243
  609. scipy/special/xsf/cephes/lanczos.h +0 -112
  610. scipy/special/xsf/cephes/ndtr.h +0 -275
  611. scipy/special/xsf/cephes/poch.h +0 -85
  612. scipy/special/xsf/cephes/polevl.h +0 -167
  613. scipy/special/xsf/cephes/psi.h +0 -194
  614. scipy/special/xsf/cephes/rgamma.h +0 -111
  615. scipy/special/xsf/cephes/scipy_iv.h +0 -811
  616. scipy/special/xsf/cephes/shichi.h +0 -248
  617. scipy/special/xsf/cephes/sici.h +0 -224
  618. scipy/special/xsf/cephes/sindg.h +0 -221
  619. scipy/special/xsf/cephes/tandg.h +0 -139
  620. scipy/special/xsf/cephes/trig.h +0 -58
  621. scipy/special/xsf/cephes/unity.h +0 -186
  622. scipy/special/xsf/cephes/zeta.h +0 -172
  623. scipy/special/xsf/config.h +0 -304
  624. scipy/special/xsf/digamma.h +0 -205
  625. scipy/special/xsf/error.h +0 -57
  626. scipy/special/xsf/evalpoly.h +0 -47
  627. scipy/special/xsf/expint.h +0 -266
  628. scipy/special/xsf/hyp2f1.h +0 -694
  629. scipy/special/xsf/iv_ratio.h +0 -173
  630. scipy/special/xsf/lambertw.h +0 -150
  631. scipy/special/xsf/loggamma.h +0 -163
  632. scipy/special/xsf/sici.h +0 -200
  633. scipy/special/xsf/tools.h +0 -427
  634. scipy/special/xsf/trig.h +0 -164
  635. scipy/special/xsf/wright_bessel.h +0 -843
  636. scipy/special/xsf/zlog1.h +0 -35
  637. scipy/stats/_mvn.cpython-313-aarch64-linux-musl.so +0 -0
  638. scipy.libs/libgcc_s-7393e603.so.1 +0 -0
  639. scipy.libs/libgfortran-eb933d8e.so.5.0.0 +0 -0
  640. {scipy-1.15.3.dist-info → scipy-1.16.0rc1.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
- }