OLD | NEW |
---|---|
1 /* | 1 /* |
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef WEBRTC_MODULES_VIDEO_CODING_UTILITY_MOVING_AVERAGE_H_ | 11 #ifndef WEBRTC_MODULES_VIDEO_CODING_UTILITY_MOVING_AVERAGE_H_ |
12 #define WEBRTC_MODULES_VIDEO_CODING_UTILITY_MOVING_AVERAGE_H_ | 12 #define WEBRTC_MODULES_VIDEO_CODING_UTILITY_MOVING_AVERAGE_H_ |
13 | 13 |
14 #include <stddef.h> | 14 #include <stddef.h> |
15 | 15 |
16 #include <list> | 16 #include <vector> |
17 | |
18 #include "webrtc/typedefs.h" | |
19 | 17 |
20 namespace webrtc { | 18 namespace webrtc { |
21 template <class T> | |
22 class MovingAverage { | 19 class MovingAverage { |
23 public: | 20 public: |
24 MovingAverage(); | 21 MovingAverage(); |
25 void AddSample(T sample); | 22 explicit MovingAverage(size_t s); |
26 bool GetAverage(size_t num_samples, T* average); | 23 void AddSample(int sample); |
24 int GetAverage() const; | |
25 int GetAverage(size_t num_samples) const; | |
27 void Reset(); | 26 void Reset(); |
28 int size(); | 27 size_t size() const; |
29 | 28 |
30 private: | 29 private: |
31 T sum_; | 30 size_t count_ = 0; |
32 std::list<T> samples_; | 31 int sum_ = 0; |
32 std::vector<int> samples_; | |
sakal
2016/09/05 12:28:11
This vector does not contain samples. Maybe sum_hi
| |
33 }; | 33 }; |
34 | |
35 template <class T> | |
36 MovingAverage<T>::MovingAverage() | |
37 : sum_(static_cast<T>(0)) {} | |
38 | |
39 template <class T> | |
40 void MovingAverage<T>::AddSample(T sample) { | |
41 samples_.push_back(sample); | |
42 sum_ += sample; | |
43 } | |
44 | |
45 template <class T> | |
46 bool MovingAverage<T>::GetAverage(size_t num_samples, T* avg) { | |
47 if (num_samples > samples_.size()) | |
48 return false; | |
49 | |
50 // Remove old samples. | |
51 while (num_samples < samples_.size()) { | |
52 sum_ -= samples_.front(); | |
53 samples_.pop_front(); | |
54 } | |
55 | |
56 *avg = sum_ / static_cast<T>(num_samples); | |
57 return true; | |
58 } | |
59 | |
60 template <class T> | |
61 void MovingAverage<T>::Reset() { | |
62 sum_ = static_cast<T>(0); | |
63 samples_.clear(); | |
64 } | |
65 | |
66 template <class T> | |
67 int MovingAverage<T>::size() { | |
68 return samples_.size(); | |
69 } | |
70 | |
71 } // namespace webrtc | 34 } // namespace webrtc |
72 | 35 |
73 #endif // WEBRTC_MODULES_VIDEO_CODING_UTILITY_MOVING_AVERAGE_H_ | 36 #endif // WEBRTC_MODULES_VIDEO_CODING_UTILITY_MOVING_AVERAGE_H_ |
OLD | NEW |