| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 <inttypes.h> | 5 #include <inttypes.h> |
| 6 #include <stddef.h> | 6 #include <stddef.h> |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 | 8 |
| 9 #include <algorithm> | 9 #include <algorithm> |
| 10 #include <memory> | 10 #include <memory> |
| (...skipping 28 matching lines...) Expand all Loading... |
| 39 #include "media/base/cdm_context.h" | 39 #include "media/base/cdm_context.h" |
| 40 #include "media/base/decoder_buffer.h" | 40 #include "media/base/decoder_buffer.h" |
| 41 #include "media/base/media_util.h" | 41 #include "media/base/media_util.h" |
| 42 #include "media/base/test_data_util.h" | 42 #include "media/base/test_data_util.h" |
| 43 #include "media/base/video_decoder.h" | 43 #include "media/base/video_decoder.h" |
| 44 #include "media/base/video_frame.h" | 44 #include "media/base/video_frame.h" |
| 45 #include "media/filters/ffmpeg_glue.h" | 45 #include "media/filters/ffmpeg_glue.h" |
| 46 #include "media/filters/ffmpeg_video_decoder.h" | 46 #include "media/filters/ffmpeg_video_decoder.h" |
| 47 #include "media/filters/h264_parser.h" | 47 #include "media/filters/h264_parser.h" |
| 48 #include "media/filters/ivf_parser.h" | 48 #include "media/filters/ivf_parser.h" |
| 49 #include "media/gpu/h264_decoder.h" | |
| 50 #include "media/gpu/h264_dpb.h" | |
| 51 #include "media/gpu/video_accelerator_unittest_helpers.h" | 49 #include "media/gpu/video_accelerator_unittest_helpers.h" |
| 52 #include "media/video/fake_video_encode_accelerator.h" | 50 #include "media/video/fake_video_encode_accelerator.h" |
| 53 #include "media/video/video_encode_accelerator.h" | 51 #include "media/video/video_encode_accelerator.h" |
| 54 #include "testing/gtest/include/gtest/gtest.h" | 52 #include "testing/gtest/include/gtest/gtest.h" |
| 55 | 53 |
| 56 #if defined(OS_CHROMEOS) | 54 #if defined(OS_CHROMEOS) |
| 57 #if defined(USE_V4L2_CODEC) | 55 #if defined(USE_V4L2_CODEC) |
| 58 #include "base/threading/thread_task_runner_handle.h" | 56 #include "base/threading/thread_task_runner_handle.h" |
| 59 #include "media/gpu/v4l2_video_encode_accelerator.h" | 57 #include "media/gpu/v4l2_video_encode_accelerator.h" |
| 60 #endif | 58 #endif |
| (...skipping 539 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 600 public: | 598 public: |
| 601 explicit H264Validator(const FrameFoundCallback& frame_cb) | 599 explicit H264Validator(const FrameFoundCallback& frame_cb) |
| 602 : StreamValidator(frame_cb), | 600 : StreamValidator(frame_cb), |
| 603 seen_sps_(false), | 601 seen_sps_(false), |
| 604 seen_pps_(false), | 602 seen_pps_(false), |
| 605 seen_idr_(false) {} | 603 seen_idr_(false) {} |
| 606 | 604 |
| 607 void ProcessStreamBuffer(const uint8_t* stream, size_t size) override; | 605 void ProcessStreamBuffer(const uint8_t* stream, size_t size) override; |
| 608 | 606 |
| 609 private: | 607 private: |
| 610 bool IsNewPicture(const H264SliceHeader& slice_hdr); | |
| 611 bool UpdateCurrentPicture(const H264SliceHeader& slice_hdr); | |
| 612 | |
| 613 // Set to true when encoder provides us with the corresponding NALU type. | 608 // Set to true when encoder provides us with the corresponding NALU type. |
| 614 bool seen_sps_; | 609 bool seen_sps_; |
| 615 bool seen_pps_; | 610 bool seen_pps_; |
| 616 bool seen_idr_; | 611 bool seen_idr_; |
| 617 | 612 |
| 618 scoped_refptr<H264Picture> curr_pic_; | |
| 619 int curr_sps_id_ = -1; | |
| 620 int curr_pps_id_ = -1; | |
| 621 | |
| 622 H264Parser h264_parser_; | 613 H264Parser h264_parser_; |
| 623 }; | 614 }; |
| 624 | 615 |
| 625 void H264Validator::ProcessStreamBuffer(const uint8_t* stream, size_t size) { | 616 void H264Validator::ProcessStreamBuffer(const uint8_t* stream, size_t size) { |
| 626 h264_parser_.SetStream(stream, static_cast<off_t>(size)); | 617 h264_parser_.SetStream(stream, static_cast<off_t>(size)); |
| 627 | 618 |
| 628 while (1) { | 619 while (1) { |
| 629 H264NALU nalu; | 620 H264NALU nalu; |
| 630 H264Parser::Result result; | 621 H264Parser::Result result; |
| 631 | 622 |
| 632 result = h264_parser_.AdvanceToNextNALU(&nalu); | 623 result = h264_parser_.AdvanceToNextNALU(&nalu); |
| 633 if (result == H264Parser::kEOStream) | 624 if (result == H264Parser::kEOStream) |
| 634 break; | 625 break; |
| 635 | 626 |
| 636 ASSERT_EQ(H264Parser::kOk, result); | 627 ASSERT_EQ(H264Parser::kOk, result); |
| 637 | 628 |
| 638 bool keyframe = false; | 629 bool keyframe = false; |
| 639 | 630 |
| 640 switch (nalu.nal_unit_type) { | 631 switch (nalu.nal_unit_type) { |
| 641 case H264NALU::kIDRSlice: | 632 case H264NALU::kIDRSlice: |
| 642 ASSERT_TRUE(seen_sps_); | 633 ASSERT_TRUE(seen_sps_); |
| 643 ASSERT_TRUE(seen_pps_); | 634 ASSERT_TRUE(seen_pps_); |
| 644 seen_idr_ = true; | 635 seen_idr_ = true; |
| 645 keyframe = true; | 636 keyframe = true; |
| 646 // fallthrough | 637 // fallthrough |
| 647 case H264NALU::kNonIDRSlice: { | 638 case H264NALU::kNonIDRSlice: { |
| 648 ASSERT_TRUE(seen_idr_); | 639 ASSERT_TRUE(seen_idr_); |
| 649 H264SliceHeader slice_hdr; | 640 seen_sps_ = seen_pps_ = false; |
| 650 ASSERT_EQ(H264Parser::kOk, | 641 if (!frame_cb_.Run(keyframe)) |
| 651 h264_parser_.ParseSliceHeader(nalu, &slice_hdr)); | 642 return; |
| 652 if (IsNewPicture(slice_hdr)) { | |
| 653 if (!frame_cb_.Run(keyframe)) | |
| 654 return; | |
| 655 ASSERT_TRUE(UpdateCurrentPicture(slice_hdr)); | |
| 656 } | |
| 657 break; | 643 break; |
| 658 } | 644 } |
| 659 | 645 |
| 660 case H264NALU::kSPS: { | 646 case H264NALU::kSPS: { |
| 661 int sps_id; | 647 int sps_id; |
| 662 ASSERT_EQ(H264Parser::kOk, h264_parser_.ParseSPS(&sps_id)); | 648 ASSERT_EQ(H264Parser::kOk, h264_parser_.ParseSPS(&sps_id)); |
| 663 seen_sps_ = true; | 649 seen_sps_ = true; |
| 664 break; | 650 break; |
| 665 } | 651 } |
| 666 | 652 |
| 667 case H264NALU::kPPS: { | 653 case H264NALU::kPPS: { |
| 668 ASSERT_TRUE(seen_sps_); | 654 ASSERT_TRUE(seen_sps_); |
| 669 int pps_id; | 655 int pps_id; |
| 670 ASSERT_EQ(H264Parser::kOk, h264_parser_.ParsePPS(&pps_id)); | 656 ASSERT_EQ(H264Parser::kOk, h264_parser_.ParsePPS(&pps_id)); |
| 671 seen_pps_ = true; | 657 seen_pps_ = true; |
| 672 break; | 658 break; |
| 673 } | 659 } |
| 674 | 660 |
| 675 default: | 661 default: |
| 676 break; | 662 break; |
| 677 } | 663 } |
| 678 } | 664 } |
| 679 } | 665 } |
| 680 | 666 |
| 681 bool H264Validator::IsNewPicture(const H264SliceHeader& slice_hdr) { | |
| 682 if (!curr_pic_) | |
| 683 return true; | |
| 684 return H264Decoder::IsNewPrimaryCodedPicture( | |
| 685 curr_pic_, curr_pps_id_, h264_parser_.GetSPS(curr_sps_id_), slice_hdr); | |
| 686 } | |
| 687 | |
| 688 bool H264Validator::UpdateCurrentPicture(const H264SliceHeader& slice_hdr) { | |
| 689 curr_pps_id_ = slice_hdr.pic_parameter_set_id; | |
| 690 const H264PPS* pps = h264_parser_.GetPPS(curr_pps_id_); | |
| 691 if (!pps) { | |
| 692 LOG(ERROR) << "Cannot parse pps."; | |
| 693 return false; | |
| 694 } | |
| 695 | |
| 696 curr_sps_id_ = pps->seq_parameter_set_id; | |
| 697 const H264SPS* sps = h264_parser_.GetSPS(curr_sps_id_); | |
| 698 if (!sps) { | |
| 699 LOG(ERROR) << "Cannot parse sps."; | |
| 700 return false; | |
| 701 } | |
| 702 | |
| 703 curr_pic_ = H264Decoder::CreateH264PictureFromSliceHeader(sps, slice_hdr); | |
| 704 if (!curr_pic_) { | |
| 705 LOG(FATAL) << "Cannot initialize current frame."; | |
| 706 return false; | |
| 707 } | |
| 708 return true; | |
| 709 } | |
| 710 | |
| 711 class VP8Validator : public StreamValidator { | 667 class VP8Validator : public StreamValidator { |
| 712 public: | 668 public: |
| 713 explicit VP8Validator(const FrameFoundCallback& frame_cb) | 669 explicit VP8Validator(const FrameFoundCallback& frame_cb) |
| 714 : StreamValidator(frame_cb), seen_keyframe_(false) {} | 670 : StreamValidator(frame_cb), seen_keyframe_(false) {} |
| 715 | 671 |
| 716 void ProcessStreamBuffer(const uint8_t* stream, size_t size) override; | 672 void ProcessStreamBuffer(const uint8_t* stream, size_t size) override; |
| 717 | 673 |
| 718 private: | 674 private: |
| 719 // Have we already got a keyframe in the stream? | 675 // Have we already got a keyframe in the stream? |
| 720 bool seen_keyframe_; | 676 bool seen_keyframe_; |
| (...skipping 1549 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 2270 VideoEncodeAcceleratorTest, | 2226 VideoEncodeAcceleratorTest, |
| 2271 ::testing::Values(std::make_tuple(3, | 2227 ::testing::Values(std::make_tuple(3, |
| 2272 false, | 2228 false, |
| 2273 0, | 2229 0, |
| 2274 false, | 2230 false, |
| 2275 false, | 2231 false, |
| 2276 false, | 2232 false, |
| 2277 false, | 2233 false, |
| 2278 false, | 2234 false, |
| 2279 false))); | 2235 false))); |
| 2280 | |
| 2281 INSTANTIATE_TEST_CASE_P( | |
| 2282 VerifyTimestamp, | |
| 2283 VideoEncodeAcceleratorTest, | |
| 2284 ::testing::Values( | |
| 2285 std::make_tuple(1, false, 0, false, false, false, false, false, true))); | |
| 2286 | |
| 2287 #if defined(OS_WIN) | 2236 #if defined(OS_WIN) |
| 2288 INSTANTIATE_TEST_CASE_P( | 2237 INSTANTIATE_TEST_CASE_P( |
| 2289 ForceBitrate, | 2238 ForceBitrate, |
| 2290 VideoEncodeAcceleratorTest, | 2239 VideoEncodeAcceleratorTest, |
| 2291 ::testing::Values( | 2240 ::testing::Values( |
| 2292 std::make_tuple(1, false, 0, true, false, false, false, false, false))); | 2241 std::make_tuple(1, false, 0, true, false, false, false, false, false))); |
| 2293 #endif // defined(OS_WIN) | 2242 #endif // defined(OS_WIN) |
| 2294 | 2243 |
| 2295 #endif // defined(OS_CHROMEOS) | 2244 #endif // defined(OS_CHROMEOS) |
| 2296 | 2245 |
| (...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 2382 | 2331 |
| 2383 media::g_env = | 2332 media::g_env = |
| 2384 reinterpret_cast<media::VideoEncodeAcceleratorTestEnvironment*>( | 2333 reinterpret_cast<media::VideoEncodeAcceleratorTestEnvironment*>( |
| 2385 testing::AddGlobalTestEnvironment( | 2334 testing::AddGlobalTestEnvironment( |
| 2386 new media::VideoEncodeAcceleratorTestEnvironment( | 2335 new media::VideoEncodeAcceleratorTestEnvironment( |
| 2387 std::move(test_stream_data), log_path, run_at_fps, | 2336 std::move(test_stream_data), log_path, run_at_fps, |
| 2388 needs_encode_latency, verify_all_output))); | 2337 needs_encode_latency, verify_all_output))); |
| 2389 | 2338 |
| 2390 return RUN_ALL_TESTS(); | 2339 return RUN_ALL_TESTS(); |
| 2391 } | 2340 } |
| OLD | NEW |