OLD | NEW |
| (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 | |
OLD | NEW |