B 0dB@sddlmZddlZddlZddlZddlZddlZddlm Z e Z e Z dddZddZd d Zd d Zd dZddZdddddZddZddZejddZddZdS))NumberN) good_sizec Csbt|tr|f}ydd|D}Wn:tk r\}z|p:d}td||Wdd}~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).0arrM/var/www/html/venv/lib/python3.7/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)r)ndim)rr )rrr r 0sz+_init_nd_shape_and_axes..c3s |]}|jkp|dkVqdS)rN)r)rr )rrr 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)rsr )rrr r Bscsg|]}j|qSr)r)rr )rrr r Gscss|]}|dkVqdS)rNr)rrrrr rIsz-invalid number of data points ({0}) specified) ranyrlensetrrangeziplistrr)rrrZnoshapeZnoaxesr)rr _init_nd_shape_and_axes)s4     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. dtypeZfc=ZALIGNED)r"copy) hasattrnpZasarrayr"Zfloat16Zfloat32kindZfloat64Z newbyteorderflagsarray)rr"r$rrr _asfarrayPs      r*cCs0||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) FZ __array__N)r r&Zndarrayr%base)Zarroriginalrrr _datacopiedds r-c Csd}tdg|j}xLt||D]>\}}|j||krFtd|||<q td|j|||<d}q Wt|}|sz||dfSt|j}xt||D]\}}|||<qWt||j} ||| |<| dfS)z5Internal auxiliary function for _raw_fft, _raw_fftnd.FNrT) slicerrrtupler r&Zzerosr") rrrZ must_copyrnZaxraxiszrrr _fix_shapeps     r3cCs&|dkrtd|t||f|fS)Nrz-invalid number of data points ({0}) specified)rrr3)rr0r1rrr _fix_shape_1ds r4)NZbackwardZorthoforwardcCsDyt|}|r|Sd|Stk r>td|ddYnXdS)zAReturns the pypocketfft normalization mode from the norm argumentr5zInvalid norm value z,, should be "backward", "ortho" or "forward"N) _NORM_MAPKeyErrorr)Znormr6Zinormrrr _normalizations r9cCs\|dkrttddS|dkrH|t kr4|dt7}qXtd|t n|dkrXtd|S)Ndefault_workersrrz>> 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_workersr?rrr<r:)r>Z old_workersrrr set_workerss  rAcCs 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 r:r)r;r<rrrr r@s r@)N)numbersrros threading contextlibnumpyr&Z pypocketfftrlocalr< cpu_countr=rr!r*r-r3r4r7r9r?contextmanagerrAr@rrrr s&   '