From f0e9c5e03d97a3307469c1b5666c2f6d84e5bb2e Mon Sep 17 00:00:00 2001
From: Jean-Marc Valin <jmvalin@jmvalin.ca>
Date: Fri, 28 Jul 2017 17:17:11 -0400
Subject: randomize gain

---
 src/denoise.c | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/denoise.c b/src/denoise.c
index 7ea5c36..9be6532 100644
--- a/src/denoise.c
+++ b/src/denoise.c
@@ -363,6 +363,8 @@ int main(int argc, char **argv) {
   float n[FRAME_SIZE];
   float xn[FRAME_SIZE];
   int vad_cnt=0;
+  int gain_change_count=0;
+  float speech_gain = 1, noise_gain = 1;
   FILE *f1, *f2, *fout;
   DenoiseState *st;
   DenoiseState *noise_state;
@@ -391,12 +393,19 @@ int main(int argc, char **argv) {
     short tmp[FRAME_SIZE];
     float vad=0;
     float E=0;
+    if (++gain_change_count > 101*300) {
+      speech_gain = pow(10., (-30+(rand()%40))/20.);
+      noise_gain = pow(10., (-30+(rand()%40))/20.);
+      if (rand()%10==0) noise_gain = 0;
+      noise_gain *= speech_gain;
+      gain_change_count = 0;
+    }
     fread(tmp, sizeof(short), FRAME_SIZE, f1);
     if (feof(f1)) break;
-    for (i=0;i<FRAME_SIZE;i++) x[i] = tmp[i];
+    for (i=0;i<FRAME_SIZE;i++) x[i] = speech_gain*tmp[i];
     fread(tmp, sizeof(short), FRAME_SIZE, f2);
     if (feof(f2)) break;
-    for (i=0;i<FRAME_SIZE;i++) n[i] = tmp[i];
+    for (i=0;i<FRAME_SIZE;i++) n[i] = noise_gain*tmp[i];
     for (i=0;i<FRAME_SIZE;i++) xn[i] = x[i] + n[i];
     for (i=0;i<FRAME_SIZE;i++) E += x[i]*(float)x[i];
     if (E > 1e9f) {
-- 
cgit v1.2.3