`iXddlZddlZejdkrddlmZnddlmZddlZddlmZddl m Z ddl m Z m Z mZmZdZdd Zd Zd Zd Zd ZdZdZ ddZdZdZdS)N2)normalize_axis_index)sparse)spsolve)_get_module_funcINTERVAL_MODULE D_BOOR_MODULEBSplinecptj|tjr tjStjS)z>Return np.complex128 for complex dtypes, np.float64 otherwise.)cupy issubdtypecomplexfloatingcomplex_float_dtypes u/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/interpolate/_bspline2.py _get_dtypers) ud233}{Fctj|}tj|}t|j}||d}|r5tj|std|S)zConvert the input into a C contiguous float array. NB: Upcasts half- and single-precision floats to double precision. F)copyz$Array must not contain infs or nans.) r asarrayascontiguousarrayrrastypeisfiniteall ValueError)x check_finitedtyps r_as_float_arrayr!s QA q!!A ag  D E""AADM!,,0022A?@@@ Hrcd}|D]}||z}|S)z Product of a sequence of numbers. Faster than np.prod for short lists like array shapes, and does not overflow if using Python integers. )iterableproductrs rprodr'(s( G 1  Nrctj|}|dzdkrtd|z|dz dz}||dz| dz }tj|df|dzz||df|dzzf}|S)zSGiven data x, construct the knot vector w/ not-a-knot BC. cf de Boor, XIII(12).r#z Odd degree for now only. Got %s.r)r rrr_)rkmts r _not_a_knotr/8s QA1uzz;a?@@@ Q1 A !A#qbd( A 11 q1R5(AaC.01A HrcTtj|df|z||df|zfS)zBConstruct a knot vector appropriate for the order-k interpolation.rr*)r r+)rr,s r_augkntr1Es* 7AaD719a!B%!+ ,,rctj|}t|}|dzdkr1tj|}|ddxx|dddz zcc<tj|}tj|d|zz}|||| <t d|D]N}|||z |||dz z dz z |||z dz <|| |zdz |||dz zz|| |z<O|S)z$Returns vector of nodes on a circle.r)rr#r*N)r rlendiffzerosrange)rr,xcndxr.is r_periodic_knotsr;Js  1B BA1uzz Yr]] 1b5 RWq[ 2B 1q1u9AAa!eH 1a[[44Qx"qAE{^a%7"88!a%!) qb1fqjMBqAE{O31"q& Hrct|trN|dkrdtj|fg}n0|dkrdtj|fg}nt d|z|S)Nclampedr#naturalr)zUnknown boundary condition : %s) isinstancestrr r5r)deriv target_shapes r_convert_string_aliasesrC\sv%H I  L1123EE i  L1123EE>FGG G Lrc|2 t|\}}n(#t$r}d}t||d}~wwxYwgg}}tj||S)Nz^Derivatives, `bc_type`, should be specified as a pair of iterables of pairs of (order, value).)zip TypeErrorrr atleast_1d)rAordsvalsemsgs r_process_deriv_specrLgsm  )eJD$$ ) ) );CS//q ( ) d ?4 & &&s  2-2Tc| |dks|dkrd\}}nFt|tr||}}n, |\}}n%#t$r} td|z| d} ~ wwxYwt j|}t ||j}t||}t||}t j ||d}|dkr2t j |d|dd std |j |j dkr-td |j |j |d d|ddkrtd |jd ks(|d d|ddkrtd|dkrtd|||fDrtdtj||df}t j|} t j| t#| j} t'j|| ||S|d kr|||tdtj|d||df}t j|} t j| t#| j} t'j|| ||St+j|}|dkr|t/d|||~|dkrt1||}nw|dkrP|d d|ddzdz }tj|df|d zz|d d|df|d zzf}n!t3||}nt5||}t||}|dkrtd|jd ks(|d d|ddkrtd|j |j |zd zkr$td|j |j |zd zfz|d||ks|d|| krtd|z|dkrt7|||||St9||j d d}t;|\} } | j d} t9||j d d}t;|\}}|j d}|j }|j |z d z }||z | |zkrtd||z d| d||j dkrDt j|f|j d dzt>} t'j|| ||St'j |||}| dks|dkrt jd|zd zft>}d }t j!||ft>}t j!d|j}tEtFd|}t j!dtj$}tEtJd }|d!d"|tj|d|df|||d#df| dkrt j&|dg|j}t j| |ft>}tO| D]L\}}|d!d!|||tQ||||||dd f |d}|d|d z||||z |d zf<MtSj*tSj+||g}|dkr|d|d<t j&|dg|j}t j||ft>}tO|D]L\}}|d!d!|||tQ||||||dd f |d}|d|d z||||z |d zf<MtSj*|tSj+|g}tY|j d d} t j!|| f|j}!| dkr| -d| |!d| <|-d| |!| ||z <|dkr|-d| |!||z d<t j.|!jtj/r/ta||!j1ta||!j2d$zz}"nta||!}"t j|"-|f|j d dz}"t'||"|S)%aiCompute the (coefficients of) interpolating B-spline. Parameters ---------- x : array_like, shape (n,) Abscissas. y : array_like, shape (n, ...) Ordinates. k : int, optional B-spline degree. Default is cubic, ``k = 3``. t : array_like, shape (nt + k + 1,), optional. Knots. The number of knots needs to agree with the number of data points and the number of derivatives at the edges. Specifically, ``nt - n`` must equal ``len(deriv_l) + len(deriv_r)``. bc_type : 2-tuple or None Boundary conditions. Default is None, which means choosing the boundary conditions automatically. Otherwise, it must be a length-two tuple where the first element (``deriv_l``) sets the boundary conditions at ``x[0]`` and the second element (``deriv_r``) sets the boundary conditions at ``x[-1]``. Each of these must be an iterable of pairs ``(order, value)`` which gives the values of derivatives of specified orders at the given edge of the interpolation interval. Alternatively, the following string aliases are recognized: * ``"clamped"``: The first derivatives at the ends are zero. This is equivalent to ``bc_type=([(1, 0.0)], [(1, 0.0)])``. * ``"natural"``: The second derivatives at ends are zero. This is equivalent to ``bc_type=([(2, 0.0)], [(2, 0.0)])``. * ``"not-a-knot"`` (default): The first and second segments are the same polynomial. This is equivalent to having ``bc_type=None``. * ``"periodic"``: The values and the first ``k-1`` derivatives at the ends are equivalent. axis : int, optional Interpolation axis. Default is 0. check_finite : bool, optional Whether to check that the input arrays contain only finite numbers. Disabling may give a performance gain, but may result in problems (crashes, non-termination) if the inputs do contain infinities or NaNs. Default is True. Returns ------- b : a BSpline object of the degree ``k`` and with knots ``t``. N not-a-knotperiodicNNUnknown boundary condition: %srr*gV瞯<)atolzAFirst and last points does not match while periodic case expectedz(Shapes of x {} and y {} are incompatibler#zExpect x to not have duplicatesz1Expect x to be a 1D strictly increasing sequence.c3K|]}|duV dS)Nr$).0_s r z%make_interp_spline..s&<<q}<<<<<>BKLL sy$"677"ch''ch''#-.tS!!  !$,,uqwqrr{/B"C"C D DD 1dA  s6 AAAc2 ||dkrd\}}nFt|tr||}}n, |\}}n%#t$r}td|z|d}~wwxYwt ||jdd}t |\}} |jd} t ||jdd}t |\} } | jd} |j}|j|z dz }t ||jdd}t |\}} |jd} t ||jdd}t |\} } | jd} |j}|j|z dz }||z | | zksJtj |tj }ttd}||jdd zdz d zfd |||||d |jdftj |dft}tj t|t!|jddf|j}t!|jdd}tj |jdd |zdzz}tt$d |}||jdd zdz d zfd |||d|||||d|jdf tj||ft}| }t)t|D]<}||}||d |zdzz|d |zdzz|zdz|||z||z |dzf<=tj dtj }| dkr|d|d<tj|dg|j}t-|D]L\}}|dd|||t/||||||ddf |d}|d|dz||||z |dzf<M| dkr|d|d<tj|dg|j}t-| D]a\}}|dd|||t/||||||ddf |d}| t|z|z}|d|dz||||z |dzf<bt!|jdd}tj ||f|j} | dkr| d|| d| <|d|| | || z <| dkr| d|| || z d<ddlm}!|!|| }"tj|"|f|jddz}"t9||"|S)z Construct the interpolating spline spl(x) = y with *full* linalg. Only useful for testing, do not call directly! This version is O(N**2) in memory and O(N**3) in flop count. NrOrQrRr#rrrZ)rFr)rYr[r*)solve)r?r@rFrrCrarLr`r empty_likerlrrrkrir3r'rr r5r6rmrnrorr cupy.linalgrrr )#rrur,r.rvrwrxrJrzr{r|r}r~rr8r intervalsrrrrrrAoffsetrrrrr-rowrrrrs# r_make_interp_spline_full_matrixr}s]'\11% GS ! !P"G P& GWW P P P=GHHa O P&gqwqrr{;;G!4W!=!=L,  q !E%gqwqrr{;;G!4W!=!=L,   "F A !aB%gqwqrr{;;G!4W!=!=L,  q !E%gqwqrr{;;G!4W!=!=L,   "F A !aB 6UV^ # # # #444I&HHOOagaj3&*s24f9aUAGAJ?AAAj"a...G * QgmABB'(() @ @ @C qrr" # #E :agajAEAI. / /D$]HgFFMMAGAJ$q(S02Fgq!Q 3eQ71:    B85)))A F 3q66]]EE|'+Aqs1uIa1Qik!m,C'D!f*d1fT!Vm #$$:atz222L qyy#A, Q Z1ag . . .l++ - -DAq M$gq#a&&"lC !Q( ) ) ) ?D"&t!t*AaaQ  zz#B- Q Z2qw / / /l++ / /DAq M$gq#a&&"lC !Q( ) ) ) ?D#a&&.1$C$(!A#JAc46$q&= ! !AGABBK  H *b(^17 3 3 3C qyy"**2x88FUF YYr844Cb6k zz(00X>>BKLL"!!!!! 5C==D  !$,,uqwqrr{/B"C"C D DD 1dA  s0 AA  AcR|j}tj|||}tjdd|zdzzt }d}tj|j|z dz |ft } tjd| j} ttd| } tj |d|df} tj |||zdz gtj } tj|dz ||zdz ft }t|dz D]k}| dd || ||dz| | | ||ddf |d|dz||d|dzf<||| dfxx|d|zdzd|zdz|zdzddzcc<ltjtj||g}t#|jdd}tj||zdz |ft }d|d|dz ddf<|jdkr||dn|d|f||dz dddf<t)||}tj|||zdz f|jddz}tj|||d | S) Nr)r#r)r)r#rYrr*r[r\rP) extrapolaterX)r`r rjr r5rirkrrr r+rmrlr6rrprqr'rarrrrrd)rrur.r,rXr8rrrrrrrrrr-matr_csrrrrs rrhrhs A  Aq ) )D :a1Qiu - - -D Ej!&1*q.%0>>>G *V7= 1 1 1C$]HgFFM 1qu B:q!a%!)nDJ???L :qsAEAI&e 4 4 4D 1Q3ZZ ; ; dD'1ac2|    TacT Q1W  QV QqS1WacAg]1_45crc:: }f/55"$%%HAGABBK  H *a!eai*% 8 8 8CC!a%!!! +,6Q;; !Q 2x.11Q!!!  8S ! !D  !$,,A |agabbk/I"J"J K KD  !!T1.8t E E EEr)F)rMNNrT)renumpy __version__numpy.core.multiarrayrnumpy.lib.array_utilsr cupyx.scipyrcupyx.scipy.sparse.linalgr cupyx.scipy.interpolate._bsplinerrr r rr!r'r/r1r;rCrLrrrhr$rrrs s::::::::::::: ------????????????            ---    $ ' ' '>?$(FFFFRBBBJ6E6E6E6E6Er