o wiY @sddlZddlmZddlmZddlmZmZmZm Z m Z ddl m Z eZdZdZd eZd Zd Zd d ZddZGdddZddedededejfddZddejdededejfddZdS)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]cCsv|tttttfvr |St|turtt| }|dvrtS|dvr%tS|dvr+tS|dvr1tS|dvr7tStt)N)vhqsoxr_vhq)hqsoxr_hq)mqsoxr_mq)lqsoxr_lq)qqsoxr_qq) r rrrrtypeint ValueError_QUALITY_ERR_STRlower)qrJ/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/soxr/__init__.py_quality_to_enums  rcCsN|tjkrtjS|tjkrtjS|tjkrtjS|tjkr tj St t |)N) npfloat32rSOXR_FLOAT32_Ifloat64SOXR_FLOAT64_Iint32 SOXR_INT32_Iint16 SOXR_INT16_I TypeError_DTYPE_ERR_STRformat)ntyperrr_to_soxr_datatype4s    r+c@sheZdZdZ ddededefddZdd ejd ejfd d Z d efddZ d efddZ dddZ dS)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`. rrin_rateout_rate num_channelscCs|dks|dkr td|dkst|krtt|t||_t|j}t|}t ||||||_ t |j d|j|_ dS)NrSample rate should be over 0rprocess_)r _CH_LIMIT_CH_EXEED_ERR_STRr)rdtype_typer+rrCSoxr_csoxrgetattr_process)selfr-r.r/r4qualitystyperrrr__init__Vs  zResampleStream.__init__FxreturncCst|tjks |j|jkrtt|jt|}|j dkr3| |ddtj f|}tj |ddS|j dkr>| ||St d)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)rrndarrayr4r5r'_DTYPE_UNMATCH_ERR_STRr)ascontiguousarrayndimr9newaxissqueezer)r:r>lastyrrrresample_chunkgs    zResampleStream.resample_chunkcC |jS)zx Clip counter. (for int I/O) Returns ------- int Count of clipped samples. )r7 num_clipsr:rrrrNs zResampleStream.num_clipscCrM)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. )r7delayrOrrrrPs zResampleStream.delayNcCs|jdS)zv Reset resampler. Ready for fresh signal, same config. This can be used to save initialization time. N)r7clearrOrrrrQszResampleStream.clear)rr)F)r?N) __name__ __module__ __qualname____doc__floatrr=rrDrLrNrPrQrrrrr,As !  r,rr>r-r.r?cCs|dks|dkr tdt|tjkrtj|tjd}z|jd|jkr.tt d|j }n tt d|j }Wnt yGt t |j wt|}|jdkrf||||ddtjf|}tj|ddS|jd kr|jd}|dksxt|krtt||||||Std ) 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. rr0)r4csoxr_split_ch_csoxr_divide_proc_rNr@rBrC)rrrrDasarrayrstridesitemsizer8rr4AttributeErrorr'r(r)rrGrHrIshaper2r3)r>r-r.r; divide_procrrKr/rrrresamples,    r_cCsz ttd|j}Wntytt|jwt|}|j dkr<||||ddtj ft |}tj |ddS||||t |S)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_rNr@) r8rr4r\r'r(r)rrFrGrHrrI)r>r-r.r;oneshotrKrrr_resample_oneshots    rb)r)numpyr numpy.typingrrrrrrr _versionr __version__libsoxr_version__libsoxr_version__r2rEr3r(rrr+r,rVrDr_rbrrrrs     e$4