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

Side by Side Diff: webrtc/base/ratetracker.cc

Issue 2877023002: Move webrtc/{base => rtc_base} (Closed)
Patch Set: update presubmit.py and DEPS include rules Created 3 years, 5 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
« no previous file with comments | « webrtc/base/ratetracker.h ('k') | webrtc/base/ratetracker_unittest.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 /*
2 * Copyright 2015 The WebRTC Project Authors. All rights reserved.
3 *
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
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include "webrtc/base/ratetracker.h"
12
13 #include <stddef.h>
14
15 #include <algorithm>
16
17 #include "webrtc/base/checks.h"
18 #include "webrtc/base/timeutils.h"
19
20 namespace rtc {
21
22 static const int64_t kTimeUnset = -1;
23
24 RateTracker::RateTracker(int64_t bucket_milliseconds, size_t bucket_count)
25 : bucket_milliseconds_(bucket_milliseconds),
26 bucket_count_(bucket_count),
27 sample_buckets_(new size_t[bucket_count + 1]),
28 total_sample_count_(0u),
29 bucket_start_time_milliseconds_(kTimeUnset) {
30 RTC_CHECK(bucket_milliseconds > 0);
31 RTC_CHECK(bucket_count > 0);
32 }
33
34 RateTracker::~RateTracker() {
35 delete[] sample_buckets_;
36 }
37
38 double RateTracker::ComputeRateForInterval(
39 int64_t interval_milliseconds) const {
40 if (bucket_start_time_milliseconds_ == kTimeUnset) {
41 return 0.0;
42 }
43 int64_t current_time = Time();
44 // Calculate which buckets to sum up given the current time. If the time
45 // has passed to a new bucket then we have to skip some of the oldest buckets.
46 int64_t available_interval_milliseconds =
47 std::min(interval_milliseconds,
48 bucket_milliseconds_ * static_cast<int64_t>(bucket_count_));
49 // number of old buckets (i.e. after the current bucket in the ring buffer)
50 // that are expired given our current time interval.
51 size_t buckets_to_skip;
52 // Number of milliseconds of the first bucket that are not a portion of the
53 // current interval.
54 int64_t milliseconds_to_skip;
55 if (current_time >
56 initialization_time_milliseconds_ + available_interval_milliseconds) {
57 int64_t time_to_skip =
58 current_time - bucket_start_time_milliseconds_ +
59 static_cast<int64_t>(bucket_count_) * bucket_milliseconds_ -
60 available_interval_milliseconds;
61 buckets_to_skip = time_to_skip / bucket_milliseconds_;
62 milliseconds_to_skip = time_to_skip % bucket_milliseconds_;
63 } else {
64 buckets_to_skip = bucket_count_ - current_bucket_;
65 milliseconds_to_skip = 0;
66 available_interval_milliseconds =
67 TimeDiff(current_time, initialization_time_milliseconds_);
68 // Let one bucket interval pass after initialization before reporting.
69 if (available_interval_milliseconds < bucket_milliseconds_) {
70 return 0.0;
71 }
72 }
73 // If we're skipping all buckets that means that there have been no samples
74 // within the sampling interval so report 0.
75 if (buckets_to_skip > bucket_count_ || available_interval_milliseconds == 0) {
76 return 0.0;
77 }
78 size_t start_bucket = NextBucketIndex(current_bucket_ + buckets_to_skip);
79 // Only count a portion of the first bucket according to how much of the
80 // first bucket is within the current interval.
81 size_t total_samples = ((sample_buckets_[start_bucket] *
82 (bucket_milliseconds_ - milliseconds_to_skip)) +
83 (bucket_milliseconds_ >> 1)) /
84 bucket_milliseconds_;
85 // All other buckets in the interval are counted in their entirety.
86 for (size_t i = NextBucketIndex(start_bucket);
87 i != NextBucketIndex(current_bucket_);
88 i = NextBucketIndex(i)) {
89 total_samples += sample_buckets_[i];
90 }
91 // Convert to samples per second.
92 return static_cast<double>(total_samples * 1000) /
93 static_cast<double>(available_interval_milliseconds);
94 }
95
96 double RateTracker::ComputeTotalRate() const {
97 if (bucket_start_time_milliseconds_ == kTimeUnset) {
98 return 0.0;
99 }
100 int64_t current_time = Time();
101 if (current_time <= initialization_time_milliseconds_) {
102 return 0.0;
103 }
104 return static_cast<double>(total_sample_count_ * 1000) /
105 static_cast<double>(
106 TimeDiff(current_time, initialization_time_milliseconds_));
107 }
108
109 size_t RateTracker::TotalSampleCount() const {
110 return total_sample_count_;
111 }
112
113 void RateTracker::AddSamples(size_t sample_count) {
114 EnsureInitialized();
115 int64_t current_time = Time();
116 // Advance the current bucket as needed for the current time, and reset
117 // bucket counts as we advance.
118 for (size_t i = 0;
119 i <= bucket_count_ &&
120 current_time >= bucket_start_time_milliseconds_ + bucket_milliseconds_;
121 ++i) {
122 bucket_start_time_milliseconds_ += bucket_milliseconds_;
123 current_bucket_ = NextBucketIndex(current_bucket_);
124 sample_buckets_[current_bucket_] = 0;
125 }
126 // Ensure that bucket_start_time_milliseconds_ is updated appropriately if
127 // the entire buffer of samples has been expired.
128 bucket_start_time_milliseconds_ += bucket_milliseconds_ *
129 ((current_time - bucket_start_time_milliseconds_) / bucket_milliseconds_);
130 // Add all samples in the bucket that includes the current time.
131 sample_buckets_[current_bucket_] += sample_count;
132 total_sample_count_ += sample_count;
133 }
134
135 int64_t RateTracker::Time() const {
136 return rtc::TimeMillis();
137 }
138
139 void RateTracker::EnsureInitialized() {
140 if (bucket_start_time_milliseconds_ == kTimeUnset) {
141 initialization_time_milliseconds_ = Time();
142 bucket_start_time_milliseconds_ = initialization_time_milliseconds_;
143 current_bucket_ = 0;
144 // We only need to initialize the first bucket because we reset buckets when
145 // current_bucket_ increments.
146 sample_buckets_[current_bucket_] = 0;
147 }
148 }
149
150 size_t RateTracker::NextBucketIndex(size_t bucket_index) const {
151 return (bucket_index + 1u) % (bucket_count_ + 1u);
152 }
153
154 } // namespace rtc
OLDNEW
« no previous file with comments | « webrtc/base/ratetracker.h ('k') | webrtc/base/ratetracker_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698