From c0a49cea32898c4805a7a94c96905f494a2aa03f Mon Sep 17 00:00:00 2001
From: Jean-Marc Valin <jmvalin@jmvalin.ca>
Date: Wed, 13 Sep 2017 19:56:46 -0400
Subject: [PATCH] Output VAD prob

---
 include/rnnoise.h | 2 +-
 src/denoise.c     | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/include/rnnoise.h b/include/rnnoise.h
index f85a522..562b34c 100644
--- a/include/rnnoise.h
+++ b/include/rnnoise.h
@@ -49,4 +49,4 @@ RNNOISE_EXPORT DenoiseState *rnnoise_create();
 
 RNNOISE_EXPORT void rnnoise_destroy(DenoiseState *st);
 
-RNNOISE_EXPORT void rnnoise_process_frame(DenoiseState *st, float *out, const float *in);
+RNNOISE_EXPORT float rnnoise_process_frame(DenoiseState *st, float *out, const float *in);
diff --git a/src/denoise.c b/src/denoise.c
index 075edc8..cb545b6 100644
--- a/src/denoise.c
+++ b/src/denoise.c
@@ -467,7 +467,7 @@ void pitch_filter(kiss_fft_cpx *X, const kiss_fft_cpx *P, const float *Ex, const
   }
 }
 
-void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) {
+float rnnoise_process_frame(DenoiseState *st, float *out, const float *in) {
   int i;
   kiss_fft_cpx X[FREQ_SIZE];
   kiss_fft_cpx P[WINDOW_SIZE];
@@ -477,7 +477,7 @@ void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) {
   float features[NB_FEATURES];
   float g[NB_BANDS];
   float gf[FREQ_SIZE]={1};
-  float vad_prob;
+  float vad_prob = 0;
   int silence;
   static const float a_hp[2] = {-1.99599, 0.99600};
   static const float b_hp[2] = {-2, 1};
@@ -497,6 +497,7 @@ void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) {
   }
 
   frame_synthesis(st, out, X);
+  return vad_prob;
 }
 
 #if TRAINING
-- 
2.7.4