o ?Hh@sddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z gdZ eZeZdddZd d Zd d Zd dZddZddZdddddZddZddZejddZddZdS))NumberN)copy_if_needed) good_sizeprev_good_size)rr set_workers get_workersc CsVt|tr|f}z dd|D}W|Sty*}z |pd}t|d|d}~ww)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]`` cSsg|]}t|qS)operatorindex.0ar r [/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/scipy/fft/_pocketfft/helper.py &z$_iterable_of_int..valuez) must be a scalar or iterable of integersN) isinstancer TypeError ValueError)xnameer r r_iterable_of_ints rcs<|du}|du}|s5t|d}fdd|D}tfdd|Dr'tdtt|t|kr5td|sot|d }|rJt|t|krJtd |rbt|jkrWtd tjt|j}fd dt||D}n|r|tj }tj}n fd d|D}tdd|Drtd|dt |t|fS)z2Handles shape and axes arguments for nd transformsNaxescs"g|] }|dkr |jn|qS)rndimr rr rr5s"z+_init_nd_shape_and_axes..c3s"|] }|jkp |dkVqdS)rNrr rr r 7s z*_init_nd_shape_and_axes..z$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 presentcs&g|]\}}|dkrj|n|qS)r)r srrr rrGs&csg|]}j|qSr r!r rr rrLrcss|]}|dkVqdS)rNr )r r"r r rrNsinvalid number of data points ( ) specified) ranyrlensetrrangeziplistrtuple)rrrnoshapenoaxesr rr_init_nd_shape_and_axes.s6     r.cCsvt|ds t|}|jtjkrt|tjS|jjdvr$t|tjS|jd}|j ds1dnt }tj |||dS)zl Convert to array with floating or complex dtype. float16 values are also promoted to float32. dtypefc=ALIGNEDT)r/copy) hasattrnpasarrayr/float16float32kindfloat64 newbyteorderflagsrarray)rr/r3r r r _asfarrayUs     r>cCs0||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)rr5ndarrayr4base)arroriginalr r r _datacopiedis  rDc Csd}tdg|j}t||D]\}}|j||kr"td|||<qtd|j|||<d}qt|}|s;||dfSt|j}t||D]\}}|||<qEt||j} ||| |<| dfS)z5Internal auxiliary function for _raw_fft, _raw_fftnd.FNrT) slicerr)rr+r*r5zerosr/) rrr must_copyr naxr"axiszr r r _fix_shapeus     rLcCs(|dkr td|dt||f|fS)Nrr#r$)rrL)rrHrJr r r _fix_shape_1ds  rM)NbackwardorthoforwardcCs>zt|}|r |WSd|WStytd|ddw)zAReturns the pypocketfft normalization mode from the norm argumentrNzInvalid norm value z,, should be "backward", "ortho" or "forward"N) _NORM_MAPKeyErrorr)normrQinormr r r_normalizations  rVcCs`|dur ttddS|dkr&|t kr|dt7}|Std|dt |dkr.td|S)Ndefault_workersrrz workers value out of range; got z, must not be less than zworkers must not be zero)getattr_config _cpu_countr)workersr r r_workerss    r\ccs4t}tt|t_z dVW|t_dS|t_w)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)rr\r r rYrW)r[ old_workersr r rrs rcCs 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 rWr)rXrYr r r rrs r)N)numbersrr os threading contextlibnumpyr5scipy._lib._utilr pypocketfftrr__all__localrY cpu_countrZrr.r>rDrLrMrRrVr\contextmanagerrrr r r rs.   '