Line data    Source code

       1             : /*
       2             :  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
       3             :  *
       4             :  *  Use of this source code is governed by a BSD-style license
       5             :  *  that can be found in the LICENSE file in the root of the source
       6             :  *  tree. An additional intellectual property rights grant can be found
       7             :  *  in the file PATENTS.  All contributing project authors may
       8             :  *  be found in the AUTHORS file in the root of the source tree.
       9             :  */
      10             : 
      11             : #include "webrtc/modules/audio_processing/transient/moving_moments.h"
      12             : 
      13             : #include <math.h>
      14             : #include <string.h>
      15             : 
      16             : #include "webrtc/base/checks.h"
      17             : 
      18             : namespace webrtc {
      19             : 
      20           0 : MovingMoments::MovingMoments(size_t length)
      21             :     : length_(length),
      22             :       queue_(),
      23             :       sum_(0.0),
      24           0 :       sum_of_squares_(0.0) {
      25           0 :   RTC_DCHECK_GT(length, 0);
      26           0 :   for (size_t i = 0; i < length; ++i) {
      27           0 :     queue_.push(0.0);
      28             :   }
      29           0 : }
      30             : 
      31           0 : MovingMoments::~MovingMoments() {}
      32             : 
      33           0 : void MovingMoments::CalculateMoments(const float* in, size_t in_length,
      34             :                                      float* first, float* second) {
      35           0 :   RTC_DCHECK(in && in_length > 0 && first && second);
      36             : 
      37           0 :   for (size_t i = 0; i < in_length; ++i) {
      38           0 :     const float old_value = queue_.front();
      39           0 :     queue_.pop();
      40           0 :     queue_.push(in[i]);
      41             : 
      42           0 :     sum_ += in[i] - old_value;
      43           0 :     sum_of_squares_ += in[i] * in[i] - old_value * old_value;
      44           0 :     first[i] = sum_ / length_;
      45           0 :     second[i] = sum_of_squares_ / length_;
      46             :   }
      47           0 : }
      48             : 
      49             : }  // namespace webrtc