U |e]@sddlmZddlZddlZddlZddlZddlZddlm Z e Z e Z dddZddZd d Zd d Zd dZddZdddddZddZddZejddZddZdS))NumberN) good_sizec Csbt|tr|f}zdd|D}Wn:tk r\}z|p:d}td||W5d}~XYnX|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]`` cSsg|]}t|qS)operatorindex.0arrX/var/www/website-v5/atlas_env/lib/python3.8/site-packages/scipy/fft/_pocketfft/helper.py sz$_iterable_of_int..valuez+{} must be a scalar or iterable of integersN) isinstancer TypeError ValueErrorformat)xnameerrr _iterable_of_ints rcs4|dk}|dk}|sjt|d}fdd|D}tfdd|DrNtdtt|t|krjtd|st|d }|rt|t|krtd |rt|jkrtd tjt|j}fd dt||D}n,|rtj }tj}nfd d|D}tdd|Dr,td |||fS)z2Handles shape and axes arguments for nd transformsNaxescs"g|]}|dkr|jn|qS)rndimrrrr r 0sz+_init_nd_shape_and_axes..c3s |]}|jkp|dkVqdS)rNrrrrr 2sz*_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 sr rrr r Bscsg|]}j|qSrrrrrr r Gscss|]}|dkVqdS)rNr)r rrrr rIs-invalid number of data points ({0}) specified) ranyrlensetrrangeziplistrr)rrrnoshapeZnoaxesrrr _init_nd_shape_and_axes)s6    r'cCspt|dst|}|jtjkr.t|tjS|jjdkrHt|tjS|jd}|j d }tj |||dS)zl Convert to array with floating or complex dtype. float16 values are also promoted to float32. dtypefc=ALIGNED)r(copy) hasattrnpasarrayr(float16float32kindfloat64 newbyteorderflagsarray)rr(r,rrr _asfarrayPs      r7cCs0||kr dSt|tjs&t|dr&dS|jdkS)z| Strict check for `arr` not sharing any data with `original`, under the assumption that arr = asarray(original) F __array__N)rr.ndarrayr-base)arroriginalrrr _datacopiedds r=c Csd}tdg|j}t||D]>\}}|j||krDtd|||<qtd|j|||<d}qt|}|sv||dfSt|j}t||D]\}}|||<qt||j} ||| |<| dfS)z5Internal auxiliary function for _raw_fft, _raw_fftnd.FNrT) slicerr$rtupler%r.zerosr() rrrZ must_copyrnaxraxiszrrr _fix_shapeps     rEcCs&|dkrtd|t||f|fS)Nrr)rrrE)rrArCrrr _fix_shape_1ds rF)NbackwardorthoforwardcCsFzt|}|r|nd|WStk r@td|ddYnXdS)zAReturns the pypocketfft normalization mode from the norm argumentrGzInvalid norm value z,, should be "backward", "ortho" or "forward"N) _NORM_MAPKeyErrorr)normrJZinormrrr _normalizations rNcCs\|dkrttddS|dkrH|t kr4|dt7}qXtd|t n|dkrXtd|S)Ndefault_workersrrz>> 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) get_workersrTrrrQrO)rSZ old_workersrrr set_workerss  rVcCs 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 rOr)rPrQrrrr rUs rU)N)numbersrros threading contextlibnumpyr. pypocketfftrlocalrQ cpu_countrRrr'r7r=rErFrKrNrTcontextmanagerrVrUrrrr s(   '