| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <stdint.h> |
| |
| #include "base/files/file_path.h" |
| #include "base/files/memory_mapped_file.h" |
| #include "base/macros.h" |
| #include "media/base/test_data_util.h" |
| #include "media/video/h264_parser.h" |
| #include "media/video/h264_poc.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace media { |
| |
| class H264POCTest : public testing::Test { |
| public: |
| H264POCTest() : sps_(), slice_hdr_() { |
| // Default every frame to be a reference frame. |
| slice_hdr_.nal_ref_idc = 1; |
| } |
| |
| protected: |
| void ComputePOC() { |
| poc_ = h264_poc_.ComputePicOrderCnt(&sps_, slice_hdr_); |
| |
| // Clear MMCO5. |
| slice_hdr_.adaptive_ref_pic_marking_mode_flag = false; |
| slice_hdr_.ref_pic_marking[0].memory_mgmnt_control_operation = 0; |
| slice_hdr_.ref_pic_marking[1].memory_mgmnt_control_operation = 0; |
| slice_hdr_.ref_pic_marking[2].memory_mgmnt_control_operation = 0; |
| } |
| |
| // Also sets as a reference frame and unsets IDR, which is required for |
| // memory management control operations to be parsed. |
| void SetMMCO5() { |
| slice_hdr_.nal_ref_idc = 1; |
| slice_hdr_.idr_pic_flag = false; |
| slice_hdr_.adaptive_ref_pic_marking_mode_flag = true; |
| slice_hdr_.ref_pic_marking[0].memory_mgmnt_control_operation = 6; |
| slice_hdr_.ref_pic_marking[1].memory_mgmnt_control_operation = 5; |
| slice_hdr_.ref_pic_marking[2].memory_mgmnt_control_operation = 0; |
| } |
| |
| base::Optional<int32_t> poc_; |
| |
| H264SPS sps_; |
| H264SliceHeader slice_hdr_; |
| H264POC h264_poc_; |
| |
| DISALLOW_COPY_AND_ASSIGN(H264POCTest); |
| }; |
| |
| TEST_F(H264POCTest, PicOrderCntType0) { |
| sps_.pic_order_cnt_type = 0; |
| sps_.log2_max_pic_order_cnt_lsb_minus4 = 0; // 16 |
| |
| // Initial IDR with POC 0. |
| slice_hdr_.idr_pic_flag = true; |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(0, *poc_); |
| |
| // Ref frame with POC lsb 8. |
| slice_hdr_.idr_pic_flag = false; |
| slice_hdr_.frame_num = 1; |
| slice_hdr_.pic_order_cnt_lsb = 8; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(8, *poc_); |
| |
| // Ref frame with POC lsb 0. This should be detected as wrapping, as the |
| // (negative) gap is at least half the maximum. |
| slice_hdr_.pic_order_cnt_lsb = 0; |
| slice_hdr_.frame_num = 2; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(16, *poc_); |
| |
| // Ref frame with POC lsb 9. This should be detected as negative wrapping, |
| // as the (positive) gap is more than half the maximum. |
| slice_hdr_.pic_order_cnt_lsb = 9; |
| slice_hdr_.frame_num = 3; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(9, *poc_); |
| } |
| |
| TEST_F(H264POCTest, PicOrderCntType0_WithMMCO5) { |
| sps_.pic_order_cnt_type = 0; |
| sps_.log2_max_pic_order_cnt_lsb_minus4 = 0; // 16 |
| |
| // Initial IDR with POC 0. |
| slice_hdr_.idr_pic_flag = true; |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(0, *poc_); |
| |
| // Skip ahead. |
| slice_hdr_.idr_pic_flag = false; |
| slice_hdr_.frame_num = 1; |
| slice_hdr_.pic_order_cnt_lsb = 8; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(8, *poc_); |
| |
| slice_hdr_.frame_num = 2; |
| slice_hdr_.pic_order_cnt_lsb = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(16, *poc_); |
| |
| slice_hdr_.frame_num = 3; |
| slice_hdr_.pic_order_cnt_lsb = 8; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(24, *poc_); |
| |
| // MMCO5 resets to 0. |
| slice_hdr_.frame_num = 4; |
| slice_hdr_.pic_order_cnt_lsb = 0; |
| SetMMCO5(); |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(0, *poc_); |
| |
| // Still detected as positive wrapping. |
| slice_hdr_.frame_num = 5; |
| slice_hdr_.pic_order_cnt_lsb = 8; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(24, *poc_); |
| } |
| |
| TEST_F(H264POCTest, PicOrderCntType1) { |
| sps_.pic_order_cnt_type = 1; |
| sps_.log2_max_frame_num_minus4 = 0; // 16 |
| sps_.num_ref_frames_in_pic_order_cnt_cycle = 2; |
| sps_.expected_delta_per_pic_order_cnt_cycle = 3; |
| sps_.offset_for_ref_frame[0] = 1; |
| sps_.offset_for_ref_frame[1] = 2; |
| |
| // Initial IDR with POC 0. |
| slice_hdr_.idr_pic_flag = true; |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(0, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.idr_pic_flag = false; |
| slice_hdr_.frame_num = 1; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(1, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.frame_num = 2; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(3, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.frame_num = 3; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(4, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.frame_num = 4; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(6, *poc_); |
| |
| // Ref frame, detected as wrapping (ie, this is frame 16). |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(24, *poc_); |
| } |
| |
| TEST_F(H264POCTest, PicOrderCntType1_WithMMCO5) { |
| sps_.pic_order_cnt_type = 1; |
| sps_.log2_max_frame_num_minus4 = 0; // 16 |
| sps_.num_ref_frames_in_pic_order_cnt_cycle = 2; |
| sps_.expected_delta_per_pic_order_cnt_cycle = 3; |
| sps_.offset_for_ref_frame[0] = 1; |
| sps_.offset_for_ref_frame[1] = 2; |
| |
| // Initial IDR with POC 0. |
| slice_hdr_.idr_pic_flag = true; |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(0, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.idr_pic_flag = false; |
| slice_hdr_.frame_num = 1; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(1, *poc_); |
| |
| // Ref frame, detected as wrapping. |
| SetMMCO5(); |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(0, *poc_); |
| |
| // Ref frame, wrapping from before has been cleared. |
| slice_hdr_.frame_num = 1; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(1, *poc_); |
| } |
| |
| // |frame_num| values may be duplicated by non-reference frames. |
| TEST_F(H264POCTest, PicOrderCntType1_DupFrameNum) { |
| sps_.pic_order_cnt_type = 1; |
| sps_.log2_max_frame_num_minus4 = 0; // 16 |
| sps_.num_ref_frames_in_pic_order_cnt_cycle = 2; |
| sps_.expected_delta_per_pic_order_cnt_cycle = 3; |
| sps_.offset_for_ref_frame[0] = 1; |
| sps_.offset_for_ref_frame[1] = 2; |
| |
| // Initial IDR with POC 0. |
| slice_hdr_.idr_pic_flag = true; |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(0, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.idr_pic_flag = false; |
| slice_hdr_.frame_num = 1; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(1, *poc_); |
| |
| // Duplicate |frame_num| frame. |
| slice_hdr_.nal_ref_idc = 0; |
| slice_hdr_.frame_num = 1; |
| slice_hdr_.delta_pic_order_cnt0 = 2; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(2, *poc_); |
| } |
| |
| TEST_F(H264POCTest, PicOrderCntType2) { |
| sps_.pic_order_cnt_type = 2; |
| |
| // Initial IDR with POC 0. |
| slice_hdr_.idr_pic_flag = true; |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(0, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.idr_pic_flag = false; |
| slice_hdr_.frame_num = 1; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(2, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.frame_num = 2; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(4, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.frame_num = 3; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(6, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.frame_num = 4; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(8, *poc_); |
| |
| // Ref frame, detected as wrapping (ie, this is frame 16). |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(32, *poc_); |
| } |
| |
| TEST_F(H264POCTest, PicOrderCntType2_WithMMCO5) { |
| sps_.pic_order_cnt_type = 2; |
| |
| // Initial IDR with POC 0. |
| slice_hdr_.idr_pic_flag = true; |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(0, *poc_); |
| |
| // Ref frame. |
| slice_hdr_.idr_pic_flag = false; |
| slice_hdr_.frame_num = 1; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(2, *poc_); |
| |
| // Ref frame, detected as wrapping. |
| SetMMCO5(); |
| slice_hdr_.frame_num = 0; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(0, *poc_); |
| |
| // Ref frame, wrapping from before has been cleared. |
| slice_hdr_.frame_num = 1; |
| ComputePOC(); |
| ASSERT_TRUE(poc_.has_value()); |
| ASSERT_EQ(2, *poc_); |
| } |
| |
| } // namespace media |