Ei/<SSKJr SSKrSSKrSSKrSSKrSSKrSSKJ r SSK J r SSK J r Jr /SQr\R "5r\R$"5rSSjrS rS rS rS rS rSSSSS.rSrSr\ "SS9\R:S55r\ "SS9S5rg))NumberN)copy_if_needed)xp_capabilities) good_sizeprev_good_size)rr set_workers get_workersc[U[5(aU4nUVs/sHn[R"U5PM nnU$s snf![anU=(d Sn[ US35UeSnAff=f)zConvert ``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]`` valuez) must be a scalar or iterable of integersN) isinstanceroperatorindex TypeError ValueError)xnameaes X/home/mande/repo/quber/.venv/lib/python3.13/site-packages/scipy/fft/_pocketfft/helper.py_iterable_of_intrsx!V DT() *1X^^A  * H + TwD6!JKLRSSTs'A AAA A0A++A0c^USLnUSLnU(d[US5nUVs/sHoUS:aUTR-OUPM nn[U4SjU55(a [S5e[ [ U55[ U5:wa [S5eU(d[US5nU(a#[ U5[ U5:wa [S5eU(aP[ U5TR:a [S 5e[ TR[ U5- TR5n[X5VVs/sHupeUS :XaTRUOUPM nnnOQU(a+[TR5n[ TR5nOUVs/sHnTRUPM nn[S U55(a[S US 35e[U5[U54$s snfs snnfs snf)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. Naxesrc3X># UHoTR:=(d US:v M! g7f)rN)ndim).0rrs r *_init_nd_shape_and_axes..Ts#2TAFF{#a!e#Ts'*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 presentc3*# UH oS:v M g7f)rN)rss rrrks %Qq5%sinvalid number of data points ( ) specified) rranyrlensetrangeziprlisttuple)rrrnoshapenoaxesrr#s` r_init_nd_shape_and_axesr//s<tmG T\F f-489Dq!eAFF *D9 2T2 2 2CD D s4y>SY &67 7  0 CIU+>? ? 5zAFF" !LMM#e*,aff5D:=e:JK:J$!qBwA-:JK QWW QVV}%)*TT* %   -eWK @B B <d ##;:$L +s G*6#G/G5c[US5(d[R"U5nUR[R:Xa%[R"U[R 5$URR S;a%[R"U[R5$URRS5nURS(dSO[n[R"XUS9$)z` Convert to array with floating or complex dtype. float16 values are also promoted to float32. dtypefc=ALIGNEDT)r1copy) hasattrnpasarrayr1float16float32kindfloat64 newbyteorderflagsrarray)rr1r5s r _asfarrayr@rs 1g   JJqMww"**zz!RZZ(( T !zz!RZZ(( GG  %Ewwy)4~D 88A ..cXLag[U[R5(d[US5(agURSL$)zp Strict check for `arr` not sharing any data with `original`, under the assumption that arr = asarray(original) F __array__N)r r7ndarrayr6base)arroriginals r _datacopiedrHs;   h + ++0N0N 88t rAcSn[S5/UR-n[X5HEupVURUU:a[SU5XF'M([SURU5XF'SnMG [ U5nU(dXS4$[ UR5n[X5H upXXWU'M [ R"XpR5n XX'U S4$)z5Internal auxiliary function for _raw_fft, _raw_fftnd.FNrT) slicerr*rr,r+r7zerosr1) rrr must_copyrnaxr#axiszs r _fix_shaperQsI4[M!&& EU! 772;! a EIa-EII " %LE x QWW Au#$$ GGAxAH d7NrAcHUS:a[SUS35e[X4U45$)Nrr$r%)rrQ)rrMrOs r _fix_shape_1drSs51u-aS <> > aw ''rA)Nbackwardorthoforwardcp[UnU(aU$SU- $![a [SU<S35Sef=f)zAReturns the pypocketfft normalization mode from the norm argumentrTzInvalid norm value z,, should be "backward", "ortho" or "forward"N) _NORM_MAPKeyErrorr)normrWinorms r_normalizationr]sQ=$u0a%i0 =!$*2 238< ==s 5cUc[[SS5$US:a/U[*:aUS[-- nU$[SUS[*35eUS:Xa [S5eU$)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_workersrds}w 1155{ zk ! q:~ %G N ?yI+,6;-9: : A344 NrAT) out_of_scopec## [5n[[R"U55[lSv U[lg!U[lf=f7f)alContext 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 rdrrrar_)rc old_workerss rr r s<(-K&x~~g'>?G. "-+s3AA A AAc$[[SS5$)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`rar"rArr r s 7-q 11rA)N) numbersrros threading contextlibnumpyr7scipy._lib._utilrscipy._lib._array_apir pypocketfftrr__all__localra cpu_countrbrr/r@rHrQrSrYr]rdcontextmanagerr r r"rArrus +13 H //  \\^  2@$F/( 6(!aA > =  d# .$.4d# 2$ 2rA