o ÒtBhbã @s¢ddlZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Zdd„Zdd„Z eeeeeeee dœZ dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd „Zd!d"„ZdS)#éNcCs| S)N©©Úrrrú{/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/scipy/interpolate/_rbfinterp_pythran.pyÚlinearsrcCs|dkrdS|dt |¡S)Nrçé)ÚnpÚlogrrrrÚthin_plate_splinesr cCs|dS)NérrrrrÚcubicsr cCs |d S)NérrrrrÚquintics rcCst |dd¡ S)Nré©r ÚsqrtrrrrÚ multiquadricsrcCsdt |dd¡S©NrrrrrrrÚinverse_multiquadricsrcCsd|ddSrrrrrrÚinverse_quadraticórcCst |d ¡S)Nr)r ÚexprrrrÚgaussian#rr)rr r rrrrrcCs4t|jdƒD]}|tj |||¡ƒ||<qdS)z5Evaluate RBFs, with centers at `y`, at the point `x`.rN©ÚrangeÚshaper ÚlinalgÚnorm)ÚxÚyÚ kernel_funcÚoutÚirrrÚ kernel_vector3sÿr$cCs.t|jdƒD] }t |||¡||<qdS)zCEvaluate monomials, with exponents from `powers`, at the point `x`.rN©rrr Úprod)rÚpowersr"r#rrrÚpolynomial_vector9sÿr(cCsbt|jdƒD]'}t|dƒD]}|tj ||||¡ƒ|||f<|||f|||f<qqdS)z+Evaluate RBFs, with centers at `x`, at `x`.rrNr)rr!r"r#ÚjrrrÚ kernel_matrix?s $þÿr*cCsJt|jdƒD]}t|jdƒD]}t ||||¡|||f<qqdS)z9Evaluate monomials, with exponents from `powers`, at `x`.rNr%)rr'r"r#r)rrrÚpolynomial_matrixGs  ÿÿr+cCs6tj|jd|jdftd}t|}t|||ƒ|S)z3Return RBFs, with centers at `x`, evaluated at `x`.r©Údtype)r ÚemptyrÚfloatÚ NAME_TO_FUNCr*)rÚkernelr"r!rrrÚ_kernel_matrixOs r2cCs.tj|jd|jdftd}t|||ƒ|S)zAReturn monomials, with exponents from `powers`, evaluated at `x`.rr,)r r.rr/r+)rr'r"rrrÚ_polynomial_matrixXs r3cCsj|jd}|jd}|jd}t|} tj|dd} tj|dd} | | d} | | d} d| | dk<||}|| | }tj||||ftdj}t|| |d|…d|…fƒt |||d|…|d…fƒ|d|…|d…fj||d…d|…f<d||d…|d…f<t |ƒD]}|||f||7<qˆtj|||ftdj}||d|…<d||d…<||| | 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. rr)Úaxisrgð?rr,N) rr0r ÚminÚmaxr.r/ÚTr*r+r)r ÚdÚ smoothingr1Úepsilonr'ÚpÚsrr!ÚminsÚmaxsÚshiftÚscaleÚyepsÚyhatÚlhsr#ÚrhsrrrÚ _build_systemes,       &    rEc Csú|jd}|jd} |jd} |jd} t|} ||} ||}|||}tj|| ftd}tj| | ftd}t|ƒD]=}t||| | |d| …ƒt||||| d…ƒt| ƒD]}t| | ƒD]}|||f|||f||7<qdq\q=|S)a½Evaluate the RBF interpolant at `x`. 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. coeffs : (P + R, S) float ndarray Coefficients for each RBF and monomial. Returns ------- (Q, S) float ndarray rrr,N) rr0r Úzerosr/r.rr$r()rr r1r:r'r?r@ÚcoeffsÚqr;rr<r!rAÚxepsÚxhatr"Úvecr#r)ÚkrrrÚ _evaluate¯s&       &ÿÿrM)Únumpyr rr r rrrrrr0r$r(r*r+r2r3rErMrrrrÚs4ø   J