_MhzdZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZ d"ddd Zejed Zd e_ejedZd e_ d"ddd Zejed Zde_ejedZde_ d"dddZejed Zde_ejedZde_d#dddZd#dddZ d$dddZejed Zde_ejedZde_ d$dddZejed Z de _ejedZ!de!_ d$dddZ"eje"d Z#de#_eje"dZ$de$_d%dZ%eje%d Z&d e&_eje%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|tdt|}|pt||}t||}t |}|t |||\}} |p| }n1|j|dkr d|j|d} t| |r|jj dkr|nd} tj ||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 r shape ValueErrordtypekindpfftc2c) forwardxnaxisnorm overwrite_xworkersr tmpcopiedmessageouts Z/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/fft/_pocketfft/basic.pyrr s !#EFF F A,,C4S!!4!4K $ ( (DwG}#CD11 V!+V 41  PCIdOPPP!!! A#).C"7"733TC 8C$'4g > >>Tfftifftc|tdt|}t||}t|}t j|st d|t|||\}} n/|j|dkrtd|j|dtj ||f||d|S)z8 Discrete Fourier transform of a real sequence. Nrx must be a real sequencerrr) rrr r np isrealobj TypeErrorr rrrr2c rrrrrrr r r!_s r%r.r.(s  !#EFF F A,,C $ ( (DwG <  53444}sAt,,QQ 41  W39T?WWWXXX 8C$'4w ? ??r&rfftihfftc z|tdt|}t||}t|}t j|r|dz}|-|j|dz dz}|dkrtd|dnt||dzdz|\}} tj ||f|||d|S)zG Return inverse discrete Fourier transform of real sequence x. NrrzInvalid number of data points (r) rrr r r+r,rrr rc2rr/s r%r6r6Fs  !#EFF F A,,C $ ( (DwG |CCi y Yt_q A % q55MqMMMNN N sQTQJ55Q 8C$!WdD' B BBr&hfftirfftr cL|tdt||||||S)z@ 2-D discrete Fourier transform of a Hermitian sequence Nr)rhfftnrsaxesrrr r s r%hfft2r@hs=  !#EFF F AtT; 8 88r&cL|tdt||||||S)zH 2-D discrete inverse Fourier transform of a Hermitian sequence Nr)rihfftnr=s r%ihfft2rCss=  !#EFF F !QdK 9 99r&c|tdt|}t|||\} }|pt||}t |}t |dkr|St || |\}} |p| }t||}|r|jj dkr|nd} tj ||||| |S)z= Return multidimensional discrete Fourier transform. Nrrr) rrrrr lenrr rrrr) rrr>r?rrr r r!rr"r$s r%c2cnrF~s  !#EFF F A,,C)#q$77KE44S!!4!4KwG 4yyA~~S%..KC'K $ ( (D A#).C"7"733TC 8Cwc7 ; ;;r&fftnifftnc|tdt|}tj|st dt |||\} }t || |\}} t||}t|}t|dkrtdtj ||||d|S)z@Return multidimensional discrete Fourier transform of real inputNrr*r#at least 1 axis must be transformed) rrr+r,r-rrr r rErrr.) rrr>r?rrr r r!rr0s r%r2cnrKs !#EFF F A,,C <  53444)#q$77KE4 UD ) )FC $ ( (DwG 4yyA~~>??? 8CwdG < <r?rrr r r!noshaperlastsizer0s r%c2rnrRs2 !#EFF F A,,C |CCi4iG)#q$77KE4 4yyA~~>??? KKE0WT"X&*a/b $ ( (DwGRyHra1$E"I :c5$// 0 0FC 8Cx$g F FFr&r<irfftnc t|}|pt||}t||}td}|jjdkrt d|t|||\}}|p|}n/|j|dkrtd|j|d|r|nd} tj ||f|||| |S)z=FFT of a real sequence, returning fftpack half complex formatNrr*rrr) rrr r rrr-r rrr r2r_fftpack) rrrrrrr!r r"r$s r%rUrUs A,,C4S!!4!4K $ ( (DtnnG y~3444}#CD11 V!+V 41  W39T?WWWXXX '334C  C$'7D#w O OOr& rfft_fftpack irfft_fftpack)Nr NFN)Nr9NFN)NNNFN)Nr NF)(__doc__numpyr+ functoolsrrhelperrrrrr r r rpartialr'__name__r(r.r1r2r6r7r8r@rCrFrGrHrKrLrBrRr<rSrUrVrWr&r%r`s!!!!!!=B?"?????.iT"" ye$$ =B@"@@@@@0yd##  #u%%=BC"CCCCC8yd##  #u%%999999::::::@E<#<<<<<4yt$$  $&&?D=#=====.  $%%  4 ' '@E G# G G G G GF  $%%  4 ' 'PPPP*!y d33 & ! !+u55 ( r&