tie slider to rnnoise probability threshold

uowuo · Aug 2, 2023 · d208c64 · d208c64
1 parent 60a8c86
commit d208c64
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/src/audio/manager.cpp b/src/audio/manager.cpp
@@ -579,6 +579,10 @@ double AudioManager::GetRNNProbThreshold() const {
     return m_prob_threshold;
 }
 
+void AudioManager::SetRNNProbThreshold(double value) {
+    m_prob_threshold = value;
+}
+
 AudioManager::type_signal_opus_packet AudioManager::signal_opus_packet() {
     return m_signal_opus_packet;
 }

diff --git a/src/audio/manager.hpp b/src/audio/manager.hpp
@@ -81,6 +81,7 @@ class AudioManager {
 
     float GetCurrentVADProbability() const;
     double GetRNNProbThreshold() const;
+    void SetRNNProbThreshold(double value);
 
 private:
     void OnCapturedPCM(const int16_t *pcm, ma_uint32 frames);

diff --git a/src/windows/voicewindow.cpp b/src/windows/voicewindow.cpp
@@ -129,6 +129,8 @@ VoiceWindow::VoiceWindow(Snowflake channel_id)
                 break;
 #ifdef WITH_RNNOISE
             case AudioManager::VADMethod::RNNoise:
+                audio.SetRNNProbThreshold(val);
+                m_vad_value.SetTick(val);
                 break;
 #endif
         };