Index: webrtc/modules/video_coding/distribution.cc |
diff --git a/webrtc/modules/video_coding/distribution.cc b/webrtc/modules/video_coding/distribution.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f433b1fb70afbc47987b0cecae3f15cc4f90b766 |
--- /dev/null |
+++ b/webrtc/modules/video_coding/distribution.cc |
@@ -0,0 +1,59 @@ |
+/* |
+ * 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 <algorithm> |
+ |
+#include "webrtc/modules/video_coding/distribution.h" |
stefan-webrtc
2016/03/01 08:52:10
this one should be first
philipel
2016/03/01 10:27:05
Done.
|
+ |
+#include "webrtc/base/mod_ops.h" |
+ |
+namespace webrtc { |
+ |
+Distribution::Distribution(int num_buckets, int max_num_values) { |
+ buckets_.resize(num_buckets); |
+ values_.reserve(max_num_values); |
+ index_ = 0; |
+} |
+ |
+void Distribution::AddValue(int value) { |
+ RTC_DCHECK_LE(0, value); |
+ value = std::min<int>(value, buckets_.size() - 1); |
+ if (index_ < values_.size()) { |
+ buckets_[values_[index_]]--; |
stefan-webrtc
2016/03/01 08:52:10
--buckets_...;
philipel
2016/03/01 10:27:05
Done.
|
+ values_[index_] = value; |
+ } else { |
+ values_.emplace_back(value); |
+ } |
+ |
+ buckets_[value]++; |
stefan-webrtc
2016/03/01 08:52:10
++buckets_...;
philipel
2016/03/01 10:27:05
Done.
|
+ index_ = (index_ + 1) % values_.capacity(); |
+} |
+ |
+size_t Distribution::InverseCDF(float probability) const { |
+ RTC_DCHECK_LE(1.f, probability); |
+ RTC_DCHECK_GE(0.f, probability); |
+ RTC_DCHECK_LE(0ul, values_.size()); |
+ |
+ size_t bucket = 0; |
+ float accumulated_probability = 0; |
+ while (accumulated_probability < probability && |
+ bucket < buckets_.size()) { |
stefan-webrtc
2016/03/01 08:52:10
git cl format
philipel
2016/03/01 10:27:05
Done.
|
+ accumulated_probability += static_cast<float>(buckets_[bucket])/ |
+ values_.size(); |
+ ++bucket; |
+ } |
+ return bucket; |
+} |
+ |
+size_t Distribution::NumValues() const { |
+ return values_.size(); |
+} |
+ |
+} // namespace webrtc |