PhNDdZddlZddlmZmZddlZddlmZddl m Z ddl m Z m Z m Z mZddlmZmZmZmZddlmZdd lmZmZmZ dd Z dd eej8ej8eeefffd Z dd Z ddZ ddZ! ddZ" ddZ# ddZ$y)anHelpers to utilize existing stft / istft tests for testing `ShortTimeFFT`. This module provides the functions stft_compare() and istft_compare(), which, compares the output between the existing (i)stft() and the shortTimeFFT based _(i)stft_wrapper() implementations in this module. For testing add the following imports to the file ``tests/test_spectral.py``:: from ._scipy_spectral_test_shim import stft_compare as stft from ._scipy_spectral_test_shim import istft_compare as istft and remove the existing imports of stft and istft. The idea of these wrappers is not to provide a backward-compatible interface but to demonstrate that the ShortTimeFFT implementation is at least as capable as the existing one and delivers comparable results. Furthermore, the wrappers highlight the different philosophies of the implementations, especially in the border handling. N)castLiteral)assert_allclose) ShortTimeFFT)csd get_windowstftistft) const_exteven_extodd_extzero_ext) FFT_MODE_TYPE)_spectral_helper_triage_segments _median_biasc h| dvrtd| dttttdd} || vr,td|ddt | j z|jd k(r\tj|jtj|jtj|jfS|t|}|d kr td t|||j|  \} }||}n||kr td t|}||dz}n t|}||k\r td||z }|j| }|| |}|||dz| }| rtj|| d}|dzd k(r|dzd k(r|dzd k(r |dddf}|jd|z |z|z}t |jdd|gz}tj|tj|fd}tj|d| }ddd| }tj |r|rd}t#t$|rdnd}t'| |||||d}|dz}d }| r|j| n||zd z}|j| |z |zd z}|durdn|}|j)|||||| }|j+|d ||z |d n|}|j,tj.tj0fvr|j3tj0}|j4||fS)aOWrapper for the SciPy `stft()` function based on `ShortTimeFFT` for unit testing. Handling the boundary and padding is where `ShortTimeFFT` and `stft()` differ in behavior. Parts of `_spectral_helper()` were copied to mimic the` stft()` behavior. This function is meant to be solely used by `stft_compare()`. )psdspectrumzParameter scaling=z not in ['spectrum', 'psd']!N)evenoddconstantzerosNzUnknown boundary option 'z', must be onez of: r"nperseg must be a positive integer input_length.nfft must be greater than or equal to nperseg.#noverlap must be less than nperseg.axis. magnituderrrFonesidedtwosidedfft_modemfftscale_to phase_shift)k_offsetr")r-) ValueErrorr r r rlistkeyssizenpemptyshapeintrmoveaxis concatenater iscomplexobjrrr stft_detrendtdtypefloat32 complex64astypef)xfswindownpersegnoverlapnfftdetrendreturn_onesidedboundarypaddedr"scalingboundary_funcswinnstepnext_funcnadd zeros_shaper+r)STk_offp0nnp1detrSxxr:s wC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\scipy/signal/tests/_scipy_spectral_test_shim.py _stft_wrapperrZ"sY)).gZ/KLMM'$"+' "N ~%4XJnM n&9&9&;!< =>?@ @vv{xx "((177"3RXXagg5FFFg, Q;AB B$FG12@LC | IJJ4yA:x=7>?? h E  A!(+ Q  .  KK4 $ q5A:'A+*x!|q/@#tt) A''"+g%&.'91773B<(D61 NNArxx 45B ? KK2t $(6w?H qoM:jQH c5"xd'T ;B qLE B agaiB ''$-' !e +a /Be#4D //!T2rE/ EC RBG8+?aUKAww2::r||,,jj& 44C<returnc |jdkr td| |k(r td|j|} |rd|j| dz z} n|j| } || }nt|}|dkr td||r || dzk(r|}n| }n||kr tdt|}||dz}n t|}||k\r td||z } t |t st |tur t||}nXtj|}t|jdk7r td |jd |k7rtd ||| dz | zz}tt|rd nd }ttdddd| }t|| ||||d}|r*|dzd k(r|n|dz }|j |dzz}||z |z}n t#d|j%|||| |}tj&||z |j(z}|j+||z d }|||j,d |ffS)a.Wrapper for the SciPy `istft()` function based on `ShortTimeFFT` for unit testing. Note that only option handling is implemented as far as to handle the unit tests. E.g., the case ``nperseg=None`` is not handled. This function is meant to be solely used by `istft_compare()`. rzInput stft must be at least 2d!z/Must specify differing time and frequency axes!rNrrr zwindow must be 1-Drzwindow must have length of r&r')r$rr$rr%r(z?? h E&#$v,%"7)jj  syy>Q 12 2 99Q<7 ":7)DE Ed1fe^+LM:ZPHG./!,U;GDFH c5"xd'T ;B{a'GWq[ XX1 $ A  "!#:; ; 9YGA "r'RTT!A b )! ,D a"%%a($/ //r[c t|||||||||| | \} } }t|jdk\r|jdkDr |jddkDr| dk(rt |jd}t j |rYt jt j|ddt jt j|dzz}nt j|d}||z}| |fS| dk(r|jd}| |fStd | t j||jd d}| |fS) zWWrapper for the `csd()` function based on `ShortTimeFFT` for unit testing. rrr#rmedianr!y?meanz(average must be "median" or "mean", got N) _csd_test_shimrhr4r1rr2r8rzrealimagr{r.reshape)r@yrArBrCrDrErFrGrJr"averagefreqs_Pxybiass rY _csd_wrapperrs= #1aVWh#*OWdLME1c 399~sxx!| 99R=1 ("#CIIbM2??3'99RWWS\;"))BGGCLr"BBCC))Cb1Ct  #:F"hhBh' #: !#KG9!UVV**S#))CR.1C #:r[c t|||||||||| | d \} } } t|||||||||| | \}}tjj || | j r-t tj| j dnd}tj| jj|z}tjj || || | | fS)aVCompare output of _spectral_helper() and ShortTimeFFT, more precisely _spect_helper_csd() for used in csd_wrapper(). The motivation of this function is to test if the ShortTimeFFT-based wrapper `_spect_helper_csd()` returns the same values as `_spectral_helper`. This function should only be usd by csd() in (unit) testing. r)moder)atol) r_spect_helper_csdr2testingrr1maxabsfinfor; resolution)r@rrArBrCrDrErFrGrJr"rr:rfreqs1Pxy1amax_Pxyrs rYr|r|s%Q2vw$%,ow*/1ME1c%Q2vw$%,owNLFDJJvu-,/HHs266#;??$a(!H 88CII  ) )H 4DJJtSt4 !S=r[c ||u} t| } tj|}| stj|}| st|j} t|j} | j | | j |  tj tj| tj| j}| rM|jdk(rtj|jtj|jfS|jdk(s|jdk(rYt|j| |j| gfz}tjtj|d| }||fS|t|}|dkr td| r|j| n%t|j| |j| }t|||\}}||}n||kr tdt|}||d z}n t|}||k\r td ||z }tj|r|rd }tt |rd nd }ddd| }t#||||||d}|j%|||d urdn|d||z |j&z|d z| j)}|rl| dkr|j*dz | zn| }tj||d}|dd|j,d zdk(rdndfxxd zcc<tj|d|}|j.|fS#t$r}td|d}~wwxYw)zWrapper for replacing _spectral_helper() by using the ShortTimeFFT for use by csd(). This function should be only used by _csd_test_shim() and is only useful for testing the ShortTimeFFT implementation. z%x and y cannot be broadcast together.Nrr#rrrrrr Fr&r'r$r)rdensityr()rWrTrVr-r".)r5r2rgr/r4pop broadcastr3r.r1minr6rrr8rrr spectrogramhopconjrbr*r?)r@rrArBrCrDrErFrGrJr" same_dataxouteryouter outershapeeoutshapeemptyoutrNrLrMr)scaleSFTrr`s rYrrs#QI t9D 1 A  JJqM aggagg 4 4 Mbhhv&68HIOOJ 66Q;88AGG$bhhqww&77 7 66Q;!&&A+!S!''$-)G%H$JJH{{288H#5r4@HX% %g, Q;AB B# AGGDM1774=(IA#FG!DLC | IJJ4ya<x=7>?? h E qoM:#%H$ 7 @E sE2t %4 9C //!QW-=T7AhJ#87A:#  %%)TV(,qA$dkk#vr* CA*24 45:5kk#r6* 55#:{ MDE1 L Ms AM M  MM c :t|||||||||| | |  } td i| \} }}td i| \}}}d}t|| d|t||d|t j |j jdz}t|||d|| ||fS) aAssert that the results from the existing `stft()` and `_stft_wrapper()` are close to each other. For comparing the STFT values an absolute tolerance of the floating point resolution was added to circumvent problems with the following tests: * For float32 the tolerances are much higher in TestSTFT.test_roundtrip_float32()). * The TestSTFT.test_roundtrip_scaling() has a high relative deviation. Interestingly this did not appear in Scipy 1.9.1 but only in the current development version. ) r@rArBrCrDrErFrGrHrIr"rJz* of `stft_wrapper()` differ from `stft()`.z Frequencies err_msgz Time slices rz STFT values )rr)dictr rZrr2rr;r)r@rArBrCrDrErFrGrHrIr"rJkwr?r:ro f_wrapper t_wrapper Zxx_wrapper e_msg_partrs rY stft_comparer{s b8/T7 LB r IAq#(5(;(;%Iy+=JIqL *EFIqL *EF 88CII  ) )A -DK4*:,79 a9r[c Lt|||||||||| |  } tdi| \} } |s| | fS tdi| \}}\}}d}t | |d|t j| jjdz}d }| jt jk(rtjd k(rd \}}n"tjd vr t|d }t |||| ||||d|| | fS#t$r"}|jddk(r | | fcYd}~S|d}~wwxYw)aAssert that the results from the existing `istft()` and `_istft_wrapper()` are close to each other. Quirks: * If ``boundary=False`` the comparison is skipped, since it does not make sense with ShortTimeFFT.istft(). Only used in test TestSTFT.test_roundtrip_boundary_extension(). * If ShortTimeFFT.istft() decides the STFT is not invertible, the comparison is skipped, since istft() only emits a warning and does not return a correct result. Only used in ShortTimeFFT.test_roundtrip_not_nola(). * For comparing the signals an absolute tolerance of the floating point resolution was added to account for the low accuracy of float32 (Occurs only in TestSTFT.test_roundtrip_float32()). ) rorArBrCrDrErprHrqrrrJrz,Short-time Fourier Transform not invertible!Nz+ of `istft_wrapper()` differ from `istft()`z Sample times rrgHz>i686)g-C6?gh㈵>)aarch64i386rg-q=zSignal values )rrtolrr)rr rxr.argsrr2rr;rr<platformmachiner)rorArBrCrDrErprHrqrrrJrr:r@r x_wrapperk_lorwvrrrs rY istft_comparersK$ #"VWd>9  B ;2;DAq !t -;-Ab-A* 9ltT ?JAyM**FG 88AGG  ' ' )D D  ww"**!1!1!3v!= d    : :4Id4(!D,T,ZL9; a4K1  66!9F Fa4Ks#C88 D#DD#DD#c t|||||||||| | |  } tdi| \} }tdi| \}}t|| t||t|| | |fS)zdAssert that the results from the existing `csd()` and `_csd_wrapper()` are close to each other. ) r@rrArBrCrDrErFrGrJr"rr)rrrr)r@rrArBrCrDrErFrGrJr"rrfreqs0Pxy0rrs rY csd_comparerss Q2fgdG-wT B99LFD%"%LFDFF#D$FF# 4<r[) ?hannNNFTrTr#r) rrNNNTTr#r) rrNNNrTrr#r{) rrNNNrTrr#)%__doc__rtypingrrnumpyr2 numpy.testingr scipy.signalrrrr r scipy.signal._arraytoolsr r r rscipy.signal._short_time_fftrscipy.signal._spectral_pyrrrrZrfndarrayr5rxrr|rrrrrr[rYrs& )%55KK6CG<@BLfRGKLN)3W0 bjj"**eCHo56W0tFJ@D5;>HLBF+-2KOEI.0[|BF;?AK>FJKM(24nEI?C4:r[