pbos@webrtc.org | 788acd1 | 2014-12-15 09:41:24 +0000 | [diff] [blame] | 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 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 11 | #ifndef MODULES_AUDIO_PROCESSING_TRANSIENT_MOVING_MOMENTS_H_ |
| 12 | #define MODULES_AUDIO_PROCESSING_TRANSIENT_MOVING_MOMENTS_H_ |
pbos@webrtc.org | 788acd1 | 2014-12-15 09:41:24 +0000 | [diff] [blame] | 13 | |
kwiberg | 85d8bb0 | 2016-02-16 20:39:36 -0800 | [diff] [blame] | 14 | #include <stddef.h> |
pbos@webrtc.org | 788acd1 | 2014-12-15 09:41:24 +0000 | [diff] [blame] | 15 | |
kwiberg | 85d8bb0 | 2016-02-16 20:39:36 -0800 | [diff] [blame] | 16 | #include <queue> |
pbos@webrtc.org | 788acd1 | 2014-12-15 09:41:24 +0000 | [diff] [blame] | 17 | |
| 18 | namespace webrtc { |
| 19 | |
| 20 | // Calculates the first and second moments for each value of a buffer taking |
| 21 | // into account a given number of previous values. |
| 22 | // It preserves its state, so it can be multiple-called. |
| 23 | // TODO(chadan): Implement a function that takes a buffer of first moments and a |
| 24 | // buffer of second moments; and calculates the variances. When needed. |
| 25 | // TODO(chadan): Add functionality to update with a buffer but only output are |
| 26 | // the last values of the moments. When needed. |
| 27 | class MovingMoments { |
| 28 | public: |
| 29 | // Creates a Moving Moments object, that uses the last |length| values |
| 30 | // (including the new value introduced in every new calculation). |
| 31 | explicit MovingMoments(size_t length); |
| 32 | ~MovingMoments(); |
| 33 | |
| 34 | // Calculates the new values using |in|. Results will be in the out buffers. |
| 35 | // |first| and |second| must be allocated with at least |in_length|. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 36 | void CalculateMoments(const float* in, |
| 37 | size_t in_length, |
| 38 | float* first, |
| 39 | float* second); |
pbos@webrtc.org | 788acd1 | 2014-12-15 09:41:24 +0000 | [diff] [blame] | 40 | |
| 41 | private: |
| 42 | size_t length_; |
| 43 | // A queue holding the |length_| latest input values. |
| 44 | std::queue<float> queue_; |
| 45 | // Sum of the values of the queue. |
| 46 | float sum_; |
| 47 | // Sum of the squares of the values of the queue. |
| 48 | float sum_of_squares_; |
| 49 | }; |
| 50 | |
| 51 | } // namespace webrtc |
| 52 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 53 | #endif // MODULES_AUDIO_PROCESSING_TRANSIENT_MOVING_MOMENTS_H_ |