From 3cb2c612fb030278aaf5e4d49d42cf689bdc9cc0 Mon Sep 17 00:00:00 2001
From: Eskil Queseth <eskilq@kth.se>
Date: Sat, 12 Jun 2021 15:58:44 +0200
Subject: re-add noise gate

---
 mumd/src/audio/input.rs | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'mumd/src/audio/input.rs')

diff --git a/mumd/src/audio/input.rs b/mumd/src/audio/input.rs
index 162dd2c..b0e621b 100644
--- a/mumd/src/audio/input.rs
+++ b/mumd/src/audio/input.rs
@@ -4,12 +4,13 @@ use log::*;
 use tokio::sync::watch;
 
 use crate::audio::SAMPLE_RATE;
+use crate::audio::transformers::{create_noise_gate};
 use crate::error::{AudioError, AudioStream};
 use crate::state::StatePhase;
 
 pub fn callback<T: Sample>(
     mut input_sender: futures_channel::mpsc::Sender<Vec<u8>>,
-    transformers: Vec<Box<dyn Fn(&mut [f32]) -> Option<&mut [f32]> + Send + 'static>>,
+    mut transformers: Vec<Box<dyn FnMut(&mut [f32]) -> Option<&mut [f32]> + Send + 'static>>,
     frame_size: u32,
     sample_rate: u32,
     channels: u16,
@@ -39,7 +40,7 @@ pub fn callback<T: Sample>(
         while buffer.len() + data.len() > buffer_size {
             buffer.extend(data.by_ref().take(buffer_size - buffer.len()));
             let encoded = transformers
-                .iter()
+                .iter_mut()
                 .try_fold(&mut buffer[..], |acc, e| e(acc))
                 .map(|buf| opus_encoder.encode_vec_float(&*buf, buffer_size).unwrap());
 
@@ -103,7 +104,7 @@ impl DefaultAudioInputDevice {
 
         let (volume_sender, input_volume_receiver) = watch::channel::<f32>(input_volume);
 
-        let transformers = vec![];
+        let transformers = vec![Box::new(create_noise_gate(10, 0.6)) as Box<dyn FnMut(&mut [f32]) -> Option<&mut [f32]> + Send + 'static>];
 
         let input_stream = match input_supported_sample_format {
             SampleFormat::F32 => input_device.build_input_stream(
-- 
cgit v1.2.1