av1_config_test.cc (6940B)
1 /* 2 * Copyright (c) 2018, Alliance for Open Media. All rights reserved. 3 * 4 * This source code is subject to the terms of the BSD 2 Clause License and 5 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License 6 * was not distributed with this source code in the LICENSE file, you can 7 * obtain it at www.aomedia.org/license/software. If the Alliance for Open 8 * Media Patent License 1.0 was not distributed with this source code in the 9 * PATENTS file, you can obtain it at www.aomedia.org/license/patent. 10 */ 11 #include <string.h> 12 13 #include "common/av1_config.h" 14 #include "gtest/gtest.h" 15 #include "test/util.h" 16 17 namespace { 18 19 // 20 // Input buffers containing exactly one Sequence Header OBU. 21 // 22 // Each buffer is named according to the OBU storage format (Annex-B vs Low 23 // Overhead Bitstream Format) and the type of Sequence Header OBU ("Full" 24 // Sequence Header OBUs vs Sequence Header OBUs with the 25 // reduced_still_image_flag set). 26 // 27 const uint8_t kAnnexBFullSequenceHeaderObu[] = { 0x0c, 0x08, 0x00, 0x00, 0x00, 28 0x04, 0x45, 0x7e, 0x3e, 0xff, 29 0xfc, 0xc0, 0x20 }; 30 const uint8_t kAnnexBReducedStillImageSequenceHeaderObu[] = { 31 0x08, 0x08, 0x18, 0x22, 0x2b, 0xf1, 0xfe, 0xc0, 0x20 32 }; 33 34 const uint8_t kLobfFullSequenceHeaderObu[] = { 0x0a, 0x0b, 0x00, 0x00, 0x00, 35 0x04, 0x45, 0x7e, 0x3e, 0xff, 36 0xfc, 0xc0, 0x20 }; 37 38 const uint8_t kLobfReducedStillImageSequenceHeaderObu[] = { 0x0a, 0x07, 0x18, 39 0x22, 0x2b, 0xf1, 40 0xfe, 0xc0, 0x20 }; 41 42 const uint8_t kAv1cAllZero[] = { 0, 0, 0, 0 }; 43 44 // The size of AV1 config when no configOBUs are present at the end of the 45 // configuration structure. 46 const size_t kAv1cNoConfigObusSize = 4; 47 48 bool VerifyAv1c(const uint8_t *const obu_buffer, size_t obu_buffer_length, 49 bool is_annexb) { 50 Av1Config av1_config; 51 memset(&av1_config, 0, sizeof(av1_config)); 52 bool parse_ok = get_av1config_from_obu(obu_buffer, obu_buffer_length, 53 is_annexb, &av1_config) == 0; 54 if (parse_ok) { 55 EXPECT_EQ(1, av1_config.marker); 56 EXPECT_EQ(1, av1_config.version); 57 EXPECT_EQ(0, av1_config.seq_profile); 58 EXPECT_EQ(0, av1_config.seq_level_idx_0); 59 EXPECT_EQ(0, av1_config.seq_tier_0); 60 EXPECT_EQ(0, av1_config.high_bitdepth); 61 EXPECT_EQ(0, av1_config.twelve_bit); 62 EXPECT_EQ(0, av1_config.monochrome); 63 EXPECT_EQ(1, av1_config.chroma_subsampling_x); 64 EXPECT_EQ(1, av1_config.chroma_subsampling_y); 65 EXPECT_EQ(0, av1_config.chroma_sample_position); 66 EXPECT_EQ(0, av1_config.initial_presentation_delay_present); 67 EXPECT_EQ(0, av1_config.initial_presentation_delay_minus_one); 68 } 69 return parse_ok && ::testing::Test::HasFailure() == false; 70 } 71 72 TEST(Av1Config, ObuInvalidInputs) { 73 Av1Config av1_config; 74 memset(&av1_config, 0, sizeof(av1_config)); 75 ASSERT_EQ(-1, get_av1config_from_obu(nullptr, 0, 0, nullptr)); 76 ASSERT_EQ(-1, get_av1config_from_obu(&kLobfFullSequenceHeaderObu[0], 0, 0, 77 nullptr)); 78 ASSERT_EQ(-1, get_av1config_from_obu(&kLobfFullSequenceHeaderObu[0], 79 sizeof(kLobfFullSequenceHeaderObu), 0, 80 nullptr)); 81 ASSERT_EQ(-1, get_av1config_from_obu( 82 nullptr, sizeof(kLobfFullSequenceHeaderObu), 0, nullptr)); 83 ASSERT_EQ(-1, get_av1config_from_obu(&kLobfFullSequenceHeaderObu[0], 0, 0, 84 &av1_config)); 85 } 86 87 TEST(Av1Config, ReadInvalidInputs) { 88 Av1Config av1_config; 89 memset(&av1_config, 0, sizeof(av1_config)); 90 size_t bytes_read = 0; 91 ASSERT_EQ(-1, read_av1config(nullptr, 0, nullptr, nullptr)); 92 ASSERT_EQ(-1, read_av1config(nullptr, 4, nullptr, nullptr)); 93 ASSERT_EQ(-1, read_av1config(&kAv1cAllZero[0], 0, nullptr, nullptr)); 94 ASSERT_EQ(-1, read_av1config(&kAv1cAllZero[0], 4, &bytes_read, nullptr)); 95 ASSERT_EQ(-1, read_av1config(nullptr, 4, &bytes_read, &av1_config)); 96 } 97 98 TEST(Av1Config, WriteInvalidInputs) { 99 Av1Config av1_config; 100 memset(&av1_config, 0, sizeof(av1_config)); 101 size_t bytes_written = 0; 102 uint8_t av1c_buffer[4] = { 0 }; 103 ASSERT_EQ(-1, write_av1config(nullptr, 0, nullptr, nullptr)); 104 ASSERT_EQ(-1, write_av1config(&av1_config, 0, nullptr, nullptr)); 105 ASSERT_EQ(-1, write_av1config(&av1_config, 0, &bytes_written, nullptr)); 106 107 ASSERT_EQ(-1, 108 write_av1config(&av1_config, 0, &bytes_written, &av1c_buffer[0])); 109 ASSERT_EQ(-1, write_av1config(&av1_config, 4, &bytes_written, nullptr)); 110 } 111 112 TEST(Av1Config, GetAv1ConfigFromLobfObu) { 113 // Test parsing of a Sequence Header OBU with the reduced_still_picture_header 114 // unset-- aka a full Sequence Header OBU. 115 ASSERT_TRUE(VerifyAv1c(kLobfFullSequenceHeaderObu, 116 sizeof(kLobfFullSequenceHeaderObu), false)); 117 118 // Test parsing of a reduced still image Sequence Header OBU. 119 ASSERT_TRUE(VerifyAv1c(kLobfReducedStillImageSequenceHeaderObu, 120 sizeof(kLobfReducedStillImageSequenceHeaderObu), 121 false)); 122 } 123 124 TEST(Av1Config, GetAv1ConfigFromAnnexBObu) { 125 // Test parsing of a Sequence Header OBU with the reduced_still_picture_header 126 // unset-- aka a full Sequence Header OBU. 127 ASSERT_TRUE(VerifyAv1c(kAnnexBFullSequenceHeaderObu, 128 sizeof(kAnnexBFullSequenceHeaderObu), true)); 129 130 // Test parsing of a reduced still image Sequence Header OBU. 131 ASSERT_TRUE(VerifyAv1c(kAnnexBReducedStillImageSequenceHeaderObu, 132 sizeof(kAnnexBReducedStillImageSequenceHeaderObu), 133 true)); 134 } 135 136 TEST(Av1Config, ReadWriteConfig) { 137 Av1Config av1_config; 138 memset(&av1_config, 0, sizeof(av1_config)); 139 140 // Test writing out the AV1 config. 141 size_t bytes_written = 0; 142 uint8_t av1c_buffer[4] = { 0 }; 143 ASSERT_EQ(0, write_av1config(&av1_config, sizeof(av1c_buffer), &bytes_written, 144 &av1c_buffer[0])); 145 ASSERT_EQ(kAv1cNoConfigObusSize, bytes_written); 146 for (size_t i = 0; i < kAv1cNoConfigObusSize; ++i) { 147 ASSERT_EQ(kAv1cAllZero[i], av1c_buffer[i]) 148 << "Mismatch in output Av1Config at offset=" << i; 149 } 150 151 // Test reading the AV1 config. 152 size_t bytes_read = 0; 153 ASSERT_EQ(0, read_av1config(&kAv1cAllZero[0], sizeof(kAv1cAllZero), 154 &bytes_read, &av1_config)); 155 ASSERT_EQ(kAv1cNoConfigObusSize, bytes_read); 156 ASSERT_EQ(0, write_av1config(&av1_config, sizeof(av1c_buffer), &bytes_written, 157 &av1c_buffer[0])); 158 for (size_t i = 0; i < kAv1cNoConfigObusSize; ++i) { 159 ASSERT_EQ(kAv1cAllZero[i], av1c_buffer[i]) 160 << "Mismatch in output Av1Config at offset=" << i; 161 } 162 } 163 164 } // namespace