Index: webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.h |
diff --git a/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.h b/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.h |
index 111b765f973485bb888785cdd6946693b4c445ba..63ae80e2c44060853bd6578e8895036609aaa581 100644 |
--- a/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.h |
+++ b/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.h |
@@ -36,7 +36,7 @@ class IntelligibilityEnhancer : public LappedTransform::Callback { |
size_t num_noise_bins); |
// Sets the capture noise magnitude spectrum estimate. |
- void SetCaptureNoiseEstimate(std::vector<float> noise); |
+ void SetCaptureNoiseEstimate(std::vector<float> noise, int gain_db); |
// Reads chunk of speech in time domain and updates with modified signal. |
void ProcessRenderAudio(float* const* audio, |
@@ -56,6 +56,8 @@ class IntelligibilityEnhancer : public LappedTransform::Callback { |
private: |
FRIEND_TEST_ALL_PREFIXES(IntelligibilityEnhancerTest, TestErbCreation); |
FRIEND_TEST_ALL_PREFIXES(IntelligibilityEnhancerTest, TestSolveForGains); |
+ FRIEND_TEST_ALL_PREFIXES(IntelligibilityEnhancerTest, |
+ TestNoiseGainHasExpectedResult); |
// Updates the SNR estimation and enables or disables this component using a |
// hysteresis. |