summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/audio/manager.cpp4
-rw-r--r--src/audio/manager.hpp1
-rw-r--r--src/windows/voicewindow.cpp2
3 files changed, 7 insertions, 0 deletions
diff --git a/src/audio/manager.cpp b/src/audio/manager.cpp
index 4f3555b..e47b220 100644
--- 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
index 6b67719..d37a281 100644
--- a/src/audio/manager.hpp
+++ b/src/audio/manager.hpp
@@ -81,6 +81,7 @@ public:
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
index 4ab9864..d9e3336 100644
--- 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
};