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