`i0ddlZddlZddlZddlZddlmZddlZdZdZ edZ edZ edZ dZ d Zd Zdd Zd ZdZdS)N) RankWarningcBdfd}tj|S)NcFt|tjr|jStj|rtj|St|tjr0tj|}|jdkr|Stdtd)Nz)Multidimensional inputs are not supportedzUnsupported type) isinstancecupypoly1d_coeffsisscalar atleast_1dndarrayndim ValueError TypeErrorxs k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/lib/_routines_poly.py _get_coeffsz'_wraps_polyroutine.._get_coeffs s a % % 9  =   &?1%% % a & & J""Av{{HII I*+++cfd|D}|}td|Dr|St|tjrtj|St|t rt d|DSJ)Nc&g|] }|Sr).0rrs r z7_wraps_polyroutine..wrapper..s!///Q++a..///rc3LK|]}t|tj V dS)N)rrr rrs r z6_wraps_polyroutine..wrapper..s1<.wrapper..!s 666Q$+a..666r)allrrr r tuple)argscoeffsoutrfuncs rwrapperz#_wraps_polyroutine..wrappers////$///dFm <> r hrBHQK"(1+5q9::jj)"b11j>> q3w| Jrctj|d}tj|d}|jdkrtjdg|j}|jdkrtjdg|j}tj||S)aDComputes the product of two polynomials. Args: a1 (scalar, cupy.ndarray or cupy.poly1d): first input polynomial. a2 (scalar, cupy.ndarray or cupy.poly1d): second input polynomial. Returns: cupy.ndarray or cupy.poly1d: The product of the inputs. .. seealso:: :func:`numpy.polymul` f)trimrg)r trim_zerosr:arrayr8convolve)rIrJs rpolymulrTs~ # & & &B # & & &B w!|| Zbh ' ' w!|| Zbh ' ' =R  rc |dkrdS|dkr|S|dzdkr&ttj|||dzStj|ttj|||dz dzS)Nrrr*)_polypowrrS)rns r_polypow_directrXs{AvvqAvv1uzz a++Q!V444 =HT]1a%8%81q5Q,GG H HHrc,|dkrdS|dkr|Stjj||d}|dkrt ||S|dkr|jjdkr#tjjtjj}}n"tjj tjj }}|j dz |zdz}tj j|}|||}tj|||}|||} | d|SJ)Nrrr.directfftc)rr?r@_choose_conv_methodrXr8kindr[ifftrfftirfftr:cupyxscipy next_fast_lenpower) rrWmethodr[r_out_sizer:fxfyys rrVrVsAvvqAvv Z_ 0 0Av > >F q!$$$ 5 7<3   dhmCC tx~CFQJ!#a'{,,X66 SD\\ ZAr " " DTNN(|urc|jjdkr!|tjdS|tjdS)Nr\Fr,)r8r^r6numpy complex128float64rs r_polyfit_typecastrosBw|sxx(ux555 88EM8 . ..rFc|jjdkr|jjdkrtd|jtjkrt dt|}t|}t|}|dkrtd|j dkrt d|j dkrt d |j dks |j d krt d |j |j dkrt d tjj||d d d d df}|}|st|}|j dkrt d|j |j krt d||d d d fz}|j d kr |d d d f}||z}|&|j tj|jjz}tjtj|d} || z}tj|||\} } } } |j dkr| dd} | | z} |dz}| |kr |sd}t1j|t4d|r+| jjdkrtj| } | | | | |fS|rtjtj|j|}|tj| | z}|dkrd}n(|j |kr| |j |z z }ntd|j dkr|d}| ||zfS| S)aReturns the least squares fit of polynomial of degree deg to the data y sampled at x. Args: x (cupy.ndarray): x-coordinates of the sample points of shape (M,). y (cupy.ndarray): y-coordinates of the sample points of shape (M,) or (M, K). deg (int): degree of the fitting polynomial. rcond (float, optional): relative condition number of the fit. The default value is ``len(x) * eps``. full (bool, optional): indicator of the return value nature. When False (default), only the coefficients are returned. When True, diagnostic information is also returned. w (cupy.ndarray, optional): weights applied to the y-coordinates of the sample points of shape (M,). cov (bool or str, optional): if given, returns the coefficients along with the covariance matrix. Returns: cupy.ndarray or tuple: p (cupy.ndarray of shape (deg + 1,) or (deg + 1, K)): Polynomial coefficients from highest to lowest degree residuals, rank, singular_values, rcond (cupy.ndarray, int, cupy.ndarray, float): Present only if ``full=True``. Sum of squared residuals of the least-squares fit, rank of the scaled Vandermonde coefficient matrix, its singular values, and the specified value of ``rcond``. V (cupy.ndarray of shape (M, M) or (M, M, K)): Present only if ``full=False`` and ``cov=True``. The covariance matrix of the polynomial coefficient estimates. .. warning:: cupy.exceptions.RankWarning: The rank of the coefficient matrix in the least-squares fit is deficient. It is raised if ``full=False``. .. seealso:: :func:`numpy.polyfit` ebz0float16 x and bool y are not currently supportedzfloat16 y are not supportedrzexpected deg >= 0rzexpected 1D vector for xzexpected non-empty vector for xr*zexpected 1D or 2D array for yz$expected x and y to have same lengthNz expected a 1-d array for weightsz(expected w and y to have the same length)axisz!Polyfit may be poorly conditioned) stacklevelr\unscaledzJthe number of data points must exceed order to scale the covariance matrix.N) r8r9r^r5rlfloat16rrointrrr:r/r polynomial polyvanderfinfoepssqrtsquaresumr3lstsqreshapewarningswarnrabsoluteinvdotr2outer)rrjdegrcondr.wcovlhsrhsscaler\residsranksordermsgbasefactors rpolyfitrs}R w|sqw|s22!#9:: :w%-5666!A!A c((C Qww,---v{{2333v{{9:::vzzQVaZZ7888v>??? / $ / /3 7 744R4 @C C} a  6Q;;>?? ? 6QV  FGG G qDz 8q==!!!T' A q }AG,,00 It{3'',,!,44 5 5E5LC**3U;;AvtQvzz b!$$JA !GE u}}T}1 c;15555 ) <  # #]6**F&$5((  {txs3344  5%((( *  FF Ve^^qv~.FF?@@ @ 6Q;; ?D$- Hrc:t|tjr|j}t|tjr |jdkrt d|jdkrtdt|tjrtj |jd}tjtj dg|}tjtj dg|}|dddD]}|||zz}||z}|Stj |j d|}| |d }|jdkrtj|j|S|tjkr ||z|dzStj|stj||d }|tj|j|z}|ddd|zd| S) aEvaluates a polynomial at specific values. Args: p (cupy.ndarray or cupy.poly1d): input polynomial. x (scalar, cupy.ndarray): values at which the polynomial is evaluated. Returns: cupy.ndarray or cupy.poly1d: polynomial evaluated at x. .. warning:: This function doesn't currently support poly1d values to evaluate. .. seealso:: :func:`numpy.polyval` rz%p must be 1d ndarray or poly1d objectrzp must be 1d array)r8NrsFr,rx)rtr8)rrr r"r rrrrlrGrRr8typer6r:r<r/bool_anyr asarrayaranger)prr8resprodr\s rpolyvalr4s$!T[!! H a & &A!&A++?@@@vzz-...!T[!!!!(A..k$*aS66677{4:qc7778844R4  Aq.C!8DD  agll1ooq 1 1E U##Av{{z!'5)))  uuww{QrU"" =  4 L% ( ( ( 3 T[u - - --A dddGaK  "E  2 22rcnt|tjr|j}|jjdkrt d|jdkrtd|j dkrtj gStj j |dddg\}|j dkrtj gS|j dkrJ|d |dz d}|ddkr$|jt j}|Stj j |}tj||jr tj|}nt d ||jS) a!Computes the roots of a polynomial with given coefficients. Args: p (cupy.ndarray or cupy.poly1d): polynomial coefficients. Returns: cupy.ndarray: polynomial roots. .. warning:: This function doesn't support currently polynomial coefficients whose companion matrices are general 2d square arrays. Only those with complex Hermitian or real symmetric 2d arrays are allowed. The current `cupy.roots` doesn't guarantee the order of results. .. seealso:: :func:`numpy.roots` rrz boolean inputs are not supportedrz"0-dimensional input is not allowedr*Nrsrr+)rrr r"r8r^r5rrr:rRr{ polyutils as_seriesrealr6rlrn polycompanionr0r1r2r3r4)rr#cmatrixs rrootsrasu(!T[!! Hw|s!"DEEEv{{<===vzzz"~~ / # - -q2wi 8 8CQvzzz"~~v{{!uqt|T" Q4199(//%-00C o(66q99G !122/k""7++!#.// / ::ag  r)NFNF)r&rrlrcupy.exceptionsrcupyx.scipy.fftrbr(rDrKrMrTrXrVrorrrrrrrs/ ''''''3338(((V*0!!!,III2/// p p p p f*3*3*3Z,,,,,r