123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512 |
- /*
- * Copyright (c) 2018, Alliance for Open Media. All rights reserved
- *
- * This source code is subject to the terms of the BSD 2 Clause License and
- * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
- * was not distributed with this source code in the LICENSE file, you can
- * obtain it at www.aomedia.org/license/software. If the Alliance for Open
- * Media Patent License 1.0 was not distributed with this source code in the
- * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
- */
- #include <stdio.h>
- #include <string.h>
- #include "aom/aom_image.h"
- #include "aom/aom_integer.h"
- #include "aom_dsp/bitreader_buffer.h"
- #include "aom_dsp/bitwriter_buffer.h"
- #include "av1/common/obu_util.h"
- #include "common/av1_config.h"
- #include "config/aom_config.h"
- // Helper macros to reduce verbosity required to check for read errors.
- //
- // Note that when using these macros, even single line if statements should use
- // curly braces to avoid unexpected behavior because all but the
- // AV1C_POP_ERROR_HANDLER_DATA() macro consist of multiple statements.
- #define AV1C_READ_BIT_OR_RETURN_ERROR(field) \
- int field = 0; \
- do { \
- field = aom_rb_read_bit(reader); \
- if (result == -1) { \
- fprintf(stderr, \
- "av1c: Error reading bit for " #field ", value=%d result=%d.\n", \
- field, result); \
- return -1; \
- } \
- } while (0)
- #define AV1C_READ_BITS_OR_RETURN_ERROR(field, length) \
- int field = 0; \
- do { \
- field = aom_rb_read_literal(reader, (length)); \
- if (result == -1) { \
- fprintf(stderr, \
- "av1c: Could not read bits for " #field \
- ", value=%d result=%d.\n", \
- field, result); \
- return -1; \
- } \
- } while (0)
- // Helper macros for setting/restoring the error handler data in
- // aom_read_bit_buffer.
- #define AV1C_PUSH_ERROR_HANDLER_DATA(new_data) \
- void *original_error_handler_data = NULL; \
- do { \
- original_error_handler_data = reader->error_handler_data; \
- reader->error_handler_data = &new_data; \
- } while (0)
- #define AV1C_POP_ERROR_HANDLER_DATA() \
- do { \
- reader->error_handler_data = original_error_handler_data; \
- } while (0)
- static const size_t kAv1cSize = 4;
- static void bitreader_error_handler(void *data) {
- int *error_val = (int *)data;
- *error_val = -1;
- }
- // Parse the AV1 timing_info() structure:
- // timing_info( ) {
- // num_units_in_display_tick f(32)
- // time_scale f(32)
- // equal_picture_interval f(1)
- // if (equal_picture_interval)
- // num_ticks_per_picture_minus_1 uvlc()
- // }
- static int parse_timing_info(struct aom_read_bit_buffer *reader) {
- int result = 0;
- AV1C_PUSH_ERROR_HANDLER_DATA(result);
- AV1C_READ_BITS_OR_RETURN_ERROR(num_units_in_display_tick, 32);
- AV1C_READ_BITS_OR_RETURN_ERROR(time_scale, 32);
- AV1C_READ_BIT_OR_RETURN_ERROR(equal_picture_interval);
- if (equal_picture_interval) {
- uint32_t num_ticks_per_picture_minus_1 = aom_rb_read_uvlc(reader);
- if (result == -1) {
- fprintf(stderr,
- "av1c: Could not read bits for "
- "num_ticks_per_picture_minus_1, value=%u.\n",
- num_ticks_per_picture_minus_1);
- return result;
- }
- }
- AV1C_POP_ERROR_HANDLER_DATA();
- return result;
- }
- // Parse the AV1 decoder_model_info() structure:
- // decoder_model_info( ) {
- // buffer_delay_length_minus_1 f(5)
- // num_units_in_decoding_tick f(32)
- // buffer_removal_time_length_minus_1 f(5)
- // frame_presentation_time_length_minus_1 f(5)
- // }
- //
- // Returns -1 upon failure, or the value of buffer_delay_length_minus_1 + 1.
- static int parse_decoder_model_info(struct aom_read_bit_buffer *reader) {
- int result = 0;
- AV1C_PUSH_ERROR_HANDLER_DATA(result);
- AV1C_READ_BITS_OR_RETURN_ERROR(buffer_delay_length_minus_1, 5);
- AV1C_READ_BITS_OR_RETURN_ERROR(num_units_in_decoding_tick, 32);
- AV1C_READ_BITS_OR_RETURN_ERROR(buffer_removal_time_length_minus_1, 5);
- AV1C_READ_BITS_OR_RETURN_ERROR(frame_presentation_time_length_minus_1, 5);
- AV1C_POP_ERROR_HANDLER_DATA();
- return buffer_delay_length_minus_1 + 1;
- }
- // Parse the AV1 operating_parameters_info() structure:
- // operating_parameters_info( op ) {
- // n = buffer_delay_length_minus_1 + 1
- // decoder_buffer_delay[ op ] f(n)
- // encoder_buffer_delay[ op ] f(n)
- // low_delay_mode_flag[ op ] f(1)
- // }
- static int parse_operating_parameters_info(struct aom_read_bit_buffer *reader,
- int buffer_delay_length_minus_1) {
- int result = 0;
- AV1C_PUSH_ERROR_HANDLER_DATA(result);
- const int buffer_delay_length = buffer_delay_length_minus_1 + 1;
- AV1C_READ_BITS_OR_RETURN_ERROR(decoder_buffer_delay, buffer_delay_length);
- AV1C_READ_BITS_OR_RETURN_ERROR(encoder_buffer_delay, buffer_delay_length);
- AV1C_READ_BIT_OR_RETURN_ERROR(low_delay_mode_flag);
- AV1C_POP_ERROR_HANDLER_DATA();
- return result;
- }
- // Parse the AV1 color_config() structure..See:
- // https://aomediacodec.github.io/av1-spec/av1-spec.pdf#page=44
- static int parse_color_config(struct aom_read_bit_buffer *reader,
- Av1Config *config) {
- int result = 0;
- AV1C_PUSH_ERROR_HANDLER_DATA(result);
- AV1C_READ_BIT_OR_RETURN_ERROR(high_bitdepth);
- config->high_bitdepth = high_bitdepth;
- int bit_depth = 0;
- if (config->seq_profile == 2 && config->high_bitdepth) {
- AV1C_READ_BIT_OR_RETURN_ERROR(twelve_bit);
- config->twelve_bit = twelve_bit;
- bit_depth = config->twelve_bit ? 12 : 10;
- } else {
- bit_depth = config->high_bitdepth ? 10 : 8;
- }
- if (config->seq_profile != 1) {
- AV1C_READ_BIT_OR_RETURN_ERROR(mono_chrome);
- config->monochrome = mono_chrome;
- }
- int color_primaries = AOM_CICP_CP_UNSPECIFIED;
- int transfer_characteristics = AOM_CICP_TC_UNSPECIFIED;
- int matrix_coefficients = AOM_CICP_MC_UNSPECIFIED;
- AV1C_READ_BIT_OR_RETURN_ERROR(color_description_present_flag);
- if (color_description_present_flag) {
- AV1C_READ_BITS_OR_RETURN_ERROR(color_primaries_val, 8);
- color_primaries = color_primaries_val;
- AV1C_READ_BITS_OR_RETURN_ERROR(transfer_characteristics_val, 8);
- transfer_characteristics = transfer_characteristics_val;
- AV1C_READ_BITS_OR_RETURN_ERROR(matrix_coefficients_val, 8);
- matrix_coefficients = matrix_coefficients_val;
- }
- if (config->monochrome) {
- AV1C_READ_BIT_OR_RETURN_ERROR(color_range);
- config->chroma_subsampling_x = 1;
- config->chroma_subsampling_y = 1;
- } else if (color_primaries == AOM_CICP_CP_BT_709 &&
- transfer_characteristics == AOM_CICP_TC_SRGB &&
- matrix_coefficients == AOM_CICP_MC_IDENTITY) {
- config->chroma_subsampling_x = 0;
- config->chroma_subsampling_y = 0;
- } else {
- AV1C_READ_BIT_OR_RETURN_ERROR(color_range);
- if (config->seq_profile == 0) {
- config->chroma_subsampling_x = 1;
- config->chroma_subsampling_y = 1;
- } else if (config->seq_profile == 1) {
- config->chroma_subsampling_x = 0;
- config->chroma_subsampling_y = 0;
- } else {
- if (bit_depth == 12) {
- AV1C_READ_BIT_OR_RETURN_ERROR(subsampling_x);
- config->chroma_subsampling_x = subsampling_x;
- if (subsampling_x) {
- AV1C_READ_BIT_OR_RETURN_ERROR(subsampling_y);
- config->chroma_subsampling_y = subsampling_y;
- } else {
- config->chroma_subsampling_y = 0;
- }
- } else {
- config->chroma_subsampling_x = 1;
- config->chroma_subsampling_y = 0;
- }
- }
- if (config->chroma_subsampling_x && config->chroma_subsampling_y) {
- AV1C_READ_BITS_OR_RETURN_ERROR(chroma_sample_position, 2);
- config->chroma_sample_position = chroma_sample_position;
- }
- }
- if (!config->monochrome) {
- AV1C_READ_BIT_OR_RETURN_ERROR(separate_uv_delta_q);
- }
- AV1C_POP_ERROR_HANDLER_DATA();
- return result;
- }
- // Parse AV1 Sequence Header OBU. See:
- // https://aomediacodec.github.io/av1-spec/av1-spec.pdf#page=41
- static int parse_sequence_header(const uint8_t *const buffer, size_t length,
- Av1Config *config) {
- int result = 0;
- // The reader instance is local to this function, but a pointer to the
- // reader instance is used within this function and throughout this file to
- // allow use of the helper macros that reduce parse error checking verbosity.
- struct aom_read_bit_buffer reader_instance = { buffer, buffer + length, 0,
- &result,
- bitreader_error_handler };
- struct aom_read_bit_buffer *reader = &reader_instance;
- AV1C_READ_BITS_OR_RETURN_ERROR(seq_profile, 3);
- config->seq_profile = seq_profile;
- AV1C_READ_BIT_OR_RETURN_ERROR(still_picture);
- AV1C_READ_BIT_OR_RETURN_ERROR(reduced_still_picture_header);
- if (reduced_still_picture_header) {
- config->initial_presentation_delay_present = 0;
- AV1C_READ_BITS_OR_RETURN_ERROR(seq_level_idx_0, 5);
- config->seq_level_idx_0 = seq_level_idx_0;
- config->seq_tier_0 = 0;
- } else {
- int has_decoder_model = 0;
- int buffer_delay_length = 0;
- AV1C_READ_BIT_OR_RETURN_ERROR(timing_info_present_flag);
- if (timing_info_present_flag) {
- if (parse_timing_info(reader) != 0) return -1;
- AV1C_READ_BIT_OR_RETURN_ERROR(decoder_model_info_present_flag);
- if (decoder_model_info_present_flag &&
- (buffer_delay_length = parse_decoder_model_info(reader)) == -1) {
- return -1;
- }
- has_decoder_model = 1;
- }
- AV1C_READ_BIT_OR_RETURN_ERROR(initial_presentation_delay_present);
- config->initial_presentation_delay_present =
- initial_presentation_delay_present;
- AV1C_READ_BITS_OR_RETURN_ERROR(operating_points_cnt_minus_1, 5);
- const int num_operating_points = operating_points_cnt_minus_1 + 1;
- for (int op_index = 0; op_index < num_operating_points; ++op_index) {
- AV1C_READ_BITS_OR_RETURN_ERROR(operating_point_idc, 12);
- AV1C_READ_BITS_OR_RETURN_ERROR(seq_level_idx, 5);
- int seq_tier = 0;
- if (seq_level_idx > 7) {
- AV1C_READ_BIT_OR_RETURN_ERROR(seq_tier_this_op);
- seq_tier = seq_tier_this_op;
- }
- if (has_decoder_model) {
- AV1C_READ_BIT_OR_RETURN_ERROR(decoder_model_present_for_op);
- if (decoder_model_present_for_op) {
- if (parse_operating_parameters_info(reader, buffer_delay_length) ==
- -1) {
- return -1;
- }
- }
- }
- if (config->initial_presentation_delay_present) {
- // Skip the initial presentation delay bits if present since this
- // function has no access to the data required to properly set the
- // field.
- AV1C_READ_BIT_OR_RETURN_ERROR(
- initial_presentation_delay_present_for_this_op);
- if (initial_presentation_delay_present_for_this_op) {
- AV1C_READ_BITS_OR_RETURN_ERROR(initial_presentation_delay_minus_1, 4);
- }
- }
- if (op_index == 0) {
- // Av1Config needs only the values from the first operating point.
- config->seq_level_idx_0 = seq_level_idx;
- config->seq_tier_0 = seq_tier;
- config->initial_presentation_delay_present = 0;
- config->initial_presentation_delay_minus_one = 0;
- }
- }
- }
- AV1C_READ_BITS_OR_RETURN_ERROR(frame_width_bits_minus_1, 4);
- AV1C_READ_BITS_OR_RETURN_ERROR(frame_height_bits_minus_1, 4);
- AV1C_READ_BITS_OR_RETURN_ERROR(max_frame_width_minus_1,
- frame_width_bits_minus_1 + 1);
- AV1C_READ_BITS_OR_RETURN_ERROR(max_frame_height_minus_1,
- frame_height_bits_minus_1 + 1);
- uint8_t frame_id_numbers_present = 0;
- if (!reduced_still_picture_header) {
- AV1C_READ_BIT_OR_RETURN_ERROR(frame_id_numbers_present_flag);
- frame_id_numbers_present = frame_id_numbers_present_flag;
- }
- if (frame_id_numbers_present) {
- AV1C_READ_BITS_OR_RETURN_ERROR(delta_frame_id_length_minus_2, 4);
- AV1C_READ_BITS_OR_RETURN_ERROR(additional_frame_id_length_minus_1, 3);
- }
- AV1C_READ_BIT_OR_RETURN_ERROR(use_128x128_superblock);
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_filter_intra);
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_intra_edge_filter);
- if (!reduced_still_picture_header) {
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_interintra_compound);
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_masked_compound);
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_warped_motion);
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_dual_filter);
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_order_hint);
- if (enable_order_hint) {
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_dist_wtd_comp);
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_ref_frame_mvs);
- }
- const int SELECT_SCREEN_CONTENT_TOOLS = 2;
- int seq_force_screen_content_tools = SELECT_SCREEN_CONTENT_TOOLS;
- AV1C_READ_BIT_OR_RETURN_ERROR(seq_choose_screen_content_tools);
- if (!seq_choose_screen_content_tools) {
- AV1C_READ_BIT_OR_RETURN_ERROR(seq_force_screen_content_tools_val);
- seq_force_screen_content_tools = seq_force_screen_content_tools_val;
- }
- if (seq_force_screen_content_tools > 0) {
- AV1C_READ_BIT_OR_RETURN_ERROR(seq_choose_integer_mv);
- if (!seq_choose_integer_mv) {
- AV1C_READ_BIT_OR_RETURN_ERROR(seq_force_integer_mv);
- }
- }
- if (enable_order_hint) {
- AV1C_READ_BITS_OR_RETURN_ERROR(order_hint_bits_minus_1, 3);
- }
- }
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_superres);
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_cdef);
- AV1C_READ_BIT_OR_RETURN_ERROR(enable_restoration);
- if (parse_color_config(reader, config) != 0) {
- fprintf(stderr, "av1c: color_config() parse failed.\n");
- return -1;
- }
- AV1C_READ_BIT_OR_RETURN_ERROR(film_grain_params_present);
- return 0;
- }
- int get_av1config_from_obu(const uint8_t *buffer, size_t length, int is_annexb,
- Av1Config *config) {
- if (!buffer || length == 0 || !config) {
- return -1;
- }
- ObuHeader obu_header;
- memset(&obu_header, 0, sizeof(obu_header));
- size_t sequence_header_length = 0;
- size_t obu_header_length = 0;
- if (aom_read_obu_header_and_size(buffer, length, is_annexb, &obu_header,
- &sequence_header_length,
- &obu_header_length) != AOM_CODEC_OK ||
- obu_header.type != OBU_SEQUENCE_HEADER ||
- sequence_header_length + obu_header_length > length) {
- return -1;
- }
- memset(config, 0, sizeof(*config));
- config->marker = 1;
- config->version = 1;
- return parse_sequence_header(buffer + obu_header_length,
- sequence_header_length, config);
- }
- int read_av1config(const uint8_t *buffer, size_t buffer_length,
- size_t *bytes_read, Av1Config *config) {
- if (!buffer || buffer_length < kAv1cSize || !bytes_read || !config) return -1;
- *bytes_read = 0;
- int result = 0;
- struct aom_read_bit_buffer reader_instance = { buffer, buffer + buffer_length,
- 0, &result,
- bitreader_error_handler };
- struct aom_read_bit_buffer *reader = &reader_instance;
- memset(config, 0, sizeof(*config));
- AV1C_READ_BIT_OR_RETURN_ERROR(marker);
- config->marker = marker;
- AV1C_READ_BITS_OR_RETURN_ERROR(version, 7);
- config->version = version;
- AV1C_READ_BITS_OR_RETURN_ERROR(seq_profile, 3);
- config->seq_profile = seq_profile;
- AV1C_READ_BITS_OR_RETURN_ERROR(seq_level_idx_0, 5);
- config->seq_level_idx_0 = seq_level_idx_0;
- AV1C_READ_BIT_OR_RETURN_ERROR(seq_tier_0);
- config->seq_tier_0 = seq_tier_0;
- AV1C_READ_BIT_OR_RETURN_ERROR(high_bitdepth);
- config->high_bitdepth = high_bitdepth;
- AV1C_READ_BIT_OR_RETURN_ERROR(twelve_bit);
- config->twelve_bit = twelve_bit;
- AV1C_READ_BIT_OR_RETURN_ERROR(monochrome);
- config->monochrome = monochrome;
- AV1C_READ_BIT_OR_RETURN_ERROR(chroma_subsampling_x);
- config->chroma_subsampling_x = chroma_subsampling_x;
- AV1C_READ_BIT_OR_RETURN_ERROR(chroma_subsampling_y);
- config->chroma_subsampling_y = chroma_subsampling_y;
- AV1C_READ_BITS_OR_RETURN_ERROR(chroma_sample_position, 2);
- config->chroma_sample_position = chroma_sample_position;
- AV1C_READ_BITS_OR_RETURN_ERROR(reserved, 3);
- AV1C_READ_BIT_OR_RETURN_ERROR(initial_presentation_delay_present);
- config->initial_presentation_delay_present =
- initial_presentation_delay_present;
- AV1C_READ_BITS_OR_RETURN_ERROR(initial_presentation_delay_minus_one, 4);
- config->initial_presentation_delay_minus_one =
- initial_presentation_delay_minus_one;
- *bytes_read = aom_rb_bytes_read(reader);
- return 0;
- }
- int write_av1config(const Av1Config *config, size_t capacity,
- size_t *bytes_written, uint8_t *buffer) {
- if (!config || !buffer || capacity < kAv1cSize || !bytes_written) return -1;
- *bytes_written = 0;
- memset(buffer, 0, kAv1cSize);
- struct aom_write_bit_buffer writer = { buffer, 0 };
- aom_wb_write_bit(&writer, config->marker);
- aom_wb_write_literal(&writer, config->version, 7);
- aom_wb_write_literal(&writer, config->seq_profile, 3);
- aom_wb_write_literal(&writer, config->seq_level_idx_0, 5);
- aom_wb_write_bit(&writer, config->seq_tier_0);
- aom_wb_write_bit(&writer, config->high_bitdepth);
- aom_wb_write_bit(&writer, config->twelve_bit);
- aom_wb_write_bit(&writer, config->monochrome);
- aom_wb_write_bit(&writer, config->chroma_subsampling_x);
- aom_wb_write_bit(&writer, config->chroma_subsampling_y);
- aom_wb_write_literal(&writer, config->chroma_sample_position, 2);
- aom_wb_write_literal(&writer, 0, 3); // reserved
- aom_wb_write_bit(&writer, config->initial_presentation_delay_present);
- if (config->initial_presentation_delay_present) {
- aom_wb_write_literal(&writer, config->initial_presentation_delay_minus_one,
- 4);
- } else {
- aom_wb_write_literal(&writer, 0, 4); // reserved
- }
- *bytes_written = aom_wb_bytes_written(&writer);
- return 0;
- }
- #undef AV1C_READ_BIT_OR_RETURN_ERROR
- #undef AV1C_READ_BITS_OR_RETURN_ERROR
- #undef AV1C_PUSH_ERROR_HANDLER_DATA
- #undef AV1C_POP_ERROR_HANDLER_DATA
|