diff options
Diffstat (limited to 'webrtc/modules/audio_processing/transient/moving_moments.cc')
-rw-r--r-- | webrtc/modules/audio_processing/transient/moving_moments.cc | 27 |
1 files changed, 14 insertions, 13 deletions
diff --git a/webrtc/modules/audio_processing/transient/moving_moments.cc b/webrtc/modules/audio_processing/transient/moving_moments.cc index aa47522..83810bf 100644 --- a/webrtc/modules/audio_processing/transient/moving_moments.cc +++ b/webrtc/modules/audio_processing/transient/moving_moments.cc @@ -8,21 +8,17 @@ * be found in the AUTHORS file in the root of the source tree. */ -#include "webrtc/modules/audio_processing/transient/moving_moments.h" +#include "modules/audio_processing/transient/moving_moments.h" -#include <math.h> -#include <string.h> +#include <algorithm> -#include "webrtc/base/scoped_ptr.h" +#include "rtc_base/checks.h" namespace webrtc { MovingMoments::MovingMoments(size_t length) - : length_(length), - queue_(), - sum_(0.0), - sum_of_squares_(0.0) { - assert(length > 0); + : length_(length), queue_(), sum_(0.0), sum_of_squares_(0.0) { + RTC_DCHECK_GT(length, 0); for (size_t i = 0; i < length; ++i) { queue_.push(0.0); } @@ -30,9 +26,14 @@ MovingMoments::MovingMoments(size_t length) MovingMoments::~MovingMoments() {} -void MovingMoments::CalculateMoments(const float* in, size_t in_length, - float* first, float* second) { - assert(in && in_length > 0 && first && second); +void MovingMoments::CalculateMoments(const float* in, + size_t in_length, + float* first, + float* second) { + RTC_DCHECK(in); + RTC_DCHECK_GT(in_length, 0); + RTC_DCHECK(first); + RTC_DCHECK(second); for (size_t i = 0; i < in_length; ++i) { const float old_value = queue_.front(); @@ -42,7 +43,7 @@ void MovingMoments::CalculateMoments(const float* in, size_t in_length, sum_ += in[i] - old_value; sum_of_squares_ += in[i] * in[i] - old_value * old_value; first[i] = sum_ / length_; - second[i] = sum_of_squares_ / length_; + second[i] = std::max(0.f, sum_of_squares_ / length_); } } |