e#ddlmZmZddlmZmZmZmZddlm Z m Z dgZ ddl Z ddlZejZejZn4#e$r, ddlZejZejZn#e$r e jZdZYnwxYwYnwxYwd d ZdS) )floorceil)DiscreteContinuousWaveletContinuousWaveletWavelet _check_dtype)integrate_waveletscale2frequencycwtNcJdttj|zS)aRound up size to the nearest power of two. Given a number of samples `n`, returns the next power of two following this number to take advantage of FFT speedup. This fallback is less efficient than `scipy.fftpack.next_fast_len` )rnplog2)ns )lib/python3.11/site-packages/pywt/_cwt.py next_fast_lenrsd271::&&& &?convcZ t|}tj||}tj|tj}t |t tfst|}tj |rtj |g}tj |stj d|j r|n|}tj tj|f|jz|} d} t!|| \} } |j rtj| n| } | jjdkr|n|} tj| | } tj| |jj} |dkrd}d}n|d kst+d |jd kr?|d|}|j}|d|jdf}t3|D]\}}| d | d z }tj|| d| d z zd z||zz }|t8}|d| jkrtj|| jk|}| |ddd}|d kr|jd krtj||}n9t?|j}|dxx|jd z z cc<tA|}tj ||}tC|jd D]$}tj|||||ddf<%ntE|jd|jzd z }||krtF$||d }|}tF$||d }tF%||zd }|dd|jd|jzd z f}tj&| tj'|d z}| jjdkr|j}|jd|jdz dz }|d kr(|dtQ|tS| f}n(|d kr"t+d*||jd kr+||}||d}|| |df<tW||| }tj |rtj |g}||z}| |fS)aH cwt(data, scales, wavelet) One dimensional Continuous Wavelet Transform. Parameters ---------- data : array_like Input signal scales : array_like The wavelet scales to use. One can use ``f = scale2frequency(wavelet, scale)/sampling_period`` to determine what physical frequency, ``f``. Here, ``f`` is in hertz when the ``sampling_period`` is given in seconds. wavelet : Wavelet object or name Wavelet to use sampling_period : float Sampling period for the frequencies output (optional). The values computed for ``coefs`` are independent of the choice of ``sampling_period`` (i.e. ``scales`` is not scaled by the sampling period). method : {'conv', 'fft'}, optional The method used to compute the CWT. Can be any of: - ``conv`` uses ``numpy.convolve``. - ``fft`` uses frequency domain convolution. - ``auto`` uses automatic selection based on an estimate of the computational complexity at each scale. The ``conv`` method complexity is ``O(len(scale) * len(data))``. The ``fft`` method is ``O(N * log2(N))`` with ``N = len(scale) + len(data) - 1``. It is well suited for large size signals but slightly slower than ``conv`` on small ones. axis: int, optional Axis over which to compute the CWT. If not given, the last axis is used. Returns ------- coefs : array_like Continuous wavelet transform of the input signal for the given scales and wavelet. The first axis of ``coefs`` corresponds to the scales. The remaining axes match the shape of ``data``. frequencies : array_like If the unit of sampling period are seconds and given, then frequencies are in hertz. Otherwise, a sampling period of 1 is assumed. Notes ----- Size of coefficients arrays depends on the length of the input array and the length of given scales. Examples -------- >>> import pywt >>> import numpy as np >>> import matplotlib.pyplot as plt >>> x = np.arange(512) >>> y = np.sin(2*np.pi*x/32) >>> coef, freqs=pywt.cwt(y,np.arange(1,129),'gaus1') >>> plt.matshow(coef) # doctest: +SKIP >>> plt.show() # doctest: +SKIP ---------- >>> import pywt >>> import numpy as np >>> import matplotlib.pyplot as plt >>> t = np.linspace(-1, 1, 200, endpoint=False) >>> sig = np.cos(2 * np.pi * 7 * t) + np.real(np.exp(-7*(t-0.4)**2)*np.exp(1j*2*np.pi*2*(t-0.4))) >>> widths = np.arange(1, 31) >>> cwtmatr, freqs = pywt.cwt(sig, widths, 'mexh') >>> plt.imshow(cwtmatr, extent=[-1, 1, 1, 31], cmap='PRGn', aspect='auto', ... vmax=abs(cwtmatr).max(), vmin=-abs(cwtmatr).max()) # doctest: +SKIP >>> plt.show() # doctest: +SKIP )dtypezaxis must be a scalar. ) precisioncfftrNrzmethod must be 'conv' or 'fft'rr)axis.g@zSelected scale of {} too small.),r rasarray result_type complex64 isinstancerrrisscalararray AxisError complex_cwtemptysizeshaper conjrkindreal ValueErrorndimswapaxesreshape enumeratearangeastypeintextractconvolvelisttupleranger fftmodulerifftsqrtdiffrrformatr )datascaleswaveletsampling_periodmethodrdtdt_cplxdt_outoutrint_psixdt_psi size_scale0fft_datadata_shape_preiscalestepj int_psi_scaler conv_shaper size_scalefft_wavcoefd frequenciess rr r %sX d  B :d" % % %DnR..G g 17; < <5+G44 {6$6(## ;t  5l3444+ 3WWF (BGFOO% 2& A A ACI"7i@@@JGQ")"5Bbgg7G *c11WWrFj///G 1DIO,,,A   v  9::: y1}}}}R&&||RB011f%%115tad{ Iequqt|,q0 1 1UT\ B HHSMM R5GL  1w|+Q//A 44R4( V  yA~~{477"$*-- 2-"4q"88":.. x &999tz!}--EEA!#T!Wm!D!DDAAAJJE' 2!33a7J[(($==z=CC$KmmM:BmGGG>>'H"42>>>DEdjn}/AAAEEEFD"'$R"8"8"88 9>S 9D Z^djn , 2 q55U1XXtAwwh../DD UU188??AA A 9q==<<//D==r**DAsF !'69==K {;.h }-- ?"K  r)rrr)mathrr_extensions._pywtrrrr _functionsr r __all__numpyr scipy.fftscipyrr:r ImportError scipy.fftpackfftpackr rrrdsH777777777777:::::::: '' I+MM''''M !/  ' ' 'F  ' ' ' ' ' ' '&ffffffs28A)AA)A# A)"A##A)(A)