o i@sddlZddlmZddlmZddlmZmZmZm Z m Z ddl m Z eZdZdZd eZd Zd Ze e d e d e e e de de eededeeededeeededeiZddZddZGdddZd%dededed ejfd!d"Zd%dejdeded ejfd#d$ZdS)&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]ZvhqZsoxr_vhqZhqZsoxr_hqZmqZsoxr_mqZlqZsoxr_lqZqqZsoxr_qqc Cs8t|tr |}zt|WSttfyttwN) isinstancestrlower_QUALITY_ENUM_DICTKeyError TypeError ValueError_QUALITY_ERR_STR)qr\/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/soxr/__init__.py_quality_to_enum&s  rcCsN|tjkrtjS|tjkrtjS|tjkrtjS|tjkr tj St t |r ) npfloat32rZSOXR_FLOAT32_IZfloat64ZSOXR_FLOAT64_IZint32Z SOXR_INT32_IZint16Z SOXR_INT16_Ir_DTYPE_ERR_STRformat)Zntyperrr_to_soxr_datatype0s    rc@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 0rZprocess_)r _CH_LIMIT_CH_EXEED_ERR_STRrrdtype_typerrrZCSoxr_csoxrgetattr_process)selfrrr r$qualitystyperrrr__init__Rs  zResampleStream.__init__FxreturncCszt|tjks |j|jkrtt|j|jdkr.| |ddtj f|}tj |ddS|jdkr9| ||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. rNZaxisInput must be 1-D or 2-D array) typerndarrayr$r%r_DTYPE_UNMATCH_ERR_STRrndimr(newaxissqueezer)r)r-lastyrrrresample_chunkcs   zResampleStream.resample_chunkcC |jS)zx Clip counter. (for int I/O) Returns ------- int Count of clipped samples. )r& num_clipsr)rrrr<s zResampleStream.num_clipscCr;)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&delayr=rrrr>s zResampleStream.delayNcCs|jdS)zv Reset resampler. Ready for fresh signal, same config. This can be used to save initialization time. N)r&clearr=rrrr?szResampleStream.clear)rr)F)r.N) __name__ __module__ __qualname____doc__floatintr,rr3r:r<r>r?rrrrr=s   rrr-rrr.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. rr!)r$Zcsoxr_split_ch_Zcsoxr_divide_proc_rNr/r0r1)rr2rr3Zasarrayrstridesitemsizer'rr$AttributeErrorrrrrr5r6r7shaper"r#)r-rrr*Z divide_procrr9r rrrresamples,    rJcCsz ttd|j}Wntytt|jw|jdkr7||||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. Zcsoxr_oneshot_rNr/) r'rr$rHrrrr5rr6rr7)r-rrr*Zoneshotr9rrr_resample_oneshots   rK)r)numpyrZ numpy.typingrrrrrrr _versionr __version__Zlibsoxr_versionZ__libsoxr_version__r"r4r#rrrrrrrDr3rJrKrrrrs,           c$4