| 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 552 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 563 bool seen_sps_; | 563 bool seen_sps_; |
| 564 bool seen_pps_; | 564 bool seen_pps_; |
| 565 bool seen_idr_; | 565 bool seen_idr_; |
| 566 | 566 |
| 567 H264Parser h264_parser_; | 567 H264Parser h264_parser_; |
| 568 }; | 568 }; |
| 569 | 569 |
| 570 void H264Validator::ProcessStreamBuffer(const uint8_t* stream, size_t size) { | 570 void H264Validator::ProcessStreamBuffer(const uint8_t* stream, size_t size) { |
| 571 h264_parser_.SetStream(stream, static_cast<off_t>(size)); | 571 h264_parser_.SetStream(stream, static_cast<off_t>(size)); |
| 572 | 572 |
| 573 // Run |frame_cb_| for only first nalu. | |
| 574 bool frame_cb_called = false; | |
| 575 | |
| 576 while (1) { | 573 while (1) { |
| 577 H264NALU nalu; | 574 H264NALU nalu; |
| 578 H264Parser::Result result; | 575 H264Parser::Result result; |
| 579 | 576 |
| 580 result = h264_parser_.AdvanceToNextNALU(&nalu); | 577 result = h264_parser_.AdvanceToNextNALU(&nalu); |
| 581 if (result == H264Parser::kEOStream) | 578 if (result == H264Parser::kEOStream) |
| 582 break; | 579 break; |
| 583 | 580 |
| 584 ASSERT_EQ(H264Parser::kOk, result); | 581 ASSERT_EQ(H264Parser::kOk, result); |
| 585 | 582 |
| 586 bool keyframe = false; | 583 bool keyframe = false; |
| 587 | 584 |
| 588 switch (nalu.nal_unit_type) { | 585 switch (nalu.nal_unit_type) { |
| 589 case H264NALU::kIDRSlice: | 586 case H264NALU::kIDRSlice: |
| 590 ASSERT_TRUE(seen_sps_); | 587 ASSERT_TRUE(seen_sps_); |
| 591 ASSERT_TRUE(seen_pps_); | 588 ASSERT_TRUE(seen_pps_); |
| 592 seen_idr_ = true; | 589 seen_idr_ = true; |
| 593 keyframe = true; | 590 keyframe = true; |
| 594 // fallthrough | 591 // fallthrough |
| 595 case H264NALU::kNonIDRSlice: { | 592 case H264NALU::kNonIDRSlice: { |
| 596 // Stream may contain at most one frame. | |
| 597 ASSERT_TRUE(seen_idr_); | 593 ASSERT_TRUE(seen_idr_); |
| 598 seen_sps_ = seen_pps_ = false; | 594 seen_sps_ = seen_pps_ = false; |
| 599 if (!frame_cb_called) { | 595 if (!frame_cb_.Run(keyframe)) |
| 600 frame_cb_called = true; | 596 return; |
| 601 if (!frame_cb_.Run(keyframe)) | |
| 602 return; | |
| 603 } | |
| 604 break; | 597 break; |
| 605 } | 598 } |
| 606 | 599 |
| 607 case H264NALU::kSPS: { | 600 case H264NALU::kSPS: { |
| 608 int sps_id; | 601 int sps_id; |
| 609 ASSERT_EQ(H264Parser::kOk, h264_parser_.ParseSPS(&sps_id)); | 602 ASSERT_EQ(H264Parser::kOk, h264_parser_.ParseSPS(&sps_id)); |
| 610 seen_sps_ = true; | 603 seen_sps_ = true; |
| 611 break; | 604 break; |
| 612 } | 605 } |
| 613 | 606 |
| (...skipping 1229 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1843 VideoEncodeAcceleratorTest, | 1836 VideoEncodeAcceleratorTest, |
| 1844 ::testing::Values(std::make_tuple(3, | 1837 ::testing::Values(std::make_tuple(3, |
| 1845 false, | 1838 false, |
| 1846 0, | 1839 0, |
| 1847 false, | 1840 false, |
| 1848 false, | 1841 false, |
| 1849 false, | 1842 false, |
| 1850 false, | 1843 false, |
| 1851 false, | 1844 false, |
| 1852 false))); | 1845 false))); |
| 1853 | |
| 1854 #if defined(OS_MACOSX) | |
| 1855 INSTANTIATE_TEST_CASE_P( | |
| 1856 VerifyTimestamp, | |
| 1857 VideoEncodeAcceleratorTest, | |
| 1858 ::testing::Values( | |
| 1859 std::make_tuple(1, false, 0, false, false, false, false, false, true))); | |
| 1860 #endif // defined(OS_MACOSX) | |
| 1861 | |
| 1862 #if defined(OS_WIN) | 1846 #if defined(OS_WIN) |
| 1863 INSTANTIATE_TEST_CASE_P( | 1847 INSTANTIATE_TEST_CASE_P( |
| 1864 ForceBitrate, | 1848 ForceBitrate, |
| 1865 VideoEncodeAcceleratorTest, | 1849 VideoEncodeAcceleratorTest, |
| 1866 ::testing::Values( | 1850 ::testing::Values( |
| 1867 std::make_tuple(1, false, 0, true, false, false, false, false, false))); | 1851 std::make_tuple(1, false, 0, true, false, false, false, false, false))); |
| 1868 #endif // defined(OS_WIN) | 1852 #endif // defined(OS_WIN) |
| 1869 | 1853 |
| 1870 #endif // defined(OS_CHROMEOS) | 1854 #endif // defined(OS_CHROMEOS) |
| 1871 | 1855 |
| (...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1957 | 1941 |
| 1958 media::g_env = | 1942 media::g_env = |
| 1959 reinterpret_cast<media::VideoEncodeAcceleratorTestEnvironment*>( | 1943 reinterpret_cast<media::VideoEncodeAcceleratorTestEnvironment*>( |
| 1960 testing::AddGlobalTestEnvironment( | 1944 testing::AddGlobalTestEnvironment( |
| 1961 new media::VideoEncodeAcceleratorTestEnvironment( | 1945 new media::VideoEncodeAcceleratorTestEnvironment( |
| 1962 std::move(test_stream_data), log_path, run_at_fps, | 1946 std::move(test_stream_data), log_path, run_at_fps, |
| 1963 needs_encode_latency, verify_all_output))); | 1947 needs_encode_latency, verify_all_output))); |
| 1964 | 1948 |
| 1965 return RUN_ALL_TESTS(); | 1949 return RUN_ALL_TESTS(); |
| 1966 } | 1950 } |
| OLD | NEW |