ó ˽÷Xc@scddlZddlmZmZddgZd„Zdefd„ƒYZdddd „Z dS( iÿÿÿÿNi(t _output_lent_applytupfirdnRcCsxt|ƒt|ƒ |}tj||jƒ}||t|ƒ*|jd|ƒjdd…ddd…fjƒ}|S(s´Store 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) iÿÿÿÿN(tlentnptzerostdtypetreshapetTtravel(thtupth_padlenth_full((s4/tmp/pip-build-7oUkmx/scipy/scipy/signal/_upfirdn.pyt_pad_h)s 4t_UpFIRDncBseZd„Zdd„ZRS(cCsètj|ƒ}|jdks-|jdkr<tdƒ‚ntj|j|tjƒ|_tj||jƒ}t |ƒ|_ t |ƒ|_ |j dks«|j dkrºtdƒ‚nt ||j ƒ|_ tj|j ƒ|_ dS(sHelper for resamplingiis!h must be 1D with non-zero lengthsBoth up and down must be >= 1N(Rtasarraytndimtsizet ValueErrort result_typeRtfloat32t _output_typetintt_upt_downRt _h_trans_fliptascontiguousarray(tselfR tx_dtypeR tdown((s4/tmp/pip-build-7oUkmx/scipy/scipy/signal/_upfirdn.pyt__init__>siÿÿÿÿcCs§tt|jƒ|j||j|jƒ}tj|jƒ}|||>> 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., 0., 0.]) >>> 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, 0. ]) >>> 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, 0. ]) 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.]]) (RRRRR((R R#R RR$tufd((s4/tmp/pip-build-7oUkmx/scipy/scipy/signal/_upfirdn.pyR[sY( tnumpyRt_upfirdn_applyRRt__all__RtobjectRR(((s4/tmp/pip-build-7oUkmx/scipy/scipy/signal/_upfirdn.pyt"s