geYddlmZddlZddlZddlZddlZddlZddlm Z ej Z ej Z ddZdZdZdZd Zd Zdddd d Zd ZdZejdZdZdS))NumberN) good_sizect|tr|f} d|D}n9#t$r,}|pd}td||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 ;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 ValueErrorformat)xnamees r _iterable_of_intrs!V D/ * * * * * ///wF &,,((-. // Hs ' A'AAcX|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 |d||fS)z2Handles shape and axes arguments for nd transformsNaxesc4g|]}|dkr |jzn|S)rndimr r rs r rz+_init_nd_shape_and_axes..0s+999qa!eeAF 999rc3:K|]}|jkp|dkVdS)rNrrs r z*_init_nd_shape_and_axes..2s222qAF{#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!)r sr rs r rz+_init_nd_shape_and_axes..Bs/KKK$!QqBwwAKKKrc*g|]}j|Srr$rs r rz+_init_nd_shape_and_axes..Gs******rc3"K|] }|dkV dS)rNr)r r%s r r z*_init_nd_shape_and_axes..Is& Q1q5 rinvalid number of data points ( ) specified) ranyrlensetrrangeziplistr!)rr!rnoshapenoaxess` r _init_nd_shape_and_axesr2)stmG 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 $;rct|dstj|}|jtjkrtj|tjS|jjdvrtj|tjS|jd}|j d }tj |||S)zl Convert to array with floating or complex dtype. float16 values are also promoted to float32. dtypefc=ALIGNED)r4copy) hasattrnpasarrayr4float16float32kindfloat64 newbyteorderflagsarray)rr4r8s r _asfarrayrCPs 1g   JqMMw"*z!RZ((( T ! !z!RZ((( G  % %Ewy! !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:ndarrayr9base)arroriginals r _datacopiedrJdsI  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) slicerr.r!tupler/r:zerosr4) rr!r must_copyr naxr%axiszs r _fix_shaperTpsI4[[M!& EUD!!2 72;!  a E"IIa--E"III %LLE x QW Aud##4$ AGAxAeH d7NrcZ|dkrtd|dt||f|fS)Nrr(r))rrT)rrPrRs r _fix_shape_1drVsC1uu > > a!w ' ''r)Nbackwardorthoforwardct t|}|r|nd|z S#t$rtd|ddwxYw)zAReturns the pypocketfft normalization mode from the norm argumentrWzInvalid norm value z,, should be "backward", "ortho" or "forward"N) _NORM_MAPKeyErrorr)normrZinorms r _normalizationr`si=$0uua%i0 === 2$ 2 2 2338< ==s7c|ttddS|dkrC|t kr|dtzz }n>td|t |dkrtd|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_workersrgr r rdrb)rf old_workerss r set_workersrks[&--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 rbr)rcrdrrr riris 7-q 1 11r)N)numbersrr os threading contextlibnumpyr: pypocketfftrlocalrd cpu_countrerr2rCrJrTrVr\r`rgcontextmanagerrkrirrr rvsD """""" )/   R\^^     4$$$N///(   6(((!aA > > ===     ...4 2 2 2 2 2r