| 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.
|
|
|