a d8@sddlmZmZddlZddlmZmZmZm Z m Z m Z m Z ddl mZmZmZmZmZmZmZddlmZmZgdZdd d Zd dZdddZddZdddZddZdS))partialreduceN)_prep_axes_wavedecnwavedecwavedec2wavedecnwaverecwaverec2waverecn)iswtiswt2iswtnswtswt2 swt_max_levelswtn)_modes_per_axis_wavelets_per_axis)mramra2mranimraimra2imranr periodizationcCsX|dkrR|dkrtdt||dd}ttf|dd|}ttfi|}d} nN|dkrt|||d}ttfd |i|}ttfi|}d } ntd |||} g} t| } | rt | d } | g| }nd d| D}t | D]j}| |||<||}|j |j kr&|t dd|j D}| || r@| ||<qt ||||<q| S)aForward 1D multiresolution analysis. It is a projection onto the wavelet subspaces. Parameters ---------- data: array_like Input data wavelet : Wavelet object or name string Wavelet to use level : int, optional Decomposition level (must be >= 0). If level is None (default) then it will be calculated using the `dwt_max_level` function. axis: int, optional Axis over which to compute the DWT. If not given, the last axis is used. Currently only available when ``transform='dwt'``. transform : {'dwt', 'swt'} Whether to use the DWT or SWT for the transforms. mode : str, optional Signal extension mode, see `Modes` (default: 'symmetric'). This option is only used when transform='dwt'. Returns ------- [cAn, {details_level_n}, ... {details_level_1}] : list For more information, see the detailed description in `wavedec` See Also -------- imra, swt Notes ----- This is sometimes referred to as an additive decomposition because the inverse transform (``imra``) is just the sum of the coefficient arrays [1]_. The decomposition using ``transform='dwt'`` corresponds to section 2.2 while that using an undecimated transform (``transform='swt'``) is described in section 3.2 and appendix A. This transform does not share the variance partition property of ``swt`` with `norm=True`. It does however, result in coefficients that are temporally aligned regardless of the symmetry of the wavelet used. The redundancy of this transform is ``(level + 1)``. References ---------- .. [1] Donald B. Percival and Harold O. Mofjeld. Analysis of Subtidal Coastal Sea Level Fluctuations Using Wavelets. Journal of the American Statistical Association Vol. 92, No. 439 (Sep., 1997), pp. 868-880. https://doi.org/10.2307/2965551 rr0transform swt only supports mode='periodization'T)waveletaxisnormlevelZ trim_approxZdwt)rmoderr"Funrecognized transform: {}rcSsg|]}t|qSnp zeros_like.0cr%r%R/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/pywt/_mra.py _zmra..cSsg|] }t|qSr%slicer*szr%r%r,r-ir.) ValueErrordictrrr rr formatlenr'r(rangeshapetupleappend)datarr"r transformr#kwargsforwardinverseZis_swt wav_coeffs mra_coeffsncztmpjrecr%r%r,r s@7     rcCstdd|S)aWInverse 1D multiresolution analysis via summation. Parameters ---------- mra_coeffs : list of ndarray Multiresolution analysis coefficients as returned by `mra`. Returns ------- rec : ndarray The reconstructed signal. See Also -------- mra References ---------- .. [1] Donald B. Percival and Harold O. Mofjeld. Analysis of Subtidal Coastal Sea Level Fluctuations Using Wavelets. Journal of the American Statistical Association Vol. 92, No. 439 (Sep., 1997), pp. 868-880. https://doi.org/10.2307/2965551 cSs||SNr%)xyr%r%r,r.zimra..)r)rAr%r%r,rtsrrrcCs|dkrj|dkrtd|dur4tdd|jD}t||dd}ttf|dd |}ttfi|}nJ|d krt|||d }ttfd |i|}ttfi|}ntd |||} g} t | } t | d} | g} t d| D]}| dd| |Dq| d| d<|| }|j|jkr>|tdd|jD}| || | d<t d| D]}g}t dD]j}| ||} | ||| ||<|| }|j|jkr|tdd|jD}||| | ||<qj| t|qZ| S)aForward 2D multiresolution analysis. It is a projection onto wavelet subspaces. Parameters ---------- data: array_like Input data wavelet : Wavelet object or name string, or 2-tuple of wavelets Wavelet to use. This can also be a tuple containing a wavelet to apply along each axis in `axes`. level : int, optional Decomposition level (must be >= 0). If level is None (default) then it will be calculated using the `dwt_max_level` function. axes : 2-tuple of ints, optional Axes over which to compute the DWT. Repeated elements are not allowed. Currently only available when ``transform='dwt2'``. transform : {'dwt2', 'swt2'} Whether to use the DWT or SWT for the transforms. mode : str or 2-tuple of str, optional Signal extension mode, see `Modes` (default: 'symmetric'). This option is only used when transform='dwt2'. Returns ------- coeffs : list For more information, see the detailed description in `wavedec2` Notes ----- This is sometimes referred to as an additive decomposition because the inverse transform (``imra2``) is just the sum of the coefficient arrays [1]_. The decomposition using ``transform='dwt'`` corresponds to section 2.2 while that using an undecimated transform (``transform='swt'``) is described in section 3.2 and appendix A. This transform does not share the variance partition property of ``swt2`` with `norm=True`. It does however, result in coefficients that are temporally aligned regardless of the symmetry of the wavelet used. The redundancy of this transform is ``3 * level + 1``. See Also -------- imra2, swt2 References ---------- .. [1] Donald B. Percival and Harold O. Mofjeld. Analysis of Subtidal Coastal Sea Level Fluctuations Using Wavelets. Journal of the American Statistical Association Vol. 92, No. 439 (Sep., 1997), pp. 868-880. https://doi.org/10.2307/2965551 rrrNcss|]}t|VqdSrGrr*sr%r%r, r.zmra2..Traxesr r!Zdwt2rr#rRr"r$rrcSsg|]}t|qSr%r&r)r%r%r,r-r.zmra2..cSsg|] }t|qSr%r/r1r%r%r,r-r.cSsg|] }t|qSr%r/r1r%r%r,r-r.)r3minr8r4rrr rr r5r6r'r(r7r:r9)r;rr"rRr<r#r=r>r?r@rArBrCrDrErFdcoeffsnr%r%r,rsP7     rcCs>|d}tdt|D]"}tdD]}||||7}q"q|S)aNInverse 2D multiresolution analysis via summation. Parameters ---------- mra_coeffs : list Multiresolution analysis coefficients as returned by `mra2`. Returns ------- rec : ndarray The reconstructed signal. See Also -------- mra2 References ---------- .. [1] Donald B. Percival and Harold O. Mofjeld. Analysis of Subtidal Coastal Sea Level Fluctuations Using Wavelets. Journal of the American Statistical Association Vol. 92, No. 439 (Sep., 1997), pp. 868-880. https://doi.org/10.2307/2965551 rrrT)r7r6)rArFrErWr%r%r,rs  rrcCst|j|\}}}t||}|dkr|dkr4td|durPtdd|jD}t||dd} ttf|dd | } ttfi| } nT|d krt ||} t|| |d } tt fd |i| } tt fi| } ntd || |} g}t | }t| d}|g}td|D]"}|dd| |Dq | d|d<| |}|j|jkrj|tdd|jD}||||d<td|D]}i}t| |}|D]h}|||}| |||||<| |}|j|jkr|tdd|jD}|||<||||<q||q|S)aForward nD multiresolution analysis. It is a projection onto the wavelet subspaces. Parameters ---------- data: array_like Input data wavelet : Wavelet object or name string, or tuple of wavelets Wavelet to use. This can also be a tuple containing a wavelet to apply along each axis in `axes`. level : int, optional Decomposition level (must be >= 0). If level is None (default) then it will be calculated using the `dwt_max_level` function. axes : tuple of ints, optional Axes over which to compute the DWT. Repeated elements are not allowed. transform : {'dwtn', 'swtn'} Whether to use the DWT or SWT for the transforms. mode : str or tuple of str, optional Signal extension mode, see `Modes` (default: 'symmetric'). This option is only used when transform='dwtn'. Returns ------- coeffs : list For more information, see the detailed description in `wavedecn`. See Also -------- imran, swtn Notes ----- This is sometimes referred to as an additive decomposition because the inverse transform (``imran``) is just the sum of the coefficient arrays [1]_. The decomposition using ``transform='dwt'`` corresponds to section 2.2 while that using an undecimated transform (``transform='swt'``) is described in section 3.2 and appendix A. This transform does not share the variance partition property of ``swtn`` with `norm=True`. It does however, result in coefficients that are temporally aligned regardless of the symmetry of the wavelet used. The redundancy of this transform is ``(2**n - 1) * level + 1`` where ``n`` corresponds to the number of axes transformed. References ---------- .. [1] Donald B. Percival and Harold O. Mofjeld. Analysis of Subtidal Coastal Sea Level Fluctuations Using Wavelets. Journal of the American Statistical Association Vol. 92, No. 439 (Sep., 1997), pp. 868-880. https://doi.org/10.2307/2965551 rrrNcss|]}t|VqdSrGrMrNr%r%r,rPZr.zmran..TrQr!ZdwtnrSr"r$rrcSsi|]\}}|t|qSr%r&)r*kvr%r%r, mr.zmran..cSsg|] }t|qSr%r/r1r%r%r,r-ur.zmran..cSsg|] }t|qSr%r/r1r%r%r,r-r.)rr8rr3rUr4rrrrrr r5r6r'r(r7r:itemsr9listkeys)r;rr"rRr<r#Z axes_shapesZndim_transformZwaveletsr=r>r?modesr@rArBrCrDrErFrVZdkeysrXr%r%r,rsX7      rcCs>|d}tdt|D]"}||D]\}}||7}q&q|S)aNInverse nD multiresolution analysis via summation. Parameters ---------- mra_coeffs : list Multiresolution analysis coefficients as returned by `mra2`. Returns ------- rec : ndarray The reconstructed signal. See Also -------- mran References ---------- .. [1] Donald B. Percival and Harold O. Mofjeld. Analysis of Subtidal Coastal Sea Level Fluctuations Using Wavelets. Journal of the American Statistical Association Vol. 92, No. 439 (Sep., 1997), pp. 868-880. https://doi.org/10.2307/2965551 rr)r7r6r[)rArFrErXrYr%r%r,rs  r)Nrrr)NrKrr)NNrr) functoolsrrnumpyr'Z _multilevelrrrrr r r Z_swtr r rrrrr_utilsrr__all__rrrrrrr%r%r%r,s"$$ g m t