scipy 1.15.3__cp312-cp312-macosx_12_0_arm64.whl → 1.16.0rc2__cp312-cp312-macosx_12_0_arm64.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.
- scipy/.dylibs/libscipy_openblas.dylib +0 -0
- scipy/__config__.py +8 -8
- scipy/__init__.py +3 -6
- scipy/_cyutility.cpython-312-darwin.so +0 -0
- scipy/_lib/_array_api.py +486 -161
- scipy/_lib/_array_api_compat_vendor.py +9 -0
- scipy/_lib/_bunch.py +4 -0
- scipy/_lib/_ccallback_c.cpython-312-darwin.so +0 -0
- scipy/_lib/_docscrape.py +1 -1
- scipy/_lib/_elementwise_iterative_method.py +15 -26
- scipy/_lib/_sparse.py +41 -0
- scipy/_lib/_test_deprecation_call.cpython-312-darwin.so +0 -0
- scipy/_lib/_test_deprecation_def.cpython-312-darwin.so +0 -0
- scipy/_lib/_testutils.py +6 -2
- scipy/_lib/_util.py +222 -125
- scipy/_lib/array_api_compat/__init__.py +4 -4
- scipy/_lib/array_api_compat/_internal.py +19 -6
- scipy/_lib/array_api_compat/common/__init__.py +1 -1
- scipy/_lib/array_api_compat/common/_aliases.py +365 -193
- scipy/_lib/array_api_compat/common/_fft.py +94 -64
- scipy/_lib/array_api_compat/common/_helpers.py +413 -180
- scipy/_lib/array_api_compat/common/_linalg.py +116 -40
- scipy/_lib/array_api_compat/common/_typing.py +179 -10
- scipy/_lib/array_api_compat/cupy/__init__.py +1 -4
- scipy/_lib/array_api_compat/cupy/_aliases.py +61 -41
- scipy/_lib/array_api_compat/cupy/_info.py +16 -6
- scipy/_lib/array_api_compat/cupy/_typing.py +24 -39
- scipy/_lib/array_api_compat/dask/array/__init__.py +6 -3
- scipy/_lib/array_api_compat/dask/array/_aliases.py +267 -108
- scipy/_lib/array_api_compat/dask/array/_info.py +105 -34
- scipy/_lib/array_api_compat/dask/array/fft.py +5 -8
- scipy/_lib/array_api_compat/dask/array/linalg.py +21 -22
- scipy/_lib/array_api_compat/numpy/__init__.py +13 -15
- scipy/_lib/array_api_compat/numpy/_aliases.py +98 -49
- scipy/_lib/array_api_compat/numpy/_info.py +36 -16
- scipy/_lib/array_api_compat/numpy/_typing.py +27 -43
- scipy/_lib/array_api_compat/numpy/fft.py +11 -5
- scipy/_lib/array_api_compat/numpy/linalg.py +75 -22
- scipy/_lib/array_api_compat/torch/__init__.py +3 -5
- scipy/_lib/array_api_compat/torch/_aliases.py +262 -159
- scipy/_lib/array_api_compat/torch/_info.py +27 -16
- scipy/_lib/array_api_compat/torch/_typing.py +3 -0
- scipy/_lib/array_api_compat/torch/fft.py +17 -18
- scipy/_lib/array_api_compat/torch/linalg.py +16 -16
- scipy/_lib/array_api_extra/__init__.py +26 -3
- scipy/_lib/array_api_extra/_delegation.py +171 -0
- scipy/_lib/array_api_extra/_lib/__init__.py +1 -0
- scipy/_lib/array_api_extra/_lib/_at.py +463 -0
- scipy/_lib/array_api_extra/_lib/_backends.py +46 -0
- scipy/_lib/array_api_extra/_lib/_funcs.py +937 -0
- scipy/_lib/array_api_extra/_lib/_lazy.py +357 -0
- scipy/_lib/array_api_extra/_lib/_testing.py +278 -0
- scipy/_lib/array_api_extra/_lib/_utils/__init__.py +1 -0
- scipy/_lib/array_api_extra/_lib/_utils/_compat.py +74 -0
- scipy/_lib/array_api_extra/_lib/_utils/_compat.pyi +45 -0
- scipy/_lib/array_api_extra/_lib/_utils/_helpers.py +559 -0
- scipy/_lib/array_api_extra/_lib/_utils/_typing.py +10 -0
- scipy/_lib/array_api_extra/_lib/_utils/_typing.pyi +105 -0
- scipy/_lib/array_api_extra/testing.py +359 -0
- scipy/_lib/decorator.py +2 -2
- scipy/_lib/doccer.py +1 -7
- scipy/_lib/messagestream.cpython-312-darwin.so +0 -0
- scipy/_lib/pyprima/__init__.py +212 -0
- scipy/_lib/pyprima/cobyla/__init__.py +0 -0
- scipy/_lib/pyprima/cobyla/cobyla.py +559 -0
- scipy/_lib/pyprima/cobyla/cobylb.py +714 -0
- scipy/_lib/pyprima/cobyla/geometry.py +226 -0
- scipy/_lib/pyprima/cobyla/initialize.py +215 -0
- scipy/_lib/pyprima/cobyla/trustregion.py +492 -0
- scipy/_lib/pyprima/cobyla/update.py +289 -0
- scipy/_lib/pyprima/common/__init__.py +0 -0
- scipy/_lib/pyprima/common/_bounds.py +34 -0
- scipy/_lib/pyprima/common/_linear_constraints.py +46 -0
- scipy/_lib/pyprima/common/_nonlinear_constraints.py +54 -0
- scipy/_lib/pyprima/common/_project.py +173 -0
- scipy/_lib/pyprima/common/checkbreak.py +93 -0
- scipy/_lib/pyprima/common/consts.py +47 -0
- scipy/_lib/pyprima/common/evaluate.py +99 -0
- scipy/_lib/pyprima/common/history.py +38 -0
- scipy/_lib/pyprima/common/infos.py +30 -0
- scipy/_lib/pyprima/common/linalg.py +435 -0
- scipy/_lib/pyprima/common/message.py +290 -0
- scipy/_lib/pyprima/common/powalg.py +131 -0
- scipy/_lib/pyprima/common/preproc.py +277 -0
- scipy/_lib/pyprima/common/present.py +5 -0
- scipy/_lib/pyprima/common/ratio.py +54 -0
- scipy/_lib/pyprima/common/redrho.py +47 -0
- scipy/_lib/pyprima/common/selectx.py +296 -0
- scipy/_lib/tests/test__util.py +105 -121
- scipy/_lib/tests/test_array_api.py +166 -35
- scipy/_lib/tests/test_bunch.py +7 -0
- scipy/_lib/tests/test_ccallback.py +2 -10
- scipy/_lib/tests/test_public_api.py +13 -0
- scipy/cluster/_hierarchy.cpython-312-darwin.so +0 -0
- scipy/cluster/_optimal_leaf_ordering.cpython-312-darwin.so +0 -0
- scipy/cluster/_vq.cpython-312-darwin.so +0 -0
- scipy/cluster/hierarchy.py +393 -223
- scipy/cluster/tests/test_hierarchy.py +273 -335
- scipy/cluster/tests/test_vq.py +45 -61
- scipy/cluster/vq.py +39 -35
- scipy/conftest.py +263 -157
- scipy/constants/_constants.py +4 -1
- scipy/constants/tests/test_codata.py +2 -2
- scipy/constants/tests/test_constants.py +11 -18
- scipy/datasets/_download_all.py +15 -1
- scipy/datasets/_fetchers.py +7 -1
- scipy/datasets/_utils.py +1 -1
- scipy/differentiate/_differentiate.py +25 -25
- scipy/differentiate/tests/test_differentiate.py +24 -25
- scipy/fft/_basic.py +20 -0
- scipy/fft/_helper.py +3 -34
- scipy/fft/_pocketfft/helper.py +29 -1
- scipy/fft/_pocketfft/tests/test_basic.py +2 -4
- scipy/fft/_pocketfft/tests/test_real_transforms.py +4 -4
- scipy/fft/_realtransforms.py +13 -0
- scipy/fft/tests/test_basic.py +27 -25
- scipy/fft/tests/test_fftlog.py +16 -7
- scipy/fft/tests/test_helper.py +18 -34
- scipy/fft/tests/test_real_transforms.py +8 -10
- scipy/fftpack/convolve.cpython-312-darwin.so +0 -0
- scipy/fftpack/tests/test_basic.py +2 -4
- scipy/fftpack/tests/test_real_transforms.py +8 -9
- scipy/integrate/_bvp.py +9 -3
- scipy/integrate/_cubature.py +3 -2
- scipy/integrate/_dop.cpython-312-darwin.so +0 -0
- scipy/integrate/_lsoda.cpython-312-darwin.so +0 -0
- scipy/integrate/_ode.py +9 -2
- scipy/integrate/_odepack.cpython-312-darwin.so +0 -0
- scipy/integrate/_quad_vec.py +21 -29
- scipy/integrate/_quadpack.cpython-312-darwin.so +0 -0
- scipy/integrate/_quadpack_py.py +11 -7
- scipy/integrate/_quadrature.py +3 -3
- scipy/integrate/_rules/_base.py +2 -2
- scipy/integrate/_tanhsinh.py +48 -47
- scipy/integrate/_test_odeint_banded.cpython-312-darwin.so +0 -0
- scipy/integrate/_vode.cpython-312-darwin.so +0 -0
- scipy/integrate/tests/test__quad_vec.py +0 -6
- scipy/integrate/tests/test_banded_ode_solvers.py +85 -0
- scipy/integrate/tests/test_cubature.py +21 -35
- scipy/integrate/tests/test_quadrature.py +6 -8
- scipy/integrate/tests/test_tanhsinh.py +56 -48
- scipy/interpolate/__init__.py +70 -58
- scipy/interpolate/_bary_rational.py +22 -22
- scipy/interpolate/_bsplines.py +119 -66
- scipy/interpolate/_cubic.py +65 -50
- scipy/interpolate/_dfitpack.cpython-312-darwin.so +0 -0
- scipy/interpolate/_dierckx.cpython-312-darwin.so +0 -0
- scipy/interpolate/_fitpack.cpython-312-darwin.so +0 -0
- scipy/interpolate/_fitpack2.py +9 -6
- scipy/interpolate/_fitpack_impl.py +32 -26
- scipy/interpolate/_fitpack_repro.py +23 -19
- scipy/interpolate/_interpnd.cpython-312-darwin.so +0 -0
- scipy/interpolate/_interpolate.py +30 -12
- scipy/interpolate/_ndbspline.py +13 -18
- scipy/interpolate/_ndgriddata.py +5 -8
- scipy/interpolate/_polyint.py +95 -31
- scipy/interpolate/_ppoly.cpython-312-darwin.so +0 -0
- scipy/interpolate/_rbf.py +2 -2
- scipy/interpolate/_rbfinterp.py +1 -1
- scipy/interpolate/_rbfinterp_pythran.cpython-312-darwin.so +0 -0
- scipy/interpolate/_rgi.py +31 -26
- scipy/interpolate/_rgi_cython.cpython-312-darwin.so +0 -0
- scipy/interpolate/dfitpack.py +0 -20
- scipy/interpolate/interpnd.py +1 -2
- scipy/interpolate/tests/test_bary_rational.py +2 -2
- scipy/interpolate/tests/test_bsplines.py +97 -1
- scipy/interpolate/tests/test_fitpack2.py +39 -1
- scipy/interpolate/tests/test_interpnd.py +32 -20
- scipy/interpolate/tests/test_interpolate.py +48 -4
- scipy/interpolate/tests/test_rgi.py +2 -1
- scipy/io/_fast_matrix_market/__init__.py +2 -0
- scipy/io/_harwell_boeing/_fortran_format_parser.py +19 -16
- scipy/io/_harwell_boeing/hb.py +7 -11
- scipy/io/_idl.py +5 -7
- scipy/io/_netcdf.py +15 -5
- scipy/io/_test_fortran.cpython-312-darwin.so +0 -0
- scipy/io/arff/tests/test_arffread.py +3 -3
- scipy/io/matlab/__init__.py +5 -3
- scipy/io/matlab/_mio.py +4 -1
- scipy/io/matlab/_mio5.py +19 -13
- scipy/io/matlab/_mio5_utils.cpython-312-darwin.so +0 -0
- scipy/io/matlab/_mio_utils.cpython-312-darwin.so +0 -0
- scipy/io/matlab/_miobase.py +4 -1
- scipy/io/matlab/_streams.cpython-312-darwin.so +0 -0
- scipy/io/matlab/tests/test_mio.py +46 -18
- scipy/io/matlab/tests/test_mio_funcs.py +1 -1
- scipy/io/tests/test_mmio.py +7 -1
- scipy/io/tests/test_wavfile.py +41 -0
- scipy/io/wavfile.py +57 -10
- scipy/linalg/_basic.py +113 -86
- scipy/linalg/_cythonized_array_utils.cpython-312-darwin.so +0 -0
- scipy/linalg/_decomp.py +22 -9
- scipy/linalg/_decomp_cholesky.py +28 -13
- scipy/linalg/_decomp_cossin.py +45 -30
- scipy/linalg/_decomp_interpolative.cpython-312-darwin.so +0 -0
- scipy/linalg/_decomp_ldl.py +4 -1
- scipy/linalg/_decomp_lu.py +18 -6
- scipy/linalg/_decomp_lu_cython.cpython-312-darwin.so +0 -0
- scipy/linalg/_decomp_polar.py +2 -0
- scipy/linalg/_decomp_qr.py +6 -2
- scipy/linalg/_decomp_qz.py +3 -0
- scipy/linalg/_decomp_schur.py +3 -1
- scipy/linalg/_decomp_svd.py +13 -2
- scipy/linalg/_decomp_update.cpython-312-darwin.so +0 -0
- scipy/linalg/_expm_frechet.py +4 -0
- scipy/linalg/_fblas.cpython-312-darwin.so +0 -0
- scipy/linalg/_flapack.cpython-312-darwin.so +0 -0
- scipy/linalg/_linalg_pythran.cpython-312-darwin.so +0 -0
- scipy/linalg/_matfuncs.py +187 -4
- scipy/linalg/_matfuncs_expm.cpython-312-darwin.so +0 -0
- scipy/linalg/_matfuncs_schur_sqrtm.cpython-312-darwin.so +0 -0
- scipy/linalg/_matfuncs_sqrtm.py +1 -99
- scipy/linalg/_matfuncs_sqrtm_triu.cpython-312-darwin.so +0 -0
- scipy/linalg/_procrustes.py +2 -0
- scipy/linalg/_sketches.py +17 -6
- scipy/linalg/_solve_toeplitz.cpython-312-darwin.so +0 -0
- scipy/linalg/_solvers.py +7 -2
- scipy/linalg/_special_matrices.py +26 -36
- scipy/linalg/cython_blas.cpython-312-darwin.so +0 -0
- scipy/linalg/cython_lapack.cpython-312-darwin.so +0 -0
- scipy/linalg/lapack.py +22 -2
- scipy/linalg/tests/_cython_examples/meson.build +7 -0
- scipy/linalg/tests/test_basic.py +31 -16
- scipy/linalg/tests/test_batch.py +588 -0
- scipy/linalg/tests/test_cythonized_array_utils.py +0 -2
- scipy/linalg/tests/test_decomp.py +40 -3
- scipy/linalg/tests/test_decomp_cossin.py +14 -0
- scipy/linalg/tests/test_decomp_ldl.py +1 -1
- scipy/linalg/tests/test_lapack.py +115 -7
- scipy/linalg/tests/test_matfuncs.py +157 -102
- scipy/linalg/tests/test_procrustes.py +0 -7
- scipy/linalg/tests/test_solve_toeplitz.py +1 -1
- scipy/linalg/tests/test_special_matrices.py +1 -5
- scipy/ndimage/__init__.py +1 -0
- scipy/ndimage/_cytest.cpython-312-darwin.so +0 -0
- scipy/ndimage/_delegators.py +8 -2
- scipy/ndimage/_filters.py +453 -5
- scipy/ndimage/_interpolation.py +36 -6
- scipy/ndimage/_measurements.py +4 -2
- scipy/ndimage/_morphology.py +5 -0
- scipy/ndimage/_nd_image.cpython-312-darwin.so +0 -0
- scipy/ndimage/_ni_docstrings.py +5 -1
- scipy/ndimage/_ni_label.cpython-312-darwin.so +0 -0
- scipy/ndimage/_ni_support.py +1 -5
- scipy/ndimage/_rank_filter_1d.cpython-312-darwin.so +0 -0
- scipy/ndimage/_support_alternative_backends.py +18 -6
- scipy/ndimage/tests/test_filters.py +370 -259
- scipy/ndimage/tests/test_fourier.py +7 -9
- scipy/ndimage/tests/test_interpolation.py +68 -61
- scipy/ndimage/tests/test_measurements.py +18 -35
- scipy/ndimage/tests/test_morphology.py +143 -131
- scipy/ndimage/tests/test_splines.py +1 -3
- scipy/odr/__odrpack.cpython-312-darwin.so +0 -0
- scipy/optimize/_basinhopping.py +13 -7
- scipy/optimize/_bglu_dense.cpython-312-darwin.so +0 -0
- scipy/optimize/_bracket.py +17 -24
- scipy/optimize/_chandrupatla.py +9 -10
- scipy/optimize/_cobyla_py.py +104 -123
- scipy/optimize/_constraints.py +14 -10
- scipy/optimize/_differentiable_functions.py +371 -230
- scipy/optimize/_differentialevolution.py +4 -3
- scipy/optimize/_direct.cpython-312-darwin.so +0 -0
- scipy/optimize/_dual_annealing.py +1 -1
- scipy/optimize/_elementwise.py +1 -4
- scipy/optimize/_group_columns.cpython-312-darwin.so +0 -0
- scipy/optimize/_lbfgsb.cpython-312-darwin.so +0 -0
- scipy/optimize/_lbfgsb_py.py +57 -16
- scipy/optimize/_linprog_doc.py +2 -2
- scipy/optimize/_linprog_highs.py +2 -2
- scipy/optimize/_linprog_ip.py +25 -10
- scipy/optimize/_linprog_util.py +14 -16
- scipy/optimize/_lsap.cpython-312-darwin.so +0 -0
- scipy/optimize/_lsq/common.py +3 -3
- scipy/optimize/_lsq/dogbox.py +16 -2
- scipy/optimize/_lsq/givens_elimination.cpython-312-darwin.so +0 -0
- scipy/optimize/_lsq/least_squares.py +198 -126
- scipy/optimize/_lsq/lsq_linear.py +6 -6
- scipy/optimize/_lsq/trf.py +35 -8
- scipy/optimize/_milp.py +3 -1
- scipy/optimize/_minimize.py +105 -36
- scipy/optimize/_minpack.cpython-312-darwin.so +0 -0
- scipy/optimize/_minpack_py.py +21 -14
- scipy/optimize/_moduleTNC.cpython-312-darwin.so +0 -0
- scipy/optimize/_nnls.py +20 -21
- scipy/optimize/_nonlin.py +34 -3
- scipy/optimize/_numdiff.py +288 -110
- scipy/optimize/_optimize.py +86 -48
- scipy/optimize/_pava_pybind.cpython-312-darwin.so +0 -0
- scipy/optimize/_remove_redundancy.py +5 -5
- scipy/optimize/_root_scalar.py +1 -1
- scipy/optimize/_shgo.py +6 -0
- scipy/optimize/_shgo_lib/_complex.py +1 -1
- scipy/optimize/_slsqp_py.py +216 -124
- scipy/optimize/_slsqplib.cpython-312-darwin.so +0 -0
- scipy/optimize/_spectral.py +1 -1
- scipy/optimize/_tnc.py +8 -1
- scipy/optimize/_trlib/_trlib.cpython-312-darwin.so +0 -0
- scipy/optimize/_trustregion.py +20 -6
- scipy/optimize/_trustregion_constr/canonical_constraint.py +7 -7
- scipy/optimize/_trustregion_constr/equality_constrained_sqp.py +1 -1
- scipy/optimize/_trustregion_constr/minimize_trustregion_constr.py +11 -3
- scipy/optimize/_trustregion_constr/projections.py +12 -8
- scipy/optimize/_trustregion_constr/qp_subproblem.py +9 -9
- scipy/optimize/_trustregion_constr/tests/test_projections.py +7 -7
- scipy/optimize/_trustregion_constr/tests/test_qp_subproblem.py +77 -77
- scipy/optimize/_trustregion_constr/tr_interior_point.py +5 -5
- scipy/optimize/_trustregion_exact.py +0 -1
- scipy/optimize/_zeros.cpython-312-darwin.so +0 -0
- scipy/optimize/_zeros_py.py +97 -17
- scipy/optimize/cython_optimize/_zeros.cpython-312-darwin.so +0 -0
- scipy/optimize/slsqp.py +0 -1
- scipy/optimize/tests/test__basinhopping.py +1 -1
- scipy/optimize/tests/test__differential_evolution.py +4 -4
- scipy/optimize/tests/test__linprog_clean_inputs.py +5 -3
- scipy/optimize/tests/test__numdiff.py +66 -22
- scipy/optimize/tests/test__remove_redundancy.py +2 -2
- scipy/optimize/tests/test__shgo.py +9 -1
- scipy/optimize/tests/test_bracket.py +36 -46
- scipy/optimize/tests/test_chandrupatla.py +133 -135
- scipy/optimize/tests/test_cobyla.py +74 -45
- scipy/optimize/tests/test_constraints.py +1 -1
- scipy/optimize/tests/test_differentiable_functions.py +226 -6
- scipy/optimize/tests/test_lbfgsb_hessinv.py +22 -0
- scipy/optimize/tests/test_least_squares.py +125 -13
- scipy/optimize/tests/test_linear_assignment.py +3 -3
- scipy/optimize/tests/test_linprog.py +3 -3
- scipy/optimize/tests/test_lsq_linear.py +6 -6
- scipy/optimize/tests/test_minimize_constrained.py +2 -2
- scipy/optimize/tests/test_minpack.py +4 -4
- scipy/optimize/tests/test_nnls.py +43 -3
- scipy/optimize/tests/test_nonlin.py +36 -0
- scipy/optimize/tests/test_optimize.py +95 -17
- scipy/optimize/tests/test_slsqp.py +36 -4
- scipy/optimize/tests/test_zeros.py +34 -1
- scipy/signal/__init__.py +12 -23
- scipy/signal/_delegators.py +568 -0
- scipy/signal/_filter_design.py +459 -241
- scipy/signal/_fir_filter_design.py +262 -90
- scipy/signal/_lti_conversion.py +3 -2
- scipy/signal/_ltisys.py +118 -91
- scipy/signal/_max_len_seq_inner.cpython-312-darwin.so +0 -0
- scipy/signal/_peak_finding_utils.cpython-312-darwin.so +0 -0
- scipy/signal/_polyutils.py +172 -0
- scipy/signal/_short_time_fft.py +519 -70
- scipy/signal/_signal_api.py +30 -0
- scipy/signal/_signaltools.py +719 -399
- scipy/signal/_sigtools.cpython-312-darwin.so +0 -0
- scipy/signal/_sosfilt.cpython-312-darwin.so +0 -0
- scipy/signal/_spectral_py.py +230 -50
- scipy/signal/_spline.cpython-312-darwin.so +0 -0
- scipy/signal/_spline_filters.py +108 -68
- scipy/signal/_support_alternative_backends.py +73 -0
- scipy/signal/_upfirdn.py +4 -1
- scipy/signal/_upfirdn_apply.cpython-312-darwin.so +0 -0
- scipy/signal/_waveforms.py +2 -11
- scipy/signal/_wavelets.py +1 -1
- scipy/signal/fir_filter_design.py +1 -0
- scipy/signal/spline.py +4 -11
- scipy/signal/tests/_scipy_spectral_test_shim.py +2 -171
- scipy/signal/tests/test_bsplines.py +114 -79
- scipy/signal/tests/test_cont2discrete.py +9 -2
- scipy/signal/tests/test_filter_design.py +721 -481
- scipy/signal/tests/test_fir_filter_design.py +332 -140
- scipy/signal/tests/test_savitzky_golay.py +4 -3
- scipy/signal/tests/test_short_time_fft.py +221 -3
- scipy/signal/tests/test_signaltools.py +2144 -1348
- scipy/signal/tests/test_spectral.py +50 -6
- scipy/signal/tests/test_splines.py +161 -96
- scipy/signal/tests/test_upfirdn.py +84 -50
- scipy/signal/tests/test_waveforms.py +20 -0
- scipy/signal/tests/test_windows.py +607 -466
- scipy/signal/windows/_windows.py +287 -148
- scipy/sparse/__init__.py +23 -4
- scipy/sparse/_base.py +270 -108
- scipy/sparse/_bsr.py +7 -4
- scipy/sparse/_compressed.py +59 -231
- scipy/sparse/_construct.py +90 -38
- scipy/sparse/_coo.py +115 -181
- scipy/sparse/_csc.py +4 -4
- scipy/sparse/_csparsetools.cpython-312-darwin.so +0 -0
- scipy/sparse/_csr.py +2 -2
- scipy/sparse/_data.py +48 -48
- scipy/sparse/_dia.py +105 -18
- scipy/sparse/_dok.py +0 -23
- scipy/sparse/_index.py +4 -4
- scipy/sparse/_matrix.py +23 -0
- scipy/sparse/_sparsetools.cpython-312-darwin.so +0 -0
- scipy/sparse/_sputils.py +37 -22
- scipy/sparse/base.py +0 -9
- scipy/sparse/bsr.py +0 -14
- scipy/sparse/compressed.py +0 -23
- scipy/sparse/construct.py +0 -6
- scipy/sparse/coo.py +0 -14
- scipy/sparse/csc.py +0 -3
- scipy/sparse/csgraph/_flow.cpython-312-darwin.so +0 -0
- scipy/sparse/csgraph/_matching.cpython-312-darwin.so +0 -0
- scipy/sparse/csgraph/_min_spanning_tree.cpython-312-darwin.so +0 -0
- scipy/sparse/csgraph/_reordering.cpython-312-darwin.so +0 -0
- scipy/sparse/csgraph/_shortest_path.cpython-312-darwin.so +0 -0
- scipy/sparse/csgraph/_tools.cpython-312-darwin.so +0 -0
- scipy/sparse/csgraph/_traversal.cpython-312-darwin.so +0 -0
- scipy/sparse/csgraph/tests/test_matching.py +14 -2
- scipy/sparse/csgraph/tests/test_pydata_sparse.py +4 -1
- scipy/sparse/csgraph/tests/test_shortest_path.py +83 -27
- scipy/sparse/csr.py +0 -5
- scipy/sparse/data.py +1 -6
- scipy/sparse/dia.py +0 -7
- scipy/sparse/dok.py +0 -10
- scipy/sparse/linalg/_dsolve/_superlu.cpython-312-darwin.so +0 -0
- scipy/sparse/linalg/_dsolve/linsolve.py +9 -0
- scipy/sparse/linalg/_dsolve/tests/test_linsolve.py +35 -28
- scipy/sparse/linalg/_eigen/arpack/_arpack.cpython-312-darwin.so +0 -0
- scipy/sparse/linalg/_eigen/arpack/arpack.py +23 -17
- scipy/sparse/linalg/_eigen/lobpcg/lobpcg.py +6 -6
- scipy/sparse/linalg/_interface.py +17 -18
- scipy/sparse/linalg/_isolve/_gcrotmk.py +4 -4
- scipy/sparse/linalg/_isolve/iterative.py +51 -45
- scipy/sparse/linalg/_isolve/lgmres.py +6 -6
- scipy/sparse/linalg/_isolve/minres.py +5 -5
- scipy/sparse/linalg/_isolve/tfqmr.py +7 -7
- scipy/sparse/linalg/_isolve/utils.py +2 -8
- scipy/sparse/linalg/_matfuncs.py +1 -1
- scipy/sparse/linalg/_norm.py +1 -1
- scipy/sparse/linalg/_propack/_cpropack.cpython-312-darwin.so +0 -0
- scipy/sparse/linalg/_propack/_dpropack.cpython-312-darwin.so +0 -0
- scipy/sparse/linalg/_propack/_spropack.cpython-312-darwin.so +0 -0
- scipy/sparse/linalg/_propack/_zpropack.cpython-312-darwin.so +0 -0
- scipy/sparse/linalg/_special_sparse_arrays.py +39 -38
- scipy/sparse/linalg/tests/test_pydata_sparse.py +14 -0
- scipy/sparse/tests/test_arithmetic1d.py +5 -2
- scipy/sparse/tests/test_base.py +214 -42
- scipy/sparse/tests/test_common1d.py +7 -7
- scipy/sparse/tests/test_construct.py +1 -1
- scipy/sparse/tests/test_coo.py +272 -4
- scipy/sparse/tests/test_sparsetools.py +5 -0
- scipy/sparse/tests/test_sputils.py +36 -7
- scipy/spatial/_ckdtree.cpython-312-darwin.so +0 -0
- scipy/spatial/_distance_pybind.cpython-312-darwin.so +0 -0
- scipy/spatial/_distance_wrap.cpython-312-darwin.so +0 -0
- scipy/spatial/_hausdorff.cpython-312-darwin.so +0 -0
- scipy/spatial/_qhull.cpython-312-darwin.so +0 -0
- scipy/spatial/_voronoi.cpython-312-darwin.so +0 -0
- scipy/spatial/distance.py +49 -42
- scipy/spatial/tests/test_distance.py +15 -1
- scipy/spatial/tests/test_kdtree.py +1 -0
- scipy/spatial/tests/test_qhull.py +7 -2
- scipy/spatial/transform/__init__.py +5 -3
- scipy/spatial/transform/_rigid_transform.cpython-312-darwin.so +0 -0
- scipy/spatial/transform/_rotation.cpython-312-darwin.so +0 -0
- scipy/spatial/transform/tests/test_rigid_transform.py +1221 -0
- scipy/spatial/transform/tests/test_rotation.py +1213 -832
- scipy/spatial/transform/tests/test_rotation_groups.py +3 -3
- scipy/spatial/transform/tests/test_rotation_spline.py +29 -8
- scipy/special/__init__.py +1 -47
- scipy/special/_add_newdocs.py +34 -772
- scipy/special/_basic.py +22 -25
- scipy/special/_comb.cpython-312-darwin.so +0 -0
- scipy/special/_ellip_harm_2.cpython-312-darwin.so +0 -0
- scipy/special/_gufuncs.cpython-312-darwin.so +0 -0
- scipy/special/_logsumexp.py +67 -58
- scipy/special/_orthogonal.pyi +1 -1
- scipy/special/_specfun.cpython-312-darwin.so +0 -0
- scipy/special/_special_ufuncs.cpython-312-darwin.so +0 -0
- scipy/special/_spherical_bessel.py +4 -4
- scipy/special/_support_alternative_backends.py +212 -119
- scipy/special/_test_internal.cpython-312-darwin.so +0 -0
- scipy/special/_testutils.py +4 -4
- scipy/special/_ufuncs.cpython-312-darwin.so +0 -0
- scipy/special/_ufuncs.pyi +1 -0
- scipy/special/_ufuncs.pyx +215 -1400
- scipy/special/_ufuncs_cxx.cpython-312-darwin.so +0 -0
- scipy/special/_ufuncs_cxx.pxd +2 -15
- scipy/special/_ufuncs_cxx.pyx +5 -44
- scipy/special/_ufuncs_cxx_defs.h +2 -16
- scipy/special/_ufuncs_defs.h +0 -8
- scipy/special/cython_special.cpython-312-darwin.so +0 -0
- scipy/special/cython_special.pxd +1 -1
- scipy/special/tests/_cython_examples/meson.build +10 -1
- scipy/special/tests/test_basic.py +153 -20
- scipy/special/tests/test_boost_ufuncs.py +3 -0
- scipy/special/tests/test_cdflib.py +35 -11
- scipy/special/tests/test_gammainc.py +16 -0
- scipy/special/tests/test_hyp2f1.py +2 -2
- scipy/special/tests/test_log1mexp.py +85 -0
- scipy/special/tests/test_logsumexp.py +206 -64
- scipy/special/tests/test_mpmath.py +1 -0
- scipy/special/tests/test_nan_inputs.py +1 -1
- scipy/special/tests/test_orthogonal.py +17 -18
- scipy/special/tests/test_sf_error.py +3 -2
- scipy/special/tests/test_sph_harm.py +6 -7
- scipy/special/tests/test_support_alternative_backends.py +211 -76
- scipy/stats/__init__.py +4 -1
- scipy/stats/_ansari_swilk_statistics.cpython-312-darwin.so +0 -0
- scipy/stats/_axis_nan_policy.py +5 -12
- scipy/stats/_biasedurn.cpython-312-darwin.so +0 -0
- scipy/stats/_continued_fraction.py +387 -0
- scipy/stats/_continuous_distns.py +277 -310
- scipy/stats/_correlation.py +1 -1
- scipy/stats/_covariance.py +6 -3
- scipy/stats/_discrete_distns.py +39 -32
- scipy/stats/_distn_infrastructure.py +39 -12
- scipy/stats/_distribution_infrastructure.py +900 -238
- scipy/stats/_entropy.py +9 -10
- scipy/{_lib → stats}/_finite_differences.py +1 -1
- scipy/stats/_hypotests.py +83 -50
- scipy/stats/_kde.py +53 -49
- scipy/stats/_ksstats.py +1 -1
- scipy/stats/_levy_stable/__init__.py +7 -15
- scipy/stats/_levy_stable/levyst.cpython-312-darwin.so +0 -0
- scipy/stats/_morestats.py +118 -73
- scipy/stats/_mstats_basic.py +13 -17
- scipy/stats/_mstats_extras.py +8 -8
- scipy/stats/_multivariate.py +89 -113
- scipy/stats/_new_distributions.py +97 -20
- scipy/stats/_page_trend_test.py +12 -5
- scipy/stats/_probability_distribution.py +265 -43
- scipy/stats/_qmc.py +14 -9
- scipy/stats/_qmc_cy.cpython-312-darwin.so +0 -0
- scipy/stats/_qmvnt.py +16 -95
- scipy/stats/_qmvnt_cy.cpython-312-darwin.so +0 -0
- scipy/stats/_quantile.py +335 -0
- scipy/stats/_rcont/rcont.cpython-312-darwin.so +0 -0
- scipy/stats/_resampling.py +4 -29
- scipy/stats/_sampling.py +1 -1
- scipy/stats/_sobol.cpython-312-darwin.so +0 -0
- scipy/stats/_stats.cpython-312-darwin.so +0 -0
- scipy/stats/_stats_mstats_common.py +21 -2
- scipy/stats/_stats_py.py +550 -476
- scipy/stats/_stats_pythran.cpython-312-darwin.so +0 -0
- scipy/stats/_unuran/unuran_wrapper.cpython-312-darwin.so +0 -0
- scipy/stats/_unuran/unuran_wrapper.pyi +2 -1
- scipy/stats/_variation.py +6 -8
- scipy/stats/_wilcoxon.py +13 -7
- scipy/stats/tests/common_tests.py +6 -4
- scipy/stats/tests/test_axis_nan_policy.py +62 -24
- scipy/stats/tests/test_continued_fraction.py +173 -0
- scipy/stats/tests/test_continuous.py +379 -60
- scipy/stats/tests/test_continuous_basic.py +18 -12
- scipy/stats/tests/test_discrete_basic.py +14 -8
- scipy/stats/tests/test_discrete_distns.py +16 -16
- scipy/stats/tests/test_distributions.py +95 -75
- scipy/stats/tests/test_entropy.py +40 -48
- scipy/stats/tests/test_fit.py +4 -3
- scipy/stats/tests/test_hypotests.py +153 -24
- scipy/stats/tests/test_kdeoth.py +109 -41
- scipy/stats/tests/test_marray.py +289 -0
- scipy/stats/tests/test_morestats.py +79 -47
- scipy/stats/tests/test_mstats_basic.py +3 -3
- scipy/stats/tests/test_multivariate.py +434 -83
- scipy/stats/tests/test_qmc.py +13 -10
- scipy/stats/tests/test_quantile.py +199 -0
- scipy/stats/tests/test_rank.py +119 -112
- scipy/stats/tests/test_resampling.py +47 -56
- scipy/stats/tests/test_sampling.py +9 -4
- scipy/stats/tests/test_stats.py +799 -939
- scipy/stats/tests/test_variation.py +8 -6
- scipy/version.py +2 -2
- {scipy-1.15.3.dist-info → scipy-1.16.0rc2.dist-info}/LICENSE.txt +4 -4
- {scipy-1.15.3.dist-info → scipy-1.16.0rc2.dist-info}/METADATA +11 -11
- {scipy-1.15.3.dist-info → scipy-1.16.0rc2.dist-info}/RECORD +561 -568
- scipy-1.16.0rc2.dist-info/WHEEL +6 -0
- scipy/_lib/array_api_extra/_funcs.py +0 -484
- scipy/_lib/array_api_extra/_typing.py +0 -8
- scipy/interpolate/_bspl.cpython-312-darwin.so +0 -0
- scipy/optimize/_cobyla.cpython-312-darwin.so +0 -0
- scipy/optimize/_cython_nnls.cpython-312-darwin.so +0 -0
- scipy/optimize/_slsqp.cpython-312-darwin.so +0 -0
- scipy/spatial/qhull_src/COPYING.txt +0 -38
- scipy/special/libsf_error_state.dylib +0 -0
- scipy/special/tests/test_log_softmax.py +0 -109
- scipy/special/tests/test_xsf_cuda.py +0 -114
- scipy/special/xsf/binom.h +0 -89
- scipy/special/xsf/cdflib.h +0 -100
- scipy/special/xsf/cephes/airy.h +0 -307
- scipy/special/xsf/cephes/besselpoly.h +0 -51
- scipy/special/xsf/cephes/beta.h +0 -257
- scipy/special/xsf/cephes/cbrt.h +0 -131
- scipy/special/xsf/cephes/chbevl.h +0 -85
- scipy/special/xsf/cephes/chdtr.h +0 -193
- scipy/special/xsf/cephes/const.h +0 -87
- scipy/special/xsf/cephes/ellie.h +0 -293
- scipy/special/xsf/cephes/ellik.h +0 -251
- scipy/special/xsf/cephes/ellpe.h +0 -107
- scipy/special/xsf/cephes/ellpk.h +0 -117
- scipy/special/xsf/cephes/expn.h +0 -260
- scipy/special/xsf/cephes/gamma.h +0 -398
- scipy/special/xsf/cephes/hyp2f1.h +0 -596
- scipy/special/xsf/cephes/hyperg.h +0 -361
- scipy/special/xsf/cephes/i0.h +0 -149
- scipy/special/xsf/cephes/i1.h +0 -158
- scipy/special/xsf/cephes/igam.h +0 -421
- scipy/special/xsf/cephes/igam_asymp_coeff.h +0 -195
- scipy/special/xsf/cephes/igami.h +0 -313
- scipy/special/xsf/cephes/j0.h +0 -225
- scipy/special/xsf/cephes/j1.h +0 -198
- scipy/special/xsf/cephes/jv.h +0 -715
- scipy/special/xsf/cephes/k0.h +0 -164
- scipy/special/xsf/cephes/k1.h +0 -163
- scipy/special/xsf/cephes/kn.h +0 -243
- scipy/special/xsf/cephes/lanczos.h +0 -112
- scipy/special/xsf/cephes/ndtr.h +0 -275
- scipy/special/xsf/cephes/poch.h +0 -85
- scipy/special/xsf/cephes/polevl.h +0 -167
- scipy/special/xsf/cephes/psi.h +0 -194
- scipy/special/xsf/cephes/rgamma.h +0 -111
- scipy/special/xsf/cephes/scipy_iv.h +0 -811
- scipy/special/xsf/cephes/shichi.h +0 -248
- scipy/special/xsf/cephes/sici.h +0 -224
- scipy/special/xsf/cephes/sindg.h +0 -221
- scipy/special/xsf/cephes/tandg.h +0 -139
- scipy/special/xsf/cephes/trig.h +0 -58
- scipy/special/xsf/cephes/unity.h +0 -186
- scipy/special/xsf/cephes/zeta.h +0 -172
- scipy/special/xsf/config.h +0 -304
- scipy/special/xsf/digamma.h +0 -205
- scipy/special/xsf/error.h +0 -57
- scipy/special/xsf/evalpoly.h +0 -47
- scipy/special/xsf/expint.h +0 -266
- scipy/special/xsf/hyp2f1.h +0 -694
- scipy/special/xsf/iv_ratio.h +0 -173
- scipy/special/xsf/lambertw.h +0 -150
- scipy/special/xsf/loggamma.h +0 -163
- scipy/special/xsf/sici.h +0 -200
- scipy/special/xsf/tools.h +0 -427
- scipy/special/xsf/trig.h +0 -164
- scipy/special/xsf/wright_bessel.h +0 -843
- scipy/special/xsf/zlog1.h +0 -35
- scipy/stats/_mvn.cpython-312-darwin.so +0 -0
- scipy-1.15.3.dist-info/WHEEL +0 -4
scipy/special/xsf/hyp2f1.h
DELETED
@@ -1,694 +0,0 @@
|
|
1
|
-
/* Implementation of Gauss's hypergeometric function for complex values.
|
2
|
-
*
|
3
|
-
* This implementation is based on the Fortran implementation by Shanjie Zhang and
|
4
|
-
* Jianming Jin included in specfun.f [1]_. Computation of Gauss's hypergeometric
|
5
|
-
* function involves handling a patchwork of special cases. By default the Zhang and
|
6
|
-
* Jin implementation has been followed as closely as possible except for situations where
|
7
|
-
* an improvement was obvious. We've attempted to document the reasons behind decisions
|
8
|
-
* made by Zhang and Jin and to document the reasons for deviating from their implementation
|
9
|
-
* when this has been done. References to the NIST Digital Library of Mathematical
|
10
|
-
* Functions [2]_ have been added where they are appropriate. The review paper by
|
11
|
-
* Pearson et al [3]_ is an excellent resource for best practices for numerical
|
12
|
-
* computation of hypergeometric functions. We have followed this review paper
|
13
|
-
* when making improvements to and correcting defects in Zhang and Jin's
|
14
|
-
* implementation. When Pearson et al propose several competing alternatives for a
|
15
|
-
* given case, we've used our best judgment to decide on the method to use.
|
16
|
-
*
|
17
|
-
* Author: Albert Steppi
|
18
|
-
*
|
19
|
-
* Distributed under the same license as Scipy.
|
20
|
-
*
|
21
|
-
* References
|
22
|
-
* ----------
|
23
|
-
* .. [1] S. Zhang and J.M. Jin, "Computation of Special Functions", Wiley 1996
|
24
|
-
* .. [2] NIST Digital Library of Mathematical Functions. http://dlmf.nist.gov/,
|
25
|
-
* Release 1.1.1 of 2021-03-15. F. W. J. Olver, A. B. Olde Daalhuis,
|
26
|
-
* D. W. Lozier, B. I. Schneider, R. F. Boisvert, C. W. Clark, B. R. Miller,
|
27
|
-
* B. V. Saunders, H. S. Cohl, and M. A. McClain, eds.
|
28
|
-
* .. [3] Pearson, J.W., Olver, S. & Porter, M.A.
|
29
|
-
* "Numerical methods for the computation of the confluent and Gauss
|
30
|
-
* hypergeometric functions."
|
31
|
-
* Numer Algor 74, 821-866 (2017). https://doi.org/10.1007/s11075-016-0173-0
|
32
|
-
* .. [4] Raimundas Vidunas, "Degenerate Gauss Hypergeometric Functions",
|
33
|
-
* Kyushu Journal of Mathematics, 2007, Volume 61, Issue 1, Pages 109-135,
|
34
|
-
* .. [5] López, J.L., Temme, N.M. New series expansions of the Gauss hypergeometric
|
35
|
-
* function. Adv Comput Math 39, 349-365 (2013).
|
36
|
-
* https://doi.org/10.1007/s10444-012-9283-y
|
37
|
-
* """
|
38
|
-
*/
|
39
|
-
|
40
|
-
#pragma once
|
41
|
-
|
42
|
-
#include "config.h"
|
43
|
-
#include "error.h"
|
44
|
-
#include "tools.h"
|
45
|
-
|
46
|
-
#include "binom.h"
|
47
|
-
#include "cephes/gamma.h"
|
48
|
-
#include "cephes/lanczos.h"
|
49
|
-
#include "cephes/poch.h"
|
50
|
-
#include "cephes/hyp2f1.h"
|
51
|
-
#include "digamma.h"
|
52
|
-
|
53
|
-
namespace xsf {
|
54
|
-
namespace detail {
|
55
|
-
constexpr double hyp2f1_EPS = 1e-15;
|
56
|
-
/* The original implementation in SciPy from Zhang and Jin used 1500 for the
|
57
|
-
* maximum number of series iterations in some cases and 500 in others.
|
58
|
-
* Through the empirical results on the test cases in
|
59
|
-
* scipy/special/_precompute/hyp2f1_data.py, it was determined that these values
|
60
|
-
* can lead to early termination of series which would have eventually converged
|
61
|
-
* at a reasonable level of accuracy. We've bumped the iteration limit to 3000,
|
62
|
-
* and may adjust it again based on further analysis. */
|
63
|
-
constexpr std::uint64_t hyp2f1_MAXITER = 3000;
|
64
|
-
|
65
|
-
XSF_HOST_DEVICE inline double four_gammas_lanczos(double u, double v, double w, double x) {
|
66
|
-
/* Compute ratio of gamma functions using lanczos approximation.
|
67
|
-
*
|
68
|
-
* Computes gamma(u)*gamma(v)/(gamma(w)*gamma(x))
|
69
|
-
*
|
70
|
-
* It is assumed that x = u + v - w, but it is left to the user to
|
71
|
-
* ensure this.
|
72
|
-
*
|
73
|
-
* The lanczos approximation takes the form
|
74
|
-
*
|
75
|
-
* gamma(x) = factor(x) * lanczos_sum_expg_scaled(x)
|
76
|
-
*
|
77
|
-
* where factor(x) = ((x + lanczos_g - 0.5)/e)**(x - 0.5).
|
78
|
-
*
|
79
|
-
* The formula above is only valid for x >= 0.5, but can be extended to
|
80
|
-
* x < 0.5 with the reflection principle.
|
81
|
-
*
|
82
|
-
* Using the lanczos approximation when computing this ratio of gamma functions
|
83
|
-
* allows factors to be combined analytically to avoid underflow and overflow
|
84
|
-
* and produce a more accurate result. The condition x = u + v - w makes it
|
85
|
-
* possible to cancel the factors in the expression
|
86
|
-
*
|
87
|
-
* factor(u) * factor(v) / (factor(w) * factor(x))
|
88
|
-
*
|
89
|
-
* by taking one factor and absorbing it into the others. Currently, this
|
90
|
-
* implementation takes the factor corresponding to the argument with largest
|
91
|
-
* absolute value and absorbs it into the others.
|
92
|
-
*
|
93
|
-
* Since this is only called internally by four_gammas. It is assumed that
|
94
|
-
* |u| >= |v| and |w| >= |x|.
|
95
|
-
*/
|
96
|
-
|
97
|
-
/* The below implementation may incorrectly return finite results
|
98
|
-
* at poles of the gamma function. Handle these cases explicitly. */
|
99
|
-
if ((u == std::trunc(u) && u <= 0) || (v == std::trunc(v) && v <= 0)) {
|
100
|
-
/* Return nan if numerator has pole. Diverges to +- infinity
|
101
|
-
* depending on direction so value is undefined. */
|
102
|
-
return std::numeric_limits<double>::quiet_NaN();
|
103
|
-
}
|
104
|
-
if ((w == std::trunc(w) && w <= 0) || (x == std::trunc(x) && x <= 0)) {
|
105
|
-
// Return 0 if denominator has pole but not numerator.
|
106
|
-
return 0.0;
|
107
|
-
}
|
108
|
-
|
109
|
-
double result = 1.0;
|
110
|
-
double ugh, vgh, wgh, xgh, u_prime, v_prime, w_prime, x_prime;
|
111
|
-
|
112
|
-
if (u >= 0.5) {
|
113
|
-
result *= cephes::lanczos_sum_expg_scaled(u);
|
114
|
-
ugh = u + cephes::lanczos_g - 0.5;
|
115
|
-
u_prime = u;
|
116
|
-
} else {
|
117
|
-
result /= cephes::lanczos_sum_expg_scaled(1 - u) * std::sin(M_PI * u) * M_1_PI;
|
118
|
-
ugh = 0.5 - u + cephes::lanczos_g;
|
119
|
-
u_prime = 1 - u;
|
120
|
-
}
|
121
|
-
|
122
|
-
if (v >= 0.5) {
|
123
|
-
result *= cephes::lanczos_sum_expg_scaled(v);
|
124
|
-
vgh = v + cephes::lanczos_g - 0.5;
|
125
|
-
v_prime = v;
|
126
|
-
} else {
|
127
|
-
result /= cephes::lanczos_sum_expg_scaled(1 - v) * std::sin(M_PI * v) * M_1_PI;
|
128
|
-
vgh = 0.5 - v + cephes::lanczos_g;
|
129
|
-
v_prime = 1 - v;
|
130
|
-
}
|
131
|
-
|
132
|
-
if (w >= 0.5) {
|
133
|
-
result /= cephes::lanczos_sum_expg_scaled(w);
|
134
|
-
wgh = w + cephes::lanczos_g - 0.5;
|
135
|
-
w_prime = w;
|
136
|
-
} else {
|
137
|
-
result *= cephes::lanczos_sum_expg_scaled(1 - w) * std::sin(M_PI * w) * M_1_PI;
|
138
|
-
wgh = 0.5 - w + cephes::lanczos_g;
|
139
|
-
w_prime = 1 - w;
|
140
|
-
}
|
141
|
-
|
142
|
-
if (x >= 0.5) {
|
143
|
-
result /= cephes::lanczos_sum_expg_scaled(x);
|
144
|
-
xgh = x + cephes::lanczos_g - 0.5;
|
145
|
-
x_prime = x;
|
146
|
-
} else {
|
147
|
-
result *= cephes::lanczos_sum_expg_scaled(1 - x) * std::sin(M_PI * x) * M_1_PI;
|
148
|
-
xgh = 0.5 - x + cephes::lanczos_g;
|
149
|
-
x_prime = 1 - x;
|
150
|
-
}
|
151
|
-
|
152
|
-
if (std::abs(u) >= std::abs(w)) {
|
153
|
-
// u has greatest absolute value. Absorb ugh into the others.
|
154
|
-
if (std::abs((v_prime - u_prime) * (v - 0.5)) < 100 * ugh and v > 100) {
|
155
|
-
/* Special case where base is close to 1. Condition taken from
|
156
|
-
* Boost's beta function implementation. */
|
157
|
-
result *= std::exp((v - 0.5) * std::log1p((v_prime - u_prime) / ugh));
|
158
|
-
} else {
|
159
|
-
result *= std::pow(vgh / ugh, v - 0.5);
|
160
|
-
}
|
161
|
-
|
162
|
-
if (std::abs((u_prime - w_prime) * (w - 0.5)) < 100 * wgh and u > 100) {
|
163
|
-
result *= std::exp((w - 0.5) * std::log1p((u_prime - w_prime) / wgh));
|
164
|
-
} else {
|
165
|
-
result *= std::pow(ugh / wgh, w - 0.5);
|
166
|
-
}
|
167
|
-
|
168
|
-
if (std::abs((u_prime - x_prime) * (x - 0.5)) < 100 * xgh and u > 100) {
|
169
|
-
result *= std::exp((x - 0.5) * std::log1p((u_prime - x_prime) / xgh));
|
170
|
-
} else {
|
171
|
-
result *= std::pow(ugh / xgh, x - 0.5);
|
172
|
-
}
|
173
|
-
} else {
|
174
|
-
// w has greatest absolute value. Absorb wgh into the others.
|
175
|
-
if (std::abs((u_prime - w_prime) * (u - 0.5)) < 100 * wgh and u > 100) {
|
176
|
-
result *= std::exp((u - 0.5) * std::log1p((u_prime - w_prime) / wgh));
|
177
|
-
} else {
|
178
|
-
result *= pow(ugh / wgh, u - 0.5);
|
179
|
-
}
|
180
|
-
if (std::abs((v_prime - w_prime) * (v - 0.5)) < 100 * wgh and v > 100) {
|
181
|
-
result *= std::exp((v - 0.5) * std::log1p((v_prime - w_prime) / wgh));
|
182
|
-
} else {
|
183
|
-
result *= std::pow(vgh / wgh, v - 0.5);
|
184
|
-
}
|
185
|
-
if (std::abs((w_prime - x_prime) * (x - 0.5)) < 100 * xgh and x > 100) {
|
186
|
-
result *= std::exp((x - 0.5) * std::log1p((w_prime - x_prime) / xgh));
|
187
|
-
} else {
|
188
|
-
result *= std::pow(wgh / xgh, x - 0.5);
|
189
|
-
}
|
190
|
-
}
|
191
|
-
// This exhausts all cases because we assume |u| >= |v| and |w| >= |x|.
|
192
|
-
|
193
|
-
return result;
|
194
|
-
}
|
195
|
-
|
196
|
-
XSF_HOST_DEVICE inline double four_gammas(double u, double v, double w, double x) {
|
197
|
-
double result;
|
198
|
-
|
199
|
-
// Without loss of generality, ensure |u| >= |v| and |w| >= |x|.
|
200
|
-
if (std::abs(v) > std::abs(u)) {
|
201
|
-
std::swap(u, v);
|
202
|
-
}
|
203
|
-
if (std::abs(x) > std::abs(w)) {
|
204
|
-
std::swap(x, w);
|
205
|
-
}
|
206
|
-
/* Direct ratio tends to be more accurate for arguments in this range. Range
|
207
|
-
* chosen empirically based on the relevant benchmarks in
|
208
|
-
* scipy/special/_precompute/hyp2f1_data.py */
|
209
|
-
if (std::abs(u) <= 100 && std::abs(v) <= 100 && std::abs(w) <= 100 && std::abs(x) <= 100) {
|
210
|
-
result = cephes::Gamma(u) * cephes::Gamma(v) * (cephes::rgamma(w) * cephes::rgamma(x));
|
211
|
-
if (std::isfinite(result) && result != 0.0) {
|
212
|
-
return result;
|
213
|
-
}
|
214
|
-
}
|
215
|
-
result = four_gammas_lanczos(u, v, w, x);
|
216
|
-
if (std::isfinite(result) && result != 0.0) {
|
217
|
-
return result;
|
218
|
-
}
|
219
|
-
// If overflow or underflow, try again with logs.
|
220
|
-
result = std::exp(cephes::lgam(v) - cephes::lgam(x) + cephes::lgam(u) - cephes::lgam(w));
|
221
|
-
result *= cephes::gammasgn(u) * cephes::gammasgn(w) * cephes::gammasgn(v) * cephes::gammasgn(x);
|
222
|
-
return result;
|
223
|
-
}
|
224
|
-
|
225
|
-
class HypergeometricSeriesGenerator {
|
226
|
-
/* Maclaurin series for hyp2f1.
|
227
|
-
*
|
228
|
-
* Series is convergent for |z| < 1 but is only practical for numerical
|
229
|
-
* computation when |z| < 0.9.
|
230
|
-
*/
|
231
|
-
public:
|
232
|
-
XSF_HOST_DEVICE HypergeometricSeriesGenerator(double a, double b, double c, std::complex<double> z)
|
233
|
-
: a_(a), b_(b), c_(c), z_(z), term_(1.0), k_(0) {}
|
234
|
-
|
235
|
-
XSF_HOST_DEVICE std::complex<double> operator()() {
|
236
|
-
std::complex<double> output = term_;
|
237
|
-
term_ = term_ * (a_ + k_) * (b_ + k_) / ((k_ + 1) * (c_ + k_)) * z_;
|
238
|
-
++k_;
|
239
|
-
return output;
|
240
|
-
}
|
241
|
-
|
242
|
-
private:
|
243
|
-
double a_, b_, c_;
|
244
|
-
std::complex<double> z_, term_;
|
245
|
-
std::uint64_t k_;
|
246
|
-
};
|
247
|
-
|
248
|
-
class Hyp2f1Transform1Generator {
|
249
|
-
/* 1 -z transformation of standard series.*/
|
250
|
-
public:
|
251
|
-
XSF_HOST_DEVICE Hyp2f1Transform1Generator(double a, double b, double c, std::complex<double> z)
|
252
|
-
: factor1_(four_gammas(c, c - a - b, c - a, c - b)),
|
253
|
-
factor2_(four_gammas(c, a + b - c, a, b) * std::pow(1.0 - z, c - a - b)),
|
254
|
-
generator1_(HypergeometricSeriesGenerator(a, b, a + b - c + 1, 1.0 - z)),
|
255
|
-
generator2_(HypergeometricSeriesGenerator(c - a, c - b, c - a - b + 1, 1.0 - z)) {}
|
256
|
-
|
257
|
-
XSF_HOST_DEVICE std::complex<double> operator()() {
|
258
|
-
return factor1_ * generator1_() + factor2_ * generator2_();
|
259
|
-
}
|
260
|
-
|
261
|
-
private:
|
262
|
-
std::complex<double> factor1_, factor2_;
|
263
|
-
HypergeometricSeriesGenerator generator1_, generator2_;
|
264
|
-
};
|
265
|
-
|
266
|
-
class Hyp2f1Transform1LimitSeriesGenerator {
|
267
|
-
/* 1 - z transform in limit as c - a - b approaches an integer m. */
|
268
|
-
public:
|
269
|
-
XSF_HOST_DEVICE Hyp2f1Transform1LimitSeriesGenerator(double a, double b, double m, std::complex<double> z)
|
270
|
-
: d1_(xsf::digamma(a)), d2_(xsf::digamma(b)), d3_(xsf::digamma(1 + m)),
|
271
|
-
d4_(xsf::digamma(1.0)), a_(a), b_(b), m_(m), z_(z), log_1_z_(std::log(1.0 - z)),
|
272
|
-
factor_(cephes::rgamma(m + 1)), k_(0) {}
|
273
|
-
|
274
|
-
XSF_HOST_DEVICE std::complex<double> operator()() {
|
275
|
-
std::complex<double> term_ = (d1_ + d2_ - d3_ - d4_ + log_1_z_) * factor_;
|
276
|
-
// Use digamma(x + 1) = digamma(x) + 1/x
|
277
|
-
d1_ += 1 / (a_ + k_); // d1 = digamma(a + k)
|
278
|
-
d2_ += 1 / (b_ + k_); // d2 = digamma(b + k)
|
279
|
-
d3_ += 1 / (1.0 + m_ + k_); // d3 = digamma(1 + m + k)
|
280
|
-
d4_ += 1 / (1.0 + k_); // d4 = digamma(1 + k)
|
281
|
-
factor_ *= (a_ + k_) * (b_ + k_) / ((k_ + 1.0) * (m_ + k_ + 1)) * (1.0 - z_);
|
282
|
-
++k_;
|
283
|
-
return term_;
|
284
|
-
}
|
285
|
-
|
286
|
-
private:
|
287
|
-
double d1_, d2_, d3_, d4_, a_, b_, m_;
|
288
|
-
std::complex<double> z_, log_1_z_, factor_;
|
289
|
-
int k_;
|
290
|
-
};
|
291
|
-
|
292
|
-
class Hyp2f1Transform2Generator {
|
293
|
-
/* 1/z transformation of standard series.*/
|
294
|
-
public:
|
295
|
-
XSF_HOST_DEVICE Hyp2f1Transform2Generator(double a, double b, double c, std::complex<double> z)
|
296
|
-
: factor1_(four_gammas(c, b - a, b, c - a) * std::pow(-z, -a)),
|
297
|
-
factor2_(four_gammas(c, a - b, a, c - b) * std::pow(-z, -b)),
|
298
|
-
generator1_(HypergeometricSeriesGenerator(a, a - c + 1, a - b + 1, 1.0 / z)),
|
299
|
-
generator2_(HypergeometricSeriesGenerator(b, b - c + 1, b - a + 1, 1.0 / z)) {}
|
300
|
-
|
301
|
-
XSF_HOST_DEVICE std::complex<double> operator()() {
|
302
|
-
return factor1_ * generator1_() + factor2_ * generator2_();
|
303
|
-
}
|
304
|
-
|
305
|
-
private:
|
306
|
-
std::complex<double> factor1_, factor2_;
|
307
|
-
HypergeometricSeriesGenerator generator1_, generator2_;
|
308
|
-
};
|
309
|
-
|
310
|
-
class Hyp2f1Transform2LimitSeriesGenerator {
|
311
|
-
/* 1/z transform in limit as a - b approaches a non-negative integer m. (Can swap a and b to
|
312
|
-
* handle the m a negative integer case. */
|
313
|
-
public:
|
314
|
-
XSF_HOST_DEVICE Hyp2f1Transform2LimitSeriesGenerator(double a, double b, double c, double m,
|
315
|
-
std::complex<double> z)
|
316
|
-
: d1_(xsf::digamma(1.0)), d2_(xsf::digamma(1 + m)), d3_(xsf::digamma(a)),
|
317
|
-
d4_(xsf::digamma(c - a)), a_(a), b_(b), c_(c), m_(m), z_(z), log_neg_z_(std::log(-z)),
|
318
|
-
factor_(xsf::cephes::poch(b, m) * xsf::cephes::poch(1 - c + b, m) *
|
319
|
-
xsf::cephes::rgamma(m + 1)),
|
320
|
-
k_(0) {}
|
321
|
-
|
322
|
-
XSF_HOST_DEVICE std::complex<double> operator()() {
|
323
|
-
std::complex<double> term = (d1_ + d2_ - d3_ - d4_ + log_neg_z_) * factor_;
|
324
|
-
// Use digamma(x + 1) = digamma(x) + 1/x
|
325
|
-
d1_ += 1 / (1.0 + k_); // d1 = digamma(1 + k)
|
326
|
-
d2_ += 1 / (1.0 + m_ + k_); // d2 = digamma(1 + m + k)
|
327
|
-
d3_ += 1 / (a_ + k_); // d3 = digamma(a + k)
|
328
|
-
d4_ -= 1 / (c_ - a_ - k_ - 1); // d4 = digamma(c - a - k)
|
329
|
-
factor_ *= (b_ + m_ + k_) * (1 - c_ + b_ + m_ + k_) / ((k_ + 1) * (m_ + k_ + 1)) / z_;
|
330
|
-
++k_;
|
331
|
-
return term;
|
332
|
-
}
|
333
|
-
|
334
|
-
private:
|
335
|
-
double d1_, d2_, d3_, d4_, a_, b_, c_, m_;
|
336
|
-
std::complex<double> z_, log_neg_z_, factor_;
|
337
|
-
std::uint64_t k_;
|
338
|
-
};
|
339
|
-
|
340
|
-
class Hyp2f1Transform2LimitSeriesCminusAIntGenerator {
|
341
|
-
/* 1/z transform in limit as a - b approaches a non-negative integer m, and c - a approaches
|
342
|
-
* a positive integer n. */
|
343
|
-
public:
|
344
|
-
XSF_HOST_DEVICE Hyp2f1Transform2LimitSeriesCminusAIntGenerator(double a, double b, double c, double m,
|
345
|
-
double n, std::complex<double> z)
|
346
|
-
: d1_(xsf::digamma(1.0)), d2_(xsf::digamma(1 + m)), d3_(xsf::digamma(a)),
|
347
|
-
d4_(xsf::digamma(n)), a_(a), b_(b), c_(c), m_(m), n_(n), z_(z), log_neg_z_(std::log(-z)),
|
348
|
-
factor_(xsf::cephes::poch(b, m) * xsf::cephes::poch(1 - c + b, m) *
|
349
|
-
xsf::cephes::rgamma(m + 1)),
|
350
|
-
k_(0) {}
|
351
|
-
|
352
|
-
XSF_HOST_DEVICE std::complex<double> operator()() {
|
353
|
-
std::complex<double> term;
|
354
|
-
if (k_ < n_) {
|
355
|
-
term = (d1_ + d2_ - d3_ - d4_ + log_neg_z_) * factor_;
|
356
|
-
// Use digamma(x + 1) = digamma(x) + 1/x
|
357
|
-
d1_ += 1 / (1.0 + k_); // d1 = digamma(1 + k)
|
358
|
-
d2_ += 1 / (1 + m_ + k_); // d2 = digamma(1 + m + k)
|
359
|
-
d3_ += 1 / (a_ + k_); // d3 = digamma(a + k)
|
360
|
-
d4_ -= 1 / (n_ - k_ - 1); // d4 = digamma(c - a - k)
|
361
|
-
factor_ *= (b_ + m_ + k_) * (1 - c_ + b_ + m_ + k_) / ((k_ + 1) * (m_ + k_ + 1)) / z_;
|
362
|
-
++k_;
|
363
|
-
return term;
|
364
|
-
}
|
365
|
-
if (k_ == n_) {
|
366
|
-
/* When c - a approaches a positive integer and k_ >= c - a = n then
|
367
|
-
* poch(1 - c + b + m + k) = poch(1 - c + a + k) = approaches zero and
|
368
|
-
* digamma(c - a - k) approaches a pole. However we can use the limit
|
369
|
-
* digamma(-n + epsilon) / gamma(-n + epsilon) -> (-1)**(n + 1) * (n+1)! as epsilon -> 0
|
370
|
-
* to continue the series.
|
371
|
-
*
|
372
|
-
* poch(1 - c + b, m + k) = gamma(1 - c + b + m + k)/gamma(1 - c + b)
|
373
|
-
*
|
374
|
-
* If a - b is an integer and c - a is an integer, then a and b must both be integers, so assume
|
375
|
-
* a and b are integers and take the limit as c approaches an integer.
|
376
|
-
*
|
377
|
-
* gamma(1 - c + epsilon + a + k)/gamma(1 - c - epsilon + b) =
|
378
|
-
* (gamma(c + epsilon - b) / gamma(c + epsilon - a - k)) *
|
379
|
-
* (sin(pi * (c + epsilon - b)) / sin(pi * (c + epsilon - a - k))) (reflection principle)
|
380
|
-
*
|
381
|
-
* In the limit as epsilon goes to zero, the ratio of sines will approach
|
382
|
-
* (-1)**(a - b + k) = (-1)**(m + k)
|
383
|
-
*
|
384
|
-
* We may then replace
|
385
|
-
*
|
386
|
-
* poch(1 - c - epsilon + b, m + k)*digamma(c + epsilon - a - k)
|
387
|
-
*
|
388
|
-
* with
|
389
|
-
*
|
390
|
-
* (-1)**(a - b + k)*gamma(c + epsilon - b) * digamma(c + epsilon - a - k) / gamma(c + epsilon - a - k)
|
391
|
-
*
|
392
|
-
* and taking the limit epsilon -> 0 gives
|
393
|
-
*
|
394
|
-
* (-1)**(a - b + k) * gamma(c - b) * (-1)**(k + a - c + 1)(k + a - c)!
|
395
|
-
* = (-1)**(c - b - 1)*Gamma(k + a - c + 1)
|
396
|
-
*/
|
397
|
-
factor_ = std::pow(-1, m_ + n_) * xsf::binom(c_ - 1, b_ - 1) *
|
398
|
-
xsf::cephes::poch(c_ - a_ + 1, m_ - 1) / std::pow(z_, static_cast<double>(k_));
|
399
|
-
}
|
400
|
-
term = factor_;
|
401
|
-
factor_ *= (b_ + m_ + k_) * (k_ + a_ - c_ + 1) / ((k_ + 1) * (m_ + k_ + 1)) / z_;
|
402
|
-
++k_;
|
403
|
-
return term;
|
404
|
-
}
|
405
|
-
|
406
|
-
private:
|
407
|
-
double d1_, d2_, d3_, d4_, a_, b_, c_, m_, n_;
|
408
|
-
std::complex<double> z_, log_neg_z_, factor_;
|
409
|
-
std::uint64_t k_;
|
410
|
-
};
|
411
|
-
|
412
|
-
class Hyp2f1Transform2LimitFinitePartGenerator {
|
413
|
-
/* Initial finite sum in limit as a - b approaches a non-negative integer m. The limiting series
|
414
|
-
* for the 1 - z transform also has an initial finite sum, but it is a standard hypergeometric
|
415
|
-
* series. */
|
416
|
-
public:
|
417
|
-
XSF_HOST_DEVICE Hyp2f1Transform2LimitFinitePartGenerator(double b, double c, double m,
|
418
|
-
std::complex<double> z)
|
419
|
-
: b_(b), c_(c), m_(m), z_(z), term_(cephes::Gamma(m) * cephes::rgamma(c - b)), k_(0) {}
|
420
|
-
|
421
|
-
XSF_HOST_DEVICE std::complex<double> operator()() {
|
422
|
-
std::complex<double> output = term_;
|
423
|
-
term_ = term_ * (b_ + k_) * (c_ - b_ - k_ - 1) / ((k_ + 1) * (m_ - k_ - 1)) / z_;
|
424
|
-
++k_;
|
425
|
-
return output;
|
426
|
-
}
|
427
|
-
|
428
|
-
private:
|
429
|
-
double b_, c_, m_;
|
430
|
-
std::complex<double> z_, term_;
|
431
|
-
std::uint64_t k_;
|
432
|
-
};
|
433
|
-
|
434
|
-
class LopezTemmeSeriesGenerator {
|
435
|
-
/* Lopez-Temme Series for Gaussian hypergeometric function [4].
|
436
|
-
*
|
437
|
-
* Converges for all z with real(z) < 1, including in the regions surrounding
|
438
|
-
* the points exp(+- i*pi/3) that are not covered by any of the standard
|
439
|
-
* transformations.
|
440
|
-
*/
|
441
|
-
public:
|
442
|
-
XSF_HOST_DEVICE LopezTemmeSeriesGenerator(double a, double b, double c, std::complex<double> z)
|
443
|
-
: n_(0), a_(a), b_(b), c_(c), phi_previous_(1.0), phi_(1 - 2 * b / c), z_(z), Z_(a * z / (z - 2.0)) {}
|
444
|
-
|
445
|
-
XSF_HOST_DEVICE std::complex<double> operator()() {
|
446
|
-
if (n_ == 0) {
|
447
|
-
++n_;
|
448
|
-
return 1.0;
|
449
|
-
}
|
450
|
-
if (n_ > 1) { // Update phi and Z for n>=2
|
451
|
-
double new_phi = ((n_ - 1) * phi_previous_ - (2.0 * b_ - c_) * phi_) / (c_ + (n_ - 1));
|
452
|
-
phi_previous_ = phi_;
|
453
|
-
phi_ = new_phi;
|
454
|
-
Z_ = Z_ * z_ / (z_ - 2.0) * ((a_ + (n_ - 1)) / n_);
|
455
|
-
}
|
456
|
-
++n_;
|
457
|
-
return Z_ * phi_;
|
458
|
-
}
|
459
|
-
|
460
|
-
private:
|
461
|
-
std::uint64_t n_;
|
462
|
-
double a_, b_, c_, phi_previous_, phi_;
|
463
|
-
std::complex<double> z_, Z_;
|
464
|
-
};
|
465
|
-
|
466
|
-
XSF_HOST_DEVICE std::complex<double> hyp2f1_transform1_limiting_case(double a, double b, double c, double m,
|
467
|
-
std::complex<double> z) {
|
468
|
-
/* 1 - z transform in limiting case where c - a - b approaches an integer m. */
|
469
|
-
std::complex<double> result = 0.0;
|
470
|
-
if (m >= 0) {
|
471
|
-
if (m != 0) {
|
472
|
-
auto series_generator = HypergeometricSeriesGenerator(a, b, 1 - m, 1.0 - z);
|
473
|
-
result += four_gammas(m, c, a + m, b + m) * series_eval_fixed_length(series_generator,
|
474
|
-
std::complex<double>{0.0, 0.0},
|
475
|
-
static_cast<std::uint64_t>(m));
|
476
|
-
}
|
477
|
-
std::complex<double> prefactor = std::pow(-1.0, m + 1) * xsf::cephes::Gamma(c) /
|
478
|
-
(xsf::cephes::Gamma(a) * xsf::cephes::Gamma(b)) *
|
479
|
-
std::pow(1.0 - z, m);
|
480
|
-
auto series_generator = Hyp2f1Transform1LimitSeriesGenerator(a + m, b + m, m, z);
|
481
|
-
result += prefactor * series_eval(series_generator, std::complex<double>{0.0, 0.0}, hyp2f1_EPS,
|
482
|
-
hyp2f1_MAXITER, "hyp2f1");
|
483
|
-
return result;
|
484
|
-
} else {
|
485
|
-
result = four_gammas(-m, c, a, b) * std::pow(1.0 - z, m);
|
486
|
-
auto series_generator1 = HypergeometricSeriesGenerator(a + m, b + m, 1 + m, 1.0 - z);
|
487
|
-
result *= series_eval_fixed_length(series_generator1, std::complex<double>{0.0, 0.0},
|
488
|
-
static_cast<std::uint64_t>(-m));
|
489
|
-
double prefactor = std::pow(-1.0, m + 1) * xsf::cephes::Gamma(c) *
|
490
|
-
(xsf::cephes::rgamma(a + m) * xsf::cephes::rgamma(b + m));
|
491
|
-
auto series_generator2 = Hyp2f1Transform1LimitSeriesGenerator(a, b, -m, z);
|
492
|
-
result += prefactor * series_eval(series_generator2, std::complex<double>{0.0, 0.0}, hyp2f1_EPS,
|
493
|
-
hyp2f1_MAXITER, "hyp2f1");
|
494
|
-
return result;
|
495
|
-
}
|
496
|
-
}
|
497
|
-
|
498
|
-
XSF_HOST_DEVICE std::complex<double> hyp2f1_transform2_limiting_case(double a, double b, double c, double m,
|
499
|
-
std::complex<double> z) {
|
500
|
-
/* 1 / z transform in limiting case where a - b approaches a non-negative integer m. Negative integer case
|
501
|
-
* can be handled by swapping a and b. */
|
502
|
-
auto series_generator1 = Hyp2f1Transform2LimitFinitePartGenerator(b, c, m, z);
|
503
|
-
std::complex<double> result = cephes::Gamma(c) * cephes::rgamma(a) * std::pow(-z, -b);
|
504
|
-
result *=
|
505
|
-
series_eval_fixed_length(series_generator1, std::complex<double>{0.0, 0.0}, static_cast<std::uint64_t>(m));
|
506
|
-
std::complex<double> prefactor = cephes::Gamma(c) * (cephes::rgamma(a) * cephes::rgamma(c - b) * std::pow(-z, -a));
|
507
|
-
double n = c - a;
|
508
|
-
if (abs(n - std::round(n)) < hyp2f1_EPS) {
|
509
|
-
auto series_generator2 = Hyp2f1Transform2LimitSeriesCminusAIntGenerator(a, b, c, m, n, z);
|
510
|
-
result += prefactor * series_eval(series_generator2, std::complex<double>{0.0, 0.0}, hyp2f1_EPS,
|
511
|
-
hyp2f1_MAXITER, "hyp2f1");
|
512
|
-
return result;
|
513
|
-
}
|
514
|
-
auto series_generator2 = Hyp2f1Transform2LimitSeriesGenerator(a, b, c, m, z);
|
515
|
-
result += prefactor *
|
516
|
-
series_eval(series_generator2, std::complex<double>{0.0, 0.0}, hyp2f1_EPS, hyp2f1_MAXITER, "hyp2f1");
|
517
|
-
return result;
|
518
|
-
}
|
519
|
-
|
520
|
-
} // namespace detail
|
521
|
-
|
522
|
-
XSF_HOST_DEVICE inline std::complex<double> hyp2f1(double a, double b, double c, std::complex<double> z) {
|
523
|
-
/* Special Cases
|
524
|
-
* -----------------------------------------------------------------------
|
525
|
-
* Takes constant value 1 when a = 0 or b = 0, even if c is a non-positive
|
526
|
-
* integer. This follows mpmath. */
|
527
|
-
if (a == 0 || b == 0) {
|
528
|
-
return 1.0;
|
529
|
-
}
|
530
|
-
double z_abs = std::abs(z);
|
531
|
-
// Equals 1 when z i 0, unless c is 0.
|
532
|
-
if (z_abs == 0) {
|
533
|
-
if (c != 0) {
|
534
|
-
return 1.0;
|
535
|
-
} else {
|
536
|
-
// Returning real part NAN and imaginary part 0 follows mpmath.
|
537
|
-
return std::complex<double>{std::numeric_limits<double>::quiet_NaN(), 0};
|
538
|
-
}
|
539
|
-
}
|
540
|
-
bool a_neg_int = a == std::trunc(a) && a < 0;
|
541
|
-
bool b_neg_int = b == std::trunc(b) && b < 0;
|
542
|
-
bool c_non_pos_int = c == std::trunc(c) and c <= 0;
|
543
|
-
/* Diverges when c is a non-positive integer unless a is an integer with
|
544
|
-
* c <= a <= 0 or b is an integer with c <= b <= 0, (or z equals 0 with
|
545
|
-
* c != 0) Cases z = 0, a = 0, or b = 0 have already been handled. We follow
|
546
|
-
* mpmath in handling the degenerate cases where any of a, b, c are
|
547
|
-
* non-positive integers. See [3] for a treatment of degenerate cases. */
|
548
|
-
if (c_non_pos_int && !((a_neg_int && c <= a && a < 0) || (b_neg_int && c <= b && b < 0))) {
|
549
|
-
return std::complex<double>{std::numeric_limits<double>::infinity(), 0};
|
550
|
-
}
|
551
|
-
/* Reduces to a polynomial when a or b is a negative integer.
|
552
|
-
* If a and b are both negative integers, we take care to terminate
|
553
|
-
* the series at a or b of smaller magnitude. This is to ensure proper
|
554
|
-
* handling of situations like a < c < b <= 0, a, b, c all non-positive
|
555
|
-
* integers, where terminating at a would lead to a term of the form 0 / 0. */
|
556
|
-
double max_degree;
|
557
|
-
if (a_neg_int || b_neg_int) {
|
558
|
-
if (a_neg_int && b_neg_int) {
|
559
|
-
max_degree = a > b ? std::abs(a) : std::abs(b);
|
560
|
-
} else if (a_neg_int) {
|
561
|
-
max_degree = std::abs(a);
|
562
|
-
} else {
|
563
|
-
max_degree = std::abs(b);
|
564
|
-
}
|
565
|
-
if (max_degree <= (double) UINT64_MAX) {
|
566
|
-
auto series_generator = detail::HypergeometricSeriesGenerator(a, b, c, z);
|
567
|
-
return detail::series_eval_fixed_length(series_generator, std::complex<double>{0.0, 0.0}, max_degree + 1);
|
568
|
-
} else {
|
569
|
-
set_error("hyp2f1", SF_ERROR_NO_RESULT, NULL);
|
570
|
-
return std::complex<double>{std::numeric_limits<double>::quiet_NaN(),
|
571
|
-
std::numeric_limits<double>::quiet_NaN()};
|
572
|
-
}
|
573
|
-
}
|
574
|
-
// Kummer's Theorem for z = -1; c = 1 + a - b (DLMF 15.4.26)
|
575
|
-
if (std::abs(z + 1.0) < detail::hyp2f1_EPS && std::abs(1 + a - b - c) < detail::hyp2f1_EPS && !c_non_pos_int) {
|
576
|
-
return detail::four_gammas(a - b + 1, 0.5 * a + 1, a + 1, 0.5 * a - b + 1);
|
577
|
-
}
|
578
|
-
std::complex<double> result;
|
579
|
-
bool c_minus_a_neg_int = c - a == std::trunc(c - a) && c - a < 0;
|
580
|
-
bool c_minus_b_neg_int = c - b == std::trunc(c - b) && c - b < 0;
|
581
|
-
/* If one of c - a or c - b is a negative integer, reduces to evaluating
|
582
|
-
* a polynomial through an Euler hypergeometric transformation.
|
583
|
-
* (DLMF 15.8.1) */
|
584
|
-
if (c_minus_a_neg_int || c_minus_b_neg_int) {
|
585
|
-
max_degree = c_minus_b_neg_int ? std::abs(c - b) : std::abs(c - a);
|
586
|
-
if (max_degree <= (double) UINT64_MAX) {
|
587
|
-
result = std::pow(1.0 - z, c - a - b);
|
588
|
-
auto series_generator = detail::HypergeometricSeriesGenerator(c - a, c - b, c, z);
|
589
|
-
result *=
|
590
|
-
detail::series_eval_fixed_length(series_generator, std::complex<double>{0.0, 0.0}, max_degree + 2);
|
591
|
-
return result;
|
592
|
-
} else {
|
593
|
-
set_error("hyp2f1", SF_ERROR_NO_RESULT, NULL);
|
594
|
-
return std::complex<double>{std::numeric_limits<double>::quiet_NaN(),
|
595
|
-
std::numeric_limits<double>::quiet_NaN()};
|
596
|
-
}
|
597
|
-
}
|
598
|
-
/* Diverges as real(z) -> 1 when c <= a + b.
|
599
|
-
* Todo: Actually check for overflow instead of using a fixed tolerance for
|
600
|
-
* all parameter combinations like in the Fortran original. */
|
601
|
-
if (std::abs(1 - z.real()) < detail::hyp2f1_EPS && z.imag() == 0 && c - a - b <= 0 && !c_non_pos_int) {
|
602
|
-
return std::complex<double>{std::numeric_limits<double>::infinity(), 0};
|
603
|
-
}
|
604
|
-
// Gauss's Summation Theorem for z = 1; c - a - b > 0 (DLMF 15.4.20).
|
605
|
-
if (z == 1.0 && c - a - b > 0 && !c_non_pos_int) {
|
606
|
-
return detail::four_gammas(c, c - a - b, c - a, c - b);
|
607
|
-
}
|
608
|
-
/* |z| < 0, z.real() >= 0. Use the Maclaurin Series.
|
609
|
-
* -----------------------------------------------------------------------
|
610
|
-
* Apply Euler Hypergeometric Transformation (DLMF 15.8.1) to reduce
|
611
|
-
* size of a and b if possible. We follow Zhang and Jin's
|
612
|
-
* implementation [1] although there is very likely a better heuristic
|
613
|
-
* to determine when this transformation should be applied. As it
|
614
|
-
* stands, this hurts precision in some cases. */
|
615
|
-
if (z_abs < 0.9 && z.real() >= 0) {
|
616
|
-
if (c - a < a && c - b < b) {
|
617
|
-
result = std::pow(1.0 - z, c - a - b);
|
618
|
-
auto series_generator = detail::HypergeometricSeriesGenerator(c - a, c - b, c, z);
|
619
|
-
result *= detail::series_eval(series_generator, std::complex<double>{0.0, 0.0}, detail::hyp2f1_EPS,
|
620
|
-
detail::hyp2f1_MAXITER, "hyp2f1");
|
621
|
-
return result;
|
622
|
-
}
|
623
|
-
auto series_generator = detail::HypergeometricSeriesGenerator(a, b, c, z);
|
624
|
-
return detail::series_eval(series_generator, std::complex<double>{0.0, 0.0}, detail::hyp2f1_EPS,
|
625
|
-
detail::hyp2f1_MAXITER, "hyp2f1");
|
626
|
-
}
|
627
|
-
/* Points near exp(iπ/3), exp(-iπ/3) not handled by any of the standard
|
628
|
-
* transformations. Use series of López and Temme [5]. These regions
|
629
|
-
* were not correctly handled by Zhang and Jin's implementation.
|
630
|
-
* -------------------------------------------------------------------------*/
|
631
|
-
if (0.9 <= z_abs && z_abs < 1.1 && std::abs(1.0 - z) >= 0.9 && z.real() >= 0) {
|
632
|
-
/* This condition for applying Euler Transformation (DLMF 15.8.1)
|
633
|
-
* was determined empirically to work better for this case than that
|
634
|
-
* used in Zhang and Jin's implementation for |z| < 0.9,
|
635
|
-
* real(z) >= 0. */
|
636
|
-
if ((c - a <= a && c - b < b) || (c - a < a && c - b <= b)) {
|
637
|
-
auto series_generator = detail::LopezTemmeSeriesGenerator(c - a, c - b, c, z);
|
638
|
-
result = std::pow(1.0 - 0.5 * z, a - c); // Lopez-Temme prefactor
|
639
|
-
result *= detail::series_eval(series_generator, std::complex<double>{0.0, 0.0}, detail::hyp2f1_EPS,
|
640
|
-
detail::hyp2f1_MAXITER, "hyp2f1");
|
641
|
-
return std::pow(1.0 - z, c - a - b) * result; // Euler transform prefactor.
|
642
|
-
}
|
643
|
-
auto series_generator = detail::LopezTemmeSeriesGenerator(a, b, c, z);
|
644
|
-
result = detail::series_eval(series_generator, std::complex<double>{0.0, 0.0}, detail::hyp2f1_EPS,
|
645
|
-
detail::hyp2f1_MAXITER, "hyp2f1");
|
646
|
-
return std::pow(1.0 - 0.5 * z, -a) * result; // Lopez-Temme prefactor.
|
647
|
-
}
|
648
|
-
/* z/(z - 1) transformation (DLMF 15.8.1). Avoids cancellation issues that
|
649
|
-
* occur with Maclaurin series for real(z) < 0.
|
650
|
-
* -------------------------------------------------------------------------*/
|
651
|
-
if (z_abs < 1.1 && z.real() < 0) {
|
652
|
-
if (0 < b && b < a && a < c) {
|
653
|
-
std::swap(a, b);
|
654
|
-
}
|
655
|
-
auto series_generator = detail::HypergeometricSeriesGenerator(a, c - b, c, z / (z - 1.0));
|
656
|
-
return std::pow(1.0 - z, -a) * detail::series_eval(series_generator, std::complex<double>{0.0, 0.0},
|
657
|
-
detail::hyp2f1_EPS, detail::hyp2f1_MAXITER, "hyp2f1");
|
658
|
-
}
|
659
|
-
/* 1 - z transformation (DLMF 15.8.4). */
|
660
|
-
if (0.9 <= z_abs && z_abs < 1.1) {
|
661
|
-
if (std::abs(c - a - b - std::round(c - a - b)) < detail::hyp2f1_EPS) {
|
662
|
-
// Removable singularity when c - a - b is an integer. Need to use limiting formula.
|
663
|
-
double m = std::round(c - a - b);
|
664
|
-
return detail::hyp2f1_transform1_limiting_case(a, b, c, m, z);
|
665
|
-
}
|
666
|
-
auto series_generator = detail::Hyp2f1Transform1Generator(a, b, c, z);
|
667
|
-
return detail::series_eval(series_generator, std::complex<double>{0.0, 0.0}, detail::hyp2f1_EPS,
|
668
|
-
detail::hyp2f1_MAXITER, "hyp2f1");
|
669
|
-
}
|
670
|
-
/* 1/z transformation (DLMF 15.8.2). */
|
671
|
-
if (std::abs(a - b - std::round(a - b)) < detail::hyp2f1_EPS) {
|
672
|
-
if (b > a) {
|
673
|
-
std::swap(a, b);
|
674
|
-
}
|
675
|
-
double m = std::round(a - b);
|
676
|
-
return detail::hyp2f1_transform2_limiting_case(a, b, c, m, z);
|
677
|
-
}
|
678
|
-
auto series_generator = detail::Hyp2f1Transform2Generator(a, b, c, z);
|
679
|
-
return detail::series_eval(series_generator, std::complex<double>{0.0, 0.0}, detail::hyp2f1_EPS,
|
680
|
-
detail::hyp2f1_MAXITER, "hyp2f1");
|
681
|
-
}
|
682
|
-
|
683
|
-
XSF_HOST_DEVICE inline std::complex<float> hyp2f1(float a, float b, float c, std::complex<float> x) {
|
684
|
-
return static_cast<std::complex<float>>(hyp2f1(static_cast<double>(a), static_cast<double>(b),
|
685
|
-
static_cast<double>(c), static_cast<std::complex<double>>(x)));
|
686
|
-
}
|
687
|
-
|
688
|
-
XSF_HOST_DEVICE inline double hyp2f1(double a, double b, double c, double x) { return cephes::hyp2f1(a, b, c, x); }
|
689
|
-
|
690
|
-
XSF_HOST_DEVICE inline float hyp2f1(float a, float b, float c, float x) {
|
691
|
-
return hyp2f1(static_cast<double>(a), static_cast<double>(b), static_cast<double>(c), static_cast<double>(x));
|
692
|
-
}
|
693
|
-
|
694
|
-
} // namespace xsf
|