Index: webrtc/modules/video_coding/utility/moving_average.cc |
diff --git a/webrtc/modules/video_coding/utility/moving_average.cc b/webrtc/modules/video_coding/utility/moving_average.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f43f7112d94fb7dd0837f5ac110b706c38775d31 |
--- /dev/null |
+++ b/webrtc/modules/video_coding/utility/moving_average.cc |
@@ -0,0 +1,50 @@ |
+/* |
+ * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
+ * |
+ * Use of this source code is governed by a BSD-style license |
+ * that can be found in the LICENSE file in the root of the source |
+ * tree. An additional intellectual property rights grant can be found |
+ * in the file PATENTS. All contributing project authors may |
+ * be found in the AUTHORS file in the root of the source tree. |
+ */ |
+ |
+#include "webrtc/modules/video_coding/utility/moving_average.h" |
+ |
+#include <algorithm> |
+ |
+namespace webrtc { |
+ |
+MovingAverage::MovingAverage(size_t s) : sum_history_(s + 1, 0) {} |
+ |
+void MovingAverage::AddSample(int sample) { |
+ count_++; |
+ sum_ += sample; |
+ sum_history_[count_ % sum_history_.size()] = sum_; |
+} |
+ |
+int MovingAverage::GetAverage() const { |
+ return *GetAverage(size()); |
+} |
+ |
+rtc::Optional<int> MovingAverage::GetAverage(size_t num_samples) const { |
+ if (num_samples > size()) |
+ return rtc::Optional<int>(); |
+ if (num_samples == 0) |
+ return rtc::Optional<int>(0); |
magjed_webrtc
2016/09/07 11:55:16
I find it more intuitive if we return rtc::Optiona
|
+ int sum = sum_ - sum_history_[(count_ - num_samples) % sum_history_.size()]; |
+ if (sum == 0) |
sakal
2016/09/07 09:47:21
What is this special case for? Looks like a bug to
kthelgason
2016/09/07 12:31:46
Well spotted. That was left in by accident when I
|
+ sum = sum_; |
+ return rtc::Optional<int>(sum / num_samples); |
+} |
+ |
+void MovingAverage::Reset() { |
+ count_ = 0; |
+ sum_ = 0; |
+ std::fill(sum_history_.begin(), sum_history_.end(), 0); |
+} |
+ |
+size_t MovingAverage::size() const { |
+ return std::min(count_, sum_history_.size() - 1); |
+} |
+ |
+} // namespace webrtc |