0;ji>ddlZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ejZdZdZd ezZd Zd Ze e d e d e e e de de eededeeededeeededeiZdZdZGddZd!dedededejfdZd!dejdededejfd 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]vhqsoxr_vhqhqsoxr_hqmqsoxr_mqlqsoxr_lqqqsoxr_qqct|tr|} t|S#tt f$rt twxYwN) isinstancestrlower_QUALITY_ENUM_DICTKeyError TypeError ValueError_QUALITY_ERR_STR)qs G/root/voice-cloning/.venv/lib/python3.11/site-packages/soxr/__init__.py_quality_to_enumr"&s`!S GGII+!!$$ i +++)***+s 8&Ac0|tjkr tjS|tjkr tjS|tjkr tjS|tjkr tj Stt |r) npfloat32rSOXR_FLOAT32_Ifloat64SOXR_FLOAT64_Iint32 SOXR_INT32_Iint16 SOXR_INT16_Ir_DTYPE_ERR_STRformat)ntypes r!_to_soxr_datatyper00sv  && "*  && "(  $$ "(  $$--e44555cpeZdZdZ ddededefdZdd ejd ejfd Z d efd Z d efd Z ddZ 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`. r%r in_rateout_rate num_channelsc|dks|dkrtd|dks t|kr'tt|t j||_t|j}t|}tj ||||||_ t|j d|j|_ dS)NrSample rate should be over 0rprocess_)r _CH_LIMIT_CH_EXEED_ERR_STRr.r$dtype_typer0r"rCSoxr_csoxrgetattr_process)selfr4r5r6r<qualitystyper s r!__init__zResampleStream.__init__Rs a<<8q==;<< < !  y<77.55lCCDD DXe__ !$*-- W % %nWh eQOO  -D -D-DEE r1Fxreturnct|tjks|j|jkr,t t |j|jdkr@| |ddtj f|}tj |dS|jdkr| ||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) typer$ndarrayr<r=r_DTYPE_UNMATCH_ERR_STRr.ndimrAnewaxissqueezer)rBrFlastys r!resample_chunkzResampleStream.resample_chunkcs( 77bj AGtz$9$9299$*EEFF F 6Q;; a2: .55A:aa((( ( Vq[[==D)) )=>> >r1c4|jS)zx Clip counter. (for int I/O) Returns ------- int Count of clipped samples. )r? num_clipsrBs r!rWzResampleStream.num_clipss{$$&&&r1c4|jS)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?delayrXs r!rZzResampleStream.delays{  """r1Nc8|jdS)zv Reset resampler. Ready for fresh signal, same config. This can be used to save initialization time. N)r?clearrXs r!r\zResampleStream.clears r1)r%r )F)rGN) __name__ __module__ __qualname____doc__floatintrEr$rNrUrWrZr\r1r!r3r3=s,+/FFF+0F@CFFFF"?? ?2:????>'3'''' #u # # # #r1r3r rFr4r5rGcJ|dks|dkrtdt|tjkr tj|tj} |jd|jkrttd|j }nttd|j }n:#t$r-tt|j wxYwt|}|jdkr8||||ddtjf|}tj|dS|jd krS|jd}|dks t(|kr'tt*||||||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. rr8)r<csoxr_split_ch_csoxr_divide_proc_rNrIrKrL)rrMr$rNasarrayr%stridesitemsizer@rr<AttributeErrorrr-r.r"rPrQrRshaper:r;)rFr4r5rC divide_procr rTr6s r!resamplerms.!||x1}}7888 Aww"* Jq + + +8 9Q<1: % %!(,Gag,G,GHHKK!(,J,J,JKKK 888--ag667778 !!Av{{ K1QQQ ]+;Q ? ?z!!$$$$ 1wqz !  y<77.55lCCDD D{7Ha3339:::s AB,,7C#c ttd|j}n:#t$r-t t |jwxYw|jdkrE||||ddtj ft|}tj |dS||||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_rNrI) r@rr<rjrr-r.rPr$rQr"rR)rFr4r5rConeshotrTs r!_resample_oneshotrqs 8($>QW$>$>?? 888--ag667778 v{{ GGXqBJ'79I'9R9R S Sz!!$$$$ 77Ha)9')B)B C CCs  7A)r )numpyr$ numpy.typingrrrrrr r _versionr __version__libsoxr_version__libsoxr_version__r:rOr;r-rrr"r0r3rarNrmrqrcr1r!rys"""""")))))))))))))),,,,,,/h.00  eH9TTAeS*cD"iD"iD"iD"i +++ 6 6 6````````F1;1; 1;E1;U1;RZ1;1;1;1;hDDDeDuDWYWaDDDDDDr1