o ?Hh@sdZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZ  d+ddd d Zeed Zd e_eedZd e_  d+ddddZeed Zde_eedZde_  d+ddddZeed Zde_eedZde_d,ddddZd,ddddZ  d-ddddZeed Zde_eedZde_  d-dddd Zeed Z d!e _eedZ!d"e!_  d-ddd#d$Z"ee"d Z#d%e#_ee"dZ$d&e$_d.d'd(Z%ee%d Z&d)e&_ee%dZ'd*e'_dS)/z( Discrete Fourier Transforms - basic.py N) pypocketfft) _asfarray_init_nd_shape_and_axes _datacopied _fix_shape _fix_shape_1d_normalization_workersF)planc Cs|durtdt|}|pt||}t||}t|}|dur-t|||\}} |p+| }n|j|dkrAd|j|d} t| |rK|jj dkrK|nd} t ||f||| |S)z@ Return discrete Fourier transform of real or complex sequence. NFPassing a precomputed plan is not yet supported by scipy.fft functionsrinvalid number of data points ( ) specifiedc) NotImplementedErrorrrr r rshape ValueErrordtypekindpfftc2c) forwardxnaxisnorm overwrite_xworkersr tmpcopiedmessageoutr#Z/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/scipy/fft/_pocketfft/basic.pyr s  rTfftifftc Cs|durtdt|}t||}t|}t|std|dur+t|||\}} n|j|dkr=t d|j|dt ||f||d|S)z8 Discrete Fourier transform of a real sequence. Nr x must be a real sequencerrr) rrr r np isrealobj TypeErrorrrrrr2c rrrrrrrr r_r#r#r$r+(s  r+rfftihfftc Cs|durtdt|}t||}t|}t|r|d}|dur8|j|dd}|dkr7td|dn t||dd|\}} t ||f|||d|S)zG Return inverse discrete Fourier transform of real sequence x. Nr rzInvalid number of data points (r) rrr r r(r)rrrrc2rr,r#r#r$r2Fs  r2hfftirfftr cC"|durtdt||||||S)z@ 2-D discrete Fourier transform of a Hermitian sequence Nr )rhfftnrsaxesrrrr r#r#r$hfft2hr<cCr7)zH 2-D discrete inverse Fourier transform of a Hermitian sequence Nr )rihfftnr9r#r#r$ihfft2sr=r?c Cs|durtdt|}t|||\} }|pt||}t|}t|dkr'|St|| |\}} |p2| }t||}|rB|jj dkrB|nd} t ||||| |S)z= Return multidimensional discrete Fourier transform. Nr rr) rrrrr lenrr rrrr) rrr:r;rrrr rrr r"r#r#r$c2cn~s  rAfftnifftnc Cs|durtdt|}t|stdt|||\} }t|| |\}} t||}t|}t |dkr8t dt ||||d|S)z@Return multidimensional discrete Fourier transform of real inputNr r'r#at least 1 axis must be transformed) rrr(r)r*rrr r r@rrr+) rrr:r;rrrr rrr-r#r#r$r2cns   rErfftnr>c Cs|durtdt|}t|r|d}|du} t|||\} }t|dkr+tdt| } | r>|j|ddd| d<t ||}t |}| d} | ddd| d<t t || |\}} t ||| ||d|S) zDMultidimensional inverse discrete fourier transform with real outputNr r0rrDr rr1)rrr(r)rr@rlistrr r tuplerrr2) rrr:r;rrrr rnoshaperlastsizer-r#r#r$c2rns$   rKr8irfftnc Cst|}|p t||}t||}td}|jjdkrtd|dur/t|||\}}|p-|}n|j|dkrAt d|j|d|rE|nd} t ||f|||| |S)z=FFT of a real sequence, returning fftpack half complex formatNrr'rrr) rrr r rrr*rrrr r2r_fftpack) rrrrrrrrr r"r#r#r$rMs    rM rfft_fftpack irfft_fftpack)Nr NFN)Nr5NFN)NNNFN)Nr NF)(__doc__numpyr( functoolsrrhelperrrrrrr r rpartialr%__name__r&r+r.r/r2r3r4r<r?rArBrCrErFr>rKr8rLrMrNrOr#r#r#r$s $               #