o Ki@spddlZddlZddlZddlZddZddfddZddfd d Zd d Zd dZdddZ dddZ dS)NcCs tt|jtdtjdS)N1)rounding)intdecimalDecimalquantize ROUND_HALF_UP)numberr R/home/ubuntu/.local/lib/python3.10/site-packages/python_speech_features/sigproc.py round_half_up s r cC t|fSNnumpyonesxr r r   rc Cst|}tt|}tt|}||krd}ndttd|||}t|d||}t||f}t||f}tt d||dftt d||||dfj } tj | tj d} || } t|||df} | | S)aFrame a signal into overlapping frames. :param sig: the audio signal to frame. :param frame_len: length of each frame measured in samples. :param frame_step: number of samples after the start of the previous frame that the next frame should begin. :param winfunc: the analysis window to apply to each frame. By default no window is applied. :returns: an array of frames. Size is NUMFRAMES by frame_len. ?rdtype) lenrr mathceilrzeros concatenatetilearangeTarrayint32) sig frame_len frame_stepwinfuncslen numframespadlenr padsignalindicesframeswinr r r framesig s  8r/cCr rrrr r r r*rc Cs@t|}t|}t|d}t|d|ksJdttd||dfttd||||dfj}tj|tjd}|d||}|dkrN|}t|f}t|f} ||} t d|D]2} | || ddf| d| || ddf<||| ddf|| ddf||| ddf<qc|| }|d|S)a Does overlap-add procedure to undo the action of framesig. :param frames: the array of frames. :param siglen: the length of the desired signal, use 0 if unknown. Output will be truncated to siglen samples. :param frame_len: length of each frame measured in samples. :param frame_step: number of samples after the start of the previous frame that the next frame should begin. :param winfunc: the analysis window to apply to each frame. By default no window is applied. :returns: a 1-D signal. rrz@"frames" matrix is wrong size, 2nd dim is not equal to frame_lenrNgV瞯<) r rshaperr r!r"r#rrange) r-siglenr%r&r'r)r,r* rec_signalwindow_correctionr.ir r r deframesig*s  8   ,6 r6cCsBt|d|krtdt|d|tj||}t|S)aCompute the magnitude spectrum of each frame in frames. If frames is an NxD matrix, output will be Nx(NFFT/2+1). :param frames: the array of frames. Each row is a frame. :param NFFT: the FFT length to use. If NFFT > frame_len, the frames are zero-padded. :returns: If frames is an NxD matrix, output will be Nx(NFFT/2+1). Each row will be the magnitude spectrum of the corresponding frame. rzaframe length (%d) is greater than FFT size (%d), frame will be truncated. Increase NFFT to avoid.)rr0loggingwarnfftrfftabsolute)r-NFFT complex_specr r r magspecJs r>cCsd|tt||S)aCompute the power spectrum of each frame in frames. If frames is an NxD matrix, output will be Nx(NFFT/2+1). :param frames: the array of frames. Each row is a frame. :param NFFT: the FFT length to use. If NFFT > frame_len, the frames are zero-padded. :returns: If frames is an NxD matrix, output will be Nx(NFFT/2+1). Each row will be the power spectrum of the corresponding frame. r)rsquarer>)r-r<r r r powspecVsr@rcCs:t||}d||dk<dt|}|r|t|S|S)aCompute the log power spectrum of each frame in frames. If frames is an NxD matrix, output will be Nx(NFFT/2+1). :param frames: the array of frames. Each row is a frame. :param NFFT: the FFT length to use. If NFFT > frame_len, the frames are zero-padded. :param norm: If norm=1, the log power spectrum is normalised so that the max value (across all frames) is 0. :returns: If frames is an NxD matrix, output will be Nx(NFFT/2+1). Each row will be the log power spectrum of the corresponding frame. gKH9 )r@rlog10max)r-r<normpslpsr r r logpowspec_s  rGffffff?cCs(t|d|dd||ddS)zperform preemphasis on the input signal. :param signal: The signal to filter. :param coeff: The preemphasis coefficient. 0 is no filter, default is 0.95. :returns: the filtered signal. rrN)rappend)signalcoeffr r r preemphasisos(rM)r)rH) rrrr7r r/r6r>r@rGrMr r r r s