Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(212)

Side by Side Diff: webrtc/modules/video_coding/utility/moving_average.h

Issue 2310853002: Refactor QualityScaler and MovingAverage (Closed)
Patch Set: Created 4 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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_
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698