ftj/TddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZgdZejZejZddZd Zd Zd Zd Zd ZdddddZdZdZe dejdZe ddZdS))NumberN)copy_if_needed)xp_capabilities) good_sizeprev_good_size)rr set_workers get_workersct|tr|f} d|D}n)#t$r}|pd}t|d|d}~wwxYw|S)aConvert ``x`` to an iterable sequence of int Parameters ---------- x : value, or sequence of values, convertible to int name : str, optional Name of the argument being converted, only used in the error message Returns ------- y : ``List[int]`` c6g|]}tj|S)operatorindex).0as Y/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/scipy/fft/_pocketfft/helper.py z$_iterable_of_int..'s" * * *1X^A   * * *valuez) must be a scalar or iterable of integersN) isinstancer TypeError ValueError)xnamees r_iterable_of_intrs!V DT * * * * * TTTwDKKKLLRSST Hs ' A AA c|du}|du}|st|d}fd|D}tfd|Drtdtt |t|krtd|st|d}|r/t|t|krtd|rQt|jkrtd t jt|z j}fd t||D}n9|r)tj }t j}nfd |D}td |Drtd |dt|t|fS)a Handle shape and axes arguments for N-D transforms. Returns the shape and axes in a standard form, taking into account negative values and checking for various potential errors. Parameters ---------- x : ndarray The input array. shape : int or array_like of ints or None The shape of the result. If both `shape` and `axes` (see below) are None, `shape` is ``x.shape``; if `shape` is None but `axes` is not None, then `shape` is ``numpy.take(x.shape, axes, axis=0)``. If `shape` is -1, the size of the corresponding dimension of `x` is used. axes : int or array_like of ints or None Axes along which the calculation is computed. The default is over all axes. Negative indices are automatically converted to their positive counterparts. Returns ------- shape : tuple The shape of the result as a tuple of integers. axes : list Axes along which the calculation is computed, as a list of integers. Naxesc4g|]}|dkr |jzn|S)rndimrrrs rrz+_init_nd_shape_and_axes..Rs+999qa!eeAF 999rc3:K|]}|jkp|dkVdS)rNr r"s r z*_init_nd_shape_and_axes..Ts222qAF{#a!e222222rz$axes exceeds dimensionality of inputzall axes must be uniqueshapezBwhen given, axes and shape arguments have to be of the same lengthz)shape requires more axes than are presentc@g|]\}}|dkr j|n|S)r%)rsrrs rrz+_init_nd_shape_and_axes..ds/KKK$!QqBwwAKKKrc*g|]}j|Sr r(r"s rrz+_init_nd_shape_and_axes..is******rc3"K|] }|dkV dS)rNr )rr)s rr$z*_init_nd_shape_and_axes..ks& Q1q5 rinvalid number of data points ( ) specified) ranyrlensetr!rangeziplistr%tuple)rr%rnoshapenoaxess` r_init_nd_shape_and_axesr7/s<tmG T\F 8f--9999D999 2222T222 2 2 ECDD D s4yy>>SYY & &677 7 + 00  ?CIIU++>?? ?  65zzAF"" !LMMM#e**,af55DKKKK#eT:J:JKKK +QW QV}}****T*** %   B @e @ @ @BB B <<d ##rct|dstj|}|jtjkrtj|tjS|jjdvrtj|tjS|jd}|j dsdnt}tj |||S)zl Convert to array with floating or complex dtype. float16 values are also promoted to float32. dtypefc=ALIGNEDT)r9copy) hasattrnpasarrayr9float16float32kindfloat64 newbyteorderflagsrarray)rr9r=s r _asfarrayrHrs 1g   JqMMw"*z!RZ((( T ! !z!RZ((( G  % %Ewy) =44~D 8AU . . ..rcx||urdSt|tjst|drdS|jduS)z| Strict check for `arr` not sharing any data with `original`, under the assumption that arr = asarray(original) F __array__N)rr?ndarrayr>base)arroriginals r _datacopiedrOsI  hu h + ++0N0Nu 8t rcd}tdg|jz}t||D]J\}}|j||krtd|||<*td|j|||<d}Kt |}|s ||dfSt |j}t||D] \}}|||< t j||j} ||| |<| dfS)z5Internal auxiliary function for _raw_fft, _raw_fftnd.FNrT) slicer!r2r%r4r3r?zerosr9) rr%r must_copyrnaxr)axiszs r _fix_shaperXsI4[[M!& EUD!!2 72;!  a E"IIa--E"III %LLE x QW Aud##4$ AGAxAeH d7NrcZ|dkrtd|dt||f|fS)Nrr,r-)rrX)rrTrVs r _fix_shape_1drZsC1uu > > a!w ' ''r)Nbackwardorthoforwardct t|}|r|nd|z S#t$rtd|ddwxYw)zAReturns the pypocketfft normalization mode from the norm argumentr[zInvalid norm value z,, should be "backward", "ortho" or "forward"N) _NORM_MAPKeyErrorr)normr^inorms r_normalizationrdsi=$0uua%i0 === 2$ 2 2 2338< ==s7c|ttddS|dkr5|t kr|dtzz }n0td|dt |dkrtd|S)Ndefault_workersrrz workers value out of range; got z, must not be less than zworkers must not be zero)getattr_config _cpu_countr)workerss r_workersrksw 11555{{ zk ! ! q:~ %GG999,6;99:: : A3444 NrT) out_of_scopec#Kt}ttj|t_ dV|t_dS#|t_wxYw)aContext manager for the default number of workers used in `scipy.fft` Parameters ---------- workers : int The default number of workers to use Examples -------- >>> import numpy as np >>> from scipy import fft, signal >>> rng = np.random.default_rng() >>> x = rng.standard_normal((128, 64)) >>> with fft.set_workers(4): ... y = signal.fftconvolve(x, x) N)r rkrrrhrf)rj old_workerss rr r s[(--K&x~g'>'>??G. "-+----s AAc.ttddS)zReturns the default number of workers within the current context Examples -------- >>> from scipy import fft >>> fft.get_workers() 1 >>> with fft.set_workers(4): ... fft.get_workers() 4 rfr)rgrhr rrr r s 7-q 1 11r)N) numbersrros threading contextlibnumpyr?scipy._lib._utilrscipy._lib._array_apir pypocketfftrr__all__localrh cpu_countrirr7rHrOrXrZr`rdrkcontextmanagerr r r rrr|s ++++++11111132222222 H G G )/   R\^^     2@$@$@$F///(   6(((!aA > > ===    d### ..$#.4d### 2 2$# 2 2 2r