Chromium Code Reviews| Index: webrtc/modules/video_coding/histogram.cc |
| diff --git a/webrtc/modules/video_coding/histogram.cc b/webrtc/modules/video_coding/histogram.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..000ae87557cf8ea0c34ad46cc272fddf62a3d3a2 |
| --- /dev/null |
| +++ b/webrtc/modules/video_coding/histogram.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 "webrtc/modules/video_coding/histogram.h" |
| + |
| +#include <algorithm> |
| + |
| +#include "webrtc/base/mod_ops.h" |
| + |
| +namespace webrtc { |
| +namespace video_coding { |
| +Histogram::Histogram(int num_buckets, int max_num_values) { |
|
stefan-webrtc
2016/03/02 14:32:05
Should these be size_t?
philipel
2016/03/02 15:35:10
Done.
|
| + buckets_.resize(num_buckets); |
| + values_.reserve(max_num_values); |
| + index_ = 0; |
| +} |
| + |
| +void Histogram::Add(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/02 14:32:05
DCHECK that values_[index_] < buckets_.size()
philipel
2016/03/02 15:35:10
Done.
|
| + values_[index_] = value; |
| + } else { |
| + values_.emplace_back(value); |
| + } |
| + |
| + ++buckets_[value]; |
| + index_ = (index_ + 1) % values_.capacity(); |
| +} |
| + |
| +size_t Histogram::InverseCdf(float probability) const { |
| + RTC_DCHECK_LE(0.f, probability); |
| + RTC_DCHECK_GE(1.f, probability); |
|
stefan-webrtc
2016/03/02 14:32:05
This is easier to read in my opinion. Feel free to
philipel
2016/03/02 15:35:10
Done.
|
| + RTC_DCHECK_LE(0ul, values_.size()); |
| + |
| + size_t bucket = 0; |
| + float accumulated_probability = 0; |
| + while (accumulated_probability < probability && bucket < buckets_.size()) { |
| + accumulated_probability += |
| + static_cast<float>(buckets_[bucket]) / values_.size(); |
| + ++bucket; |
| + } |
| + return bucket; |
| +} |
| + |
| +size_t Histogram::NumValues() const { |
| + return values_.size(); |
| +} |
| + |
| +} // namespace video_coding |
| +} // namespace webrtc |