bi hddlZddlmZddlmZddlmZddlm Z m Z m ZdZ dZd Zd Z d Zd Zy) N) LinAlgError)dgesv)_monomial_powers_impl) _build_system_build_evaluation_coefficients_polynomial_matrixc $t|||||||SN)&_pythran_build_evaluation_coefficients)xykernelepsilonpowersshiftscalexps Z/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/scipy/interpolate/_rbfinterp_np.pyrrs  2 1fgvue ct||Sr )_pythran_polynomial_matrix)r rrs rpolynomial_matrixrs %a 00rct||}tj|tj}t |dk(r|j d|}|S)N)dtyper)rnpasarrayint64lenreshape)ndimdegreerouts r_monomial_powersr$sB f -C **S )C 3x1}kk!T" Jrc"t||||||Sr )_pythran_build_system)rd smoothingrrrrs rrr!s Ay&'6 JJrc Vt|||||||\}}} } t||dd\} } } } | dkrtd| d| dkDrad}|jd}|dkDr@t || z | z ||}t j j|}||kr d|d|d }t|| | | fS) aBuild and solve the RBF interpolation system of equations. Parameters ---------- y : (P, N) float ndarray Data point coordinates. d : (P, S) float ndarray Data values at `y`. smoothing : (P,) float ndarray Smoothing parameter for each data point. kernel : str Name of the RBF. epsilon : float Shape parameter. powers : (R, N) int ndarray The exponents for each monomial in the polynomial. Returns ------- coeffs : (P + R, S) float ndarray Coefficients for each RBF and monomial. shift : (N,) float ndarray Domain shift used to create the polynomial matrix. scale : (N,) float ndarray Domain scaling used to create the polynomial matrix. T) overwrite_a overwrite_brzThe z"-th argument had an illegal value.zSingular matrix.zqSingular matrix. The matrix of monomials evaluated at the data point coordinates does not have full column rank (/z).) rr ValueErrorshaperrlinalg matrix_rankr)rr'r(rrrrlhsrhsrr_coeffsinfomsgnmonospmatranks r_build_and_solve_systemr:%s8+ 1i&" CeUsCTtLAq&$ ax4w&HIJJ  a A:$a%i%6CD99((.Df}!F!F82/ # % rc 0t||||||||} | |zSr )r) r rrrrrrr4rvecs rcompute_interpolationr=Xs) ( 1fgvueR C <r)numpyr numpy.linalgrscipy.linalg.lapackr_rbfinterp_commonr_rbfinterp_pythranrr&rr r rrr$r:r=rrrDs;$%41K1 fr