o X۷i@sddlmZddlZeddddZd"dd Zejd d d d dddZd"ddZejddddddZddZ ejddddddZ d#d d!Z dS)$) annotationsNz T spectrumzT outputz* output = log( abs( spectrum ) ); _real_cepstrum_kernelcCs8t|}tjj|||d}t|}tjj|||djS)aa Calculates the real cepstrum of an input sequence x where the cepstrum is defined as the inverse Fourier transform of the log magnitude DFT (spectrum) of a signal. It's primarily used for source/speaker separation in speech signal processing Parameters ---------- x : ndarray Input sequence, if x is a matrix, return cepstrum in direction of axis n : int Size of Fourier Transform; If none, will use length of input array axis: int Direction for cepstrum calculation Returns ------- ceps : ndarray Complex cepstrum result naxis)cupyasarrayfftrifftreal)xrrspectrumrW/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/signal/_acoustics/_cepstrum.py real_cepstrum$s rzC spectrum, raw T unwrappedzC output, T ndelayz ndelay = round( unwrapped[center] / M_PI ); const T temp { unwrapped[i] - ( M_PI * ndelay * i / center ) }; output = log( abs( spectrum ) ) + C( 0, temp ); _complex_cepstrum_kernelTzKconst int center { static_cast( 0.5 * ( _ind.size() + 1 ) ) };) return_tuple loop_prepcCsVt|}tjj|||d}tt|}t||\}}tjj|||dj}||fS)a Calculates the complex cepstrum of a real valued input sequence x where the cepstrum is defined as the inverse Fourier transform of the log magnitude DFT (spectrum) of a signal. It's primarily used for source/speaker separation in speech signal processing. The input is altered to have zero-phase at pi radians (180 degrees) Parameters ---------- x : ndarray Input sequence, if x is a matrix, return cepstrum in direction of axis n : int Size of Fourier Transform; If none, will use length of input array axis: int Direction for cepstrum calculation Returns ------- ceps : ndarray Complex cepstrum result r)rr r unwrapanglerr r )r rrr unwrapped log_spectrumndelaycepsrrrcomplex_cepstrumOs rz(C log_spectrum, int32 ndelay, float64 piz C spectrumz const double wrapped { log_spectrum.imag() + M_PI * ndelay * i / center }; spectrum = exp( C( log_spectrum.real(), wrapped ) ) _inverse_complex_cepstrum_kernelz2const double center { 0.5 * ( _ind.size() + 1 ) };)rcCs6t|}tj|}t||tj}tj|j}|S)a6Compute the inverse complex cepstrum of a real sequence. ceps : ndarray Real sequence to compute inverse complex cepstrum of. ndelay: int The amount of samples of circular delay added to `x`. Returns ------- x : ndarray The inverse complex cepstrum of the real sequence `ceps`. The inverse complex cepstrum is given by .. math:: x[n] = F^{-1}\left{\exp(F(c[n]))\right} where :math:`c_[n]` is the input signal and :math:`F` and :math:`F_{-1} are respectively the forward and backward Fourier transform. )rr r rpir r )rrrricepsrrrinverse_complex_cepstrumzs  rzT cepszT windowz if ( !i ) { window = ceps; } else if ( i < bend ) { window = ceps * 2.0; } else if ( i == bend ) { window = ceps * ( 1 - odd ); } else { window = 0; } _minimum_phase_kernelzconst bool odd { (_ind.size() & 1) != 0 }; const int bend { static_cast( 0.5 * ( _ind.size() + odd ) ) };cCsD|durt|}t||d}t|}tjttj|j}|S)aCompute the minimum phase reconstruction of a real sequence. x : ndarray Real sequence to compute the minimum phase reconstruction of. n : {None, int}, optional Length of the Fourier transform. Compute the minimum phase reconstruction of a real sequence using the real cepstrum. Returns ------- m : ndarray The minimum phase reconstruction of the real sequence `x`. N)r)lenrr rr r expr )r rrwindowmrrr minimum_phases   r%)Nr)N) __future__rrElementwiseKernelrrrrrrr r%rrrrsD