tKgY ddlZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ejZdZdZd ezZd Zd Zd Zd ZGddZddedededej2fdZddej2dededej2fdZy)N) ArrayLike)soxr_ext)QQLQMQHQVHQ)versionizJInput should be a `np.ndarray` with matching dtype for ResampleStream({}).z2Channel num({}) out of limit. Should be in [1, %d]zAData type must be one of [float32, float64, int16, int32], not {}z,Quality must be one of [QQ, LQ, MQ, HQ, VHQ]c"|tttttfvr|St |t urtt|j}|dvrtS|dvrtS|dvrtS|dvrtS|dvrtStt)N)vhqsoxr_vhq)hqsoxr_hq)mqsoxr_mq)lqsoxr_lq)qqsoxr_qq) r r rrrtypeint ValueError_QUALITY_ERR_STRlower)qs Q/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/soxr/__init__.py_quality_to_enumrsS"b"b !! Aw#~)**  A          % &&cV|tjk(rtjS|tjk(rtj S|tj k(rtjS|tjk(rtjSttj|)N) npfloat32rSOXR_FLOAT32_Ifloat64SOXR_FLOAT64_Iint32 SOXR_INT32_Iint16 SOXR_INT16_I TypeError_DTYPE_ERR_STRformat)ntypes r_to_soxr_datatyper.4sx  &&& "** &&& "(( $$$ "(( $$$--e455rceZdZdZ d dededefdZddejdejfdZ defd Z defd Z dd Z y )ResampleStreama? Streaming resampler Use `ResampleStream` for real-time processing or very long signal. Parameters ---------- in_rate : float Input sample-rate. out_rate : float Output sample-rate. num_channels : int Number of channels. dtype : type or str, optional Internal data type processed with. Should be one of float32, float64, int16, int32. quality : int or str, optional Quality setting. One of `QQ`, `LQ`, `MQ`, `HQ`, `VHQ`. in_rateout_rate num_channelsc|dks|dkr td|dks t|krttj|t j ||_t|j }t|}tj||||||_ t|jd|j |_ y)NrSample rate should be over 0rprocess_)r _CH_LIMIT_CH_EXEED_ERR_STRr,r!dtype_typer.rrCSoxr_csoxrgetattr_process)selfr1r2r3r9qualitystypers r__init__zResampleStream.__init__Vs a<8q=;< < ! y<7.55lCD DXXe_ !$**- W %nnWh eQO  x |-DE rxreturnct|tjk7s|j|jk7r(t t j|jtj|}|jdk(r>|j|ddtjf|}tj|dS|jdk(r|j||Std)a/ Resample chunk with streaming resampler Parameters ---------- x : np.ndarray Input array. Input can be mono(1D) or multi-channel(2D of [frame, channel]). dtype should match with constructor. last : bool, optional Set True at final chunk to flush last outputs. It should be `True` only once at the end of a continuous sequence. Returns ------- np.ndarray Resampled data. Output is np.ndarray with same ndim with input. rNaxisInput must be 1-D or 2-D array)rr!ndarrayr9r:r*_DTYPE_UNMATCH_ERR_STRr,ascontiguousarrayndimr>newaxissqueezer)r?rClastys rresample_chunkzResampleStream.resample_chunkgs( 7bjj AGGtzz$9299$**EF F   # 66Q; a2:: .5A::aa( ( VVq[==D) )=> >rc6|jjS)zx Clip counter. (for int I/O) Returns ------- int Count of clipped samples. )r< num_clipsr?s rrTzResampleStream.num_clipss{{$$&&rc6|jjS)z Get current delay. SoXR output has an algorithmic delay. This function returns the length of current pending output. Returns ------- float Current delay in output samples. )r<delayrUs rrWzResampleStream.delays{{  ""rNc8|jjy)zv Reset resampler. Ready for fresh signal, same config. This can be used to save initialization time. N)r<clearrUs rrYzResampleStream.clears r)r"r )F)rDN) __name__ __module__ __qualname____doc__floatrrBr!rJrRrTrWrYrrr0r0Asf,+/FF+0F@CF"? ?2::?B'3' #u #rr0rCr1r2rDc*|dks|dkr tdt|tjk7r%tj|tj } |j d|jk(rttd|j}nttd|j}t|}|j dk(r7||||ddtj"f|}tj$|dS|j d k(rF|j&d}|dks t(|krtt*j||||||Std #t$r)ttj|jwxYw) a Resample signal Parameters ---------- x : array_like Input array. Input can be mono(1D) or multi-channel(2D of [frame, channel]). If input is not `np.ndarray`, it will be converted to `np.ndarray(dtype='float32')`. Its dtype should be one of float32, float64, int16, int32. in_rate : float Input sample-rate. out_rate : float Output sample-rate. quality : int or str, optional Quality setting. One of `QQ`, `LQ`, `MQ`, `HQ`, `VHQ`. Returns ------- np.ndarray Resampled data. Output is `np.ndarray` with same ndim and dtype with input. rr5)r9csoxr_split_ch_csoxr_divide_proc_rNrFrHrI)rrr!rJasarrayr"stridesitemsizer=rr9AttributeErrorr*r+r,rrMrNrOshaper7r8)rCr1r2r@ divide_procrrQr3s rresamplerisT.!|x1}788 Aw"** JJq +8 99Q<1:: %!(oaggY,GHK!(.@ ,JKK !Avv{ 1Q ]+;Q ?zz!!$$ 1wwqz ! y<7.55lCD D7Ha339:: 8--agg6778s AE 2Fc ttd|j}tj|}|jdk(r@||||ddtjft|}tj|dS||||t|S#t$r)t t j |jwxYw)z Resample using libsoxr's `soxr_oneshot()`. Use `resample()` for general use. `soxr_oneshot()` becomes slow with long input. This function exists for test purpose. csoxr_oneshot_rNrF) r=rr9rfr*r+r,r!rLrMrNrrO)rCr1r2r@oneshotrQs r_resample_oneshotrms 8(nQWWI$>? QAvv{ GXqBJJ'79I'9R Szz!!$$ 7Ha)9')B CC 8--agg6778s B2C )r )numpyr! numpy.typingrrrrrr r _versionr __version__libsoxr_version__libsoxr_version__r7rKr8r+rrr.r0r^rJrirmr_rrrus")),/h..0  eH9TTA', 6bbJ1; 1;E1;U1;RZZ1;hDDeDuDWYWaWaDr