OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 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/modules/video_coding/frame_buffer2.h" | 11 #include "webrtc/modules/video_coding/frame_buffer2.h" |
12 | 12 |
13 #include <algorithm> | 13 #include <algorithm> |
14 #include <cstring> | 14 #include <cstring> |
15 #include <limits> | 15 #include <limits> |
16 #include <vector> | 16 #include <vector> |
17 | 17 |
18 #include "webrtc/base/platform_thread.h" | 18 #include "webrtc/base/platform_thread.h" |
19 #include "webrtc/base/random.h" | 19 #include "webrtc/base/random.h" |
20 #include "webrtc/modules/video_coding/frame_object.h" | 20 #include "webrtc/modules/video_coding/frame_object.h" |
21 #include "webrtc/modules/video_coding/jitter_estimator.h" | 21 #include "webrtc/modules/video_coding/jitter_estimator.h" |
22 #include "webrtc/modules/video_coding/sequence_number_util.h" | 22 #include "webrtc/modules/video_coding/sequence_number_util.h" |
23 #include "webrtc/modules/video_coding/timing.h" | 23 #include "webrtc/modules/video_coding/timing.h" |
24 #include "webrtc/system_wrappers/include/clock.h" | 24 #include "webrtc/system_wrappers/include/clock.h" |
25 #include "webrtc/test/gmock.h" | 25 #include "webrtc/test/gmock.h" |
26 #include "webrtc/test/gtest.h" | 26 #include "webrtc/test/gtest.h" |
27 | 27 |
28 using testing::_; | |
29 using testing::Return; | |
30 | |
31 namespace webrtc { | 28 namespace webrtc { |
32 namespace video_coding { | 29 namespace video_coding { |
33 | 30 |
34 class VCMTimingFake : public VCMTiming { | 31 class VCMTimingFake : public VCMTiming { |
35 public: | 32 public: |
36 explicit VCMTimingFake(Clock* clock) : VCMTiming(clock) {} | 33 explicit VCMTimingFake(Clock* clock) : VCMTiming(clock) {} |
37 | 34 |
38 int64_t RenderTimeMs(uint32_t frame_timestamp, | 35 int64_t RenderTimeMs(uint32_t frame_timestamp, |
39 int64_t now_ms) const override { | 36 int64_t now_ms) const override { |
40 if (last_ms_ == -1) { | 37 if (last_ms_ == -1) { |
41 last_ms_ = now_ms + kDelayMs; | 38 last_ms_ = now_ms + kDelayMs; |
42 last_timestamp_ = frame_timestamp; | 39 last_timestamp_ = frame_timestamp; |
43 } | 40 } |
44 | 41 |
45 uint32_t diff = MinDiff(frame_timestamp, last_timestamp_); | 42 uint32_t diff = MinDiff(frame_timestamp, last_timestamp_); |
46 if (AheadOf(frame_timestamp, last_timestamp_)) | 43 if (AheadOf(frame_timestamp, last_timestamp_)) |
47 last_ms_ += diff / 90; | 44 last_ms_ += diff / 90; |
48 else | 45 else |
49 last_ms_ -= diff / 90; | 46 last_ms_ -= diff / 90; |
50 | 47 |
51 last_timestamp_ = frame_timestamp; | 48 last_timestamp_ = frame_timestamp; |
52 return last_ms_; | 49 return last_ms_; |
53 } | 50 } |
54 | 51 |
55 uint32_t MaxWaitingTime(int64_t render_time_ms, | 52 uint32_t MaxWaitingTime(int64_t render_time_ms, |
56 int64_t now_ms) const override { | 53 int64_t now_ms) const override { |
57 return std::max<int>(0, render_time_ms - now_ms - kDecodeTime); | 54 return std::max<int>(0, render_time_ms - now_ms - kDecodeTime); |
58 } | 55 } |
59 | 56 |
60 bool GetTimings(int* decode_ms, | |
61 int* max_decode_ms, | |
62 int* current_delay_ms, | |
63 int* target_delay_ms, | |
64 int* jitter_buffer_ms, | |
65 int* min_playout_delay_ms, | |
66 int* render_delay_ms) const override { | |
67 return true; | |
68 } | |
69 | |
70 private: | 57 private: |
71 static constexpr int kDelayMs = 50; | 58 static constexpr int kDelayMs = 50; |
72 static constexpr int kDecodeTime = kDelayMs / 2; | 59 static constexpr int kDecodeTime = kDelayMs / 2; |
73 mutable uint32_t last_timestamp_ = 0; | 60 mutable uint32_t last_timestamp_ = 0; |
74 mutable int64_t last_ms_ = -1; | 61 mutable int64_t last_ms_ = -1; |
75 }; | 62 }; |
76 | 63 |
77 class VCMJitterEstimatorMock : public VCMJitterEstimator { | 64 class VCMJitterEstimatorMock : public VCMJitterEstimator { |
78 public: | 65 public: |
79 explicit VCMJitterEstimatorMock(Clock* clock) : VCMJitterEstimator(clock) {} | 66 explicit VCMJitterEstimatorMock(Clock* clock) : VCMJitterEstimator(clock) {} |
80 | 67 |
81 MOCK_METHOD1(UpdateRtt, void(int64_t rttMs)); | 68 MOCK_METHOD1(UpdateRtt, void(int64_t rttMs)); |
82 MOCK_METHOD3(UpdateEstimate, | 69 MOCK_METHOD3(UpdateEstimate, |
83 void(int64_t frameDelayMs, | 70 void(int64_t frameDelayMs, |
84 uint32_t frameSizeBytes, | 71 uint32_t frameSizeBytes, |
85 bool incompleteFrame)); | 72 bool incompleteFrame)); |
86 MOCK_METHOD1(GetJitterEstimate, int(double rttMultiplier)); | 73 MOCK_METHOD1(GetJitterEstimate, int(double rttMultiplier)); |
87 }; | 74 }; |
88 | 75 |
89 class FrameObjectFake : public FrameObject { | 76 class FrameObjectFake : public FrameObject { |
90 public: | 77 public: |
91 bool GetBitstream(uint8_t* destination) const override { return true; } | 78 bool GetBitstream(uint8_t* destination) const override { return true; } |
92 | 79 |
93 uint32_t Timestamp() const override { return timestamp; } | 80 uint32_t Timestamp() const override { return timestamp; } |
94 | 81 |
95 int64_t ReceivedTime() const override { return 0; } | 82 int64_t ReceivedTime() const override { return 0; } |
96 | 83 |
97 int64_t RenderTime() const override { return _renderTimeMs; } | 84 int64_t RenderTime() const override { return _renderTimeMs; } |
98 | |
99 // In EncodedImage |_length| is used to descibe its size and |_size| to | |
100 // describe its capacity. | |
101 void SetSize(int size) { _length = size; } | |
102 }; | |
103 | |
104 class VCMReceiveStatisticsCallbackMock : public VCMReceiveStatisticsCallback { | |
105 public: | |
106 MOCK_METHOD2(OnReceiveRatesUpdated, | |
107 void(uint32_t bitRate, uint32_t frameRate)); | |
108 MOCK_METHOD2(OnCompleteFrame, void(bool is_keyframe, size_t size_bytes)); | |
109 MOCK_METHOD1(OnDiscardedPacketsUpdated, void(int discarded_packets)); | |
110 MOCK_METHOD1(OnFrameCountsUpdated, void(const FrameCounts& frame_counts)); | |
111 MOCK_METHOD7(OnFrameBufferTimingsUpdated, | |
112 void(int decode_ms, | |
113 int max_decode_ms, | |
114 int current_delay_ms, | |
115 int target_delay_ms, | |
116 int jitter_buffer_ms, | |
117 int min_playout_delay_ms, | |
118 int render_delay_ms)); | |
119 }; | 85 }; |
120 | 86 |
121 class TestFrameBuffer2 : public ::testing::Test { | 87 class TestFrameBuffer2 : public ::testing::Test { |
122 protected: | 88 protected: |
123 static constexpr int kMaxReferences = 5; | 89 static constexpr int kMaxReferences = 5; |
124 static constexpr int kFps1 = 1000; | 90 static constexpr int kFps1 = 1000; |
125 static constexpr int kFps10 = kFps1 / 10; | 91 static constexpr int kFps10 = kFps1 / 10; |
126 static constexpr int kFps20 = kFps1 / 20; | 92 static constexpr int kFps20 = kFps1 / 20; |
127 | 93 |
128 TestFrameBuffer2() | 94 TestFrameBuffer2() |
129 : clock_(0), | 95 : clock_(0), |
130 timing_(&clock_), | 96 timing_(&clock_), |
131 jitter_estimator_(&clock_), | 97 jitter_estimator_(&clock_), |
132 buffer_(&clock_, &jitter_estimator_, &timing_, &stats_callback_), | 98 buffer_(&clock_, &jitter_estimator_, &timing_), |
133 rand_(0x34678213), | 99 rand_(0x34678213), |
134 tear_down_(false), | 100 tear_down_(false), |
135 extract_thread_(&ExtractLoop, this, "Extract Thread"), | 101 extract_thread_(&ExtractLoop, this, "Extract Thread"), |
136 trigger_extract_event_(false, false), | 102 trigger_extract_event_(false, false), |
137 crit_acquired_event_(false, false) {} | 103 crit_acquired_event_(false, false) {} |
138 | 104 |
139 void SetUp() override { extract_thread_.Start(); } | 105 void SetUp() override { extract_thread_.Start(); } |
140 | 106 |
141 void TearDown() override { | 107 void TearDown() override { |
142 tear_down_ = true; | 108 tear_down_ = true; |
(...skipping 74 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
217 } | 183 } |
218 | 184 |
219 uint32_t Rand() { return rand_.Rand<uint32_t>(); } | 185 uint32_t Rand() { return rand_.Rand<uint32_t>(); } |
220 | 186 |
221 SimulatedClock clock_; | 187 SimulatedClock clock_; |
222 VCMTimingFake timing_; | 188 VCMTimingFake timing_; |
223 ::testing::NiceMock<VCMJitterEstimatorMock> jitter_estimator_; | 189 ::testing::NiceMock<VCMJitterEstimatorMock> jitter_estimator_; |
224 FrameBuffer buffer_; | 190 FrameBuffer buffer_; |
225 std::vector<std::unique_ptr<FrameObject>> frames_; | 191 std::vector<std::unique_ptr<FrameObject>> frames_; |
226 Random rand_; | 192 Random rand_; |
227 ::testing::NiceMock<VCMReceiveStatisticsCallbackMock> stats_callback_; | |
228 | 193 |
229 int64_t max_wait_time_; | 194 int64_t max_wait_time_; |
230 bool tear_down_; | 195 bool tear_down_; |
231 rtc::PlatformThread extract_thread_; | 196 rtc::PlatformThread extract_thread_; |
232 rtc::Event trigger_extract_event_; | 197 rtc::Event trigger_extract_event_; |
233 rtc::Event crit_acquired_event_; | 198 rtc::Event crit_acquired_event_; |
234 rtc::CriticalSection crit_; | 199 rtc::CriticalSection crit_; |
235 }; | 200 }; |
236 | 201 |
237 // Following tests are timing dependent. Either the timeouts have to | 202 // Following tests are timing dependent. Either the timeouts have to |
(...skipping 226 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
464 CheckFrame(0, pid, 0); | 429 CheckFrame(0, pid, 0); |
465 | 430 |
466 // Jump back in pid but increase ts. | 431 // Jump back in pid but increase ts. |
467 EXPECT_EQ(pid - 1, InsertFrame(pid - 1, 0, ts + 2, false)); | 432 EXPECT_EQ(pid - 1, InsertFrame(pid - 1, 0, ts + 2, false)); |
468 ExtractFrame(); | 433 ExtractFrame(); |
469 ExtractFrame(); | 434 ExtractFrame(); |
470 CheckFrame(1, pid - 1, 0); | 435 CheckFrame(1, pid - 1, 0); |
471 CheckNoFrame(2); | 436 CheckNoFrame(2); |
472 } | 437 } |
473 | 438 |
474 TEST_F(TestFrameBuffer2, StatsCallback) { | |
475 uint16_t pid = Rand(); | |
476 uint32_t ts = Rand(); | |
477 const int kFrameSize = 5000; | |
478 | |
479 EXPECT_CALL(stats_callback_, OnCompleteFrame(true, kFrameSize)); | |
480 EXPECT_CALL(stats_callback_, | |
481 OnFrameBufferTimingsUpdated(_, _, _, _, _, _, _)); | |
482 | |
483 { | |
484 std::unique_ptr<FrameObjectFake> frame(new FrameObjectFake()); | |
485 frame->SetSize(kFrameSize); | |
486 frame->picture_id = pid; | |
487 frame->spatial_layer = 0; | |
488 frame->timestamp = ts; | |
489 frame->num_references = 0; | |
490 frame->inter_layer_predicted = false; | |
491 | |
492 EXPECT_EQ(buffer_.InsertFrame(std::move(frame)), pid); | |
493 } | |
494 | |
495 ExtractFrame(); | |
496 CheckFrame(0, pid, 0); | |
497 } | |
498 | |
499 } // namespace video_coding | 439 } // namespace video_coding |
500 } // namespace webrtc | 440 } // namespace webrtc |
OLD | NEW |