Index: webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.cc |
diff --git a/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.cc b/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.cc |
index 3029e21619a981917f377afb78075bfd66def952..3e5063bbf82287c10a6c4b8848fa5171d992b474 100644 |
--- a/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.cc |
+++ b/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.cc |
@@ -42,6 +42,9 @@ const float IntelligibilityEnhancer::kClipFreq = 200.0f; |
const float IntelligibilityEnhancer::kConfigRho = 0.02f; |
const float IntelligibilityEnhancer::kKbdAlpha = 1.5f; |
hlundin-webrtc
2015/06/30 14:00:52
Delete empty line.
ekm
2015/07/01 23:48:25
Done.
|
+const float IntelligibilityEnhancer::kLambdaBot = -1.0; |
hlundin-webrtc
2015/06/30 14:00:52
Are all of these constants only used locally in th
ekm
2015/07/01 23:48:25
Done.
|
+const float IntelligibilityEnhancer::kLambdaTop = -10e-18f; |
+ |
// To disable gain update smoothing, set gain limit to be VERY high. |
// TODO(ekmeyerson): Add option to disable gain smoothing altogether |
// to avoid the extra computation. |
@@ -239,18 +242,25 @@ void IntelligibilityEnhancer::AnalyzeClearBlock(float power_target) { |
FilterVariance(clear_variance_.variance(), filtered_clear_var_.get()); |
FilterVariance(noise_variance_.variance(), filtered_noise_var_.get()); |
- // Bisection search for optimal |lambda| |
- |
- float lambda_bot = -1.0f, lambda_top = -10e-18f, lambda; |
- float power_bot, power_top, power; |
- SolveForGainsGivenLambda(lambda_top, start_freq_, gains_eq_.get()); |
+ float power_bot, power_top; |
hlundin-webrtc
2015/06/30 14:00:51
Premature declaration of variables. Declare them w
ekm
2015/07/01 23:48:25
Done.
|
+ SolveForGainsGivenLambda(kLambdaTop, start_freq_, gains_eq_.get()); |
power_top = |
DotProduct(gains_eq_.get(), filtered_clear_var_.get(), bank_size_); |
- SolveForGainsGivenLambda(lambda_bot, start_freq_, gains_eq_.get()); |
+ SolveForGainsGivenLambda(kLambdaBot, start_freq_, gains_eq_.get()); |
power_bot = |
DotProduct(gains_eq_.get(), filtered_clear_var_.get(), bank_size_); |
- DCHECK(power_target >= power_bot && power_target <= power_top); |
+ if(power_target >= power_bot && power_target <= power_top) { |
+ SolveForLambda(power_target, power_bot, power_top); |
+ UpdateErbGains(); |
+ } // Else experiencing variance underflow, so do nothing. |
+} |
+void IntelligibilityEnhancer::SolveForLambda(float power_target, |
+ float power_bot, |
+ float power_top) { |
+ float lambda_bot = kLambdaBot; |
+ float lambda_top = kLambdaTop; |
+ float lambda, power; |
hlundin-webrtc
2015/06/30 14:00:52
Skip declaration of lambda and power here and decl
ekm
2015/07/01 23:48:25
Done.
|
float power_ratio = 2.0f; // Ratio of achieved power to target power. |
const float kConvergeThresh = 0.001f; // TODO(ekmeyerson): Find best values |
const int kMaxIters = 100; // for these, based on experiments. |
@@ -267,7 +277,9 @@ void IntelligibilityEnhancer::AnalyzeClearBlock(float power_target) { |
power_ratio = fabs(power / power_target); |
++iters; |
} |
+} |
+void IntelligibilityEnhancer::UpdateErbGains() { |
// (ERB gain) = filterbank' * (freq gain) |
float* gains = gain_applier_.target(); |
for (int i = 0; i < freqs_; ++i) { |