From 5f646c195fbbca32db5a62d6a80f3bc0d262012e Mon Sep 17 00:00:00 2001 From: Jean-Marc Valin Date: Tue, 15 Aug 2017 00:47:22 -0400 Subject: minor refactoring --- src/denoise.c | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/denoise.c b/src/denoise.c index 1d6f40f..7e2cb6a 100644 --- a/src/denoise.c +++ b/src/denoise.c @@ -269,14 +269,14 @@ static void frame_analysis(DenoiseState *st, kiss_fft_cpx *X, float *Ex, const f compute_band_energy(Ex, X); } -static int compute_frame_features(DenoiseState *st, kiss_fft_cpx *X, kiss_fft_cpx *P, float *Ex, float *Ep, float *features, const float *in) { +static int compute_frame_features(DenoiseState *st, kiss_fft_cpx *X, kiss_fft_cpx *P, + float *Ex, float *Ep, float *Exp, float *features, const float *in) { int i; float E = 0; float *ceps_0, *ceps_1, *ceps_2; float spec_variability = 0; float Ly[NB_BANDS]; float p[WINDOW_SIZE]; - float Exp[NB_BANDS]; float pitch_buf[PITCH_BUF_SIZE>>1]; int pitch_index; float gain; @@ -382,6 +382,7 @@ void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) { kiss_fft_cpx P[WINDOW_SIZE]; float x[FRAME_SIZE]; float Ex[NB_BANDS], Ep[NB_BANDS]; + float Exp[NB_BANDS]; float features[NB_FEATURES]; float g[NB_BANDS]; float gf[FREQ_SIZE]={1}; @@ -390,7 +391,7 @@ void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) { static const float a_hp[2] = {-1.99599, 0.99600}; static const float b_hp[2] = {-2, 1}; biquad(x, st->mem_hp_x, in, b_hp, a_hp, FRAME_SIZE); - silence = compute_frame_features(st, X, P, Ex, Ep, features, x); + silence = compute_frame_features(st, X, P, Ex, Ep, Exp, features, x); if (!silence) { compute_rnn(&st->rnn, g, &vad_prob, features); @@ -459,6 +460,7 @@ int main(int argc, char **argv) { while (1) { kiss_fft_cpx X[FREQ_SIZE], Y[FREQ_SIZE], N[FREQ_SIZE], P[WINDOW_SIZE]; float Ex[NB_BANDS], Ey[NB_BANDS], En[NB_BANDS], Ep[NB_BANDS]; + float Exp[NB_BANDS]; float Ln[NB_BANDS]; float features[NB_FEATURES]; float g[NB_BANDS]; @@ -505,7 +507,7 @@ int main(int argc, char **argv) { frame_analysis(st, X, Ex, x); frame_analysis(noise_state, N, En, n); for (i=0;ilast_gain, noisy->last_period); for (i=0;i