bi dZddlmZddlmZdZdZdZdZd Z d Z d Z d Z d Z dZeee e e e e edZdZdZdZdZdZy)aj 'Generic' Array API backend for RBF interpolation. The general logic is this: `_rbfinterp.py` implements the user API and calls into either `_rbfinterp_np` (the "numpy backend"), or `_rbfinterp_xp` (the "generic backend". The numpy backend offloads performance-critical computations to the pythran-compiled `_rbfinterp_pythran` extension. This way, the call chain is _rbfinterp.py <-- _rbfinterp_np.py <-- _rbfinterp_pythran.py The "generic" backend here is a drop-in replacement of the API of `_rbfinterp_np.py` for use in `_rbfinterp.py` with non-numpy arrays. The implementation closely follows `_rbfinterp_np + _rbfinterp_pythran`, with the following differences: - We used vectorized code not explicit loops in `_build_system` and `_build_evaluation_coefficients`; this is more torch/jax friendly; - RBF kernels are also "vectorized" and not scalar: they receive an array of norms not a single norm; - RBF kernels accept an extra xp= argument; In general, we would prefer less code duplication. The main blocker ATM is that pythran cannot compile functions with an xp= argument where xp is numpy. ) LinAlgError)_monomial_powers_implct||}|j|}|jddk(r|j|d|f}|S)Nr)rasarrayshapereshape)ndimdegreexpouts Z/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/scipy/interpolate/_rbfinterp_xp.py_monomial_powersr sD f -C **S/C yy|qjjq$i( Jc Ht|||||||\}}} } |jj||} | | | fS#t$r_d} |jd} | dkDr=t || z | z ||}|jj |}|| kr d|d| d} t| wxYw)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. zSingular matrixr)r zqSingular matrix. The matrix of monomials evaluated at the data point coordinates does not have full column rank (/z).) _build_systemlinalgsolve Exceptionrpolynomial_matrix matrix_rankr)yd smoothingkernelepsilonpowersr lhsrhsshiftscalecoeffsmsgnmonospmatranks r_build_and_solve_systemr((s8+ 1i&" CeUc*& % %   a A:$a%i%62FD99((.Df}!F!F82/ #!s 9A(B!c| SNrr s rlinearr.^s 2IrcX|j|dk(d|dz|j|zS)Nr)wherelogr,s rthin_plate_spliner3bs* 88AFAq!tbffQi/ 00rc |dzS)Nr+r,s rcubicr6gs a4Krc|dz S)Nr+r,s rquinticr9ks qD5Lrc2|j|dzdz S)Nr0rsqrtr,s r multiquadricr=os GGAqD1H  rc6d|j|dzdzz SN?r0r;r,s rinverse_multiquadricrAss A$ $$rcd|dzdzz Sr?r+r,s rinverse_quadraticrCws !Q$* rc,|j|dz S)Nr0)expr,s rgaussianrF{s 661a4%=r)r.r3r6r9r=rArCrFc x||jj|dddddf|dddddfz d|S)z+Evaluate RBFs, with centers at `x`, at `x`.Naxis)r vector_norm)x kernel_funcr s r kernel_matrixrNsA  aa ma4 m;"Er rcB|j|dddddf|zdS)z9Evaluate monomials, with exponents from `powers`, at `x`.NrHrI)prod)rLrr s rrrs$ 771QaZ=F*7 44rc |jd}|jd}t|} |j|d} |j|d} | | zdz } | | z dz } |j | dk(d| } ||z}|| z | z }t || |}t |||}|j|j||fd|j|j|j||ffdgd|j|j||j|gz}|j||j||fgd}||| | fS)a=Build the system used to solve for the RBF interpolant coefficients. 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 ------- lhs : (P + R, P + R) float ndarray Left-hand side matrix. rhs : (P + R, S) float ndarray Right-hand side matrix. shift : (N,) float ndarray Domain shift used to create the polynomial matrix. scale : (N,) float ndarray Domain scaling used to create the polynomial matrix. rrrIr0gr@) r NAME_TO_FUNCminmaxr1rNrconcatTzerosdiag)rrrrrrr sr-rMminsmaxsr!r"yepsyhat out_kernelsout_polyrr s rrrsi<  A QAv&K 66!!6 D 66!!6 D D[!OE D[!OE HHUc\3 .E W9D Iu D {B7K vr2H )) K* 3 HJJ!Q 01 :     9bhhqk":;<  =C ))Q!Q()) 2C UE !!rc t|}||z} ||z} ||z |z } |j||jj| dddddf| dddddfz d||j | dddddf|zdgd} | S)aConstruct the coefficients needed to evaluate the RBF. Parameters ---------- x : (Q, N) float ndarray Evaluation point coordinates. y : (P, N) float ndarray Data point coordinates. kernel : str Name of the RBF. epsilon : float Shape parameter. powers : (R, N) int ndarray The exponents for each monomial in the polynomial. shift : (N,) float ndarray Shifts the polynomial domain for numerical stability. scale : (N,) float ndarray Scales the polynomial domain for numerical stability. Returns ------- (Q, P + R) float ndarray NrHrI)rRrUrrKrP) rLrrrrr!r"r rMr\xepsxhatvecs r_build_evaluation_coefficientsrds8v&K W9D W9D Iu D ))  %%D!$tD!QJ'77b&  GGDD!$.RG 8    C Jrc 0t||||||||} | |zSr*)rd) rLrrrrr!r"r#r rcs rcompute_interpolationrfs) ( 1fgvueR C <rN)__doc__ numpy.linalgr_rbfinterp_commonrrr(r.r3r6r9r=rArCrFrRrNrrrdrfr+rrrjs6%43 l1 % ) /)  5 ;"|.br