OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <stddef.h> | 5 #include <stddef.h> |
6 #include <stdint.h> | 6 #include <stdint.h> |
7 | 7 |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
11 #include "base/macros.h" | 11 #include "base/macros.h" |
12 #include "base/memory/scoped_ptr.h" | 12 #include "base/memory/ptr_util.h" |
13 #include "base/test/simple_test_tick_clock.h" | 13 #include "base/test/simple_test_tick_clock.h" |
14 #include "media/base/fake_single_thread_task_runner.h" | 14 #include "media/base/fake_single_thread_task_runner.h" |
15 #include "media/base/video_frame.h" | 15 #include "media/base/video_frame.h" |
16 #include "media/cast/cast_environment.h" | 16 #include "media/cast/cast_environment.h" |
17 #include "media/cast/constants.h" | 17 #include "media/cast/constants.h" |
18 #include "media/cast/logging/simple_event_subscriber.h" | 18 #include "media/cast/logging/simple_event_subscriber.h" |
19 #include "media/cast/net/cast_transport_config.h" | 19 #include "media/cast/net/cast_transport_config.h" |
20 #include "media/cast/net/cast_transport_impl.h" | 20 #include "media/cast/net/cast_transport_impl.h" |
21 #include "media/cast/net/pacing/paced_sender.h" | 21 #include "media/cast/net/pacing/paced_sender.h" |
22 #include "media/cast/sender/fake_video_encode_accelerator_factory.h" | 22 #include "media/cast/sender/fake_video_encode_accelerator_factory.h" |
(...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
100 bool paused_; | 100 bool paused_; |
101 base::Closure callback_; | 101 base::Closure callback_; |
102 PacketRef stored_packet_; | 102 PacketRef stored_packet_; |
103 | 103 |
104 DISALLOW_COPY_AND_ASSIGN(TestPacketSender); | 104 DISALLOW_COPY_AND_ASSIGN(TestPacketSender); |
105 }; | 105 }; |
106 | 106 |
107 void IgnorePlayoutDelayChanges(base::TimeDelta unused_playout_delay) { | 107 void IgnorePlayoutDelayChanges(base::TimeDelta unused_playout_delay) { |
108 } | 108 } |
109 | 109 |
110 class PeerVideoSender : public VideoSender { | |
111 public: | |
112 PeerVideoSender( | |
113 scoped_refptr<CastEnvironment> cast_environment, | |
114 const VideoSenderConfig& video_config, | |
115 const StatusChangeCallback& status_change_cb, | |
116 const CreateVideoEncodeAcceleratorCallback& create_vea_cb, | |
117 const CreateVideoEncodeMemoryCallback& create_video_encode_mem_cb, | |
118 CastTransport* const transport_sender) | |
119 : VideoSender(cast_environment, | |
120 video_config, | |
121 status_change_cb, | |
122 create_vea_cb, | |
123 create_video_encode_mem_cb, | |
124 transport_sender, | |
125 base::Bind(&IgnorePlayoutDelayChanges)) {} | |
126 using VideoSender::OnReceivedCastFeedback; | |
127 using VideoSender::OnReceivedPli; | |
128 }; | |
129 | |
130 class TransportClient : public CastTransport::Client { | 110 class TransportClient : public CastTransport::Client { |
131 public: | 111 public: |
132 TransportClient() {} | 112 TransportClient() {} |
133 | 113 |
134 void OnStatusChanged(CastTransportStatus status) final { | 114 void OnStatusChanged(CastTransportStatus status) final { |
135 EXPECT_EQ(TRANSPORT_VIDEO_INITIALIZED, status); | 115 EXPECT_EQ(TRANSPORT_VIDEO_INITIALIZED, status); |
136 }; | 116 }; |
137 void OnLoggingEventsReceived( | 117 void OnLoggingEventsReceived( |
138 scoped_ptr<std::vector<FrameEvent>> frame_events, | 118 std::unique_ptr<std::vector<FrameEvent>> frame_events, |
139 scoped_ptr<std::vector<PacketEvent>> packet_events) final{}; | 119 std::unique_ptr<std::vector<PacketEvent>> packet_events) final{}; |
140 void ProcessRtpPacket(scoped_ptr<Packet> packet) final{}; | 120 void ProcessRtpPacket(std::unique_ptr<Packet> packet) final{}; |
141 | 121 |
142 DISALLOW_COPY_AND_ASSIGN(TransportClient); | 122 DISALLOW_COPY_AND_ASSIGN(TransportClient); |
143 }; | 123 }; |
144 | 124 |
145 } // namespace | 125 } // namespace |
146 | 126 |
147 class VideoSenderTest : public ::testing::Test { | 127 class VideoSenderTest : public ::testing::Test { |
148 protected: | 128 protected: |
149 VideoSenderTest() | 129 VideoSenderTest() |
150 : testing_clock_(new base::SimpleTestTickClock()), | 130 : testing_clock_(new base::SimpleTestTickClock()), |
151 task_runner_(new FakeSingleThreadTaskRunner(testing_clock_)), | 131 task_runner_(new FakeSingleThreadTaskRunner(testing_clock_)), |
152 cast_environment_( | 132 cast_environment_(new CastEnvironment( |
153 new CastEnvironment(scoped_ptr<base::TickClock>(testing_clock_), | 133 std::unique_ptr<base::TickClock>(testing_clock_), |
154 task_runner_, | 134 task_runner_, |
155 task_runner_, | 135 task_runner_, |
156 task_runner_)), | 136 task_runner_)), |
157 operational_status_(STATUS_UNINITIALIZED), | 137 operational_status_(STATUS_UNINITIALIZED), |
158 vea_factory_(task_runner_) { | 138 vea_factory_(task_runner_) { |
159 testing_clock_->Advance(base::TimeTicks::Now() - base::TimeTicks()); | 139 testing_clock_->Advance(base::TimeTicks::Now() - base::TimeTicks()); |
160 vea_factory_.SetAutoRespond(true); | 140 vea_factory_.SetAutoRespond(true); |
161 last_pixel_value_ = kPixelValue; | 141 last_pixel_value_ = kPixelValue; |
162 transport_ = new TestPacketSender(); | 142 transport_ = new TestPacketSender(); |
163 transport_sender_.reset( | 143 transport_sender_.reset( |
164 new CastTransportImpl(testing_clock_, base::TimeDelta(), | 144 new CastTransportImpl(testing_clock_, base::TimeDelta(), |
165 make_scoped_ptr(new TransportClient()), | 145 base::WrapUnique(new TransportClient()), |
166 make_scoped_ptr(transport_), task_runner_)); | 146 base::WrapUnique(transport_), task_runner_)); |
167 } | 147 } |
168 | 148 |
169 ~VideoSenderTest() override {} | 149 ~VideoSenderTest() override {} |
170 | 150 |
171 void TearDown() final { | 151 void TearDown() final { |
172 video_sender_.reset(); | 152 video_sender_.reset(); |
173 task_runner_->RunTasks(); | 153 task_runner_->RunTasks(); |
174 } | 154 } |
175 | 155 |
176 // If |external| is true then external video encoder (VEA) is used. | 156 // If |external| is true then external video encoder (VEA) is used. |
177 // |expect_init_success| is true if initialization is expected to succeed. | 157 // |expect_init_success| is true if initialization is expected to succeed. |
178 void InitEncoder(bool external, bool expect_init_success) { | 158 void InitEncoder(bool external, bool expect_init_success) { |
179 VideoSenderConfig video_config = GetDefaultVideoSenderConfig(); | 159 VideoSenderConfig video_config = GetDefaultVideoSenderConfig(); |
180 video_config.use_external_encoder = external; | 160 video_config.use_external_encoder = external; |
181 | 161 |
182 ASSERT_EQ(operational_status_, STATUS_UNINITIALIZED); | 162 ASSERT_EQ(operational_status_, STATUS_UNINITIALIZED); |
183 | 163 |
184 if (external) { | 164 if (external) { |
185 vea_factory_.SetInitializationWillSucceed(expect_init_success); | 165 vea_factory_.SetInitializationWillSucceed(expect_init_success); |
186 video_sender_.reset(new PeerVideoSender( | 166 video_sender_.reset(new VideoSender( |
187 cast_environment_, | 167 cast_environment_, video_config, |
188 video_config, | |
189 base::Bind(&SaveOperationalStatus, &operational_status_), | 168 base::Bind(&SaveOperationalStatus, &operational_status_), |
190 base::Bind( | 169 base::Bind( |
191 &FakeVideoEncodeAcceleratorFactory::CreateVideoEncodeAccelerator, | 170 &FakeVideoEncodeAcceleratorFactory::CreateVideoEncodeAccelerator, |
192 base::Unretained(&vea_factory_)), | 171 base::Unretained(&vea_factory_)), |
193 base::Bind(&FakeVideoEncodeAcceleratorFactory::CreateSharedMemory, | 172 base::Bind(&FakeVideoEncodeAcceleratorFactory::CreateSharedMemory, |
194 base::Unretained(&vea_factory_)), | 173 base::Unretained(&vea_factory_)), |
195 transport_sender_.get())); | 174 transport_sender_.get(), base::Bind(&IgnorePlayoutDelayChanges))); |
196 } else { | 175 } else { |
197 video_sender_.reset(new PeerVideoSender( | 176 video_sender_.reset(new VideoSender( |
198 cast_environment_, | 177 cast_environment_, video_config, |
199 video_config, | |
200 base::Bind(&SaveOperationalStatus, &operational_status_), | 178 base::Bind(&SaveOperationalStatus, &operational_status_), |
201 CreateDefaultVideoEncodeAcceleratorCallback(), | 179 CreateDefaultVideoEncodeAcceleratorCallback(), |
202 CreateDefaultVideoEncodeMemoryCallback(), | 180 CreateDefaultVideoEncodeMemoryCallback(), transport_sender_.get(), |
203 transport_sender_.get())); | 181 base::Bind(&IgnorePlayoutDelayChanges))); |
204 } | 182 } |
205 task_runner_->RunTasks(); | 183 task_runner_->RunTasks(); |
206 } | 184 } |
207 | 185 |
208 scoped_refptr<media::VideoFrame> GetNewVideoFrame() { | 186 scoped_refptr<media::VideoFrame> GetNewVideoFrame() { |
209 if (first_frame_timestamp_.is_null()) | 187 if (first_frame_timestamp_.is_null()) |
210 first_frame_timestamp_ = testing_clock_->NowTicks(); | 188 first_frame_timestamp_ = testing_clock_->NowTicks(); |
211 gfx::Size size(kWidth, kHeight); | 189 gfx::Size size(kWidth, kHeight); |
212 scoped_refptr<media::VideoFrame> video_frame = | 190 scoped_refptr<media::VideoFrame> video_frame = |
213 media::VideoFrame::CreateFrame( | 191 media::VideoFrame::CreateFrame( |
(...skipping 18 matching lines...) Expand all Loading... |
232 void RunTasks(int during_ms) { | 210 void RunTasks(int during_ms) { |
233 task_runner_->Sleep(base::TimeDelta::FromMilliseconds(during_ms)); | 211 task_runner_->Sleep(base::TimeDelta::FromMilliseconds(during_ms)); |
234 } | 212 } |
235 | 213 |
236 base::SimpleTestTickClock* const testing_clock_; // Owned by CastEnvironment. | 214 base::SimpleTestTickClock* const testing_clock_; // Owned by CastEnvironment. |
237 const scoped_refptr<FakeSingleThreadTaskRunner> task_runner_; | 215 const scoped_refptr<FakeSingleThreadTaskRunner> task_runner_; |
238 const scoped_refptr<CastEnvironment> cast_environment_; | 216 const scoped_refptr<CastEnvironment> cast_environment_; |
239 OperationalStatus operational_status_; | 217 OperationalStatus operational_status_; |
240 FakeVideoEncodeAcceleratorFactory vea_factory_; | 218 FakeVideoEncodeAcceleratorFactory vea_factory_; |
241 TestPacketSender* transport_; // Owned by CastTransport. | 219 TestPacketSender* transport_; // Owned by CastTransport. |
242 scoped_ptr<CastTransportImpl> transport_sender_; | 220 std::unique_ptr<CastTransportImpl> transport_sender_; |
243 scoped_ptr<PeerVideoSender> video_sender_; | 221 std::unique_ptr<VideoSender> video_sender_; |
244 int last_pixel_value_; | 222 int last_pixel_value_; |
245 base::TimeTicks first_frame_timestamp_; | 223 base::TimeTicks first_frame_timestamp_; |
246 | 224 |
247 private: | 225 private: |
248 DISALLOW_COPY_AND_ASSIGN(VideoSenderTest); | 226 DISALLOW_COPY_AND_ASSIGN(VideoSenderTest); |
249 }; | 227 }; |
250 | 228 |
251 TEST_F(VideoSenderTest, BuiltInEncoder) { | 229 TEST_F(VideoSenderTest, BuiltInEncoder) { |
252 InitEncoder(false, true); | 230 InitEncoder(false, true); |
253 ASSERT_EQ(STATUS_INITIALIZED, operational_status_); | 231 ASSERT_EQ(STATUS_INITIALIZED, operational_status_); |
(...skipping 380 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
634 video_frame, | 612 video_frame, |
635 testing_clock_->NowTicks() + base::TimeDelta::FromMilliseconds(1000)); | 613 testing_clock_->NowTicks() + base::TimeDelta::FromMilliseconds(1000)); |
636 RunTasks(33); | 614 RunTasks(33); |
637 transport_->SetPause(false); | 615 transport_->SetPause(false); |
638 RunTasks(33); | 616 RunTasks(33); |
639 EXPECT_EQ(2, transport_->number_of_rtp_packets()); | 617 EXPECT_EQ(2, transport_->number_of_rtp_packets()); |
640 } | 618 } |
641 | 619 |
642 } // namespace cast | 620 } // namespace cast |
643 } // namespace media | 621 } // namespace media |
OLD | NEW |