admin/webrtc_m130
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix division by zero errors in IntelligibilityEnhancer

Browse Source 
BUG=webrtc:5509
R=henrik.lundin@webrtc.org

Review URL: https://codereview.webrtc.org/1686333002 .

Cr-Commit-Position: refs/heads/master@{#11584}
This commit is contained in:
Alejandro Luebs 2016-02-11 10:37:12 -08:00
parent 6ecee07bab
commit 66d2481106
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer.cc
View File

@ -20,6 +20,7 @@
#include <math.h>
#include <stdlib.h>
#include <algorithm>
#include <limits>
#include <numeric>
#include "webrtc/base/checks.h"
@ -225,7 +226,8 @@ void IntelligibilityEnhancer::SolveForLambda(float power_target,
const float kConvergeThresh = 0.001f; // TODO(ekmeyerson): Find best values
const int kMaxIters = 100; // for these, based on experiments.
const float reciprocal_power_target = 1.f / power_target;
const float reciprocal_power_target =
1.f / (power_target + std::numeric_limits<float>::epsilon());
float lambda_bot = kLambdaBot;
float lambda_top = kLambdaTop;
float power_ratio = 2.0f; // Ratio of achieved power to target power.
@ -307,13 +309,13 @@ std::vector<std::vector<float>> IntelligibilityEnhancer::CreateErbBank(
float step, element;
step = 1.0f / (ll - lll);
step = ll == lll ? 0.f : 1.f / (ll - lll);
element = 0.0f;
for (size_t j = lll; j <= ll; ++j) {
filter_bank[i - 1][j] = element;
element += step;
}
step = 1.0f / (rrr - rr);
step = rr == rrr ? 0.f : 1.f / (rrr - rr);
element = 1.0f;
for (size_t j = rr; j <= rrr; ++j) {
filter_bank[i - 1][j] = element;
@ -357,7 +359,8 @@ void IntelligibilityEnhancer::SolveForGainsGivenLambda(float lambda,
if (quadratic) {
alpha0 = lambda * var_x0[n] * (1 - rho_[n]) * var_x0[n] * var_x0[n];
sols[n] =
(-beta0 - sqrtf(beta0 * beta0 - 4 * alpha0 * gamma0)) / (2 * alpha0);
(-beta0 - sqrtf(beta0 * beta0 - 4 * alpha0 * gamma0)) /
(2 * alpha0 + std::numeric_limits<float>::epsilon());
} else {
sols[n] = -gamma0 / beta0;
}