qiU#XdZddlZddlZddlZddlmZddlmZm Z ddgZ d dZ d d Z y) zCore resampling interfaceN) get_filter) resample_f_s resample_f_presample resample_nuc |dkrtdj||dkrtdj|||k(r|jSt||z }t |j }t ||t|zt|z ||<||dkr)tdj|j |||tj|jtjrtj} n |j} tj|| |} t|fi|\} } } |dkr|| z} tj| | d}td |}d |z }tj |||z}|r4 t#|j%d||| || || j%d|| St/|j%d||| || || j%d|| S#t&j($rV}t+j,|d d t/|j%d||| || || j%d|Yd }~| Sd }~wwxYw)a Resample a signal x from sr_orig to sr_new along a given axis. Parameters ---------- x : np.ndarray, dtype=np.float* The input signal(s) to resample. sr_orig : int > 0 The sampling rate of x sr_new : int > 0 The target sampling rate of the output signal(s) If `sr_new == sr_orig`, then a copy of `x` is returned with no interpolation performed. axis : int The target axis along which to resample `x` filter : optional, str or callable The resampling filter to use. By default, uses the `kaiser_best` (pre-computed filter). parallel : optional, bool Enable/disable parallel computation exploiting multi-threading. Default: False. **kwargs additional keyword arguments provided to the specified filter Returns ------- y : np.ndarray `x` resampled to `sr_new` Raises ------ ValueError if `sr_orig` or `sr_new` is not positive TypeError if the input signal `x` has an unsupported data type. Examples -------- >>> import resampy >>> np.set_printoptions(precision=3, suppress=True) >>> # Generate a sine wave at 440 Hz for 5 seconds >>> sr_orig = 44100.0 >>> x = np.sin(2 * np.pi * 440.0 / sr_orig * np.arange(5 * sr_orig)) >>> x array([ 0. , 0.063, ..., -0.125, -0.063]) >>> # Resample to 22050 with default parameters >>> resampy.resample(x, sr_orig, 22050) array([ 0.011, 0.122, 0.25 , ..., -0.366, -0.25 , -0.122]) >>> # Resample using the fast (low-quality) filter >>> resampy.resample(x, sr_orig, 22050, filter='kaiser_fast') array([ 0.012, 0.121, 0.251, ..., -0.365, -0.251, -0.121]) >>> # Resample using a high-quality filter >>> resampy.resample(x, sr_orig, 22050, filter='kaiser_best') array([ 0.011, 0.122, 0.25 , ..., -0.366, -0.25 , -0.122]) >>> # Resample using a Hann-windowed sinc filter >>> import scipy.signal >>> resampy.resample(x, sr_orig, 22050, filter='sinc_window', ... window=scipy.signal.hann) array([ 0.011, 0.123, 0.25 , ..., -0.366, -0.25 , -0.123]) >>> # Generate stereo data >>> x_right = np.sin(2 * np.pi * 880.0 / sr_orig * np.arange(len(x))) >>> x_stereo = np.stack([x, x_right]) >>> x_stereo.shape (2, 220500) >>> # Resample along the time axis (1) >>> y_stereo = resampy.resample(x_stereo, sr_orig, 22050, axis=1) >>> y_stereo.shape (2, 110250) rInvalid sample rate: sr_orig={}zInvalid sample rate: sr_new={}rz;Input signal length={} is too small to resample from {}->{}dtypeshapeappend?$ Fallback to the sequential version. stacklevelN) ValueErrorformatcopyfloatlistr intnp issubdtyper integerfloat32 zeros_likerdiffminarangerswapaxesnumba TypingErrorwarningswarnr)xsr_origsr_newaxisfilterparallelkwargs sample_ratior r y interp_win precision_ interp_deltascaletime_incrementt_outexcs >/opt/pipecat/venv/lib/python3.12/site-packages/resampy/core.pyrrsad!|:AA'JKK {9@@HII&vvx=7*L MEeDkE&M1E'NBCE$K T{Q ##)6!''$-&#I   }}QWWbjj)  auE2A)&;F;J 1a!J. 77:jn=L \ "E<'N IIeDk "^ 3E   2t$ 2t$ D H  JJr4      JJr4   H3   MM%<=   2t$ 2t$  ( H3 s1G==I&A I!!I&c |dkrtdj|tj|}|jdk7r$tdj|j tj |dks+tj||j |dz |z kDrUtdjtj |tj||j |dz |z t|j }t|||<tj|jtjrtj}n |j}tj|||} t|fi|\} } } tj | | d} |d k7r||z}|r4 t#|j%d||| | | d | j%d|| St/|j%d||| | | d | j%d|| S#t&j($rV}t+j,|d d t/|j%d||| | | d | j%d|Yd }~| Sd }~wwxYw)a'Interpolate a signal x at specified positions (t_out) along a given axis. Parameters ---------- x : np.ndarray, dtype=np.float* The input signal(s) to resample. sr_orig : float Sampling rate of the input signal (x). t_out : np.ndarray, dtype=np.float* Position of the output samples. axis : int The target axis along which to resample `x` filter : optional, str or callable The resampling filter to use. By default, uses the `kaiser_best` (pre-computed filter). parallel : optional, bool Enable/disable parallel computation exploiting multi-threading. Default: True. **kwargs additional keyword arguments provided to the specified filter Returns ------- y : np.ndarray `x` resampled to `t_out` Raises ------ TypeError if the input signal `x` has an unsupported data type. Notes ----- Differently form the `resample` function the filter `rolloff` is not automatically adapted in case of subsampling. For this reason results obtained with the `resample_nu` could be slightly different form the ones obtained with `resample` if the filter parameters are not carefully set by the user. Examples -------- >>> import resampy >>> np.set_printoptions(precision=3, suppress=True) >>> # Generate a sine wave at 100 Hz for 5 seconds >>> sr_orig = 100.0 >>> f0 = 1 >>> t = np.arange(5 * sr_orig) / sr_orig >>> x = np.sin(2 * np.pi * f0 * t) >>> x array([ 0. , 0.063, 0.125, ..., -0.187, -0.125, -0.063]) >>> # Resample to non-uniform sampling >>> t_new = np.log2(1 + t)[::5] - t[0] >>> resampy.resample_nu(x, sr_orig, t_new) array([ 0.001, 0.427, 0.76 , ..., -0.3 , -0.372, -0.442]) rr rz+Invalid t_out shape ({}), 1D array expectedz6Output domain [{}, {}] exceeds the data domain [0, {}]r rrrrrrN)rrrasarrayndimr r"maxrlenrr rrr rr!rr$r%r&r'r(r)r)r*r8r,r-r.r/r r r1r2r3r4r5r9s r:rrsVD!|:AA'JKK JJu E zzQ 9 @ @ M   vve}qBFF5MQWWT]Q->',II D K Ku rvve}qwwt}q/@G.K    MEe*E$K }}QWWbjj)  auE2A)&;F;J 177:jn=L#~   2t$ 2t$ D H  JJr4      JJr4   H3   MM%<=   2t$ 2t$  ( H3 s=1H##J 6A JJ )r kaiser_bestF) __doc__r'numpyrr%filtersrinterpnrr__all__rrr:rHs>  / } %AFb L@EG rG