OLD | NEW |
1 /* | 1 /* |
2 * Copyright 2010 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2015 The WebRTC Project Authors. All rights reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/base/gunit.h" | 11 #include "webrtc/base/gunit.h" |
12 #include "webrtc/base/ratetracker.h" | 12 #include "webrtc/base/ratetracker.h" |
13 | 13 |
14 namespace rtc { | 14 namespace rtc { |
15 | 15 |
16 class RateTrackerForTest : public RateTracker { | 16 class RateTrackerForTest : public RateTracker { |
17 public: | 17 public: |
18 RateTrackerForTest() : time_(0) {} | 18 RateTrackerForTest() : RateTracker(100u, 10u), time_(0) {} |
19 virtual uint32 Time() const { return time_; } | 19 virtual uint32 Time() const { return time_; } |
20 void AdvanceTime(uint32 delta) { time_ += delta; } | 20 void AdvanceTime(uint32 delta) { time_ += delta; } |
21 | 21 |
22 private: | 22 private: |
23 uint32 time_; | 23 uint32 time_; |
24 }; | 24 }; |
25 | 25 |
26 TEST(RateTrackerTest, TestBasics) { | 26 TEST(RateTrackerTest, Test30FPS) { |
27 RateTrackerForTest tracker; | 27 RateTrackerForTest tracker; |
28 EXPECT_EQ(0U, tracker.total_units()); | 28 |
29 EXPECT_EQ(0U, tracker.units_second()); | 29 for (int i = 0; i < 300; ++i) { |
| 30 tracker.AddSamples(1); |
| 31 tracker.AdvanceTime(33); |
| 32 if (i % 3 == 0) { |
| 33 tracker.AdvanceTime(1); |
| 34 } |
| 35 } |
| 36 EXPECT_DOUBLE_EQ(30.0, tracker.ComputeRateForInterval(50000u)); |
| 37 } |
| 38 |
| 39 TEST(RateTrackerTest, Test60FPS) { |
| 40 RateTrackerForTest tracker; |
| 41 |
| 42 for (int i = 0; i < 300; ++i) { |
| 43 tracker.AddSamples(1); |
| 44 tracker.AdvanceTime(16); |
| 45 if (i % 3 != 0) { |
| 46 tracker.AdvanceTime(1); |
| 47 } |
| 48 } |
| 49 EXPECT_DOUBLE_EQ(60.0, tracker.ComputeRateForInterval(1000u)); |
| 50 } |
| 51 |
| 52 TEST(RateTrackerTest, TestRateTrackerBasics) { |
| 53 RateTrackerForTest tracker; |
| 54 EXPECT_DOUBLE_EQ(0.0, tracker.ComputeRateForInterval(1000u)); |
30 | 55 |
31 // Add a sample. | 56 // Add a sample. |
32 tracker.Update(1234); | 57 tracker.AddSamples(1234); |
33 // Advance the clock by 100 ms. | 58 // Advance the clock by 100 ms. |
34 tracker.AdvanceTime(100); | 59 tracker.AdvanceTime(100); |
35 // total_units should advance, but units_second should stay 0. | 60 EXPECT_DOUBLE_EQ(12340.0, tracker.ComputeRateForInterval(1000u)); |
36 EXPECT_EQ(1234U, tracker.total_units()); | 61 EXPECT_DOUBLE_EQ(12340.0, tracker.ComputeRate()); |
37 EXPECT_EQ(0U, tracker.units_second()); | 62 EXPECT_EQ(1234U, tracker.TotalSampleCount()); |
| 63 EXPECT_DOUBLE_EQ(12340.0, tracker.ComputeTotalRate()); |
38 | 64 |
39 // Repeat. | 65 // Repeat. |
40 tracker.Update(1234); | 66 tracker.AddSamples(1234); |
41 tracker.AdvanceTime(100); | 67 tracker.AdvanceTime(100); |
42 EXPECT_EQ(1234U * 2, tracker.total_units()); | 68 EXPECT_DOUBLE_EQ(12340.0, tracker.ComputeRateForInterval(1000u)); |
43 EXPECT_EQ(0U, tracker.units_second()); | 69 EXPECT_DOUBLE_EQ(12340.0, tracker.ComputeRate()); |
| 70 EXPECT_EQ(1234U * 2, tracker.TotalSampleCount()); |
| 71 EXPECT_DOUBLE_EQ(12340.0, tracker.ComputeTotalRate()); |
44 | 72 |
45 // Advance the clock by 800 ms, so we've elapsed a full second. | 73 // Advance the clock by 800 ms, so we've elapsed a full second. |
46 // units_second should now be filled in properly. | 74 // units_second should now be filled in properly. |
47 tracker.AdvanceTime(800); | 75 tracker.AdvanceTime(800); |
48 EXPECT_EQ(1234U * 2, tracker.total_units()); | 76 EXPECT_DOUBLE_EQ(1234.0 * 2.0, tracker.ComputeRateForInterval(1000u)); |
49 EXPECT_EQ(1234U * 2, tracker.units_second()); | 77 EXPECT_DOUBLE_EQ(1234.0 * 2.0, tracker.ComputeRate()); |
| 78 EXPECT_EQ(1234U * 2, tracker.TotalSampleCount()); |
| 79 EXPECT_DOUBLE_EQ(1234.0 * 2.0, tracker.ComputeTotalRate()); |
50 | 80 |
51 // Poll the tracker again immediately. The reported rate should stay the same. | 81 // Poll the tracker again immediately. The reported rate should stay the same. |
52 EXPECT_EQ(1234U * 2, tracker.total_units()); | 82 EXPECT_DOUBLE_EQ(1234.0 * 2.0, tracker.ComputeRateForInterval(1000u)); |
53 EXPECT_EQ(1234U * 2, tracker.units_second()); | 83 EXPECT_DOUBLE_EQ(1234.0 * 2.0, tracker.ComputeRate()); |
| 84 EXPECT_EQ(1234U * 2, tracker.TotalSampleCount()); |
| 85 EXPECT_DOUBLE_EQ(1234.0 * 2.0, tracker.ComputeTotalRate()); |
54 | 86 |
55 // Do nothing and advance by a second. We should drop down to zero. | 87 // Do nothing and advance by a second. We should drop down to zero. |
56 tracker.AdvanceTime(1000); | 88 tracker.AdvanceTime(1000); |
57 EXPECT_EQ(1234U * 2, tracker.total_units()); | 89 EXPECT_DOUBLE_EQ(0.0, tracker.ComputeRateForInterval(1000u)); |
58 EXPECT_EQ(0U, tracker.units_second()); | 90 EXPECT_DOUBLE_EQ(0.0, tracker.ComputeRate()); |
| 91 EXPECT_EQ(1234U * 2, tracker.TotalSampleCount()); |
| 92 EXPECT_DOUBLE_EQ(1234.0, tracker.ComputeTotalRate()); |
59 | 93 |
60 // Send a bunch of data at a constant rate for 5.5 "seconds". | 94 // Send a bunch of data at a constant rate for 5.5 "seconds". |
61 // We should report the rate properly. | 95 // We should report the rate properly. |
62 for (int i = 0; i < 5500; i += 100) { | 96 for (int i = 0; i < 5500; i += 100) { |
63 tracker.Update(9876U); | 97 tracker.AddSamples(9876U); |
64 tracker.AdvanceTime(100); | 98 tracker.AdvanceTime(100); |
65 } | 99 } |
66 EXPECT_EQ(9876U * 10, tracker.units_second()); | 100 EXPECT_DOUBLE_EQ(9876.0 * 10.0, tracker.ComputeRateForInterval(1000u)); |
| 101 EXPECT_DOUBLE_EQ(9876.0 * 10.0, tracker.ComputeRate()); |
| 102 EXPECT_EQ(1234U * 2 + 9876U * 55, tracker.TotalSampleCount()); |
| 103 EXPECT_DOUBLE_EQ((1234.0 * 2.0 + 9876.0 * 55.0) / 7.5, |
| 104 tracker.ComputeTotalRate()); |
67 | 105 |
68 // Advance the clock by 500 ms. Since we sent nothing over this half-second, | 106 // Advance the clock by 500 ms. Since we sent nothing over this half-second, |
69 // the reported rate should be reduced by half. | 107 // the reported rate should be reduced by half. |
70 tracker.AdvanceTime(500); | 108 tracker.AdvanceTime(500); |
71 EXPECT_EQ(9876U * 5, tracker.units_second()); | 109 EXPECT_DOUBLE_EQ(9876.0 * 5.0, tracker.ComputeRateForInterval(1000u)); |
| 110 EXPECT_DOUBLE_EQ(9876.0 * 5.0, tracker.ComputeRate()); |
| 111 EXPECT_EQ(1234U * 2 + 9876U * 55, tracker.TotalSampleCount()); |
| 112 EXPECT_DOUBLE_EQ((1234.0 * 2.0 + 9876.0 * 55.0) / 8.0, |
| 113 tracker.ComputeTotalRate()); |
| 114 |
| 115 // Rate over the last half second should be zero. |
| 116 EXPECT_DOUBLE_EQ(0.0, tracker.ComputeRateForInterval(500u)); |
| 117 } |
| 118 |
| 119 TEST(RateTrackerTest, TestLongPeriodBetweenSamples) { |
| 120 RateTrackerForTest tracker; |
| 121 tracker.AddSamples(1); |
| 122 tracker.AdvanceTime(1000); |
| 123 EXPECT_DOUBLE_EQ(1.0, tracker.ComputeRate()); |
| 124 |
| 125 tracker.AdvanceTime(2000); |
| 126 EXPECT_DOUBLE_EQ(0.0, tracker.ComputeRate()); |
| 127 |
| 128 tracker.AdvanceTime(2000); |
| 129 tracker.AddSamples(1); |
| 130 EXPECT_DOUBLE_EQ(1.0, tracker.ComputeRate()); |
| 131 } |
| 132 |
| 133 TEST(RateTrackerTest, TestRolloff) { |
| 134 RateTrackerForTest tracker; |
| 135 for (int i = 0; i < 10; ++i) { |
| 136 tracker.AddSamples(1U); |
| 137 tracker.AdvanceTime(100); |
| 138 } |
| 139 EXPECT_DOUBLE_EQ(10.0, tracker.ComputeRate()); |
| 140 |
| 141 for (int i = 0; i < 10; ++i) { |
| 142 tracker.AddSamples(1U); |
| 143 tracker.AdvanceTime(50); |
| 144 } |
| 145 EXPECT_DOUBLE_EQ(15.0, tracker.ComputeRate()); |
| 146 EXPECT_DOUBLE_EQ(20.0, tracker.ComputeRateForInterval(500u)); |
| 147 |
| 148 for (int i = 0; i < 10; ++i) { |
| 149 tracker.AddSamples(1U); |
| 150 tracker.AdvanceTime(50); |
| 151 } |
| 152 EXPECT_DOUBLE_EQ(20.0, tracker.ComputeRate()); |
72 } | 153 } |
73 | 154 |
74 TEST(RateTrackerTest, TestGetUnitSecondsAfterInitialValue) { | 155 TEST(RateTrackerTest, TestGetUnitSecondsAfterInitialValue) { |
75 RateTrackerForTest tracker; | 156 RateTrackerForTest tracker; |
76 tracker.Update(1234); | 157 tracker.AddSamples(1234); |
77 tracker.AdvanceTime(1000); | 158 tracker.AdvanceTime(1000); |
78 EXPECT_EQ(1234u, tracker.units_second()); | 159 EXPECT_DOUBLE_EQ(1234.0, tracker.ComputeRateForInterval(1000u)); |
79 } | 160 } |
80 | 161 |
81 } // namespace rtc | 162 } // namespace rtc |
OLD | NEW |