| 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 <vector> | 5 #include <vector> |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/memory/ref_counted.h" | 8 #include "base/memory/ref_counted.h" |
| 9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "media/base/video_frame.h" | 10 #include "media/base/video_frame.h" |
| 11 #include "media/cast/cast_defines.h" | 11 #include "media/cast/cast_defines.h" |
| 12 #include "media/cast/cast_environment.h" | 12 #include "media/cast/cast_environment.h" |
| 13 #include "media/cast/sender/external_video_encoder.h" | 13 #include "media/cast/sender/external_video_encoder.h" |
| 14 #include "media/cast/test/fake_single_thread_task_runner.h" | 14 #include "media/cast/test/fake_single_thread_task_runner.h" |
| 15 #include "media/cast/test/fake_video_encode_accelerator.h" | |
| 16 #include "media/cast/test/utility/video_utility.h" | 15 #include "media/cast/test/utility/video_utility.h" |
| 16 #include "media/video/fake_video_encode_accelerator.h" |
| 17 #include "testing/gmock/include/gmock/gmock.h" | 17 #include "testing/gmock/include/gmock/gmock.h" |
| 18 | 18 |
| 19 namespace media { | 19 namespace media { |
| 20 namespace cast { | 20 namespace cast { |
| 21 | 21 |
| 22 using testing::_; | 22 using testing::_; |
| 23 | 23 |
| 24 namespace { | 24 namespace { |
| 25 | 25 |
| 26 void IgnoreInitializationStatus(CastInitializationStatus status) {} | 26 void IgnoreInitializationStatus(CastInitializationStatus status) {} |
| (...skipping 100 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 127 | 127 |
| 128 testing_clock_ = new base::SimpleTestTickClock(); | 128 testing_clock_ = new base::SimpleTestTickClock(); |
| 129 testing_clock_->Advance(base::TimeTicks::Now() - base::TimeTicks()); | 129 testing_clock_->Advance(base::TimeTicks::Now() - base::TimeTicks()); |
| 130 task_runner_ = new test::FakeSingleThreadTaskRunner(testing_clock_); | 130 task_runner_ = new test::FakeSingleThreadTaskRunner(testing_clock_); |
| 131 cast_environment_ = | 131 cast_environment_ = |
| 132 new CastEnvironment(scoped_ptr<base::TickClock>(testing_clock_).Pass(), | 132 new CastEnvironment(scoped_ptr<base::TickClock>(testing_clock_).Pass(), |
| 133 task_runner_, | 133 task_runner_, |
| 134 task_runner_, | 134 task_runner_, |
| 135 task_runner_); | 135 task_runner_); |
| 136 | 136 |
| 137 fake_vea_ = new test::FakeVideoEncodeAccelerator(task_runner_, | 137 fake_vea_ = new media::FakeVideoEncodeAccelerator(task_runner_); |
| 138 &stored_bitrates_); | |
| 139 scoped_ptr<VideoEncodeAccelerator> fake_vea(fake_vea_); | 138 scoped_ptr<VideoEncodeAccelerator> fake_vea(fake_vea_); |
| 140 VEAFactory vea_factory(task_runner_, fake_vea.Pass()); | 139 VEAFactory vea_factory(task_runner_, fake_vea.Pass()); |
| 141 video_encoder_.reset(new ExternalVideoEncoder( | 140 video_encoder_.reset(new ExternalVideoEncoder( |
| 142 cast_environment_, | 141 cast_environment_, |
| 143 video_config_, | 142 video_config_, |
| 144 base::Bind(&IgnoreInitializationStatus), | 143 base::Bind(&IgnoreInitializationStatus), |
| 145 base::Bind(&VEAFactory::CreateVideoEncodeAccelerator, | 144 base::Bind(&VEAFactory::CreateVideoEncodeAccelerator, |
| 146 base::Unretained(&vea_factory)), | 145 base::Unretained(&vea_factory)), |
| 147 base::Bind(&CreateSharedMemory))); | 146 base::Bind(&CreateSharedMemory))); |
| 148 vea_factory.FinishCreatingVideoEncodeAccelerator(); | 147 vea_factory.FinishCreatingVideoEncodeAccelerator(); |
| 149 } | 148 } |
| 150 | 149 |
| 151 ~ExternalVideoEncoderTest() override {} | 150 ~ExternalVideoEncoderTest() override {} |
| 152 | 151 |
| 153 void AdvanceClockAndVideoFrameTimestamp() { | 152 void AdvanceClockAndVideoFrameTimestamp() { |
| 154 testing_clock_->Advance(base::TimeDelta::FromMilliseconds(33)); | 153 testing_clock_->Advance(base::TimeDelta::FromMilliseconds(33)); |
| 155 video_frame_->set_timestamp( | 154 video_frame_->set_timestamp( |
| 156 video_frame_->timestamp() + base::TimeDelta::FromMilliseconds(33)); | 155 video_frame_->timestamp() + base::TimeDelta::FromMilliseconds(33)); |
| 157 } | 156 } |
| 158 | 157 |
| 159 base::SimpleTestTickClock* testing_clock_; // Owned by CastEnvironment. | 158 base::SimpleTestTickClock* testing_clock_; // Owned by CastEnvironment. |
| 160 test::FakeVideoEncodeAccelerator* fake_vea_; // Owned by video_encoder_. | 159 media::FakeVideoEncodeAccelerator* fake_vea_; // Owned by video_encoder_. |
| 161 std::vector<uint32> stored_bitrates_; | |
| 162 scoped_refptr<TestVideoEncoderCallback> test_video_encoder_callback_; | 160 scoped_refptr<TestVideoEncoderCallback> test_video_encoder_callback_; |
| 163 VideoSenderConfig video_config_; | 161 VideoSenderConfig video_config_; |
| 164 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_; | 162 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_; |
| 165 scoped_ptr<VideoEncoder> video_encoder_; | 163 scoped_ptr<VideoEncoder> video_encoder_; |
| 166 scoped_refptr<media::VideoFrame> video_frame_; | 164 scoped_refptr<media::VideoFrame> video_frame_; |
| 167 scoped_refptr<CastEnvironment> cast_environment_; | 165 scoped_refptr<CastEnvironment> cast_environment_; |
| 168 | 166 |
| 169 DISALLOW_COPY_AND_ASSIGN(ExternalVideoEncoderTest); | 167 DISALLOW_COPY_AND_ASSIGN(ExternalVideoEncoderTest); |
| 170 }; | 168 }; |
| 171 | 169 |
| (...skipping 20 matching lines...) Expand all Loading... |
| 192 TimeDeltaToRtpDelta(video_frame_->timestamp(), kVideoFrequency), | 190 TimeDeltaToRtpDelta(video_frame_->timestamp(), kVideoFrequency), |
| 193 testing_clock_->NowTicks()); | 191 testing_clock_->NowTicks()); |
| 194 EXPECT_TRUE(video_encoder_->EncodeVideoFrame( | 192 EXPECT_TRUE(video_encoder_->EncodeVideoFrame( |
| 195 video_frame_, testing_clock_->NowTicks(), frame_encoded_callback)); | 193 video_frame_, testing_clock_->NowTicks(), frame_encoded_callback)); |
| 196 task_runner_->RunTasks(); | 194 task_runner_->RunTasks(); |
| 197 } | 195 } |
| 198 // We need to run the task to cleanup the GPU instance. | 196 // We need to run the task to cleanup the GPU instance. |
| 199 video_encoder_.reset(NULL); | 197 video_encoder_.reset(NULL); |
| 200 task_runner_->RunTasks(); | 198 task_runner_->RunTasks(); |
| 201 | 199 |
| 202 ASSERT_EQ(1u, stored_bitrates_.size()); | 200 ASSERT_EQ(1u, fake_vea_->stored_bitrates().size()); |
| 203 EXPECT_EQ(2000u, stored_bitrates_[0]); | 201 EXPECT_EQ(2000u, fake_vea_->stored_bitrates()[0]); |
| 204 } | 202 } |
| 205 | 203 |
| 206 TEST_F(ExternalVideoEncoderTest, StreamHeader) { | 204 TEST_F(ExternalVideoEncoderTest, StreamHeader) { |
| 207 task_runner_->RunTasks(); // Run the initializer on the correct thread. | 205 task_runner_->RunTasks(); // Run the initializer on the correct thread. |
| 208 | 206 |
| 209 VideoEncoder::FrameEncodedCallback frame_encoded_callback = | 207 VideoEncoder::FrameEncodedCallback frame_encoded_callback = |
| 210 base::Bind(&TestVideoEncoderCallback::DeliverEncodedVideoFrame, | 208 base::Bind(&TestVideoEncoderCallback::DeliverEncodedVideoFrame, |
| 211 test_video_encoder_callback_.get()); | 209 test_video_encoder_callback_.get()); |
| 212 | 210 |
| 213 // Force the FakeVideoEncodeAccelerator to return a dummy non-key frame first. | 211 // Force the FakeVideoEncodeAccelerator to return a dummy non-key frame first. |
| (...skipping 18 matching lines...) Expand all Loading... |
| 232 VideoSenderConfig video_config; | 230 VideoSenderConfig video_config; |
| 233 base::SimpleTestTickClock* testing_clock = new base::SimpleTestTickClock(); | 231 base::SimpleTestTickClock* testing_clock = new base::SimpleTestTickClock(); |
| 234 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner( | 232 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner( |
| 235 new test::FakeSingleThreadTaskRunner(testing_clock)); | 233 new test::FakeSingleThreadTaskRunner(testing_clock)); |
| 236 scoped_refptr<CastEnvironment> cast_environment( | 234 scoped_refptr<CastEnvironment> cast_environment( |
| 237 new CastEnvironment(scoped_ptr<base::TickClock>(testing_clock).Pass(), | 235 new CastEnvironment(scoped_ptr<base::TickClock>(testing_clock).Pass(), |
| 238 task_runner, | 236 task_runner, |
| 239 task_runner, | 237 task_runner, |
| 240 task_runner)); | 238 task_runner)); |
| 241 | 239 |
| 242 std::vector<uint32> stored_bitrates; | |
| 243 scoped_ptr<VideoEncodeAccelerator> fake_vea( | 240 scoped_ptr<VideoEncodeAccelerator> fake_vea( |
| 244 new test::FakeVideoEncodeAccelerator(task_runner, &stored_bitrates)); | 241 new media::FakeVideoEncodeAccelerator(task_runner)); |
| 245 VEAFactory vea_factory(task_runner, fake_vea.Pass()); | 242 VEAFactory vea_factory(task_runner, fake_vea.Pass()); |
| 246 | 243 |
| 247 scoped_ptr<ExternalVideoEncoder> video_encoder(new ExternalVideoEncoder( | 244 scoped_ptr<ExternalVideoEncoder> video_encoder(new ExternalVideoEncoder( |
| 248 cast_environment, | 245 cast_environment, |
| 249 video_config, | 246 video_config, |
| 250 base::Bind(&IgnoreInitializationStatus), | 247 base::Bind(&IgnoreInitializationStatus), |
| 251 base::Bind(&VEAFactory::CreateVideoEncodeAccelerator, | 248 base::Bind(&VEAFactory::CreateVideoEncodeAccelerator, |
| 252 base::Unretained(&vea_factory)), | 249 base::Unretained(&vea_factory)), |
| 253 base::Bind(&CreateSharedMemory))); | 250 base::Bind(&CreateSharedMemory))); |
| 254 | 251 |
| 255 video_encoder.reset(); | 252 video_encoder.reset(); |
| 256 vea_factory.FinishCreatingVideoEncodeAccelerator(); | 253 vea_factory.FinishCreatingVideoEncodeAccelerator(); |
| 257 task_runner->RunTasks(); | 254 task_runner->RunTasks(); |
| 258 } | 255 } |
| 259 | 256 |
| 260 } // namespace cast | 257 } // namespace cast |
| 261 } // namespace media | 258 } // namespace media |
| OLD | NEW |