_Mh`ddlZddlmZmZmZddgZgdZdZdZ Gd d Z dd Z dS)N) _output_len_apply mode_enumupfirdnr) constantwrapedgesmooth symmetricreflect antisymmetric antireflectlinect|t| |zz}tj||j}||dt|<|d|jdddddf}|S)aStore coefficients in a transposed, flipped arrangement. For example, suppose upRate is 3, and the input number of coefficients is 10, represented as h[0], ..., h[9]. Then the internal buffer will look like this:: h[9], h[6], h[3], h[0], // flipped phase 0 coefs 0, h[7], h[4], h[1], // flipped phase 1 coefs (zero-padded) 0, h[8], h[5], h[2], // flipped phase 2 coefs (zero-padded) N)lennpzerosdtypereshapeTravel)huph_padlenh_fulls U/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/signal/_upfirdn.py_pad_hr.s1vv#a&&2&H Xh ( (FF7CFF7O ^^B # # %aaa2g . 4 4 6 6F McL|}t|}|S)N)lowerr)modeenums r _check_moder%Bs ::<= 1)rasarrayndimsize ValueError result_typerfloat32 _output_typeint_up_downr _h_trans_flipascontiguousarrayr _h_len_orig)selfrx_dtyperdowns r__init__z_UpFIRDn.__init__Ks JqMM 6Q;;!&A++ABB BN17GRZHH Jq$+ , ,r77YY 8a<<4:>><== =#Atx001$2DEEq66r rrrc t|j|j||j|j}t j|jt j}|||<t j||j d}||j z}t|}tt j||j |j ||j|j||||S)z@Apply the prepared filter to the specified axis of N-D signal x.)rC)rorder)rr5shaper1r2rr)int64rr/r*r%rr3)r6xaxisr#cval output_len output_shapeouts r apply_filterz_UpFIRDn.apply_filterZs !1174=!%4:77 z!'::: ' Th|4+>> import numpy as np >>> from scipy.signal import upfirdn >>> upfirdn([1, 1, 1], [1, 1, 1]) # FIR filter array([ 1., 2., 3., 2., 1.]) >>> upfirdn([1], [1, 2, 3], 3) # upsampling with zeros insertion array([ 1., 0., 0., 2., 0., 0., 3.]) >>> upfirdn([1, 1, 1], [1, 2, 3], 3) # upsampling with sample-and-hold array([ 1., 1., 1., 2., 2., 2., 3., 3., 3.]) >>> upfirdn([.5, 1, .5], [1, 1, 1], 2) # linear interpolation array([ 0.5, 1. , 1. , 1. , 1. , 1. , 0.5]) >>> upfirdn([1], np.arange(10), 1, 3) # decimation by 3 array([ 0., 3., 6., 9.]) >>> upfirdn([.5, 1, .5], np.arange(10), 2, 3) # linear interp, rate 2/3 array([ 0. , 1. , 2.5, 4. , 5.5, 7. , 8.5]) Apply a single filter to multiple signals: >>> x = np.reshape(np.arange(8), (4, 2)) >>> x array([[0, 1], [2, 3], [4, 5], [6, 7]]) Apply along the last dimension of ``x``: >>> h = [1, 1] >>> upfirdn(h, x, 2) array([[ 0., 0., 1., 1.], [ 2., 2., 3., 3.], [ 4., 4., 5., 5.], [ 6., 6., 7., 7.]]) Apply along the 0th dimension of ``x``: >>> upfirdn(h, x, 2, axis=0) array([[ 0., 1.], [ 0., 1.], [ 2., 3.], [ 2., 3.], [ 4., 5.], [ 4., 5.], [ 6., 7.], [ 6., 7.]]) )rr)r'rrE)rr?rr8r@r#rAufds rrrksAT 1 A 1agr4 ( (C   AtT4 0 00r )rrrrr) numpyr_upfirdn_applyrrr__all___upfirdn_modesrr%r'rrJr rrQsD:::::::::: m $ (        Fm1m1m1m1m1m1r