| 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" |
| (...skipping 95 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 106 PopulateVideoFrame(video_frame_, 123); | 106 PopulateVideoFrame(video_frame_, 123); |
| 107 | 107 |
| 108 testing_clock_ = new base::SimpleTestTickClock(); | 108 testing_clock_ = new base::SimpleTestTickClock(); |
| 109 task_runner_ = new test::FakeSingleThreadTaskRunner(testing_clock_); | 109 task_runner_ = new test::FakeSingleThreadTaskRunner(testing_clock_); |
| 110 cast_environment_ = | 110 cast_environment_ = |
| 111 new CastEnvironment(scoped_ptr<base::TickClock>(testing_clock_).Pass(), | 111 new CastEnvironment(scoped_ptr<base::TickClock>(testing_clock_).Pass(), |
| 112 task_runner_, | 112 task_runner_, |
| 113 task_runner_, | 113 task_runner_, |
| 114 task_runner_); | 114 task_runner_); |
| 115 | 115 |
| 116 fake_vea_ = new test::FakeVideoEncodeAccelerator(task_runner_); | 116 fake_vea_ = new test::FakeVideoEncodeAccelerator(task_runner_, |
| 117 &stored_bitrates_); |
| 117 scoped_ptr<VideoEncodeAccelerator> fake_vea(fake_vea_); | 118 scoped_ptr<VideoEncodeAccelerator> fake_vea(fake_vea_); |
| 118 video_encoder_.reset( | 119 video_encoder_.reset( |
| 119 new ExternalVideoEncoder(cast_environment_, | 120 new ExternalVideoEncoder(cast_environment_, |
| 120 video_config_, | 121 video_config_, |
| 121 base::Bind(&CreateVideoEncodeAccelerator, | 122 base::Bind(&CreateVideoEncodeAccelerator, |
| 122 task_runner_, | 123 task_runner_, |
| 123 base::Passed(&fake_vea)), | 124 base::Passed(&fake_vea)), |
| 124 base::Bind(&CreateSharedMemory))); | 125 base::Bind(&CreateSharedMemory))); |
| 125 } | 126 } |
| 126 | 127 |
| 127 virtual ~ExternalVideoEncoderTest() {} | 128 virtual ~ExternalVideoEncoderTest() {} |
| 128 | 129 |
| 129 base::SimpleTestTickClock* testing_clock_; // Owned by CastEnvironment. | 130 base::SimpleTestTickClock* testing_clock_; // Owned by CastEnvironment. |
| 130 test::FakeVideoEncodeAccelerator* fake_vea_; // Owned by video_encoder_. | 131 test::FakeVideoEncodeAccelerator* fake_vea_; // Owned by video_encoder_. |
| 132 std::vector<uint32> stored_bitrates_; |
| 131 scoped_refptr<TestVideoEncoderCallback> test_video_encoder_callback_; | 133 scoped_refptr<TestVideoEncoderCallback> test_video_encoder_callback_; |
| 132 VideoSenderConfig video_config_; | 134 VideoSenderConfig video_config_; |
| 133 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_; | 135 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_; |
| 134 scoped_ptr<VideoEncoder> video_encoder_; | 136 scoped_ptr<VideoEncoder> video_encoder_; |
| 135 scoped_refptr<media::VideoFrame> video_frame_; | 137 scoped_refptr<media::VideoFrame> video_frame_; |
| 136 scoped_refptr<CastEnvironment> cast_environment_; | 138 scoped_refptr<CastEnvironment> cast_environment_; |
| 137 | 139 |
| 138 DISALLOW_COPY_AND_ASSIGN(ExternalVideoEncoderTest); | 140 DISALLOW_COPY_AND_ASSIGN(ExternalVideoEncoderTest); |
| 139 }; | 141 }; |
| 140 | 142 |
| 141 TEST_F(ExternalVideoEncoderTest, EncodePattern30fpsRunningOutOfAck) { | 143 TEST_F(ExternalVideoEncoderTest, EncodePattern30fpsRunningOutOfAck) { |
| 142 task_runner_->RunTasks(); // Run the initializer on the correct thread. | 144 task_runner_->RunTasks(); // Run the initializer on the correct thread. |
| 143 | 145 |
| 144 VideoEncoder::FrameEncodedCallback frame_encoded_callback = | 146 VideoEncoder::FrameEncodedCallback frame_encoded_callback = |
| 145 base::Bind(&TestVideoEncoderCallback::DeliverEncodedVideoFrame, | 147 base::Bind(&TestVideoEncoderCallback::DeliverEncodedVideoFrame, |
| 146 test_video_encoder_callback_.get()); | 148 test_video_encoder_callback_.get()); |
| 147 | 149 |
| 148 base::TimeTicks capture_time; | 150 base::TimeTicks capture_time; |
| 149 capture_time += base::TimeDelta::FromMilliseconds(33); | 151 capture_time += base::TimeDelta::FromMilliseconds(33); |
| 150 test_video_encoder_callback_->SetExpectedResult(0, 0, capture_time); | 152 test_video_encoder_callback_->SetExpectedResult(0, 0, capture_time); |
| 153 video_encoder_->SetBitRate(2000); |
| 151 EXPECT_TRUE(video_encoder_->EncodeVideoFrame( | 154 EXPECT_TRUE(video_encoder_->EncodeVideoFrame( |
| 152 video_frame_, capture_time, frame_encoded_callback)); | 155 video_frame_, capture_time, frame_encoded_callback)); |
| 153 task_runner_->RunTasks(); | 156 task_runner_->RunTasks(); |
| 154 | 157 |
| 155 for (int i = 0; i < 6; ++i) { | 158 for (int i = 0; i < 6; ++i) { |
| 156 capture_time += base::TimeDelta::FromMilliseconds(33); | 159 capture_time += base::TimeDelta::FromMilliseconds(33); |
| 157 test_video_encoder_callback_->SetExpectedResult(i + 1, i, capture_time); | 160 test_video_encoder_callback_->SetExpectedResult(i + 1, i, capture_time); |
| 158 EXPECT_TRUE(video_encoder_->EncodeVideoFrame( | 161 EXPECT_TRUE(video_encoder_->EncodeVideoFrame( |
| 159 video_frame_, capture_time, frame_encoded_callback)); | 162 video_frame_, capture_time, frame_encoded_callback)); |
| 160 task_runner_->RunTasks(); | 163 task_runner_->RunTasks(); |
| 161 } | 164 } |
| 162 // We need to run the task to cleanup the GPU instance. | 165 // We need to run the task to cleanup the GPU instance. |
| 163 video_encoder_.reset(NULL); | 166 video_encoder_.reset(NULL); |
| 164 task_runner_->RunTasks(); | 167 task_runner_->RunTasks(); |
| 168 |
| 169 ASSERT_EQ(1u, stored_bitrates_.size()); |
| 170 EXPECT_EQ(2000u, stored_bitrates_[0]); |
| 165 } | 171 } |
| 166 | 172 |
| 167 TEST_F(ExternalVideoEncoderTest, StreamHeader) { | 173 TEST_F(ExternalVideoEncoderTest, StreamHeader) { |
| 168 task_runner_->RunTasks(); // Run the initializer on the correct thread. | 174 task_runner_->RunTasks(); // Run the initializer on the correct thread. |
| 169 | 175 |
| 170 VideoEncoder::FrameEncodedCallback frame_encoded_callback = | 176 VideoEncoder::FrameEncodedCallback frame_encoded_callback = |
| 171 base::Bind(&TestVideoEncoderCallback::DeliverEncodedVideoFrame, | 177 base::Bind(&TestVideoEncoderCallback::DeliverEncodedVideoFrame, |
| 172 test_video_encoder_callback_.get()); | 178 test_video_encoder_callback_.get()); |
| 173 | 179 |
| 174 // Force the FakeVideoEncodeAccelerator to return a dummy non-key frame first. | 180 // Force the FakeVideoEncodeAccelerator to return a dummy non-key frame first. |
| 175 fake_vea_->SendDummyFrameForTesting(false); | 181 fake_vea_->SendDummyFrameForTesting(false); |
| 176 | 182 |
| 177 // Verify the first returned bitstream buffer is still a key frame. | 183 // Verify the first returned bitstream buffer is still a key frame. |
| 178 base::TimeTicks capture_time; | 184 base::TimeTicks capture_time; |
| 179 capture_time += base::TimeDelta::FromMilliseconds(33); | 185 capture_time += base::TimeDelta::FromMilliseconds(33); |
| 180 test_video_encoder_callback_->SetExpectedResult(0, 0, capture_time); | 186 test_video_encoder_callback_->SetExpectedResult(0, 0, capture_time); |
| 181 EXPECT_TRUE(video_encoder_->EncodeVideoFrame( | 187 EXPECT_TRUE(video_encoder_->EncodeVideoFrame( |
| 182 video_frame_, capture_time, frame_encoded_callback)); | 188 video_frame_, capture_time, frame_encoded_callback)); |
| 183 task_runner_->RunTasks(); | 189 task_runner_->RunTasks(); |
| 184 | 190 |
| 185 // We need to run the task to cleanup the GPU instance. | 191 // We need to run the task to cleanup the GPU instance. |
| 186 video_encoder_.reset(NULL); | 192 video_encoder_.reset(NULL); |
| 187 task_runner_->RunTasks(); | 193 task_runner_->RunTasks(); |
| 188 } | 194 } |
| 189 | 195 |
| 190 } // namespace cast | 196 } // namespace cast |
| 191 } // namespace media | 197 } // namespace media |
| OLD | NEW |