| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "remoting/test/test_video_renderer.h" | 5 #include "remoting/test/test_video_renderer.h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 | 8 |
| 9 #include <cmath> | 9 #include <cmath> |
| 10 #include <utility> | 10 #include <utility> |
| (...skipping 122 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 133 int screen_height, | 133 int screen_height, |
| 134 double error_limit) { | 134 double error_limit) { |
| 135 DCHECK(encoder_); | 135 DCHECK(encoder_); |
| 136 DCHECK(test_video_renderer_); | 136 DCHECK(test_video_renderer_); |
| 137 | 137 |
| 138 // Generate a frame containing a gradient. | 138 // Generate a frame containing a gradient. |
| 139 std::unique_ptr<webrtc::DesktopFrame> original_frame = | 139 std::unique_ptr<webrtc::DesktopFrame> original_frame = |
| 140 CreateDesktopFrameWithGradient(screen_width, screen_height); | 140 CreateDesktopFrameWithGradient(screen_width, screen_height); |
| 141 EXPECT_TRUE(original_frame); | 141 EXPECT_TRUE(original_frame); |
| 142 | 142 |
| 143 std::unique_ptr<VideoPacket> packet = encoder_->Encode(*original_frame.get()); | 143 std::unique_ptr<VideoPacket> packet = |
| 144 encoder_->Encode(*original_frame.get(), 0); |
| 144 | 145 |
| 145 DCHECK(!run_loop_ || !run_loop_->running()); | 146 DCHECK(!run_loop_ || !run_loop_->running()); |
| 146 DCHECK(!timer_->IsRunning()); | 147 DCHECK(!timer_->IsRunning()); |
| 147 run_loop_.reset(new base::RunLoop()); | 148 run_loop_.reset(new base::RunLoop()); |
| 148 | 149 |
| 149 // Set an extremely long time: 10 min to prevent bugs from hanging the system. | 150 // Set an extremely long time: 10 min to prevent bugs from hanging the system. |
| 150 // NOTE: We've seen cases which take up to 1 min to process a packet, so an | 151 // NOTE: We've seen cases which take up to 1 min to process a packet, so an |
| 151 // extremely long time as 10 min is chosen to avoid being variable/flaky. | 152 // extremely long time as 10 min is chosen to avoid being variable/flaky. |
| 152 timer_->Start(FROM_HERE, base::TimeDelta::FromMinutes(10), | 153 timer_->Start(FROM_HERE, base::TimeDelta::FromMinutes(10), |
| 153 run_loop_->QuitClosure()); | 154 run_loop_->QuitClosure()); |
| (...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 225 int screen_height, | 226 int screen_height, |
| 226 const webrtc::DesktopRect& expected_rect, | 227 const webrtc::DesktopRect& expected_rect, |
| 227 bool expect_to_match) { | 228 bool expect_to_match) { |
| 228 DCHECK(encoder_); | 229 DCHECK(encoder_); |
| 229 DCHECK(test_video_renderer_); | 230 DCHECK(test_video_renderer_); |
| 230 | 231 |
| 231 std::unique_ptr<webrtc::DesktopFrame> frame = | 232 std::unique_ptr<webrtc::DesktopFrame> frame = |
| 232 CreateDesktopFrameWithGradient(screen_width, screen_height); | 233 CreateDesktopFrameWithGradient(screen_width, screen_height); |
| 233 RGBValue expected_average_color = | 234 RGBValue expected_average_color = |
| 234 CalculateAverageColorValueForFrame(frame.get(), expected_rect); | 235 CalculateAverageColorValueForFrame(frame.get(), expected_rect); |
| 235 std::unique_ptr<VideoPacket> packet = encoder_->Encode(*frame.get()); | 236 std::unique_ptr<VideoPacket> packet = encoder_->Encode(*frame.get(), 0); |
| 236 | 237 |
| 237 if (expect_to_match) { | 238 if (expect_to_match) { |
| 238 EXPECT_TRUE(SendPacketAndWaitForMatch(std::move(packet), expected_rect, | 239 EXPECT_TRUE(SendPacketAndWaitForMatch(std::move(packet), expected_rect, |
| 239 expected_average_color)); | 240 expected_average_color)); |
| 240 } else { | 241 } else { |
| 241 // Shift each channel by 128. | 242 // Shift each channel by 128. |
| 242 // e.g. (10, 127, 200) -> (138, 255, 73). | 243 // e.g. (10, 127, 200) -> (138, 255, 73). |
| 243 // In this way, the error between expected color and true value is always | 244 // In this way, the error between expected color and true value is always |
| 244 // around 0.5. | 245 // around 0.5. |
| 245 int red_shift = (expected_average_color.red + 128) % 255; | 246 int red_shift = (expected_average_color.red + 128) % 255; |
| (...skipping 151 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 397 // Post multiple tasks to |test_video_renderer_|, and it should not crash. | 398 // Post multiple tasks to |test_video_renderer_|, and it should not crash. |
| 398 // 20 is chosen because it's large enough to make sure that there will be | 399 // 20 is chosen because it's large enough to make sure that there will be |
| 399 // more than one task on the video decode thread, while not too large to wait | 400 // more than one task on the video decode thread, while not too large to wait |
| 400 // for too long for the unit test to complete. | 401 // for too long for the unit test to complete. |
| 401 const int task_num = 20; | 402 const int task_num = 20; |
| 402 ScopedVector<VideoPacket> video_packets; | 403 ScopedVector<VideoPacket> video_packets; |
| 403 for (int i = 0; i < task_num; ++i) { | 404 for (int i = 0; i < task_num; ++i) { |
| 404 std::unique_ptr<webrtc::DesktopFrame> original_frame = | 405 std::unique_ptr<webrtc::DesktopFrame> original_frame = |
| 405 CreateDesktopFrameWithGradient(kDefaultScreenWidthPx, | 406 CreateDesktopFrameWithGradient(kDefaultScreenWidthPx, |
| 406 kDefaultScreenHeightPx); | 407 kDefaultScreenHeightPx); |
| 407 video_packets.push_back(encoder_->Encode(*original_frame.get())); | 408 video_packets.push_back(encoder_->Encode(*original_frame.get(), 0)); |
| 408 } | 409 } |
| 409 | 410 |
| 410 for (int i = 0; i < task_num; ++i) { | 411 for (int i = 0; i < task_num; ++i) { |
| 411 // Transfer ownership of video packet. | 412 // Transfer ownership of video packet. |
| 412 VideoPacket* packet = video_packets[i]; | 413 VideoPacket* packet = video_packets[i]; |
| 413 video_packets[i] = nullptr; | 414 video_packets[i] = nullptr; |
| 414 test_video_renderer_->ProcessVideoPacket(base::WrapUnique(packet), | 415 test_video_renderer_->ProcessVideoPacket(base::WrapUnique(packet), |
| 415 base::Bind(&base::DoNothing)); | 416 base::Bind(&base::DoNothing)); |
| 416 } | 417 } |
| 417 } | 418 } |
| (...skipping 28 matching lines...) Expand all Loading... |
| 446 | 447 |
| 447 // Since we don't care whether expected image pattern is matched or not in | 448 // Since we don't care whether expected image pattern is matched or not in |
| 448 // this case, an expected color is chosen arbitrarily. | 449 // this case, an expected color is chosen arbitrarily. |
| 449 RGBValue black_color = RGBValue(); | 450 RGBValue black_color = RGBValue(); |
| 450 | 451 |
| 451 // Set expected image pattern. | 452 // Set expected image pattern. |
| 452 test_video_renderer_->ExpectAverageColorInRect( | 453 test_video_renderer_->ExpectAverageColorInRect( |
| 453 kDefaultExpectedRect, black_color, base::Bind(&base::DoNothing)); | 454 kDefaultExpectedRect, black_color, base::Bind(&base::DoNothing)); |
| 454 | 455 |
| 455 // Post test video packet. | 456 // Post test video packet. |
| 456 test_video_renderer_->ProcessVideoPacket(encoder_->Encode(*frame.get()), | 457 test_video_renderer_->ProcessVideoPacket(encoder_->Encode(*frame.get(), 0), |
| 457 base::Bind(&base::DoNothing)); | 458 base::Bind(&base::DoNothing)); |
| 458 } | 459 } |
| 459 | 460 |
| 460 // Verify correct image pattern can be matched for VP8. | 461 // Verify correct image pattern can be matched for VP8. |
| 461 TEST_F(TestVideoRendererTest, VerifyImagePatternMatchForVP8) { | 462 TEST_F(TestVideoRendererTest, VerifyImagePatternMatchForVP8) { |
| 462 encoder_ = VideoEncoderVpx::CreateForVP8(); | 463 encoder_ = VideoEncoderVpx::CreateForVP8(); |
| 463 test_video_renderer_->SetCodecForDecoding( | 464 test_video_renderer_->SetCodecForDecoding( |
| 464 protocol::ChannelConfig::Codec::CODEC_VP8); | 465 protocol::ChannelConfig::Codec::CODEC_VP8); |
| 465 TestImagePatternMatch(kDefaultScreenWidthPx, kDefaultScreenHeightPx, | 466 TestImagePatternMatch(kDefaultScreenWidthPx, kDefaultScreenHeightPx, |
| 466 kDefaultExpectedRect, true); | 467 kDefaultExpectedRect, true); |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 506 TEST_F(TestVideoRendererTest, VerifyImagePatternNotMatchForVERBATIM) { | 507 TEST_F(TestVideoRendererTest, VerifyImagePatternNotMatchForVERBATIM) { |
| 507 encoder_.reset(new VideoEncoderVerbatim()); | 508 encoder_.reset(new VideoEncoderVerbatim()); |
| 508 test_video_renderer_->SetCodecForDecoding( | 509 test_video_renderer_->SetCodecForDecoding( |
| 509 protocol::ChannelConfig::Codec::CODEC_VERBATIM); | 510 protocol::ChannelConfig::Codec::CODEC_VERBATIM); |
| 510 TestImagePatternMatch(kDefaultScreenWidthPx, kDefaultScreenHeightPx, | 511 TestImagePatternMatch(kDefaultScreenWidthPx, kDefaultScreenHeightPx, |
| 511 kDefaultExpectedRect, false); | 512 kDefaultExpectedRect, false); |
| 512 } | 513 } |
| 513 | 514 |
| 514 } // namespace test | 515 } // namespace test |
| 515 } // namespace remoting | 516 } // namespace remoting |
| OLD | NEW |